CorrFit.h

/*
 * CorrFit.h
 *
 *  Created on: 24 mar 2018
 *      Author: Daniel Wielanek
 *      E-mail: daniel.wielanek@gmail.com
 *      Warsaw University of Technology, Faculty of Physics
 */
#ifndef HALCORRFIT_H_
#define HALCORRFIT_H_
 
#include "Array.h"
#include "FitParam.h"
 
#include <Rtypes.h>
#include <RtypesCore.h>
#include <TString.h>
#include <utility>
#include <vector>
 
class TLegend;
class TLegendEntry;
class TF1;
class TVirtualPad;
 
namespace Hal {
  class CorrFitGUI;
  class CorrFitPainter;
  class CorrFit : public TObject {
    friend class CorrFitGUI;
 
  private:
    Color_t fLineColor;
    Style_t fLineStyle;
    Width_t fLineWidth;
    Int_t fParametersNo;
 
  public:
    enum EMinFunc {
      kChi, 
      kLog, 
      kChi2 
    };
    enum EMinAlgo {
      kMinuitMigrad,
      kMinuitSimplex,
      kMinuitCombined,
      kMinuitScan,
      kMinuitFumili,
      kGLSMultiMinConjungateFR,
      kGLSMultiMinConjugatePR,
      kGLSMultiMinBFGS,
      kGLSMultiMinBFGS2,
      kGLSMultiMinSteppestDescent,
      kGLSMultiFit,
      kGLSSimAn,
      kDefaultAlgo,
      kHalScan,
      kHalAnt,
      kHalScanMigrad,
      kHalScanScan
    };
    enum ECalcOption {
      kSimple,      
      kExtrapolated 
    };
 
  protected:
    ECalcOption fBinCalc;
    EMinFunc fMinFunc;
    EMinAlgo fMinAlgo;
    CorrFitPainter* fPainter = {nullptr};
    std::vector<TString> AlgoToOptions(EMinAlgo algo) const;
    Int_t fNDF;
    Double_t fThreshold;
    std::vector<FitParam> fParameters;
    std::vector<Int_t> fFitOrder;
    std::vector<std::pair<TF1*, TVirtualPad*>> fDrawFunc;  
    mutable Double_t* fTempParamsEval;  // [fParametersNo]
    void CheckOrder();
    void ChangeParametersNo();
    virtual void ParametersChanged() const {};
    Color_t GetLineColor() const { return fLineColor; };
    Style_t GetLineStyle() const { return fLineStyle; };
    Width_t GetLineWidth() const { return fLineWidth; };
    TF1* GetTF1(Int_t no) const { return fDrawFunc[no].first; };
    std::vector<std::pair<TF1*, TVirtualPad*>> GetDrawFunctions() const { return fDrawFunc; };
    void OverwriteParam(Int_t par, Double_t val) { fParameters[par].SetFittedValue(val); }
    void CopyParamsToTF1(TF1* f, Bool_t copyNumPar = kTRUE, Bool_t copyGrapPar = kTRUE) const;
 
    void SetErrorsNegative();
    void SetFitOrder(std::initializer_list<Int_t> order);
 
  public:
    CorrFit(Int_t parameters_no = 2);
    void SetParLimits(Int_t par, Double_t min, Double_t max);
    void FixParameter(Int_t par, Double_t val);
    void SetParameterName(Int_t par, TString name);
    void SetLineColor(Color_t color) { fLineColor = color; };
    void SetLineStyle(Style_t style) { fLineStyle = style; };
    void SetLineWidth(Width_t width) { fLineWidth = width; };
    void SetThreshold(Double_t thres) { fThreshold = thres; };
    void PrintFitResults() const;
    Bool_t IsParFixed(Int_t par) const;
    Bool_t IsParDiscrete(Int_t par) const;
    Double_t GetParMin(Int_t par) const;
    Double_t GetParMax(Int_t par) const;
    Double_t GetParError(Int_t par) const;
    Double_t GetParameter(Int_t par) const;
    const FitParam& GetParamConf(Int_t par) const { return fParameters[par]; }
    Double_t GetNDF() const { return fNDF; };
    Int_t GetParametersNo() const { return fParametersNo; };
    Int_t GetParameterIndex(TString name) const;
    TString GetParameterName(Int_t no) const { return fParameters[no].GetParName(); };
    virtual void SetMinimizer(EMinAlgo alg) { fMinAlgo = alg; };
    void SetMinimizedFunc(EMinFunc f) { fMinFunc = f; };
    void SetCalculationOption(ECalcOption f) { fBinCalc = f; }
    CorrFitPainter* GetPainter() const { return fPainter; }
    virtual ~CorrFit();
    ClassDef(CorrFit, 2)
  };
}  // namespace Hal
 
#endif /* HALCORRFIT_H_ */