Corrected protection.
[u/mrichter/AliRoot.git] / STEER / AliESDCaloCells.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 //     ESD class to store calorimeter cell data
20 //     Clone of AliAODCaloCells made by Markus Oldenburg, CERN
21 //     Author: Gustavo Conesa Balbastre INFN-LNF
22 //-------------------------------------------------------------------------
23
24 #include "AliESDCaloCells.h"
25
26 ClassImp(AliESDCaloCells)
27
28 //_______________________________________________________________________
29 AliESDCaloCells::AliESDCaloCells() : 
30   AliVCaloCells(), fNCells(0), fCellNumber(0), fAmplitude(0), fTime(0), fIsSorted(kTRUE), fType(kUndef)
31 {
32   // default constructor
33 }
34 //_______________________________________________________________________
35  AliESDCaloCells::AliESDCaloCells(const char* name, const char* title, VCells_t ttype) : 
36    AliVCaloCells(name, title), fNCells(0), fCellNumber(0), fAmplitude(0),  fTime(0), fIsSorted(kTRUE), fType(ttype)
37 {
38    // AliVCaloCells constructor
39  }
40
41 //_______________________________________________________________________
42 AliESDCaloCells::AliESDCaloCells(const AliESDCaloCells& c) : 
43   AliVCaloCells(c), fNCells(c.fNCells),  fCellNumber(0), fAmplitude(0), fTime(0), fIsSorted(c.fIsSorted), fType(c.fType)
44 {
45   // copy constructor
46
47   fCellNumber = new Short_t[fNCells];
48   fAmplitude  = new Double32_t[fNCells];
49   
50   for(Int_t i = 0; i < fNCells; i++){
51     fCellNumber[i]    = c.fCellNumber[i];
52     fAmplitude[i]     = c.fAmplitude[i];
53     fTime[i]          = c.fTime[i];
54     fAmplitude[i]     = c.fAmplitude[i];
55
56   }
57 }
58
59 //_______________________________________________________________________
60 AliESDCaloCells & AliESDCaloCells::operator =(const AliESDCaloCells& source)  
61 {
62   // assignment operator
63
64   if(&source == this) return *this;
65
66   if(fNCells != source.fNCells){
67     DeleteContainer();
68     CreateContainer(source.fNCells);
69   }
70
71   fNCells = source.fNCells; 
72   fIsSorted = source.fIsSorted;
73   fType = source.fType;
74
75
76
77   for(Int_t i = 0; i < fNCells; i++){
78     fCellNumber[i]    = source.fCellNumber[i];
79     fAmplitude[i]     = source.fAmplitude[i];
80     fTime[i]          = source.fTime[i];
81   }
82
83   return *this;
84
85 }
86
87 //_______________________________________________________________________
88 void AliESDCaloCells::Copy(TObject &obj) const {
89   
90   // this overwrites the virtual TOBject::Copy()
91   // to allow run time copying without casting
92   // in AliESDEvent
93
94   if(this==&obj)return;
95   AliESDCaloCells *robj = dynamic_cast<AliESDCaloCells*>(&obj);
96   if(!robj)return; // not an AliESDCaloCells
97   *robj = *this;
98
99 }
100
101 //_______________________________________________________________________
102 AliVCaloCells* AliESDCaloCells::CopyCaloCells(Bool_t all = kTRUE) const {
103   
104   // copy the calo cells into a new object. If option all=FALSE, just the object type, 
105   // for mixing
106   
107   AliVCaloCells *obj = new AliESDCaloCells();
108   
109   if(all){
110     obj->SetName (GetName()) ; 
111     obj->SetTitle(GetTitle()) ; 
112     obj->SetType (GetType()) ; 
113     
114     obj->SetNumberOfCells(fNCells);
115     for (Short_t i = 0; i < fNCells; i++) 
116       obj->SetCell(i,fCellNumber[i],fAmplitude[i],fTime[i]);
117   }
118
119   return obj;
120 }
121
122
123 //_______________________________________________________________________
124 AliESDCaloCells::~AliESDCaloCells()
125 {
126   // destructor
127
128   DeleteContainer();
129 }
130
131 //_______________________________________________________________________
132 void AliESDCaloCells::CreateContainer(Short_t nCells)
133 {
134   // function that creates container to store calorimeter cell data
135
136   DeleteContainer();
137   
138   if (nCells <= 0) {
139     fNCells = 0;
140     return;
141   }
142
143   fNCells = nCells;
144
145   fCellNumber = new Short_t[fNCells];
146   fAmplitude  = new Double32_t[fNCells];
147   fTime       = new Double32_t[fNCells];
148   
149     // set to zero
150   for(int i = 0;i<fNCells;++i){
151     fAmplitude[i] = fCellNumber[i] = 0 ;
152   }
153 }
154
155 //_______________________________________________________________________
156 void AliESDCaloCells::DeleteContainer()
157 {
158   // deletes allocated memory
159
160   if (fCellNumber)
161   {
162     delete[] fCellNumber;
163     fCellNumber = 0;
164   }
165
166   if (fAmplitude)
167   {
168     delete[] fAmplitude;
169     fAmplitude = NULL;
170   }
171
172   if (fTime)
173   {
174     delete[] fTime;
175     fTime = NULL;
176   }
177   
178   fNCells = 0;
179   fIsSorted = kFALSE;
180 }
181
182 //_______________________________________________________________________
183 void AliESDCaloCells::Sort() 
184 {
185   // sort the cell array by cell number
186   
187   Int_t *idxArray = new Int_t[fNCells];
188   TMath::Sort(fNCells,fCellNumber,idxArray,kFALSE);
189   
190   Short_t *newIndex        = new Short_t[fNCells];
191   Double32_t *newAmplitude = new Double32_t[fNCells];
192   Double32_t *newTime      = new Double32_t[fNCells];
193   for (Int_t i=0; i < fNCells; i++) {
194     newIndex[i]     = fCellNumber[idxArray[i]];
195     newAmplitude[i] = fAmplitude[idxArray[i]];
196     newTime[i]      = fTime[idxArray[i]];
197   }
198   delete [] fCellNumber;
199   delete [] fAmplitude;
200   delete [] fTime;
201
202   fCellNumber = newIndex;
203   fAmplitude  = newAmplitude;
204   fTime       = newTime;
205
206   delete [] idxArray;
207   
208   fIsSorted = kTRUE;
209
210
211 //_______________________________________________________________________
212 Bool_t AliESDCaloCells::SetCell(Short_t pos, Short_t cellNumber, Double32_t amplitude, Double32_t  time)
213 {
214   // Sets a cell at the given position
215
216   if (pos>=0 && pos < fNCells) {
217     fCellNumber[pos] = cellNumber;
218     fAmplitude[pos]  = amplitude;
219     fTime[pos]       = time;
220
221     fIsSorted = kFALSE;
222     return kTRUE;
223   } else {
224     return kFALSE;
225   }
226 }