screen mines out of offsetvector when grabbing from pss

etternagame · Nov 25, 2018 · a19738c · a19738c
1 parent 8c431f3
commit a19738c
Showing 1 changed file with 6 additions and 3 deletions.
diff --git a/src/PlayerStageStats.cpp b/src/PlayerStageStats.cpp
@@ -1037,9 +1037,12 @@ LuaFunction(GetGradeFromPercent, GetGradeFromPercent(FArg(1)))
 	// Convert to MS so lua doesn't have to
 	static int GetOffsetVector(T* p, lua_State* L)
 	{
-		vector<float> doot = p->m_vOffsetVector;
-		for (size_t i = 0; i < doot.size(); ++i)
-			doot[i] = doot[i] * 1000;
+		auto& offs = p->m_vOffsetVector;
+		auto& type = p->m_vTapNoteTypeVector;
+		vector<float> doot;
+		for (size_t i = 0; i < offs.size(); ++i)
+			if (type[i] != TapNoteType_Mine)
+				doot.emplace_back(offs[i] * 1000);
 		LuaHelpers::CreateTableFromArray(doot, L);
 		return 1;
 	}