mirror of
https://github.com/isledecomp/LEGOIslandRebuilder.git
synced 2024-11-30 02:55:55 -05:00
418 lines
13 KiB
C++
418 lines
13 KiB
C++
#include "hooks.h"
|
|
|
|
#include <MATH.H>
|
|
#include <STDIO.H>
|
|
|
|
#include "../cmn/path.h"
|
|
#include "config.h"
|
|
#include "util.h"
|
|
|
|
HWND isleWindow = NULL;
|
|
|
|
void InterceptOutputDebugStringA(LPCSTR s)
|
|
{
|
|
printf("%s\n", s);
|
|
MessageBoxA(isleWindow, s, "LEGO Island sez", 0);
|
|
}
|
|
|
|
HWND WINAPI InterceptCreateWindowExA(DWORD dwExStyle, LPCSTR lpClassName, LPCSTR lpWindowName, DWORD dwStyle, int X, int Y, int nWidth, int nHeight, HWND hWndParent, HMENU hMenu, HINSTANCE hInstance, LPVOID lpParam)
|
|
{
|
|
// If full screen, hide the titlebar
|
|
if (config.GetInt(_T("FullScreen"))) {
|
|
dwStyle = WS_POPUP;
|
|
dwExStyle = 0;
|
|
}
|
|
|
|
// Grab a copy of the ISLE window so we can do stuff with it
|
|
isleWindow = CreateWindowExA(dwExStyle, lpClassName, "LEGO Island: Rebuilt", dwStyle, X, Y, nWidth, nHeight, hWndParent, hMenu, hInstance, lpParam);
|
|
return isleWindow;
|
|
}
|
|
|
|
HWND WINAPI InterceptFindWindowA(LPCSTR lpClassName, LPCSTR lpWindowName)
|
|
{
|
|
return NULL;
|
|
}
|
|
|
|
void ForceDDPixelFormatTo16(LPDDPIXELFORMAT lpDDPixelFormat)
|
|
{
|
|
if (lpDDPixelFormat->dwRGBBitCount == 32) {
|
|
lpDDPixelFormat->dwRGBBitCount = 16;
|
|
lpDDPixelFormat->dwRBitMask = 0xF800;
|
|
lpDDPixelFormat->dwGBitMask = 0x07E0;
|
|
lpDDPixelFormat->dwBBitMask = 0x001F;
|
|
}
|
|
}
|
|
|
|
void ForceDDSurfaceDescTo16(LPDDSURFACEDESC lpDDSurfaceDesc)
|
|
{
|
|
DDPIXELFORMAT &pixfmt = lpDDSurfaceDesc->ddpfPixelFormat;
|
|
if (pixfmt.dwRGBBitCount == 32) {
|
|
// LEGO Island has no support for 32-bit surfaces, so we tell it we're on a 16-bit surface
|
|
lpDDSurfaceDesc->lPitch /= 2;
|
|
ForceDDPixelFormatTo16(&pixfmt);
|
|
}
|
|
}
|
|
|
|
typedef HRESULT (WINAPI *ddSurfaceGetDescFunction)(LPDIRECTDRAWSURFACE lpDDSurface, LPDDSURFACEDESC lpDDSurfaceDesc);
|
|
ddSurfaceGetDescFunction originalDDSurfaceGetDescFunction = NULL;
|
|
HRESULT WINAPI InterceptSurfaceGetDesc(LPDIRECTDRAWSURFACE lpDDSurface, LPDDSURFACEDESC lpDDSurfaceDesc)
|
|
{
|
|
HRESULT res = originalDDSurfaceGetDescFunction(lpDDSurface, lpDDSurfaceDesc);
|
|
|
|
if (res == DD_OK) {
|
|
//ForceDDSurfaceDescTo16(lpDDSurfaceDesc);
|
|
}
|
|
|
|
return res;
|
|
}
|
|
|
|
typedef HRESULT (WINAPI *ddSurfaceGetPixelFormatFunction)(LPDIRECTDRAWSURFACE lpDDSurface, LPDDPIXELFORMAT lpDDPixelFormat);
|
|
ddSurfaceGetPixelFormatFunction originalDDSurfaceGetPixelFunction = NULL;
|
|
HRESULT WINAPI InterceptSurfaceGetPixelFormatFunction(LPDIRECTDRAWSURFACE lpDDSurface, LPDDPIXELFORMAT lpDDPixelFormat)
|
|
{
|
|
HRESULT res = originalDDSurfaceGetPixelFunction(lpDDSurface, lpDDPixelFormat);
|
|
|
|
if (res == DD_OK) {
|
|
//ForceDDPixelFormatTo16(lpDDPixelFormat);
|
|
}
|
|
|
|
return res;
|
|
}
|
|
|
|
typedef HRESULT (WINAPI *ddCreateSurfaceFunction)(LPDIRECTDRAW lpDD, LPDDSURFACEDESC lpDDSurfaceDesc, LPDIRECTDRAWSURFACE *lplpDDSurface, IUnknown *unknown);
|
|
ddCreateSurfaceFunction originalCreateSurfaceFunction = NULL;
|
|
HRESULT WINAPI InterceptCreateSurface(LPDIRECTDRAW lpDD, LPDDSURFACEDESC lpDDSurfaceDesc, LPDIRECTDRAWSURFACE *lplpDDSurface, IUnknown *unknown)
|
|
{
|
|
//ForceDDSurfaceDescTo16(lpDDSurfaceDesc);
|
|
|
|
HRESULT res = originalCreateSurfaceFunction(lpDD, lpDDSurfaceDesc, lplpDDSurface, unknown);
|
|
|
|
if (res == DD_OK) {
|
|
if (!originalDDSurfaceGetPixelFunction) {
|
|
originalDDSurfaceGetPixelFunction = (ddSurfaceGetPixelFormatFunction)OverwriteVirtualTable(*lplpDDSurface, 0x15, (LPVOID)InterceptSurfaceGetPixelFormatFunction);
|
|
}
|
|
|
|
if (!originalDDSurfaceGetDescFunction) {
|
|
//originalDDSurfaceGetDescFunction = (ddSurfaceGetDescFunction)OverwriteVirtualTable(*lplpDDSurface, 0x16, (LPVOID)InterceptSurfaceGetDesc);
|
|
originalDDSurfaceGetDescFunction = (ddSurfaceGetDescFunction)OverwriteVirtualTable(*lplpDDSurface, 0x16, NULL);
|
|
}
|
|
}
|
|
|
|
return res;
|
|
}
|
|
|
|
typedef HRESULT (WINAPI *ddGetDisplayModeFunction)(LPDIRECTDRAW lpDD, LPDDSURFACEDESC lpDDSurfaceDesc);
|
|
ddGetDisplayModeFunction originalGetDisplayMode = NULL;
|
|
HRESULT WINAPI InterceptGetDisplayMode(LPDIRECTDRAW lpDD, LPDDSURFACEDESC lpDDSurfaceDesc)
|
|
{
|
|
HRESULT res = originalGetDisplayMode(lpDD, lpDDSurfaceDesc);
|
|
|
|
//ForceDDSurfaceDescTo16(lpDDSurfaceDesc);
|
|
|
|
return res;
|
|
}
|
|
|
|
ddCreateFunction ddCreateOriginal = NULL;
|
|
HRESULT WINAPI InterceptDirectDrawCreate(GUID *lpGUID, LPDIRECTDRAW *lplpDD, IUnknown *pUnkOuter)
|
|
{
|
|
HRESULT res = ddCreateOriginal(lpGUID, lplpDD, pUnkOuter);
|
|
|
|
if (res == DD_OK) {
|
|
if (!originalGetDisplayMode) {
|
|
originalGetDisplayMode = (ddGetDisplayModeFunction)OverwriteVirtualTable(*lplpDD, 0xC, (LPVOID)InterceptGetDisplayMode);
|
|
}
|
|
|
|
ddCreateSurfaceFunction f = (ddCreateSurfaceFunction)OverwriteVirtualTable(*lplpDD, 0x6, (LPVOID)InterceptCreateSurface);
|
|
if (f != InterceptCreateSurface) {
|
|
originalCreateSurfaceFunction = f;
|
|
}
|
|
}
|
|
|
|
return res;
|
|
}
|
|
|
|
void ReturnRegistryYESNOFromBool(LPBYTE lpData, BOOL value)
|
|
{
|
|
strcpy((char*)lpData, value ? "YES" : "NO");
|
|
}
|
|
|
|
LONG WINAPI InterceptRegQueryValueExA(HKEY hKey, LPCSTR lpValueName, LPDWORD lpReserved, LPDWORD lpType, LPBYTE lpData, LPDWORD lpcbData)
|
|
{
|
|
if (!strcmp(lpValueName, "Music")) {
|
|
|
|
// Music option
|
|
ReturnRegistryYESNOFromBool(lpData, config.GetInt(_T("MusicToggle"), 1));
|
|
return ERROR_SUCCESS;
|
|
|
|
} else if (!strcmp(lpValueName, "UseJoystick")) {
|
|
|
|
ReturnRegistryYESNOFromBool(lpData, config.GetInt(_T("UseJoystick"), 0));
|
|
return ERROR_SUCCESS;
|
|
|
|
} else if (!strcmp(lpValueName, "Full Screen")) {
|
|
|
|
ReturnRegistryYESNOFromBool(lpData, config.GetInt(_T("FullScreen"), 1));
|
|
return ERROR_SUCCESS;
|
|
|
|
} else if (!strcmp(lpValueName, "Draw Cursor")) {
|
|
|
|
ReturnRegistryYESNOFromBool(lpData, config.GetInt(_T("DrawCursor"), 1));
|
|
return ERROR_SUCCESS;
|
|
|
|
} else if (!strcmp(lpValueName, "savepath")) {
|
|
|
|
// If enabled, return a safe %APPDATA% based save location rather than its default
|
|
// "C:\Program Files" location
|
|
if (config.GetInt(_T("RedirectSaveData"))) {
|
|
// Generate directory
|
|
TCHAR save_path[MAX_PATH];
|
|
if (GetSafeLEGOIslandSavePath(save_path)) {
|
|
strcpy((char*)lpData, save_path);
|
|
return ERROR_SUCCESS;
|
|
} else {
|
|
MessageBoxA(isleWindow, "Failed to redirect save path. Default will be used instead.", 0, 0);
|
|
}
|
|
}
|
|
|
|
} else if (!strcmp(lpValueName, "diskpath") || !strcmp(lpValueName, "cdpath")) {
|
|
|
|
// Pass through
|
|
|
|
} else {
|
|
|
|
printf("Passed through requested registry key \"%s\"\n", lpValueName);
|
|
|
|
}
|
|
|
|
// Pass these through
|
|
return RegQueryValueExA(hKey, lpValueName, lpReserved, lpType, lpData, lpcbData);
|
|
}
|
|
|
|
void WINAPI InterceptSleep(DWORD dwMilliseconds)
|
|
{
|
|
// If uncapped, do nothing. Otherwise pass through as normal.
|
|
if (config.GetString(_T("FPSLimit")) != "Uncapped") {
|
|
Sleep(dwMilliseconds);
|
|
}
|
|
}
|
|
|
|
LPDIRECT3DRMVIEWPORT last_viewport = NULL;
|
|
D3DVALUE last_fov = 0.0f;
|
|
typedef HRESULT (WINAPI *d3drmViewportSetFieldFunction)(LPDIRECT3DRMVIEWPORT viewport, D3DVALUE field);
|
|
d3drmViewportSetFieldFunction d3drmViewportSetFieldOriginal = NULL;
|
|
HRESULT WINAPI InterceptD3DRMViewportSetField(LPDIRECT3DRMVIEWPORT viewport, D3DVALUE field)
|
|
{
|
|
last_viewport = viewport;
|
|
last_fov = field;
|
|
|
|
return d3drmViewportSetFieldOriginal(viewport, field);
|
|
}
|
|
|
|
const SIZE_T max_frame_numbers = 60;
|
|
SIZE_T current_frame = 0;
|
|
DWORD frame_deltas[max_frame_numbers];
|
|
DWORD last_time = 0;
|
|
|
|
typedef HRESULT (WINAPI *d3drmDeviceUpdateFunction)(LPDIRECT3DRMDEVICE device);
|
|
d3drmDeviceUpdateFunction d3drmDeviceUpdateOriginal = NULL;
|
|
HRESULT WINAPI InterceptD3DRMDeviceUpdate(LPDIRECT3DRMDEVICE device)
|
|
{
|
|
DWORD now = timeGetTime();
|
|
|
|
if (last_time != 0) {
|
|
DWORD diff = now - last_time;
|
|
frame_deltas[current_frame%max_frame_numbers] = diff;
|
|
current_frame++;
|
|
|
|
if (current_frame >= max_frame_numbers) {
|
|
double avg_delta = 0;
|
|
for (size_t i=0; i<max_frame_numbers; i++) {
|
|
avg_delta += frame_deltas[i];
|
|
}
|
|
avg_delta /= max_frame_numbers;
|
|
printf("Avg FPS: %f\n", 1000.0 / avg_delta);
|
|
}
|
|
}
|
|
|
|
last_time = now;
|
|
|
|
return d3drmDeviceUpdateOriginal(device);
|
|
}
|
|
|
|
typedef HRESULT (WINAPI *d3drmCreateViewportFunction)(LPDIRECT3DRM d3drm, LPDIRECT3DRMDEVICE device, LPDIRECT3DRMFRAME frame, DWORD x, DWORD y, DWORD w, DWORD h, LPDIRECT3DRMVIEWPORT *viewport);
|
|
d3drmCreateViewportFunction d3drmCreateViewportOriginal = NULL;
|
|
HRESULT WINAPI InterceptD3DRMCreateViewport(LPDIRECT3DRM d3drm, LPDIRECT3DRMDEVICE device, LPDIRECT3DRMFRAME frame, DWORD x, DWORD y, DWORD w, DWORD h, LPDIRECT3DRMVIEWPORT *viewport)
|
|
{
|
|
HRESULT res = d3drmCreateViewportOriginal(d3drm, device, frame, x, y, w, h, viewport);
|
|
|
|
if (res == DD_OK) {
|
|
if (!d3drmDeviceUpdateOriginal) {
|
|
d3drmDeviceUpdateOriginal = (d3drmDeviceUpdateFunction)OverwriteVirtualTable(device, 14, (LPVOID)InterceptD3DRMDeviceUpdate);
|
|
}
|
|
|
|
if (!d3drmViewportSetFieldOriginal) {
|
|
d3drmViewportSetFieldOriginal = (d3drmViewportSetFieldFunction)OverwriteVirtualTable(*viewport, 0x10, (LPVOID)InterceptD3DRMViewportSetField);
|
|
}
|
|
}
|
|
|
|
return res;
|
|
}
|
|
|
|
d3drmCreateFunction d3drmCreateOriginal = NULL;
|
|
HRESULT WINAPI InterceptDirect3DRMCreate(LPDIRECT3DRM FAR *lplpDirect3DRM)
|
|
{
|
|
HRESULT result = d3drmCreateOriginal(lplpDirect3DRM);
|
|
|
|
if (result == DD_OK) {
|
|
if (!d3drmCreateViewportOriginal) {
|
|
d3drmCreateViewportOriginal = (d3drmCreateViewportFunction)OverwriteVirtualTable(*lplpDirect3DRM, 0x38, (LPVOID)InterceptD3DRMCreateViewport);
|
|
}
|
|
}
|
|
|
|
return result;
|
|
}
|
|
|
|
#ifndef WM_MOUSEWHEEL
|
|
#define WM_MOUSEWHEEL 0x020A
|
|
#endif
|
|
|
|
#ifndef WHEEL_DELTA
|
|
#define WHEEL_DELTA 120
|
|
#endif
|
|
|
|
WNDPROC originalWndProc = NULL;
|
|
LRESULT CALLBACK InterceptWindowProc(HWND hwnd, UINT uMsg, WPARAM wParam, LPARAM lParam)
|
|
{
|
|
if (uMsg == WM_MOUSEWHEEL) {
|
|
SHORT distance = HIWORD(wParam);
|
|
distance /= WHEEL_DELTA;
|
|
|
|
float multiplier = 0.005 * distance;
|
|
|
|
InterceptD3DRMViewportSetField(last_viewport, last_fov + multiplier);
|
|
|
|
return 0;
|
|
}
|
|
|
|
return originalWndProc(hwnd, uMsg, wParam, lParam);
|
|
}
|
|
|
|
ATOM WINAPI InterceptRegisterClassA(const WNDCLASSA *c)
|
|
{
|
|
WNDCLASSA copy = *c;
|
|
originalWndProc = copy.lpfnWndProc;
|
|
copy.lpfnWndProc = InterceptWindowProc;
|
|
|
|
return RegisterClassA(©);
|
|
}
|
|
|
|
typedef HRESULT (WINAPI *dsCreateSoundBufferFunction)(LPDIRECTSOUND lpDS, LPCDSBUFFERDESC desc, LPDIRECTSOUNDBUFFER *buffer, LPUNKNOWN unk);
|
|
dsCreateSoundBufferFunction dsCreateSoundBufferOriginal = NULL;
|
|
HRESULT WINAPI InterceptDirectSoundCreateSoundBuffer(LPDIRECTSOUND lpDS, LPCDSBUFFERDESC desc, LPDIRECTSOUNDBUFFER *buffer, LPUNKNOWN unk)
|
|
{
|
|
DSBUFFERDESC copy = *desc;
|
|
|
|
if (config.GetInt(_T("StayActiveWhenDefocused")) && !(copy.dwFlags & DSBCAPS_PRIMARYBUFFER)) {
|
|
copy.dwFlags |= DSBCAPS_GLOBALFOCUS;
|
|
}
|
|
|
|
return dsCreateSoundBufferOriginal(lpDS, ©, buffer, unk);
|
|
}
|
|
|
|
dsCreateFunction dsCreateOriginal = NULL;
|
|
HRESULT WINAPI InterceptDirectSoundCreate(LPGUID lpGuid, LPDIRECTSOUND* ppDS, LPUNKNOWN pUnkOuter )
|
|
{
|
|
HRESULT res = dsCreateOriginal(lpGuid, ppDS, pUnkOuter);
|
|
|
|
if (res == DD_OK) {
|
|
if (!dsCreateSoundBufferOriginal) {
|
|
dsCreateSoundBufferOriginal = (dsCreateSoundBufferFunction)OverwriteVirtualTable(*ppDS, 0x3, (LPVOID)InterceptDirectSoundCreateSoundBuffer);
|
|
}
|
|
}
|
|
|
|
return res;
|
|
}
|
|
|
|
SHORT WINAPI InterceptGetAsyncKeyState(int vKey)
|
|
{
|
|
if (config.GetInt("UseWASD")) {
|
|
switch (vKey) {
|
|
case VK_UP:
|
|
vKey = 'W';
|
|
break;
|
|
case VK_LEFT:
|
|
vKey = 'A';
|
|
break;
|
|
case VK_DOWN:
|
|
vKey = 'S';
|
|
break;
|
|
case VK_RIGHT:
|
|
vKey = 'D';
|
|
break;
|
|
case 'W':
|
|
vKey = VK_UP;
|
|
break;
|
|
case 'A':
|
|
vKey = VK_LEFT;
|
|
break;
|
|
case 'S':
|
|
vKey = VK_DOWN;
|
|
break;
|
|
case 'D':
|
|
vKey = VK_RIGHT;
|
|
break;
|
|
}
|
|
}
|
|
|
|
return GetAsyncKeyState(vKey);
|
|
}
|
|
|
|
typedef HRESULT (WINAPI *dinputGetDeviceStateFunction)(LPDIRECTINPUTDEVICEA lpDevice, DWORD cbData, LPVOID lpvData);
|
|
dinputGetDeviceStateFunction dinputGetDeviceStateOriginal = NULL;
|
|
HRESULT WINAPI InterceptDirectInputGetDeviceStateA(LPDIRECTINPUTDEVICEA lpDevice, DWORD cbData, LPVOID lpvData)
|
|
{
|
|
HRESULT res = dinputGetDeviceStateOriginal(lpDevice, cbData, lpvData);
|
|
|
|
if (config.GetInt("UseWASD") && res == DD_OK) {
|
|
if (cbData == 256) {
|
|
unsigned char *keys = (unsigned char *)lpvData;
|
|
|
|
// Swap state of WASD with arrow keys
|
|
std::swap(keys[DIK_W], keys[DIK_UP]);
|
|
std::swap(keys[DIK_A], keys[DIK_LEFT]);
|
|
std::swap(keys[DIK_S], keys[DIK_DOWN]);
|
|
std::swap(keys[DIK_D], keys[DIK_RIGHT]);
|
|
}
|
|
}
|
|
|
|
return res;
|
|
}
|
|
|
|
typedef HRESULT (WINAPI *dinputCreateDeviceFunction)(LPDIRECTINPUTA lpDI, REFGUID, LPDIRECTINPUTDEVICEA *, LPUNKNOWN);
|
|
dinputCreateDeviceFunction dinputCreateDeviceOriginal = NULL;
|
|
HRESULT WINAPI InterceptDirectInputCreateDeviceA(LPDIRECTINPUTA lpDI, REFGUID guid, LPDIRECTINPUTDEVICEA *ppDevice, LPUNKNOWN unk)
|
|
{
|
|
HRESULT res = dinputCreateDeviceOriginal(lpDI, guid, ppDevice, unk);
|
|
|
|
if (res == DD_OK) {
|
|
if (!dinputGetDeviceStateOriginal) {
|
|
dinputGetDeviceStateOriginal = (dinputGetDeviceStateFunction)OverwriteVirtualTable(*ppDevice, 0x9, (LPVOID)InterceptDirectInputGetDeviceStateA);
|
|
}
|
|
}
|
|
|
|
return res;
|
|
}
|
|
|
|
dinputCreateFunction dinputCreateOriginal = NULL;
|
|
HRESULT WINAPI InterceptDirectInputCreateA(HINSTANCE hinst, DWORD dwVersion, LPDIRECTINPUTA *ppDI, LPUNKNOWN punkOuter)
|
|
{
|
|
HRESULT res = dinputCreateOriginal(hinst, dwVersion, ppDI, punkOuter);
|
|
|
|
if (res == DD_OK) {
|
|
if (!dinputCreateDeviceOriginal) {
|
|
dinputCreateDeviceOriginal = (dinputCreateDeviceFunction)OverwriteVirtualTable(*ppDI, 0x3, (LPVOID)InterceptDirectInputCreateDeviceA);
|
|
}
|
|
}
|
|
|
|
return res;
|
|
}
|