CorrFitVerticalSlices.h

/*
 * CorrFitSlices.h
 *
 *  Created on: 2 wrz 2022
 *      Author: Daniel Wielanek
 *      E-mail: daniel.wielanek@gmail.com
 *      Warsaw University of Technology, Faculty of Physics
 */
#ifndef HAL_ANALYSIS_FEMTO_CORRFIT_MAPGENERATOR_MULTIDIM_CORRFITVERTICALSLICES_H_
#define HAL_ANALYSIS_FEMTO_CORRFIT_MAPGENERATOR_MULTIDIM_CORRFITVERTICALSLICES_H_
 
#include <TObject.h>
#include <vector>
 
#include "FemtoYlmIndexes.h"
#include "FemtoYlmMath.h"
 
namespace Hal {
  class FemtoPair;
  class FemtoSHCF;
  class Femto1DCF;
  class Femto3DCF;
  class DividedHisto1D;
  class CorrFitVerticalSlices : public TObject {
  protected:
  public:
    CorrFitVerticalSlices() {};
    virtual void FillNum(Int_t bin, FemtoPair* pair) = 0;
    virtual void FillDen(Int_t bin, FemtoPair* pair) = 0;
    virtual ~CorrFitVerticalSlices() {};
    ClassDef(CorrFitVerticalSlices, 1)
  };
 
  class CorrFitVerticalSlices1D : public CorrFitVerticalSlices {
    friend class FemtoSerializationInterface1D;
    std::vector<Double_t> fNum;
    std::vector<Double_t> fDen;
 
  public:
    CorrFitVerticalSlices1D() {};  // only for root
    CorrFitVerticalSlices1D(const Hal::Femto1DCF& h, Int_t nSamples);
    void FillNum(Int_t bin, FemtoPair* pair);
    void FillDen(Int_t bin, FemtoPair* pair);
    void FillDirectNum(Int_t bin, Double_t val) { fNum[bin] += val; }
    void FillDirectDen(Int_t bin, Double_t val) { fDen[bin] += val; }
    virtual ~CorrFitVerticalSlices1D() {};
    ClassDef(CorrFitVerticalSlices1D, 1)
  };
 
  class CorrFitVerticalSlices3D : public CorrFitVerticalSlices {
    friend class FemtoSerializationInterface3D;
    std::vector<std::vector<std::vector<Double_t>>> fNum;
    std::vector<std::vector<std::vector<Double_t>>> fDen;
    Int_t fOutBins        = {0};
    Int_t fSideBins       = {0};
    Double_t fMin[2]      = {0, 0};
    Double_t fOverStep[2] = {0, 0};
 
  public:
    CorrFitVerticalSlices3D() {};  // only for root
    CorrFitVerticalSlices3D(const Hal::Femto3DCF& h, Int_t nSamples);
    std::pair<int, int> FindBin(FemtoPair* pair);
    void FillNum(Int_t bin, FemtoPair* pair);
    void FillDen(Int_t bin, FemtoPair* pair);
    void FillNumDirect(Int_t bin, std::pair<int, int> coord, Double_t w) { fNum[bin][coord.first][coord.second] += w; }
    void FillDenDirect(Int_t bin, std::pair<int, int> coord, Double_t w) { fDen[bin][coord.first][coord.second] += w; }
    virtual ~CorrFitVerticalSlices3D() {};
    ClassDef(CorrFitVerticalSlices3D, 1)
  };
 
  class CorrFitVerticalSlicesSH : public CorrFitVerticalSlices {
    friend class FemtoSerializationInterfaceSH;
    FemtoYlmIndexes fLmVals;
    FemtoYlmMath fLmMath;
 
  public:
    const Int_t fMaxJM;
    std::vector<Double_t> fNum;
    std::vector<Double_t> fDen;
    std::vector<std::vector<Double_t>> fShNumReal;
    std::vector<std::vector<Double_t>> fShDenReal;
    std::vector<std::vector<Double_t>> fShNumImag;
    std::vector<std::vector<Double_t>> fShDenImag;
    std::vector<std::vector<Double_t>> fShNumRealE;
    std::vector<std::vector<Double_t>> fShDenRealE;
    std::vector<std::vector<Double_t>> fShNumImagE;
    std::vector<std::vector<Double_t>> fShDenImagE;
    std::vector<std::vector<std::vector<Double_t>>> fCovMatrix;
 
  public:
    CorrFitVerticalSlicesSH() : fMaxJM(3) {};  // only for root
    Int_t GetMaxL() const { return fLmVals.GetMaxL(); }
    CorrFitVerticalSlicesSH(const Hal::FemtoSHCF& h, Int_t nSamples);
    void FillNum(Int_t bin, FemtoPair* pair);
    void FillDen(Int_t bin, FemtoPair* pair);
    std::complex<double>* GetBufferCalc(FemtoPair* pair);
    void FillNumBuffer(std::complex<double>* shCoord, Double_t weight, Int_t paramBin);
    void FillDenBuffer(std::complex<double>* shCoord, Double_t weight, Int_t paramBin);
    void FillNumBuffer10(std::complex<double>* shCoord, std::vector<Double_t>& weight, Int_t paramBin);
    void FillDenBuffer10(std::complex<double>* shCoord, std::vector<Double_t>& weight, Int_t paramBin);
    const std::vector<Double_t>& GetNumReal(Int_t bin) const { return fShNumReal[bin]; };
    const std::vector<Double_t>& GetNumImag(Int_t bin) const { return fShNumImag[bin]; };
    const std::vector<Double_t>& GetDenReal(Int_t bin) const { return fShDenReal[bin]; };
    const std::vector<Double_t>& GetDenImag(Int_t bin) const { return fShDenImag[bin]; };
    void SetNumReal(Int_t parLoc, Int_t lmbin, Double_t re, Double_t im) {
      fShNumReal[parLoc][lmbin] = re;
      fShNumImag[parLoc][lmbin] = im;
    }
    virtual ~CorrFitVerticalSlicesSH() {};
    ClassDef(CorrFitVerticalSlicesSH, 1)
 
      virtual void Print(Option_t* option) const;
  };
} /* namespace Hal */
 
#endif /* HAL_ANALYSIS_FEMTO_CORRFIT_MAPGENERATOR_MULTIDIM_CORRFITVERTICALSLICES_H_ */