CorrFitPairGeneratorYPtKt.h

/*
 * CorrFitPairGeneratorYPtKt.h
 *
 *  Created on: 10 kwi 2020
 *      Author: Daniel Wielanek
 *      E-mail: daniel.wielanek@gmail.com
 *      Warsaw University of Technology, Faculty of Physics
 */
#ifndef HALCorrFitPairGeneratorYPtKt_H_
#define HALCorrFitPairGeneratorYPtKt_H_
 
#include "CorrFitPairGenerator.h"
 
#include "FemtoMiniPair.h"
 
#include <utility>
 
#include <Rtypes.h>
#include <RtypesCore.h>
#include <TH2.h>
 
class TFile;
class TTree;
class TClonesArray;
namespace Hal {
  class FemtoMicroPair;
  class CorrFitPairGeneratorYPtKt : public CorrFitPairGenerator {
  protected:
    Float_t f2Kt2[2];
    TH2D fHist1, fHist2;
    FemtoMicroPair fPair;
    struct fourVect {
      int t, x, y, z;
    };
    void GenerateFreeRandomPair();
    Int_t GeneratePairOneDim();
    fourVect GeneratePairThreeDim();
    virtual void GenerateEvent();
 
  public:
    CorrFitPairGeneratorYPtKt();
    CorrFitPairGeneratorYPtKt& operator=(const CorrFitPairGeneratorYPtKt&) = delete;
 
    void SetKt(Double_t min, Double_t max) {
      f2Kt2[0] = min * min * 4.0;
      f2Kt2[1] = max * max * 4.0;
    };
    void SetHist(const TH2D& hist1, const TH2D& hist2);
    Bool_t Init();
    virtual ~CorrFitPairGeneratorYPtKt();
    ClassDef(CorrFitPairGeneratorYPtKt, 1)
  };
}  // namespace Hal
#endif /* HALCorrFitPairGeneratorYPtKt_H_ */