diff options
-rw-r--r-- | Tools/CodeGen/KrancThorn.m | 11 |
1 files changed, 8 insertions, 3 deletions
diff --git a/Tools/CodeGen/KrancThorn.m b/Tools/CodeGen/KrancThorn.m index 85ebd4c..ee656e1 100644 --- a/Tools/CodeGen/KrancThorn.m +++ b/Tools/CodeGen/KrancThorn.m @@ -189,9 +189,14 @@ CreateKrancThorn[groupsOrig_, parentDirectory_, thornName_, opts:OptionsPattern[ (* 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[ |