Stage mods.

2006-08-18 01:27:14 +00:00
parent be1e448cb2
commit fa0bc95823
2 changed files with 31 additions and 26 deletions
@@ -1,31 +1,34 @@
-function SetupStageMods()
-	if GAMESTATE:IsAnExtraStage() then
-		if GAMESTATE:GetPreferredSongGroup() == "---Group All---" and
-		   not PREFSMAN:GetPreference("PickExtraStage") then
-			local song = GAMESTATE:GetCurrentSong()
-			GAMESTATE:SetPreferredSongGroup( song:GetGroupName() )
-		end
+function AreStageModsForced()
+	if GAMESTATE:IsAnExtraStage() then return true end
+	return false
+end

-		local bExtra2 = GAMESTATE:IsExtraStage2()
-		local style = GAMESTATE:GetCurrentStyle()
-		local song, steps, po, so = SONGMAN:GetExtraStageInfo( bExtra2, style )
-		local difficulty = steps:GetDifficulty()
-		
-		GAMESTATE:SetCurrentSong( song )
-		GAMESTATE:SetPreferredSong( song )
-
-		local fun = function( dummy, pn )
-			GAMESTATE:SetCurrentSteps( pn, steps )
-			GAMESTATE:GetPlayerState(pn):SetPlayerOptions( ModsLevel_Stage, po )
-			GAMESTATE:SetPreferredDifficulty( pn, difficulty )
-			MESSAGEMAN:Broadcast( "PlayerOptionsChangedP" .. (pn+1) )
-		end
-
-		table.foreach( GAMESTATE:GetHumanPlayers(), fun )
-		GAMESTATE:SetSongOptions( ModsLevel_Stage, so )
-		MESSAGEMAN:Broadcast( "SongOptionsChanged" )
-		return
+function ScreenSelectMusic:setupstagemods()
+	if not GAMESTATE:IsAnExtraStage() then return end
+	if GAMESTATE:GetPreferredSongGroup() == "---Group All---" and
+	   not PREFSMAN:GetPreference("PickExtraStage") then
+		local song = GAMESTATE:GetCurrentSong()
+		GAMESTATE:SetPreferredSongGroup( song:GetGroupName() )
 	end
+
+	local bExtra2 = GAMESTATE:IsExtraStage2()
+	local style = GAMESTATE:GetCurrentStyle()
+	local song, steps, po, so = SONGMAN:GetExtraStageInfo( bExtra2, style )
+	local difficulty = steps:GetDifficulty()
+	
+	GAMESTATE:SetCurrentSong( song )
+	GAMESTATE:SetPreferredSong( song )
+
+	local fun = function( dummy, pn )
+		GAMESTATE:SetCurrentSteps( pn, steps )
+		GAMESTATE:GetPlayerState(pn):SetPlayerOptions( ModsLevel_Stage, po )
+		GAMESTATE:SetPreferredDifficulty( pn, difficulty )
+		MESSAGEMAN:Broadcast( "PlayerOptionsChangedP" .. (pn+1) )
+	end
+
+	table.foreach( GAMESTATE:GetHumanPlayers(), fun )
+	GAMESTATE:SetSongOptions( ModsLevel_Stage, so )
+	MESSAGEMAN:Broadcast( "SongOptionsChanged" )
 end
 -- 
 -- (c) 2006 Steve Checkoway
@@ -516,6 +516,7 @@ Codes=
 OptionsMenuAvailable=OptionsMenuAvailable()
 SelectMenuAvailable=false
 ModeMenuAvailable=ModeMenuAvailable()
+ScreenBeginCommand=setupstagemods

 [ScreenSelectCourse]
 Class=ScreenSelectMusic
@@ -4387,6 +4388,7 @@ Answer3Of3OffCommand=
 UseNameBlacklist=0
 DefaultSort=Group
 DefaultSong=
+AreStageModsForced=AreStageModsForced

 [ScreenMiniMenuContext]
 Fallback=ScreenMiniMenu