#include "dxhk.h"
#include "inter.h"

HOOK_DATA* hk;
D3D11_RENDERDATA* dx11;

U64 __fastcall hk_multiplane_draw( 
  void* dxdev, 
  IDXGISwapChain* swpchain, 
  U32 a3, 
  U32 a4, 
  U32 a5, 
  U64 a6, 
  U64 a7, 
  U32 a8 
) {
  static decltype( &hk_multiplane_draw ) original = (decltype( &hk_multiplane_draw ))hk->multiplane_o;
  
  U64 ret = original( dxdev, swpchain, a3, a4, a5, a6, a7, a8 );
  return ret;
}

U64 __fastcall hk_present( 
  void* dxdev, 
  IDXGISwapChain* swpchain, 
  U32 a3, 
  U32 a4, 
  const RECT* rect, 
  U64 a6, 
  void* a7, 
  U32 a8, 
  void* rsrc, 
  U32 a10 
) {
  static decltype( &hk_present ) original = (decltype( &hk_present ))hk->present_o;

  U64 ret = original( dxdev, swpchain, a3, a4, rect, a6, a7, a8, rsrc, a10 );
  return ret;
}

U64 __stdcall hk_wndproc( HWND hwnd, UINT msg, WPARAM wparam, LPARAM lparam ) {
  U64 ret = CallWindowProcA( (WNDPROC)hk->wndproc_o, hwnd, msg, wparam, lparam );


  return ret;
}

HOOK_DATA* dxhk_init() {
  HOOK_DATA* hkd = new HOOK_DATA;

  U64 dxdev = u_find_pattern( (U64)GetModuleHandleA( "d2d1.dll" ), "48 8D 05 ? ? ? ? 33 ED 48 8D 71 08" );
  if( !dxdev ) {
    delete hkd;
    return 0;
  }

  dxdev = dxdev + *(U64*)( dxdev + 3 ) + 7;
  VMT vmt( *(U64*)dxdev );

  hkd->multiplane_o = vmt.set( 7, (U64)&hk_multiplane_draw );
  hkd->present_o = vmt.set( 6, (U64)&hk_present );
  hkd->wndproc_o = SetWindowLongPtrA( FindWindowA( "Progman", "Program Manager" ), GWLP_WNDPROC, (LONG_PTR)&hk_wndproc );

  dx11 = d3d11_init();

  return hkd;
}