add the apply() function to the constraint

in order to make it callable in the initial guesser Performing an initial guess via run_to() and apply() yields a very good first guess.
FlexibleSUSY · Jul 15, 2015 · 4647dc0 · 4647dc0
1 parent 2653da4
commit 4647dc0
Show file tree

Hide file tree

Showing 8 changed files with 72 additions and 1 deletion.
diff --git a/meta/FlexibleSUSY.m b/meta/FlexibleSUSY.m
@@ -641,7 +641,8 @@
 
 WriteRootConstraintClass[condition_, settings_List, scaleFirstGuess_,
                          {minimumScale_, maximumScale_}, files_List] :=
-   Module[{calculateFunction = "", calculateScale, scaleGuess,
+   Module[{applyConstraint = "", calculateFunction = "",
+           calculateScale, scaleGuess,
            restrictScale,
            numberOfFixedParameters,
            setDRbarYukawaCouplings,
@@ -654,6 +655,7 @@
           Constraint`SetBetaFunctions[GetBetaFunctions[]];
           numberOfFixedParameters = Constraint`CalculateNumberOfParameters[settings];
           calculateFunction = Constraint`CalculateDifferenceFunction[settings];
+          applyConstraint = Constraint`ApplyConstraints[settings];
           calculateScale  = Constraint`CalculateScale[condition, "scale"];
           scaleGuess      = Constraint`CalculateScale[scaleFirstGuess, "initial_scale_guess"];
           restrictScale   = Constraint`RestrictScale[{minimumScale, maximumScale}];
@@ -676,6 +678,7 @@
             ];
           WriteOut`ReplaceInFiles[files,
                  { "@numberOfFixedParameters@" -> numberOfFixedParameters,
+                   "@applyConstraint@"      -> IndentText[WrapLines[applyConstraint]],
                    "@calculateFunction@"    -> IndentText[WrapLines[calculateFunction]],
                    "@calculateScale@"       -> IndentText[WrapLines[calculateScale]],
                    "@scaleGuess@"           -> IndentText[WrapLines[scaleGuess]],

diff --git a/src/root_constraint.hpp b/src/root_constraint.hpp
@@ -34,6 +34,7 @@ template<>
 class Constraint<Root> {
 public:
    virtual ~Constraint() {}
+   virtual void apply() = 0;
    virtual Eigen::ArrayXd calculate_function(const Eigen::ArrayXd&) = 0;
    virtual std::size_t get_number_of_fixed_parameters() const = 0;
    virtual double get_scale() const = 0;        ///< get scale where to apply

diff --git a/templates/root_high_scale_constraint.cpp.in b/templates/root_high_scale_constraint.cpp.in
@@ -70,6 +70,40 @@ namespace flexiblesusy {
 {
 }
 
+void @ModelName@_high_scale_constraint<Root>::apply()
+{
+   assert(model && "Error: @ModelName@_high_scale_constraint::apply():"
+          " model pointer must not be zero");
+
+   if (std::fabs(model->get_@hyperchargeCoupling@()) > @perturbativityThreshold@) {
+#ifdef ENABLE_VERBOSE
+      ERROR("@ModelName@_high_scale_constraint: Non-perturbative gauge "
+            "coupling @hyperchargeCoupling@ = " << model->get_@hyperchargeCoupling@());
+#endif
+      model->set_@hyperchargeCoupling@(@perturbativityThreshold@);
+   }
+   if (std::fabs(model->get_@leftCoupling@()) > @perturbativityThreshold@) {
+#ifdef ENABLE_VERBOSE
+      ERROR("@ModelName@_high_scale_constraint: Non-perturbative gauge "
+            "coupling @leftCoupling@ = " << model->get_@leftCoupling@());
+#endif
+      model->set_@leftCoupling@(@perturbativityThreshold@);
+   }
+   if (std::fabs(model->get_@strongCoupling@()) > @perturbativityThreshold@) {
+#ifdef ENABLE_VERBOSE
+      ERROR("@ModelName@_high_scale_constraint: Non-perturbative gauge "
+            "coupling @strongCoupling@ = " << model->get_@strongCoupling@());
+#endif
+      model->set_@strongCoupling@(@perturbativityThreshold@);
+   }
+
+   update_scale();
+
+@applyConstraint@
+
+   non_perturbative();
+}
+
 Eigen::ArrayXd @ModelName@_high_scale_constraint<Root>::calculate_function(
    const Eigen::ArrayXd& arg_)
 {

diff --git a/templates/root_high_scale_constraint.hpp.in b/templates/root_high_scale_constraint.hpp.in
@@ -38,6 +38,7 @@ public:
    @ModelName@_high_scale_constraint();
    @ModelName@_high_scale_constraint(@ModelName@<Root>*);
    virtual ~@ModelName@_high_scale_constraint();
+   virtual void apply();
    virtual Eigen::ArrayXd calculate_function(const Eigen::ArrayXd&);
    virtual std::size_t get_number_of_fixed_parameters() const;
    virtual double get_scale() const;

diff --git a/templates/root_low_scale_constraint.cpp.in b/templates/root_low_scale_constraint.cpp.in
@@ -97,6 +97,24 @@ namespace flexiblesusy {
 {
 }
 
+void @ModelName@_low_scale_constraint<Root>::apply()
+{
+   assert(model && "Error: @ModelName@_low_scale_constraint::apply():"
+          " model pointer must not be zero");
+
+   model->calculate_DRbar_masses();
+   update_scale();
+   calculate_DRbar_gauge_couplings();
+
+@applyConstraint@
+
+   model->set_@hyperchargeCoupling@(new_@hyperchargeCoupling@);
+   model->set_@leftCoupling@(new_@leftCoupling@);
+   model->set_@strongCoupling@(new_@strongCoupling@);
+
+   recalculate_mw_pole();
+}
+
 Eigen::ArrayXd @ModelName@_low_scale_constraint<Root>::calculate_function(
    const Eigen::ArrayXd& arg_)
 {

diff --git a/templates/root_low_scale_constraint.hpp.in b/templates/root_low_scale_constraint.hpp.in
@@ -40,6 +40,7 @@ public:
    @ModelName@_low_scale_constraint();
    @ModelName@_low_scale_constraint(@ModelName@<Root>*, const QedQcd&);
    virtual ~@ModelName@_low_scale_constraint();
+   virtual void apply();
    virtual Eigen::ArrayXd calculate_function(const Eigen::ArrayXd&);
    virtual std::size_t get_number_of_fixed_parameters() const;
    virtual double get_scale() const;

diff --git a/templates/root_susy_scale_constraint.cpp.in b/templates/root_susy_scale_constraint.cpp.in
@@ -65,6 +65,18 @@ namespace flexiblesusy {
 {
 }
 
+void @ModelName@_susy_scale_constraint<Root>::apply()
+{
+   assert(model && "Error: @ModelName@_susy_scale_constraint::apply():"
+          " model pointer must not be zero");
+
+   model->calculate_DRbar_masses();
+   update_scale();
+
+   // apply user-defined susy scale constraints
+@applyConstraint@
+}
+
 Eigen::ArrayXd @ModelName@_susy_scale_constraint<Root>::calculate_function(
    const Eigen::ArrayXd& arg_)
 {

diff --git a/templates/root_susy_scale_constraint.hpp.in b/templates/root_susy_scale_constraint.hpp.in
@@ -38,6 +38,7 @@ public:
    @ModelName@_susy_scale_constraint();
    @ModelName@_susy_scale_constraint(@ModelName@<Root>*);
    virtual ~@ModelName@_susy_scale_constraint();
+   virtual void apply();
    virtual Eigen::ArrayXd calculate_function(const Eigen::ArrayXd&);
    virtual std::size_t get_number_of_fixed_parameters() const;
    virtual double get_scale() const;