FemtoMiniPair.h

/*
 * FemtoMiniPair.h
 *
 *  Created on: 9 lut 2019
 *      Author: Daniel Wielanek
 *      E-mail: daniel.wielanek@gmail.com
 *      Warsaw University of Technology, Faculty of Physics
 */
#ifndef HALFEMTOMINIPAIR_H_
#define HALFEMTOMINIPAIR_H_
 
#include "FemtoPair.h"
 
#include <TMath.h>
 
namespace Hal {
  class FemtoPicoPair : public TObject {
  protected:
    Float_t fPx1, fPy1, fPz1, fE1;  // true momenta
    Float_t fPx2, fPy2, fPz2, fE2;  // true momenta
  public:
    FemtoPicoPair();
    FemtoPicoPair& operator=(const FemtoPicoPair& other);
    inline Float_t TruePx1() const { return fPx1; };
    inline Float_t TruePy1() const { return fPy1; };
    inline Float_t TruePz1() const { return fPz1; };
    inline Float_t TrueE1() const { return fE1; };
    inline Float_t TruePhi1() const { return TMath::ATan2(fPy1, fPx1); }
    inline Float_t TruePx2() const { return fPx2; };
    inline Float_t TruePy2() const { return fPy2; };
    inline Float_t TruePz2() const { return fPz2; };
    inline Float_t TrueE2() const { return fE2; };
    inline Float_t TruePhi2() const { return TMath::ATan2(fPy2, fPx2); };
    inline Float_t GetPhi() const { return TMath::ATan2(fPy1 + fPy2, fPx1 + fPx2); };
    virtual ~FemtoPicoPair();
    ClassDef(FemtoPicoPair, 1)
  };
 
  class FemtoMicroPair : public FemtoPicoPair {
  protected:
    Float_t fpx1, fpy1, fpz1, fe1;  // fake momenta
    Float_t fpx2, fpy2, fpz2, fe2;  // fake momenta
    Int_t fPdg1, fPdg2;
 
  public:
    FemtoMicroPair();
    inline Float_t GetPx1() const { return fpx1; };
    inline Float_t GetPy1() const { return fpy1; };
    inline Float_t GetPz1() const { return fpz1; };
    inline Float_t GetE1() const { return fe1; };
    inline Float_t GetPhi1() const { return TMath::ATan2(fpy1, fpx1); };
    inline Float_t GetPx2() const { return fpx2; };
    inline Float_t GetPy2() const { return fpy2; };
    inline Float_t GetPz2() const { return fpz2; };
    inline Float_t GetE2() const { return fe2; };
    inline Float_t GetPhi2() const { return TMath::ATan2(fpy2, fpx2); };
    inline Int_t GetPdg1() const { return fPdg1; };
    inline Int_t GetPdg2() const { return fPdg2; };
    inline void SetPdg1(Int_t val) { fPdg1 = val; };
    inline void SetPdg2(Int_t val) { fPdg2 = val; };
    void SetTrueMomenta1(Float_t px, Float_t py, Float_t pz, Float_t e);
    void SetTrueMomenta2(Float_t px, Float_t py, Float_t pz, Float_t e);
 
    void SetMomenta1(Float_t px, Float_t py, Float_t pz, Float_t e);
    void SetMomenta2(Float_t px, Float_t py, Float_t pz, Float_t e);
    FemtoMicroPair& operator=(const FemtoPair& other);
    virtual ~FemtoMicroPair();
    ClassDef(FemtoMicroPair, 1)
  };
 
  class FemtoMiniPair : public FemtoMicroPair {
    Float_t fX1, fY1, fZ1, fT1;
    Float_t fX2, fY2, fZ2, fT2;
 
  public:
    FemtoMiniPair();
    FemtoMiniPair& operator=(const FemtoPair& other);
    void PrintInfo() const;
    virtual ~FemtoMiniPair();
    inline Float_t GetX1() const { return fX1; };
    inline Float_t GetY1() const { return fY1; };
    inline Float_t GetZ1() const { return fZ1; };
    inline Float_t GetT1() const { return fT1; };
    inline Float_t GetX2() const { return fX2; };
    inline Float_t GetY2() const { return fY2; };
    inline Float_t GetZ2() const { return fZ2; };
    inline Float_t GetT2() const { return fT2; };
 
    inline void SetX1(Float_t val) { fX1 = val; };
    inline void SetY1(Float_t val) { fY1 = val; };
    inline void SetZ1(Float_t val) { fZ1 = val; };
    inline void SetT1(Float_t val) { fT1 = val; };
    inline void SetX2(Float_t val) { fX2 = val; };
    inline void SetY2(Float_t val) { fY2 = val; };
    inline void SetZ2(Float_t val) { fZ2 = val; };
    inline void SetT2(Float_t val) { fT2 = val; };
    ClassDef(FemtoMiniPair, 1);
  };
}  // namespace Hal
#endif /* HALFEMTOMINIPAIR_H_ */