From 2f8e6214398ed4470e4cfad1af89d7262e7a5d1f Mon Sep 17 00:00:00 2001
From: Glenn Maynard <gmaynard@users.sourceforge.net>
Date: Tue, 24 Apr 2007 00:25:54 +0000
Subject: [PATCH] store stage in StageStats

---
 stepmania/src/ScreenGameplay.cpp | 1 +
 stepmania/src/StageStats.cpp     | 4 ++++
 stepmania/src/StageStats.h       | 1 +
 3 files changed, 6 insertions(+)

diff --git a/stepmania/src/ScreenGameplay.cpp b/stepmania/src/ScreenGameplay.cpp
index 6402fa5143..efb7ddbe57 100644
--- a/stepmania/src/ScreenGameplay.cpp
+++ b/stepmania/src/ScreenGameplay.cpp
@@ -404,6 +404,7 @@ void ScreenGameplay::Init()
 		FOREACH_EnabledPlayer(p)
 			PROFILEMAN->IncrementCoursePlayCount( GAMESTATE->m_pCurCourse, GAMESTATE->m_pCurTrail[p], p );
 
+	STATSMAN->m_CurStageStats.m_Stage = GAMESTATE->GetCurrentStage();
 	STATSMAN->m_CurStageStats.m_playMode = GAMESTATE->m_PlayMode;
 	STATSMAN->m_CurStageStats.m_pStyle = GAMESTATE->GetCurrentStyle();
 
diff --git a/stepmania/src/StageStats.cpp b/stepmania/src/StageStats.cpp
index dc1c763997..716f4a67f3 100644
--- a/stepmania/src/StageStats.cpp
+++ b/stepmania/src/StageStats.cpp
@@ -19,6 +19,7 @@
 StageStats::StageStats()
 {
 	m_playMode = PlayMode_Invalid;
+	m_Stage = Stage_Invalid;
 	m_pStyle = NULL;
 	m_vpPlayedSongs.clear();
 	m_vpPossibleSongs.clear();
@@ -90,6 +91,7 @@ void StageStats::AddStats( const StageStats& other )
 		m_vpPlayedSongs.push_back( *s );
 	FOREACH_CONST( Song*, other.m_vpPossibleSongs, s )
 		m_vpPossibleSongs.push_back( *s );
+	m_Stage = Stage_Invalid; // meaningless
 	m_StageType = StageType_Invalid; // meaningless
 
 	m_bGaveUp |= other.m_bGaveUp;
@@ -341,6 +343,7 @@ public:
 	static int GetGameplaySeconds( T* p, lua_State *L )		{ lua_pushnumber(L, p->m_fGameplaySeconds); return 1; }
 	static int OnePassed( T* p, lua_State *L )			{ lua_pushboolean(L, p->OnePassed()); return 1; }
 	static int AllFailed( T* p, lua_State *L )			{ lua_pushboolean(L, p->AllFailed()); return 1; }
+	static int GetStage( T* p, lua_State *L )			{ LuaHelpers::Push( L, p->m_Stage ); return 1; }
 	static int PlayerHasHighScore( T* p, lua_State *L )
 	{
 		lua_pushboolean(L, p->PlayerHasHighScore(Enum::Check<PlayerNumber>(L, 1)));
@@ -356,6 +359,7 @@ public:
 		ADD_METHOD( GetGameplaySeconds );
 		ADD_METHOD( OnePassed );
 		ADD_METHOD( AllFailed );
+		ADD_METHOD( GetStage );
 		ADD_METHOD( PlayerHasHighScore );
 	}
 };
diff --git a/stepmania/src/StageStats.h b/stepmania/src/StageStats.h
index 5c6696a109..f4f8b97abb 100644
--- a/stepmania/src/StageStats.h
+++ b/stepmania/src/StageStats.h
@@ -26,6 +26,7 @@ public:
 
 	int		GetAverageMeter( PlayerNumber pn ) const;
 
+	Stage		m_Stage;
 	PlayMode	m_playMode;
 	const Style*	m_pStyle;
 	vector<Song*>	m_vpPlayedSongs;