1 files changed, 77 insertions, 0 deletions
diff --git a/ML_CCZ4_Helper/src/SetGroupTags.c b/ML_CCZ4_Helper/src/SetGroupTags.c
new file mode 100644
index 0000000..00878ee
--- /dev/null
+++ b/ML_CCZ4_Helper/src/SetGroupTags.c
@@ -0,0 +1,77 @@
+#include <cctk.h>
+#include <cctk_Parameters.h>
+#include <util_Table.h>
+
+#include <assert.h>
+
+static void
+set_group_tags (int const checkpoint,
+                int const persistent,
+                int const prolongate,
+                char const * restrict const gn);
+
+int
+ML_CCZ4_SetGroupTags (void)
+{
+  DECLARE_CCTK_PARAMETERS;
+  
+  int const checkpoint = timelevels > 1;
+  set_group_tags (checkpoint, checkpoint, 1, "ADMBase::metric");
+  set_group_tags (checkpoint, checkpoint, 1, "ADMBase::curv");
+  set_group_tags (checkpoint, checkpoint, 1, "ADMBase::lapse");
+  set_group_tags (checkpoint, checkpoint, 1, "ADMBase::shift");
+  set_group_tags (checkpoint, checkpoint, 1, "ADMBase::dtlapse");
+  set_group_tags (checkpoint, checkpoint, 1, "ADMBase::dtshift");
+  
+  set_group_tags (checkpoint, checkpoint, 0, "ML_CCZ4::ML_cons_detg");
+  set_group_tags (checkpoint, checkpoint, 0, "ML_CCZ4::ML_cons_Gamma");
+  set_group_tags (checkpoint, checkpoint, 0, "ML_CCZ4::ML_cons_traceA");
+  set_group_tags (checkpoint, checkpoint, 0, "ML_CCZ4::ML_Ham");
+  set_group_tags (checkpoint, checkpoint, 0, "ML_CCZ4::ML_mom");
+  
+  int const rhs_checkpoint = rhs_timelevels > 1;
+  set_group_tags (rhs_checkpoint, rhs_checkpoint, 0, "ML_CCZ4::ML_log_confacrhs");
+  set_group_tags (rhs_checkpoint, rhs_checkpoint, 0, "ML_CCZ4::ML_metricrhs");
+  set_group_tags (rhs_checkpoint, rhs_checkpoint, 0, "ML_CCZ4::ML_Gammarhs");
+  set_group_tags (rhs_checkpoint, rhs_checkpoint, 0, "ML_CCZ4::ML_trace_curvrhs");
+  set_group_tags (rhs_checkpoint, rhs_checkpoint, 0, "ML_CCZ4::ML_curvrhs");
+  set_group_tags (rhs_checkpoint, rhs_checkpoint, 0, "ML_CCZ4::ML_lapserhs");
+  set_group_tags (rhs_checkpoint, rhs_checkpoint, 0, "ML_CCZ4::ML_dtlapserhs");
+  set_group_tags (rhs_checkpoint, rhs_checkpoint, 0, "ML_CCZ4::ML_shiftrhs");
+  set_group_tags (rhs_checkpoint, rhs_checkpoint, 0, "ML_CCZ4::ML_dtshiftrhs");
+  set_group_tags (rhs_checkpoint, rhs_checkpoint, 0, "ML_CCZ4::ML_Thetarhs");
+  
+  return 0;
+}
+
+static void
+set_group_tags (int const checkpoint,
+                int const persistent,
+                int const prolongate,
+                char const * restrict const gn)
+{
+  assert (gn);
+  
+  int const gi = CCTK_GroupIndex (gn);
+  assert (gi >= 0);
+  
+  int const table = CCTK_GroupTagsTableI (gi);
+  assert (table >= 0);
+  
+  if (! checkpoint) {
+    int const ierr = Util_TableSetString (table, "no", "Checkpoint");
+    assert (! ierr);
+  }
+  
+  if (! persistent) {
+    int const ierr = Util_TableSetString (table, "no", "Persistent");
+    assert (! ierr);
+  }
+  
+  if (! prolongate) {
+    int const iret = Util_TableDeleteKey (table, "ProlongationParameter");
+    assert (iret == 0 || iret == UTIL_ERROR_TABLE_NO_SUCH_KEY);
+    int const ierr = Util_TableSetString (table, "none", "Prolongation");
+    assert (! ierr);
+  }
+}