diff options
Diffstat (limited to 'Tools')
| -rw-r--r-- | Tools/CodeGen/CalculationFunction.m | 6 |
1 files changed, 6 insertions, 0 deletions
diff --git a/Tools/CodeGen/CalculationFunction.m b/Tools/CodeGen/CalculationFunction.m index 1fd65a9..58e7a5f 100644 --- a/Tools/CodeGen/CalculationFunction.m +++ b/Tools/CodeGen/CalculationFunction.m @@ -25,6 +25,7 @@ BeginPackage["CalculationFunction`", {"CodeGen`", CreateCalculationFunction::usage = ""; VerifyCalculation::usage = ""; calculationSymbols::usage = ""; +GridFunctionsInExpression; Begin["`Private`"]; @@ -565,6 +566,11 @@ equationLoop[eqs_, cleancalc_, gfs_, shorts_, incs_, groups_, pddefs_, If[debugInLoop, Map[InfoVariable[GridName[#]] &, gfsInLHS], ""]}, opts]]; +(* Unsorted *) + +GridFunctionsInExpression[x_, groups_] := + Union[Cases[x, _ ? (MemberQ[allGroupVariables[groups],#] &), Infinity]]; + End[]; EndPackage[]; |