2 files changed, 18 insertions, 0 deletions
diff --git a/m/prototype/ML_BSSN_Helper/src/NewRad.c b/m/prototype/ML_BSSN_Helper/src/NewRad.c
index 9c576a1..c5529a1 100644
--- a/m/prototype/ML_BSSN_Helper/src/NewRad.c
+++ b/m/prototype/ML_BSSN_Helper/src/NewRad.c
@@ -4,6 +4,11 @@
 #include <cctk_Arguments.h>
 #include <cctk_Parameters.h>
 
+#define THORN_ML_BSSN           // "ML_BSSN" will be replaced
+#ifdef THORN_ML_CCZ4
+#  define HAVE_THETA
+#endif
+
 static void
 newrad (cGH const * restrict cctkGH,
         CCTK_REAL const * restrict var,
@@ -32,6 +37,10 @@ ML_BSSN_NewRad (CCTK_ARGUMENTS)
   newrad (cctkGH, Xt2  , Xt2rhs  , 0.0, 1.0);
   newrad (cctkGH, Xt3  , Xt3rhs  , 0.0, 1.0);
   
+#ifdef HAVE_THETA
+  newrad (cctkGH, Theta, Thetarhs, 0.0, v0 );
+#endif
+  
   newrad (cctkGH, trK  , trKrhs  , 0.0, v0 );
   
   newrad (cctkGH, At11 , At11rhs , 0.0, 1.0);
diff --git a/m/prototype/ML_BSSN_Helper/src/SetGroupTags.c b/m/prototype/ML_BSSN_Helper/src/SetGroupTags.c
index 7ae24bb..c85b0b7 100644
--- a/m/prototype/ML_BSSN_Helper/src/SetGroupTags.c
+++ b/m/prototype/ML_BSSN_Helper/src/SetGroupTags.c
@@ -4,6 +4,11 @@
 
 #include <assert.h>
 
+#define THORN_ML_BSSN           // "ML_BSSN" will be replaced
+#ifdef THORN_ML_CCZ4
+#  define HAVE_THETA
+#endif
+
 static void
 set_group_tags (int const checkpoint,
                 int const persistent,
@@ -33,6 +38,10 @@ ML_BSSN_SetGroupTags (void)
   set_group_tags (rhs_checkpoint, rhs_checkpoint, 0, "ML_BSSN::ML_log_confacrhs");
   set_group_tags (rhs_checkpoint, rhs_checkpoint, 0, "ML_BSSN::ML_metricrhs");
   set_group_tags (rhs_checkpoint, rhs_checkpoint, 0, "ML_BSSN::ML_Gammarhs");
+  
+#ifdef HAVE_THETA
+  set_group_tags (rhs_checkpoint, rhs_checkpoint, 0, "ML_BSSN::ML_Thetarhs");
+#endif
   set_group_tags (rhs_checkpoint, rhs_checkpoint, 0, "ML_BSSN::ML_trace_curvrhs");
   set_group_tags (rhs_checkpoint, rhs_checkpoint, 0, "ML_BSSN::ML_curvrhs");
   set_group_tags (rhs_checkpoint, rhs_checkpoint, 0, "ML_BSSN::ML_lapserhs");