From b9bc904d671d9c5eeeb7dc49f2549bf944bf0d0e Mon Sep 17 00:00:00 2001
From: Federico Cecchetto <fedecek3@gmail.com>
Date: Thu, 2 Jun 2022 22:49:09 +0200
Subject: [PATCH] Adjust red-dot brightness on some maps

---
 src/client/component/renderer.cpp | 92 +++++++++++++++++++++++++++++++
 1 file changed, 92 insertions(+)

diff --git a/src/client/component/renderer.cpp b/src/client/component/renderer.cpp
index eed4cd4e..fc65a0a2 100644
--- a/src/client/component/renderer.cpp
+++ b/src/client/component/renderer.cpp
@@ -15,6 +15,36 @@ namespace renderer
 		utils::hook::detour r_init_draw_method_hook;
 		utils::hook::detour r_update_front_end_dvar_options_hook;
 
+		utils::hook::detour db_load_xassets_hook;
+
+		game::dvar_t* r_red_dot_brightness_scale;
+		game::dvar_t* r_use_custom_red_dot_brightness;
+		float tonemap_highlight_range = 16.f;
+
+		std::unordered_map<std::string, float> tonemap_highlight_range_overrides =
+		{
+			// all these are 16 by default (except mp_bog & mp_cargoship which have it at 0) which makes red dots hard to see
+			{"mp_convoy", 22.f},
+			{"mp_backlot", 20.f},
+			{"mp_bog", 12.f},
+			{"mp_bloc", 30.f},
+			{"mp_countdown", 20.f},
+			{"mp_crash", 20.f},
+			{"mp_creek", 20.f},
+			{"mp_crossfire", 26.f},
+			{"mp_citystreets", 30.f},
+			{"mp_farm", 20.f},
+			{"mp_overgrown", 22.f},
+			{"mp_pipeline", 26.f},
+			{"mp_shipment", 20.f},
+			{"mp_showdown", 24.f},
+			{"mp_strike", 24.f},
+			{"mp_vacant", 20.f},
+			{"mp_cargoship", 14.f},
+			{"mp_crash_snow", 24.f},
+			{"mp_bog_summer", 24.f},
+		};
+
 		int get_fullbright_technique()
 		{
 			switch (dvars::r_fullbright->current.integer)
@@ -55,6 +85,55 @@ namespace renderer
 
 			return r_update_front_end_dvar_options_hook.invoke<bool>();
 		}
+
+		void set_tonemap_highlight_range()
+		{
+			auto* mapname = game::Dvar_FindVar("mapname");
+			if (mapname != nullptr && tonemap_highlight_range_overrides.find(mapname->current.string)
+				!= tonemap_highlight_range_overrides.end())
+			{
+				tonemap_highlight_range = tonemap_highlight_range_overrides[mapname->current.string];
+			}
+			else
+			{
+				tonemap_highlight_range = 16.f;
+			}
+		}
+
+		void db_load_xassets_stub(void* a1, void* a2, void* a3)
+		{
+			set_tonemap_highlight_range();
+			db_load_xassets_hook.invoke<void>(a1, a2, a3);
+		}
+
+		int get_red_dot_brightness()
+		{
+			static auto* r_tonemap_highlight_range = game::Dvar_FindVar("r_tonemapHighlightRange");
+			auto value = r_tonemap_highlight_range->current.value;
+			if (r_use_custom_red_dot_brightness->current.enabled)
+			{
+				value = tonemap_highlight_range * r_red_dot_brightness_scale->current.value;
+			}
+
+			return *reinterpret_cast<int*>(&value);
+		}
+
+		void* get_tonemap_highlight_range_stub()
+		{
+			return utils::hook::assemble([](utils::hook::assembler& a)
+			{
+				a.push(rax);
+				a.pushad64();
+				a.call_aligned(get_red_dot_brightness);
+				a.mov(qword_ptr(rsp, 0x80), rax);
+				a.popad64();
+				a.pop(rax);
+
+				a.mov(dword_ptr(r8, 0x1E84), eax);
+
+				a.jmp(0x1C4136_b);
+			});
+		}
 	}
 	
 	class component final : public component_interface
@@ -76,6 +155,19 @@ namespace renderer
 			dvars::override::register_enum("r_normalMap", game::DVAR_FLAG_SAVED);
 			dvars::override::register_enum("r_specularMap", game::DVAR_FLAG_SAVED);
 			dvars::override::register_enum("r_specOccMap", game::DVAR_FLAG_SAVED);
+
+			if (game::environment::is_mp())
+			{
+				// Adjust red dot brightness
+				utils::hook::jump(0x1C4125_b, get_tonemap_highlight_range_stub(), true);
+				db_load_xassets_hook.create(0x397500_b, db_load_xassets_stub);
+
+				r_red_dot_brightness_scale = dvars::register_float("r_redDotBrightnessScale", 
+					1.f, 0.1f, 5.f, game::DVAR_FLAG_SAVED, "Adjust red-dot reticle brightness");
+
+				r_use_custom_red_dot_brightness = dvars::register_bool("r_useCustomRedDotBrightness",
+					true, game::DVAR_FLAG_SAVED, "Use custom red-dot brightness values");
+			}
 		}
 	};
 }