algebraic_fitting_function.h

#ifndef ALGEBRAIC_FITTING_FUNCTION_H
#define ALGEBRAIC_FITTING_FUNCTION_H
 
#include "ipde_fit_parameter.h"
#include <TH1.h>
namespace KVImpactParameters {
   class algebraic_fitting_function {
      class bce_fit_results {
      public:
         ipde_fit_parameter k0; // minimum of k at c=1
         ipde_fit_parameter alpha; // power of centrality
         ipde_fit_parameter gamma; // power of (1-pow(c,alpha))
         ipde_fit_parameter theta; // the (b-independent) fluctuation width Eqs(1),(2)
         ipde_fit_parameter kmax; // value of k at b=0 Eq(8)
         ipde_fit_parameter Xmax; // value of X at b=0
         double chisquare;
         double ndf;
 
         bce_fit_results() = default;
         bce_fit_results(const bce_fit_results& other)
            : k0(other.k0), alpha(other.alpha), gamma(other.gamma), theta(other.theta), kmax(other.kmax), Xmax(other.Xmax)
         {}
         bce_fit_results& operator=(const bce_fit_results& other)
         {
            if (&other != this) {
               k0 = (other.k0);
               alpha = (other.alpha);
               gamma = (other.gamma);
               theta = (other.theta);
               kmax = (other.kmax);
               Xmax = (other.Xmax);
            }
            return *this;
         }
         void Print(Option_t* = "") const
         {
            std:: cout << "   kmax = ";
            kmax.print();
            std::cout << "   k0 = ";
            k0.print();
            std::cout << "   alpha = ";
            alpha.print();
            std:: cout << "   gamma = ";
            gamma.print();
            std::cout << "   theta = ";
            theta.print();
            std::cout << std::endl;
         }
      } params, param_backup;
   public:
      algebraic_fitting_function() {}
      algebraic_fitting_function(const algebraic_fitting_function&);
      algebraic_fitting_function(double alph, double gam, double thet, double Xmi, double Xma);
      void backup_params()
      {
         param_backup = params;
      }
      void restore_params()
      {
         params = param_backup;
      }
      void normalise_shape_function()
      {
         params.kmax.value = 1. / params.theta.value;
         params.k0.value = 0;
         params.Xmax.value = 1;
      }
 
      int npar() const
      {
         return 5;
      }
      double k_cb(double cb) const;
      void fill_params_from_array(double* p);
      void fill_array_from_params(double* p) const;
      double theta() const
      {
         return params.theta.value;
      }
      void set_par_names(TF1& f) const;
      void set_initial_parameters(TH1* h, TF1& f);
      void print_fit_params() const
      {
         params.Print();
      }
      double meanX(double cb) const
      {
         return theta() * k_cb(cb);
      }
      double redVar(double) const
      {
         return theta();
      }
   };
 
}
 
#endif // ALGEBRAIC_FITTING_FUNCTION_H