fix: Don't do cgaz updates on modes that aren't defrag

momentum-mod · Oct 15, 2023 · 79fe93e · 79fe93e
1 parent 748cf08
commit 79fe93e
Showing 1 changed file with 21 additions and 12 deletions.
diff --git a/scripts/hud/cgaz.js b/scripts/hud/cgaz.js
@@ -121,17 +121,28 @@ class Cgaz {
 	static snapAccel = 0;
 	static bShouldUpdateStyles = false;
 
+	static updateHandle = null;
+
 	static onLoad() {
-		if (GameModeAPI.GetCurrentGameMode() !== GameMode.DEFRAG) return;
-
-		this.onAccelConfigChange();
-		this.onSnapConfigChange();
-		this.onPrimeConfigChange();
-		this.onProjectionChange();
-		this.onHudFovChange();
-		this.onSnapConfigChange();
-		this.onWindicatorConfigChange();
-		this.onCompassConfigChange();
+		if (GameModeAPI.GetCurrentGameMode() === GameMode.DEFRAG) {
+			this.updateHandle = $.RegisterEventHandler(
+				'ChaosHudProcessInput',
+				$.GetContextPanel(),
+				this.onUpdate.bind(this)
+			);
+
+			this.onAccelConfigChange();
+			this.onSnapConfigChange();
+			this.onPrimeConfigChange();
+			this.onProjectionChange();
+			this.onHudFovChange();
+			this.onSnapConfigChange();
+			this.onWindicatorConfigChange();
+			this.onCompassConfigChange();
+		} else if (this.updateHandle) {
+			$.UnregisterEventHandler('ChaosHudProcessInput', $.GetContextPanel(), this.updateHandle);
+			this.updateHandle = null;
+		}
 	}
 
 	static onProjectionChange() {
@@ -1333,8 +1344,6 @@ class Cgaz {
 	}
 
 	static {
-		$.RegisterEventHandler('ChaosHudProcessInput', $.GetContextPanel(), this.onUpdate.bind(this));
-
 		$.RegisterForUnhandledEvent('ChaosLevelInitPostEntity', this.onLoad.bind(this));
 		$.RegisterForUnhandledEvent('OnDefragHUDProjectionChange', this.onProjectionChange.bind(this));
 		$.RegisterForUnhandledEvent('OnDefragHUDFOVChange', this.onHudFovChange.bind(this));