CalculationFunction.m: Add function GridFunctionsInExpression

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[];