]>
Commit | Line | Data |
---|---|---|
58e15fc1 | 1 | // $Id: MUON_display.C 24485 2008-03-13 15:27:38Z mtadel $ |
2 | // Main authors: Matevz Tadel & Alja Mrak-Tadel: 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 | class AliEveMUONData; | |
11 | class AliEveEventManager; | |
12 | ||
13 | AliEveMUONData *g_muon_data = 0; | |
14 | AliEveEventManager *g_muon_last_event = 0; | |
15 | ||
16 | Int_t g_currentEvent = -1; | |
17 | Bool_t g_fromRaw = kFALSE; | |
18 | ||
19 | AliMagFMaps *g_field = 0; | |
20 | ||
af2d8a68 | 21 | void MUON_displayData(Bool_t fromRaw = kFALSE, Bool_t showTracks = kTRUE, Bool_t clustersFromESD = kTRUE) |
58e15fc1 | 22 | { |
23 | // | |
24 | // display from real data, eventually with recreated digits | |
25 | // tracks: ESD | |
26 | // | |
27 | ||
28 | if (!AliMpSegmentation::Instance()) AliMpCDB::LoadMpSegmentation(); | |
29 | if (!AliMpDDLStore::Instance()) AliMpCDB::LoadDDLStore(); | |
30 | ||
31 | if (g_field == 0) { | |
32 | printf("Loading field map...\n"); | |
33 | g_field = new AliMagFMaps("Maps","Maps", 1, 1., 10., AliMagFMaps::k5kG); | |
34 | AliTracker::SetFieldMap(g_field, kFALSE); | |
35 | AliMUONTrackExtrap::SetField(AliTracker::GetFieldMap()); | |
36 | } | |
37 | ||
38 | TTree* dt = 0; | |
39 | TTree* ct = 0; | |
40 | TTree* ht = 0; | |
41 | ||
42 | if (gAliEveEvent == 0) { | |
43 | printf("No alieve event: use alieve_init(...) \n"); | |
44 | return; | |
45 | } | |
46 | ||
47 | if (g_currentEvent == gAliEveEvent->GetEventId()) { | |
48 | if (g_fromRaw == fromRaw) { | |
49 | printf("Same event... \n"); | |
50 | return; | |
51 | } else { | |
52 | if (g_fromRaw) { | |
53 | printf("Same event with digits.\n"); | |
54 | gAliEveEvent->GotoEvent(g_currentEvent); | |
55 | } else { | |
56 | printf("Same event with raw.\n"); | |
57 | gAliEveEvent->GotoEvent(g_currentEvent); | |
58 | } | |
59 | } | |
60 | } | |
61 | ||
62 | g_fromRaw = fromRaw; | |
63 | ||
64 | TString dataPath = TString(gAliEveEvent->GetTitle()); | |
65 | dataPath.Append("/rawmuon.root"); | |
66 | ||
67 | AliRunLoader* rl = AliEveEventManager::AssertRunLoader(); | |
68 | g_muon_data = new AliEveMUONData; | |
69 | ||
70 | if (!fromRaw) { | |
71 | rl->LoadDigits("MUON"); | |
72 | dt = rl->GetTreeD("MUON", false); | |
73 | if (dt == 0) { | |
74 | cout << "No digits produced!" << endl; | |
75 | } else { | |
76 | cout << "With aliroot digits!" << endl; | |
77 | g_muon_data->LoadDigits(dt); | |
78 | } | |
79 | } else { | |
80 | if (gSystem->AccessPathName(dataPath.Data(),kFileExists)) { | |
81 | cout << "No raw data produced!" << endl; | |
82 | } else { | |
83 | cout << "With raw digits!" << endl; | |
84 | g_muon_data->LoadRaw(dataPath.Data()); | |
85 | } | |
86 | } | |
87 | ||
58e15fc1 | 88 | TString esdDataPath = TString(gAliEveEvent->GetTitle()); |
89 | esdDataPath.Append("/AliESDs.root"); | |
af2d8a68 | 90 | if (clustersFromESD) { |
91 | g_muon_data->LoadRecPointsFromESD(esdDataPath.Data()); | |
92 | } else { | |
93 | rl->LoadRecPoints("MUON"); | |
94 | ct = rl->GetTreeR("MUON", false); | |
95 | g_muon_data->LoadRecPoints(ct); | |
96 | } | |
97 | ||
58e15fc1 | 98 | g_muon_last_event = gAliEveEvent; |
99 | ||
100 | g_currentEvent = g_muon_last_event->GetEventId(); | |
101 | ||
102 | gStyle->SetPalette(1, 0); | |
103 | ||
104 | gEve->DisableRedraw(); | |
105 | ||
106 | TEveElementList* l = new TEveElementList("MUONChambers"); | |
107 | l->SetTitle("MUON chambers"); | |
108 | l->SetMainColor(Color_t(2)); | |
109 | gEve->AddElement(l); | |
110 | ||
111 | for (Int_t ic = 0; ic < 14; ic++) | |
112 | { | |
113 | AliEveMUONChamber* mucha = new AliEveMUONChamber(ic); | |
114 | ||
115 | mucha->SetFrameColor(2); | |
116 | mucha->SetChamberID(ic); | |
117 | ||
118 | mucha->SetDataSource(g_muon_data); | |
119 | ||
120 | gEve->AddElement(mucha, l); | |
121 | } | |
122 | ||
123 | if (showTracks) { | |
124 | MUON_ESD_tracks(); | |
125 | } | |
126 | ||
127 | gEve->Redraw3D(kTRUE); | |
128 | gEve->EnableRedraw(); | |
129 | } | |
130 | ||
131 | //______________________________________________________________________________ | |
132 | void MUON_ESD_tracks() | |
133 | { | |
134 | AliESDEvent* esd = AliEveEventManager::AssertESD(); | |
135 | ||
136 | TEveTrackList* lt = new TEveTrackList("ESD-Tracks"); | |
137 | lt->SetMainColor(Color_t(6)); | |
138 | //lt->SetMUON(); | |
139 | ||
140 | gEve->AddElement(lt); | |
141 | ||
142 | AliESDMuonTrack *mt; | |
143 | TEveRecTrack rt; | |
144 | Int_t nMuonTracks = esd->GetNumberOfMuonTracks(); | |
145 | Int_t nTrack = 0; | |
146 | for (Int_t n = 0; n < nMuonTracks; n++) | |
147 | { | |
148 | mt = esd->GetMuonTrack(n); | |
149 | ||
150 | if (mt->GetNHit() == 0) continue; | |
151 | nTrack++; | |
152 | ||
153 | rt.fLabel = n; | |
154 | ||
155 | AliEveMUONTrack* track = new AliEveMUONTrack(&rt, lt->GetPropagator()); | |
156 | ||
157 | track->MakeESDTrack(mt); | |
158 | ||
159 | gEve->AddElement(track, lt); | |
160 | } | |
161 | ||
162 | } | |
163 |