RapidityFit.h

Go to the documentation of this file.

#ifndef RAPIDITYFIT_H
#define RAPIDITYFIT_H
 
#include <cmath>
#include <string>
#include <vector>
 
struct RapidityFitParameter {
  bool toFit;
  double value;
  double error;
  double xmin, xmax;
  std::string name;
  RapidityFitParameter(std::string name_ = "",
                       bool toFit_       = true,
                       double val        = 1.,
                       double err        = 0.,
                       double vmin       = -2.,
                       double vmax       = 2.)
    : toFit(toFit_)
    , value(val)
    , error(err)
    , xmin(vmin)
    , xmax(vmax)
    , name(name_) {}
};
 
struct RapidityFitParameters {
  RapidityFitParameter A, sig, yav;
  double chi2ndf;
  RapidityFitParameters(double A_ = 200., double sig_ = 1., double yav_ = 0.)
    : A(RapidityFitParameter("A", true, A_, 100., 0., 2000.))
    , sig(RapidityFitParameter("sig", true, sig_, 0.5, 0., 10.))
    , yav(RapidityFitParameter("yav", true, yav_, 0.3, 0., 10.))
    , chi2ndf(1.) {
    // A = RapidityFitParameter("A", true, A_, 100., 0., 2000.);
    // sig = RapidityFitParameter("sig", true, sig_, 0.5, 0., 10.);
    // yav = RapidityFitParameter("yav", true, yav_, 0.3, 0., 10.);
  }
  RapidityFitParameter GetParamater(const std::string& name) const {
    if (A.name == name) return A;
    if (sig.name == name) return sig;
    if (yav.name == name) return yav;
    return RapidityFitParameter();
  }
  void SetParameter(const std::string& name,
                    const RapidityFitParameter& param) {
    if (A.name == name) A = param;
    if (sig.name == name) sig = param;
    if (yav.name == name) yav = param;
  }
  void SetParameter(const std::string& name,
                    double val,
                    double err,
                    double xmin,
                    double xmax) {
    if (A.name == name)
      A = RapidityFitParameter(name, true, val, err, xmin, xmax);
    if (sig.name == name)
      sig = RapidityFitParameter(name, true, val, err, xmin, xmax);
    if (yav.name == name)
      yav = RapidityFitParameter(name, true, val, err, xmin, xmax);
  }
  void SetParameterFitFlag(const std::string& name, bool toFit) {
    if (A.name == name) A.toFit = toFit;
    if (sig.name == name) sig.toFit = toFit;
    if (yav.name == name) yav.toFit = toFit;
  }
  RapidityFitParameters GetRapidityFitParameters() {
    return RapidityFitParameters(A.value, sig.value, yav.value);
  }
};
 
namespace RapidityFitNamespace {
  double TwoGauss(double y, double A, double sig, double y0);
};
 
class RapidityFit {
  std::vector<double> y, dndy, err;
  RapidityFitParameters Parameters;
 
public:
  RapidityFit(const std::vector<double>& y_,
              const std::vector<double>& dndy_,
              const std::vector<double>& err_,
              double A_   = 200.,
              double sig_ = 1.,
              double yav_ = 0.2)
    : y(y_), dndy(dndy_), err(err_), Parameters(A_, sig_, yav_) {}
  ~RapidityFit(void);
  void SetFitFlag(const std::string& name, bool flag) {
    Parameters.SetParameterFitFlag(name, flag);
  }
  void SetData(const std::vector<double>& y_,
               const std::vector<double>& dndy_,
               const std::vector<double>& err_) {
    y    = y_;
    dndy = dndy_;
    err  = err_;
  }
 
  RapidityFitParameters PerformFit();
};
 
#endif