Screen, ScreenWithMenuElements bindings

stepmania/src/Screen.cpp

@@ -255,6 +255,23 @@ void Screen::ClearMessageQueue( const ScreenMessage SM )
 			m_QueuedMessages.erase( m_QueuedMessages.begin()+i ); 
 }
 
+// lua start
+#include "LuaBinding.h"
+
+class LunaScreen: public Luna<Screen>
+{
+public:
+	LunaScreen() { LUA->Register( Register ); }
+
+	static void Register( Lua *L )
+	{
+		Luna<T>::Register( L );
+	}
+};
+
+LUA_REGISTER_DERIVED_CLASS( Screen, ActorFrame )
+// lua end
+
 /*
  * (c) 2001-2004 Chris Danford, Glenn Maynard
  * All rights reserved.

stepmania/src/Screen.h

@@ -53,6 +53,11 @@ public:
 
 	static bool JoinInput( const MenuInput &MenuI );	// return true if a player joined
 
+	//
+	// Lua
+	//
+	virtual void PushSelf( lua_State *L );
+
 protected:
 	// structure for holding messages sent to a Screen
 	struct QueuedScreenMessage {

stepmania/src/ScreenWithMenuElements.cpp

@@ -269,6 +269,23 @@ void ScreenWithMenuElements::UpdateStage()
 		m_sprStage[s]->SetHidden( s != GAMESTATE->GetCurrentStage() );
 }
 
+// lua start
+#include "LuaBinding.h"
+
+class LunaScreenWithMenuElements: public Luna<ScreenWithMenuElements>
+{
+public:
+	LunaScreenWithMenuElements() { LUA->Register( Register ); }
+
+	static void Register( Lua *L )
+	{
+		Luna<T>::Register( L );
+	}
+};
+
+LUA_REGISTER_DERIVED_CLASS( ScreenWithMenuElements, Screen )
+// lua end
+
 /*
  * (c) 2004 Chris Danford
  * All rights reserved.

stepmania/src/ScreenWithMenuElements.h

@@ -30,6 +30,11 @@ public:
 
 	void TweenOffScreen();
 
+	//
+	// Lua
+	//
+	virtual void PushSelf( lua_State *L );
+
 protected:
 	virtual void HandleMessage( const CString& sMessage );