First pass of changes required for visualization of event-embedding.
[u/mrichter/AliRoot.git] / EVE / EveDet / AliEveMUONData.cxx
1 // $Id$
2 // Main authors: Matevz Tadel & Alja Mrak-Tadel & Bogdan Vulpescu: 2006, 2007
3
4 /**************************************************************************
5  * Copyright(c) 1998-2008, ALICE Experiment at CERN, all rights reserved. *
6  * See http://aliceinfo.cern.ch/Offline/AliRoot/License.html for          *
7  * full copyright notice.                                                 *
8  **************************************************************************/
9
10 #include "AliEveMUONData.h"
11
12 #include <EveDet/AliEveMUONChamberData.h>
13 #include <EveBase/AliEveEventManager.h>
14
15 #include <AliRawReader.h>
16 #include <AliRawReaderFile.h>
17 #include <AliRawReaderDate.h>
18 #include <AliRawReaderRoot.h>
19
20 #include <AliMUONDigitMaker.h>
21 #include <AliMUONHit.h>
22 #include <AliMUONVCluster.h>
23 #include "AliMUONVClusterStore.h"
24 #include <AliMUONVDigit.h>
25 #include "AliMUONDigitStoreV1.h"
26 #include "AliMUONVDigitStore.h"
27 #include "AliMUONTrackParam.h"
28 #include "AliMUONTrack.h"
29 #include "AliMUONESDInterface.h"
30 #include "AliESDMuonTrack.h"
31 #include "AliESDEvent.h"
32 #include "TTree.h"
33 #include "TString.h"
34 #include "TClonesArray.h"
35 #include "TFile.h"
36
37
38 //______________________________________________________________________________
39 // AliEveMUONData
40 //
41
42 ClassImp(AliEveMUONData)
43
44 AliRawReader*            AliEveMUONData::fgRawReader        = 0;
45
46 //______________________________________________________________________________
47 AliEveMUONData::AliEveMUONData() :
48   fChambers(14),
49   fNTrackList(0)
50 {
51   //
52   // Constructor
53   //
54
55   for (Int_t i = 0; i < 256; i++) {
56     fTrackList[i] = -1;
57   }
58
59   CreateAllChambers();
60
61 }
62
63 //______________________________________________________________________________
64 AliEveMUONData::~AliEveMUONData()
65 {
66   //
67   // Destructor
68   //
69
70   DeleteAllChambers();
71
72 }
73
74 //______________________________________________________________________________
75 void AliEveMUONData::Reset()
76 {
77   //
78   // Reset data
79   //
80
81   //DropAllChambers();
82
83   fNTrackList = 0;
84   for (Int_t i = 0; i < 256; i++) {
85     fTrackList[i] = -1;
86   }
87
88 }
89
90 //______________________________________________________________________________
91 AliEveMUONData::AliEveMUONData(const AliEveMUONData &mdata) :
92   TObject(mdata),
93   TEveRefCnt(),
94   fChambers(14),
95   fNTrackList(0)
96 {
97   //
98   // Copy constructor
99   //
100
101 }
102
103 //______________________________________________________________________________
104 AliEveMUONData& AliEveMUONData::operator=(const AliEveMUONData &mdata)
105 {
106   //
107   // Assignment operator
108   //
109
110   if (this != &mdata) {
111
112   }
113
114   return *this;
115
116 }
117
118 //______________________________________________________________________________
119 void AliEveMUONData::CreateChamber(Int_t chamber)
120 {
121   //
122   // create data for the chamber with id=chamber (0 to 13)
123   //
124
125   if (fChambers[chamber] == 0)
126     fChambers[chamber] = new AliEveMUONChamberData(chamber);
127
128 }
129
130 //______________________________________________________________________________
131 void AliEveMUONData::CreateAllChambers()
132 {
133   //
134   // create all 14 chambers data
135   //
136
137   for (Int_t c = 0; c < 14; ++c)
138     CreateChamber(c);
139
140 }
141
142 //______________________________________________________________________________
143 void AliEveMUONData::DropAllChambers()
144 {
145   //
146   // release data from all chambers
147   //
148
149   for (Int_t c = 0; c < 14; ++c) {
150
151     if (fChambers[c] != 0)
152       fChambers[c]->DropData();
153
154   }
155
156 }
157
158 //______________________________________________________________________________
159 void AliEveMUONData::DeleteAllChambers()
160 {
161   //
162   // delete all chambers data
163   //
164
165   for (Int_t c = 0; c < 14; ++c) {
166
167     delete fChambers[c];
168     fChambers[c] = 0;
169
170   }
171
172 }
173
174 //______________________________________________________________________________
175 void AliEveMUONData::RegisterTrack(Int_t track)
176 {
177   //
178   // register (in a list) a track with hits in the chambers
179   //
180
181   if (fNTrackList == (256-1)) {
182     cout << "Maximum of registered tracks reached..." << endl;
183     return;
184   }
185
186   Bool_t inList = kFALSE;
187   for (Int_t i = 0; i < fNTrackList; i++) {
188     if (track == fTrackList[i]) {
189       inList = kTRUE;
190       break;
191     }
192   }
193   if (!inList) {
194     fTrackList[fNTrackList] = track;
195     fNTrackList++;
196   }
197
198 }
199
200 //______________________________________________________________________________
201 void AliEveMUONData::LoadRecPoints(TTree* tree)
202 {
203   //
204   // load reconstructed points from the TreeR
205   // load local trigger information
206   //
207
208   AliMUONVClusterStore *clusterStore = AliMUONVClusterStore::Create(*tree);
209   clusterStore->Clear();
210   clusterStore->Connect(*tree,kFALSE);
211
212   tree->GetEvent(0);
213
214   AliMUONVCluster *cluster;
215   Int_t detElemId;
216   Double_t clsX, clsY, clsZ, charge;
217
218   for (Int_t ch = 0; ch < 10; ++ch) {
219
220     if (fChambers[ch] == 0) continue;
221
222     TIter next(clusterStore->CreateChamberIterator(ch,ch));
223
224     while ( ( cluster = static_cast<AliMUONVCluster*>(next()) ) ) {
225
226       detElemId = cluster->GetDetElemId();
227
228       clsX   = cluster->GetX();
229       clsY   = cluster->GetY();
230       clsZ   = cluster->GetZ();
231       charge = cluster->GetCharge();
232
233       fChambers[ch]->RegisterCluster(detElemId,0,clsX,clsY,clsZ,charge);
234       fChambers[ch]->RegisterCluster(detElemId,1,clsX,clsY,clsZ,charge);
235
236     }
237
238   }
239
240   delete clusterStore;
241
242 }
243
244 //______________________________________________________________________________
245 void AliEveMUONData::LoadRecPointsFromESD(Char_t *fileName)
246 {
247   //
248   // load reconstructed points stored in AliESDs.root
249   // load local trigger information
250   //
251
252   TFile* esdFile = TFile::Open(fileName);
253   if (!esdFile || !esdFile->IsOpen()) {
254     cout << "opening ESD file " << fileName << "failed" << endl;
255     return;
256   }
257   TTree* esdTree = (TTree*) esdFile->Get("esdTree");
258   if (!esdTree) {
259     cout << "no ESD tree found" << endl;
260     esdFile->Close();
261     return;
262   }
263   AliESDEvent* esdEvent = new AliESDEvent();
264   esdEvent->ReadFromTree(esdTree);
265
266   AliMUONVCluster *cluster;
267   AliMUONTrackParam *trackParam;
268   AliESDMuonTrack *esdTrack;
269   AliMUONTrack muonTrack;
270   Int_t detElemId, chamber, nTrackParam;
271   Double_t clsX, clsY, clsZ, charge;
272   
273   if (esdTree->GetEvent(AliEveEventManager::GetMaster()->GetEventId()) <= 0) {
274     cout << "fails to read ESD object for event " << AliEveEventManager::GetMaster()->GetEventId() << endl;
275     return;
276   }
277     
278   Int_t nTracks = Int_t(esdEvent->GetNumberOfMuonTracks());
279   for (Int_t iTrack = 0; iTrack < nTracks; iTrack++) {
280     esdTrack = esdEvent->GetMuonTrack(iTrack);
281     if (!esdTrack->ClustersStored()) continue;
282     AliMUONESDInterface::ESDToMUON(*esdTrack,muonTrack);
283     nTrackParam = muonTrack.GetTrackParamAtCluster()->GetEntries();
284     for(Int_t iCluster = 0; iCluster < nTrackParam; iCluster++) {
285       trackParam = (AliMUONTrackParam *) muonTrack.GetTrackParamAtCluster()->At(iCluster);
286       cluster = trackParam->GetClusterPtr();
287       chamber   = cluster->GetChamberId();
288       detElemId = cluster->GetDetElemId();
289       charge  = cluster->GetCharge();
290       clsX = cluster->GetX();
291       clsY = cluster->GetY();
292       clsZ = cluster->GetZ();
293       
294       fChambers[chamber]->RegisterCluster(detElemId,0,clsX,clsY,clsZ,charge);
295       fChambers[chamber]->RegisterCluster(detElemId,1,clsX,clsY,clsZ,charge);
296       
297     }
298   }
299
300   delete esdEvent;
301   
302   esdFile->Close();
303
304 }
305
306 //______________________________________________________________________________
307 void AliEveMUONData::LoadHits(TTree* tree)
308 {
309   //
310   // load simulation hits from the TreeH
311   //
312
313   TClonesArray *hits = 0;
314   AliMUONHit  *mhit;
315   Int_t cha, detElemId, nhits, ntracks;
316   Float_t hitX, hitY, hitZ;
317
318   ntracks = tree->GetEntries();
319   tree->SetBranchAddress("MUONHits",&hits);
320
321   for (Int_t it = 0; it < ntracks; it++) {
322
323     tree->GetEvent(it);
324     nhits = hits->GetEntriesFast();
325
326     for (Int_t ih = 0; ih < nhits; ih++) {
327
328       mhit = (AliMUONHit*)hits->UncheckedAt(ih);
329       hitX = mhit->X();
330       hitY = mhit->Y();
331       hitZ = mhit->Z();
332       detElemId = mhit->DetElemId();
333       cha = mhit->Chamber();
334
335       RegisterTrack(mhit->GetTrack());
336
337       fChambers[cha-1]->RegisterHit(detElemId,hitX,hitY,hitZ);
338
339     }
340   }
341
342 }
343
344 //______________________________________________________________________________
345 void AliEveMUONData::LoadDigits(TTree* tree)
346 {
347   //
348   // load digits from the TreeD
349   //
350
351   AliMUONVDigitStore *digitStore = AliMUONVDigitStore::Create(*tree);
352   digitStore->Clear();
353   digitStore->Connect(*tree,0);
354
355   tree->GetEvent(0);
356
357   AliMUONVDigit* digit;
358   TIter next(digitStore->CreateIterator());
359
360   Int_t cathode, detElemId, ix, iy, charge, chamber, adc;
361
362   while ( ( digit = static_cast<AliMUONVDigit*>(next() ) ) )
363     {
364       cathode   = digit->Cathode();
365       ix        = digit->PadX();
366       iy        = digit->PadY();
367       detElemId = digit->DetElemId();
368       charge    = (Int_t)digit->Charge();
369       adc       = digit->ADC();
370       chamber   = detElemId/100 - 1;
371       if (chamber > 9) {
372         fChambers[chamber]->RegisterDigit(detElemId,cathode,ix,iy,charge);
373       } else {
374         fChambers[chamber]->RegisterDigit(detElemId,cathode,ix,iy,adc);
375       }
376     }
377
378   delete digitStore;
379
380 }
381
382 //______________________________________________________________________________
383 void AliEveMUONData::LoadRaw(TString fileName)
384 {
385   //
386   // load raw data from fileName; tracker and trigger data
387   //
388
389   if (fgRawReader == 0) {
390     // check extention to choose the rawdata file format
391     if (fileName.EndsWith("/")) {
392       fgRawReader = new AliRawReaderFile(fileName); // DDL files
393     } else if (fileName.EndsWith(".root")) {
394       fgRawReader = new AliRawReaderRoot(fileName); // ROOT file
395     } else if (!fileName.IsNull()) {
396       fgRawReader = new AliRawReaderDate(fileName); // DATE file
397     }
398   }
399
400   fgRawReader->RewindEvents();
401   fgRawReader->Reset();
402
403   Int_t iEvent = 0;
404   while (fgRawReader->NextEvent())
405   {
406     if (iEvent != AliEveEventManager::GetMaster()->GetEventId())
407     {
408       iEvent++;
409       continue;
410     }
411     break;
412   }
413
414   AliMUONDigitMaker digitMaker;
415
416   digitMaker.SetMakeTriggerDigits(kTRUE);
417
418   AliMUONDigitStoreV1 digitStore;
419
420   digitMaker.Raw2Digits(fgRawReader,&digitStore);
421
422   AliMUONVDigit* digit;
423   TIter next(digitStore.CreateIterator());
424
425   Int_t cathode, detElemId, ix, iy, charge, chamber, adc;
426
427   while ( ( digit = static_cast<AliMUONVDigit*>(next() ) ) )
428   {
429       cathode   = digit->Cathode();
430       ix        = digit->PadX();
431       iy        = digit->PadY();
432       detElemId = digit->DetElemId();
433       charge    = (Int_t)digit->Charge();
434       adc       = digit->ADC();
435       chamber   = detElemId/100 - 1;
436       if (chamber > 9) {
437         fChambers[chamber]->RegisterDigit(detElemId,cathode,ix,iy,charge);
438       } else {
439         fChambers[chamber]->RegisterDigit(detElemId,cathode,ix,iy,adc);
440       }
441   }
442 }
443
444 //______________________________________________________________________________
445 Int_t AliEveMUONData::GetTrack(Int_t index) const
446 {
447   //
448   // return track stack number for "index"-th track with hits in the chambers
449   //
450
451   if (index < 256) {
452     return fTrackList[index];
453   } else {
454     return -1;
455   }
456 }
457
458 //______________________________________________________________________________
459 AliEveMUONChamberData* AliEveMUONData::GetChamberData(Int_t chamber)
460 {
461   //
462   // return chamber data
463   //
464
465   if (chamber < 0 || chamber > 13) return 0;
466
467   //if (fChambers[chamber] == 0) CreateChamber(chamber);
468
469   return fChambers[chamber];
470 }