InGameUI.cpp

#include "stdafx.h"
#include "InGameUI.h"
#include "d3d9.h"
#include "dinput8.h"
#include "Hotkeys.h"
#include "ImGui/imgui_impl_dx9.h"
#include "ImGui/imgui_internal.h"

namespace
{

struct fontInfo
{
	LPCSTR name;
	float size;
	ImFont **font;
	fontInfo(LPCSTR n, float s, ImFont **f) : name(n), size(s), font(f) { }
};

vector<void(*)()> content;
vector<void(*)(bool)> windows;
vector<fontInfo> fonts;
void onLostDevice() { ImGui_ImplDX9_InvalidateDeviceObjects(); }
void onResetDevice() { ImGui_ImplDX9_CreateDeviceObjects(); }
void onCreateDevice(LPDIRECT3DDEVICE9 pD3DDevice)
{
	ImGui_ImplDX9_Init(pD3DDevice);
	ImGui::GetIO().IniFilename = nullptr;
	ImGui::GetStyle().WindowTitleAlign = ImGuiAlign_Center;
	ImGui::GetStyle().WindowFillAlphaDefault = 0.95f;
	ImGui::GetIO().Fonts->AddFontDefault();
	for (size_t i = 0; i < fonts.size(); i++)
	{
		CHAR syspath[MAX_PATH];
		GetWindowsDirectory(syspath, MAX_PATH);
		strcat_s(syspath, "\\Fonts\\");
		strcat_s(syspath, fonts[i].name);

		ImFont **font = fonts[i].font;
		*font = ImGui::GetIO().Fonts->AddFontFromFileTTF(syspath, fonts[i].size);
		if (!*font)
			LOG("InGameClock: failed to load font - %s\n", syspath);
	}
}

bool inGameUIEnabled = false;
void onEndScene()
{
	ImGui_ImplDX9_NewFrame();
	for (size_t i = 0; i < windows.size(); i++)
		windows[i](inGameUIEnabled);
	ImGui::Render();
}

void renderDDDAFixUI(bool getsInput)
{
	if (!getsInput)
		return;

	static char titleBuffer[64];
	sprintf_s(titleBuffer, "DDDAFix - %.1f FPS###DDDAFix", ImGui::GetIO().Framerate);
	ImGui::SetNextWindowPos(ImVec2(5, 5), ImGuiSetCond_Once);
	ImGui::Begin(titleBuffer, nullptr, ImVec2(450, 600));
	for (size_t i = 0; i < content.size(); i++)
	{
		ImGui::PushID(i);
		content[i]();
		ImGui::PopID();
	}
	ImGui::End();
}

LPBYTE pInGameUI, oInGameUI;
SHORT WINAPI HGetAsyncKeyState(int vKey) { return ImGui::IsAnyItemActive() ? 0 : GetAsyncKeyState(vKey); }
void __declspec(naked) HInGameUI()
{
	asm("mov	$%0,%%ebp" : : "m"(HGetAsyncKeyState));
	asm("jmp	*%0" : : "m"(oInGameUI));
}

WPARAM inGameUIHotkey;
LRESULT CALLBACK inGameUIEvent(HWND hwnd, UINT msg, WPARAM wParam, LPARAM lParam)
{
	if (msg == WM_KEYDOWN && (HIWORD(lParam) & KF_REPEAT) == 0 && wParam == inGameUIHotkey)
		Hooks::SwitchHook("InGameUI", pInGameUI, inGameUIEnabled = !inGameUIEnabled);
	return inGameUIEnabled ? ImGui_ImplDX9_WndProcHandler(hwnd, msg, wParam, lParam) : 0;
}

LRESULT CALLBACK inGameUIInit(HWND hwnd, UINT msg, WPARAM wParam, LPARAM lParam)
{
	if (Hooks::D3D9(onCreateDevice, onLostDevice, onResetDevice, onEndScene))
		Hooks::HotkeysHandler(inGameUIEvent);
	return 0;
}

}

bool Hooks::InGameUI()
{
	if (!config.getBool("inGameUI", "enabled", false))
	{
		LOG("InGameUI: disabled\n");
		return false;
	}

	BYTE sigRun[] = { 0x8B, 0x2D, 0xCC, 0xCC, 0xCC, 0xCC,	//mov	ebp, ds:GetAsyncKeyState
					0x8D, 0x7E, 0x01 };						//lea	edi, [esi+1]
	if (FindSignature("InGameUI", sigRun, &pInGameUI))
	{
		CreateHook("InGameUI", pInGameUI, (LPVOID)HInGameUI, &oInGameUI, inGameUIEnabled);
		oInGameUI += 6;
	}

	inGameUIHotkey = config.getUInt("hotkeys", "keyUI", VK_F12) & 0xFF;
	InGameUIAddWindow(renderDDDAFixUI);
	HotkeysHandler(inGameUIInit);
	return true;
}

void Hooks::InGameUIAdd(void(*callback)()) { content.push_back(callback); }
void Hooks::InGameUIAddWindow(void(*callback)(bool getsInput)) { windows.push_back(callback); }
void Hooks::InGameUIAddFont(const char *filename, float size_pixels, ImFont **font) { fonts.emplace_back(filename, size_pixels, font); }

namespace ImGui
{
	bool InputFloatN(const char* label, float* v, int count, float item_width, float min, float max, int precision)
	{
		if (item_width > 0.0f)
			PushItemWidth(item_width * count);
		bool changed = InputFloatN(label, v, count, precision, 0);
		if (item_width > 0.0f)
			PopItemWidth();
		if (changed)
			for (int i = 0; i < count; i++)
			{
				if (v[i] < min)
					v[i] = min;
				if (v[i] > max)
					v[i] = max;
			}
		return changed;
	}

	bool InputFloatEx(const char* label, float* v, float step, float min, float max, int precision)
	{
		if (!InputFloat(label, v, step, step * 100, precision, 0))
			return false;
		if (*v < min)
			*v = min;
		if (*v > max)
			*v = max;
		return true;
	}

	void TextUnformatted(const char* label, float pos)
	{
		SameLine(pos);
		TextUnformatted(label);
	}
}