diff options
author | Ian Hinder <ian.hinder@aei.mpg.de> | 2012-01-27 23:08:19 -0600 |
---|---|---|
committer | Ian Hinder <ian.hinder@aei.mpg.de> | 2012-01-27 23:08:19 -0600 |
commit | a3693e49061e53960334fbf7f0cd476be290fe2a (patch) | |
tree | 0a0f894f0353ecadbb090281d859629b121e6ce8 | |
parent | 294e59653c5b1c23021c116be5b11ca26c6bb3a2 (diff) |
Add a thorn-global Shorthands option
This is added to the Shorthands option of each calculation. Might want to make it a union.
-rw-r--r-- | Tools/CodeGen/KrancTensor.m | 1 | ||||
-rw-r--r-- | Tools/CodeGen/KrancThorn.m | 2 |
2 files changed, 3 insertions, 0 deletions
diff --git a/Tools/CodeGen/KrancTensor.m b/Tools/CodeGen/KrancTensor.m index ea56d38..d0157c9 100644 --- a/Tools/CodeGen/KrancTensor.m +++ b/Tools/CodeGen/KrancTensor.m @@ -55,6 +55,7 @@ CreateKrancThornTT[groups_, parentDirectory_, thornName_, opts___] := expDerivs = Flatten[Map[ExpandComponents,derivs],1]; expGroups = Map[makeGroupExplicit, groups]; options = Join[DeleteCases[{opts}, Calculations -> _], {Calculations -> expCalcs}]; + options = mapReplace[options, Shorthands, ExpandComponents[lookup[options,Shorthands,{}]]]; options = Join[DeleteCases[options, ConservationCalculations -> _], {ConservationCalculations -> expConsCalcs}]; options = Join[DeleteCases[options, PartialDerivatives -> _], {PartialDerivatives -> expDerivs}]; diff --git a/Tools/CodeGen/KrancThorn.m b/Tools/CodeGen/KrancThorn.m index 0a4003a..c4e6c93 100644 --- a/Tools/CodeGen/KrancThorn.m +++ b/Tools/CodeGen/KrancThorn.m @@ -103,6 +103,8 @@ CreateKrancThorn[groupsOrig_, parentDirectory_, thornName_, opts:OptionsPattern[ cktCheckNamedArgs[{opts}]; calcs = OptionValue[Calculations]; + calcs = Map[mapReplaceAdd[#, Shorthands, Join[lookup[#,Shorthands,{}],OptionValue[Shorthands]]] &, calcs]; + declaredGroups = OptionValue[DeclaredGroups]; odeGroups = OptionValue[ODEGroups]; implementation = |