From a2f96a215c995c3f068aba93fa16e5ae36cda799 Mon Sep 17 00:00:00 2001
From: Glenn Maynard <gmaynard@users.sourceforge.net>
Date: Tue, 4 Feb 2003 21:44:58 +0000
Subject: [PATCH] handle no extra stage in event mode in HasEarnedExtraStage
 challenge is good enough for an extra stage

---
 stepmania/src/GameState.cpp | 9 ++++++++-
 1 file changed, 8 insertions(+), 1 deletion(-)

diff --git a/stepmania/src/GameState.cpp b/stepmania/src/GameState.cpp
index e29d11ab50..a2fb9c5810 100644
--- a/stepmania/src/GameState.cpp
+++ b/stepmania/src/GameState.cpp
@@ -202,6 +202,9 @@ bool GameState::IsPlayerEnabled( PlayerNumber pn )
 
 bool GameState::HasEarnedExtraStage()
 {
+	if( PREFSMAN->m_bEventMode )
+		return false;
+
 	if( (GAMESTATE->IsFinalStage() || GAMESTATE->IsExtraStage()) )
 	{
 		for( int p=0; p<NUM_PLAYERS; p++ )
@@ -209,7 +212,11 @@ bool GameState::HasEarnedExtraStage()
 			if( !GAMESTATE->IsPlayerEnabled(p) )
 				continue;	// skip
 
-			if( GAMESTATE->m_pCurNotes[p]->GetDifficulty()==DIFFICULTY_HARD && m_CurStageStats.GetGrade((PlayerNumber)p)==GRADE_AA )
+			if( GAMESTATE->m_pCurNotes[p]->GetDifficulty() != DIFFICULTY_HARD && 
+				GAMESTATE->m_pCurNotes[p]->GetDifficulty() != DIFFICULTY_CHALLENGE )
+				continue; /* not hard enough! */
+
+			if( m_CurStageStats.GetGrade((PlayerNumber)p) >= GRADE_AA )
 				return true;
 		}
 	}