RapidityFitBlastWave.h

Go to the documentation of this file.

#ifndef RAPIDITYFIT_H
#define RAPIDITYFIT_H
 
#include <cmath>
#include <string>
#include <vector>
 
struct RapidityFitBlastWaveParameter {
  bool toFit;
  double value;
  double error;
  double xmin, xmax;
  std::string name;
  RapidityFitBlastWaveParameter(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 RapidityFitBlastWaveParameters {
  RapidityFitBlastWaveParameter A, eta;
  double chi2ndf;
  RapidityFitBlastWaveParameters(double A_ = 200., double eta_ = 0.3)
    : A(RapidityFitBlastWaveParameter("A", true, A_, 1000., 0., 200000.))
    , eta(RapidityFitBlastWaveParameter("eta", true, eta_, 0.2, 0., 10.))
    , chi2ndf(1.) {
    // A = RapidityFitBlastWaveParameter("A", true, A_, 1000., 0., 200000.);
    // eta = RapidityFitBlastWaveParameter("eta", true, eta_, 0.2, 0., 10.);
  }
  RapidityFitBlastWaveParameter GetParamater(const std::string& name) const {
    if (A.name == name) return A;
    if (eta.name == name) return eta;
    return RapidityFitBlastWaveParameter();
  }
  void SetParameter(const std::string& name,
                    const RapidityFitBlastWaveParameter& param) {
    if (A.name == name) A = param;
    if (eta.name == name) eta = param;
  }
  void SetParameter(const std::string& name,
                    double val,
                    double err,
                    double xmin,
                    double xmax) {
    if (A.name == name)
      A = RapidityFitBlastWaveParameter(name, true, val, err, xmin, xmax);
    if (eta.name == name)
      eta = RapidityFitBlastWaveParameter(name, true, val, err, xmin, xmax);
  }
  void SetParameterFitFlag(const std::string& name, bool toFit) {
    if (A.name == name) A.toFit = toFit;
    if (eta.name == name) eta.toFit = toFit;
  }
};
 
namespace RapidityFitBlastWaveNamespace {
  double dndy(double y, double A, double eta, double T, double mass);
};
 
class RapidityFitBlastWave {
  std::vector<double> y, dndy, err;
  RapidityFitBlastWaveParameters Parameters;
  double fT, fMass;
 
public:
  RapidityFitBlastWave(const std::vector<double>& y_,
                       const std::vector<double>& dndy_,
                       const std::vector<double>& err_,
                       double T,
                       double mass,
                       double A_   = 200.,
                       double eta_ = 0.3)
    : y(y_), dndy(dndy_), err(err_), Parameters(A_, eta_), fT(T), fMass(mass) {}
  ~RapidityFitBlastWave(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_;
  }
 
  RapidityFitBlastWaveParameters PerformFit();
};
 
#endif