From e74ea267f991c077af44534fd42071074db8e896 Mon Sep 17 00:00:00 2001
From: Glenn Maynard <gmaynard@users.sourceforge.net>
Date: Sun, 29 Feb 2004 07:36:03 +0000
Subject: [PATCH] SongManager::GetSongColor: ignore edits

---
 stepmania/src/SongManager.cpp | 9 ++++++---
 1 file changed, 6 insertions(+), 3 deletions(-)

diff --git a/stepmania/src/SongManager.cpp b/stepmania/src/SongManager.cpp
index 3d641e7f5d..ee9c9fd15d 100644
--- a/stepmania/src/SongManager.cpp
+++ b/stepmania/src/SongManager.cpp
@@ -380,14 +380,17 @@ RageColor SongManager::GetSongColor( const Song* pSong )
 	 *
 	 * XXX: Ack.  This means this function can only be called when we have a style
 	 * set up, which is too restrictive.  How to handle this?
-	 *
-	 * XXX: Once we support edits, ignore them, too. */
+	 */
 //	const StepsType nt = GAMESTATE->GetCurrentStyleDef()->m_StepsType;
 	for( unsigned i=0; i<pSong->m_apNotes.size(); i++ )
 	{
 		const Steps* pNotes = pSong->m_apNotes[i];
-		if( pNotes->GetDifficulty() == DIFFICULTY_CHALLENGE )
+		switch( pNotes->GetDifficulty() )
+		{
+		case DIFFICULTY_CHALLENGE:
+		case DIFFICULTY_EDIT:
 			continue;
+		}
 
 //		if(pNotes->m_StepsType != nt)
 //			continue;