PWG/EMCAL/AliPicoTrack.h

   1 #ifndef AliPicoTrack_H
   2 #define AliPicoTrack_H
   3
   4 #include <TMath.h>
   5
   6 #include "AliVTrack.h"
   7 #include "AliAODMCParticle.h"
   8
   9 class AliVCluster;
  10
  11 class AliPicoTrack: public AliVTrack {
  12  public:
  13   AliPicoTrack();
  14   AliPicoTrack(Double_t pt, Double_t eta, Double_t phi, Byte_t q, Int_t label, Byte_t type,
  15                Double_t etaemc=0, Double_t phiemc=0, Double_t ptemc=0, Bool_t ise=0, Double_t mass=0.13957);
  16   AliPicoTrack(const AliPicoTrack &pc);
  17   AliPicoTrack &operator=(const AliPicoTrack &pc);
  18
  19   Double_t        Px()                        const { return fPt*TMath::Cos(fPhi);  }
  20   Double_t        Py()                        const { return fPt*TMath::Sin(fPhi);  }
  21   Double_t        Pz()                        const { return fPt*TMath::SinH(fEta); }
  22   Double_t        Pt()                        const { return fPt;                   }
  23   Double_t        P()                         const { return fPt*TMath::CosH(fEta); }
  24
  25   Bool_t          PxPyPz(Double_t p[3])       const { p[0]=Px();p[1]=Py();p[2]=Pz(); return 1;         }
  26   Bool_t          GetPxPyPz(Double_t p[3])    const { p[0]=Px();p[1]=Py();p[2]=Pz(); return 1;         }
  27   Double_t        Xv()                        const { return 0.;      }
  28   Double_t        Yv()                        const { return 0.;      }
  29   Double_t        Zv()                        const { return 0.;      }
  30   Bool_t          XvYvZv(Double_t x[3])       const { x[0]=0;x[1]=0;x[2]=0; return 1;                  }
  31   Double_t        OneOverPt()                 const { return 1./fPt;  }
  32   Double_t        Phi()                       const { return fPhi;    }
  33   Double_t        Theta()                     const { return 2*TMath::ATan(TMath::Exp(-fEta));         }
  34   Double_t        E()                         const { Double_t p=P(); return TMath::Sqrt(M()*M()+p*p); }
  35   Double_t        M()                         const { return fM; }
  36   Double_t        Eta()                       const { return fEta;    }
  37   Double_t        Y()                         const { return 0.5*TMath::Log((E()+Pz())/(E()-Pz()));  }
  38   Short_t         Charge()                    const { return (char)fQ;}
  39   Int_t           GetLabel()                  const { return fLabel;  }
  40   void            SetLabel(Int_t label)             { fLabel = label; }
  41   Byte_t          GetTrackType()              const { return fTrackType;}
  42   void            SetTrackType(Byte_t type)         { fTrackType = type;}
  43   Int_t           PdgCode()                   const { return 0;       }
  44   const Double_t *PID()                const { return 0;       }
  45   Int_t           GetID()                     const { return 0;       }
  46   UChar_t         GetITSClusterMap()          const { return 0;       }
  47   Int_t           GetEMCALcluster()           const { return fClusId; }
  48   void            SetEMCALcluster(Int_t id)         { fClusId = id;   }
  49   Double_t        GetEtaEmc()                 const { return GetTrackEtaOnEMCal(); }
  50   Double_t        GetPhiEmc()                 const { return GetTrackPhiOnEMCal(); }
  51   Bool_t          IsEMCAL()                   const { return fEmcal;  }
  52   ULong_t         GetStatus()                 const { return 0;       }
  53   Bool_t          GetXYZ(Double_t *v)         const { v[0]=0; v[1]=0; v[2]=0; return 0; }
  54   using AliVTrack::GetXYZ;
  55   Double_t        GetBz()                     const { return 0;       }
  56   void            GetBxByBz(Double_t b[3])    const { b[0]=0;b[1]=0;b[2]=0; }
  57   Bool_t          GetCovarianceXYZPxPyPz(Double_t /*cv*/[21]) const { return 0; }
  58   AliVTrack      *GetTrack()                  const { return fOrig;   }
  59   void            SetTrack(AliVTrack *t)            { fOrig = t;      }
  60   Double_t        GetTrackPhiOnEMCal()        const { return fPhiEmc ; }
  61   Double_t        GetTrackEtaOnEMCal()        const { return fEtaEmc ; }
  62   Double_t        GetTrackPtOnEMCal()         const { return fPtEmc  ; }
  63   Double_t        GetTrackPOnEMCal()          const { return fPtEmc*TMath::CosH(fEtaEmc);}
  64   void            SetTrackPhiEtaPtOnEMCal(Double_t eta, Double_t phi, Double_t pt) { fEtaEmc = eta; fPhiEmc = phi; fPtEmc = pt;}
  65   void            Clear(Option_t * /*option*/ ="")  { fClusId = -1; fOrig = 0; }
  66   Int_t           Compare(const TObject* obj) const;
  67   Bool_t          PropagateToDCA(const AliVVertex *, Double_t, Double_t, Double_t *, Double_t *) { return 0; }
  68   void            SetFlag(UInt_t flag) {fFlag = flag;}
  69   UInt_t          GetFlag() const {return fFlag;}
  70   void SetPrimary(Bool_t b = kTRUE){
  71     if(b)fFlag |= AliAODMCParticle::kPrimary;
  72     else fFlag &= ~AliAODMCParticle::kPrimary;
  73   }
  74   Bool_t IsPrimary() const {return ((fFlag&AliAODMCParticle::kPrimary)==AliAODMCParticle::kPrimary);}
  75
  76   void SetPhysicalPrimary(Bool_t b = kTRUE){
  77     if(b)fFlag |= AliAODMCParticle::kPhysicalPrim;
  78     else fFlag &= ~AliAODMCParticle::kPhysicalPrim;
  79   }
  80   Bool_t IsPhysicalPrimary() const {return ((fFlag&AliAODMCParticle::kPhysicalPrim)==AliAODMCParticle::kPhysicalPrim);}
  81
  82   void SetSecondaryFromWeakDecay(Bool_t b = kTRUE){
  83     if(b)fFlag |= AliAODMCParticle::kSecondaryFromWeakDecay;
  84     else fFlag &= ~AliAODMCParticle::kSecondaryFromWeakDecay;
  85   }
  86   Bool_t IsSecondaryFromWeakDecay() const {return ((fFlag&AliAODMCParticle::kSecondaryFromWeakDecay)==AliAODMCParticle::kSecondaryFromWeakDecay);}
  87
  88   void SetSecondaryFromMaterial(Bool_t b = kTRUE){
  89     if(b)fFlag |= AliAODMCParticle::kSecondaryFromMaterial;
  90     else fFlag &= ~AliAODMCParticle::kSecondaryFromMaterial;
  91   }
  92   Bool_t IsSecondaryFromMaterial() const {return ((fFlag&AliAODMCParticle::kSecondaryFromMaterial)==AliAODMCParticle::kSecondaryFromMaterial);}
  93
  94   Bool_t  IsFromHIJING() const { return (fGeneratorIndex == 0); }
  95   void    SetGeneratorIndex(Short_t i) {fGeneratorIndex = i;}
  96   Short_t GetGeneratorIndex() const {return fGeneratorIndex;}
  97
  98   static void     GetEtaPhiDiff(const AliVTrack *t, const AliVCluster *v, Double_t &phidiff, Double_t &etadiff);
  99   static Byte_t   GetTrackType(const AliVTrack *t);
 100
 101  protected:
 102   Double32_t      fPt;             //[0,0,12]   pt at vertex
 103   Double32_t      fEta;            //[-1,1,12]  eta at vertex
 104   Double32_t      fPhi;            //[0,6.3,12] phi at vertex
 105   Double32_t      fM;              //           mass
 106   Byte_t          fQ;              //           charge
 107   Int_t           fLabel;          //           label
 108   Byte_t          fTrackType;      //           0=global track; 1=w/o SPD, w/ ITS refit; 2=w/o SPD, w/o ITS refit
 109   Double32_t      fEtaEmc;         //[-1,1,12]  eta at emcal surface
 110   Double32_t      fPhiEmc;         //[0,6.3,12] phi at emcal surface
 111   Double32_t      fPtEmc;          //[0,0,12]   pt at emcal surface
 112   Bool_t          fEmcal;          //           is true if track propagated to emcal
 113   UInt_t          fFlag;           //!          Flag for indication of primary etc (MC)
 114   Short_t         fGeneratorIndex; //!          Index of generator in cocktail
 115   Short_t         fClusId;         //!          cluster id of matched cluster; -1 if not set
 116   AliVTrack      *fOrig;           //!          ptr to original track
 117
 118   ClassDef(AliPicoTrack, 8) // Pico track class
 119 };
 120 #endif