MoL.m: Add MoLProcessGroups

This function is called to modify the groups structure for MoL's requirements.  At the moment, it adds an InterfaceTimelevels option to the group with a value of EvolutionTimelevels (passed in as an argument).

2 files changed, 18 insertions, 0 deletions

diff --git a/Tools/CodeGen/Kranc.m b/Tools/CodeGen/Kranc.m
index c83fdde..31ff109 100644
--- a/Tools/CodeGen/Kranc.m
+++ b/Tools/CodeGen/Kranc.m
@@ -57,6 +57,10 @@ SimpleCode};
 
 {Name, Definitions, shift, spacing, SBPDerivative};
 
+(* Interface.m *)
+
+{InterfaceTimelevels};
+
 (* KrancThorn.m *)
 
 ThornOptions =
diff --git a/Tools/CodeGen/MoL.m b/Tools/CodeGen/MoL.m
index 83b91fd..0425412 100644
--- a/Tools/CodeGen/MoL.m
+++ b/Tools/CodeGen/MoL.m
@@ -34,6 +34,7 @@ EvolvedGroupToRHSGroup::usage = "";
 MoLRHSGroupDefinitions;
 MoLRHSODEGroupDefinitions;
 MoLUsedFunctions;
+MoLProcessGroups;
 
 Begin["`Private`"];
 
@@ -597,6 +598,19 @@ DefFn[
     *)
   }];
 
+DefFn[
+  MoLProcessGroups[declaredGroups_List, calcs_List, groups_List,
+                   evolutionTimelevels_Integer] :=
+  Module[
+    {evolvedGroups, groups2},
+    evolvedGroups = MoLEvolvedGroups[declaredGroups, calcs, groups];
+
+    groups2 = Map[If[MemberQ[evolvedGroups, groupName[#]],
+                     (* Print["Adding InterfaceTimelevels to ", groupName[#]]; *)
+                     EnsureInterfaceTimelevels[#, evolutionTimelevels],
+                     #] &, groups];
+    groups2]];
+
 End[];
 
 EndPackage[];