]>
Commit | Line | Data |
---|---|---|
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; |
11 | ||
e9b9f7d2 | 12 | class TEveProjectionManager; |
13 | class TEveGeoShape; | |
14 | class TEveUtil; | |
15 | ||
5e078eda | 16 | TEveGeoShape *gGeomGentle = 0; |
17 | TEveGeoShape *gGeomGentleRPhi = 0; | |
18 | TEveGeoShape *gGeomGentleRhoZ = 0; | |
19 | TEveGeoShape *gGeomGentleTRD = 0; | |
20dae051 | 20 | TEveGeoShape *gGeomGentleMUON = 0; |
21 | ||
22 | TEveScene *gRPhiGeomScene = 0; | |
23 | TEveScene *gRhoZGeomScene = 0; | |
24 | TEveScene *gRPhiEventScene = 0; | |
25 | TEveScene *gRhoZEventScene = 0; | |
26 | ||
27 | TEveProjectionManager *gRPhiMgr = 0; | |
28 | TEveProjectionManager *gRhoZMgr = 0; | |
29 | ||
30 | TEveViewer *g3DView = 0; | |
31 | TEveViewer *gRPhiView = 0; | |
32 | TEveViewer *gRhoZView = 0; | |
503bfbc8 | 33 | |
20dae051 | 34 | Bool_t gShowTRD = kFALSE; |
35 | Bool_t gShowMUON = kTRUE; | |
36 | Bool_t gShowMUONRPhi = kFALSE; | |
115b6655 | 37 | Bool_t gShowMUONRhoZ = kTRUE; |
20dae051 | 38 | |
39 | Bool_t gCenterProjectionsAtPrimaryVertex = kFALSE; | |
e9b9f7d2 | 40 | |
115b6655 | 41 | void visscan_init(Bool_t show_extra_geo=kFALSE) |
e9b9f7d2 | 42 | { |
115b6655 | 43 | if (!show_extra_geo) |
44 | { | |
45 | gShowTRD = gShowMUON = gShowMUONRPhi = gShowMUONRhoZ = kFALSE; | |
46 | } | |
47 | ||
f8d5a389 | 48 | AliEveEventManager::AddAODfriend("AliAOD.VertexingHF.root"); |
49 | ||
e9b9f7d2 | 50 | TEveUtil::LoadMacro("alieve_init.C"); |
20dae051 | 51 | alieve_init(".", -1); |
e9b9f7d2 | 52 | |
20dae051 | 53 | // TEveLine::SetDefaultSmooth(1); |
e9b9f7d2 | 54 | |
20dae051 | 55 | TEveUtil::AssertMacro("VizDB_scan.C"); |
56 | ||
57 | AliEveMacroExecutor *exec = AliEveEventManager::GetMaster()->GetExecutor(); | |
58 | TEveBrowser *browser = gEve->GetBrowser(); | |
59 | ||
60 | //============================================================================== | |
61 | // Geometry, scenes, projections and viewers | |
62 | //============================================================================== | |
63 | ||
64 | browser->ShowCloseTab(kFALSE); | |
e9b9f7d2 | 65 | |
68ca2fe7 | 66 | // Geometry |
20dae051 | 67 | |
f6afd0e1 | 68 | TEveUtil::LoadMacro("geom_gentle.C"); |
20dae051 | 69 | gGeomGentle = geom_gentle(); |
5e078eda | 70 | gGeomGentleRPhi = geom_gentle_rphi(); gGeomGentleRPhi->IncDenyDestroy(); |
71 | gGeomGentleRhoZ = geom_gentle_rhoz(); gGeomGentleRhoZ->IncDenyDestroy(); | |
f6afd0e1 | 72 | if (gShowTRD) { |
73 | TEveUtil::LoadMacro("geom_gentle_trd.C"); | |
74 | gGeomGentleTRD = geom_gentle_trd(); | |
75 | } | |
20dae051 | 76 | if (gShowMUON) { |
77 | TEveUtil::LoadMacro("geom_gentle_muon.C"); | |
ae8fef5f | 78 | gGeomGentleMUON = geom_gentle_muon(kFALSE); |
20dae051 | 79 | } |
f6afd0e1 | 80 | |
20dae051 | 81 | // Scenes |
68ca2fe7 | 82 | |
20dae051 | 83 | gRPhiGeomScene = gEve->SpawnNewScene("RPhi Geometry", |
84 | "Scene holding projected geometry for the RPhi view."); | |
85 | gRhoZGeomScene = gEve->SpawnNewScene("RhoZ Geometry", | |
86 | "Scene holding projected geometry for the RhoZ view."); | |
87 | gRPhiEventScene = gEve->SpawnNewScene("RPhi Event Data", | |
88 | "Scene holding projected geometry for the RPhi view."); | |
89 | gRhoZEventScene = gEve->SpawnNewScene("RhoZ Event Data", | |
90 | "Scene holding projected geometry for the RhoZ view."); | |
f6afd0e1 | 91 | |
20dae051 | 92 | // Projection managers |
f6afd0e1 | 93 | |
20dae051 | 94 | gRPhiMgr = new TEveProjectionManager(); |
95 | gRPhiMgr->SetProjection(TEveProjection::kPT_RPhi); | |
96 | gEve->AddToListTree(gRPhiMgr, kFALSE); | |
97 | { | |
68ca2fe7 | 98 | TEveProjectionAxes* a = new TEveProjectionAxes(gRPhiMgr); |
99 | a->SetMainColor(kWhite); | |
100 | a->SetTitle("R-Phi"); | |
101 | a->SetTitleSize(0.05); | |
115b6655 | 102 | a->SetTitleFont(102); |
68ca2fe7 | 103 | a->SetLabelSize(0.025); |
115b6655 | 104 | a->SetLabelFont(102); |
20dae051 | 105 | gRPhiGeomScene->AddElement(a); |
68ca2fe7 | 106 | } |
9851a663 | 107 | gRPhiMgr->SetCurrentDepth(-10); |
20dae051 | 108 | gRPhiMgr->ImportElements(gGeomGentleRPhi, gRPhiGeomScene); |
9851a663 | 109 | gRPhiMgr->SetCurrentDepth(0); |
20dae051 | 110 | if (gShowTRD) gRPhiMgr->ImportElements(gGeomGentleTRD, gRPhiGeomScene); |
111 | if (gShowMUONRPhi) gRPhiMgr->ImportElements(gGeomGentleMUON, gRPhiGeomScene); | |
112 | ||
113 | gRhoZMgr = new TEveProjectionManager(); | |
114 | gRhoZMgr->SetProjection(TEveProjection::kPT_RhoZ); | |
115 | gEve->AddToListTree(gRhoZMgr, kFALSE); | |
116 | { | |
68ca2fe7 | 117 | TEveProjectionAxes* a = new TEveProjectionAxes(gRhoZMgr); |
118 | a->SetMainColor(kWhite); | |
119 | a->SetTitle("Rho-Z"); | |
120 | a->SetTitleSize(0.05); | |
115b6655 | 121 | a->SetTitleFont(102); |
68ca2fe7 | 122 | a->SetLabelSize(0.025); |
115b6655 | 123 | a->SetLabelFont(102); |
20dae051 | 124 | gRhoZGeomScene->AddElement(a); |
68ca2fe7 | 125 | } |
9851a663 | 126 | gRhoZMgr->SetCurrentDepth(-10); |
20dae051 | 127 | gRhoZMgr->ImportElements(gGeomGentleRhoZ, gRhoZGeomScene); |
9851a663 | 128 | gRhoZMgr->SetCurrentDepth(0); |
20dae051 | 129 | if (gShowTRD) gRhoZMgr->ImportElements(gGeomGentleTRD, gRhoZGeomScene); |
130 | if (gShowMUONRhoZ) gRhoZMgr->ImportElements(gGeomGentleMUON, gRhoZGeomScene); | |
131 | ||
132 | // Viewers | |
133 | ||
134 | TEveWindowSlot *slot = 0; | |
135 | TEveWindowPack *pack = 0; | |
136 | ||
137 | slot = TEveWindow::CreateWindowInTab(browser->GetTabRight()); | |
138 | pack = slot->MakePack(); | |
139 | pack->SetElementName("Multi View"); | |
140 | pack->SetHorizontal(); | |
141 | pack->SetShowTitleBar(kFALSE); | |
142 | pack->NewSlot()->MakeCurrent(); | |
143 | g3DView = gEve->SpawnNewViewer("3D View", ""); | |
144 | g3DView->AddScene(gEve->GetGlobalScene()); | |
145 | g3DView->AddScene(gEve->GetEventScene()); | |
146 | ||
147 | pack = pack->NewSlot()->MakePack(); | |
148 | pack->SetShowTitleBar(kFALSE); | |
149 | pack->NewSlot()->MakeCurrent(); | |
150 | gRPhiView = gEve->SpawnNewViewer("RPhi View", ""); | |
151 | gRPhiView->GetGLViewer()->SetCurrentCamera(TGLViewer::kCameraOrthoXOY); | |
152 | gRPhiView->AddScene(gRPhiGeomScene); | |
153 | gRPhiView->AddScene(gRPhiEventScene); | |
154 | ||
155 | pack->NewSlot()->MakeCurrent(); | |
156 | gRhoZView = gEve->SpawnNewViewer("RhoZ View", ""); | |
157 | gRhoZView->GetGLViewer()->SetCurrentCamera(TGLViewer::kCameraOrthoXOY); | |
158 | gRhoZView->AddScene(gRhoZGeomScene); | |
159 | gRhoZView->AddScene(gRhoZEventScene); | |
160 | ||
161 | ||
162 | //============================================================================== | |
163 | // Registration of per-event macros | |
164 | //============================================================================== | |
165 | ||
166 | exec->AddMacro(new AliEveMacro(AliEveMacro::kRunLoader, "SIM Track", "kine_tracks.C", "kine_tracks", "", kFALSE)); | |
167 | ||
168 | exec->AddMacro(new AliEveMacro(AliEveMacro::kRunLoader, "SIM Hit ITS", "its_hits.C", "its_hits", "", kFALSE)); | |
169 | exec->AddMacro(new AliEveMacro(AliEveMacro::kRunLoader, "SIM Hit TPC", "tpc_hits.C", "tpc_hits", "", kFALSE)); | |
170 | exec->AddMacro(new AliEveMacro(AliEveMacro::kRunLoader, "SIM Hit T0", "t0_hits.C", "t0_hits", "", kFALSE)); | |
171 | exec->AddMacro(new AliEveMacro(AliEveMacro::kRunLoader, "SIM Hit FMD", "fmd_hits.C", "fmd_hits", "", kFALSE)); | |
172 | ||
173 | exec->AddMacro(new AliEveMacro(AliEveMacro::kRunLoader, "DIG FMD", "fmd_digits.C", "fmd_digits", "", kFALSE)); | |
174 | ||
175 | exec->AddMacro(new AliEveMacro(AliEveMacro::kRawReader, "RAW TPC", "tpc_raw.C", "tpc_raw", "", kFALSE)); | |
176 | exec->AddMacro(new AliEveMacro(AliEveMacro::kRawReader, "RAW T0", "t0_raw.C", "t0_raw", "", kFALSE)); | |
177 | exec->AddMacro(new AliEveMacro(AliEveMacro::kRawReader, "RAW FMD", "fmd_raw.C", "fmd_raw", "", kFALSE)); | |
178 | exec->AddMacro(new AliEveMacro(AliEveMacro::kRawReader, "RAW VZERO", "vzero_raw.C", "vzero_raw", "", kFALSE)); | |
179 | exec->AddMacro(new AliEveMacro(AliEveMacro::kRawReader, "RAW ACORDE", "acorde_raw.C", "acorde_raw", "", kFALSE)); | |
180 | ||
181 | exec->AddMacro(new AliEveMacro(AliEveMacro::kESD, "REC PVTX", "primary_vertex.C", "primary_vertex", "", kTRUE)); | |
182 | exec->AddMacro(new AliEveMacro(AliEveMacro::kESD, "REC PVTX Ellipse", "primary_vertex.C", "primary_vertex_ellipse", "", kTRUE)); | |
183 | exec->AddMacro(new AliEveMacro(AliEveMacro::kESD, "REC PVTX Box", "primary_vertex.C", "primary_vertex_box", "kFALSE, 3, 3, 3", kFALSE)); | |
184 | exec->AddMacro(new AliEveMacro(AliEveMacro::kESD, "REC PVTX", "primary_vertex.C", "primary_vertex_spd", "", kTRUE)); | |
185 | exec->AddMacro(new AliEveMacro(AliEveMacro::kESD, "REC PVTX Ellipse", "primary_vertex.C", "primary_vertex_ellipse_spd", "", kTRUE)); | |
186 | exec->AddMacro(new AliEveMacro(AliEveMacro::kESD, "REC PVTX Box", "primary_vertex.C", "primary_vertex_box_spd", "kFALSE, 3, 3, 3", kFALSE)); | |
187 | exec->AddMacro(new AliEveMacro(AliEveMacro::kESD, "REC PVTX", "primary_vertex.C", "primary_vertex_tpc", "", kFALSE)); | |
188 | exec->AddMacro(new AliEveMacro(AliEveMacro::kESD, "REC PVTX Ellipse", "primary_vertex.C", "primary_vertex_ellipse_tpc", "", kFALSE)); | |
189 | exec->AddMacro(new AliEveMacro(AliEveMacro::kESD, "REC PVTX Box", "primary_vertex.C", "primary_vertex_box_tpc", "kFALSE, 3, 3, 3", kFALSE)); | |
190 | ||
191 | exec->AddMacro(new AliEveMacro(AliEveMacro::kESD, "REC V0", "esd_V0_points.C", "esd_V0_points_onfly")); | |
192 | exec->AddMacro(new AliEveMacro(AliEveMacro::kESD, "REC V0", "esd_V0_points.C", "esd_V0_points_offline")); | |
193 | exec->AddMacro(new AliEveMacro(AliEveMacro::kESD, "REC V0", "esd_V0.C", "esd_V0")); | |
194 | exec->AddMacro(new AliEveMacro(AliEveMacro::kESD, "REC CSCD", "esd_cascade_points.C", "esd_cascade_points")); | |
195 | exec->AddMacro(new AliEveMacro(AliEveMacro::kESD, "REC CSCD", "esd_cascade.C", "esd_cascade")); | |
4711ad43 | 196 | exec->AddMacro(new AliEveMacro(AliEveMacro::kESD, "REC KINK", "esd_kink_points.C", "esd_kink_points")); |
197 | exec->AddMacro(new AliEveMacro(AliEveMacro::kESD, "REC KINK", "esd_kink.C", "esd_kink")); | |
20dae051 | 198 | |
199 | exec->AddMacro(new AliEveMacro(AliEveMacro::kESD, "REC Track", "esd_tracks.C", "esd_tracks", "", kFALSE)); | |
200 | exec->AddMacro(new AliEveMacro(AliEveMacro::kESD, "REC Track", "esd_tracks.C", "esd_tracks_MI", "", kFALSE)); | |
201 | exec->AddMacro(new AliEveMacro(AliEveMacro::kESD, "REC Track", "esd_tracks.C", "esd_tracks_by_category", "", kTRUE)); | |
202 | ||
203 | exec->AddMacro(new AliEveMacro(AliEveMacro::kESD, "REC Tracklet", "esd_spd_tracklets.C", "esd_spd_tracklets", "", kFALSE)); | |
204 | ||
205 | exec->AddMacro(new AliEveMacro(AliEveMacro::kESD, "REC ZDC", "esd_zdc.C", "esd_zdc", "", kFALSE)); | |
206 | ||
207 | exec->AddMacro(new AliEveMacro(AliEveMacro::kRunLoader, "REC Clus", "clusters.C+", "clusters", "", kFALSE)); | |
208 | exec->AddMacro(new AliEveMacro(AliEveMacro::kRunLoader, "REC Clus ITS", "its_clusters.C+", "its_clusters")); | |
209 | exec->AddMacro(new AliEveMacro(AliEveMacro::kRunLoader, "REC Clus TPC", "tpc_clusters.C+", "tpc_clusters")); | |
210 | exec->AddMacro(new AliEveMacro(AliEveMacro::kRunLoader, "REC Clus TRD", "trd_clusters.C+", "trd_clusters")); | |
211 | exec->AddMacro(new AliEveMacro(AliEveMacro::kRunLoader, "REC Clus TOF", "tof_clusters.C+", "tof_clusters")); | |
212 | ||
08b0f222 | 213 | exec->AddMacro(new AliEveMacro(AliEveMacro::kRunLoader, "REC Clus TPC", "vplot_tpc.C+", "vplot_tpc", "", kFALSE)); |
214 | ||
f8d5a389 | 215 | exec->AddMacro(new AliEveMacro(AliEveMacro::kAOD, "ANA HF", "aod_HF.C", "aod_HF", "", kFALSE)); |
08b0f222 | 216 | exec->AddMacro(new AliEveMacro(AliEveMacro::kAOD, "ANA Jets", "jetplane.C", "jetplane", "", kFALSE)); |
20dae051 | 217 | |
218 | //============================================================================== | |
219 | // Additional GUI components | |
220 | //============================================================================== | |
221 | ||
222 | slot = TEveWindow::CreateWindowInTab(browser->GetTabRight()); | |
223 | slot->StartEmbedding(); | |
224 | AliEveMacroExecutorWindow* exewin = new AliEveMacroExecutorWindow(exec); | |
225 | slot->StopEmbedding("DataSelection"); | |
226 | exewin->PopulateMacros(); | |
227 | ||
228 | slot = TEveWindow::CreateWindowInTab(browser->GetTabRight()); | |
229 | slot->StartEmbedding(); | |
230 | new AliQAHistViewer(gClient->GetRoot(), 600, 400, kTRUE); | |
231 | slot->StopEmbedding("QA histograms"); | |
232 | ||
008ac94c | 233 | //event selection tab |
234 | slot = TEveWindow::CreateWindowInTab(browser->GetTabRight()); | |
235 | slot->StartEmbedding(); | |
236 | new AliEveEventSelectorWindow(gClient->GetRoot(), 600, 400, AliEveEventManager::GetMaster()->GetEventSelector()); | |
237 | slot->StopEmbedding("Selections"); | |
238 | ||
20dae051 | 239 | browser->GetTabRight()->SetTab(1); |
240 | ||
241 | browser->StartEmbedding(TRootBrowser::kBottom); | |
242 | new AliEveEventManagerWindow(AliEveEventManager::GetMaster()); | |
243 | browser->StopEmbedding("EventCtrl"); | |
244 | ||
245 | slot = TEveWindow::CreateWindowInTab(browser->GetTabRight()); | |
246 | TEveWindowTab *store_tab = slot->MakeTab(); | |
247 | store_tab->SetElementNameTitle("WindowStore", | |
248 | "Undocked windows whose previous container is not known\n" | |
249 | "are placed here when the main-frame is closed."); | |
250 | gEve->GetWindowManager()->SetDefaultContainer(store_tab); | |
68ca2fe7 | 251 | |
20dae051 | 252 | //============================================================================== |
253 | // AliEve objects - global tools | |
254 | //============================================================================== | |
255 | ||
256 | AliEveTrackFitter* fitter = new AliEveTrackFitter(); | |
257 | gEve->AddToListTree(fitter, 1); | |
258 | gEve->AddElement(fitter, gEve->GetEventScene()); | |
259 | ||
260 | AliEveTrackCounter* g_trkcnt = new AliEveTrackCounter("Primary Counter"); | |
261 | gEve->AddToListTree(g_trkcnt, kFALSE); | |
262 | ||
263 | ||
264 | //============================================================================== | |
265 | // Final stuff | |
266 | //============================================================================== | |
267 | ||
268 | // A refresh to show proper window. | |
269 | gEve->Redraw3D(kTRUE); | |
270 | gSystem->ProcessEvents(); | |
271 | ||
272 | // Register command to call on each event. | |
4d62585e | 273 | AliEveEventManager::GetMaster()->AddNewEventCommand("on_new_event();"); |
274 | AliEveEventManager::GetMaster()->GotoEvent(0); | |
e9b9f7d2 | 275 | |
ca49b003 | 276 | gEve->EditElement(g_trkcnt); |
277 | ||
e9b9f7d2 | 278 | gEve->Redraw3D(kTRUE); |
279 | } | |
280 | ||
281 | /******************************************************************************/ | |
282 | ||
283 | void on_new_event() | |
284 | { | |
f76c9e9b | 285 | AliEveTrackCounter* g_trkcnt = AliEveTrackCounter::fgInstance; |
e9b9f7d2 | 286 | g_trkcnt->Reset(); |
4d62585e | 287 | g_trkcnt->SetEventId(AliEveEventManager::GetMaster()->GetEventId()); |
29207369 | 288 | |
20dae051 | 289 | if (g_esd_tracks_by_category_container != 0) |
df56539f | 290 | { |
20dae051 | 291 | TEveElementList* cont = g_esd_tracks_by_category_container; |
292 | ||
293 | // Here we expect several TEveTrackList containers. | |
294 | // First two have reasonable primaries (sigma-to-prim-vertex < 5). | |
295 | // Others are almost certainly secondaries. | |
296 | Int_t count = 1; | |
297 | TEveElement::List_i i = cont->BeginChildren(); | |
298 | while (i != cont->EndChildren()) | |
29207369 | 299 | { |
20dae051 | 300 | TEveTrackList* l = dynamic_cast<TEveTrackList*>(*i); |
301 | if (l != 0) | |
302 | { | |
303 | g_trkcnt->RegisterTracks(l, (count <= 2)); | |
304 | ++count; | |
305 | } | |
306 | ++i; | |
e9b9f7d2 | 307 | } |
20dae051 | 308 | |
309 | // Set it to zero, so that we do not reuse an old one. | |
310 | g_esd_tracks_by_category_container = 0; | |
311 | } | |
312 | else | |
313 | { | |
314 | Warning("on_new_event", "g_esd_tracks_by_category_container not initialized."); | |
e9b9f7d2 | 315 | } |
316 | ||
20dae051 | 317 | Double_t x[3] = { 0, 0, 0 }; |
318 | ||
319 | if (AliEveEventManager::HasESD()) | |
a27140cb | 320 | { |
20dae051 | 321 | AliESDEvent* esd = AliEveEventManager::AssertESD(); |
322 | esd->GetPrimaryVertex()->GetXYZ(x); | |
323 | ||
a27140cb | 324 | TTimeStamp ts(esd->GetTimeStamp()); |
325 | TString win_title("Eve Main Window -- Timestamp: "); | |
326 | win_title += ts.AsString("s"); | |
5858bcbe | 327 | win_title += "; Event # in ESD file: "; |
a27140cb | 328 | win_title += esd->GetEventNumberInFile(); |
329 | gEve->GetBrowser()->SetWindowName(win_title); | |
330 | } | |
e9b9f7d2 | 331 | |
332 | TEveElement* top = gEve->GetCurrentEvent(); | |
333 | ||
20dae051 | 334 | if (gRPhiMgr && top) |
335 | { | |
336 | gRPhiEventScene->DestroyElements(); | |
337 | if (gCenterProjectionsAtPrimaryVertex) | |
338 | gRPhiMgr->SetCenter(x[0], x[1], x[2]); | |
339 | gRPhiMgr->ImportElements(top, gRPhiEventScene); | |
e9b9f7d2 | 340 | } |
20dae051 | 341 | if (gRhoZMgr && top) |
342 | { | |
343 | gRhoZEventScene->DestroyElements(); | |
344 | if (gCenterProjectionsAtPrimaryVertex) | |
345 | gRhoZMgr->SetCenter(x[0], x[1], x[2]); | |
346 | gRhoZMgr->ImportElements(top, gRhoZEventScene); | |
e9b9f7d2 | 347 | } |
e9b9f7d2 | 348 | } |