From c8c47c5e308d8a0d50764d15296ab92c379f08c6 Mon Sep 17 00:00:00 2001
From: Oleg Smirnov <osmirnov@gmail.com>
Date: Mon, 2 Dec 2024 16:31:48 +0200
Subject: [PATCH] backed out
 https://github.com/caracal-pipeline/stimela/tree/profiling-hacks which were
 merged in by accident

---
 stimela/task_stats.py              | 36 +++++-------------------------
 tests/stimela_tests/test_recipe.py |  8 +++----
 2 files changed, 10 insertions(+), 34 deletions(-)

diff --git a/stimela/task_stats.py b/stimela/task_stats.py
index 0c27fef8..0d865ac7 100644
--- a/stimela/task_stats.py
+++ b/stimela/task_stats.py
@@ -237,7 +237,7 @@ def update_stats(now: datetime, sample: TaskStatsDatum):
         _taskstats[key][0] = (now - start).total_seconds()
 
 
-def update_process_status(stimela_process=None, child_processes=None):
+def update_process_status():
     # current subtask info
     ti = _task_stack[-1] if _task_stack else None
 
@@ -247,27 +247,11 @@ def update_process_status(stimela_process=None, child_processes=None):
 
     # form up sample datum
     s = TaskStatsDatum(num_samples=1)
-
-    # If we have the process objects, we don't need to block.
-    interval = 0 if child_processes else 1
-
-    # Grab the stimela process and its children (recursively).
-    stimela_process = stimela_process or psutil.Process()
-    stimela_children = child_processes or stimela_process.children(recursive=True)
-
-    # Assume that all child processes belong to the same task.
-    # TODO: Handling of children is rudimentary at present.
-    # How would this work for scattered/parallel steps?
-    if stimela_children and ti:
-        processes = stimela_children
-    else:
-        processes = [stimela_process]
-
     # CPU and memory
-    s.cpu = sum(p.cpu_percent(interval=interval) for p in processes)
-    system_memory = psutil.virtual_memory().total
-    s.mem_used = round(sum(p.memory_info().rss for p in processes) / 2**30)
-    s.mem_total = round(system_memory / 2**30)
+    s.cpu = psutil.cpu_percent()
+    mem = psutil.virtual_memory()
+    s.mem_used = round(mem.total*mem.percent/100 / 2**30)
+    s.mem_total = round(mem.total / 2**30)
     # load
     s.load, _, _ = psutil.getloadavg()
 
@@ -335,16 +319,8 @@ def update_process_status(stimela_process=None, child_processes=None):
 async def run_process_status_update():
     if progress_bar:
         with contextlib.suppress(asyncio.CancelledError):
-            stimela_process = psutil.Process()
-            child_processes = set()
             while True:
-                new_children = {c for c in stimela_process.children(recursive=True) if c not in child_processes}
-                old_children = {c for c in child_processes if c not in stimela_process.children(recursive=True)}
-                child_processes = child_processes.union(new_children) - old_children
-                try:
-                    update_process_status(stimela_process, child_processes)
-                except psutil.NoSuchProcess:
-                    continue
+                update_process_status()
                 await asyncio.sleep(1)
 
 _printed_stats = dict(
diff --git a/tests/stimela_tests/test_recipe.py b/tests/stimela_tests/test_recipe.py
index 0c007e4d..feb1b1dd 100644
--- a/tests/stimela_tests/test_recipe.py
+++ b/tests/stimela_tests/test_recipe.py
@@ -89,7 +89,7 @@ def test_test_loop_recipe():
     assert retcode == 0
 
     print("===== expecting no errors now =====")
-    retcode = os.system("stimela exec test_loop_recipe.yml same_as_cubical_image_loop ms=foo")
+    retcode = os.system("stimela -v -b native exec test_loop_recipe.yml same_as_cubical_image_loop ms=foo")
     assert retcode == 0
 
     for name in "abc":
@@ -97,14 +97,14 @@ def test_test_loop_recipe():
         if not os.path.exists(msname):
             os.mkdir(msname)
     print("===== expecting no errors now =====")
-    retcode = os.system("stimela exec test_loop_recipe.yml loop_recipe")
+    retcode = os.system("stimela -v -b native exec test_loop_recipe.yml loop_recipe")
     assert retcode == 0
 
 def test_scatter():
     print("===== expecting no errors now =====")
-    retcode = os.system("stimela exec test_scatter.yml basic_loop")
+    retcode = os.system("stimela -v -b native exec test_scatter.yml basic_loop")
     assert retcode == 0
 
     print("===== expecting no errors now =====")
-    retcode = os.system("stimela exec test_scatter.yml nested_loop")
+    retcode = os.system("stimela -v -b native exec test_scatter.yml nested_loop")
     assert retcode == 0