* EVE/macros/alieve_online.C
[u/mrichter/AliRoot.git] / EVE / macros / visscan_init.C
1 // $Id$
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 AliEveMacroExecutor;
11 class MultiView;
12 class TEveProjectionManager;
13 class TEveGeoShape;
14 class TEveUtil;
15
16
17 Bool_t gShowTrd      = kTRUE;
18 Bool_t gShowMuonRPhi = kFALSE;
19 Bool_t gShowMuonRhoZ = kTRUE;
20
21 Bool_t gCenterProjectionsAtPrimaryVertex = kFALSE;
22
23
24 void visscan_init(const TString& cdburi = "",
25                   const TString& path   = ".", Bool_t show_extra_geo = kFALSE)
26 {
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
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
39   if (!show_extra_geo)
40   {
41     gShowTrd = gShowMuonRPhi = gShowMuonRhoZ = kFALSE;
42   }
43
44   AliEveEventManager::AddAODfriend("AliAOD.VertexingHF.root");
45
46   TEveUtil::LoadMacro("alieve_init.C");
47   alieve_init(cdburi, path, -1);
48
49   // TEveLine::SetDefaultSmooth(1);
50
51   TEveUtil::AssertMacro("VizDB_scan.C");
52
53   AliEveMacroExecutor *exec    = AliEveEventManager::GetMaster()->GetExecutor();
54   TEveBrowser         *browser = gEve->GetBrowser();
55   browser->ShowCloseTab(kFALSE);
56
57
58   //==============================================================================
59   // Geometry, scenes, projections and viewers
60   //==============================================================================
61
62   gMultiView = new MultiView;
63
64   TEveUtil::LoadMacro("geom_gentle.C");
65   gMultiView->InitGeomGentle(geom_gentle(),
66                              geom_gentle_rphi(), 
67                              geom_gentle_rhoz());
68
69   if (gShowTrd) {
70     TEveUtil::LoadMacro("geom_gentle_trd.C");
71     gMultiView->InitGeomGentleTrd(geom_gentle_trd());
72   }
73
74   if (gShowMuonRPhi || gShowMuonRhoZ) {
75     TEveUtil::LoadMacro("geom_gentle_muon.C");
76     gMultiView->InitGeomGentleMuon(geom_gentle_muon(kFALSE), gShowMuonRPhi, gShowMuonRhoZ);
77   }
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
91   exec->AddMacro(new AliEveMacro(AliEveMacro::kRunLoader, "DIG ITS",     "its_digits.C",  "its_digits",  "", kFALSE));
92   exec->AddMacro(new AliEveMacro(AliEveMacro::kRunLoader, "DIG FMD",     "fmd_digits.C",  "fmd_digits",  "", kFALSE));
93
94   exec->AddMacro(new AliEveMacro(AliEveMacro::kRawReader, "RAW ITS",     "its_raw.C",     "its_raw",     "", kFALSE));
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"));
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"));
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
123   exec->AddMacro(new AliEveMacro(AliEveMacro::kESD, "REC Tracklet", "esd_spd_tracklets.C", "esd_spd_tracklets", "", kTRUE));
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
133   exec->AddMacro(new AliEveMacro(AliEveMacro::kRunLoader, "REC Clus TPC", "vplot_tpc.C+",    "vplot_tpc", "", kFALSE));
134
135   exec->AddMacro(new AliEveMacro(AliEveMacro::kAOD, "ANA HF",   "aod_HF.C",   "aod_HF",   "", kFALSE));
136   exec->AddMacro(new AliEveMacro(AliEveMacro::kAOD, "ANA Jets", "jetplane.C", "jetplane", "", kFALSE));
137
138
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
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
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);
172
173
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.
195   AliEveEventManager::GetMaster()->AddNewEventCommand("on_new_event();");
196   AliEveEventManager::GetMaster()->GotoEvent(0);
197
198   gEve->EditElement(g_trkcnt);
199
200   gEve->Redraw3D(kTRUE);
201 }
202
203 /******************************************************************************/
204
205 void on_new_event()
206 {
207   AliEveTrackCounter* g_trkcnt = AliEveTrackCounter::fgInstance;
208   g_trkcnt->Reset();
209   g_trkcnt->SetEventId(AliEveEventManager::GetMaster()->GetEventId());
210
211   if (g_esd_tracks_by_category_container != 0)
212   {
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())
221     {
222       TEveTrackList* l = dynamic_cast<TEveTrackList*>(*i);
223       if (l != 0)
224       {
225         g_trkcnt->RegisterTracks(l, (count <= 2));
226         ++count;
227       }
228       ++i;
229     }
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.");
237   }
238
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
249   Double_t x[3] = { 0, 0, 0 };
250
251   if (AliEveEventManager::HasESD())
252   {
253     AliESDEvent* esd = AliEveEventManager::AssertESD();
254     esd->GetPrimaryVertex()->GetXYZ(x);
255
256     TTimeStamp ts(esd->GetTimeStamp());
257     TString win_title("Eve Main Window -- Timestamp: ");
258     win_title += ts.AsString("s");
259     win_title += "; Event # in ESD file: ";
260     win_title += esd->GetEventNumberInFile();
261     gEve->GetBrowser()->SetWindowName(win_title);
262   }
263
264   TEveElement* top = gEve->GetCurrentEvent();
265
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);
275 }