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 &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 &arrayWheelItemDatas case SORT_ALL_COURSES: SONGMAN->GetAllCourses( apCourses, PREFSMAN->m_bAutogenGroupCourses ); break; default: ASSERT(0); break; } - LOG->Trace("generate %i", apCourses.size()); SortCoursePointerArrayByDifficulty( apCourses );