more readable formatting of self-energies

FlexibleSUSY · Oct 31, 2016 · abd4d30 · abd4d30
1 parent f9923e8
commit abd4d30
Showing 1 changed file with 10 additions and 4 deletions.
diff --git a/meta/SelfEnergies.m b/meta/SelfEnergies.m
@@ -419,6 +419,12 @@
     CreateFunctionName[tadpole] <>
     "(" <> DeclareFieldIndices[GetField[tadpole]] <> ") const";
 
+ExpressionToStringSequentially[expr_Plus, heads_, result_String] :=
+    StringJoin[(result <> " += " <> ExpressionToString[#,heads] <> ";\n")& /@ (List @@ expr)];
+
+ExpressionToStringSequentially[expr_, heads_, result_String] :=
+    result <> " = " <> ExpressionToString[expr, heads] <> ";\n";
+
 CreateNPointFunction[nPointFunction_, vertexRules_List] :=
     Module[{decl, expr, field, prototype, body, functionName},
            expr = GetExpression[nPointFunction];
@@ -427,14 +433,14 @@
            type = CConversion`CreateCType[CConversion`ScalarType[CConversion`complexScalarCType]];
            prototype = type <> " " <> functionName <> ";\n";
            decl = "\n" <> type <> " CLASSNAME::" <> functionName <> "\n{\n";
-           body = "const " <> type <> " result = " <>
-                  ExpressionToString[expr /.
+           body = type <> " result;\n\n" <>
+                  ExpressionToStringSequentially[expr /.
                                      vertexRules /.
                                      a_[List[i__]] :> a[i] /.
                                      ReplaceGhosts[FlexibleSUSY`FSEigenstates] /.
                                      C -> 1,
-                                     TreeMasses`GetParticles[]]  <>
-                  ";\n\nreturn result * oneOver16PiSqr;";
+                                     TreeMasses`GetParticles[], "result"]  <>
+                  "\nreturn result * oneOver16PiSqr;";
            body = IndentText[WrapLines[body]];
            decl = decl <> body <> "\n}\n";
            Return[{prototype, decl}];