Move symmetries code processing from KrancThorn.m into CodeGenSymmetr…

…ies.m
ianhinder · Sep 13, 2013 · a9281ab · a9281ab
1 parent 6905628
commit a9281ab
Show file tree

Hide file tree

Showing 2 changed files with 19 additions and 12 deletions.
diff --git a/Tools/CodeGen/CodeGenSymmetries.m b/Tools/CodeGen/CodeGenSymmetries.m
@@ -21,9 +21,10 @@
 BeginPackage[
   "CodeGenSymmetries`",
   {"Errors`", "Helpers`", "Kranc`", "CodeGenCactus`", "MapLookup`", "CodeGenKranc`",
-   "CodeGenC`", "CodeGen`", "KrancGroups`"}];
+   "CodeGenC`", "CodeGen`", "KrancGroups`", "Code`", "Object`"}];
 
 CreateSymmetriesRegistrationSource::usage = "";
+SymmetriesProcessCode;
 
 Begin["`Private`"];
 
@@ -107,6 +108,22 @@
     CodeGenC`SOURCELANGUAGE = lang;
     tmp]];
 
+Options[SymmetriesProcessCode] = ThornOptions;
+
+DefFn[
+  SymmetriesProcessCode[cIn_Code, opts:OptionsPattern[]] :=
+  Module[
+    {c = cIn},
+    InfoMessage[Terse, "Creating symmetry registration file"];
+    c = AppendObjectField[c, "IncludeFiles", "Symmetry.h"];
+    c = AppendObjectField[
+      c, "Sources", 
+      {Filename -> "RegisterSymmetries.cc",
+       Contents -> CreateSymmetriesRegistrationSource[
+         GetObjectField[c, "Name"], GetObjectField[c,"Implementation"], 
+         GetObjectField[c, "DeclaredGroups"], GetObjectField[c, "Groups"],
+         OptionValue[ReflectionSymmetries], False]}]]];
+
 End[];
 
 EndPackage[];
diff --git a/Tools/CodeGen/KrancThorn.m b/Tools/CodeGen/KrancThorn.m
@@ -243,17 +243,7 @@ Thorn generation (main entry point for non-tensorial thorns)
        Symmetries
        ------------------------------------------------------------------------ *)
 
-    c = AppendObjectField[c, "IncludeFiles", "Symmetry.h"];
-
-    InfoMessage[Terse, "Creating symmetry registration file"];
-
-    c = AppendObjectField[
-      c, "Sources", 
-      {Filename -> "RegisterSymmetries.cc",
-       Contents -> CreateSymmetriesRegistrationSource[
-         GetObjectField[c, "Name"], GetObjectField[c,"Implementation"], 
-         GetObjectField[c, "DeclaredGroups"], GetObjectField[c, "Groups"],
-         OptionValue[ReflectionSymmetries], False]}];
+    c = SymmetriesProcessCode[c, opts];
 
     (* ------------------------------------------------------------------------ 
        Boundary thorn