Move parameter checking code processing from KrancThorn.m to ParamCheck.m

1 files changed, 16 insertions, 1 deletions

diff --git a/Tools/CodeGen/ParamCheck.m b/Tools/CodeGen/ParamCheck.m
index 551c433..ac6ab22 100644
--- a/Tools/CodeGen/ParamCheck.m
+++ b/Tools/CodeGen/ParamCheck.m
@@ -19,10 +19,11 @@
 *)
 
 BeginPackage["ParamCheck`", {"Errors`", "Helpers`", "Kranc`", "CodeGenCactus`",
-                             "CodeGenC`", "CodeGen`", "CodeGenKranc`"}];
+                             "CodeGenC`", "CodeGen`", "CodeGenKranc`", "Code`", "Object`"}];
 
 ParameterCheckSource;
 ParameterCheckSchedule;
+ParamCheckProcessCode;
 
 Begin["`Private`"];
 
@@ -81,6 +82,20 @@ DefFn[checkCondition[{cond_, error_String}] :=
       renderbool[cond],
       {"CCTK_WARN(CCTK_WARN_ABORT, ", StringDrop[Stringify[error],-1], ");\n"}]]];
 
+Options[ParamCheckProcessCode] = ThornOptions;
+
+DefFn[
+  ParamCheckProcessCode[cIn_Code, opts:OptionsPattern[]] :=
+  Module[
+    {c = cIn},
+    If[Length[OptionValue[ParameterConditions]] > 0,
+       c = AppendObjectField[
+         c, "Sources",
+         {Filename -> "ParamCheck.cc",
+          Contents -> ParameterCheckSource[GetObjectField[c, "Name"], 
+                                           OptionValue[ParameterConditions]]}]];
+    c]];
+
 End[];
 
 EndPackage[];