FemtoPair.h

/*
 * FemtoPair.h
 *
 *  Created on: 29-05-2014
 *      Author: wielanek
 */
 
#ifndef HALFEMTOPAIRSTRUCT_H_
#define HALFEMTOPAIRSTRUCT_H_
 
#include "FemtoConst.h"
 
#include <TMath.h>
#include <TObject.h>
 
class TLorentzVector;
namespace Hal {
  class Track;
  class TwoTrack;
 
  //#define _kinetic_debug
  class FemtoMiniPair;
  class FemtoMicroPair;
  class FemtoPicoPair;
  class FemtoPair : public TObject {
  protected:
    Track* fTrack1;
    Track* fTrack2;
    Double_t fPx1, fPy1, fPz1, fE1;  // true momenta
    Double_t fPx2, fPy2, fPz2, fE2;  // true momenta
    Double_t fpx1, fpy1, fpz1, fe1;  // fake momenta
    Double_t fpx2, fpy2, fpz2, fe2;  // fake momenta
    Double_t fX1, fY1, fZ1, fT1;
    Double_t fX2, fY2, fZ2, fT2;
    Int_t fPdg1, fPdg2;
    Double_t fM12, fM22;
    Bool_t fUseImgMomenta;
    Double_t fX;
    Double_t fY;
    Double_t fZ;
    Double_t fT;
    Double_t fWeight;
    enum eWorkingMode { kMC, kComplexRe, kComplexIm, kNoMC };
    eWorkingMode fMode;
    Bool_t fUseAbs;
    void SetTrueMomenta1(Double_t px, Double_t py, Double_t pz);
    void SetTrueMomenta2(Double_t px, Double_t py, Double_t pz);
    void SetMomenta1(Double_t px, Double_t py, Double_t pz);
    void SetMomenta2(Double_t px, Double_t py, Double_t pz);
 
  public:
    FemtoPair(Bool_t enable_fake = kFALSE);
    void UseAbs() { fUseAbs = kTRUE; };
    FemtoPair& operator=(const FemtoMiniPair& other);
    FemtoPair& operator=(const FemtoMicroPair& other);
    FemtoPair& operator=(const FemtoPicoPair& other);
    inline Bool_t IsAbs() const { return fUseAbs; };
    void SetMass(Double_t m1, Double_t m2);
    virtual FemtoPair* MakeCopy() const         = 0;
    virtual Femto::EKinematics GetFrame() const = 0;
    Bool_t Init(Int_t task_id);
    inline void Build(TwoTrack* tracks) { BuildMomenta(tracks); }
    virtual ~FemtoPair();
    inline Double_t TruePx1() const { return fPx1; };
    inline Double_t TruePy1() const { return fPy1; };
    inline Double_t TruePz1() const { return fPz1; };
    inline Double_t TrueE1() const { return fE1; };
    inline Double_t TruePhi1() const { return TMath::ATan2(fPy1, fPx1); }
    inline Double_t TruePx2() const { return fPx2; };
    inline Double_t TruePy2() const { return fPy2; };
    inline Double_t TruePz2() const { return fPz2; };
    inline Double_t TrueE2() const { return fE2; };
    inline Double_t TruePhi2() const { return TMath::ATan2(fPy2, fPx2); };
    inline Double_t GetX1() const { return fX1; };
    inline Double_t GetY1() const { return fY1; };
    inline Double_t GetZ1() const { return fZ1; };
    inline Double_t GetT1() const { return fT1; };
    inline Double_t GetX2() const { return fX2; };
    inline Double_t GetY2() const { return fY2; };
    inline Double_t GetZ2() const { return fZ2; };
    inline Double_t GetT2() const { return fT2; };
    inline Double_t GetPx1() const { return fpx1; };
    inline Double_t GetPy1() const { return fpy1; };
    inline Double_t GetPz1() const { return fpz1; };
    inline Double_t GetE1() const { return fe1; };
    inline Double_t GetPhi1() const { return TMath::ATan2(fpy1, fpx1); };
    inline Double_t GetPx2() const { return fpx2; };
    inline Double_t GetPy2() const { return fpy2; };
    inline Double_t GetPz2() const { return fpz2; };
    inline Double_t GetE2() const { return fe2; };
    inline Double_t GetPhi2() const { return TMath::ATan2(fpy2, fpx2); };
    inline Double_t GetWeight() const { return fWeight; };
    inline Double_t GetM1() const { return TMath::Sqrt(fM12); };
    inline Double_t GetM2() const { return TMath::Sqrt(fM22); };
    inline Int_t GetPdg1() const { return fPdg1; };
    inline Int_t GetPdg2() const { return fPdg2; };
    inline Double_t GetX() const { return fX; };
    inline Double_t GetY() const { return fY; };
    inline Double_t GetZ() const { return fZ; };
    inline Double_t GetT() const { return fT; };
    Double_t GetKt() const;
    Double_t GetTrueKt() const;
    inline Track* GetTrack1() const { return fTrack1; };
    inline Track* GetTrack2() const { return fTrack2; };
    inline void SetX1(Double_t val) { fX1 = val; };
    inline void SetY1(Double_t val) { fY1 = val; };
    inline void SetZ1(Double_t val) { fZ1 = val; };
    inline void SetT1(Double_t val) { fT1 = val; };
    inline void SetX2(Double_t val) { fX2 = val; };
    inline void SetY2(Double_t val) { fY2 = val; };
    inline void SetZ2(Double_t val) { fZ2 = val; };
    inline void SetT2(Double_t val) { fT2 = val; };
    inline void SetPdg1(Int_t val) { fPdg1 = val; };
    inline void SetPdg2(Int_t val) { fPdg2 = val; };
    inline void SetWeight(Double_t weight) { fWeight = weight; };
    void SetTrueMomenta1(Double_t px, Double_t py, Double_t pz, Double_t e);
    void SetTrueMomenta2(Double_t px, Double_t py, Double_t pz, Double_t e);
    void PrintInfo() const;
    void SetMomenta1(Double_t px, Double_t py, Double_t pz, Double_t e);
    void SetMomenta2(Double_t px, Double_t py, Double_t pz, Double_t e);
    void SetFreezoutCoord1(Double_t x, Double_t y, Double_t z, Double_t t);
    void SetFreezoutCoord2(Double_t x, Double_t y, Double_t z, Double_t t);
    void SetMomenta(const TLorentzVector& p1, const TLorentzVector& p2);
    void SetTrueMomenta(const TLorentzVector& p1, const TLorentzVector& p2);
    void SetFreezouts(const TLorentzVector& x1, const TLorentzVector& x2);
    inline Double_t GetPhi() const { return TMath::ATan2(fPy1 + fPy2, fPx1 + fPx2); };
    Bool_t IsUseImgMomenta() const { return fUseImgMomenta; }
    virtual void Compute() = 0;
    virtual void Compute_Rotated() = 0;
    virtual void Compute_Hemisphere() = 0;
    virtual void Compute_Charged() { Compute(); };
    virtual void Compute_Mixed() { Compute(); };
    void BuildMomenta(TwoTrack* tracks);
    virtual TString GetTags() const = 0;
    ClassDef(FemtoPair, 1)
  };
}  // namespace Hal
 
#endif /* HALFEMTOPAIR_H_ */