HBTAN/AliHBTParticle.h

   1 #ifndef ALIHBTPARTICLE
   2 #define ALIHBTPARTICLE
   3
   4 //Ali HBT Particle: simplified class TParticle
   5 //Simplified in order to minimize the size of object
   6 //  - we want to keep a lot of such a objects in memory
   7
   8 #include <TObject.h>
   9 #include <TLorentzVector.h>
  10 #include <TMath.h>
  11 #include <TDatabasePDG.h>
  12
  13 #include "AliConst.h"
  14
  15 class TParticle;
  16
  17 class AliHBTParticle : public TObject
  18 {
  19
  20 public:
  21                                 // ****** constructors and destructor
  22   AliHBTParticle();
  23   AliHBTParticle(const AliHBTParticle& in);
  24
  25   AliHBTParticle(Int_t pdg, Int_t idx, Double_t px, Double_t py, Double_t pz, Double_t etot,
  26                  Double_t vx, Double_t vy, Double_t vz, Double_t time);
  27
  28   AliHBTParticle(Int_t pdg, Float_t prob, Int_t idx, Double_t px, Double_t py, Double_t pz, Double_t etot,
  29                  Double_t vx, Double_t vy, Double_t vz, Double_t time);
  30
  31   AliHBTParticle(const TParticle& p,Int_t idx);
  32
  33   virtual ~AliHBTParticle();
  34
  35   AliHBTParticle& operator=(const AliHBTParticle& in);
  36
  37   void           SetPIDprobability(Int_t pdg, Float_t prob = 1.0);
  38   Float_t        GetPIDprobability(Int_t pdg);
  39
  40   Int_t          GetPdgCode      () const { return (fPids)?fPids[fPdgIdx]:0;}
  41   Int_t          GetPid          () const { return GetPdgCode();}
  42   Float_t        GetPidProb      () const { return (fPidProb)?fPidProb[fPdgIdx]:0;}
  43
  44   Int_t          GetUID          () const { return fIdxInEvent;}
  45   Int_t          GetNumberOfPids () const { return fNPids;}
  46   Int_t          GetNthPid         (Int_t idx) const;
  47   Float_t        GetNthPidProb     (Int_t idx) const;
  48
  49   void           SetPdgCode(Int_t pdg, Float_t prob = 1.0);
  50   Double_t       GetCalcMass     () const { return fCalcMass; }
  51   Double_t       GetMass         ()       { return (GetPDG())?GetPDG()->Mass():-1.;}
  52
  53   TParticlePDG*  GetPDG          (){return TDatabasePDG::Instance()->GetParticle(GetPdgCode());}
  54
  55   Int_t          Beauty          ()  { return GetPDG()->Beauty(); }
  56   Int_t          Charm           ()  { return GetPDG()->Charm(); }
  57   Int_t          Strangeness     ()  { return GetPDG()->Strangeness();}
  58   void ProductionVertex(TLorentzVector &v) { v.SetXYZT(fVx,fVy,fVz,fVt);}
  59
  60
  61   Double_t         Vx    () const { return fVx;}
  62   Double_t         Vy    () const { return fVy;}
  63   Double_t         Vz    () const { return fVz;}
  64   Double_t         T     () const { return fVt;}
  65
  66   Double_t         Px    () const { return fPx; } //X coordinate of the momentum
  67   Double_t         Py    () const { return fPy; } //Y coordinate of the momentum
  68   Double_t         Pz    () const { return fPz; } //Z coordinate of the momentum
  69   Double_t         P     () const                 //momentum
  70     { return TMath::Sqrt(fPx*fPx+fPy*fPy+fPz*fPz); }
  71
  72   void Momentum(TLorentzVector &v) { v.SetPxPyPzE(fPx,fPy,fPz,fE);}
  73
  74   Double_t         Pt    () const  //transverse momentum
  75     { return TMath::Sqrt(fPx*fPx+fPy*fPy); }
  76   Double_t         Energy() const { return fE; }
  77
  78                                    //Pseudo Rapidity
  79   Double_t         Eta   () const { if (P() != fPz) return 0.5*TMath::Log((P()+fPz)/(P()-fPz));
  80                                     else return 1.e30;}
  81
  82                                    //Rapidity
  83   Double_t         Y     () const { if (fE  != fPz) return 0.5*TMath::Log((fE+fPz)/(fE-fPz));
  84                                     else return 1.e30;}
  85
  86   Double_t         Phi   () const { return kPI+TMath::ATan2(-fPy,-fPx); }
  87
  88   Double_t         Theta () const { return (fPz==0)?kPI/2:TMath::ACos(fPz/P()); }
  89
  90                                 // setters
  91
  92
  93   void           SetMomentum(Double_t px, Double_t py, Double_t pz, Double_t e)
  94                              {fPx=px; fPy=py; fPz=pz; fE=e;}
  95   void           SetMomentum(const TLorentzVector& p)
  96                              {SetMomentum(p.Px(),p.Py(),p.Pz(),p.Energy());}
  97
  98   void           SetProductionVertex(Double_t vx, Double_t vy, Double_t vz, Double_t t)
  99                              {fVx=vx; fVy=vy; fVz=vz; fVt=t;}
 100   void           SetProductionVertex(const TLorentzVector& v)
 101                              {SetProductionVertex(v.X(),v.Y(),v.Z(),v.T());}
 102   void           SetCalcMass(Double_t mass) {fCalcMass = mass;}
 103   const Char_t*  GetName() const;
 104   void           Print() const;
 105
 106   static void    SetDebug(Int_t dbg=1){fgDebug=dbg;}
 107   static Int_t   GetDebug(){return fgDebug;}
 108   static Int_t   fgDebug; //debug printout level
 109
 110 protected:
 111   Int_t          GetPidSlot(Int_t pdg) const;//returns position of the given PID in fPids (and fPidProb) array.
 112
 113 private:
 114   Char_t         fPdgIdx;               // index of PDG code of the particle in fPids
 115   Int_t          fIdxInEvent;           // index of a particle: the same particle can appear in the event
 116                                         //  many times with different pid's. Idx allows to check that they are the same particles
 117   Int_t          fNPids;                // number of non-zero proboble Pids
 118   Int_t         *fPids;                 // [fNPids] Array with PIDs
 119   Float_t       *fPidProb;              // [fNPids] PIDs probabilities
 120   Double_t       fCalcMass;             // Calculated mass
 121
 122   Double_t       fPx;                   // x component of momentum
 123   Double_t       fPy;                   // y component of momentum
 124   Double_t       fPz;                   // z component of momentum
 125   Double_t       fE;                    // Energy
 126
 127   Double_t       fVx;                   // x of production vertex
 128   Double_t       fVy;                   // y of production vertex
 129   Double_t       fVz;                   // z of production vertex
 130   Double_t       fVt;                   // t of production vertex
 131
 132   ClassDef(AliHBTParticle,2)  // TParticle vertex particle information
 133 };
 134
 135 #endif