Allow access to Wifescore via ReplaySnapshots from Evaluation

poco0317 · Nov 10, 2019 · 92eb552 · 92eb552
1 parent cdc0abb
commit 92eb552
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/src/Etterna/Screen/Others/ScreenEvaluation.cpp b/src/Etterna/Screen/Others/ScreenEvaluation.cpp
@@ -389,12 +389,21 @@ class LunaScreenEvaluation : public Luna<ScreenEvaluation>
 		LuaHelpers::CreateTableFromArray(toPush, L);
 		return 1;
 	}
+	static int GetReplaySnapshotWifePercentForNoterow(T* p, lua_State* L)
+	{
+		int row = IArg(1);
+		auto rs = PlayerAI::GetReplaySnapshotForNoterow(row);
+
+		lua_pushnumber(L, rs->curwifescore / rs->maxwifescore);
+		return 1;
+	}
 
 	LunaScreenEvaluation()
 	{
 		ADD_METHOD(GetStageStats);
 		ADD_METHOD(SetPlayerStageStatsFromReplayData);
 		ADD_METHOD(GetReplaySnapshotJudgmentsForNoterow);
+		ADD_METHOD(GetReplaySnapshotWifePercentForNoterow);
 	}
 };