FitResult.h

Go to the documentation of this file.

#ifndef AMPGEN_FITRESULT_H
#define AMPGEN_FITRESULT_H
 
#include <TMatrixD.h>
 
#include "AmpGen/FitFraction.h"
#include "AmpGen/MinuitParameter.h"
#include "AmpGen/Utilities.h"
 
namespace AmpGen {
  class Minimiser;
  class LinearErrorPropagator;
 
  class FitResult {
  public:
    ~FitResult() {};
    FitResult();
    explicit FitResult(const FitResult &other);
    explicit FitResult(const std::string &filename);
    explicit FitResult(const Minimiser &mini);
    FitResult(MinuitParameterSet &mps, const TMatrixD &covMini);
 
    void addObservable(const std::string &name, const double &F);
    void addChi2(const double &chi2, const double &nBins);
    void addFractions(const std::vector<FitFraction> &fractions);
    void addFraction(const std::string &name, const double &frac, const double &err);
    void setCov(const size_t &x, const size_t &y, const double &F);
    void writeToFile(const std::string &fname);
    void clearFitFractions();
 
    bool readFile(const std::string &fname);
 
    double chi2() const;
    double LL() const;
    double Edm() const;
    double NCalls() const;
    double dof() const;
    double cov(const size_t &x, const size_t &y) const;
    double cov(const std::string &x, const std::string &y) const;
    double correlation(const std::string &x, const std::string &y) const;
 
    int status() const;
    int nParam() const;
    int nBins() const;
 
    std::map<std::string, double> observables() const;
    MinuitParameterSet *mps() const;
 
    std::vector<FitFraction> fitFractions() const;
    std::vector<MinuitParameter *> parameters() const;
    std::vector<MinuitParameter *> floating(const bool &extended = false) const;
    TMatrixD cov() const;
 
    void print() const;
 
    TMatrixD getReducedCovariance(const bool &extended = false) const;
    LinearErrorPropagator getErrorPropagator(const bool &extended = false) const;
 
    void writeOptions(const std::string &output, const std::string &input = "");
    void writeOptions(std::ostream &, const std::string &input = "");
 
  private:
    MinuitParameterSet *m_mps = {nullptr};
    double m_chi2 = {0};
    double m_LL = {-999};
    double m_Edm = {-999};
    double m_NCalls = {-999};
    double m_nBins = {0};
    double m_nParam = {0};
    int m_status = {-1};
    bool m_fitted = {false};
    std::map<std::string, double> m_observables;
    std::vector<FitFraction> m_fitFractions;
    TMatrixD m_covarianceMatrix;
    std::map<std::string, unsigned int> m_covMapping;
 
    void addToParameters(const std::string &line);
    void addToObservables(const std::string &line);
    void setFitQuality(const std::string &line);
  };
} // namespace AmpGen
 
#endif