csgo/cstrike15_src/materialsystem/shaderapidx9/dx9hook.h

/* 
TODO: add option to null out drawprim calls.
 
Also maybe hook the PIX_ENABLE stuff to Telemetry
 
		In imaterialsystem.h:
		#define PIX_ENABLE 1                   // set this to 1 and build engine/studiorender to enable pix events in the engine

		And in shaderapidx8.h:
    	#define PIX_ENABLE 1                   // set this to 1 and build engine/studiorender to enable pix events in the engine
 
Might be interesting to make it so dx9hook.h paid attention to the PIX_ENABLE names
	and allowed you to filter drawprim calls based on those?
*/

#ifndef _DX9HOOK_H_
#define _DX9HOOK_H_

#if D3D_BATCH_PERF_ANALYSIS
	#include "../../thirdparty/miniz/miniz.c"
	#include "../../thirdparty/miniz/simple_bitmap.h"

	extern ConVar d3d_batch_vis, d3d_batch_vis_abs_scale, d3d_batch_vis_y_scale, d3d_present_vis_abs_scale;
	extern uint64 g_nTotalD3DCalls, g_nTotalD3DCycles;

	class CD3DCallTimer
	{
	public:
		inline CD3DCallTimer() { g_nTotalD3DCalls++; g_nTotalD3DCycles -= TM_FAST_TIME(); }
		inline ~CD3DCallTimer() {  g_nTotalD3DCycles += TM_FAST_TIME(); }
	};
	
	#define D3D_BATCH_PERF(...) __VA_ARGS__
#else
	#define D3D_BATCH_PERF(...)
#endif

#if D3D_BATCH_PERF_ANALYSIS
	#define XXX \
		tmZone( TELEMETRY_LEVEL3, TMZF_NONE, "D3D9: %s", __FUNCTION__ ); \
		CD3DCallTimer scopedCallTimer;
#else
	#define XXX																													\
		if( ThreadInMainThread() )  																							\
		{   																													\
			tmMessage( TELEMETRY_LEVEL0, TMMF_ICON_NOTE | TMMF_SEVERITY_WARNING, "(dota/d3d)%s", __FUNCTION__ );				\
			tmZoneFiltered( TELEMETRY_LEVEL0, 50, TMZF_NONE, "%s", __FUNCTION__ );												\
		}
#endif

// Hooks for routines which return values.
#define _DOCALL0( _member )				( m_Data.pHWObj->_member )()
#define _DOCALL( _member, ... )			( m_Data.pHWObj->_member )( __VA_ARGS__ )
// And hooks for routines which return squatola.
#define _DOCALL0_NORET( _member )		( m_Data.pHWObj->_member )()
#define _DOCALL_NORET( _member, ... )	( m_Data.pHWObj->_member )( __VA_ARGS__)

#define DEF_HOOKCLASSES( X )												\
    template<> class CDx9HookBase< struct I ## X >							\
    {                                                   					\
    public:                                             					\
        typedef I ## X 					_D3DINTERFACE;                		\
        typedef class C ## X ## Hook 	_HOOKCLASS;                			\
																			\
	public:  																\
		CDx9HookBase() { memset( &m_Data, 0, sizeof( m_Data ) ); }  		\
		virtual ~CDx9HookBase() {}  										\
																			\
	public: 																\
		struct DATA 														\
		{   																\
			class CDirect3DDevice9Hook *pDevice;   							\
			_D3DINTERFACE	*pHWObj;    									\
		};  																\
																			\
		DATA m_Data;														\
    }

template < class T > class CDx9HookBase;
DEF_HOOKCLASSES( Direct3DVertexDeclaration9 );
DEF_HOOKCLASSES( Direct3DPixelShader9 );
DEF_HOOKCLASSES( Direct3DVertexShader9 );
DEF_HOOKCLASSES( Direct3DVertexBuffer9 );
DEF_HOOKCLASSES( Direct3DIndexBuffer9 );
DEF_HOOKCLASSES( Direct3DQuery9 );
DEF_HOOKCLASSES( Direct3DStateBlock9 );
DEF_HOOKCLASSES( Direct3DSurface9 );
DEF_HOOKCLASSES( Direct3DBaseTexture9 );
DEF_HOOKCLASSES( Direct3DTexture9 );
DEF_HOOKCLASSES( Direct3DCubeTexture9 );
DEF_HOOKCLASSES( Direct3DVolume9 );
DEF_HOOKCLASSES( Direct3DVolumeTexture9 );
DEF_HOOKCLASSES( Direct3DSwapChain9 );
DEF_HOOKCLASSES( Direct3D9 );
DEF_HOOKCLASSES( Direct3DDevice9 );

template < class _D3DINTERFACE > HRESULT AllocOverride( HRESULT *hr, class CDirect3DDevice9Hook *pDevice, _D3DINTERFACE **ppHWObj )
{
    if( SUCCEEDED(*hr) && ppHWObj && *ppHWObj )
    {
        CDx9HookBase< _D3DINTERFACE >::_HOOKCLASS *pClass = new CDx9HookBase< _D3DINTERFACE >::_HOOKCLASS;

        if(!pClass)
        {
            ( *ppHWObj )->Release();
            *hr = E_OUTOFMEMORY;
            return NULL;
        }
        pClass->m_Data.pDevice = pDevice;
		pClass->m_Data.pHWObj = *ppHWObj;
		*ppHWObj = pClass;
        return *hr;
    }

    return *hr;
}

template <class _D3DINTERFACE> _D3DINTERFACE *GetHWPtr( _D3DINTERFACE *pD3DInterface )
{
	if( pD3DInterface )
	{
        CDx9HookBase< _D3DINTERFACE >::_HOOKCLASS *pClass = ( CDx9HookBase< _D3DINTERFACE >::_HOOKCLASS * )pD3DInterface;
		return pClass->m_Data.pHWObj;
	}
	return NULL;
}

template <class _D3DINTERFACE> CDirect3DDevice9Hook *GetHookDevice( _D3DINTERFACE *pD3DInterface )
{
	if( pD3DInterface )
	{
        CDx9HookBase< _D3DINTERFACE >::_HOOKCLASS *pClass = ( CDx9HookBase< _D3DINTERFACE >::_HOOKCLASS * )pD3DInterface;
		return pClass->m_Data.pDevice;
	}
	return NULL;
}

//$ TODO: if(riid == IID_IDirect3DDevice9Ex, IID_IDirect3DDevice9, etc.
#define IMPL_QUERYINTERFACE()                                                   \
    STDMETHOD(QueryInterface)(THIS_ REFIID riid, void** ppvObj)                 \
	{																			\
		__debugbreak();															\
		XXX; return _DOCALL(QueryInterface, riid, ppvObj);						\
	}

#define IMPL_IUNKOWN()                                                          \
    IMPL_QUERYINTERFACE();                                                      \
    STDMETHOD_(ULONG,AddRef)(THIS)                                              \
        { XXX; return _DOCALL0(AddRef); }										\
    STDMETHOD_(ULONG,Release)(THIS)                                             \
	{																			\
		XXX;																	\
		ULONG retval = _DOCALL0(Release);   									\
		if(retval == 0) 														\
			delete this;														\
		return retval;  														\
	}

#define IMPL_GETDEVICE()                                                        \
    STDMETHOD(GetDevice)(THIS_ IDirect3DDevice9** ppDevice)                     \
	{																			\
		XXX;																	\
		HRESULT hr = _DOCALL( GetDevice, ppDevice );							\
		return AllocOverride( &hr, m_Data.pDevice, ppDevice );					\
	}

#define IMPL_SETPRIVATEDATA()                                                                       \
    STDMETHOD(SetPrivateData)(THIS_ REFGUID refguid,CONST void* pData,DWORD SizeOfData,DWORD Flags) \
        { XXX; return _DOCALL(SetPrivateData, refguid, pData, SizeOfData, Flags); }

#define IMPL_GETPRIVATEDATA()                                                           \
    STDMETHOD(GetPrivateData)(THIS_ REFGUID refguid,void* pData,DWORD* pSizeOfData)     \
        { XXX; return _DOCALL(GetPrivateData, refguid, pData, pSizeOfData); }

#define IMPL_FREEPRIVATEDATA()                                                  \
    STDMETHOD(FreePrivateData)(THIS_ REFGUID refguid)                           \
        { XXX; return _DOCALL(FreePrivateData, refguid); }

#define IMPL_IDIRECT3DRESOURCE9()                                               \
    IMPL_GETDEVICE();                                                           \
    IMPL_SETPRIVATEDATA();                                                      \
    IMPL_GETPRIVATEDATA();                                                      \
    IMPL_FREEPRIVATEDATA();                                                     \
    STDMETHOD_(DWORD, SetPriority)(THIS_ DWORD PriorityNew)                     \
		{ XXX; return _DOCALL(SetPriority, PriorityNew); }                      \
    STDMETHOD_(DWORD, GetPriority)(THIS)                                        \
        { XXX; return _DOCALL0(GetPriority); }                                  \
    STDMETHOD_(void, PreLoad)(THIS)                                             \
        { XXX; _DOCALL0_NORET(PreLoad); }                                     	\
    STDMETHOD_(D3DRESOURCETYPE, GetType)(THIS)                                  \
        { XXX; return _DOCALL0(GetType); }

class CDirect3DSwapChain9Hook : public CDx9HookBase<IDirect3DSwapChain9>, public IDirect3DSwapChain9
{
public:
    /*** IUnknown methods ***/
    IMPL_IUNKOWN();

    /*** IDirect3DSwapChain9 methods ***/
    STDMETHOD(Present)(THIS_ CONST RECT* pSourceRect,CONST RECT* pDestRect,HWND hDestWindowOverride,CONST RGNDATA* pDirtyRegion,DWORD dwFlags)
		{ XXX; return _DOCALL(Present, pSourceRect, pDestRect, hDestWindowOverride, pDirtyRegion, dwFlags); }

    STDMETHOD(GetFrontBufferData)(THIS_ IDirect3DSurface9* pDestSurface)
	{
		XXX; return _DOCALL(GetFrontBufferData, GetHWPtr(pDestSurface));
	}
    STDMETHOD(GetBackBuffer)(THIS_ UINT iBackBuffer,D3DBACKBUFFER_TYPE Type,IDirect3DSurface9** ppBackBuffer)
	{
		XXX;
		HRESULT hr = _DOCALL(GetBackBuffer, iBackBuffer, Type, ppBackBuffer);
		return AllocOverride( &hr, m_Data.pDevice, ppBackBuffer );
	}

    STDMETHOD(GetRasterStatus)(THIS_ D3DRASTER_STATUS* pRasterStatus)
        { XXX; return _DOCALL(GetRasterStatus, pRasterStatus); }
    STDMETHOD(GetDisplayMode)(THIS_ D3DDISPLAYMODE* pMode)
        { XXX; return _DOCALL(GetDisplayMode, pMode); }

    IMPL_GETDEVICE();

    STDMETHOD(GetPresentParameters)(THIS_ D3DPRESENT_PARAMETERS* pPresentationParameters)
        { XXX; return _DOCALL(GetPresentParameters, pPresentationParameters); }
};

class CDirect3DVertexDeclaration9Hook : public CDx9HookBase<IDirect3DVertexDeclaration9>, public IDirect3DVertexDeclaration9
{
public:
    /*** IUnknown methods ***/
    IMPL_IUNKOWN();

    // IDirect3DVertexDeclaration9 methods
    IMPL_GETDEVICE();

    STDMETHOD(GetDeclaration)(THIS_ D3DVERTEXELEMENT9 *pVertElem, UINT* pNumElements)
        { XXX; return _DOCALL(GetDeclaration, pVertElem, pNumElements); }
};

class CDirect3DPixelShader9Hook : public CDx9HookBase<IDirect3DPixelShader9>, public IDirect3DPixelShader9
{
public:
    /*** IUnknown methods ***/
    IMPL_IUNKOWN();

    // IDirect3DPixelShader9 methods
    IMPL_GETDEVICE();

    STDMETHOD(GetFunction)(THIS_ void *pData,UINT* pSizeOfData)
        { XXX; return _DOCALL(GetFunction, pData, pSizeOfData); }
};

class CDirect3DVertexShader9Hook : public CDx9HookBase<IDirect3DVertexShader9>, public IDirect3DVertexShader9
{
public:
    /*** IUnknown methods ***/
    IMPL_IUNKOWN();

    // IDirect3DVertexDeclaration9 methods
    IMPL_GETDEVICE();

    STDMETHOD(GetFunction)(THIS_ void *pData,UINT* pSizeOfData)
        { XXX; return _DOCALL(GetFunction, pData, pSizeOfData); }
};

class CDirect3DVertexBuffer9Hook : public CDx9HookBase<IDirect3DVertexBuffer9>, public IDirect3DVertexBuffer9
{
public:
    /*** IUnknown methods ***/
    IMPL_IUNKOWN();

    // IDirect3DResource9 methods
    IMPL_IDIRECT3DRESOURCE9();

    // IDirect3DVertexBuffer9
    STDMETHOD(Lock)(THIS_ UINT OffsetToLock,UINT SizeToLock,void** ppbData,DWORD Flags)
		{ XXX; return _DOCALL(Lock, OffsetToLock, SizeToLock, ppbData, Flags); }
    STDMETHOD(Unlock)(THIS)
		{ XXX; return _DOCALL0(Unlock); }

    STDMETHOD(GetDesc)(THIS_ D3DVERTEXBUFFER_DESC *pDesc)
        { XXX; return _DOCALL(GetDesc, pDesc); }
};

class CDirect3DIndexBuffer9Hook : public CDx9HookBase<IDirect3DIndexBuffer9>, public IDirect3DIndexBuffer9
{
public:
    /*** IUnknown methods ***/
    IMPL_IUNKOWN();

    // IDirect3DResource9 methods
    IMPL_IDIRECT3DRESOURCE9();

    // IDirect3DIndexBuffer9
    STDMETHOD(Lock)(THIS_ UINT OffsetToLock,UINT SizeToLock,void** ppbData,DWORD Flags)
		{ XXX; return _DOCALL(Lock, OffsetToLock, SizeToLock, ppbData, Flags); }
    STDMETHOD(Unlock)(THIS)
		{ XXX; return _DOCALL0(Unlock); }

    STDMETHOD(GetDesc)(THIS_ D3DINDEXBUFFER_DESC *pDesc)
        { XXX; return _DOCALL(GetDesc, pDesc); }
};

class CDirect3DQuery9Hook : public CDx9HookBase<IDirect3DQuery9>, public IDirect3DQuery9
{
public:
    /*** IUnknown methods ***/
    IMPL_IUNKOWN();

    /*** IDirect3DQuery9 methods ***/
    IMPL_GETDEVICE();

    // IDirect3DQuery9
    STDMETHOD_(D3DQUERYTYPE, GetType)(THIS)
        { XXX; return _DOCALL0(GetType); }
    STDMETHOD_(DWORD, GetDataSize)(THIS)
        { XXX; return _DOCALL0(GetDataSize); }
    STDMETHOD(Issue)(THIS_ DWORD dwIssueFlags)
        { XXX; return _DOCALL(Issue, dwIssueFlags); }
    STDMETHOD(GetData)(THIS_ void* pData,DWORD dwSize,DWORD dwGetDataFlags)
        { XXX; return _DOCALL(GetData, pData, dwSize, dwGetDataFlags); }
};

class CDirect3DStateBlock9Hook : public CDx9HookBase<IDirect3DStateBlock9>, public IDirect3DStateBlock9
{
public:
    CDirect3DStateBlock9Hook() {}
    virtual ~CDirect3DStateBlock9Hook() {}

    /*** IUnknown methods ***/
    IMPL_IUNKOWN();

    // IDirect3DStateBlock9 methods
    IMPL_GETDEVICE();

    STDMETHOD(Capture)(THIS)
		{ XXX; return _DOCALL0(Capture); }
    STDMETHOD(Apply)(THIS)
		{ XXX; return _DOCALL0(Apply); }
};

class CDirect3DSurface9Hook : public CDx9HookBase<IDirect3DSurface9>, public IDirect3DSurface9
{
public:
    /*** IUnknown methods ***/
    IMPL_IUNKOWN();

    // IDirect3DResource9 methods
    IMPL_IDIRECT3DRESOURCE9();

    STDMETHOD(GetContainer)(THIS_ REFIID riid,void** ppContainer)
	{
		//$ TODO: do the call, check riid, and wrap the returned ppContainer
		__debugbreak();
		XXX; return _DOCALL(GetContainer, riid, ppContainer);
	}
    STDMETHOD(GetDesc)(THIS_ D3DSURFACE_DESC *pDesc)
        { XXX; return _DOCALL(GetDesc, pDesc); }

    STDMETHOD(LockRect)(THIS_ D3DLOCKED_RECT* pLockedRect,CONST RECT* pRect,DWORD Flags)
		{ XXX; return _DOCALL(LockRect, pLockedRect, pRect, Flags); }
    STDMETHOD(UnlockRect)(THIS)
		{ XXX; return _DOCALL0(UnlockRect); }

    STDMETHOD(GetDC)(THIS_ HDC *phdc)
        { XXX; return _DOCALL(GetDC, phdc); }
    STDMETHOD(ReleaseDC)(THIS_ HDC hdc)
        { XXX; return _DOCALL(ReleaseDC, hdc); }
};

class CDirect3DBaseTexture9Hook : public CDx9HookBase<IDirect3DBaseTexture9>, public IDirect3DBaseTexture9
{
public:
    /*** IUnknown methods ***/
    IMPL_IUNKOWN();

    /*** IDirect3DBaseTexture9 methods ***/
    IMPL_IDIRECT3DRESOURCE9();

	STDMETHOD_(DWORD, SetLOD)(THIS_ DWORD LODNew)
		{ XXX; return _DOCALL(SetLOD, LODNew); }
	STDMETHOD_(DWORD, GetLOD)(THIS)
		{ XXX; return _DOCALL0(GetLOD); }
	STDMETHOD_(DWORD, GetLevelCount)(THIS)
		{ XXX; return _DOCALL0(GetLevelCount); }
	STDMETHOD(SetAutoGenFilterType)(THIS_ D3DTEXTUREFILTERTYPE FilterType)
		{ XXX; return _DOCALL(SetAutoGenFilterType, FilterType); }
	STDMETHOD_(D3DTEXTUREFILTERTYPE, GetAutoGenFilterType)(THIS)
		{ XXX; return _DOCALL0(GetAutoGenFilterType); }
	STDMETHOD_(void, GenerateMipSubLevels)(THIS)
		{ XXX; return _DOCALL0(GenerateMipSubLevels); }
};

class CDirect3DTexture9Hook : public CDx9HookBase<IDirect3DTexture9>, public IDirect3DTexture9
{
public:
    /*** IUnknown methods ***/
    IMPL_IUNKOWN();

    /*** IDirect3DBaseTexture9 methods ***/
    IMPL_IDIRECT3DRESOURCE9();

    STDMETHOD_(DWORD, SetLOD)(THIS_ DWORD LODNew)
		{ XXX; return _DOCALL(SetLOD, LODNew); }
    STDMETHOD_(DWORD, GetLOD)(THIS)
        { XXX; return _DOCALL0(GetLOD); }
    STDMETHOD_(DWORD, GetLevelCount)(THIS)
        { XXX; return _DOCALL0(GetLevelCount); }
    STDMETHOD(SetAutoGenFilterType)(THIS_ D3DTEXTUREFILTERTYPE FilterType)
		{ XXX; return _DOCALL(SetAutoGenFilterType, FilterType); }
    STDMETHOD_(D3DTEXTUREFILTERTYPE, GetAutoGenFilterType)(THIS)
        { XXX; return _DOCALL0(GetAutoGenFilterType); }

    STDMETHOD_(void, GenerateMipSubLevels)(THIS)
		{ XXX; return _DOCALL0(GenerateMipSubLevels); }

    STDMETHOD(GetLevelDesc)(THIS_ UINT Level,D3DSURFACE_DESC *pDesc)
        { XXX; return _DOCALL(GetLevelDesc, Level, pDesc); }

    STDMETHOD(GetSurfaceLevel)(THIS_ UINT Level,IDirect3DSurface9** ppSurfaceLevel)
	{
		XXX;
		HRESULT hr = _DOCALL(GetSurfaceLevel, Level, ppSurfaceLevel);
		return AllocOverride( &hr, m_Data.pDevice, ppSurfaceLevel );
	}
    STDMETHOD(LockRect)(THIS_ UINT Level,D3DLOCKED_RECT* pLockedRect,CONST RECT* pRect,DWORD Flags)
		{ XXX; return _DOCALL(LockRect, Level, pLockedRect, pRect, Flags); }
    STDMETHOD(UnlockRect)(THIS_ UINT Level)
		{ XXX; return _DOCALL(UnlockRect, Level); }

    STDMETHOD(AddDirtyRect)(THIS_ CONST RECT* pDirtyRect)
        { XXX; return _DOCALL(AddDirtyRect, pDirtyRect); }
};

class CDirect3DCubeTexture9Hook : public CDx9HookBase<IDirect3DCubeTexture9>, public IDirect3DCubeTexture9
{
public:
    /*** IUnknown methods ***/
    IMPL_IUNKOWN();

    /*** IDirect3DBaseTexture9 methods ***/
    IMPL_IDIRECT3DRESOURCE9();

    STDMETHOD_(DWORD, SetLOD)(THIS_ DWORD LODNew)
		{ XXX; return _DOCALL(SetLOD, LODNew); }
    STDMETHOD_(DWORD, GetLOD)(THIS)
        { XXX; return _DOCALL0(GetLOD); }
    STDMETHOD_(DWORD, GetLevelCount)(THIS)
        { XXX; return _DOCALL0(GetLevelCount); }
    STDMETHOD(SetAutoGenFilterType)(THIS_ D3DTEXTUREFILTERTYPE FilterType)
		{ XXX; return _DOCALL(SetAutoGenFilterType, FilterType); }
    STDMETHOD_(D3DTEXTUREFILTERTYPE, GetAutoGenFilterType)(THIS)
        { XXX; return _DOCALL0(GetAutoGenFilterType); }

    STDMETHOD_(void, GenerateMipSubLevels)(THIS)
		{ XXX; return _DOCALL0(GenerateMipSubLevels); }

    STDMETHOD(GetLevelDesc)(THIS_ UINT Level,D3DSURFACE_DESC *pDesc)
        { XXX; return _DOCALL(GetLevelDesc, Level, pDesc); }

    STDMETHOD(GetCubeMapSurface)(THIS_ D3DCUBEMAP_FACES FaceType,UINT Level,IDirect3DSurface9** ppCubeMapSurface)
	{
		XXX;
		HRESULT hr = _DOCALL(GetCubeMapSurface, FaceType, Level, ppCubeMapSurface);
		return AllocOverride(&hr, m_Data.pDevice, ppCubeMapSurface);
	}
    STDMETHOD(LockRect)(THIS_ D3DCUBEMAP_FACES FaceType,UINT Level,D3DLOCKED_RECT* pLockedRect,CONST RECT* pRect,DWORD Flags)
		{ XXX; return _DOCALL(LockRect, FaceType, Level, pLockedRect, pRect, Flags); }
    STDMETHOD(UnlockRect)(THIS_ D3DCUBEMAP_FACES FaceType,UINT Level)
		{ XXX; return _DOCALL(UnlockRect, FaceType, Level); }

    STDMETHOD(AddDirtyRect)(THIS_ D3DCUBEMAP_FACES FaceType,CONST RECT* pDirtyRect)
        { XXX; return _DOCALL(AddDirtyRect, FaceType, pDirtyRect); }
};

class CDirect3DVolume9Hook : public CDx9HookBase<IDirect3DVolume9>, public IDirect3DVolume9
{
public:
    /*** IUnknown methods ***/
    IMPL_IUNKOWN();

    /*** IDirect3DVolume9 methods ***/
    IMPL_GETDEVICE();

    IMPL_SETPRIVATEDATA();
    IMPL_GETPRIVATEDATA();
    IMPL_FREEPRIVATEDATA()

    STDMETHOD(GetContainer)(THIS_ REFIID riid,void** ppContainer)
	{
		//$ TODO: do the call, check riid, and wrap the returned ppContainer
		__debugbreak();
		XXX; return _DOCALL(GetContainer, riid, ppContainer);
	}

    STDMETHOD(GetDesc)(THIS_ D3DVOLUME_DESC *pDesc)
        { XXX; return _DOCALL(GetDesc, pDesc); }

    STDMETHOD(LockBox)(THIS_ D3DLOCKED_BOX * pLockedVolume,CONST D3DBOX* pBox,DWORD Flags)
		{ XXX; return _DOCALL(LockBox, pLockedVolume, pBox, Flags); }
    STDMETHOD(UnlockBox)(THIS)
		{ XXX; return _DOCALL0(UnlockBox); }
};

class CDirect3DVolumeTexture9Hook : public CDx9HookBase<IDirect3DVolumeTexture9>, public IDirect3DVolumeTexture9
{
public:
    /*** IUnknown methods ***/
    IMPL_IUNKOWN();

    /*** IDirect3DBaseTexture9 methods ***/
    IMPL_IDIRECT3DRESOURCE9();

    STDMETHOD_(DWORD, SetLOD)(THIS_ DWORD LODNew)
		{ XXX; return _DOCALL(SetLOD, LODNew); }
    STDMETHOD_(DWORD, GetLOD)(THIS)
        { XXX; return _DOCALL0(GetLOD); }
    STDMETHOD_(DWORD, GetLevelCount)(THIS)
        { XXX; return _DOCALL0(GetLevelCount); }
    STDMETHOD(SetAutoGenFilterType)(THIS_ D3DTEXTUREFILTERTYPE FilterType)
		{ XXX; return _DOCALL(SetAutoGenFilterType, FilterType); }
    STDMETHOD_(D3DTEXTUREFILTERTYPE, GetAutoGenFilterType)(THIS)
        { XXX; return _DOCALL0(GetAutoGenFilterType); }

    STDMETHOD_(void, GenerateMipSubLevels)(THIS)
		{ XXX; return _DOCALL0(GenerateMipSubLevels); }

    STDMETHOD(GetLevelDesc)(THIS_ UINT Level,D3DVOLUME_DESC *pDesc)
        { XXX; return _DOCALL(GetLevelDesc, Level, pDesc); }

    STDMETHOD(GetVolumeLevel)(THIS_ UINT Level,IDirect3DVolume9** ppVolumeLevel)
	{
		XXX;
		HRESULT hr = _DOCALL(GetVolumeLevel, Level, ppVolumeLevel);
		return AllocOverride( &hr, m_Data.pDevice, ppVolumeLevel );
	}
    STDMETHOD(LockBox)(THIS_ UINT Level,D3DLOCKED_BOX* pLockedVolume,CONST D3DBOX* pBox,DWORD Flags)
		{ XXX; return _DOCALL(LockBox, Level, pLockedVolume, pBox, Flags); }
    STDMETHOD(UnlockBox)(THIS_ UINT Level)
		{ XXX; return _DOCALL(UnlockBox, Level); }

    STDMETHOD(AddDirtyBox)(THIS_ CONST D3DBOX* pDirtyBox)
        { XXX; return _DOCALL(AddDirtyBox, pDirtyBox); }
};

class CDirect3DDevice9Hook : public CDx9HookBase<IDirect3DDevice9>, public IDirect3DDevice9
{
public:
    /*** IUnknown methods ***/
    IMPL_IUNKOWN();

	CDirect3DDevice9Hook() : CDx9HookBase<IDirect3DDevice9>(), IDirect3DDevice9()
	{
		D3D_BATCH_PERF( g_nTotalD3DCycles = 0; g_nTotalD3DCalls = 0; m_batch_state.Clear(); m_nTotalDraws = 0; m_nTotalPrims = 0; m_nTotalD3DCalls = 0; m_flTotalD3DTime = 0; m_nOverallDraws = 0; m_nOverallPrims = 0; m_nOverallD3DCalls = 0; m_flOverallD3DTime = 0; m_nTotalFrames = 0; m_pPrevRenderTarget0 = NULL; )
	}

    // IDirect3DDevice9 methods
    STDMETHOD(TestCooperativeLevel)(THIS)
        { XXX; return _DOCALL0(TestCooperativeLevel); }
    STDMETHOD_(UINT, GetAvailableTextureMem)(THIS)
        {  XXX; return _DOCALL0(GetAvailableTextureMem); }
    STDMETHOD(EvictManagedResources)(THIS)
        { XXX; return _DOCALL0(EvictManagedResources); }
    STDMETHOD(GetDirect3D)(THIS_ IDirect3D9** ppD3D9)
	{
		XXX;
		HRESULT hr = _DOCALL(GetDirect3D, ppD3D9);
		return AllocOverride(&hr, this, ppD3D9);
	}
    STDMETHOD(GetDeviceCaps)(THIS_ D3DCAPS9* pCaps)
		{ XXX; return _DOCALL(GetDeviceCaps, pCaps); }
    STDMETHOD(GetDisplayMode)(THIS_ UINT iSwapChain,D3DDISPLAYMODE* pMode)
        { XXX; return _DOCALL(GetDisplayMode, iSwapChain, pMode); }
    STDMETHOD(GetCreationParameters)(THIS_ D3DDEVICE_CREATION_PARAMETERS *pParameters)
        { XXX; return _DOCALL(GetCreationParameters, pParameters); }
    STDMETHOD(SetCursorProperties)(THIS_ UINT XHotSpot,UINT YHotSpot,IDirect3DSurface9* pCursorBitmap)
		{ XXX; return _DOCALL(SetCursorProperties, XHotSpot, YHotSpot, GetHWPtr(pCursorBitmap)); }
    STDMETHOD_(void, SetCursorPosition)(THIS_ int X,int Y,DWORD Flags)
        { XXX; _DOCALL_NORET(SetCursorPosition, X, Y, Flags); }
    STDMETHOD_(BOOL, ShowCursor)(THIS_ BOOL bShow)
        { XXX; return _DOCALL(ShowCursor, bShow); }
    STDMETHOD(CreateAdditionalSwapChain)(THIS_ D3DPRESENT_PARAMETERS* pPresentationParameters,IDirect3DSwapChain9** ppSwapChain)
	{
		XXX;
		HRESULT hr = _DOCALL(CreateAdditionalSwapChain, pPresentationParameters, ppSwapChain);
		return AllocOverride(&hr, this, ppSwapChain);
	}
    STDMETHOD(GetSwapChain)(THIS_ UINT iSwapChain,IDirect3DSwapChain9** ppSwapChain)
	{
		XXX;
		HRESULT hr = _DOCALL(GetSwapChain, iSwapChain, ppSwapChain);
		return AllocOverride(&hr, this, ppSwapChain);
	}
    STDMETHOD_(UINT, GetNumberOfSwapChains)(THIS)
        { XXX; return _DOCALL0(GetNumberOfSwapChains); }
    STDMETHOD(Reset)(THIS_ D3DPRESENT_PARAMETERS* pPresentationParameters)
		{ XXX; return _DOCALL( Reset, pPresentationParameters ); }
    STDMETHOD(Present)(THIS_ CONST RECT* pSourceRect,CONST RECT* pDestRect,HWND hDestWindowOverride,CONST RGNDATA* pDirtyRegion)
	{ 
		HRESULT hres;

#if D3D_BATCH_PERF_ANALYSIS
		uint64 nStartCycles = g_nTotalD3DCycles;

		CFastTimer tm;
		tm.Start();

		g_nTotalD3DCalls++;
#endif

		{
			XXX; 
			hres = _DOCALL(Present, pSourceRect, pDestRect, hDestWindowOverride, pDirtyRegion); 
		}
				
#if D3D_BATCH_PERF_ANALYSIS
		const uint nFrameIndex = m_nTotalFrames;
		m_nTotalFrames++;

		if (nFrameIndex >= 5)
		{
			double flPresentTime = tm.GetDurationInProgress().GetMillisecondsF();
			uint64 nEndCycles = g_nTotalD3DCycles;

			double flTotalPresentTime = ( nEndCycles - nStartCycles ) * s_rdtsc_to_ms;

			m_nTotalD3DCalls += g_nTotalD3DCalls;
			m_flTotalD3DTime += g_nTotalD3DCycles * s_rdtsc_to_ms;
				
			static int bPrevBatchVis = -1;

			if ((bPrevBatchVis == 1) && m_batch_vis_bitmap.is_valid())
			{
				m_nOverallDraws += m_nTotalDraws;
				m_nOverallPrims += m_nTotalPrims;
				m_nOverallD3DCalls += m_nTotalD3DCalls;
				m_flOverallD3DTime += m_flTotalD3DTime;

				m_batch_vis_bitmap.fill_box(0, m_nBatchVisY, (uint)(.5f + flPresentTime / d3d_present_vis_abs_scale.GetFloat() * m_batch_vis_bitmap.width()), 10, 255, 16, 128);
				m_batch_vis_bitmap.additive_fill_box(0, m_nBatchVisY, (uint)(.5f + flTotalPresentTime / d3d_present_vis_abs_scale.GetFloat() * m_batch_vis_bitmap.width()), 10, 0, 255, 128);
				m_nBatchVisY += 10;
			
				uint y = MAX( 600, m_nBatchVisY + 20 ), l = 0;
				m_batch_vis_bitmap.draw_formatted_text(0, y+8*(l++), 1, 255, 255, 255, "Frame: %u, Batches: %u, Prims: %u", nFrameIndex, m_nTotalDraws, m_nTotalPrims );
				m_batch_vis_bitmap.draw_formatted_text(0, y+8*(l++), 1, 255, 255, 255, "Frame: D3D Calls: %u, D3D Time: %3.3fms", m_nTotalD3DCalls, m_flTotalD3DTime);
				l++;
				m_batch_vis_bitmap.draw_formatted_text(0, y+8*(l++), 1, 255, 255, 255, "Overall: Batches: %u, Prims: %u", m_nOverallDraws, m_nOverallPrims );
				m_batch_vis_bitmap.draw_formatted_text(0, y+8*(l++), 1, 255, 255, 255, "Overall: D3D Calls: %u D3D Time: %4.3fms", m_nOverallD3DCalls, m_flOverallD3DTime );

				size_t png_size = 0;
				void *pPNG_data = tdefl_write_image_to_png_file_in_memory(m_batch_vis_bitmap.get_ptr(), m_batch_vis_bitmap.width(), m_batch_vis_bitmap.height(), 3, &png_size, true);
				if (pPNG_data)
				{
					char filename[256];
					V_snprintf(filename, sizeof(filename), "left4dead2/batchvis_%u_%u.png", m_nBatchVisFileIdx, m_nBatchVisFrameIndex);
					FILE* pFile = fopen(filename, "wb");
					if (pFile)
					{
						fwrite(pPNG_data, png_size, 1, pFile);
						fclose(pFile);
					}
					free(pPNG_data);
				}
				m_nBatchVisFrameIndex++;
				m_nBatchVisY = 0;
				m_batch_vis_bitmap.cls();
			}

			if (bPrevBatchVis != (int)d3d_batch_vis.GetBool())
			{
				bPrevBatchVis = d3d_batch_vis.GetBool();
				if (!bPrevBatchVis)
				{
					m_batch_vis_bitmap.clear();
				}
				else
				{
					m_batch_vis_bitmap.init(768, 1024);
				}
				m_nBatchVisY = 0;
				m_nBatchVisFrameIndex = 0;
				m_nBatchVisFileIdx = (uint)time(NULL); //rand();

				m_nOverallDraws = 0;
				m_nOverallPrims = 0;
				m_nOverallD3DCalls = 0;
				m_flOverallD3DTime = 0;
			}
		}

		m_nTotalD3DCalls = 0;
		m_nTotalPrims = 0;
		m_flTotalD3DTime = 0;
		g_nTotalD3DCycles = 0;
		g_nTotalD3DCalls = 0;
		m_nTotalDraws = 0;
#else
		if ( d3d_batch_vis.GetBool() )
		{
			d3d_batch_vis.SetValue( false );

			ConMsg( "Must define D3D_BATCH_PERF_ANALYSIS to use this feature" );
		}
#endif

		return hres;
	}

    STDMETHOD(GetBackBuffer)(THIS_ UINT iSwapChain,UINT iBackBuffer,D3DBACKBUFFER_TYPE Type,IDirect3DSurface9** ppBackBuffer)
	{
		XXX;
		HRESULT hr = _DOCALL(GetBackBuffer, iSwapChain, iBackBuffer, Type, ppBackBuffer);
		return AllocOverride(&hr, this, ppBackBuffer);
	}
    STDMETHOD(GetRasterStatus)(THIS_ UINT iSwapChain,D3DRASTER_STATUS* pRasterStatus)
        { XXX; return _DOCALL(GetRasterStatus, iSwapChain, pRasterStatus); }
    STDMETHOD(SetDialogBoxMode)(THIS_ BOOL bEnableDialogs)
        { XXX; return _DOCALL(SetDialogBoxMode, bEnableDialogs); }
    STDMETHOD_(void, SetGammaRamp)(THIS_ UINT iSwapChain,DWORD Flags,CONST D3DGAMMARAMP* pRamp)
        { XXX; _DOCALL_NORET(SetGammaRamp, iSwapChain, Flags, pRamp); }
    STDMETHOD_(void, GetGammaRamp)(THIS_ UINT iSwapChain,D3DGAMMARAMP* pRamp)
        { XXX; _DOCALL_NORET(GetGammaRamp, iSwapChain, pRamp); }
    STDMETHOD(CreateTexture)(THIS_ UINT Width,UINT Height,UINT Levels,DWORD Usage,D3DFORMAT Format,D3DPOOL Pool,IDirect3DTexture9** ppTexture,HANDLE* pSharedHandle)
	{
		XXX;
		HRESULT hr = _DOCALL(CreateTexture, Width, Height, Levels, Usage, Format, Pool, ppTexture, pSharedHandle);
		return AllocOverride(&hr, this, ppTexture);
	}
    STDMETHOD(CreateVolumeTexture)(THIS_ UINT Width,UINT Height,UINT Depth,UINT Levels,DWORD Usage,D3DFORMAT Format,D3DPOOL Pool,IDirect3DVolumeTexture9** ppVolumeTexture,HANDLE* pSharedHandle)
	{
		XXX;
		HRESULT hr = _DOCALL(CreateVolumeTexture, Width, Height, Depth, Levels, Usage, Format, Pool, ppVolumeTexture, pSharedHandle);
		return AllocOverride(&hr, this, ppVolumeTexture);
	}
    STDMETHOD(CreateCubeTexture)(THIS_ UINT EdgeLength,UINT Levels,DWORD Usage,D3DFORMAT Format,D3DPOOL Pool,IDirect3DCubeTexture9** ppCubeTexture,HANDLE* pSharedHandle)
	{
		XXX;
		HRESULT hr = _DOCALL(CreateCubeTexture, EdgeLength, Levels, Usage, Format, Pool, ppCubeTexture, pSharedHandle);
		return AllocOverride(&hr, this, ppCubeTexture);
	}
    STDMETHOD(CreateVertexBuffer)(THIS_ UINT Length,DWORD Usage,DWORD FVF,D3DPOOL Pool,IDirect3DVertexBuffer9** ppVertexBuffer,HANDLE* pSharedHandle)
	{
		XXX;
		HRESULT hr = _DOCALL(CreateVertexBuffer, Length, Usage, FVF, Pool, ppVertexBuffer, pSharedHandle);
		return AllocOverride(&hr, this, ppVertexBuffer);
	}
    STDMETHOD(CreateIndexBuffer)(THIS_ UINT Length,DWORD Usage,D3DFORMAT Format,D3DPOOL Pool,IDirect3DIndexBuffer9** ppIndexBuffer,HANDLE* pSharedHandle)
	{
		XXX;
		HRESULT hr = _DOCALL(CreateIndexBuffer, Length, Usage, Format, Pool, ppIndexBuffer, pSharedHandle);
		return AllocOverride(&hr, this, ppIndexBuffer);
	}
    STDMETHOD(CreateRenderTarget)(THIS_ UINT Width,UINT Height,D3DFORMAT Format,D3DMULTISAMPLE_TYPE MultiSample,DWORD MultiSampleQuality,BOOL Lockable,IDirect3DSurface9** ppSurface,HANDLE* pSharedHandle)
	{
		XXX;
		HRESULT hr = _DOCALL(CreateRenderTarget, Width, Height, Format, MultiSample, MultiSampleQuality, Lockable, ppSurface, pSharedHandle);
		return AllocOverride(&hr, this, ppSurface);
	}
    STDMETHOD(CreateDepthStencilSurface)(THIS_ UINT Width,UINT Height,D3DFORMAT Format,D3DMULTISAMPLE_TYPE MultiSample,DWORD MultiSampleQuality,BOOL Discard,IDirect3DSurface9** ppSurface,HANDLE* pSharedHandle)
	{
		XXX;
        HRESULT hr = _DOCALL(CreateDepthStencilSurface, Width, Height, Format, MultiSample, MultiSampleQuality, Discard, ppSurface, pSharedHandle);
		return AllocOverride(&hr, this, ppSurface);
	}
    STDMETHOD(UpdateSurface)(THIS_ IDirect3DSurface9* pSourceSurface,CONST RECT* pSourceRect,IDirect3DSurface9* pDestinationSurface,CONST POINT* pDestPoint)
		{ XXX; return _DOCALL(UpdateSurface, GetHWPtr(pSourceSurface), pSourceRect, GetHWPtr(pDestinationSurface), pDestPoint); }
    STDMETHOD(UpdateTexture)(THIS_ IDirect3DBaseTexture9* pSourceTexture,IDirect3DBaseTexture9* pDestinationTexture)
		{ XXX; return _DOCALL(UpdateTexture, GetHWPtr(pSourceTexture), GetHWPtr(pDestinationTexture)); }
    STDMETHOD(GetRenderTargetData)(THIS_ IDirect3DSurface9* pRenderTarget,IDirect3DSurface9* pDestSurface)
		{ XXX; return _DOCALL(GetRenderTargetData, GetHWPtr(pRenderTarget), GetHWPtr(pDestSurface)); }
    STDMETHOD(GetFrontBufferData)(THIS_ UINT iSwapChain,IDirect3DSurface9* pDestSurface)
		{ XXX; return _DOCALL(GetFrontBufferData, iSwapChain, GetHWPtr(pDestSurface)); }
    STDMETHOD(StretchRect)(THIS_ IDirect3DSurface9* pSourceSurface,CONST RECT* pSourceRect,IDirect3DSurface9* pDestSurface,CONST RECT* pDestRect,D3DTEXTUREFILTERTYPE Filter)
		{ XXX; return _DOCALL(StretchRect, GetHWPtr(pSourceSurface), pSourceRect, GetHWPtr(pDestSurface), pDestRect, Filter); }
    STDMETHOD(ColorFill)(THIS_ IDirect3DSurface9* pSurface,CONST RECT* pRect,D3DCOLOR color)
		{ XXX; return _DOCALL(ColorFill, GetHWPtr(pSurface), pRect, color); }
    STDMETHOD(CreateOffscreenPlainSurface)(THIS_ UINT Width,UINT Height,D3DFORMAT Format,D3DPOOL Pool,IDirect3DSurface9** ppSurface,HANDLE* pSharedHandle)
	{
		XXX;
		HRESULT hr = _DOCALL(CreateOffscreenPlainSurface, Width, Height, Format, Pool, ppSurface, pSharedHandle);
		return AllocOverride(&hr, this, ppSurface);
	}
    
	STDMETHOD(SetRenderTarget)(THIS_ DWORD RenderTargetIndex,IDirect3DSurface9* pRenderTarget)
	{ 
		HRESULT hres;
		{
			XXX; 
			hres = _DOCALL(SetRenderTarget, RenderTargetIndex, GetHWPtr(pRenderTarget)); 
		}
#if D3D_BATCH_PERF_ANALYSIS
		if ( m_batch_vis_bitmap.is_valid() && !RenderTargetIndex )
		{
			if ( pRenderTarget != m_pPrevRenderTarget0 )
			{
				m_batch_vis_bitmap.fill_box(0, m_nBatchVisY, m_batch_vis_bitmap.width(), 1, 30, 20, 20);
				m_nBatchVisY += 1;
			}
		}
		if ( !RenderTargetIndex )
		{
			m_pPrevRenderTarget0 = pRenderTarget;
		}
#endif
		return hres;
	}

    STDMETHOD(GetRenderTarget)(THIS_ DWORD RenderTargetIndex,IDirect3DSurface9** ppRenderTarget)
	{
		XXX;
		HRESULT hr =  _DOCALL(GetRenderTarget, RenderTargetIndex, ppRenderTarget);
		return AllocOverride(&hr, this, ppRenderTarget);
	}
    STDMETHOD(SetDepthStencilSurface)(THIS_ IDirect3DSurface9* pNewZStencil)
		{ XXX; return _DOCALL(SetDepthStencilSurface, GetHWPtr(pNewZStencil)); }
    STDMETHOD(GetDepthStencilSurface)(THIS_ IDirect3DSurface9** ppZStencilSurface)
	{
		XXX;
		HRESULT hr = _DOCALL(GetDepthStencilSurface, ppZStencilSurface);
		return AllocOverride(&hr, this, ppZStencilSurface);
	}
    STDMETHOD(BeginScene)(THIS)
        { XXX; return _DOCALL0(BeginScene); }
    STDMETHOD(EndScene)(THIS)
        { XXX; return _DOCALL0(EndScene); }
    STDMETHOD(Clear)(THIS_ DWORD Count,CONST D3DRECT* pRects,DWORD Flags,D3DCOLOR Color,float Z,DWORD Stencil)
		{ XXX; return _DOCALL(Clear, Count, pRects, Flags, Color, Z, Stencil); }
    STDMETHOD(SetTransform)(THIS_ D3DTRANSFORMSTATETYPE State,CONST D3DMATRIX* pMatrix)
		{ XXX; return _DOCALL(SetTransform, State, pMatrix); }
    STDMETHOD(GetTransform)(THIS_ D3DTRANSFORMSTATETYPE State,D3DMATRIX* pMatrix)
        { XXX; return _DOCALL(GetTransform, State, pMatrix); }
    STDMETHOD(MultiplyTransform)(THIS_ D3DTRANSFORMSTATETYPE State,CONST D3DMATRIX *pMatrix)
        { XXX; return _DOCALL(MultiplyTransform, State, pMatrix); }
    STDMETHOD(SetViewport)(THIS_ CONST D3DVIEWPORT9* pViewport)
        { XXX; return _DOCALL(SetViewport, pViewport); }
    STDMETHOD(GetViewport)(THIS_ D3DVIEWPORT9* pViewport)
        { XXX; return _DOCALL(GetViewport, pViewport); }
    STDMETHOD(SetMaterial)(THIS_ CONST D3DMATERIAL9* pMaterial)
        { XXX; return _DOCALL(SetMaterial, pMaterial); }
    STDMETHOD(GetMaterial)(THIS_ D3DMATERIAL9* pMaterial)
        { XXX; return _DOCALL(GetMaterial, pMaterial); }
    STDMETHOD(SetLight)(THIS_ DWORD Index,CONST D3DLIGHT9* pLight)
		{ XXX; return _DOCALL(SetLight, Index, pLight); }
    STDMETHOD(GetLight)(THIS_ DWORD Index,D3DLIGHT9* pLight)
        { XXX; return _DOCALL(GetLight, Index, pLight); }
    STDMETHOD(LightEnable)(THIS_ DWORD Index,BOOL Enable)
		{ XXX; return _DOCALL(LightEnable, Index, Enable); }
    STDMETHOD(GetLightEnable)(THIS_ DWORD Index,BOOL* pEnable)
        { XXX; return _DOCALL(GetLightEnable, Index, pEnable); }
    STDMETHOD(SetClipPlane)(THIS_ DWORD Index,CONST float* pPlane)
        { XXX; return _DOCALL(SetClipPlane, Index, pPlane);}
    STDMETHOD(GetClipPlane)(THIS_ DWORD Index,float* pPlane)
        { XXX; return _DOCALL(GetClipPlane, Index, pPlane); }
    STDMETHOD(SetRenderState)(THIS_ D3DRENDERSTATETYPE State,DWORD Value)
        { XXX; return _DOCALL(SetRenderState, State, Value); }
    STDMETHOD(GetRenderState)(THIS_ D3DRENDERSTATETYPE State,DWORD* pValue)
        { XXX; return _DOCALL(GetRenderState, State, pValue); }
    STDMETHOD(CreateStateBlock)(THIS_ D3DSTATEBLOCKTYPE Type,IDirect3DStateBlock9** ppSB)
	{
		XXX;
		HRESULT hr = _DOCALL(CreateStateBlock, Type, ppSB);
		return AllocOverride(&hr, this, ppSB);
	}
    STDMETHOD(BeginStateBlock)(THIS)
		{ XXX; return _DOCALL0(BeginStateBlock); }
    STDMETHOD(EndStateBlock)(THIS_ IDirect3DStateBlock9** ppSB)
	{
		XXX;
		HRESULT hr = _DOCALL(EndStateBlock, ppSB);
		return AllocOverride(&hr, this, ppSB);
	}
    STDMETHOD(SetClipStatus)(THIS_ CONST D3DCLIPSTATUS9* pClipStatus)
        { XXX; return _DOCALL(SetClipStatus, pClipStatus);}
    STDMETHOD(GetClipStatus)(THIS_ D3DCLIPSTATUS9* pClipStatus)
        { XXX; return _DOCALL(GetClipStatus, pClipStatus);}
    STDMETHOD(GetTexture)(THIS_ DWORD Stage,IDirect3DBaseTexture9** ppTexture)
	{
		XXX;
		HRESULT hr = _DOCALL(GetTexture, Stage, ppTexture);
		return AllocOverride(&hr, this, ppTexture);
	}
    
	STDMETHOD(SetTexture)(THIS_ DWORD Stage,IDirect3DBaseTexture9* pTexture)
	{ 
		HRESULT hres;
		{
			XXX; 
			hres = _DOCALL(SetTexture, Stage, GetHWPtr(pTexture)); 
		}
		D3D_BATCH_PERF( m_batch_state.m_nNumSamplersChanged++; )
		return hres;
	}

    STDMETHOD(GetTextureStageState)(THIS_ DWORD Stage,D3DTEXTURESTAGESTATETYPE Type,DWORD* pValue)
        { XXX; return _DOCALL(GetTextureStageState, Stage, Type, pValue); }
    STDMETHOD(SetTextureStageState)(THIS_ DWORD Stage,D3DTEXTURESTAGESTATETYPE Type,DWORD Value)
        { XXX; return _DOCALL(SetTextureStageState, Stage, Type, Value); }
    STDMETHOD(GetSamplerState)(THIS_ DWORD Sampler,D3DSAMPLERSTATETYPE Type,DWORD* pValue)
        { XXX; return _DOCALL(GetSamplerState, Sampler, Type, pValue); }
    STDMETHOD(SetSamplerState)(THIS_ DWORD Sampler,D3DSAMPLERSTATETYPE Type,DWORD Value)
	{ 
		HRESULT hres;
		{
			XXX; 
			hres = _DOCALL(SetSamplerState, Sampler, Type, Value); 
		}
		D3D_BATCH_PERF( m_batch_state.m_nNumSamplerStatesChanged++; )
		return hres;
	}

    STDMETHOD(ValidateDevice)(THIS_ DWORD* pNumPasses)
        { XXX; return _DOCALL(ValidateDevice, pNumPasses); }
    STDMETHOD(SetPaletteEntries)(THIS_ UINT PaletteNumber,CONST PALETTEENTRY* pEntries)
        { XXX; return _DOCALL(SetPaletteEntries, PaletteNumber, pEntries); }
    STDMETHOD(GetPaletteEntries)(THIS_ UINT PaletteNumber,PALETTEENTRY* pEntries)
        { XXX; return _DOCALL(GetPaletteEntries, PaletteNumber, pEntries); }
    STDMETHOD(SetCurrentTexturePalette)(THIS_ UINT PaletteNumber)
        { XXX; return _DOCALL(SetCurrentTexturePalette, PaletteNumber);}
    STDMETHOD(GetCurrentTexturePalette)(THIS_ UINT *PaletteNumber)
        { XXX; return _DOCALL(GetCurrentTexturePalette, PaletteNumber); }
    STDMETHOD(SetScissorRect)(THIS_ CONST RECT* pRect)
        { XXX; return _DOCALL(SetScissorRect, pRect); }
    STDMETHOD(GetScissorRect)(THIS_ RECT* pRect)
        { XXX; return _DOCALL(GetScissorRect, pRect); }
    STDMETHOD(SetSoftwareVertexProcessing)(THIS_ BOOL bSoftware)
        { XXX; return _DOCALL(SetSoftwareVertexProcessing, bSoftware); }
    STDMETHOD_(BOOL, GetSoftwareVertexProcessing)(THIS)
        { XXX; return _DOCALL0(GetSoftwareVertexProcessing); }
    STDMETHOD(SetNPatchMode)(THIS_ float nSegments)
        { XXX; return _DOCALL(SetNPatchMode, nSegments); }
    STDMETHOD_(float, GetNPatchMode)(THIS)
        { XXX; return _DOCALL0(GetNPatchMode);}
    STDMETHOD(DrawPrimitive)(THIS_ D3DPRIMITIVETYPE PrimitiveType,UINT StartVertex,UINT PrimitiveCount)
		{ XXX; m_nTotalDraws++; m_nTotalPrims += PrimitiveCount; return _DOCALL(DrawPrimitive, PrimitiveType, StartVertex, PrimitiveCount); }
    STDMETHOD(DrawIndexedPrimitive)(THIS_ D3DPRIMITIVETYPE PrimitiveType,INT BaseVertexIndex,UINT MinVertexIndex,UINT NumVertices,UINT startIndex,UINT primCount)
	{ 
		m_nTotalDraws++;
		m_nTotalPrims += primCount; 

#if D3D_BATCH_PERF_ANALYSIS
		CFastTimer tm;
		if ( m_batch_vis_bitmap.is_valid() )
		{
			tm.Start();
		}
#endif

		HRESULT hres;
		{
			XXX; 
			hres = _DOCALL(DrawIndexedPrimitive, PrimitiveType, BaseVertexIndex, MinVertexIndex, NumVertices, startIndex, primCount); 
		}
		
#if D3D_BATCH_PERF_ANALYSIS
		if ( m_batch_vis_bitmap.is_valid() )
		{
			double t = tm.GetDurationInProgress().GetMillisecondsF();

			uint h = 1;
			if ( d3d_batch_vis_y_scale.GetFloat() > 0.0f)
			{
				h = ceil( t / d3d_batch_vis_y_scale.GetFloat() );
				h = MAX(h, 1);
			}

			m_batch_vis_bitmap.fill_box(0, m_nBatchVisY, (uint)(.5f + t / d3d_batch_vis_abs_scale.GetFloat() * m_batch_vis_bitmap.width()), h, 64, 64, 64);

			if ( s_rdtsc_to_ms == 0.0f )
			{
				TmU64 t0 = TM_FAST_TIME();
				double d0 = Plat_FloatTime();

				ThreadSleep( 1 );

				TmU64 t1 = TM_FAST_TIME();
				double d1 = Plat_FloatTime();

				s_rdtsc_to_ms = ( 1000.0f * ( d1 - d0 ) ) / ( t1 - t0 );
			}

			double flTotalD3DCallMS = g_nTotalD3DCycles * s_rdtsc_to_ms;

			m_batch_vis_bitmap.additive_fill_box(0, m_nBatchVisY, (uint)(.5f + flTotalD3DCallMS / d3d_batch_vis_abs_scale.GetFloat() * m_batch_vis_bitmap.width()), h, 96, 96, 128);

			if (m_batch_state.m_bVertexShaderChanged) m_batch_vis_bitmap.additive_fill_box(0, m_nBatchVisY, 8, h, 0, 0, 64);
			if (m_batch_state.m_bPixelShaderChanged) m_batch_vis_bitmap.additive_fill_box(32, m_nBatchVisY, 8, h, 64, 0, 64);

			int lm = 80;
			m_batch_vis_bitmap.fill_box(lm+0+0, m_nBatchVisY, m_batch_state.m_nNumVSConstants, h, 64, 255, 255);
			m_batch_vis_bitmap.fill_box(lm+64+256+0, m_nBatchVisY, m_batch_state.m_nNumPSConstants, h, 64, 64, 255);

			m_batch_vis_bitmap.fill_box(lm+64+256+32, m_nBatchVisY, m_batch_state.m_nNumSamplersChanged, h, 255, 255, 255);
			m_batch_vis_bitmap.fill_box(lm+64+256+32+16, m_nBatchVisY, m_batch_state.m_nNumSamplerStatesChanged, h, 92, 128, 255);

			if ( m_batch_state.m_bStreamSourceChanged) m_batch_vis_bitmap.fill_box(lm+64+256+32+16+64, m_nBatchVisY, 16, h, 128, 128, 128);
			if ( m_batch_state.m_bIndicesChanged ) m_batch_vis_bitmap.fill_box(lm+64+256+32+16+64+16, m_nBatchVisY, 16, h, 128, 128, 255);

			m_nBatchVisY += h;
			
			m_nTotalD3DCalls += g_nTotalD3DCalls;
			m_flTotalD3DTime += flTotalD3DCallMS;
			
			g_nTotalD3DCycles = 0;
			g_nTotalD3DCalls = 0;
			
			m_batch_state.Clear();
		}
#endif

		return hres;
	}

    STDMETHOD(DrawPrimitiveUP)(THIS_ D3DPRIMITIVETYPE PrimitiveType,UINT PrimitiveCount,CONST void* pVertexStreamZeroData,UINT VertexStreamZeroStride)
		{ XXX; m_nTotalDraws++; m_nTotalPrims += PrimitiveCount; return _DOCALL(DrawPrimitiveUP, PrimitiveType, PrimitiveCount, pVertexStreamZeroData, VertexStreamZeroStride); }
    STDMETHOD(DrawIndexedPrimitiveUP)(THIS_ D3DPRIMITIVETYPE PrimitiveType,UINT MinVertexIndex,UINT NumVertices,UINT PrimitiveCount,CONST void* pIndexData,D3DFORMAT IndexDataFormat,CONST void* pVertexStreamZeroData,UINT VertexStreamZeroStride)
		{ XXX; m_nTotalDraws++; m_nTotalPrims += PrimitiveCount; return _DOCALL(DrawIndexedPrimitiveUP, PrimitiveType, MinVertexIndex, NumVertices, PrimitiveCount, pIndexData, IndexDataFormat, pVertexStreamZeroData, VertexStreamZeroStride); }
    STDMETHOD(ProcessVertices)(THIS_ UINT SrcStartIndex,UINT DestIndex,UINT VertexCount,IDirect3DVertexBuffer9* pDestBuffer,IDirect3DVertexDeclaration9* pVertexDecl,DWORD Flags)
		{ XXX; return _DOCALL(ProcessVertices, SrcStartIndex, DestIndex, VertexCount, GetHWPtr(pDestBuffer), GetHWPtr(pVertexDecl), Flags); }
    STDMETHOD(CreateVertexDeclaration)(THIS_ CONST D3DVERTEXELEMENT9* pVertexElements,IDirect3DVertexDeclaration9** ppDecl)
	{
		XXX;
		HRESULT hr = _DOCALL(CreateVertexDeclaration, pVertexElements, ppDecl);
		return AllocOverride(&hr, this, ppDecl);
	}
    STDMETHOD(SetVertexDeclaration)(THIS_ IDirect3DVertexDeclaration9* pDecl)
		{ XXX; return _DOCALL(SetVertexDeclaration, GetHWPtr(pDecl)); }
    STDMETHOD(GetVertexDeclaration)(THIS_ IDirect3DVertexDeclaration9** ppDecl)
	{
		XXX;
		HRESULT hr = _DOCALL(GetVertexDeclaration, ppDecl);
		return AllocOverride(&hr, this, ppDecl);
	}
    STDMETHOD(SetFVF)(THIS_ DWORD FVF)
		{ XXX; return _DOCALL(SetFVF, FVF); }
    STDMETHOD(GetFVF)(THIS_ DWORD* pFVF)
        { XXX; return _DOCALL(GetFVF, pFVF); }
    STDMETHOD(CreateVertexShader)(THIS_ CONST DWORD* pFunction,IDirect3DVertexShader9** ppShader)
	{
		XXX;
		HRESULT hr = _DOCALL(CreateVertexShader, pFunction, ppShader);
		return AllocOverride(&hr, this, ppShader);
	}
    
	STDMETHOD(SetVertexShader)(THIS_ IDirect3DVertexShader9* pShader)
	{ 
		HRESULT hres;
		{
			XXX; 
			hres = _DOCALL(SetVertexShader, GetHWPtr(pShader)); 
		}
		D3D_BATCH_PERF( m_batch_state.m_bVertexShaderChanged = true; )
		return hres;
	}

    STDMETHOD(GetVertexShader)(THIS_ IDirect3DVertexShader9** ppShader)
	{
		XXX;
		HRESULT hr = _DOCALL(GetVertexShader, ppShader);
		return AllocOverride(&hr, this, ppShader);
	}
    
	STDMETHOD(SetVertexShaderConstantF)(THIS_ UINT StartRegister,CONST float* pConstantData,UINT Vector4fCount)
	{ 
		HRESULT hres;
		{
			XXX; 
			hres = _DOCALL(SetVertexShaderConstantF, StartRegister, pConstantData, Vector4fCount); 
		}
		D3D_BATCH_PERF( m_batch_state.m_nNumVSConstants += Vector4fCount; )
		return hres;
	}

    STDMETHOD(GetVertexShaderConstantF)(THIS_ UINT StartRegister,float* pConstantData,UINT Vector4fCount)
		{ XXX; return _DOCALL(GetVertexShaderConstantF, StartRegister, pConstantData, Vector4fCount);}
    STDMETHOD(SetVertexShaderConstantI)(THIS_ UINT StartRegister,CONST int* pConstantData,UINT Vector4iCount)
		{ XXX; return _DOCALL(SetVertexShaderConstantI, StartRegister, pConstantData, Vector4iCount); }
    STDMETHOD(GetVertexShaderConstantI)(THIS_ UINT StartRegister,int* pConstantData,UINT Vector4iCount)
		{ XXX; return _DOCALL(GetVertexShaderConstantI, StartRegister, pConstantData, Vector4iCount); }
    STDMETHOD(SetVertexShaderConstantB)(THIS_ UINT StartRegister,CONST BOOL* pConstantData,UINT  BoolCount)
		{ XXX; return _DOCALL(SetVertexShaderConstantB, StartRegister, pConstantData, BoolCount); }
    STDMETHOD(GetVertexShaderConstantB)(THIS_ UINT StartRegister,BOOL* pConstantData,UINT BoolCount)
		{ XXX; return _DOCALL(GetVertexShaderConstantB, StartRegister, pConstantData, BoolCount); }
		    
	STDMETHOD(SetStreamSource)(THIS_ UINT StreamNumber,IDirect3DVertexBuffer9* pStreamData,UINT OffsetInBytes,UINT Stride)
		
	{ 
		HRESULT hres;
		{
			XXX; 
			hres = _DOCALL(SetStreamSource, StreamNumber, GetHWPtr(pStreamData), OffsetInBytes, Stride);
		}

		D3D_BATCH_PERF( m_batch_state.m_bStreamSourceChanged = true; )

		return hres;
	}

    STDMETHOD(GetStreamSource)(THIS_ UINT StreamNumber,IDirect3DVertexBuffer9** ppStreamData,UINT* OffsetInBytes,UINT* pStride)
	{
		XXX;
		HRESULT hr = _DOCALL(GetStreamSource, StreamNumber, ppStreamData, OffsetInBytes, pStride);
		return AllocOverride(&hr, this, ppStreamData);
	}
    STDMETHOD(SetStreamSourceFreq)(THIS_ UINT StreamNumber,UINT Divider)
        { XXX; return _DOCALL(SetStreamSourceFreq, StreamNumber, Divider); }

    STDMETHOD(GetStreamSourceFreq)(THIS_ UINT StreamNumber,UINT* Divider)
        { XXX; return _DOCALL(GetStreamSourceFreq, StreamNumber, Divider); }
    
	STDMETHOD(SetIndices)(THIS_ IDirect3DIndexBuffer9* pIndexData)
	{ 
		HRESULT hres;

		{
			XXX; 
			hres = _DOCALL(SetIndices, GetHWPtr(pIndexData)); 
		}

		D3D_BATCH_PERF( m_batch_state.m_bIndicesChanged = true; )

		return hres;
	}

    STDMETHOD(GetIndices)(THIS_ IDirect3DIndexBuffer9** ppIndexData)
	{
		XXX;
		HRESULT hr = _DOCALL(GetIndices, ppIndexData);
		return AllocOverride(&hr, this, ppIndexData);
	}
    STDMETHOD(CreatePixelShader)(THIS_ CONST DWORD* pFunction,IDirect3DPixelShader9** ppShader)
	{
		XXX;
		HRESULT hr = _DOCALL(CreatePixelShader, pFunction, ppShader);
		return AllocOverride(&hr, this, ppShader);
	}
    
	STDMETHOD(SetPixelShader)(THIS_ IDirect3DPixelShader9* pShader)
	{ 
		HRESULT hres;
		{
			XXX; 
			hres = _DOCALL(SetPixelShader, GetHWPtr(pShader)); 
		}
		D3D_BATCH_PERF( m_batch_state.m_bPixelShaderChanged = true; )
		return hres;
	}

    STDMETHOD(GetPixelShader)(THIS_ IDirect3DPixelShader9** ppShader)
	{
		XXX;
		HRESULT hr = _DOCALL(GetPixelShader, ppShader);
		return AllocOverride(&hr, this, ppShader);
	}
    
	STDMETHOD(SetPixelShaderConstantF)(THIS_ UINT StartRegister,CONST float* pConstantData,UINT Vector4fCount)
	{ 
		HRESULT hres;
		{
			XXX; 
			hres = _DOCALL(SetPixelShaderConstantF, StartRegister, pConstantData, Vector4fCount); 
		}
		D3D_BATCH_PERF( m_batch_state.m_nNumPSConstants += Vector4fCount; )
		return hres;
	}

    STDMETHOD(GetPixelShaderConstantF)(THIS_ UINT StartRegister,float* pConstantData,UINT Vector4fCount)
		{ XXX; return _DOCALL(GetPixelShaderConstantF, StartRegister, pConstantData, Vector4fCount); }
    STDMETHOD(SetPixelShaderConstantI)(THIS_ UINT StartRegister,CONST int* pConstantData,UINT Vector4iCount)
		{ XXX; return _DOCALL(SetPixelShaderConstantI, StartRegister, pConstantData, Vector4iCount); }
    STDMETHOD(GetPixelShaderConstantI)(THIS_ UINT StartRegister,int* pConstantData,UINT Vector4iCount)
		{ XXX; return _DOCALL(GetPixelShaderConstantI, StartRegister, pConstantData, Vector4iCount); }
    STDMETHOD(SetPixelShaderConstantB)(THIS_ UINT StartRegister,CONST BOOL* pConstantData,UINT  BoolCount)
		{ XXX; return _DOCALL(SetPixelShaderConstantB, StartRegister, pConstantData, BoolCount); }
    STDMETHOD(GetPixelShaderConstantB)(THIS_ UINT StartRegister,BOOL* pConstantData,UINT BoolCount)
		{ XXX; return _DOCALL(GetPixelShaderConstantB, StartRegister, pConstantData, BoolCount); }
    STDMETHOD(DrawRectPatch)(THIS_ UINT Handle,CONST float* pNumSegs,CONST D3DRECTPATCH_INFO* pRectPatchInfo)
        { XXX; return _DOCALL(DrawRectPatch, Handle, pNumSegs, pRectPatchInfo);}
    STDMETHOD(DrawTriPatch)(THIS_ UINT Handle,CONST float* pNumSegs,CONST D3DTRIPATCH_INFO* pTriPatchInfo)
        { XXX; return _DOCALL(DrawTriPatch, Handle, pNumSegs, pTriPatchInfo); }
    STDMETHOD(DeletePatch)(THIS_ UINT Handle)
        { XXX; return _DOCALL(DeletePatch, Handle);}
    STDMETHOD(CreateQuery)(THIS_ D3DQUERYTYPE Type,IDirect3DQuery9** ppQuery)
	{
		XXX;
		HRESULT hr = _DOCALL(CreateQuery, Type, ppQuery);
		return AllocOverride(&hr, this, ppQuery);
	}

private:

#if D3D_BATCH_PERF_ANALYSIS
	IDirect3DSurface9* m_pPrevRenderTarget0;
	simple_bitmap m_batch_vis_bitmap;
	uint m_nBatchVisY;
	uint m_nBatchVisFrameIndex, m_nBatchVisFileIdx;
	
	struct BatchState_t
	{
		void Clear() { memset(this, 0, sizeof(*this)); }

		bool m_bStreamSourceChanged;
		bool m_bIndicesChanged;
		bool m_bPixelShaderChanged;
		bool m_bVertexShaderChanged;
		uint m_nNumPSConstants;
		uint m_nNumVSConstants;
		uint m_nNumSamplersChanged;
		uint m_nNumSamplerStatesChanged;
	};

	BatchState_t m_batch_state;
	
	uint m_nTotalFrames;

	uint m_nTotalDraws;
	uint m_nTotalPrims;
	uint m_nTotalD3DCalls;
	double m_flTotalD3DTime;

	uint m_nOverallDraws;
	uint m_nOverallPrims;
	uint m_nOverallD3DCalls;
	double m_flOverallD3DTime;
#endif
};

class CDirect3D9Hook : public CDx9HookBase<IDirect3D9>, public IDirect3D9
{
public:
    /*** IUnknown methods ***/
    IMPL_IUNKOWN();

    // IDirect3D9 methods
    STDMETHOD(RegisterSoftwareDevice)(THIS_ void* pInitializeFunction)
		{ XXX; return _DOCALL(RegisterSoftwareDevice, pInitializeFunction); }
    STDMETHOD_(UINT, GetAdapterCount)(THIS)
		{ XXX; return _DOCALL0(GetAdapterCount); }
    STDMETHOD(GetAdapterIdentifier)(THIS_ UINT Adapter,DWORD Flags,D3DADAPTER_IDENTIFIER9* pIdentifier)
		{ XXX; return _DOCALL(GetAdapterIdentifier, Adapter, Flags, pIdentifier);}
    STDMETHOD_(UINT, GetAdapterModeCount)(THIS_ UINT Adapter,D3DFORMAT Format)
		{ XXX; return _DOCALL(GetAdapterModeCount, Adapter, Format);}
    STDMETHOD(EnumAdapterModes)(THIS_ UINT Adapter,D3DFORMAT Format,UINT Mode,D3DDISPLAYMODE* pMode)
		{ XXX; return _DOCALL(EnumAdapterModes, Adapter, Format, Mode, pMode); }
    STDMETHOD(GetAdapterDisplayMode)(THIS_ UINT Adapter,D3DDISPLAYMODE* pMode)
		{ XXX; return _DOCALL(GetAdapterDisplayMode, Adapter, pMode); }
    STDMETHOD(CheckDeviceType)(THIS_ UINT iAdapter,D3DDEVTYPE DevType,D3DFORMAT DisplayFormat,D3DFORMAT BackBufferFormat,BOOL bWindowed)
		{ XXX; return _DOCALL(CheckDeviceType, iAdapter, DevType, DisplayFormat, BackBufferFormat, bWindowed); }
    STDMETHOD(CheckDeviceFormat)(THIS_ UINT Adapter,D3DDEVTYPE DeviceType,D3DFORMAT AdapterFormat,DWORD Usage,D3DRESOURCETYPE RType,D3DFORMAT CheckFormat)
		{ XXX; return _DOCALL(CheckDeviceFormat, Adapter, DeviceType, AdapterFormat, Usage, RType, CheckFormat); }
    STDMETHOD(CheckDeviceMultiSampleType)(THIS_ UINT Adapter,D3DDEVTYPE DeviceType,D3DFORMAT SurfaceFormat,BOOL Windowed,D3DMULTISAMPLE_TYPE MultiSampleType,DWORD* pQualityLevels)
		{ XXX; return _DOCALL(CheckDeviceMultiSampleType, Adapter, DeviceType, SurfaceFormat, Windowed, MultiSampleType, pQualityLevels); }
    STDMETHOD(CheckDepthStencilMatch)(THIS_ UINT Adapter,D3DDEVTYPE DeviceType,D3DFORMAT AdapterFormat,D3DFORMAT RenderTargetFormat,D3DFORMAT DepthStencilFormat)
		{ XXX; return _DOCALL(CheckDepthStencilMatch, Adapter, DeviceType, AdapterFormat, RenderTargetFormat, DepthStencilFormat); }
    STDMETHOD(CheckDeviceFormatConversion)(THIS_ UINT Adapter,D3DDEVTYPE DeviceType,D3DFORMAT SourceFormat,D3DFORMAT TargetFormat)
		{ XXX; return _DOCALL(CheckDeviceFormatConversion, Adapter, DeviceType, SourceFormat, TargetFormat); }
    STDMETHOD(GetDeviceCaps)(THIS_ UINT Adapter,D3DDEVTYPE DeviceType,D3DCAPS9* pCaps)
		{ XXX; return _DOCALL(GetDeviceCaps, Adapter, DeviceType, pCaps); }
    STDMETHOD_(HMONITOR, GetAdapterMonitor)(THIS_ UINT Adapter)
		{ XXX; return _DOCALL(GetAdapterMonitor, Adapter); }
    STDMETHOD(CreateDevice)(THIS_ UINT Adapter,D3DDEVTYPE DeviceType,HWND hFocusWindow,DWORD BehaviorFlags,D3DPRESENT_PARAMETERS* pPresentationParameters,IDirect3DDevice9** ppReturnedDeviceInterface)
	{
		XXX;
		HRESULT hr = _DOCALL(CreateDevice, Adapter, DeviceType, hFocusWindow, BehaviorFlags, pPresentationParameters, ppReturnedDeviceInterface);

		if( SUCCEEDED( hr ) && ppReturnedDeviceInterface && *ppReturnedDeviceInterface )
		{
			CDirect3DDevice9Hook *pDevice = new CDirect3DDevice9Hook;
			if(!pDevice)
			{
				( *ppReturnedDeviceInterface )->Release();
				hr = E_OUTOFMEMORY;
				return NULL;
			}
			pDevice->m_Data.pDevice = pDevice;
			pDevice->m_Data.pHWObj = *ppReturnedDeviceInterface;
			*ppReturnedDeviceInterface = pDevice;
		}
		return hr;
	}
};

inline IDirect3D9 *Direct3DCreate9Hook( UINT SDKVersion )
{
	HRESULT hr = S_OK;
	IDirect3D9 *pD3D = Direct3DCreate9( D3D_SDK_VERSION );
	AllocOverride( &hr, NULL, &pD3D );
	return pD3D;
}

#endif // _DX9HOOK_H_