Stage mods.

stepmania/Themes/default/Scripts/StageMods.lua

@@ -1,31 +1,34 @@
-function SetupStageMods()
+function AreStageModsForced()
-	if GAMESTATE:IsAnExtraStage() then
+	if GAMESTATE:IsAnExtraStage() then return true end
-		if GAMESTATE:GetPreferredSongGroup() == "---Group All---" and
+	return false
-		   not PREFSMAN:GetPreference("PickExtraStage") then
+end
-			local song = GAMESTATE:GetCurrentSong()
-			GAMESTATE:SetPreferredSongGroup( song:GetGroupName() )
-		end
 
-		local bExtra2 = GAMESTATE:IsExtraStage2()
+function ScreenSelectMusic:setupstagemods()
-		local style = GAMESTATE:GetCurrentStyle()
+	if not GAMESTATE:IsAnExtraStage() then return end
-		local song, steps, po, so = SONGMAN:GetExtraStageInfo( bExtra2, style )
+	if GAMESTATE:GetPreferredSongGroup() == "---Group All---" and
-		local difficulty = steps:GetDifficulty()
+	   not PREFSMAN:GetPreference("PickExtraStage") then
-		
+		local song = GAMESTATE:GetCurrentSong()
-		GAMESTATE:SetCurrentSong( song )
+		GAMESTATE:SetPreferredSongGroup( song:GetGroupName() )
-		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
 	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

stepmania/Themes/default/metrics.ini

@@ -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