fix #104: Properly clean up before loading another project

Author: adazem009

src/projectloader.cpp

@@ -64,6 +64,40 @@ void ProjectLoader::setFileName(const QString &newFileName)
 
     m_fileName = newFileName;
 
+    // Stop the project
+    if (m_engine)
+        m_engine->stop();
+
+    // Delete old sprites
+    for (SpriteModel *sprite : m_sprites)
+        sprite->deleteLater();
+
+    m_sprites.clear();
+    emit spritesChanged();
+
+    // Delete old clones
+    for (SpriteModel *clone : m_clones)
+        deleteCloneObject(clone);
+
+    m_clones.clear();
+    emit clonesChanged();
+
+    // Delete old monitors
+    for (MonitorModel *monitor : m_monitors) {
+        emit monitorRemoved(monitor);
+        monitor->deleteLater();
+    }
+
+    m_monitors.clear();
+    emit monitorsChanged();
+
+    // Clear the engine
+    if (m_engine)
+        m_engine->clear();
+
+    m_engine = nullptr;
+    emit engineChanged();
+
     m_project.setScratchVersion(ScratchVersion::Scratch3);
     m_project.setFileName(m_fileName.toStdString());
     m_loadStatus = false;
@@ -198,12 +232,6 @@ void ProjectLoader::load()
     m_engineMutex.lock();
     m_engine = m_project.engine().get();
 
-    // Delete old sprites
-    for (SpriteModel *sprite : m_sprites)
-        sprite->deleteLater();
-
-    m_sprites.clear();
-
     if (!m_engine || m_stopLoading) {
         m_engineMutex.unlock();
         emit fileNameChanged();
@@ -313,13 +341,18 @@ void ProjectLoader::addClone(SpriteModel *model)
     emit clonesChanged();
 }
 
-void ProjectLoader::deleteClone(SpriteModel *model)
+void ProjectLoader::deleteCloneObject(SpriteModel *model)
 {
-    m_clones.removeAll(model);
     emit cloneDeleted(model);
     Q_ASSERT(model->renderedTarget());
     model->renderedTarget()->deinitClone();
     model->deleteLater();
+}
+
+void ProjectLoader::deleteClone(SpriteModel *model)
+{
+    m_clones.removeAll(model);
+    deleteCloneObject(model);
     emit clonesChanged();
 }
 

src/projectloader.h

@@ -120,6 +120,7 @@ class ProjectLoader : public QObject
         void initTimer();
         void redraw();
         void addClone(SpriteModel *model);
+        void deleteCloneObject(SpriteModel *model);
         void deleteClone(SpriteModel *model);
         void addMonitor(libscratchcpp::Monitor *monitor);
         void removeMonitor(libscratchcpp::Monitor *monitor, libscratchcpp::IMonitorHandler *iface);

test/projectloader/projectloader_test.cpp

@@ -29,6 +29,7 @@ class ProjectLoaderTest : public testing::Test
             QSignalSpy engineSpy(loader, &ProjectLoader::engineChanged);
             QSignalSpy stageSpy(loader, &ProjectLoader::stageChanged);
             QSignalSpy spritesSpy(loader, &ProjectLoader::spritesChanged);
+            QSignalSpy clonesSpy(loader, &ProjectLoader::clonesChanged);
             QSignalSpy monitorsSpy(loader, &ProjectLoader::monitorsChanged);
             QSignalSpy monitorAddedSpy(loader, &ProjectLoader::monitorAdded);
 
@@ -37,10 +38,11 @@ class ProjectLoaderTest : public testing::Test
             ASSERT_EQ(fileNameSpy.count(), 1);
             ASSERT_EQ(loadStatusSpy.count(), 1);
             ASSERT_TRUE(loadingFinishedSpy.empty());
-            ASSERT_TRUE(engineSpy.empty());
+            ASSERT_EQ(engineSpy.count(), 1);
             ASSERT_TRUE(stageSpy.empty());
-            ASSERT_TRUE(spritesSpy.empty());
-            ASSERT_TRUE(monitorsSpy.empty());
+            ASSERT_EQ(spritesSpy.count(), 1);
+            ASSERT_EQ(clonesSpy.count(), 1);
+            ASSERT_EQ(monitorsSpy.count(), 1);
             ASSERT_TRUE(monitorAddedSpy.empty());
             ASSERT_EQ(loader->fileName(), fileName);
             ASSERT_FALSE(loader->loadStatus());
@@ -52,10 +54,11 @@ class ProjectLoaderTest : public testing::Test
             ASSERT_EQ(fileNameSpy.count(), 1);
             ASSERT_EQ(loadStatusSpy.count(), 2);
             ASSERT_EQ(loadingFinishedSpy.count(), 1);
-            ASSERT_EQ(engineSpy.count(), 1);
+            ASSERT_EQ(engineSpy.count(), 2);
             ASSERT_EQ(stageSpy.count(), 1);
-            ASSERT_EQ(spritesSpy.count(), 1);
-            ASSERT_EQ(monitorsSpy.count(), loader->monitorList().size());
+            ASSERT_EQ(spritesSpy.count(), 2);
+            ASSERT_EQ(clonesSpy.count(), 1);
+            ASSERT_EQ(monitorsSpy.count(), loader->monitorList().size() + 1);
             ASSERT_EQ(monitorAddedSpy.count(), loader->monitorList().size());
         }
 };
@@ -113,13 +116,13 @@ TEST_F(ProjectLoaderTest, Clones)
     load(&loader, "clones.sb3");
     ASSERT_TRUE(cloneCreatedSpy.empty());
     ASSERT_TRUE(cloneDeletedSpy.empty());
-    ASSERT_TRUE(clonesChangedSpy.empty());
+    ASSERT_EQ(clonesChangedSpy.count(), 1);
 
     auto engine = loader.engine();
     engine->run();
     ASSERT_EQ(cloneCreatedSpy.count(), 3);
     ASSERT_EQ(cloneDeletedSpy.count(), 0);
-    ASSERT_EQ(clonesChangedSpy.count(), 3);
+    ASSERT_EQ(clonesChangedSpy.count(), 4);
 
     const auto &sprites = loader.spriteList();
     const auto &clones = loader.cloneList();
@@ -138,7 +141,7 @@ TEST_F(ProjectLoaderTest, Clones)
     clones[1]->sprite()->deleteClone();
     ASSERT_EQ(cloneCreatedSpy.count(), 3);
     ASSERT_EQ(cloneDeletedSpy.count(), 1);
-    ASSERT_EQ(clonesChangedSpy.count(), 4);
+    ASSERT_EQ(clonesChangedSpy.count(), 5);
     ASSERT_EQ(clones.size(), 2);
 }