From 81e16cf7a698ab15366c41c650c82d58641fb4ed Mon Sep 17 00:00:00 2001
From: Glenn Maynard <gmaynard@users.sourceforge.net>
Date: Sun, 20 Jul 2003 09:15:46 +0000
Subject: [PATCH] Don't show course modes after the first stage.

---
 stepmania/src/MusicWheel.cpp | 12 +++++++++++-
 1 file changed, 11 insertions(+), 1 deletion(-)

diff --git a/stepmania/src/MusicWheel.cpp b/stepmania/src/MusicWheel.cpp
index d5dbe343f9..d91608d00b 100644
--- a/stepmania/src/MusicWheel.cpp
+++ b/stepmania/src/MusicWheel.cpp
@@ -400,6 +400,17 @@ void MusicWheel::BuildWheelItemDatas( vector<WheelItemData> &arrayWheelItemDatas
 			wid.m_sLabel = Names[i];
 			wid.m_Action.FromString( Actions[i], true );
 
+			switch( so )
+			{
+			case SORT_ALL_COURSES:
+			case SORT_NONSTOP_COURSES:
+			case SORT_ONI_COURSES:
+			case SORT_ENDLESS_COURSES:
+				/* Don't display course modes after the first stage. */
+				if( !PREFSMAN->m_bEventMode && GAMESTATE->m_iCurrentStageIndex )
+					continue;
+			}
+
 			arrayWheelItemDatas.push_back( wid );
 		}		
 		break;
@@ -565,7 +576,6 @@ void MusicWheel::BuildWheelItemDatas( vector<WheelItemData> &arrayWheelItemDatas
 		case SORT_ALL_COURSES:		SONGMAN->GetAllCourses( apCourses, PREFSMAN->m_bAutogenGroupCourses );	break;
 		default: ASSERT(0); break;
 		}
-		LOG->Trace("generate %i", apCourses.size());
 
 		SortCoursePointerArrayByDifficulty( apCourses );