From e0c8a8ebc7a5266890b6c513d8895b193055030f Mon Sep 17 00:00:00 2001
From: Glenn Maynard <gmaynard@users.sourceforge.net>
Date: Thu, 16 Mar 2006 07:35:47 +0000
Subject: [PATCH] fix HandleSongTimer not reset

---
 stepmania/src/ScreenEdit.cpp | 14 +++++++++++---
 stepmania/src/ScreenEdit.h   |  3 +++
 2 files changed, 14 insertions(+), 3 deletions(-)

diff --git a/stepmania/src/ScreenEdit.cpp b/stepmania/src/ScreenEdit.cpp
index ce58bfba04..28668c0424 100644
--- a/stepmania/src/ScreenEdit.cpp
+++ b/stepmania/src/ScreenEdit.cpp
@@ -819,6 +819,17 @@ ScreenEdit::~ScreenEdit()
 	m_soundMusic.StopPlaying();
 }
 
+void ScreenEdit::BeginScreen()
+{
+	/* We do this ourself. */
+	SOUND->HandleSongTimer( false );
+}
+
+void ScreenEdit::EndScreen()
+{
+	SOUND->HandleSongTimer( true );
+}
+
 // play assist ticks
 void ScreenEdit::PlayTicks()
 {
@@ -2581,9 +2592,6 @@ void ScreenEdit::HandleScreenMessage( const ScreenMessage SM )
 	}
 	else if( SM == SM_GainFocus )
 	{
-		/* We do this ourself. */
-		SOUND->HandleSongTimer( false );
-
 		/* When another screen comes up, RageSounds takes over the sound timer.  When we come
 		 * back, put the timer back to where it was. */
 		GAMESTATE->m_fSongBeat = m_fTrailingBeat;
diff --git a/stepmania/src/ScreenEdit.h b/stepmania/src/ScreenEdit.h
index f88605cdc7..aeda726a27 100644
--- a/stepmania/src/ScreenEdit.h
+++ b/stepmania/src/ScreenEdit.h
@@ -165,6 +165,9 @@ class ScreenEdit : public ScreenWithMenuElements
 {
 public:
 	virtual void Init();
+	virtual void BeginScreen();
+	virtual void EndScreen();
+
 	virtual ~ScreenEdit();
 	virtual void Update( float fDeltaTime );
 	virtual void DrawPrimitives();