/*
* All or portions of this file Copyright (c) Amazon.com, Inc. or its affiliates or
* its licensors.
*
* For complete copyright and license terms please see the LICENSE at the root of this
* distribution (the "License"). All use of this software is governed by the License,
* or, if provided, by the license below or the license accompanying this file. Do not
* remove or modify any license notices. This file is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
*
*/
// Original file Copyright Crytek GMBH or its affiliates, used under license.

#ifndef CRYINCLUDE_CRYENGINE_RENDERDLL_XRENDERD3D9_GPUTIMER_H
#define CRYINCLUDE_CRYENGINE_RENDERDLL_XRENDERD3D9_GPUTIMER_H
#pragma once

class IGPUTimer
{
public:
    virtual ~IGPUTimer() = default;
    virtual void Start(const char* name) = 0;
    virtual void Stop() = 0;
    virtual void UpdateTime() = 0;
    virtual bool Init() = 0;
    virtual float GetTime() const = 0;
};

class CNullGPUTimer : public IGPUTimer
{
public:
    AZ_CLASS_ALLOCATOR(CNullGPUTimer, AZ::OSAllocator, 0)
    void Start(const char* name) {}
    void Stop() {}
    void UpdateTime() {}
    bool Init() { return false; }
    float GetTime() const { return 0; }
};

class CD3DGPUTimer : public IGPUTimer
{
public:
    AZ_CLASS_ALLOCATOR(CD3DGPUTimer, AZ::OSAllocator, 0)
    CD3DGPUTimer();
    virtual ~CD3DGPUTimer();

    virtual void Start(const char* name) override;
    virtual void Stop() override;
    virtual void UpdateTime() override;
    float GetTime() const override { return m_time; }


    float GetSmoothedTime() const { return m_smoothedTime; }
    bool IsStarted() const { return m_bStarted; }
    bool HasPendingQueries() const { return m_bWaiting; }

    void Release();
    virtual bool Init();

private:
    float  m_time;
    float  m_smoothedTime;

    bool   m_bInitialized;
    bool   m_bStarted;
    bool   m_bWaiting;

    AZStd::string m_Name;

#if GPUTIMER_H_TRAIT_DEFINEQUERIES
    ID3D11Query* m_pQueryStart, *m_pQueryStop, *m_pQueryFreq;
#elif defined(AZ_RESTRICTED_PLATFORM)
    #if defined(AZ_PLATFORM_XENIA)
        #include "Xenia/GPUTimer_h_xenia.inl"
    #elif defined(AZ_PLATFORM_PROVO)
        #include "Provo/GPUTimer_h_provo.inl"
    #elif defined(AZ_PLATFORM_SALEM)
        #include "Salem/GPUTimer_h_salem.inl"
    #endif
#endif

    void RecordSlice(AZ::u64 timeStart, AZ::u64 timeStop, AZ::u64 frequency);
};

class CD3DProfilingGPUTimer : public CD3DGPUTimer
{
public:
    static void EnableTiming();
    static void DisableTiming();

    static void AllowTiming();
    static void DisallowTiming();

    static bool IsTimingEnabled() { return s_bTimingEnabled; }
    static bool IsTimingAllowed() { return s_bTimingAllowed; }

    void Start(const char* name) override;
    void Stop() override;
    void UpdateTime() override;
    bool Init() override;

private:
    static bool s_bTimingEnabled;
    static bool s_bTimingAllowed;
};
#endif