PWGGA/EMCALTasks/AliAODParticle.h

   1 #ifndef ALIAODPARTICLE_H
   2 #define ALIAODPARTICLE_H
   3 /* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
   4  * See cxx source for full Copyright notice     */
   5 /* $Id: $ */
   6
   7 //_________________________________________________________________________
   8 // Class to fill photon and track for correlation
   9
  10 #include "TLorentzVector.h"
  11
  12 class AliAODParticle :public TLorentzVector{
  13
  14  public:
  15
  16   AliAODParticle() ;
  17   AliAODParticle(Double_t px,Double_t py,Double_t pz,Double_t E) ;
  18   ~AliAODParticle(){}
  19
  20    Int_t    GetChargedSign() const { return fChargedSign ;}
  21
  22    Float_t  GetLambda0(void)        const {return fL0      ;}
  23    Float_t  GetLambda1(void)        const {return fL1      ;}
  24    Int_t    GetModule(void)         const {return fModule  ;}
  25    Double_t GetDistBad(void)        const {return fBadDist ;}
  26    Int_t    GetNCells(void)         const {return fNCells  ;}
  27    Double_t GetTOF(void)            const {return fClusterTime;}
  28    Int_t    GetClusterID(void)      const {return fClusterID;}
  29    Int_t    GetAODClusterID(void)   const {return fAODClusterID;}
  30
  31    Float_t  GetPhotonPairDTime(void)    const {return fPhotonPairDTime   ;}
  32    Int_t    GetPhotonPairDModule(void)  const {return fPhotonPairDModule ;}
  33    Float_t  GetPhotonPairAsy(void)      const {return fPhotonPairAsy     ;}
  34    Float_t  GetPhotonPairAngle(void)    const {return fPhotonPairAngle   ;}
  35    Int_t    GetPhotonPairID(Int_t i)    const {if(i ==0 ) return fPhotonPairID0;   else if(i ==1) return fPhotonPairID1; else return -999;}
  36    Int_t    GetAODPhotonPairID(Int_t i)    const {if(i ==0 ) return fAODPhotonPairID0;   else if(i ==1) return fAODPhotonPairID1; else return -999;}
  37
  38    Bool_t   IsInSSA(void)          const {return kInSSA           ;}
  39    Bool_t   IsInTrackMatched(void) const {return kInTrackMatched  ;}
  40    Bool_t   IsInTOF(void)          const {return kInTOF           ;}
  41    Bool_t   IsLeading(void)        const {return kIsLeading       ;}
  42    Bool_t   IsIsolated(void)       const {return kIsIsolated      ;}
  43
  44    Bool_t   IsPIDOK(const Int_t fPid) const ;
  45
  46    void SetChargedSign(Int_t n){fChargedSign = n;}
  47
  48    void SetLambdas(Float_t l0,Float_t l1){fL0=l0,fL1=l1;}
  49    void SetModule(Int_t mod){fModule = mod;}
  50    void SetDistBad(Double_t nbad){fBadDist = nbad;}
  51    void SetNCells(Int_t ncell){fNCells = ncell;}
  52    void SetTOF(Double_t time){fClusterTime = time;}
  53    void SetClusterID(Int_t id){fClusterID = id;}
  54    void SetAODClusterID(Int_t aodid){fAODClusterID = aodid;}
  55    void SetSSABit(Bool_t ssa){kInSSA = ssa;}
  56    void SetTOFBit(Bool_t tof){kInTOF = tof;}
  57    void SetTrackMatchedBit(Bool_t matched){kInTrackMatched = matched;}
  58    void SetDistToBad(Int_t dist){fBadDist=dist;}
  59    void SetPhotonPairDTime(Float_t pairtime){fPhotonPairDTime = pairtime;}
  60    void SetPhotonPairDModule(Int_t pairmodule){fPhotonPairDModule = pairmodule;}
  61    void SetPhotonPairAsy(Float_t asy){fPhotonPairAsy = asy;}
  62    void SetPhotonPairAngle(Float_t angle){fPhotonPairAngle = angle;}
  63    void SetPhotonPairID(Int_t id0, Int_t id1){fPhotonPairID0 =id0, fPhotonPairID1=id1 ; }
  64    void SetAODPhotonPairID(Int_t aodid0, Int_t aodid1){fAODPhotonPairID0 =aodid0, fAODPhotonPairID1=aodid1 ; }
  65    void SetIsLeading(Bool_t leading){kIsLeading = leading;}
  66    void SetIsolated(Bool_t iso){kIsIsolated = iso;}
  67 private:
  68
  69   Int_t     fChargedSign ;   //sign of charged track
  70
  71   Float_t   fL0               ;
  72   Float_t   fL1               ;
  73   Int_t     fModule           ;
  74   Int_t     fBadDist          ;
  75   Int_t     fNCells           ;
  76   Double_t  fClusterTime      ;
  77   Int_t     fClusterID        ;
  78   Int_t     fAODClusterID        ;
  79   Bool_t    kInSSA            ;
  80   Bool_t    kInTOF            ;
  81   Bool_t    kInTrackMatched   ;
  82   Double_t  fPhotonPairDTime  ;
  83   Int_t     fPhotonPairDModule;
  84   Float_t   fPhotonPairAsy    ;
  85   Float_t   fPhotonPairAngle  ;
  86   Int_t     fPhotonPairID0    ;
  87   Int_t     fAODPhotonPairID0    ;
  88   Int_t     fPhotonPairID1    ;
  89   Int_t     fAODPhotonPairID1    ;
  90   Bool_t    kIsLeading        ;
  91   Bool_t    kIsIsolated       ;
  92
  93   ClassDef(AliAODParticle,2)
  94
  95 };
  96
  97 #endif // #ifdef ALIAODPARTICLE_H
  98
  99