use SARAH` context and make result head configurable

FlexibleSUSY · Mar 19, 2017 · 984dc98 · 984dc98
1 parent 3eb6bdf
commit 984dc98
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/meta/SelfEnergies2L.m b/meta/SelfEnergies2L.m
@@ -69,10 +69,10 @@
 
 SumTadpoleType[{type_, diags_}] := Total[CreateTadpoleDiag /@ diags];
 
-ConvertSarah2LDiagramList[tad_List] := 
-    Total[SumTadpoleType /@ tad] //. {
-        (m : (Mass | SARAH`Mass2))[(bar | Susyno`LieGroups`conj)[p_], idx___] :> m[p, idx],
-        (m : (Mass | SARAH`Mass2))[p_, idx__] :> m[p[{idx}]],
+ConvertSarah2LDiagramList[tad_List, head_:Total] :=
+    head[SumTadpoleType /@ tad] //. {
+        (m : (SARAH`Mass | SARAH`Mass2))[(bar | Susyno`LieGroups`conj)[p_], idx___] :> m[p, idx],
+        (m : (SARAH`Mass | SARAH`Mass2))[p_, idx__] :> m[p[{idx}]],
         C[p__] :> Cp[p]
     };