FemtoSHCF.h

/*
 * FemtoSHCF.h
 *
 *  Created on: 28 kwi 2016
 *      Author: Daniel Wielanek
 *      E-mail: daniel.wielanek@gmail.com
 *      Warsaw University of Technology, Faculty of Physics
 */
#ifndef HALFEMTOSHCF_H_
#define HALFEMTOSHCF_H_
 
#include "FemtoConst.h"
 
#include <initializer_list>
 
#ifdef __CIA__
#include "CorrFitMaskSH.h"
#endif
#include "Array.h"
#include "DividedHisto.h"
#include "FemtoYlmIndexes.h"
#include "FemtoYlmMath.h"
#include "Style.h"
 
#include <TMath.h>
#include <TString.h>
#include <complex>
 
namespace Hal {
#ifdef __CIA__
  class CorrFitSHCF;
#endif
  class FemtoSHSlice;
  class FemtoYlmSolver;
  class FemtoCFPainter;
 
  class FemtoSHCF : public DividedHisto1D {
#ifdef __CIA__
    friend class CorrFitSCHF;
#endif
    friend class FemtoSerializationInterfaceSH;
    friend class FemtoSHSlice;
    friend class FemtoYlmSolver;
    friend class FemtoSHCFPainter;
    const Int_t fMaxJM;
    TH1D** fNumReal = {nullptr};  //[fMaxJM] Real parts of Ylm components of the numerator
    TH1D** fNumImag = {nullptr};  // [fMaxJM]Imaginary parts of Ylm components of the numerator
    TH1D** fDenReal = {nullptr};  //[fMaxJM] Real parts of Ylm components of the denominator
    TH1D** fDenImag = {nullptr};  //[fMaxJM] Imaginary parts of Ylm components of the denominator
    Femto::EKinematics fFrame;
 
    TH1D** fCFReal = {nullptr};  //[fMaxJM] real CF's
    TH1D** fCFImag = {nullptr};  //[fMaxJM] img CF's
    Array_3<Double_t> fCovNum;
    Array_3<Double_t> fCovDen;
    Array_3<Double_t> fCovCf;
    Double_t fNormPurity;                  //
    Double_t fNormRadius;                  //
    Double_t fNormBohr;                    //
    TH3D* fCfcov             = {nullptr};  //
    FemtoCFPainter* fPainter = {nullptr};  
    FemtoYlmIndexes fLmVals;
    FemtoYlmMath fLmMath;
    Bool_t fColzSet = {kFALSE};
    Color_t fColRe  = {kBlue};
    Color_t fColIm  = {kRed};
 
    Double_t Sqr(Double_t val1, Double_t val2) const;
    TH1D* Histo(int ilm, int em, Option_t* opt, Double_t scale) const;
    void PackCfcCovariance();
    void UnpackCovariances() {};
    TH1D* GetHisto(int el, int em, Bool_t norm, Option_t* opt = "") const;
 
  protected:
    Int_t GetMaxJM() const { return fMaxJM; };
    virtual void FastAdd(const FemtoSHCF* obj);
    virtual void ApplyStyle(const Hal::HistoStyle& h);
 
  public:
    FemtoSHCF();
    FemtoSHCF(TString name, Int_t maxL, Int_t bins, Double_t min, Double_t max, Femto::EKinematics kinematics);
    FemtoSHCF(const FemtoSHCF& other);
    void SetCFColz(Color_t re, Color_t im) {
      fColRe   = re;
      fColIm   = im;
      fColzSet = kTRUE;
    }
    virtual void FillNumObj(TObject* obj);
    virtual void FillDenObj(TObject* obj);
    FemtoYlmIndexes GetYlmIndexes() const { return fLmVals; }
    /* TODO implenet this function in proper way
     FemtoSHCF(Femto3DCF *cf, Int_t maxL=3);
     */
    void SetComment(TString comment) { fComment = comment; };
    void SetNormPurity(Double_t purity) { fNormPurity = purity; }
    void SetNormRadius(Double_t radius) { fNormRadius = radius; };
    void SettNormBohr(Double_t bohr) { fNormBohr = bohr; };
    void AddLabel(TString label);
    void RecalculateCF(Int_t debugBin = -1, Bool_t suwm = kFALSE);
    void SetNumRe(TH1D** histograms, Bool_t clone = kTRUE);
    void SetNumIm(TH1D** histograms, Bool_t clone = kTRUE);
    void SetDenRe(TH1D** histograms, Bool_t clone = kTRUE);
    void SetDenIm(TH1D** histograms, Bool_t clone = kTRUE);
    void Draw(Option_t* opt = "");
    void NormalizeBy(int el, int em, Option_t* opt = "re");
    void SetScale(Double_t scale) { fScale = scale; };
    void Browse(TBrowser* b);
#ifdef __CIA__
    void Fit(CorrFitSHCF* fit);
    void FitDummy(CorrFitSHCF* fit);
#endif
    TH1D* GetRawCFRe(int pos) const { return fCFReal[pos]; };
    TH1D* GetRawCFIm(int pos) const { return fCFImag[pos]; };
    TH1D* GetCFRe(int el, int em) const;
    TH1D* GetCFIm(int el, int em) const;
    TH1D* GetNumRe(int el, int em) const;
    TH1D* GetNumIm(int el, int em) const;
    TH1D* GetDenRe(int el, int em) const;
    TH1D* GetDenIm(int el, int em) const;
    TH3D* GetCovCF() const;
    TH1D* GetCubic(Option_t* opt, Bool_t R = kFALSE) const;
    Femto::EKinematics GetFrame() const { return fFrame; }
    Int_t GetLMax() const { return TMath::Sqrt(fMaxJM - 1); };
    void Rebin(Int_t ngroup, Option_t* opt = "");
    TH2D* GetMaskDraw() const;
#ifdef __CIA__
    CorrFitMaskSH MakeEmptyMask() const;
#endif
    Array_1<Float_t>* ExportToFlatNum() const;
    void ExportIntoToFlatNum(Array_1<Float_t>* output) const;
    void ExportIntoToFlatNumValkyria(Array_1<Float_t>* output) const;
#ifdef __CIA__
#endif
    virtual void ImportSlice(Array_1<Float_t>* array, Int_t bin);
    virtual void Add(const Object* pack);
    virtual Long64_t Merge(TCollection* collection);
    void MakeDummyCov();
    Array_3<Double_t> GetCovNum() { return fCovNum; }
    virtual TString HTMLExtract(Int_t counter = 0, TString dir = " ") const;
    virtual TObject* GetSpecial(TString opt) const;
    FemtoCFPainter* GetPainter() const { return fPainter; }  // KURWA
    virtual ~FemtoSHCF();
    ClassDef(FemtoSHCF, 6)
  };
}  // namespace Hal
#endif /* HALFEMTOSHCF_H_ */