From 80de01e8129e9be2d4a0f96c51a6642e92e908e3 Mon Sep 17 00:00:00 2001
From: Matthieu Schaller <schaller@strw.leidenuniv.nl>
Date: Tue, 11 Feb 2020 16:29:26 +0100
Subject: [PATCH] Add the missing task and subtask types to the python
 analsysis scripts

---
 tools/task_plots/analyse_tasks.py | 15 ++++++++++++---
 tools/task_plots/plot_tasks.py    | 13 +++++++++++--
 2 files changed, 23 insertions(+), 5 deletions(-)

diff --git a/tools/task_plots/analyse_tasks.py b/tools/task_plots/analyse_tasks.py
index c093d18c2d..4a87ebad3c 100755
--- a/tools/task_plots/analyse_tasks.py
+++ b/tools/task_plots/analyse_tasks.py
@@ -93,6 +93,7 @@ TASKTYPES = [
     "kick2",
     "timestep",
     "timestep_limiter",
+    "timestep_sync",
     "send",
     "recv",
     "grav_long_range",
@@ -116,6 +117,9 @@ TASKTYPES = [
     "bh_in",
     "bh_out",
     "bh_ghost",
+    "bh_swallow_ghost1",
+    "bh_swallow_ghost2",
+    "bh_swallow_ghost3",
     "fof_self",
     "fof_pair",
     "count",
@@ -135,16 +139,21 @@ SUBTYPES = [
     "tend_bpart",
     "xv",
     "rho",
+    "part_swallow",
+    "bpart_merger",
     "gpart",
     "multipole",
     "spart",
     "stars_density",
     "stars_feedback",
-    "sf_counts"
-    "bpart",
+    "sf_counts",
+    "bpart_rho",
+    "bpart_swallow",
+    "bpart_feedback",
     "bh_density",
     "bh_swallow",
-    "do_swallow",
+    "do_gas_swallow",
+    "do_bh_swallow",
     "bh_feedback",
     "count",
 ]
diff --git a/tools/task_plots/plot_tasks.py b/tools/task_plots/plot_tasks.py
index 64d5529cc1..b0b9f0e511 100755
--- a/tools/task_plots/plot_tasks.py
+++ b/tools/task_plots/plot_tasks.py
@@ -170,6 +170,7 @@ TASKTYPES = [
     "kick2",
     "timestep",
     "timestep_limiter",
+    "timestep_sync",
     "send",
     "recv",
     "grav_long_range",
@@ -193,6 +194,9 @@ TASKTYPES = [
     "bh_in",
     "bh_out",
     "bh_ghost",
+    "bh_swallow_ghost1",
+    "bh_swallow_ghost2",
+    "bh_swallow_ghost3",
     "fof_self",
     "fof_pair",
     "count",
@@ -212,16 +216,21 @@ SUBTYPES = [
     "tend_bpart",
     "xv",
     "rho",
+    "part_swallow",
+    "bpart_merger",
     "gpart",
     "multipole",
     "spart",
     "stars_density",
     "stars_feedback",
     "sf_counts",
-    "bpart",
+    "bpart_rho",
+    "bpart_swallow",
+    "bpart_feedback",
     "bh_density",
     "bh_swallow",
-    "do_swallow",
+    "do_gas_swallow",
+    "do_bh_swallow",
     "bh_feedback",
     "count",
 ]
-- 
GitLab