Move CaKernel code processing from KrancThorn.m to CaKernel.m

ianhinder · Sep 13, 2013 · a8e14b3 · a8e14b3
1 parent c712d76
commit a8e14b3
Show file tree

Hide file tree

Showing 2 changed files with 31 additions and 22 deletions.
diff --git a/Tools/CodeGen/CaKernel.m b/Tools/CodeGen/CaKernel.m
@@ -19,7 +19,7 @@
 *)
 
 BeginPackage["CaKernel`", {"Errors`", "Helpers`", "Kranc`", "CodeGenCactus`", "CodeGenKranc`", "MapLookup`",
-                           "Calculation`", "CodeGen`", "CodeGenCalculation`", "CodeGenC`"}];
+                           "Calculation`", "CodeGen`", "CodeGenCalculation`", "CodeGenC`", "Code`", "Object`"}];
 
 CaKernelCCL;
 CaKernelCode;
@@ -28,6 +28,7 @@
 CaKernelConfigurationCLL;
 CaKernelInterfaceCLL;
 WithHostCalculations;
+CaKernelProcessCode;
 
 Begin["`Private`"];
 
@@ -246,6 +247,34 @@ CCTK_POINTER FUNCTION Device_GetVarI (CCTK_POINTER IN cctkGH, CCTK_INT IN vi, CC
   WithHostCalculations[calcs_List] :=
     Flatten[Map[splitHostCaKernel, calcs],1]];
 
+Options[CaKernelProcessCode] = ThornOptions;
+
+DefFn[
+  CaKernelProcessCode[c_Code, opts:OptionsPattern[]] :=
+  Module[
+    {calcs = GetObjectField[c, "Calculations"], c2},
+
+    (* Make the CaKernel option calculation-specific *)
+    calcs = Map[Append[#,UseCaKernel -> OptionValue[UseCaKernel]] &, calcs];
+
+    If[OptionValue[GenerateHostCode] && OptionValue[UseCaKernel],
+       calcs = WithHostCalculations[calcs]];
+
+    If[!And@@Map[ListQ, calcs], Print[Short[calcs//InputForm]]; ThrowError["Result of WithHostCalculations is not a list of lists"]];
+
+    (* Add ExecuteOn -> Device to any CaKernel calculation that has no ExecuteOn option *)
+    calcs = Map[If[!lookup[#,UseCaKernel,False], #, If[mapContains[#,ExecuteOn], #, Append[#,ExecuteOn->Device]]] &, calcs];
+
+    c2 = SetObjectField[c, "Calculations", calcs];
+
+    If[OptionValue[UseCaKernel],
+       c2 = AppendObjectField[c2, "IncludeFiles", "CaCUDALib_driver_support.h"]];
+
+    If[OptionValue[UseCaKernel],
+       c2 = AppendObjectField[c2, "InheritedImplementations", "Accelerator"]];
+    c2]];
+
+
 End[];
 
 EndPackage[];
diff --git a/Tools/CodeGen/KrancThorn.m b/Tools/CodeGen/KrancThorn.m
@@ -125,27 +125,7 @@ Thorn generation (main entry point for non-tensorial thorns)
        CaKernel
        ------------------------------------------------------------------------ *)
 
-    c = Module[
-      {calcs = GetObjectField[c, "Calculations"]},
-
-      (* Make the CaKernel option calculation-specific *)
-      calcs = Map[Append[#,UseCaKernel -> OptionValue[UseCaKernel]] &, calcs];
-
-      If[OptionValue[GenerateHostCode] && OptionValue[UseCaKernel],
-         calcs = WithHostCalculations[calcs]];
-
-      If[!And@@Map[ListQ, calcs], Print[Short[calcs//InputForm]]; ThrowError["Result of WithHostCalculations is not a list of lists"]];
-
-      (* Add ExecuteOn -> Device to any CaKernel calculation that has no ExecuteOn option *)
-      calcs = Map[If[!lookup[#,UseCaKernel,False], #, If[mapContains[#,ExecuteOn], #, Append[#,ExecuteOn->Device]]] &, calcs];
-
-      SetObjectField[c, "Calculations", calcs]];
-
-    If[OptionValue[UseCaKernel],
-       c = AppendObjectField[c, "IncludeFiles", "CaCUDALib_driver_support.h"]];
-
-    If[OptionValue[UseCaKernel],
-       c = AppendObjectField[c, "InheritedImplementations", "Accelerator"]];
+    c = CaKernelProcessCode[c, opts];
 
     (* ------------------------------------------------------------------------ 
        Add coordinates group