fix Final Stage not appearing on StageInformation if a Long or Marathon song is played as the last song; add Song:GetStageCost()

2012-12-18 21:35:54 -06:00
parent bfba3a6fe8
commit a00ce73dd3
2 changed files with 7 additions and 1 deletions
@@ -133,6 +133,10 @@ function TextBannerAfterSet(self,param)
 	end
 end

+function Song:GetStageCost()
+	return self:IsMarathon() and 3 or self:IsLong() and 2 or 1
+end
+
 -- (c) 2005 Chris Danford
 -- All rights reserved.
 -- 
@@ -12,7 +12,9 @@ local tRemap = {
 	Stage_6th		= 6,
 };

-if tRemap[sStage] == PREFSMAN:GetPreference("SongsPerPlay") then
+local nSongCount = tRemap[sStage] + (GAMESTATE:GetCurrentSong():GetStageCost()-1);
+
+if nSongCount >= PREFSMAN:GetPreference("SongsPerPlay") then
 	sStage = "Stage_Final";
 else
 	sStage = sStage;