mirror of https://github.com/PCSX2/pcsx2.git
72 lines
2.0 KiB
C++
72 lines
2.0 KiB
C++
|
#include "WndProcEater.h"
|
||
|
|
|
||
|
|
static HWND hWndEaten = 0;
|
||
|
|
static WNDPROC eatenWndProc = 0;
|
||
|
|
static ExtraWndProc* extraProcs = 0;
|
||
|
|
static int numExtraProcs = 0;
|
||
|
|
|
||
|
|
void ReleaseExtraProc(ExtraWndProc proc) {
|
||
|
|
for (int i=0; i<numExtraProcs; i++) {
|
||
|
|
if (extraProcs[i] == proc) {
|
||
|
|
extraProcs[i] = extraProcs[--numExtraProcs];
|
||
|
|
break;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
if (!numExtraProcs && eatenWndProc) {
|
||
|
|
free(extraProcs);
|
||
|
|
extraProcs = 0;
|
||
|
|
if (hWndEaten && IsWindow(hWndEaten)) {
|
||
|
|
SetWindowLongPtr(hWndEaten, GWLP_WNDPROC, (LONG_PTR)eatenWndProc);
|
||
|
|
}
|
||
|
|
hWndEaten = 0;
|
||
|
|
eatenWndProc = 0;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
void ReleaseEatenProc() {
|
||
|
|
while (numExtraProcs) ReleaseExtraProc(extraProcs[0]);
|
||
|
|
}
|
||
|
|
|
||
|
|
LRESULT CALLBACK OverrideWndProc(HWND hWnd, UINT uMsg, WPARAM wParam, LPARAM lParam) {
|
||
|
|
ExtraWndProcResult res = CONTINUE_BLISSFULLY;
|
||
|
|
LRESULT out = 0;
|
||
|
|
// Here because want it for binding, even when no keyboard mode is selected.
|
||
|
|
if (uMsg == WM_GETDLGCODE) {
|
||
|
|
return DLGC_WANTALLKEYS | CallWindowProc(eatenWndProc, hWnd, uMsg, wParam, lParam);
|
||
|
|
}
|
||
|
|
for (int i=0; i<numExtraProcs; i++) {
|
||
|
|
ExtraWndProcResult res2 = extraProcs[i](hWnd, uMsg, wParam, lParam, &out);
|
||
|
|
if (res2 != res) {
|
||
|
|
if (res2 == CONTINUE_BLISSFULLY_AND_RELEASE_PROC) {
|
||
|
|
ReleaseExtraProc(extraProcs[i]);
|
||
|
|
i--;
|
||
|
|
}
|
||
|
|
else if (res2 > res) res = res2;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
if (res != NO_WND_PROC) {
|
||
|
|
if (out == WM_DESTROY) {
|
||
|
|
ReleaseEatenProc();
|
||
|
|
}
|
||
|
|
if (res == CONTINUE_BLISSFULLY)
|
||
|
|
out = CallWindowProc(eatenWndProc, hWnd, uMsg, wParam, lParam);
|
||
|
|
else if (res == USE_DEFAULT_WND_PROC)
|
||
|
|
out = DefWindowProc(hWnd, uMsg, wParam, lParam);
|
||
|
|
}
|
||
|
|
return out;
|
||
|
|
}
|
||
|
|
|
||
|
|
|
||
|
|
int EatWndProc(HWND hWnd, ExtraWndProc proc) {
|
||
|
|
if (hWnd != hWndEaten) {
|
||
|
|
ReleaseEatenProc();
|
||
|
|
eatenWndProc = (WNDPROC) SetWindowLongPtr(hWnd, GWLP_WNDPROC, (LONG_PTR)OverrideWndProc);
|
||
|
|
// ???
|
||
|
|
if (!eatenWndProc) return 0;
|
||
|
|
hWndEaten = hWnd;
|
||
|
|
}
|
||
|
|
extraProcs = (ExtraWndProc*) realloc(extraProcs, sizeof(ExtraWndProc)*(numExtraProcs+1));
|
||
|
|
extraProcs[numExtraProcs++] = proc;
|
||
|
|
return 1;
|
||
|
|
}
|