update pipeline.cpp to stop and terminate control module with pipeline

Apra-Labs · Sep 17, 2024 · 42550e3 · kumaakh · Sep 18, 2024 · 42550e3
1 parent b9a5ea2
commit 42550e3
Showing 1 changed file with 16 additions and 11 deletions.
diff --git a/base/src/PipeLine.cpp b/base/src/PipeLine.cpp
@@ -232,6 +232,11 @@ void PipeLine::stop()
 			i->get()->stop();
 		}
 	}
+
+	if ((modules[0]->controlModule) != nullptr)
+	{
+		modules[0]->controlModule->stop();
+	}
 }
 
 void PipeLine::wait_for_all(bool ignoreStatus)
@@ -242,15 +247,15 @@ void PipeLine::wait_for_all(bool ignoreStatus)
 		return;
 	}
 
-	for (auto i = modules.begin(); i != modules.end(); i++)
+	if ((modules[0]->controlModule) != nullptr)
 	{
-		Module& m = *(i->get());
+		Module& m = *(modules[0]->controlModule);
 		m.myThread.join();
 	}
 
-	if ((modules[0]->controlModule) != nullptr)
+	for (auto i = modules.begin(); i != modules.end(); i++)
 	{
-		Module& m = *(modules[0]->controlModule);
+		Module& m = *(i->get());
 		m.myThread.join();
 	}
 }
@@ -264,12 +269,6 @@ void PipeLine::interrupt_wait_for_all()
 		return;
 	}
 
-	for (auto i = modules.begin(); i != modules.end(); i++)
-	{
-		Module& m = *(i->get());
-		m.myThread.interrupt();
-	}
-
 	if ((modules[0]->controlModule) != nullptr)
 	{
 		Module& m = *(modules[0]->controlModule);
@@ -279,7 +278,7 @@ void PipeLine::interrupt_wait_for_all()
 	for (auto i = modules.begin(); i != modules.end(); i++)
 	{
 		Module& m = *(i->get());
-		m.myThread.join();
+		m.myThread.interrupt();
 	}
 
 	if ((modules[0]->controlModule) != nullptr)
@@ -288,6 +287,12 @@ void PipeLine::interrupt_wait_for_all()
 		m.myThread.join();
 	}
 
+	for (auto i = modules.begin(); i != modules.end(); i++)
+	{
+		Module& m = *(i->get());
+		m.myThread.join();
+	}
+
 	myStatus = PL_STOPPED;
 }