STEER/AOD/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 #include "AliPID.h"
  27 #include "AliTPCdEdxInfo.h"
  28
  29 ClassImp(AliAODPid)
  30
  31
  32 //______________________________________________________________________________
  33 AliAODPid::AliAODPid():
  34     fITSsignal(0),
  35     fTPCsignal(0),
  36     fTPCsignalN(0),
  37     fTPCmomentum(0),
  38     fTRDnSlices(0),
  39     fTRDntls(0),
  40     fTRDslices(0x0),
  41     fTRDChi2(0x0),
  42     fTOFesdsignal(0),
  43     fTPCdEdxInfo(0)
  44 {
  45   // default constructor
  46     for(Int_t i=0; i<AliPID::kSPECIES; i++) fIntTime[i]   = 0;
  47     for(Int_t i=0; i<5; i++) fTOFpidResolution[i] = 0.;
  48     for(Int_t i=0; i<6; i++) {
  49       fTRDmomentum[i]      = 0.;
  50       fTRDncls[i]          = 0;
  51     }
  52     for(Int_t i=0; i<4; i++) fITSdEdxSamples[i]   = 0.;
  53 }
  54
  55 //______________________________________________________________________________
  56 AliAODPid::~AliAODPid()
  57 {
  58   delete [] fTRDslices;
  59   fTRDslices = 0;
  60   delete fTPCdEdxInfo;
  61   // destructor
  62 }
  63
  64
  65 //______________________________________________________________________________
  66 AliAODPid::AliAODPid(const AliAODPid& pid) :
  67   TObject(pid),
  68   fITSsignal(pid.fITSsignal),
  69   fTPCsignal(pid.fTPCsignal),
  70   fTPCsignalN(pid.fTPCsignalN),
  71   fTPCmomentum(pid.fTPCmomentum),
  72   fTRDnSlices(pid.fTRDnSlices),
  73   fTRDntls(pid.fTRDntls),
  74   fTRDslices(0x0),
  75   fTRDChi2(pid.fTRDChi2),
  76   fTOFesdsignal(pid.fTOFesdsignal),
  77   fTPCdEdxInfo(0x0)
  78 {
  79   // Copy constructor
  80   SetTRDsignal(fTRDnSlices, pid.fTRDslices);
  81     for(Int_t i=0; i<AliPID::kSPECIES; i++) fIntTime[i]=pid.fIntTime[i];
  82
  83     for(Int_t i=0; i<6; i++){
  84       fTRDmomentum[i]=pid.fTRDmomentum[i];
  85       fTRDncls[i] = 0;
  86     }
  87
  88     for(Int_t i=0; i<5; i++) fTOFpidResolution[i]=pid.fTOFpidResolution[i];
  89
  90     for(Int_t i=0; i<4; i++) fITSdEdxSamples[i]=pid.fITSdEdxSamples[i];
  91
  92     if (pid.fTPCdEdxInfo) fTPCdEdxInfo=new AliTPCdEdxInfo(*pid.fTPCdEdxInfo);
  93 }
  94
  95 //______________________________________________________________________________
  96 AliAODPid& AliAODPid::operator=(const AliAODPid& pid)
  97 {
  98   // Assignment operator
  99   if(this!=&pid) {
 100     // copy stuff
 101     TObject::operator=(pid);
 102
 103     fITSsignal   = pid.fITSsignal;
 104     for (Int_t i = 0; i < 4; i++) fITSdEdxSamples[i]=pid.fITSdEdxSamples[i];
 105     fTPCsignal   = pid.fTPCsignal;
 106     fTPCsignalN  = pid.fTPCsignalN;
 107     fTPCmomentum = pid.fTPCmomentum;
 108
 109
 110     if(fTRDnSlices != pid.fTRDnSlices) {
 111       // only delete if number changed or is 0
 112       delete [] fTRDslices;
 113       fTRDslices = 0;
 114       fTRDnSlices = pid.fTRDnSlices;
 115       if(pid.fTRDnSlices > 0) fTRDslices = new Double32_t[fTRDnSlices];
 116     }
 117
 118     if (fTRDslices && pid.fTRDslices)
 119       memcpy(fTRDslices, pid.fTRDslices, fTRDnSlices*sizeof(Double32_t));
 120
 121     fTRDntls = pid.fTRDntls;
 122     for(Int_t i = 0; i < 6; i++){
 123         fTRDmomentum[i] = pid.fTRDmomentum[i];
 124         fTRDncls[i]     = pid.fTRDncls[i];
 125     }
 126
 127     fTRDChi2 = pid.fTRDChi2;
 128
 129     fTOFesdsignal=pid.fTOFesdsignal;
 130     for (Int_t i = 0; i < 5; i++) fTOFpidResolution[i]=pid.fTOFpidResolution[i];
 131     for (Int_t i = 0; i < 5; i++) fIntTime[i]=pid.fIntTime[i];
 132
 133      SetTPCdEdxInfo(pid.fTPCdEdxInfo);
 134   }
 135
 136   return *this;
 137 }
 138 //_______________________________________________________________________________
 139 void AliAODPid::GetIntegratedTimes(Double_t timeint[AliPID::kSPECIES]) const
 140 {
 141  // Returns the array with integrated times for each particle hypothesis
 142  for(Int_t i=0; i<AliPID::kSPECIES; i++) timeint[i]=fIntTime[i];
 143 }
 144 //_______________________________________________________________________________
 145 void AliAODPid::SetIntegratedTimes(Double_t timeint[AliPID::kSPECIES])
 146 {
 147  // Returns the array with integrated times for each particle hypothesis
 148  for(Int_t i=0; i<AliPID::kSPECIES; i++) fIntTime[i]=timeint[i];
 149 }
 150 //______________________________________________________________________________
 151 void AliAODPid::SetTOFpidResolution(Double_t tofPIDres[5])
 152 {
 153   for (Int_t i=0; i<5; i++) fTOFpidResolution[i]=tofPIDres[i];
 154
 155 }
 156 //______________________________________________________________________________
 157 void AliAODPid::GetTOFpidResolution(Double_t tofRes[5]) const
 158 {
 159   for (Int_t i=0; i<5; i++) tofRes[i]=fTOFpidResolution[i];
 160 }
 161
 162 //______________________________________________________________________________
 163 void AliAODPid::SetITSdEdxSamples(const Double_t s[4])
 164 {
 165   //
 166   // Set the 4 values of dE/dx from individual ITS layers that are read from ESD
 167   //
 168   for (Int_t i=0; i<4; i++) fITSdEdxSamples[i]=s[i];
 169 }
 170 //______________________________________________________________________________
 171 void AliAODPid::GetITSdEdxSamples(Double_t s[4]) const
 172 {
 173   //
 174   // Get the 4 values of dE/dx from individual ITS layers that are read from ESD
 175   //
 176   for (Int_t i=0; i<4; i++) s[i]=fITSdEdxSamples[i];
 177 }
 178 //______________________________________________________________________________
 179 void AliAODPid::SetTPCdEdxInfo(AliTPCdEdxInfo * dEdxInfo)
 180 {
 181   //
 182   // Set TPC dEdx info
 183   //
 184   if (dEdxInfo==0x0){
 185     delete fTPCdEdxInfo;
 186     fTPCdEdxInfo=0x0;
 187     return;
 188   }
 189   if (!fTPCdEdxInfo) fTPCdEdxInfo=new AliTPCdEdxInfo;
 190   (*fTPCdEdxInfo)=(*dEdxInfo);
 191 }
 192