Merge pull request #1619 from kyzentun/5_0_course_difficulty_crash_fix

SSM needs to prune playable trails.
2018-01-24 15:30:49 -06:00
parent 0680ab5e8e 9210ee4642
commit c526fa6789
1 changed files with 15 additions and 0 deletions
@@ -1885,6 +1885,21 @@ void ScreenSelectMusic::AfterMusicChange()
 			if(pStyle == NULL)
 			{
 				lCourse->GetAllTrails(m_vpTrails);
+				vector<Trail*>::iterator tra= m_vpTrails.begin();
+				Game const* cur_game= GAMESTATE->GetCurrentGame();
+				int num_players= GAMESTATE->GetNumPlayersEnabled();
+				while(tra != m_vpTrails.end())
+				{
+					if(GAMEMAN->GetFirstCompatibleStyle(cur_game, num_players,
+							(*tra)->m_StepsType) == NULL)
+					{
+						tra= m_vpTrails.erase(tra);
+					}
+					else
+					{
+						++tra;
+					}
+				}
 			}
 			else
 			{