From b79888bfa49acc5d92a2172943c13484ee52a1f4 Mon Sep 17 00:00:00 2001
From: TaroNuke <boomaloo1@gmail.com>
Date: Tue, 1 May 2012 15:15:21 +0000
Subject: [PATCH] Reload the wheel on late join.

---
 src/MusicWheel.cpp | 11 +++++++++--
 1 file changed, 9 insertions(+), 2 deletions(-)

diff --git a/src/MusicWheel.cpp b/src/MusicWheel.cpp
index 1b5224e21b..f27ecbb473 100644
--- a/src/MusicWheel.cpp
+++ b/src/MusicWheel.cpp
@@ -1511,13 +1511,20 @@ void MusicWheel::PlayerJoined()
 	// a song that has an illegal stepstype for the current amount of players.
 	// (e.g. a song that only has doubles difficulties and a second player joins.)
 	// We need to rebuild the wheel item data in this situation. -aj
+  // TaroNuke: Is at some point this going to remove the non selectable song/course choices?
 	if( !GAMESTATE->IsCourseMode() && !PREFSMAN->m_bAutogenSteps )
 	{
 		FOREACH_ENUM(SortOrder, so)
 		{
 			m_WheelItemDatasStatus[so] = INVALID;
-		}
-		RebuildWheelItems();
+    }
+    // TaroNuke: Refresh the screen - feel free to replace this with an actual reload screen function
+    SCREENMAN->SetNewScreen( "ScreenSelectMusic" );
+    // RebuildWheelItems();
+  }
+  else
+  {
+    SCREENMAN->SetNewScreen( "ScreenSelectCourse" );
 	}
 	SetOpenSection( m_sExpandedSectionName );
 }