Merge pull request #1563 from kyzentun/music_wheel_handle_message_crash

Fix extra stage MusicWheelItem crash
2017-11-21 22:43:22 -08:00
parent fe404b2950 87dfd25cfc
commit 737ab1bbf9
1 changed files with 2 additions and 0 deletions
@@ -305,6 +305,7 @@ void MusicWheelItem::LoadFromWheelItemData( const WheelItemBaseData *pData, int

 void MusicWheelItem::RefreshGrades()
 {
+	if(!IsLoaded()) { return; }
 	const MusicWheelItemData *pWID = dynamic_cast<const MusicWheelItemData*>( m_pData );

 	if( pWID == NULL )
@@ -375,6 +376,7 @@ void MusicWheelItem::RefreshGrades()

 void MusicWheelItem::HandleMessage( const Message &msg )
 {
+	if(!IsLoaded()) { return; }
 	if( msg == Message_CurrentStepsP1Changed ||
 	    msg == Message_CurrentStepsP2Changed ||
 	    msg == Message_CurrentTrailP1Changed ||