werwerwerwerwerwerwer's birthday

jameskr97 · Nov 25, 2018 · 584f2a8 · 584f2a8
1 parent f9a0e1f
commit 584f2a8
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/src/NoteField.cpp b/src/NoteField.cpp
@@ -361,6 +361,11 @@ NoteField::Update(float fDeltaTime)
 {
 	if (!this->GetVisible())
 		return;
+	if (!GAMESTATE->m_pCurSong) {
+		this->DeleteAllChildren();
+		return;
+	}
+
 	if (m_bFirstUpdate) {
 		m_pCurDisplay->m_ReceptorArrowRow.PlayCommand("On");
 	}
@@ -824,6 +829,11 @@ NoteField::CalcPixelsBeforeAndAfterTargets()
 void
 NoteField::DrawPrimitives()
 {
+	if (!GAMESTATE->m_pCurSong) {
+		this->DeleteAllChildren();
+		return;
+	}
+
 	// LOG->Trace( "NoteField::DrawPrimitives()" );
 
 	// This should be filled in on the first update.