Bug corrected.
[u/mrichter/AliRoot.git] / 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   
45 }
46
47 //______________________________________________________________________________
48 AliAODPid::~AliAODPid() 
49 {
50   delete [] fTRDslices;
51   fTRDslices = 0;
52   // destructor
53 }
54
55
56 //______________________________________________________________________________
57 AliAODPid::AliAODPid(const AliAODPid& pid) : 
58   TObject(pid),
59   fITSsignal(pid.fITSsignal), 
60   fTPCsignal(pid.fTPCsignal),
61   fTPCsignalN(pid.fTPCsignalN),
62   fTPCmomentum(pid.fTPCmomentum),
63   fTRDnSlices(pid.fTRDnSlices),
64   fTRDslices(0x0),
65   fTOFesdsignal(pid.fTOFesdsignal),
66   fHMPIDsignal(pid.fHMPIDsignal)
67 {
68   // Copy constructor
69     fTRDslices = new Double32_t[fTRDnSlices];
70     for(Int_t i=0; i< fTRDnSlices; i++) fTRDslices[i]=pid.fTRDslices[i];
71     for(Int_t i=0; i<kSPECIES; i++) fIntTime[i]=pid.fIntTime[i];
72     for(Int_t i=0; i<3; i++) fEMCALPosition[i]=pid.fEMCALPosition[i];
73 }
74
75 //______________________________________________________________________________
76 AliAODPid& AliAODPid::operator=(const AliAODPid& pid)
77 {
78   // Assignment operator
79   if(this!=&pid) {
80     // copy stuff
81     fITSsignal=pid.fITSsignal; 
82     fTPCsignal=pid.fTPCsignal;
83     
84     if(pid.fTRDnSlices<=0||(fTRDnSlices!=pid.fTRDnSlices)){
85       // only delete if number changed or is 0
86       delete [] fTRDslices;
87       fTRDslices = 0;
88       if(pid.fTRDnSlices>0) fTRDslices = new Double32_t[fTRDnSlices];
89     }
90     fTRDnSlices=pid.fTRDnSlices;
91     
92     for(Int_t i=0; i< fTRDnSlices; i++) fTRDslices[i]=pid.fTRDslices[i];
93     fTOFesdsignal=pid.fTOFesdsignal;
94     fHMPIDsignal=pid.fHMPIDsignal;
95     for(Int_t i=0; i<kSPECIES; i++) fIntTime[i]=pid.fIntTime[i];
96     for(Int_t i=0; i<3; i++) fEMCALPosition[i]=pid.fEMCALPosition[i];
97   }
98
99   return *this;
100 }
101 //_______________________________________________________________________________
102 void AliAODPid::GetIntegratedTimes(Double_t timeint[kSPECIES]) const
103 {
104  // Returns the array with integrated times for each particle hypothesis
105 for(Int_t i=0; i<kSPECIES; i++) timeint[i]=fIntTime[i];
106 }
107 //_______________________________________________________________________________
108 void AliAODPid::SetIntegratedTimes(Double_t timeint[kSPECIES])
109 {
110  // Returns the array with integrated times for each particle hypothesis
111 for(Int_t i=0; i<kSPECIES; i++) fIntTime[i]=timeint[i];
112 }
113 //_______________________________________________________________________________
114 void AliAODPid::GetEMCALPosition(Double_t emcalpos[3]) const
115 {
116  // Returns the array with extrapolated track position at the EMCAL surface
117   for(Int_t i=0; i<3; i++) emcalpos[i]=fEMCALPosition[i];
118 }
119 //_______________________________________________________________________________
120 void AliAODPid::SetEMCALPosition(Double_t emcpos[3])
121 {
122  // Sets the array with extrapolated track position at the EMCAL surface
123   for(Int_t i=0; i<3; i++) fEMCALPosition[i]=emcpos[i];
124 }
125 //_______________________________________________________________________________
126 void AliAODPid::GetEMCALMomentum(Double_t emcalmom[3]) const
127 {
128  // Returns the array with extrapolated track momentum at the EMCAL surface
129   for(Int_t i=0; i<3; i++) emcalmom[i]=fEMCALMomentum[i];
130 }
131 //_______________________________________________________________________________
132 void AliAODPid::SetEMCALMomentum(Double_t emcmom[3])
133 {
134  // Sets the array with extrapolated track momentum at the EMCAL surface
135   for(Int_t i=0; i<3; i++) fEMCALMomentum[i]=emcmom[i];
136 }