CorrFit3DCF.h

/*
 * CorrFit3DCF.h
 *
 *  Created on: 08-04-2015
 *      Author: Daniel Wielanek
 *      E-mail: daniel.wielanek@gmail.com
 *      Warsaw University of Technology, Faculty of Physics
 */
 
#ifndef HALCORRFIT3DCF_H_
#define HALCORRFIT3DCF_H_
 
#include "CorrFitFunc3D.h"
 
#include "CorrFitMask.h"
class TH3;
class TH1D;
namespace Hal {
  class CorrFitMask3D;
  class CorrFitMath3DCF;
  class CorrFitSHCF;
  class CorrFit3DCFPainter;
  class CorrFit3DCF : public CorrFitFunc3D {
 
  public:
  private:
    enum class EDrawMode { kNormal, kDiagonal1, kDiagonal2 };
    friend class CorrFitMath3DCF;
    friend class CorrFitSHCF;
    friend class CorrFit3DCFPainter;
    Array_1<Double_t> fXbins;
    Array_1<Double_t> fYbins;
    Array_1<Double_t> fZbins;
    Double_t fXBinf  = {0};
    Double_t fYBinf  = {0};
    Double_t fZBinf  = {0};
    Double_t fXAxisf = {0};
    Double_t fYAxisf = {0};
    Double_t fZAxisf = {0};
    Double_t GetFunX(Double_t* x, Double_t* params) const;
    Double_t GetFunY(Double_t* x, Double_t* params) const;
    Double_t GetFunZ(Double_t* x, Double_t* params) const;
    Double_t GetFunXYpp(Double_t* x, Double_t* params) const;
    Double_t GetFunXYpm(Double_t* x, Double_t* params) const;
    Double_t GetFunXZpp(Double_t* x, Double_t* params) const;
    Double_t GetFunXZpm(Double_t* x, Double_t* params) const;
    Double_t GetFunYZpp(Double_t* x, Double_t* params) const;
    Double_t GetFunYZpm(Double_t* x, Double_t* params) const;
    Double_t GetFunXYZppp(Double_t* x, Double_t* params) const;
    Double_t GetFunXYZpmp(Double_t* x, Double_t* params) const;
    Double_t GetFunXYZppm(Double_t* x, Double_t* params) const;
    Double_t GetFunXYZpmm(Double_t* x, Double_t* params) const;
    Double_t GetFunXY2d(Double_t* x, Double_t* params) const;
    Double_t GetFunXZ2d(Double_t* x, Double_t* params) const;
    Double_t GetFunYZ2d(Double_t* x, Double_t* params) const;
    Double_t GetScaledValue(Double_t x, Double_t* params) const;
    void SetParametersToTF1(TF1* f) const;
    void Calculatef(Double_t width);
 
  protected:
    mutable Int_t fBinX = {0}, fBinY = {0}, fBinZ = {0};
 
    virtual void MakePainter(TString options);
 
    CorrFitMask3D* GetMask() const { return (CorrFitMask3D*) fMask; };
    virtual void RecalculateFunction() const {};
    virtual void RecalculateSmoothFunction() const;
    void SetErrors(TH1* num, const TH1* den) const;
    virtual void EstimateActiveBins();
    void DrawPureFunction(TString option);
    void DrawFunctionWithCF(TString option);
    void DrawOther(TString option);
    void DrawDiagonalOther(TString option);
    void DrawDiagonalWithCF(TString option);
    void PrepareRaw();
    TF1* GetDrawableFunc(TString option);
    double GetChiTFD(const double* par) const;
    double GetChiTF(const double* par) const;
    double GetLogTFD(const double* par) const;
    Double_t EvalDenominator(Double_t x, Double_t y, Double_t z) const;
    virtual Double_t EvalCF(const Double_t* x, const Double_t* params) const;
    virtual Double_t CalculateCF(const Double_t* x, const Double_t* params) const = 0;
    virtual Double_t GetNumericalError(Int_t /*x*/, Int_t /*y*/, Int_t /*z*/) const { return 0; };
    CorrFit3DCF(e3DMode mode, Int_t parameters = 3);
 
  public:
    CorrFit3DCF(Int_t parameters = 4);
    void SetFuncRange(Double_t x_min, Double_t x_max, Double_t y_min, Double_t y_max, Double_t z_min, Double_t z_max);
    void SetRadiusLimits(Double_t min, Double_t max);
 
    void SetFittingMask(const CorrFitMask& map);
    Double_t Eval(Double_t x, Double_t y, Double_t z);
    virtual ~CorrFit3DCF();
    ClassDef(CorrFit3DCF, 1)
  };
}  // namespace Hal
#endif /* HALCORRFIT3DCF_H_ */