Draw gravity zones when using grav wall

1 files changed, 20 insertions, 2 deletions

diff --git a/src/game/QuickOptions.h b/src/game/QuickOptions.h
index f194c1c..b22834c 100644
--- a/src/game/QuickOptions.h
+++ b/src/game/QuickOptions.h
@@ -29,11 +29,11 @@ public:
 	}
 	virtual bool GetToggle() 
 	{
-		return m->GetRenderer()->gravifyFieldEnabled;
+		return m->GetRenderer()->gravityFieldEnabled;
 	}
 	virtual void perform()
 	{
-		m->GetRenderer()->gravifyFieldEnabled = !m->GetRenderer()->gravifyFieldEnabled;
+		m->GetRenderer()->gravityFieldEnabled = !m->GetRenderer()->gravityFieldEnabled;
 	}
 };
 
@@ -96,4 +96,22 @@ public:
 	{
 		m->GetSimulation()->aheat_enable = !m->GetSimulation()->aheat_enable;
 	}
+};
+
+class DrawGravWallOption: public QuickOption
+{
+public:
+	DrawGravWallOption(GameModel * m):
+	QuickOption("Z", "Draw gravity zones", m, Toggle)
+	{
+
+	}
+	virtual bool GetToggle() 
+	{
+		return m->GetRenderer()->gravityZonesEnabled;
+	}
+	virtual void perform()
+	{
+		m->GetRenderer()->gravityZonesEnabled = !m->GetRenderer()->gravityZonesEnabled;
+	}
 };
\ No newline at end of file