Make sure the capture thread gets terminated

RetGal · Dec 15, 2024 · 26681d3 · 26681d3
1 parent afa6255
commit 26681d3
Showing 1 changed file with 11 additions and 1 deletion.
diff --git a/src/main/java/mpo/dayon/assisted/capture/CaptureEngine.java b/src/main/java/mpo/dayon/assisted/capture/CaptureEngine.java
@@ -43,13 +43,16 @@ public class CaptureEngine implements ReConfigurable<CaptureEngineConfiguration>
 
     private boolean reconfigured;
 
+    private boolean running;
+
     public CaptureEngine(CaptureFactory captureFactory) {
         this.captureFactory = captureFactory;
         this.captureDimension = captureFactory.getDimension();
         final int x = (captureDimension.width + TILE_DIMENSION.width -1) / TILE_DIMENSION.width;
         final int y = (captureDimension.height + TILE_DIMENSION.height -1) / TILE_DIMENSION.height;
         this.previousCapture = new long[x * y];
         resetPreviousCapture();
+        running = true;
 
         this.thread = new Thread(new RunnableEx() {
             @Override
@@ -83,12 +86,19 @@ public void addListener(CaptureEngineListener listener) {
 
     public void start() {
         Log.debug("CaptureEngine start");
+        running = true;
         thread.start();
     }
 
     public void stop() {
         Log.debug("CaptureEngine stop");
+        running = false;
         thread.interrupt();
+        try {
+            thread.join();
+        } catch (InterruptedException e) {
+            Thread.currentThread().interrupt();
+        }
     }
 
     private void mainLoop() {
@@ -101,7 +111,7 @@ private void mainLoop() {
         int skipped = 0;
         AtomicBoolean reset = new AtomicBoolean(false);
 
-        while (true) {
+        while (running) {
             synchronized (reconfigurationLOCK) {
                 if (reconfigured) {
                     // assuming everything has changed (!)