]> git.uio.no Git - u/mrichter/AliRoot.git/blame - STEER/STEERBase/AliMixedEvent.cxx
speed up with binary search
[u/mrichter/AliRoot.git] / STEER / STEERBase / AliMixedEvent.cxx
CommitLineData
7eb061fb 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//-------------------------------------------------------------------------
20// Class AliMixedEvent
21// VEvent which is the container of several VEvents
22// Use Case: Event Mixing
23// Origin: Andreas Morsch, CERN, Andreas.Morsch@cern.ch
24//-------------------------------------------------------------------------
25
26
27#include "AliMixedEvent.h"
bbeb9cae 28#include "AliExternalTrackParam.h"
bbeb9cae 29#include "TVector3.h"
c79face6 30#include "AliVVertex.h"
7eb061fb 31#include <TMath.h>
bbeb9cae 32#include <TMatrix.h>
33#include <TMatrixD.h>
c8fe2783 34#include "AliLog.h"
35#include "AliVCaloCells.h"
36
7eb061fb 37
38ClassImp(AliMixedEvent)
39
40
41AliMixedEvent::AliMixedEvent() :
c8fe2783 42 AliVEvent(),
43 fEventList(),
44 fNEvents(0),
45 fNumberOfTracks(0),
46 fNumberOfCaloClusters(0),
47 fNumberOfPHOSCells(0),
48 fNumberOfEMCALCells(0),
49 fNTracksCumul(0),
50 fNCaloClustersCumul(0),
51 fNPHOSCellsCumul(0),
52 fNEMCALCellsCumul(0),
53 fPHOSCells(NULL),
54 fEMCALCells(NULL),
55 fMeanVertex(0)
7eb061fb 56{
57 // Default constructor
58}
59
7eb061fb 60AliMixedEvent::AliMixedEvent(const AliMixedEvent& Evnt) :
c8fe2783 61 AliVEvent(Evnt),
62 fEventList(),
63 fNEvents(0),
64 fNumberOfTracks(0),
65 fNumberOfCaloClusters(0),
66 fNumberOfPHOSCells(0),
67 fNumberOfEMCALCells(0),
68 fNTracksCumul(0),
69 fNCaloClustersCumul(0),
70 fNPHOSCellsCumul(0),
71 fNEMCALCellsCumul(0),
72 fPHOSCells(NULL),
73 fEMCALCells(NULL),
74 fMeanVertex(0)
a2813d8b 75{ } // Copy constructor
7eb061fb 76
77AliMixedEvent& AliMixedEvent::operator=(const AliMixedEvent& vEvnt)
aa7e002c 78{
79// Assignment operator
80 if (this!=&vEvnt) {
7eb061fb 81 AliVEvent::operator=(vEvnt);
aa7e002c 82}
7eb061fb 83
84 return *this;
85}
86
c8fe2783 87AliMixedEvent::~AliMixedEvent()
88{
89 // dtor
90 Reset();
91 delete fPHOSCells ;
92 delete fEMCALCells ;
93}
94
7eb061fb 95
96void AliMixedEvent::AddEvent(AliVEvent* evt)
97{
98 // Add a new event to the list
bbeb9cae 99 fEventList.AddLast(evt);
7eb061fb 100}
101
102
103void AliMixedEvent::Init()
104{
105 // Initialize meta information
c8fe2783 106 fNEvents = fEventList.GetEntries();
107 fNTracksCumul = new Int_t[fNEvents];
108 fNumberOfTracks = 0;
109 fNCaloClustersCumul = new Int_t[fNEvents];
110 fNumberOfCaloClusters = 0;
111 fNumberOfPHOSCells = 0;
112 fNumberOfEMCALCells = 0;
113 fNPHOSCellsCumul = new Int_t[fNEvents];
114 fNEMCALCellsCumul = new Int_t[fNEvents];
115
116 TIter next(&fEventList);
117 AliVEvent* event;
118 Int_t iev = 0;
7eb061fb 119
c8fe2783 120 while((event = (AliVEvent*)next())) {
121 fNTracksCumul[iev] = fNumberOfTracks;
122 fNumberOfTracks += (event->GetNumberOfTracks());
123 fNCaloClustersCumul[iev] = fNumberOfCaloClusters;
124 fNumberOfCaloClusters += event->GetNumberOfCaloClusters();
125 fNPHOSCellsCumul[iev] = fNumberOfPHOSCells;
126 if (event->GetPHOSCells())
127 fNumberOfPHOSCells += event->GetPHOSCells()->GetNumberOfCells();
128 fNEMCALCellsCumul[iev] = fNumberOfEMCALCells;
129 if (event->GetEMCALCells())
130 fNumberOfEMCALCells += event->GetEMCALCells()->GetNumberOfCells();
131 iev++ ;
132 }
7eb061fb 133
c8fe2783 134 next.Reset() ;
135 Short_t phosPos = 0, emcalPos = 0;
a02655c6 136 Int_t firstPHOSEvent = kTRUE;
137 Int_t firstEMCALEvent = kTRUE;
138
c8fe2783 139 while((event = (AliVEvent*)next())) {
ecff8f07 140 AliVCaloCells * phosCells = event->GetPHOSCells() ;
141 if (phosCells) {
a02655c6 142
143 //Create the container
144 if(firstPHOSEvent)
145 {
ecff8f07 146 if(!fPHOSCells) fPHOSCells = phosCells->CopyCaloCells(kFALSE) ;// Just recover the first event type: ESD/AOD
147 else fPHOSCells->DeleteContainer(); //delete the previous container
148 //Now create a new container with the adequate size
a02655c6 149 fPHOSCells->SetType(AliVCaloCells::kPHOSCell) ;
150 fPHOSCells->CreateContainer(fNumberOfPHOSCells) ;
a02655c6 151 firstPHOSEvent=kFALSE;
ecff8f07 152
153 }//First event
154
155 Int_t ncells = event->GetPHOSCells()->GetNumberOfCells() ;
156 for (Int_t icell = 0; icell < ncells; icell++) {
77e93dc2 157 fPHOSCells->SetCell(phosPos++, phosCells->GetCellNumber(icell), phosCells->GetAmplitude(icell),
158 phosCells->GetTime(icell),phosCells->GetMCLabel(icell),phosCells->GetEFraction(icell)) ;
ecff8f07 159 }
160
161 }// phos cells
162
163 AliVCaloCells * emcalCells = event->GetEMCALCells() ;
164 if (emcalCells) {
a02655c6 165
166 //Create the container
167 if(firstEMCALEvent)
168 {
ecff8f07 169 if(!fEMCALCells)fEMCALCells = emcalCells->CopyCaloCells(kFALSE) ; // Just recover the first event type: ESD/AOD
170 else fEMCALCells->DeleteContainer(); // delete the previous container
171 //Now create a new container with the adequate size
a02655c6 172 fEMCALCells->SetType(AliVCaloCells::kEMCALCell) ;
173 fEMCALCells->CreateContainer(fNumberOfEMCALCells) ;
a02655c6 174 firstEMCALEvent=kFALSE;
ecff8f07 175 }//First event
176
177 Int_t ncells = emcalCells->GetNumberOfCells() ;
178 for (Int_t icell = 0; icell < ncells; icell++) {
77e93dc2 179 fEMCALCells->SetCell(emcalPos++, emcalCells->GetCellNumber(icell), emcalCells->GetAmplitude(icell),
180 emcalCells->GetTime(icell),emcalCells->GetMCLabel(icell),emcalCells->GetEFraction(icell)) ;
ecff8f07 181 }
182 }//EMCAL cells
183 }//while event
184
c8fe2783 185}
7eb061fb 186
187AliVParticle* AliMixedEvent::GetTrack(Int_t i) const
188{
189 // Return track # i
190 Int_t iEv = TMath::BinarySearch(fNEvents, fNTracksCumul, i);
18476ce9 191 while((iEv < (fNEvents - 1)) && (fNTracksCumul[iEv] == fNTracksCumul[iEv+1])) {iEv++;}
bbeb9cae 192
7eb061fb 193 Int_t irel = i - fNTracksCumul[iEv];
194 AliVEvent* evt = (AliVEvent*) (fEventList.At(iEv));
195 return (evt->GetTrack(irel));
196}
197
c8fe2783 198AliVCluster* AliMixedEvent::GetCaloCluster(Int_t i) const
199{
200 // Return calo cluster # i
201 Int_t iEv = TMath::BinarySearch(fNEvents, fNCaloClustersCumul, i);
202 while((iEv < (fNEvents - 1)) && (fNCaloClustersCumul[iEv] == fNCaloClustersCumul[iEv+1])) {iEv++;}
203
204 Int_t irel = i - fNCaloClustersCumul[iEv];
205 AliVEvent* evt = (AliVEvent*) (fEventList.At(iEv));
206 return (evt->GetCaloCluster(irel));
207}
208
bbeb9cae 209const AliVVertex* AliMixedEvent::GetEventVertex(Int_t i) const
210{
c8fe2783 211 // Return vertex of track # i
bbeb9cae 212 Int_t iEv = TMath::BinarySearch(fNEvents, fNTracksCumul, i);
213 while((iEv < (fNEvents - 1)) && (fNTracksCumul[iEv] == fNTracksCumul[iEv+1])) {iEv++;}
214 AliVEvent* evt = (AliVEvent*) (fEventList.At(iEv));
215 return (evt->GetPrimaryVertex());
216}
7eb061fb 217
c8fe2783 218const AliVVertex* AliMixedEvent::GetVertexOfEvent(Int_t i) const
219{
220 // Return vertex of event # i
221 if (i > fNEvents)
222 AliFatal(Form("%d events in buffer, event %d requested", fNEvents, i)) ;
223 AliVEvent* evt = (AliVEvent*) (fEventList.At(i));
224 return (evt->GetPrimaryVertex());
225}
226
7eb061fb 227void AliMixedEvent::Reset()
228{
229 // Reset the event
c8fe2783 230 fEventList.Clear();
231 fNEvents = 0;
232 fNumberOfTracks = 0;
233 fNumberOfCaloClusters = 0;
234 fNumberOfPHOSCells = 0;
235 fNumberOfEMCALCells = 0;
236 if (fNTracksCumul) {
237 delete[] fNTracksCumul;
238 fNTracksCumul = 0;
239 }
240 if (fNCaloClustersCumul) {
241 delete[] fNCaloClustersCumul;
242 fNCaloClustersCumul = 0;
243 }
244 if (fNPHOSCellsCumul) {
245 delete[] fNPHOSCellsCumul;
246 fNPHOSCellsCumul = 0;
247 }
248 if (fNEMCALCellsCumul) {
249 delete[] fNEMCALCellsCumul;
250 fNEMCALCellsCumul = 0;
251 }
ecff8f07 252
253 if (fPHOSCells) {
254 fPHOSCells->DeleteContainer();
255 }
256 if (fEMCALCells) {
257 fEMCALCells->DeleteContainer();
258 }
259
7eb061fb 260}
261
c8fe2783 262Int_t AliMixedEvent::EventIndex(Int_t itrack) const
7eb061fb 263{
264 // Return the event index for track #itrack
265 return TMath::BinarySearch(fNEvents, fNTracksCumul, itrack);
266}
bbeb9cae 267
c8fe2783 268Int_t AliMixedEvent::EventIndexForCaloCluster(Int_t icluster) const
269{
270 // Return the event index for track #itrack
271 return TMath::BinarySearch(fNEvents, fNCaloClustersCumul, icluster);
272}
273
274Int_t AliMixedEvent::EventIndexForPHOSCell(Int_t icell) const
275{
276 // Return the event index for track #itrack
277 return TMath::BinarySearch(fNEvents, fNPHOSCellsCumul, icell);
278}
279
280Int_t AliMixedEvent::EventIndexForEMCALCell(Int_t icell) const
281{
282 // Return the event index for track #itrack
283 return TMath::BinarySearch(fNEvents, fNEMCALCellsCumul, icell);
284}
285
aa7e002c 286Bool_t AliMixedEvent::ComputeVtx(const TObjArray *vertices, Double_t *pos,Double_t *sig,Int_t *nContributors) {
bbeb9cae 287//
288// Calculate the mean vertex psoitions from events in the buffer
289
290 Int_t nentries = vertices->GetEntriesFast();
291 Double_t sum[3]={0.,0.,0.};
292 Double_t sumsigma[6]={0.,0.,0.,0.,0.,0.};
b4a900dc 293
bbeb9cae 294
295 for(Int_t ivtx = 0; ivtx < nentries; ivtx++){
c79face6 296 AliVVertex *vtx=(AliVVertex*)vertices->UncheckedAt(ivtx);
bbeb9cae 297 Double_t covariance[6];
c79face6 298 vtx->GetCovarianceMatrix(covariance);
bbeb9cae 299 Double_t vtxPos[3];
300 vtx->GetXYZ(vtxPos);
b4a900dc 301 if(TMath::Abs(covariance[0])<1.e-13) {
302 return kFALSE;
303 }else{
bbeb9cae 304 sum[0]+=vtxPos[0]*(1./covariance[0]);
305 sumsigma[0]+=(1./covariance[0]);
b4a900dc 306 }
307 if(TMath::Abs(covariance[2])<1.e-13) {
308 return kFALSE;
309 }else{
bbeb9cae 310 sum[1]+=vtxPos[1]*(1./covariance[2]);
311 sumsigma[2]+=(1./covariance[2]);
b4a900dc 312 }
313 if(TMath::Abs(covariance[5])<1.e-13) {
314 return kFALSE;
315 }else{
bbeb9cae 316 sum[2]+=vtxPos[2]*(1./covariance[5]);
317 sumsigma[5]+=(1./covariance[5]);
b4a900dc 318 }
319 if(TMath::Abs(covariance[1])<1.e-13) {
320 sumsigma[1]+=0.;
321 }else{
bbeb9cae 322 sumsigma[1]+=(1./covariance[1]);
b4a900dc 323 }
324 if(TMath::Abs(covariance[3])<1.e-13) {
325 sumsigma[3]+=0.;
326 }else{
bbeb9cae 327 sumsigma[3]+=(1./covariance[3]);
b4a900dc 328 }
329 if(TMath::Abs(covariance[4])<1.e-13) {
330 sumsigma[4]+=0.;
331 }else{
bbeb9cae 332 sumsigma[4]+=(1./covariance[4]);
b4a900dc 333 }
334
335 nContributors[0]=nContributors[0]+vtx->GetNContributors();
bbeb9cae 336 }
337
338 for(Int_t i=0;i<3;i++){
b4a900dc 339 if(TMath::Abs(sumsigma[i])<1.e-13) continue;
bbeb9cae 340 pos[i]=sum[i]/sumsigma[i];
341 }
531cd64c 342 for(Int_t i2=0;i2<3;i2++){
b4a900dc 343 if(TMath::Abs(sumsigma[i2])<1.e-13) {sig[i2]=0.; continue;}
bbeb9cae 344 sig[i2]=1./sumsigma[i2];
345 }
b4a900dc 346 return kTRUE;
bbeb9cae 347}
348
a8f9624e 349
350Double_t AliMixedEvent::GetMagneticField() const
351{
352 // Return magnetic field of the first event in the list
353 if (fEventList.GetEntries() == 0) return -999.;
354
355 AliVEvent* evt = (AliVEvent*) (fEventList.At(0));
356 return evt->GetMagneticField();
357}