isledecomp/LEGOIslandRebuilder
1
0
Fork 0
mirror of https://github.com/isledecomp/LEGOIslandRebuilder.git synced 2024-11-30 02:55:55 -05:00
Code Issues Projects Releases Packages Wiki Activity Actions

lib: improved pick/getpickrecords implementation

Browse source
This commit is contained in:
itsmattkc 2022-07-20 00:14:36 -07:00
parent 13701075d8
commit 16b69d91d1
1 changed files with 52 additions and 44 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

96
lib/hooks.cpp
View file

@ -286,8 +286,6 @@ HRESULT WINAPI InterceptD3DRMDeviceUpdate(LPDIRECT3DRMDEVICE device)
return d3drmDeviceUpdateOriginal(device); return d3drmDeviceUpdateOriginal(device);
} }
std::map<LPDIRECT3DDEVICE, std::vector<D3DPICKRECORD> > g_PickRecords;
void MultiplyMatrix(D3DMATRIX &c, const D3DMATRIX &a, const D3DMATRIX &b) void MultiplyMatrix(D3DMATRIX &c, const D3DMATRIX &a, const D3DMATRIX &b)
{ {
c._11 = (a._11 * b._11) + (a._21 * b._12) + (a._31 * b._13) + (a._41 * b._14); c._11 = (a._11 * b._11) + (a._21 * b._12) + (a._31 * b._13) + (a._41 * b._14);
@ -311,9 +309,22 @@ void MultiplyMatrix(D3DMATRIX &c, const D3DMATRIX &a, const D3DMATRIX &b)
c._44 = (a._14 * b._41) + (a._24 * b._42) + (a._34 * b._43) + (a._44 * b._44); c._44 = (a._14 * b._41) + (a._24 * b._42) + (a._34 * b._43) + (a._44 * b._44);
} }
// Adapted from: http://totologic.blogspot.com/2014/01/accurate-point-in-triangle-test.html
bool BarycentricPointInTriangle(float x1, float y1, float x2, float y2, float x3, float y3, float x, float y)
{
float denominator = ((y2 - y3)*(x1 - x3) + (x3 - x2)*(y1 - y3));
float a = ((y2 - y3)*(x - x3) + (x3 - x2)*(y - y3)) / denominator;
float b = ((y3 - y1)*(x - x3) + (x1 - x3)*(y - y3)) / denominator;
float c = 1 - a - b;
return 0 <= a && a <= 1 && 0 <= b && b <= 1 && 0 <= c && c <= 1;
}
std::vector<D3DPICKRECORD> g_PickRecords;
typedef HRESULT (WINAPI *d3dimPickFunction) (LPDIRECT3DDEVICE, LPDIRECT3DEXECUTEBUFFER, LPDIRECT3DVIEWPORT, DWORD, LPD3DRECT); typedef HRESULT (WINAPI *d3dimPickFunction) (LPDIRECT3DDEVICE, LPDIRECT3DEXECUTEBUFFER, LPDIRECT3DVIEWPORT, DWORD, LPD3DRECT);
d3dimPickFunction d3dimPickOriginal = NULL; d3dimPickFunction d3dimPickOriginal = NULL;
HRESULT WINAPI InterceptD3DIMPick(LPDIRECT3DDEVICE lpDevice, LPDIRECT3DEXECUTEBUFFER lpBuffer, LPDIRECT3DVIEWPORT lpViewport, DWORD, LPD3DRECT lpRect) HRESULT WINAPI InterceptD3DIMPick(LPDIRECT3DDEVICE lpDevice, LPDIRECT3DEXECUTEBUFFER lpBuffer, LPDIRECT3DVIEWPORT lpViewport, DWORD dwFlags, LPD3DRECT lpRect)
{ {
D3DEXECUTEBUFFERDESC desc; D3DEXECUTEBUFFERDESC desc;
ZeroMemory(&desc, sizeof(D3DEXECUTEBUFFERDESC)); ZeroMemory(&desc, sizeof(D3DEXECUTEBUFFERDESC));
@ -334,11 +345,10 @@ HRESULT WINAPI InterceptD3DIMPick(LPDIRECT3DDEVICE lpDevice, LPDIRECT3DEXECUTEBU
char *instr = (char *)desc.lpData + data.dwInstructionOffset; char *instr = (char *)desc.lpData + data.dwInstructionOffset;
D3DVERTEX *vertex = (D3DVERTEX *) ((char *)desc.lpData + data.dwVertexOffset); D3DVERTEX *vertex = (D3DVERTEX *) ((char *)desc.lpData + data.dwVertexOffset);
std::vector<D3DVERTEX> dst_buf; std::vector<D3DTLVERTEX> dst_buf;
dst_buf.resize(data.dwVertexCount); dst_buf.resize(data.dwVertexCount);
std::vector<D3DPICKRECORD> &records = g_PickRecords[lpDevice]; g_PickRecords.clear();
records.clear();
LPD3DINSTRUCTION current; LPD3DINSTRUCTION current;
while (current = (LPD3DINSTRUCTION)instr, current->bOpcode != D3DOP_EXIT) { while (current = (LPD3DINSTRUCTION)instr, current->bOpcode != D3DOP_EXIT) {
@ -353,40 +363,27 @@ HRESULT WINAPI InterceptD3DIMPick(LPDIRECT3DDEVICE lpDevice, LPDIRECT3DEXECUTEBU
for (WORD i=0; i<count; i++) { for (WORD i=0; i<count; i++) {
D3DTRIANGLE *ci = (D3DTRIANGLE*)instr; D3DTRIANGLE *ci = (D3DTRIANGLE*)instr;
bool inside_x = lpRect->x1 >= min(dst_buf[ci->v1].x, min(dst_buf[ci->v2].x, dst_buf[ci->v3].x)) && lpRect->x1 <= max(dst_buf[ci->v1].x, max(dst_buf[ci->v2].x, dst_buf[ci->v3].x)); D3DTLVERTEX *verts[3];
bool inside_y = lpRect->y1 >= min(dst_buf[ci->v1].y, min(dst_buf[ci->v2].y, dst_buf[ci->v3].y)) && lpRect->y1 <= max(dst_buf[ci->v1].y, max(dst_buf[ci->v2].y, dst_buf[ci->v3].y)); verts[0] = &dst_buf[ci->v1];
verts[1] = &dst_buf[ci->v2];
verts[2] = &dst_buf[ci->v3];
if (inside_x && inside_y) { LONG x = lpRect->x1;
LONG y = lpRect->y1;
if (BarycentricPointInTriangle(verts[0]->sx, verts[0]->sy, verts[1]->sx, verts[1]->sy,
verts[2]->sx, verts[2]->sy, x, y)) {
D3DPICKRECORD record; D3DPICKRECORD record;
record.bOpcode = current->bOpcode; record.bOpcode = current->bOpcode;
record.bPad = 0;
// Write current instruction offset into file // Write current instruction offset into file
record.dwOffset = (DWORD)((LPBYTE)current + sizeof(D3DINSTRUCTION) - ((LPBYTE)desc.lpData + data.dwInstructionOffset)); record.dwOffset = (DWORD)ci - ((DWORD)desc.lpData + data.dwInstructionOffset);
// Just get the center Z // Just get the center Z
record.dvZ = (vertex[ci->v1].z + vertex[ci->v2].z + vertex[ci->v3].z) / 3; record.dvZ = (verts[0]->sz + verts[1]->sz + verts[2]->sz) / 3;
if (records.empty()) { g_PickRecords.push_back(record);
records.push_back(record);
} else if (record.dvZ > records[0].dvZ) {
records[0] = record;
}
printf("Clicked %u,%u %f to %f,%f %f,%f %f,%f\n",
lpRect->x1, lpRect->y1, record.dvZ,
dst_buf[ci->v1].x, dst_buf[ci->v1].y,
dst_buf[ci->v2].x, dst_buf[ci->v2].y,
dst_buf[ci->v3].x, dst_buf[ci->v3].y
);
} else {
/*printf("Comparing %u,%u to %f,%f %f,%f %f,%f failed\n",
lpRect->x1, lpRect->y1,
dst_buf[ci->v1].x, dst_buf[ci->v1].y,
dst_buf[ci->v2].x, dst_buf[ci->v2].y,
dst_buf[ci->v3].x, dst_buf[ci->v3].y
);*/
} }
instr += size; instr += size;
@ -446,7 +443,7 @@ HRESULT WINAPI InterceptD3DIMPick(LPDIRECT3DDEVICE lpDevice, LPDIRECT3DEXECUTEBU
MultiplyMatrix(mat, proj, mat); MultiplyMatrix(mat, proj, mat);
for (DWORD j=0; j<ci->dwCount; j++) { for (DWORD j=0; j<ci->dwCount; j++) {
D3DVERTEX in = vertex[ci->wStart+j]; const D3DVERTEX &in = vertex[ci->wStart+j];
float x, y, z, rhw; float x, y, z, rhw;
@ -461,19 +458,21 @@ HRESULT WINAPI InterceptD3DIMPick(LPDIRECT3DDEVICE lpDevice, LPDIRECT3DEXECUTEBU
y *= -1; y *= -1;
x *= vp.dwWidth / 2; x *= vp.dvScaleX;
y *= vp.dwHeight / 2; y *= vp.dvScaleY;
z *= vp.dvMaxZ - vp.dvMinZ; z *= vp.dvMaxZ - vp.dvMinZ;
x += vp.dwWidth / 2 + vp.dwX; x += vp.dwWidth / 2 + vp.dwX;
y += vp.dwHeight / 2 + vp.dwY; y += vp.dwHeight / 2 + vp.dwY;
z += vp.dvMinZ; z += vp.dvMinZ;
D3DVERTEX &out = dst_buf[ci->wDest+j]; rhw = 1 / rhw;
out = in;
out.x = x; D3DTLVERTEX &out = dst_buf[ci->wDest+j];
out.y = y; out.sx = x;
out.z = z; out.sy = y;
out.sz = in.z; // Use original Z so we can Z order on return
out.rhw = rhw;
} }
break; break;
} }
@ -484,10 +483,15 @@ HRESULT WINAPI InterceptD3DIMPick(LPDIRECT3DDEVICE lpDevice, LPDIRECT3DEXECUTEBU
instr += size; instr += size;
} }
d3dev2->Release();
break; break;
} }
default: default:
printf("unhandled op: %lx\n", current->bOpcode); printf("unhandled op: %lx\n", current->bOpcode);
case D3DOP_STATELIGHT:
case D3DOP_STATERENDER:
case D3DOP_SETSTATUS:
instr += size * count; instr += size * count;
break; break;
} }
@ -507,11 +511,10 @@ HRESULT WINAPI InterceptD3DIMGetPickRecords(LPDIRECT3DDEVICE lpDevice, LPDWORD d
return DDERR_INVALIDPARAMS; return DDERR_INVALIDPARAMS;
} }
std::vector<D3DPICKRECORD> &records = g_PickRecords[lpDevice]; if (lpRecords && *dwCount >= g_PickRecords.size()) {
if (lpRecords && *dwCount >= records.size()) { memcpy(lpRecords, &g_PickRecords[0], sizeof(D3DPICKRECORD) * g_PickRecords.size());
memcpy(lpRecords, &records[0], records.size() * sizeof(D3DPICKRECORD));
} }
*dwCount = records.size(); *dwCount = g_PickRecords.size();
return D3D_OK; return D3D_OK;
} }
@ -589,9 +592,14 @@ LRESULT CALLBACK InterceptWindowProc(HWND hwnd, UINT uMsg, WPARAM wParam, LPARAM
return 0; return 0;
} else if (uMsg == WM_KEYDOWN) { } else if (uMsg == WM_KEYDOWN) {
if (wParam == '5') { switch (wParam) {
case '5':
d3drm_device->SetQuality(D3DRMRENDER_WIREFRAME); d3drm_device->SetQuality(D3DRMRENDER_WIREFRAME);
printf("quality should have changed\n"); printf("quality should have changed\n");
break;
case VK_BACK: // Clear console, useful for debugging
system("cls");
break;
} }
} }