20dae051 |
1 | // $Id$ |
e9b9f7d2 |
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 | |
20dae051 |
10 | class AliEveMacroExecutor; |
3545fba9 |
11 | class MultiView; |
e9b9f7d2 |
12 | class TEveProjectionManager; |
13 | class TEveGeoShape; |
14 | class TEveUtil; |
15 | |
20dae051 |
16 | |
3545fba9 |
17 | Bool_t gShowTrd = kTRUE; |
18 | Bool_t gShowMuonRPhi = kFALSE; |
19 | Bool_t gShowMuonRhoZ = kTRUE; |
20dae051 |
20 | |
21 | Bool_t gCenterProjectionsAtPrimaryVertex = kFALSE; |
e9b9f7d2 |
22 | |
c63f2997 |
23 | |
24 | void visscan_init(const TString& cdburi = "", |
25 | const TString& path = ".", Bool_t show_extra_geo = kFALSE) |
e9b9f7d2 |
26 | { |
c63f2997 |
27 | if (cdburi.IsNull() && ! AliCDBManager::Instance()->IsDefaultStorageSet()) |
28 | { |
29 | gEnv->SetValue("Root.Stacktrace", "no"); |
30 | Fatal("visscan_init.C", "OCDB path MUST be specified as the first argument."); |
31 | } |
32 | |
3545fba9 |
33 | if (gROOT->LoadMacro("MultiView.C+") != 0) |
34 | { |
35 | gEnv->SetValue("Root.Stacktrace", "no"); |
36 | Fatal("visscan_init.C", "Failed loading MultiView.C in compiled mode."); |
37 | } |
38 | |
115b6655 |
39 | if (!show_extra_geo) |
40 | { |
3545fba9 |
41 | gShowTrd = gShowMuonRPhi = gShowMuonRhoZ = kFALSE; |
115b6655 |
42 | } |
43 | |
f8d5a389 |
44 | AliEveEventManager::AddAODfriend("AliAOD.VertexingHF.root"); |
45 | |
e9b9f7d2 |
46 | TEveUtil::LoadMacro("alieve_init.C"); |
c63f2997 |
47 | alieve_init(cdburi, path, -1); |
e9b9f7d2 |
48 | |
20dae051 |
49 | // TEveLine::SetDefaultSmooth(1); |
e9b9f7d2 |
50 | |
20dae051 |
51 | TEveUtil::AssertMacro("VizDB_scan.C"); |
52 | |
53 | AliEveMacroExecutor *exec = AliEveEventManager::GetMaster()->GetExecutor(); |
54 | TEveBrowser *browser = gEve->GetBrowser(); |
3545fba9 |
55 | browser->ShowCloseTab(kFALSE); |
56 | |
20dae051 |
57 | |
58 | //============================================================================== |
59 | // Geometry, scenes, projections and viewers |
60 | //============================================================================== |
61 | |
3545fba9 |
62 | gMultiView = new MultiView; |
20dae051 |
63 | |
f6afd0e1 |
64 | TEveUtil::LoadMacro("geom_gentle.C"); |
3545fba9 |
65 | gMultiView->InitGeomGentle(geom_gentle(), |
66 | geom_gentle_rphi(), |
67 | geom_gentle_rhoz()); |
68 | |
69 | if (gShowTrd) { |
f6afd0e1 |
70 | TEveUtil::LoadMacro("geom_gentle_trd.C"); |
3545fba9 |
71 | gMultiView->InitGeomGentleTrd(geom_gentle_trd()); |
f6afd0e1 |
72 | } |
f6afd0e1 |
73 | |
3545fba9 |
74 | if (gShowMuonRPhi || gShowMuonRhoZ) { |
75 | TEveUtil::LoadMacro("geom_gentle_muon.C"); |
76 | gMultiView->InitGeomGentleMuon(geom_gentle_muon(kFALSE), gShowMuonRPhi, gShowMuonRhoZ); |
68ca2fe7 |
77 | } |
20dae051 |
78 | |
79 | |
80 | //============================================================================== |
81 | // Registration of per-event macros |
82 | //============================================================================== |
83 | |
84 | exec->AddMacro(new AliEveMacro(AliEveMacro::kRunLoader, "SIM Track", "kine_tracks.C", "kine_tracks", "", kFALSE)); |
85 | |
86 | exec->AddMacro(new AliEveMacro(AliEveMacro::kRunLoader, "SIM Hit ITS", "its_hits.C", "its_hits", "", kFALSE)); |
87 | exec->AddMacro(new AliEveMacro(AliEveMacro::kRunLoader, "SIM Hit TPC", "tpc_hits.C", "tpc_hits", "", kFALSE)); |
88 | exec->AddMacro(new AliEveMacro(AliEveMacro::kRunLoader, "SIM Hit T0", "t0_hits.C", "t0_hits", "", kFALSE)); |
89 | exec->AddMacro(new AliEveMacro(AliEveMacro::kRunLoader, "SIM Hit FMD", "fmd_hits.C", "fmd_hits", "", kFALSE)); |
90 | |
1eda7a67 |
91 | exec->AddMacro(new AliEveMacro(AliEveMacro::kRunLoader, "DIG ITS", "its_digits.C", "its_digits", "", kFALSE)); |
20dae051 |
92 | exec->AddMacro(new AliEveMacro(AliEveMacro::kRunLoader, "DIG FMD", "fmd_digits.C", "fmd_digits", "", kFALSE)); |
93 | |
1eda7a67 |
94 | exec->AddMacro(new AliEveMacro(AliEveMacro::kRawReader, "RAW ITS", "its_raw.C", "its_raw", "", kFALSE)); |
20dae051 |
95 | exec->AddMacro(new AliEveMacro(AliEveMacro::kRawReader, "RAW TPC", "tpc_raw.C", "tpc_raw", "", kFALSE)); |
96 | exec->AddMacro(new AliEveMacro(AliEveMacro::kRawReader, "RAW T0", "t0_raw.C", "t0_raw", "", kFALSE)); |
97 | exec->AddMacro(new AliEveMacro(AliEveMacro::kRawReader, "RAW FMD", "fmd_raw.C", "fmd_raw", "", kFALSE)); |
98 | exec->AddMacro(new AliEveMacro(AliEveMacro::kRawReader, "RAW VZERO", "vzero_raw.C", "vzero_raw", "", kFALSE)); |
99 | exec->AddMacro(new AliEveMacro(AliEveMacro::kRawReader, "RAW ACORDE", "acorde_raw.C", "acorde_raw", "", kFALSE)); |
100 | |
101 | exec->AddMacro(new AliEveMacro(AliEveMacro::kESD, "REC PVTX", "primary_vertex.C", "primary_vertex", "", kTRUE)); |
102 | exec->AddMacro(new AliEveMacro(AliEveMacro::kESD, "REC PVTX Ellipse", "primary_vertex.C", "primary_vertex_ellipse", "", kTRUE)); |
103 | exec->AddMacro(new AliEveMacro(AliEveMacro::kESD, "REC PVTX Box", "primary_vertex.C", "primary_vertex_box", "kFALSE, 3, 3, 3", kFALSE)); |
104 | exec->AddMacro(new AliEveMacro(AliEveMacro::kESD, "REC PVTX", "primary_vertex.C", "primary_vertex_spd", "", kTRUE)); |
105 | exec->AddMacro(new AliEveMacro(AliEveMacro::kESD, "REC PVTX Ellipse", "primary_vertex.C", "primary_vertex_ellipse_spd", "", kTRUE)); |
106 | exec->AddMacro(new AliEveMacro(AliEveMacro::kESD, "REC PVTX Box", "primary_vertex.C", "primary_vertex_box_spd", "kFALSE, 3, 3, 3", kFALSE)); |
107 | exec->AddMacro(new AliEveMacro(AliEveMacro::kESD, "REC PVTX", "primary_vertex.C", "primary_vertex_tpc", "", kFALSE)); |
108 | exec->AddMacro(new AliEveMacro(AliEveMacro::kESD, "REC PVTX Ellipse", "primary_vertex.C", "primary_vertex_ellipse_tpc", "", kFALSE)); |
109 | exec->AddMacro(new AliEveMacro(AliEveMacro::kESD, "REC PVTX Box", "primary_vertex.C", "primary_vertex_box_tpc", "kFALSE, 3, 3, 3", kFALSE)); |
110 | |
111 | exec->AddMacro(new AliEveMacro(AliEveMacro::kESD, "REC V0", "esd_V0_points.C", "esd_V0_points_onfly")); |
112 | exec->AddMacro(new AliEveMacro(AliEveMacro::kESD, "REC V0", "esd_V0_points.C", "esd_V0_points_offline")); |
113 | exec->AddMacro(new AliEveMacro(AliEveMacro::kESD, "REC V0", "esd_V0.C", "esd_V0")); |
114 | exec->AddMacro(new AliEveMacro(AliEveMacro::kESD, "REC CSCD", "esd_cascade_points.C", "esd_cascade_points")); |
115 | exec->AddMacro(new AliEveMacro(AliEveMacro::kESD, "REC CSCD", "esd_cascade.C", "esd_cascade")); |
4711ad43 |
116 | exec->AddMacro(new AliEveMacro(AliEveMacro::kESD, "REC KINK", "esd_kink_points.C", "esd_kink_points")); |
117 | exec->AddMacro(new AliEveMacro(AliEveMacro::kESD, "REC KINK", "esd_kink.C", "esd_kink")); |
20dae051 |
118 | |
119 | exec->AddMacro(new AliEveMacro(AliEveMacro::kESD, "REC Track", "esd_tracks.C", "esd_tracks", "", kFALSE)); |
120 | exec->AddMacro(new AliEveMacro(AliEveMacro::kESD, "REC Track", "esd_tracks.C", "esd_tracks_MI", "", kFALSE)); |
121 | exec->AddMacro(new AliEveMacro(AliEveMacro::kESD, "REC Track", "esd_tracks.C", "esd_tracks_by_category", "", kTRUE)); |
122 | |
3a20f984 |
123 | exec->AddMacro(new AliEveMacro(AliEveMacro::kESD, "REC Tracklet", "esd_spd_tracklets.C", "esd_spd_tracklets", "", kTRUE)); |
20dae051 |
124 | |
125 | exec->AddMacro(new AliEveMacro(AliEveMacro::kESD, "REC ZDC", "esd_zdc.C", "esd_zdc", "", kFALSE)); |
126 | |
127 | exec->AddMacro(new AliEveMacro(AliEveMacro::kRunLoader, "REC Clus", "clusters.C+", "clusters", "", kFALSE)); |
128 | exec->AddMacro(new AliEveMacro(AliEveMacro::kRunLoader, "REC Clus ITS", "its_clusters.C+", "its_clusters")); |
129 | exec->AddMacro(new AliEveMacro(AliEveMacro::kRunLoader, "REC Clus TPC", "tpc_clusters.C+", "tpc_clusters")); |
130 | exec->AddMacro(new AliEveMacro(AliEveMacro::kRunLoader, "REC Clus TRD", "trd_clusters.C+", "trd_clusters")); |
131 | exec->AddMacro(new AliEveMacro(AliEveMacro::kRunLoader, "REC Clus TOF", "tof_clusters.C+", "tof_clusters")); |
132 | |
08b0f222 |
133 | exec->AddMacro(new AliEveMacro(AliEveMacro::kRunLoader, "REC Clus TPC", "vplot_tpc.C+", "vplot_tpc", "", kFALSE)); |
134 | |
f8d5a389 |
135 | exec->AddMacro(new AliEveMacro(AliEveMacro::kAOD, "ANA HF", "aod_HF.C", "aod_HF", "", kFALSE)); |
08b0f222 |
136 | exec->AddMacro(new AliEveMacro(AliEveMacro::kAOD, "ANA Jets", "jetplane.C", "jetplane", "", kFALSE)); |
20dae051 |
137 | |
3545fba9 |
138 | |
20dae051 |
139 | //============================================================================== |
140 | // Additional GUI components |
141 | //============================================================================== |
142 | |
143 | slot = TEveWindow::CreateWindowInTab(browser->GetTabRight()); |
144 | slot->StartEmbedding(); |
145 | AliEveMacroExecutorWindow* exewin = new AliEveMacroExecutorWindow(exec); |
146 | slot->StopEmbedding("DataSelection"); |
147 | exewin->PopulateMacros(); |
148 | |
149 | slot = TEveWindow::CreateWindowInTab(browser->GetTabRight()); |
150 | slot->StartEmbedding(); |
151 | new AliQAHistViewer(gClient->GetRoot(), 600, 400, kTRUE); |
152 | slot->StopEmbedding("QA histograms"); |
153 | |
008ac94c |
154 | //event selection tab |
155 | slot = TEveWindow::CreateWindowInTab(browser->GetTabRight()); |
156 | slot->StartEmbedding(); |
157 | new AliEveEventSelectorWindow(gClient->GetRoot(), 600, 400, AliEveEventManager::GetMaster()->GetEventSelector()); |
158 | slot->StopEmbedding("Selections"); |
159 | |
20dae051 |
160 | browser->GetTabRight()->SetTab(1); |
161 | |
162 | browser->StartEmbedding(TRootBrowser::kBottom); |
163 | new AliEveEventManagerWindow(AliEveEventManager::GetMaster()); |
164 | browser->StopEmbedding("EventCtrl"); |
165 | |
166 | slot = TEveWindow::CreateWindowInTab(browser->GetTabRight()); |
167 | TEveWindowTab *store_tab = slot->MakeTab(); |
168 | store_tab->SetElementNameTitle("WindowStore", |
169 | "Undocked windows whose previous container is not known\n" |
170 | "are placed here when the main-frame is closed."); |
171 | gEve->GetWindowManager()->SetDefaultContainer(store_tab); |
68ca2fe7 |
172 | |
3545fba9 |
173 | |
20dae051 |
174 | //============================================================================== |
175 | // AliEve objects - global tools |
176 | //============================================================================== |
177 | |
178 | AliEveTrackFitter* fitter = new AliEveTrackFitter(); |
179 | gEve->AddToListTree(fitter, 1); |
180 | gEve->AddElement(fitter, gEve->GetEventScene()); |
181 | |
182 | AliEveTrackCounter* g_trkcnt = new AliEveTrackCounter("Primary Counter"); |
183 | gEve->AddToListTree(g_trkcnt, kFALSE); |
184 | |
185 | |
186 | //============================================================================== |
187 | // Final stuff |
188 | //============================================================================== |
189 | |
190 | // A refresh to show proper window. |
191 | gEve->Redraw3D(kTRUE); |
192 | gSystem->ProcessEvents(); |
193 | |
194 | // Register command to call on each event. |
4d62585e |
195 | AliEveEventManager::GetMaster()->AddNewEventCommand("on_new_event();"); |
196 | AliEveEventManager::GetMaster()->GotoEvent(0); |
e9b9f7d2 |
197 | |
ca49b003 |
198 | gEve->EditElement(g_trkcnt); |
199 | |
e9b9f7d2 |
200 | gEve->Redraw3D(kTRUE); |
201 | } |
202 | |
203 | /******************************************************************************/ |
204 | |
205 | void on_new_event() |
206 | { |
f76c9e9b |
207 | AliEveTrackCounter* g_trkcnt = AliEveTrackCounter::fgInstance; |
e9b9f7d2 |
208 | g_trkcnt->Reset(); |
4d62585e |
209 | g_trkcnt->SetEventId(AliEveEventManager::GetMaster()->GetEventId()); |
29207369 |
210 | |
20dae051 |
211 | if (g_esd_tracks_by_category_container != 0) |
df56539f |
212 | { |
20dae051 |
213 | TEveElementList* cont = g_esd_tracks_by_category_container; |
214 | |
215 | // Here we expect several TEveTrackList containers. |
216 | // First two have reasonable primaries (sigma-to-prim-vertex < 5). |
217 | // Others are almost certainly secondaries. |
218 | Int_t count = 1; |
219 | TEveElement::List_i i = cont->BeginChildren(); |
220 | while (i != cont->EndChildren()) |
29207369 |
221 | { |
20dae051 |
222 | TEveTrackList* l = dynamic_cast<TEveTrackList*>(*i); |
223 | if (l != 0) |
224 | { |
225 | g_trkcnt->RegisterTracks(l, (count <= 2)); |
226 | ++count; |
227 | } |
228 | ++i; |
e9b9f7d2 |
229 | } |
20dae051 |
230 | |
231 | // Set it to zero, so that we do not reuse an old one. |
232 | g_esd_tracks_by_category_container = 0; |
233 | } |
234 | else |
235 | { |
236 | Warning("on_new_event", "g_esd_tracks_by_category_container not initialized."); |
e9b9f7d2 |
237 | } |
238 | |
3a20f984 |
239 | TEveTrackList *tracklets = dynamic_cast<TEveTrackList*>(gEve->GetCurrentEvent()->FindChild("SPD Tracklets")); |
240 | if (tracklets) |
241 | { |
242 | g_trkcnt->RegisterTracklets(tracklets, kTRUE); |
243 | } |
244 | else |
245 | { |
246 | Warning("on_new_event", "'SPD Tracklets' not found."); |
247 | } |
248 | |
20dae051 |
249 | Double_t x[3] = { 0, 0, 0 }; |
250 | |
251 | if (AliEveEventManager::HasESD()) |
a27140cb |
252 | { |
20dae051 |
253 | AliESDEvent* esd = AliEveEventManager::AssertESD(); |
254 | esd->GetPrimaryVertex()->GetXYZ(x); |
255 | |
a27140cb |
256 | TTimeStamp ts(esd->GetTimeStamp()); |
257 | TString win_title("Eve Main Window -- Timestamp: "); |
258 | win_title += ts.AsString("s"); |
5858bcbe |
259 | win_title += "; Event # in ESD file: "; |
a27140cb |
260 | win_title += esd->GetEventNumberInFile(); |
261 | gEve->GetBrowser()->SetWindowName(win_title); |
262 | } |
e9b9f7d2 |
263 | |
264 | TEveElement* top = gEve->GetCurrentEvent(); |
265 | |
3545fba9 |
266 | gMultiView->DestroyEventRPhi(); |
267 | if (gCenterProjectionsAtPrimaryVertex) |
268 | gMultiView->SetCenterRPhi(x[0], x[1], x[2]); |
269 | gMultiView->ImportEventRPhi(top); |
270 | |
271 | gMultiView->DestroyEventRhoZ(); |
272 | if (gCenterProjectionsAtPrimaryVertex) |
273 | gMultiView->SetCenterRhoZ(x[0], x[1], x[2]); |
274 | gMultiView->ImportEventRhoZ(top); |
e9b9f7d2 |
275 | } |