From e6deb22af75252f0a8b2602bb9b39509cc24e80c Mon Sep 17 00:00:00 2001
From: Andrew Wong <curewater@users.sourceforge.net>
Date: Sun, 27 Jul 2003 14:54:56 +0000
Subject: [PATCH] course coloring and minor changes

---
 stepmania/src/Course.cpp | 62 +++++++++++++++++++++++++++++++++++-----
 1 file changed, 55 insertions(+), 7 deletions(-)

diff --git a/stepmania/src/Course.cpp b/stepmania/src/Course.cpp
index da452842e2..334a21943e 100644
--- a/stepmania/src/Course.cpp
+++ b/stepmania/src/Course.cpp
@@ -582,13 +582,51 @@ void Course::GetCourseInfo( NotesType nt, vector<Course::Info> &ci ) const
 
 RageColor Course::GetColor() const
 {
-	// This could be made smarter
-	if( m_entries.size() >= 7 )
-		return RageColor(1,0,0,1);	// red
-	else if( m_entries.size() >= 4 )
-		return RageColor(1,0.5f,0,1);	// orange
-	else
-		return RageColor(0,1,0,1);	// green
+	switch (PREFSMAN->m_iCourseSortOrder)
+	{
+	case 0:	
+		if( m_entries.size() >= 7 )
+			return RageColor(1,0,0,1);	// red
+		else if( m_entries.size() >= 4 )
+			return RageColor(1,1,0,1);	// yellow
+		else
+			return RageColor(0,1,0,1);	// green
+		// never should get here
+		break;
+
+	case 1:
+		if (SortOrder_AvgDifficulty > 100)
+			return RageColor(0,0,1,1);  // blue
+		if (SortOrder_AvgDifficulty > 8.5)
+			return RageColor(1,0,0,1);  // red
+		if (SortOrder_AvgDifficulty >= 7)
+			return RageColor(1,0.5f,0,1); // orange
+		if (SortOrder_AvgDifficulty >= 5)
+			return RageColor(1,1,0,1);  // yellow
+		return RageColor(0,1,0,1); // green
+
+	case 2:
+		if (SortOrder_TotalDifficulty > 100000)
+			return RageColor(0,0,1,1);  // blue
+		if (SortOrder_TotalDifficulty >= 40)
+			return RageColor(1,0,0,1);  // red
+		if (SortOrder_TotalDifficulty >= 30)
+			return RageColor(1,0.5f,0,1); // orange
+		if (SortOrder_TotalDifficulty >= 20)
+			return RageColor(1,1,0,1);  // yellow
+		return RageColor(0,1,0,1); // green
+
+	case 3:
+		if (SortOrder_Ranking == 3)
+			return RageColor(0,0,1,1);  // blue
+		if (SortOrder_Ranking == 2)
+			return RageColor(1,0.5f,0,1); // orange
+		if (SortOrder_Ranking == 1)
+			return RageColor(0,1,0,1); // green
+		return RageColor(1,1,0,1); // yellow, never should get here
+	}
+
+	return RageColor(1,1,1,1);  // white, never should reach here
 }
 
 
@@ -853,6 +891,16 @@ void Course::UpdateCourseStats()
 
 	for(i = 0; i < m_entries.size(); i++)
 	{
+		// courses with random/players best-worst songs should go at the end
+		if (m_entries[i].type != Entry::fixed)
+		{
+			SortOrder_AvgDifficulty = 9999999; // large number
+			SortOrder_NumStages = ci.size();
+			SortOrder_Ranking = 3;
+			SortOrder_TotalDifficulty = 999999;     // large number
+			return;
+		}
+
 		SortOrder_NumStages++;
 		SortOrder_TotalDifficulty += ci[i].pNotes->GetMeter();
 	}