KrancGroups.m: Add EnsureTimelevels function

1 files changed, 10 insertions, 0 deletions

diff --git a/Tools/CodeGen/KrancGroups.m b/Tools/CodeGen/KrancGroups.m
index f1ee2ea..9140eb2 100644
--- a/Tools/CodeGen/KrancGroups.m
+++ b/Tools/CodeGen/KrancGroups.m
@@ -54,6 +54,7 @@ NonevolvedTimelevels;
 CheckGroups;
 VerifyGroupNames;
 VerifyGroups;
+EnsureTimelevels;
 
 Begin["`Private`"];
 
@@ -258,6 +259,15 @@ VerifyGroupNames[gns_] :=
    ThrowError["Not a list of group names: ", gns],
    Map[VerifyGroupName, gns]];
 
+EnsureTimelevels[g_, n_] :=
+  Module[
+    {tls, g2},
+    tls = GroupTimelevels[g];
+    If[tls === False,
+       AddGroupExtra[g, Timelevels -> n],
+       (* else *)
+       g /. {(Timelevels -> x_) :> (Timelevels -> Max[tls, n])}]];
+
 End[];
 
 EndPackage[];