2 // Main authors: Matevz Tadel & Alja Mrak-Tadel: 2006, 2007
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 **************************************************************************/
10 /// \ingroup evemacros
11 /// \file MUON_displaySimu.C
13 /// \author B. Vulpescu, LPC
16 class AliEveEventManager;
18 AliEveMUONData *g_muon_data = 0;
20 Int_t g_currentEvent = -1;
21 Bool_t g_fromRaw = kFALSE;
24 void MUON_displaySimu(Bool_t fromRaw = kFALSE, Bool_t showTracks = kTRUE, Bool_t clustersFromESD = kTRUE)
27 // display from simulated digits (or produced raw data)
28 // tracks: ESD, Refs, MC
31 if (!AliMpSegmentation::Instance()) AliMpCDB::LoadMpSegmentation();
32 if (!AliMpDDLStore::Instance()) AliMpCDB::LoadDDLStore();
34 // set the magnetic field for track extrapolations
35 AliEveEventManager::AssertMagField();
36 AliMUONTrackExtrap::SetField();
42 if (AliEveEventManager::GetMaster() == 0) {
43 printf("No alieve event: use alieve_init(...) \n");
47 if (g_currentEvent == AliEveEventManager::GetMaster()->GetEventId()) {
48 if (g_fromRaw == fromRaw) {
49 printf("Same event... \n");
53 printf("Same event with digits.\n");
54 AliEveEventManager::GetMaster()->GotoEvent(g_currentEvent);
56 printf("Same event with raw.\n");
57 AliEveEventManager::GetMaster()->GotoEvent(g_currentEvent);
64 TString dataPath = TString(AliEveEventManager::GetMaster()->GetTitle());
65 dataPath.Append("/raw.root");
67 AliRunLoader* rl = AliEveEventManager::AssertRunLoader();
68 g_muon_data = new AliEveMUONData;
71 rl->LoadDigits("MUON");
72 dt = rl->GetTreeD("MUON", false);
74 cout << "No digits produced!" << endl;
76 cout << "With aliroot digits!" << endl;
77 g_muon_data->LoadDigits(dt);
80 if (gSystem->AccessPathName(dataPath.Data(),kFileExists)) {
81 cout << "No raw data produced!" << endl;
83 cout << "With raw digits!" << endl;
84 g_muon_data->LoadRaw(dataPath.Data());
88 TString esdDataPath = TString(AliEveEventManager::GetMaster()->GetTitle());
89 esdDataPath.Append("/AliESDs.root");
90 if (clustersFromESD) {
91 g_muon_data->LoadRecPointsFromESD(esdDataPath.Data());
93 rl->LoadRecPoints("MUON");
94 ct = rl->GetTreeR("MUON", false);
95 g_muon_data->LoadRecPoints(ct);
99 ht = rl->GetTreeH("MUON", false);
100 g_muon_data->LoadHits(ht);
102 g_currentEvent = AliEveEventManager::GetMaster()->GetEventId();
104 gStyle->SetPalette(1, 0);
106 gEve->DisableRedraw();
108 TEveElementList* l = new TEveElementList("MUONChambers");
109 l->SetTitle("MUON chambers");
113 for (Int_t ic = 0; ic < 14; ic++)
115 AliEveMUONChamber* mucha = new AliEveMUONChamber(ic);
117 mucha->SetFrameColor(2);
118 mucha->SetChamberID(ic);
120 mucha->SetDataSource(g_muon_data);
122 gEve->AddElement(mucha, l);
131 gEve->Redraw3D(kTRUE);
132 gEve->EnableRedraw();
135 //______________________________________________________________________________
136 void MUON_ESD_tracks()
138 AliESDEvent* esd = AliEveEventManager::AssertESD();
140 TEveTrackList* lt = new TEveTrackList("ESD-Tracks");
144 gEve->AddElement(lt);
148 Int_t nMuonTracks = esd->GetNumberOfMuonTracks();
150 for (Int_t n = 0; n < nMuonTracks; n++)
152 mt = esd->GetMuonTrack(n);
154 if (mt->GetNHit() == 0) continue;
159 AliEveMUONTrack* track = new AliEveMUONTrack(&rt, lt->GetPropagator());
161 track->MakeESDTrack(mt);
163 gEve->AddElement(track, lt);
168 //______________________________________________________________________________
169 void MUON_Ref_tracks()
171 TString dataPathESD = TString(AliEveEventManager::GetMaster()->GetTitle());
172 dataPathESD.Append("/AliESDs.root");
173 TString dataPathSIM = TString(AliEveEventManager::GetMaster()->GetTitle());
174 dataPathSIM.Append("/");
176 AliMUONRecoCheck recoCheck(dataPathESD.Data(),dataPathSIM.Data());
177 AliMUONVTrackStore* trackRefStore = recoCheck.ReconstructibleTracks(AliEveEventManager::GetMaster()->GetEventId());
178 TIter next(trackRefStore->CreateIterator());
179 AliMUONTrack* trackRef;
181 TEveTrackList* lt = new TEveTrackList("Ref-Tracks");
184 gEve->AddElement(lt);
188 while ( ( trackRef = static_cast<AliMUONTrack*>(next()) ) )
192 AliEveMUONTrack* track = new AliEveMUONTrack(&rt, lt->GetPropagator());
194 track->MakeRefTrack(trackRef);
196 gEve->AddElement(track, lt);
201 //______________________________________________________________________________
202 void MUON_MC_tracks()
204 Double_t RADDEG = 180.0/TMath::Pi();
206 AliRunLoader* rl = AliEveEventManager::AssertRunLoader();
207 rl->LoadKinematics();
208 AliStack* stack = rl->Stack();
210 Int_t nPrimary = stack->GetNprimary();
211 Int_t nTracks = stack->GetNtrack();
213 TEveTrackList* lt = new TEveTrackList("MC-Tracks");
217 gEve->AddElement(lt);
223 Int_t nHitTracks = g_muon_data->GetNTrackList();
225 for (Int_t i = 0; i < nHitTracks; i++)
227 index = g_muon_data->GetTrack(i);
228 if (index >= nTracks) {
229 cout << "TEveHit track index larger than number in stack!" << endl;
233 part = stack->Particle(index);
234 if (part->P() < 0.001) continue; // skip momenta < 1.0 MeV/c
237 AliEveMUONTrack* track = new AliEveMUONTrack(&rt, lt->GetPropagator());
239 track->MakeMCTrack(part);
241 gEve->AddElement(track, lt);
244 rl->UnloadKinematics();