From b960e51f4276e1fe5794546785a2028af633dda2 Mon Sep 17 00:00:00 2001
From: Glenn Maynard <gmaynard@users.sourceforge.net>
Date: Mon, 21 Feb 2005 06:59:50 +0000
Subject: [PATCH] Profile::IsCodeUnlocked

---
 stepmania/src/Profile.cpp | 7 +++++++
 stepmania/src/Profile.h   | 1 +
 2 files changed, 8 insertions(+)

diff --git a/stepmania/src/Profile.cpp b/stepmania/src/Profile.cpp
index 628715f9ab..522173c2bd 100644
--- a/stepmania/src/Profile.cpp
+++ b/stepmania/src/Profile.cpp
@@ -435,6 +435,11 @@ void Profile::SetDefaultModifiers( const Game* pGameType, const CString &sModifi
 		m_sDefaultModifiers[pGameType->m_szName] = sModifiers;
 }
 
+bool Profile::IsCodeUnlocked( int iCode ) const
+{
+	return m_UnlockedSongs.find( iCode ) != m_UnlockedSongs.end();
+}
+
 //
 // Steps high scores
 //
@@ -1694,6 +1699,7 @@ public:
 	static int GetCaloriesBurnedToday( T* p, lua_State *L )	{ lua_pushnumber(L, p->GetCaloriesBurnedToday() ); return 1; }
 	static int GetSaved( T* p, lua_State *L )				{ p->m_SavedLuaData.PushSelf(L); return 1; }
 	static int GetTotalNumSongsPlayed( T* p, lua_State *L )	{ lua_pushnumber(L, p->GetTotalNumSongsPlayed() ); return 1; }
+	static int IsCodeUnlocked( T* p, lua_State *L )			{ lua_pushboolean(L, p->IsCodeUnlocked(IArg(1)) ); return 1; }
 	
 
 	static void Register(lua_State *L)
@@ -1709,6 +1715,7 @@ public:
 		ADD_METHOD( GetCaloriesBurnedToday )
 		ADD_METHOD( GetSaved )
 		ADD_METHOD( GetTotalNumSongsPlayed )
+		ADD_METHOD( IsCodeUnlocked )
 		Luna<T>::Register( L );
 	}
 };
diff --git a/stepmania/src/Profile.h b/stepmania/src/Profile.h
index 2587ee4e0c..f6af3c3f2d 100644
--- a/stepmania/src/Profile.h
+++ b/stepmania/src/Profile.h
@@ -79,6 +79,7 @@ public:
 	int GetSongNumTimesPlayed( const SongID& songID ) const;
 	bool GetDefaultModifiers( const Game* pGameType, CString &sModifiersOut ) const;
 	void SetDefaultModifiers( const Game* pGameType, const CString &sModifiers );
+	bool IsCodeUnlocked( int iCode ) const;
 	
 	void AddStepTotals( int iNumTapsAndHolds, int iNumJumps, int iNumHolds, int iNumMines, int iNumHands, float fCaloriesBurned );