STEER/AliAODPid.cxx

   1 /**************************************************************************
   2  * Copyright(c) 1998-2007, ALICE Experiment at CERN, All rights reserved. *
   3  *                                                                        *
   4  * Author: The ALICE Off-line Project.                                    *
   5  * Contributors are mentioned in the code where appropriate.              *
   6  *                                                                        *
   7  * Permission to use, copy, modify and distribute this software and its   *
   8  * documentation strictly for non-commercial purposes is hereby granted   *
   9  * without fee, provided that the above copyright notice appears in all   *
  10  * copies and that both the copyright notice and this permission notice   *
  11  * appear in the supporting documentation. The authors make no claims     *
  12  * about the suitability of this software for any purpose. It is          *
  13  * provided "as is" without express or implied warranty.                  *
  14  **************************************************************************/
  15
  16 /* $Id$ */
  17
  18 //-------------------------------------------------------------------------
  19 //     AOD Pid class to store additional pid information
  20 //     Author: Annalisa Mastroserio
  21 //-------------------------------------------------------------------------
  22
  23 #include "AliAODPid.h"
  24 #include "AliESDtrack.h"
  25 #include "AliLog.h"
  26
  27 ClassImp(AliAODPid)
  28
  29
  30 //______________________________________________________________________________
  31 AliAODPid::AliAODPid():
  32     fITSsignal(0),
  33     fTPCsignal(0),
  34     fTPCsignalN(0),
  35     fTPCmomentum(0),
  36     fTRDnSlices(0),
  37     fTRDslices(0x0),
  38     fTOFesdsignal(0),
  39     fHMPIDsignal(0)
  40 {
  41   // default constructor
  42     for(Int_t i=0; i<kSPECIES; i++) fIntTime[i]=0;
  43     for(Int_t i=0; i<3; i++) fEMCALPosition[i] = 0.;
  44     for(Int_t i=0; i<5; i++) fTOFpidResolution[i] = 0.;
  45
  46 }
  47
  48 //______________________________________________________________________________
  49 AliAODPid::~AliAODPid()
  50 {
  51   delete [] fTRDslices;
  52   fTRDslices = 0;
  53   // destructor
  54 }
  55
  56
  57 //______________________________________________________________________________
  58 AliAODPid::AliAODPid(const AliAODPid& pid) :
  59   TObject(pid),
  60   fITSsignal(pid.fITSsignal),
  61   fTPCsignal(pid.fTPCsignal),
  62   fTPCsignalN(pid.fTPCsignalN),
  63   fTPCmomentum(pid.fTPCmomentum),
  64   fTRDnSlices(pid.fTRDnSlices),
  65   fTRDslices(0x0),
  66   fTOFesdsignal(pid.fTOFesdsignal),
  67   fHMPIDsignal(pid.fHMPIDsignal)
  68 {
  69   // Copy constructor
  70     fTRDslices = new Double32_t[fTRDnSlices];
  71     for(Int_t i=0; i< fTRDnSlices; i++) fTRDslices[i]=pid.fTRDslices[i];
  72     for(Int_t i=0; i<kSPECIES; i++) fIntTime[i]=pid.fIntTime[i];
  73     for(Int_t i=0; i<3; i++) {
  74       fEMCALPosition[i]=pid.fEMCALPosition[i];
  75       fEMCALMomentum[i]=pid.fEMCALMomentum[i];
  76     }
  77     for(Int_t i=0; i<6; i++) fTRDmomentum[i]=pid.fTRDmomentum[i];
  78
  79     for(Int_t i=0; i<5; i++) fTOFpidResolution[i]=pid.fTOFpidResolution[i];
  80 }
  81
  82 //______________________________________________________________________________
  83 AliAODPid& AliAODPid::operator=(const AliAODPid& pid)
  84 {
  85   // Assignment operator
  86   if(this!=&pid) {
  87     // copy stuff
  88     fITSsignal=pid.fITSsignal;
  89     fTPCsignal=pid.fTPCsignal;
  90
  91     if(pid.fTRDnSlices<=0||(fTRDnSlices!=pid.fTRDnSlices)){
  92       // only delete if number changed or is 0
  93       delete [] fTRDslices;
  94       fTRDslices = 0;
  95       if(pid.fTRDnSlices>0) fTRDslices = new Double32_t[fTRDnSlices];
  96     }
  97     fTRDnSlices=pid.fTRDnSlices;
  98
  99     for(Int_t i=0; i< fTRDnSlices; i++) fTRDslices[i]=pid.fTRDslices[i];
 100     fTOFesdsignal=pid.fTOFesdsignal;
 101     fHMPIDsignal=pid.fHMPIDsignal;
 102     for(Int_t i=0; i<kSPECIES; i++) fIntTime[i]=pid.fIntTime[i];
 103     for(Int_t i=0; i<6; i++) fTRDmomentum[i]=pid.fTRDmomentum[i];
 104     for(Int_t i=0; i<3; i++) {
 105       fEMCALPosition[i]=pid.fEMCALPosition[i];
 106       fEMCALMomentum[i]=pid.fEMCALMomentum[i];
 107     }
 108     for (Int_t i=0; i<5; i++) fTOFpidResolution[i]=pid.fTOFpidResolution[i];
 109   }
 110
 111   return *this;
 112 }
 113 //_______________________________________________________________________________
 114 void AliAODPid::GetIntegratedTimes(Double_t timeint[kSPECIES]) const
 115 {
 116  // Returns the array with integrated times for each particle hypothesis
 117 for(Int_t i=0; i<kSPECIES; i++) timeint[i]=fIntTime[i];
 118 }
 119 //_______________________________________________________________________________
 120 void AliAODPid::SetIntegratedTimes(Double_t timeint[kSPECIES])
 121 {
 122  // Returns the array with integrated times for each particle hypothesis
 123 for(Int_t i=0; i<kSPECIES; i++) fIntTime[i]=timeint[i];
 124 }
 125 //_______________________________________________________________________________
 126 void AliAODPid::GetEMCALPosition(Double_t emcalpos[3]) const
 127 {
 128  // Returns the array with extrapolated track position at the EMCAL surface
 129   for(Int_t i=0; i<3; i++) emcalpos[i]=fEMCALPosition[i];
 130 }
 131 //_______________________________________________________________________________
 132 void AliAODPid::SetEMCALPosition(Double_t emcpos[3])
 133 {
 134  // Sets the array with extrapolated track position at the EMCAL surface
 135   for(Int_t i=0; i<3; i++) fEMCALPosition[i]=emcpos[i];
 136 }
 137 //_______________________________________________________________________________
 138 void AliAODPid::GetEMCALMomentum(Double_t emcalmom[3]) const
 139 {
 140  // Returns the array with extrapolated track momentum at the EMCAL surface
 141   for(Int_t i=0; i<3; i++) emcalmom[i]=fEMCALMomentum[i];
 142 }
 143 //_______________________________________________________________________________
 144 void AliAODPid::SetEMCALMomentum(Double_t emcmom[3])
 145 {
 146  // Sets the array with extrapolated track momentum at the EMCAL surface
 147   for(Int_t i=0; i<3; i++) fEMCALMomentum[i]=emcmom[i];
 148 }
 149 //______________________________________________________________________________
 150 void AliAODPid::SetTOFpidResolution(Double_t tofPIDres[5])
 151 {
 152   for (Int_t i=0; i<5; i++) fTOFpidResolution[i]=tofPIDres[i];
 153
 154 }
 155 //______________________________________________________________________________
 156 void AliAODPid::GetTOFpidResolution(Double_t tofRes[5]) const
 157 {
 158   for (Int_t i=0; i<5; i++) tofRes[i]=fTOFpidResolution[i];
 159 }
 160