Running also stops when process finishes

Author: mklomp

mirte_robot/linetrace.py

@@ -16,14 +16,19 @@
 # Global shared memory objects (TODO: check if we need shared memory, why is server working?)
 stepper = multiprocessing.Value('b', True)
 do_step = multiprocessing.Value('b', False)
-is_running = False
+running = multiprocessing.Value('b', False)
 
-def stop_mirte():
-     global is_running
-     process.terminate()
-     is_running = False
+# the stop be stopped when:
+# 1) the code finishes
+# 2) the user stopped the process
+# 3) the websocket connection is closed
+def stop_mirte(terminate = True):
+     global running
+     if terminate:
+        process.terminate()
+     running.value = False
 
-def load_mirte_module(stepper, do_step):
+def load_mirte_module(stepper, do_step, running):
 
     def trace_lines(frame, event, arg):
        global do_step
@@ -64,6 +69,8 @@ def traceit(frame, event, arg):
     # Sending the linetrace 0 to the client
     server.send_message_to_all("0")
 
+    stop_mirte(False)
+
 process = multiprocessing.Process(target = load_mirte_module, args=(stepper, do_step))
 
 def start_mirte():
@@ -72,22 +79,22 @@ def start_mirte():
     # but, just in case make sure to stop this
     if process.is_alive():
       process.terminate()
-    process = multiprocessing.Process(target = load_mirte_module, args=(stepper, do_step))
+    process = multiprocessing.Process(target = load_mirte_module, args=(stepper, do_step, running))
     process.start()
 
 def client_left(client, server):
     stop_mirte()
 
 def message_received(client, server, message):
-   global stepper, do_step, is_running
+   global stepper, do_step, running
 
    if message == "b": #break (pause)
       stepper.value = True
    if message == "c": #continue (play)
       stepper.value = False
-      if not is_running:
+      if not running.value:
          start_mirte()
-         is_running = True
+         running.value = True
    if message == "s": #step (step)
       do_step.value = True
    if message == "e": #exit (stop)