Calculation.m: Add GetSchedule function

1 files changed, 10 insertions, 0 deletions

diff --git a/Tools/CodeGen/Calculation.m b/Tools/CodeGen/Calculation.m
index 55644ed..7169b05 100644
--- a/Tools/CodeGen/Calculation.m
+++ b/Tools/CodeGen/Calculation.m
@@ -37,6 +37,7 @@ AddCondition;
 AddConditionSuffix;
 InNewScheduleGroup;
 BoundaryCalculationQ;
+GetSchedule;
 
 Begin["`Private`"];
 
@@ -329,6 +330,15 @@ InNewScheduleGroup[groupName_String, calc_List] :=
         Schedule, {"in "<>groupName}],
       ScheduleGroups, Append[lookup[calc, ScheduleGroups, {}],newGroup]]];
 
+DefFn[
+  GetSchedule[calc_List] :=
+  Module[
+    {s = lookup[calc,Schedule,Automatic]},
+    If[s =!= Automatic && !ListQ[s],
+       ThrowError["Calculation "<>lookup[calc,Name]<>" has an invalid Schedule entry: ",
+                  s]];
+    s]];
+
 End[];
 
 EndPackage[];