CorrFitSmear1DCF.h

/*
 * CorrFitSmear1DCF.h
 *
 *  Created on: 4 sty 2019
 *      Author: Daniel Wielanek
 *      E-mail: daniel.wielanek@gmail.com
 *      Warsaw University of Technology, Faculty of Physics
 */
#ifndef HALCORRFITSMEAR1DCF_H_
#define HALCORRFITSMEAR1DCF_H_
 
#include "CorrFitKisiel.h"
 
#include <TMatrixD.h>
class TAxis;
namespace Hal {
  class CorrFitMapKstarRstar;
  class Femto1DCF;
  class Spline1D;
  class CorrFitSmear1DCF : public TObject {
    Bool_t fSeparateSmear      = {kFALSE};
    Bool_t fAutoFill           = {kFALSE};
    Int_t fThresBin            = {0};
    TF1* fFunc                 = {nullptr};
    TF1* fSplot                = {nullptr};
    TH2D* fSmearMatrix         = {nullptr};
    Femto1DCF* fCF             = {nullptr};
    CorrFitMapKstarRstar* fMap = {nullptr};
    Hal::Spline1D* fSpline     = {nullptr};
    Double_t GaussNS(Double_t* x, Double_t* params) const;
    Double_t CorrFunc(Double_t* x, Double_t* params) const;
    Double_t Splot(Double_t* x, Double_t* params) const;
    TMatrixD GetVec(const TH2D& vec, Int_t bin) const;
    TMatrixD GetVec(const TH1D& vec) const;
    TMatrixD Multiply(TMatrixD A, TMatrixD B) const;
    TMatrixD ReverseDenominator(TF1* smearfunc, TH2D* smearMatrix) const;
    std::vector<double> GetAutoFill(const TMatrixD& matrix, Bool_t lower) const;
    std::vector<double> GetGausNorms(TF1* f, TAxis* x) const;
    void NormalizeMatrix(TMatrixD& matrix) const;
    Bool_t CheckCompatiblityMap() const;
    Bool_t CheckCompatibilityCFMap() const;
 
  public:
    CorrFitSmear1DCF();
    void SetThresBin(Int_t no) { fThresBin = no; }
    TH2D RecalcSmearMatrix(TH2D& matrix);
    static TH2D TransposeHistogram(TH2D& matrix);
    void SmearSeparately(Bool_t smear = kTRUE) { fSeparateSmear = smear; };
    void AutoFillLastBin(Bool_t autofill = kTRUE) { fAutoFill = autofill; };
    void SetCF(Hal::Femto1DCF& cf);
    void SetMap(CorrFitMapKstarRstar* map) { fMap = map; };
    CorrFitMapKstarRstar* SmearMatrix(const TH2D& smearMatrix);
    CorrFitMapKstarRstar* SmearFunction(TF1* func);
    ~CorrFitSmear1DCF();
    TF1* GetFunc() { return fSplot; }
    ClassDef(CorrFitSmear1DCF, 1)
  };
}  // namespace Hal
#endif /* HALCORRFITSMEAR1DCF_H_ */