]> git.uio.no Git - u/mrichter/AliRoot.git/blobdiff - STEER/AliAODPid.cxx
new functions, mc analysis can be performed also for pi0 and k0. eta of seed can...
[u/mrichter/AliRoot.git] / STEER / AliAODPid.cxx
index 4293476f18953fb6397b682c5d85c9b4e33a854c..41a5e2a4f65b5f8b7b4b0525170974de3ce070a8 100644 (file)
 
 //-------------------------------------------------------------------------
 //     AOD Pid class to store additional pid information
-//     Author: Markus Oldenburg, CERN
+//     Author: Annalisa Mastroserio
 //-------------------------------------------------------------------------
 
 #include "AliAODPid.h"
+#include "AliESDtrack.h"
+#include "AliLog.h"
 
 ClassImp(AliAODPid)
 
+
 //______________________________________________________________________________
-  AliAODPid::AliAODPid()
+AliAODPid::AliAODPid():
+    fITSsignal(0), 
+    fTPCsignal(0),
+    fTPCsignalN(0),
+    fTPCmomentum(0),
+    fTRDnSlices(0),
+    fTRDslices(0x0),
+    fTOFesdsignal(0),
+    fHMPIDsignal(0)
 {
   // default constructor
+    for(Int_t i=0; i<kSPECIES; i++) fIntTime[i]=0; 
+    for(Int_t i=0; i<3; i++) fEMCALPosition[i] = 0.;
+    for(Int_t i=0; i<5; i++) fTOFpidResolution[i] = 0.;
+  
 }
 
 //______________________________________________________________________________
 AliAODPid::~AliAODPid() 
 {
+  delete [] fTRDslices;
+  fTRDslices = 0;
   // destructor
 }
 
 
 //______________________________________________________________________________
 AliAODPid::AliAODPid(const AliAODPid& pid) : 
-  TObject(pid)
+  TObject(pid),
+  fITSsignal(pid.fITSsignal), 
+  fTPCsignal(pid.fTPCsignal),
+  fTPCsignalN(pid.fTPCsignalN),
+  fTPCmomentum(pid.fTPCmomentum),
+  fTRDnSlices(pid.fTRDnSlices),
+  fTRDslices(0x0),
+  fTOFesdsignal(pid.fTOFesdsignal),
+  fHMPIDsignal(pid.fHMPIDsignal)
 {
   // Copy constructor
+    fTRDslices = new Double32_t[fTRDnSlices];
+    for(Int_t i=0; i< fTRDnSlices; i++) fTRDslices[i]=pid.fTRDslices[i];
+    for(Int_t i=0; i<kSPECIES; i++) fIntTime[i]=pid.fIntTime[i];
+    for(Int_t i=0; i<3; i++) {
+      fEMCALPosition[i]=pid.fEMCALPosition[i];
+      fEMCALMomentum[i]=pid.fEMCALMomentum[i];
+    }
+    for(Int_t i=0; i<6; i++) fTRDmomentum[i]=pid.fTRDmomentum[i];
+
+    for(Int_t i=0; i<5; i++) fTOFpidResolution[i]=pid.fTOFpidResolution[i];
 }
 
 //______________________________________________________________________________
@@ -50,7 +85,76 @@ AliAODPid& AliAODPid::operator=(const AliAODPid& pid)
   // Assignment operator
   if(this!=&pid) {
     // copy stuff
+    fITSsignal=pid.fITSsignal; 
+    fTPCsignal=pid.fTPCsignal;
+    
+    if(pid.fTRDnSlices<=0||(fTRDnSlices!=pid.fTRDnSlices)){
+      // only delete if number changed or is 0
+      delete [] fTRDslices;
+      fTRDslices = 0;
+      if(pid.fTRDnSlices>0) fTRDslices = new Double32_t[fTRDnSlices];
+    }
+    fTRDnSlices=pid.fTRDnSlices;
+    
+    for(Int_t i=0; i< fTRDnSlices; i++) fTRDslices[i]=pid.fTRDslices[i];
+    fTOFesdsignal=pid.fTOFesdsignal;
+    fHMPIDsignal=pid.fHMPIDsignal;
+    for(Int_t i=0; i<kSPECIES; i++) fIntTime[i]=pid.fIntTime[i];
+    for(Int_t i=0; i<6; i++) fTRDmomentum[i]=pid.fTRDmomentum[i];
+    for(Int_t i=0; i<3; i++) {
+      fEMCALPosition[i]=pid.fEMCALPosition[i];
+      fEMCALMomentum[i]=pid.fEMCALMomentum[i];
+    }
+    for (Int_t i=0; i<5; i++) fTOFpidResolution[i]=pid.fTOFpidResolution[i];
   }
 
   return *this;
 }
+//_______________________________________________________________________________
+void AliAODPid::GetIntegratedTimes(Double_t timeint[kSPECIES]) const
+{
+ // Returns the array with integrated times for each particle hypothesis
+for(Int_t i=0; i<kSPECIES; i++) timeint[i]=fIntTime[i];
+}
+//_______________________________________________________________________________
+void AliAODPid::SetIntegratedTimes(Double_t timeint[kSPECIES])
+{
+ // Returns the array with integrated times for each particle hypothesis
+for(Int_t i=0; i<kSPECIES; i++) fIntTime[i]=timeint[i];
+}
+//_______________________________________________________________________________
+void AliAODPid::GetEMCALPosition(Double_t emcalpos[3]) const
+{
+ // Returns the array with extrapolated track position at the EMCAL surface
+  for(Int_t i=0; i<3; i++) emcalpos[i]=fEMCALPosition[i];
+}
+//_______________________________________________________________________________
+void AliAODPid::SetEMCALPosition(Double_t emcpos[3])
+{
+ // Sets the array with extrapolated track position at the EMCAL surface
+  for(Int_t i=0; i<3; i++) fEMCALPosition[i]=emcpos[i];
+}
+//_______________________________________________________________________________
+void AliAODPid::GetEMCALMomentum(Double_t emcalmom[3]) const
+{
+ // Returns the array with extrapolated track momentum at the EMCAL surface
+  for(Int_t i=0; i<3; i++) emcalmom[i]=fEMCALMomentum[i];
+}
+//_______________________________________________________________________________
+void AliAODPid::SetEMCALMomentum(Double_t emcmom[3])
+{
+ // Sets the array with extrapolated track momentum at the EMCAL surface
+  for(Int_t i=0; i<3; i++) fEMCALMomentum[i]=emcmom[i];
+}
+//______________________________________________________________________________
+void AliAODPid::SetTOFpidResolution(Double_t tofPIDres[5])
+{
+  for (Int_t i=0; i<5; i++) fTOFpidResolution[i]=tofPIDres[i];
+
+}
+//______________________________________________________________________________
+void AliAODPid::GetTOFpidResolution(Double_t tofRes[5]) const
+{
+  for (Int_t i=0; i<5; i++) tofRes[i]=fTOFpidResolution[i];
+}
+