1 /**************************************************************************
2 * Copyright(c) 1998-2007, ALICE Experiment at CERN, All rights reserved. *
4 * Author: The ALICE Off-line Project. *
5 * Contributors are mentioned in the code where appropriate. *
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 **************************************************************************/
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 //-------------------------------------------------------------------------
27 #include "AliMixedEvent.h"
28 #include "AliExternalTrackParam.h"
30 #include "AliVVertex.h"
35 #include "AliVCaloCells.h"
38 ClassImp(AliMixedEvent)
41 AliMixedEvent::AliMixedEvent() :
46 fNumberOfCaloClusters(0),
47 fNumberOfPHOSCells(0),
48 fNumberOfEMCALCells(0),
50 fNCaloClustersCumul(0),
57 // Default constructor
60 AliMixedEvent::AliMixedEvent(const AliMixedEvent& Evnt) :
65 fNumberOfCaloClusters(0),
66 fNumberOfPHOSCells(0),
67 fNumberOfEMCALCells(0),
69 fNCaloClustersCumul(0),
75 { } // Copy constructor
77 AliMixedEvent& AliMixedEvent::operator=(const AliMixedEvent& vEvnt)
79 // Assignment operator
81 AliVEvent::operator=(vEvnt);
87 AliMixedEvent::~AliMixedEvent()
96 void AliMixedEvent::AddEvent(AliVEvent* evt)
98 // Add a new event to the list
99 fEventList.AddLast(evt);
103 void AliMixedEvent::Init()
105 // Initialize meta information
106 fNEvents = fEventList.GetEntries();
107 fNTracksCumul = new Int_t[fNEvents];
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];
116 TIter next(&fEventList);
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();
135 Short_t phosPos = 0, emcalPos = 0;
136 Int_t firstPHOSEvent = kTRUE;
137 Int_t firstEMCALEvent = kTRUE;
139 while((event = (AliVEvent*)next())) {
140 AliVCaloCells * phosCells = event->GetPHOSCells() ;
143 //Create the container
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
149 fPHOSCells->SetType(AliVCaloCells::kPHOSCell) ;
150 fPHOSCells->CreateContainer(fNumberOfPHOSCells) ;
151 firstPHOSEvent=kFALSE;
155 Int_t ncells = event->GetPHOSCells()->GetNumberOfCells() ;
156 for (Int_t icell = 0; icell < ncells; icell++) {
157 fPHOSCells->SetCell(phosPos++, phosCells->GetCellNumber(icell), phosCells->GetAmplitude(icell), phosCells->GetTime(icell)) ;
162 AliVCaloCells * emcalCells = event->GetEMCALCells() ;
165 //Create the container
168 if(!fEMCALCells)fEMCALCells = emcalCells->CopyCaloCells(kFALSE) ; // Just recover the first event type: ESD/AOD
169 else fEMCALCells->DeleteContainer(); // delete the previous container
170 //Now create a new container with the adequate size
171 fEMCALCells->SetType(AliVCaloCells::kEMCALCell) ;
172 fEMCALCells->CreateContainer(fNumberOfEMCALCells) ;
173 firstEMCALEvent=kFALSE;
176 Int_t ncells = emcalCells->GetNumberOfCells() ;
177 for (Int_t icell = 0; icell < ncells; icell++) {
178 fEMCALCells->SetCell(emcalPos++, emcalCells->GetCellNumber(icell), emcalCells->GetAmplitude(icell), emcalCells->GetTime(icell)) ;
185 AliVParticle* AliMixedEvent::GetTrack(Int_t i) const
188 Int_t iEv = TMath::BinarySearch(fNEvents, fNTracksCumul, i);
189 while((iEv < (fNEvents - 1)) && (fNTracksCumul[iEv] == fNTracksCumul[iEv+1])) {iEv++;}
191 Int_t irel = i - fNTracksCumul[iEv];
192 AliVEvent* evt = (AliVEvent*) (fEventList.At(iEv));
193 return (evt->GetTrack(irel));
196 AliVCluster* AliMixedEvent::GetCaloCluster(Int_t i) const
198 // Return calo cluster # i
199 Int_t iEv = TMath::BinarySearch(fNEvents, fNCaloClustersCumul, i);
200 while((iEv < (fNEvents - 1)) && (fNCaloClustersCumul[iEv] == fNCaloClustersCumul[iEv+1])) {iEv++;}
202 Int_t irel = i - fNCaloClustersCumul[iEv];
203 AliVEvent* evt = (AliVEvent*) (fEventList.At(iEv));
204 return (evt->GetCaloCluster(irel));
207 const AliVVertex* AliMixedEvent::GetEventVertex(Int_t i) const
209 // Return vertex of track # i
210 Int_t iEv = TMath::BinarySearch(fNEvents, fNTracksCumul, i);
211 while((iEv < (fNEvents - 1)) && (fNTracksCumul[iEv] == fNTracksCumul[iEv+1])) {iEv++;}
212 AliVEvent* evt = (AliVEvent*) (fEventList.At(iEv));
213 return (evt->GetPrimaryVertex());
216 const AliVVertex* AliMixedEvent::GetVertexOfEvent(Int_t i) const
218 // Return vertex of event # i
220 AliFatal(Form("%d events in buffer, event %d requested", fNEvents, i)) ;
221 AliVEvent* evt = (AliVEvent*) (fEventList.At(i));
222 return (evt->GetPrimaryVertex());
225 void AliMixedEvent::Reset()
231 fNumberOfCaloClusters = 0;
232 fNumberOfPHOSCells = 0;
233 fNumberOfEMCALCells = 0;
235 delete[] fNTracksCumul;
238 if (fNCaloClustersCumul) {
239 delete[] fNCaloClustersCumul;
240 fNCaloClustersCumul = 0;
242 if (fNPHOSCellsCumul) {
243 delete[] fNPHOSCellsCumul;
244 fNPHOSCellsCumul = 0;
246 if (fNEMCALCellsCumul) {
247 delete[] fNEMCALCellsCumul;
248 fNEMCALCellsCumul = 0;
252 fPHOSCells->DeleteContainer();
255 fEMCALCells->DeleteContainer();
260 Int_t AliMixedEvent::EventIndex(Int_t itrack) const
262 // Return the event index for track #itrack
263 return TMath::BinarySearch(fNEvents, fNTracksCumul, itrack);
266 Int_t AliMixedEvent::EventIndexForCaloCluster(Int_t icluster) const
268 // Return the event index for track #itrack
269 return TMath::BinarySearch(fNEvents, fNCaloClustersCumul, icluster);
272 Int_t AliMixedEvent::EventIndexForPHOSCell(Int_t icell) const
274 // Return the event index for track #itrack
275 return TMath::BinarySearch(fNEvents, fNPHOSCellsCumul, icell);
278 Int_t AliMixedEvent::EventIndexForEMCALCell(Int_t icell) const
280 // Return the event index for track #itrack
281 return TMath::BinarySearch(fNEvents, fNEMCALCellsCumul, icell);
284 Bool_t AliMixedEvent::ComputeVtx(const TObjArray *vertices, Double_t *pos,Double_t *sig,Int_t *nContributors) {
286 // Calculate the mean vertex psoitions from events in the buffer
288 Int_t nentries = vertices->GetEntriesFast();
289 Double_t sum[3]={0.,0.,0.};
290 Double_t sumsigma[6]={0.,0.,0.,0.,0.,0.};
293 for(Int_t ivtx = 0; ivtx < nentries; ivtx++){
294 AliVVertex *vtx=(AliVVertex*)vertices->UncheckedAt(ivtx);
295 Double_t covariance[6];
296 vtx->GetCovarianceMatrix(covariance);
299 if(TMath::Abs(covariance[0])<1.e-13) {
302 sum[0]+=vtxPos[0]*(1./covariance[0]);
303 sumsigma[0]+=(1./covariance[0]);
305 if(TMath::Abs(covariance[2])<1.e-13) {
308 sum[1]+=vtxPos[1]*(1./covariance[2]);
309 sumsigma[2]+=(1./covariance[2]);
311 if(TMath::Abs(covariance[5])<1.e-13) {
314 sum[2]+=vtxPos[2]*(1./covariance[5]);
315 sumsigma[5]+=(1./covariance[5]);
317 if(TMath::Abs(covariance[1])<1.e-13) {
320 sumsigma[1]+=(1./covariance[1]);
322 if(TMath::Abs(covariance[3])<1.e-13) {
325 sumsigma[3]+=(1./covariance[3]);
327 if(TMath::Abs(covariance[4])<1.e-13) {
330 sumsigma[4]+=(1./covariance[4]);
333 nContributors[0]=nContributors[0]+vtx->GetNContributors();
336 for(Int_t i=0;i<3;i++){
337 if(TMath::Abs(sumsigma[i])<1.e-13) continue;
338 pos[i]=sum[i]/sumsigma[i];
340 for(Int_t i2=0;i2<3;i2++){
341 if(TMath::Abs(sumsigma[i2])<1.e-13) {sig[i2]=0.; continue;}
342 sig[i2]=1./sumsigma[i2];
348 Double_t AliMixedEvent::GetMagneticField() const
350 // Return magnetic field of the first event in the list
351 if (fEventList.GetEntries() == 0) return -999.;
353 AliVEvent* evt = (AliVEvent*) (fEventList.At(0));
354 return evt->GetMagneticField();