From 8ba4dafef94c1628f16659b874bd356c9d9e9532 Mon Sep 17 00:00:00 2001
From: Henry Pinkard <7969470+henrypinkard@users.noreply.github.com>
Date: Thu, 7 Sep 2023 07:49:11 -0700
Subject: [PATCH] add prints to stopping headless

---
 pycromanager/headless.py      | 18 +++++++++++++++++-
 pycromanager/test/conftest.py |  2 +-
 2 files changed, 18 insertions(+), 2 deletions(-)

diff --git a/pycromanager/headless.py b/pycromanager/headless.py
index 52d27b74..c58d0716 100644
--- a/pycromanager/headless.py
+++ b/pycromanager/headless.py
@@ -87,15 +87,31 @@ def get_tagged_image(self, cam_index, camera, height, width, binning=None, pixel
 _JAVA_HEADLESS_SUBPROCESSES = []
 _PYMMCORES = []
 
-def stop_headless():
+def stop_headless(debug=False):
     for p in _JAVA_HEADLESS_SUBPROCESSES:
+        if debug:
+            print('Stopping headless process with pid {}'.format(p.pid))
         p.terminate()
+        if debug:
+            print('Waiting for process with pid {} to terminate'.format(p.pid))
         p.wait()  # wait for process to terminate
+        if debug:
+            print('Process with pid {} terminated'.format(p.pid))
     _JAVA_HEADLESS_SUBPROCESSES.clear()
+    if debug:
+        print('Stopping {} pymmcore instances'.format(len(_PYMMCORES)))
     for c in _PYMMCORES:
+        if debug:
+            print('Stopping pymmcore instance')
         c.unloadAllDevices()
+        if debug:
+            print('Unloaded all devices')
         Engine.get_instance().shutdown()
+        if debug:
+            print('Engine shut down')
     _PYMMCORES.clear()
+    if debug:
+        print('Headless stopped')
 
 # make sure any Java processes are cleaned up when Python exits
 atexit.register(stop_headless)
diff --git a/pycromanager/test/conftest.py b/pycromanager/test/conftest.py
index 468e041b..1b464242 100644
--- a/pycromanager/test/conftest.py
+++ b/pycromanager/test/conftest.py
@@ -192,4 +192,4 @@ def launch_mm_headless(install_mm):
 
         yield None
 
-        stop_headless()
+        stop_headless(debug=True)