KrancThorn.m: Move RHS ODE group definitions into MoL.m

ianhinder · Sep 9, 2013 · 492ae49 · 492ae49
1 parent 59fed20
commit 492ae49
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 2 deletions.
diff --git a/Tools/CodeGen/KrancThorn.m b/Tools/CodeGen/KrancThorn.m
@@ -179,8 +179,7 @@ Thorn generation (main entry point for non-tensorial thorns)
     rhsGroupDefinitions = MoLRHSGroupDefinitions[groups, evolvedGroups];
     groups = Join[groups, rhsGroupDefinitions];
 
-    evolvedODEGroupDefinitions = Map[groupFromName[#, groups] &, evolvedODEGroups];
-    rhsODEGroupDefinitions = Map[EvolvedGroupToRHSGroup[#, evolvedODEGroupDefinitions] &, evolvedODEGroups];
+    rhsODEGroupDefinitions = MoLRHSODEGroupDefinitions[groups, evolvedODEGroups];
     groups = Join[groups, rhsODEGroupDefinitions];
 
     (* Add the groups into the calcs *)

diff --git a/Tools/CodeGen/MoL.m b/Tools/CodeGen/MoL.m
@@ -32,6 +32,7 @@
 MoLNonevolvedGroups;
 EvolvedGroupToRHSGroup::usage = "";
 MoLRHSGroupDefinitions;
+MoLRHSODEGroupDefinitions;
 
 Begin["`Private`"];
 
@@ -570,6 +571,13 @@
     evolvedGroupDefinitions = Map[groupFromName[#, groups] &, evolvedGroups];
     Map[EvolvedGroupToRHSGroup[#, evolvedGroupDefinitions] &, evolvedGroups]]];
 
+DefFn[
+  MoLRHSODEGroupDefinitions[groups_List, evolvedODEGroups_List] :=
+  Module[
+    {evolvedODEGroupDefinitions},
+    evolvedODEGroupDefinitions = Map[groupFromName[#, groups] &, evolvedODEGroups];
+    Map[EvolvedGroupToRHSGroup[#, evolvedODEGroupDefinitions] &, evolvedODEGroups]]];
+
 End[];
 
 EndPackage[];