Store boundary values in model class

FlexibleSUSY · Dec 19, 2016 · 2614736 · 2614736
1 parent 12e7585
commit 2614736
Show file tree

Hide file tree

Showing 4 changed files with 42 additions and 1 deletion.
diff --git a/meta/FlexibleSUSY.m b/meta/FlexibleSUSY.m
@@ -1277,11 +1277,16 @@ corresponding tadpole is real or imaginary (only in models with CP
     WriteOut`ReplaceInFiles[files, { Sequence @@ GeneralReplacementRules[] }];
 
 WriteSemiAnalyticModelClass[semiAnalyticBCs_List, semiAnalyticSolns_List, files_List] :=
-    Module[{semiAnalyticSolutionsDefs = "", semiAnalyticSolutionsInit = ""},
+    Module[{semiAnalyticSolutionsDefs = "", semiAnalyticSolutionsInit = "",
+            boundaryValuesDefs = "", boundaryValuesInit = ""},
            semiAnalyticSolutionsDefs = SemiAnalytic`CreateSemiAnalyticSolutionsDefinitions[semiAnalyticSolns];
            semiAnalyticSolutionsInit = SemiAnalytic`CreateSemiAnalyticSolutionsInitialization[semiAnalyticSolns];
+           boundaryValuesDefs = SemiAnalytic`CreateBoundaryValuesDefinitions[semiAnalyticSolns];
+           boundaryValuesInit = SemiAnalytic`CreateBoundaryValuesInitialization[semiAnalyticSolns];
            WriteOut`ReplaceInFiles[files, { "@semiAnalyticSolutionsDefs@" -> IndentText[WrapLines[semiAnalyticSolutionsDefs]],
                                             "@semiAnalyticSolutionsInit@" -> IndentText[WrapLines[semiAnalyticSolutionsInit]],
+                                            "@boundaryValuesDefs@" -> IndentText[WrapLines[boundaryValuesDefs]],
+                                            "@boundaryValuesInit@" -> IndentText[WrapLines[boundaryValuesInit]],
                                             Sequence @@ GeneralReplacementRules[] }];
           ];
 

diff --git a/meta/SemiAnalytic.m b/meta/SemiAnalytic.m
@@ -20,6 +20,8 @@
 
 CreateSemiAnalyticSolutionsDefinitions::usage="";
 CreateSemiAnalyticSolutionsInitialization::usage="";
+CreateBoundaryValuesDefinitions::usage="";
+CreateBoundaryValuesInitialization::usage="";
 
 Begin["`Private`"];
 
@@ -203,6 +205,7 @@
                       {_, _},
                       rules = GetBoundaryValueFromSetting[noTemp[[i,1]], noTemp[[i,2]]];
                       finalRules = Last[(finalRules = Utils`AppendOrReplaceInList[finalRules, #, test])& /@ rules];,
+                      (* @todo intermediate boundary values can be added to a list of extra parameters *)
                       FlexibleSUSY`FSMinimize[__],
                       fixedPars = Constraint`FindFixedParametersFromConstraint[{noTemp[[i]]}];
                       boundaryValues = Symbol[CConversion`ToValidCSymbolString[#] <> "MinSol"]& /@ fixedPars;
@@ -461,6 +464,25 @@
            Return[def];
           ];
 
+GetBoundaryValueParameters[solutions_List] :=
+    DeleteDuplicates[Flatten[(Parameters`FindAllParameters[GetBasis[#]])& /@ solutions]];
+
+CreateBoundaryValuesDefinitions[solutions_List] :=
+    Module[{boundaryValues, defns},
+           boundaryValues = GetBoundaryValueParameters[solutions];
+           defns = (CConversion`CreateCType[Parameters`GetType[#]]
+                    <> " basis_" <> CConversion`ToValidCSymbolString[#])& /@ boundaryValues;
+           Utils`StringJoinWithSeparator[defns, ";\n"] <> ";\n"
+          ];
+
+CreateBoundaryValuesInitialization[solutions_List] :=
+    Module[{boundaryValues, def = ""},
+           boundaryValues = GetBoundaryValueParameters[solutions];
+           (def = def <> "," <> CConversion`CreateDefaultConstructor["basis_" <> CConversion`ToValidCSymbolString[#],
+                                                                     Parameters`GetType[#]])& /@ boundaryValues;
+           Return[def];
+          ];
+
 DependsAtMostOn[num_?NumericQ, pars_List] := True;
 DependsAtMostOn[True, pars_List] := True;
 DependsAtMostOn[False, pars_List] := True;

diff --git a/templates/semi_analytic_model.cpp.in b/templates/semi_analytic_model.cpp.in
@@ -41,6 +41,7 @@ using namespace @ModelName@_info;
 CLASSNAME::@ModelName@(const @ModelName@_input_parameters& input_)
    : @ModelName@_mass_eigenstates(input_)
 @semiAnalyticSolutionsInit@
+@boundaryValuesInit@
 {
 }
 
@@ -78,6 +79,13 @@ void CLASSNAME::set_precision(double p)
    @ModelName@_mass_eigenstates::set_precision(p);
 }
 
+void CLASSNAME::set_model_parameters_to_boundary_values(@ModelName@_soft_parameters& model) const
+{
+@applySemiAnalyticBCs@
+}
+
+@calculateCoeffFunctions@
+
 std::ostream& operator<<(std::ostream& ostr, const @ModelName@<Semi_analytic>& model)
 {
    model.print(ostr);

diff --git a/templates/semi_analytic_model.hpp.in b/templates/semi_analytic_model.hpp.in
@@ -59,9 +59,15 @@ public:
    virtual void print(std::ostream& out = std::cout) const;
    virtual void set_precision(double);
 
+   void calculate_coefficients(double);
+
 private:
    // semi-analytic solution coefficients
 @semiAnalyticSolutionsDefs@
+   // boundary values
+@boundaryValuesDefs@
+
+   void set_parameters_to_boundary_values(@ModelName@_soft_parameters&) const;
 };
 
 std::ostream& operator<<(std::ostream&, const @ModelName@<Semi_analytic>&);