STEER/AliAODPWG4Particle.cxx

   1 /**************************************************************************\r
   2  * Copyright(c) 1998-2007, ALICE Experiment at CERN, All rights reserved. *\r
   3  *                                                                        *\r
   4  * Author: The ALICE Off-line Project.                                    *\r
   5  * Contributors are mentioned in the code where appropriate.              *\r
   6  *                                                                        *\r
   7  * Permission to use, copy, modify and distribute this software and its   *\r
   8  * documentation strictly for non-commercial purposes is hereby granted   *\r
   9  * without fee, provided that the above copyright notice appears in all   *\r
  10  * copies and that both the copyright notice and this permission notice   *\r
  11  * appear in the supporting documentation. The authors make no claims     *\r
  12  * about the suitability of this software for any purpose. It is          *\r
  13  * provided "as is" without express or implied warranty.                  *\r
  14  **************************************************************************/\r
  15 \r
  16 /* $Id:   AliAODPWG4Particle.h $ */\r
  17 \r
  18 //-------------------------------------------------------------------------\r
  19 //     AOD class for photon and other particles storage and \r
  20 //     correlation studies\r
  21 //     Author: Yves Schutz, CERN, Gustavo Conesa, INFN\r
  22 //-------------------------------------------------------------------------\r
  23 \r
  24 //-- ROOT system --\r
  25 \r
  26 //-- Analysis system\r
  27 #include "AliAODPWG4Particle.h"\r
  28 \r
  29 ClassImp(AliAODPWG4Particle)\r
  30 \r
  31 \r
  32 //______________________________________________________________________________\r
  33 AliAODPWG4Particle::AliAODPWG4Particle() :\r
  34 AliVParticle(),\r
  35 fMomentum(0),fPdg(-1), fTag(0), fBtag(-1), fLabel(-1), fCaloLabel(), fTrackLabel(),\r
  36 fDetector(""), fDisp(0), fTof(0), fCharged(0), fTagged(0), fBadDist(0), fFidArea(0), fInputFileIndex(0)\r
  37 {\r
  38   // constructor\r
  39   fCaloLabel [0] = -1;\r
  40   fCaloLabel [1] = -1;\r
  41   fTrackLabel[0] = -1;\r
  42   fTrackLabel[1] = -1;\r
  43 }\r
  44 \r
  45 //______________________________________________________________________________\r
  46 AliAODPWG4Particle::AliAODPWG4Particle(Double_t px, Double_t py, Double_t pz, Double_t e):\r
  47   AliVParticle(),\r
  48   fMomentum(0),fPdg(-1), fTag(0), fBtag(-1), fLabel(-1),fCaloLabel(), fTrackLabel(),\r
  49   fDetector(""), fDisp(0), fTof(0),fCharged(0), fTagged(0), fBadDist(0), fFidArea(0), fInputFileIndex(0)\r
  50 {\r
  51   // constructor\r
  52   fMomentum = new TLorentzVector(px, py, pz, e);\r
  53   \r
  54   fCaloLabel [0] = -1;\r
  55   fCaloLabel [1] = -1;\r
  56   fTrackLabel[0] = -1;\r
  57   fTrackLabel[1] = -1;  \r
  58 }\r
  59 \r
  60 //______________________________________________________________________________\r
  61 AliAODPWG4Particle::AliAODPWG4Particle(TLorentzVector & p):\r
  62   AliVParticle(),\r
  63   fMomentum(0),fPdg(-1), fTag(0), fBtag(-1), fLabel(-1),fCaloLabel(), fTrackLabel(),\r
  64   fDetector(""), fDisp(0), fTof(0), fCharged(0), fTagged(0), fBadDist(0), fFidArea(0), fInputFileIndex(0)\r
  65 {\r
  66   // constructor\r
  67   fMomentum = new TLorentzVector(p);\r
  68   \r
  69   fCaloLabel [0] = -1;\r
  70   fCaloLabel [1] = -1;\r
  71   fTrackLabel[0] = -1;\r
  72   fTrackLabel[1] = -1;\r
  73 }\r
  74 \r
  75 \r
  76 //______________________________________________________________________________\r
  77 AliAODPWG4Particle::~AliAODPWG4Particle() \r
  78 {\r
  79   // destructor\r
  80     delete fMomentum;\r
  81 }\r
  82 \r
  83 //______________________________________________________________________________\r
  84 AliAODPWG4Particle::AliAODPWG4Particle(const AliAODPWG4Particle& part) :\r
  85   AliVParticle(part),\r
  86   fMomentum(0), fPdg(part.fPdg), fTag(part.fTag), fBtag(part.fBtag), fLabel(part.fLabel), \r
  87   fCaloLabel(), fTrackLabel(), fDetector(part.fDetector),fDisp(part.fDisp), \r
  88   fTof(part.fTof), fCharged(part.fCharged), fTagged(part.fTagged), fBadDist(part.fBadDist), \r
  89   fFidArea(part.fFidArea), fInputFileIndex(part.fInputFileIndex)\r
  90 {\r
  91   // Copy constructor\r
  92   fMomentum = new TLorentzVector(*part.fMomentum);\r
  93   \r
  94   fCaloLabel [0] = part.fCaloLabel[0];\r
  95   fCaloLabel [1] = part.fCaloLabel[1];\r
  96   fTrackLabel[0] = part.fTrackLabel[0];\r
  97   fTrackLabel[1] = part.fTrackLabel[1];\r
  98 }\r
  99 \r
 100 //______________________________________________________________________________\r
 101 AliAODPWG4Particle& AliAODPWG4Particle::operator=(const AliAODPWG4Particle& part)\r
 102 {\r
 103   // Assignment operator\r
 104   if(this!=&part) {\r
 105     \r
 106     fPdg   = part.fPdg;\r
 107     fTag   = part.fTag;\r
 108     fBtag  = part.fBtag;\r
 109     fLabel = part.fLabel;\r
 110         \r
 111     fCaloLabel [0] = part.fCaloLabel[0];\r
 112     fCaloLabel [1] = part.fCaloLabel[1];\r
 113     fTrackLabel[0] = part.fTrackLabel[0];\r
 114     fTrackLabel[1] = part.fTrackLabel[1];\r
 115         \r
 116     fDetector = part.fDetector;\r
 117     fDisp     = part.fDisp; \r
 118     fTof      = part.fTof; \r
 119     fCharged  = part.fCharged; \r
 120         fTagged   = part.fTagged;\r
 121     fBadDist  = part.fBadDist;\r
 122         fFidArea  = part.fFidArea;\r
 123         fInputFileIndex =  part.fInputFileIndex;\r
 124           \r
 125     if (fMomentum ) delete fMomentum;   \r
 126     fMomentum = new TLorentzVector(*part.fMomentum);\r
 127   }\r
 128   \r
 129   return *this;\r
 130 }\r
 131 \r
 132 \r
 133 //_______________________________________________________________\r
 134 Bool_t AliAODPWG4Particle::IsPIDOK(const Int_t ipid, const Int_t pdgwanted) const{\r
 135   // returns true if particle satisfies given PID criterium\r
 136         switch(ipid){\r
 137         case 0: return kTRUE ; //No PID at all\r
 138         case 1: \r
 139           {\r
 140             if (fPdg == pdgwanted) return kTRUE; \r
 141             else return kFALSE; //Overall PID calculated with bayesian methods.\r
 142           }\r
 143         case 2: return fDisp ;   //only dispersion cut\r
 144         case 3: return fTof ;    //Only TOF cut\r
 145         case 4: return fCharged ;    //Only Charged cut\r
 146         case 5: return fDisp && fTof ;  //Dispersion and TOF\r
 147         case 6: return fDisp && fCharged ;  //Dispersion and Charged\r
 148         case 7: return fTof  && fCharged ;  //TOF and Charged\r
 149         case 8: return fDisp && fTof && fCharged ; // all 3 cuts\r
 150         default: return kFALSE ; //Not known combination\r
 151         }\r
 152 }\r
 153 \r
 154 //______________________________________________________________________________\r
 155 void AliAODPWG4Particle::Print(Option_t* /*option*/) const \r
 156 {\r
 157   // Print information of all data members\r
 158   printf("Particle 4-vector:\n");\r
 159   printf("     E  = %13.3f", E() );\r
 160   printf("     Px = %13.3f", Px());\r
 161   printf("     Py = %13.3f", Py());\r
 162   printf("     Pz = %13.3f\n", Pz());\r
 163   printf("PID bits :\n");\r
 164   printf("     TOF        : %d",fTof);\r
 165   printf("     Charged    : %d",fCharged);\r
 166   printf("     Dispersion : %d\n",fDisp);\r
 167   printf("PDG       : %d\n",fPdg);\r
 168   printf("Tag       : %d\n",fTag); \r
 169   printf("Btag      : %d\n",fBtag);  \r
 170   printf("Pi0 Tag   : %d\n",fTagged);  \r
 171   printf("Dist. to bad channel : %d\n",fBadDist);  \r
 172   printf("Fid Area  : %d\n",fFidArea);  \r
 173   printf("Input File Index : %d\n",fInputFileIndex);  \r
 174   printf("Detector  : %s\n",fDetector.Data());\r
 175   \r
 176 }\r