diff --git a/stepmania/src/MusicWheel.cpp b/stepmania/src/MusicWheel.cpp index f3eb8fde67..e98453c518 100644 --- a/stepmania/src/MusicWheel.cpp +++ b/stepmania/src/MusicWheel.cpp @@ -696,7 +696,11 @@ void MusicWheel::BuildWheelItemDatas( vector &arrayWheelIt if( WID.m_pSong != NULL ) { WID.m_Flags.bHasBeginnerOr1Meter = WID.m_pSong->IsEasy( GAMESTATE->GetCurrentStyle()->m_StepsType ) && SHOW_EASY_FLAG; - WID.m_Flags.bEdits = WID.m_pSong->HasEdits( GAMESTATE->GetCurrentStyle()->m_StepsType ); + WID.m_Flags.bEdits = false; + set vStepsType; + SongUtil::GetPlayableStepsTypes( WID.m_pSong, vStepsType ); + FOREACHS( StepsType, vStepsType, st ) + WID.m_Flags.bEdits |= WID.m_pSong->HasEdits( *st ); WID.m_Flags.iStagesForSong = GameState::GetNumStagesMultiplierForSong( WID.m_pSong ); } else if( WID.m_pCourse != NULL ) diff --git a/stepmania/src/SongUtil.cpp b/stepmania/src/SongUtil.cpp index 4fc7a0578d..b327a625c1 100644 --- a/stepmania/src/SongUtil.cpp +++ b/stepmania/src/SongUtil.cpp @@ -824,7 +824,7 @@ void SongUtil::FilterSongs( const SongCriteria &sc, const vector &in, vec } } -static void GetPlayableStepsTypes( const Song *pSong, set &vOut ) +void SongUtil::GetPlayableStepsTypes( const Song *pSong, set &vOut ) { vector vpPossibleStyles; if( CommonMetrics::AUTO_SET_STYLE ) diff --git a/stepmania/src/SongUtil.h b/stepmania/src/SongUtil.h index a0664af993..b08a36caf4 100644 --- a/stepmania/src/SongUtil.h +++ b/stepmania/src/SongUtil.h @@ -6,6 +6,7 @@ #include "GameConstantsAndTypes.h" #include "Difficulty.h" #include "RageUtil_CachedObject.h" +#include class Song; class Steps; @@ -112,6 +113,7 @@ namespace SongUtil void GetAllSongGenres( vector &vsOut ); void FilterSongs( const SongCriteria &sc, const vector &in, vector &out ); + void GetPlayableStepsTypes( const Song *pSong, set &vOut ); void GetPlayableSteps( const Song *pSong, vector &vOut ); bool IsStepsTypePlayable( Song *pSong, StepsType st ); bool IsStepsPlayable( Song *pSong, Steps *pSteps );