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