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