diff options
Diffstat (limited to 'Tools/CodeGen/KrancTensor.m')
-rw-r--r-- | Tools/CodeGen/KrancTensor.m | 12 |
1 files changed, 10 insertions, 2 deletions
diff --git a/Tools/CodeGen/KrancTensor.m b/Tools/CodeGen/KrancTensor.m index 204bc60..e9c9876 100644 --- a/Tools/CodeGen/KrancTensor.m +++ b/Tools/CodeGen/KrancTensor.m @@ -26,6 +26,7 @@ If[!ValueQ[$KrancTensorPackage], $KrancTensorPackage = "TensorToolsKranc`"]; BeginPackage["KrancTensor`", {"Errors`", "KrancThorn`", "MapLookup`", "KrancGroups`", "Kranc`", $KrancTensorPackage}]; +BeginPackage["KrancTensor`", {"Errors`", "KrancThorn`", "MapLookup`", "KrancGroups`", "Kranc`", $KrancTensorPackage, "ConservationCalculation`", "TensorTools`"}]; CreateKrancThornTT::usage = "Construct a Kranc thorn using tensor expressions."; @@ -38,12 +39,14 @@ Begin["`Private`"]; -------------------------------------------------------------------------- *) CreateKrancThornTT[groups_, parentDirectory_, thornName_, opts___] := - Module[{calcs, expCalcs, expGroups, options, derivs, expDerivs, reflectionSymmetries, declaredGroups}, + Module[{calcs, expCalcs, expGroups, options, derivs, expDerivs, reflectionSymmetries, declaredGroups, consCalcs, expConsCalcs}, InfoMessage[Terse, "Processing tensorial arguments"]; calcs = lookup[{opts}, Calculations]; + consCalcs = lookupDefault[{opts}, ConservationCalculations, {}]; derivs = lookupDefault[{opts}, PartialDerivatives, {}]; Map[CheckCalculationTensors, calcs]; expCalcs = Map[makeCalculationExplicit, calcs]; + expConsCalcs = Map[makeCalculationExplicit, consCalcs]; InfoMessage[Info, "Group definitions:", groups]; VerifyGroups[groups]; @@ -51,6 +54,8 @@ CreateKrancThornTT[groups_, parentDirectory_, thornName_, opts___] := expDerivs = Flatten[Map[ExpandComponents,derivs],1]; expGroups = Map[makeGroupExplicit, groups]; options = Join[DeleteCases[{opts}, Calculations -> _], {Calculations -> expCalcs}]; + options = Join[DeleteCases[options, ConservationCalculations -> _], + {ConservationCalculations -> expConsCalcs}]; options = Join[DeleteCases[options, PartialDerivatives -> _], {PartialDerivatives -> expDerivs}]; declaredGroups = lookupDefault[{opts}, DeclaredGroups, {}]; @@ -77,7 +82,10 @@ makeCalculationExplicit[calc_] := mapValueMapMultiple[calc, {Shorthands -> ExpandComponents, CollectList -> ExpandComponents, - Equations -> ExpandComponents}]; + Equations -> ExpandComponents, + PrimitiveEquations -> MakeExplicit, + ConservedEquations -> MakeExplicit, + Primitives -> MakeExplicit}]; makeGroupExplicit[g_] := Module[{variables, newVariables, newGroup}, |