Properly handle input while menu is visible.

* INSERT toggles the menu.

Signed-off-by: aixxe <me@aixxe.net>

include/cstrike/Interfaces/IInputInternal.h

@@ -11,8 +11,8 @@ enum MouseCodeState_t {
 
 class IInputInternal {
 	public:
-		void SetKeyCodeState(KeyCode code, bool pressed) {
-			GetVirtualFunction<void(*)(IInputInternal*, KeyCode, bool)>(this, 83)(this, code, pressed);
+		void SetKeyCodeState(KeyCode code, bool down) {
+			GetVirtualFunction<void(*)(IInputInternal*, KeyCode, bool)>(this, 83)(this, code, down);
 		}
 
 		void SetMouseCodeState(MouseCode code, MouseCodeState_t state) {

src/GUI/Components.cpp

@@ -0,0 +1,14 @@
+#include "GUI.h"
+
+void GUI::DrawFramerateCounter() {
+	ImGui::SetNextWindowPos(ImVec2(10, 10));
+	ImGui::Begin("FPS", nullptr, ImVec2(0, 0), 0.3f, ImGuiWindowFlags_NoTitleBar | ImGuiWindowFlags_NoResize);
+	ImGui::Text("FPS: %.2f", ImGui::GetIO().Framerate);
+	ImGui::End();
+}
+
+void GUI::DrawConfigurationWindow() {
+	ImGui::Begin("cstrike-basehook-linux", nullptr);
+	ImGui::Text("Hello, world!");
+	ImGui::End();
+}

src/GUI/GUI.cpp

@@ -0,0 +1,12 @@
+#include "GUI.h"
+
+bool GUI::IsVisible = false;
+
+void GUI::Render() {
+	// Draw various global components.
+	GUI::DrawFramerateCounter();
+
+	// Draw the actual configuration window.
+	if (GUI::IsVisible)
+		GUI::DrawConfigurationWindow();
+}

src/GUI/GUI.h

@@ -0,0 +1,13 @@
+#pragma once
+
+#include <imgui/imgui.h>
+#include <imgui/imgui_impl_sdl.h>
+
+namespace GUI {
+	extern bool IsVisible;
+
+	void DrawFramerateCounter();
+	void DrawConfigurationWindow();
+
+	void Render();
+}

src/Hooks/Hooks.h

@@ -1,6 +1,7 @@
 #pragma once
 
 #include "../Basehook.h"
+#include "../GUI/GUI.h"
 
 namespace Hooks {
 	// IBaseClientDLL

src/Hooks/PumpWindowsMessageLoop.cpp

@@ -6,6 +6,10 @@ void Hooks::PumpWindowsMessageLoop(ILauncherMgr* thisptr) {
 	// Get the original function and store it statically.
 	static PumpWindowsMessageLoop_t oPumpWindowsMessageLoop = sdl_hook->GetOriginalFunction<PumpWindowsMessageLoop_t>(15);
 
+	// Block game input while the GUI is visible.
+	if (GUI::IsVisible)
+		return;
+
 	// Call original 'ILauncherMgr::PumpWindowsMessageLoop'.
 	oPumpWindowsMessageLoop(thisptr);
 }

src/Hooks/SetKeyCodeState.cpp

@@ -2,13 +2,17 @@
 
 typedef void (*SetKeyCodeState_t) (IInputInternal*, KeyCode, bool);
 
-void Hooks::SetKeyCodeState(IInputInternal* thisptr, KeyCode code, bool pressed) {
+void Hooks::SetKeyCodeState(IInputInternal* thisptr, KeyCode code, bool down) {
 	// Get the original function and store it statically.
 	static SetKeyCodeState_t oSetKeyCodeState = inputinternal_hook->GetOriginalFunction<SetKeyCodeState_t>(83);
 
 	// Print to console every time a key is pressed or released.
-	cvar->ConsoleColorPrintf(Color(150, 150, 255, 255), "IInputInternal::SetKeyCodeState - code: %i, pressed: %i\n", code, pressed);
+	cvar->ConsoleColorPrintf(Color(150, 150, 255, 255), "IInputInternal::SetKeyCodeState - code: %i, down: %i\n", code, down);
+
+	// Check if we should open the menu.
+	if (code == KeyCode::KEY_INSERT && down == false)
+		GUI::IsVisible = true;
 
 	// Call original 'IInputInternal::SetKeyCodeState'.
-	oSetKeyCodeState(thisptr, code, pressed);
+	oSetKeyCodeState(thisptr, code, down);
 }

src/Hooks/ShowPixels.cpp

@@ -1,7 +1,5 @@
 #include "Hooks.h"
-
-#include <imgui/imgui.h>
-#include <imgui/imgui_impl_sdl.h>
+#include "../GUI/GUI.h"
 
 typedef void (*ShowPixels_t) (ILauncherMgr*, CShowPixelsParams*);
 
@@ -30,8 +28,25 @@ void Hooks::ShowPixels(ILauncherMgr* thisptr, CShowPixelsParams* params) {
 	// Start ImGui rendering.
 	ImGui_ImplSdl_NewFrame(window);
 
-	// Draw some test stuff.
-	ImGui::Text("Hello, world!");
+	// Enable or disable the ImGui cursor depending on the GUI visibility.
+	ImGui::GetIO().MouseDrawCursor = GUI::IsVisible;
+	ImGui::GetIO().WantCaptureMouse = GUI::IsVisible;
+	ImGui::GetIO().WantCaptureKeyboard = GUI::IsVisible;
+
+	// Handle incoming input while the menu is active.
+	if (GUI::IsVisible) {
+		SDL_Event event = {};
+
+		while (SDL_PollEvent(&event)) {
+			if (event.key.keysym.sym == SDLK_INSERT && event.type == SDL_KEYDOWN)
+				GUI::IsVisible = !GUI::IsVisible;
+
+			ImGui_ImplSdl_ProcessEvent(&event);
+		}
+	} 
+
+	// Draw GUI components.
+	GUI::Render();
 
 	// Finish ImGui rendering.
 	ImGui::Render();