adding helper function to determine number of loops in n-point function

FlexibleSUSY · Mar 22, 2017 · 1613ba1 · 1613ba1
1 parent 0e76fe6
commit 1613ba1
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/meta/SelfEnergies.m b/meta/SelfEnergies.m
@@ -54,6 +54,14 @@
 
 Begin["`Private`"];
 
+NPointFunctionQ[_SelfEnergies`Tadpole]                  := True;
+NPointFunctionQ[_SelfEnergies`FSSelfEnergy]             := True;
+NPointFunctionQ[_SelfEnergies`FSHeavySelfEnergy]        := True;
+NPointFunctionQ[_SelfEnergies`FSHeavyRotatedSelfEnergy] := True;
+NPointFunctionQ[_]                                      := False;
+
+NumberOfLoops[(_[_, expr___])?NPointFunctionQ] := Length[{expr}];
+
 GetExpression[selfEnergy_SelfEnergies`FSSelfEnergy, loops_:1] :=
     selfEnergy[[1 + loops]];