diff --git a/src/GameState.cpp b/src/GameState.cpp index d9caa4f9b6..22cea682f3 100644 --- a/src/GameState.cpp +++ b/src/GameState.cpp @@ -1459,6 +1459,12 @@ void GameState::RemoveAllActiveAttacks() // called on end of song m_pPlayerState[p]->RemoveActiveAttacks(); } +void GameState::AddStageToPlayer( PlayerNumber pn ) +{ + // Add one stage more to player (bonus) -cerbo + ++m_iPlayerStageTokens[pn]; +} + template void setmin( T &a, const T &b ) { @@ -2430,6 +2436,7 @@ public: lua_pushboolean(L, p->m_bDopefish); return 1; } + static int AddStageToPlayer( T* p, lua_State *L ) { p->AddStageToPlayer(Enum::Check(L, 1)); return 0; } LunaGameState() { @@ -2531,6 +2538,7 @@ public: ADD_METHOD( SetCharacter ); ADD_METHOD( GetExpandedSectionName ); ADD_METHOD( Dopefish ); + ADD_METHOD( AddStageToPlayer ); } }; diff --git a/src/GameState.h b/src/GameState.h index c78f4ce7f2..fadb709559 100644 --- a/src/GameState.h +++ b/src/GameState.h @@ -62,6 +62,7 @@ public: bool HaveProfileToLoad(); bool HaveProfileToSave(); void SaveLocalData(); + void AddStageToPlayer( PlayerNumber pn ); void LoadCurrentSettingsFromProfile( PlayerNumber pn ); /** * @brief Save the specified player's settings to his/her profile.