KrancThorn.m: Ensure that conservation calculations have required ent…

…ries
ianhinder · Nov 14, 2012 · 72b0c57 · 72b0c57
1 parent 052fa46
commit 72b0c57
Showing 1 changed file with 8 additions and 3 deletions.
diff --git a/Tools/CodeGen/KrancThorn.m b/Tools/CodeGen/KrancThorn.m
@@ -189,9 +189,14 @@ Thorn generation (main entry point for non-tensorial thorns)
 
     (* Add in calculations to solve any conservation laws that have
        been provided *)
-    calcs = Join[calcs,
-      consCalcs = Flatten[Map[ProcessConservationCalculation[#,thornName] &,
-        consCalcsIn],1]];
+
+    consCalcs = Flatten[Map[ProcessConservationCalculation[#,thornName] &,
+                            consCalcsIn],1];
+
+    consCalcs = Map[Join[#, {PartialDerivatives -> partialDerivs,
+                             Implementation -> implementation}] &, consCalcs];
+
+    calcs = Join[calcs,consCalcs];
     (* Print["consCalcs = ", consCalcs]; *)
 
     consGroups = Union@Flatten[