From d7c63491dc1284ad57860d6cca6ada97d732e014 Mon Sep 17 00:00:00 2001 From: Ian Hinder Date: Fri, 1 Oct 2010 14:06:32 +0200 Subject: CalculationFunction.m: Add function GridFunctionsInExpression --- Tools/CodeGen/CalculationFunction.m | 6 ++++++ 1 file changed, 6 insertions(+) (limited to 'Tools') 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[]; -- cgit v1.2.3