#define CINTERFACE
#include <d3d9.h>

#include "Logger.h"
#include "MinHook.h"

namespace cinterface
{
	HRESULT(STDMETHODCALLTYPE* TrueLock)(IDirect3DVertexBuffer9* This, UINT OffsetToLock, UINT SizeToLock, void** ppbData, DWORD Flags) = nullptr;

	HRESULT STDMETHODCALLTYPE HookLock(IDirect3DVertexBuffer9* This, UINT OffsetToLock, UINT SizeToLock, void** ppbData, DWORD Flags)
	{
		//PrintLog(__FUNCTION__" %u", SizeToLock);
		if (SizeToLock == 358400)
			Flags = D3DLOCK_DISCARD;
		return TrueLock(This, OffsetToLock, SizeToLock, ppbData, Flags);
	}

	void VertexBufferFix(IDirect3DVertexBuffer9* pVertexBuffer)
	{
		if (pVertexBuffer->lpVtbl->Lock && TrueLock == nullptr)
		{
			PrintLog("WARNING: Experimental DiscardUIVertexBuffer enabled!");
			const MH_STATUS createHookLock = MH_CreateHook(pVertexBuffer->lpVtbl->Lock, HookLock, reinterpret_cast<void**>(&TrueLock));
			PrintLog("CreateHookLock = %d", createHookLock);
			const MH_STATUS enableHookLock = MH_EnableHook(pVertexBuffer->lpVtbl->Lock);
			PrintLog("EnableHookLock = %d", enableHookLock);
		}
	}
}