Added CheatHUD

4 files changed, 77 insertions, 3 deletions

diff --git a/src/client/game.cpp b/src/client/game.cpp
index 491d55a34..479484ae9 100644
--- a/src/client/game.cpp
+++ b/src/client/game.cpp
@@ -3206,9 +3206,12 @@ void Game::updateFrame(ProfilerGraph *graph, RunStats *stats, f32 dtime,
 		Cheat menu
 	*/
 
-	if (m_game_ui->m_flags.show_cheat_menu && ! gui_chat_console->isOpen())
-		m_cheat_menu->draw(driver, m_game_ui->m_flags.show_debug);
-		
+	if (! gui_chat_console->isOpen()) {
+		if (m_game_ui->m_flags.show_cheat_menu)
+			m_cheat_menu->draw(driver, m_game_ui->m_flags.show_debug);
+		if (g_settings->getBool("cheat_hud"))
+			m_cheat_menu->drawHUD(driver, dtime);
+	}
 	/*
 		Damage flash
 	*/
diff --git a/src/defaultsettings.cpp b/src/defaultsettings.cpp
index bcdd80074..9e32a11fe 100644
--- a/src/defaultsettings.cpp
+++ b/src/defaultsettings.cpp
@@ -119,6 +119,7 @@ void set_default_settings(Settings *settings)
 	settings->setDefault("chat_reverse", "false");
 	settings->setDefault("forcefield", "false");
 	settings->setDefault("friendlist", "");
+	settings->setDefault("cheat_hud", "true");
 
 	// Keymap
 	settings->setDefault("remote_port", "30000");
diff --git a/src/gui/cheatMenu.cpp b/src/gui/cheatMenu.cpp
index f796fb0ba..1485541c2 100644
--- a/src/gui/cheatMenu.cpp
+++ b/src/gui/cheatMenu.cpp
@@ -94,6 +94,72 @@ void CheatMenu::draw(video::IVideoDriver *driver, bool show_debug)
 	}
 }
 
+void CheatMenu::drawHUD(video::IVideoDriver *driver, double dtime)
+{
+	CHEAT_MENU_GET_SCRIPTPTR
+	
+	m_rainbow_offset += dtime;
+
+	m_rainbow_offset = fmod(m_rainbow_offset, 6.0f);
+	
+	std::vector<std::string> enabled_cheats;
+	
+	int cheat_count = 0;
+	
+	for (auto category = script->m_cheat_categories.begin(); category != script->m_cheat_categories.end(); category++) {
+		for (auto cheat = (*category)->m_cheats.begin(); cheat != (*category)->m_cheats.end(); cheat++) {
+			if ((*cheat)->is_enabled()) {
+				enabled_cheats.push_back((*cheat)->m_name);
+				cheat_count++;
+			}
+		}
+	}
+	
+	if (enabled_cheats.empty())
+		return;
+	
+	std::vector<video::SColor> colors;
+	
+	for (int i = 0; i < cheat_count; i++) {
+		video::SColor color;
+		f32 h = (f32)i * 2.0f / (f32)cheat_count - m_rainbow_offset;
+		if (h < 0)
+			h = 6.0f + h;
+		f32 x = (1 - fabs(fmod(h, 2.0f) - 1.0f)) * 255.0f;
+		switch((int)h) {
+		case 0:
+			color = video::SColor(255, 255, x, 0); break;
+		case 1:
+			color = video::SColor(255, x, 255, 0); break;
+		case 2:
+			color = video::SColor(255, 0, 255, x); break;
+		case 3:
+			color = video::SColor(255, 0, x, 255); break;
+		case 4:
+			color = video::SColor(255, x, 0, 255); break;
+		case 5:
+			color = video::SColor(255, 255, 0, x); break;
+		}
+		colors.push_back(color);
+	}
+	
+	core::dimension2d<u32> screensize = driver->getScreenSize();
+	
+	u32 y = 5;
+	
+	int i = 0;
+	for (std::string cheat : enabled_cheats) {
+		core::dimension2d<u32> dim = m_font->getDimension(utf8_to_wide(cheat).c_str());
+		u32 x = screensize.Width - 5 - dim.Width;
+		
+		core::rect<s32> fontbounds(x, y, x + dim.Width, y + dim.Height);
+		m_font->draw(cheat.c_str(), fontbounds, colors[i], false, false);
+		
+		y += dim.Height;
+		i++;
+	}
+}
+
 void CheatMenu::selectUp()
 {
 	CHEAT_MENU_GET_SCRIPTPTR
diff --git a/src/gui/cheatMenu.h b/src/gui/cheatMenu.h
index ea9a9d853..f67cdea5a 100644
--- a/src/gui/cheatMenu.h
+++ b/src/gui/cheatMenu.h
@@ -42,6 +42,8 @@ public:
 	CheatMenu(Client *client);
 
 	void draw(video::IVideoDriver *driver, bool show_debug);
+	
+	void drawHUD(video::IVideoDriver *driver, double dtime);
 
 	void drawEntry(video::IVideoDriver *driver, std::string name, int number,
 			bool selected, bool active,
@@ -72,4 +74,6 @@ private:
 
 	gui::IGUIFont *m_font = nullptr;
 	v2u32 m_fontsize;
+	
+	float m_rainbow_offset = 0.0;
 };