generate 2-loop self-energy and tadpole functions

FlexibleSUSY · Mar 22, 2017 · 5841471 · 5841471
1 parent c588073
commit 5841471
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 8 deletions.
diff --git a/meta/FlexibleSUSY.m b/meta/FlexibleSUSY.m
@@ -1175,6 +1175,9 @@ corresponding tadpole is real or imaginary (only in models with CP
               calculateTwoLoopTadpoles  = SelfEnergies`FillArrayWithTwoLoopTadpoles[SARAH`HiggsBoson, "tadpole", "-"];
               calculateTwoLoopTadpolesNoStruct = SelfEnergies`FillArrayWithTwoLoopTadpoles[SARAH`HiggsBoson, "tadpole", "+"];
               {thirdGenerationHelperPrototypes, thirdGenerationHelperFunctions} = TreeMasses`CreateThirdGenerationHelpers[];
+              ,
+              calculateTwoLoopTadpoles     = SelfEnergies`FillArrayWithLoopTadpoles[2, higgsToEWSBEqAssociation, "tadpole", "-"];
+              calculateTwoLoopTadpolesNoStruct = SelfEnergies`FillArrayWithLoopTadpoles[2, higgsToEWSBEqAssociation, "tadpole", "+"];
              ];
            If[FlexibleSUSY`UseHiggs2LoopSM === True,
               {twoLoopSelfEnergyPrototypes, twoLoopSelfEnergyFunctions} = SelfEnergies`CreateTwoLoopSelfEnergiesSM[{SARAH`HiggsBoson}];

diff --git a/meta/SelfEnergies.m b/meta/SelfEnergies.m
@@ -529,7 +529,7 @@ therefore not be accessed in the form Glu(gO2).
           ];
 
 CreateNPointFunctions[nPointFunctions_List, vertexRules_List] :=
-    Module[{prototypes = "", defs = "", vertexFunctionNames = {}, p, d,
+    Module[{prototypes = "", defs = "", vertexFunctionNames = {}, p, d, l,
             relevantVertexRules},
            (* create coupling functions for all vertices in the list *)
            Print["Converting vertex functions ..."];
@@ -540,13 +540,15 @@ therefore not be accessed in the form Glu(gO2).
            Print["Converting self energies ..."];
            Utils`StartProgressBar[Dynamic[k], Length[nPointFunctions]];
            For[k = 1, k <= Length[nPointFunctions], k++,
-               Utils`UpdateProgressBar[k, Length[nPointFunctions]];
-               {p,d} = CreateNPointFunction[nPointFunctions[[k]], vertexFunctionNames, 1];
-               prototypes = prototypes <> p;
-               defs = defs <> d;
-               {p,d} = CreateNPointFunctionMatrix[nPointFunctions[[k]], 1];
-               prototypes = prototypes <> p;
-               defs = defs <> d;
+               For[loops = 1, loops <= NumberOfLoops[nPointFunctions[[k]]], loops++,
+                   Utils`UpdateProgressBar[k, Length[nPointFunctions]];
+                   {p,d} = CreateNPointFunction[nPointFunctions[[k]], vertexFunctionNames, loops];
+                   prototypes = prototypes <> p;
+                   defs = defs <> d;
+                   {p,d} = CreateNPointFunctionMatrix[nPointFunctions[[k]], loops];
+                   prototypes = prototypes <> p;
+                   defs = defs <> d;
+                  ];
               ];
            Utils`StopProgressBar[Length[nPointFunctions]];
            {prototypes, defs}