fix HandleSongTimer not reset

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;

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();