Prometheus/Main.h at master · morrx/Prometheus · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
#pragma once
#include "HookIncludes.h"
#include "Interface.h"
#include "Parser.h"
#include "Variables.h"
#include <Windows.h>
#include <d3d9.h>
#include <iostream>
#include <d3dx9.h>
#pragma comment(lib,"d3d9.lib")
#pragma comment(lib,"d3dx9.lib")


typedef HRESULT(__stdcall *EndScene_t) (IDirect3DDevice9*);
typedef HRESULT(__stdcall *Reset_t) (IDirect3DDevice9*, D3DPRESENT_PARAMETERS*);
extern LRESULT ImGui_ImplDX9_WndProcHandler(HWND hWnd, UINT msg, WPARAM wParam, LPARAM lParam);


EndScene_t oEndScene = nullptr;
Reset_t	oReset = nullptr;
HWND window = nullptr;

namespace handlers
{


	LRESULT __stdcall Hooked_WndProc(HWND hWnd, UINT uMsg, WPARAM wParam, LPARAM lParam)
	{

		switch (uMsg)
		{
		case WM_LBUTTONDOWN:
			G::PressedKeys[VK_LBUTTON] = true;
			break;
		case WM_LBUTTONUP:
			G::PressedKeys[VK_LBUTTON] = false;
			break;
		case WM_RBUTTONDOWN:
			G::PressedKeys[VK_RBUTTON] = true;
			break;
		case WM_RBUTTONUP:
			G::PressedKeys[VK_RBUTTON] = false;
			break;
		case WM_KEYDOWN:
			G::PressedKeys[wParam] = true;
			break;
		case WM_KEYUP:
			G::PressedKeys[wParam] = false;
			break;
		default: break;
		}

		static bool isDown = false;
		static bool isClicked = false;
		if (G::PressedKeys[VK_INSERT])
		{
			isClicked = false;
			isDown = true;
		}
		else if (!G::PressedKeys[VK_INSERT] && isDown)
		{
			isClicked = true;
			isDown = false;
		}
		else
		{
			isClicked = false;
			isDown = false;
		}

		if (isClicked)
		{
			g_Options.Menu.Opened = !g_Options.Menu.Opened;
			static auto cl_mouseenable = g_CVar->FindVar("cl_mouseenable");
			cl_mouseenable->SetValue(!g_Options.Menu.Opened);
		}

		if (G::d3dinit && g_Options.Menu.Opened && ImGui_ImplDX9_WndProcHandler(hWnd, uMsg, wParam, lParam))
			return true;

		return CallWindowProc(G::OldWindow, hWnd, uMsg, wParam, lParam);
	}


	HRESULT __stdcall Hooked_EndScene(IDirect3DDevice9* pDevice)
	{

		pDevice->SetRenderState(D3DRS_COLORWRITEENABLE, 0xFFFFFFFF);

			if (!G::d3dinit)
			{
				prometheus::GUI_Init(window, pDevice);
			}


			ImGui::GetIO().MouseDrawCursor = g_Options.Menu.Opened;

			ImGui_ImplDX9_NewFrame();

			if (g_Options.Menu.Opened)
			{
				prometheus::mainWindow();
			}


			if (g_Options.Misc.SpecList)
			{
				speclist();
			}

			if (g_Options.Misc.radarwindow)
			{
				DrawRadar();
			}


			ImGui::Render();


		return hooks::d3d.get_original<EndScene_t>(42)(pDevice);
	}


	HRESULT __stdcall Hooked_Reset(IDirect3DDevice9* pDevice, D3DPRESENT_PARAMETERS* params)
	{
		if (!G::d3dinit)
			return hooks::d3d.get_original<Reset_t>(16)(pDevice, params);

		ImGui_ImplDX9_InvalidateDeviceObjects();

		auto hr = hooks::d3d.get_original<Reset_t>(16)(pDevice, params);


		if (hr >= 0)
		{
			ImGui_ImplDX9_CreateDeviceObjects();
			g_Render->SetupFonts();
		}

		return hr;
	}

}