improved menu ESC handling

1 files changed, 15 insertions, 0 deletions

diff --git a/src/client/entitymenu.h b/src/client/entitymenu.h
index adc31a9..db69867 100644
--- a/src/client/entitymenu.h
+++ b/src/client/entitymenu.h
@@ -22,6 +22,16 @@ public:
 	/// create a new menu
 	EntityMenu(ui::Widget *parent, const char * label = 0);
 	~EntityMenu();
+
+	/* ---- inspectors ----------------------------------------- */
+
+	/// entity the current menu was generated for
+	inline core::Entity *generated_entity() { return menu_generated_entity; }
+
+	/// the current generated menu
+	inline const std::string & generated_menu() const { return menu_generated_menu; }
+
+	/* ---- mutators ------------------------------------------- */
 	
 	/// generate a menu from menu descriptions
 	void generate(core::Entity *entity, const char *menulabel);
@@ -33,8 +43,13 @@ protected:
 	/// resize event
 	virtual void resize();
 
+	/// keypress event handler
+	virtual bool on_keypress(const int key, const unsigned int modifier);
+
 private:
 	ui::Container		*menu_container;
+	core::Entity 		*menu_generated_entity;
+	std::string		menu_generated_menu;
 };
 
 }