Coverity 16571
[u/mrichter/AliRoot.git] / STEER / AliAODCluster.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 cluster base class
20 //     Author: Markus Oldenburg, CERN
21 //-------------------------------------------------------------------------
22
23 #include "AliAODCluster.h"
24
25 ClassImp(AliAODCluster)
26
27 //______________________________________________________________________________
28 AliAODCluster::AliAODCluster() :
29   AliVCluster(),
30   fEnergy(0),
31   fChi2(-999.),
32   fID(-999),
33   fNLabel(0),
34   fLabel(0x0),
35   fFilterMap(0),
36   fType(kUndef)
37 {
38   // default constructor
39
40   SetPosition(NULL);
41   SetPID(NULL);
42 }
43
44 //______________________________________________________________________________
45 AliAODCluster::AliAODCluster(Int_t id,
46                              UInt_t nLabel,
47                              Int_t *label, 
48                              Double_t energy,
49                              Double_t x[3],
50                              Double_t pid[13],
51                              Char_t ttype,
52                              UInt_t selectInfo) :
53   AliVCluster(),
54   fEnergy(energy),
55   fChi2(-999.),
56   fID(id),
57   fNLabel(0),
58   fLabel(0x0),
59   fFilterMap(selectInfo),
60   fType(ttype)
61 {
62   // constructor
63   for (Int_t i = 0; i <  3; i++) fPosition[i] = 0.;
64   for (Int_t i = 0; i < 13; i++) fPID[i]      = 0;
65  
66   if(x)   {for (Int_t i = 0; i < 3  ; i++) SetPositionAt(x[i]  ,i);}
67   if(pid) {for (Int_t i = 0; i < 13 ; i++) SetPIDAt     (pid[i],i);}
68   SetLabel(label, nLabel);
69 }
70
71 //______________________________________________________________________________
72 AliAODCluster::AliAODCluster(Int_t id,
73                              UInt_t nLabel,
74                              Int_t *label, 
75                              Float_t energy,
76                              Float_t x[3],
77                              Float_t pid[13],
78                              Char_t ttype,
79                              UInt_t selectInfo) :
80   AliVCluster(),
81   fEnergy(energy),
82   fChi2(-999.),
83   fID(id),
84   fNLabel(0),
85   fLabel(0x0),
86   fFilterMap(selectInfo),
87   fType(ttype)
88 {
89   // constructor
90   for (Int_t i = 0; i <  3; i++) fPosition[i] = 0.;
91   for (Int_t i = 0; i < 13; i++) fPID[i]      = 0;
92
93   if(x)   {for (Int_t i = 0; i < 3  ; i++) SetPositionAt(x[i]  ,i);}
94   if(pid) {for (Int_t i = 0; i < 13 ; i++) SetPIDAt     (pid[i],i);}
95   SetLabel(label, nLabel);
96 }
97
98
99 //______________________________________________________________________________
100 AliAODCluster::~AliAODCluster() 
101 {
102   // destructor
103
104   RemoveLabel();
105 }
106
107 //______________________________________________________________________________
108 void AliAODCluster::Clear(const Option_t*) 
109 {
110   // Clear
111   
112   RemoveLabel();
113 }
114
115
116 //______________________________________________________________________________
117 AliAODCluster::AliAODCluster(const AliAODCluster& clus) :
118   AliVCluster(clus),
119   fEnergy(clus.fEnergy),
120   fChi2(clus.fChi2),
121   fID(clus.fID),
122   fNLabel(0),
123   fLabel(0x0),
124   fFilterMap(clus.fFilterMap),
125   fType(clus.fType)
126 {
127   // Copy constructor
128
129   for(Int_t i = 0; i < 3  ; i++) fPosition[i]  = clus.fPosition[i];
130   for(Int_t i = 0; i < 13 ; i++)  fPID[i]      = clus.fPID[i];
131
132   SetLabel(clus.fLabel, clus.fNLabel);
133 }
134
135 //______________________________________________________________________________
136 AliAODCluster& AliAODCluster::operator=(const AliAODCluster& clus)
137 {
138   // Assignment operator
139   if(this!=&clus) {
140     
141         for(Int_t i = 0; i < 3 ;  i++) fPosition[i] = clus.fPosition[i];
142         for(Int_t i = 0; i < 13 ; i++) fPID[i]      = clus.fPID[i];
143     
144     fEnergy = clus.fEnergy;
145     fChi2 = clus.fChi2;
146
147     fID = clus.fID;
148     SetLabel(clus.fLabel, clus.fNLabel);
149     fFilterMap = clus.fFilterMap;
150
151     fType = clus.fType;
152   }
153
154   return *this;
155 }
156
157 //______________________________________________________________________________
158 void AliAODCluster::SetPosition(Float_t *x) 
159 {
160   // set the position
161   
162   if (x) {
163     fPosition[0] = x[0];
164     fPosition[1] = x[1];
165     fPosition[2] = x[2];
166   } else {
167     fPosition[0] = -999.;
168     fPosition[1] = -999.;
169     fPosition[2] = -999.;
170   }
171 }
172
173 //______________________________________________________________________________
174 UShort_t AliAODCluster::GetMostProbablePID() const 
175 {
176   // Returns the most probable PID array element.
177   
178   Int_t nPID = 13;
179   UShort_t unknown = AliVCluster::kUnknown;
180   
181   UShort_t loc = unknown;
182   Double_t max = 0.;
183   Bool_t allTheSame = kTRUE;
184   
185   for (Int_t iPID = 0; iPID < nPID; iPID++) {
186     if (fPID[iPID] >= max) {
187       if (fPID[iPID] > max) {
188         allTheSame = kFALSE;
189         max = fPID[iPID];
190         loc = (UShort_t)iPID;
191       } else {
192         allTheSame = kTRUE;
193       }
194     }
195   }
196   return allTheSame ? unknown : loc;
197 }
198
199 //______________________________________________________________________________
200 void AliAODCluster::SetLabel(Int_t *label, UInt_t size) 
201 {
202   if (label && size>0) {
203     if (size != (UInt_t)fNLabel) {
204       RemoveLabel();
205       fNLabel = size;
206       fLabel = new Int_t[fNLabel];
207     }
208     
209     for (Int_t i = 0; i < fNLabel; i++) {
210       fLabel[i] = label[i];
211     }
212   } else {
213     RemoveLabel();
214   }
215
216   return;
217 }
218
219 //______________________________________________________________________________
220 Int_t AliAODCluster::GetLabelAt(UInt_t i) const
221 {
222   if (fLabel && i < (UInt_t)fNLabel) {
223     return fLabel[i];
224   } else {
225     return -999;
226   }
227 }
228
229 //______________________________________________________________________________
230 void AliAODCluster::RemoveLabel()
231 {
232   delete[] fLabel;
233   fLabel = 0x0;
234   fNLabel = 0;
235
236   return;
237 }
238
239 //______________________________________________________________________________
240 void AliAODCluster::Print(Option_t* /* option */) const
241 {
242   // prints information about AliAODCluster
243
244   printf("Cluster type: %d\n", GetType()); 
245   printf("     energy = %f\n", E());
246   printf("       chi2 = %f\n", Chi2());
247   const Double_t *pid = GetPID();
248   printf("PID weights: photon %0.2f, pi0 %0.2f, electron %0.2f, conversion electron %0.2f\n, hadrons: pion %0.2f, kaon %0.2f, proton %0.2f , neutron %0.2f, kaon %0.2f \n",
249          pid[AliVCluster::kPhoton],   pid[AliVCluster::kPi0],
250          pid[AliVCluster::kElectron], pid[AliVCluster::kEleCon],
251          pid[AliVCluster::kPion],     pid[AliVCluster::kKaon],   pid[AliVCluster::kProton],
252          pid[AliVCluster::kNeutron],  pid[AliVCluster::kKaon0]);
253 }