diff --git a/stepmania/src/Difficulty.cpp b/stepmania/src/Difficulty.cpp index 33ae642ea3..535fd1510c 100644 --- a/stepmania/src/Difficulty.cpp +++ b/stepmania/src/Difficulty.cpp @@ -2,6 +2,7 @@ #include "Difficulty.h" #include "GameState.h" #include "ThemeMetric.h" +#include "LuaManager.h" static const CString DifficultyNames[NUM_DIFFICULTIES] = { @@ -42,6 +43,19 @@ Difficulty StringToDifficulty( const CString& sDC ) else return DIFFICULTY_INVALID; } +static void LuaDifficulty(lua_State* L) +{ + FOREACH_Difficulty( d ) + { + CString s = DifficultyNames[d]; + s.MakeUpper(); + LUA->SetGlobal( "DIFFICULTY_"+s, d ); + } + LUA->SetGlobal( "NUM_DIFFICULTIES", NUM_DIFFICULTIES ); +} +REGISTER_WITH_LUA_FUNCTION( LuaDifficulty ); + + static const CString CourseDifficultyNames[NUM_DIFFICULTIES] = {