allow lua access to get song/steps pointers by chartkey

etternagame · Apr 28, 2017 · d0e0c2f · d0e0c2f
1 parent 7877fe3
commit d0e0c2f
Showing 1 changed file with 14 additions and 0 deletions.
diff --git a/src/SongManager.cpp b/src/SongManager.cpp
@@ -2291,6 +2291,18 @@ class LunaSongManager: public Luna<SongManager>
 		lua_pushboolean(L, p->WasLoadedFromAdditionalCourses(pCourse));
 		return 1;
 	}
+	static int GetSongByChartKey(T* p, lua_State *L)
+	{
+		Song* pSong = p->GetSongByChartkey(SArg(1));
+		pSong->PushSelf(L);
+		return 1;
+	}
+	static int GetStepsByChartKey(T* p, lua_State *L)
+	{
+		Steps* pSteps = p->GetStepsByChartkey(SArg(1));
+		pSteps->PushSelf(L);
+		return 1;
+	}
 
 	LunaSongManager()
 	{
@@ -2333,6 +2345,8 @@ class LunaSongManager: public Luna<SongManager>
 		ADD_METHOD( SongToPreferredSortSectionName );
 		ADD_METHOD( WasLoadedFromAdditionalSongs );
 		ADD_METHOD( WasLoadedFromAdditionalCourses );
+		ADD_METHOD( GetSongByChartKey );
+		ADD_METHOD( GetStepsByChartKey );
 	}
 };