ThermalModelFitParameters.h

Go to the documentation of this file.

/*
 * Thermal-FIST package
 * 
 * Copyright (c) 2019 Volodymyr Vovchenko
 *
 * GNU General Public License (GPLv3 or later)
 */
#ifndef THERMALMODELFITPARAMETERS_H
#define THERMALMODELFITPARAMETERS_H

#include <string>

#include "HRGBase/ThermalModelBase.h"

namespace thermalfist {

  struct FitParameter {
    double value;       

    double error;       
    double errp, errm;  
    double xmin, xmax;  
    bool toFit;         
    std::string name;   

    FitParameter(std::string namep = "", bool toFitp = true, double val = 1., double err = 0., double vmin = -2., double vmax = 2.) :
      toFit(toFitp), value(val), error(err), errp(err), errm(err), xmin(vmin), xmax(vmax), name(namep) {
    }
  };

  struct ThermalModelFitParametersExtended {
    FitParameter T;
    FitParameter muB;
    FitParameter muS;
    FitParameter muQ;
    FitParameter muC;
    FitParameter gammaq;
    FitParameter gammaS;
    FitParameter gammaC;
    FitParameter R;
    FitParameter Rc;

    FitParameter Tkin;

    double chi2ndf;

    FitParameter nH, rhoB, rhoQ, en, entropy, pressure;
    ThermalModelFitParametersExtended() { }
    ThermalModelFitParametersExtended(ThermalModelBase *model);
  };

  struct ThermalModelFitParameters {
    static const int ParameterCount = 11;

    bool GCE;  

    FitParameter T, muB, muS, muQ, muC, gammaq, gammaS, gammaC, R, Rc;

    FitParameter Tkin;

    std::vector<FitParameter*> ParameterList;

    int B, S, Q, C;

    double chi2; 
    
    double chi2ndf;

    int ndf;

    ThermalModelFitParameters(const ThermalModelParameters &params = ThermalModelParameters());

    ThermalModelFitParameters(const ThermalModelFitParameters& op);

    ThermalModelFitParameters& operator=(const ThermalModelFitParameters& op);

    void FillParameterList();

    int IndexByName(const std::string& name) const;


    FitParameter GetParameter(const std::string& name) const;
    FitParameter& GetParameter(const std::string& name);


    FitParameter GetParameter(const int index) const;
    FitParameter& GetParameter(const int index);

    void SetParameter(const std::string& name, const FitParameter& param);

    void SetParameter(const std::string& name, double val, double err, double xmin, double xmax);

    void SetParameterValue(const std::string& name, double value);

    void SetParameterFitFlag(const std::string& name, bool toFit);

    ThermalModelParameters GetThermalModelParameters();
  };

} // namespace thermalfist

#endif