EVE/macros/online.C

   1 /**************************************************************************
   2  * Copyright(c) 1998-2008, ALICE Experiment at CERN, all rights reserved. *
   3  * See http://aliceinfo.cern.ch/Offline/AliRoot/License.html for          *
   4  * full copyright notice.                                                 *
   5  **************************************************************************/
   6
   7 #include <zmq.hpp>
   8
   9 class TEveProjectionManager;
  10 class TEveGeoShape;
  11 class TEveUtil;
  12 class AliTriggerAnalysis;
  13 class AliSysInfo;
  14
  15 TH2D* V0StateHistogram;
  16
  17 Bool_t gCenterProjectionsAtPrimaryVertex = kFALSE;
  18 //Container for gGeoManager till it is broken
  19 TGeoManager *fGeoManager = 0;
  20
  21 void online()
  22 {
  23     printf("online() ...\n");
  24
  25   if (gSystem->Getenv("ALICE_ROOT") != 0)
  26   {
  27     gInterpreter->AddIncludePath(Form("%s/MUON", gSystem->Getenv("ALICE_ROOT")));
  28     gInterpreter->AddIncludePath(Form("%s/MUON/mapping", gSystem->Getenv("ALICE_ROOT")));
  29   }
  30
  31     Info("alieve_init", "Adding standard macros.");
  32     TString  hack = gSystem->pwd(); // Problem with TGFileBrowser cding
  33     alieve_init_import_macros();
  34     gSystem->cd(hack);
  35
  36   TEveUtil::LoadMacro("alieve_init.C");
  37   alieve_init("local:///opt/OCDB/2013", ".", -1);
  38
  39
  40   TEveUtil::AssertMacro("VizDB_scan.C");
  41
  42   AliEveMacroExecutor *exec    = AliEveEventManager::GetMaster()->GetExecutor();
  43   TEveBrowser         *browser = gEve->GetBrowser();
  44   browser->ShowCloseTab(kFALSE);
  45
  46
  47   //==============================================================================
  48   // Geometry, scenes, projections and viewers
  49   //==============================================================================
  50
  51   AliEveMultiView *mv = new AliEveMultiView;
  52
  53   mv->SetDepth(-10);
  54
  55   TEveUtil::LoadMacro("geom_gentle.C");
  56   mv->InitGeomGentle(geom_gentle(), geom_gentle_rphi(), geom_gentle_rhoz(), 0);
  57
  58
  59     TEveUtil::LoadMacro("geom_gentle_trd.C");
  60     mv->InitGeomGentleTrd(geom_gentle_trd());
  61
  62
  63
  64     TEveUtil::LoadMacro("geom_gentle_muon.C");
  65     mv->InitGeomGentleMuon(geom_gentle_muon(kFALSE), kFALSE, kFALSE, kFALSE);
  66
  67
  68   mv->SetDepth(0);
  69
  70   //==============================================================================
  71   // Registration of per-event macros
  72   //==============================================================================
  73
  74   exec->AddMacro(new AliEveMacro(AliEveMacro::kRunLoader, "SIM Track",   "kine_tracks.C", "kine_tracks", "", kFALSE));
  75
  76   exec->AddMacro(new AliEveMacro(AliEveMacro::kRunLoader, "SIM Hits ITS", "its_hits.C",    "its_hits",    "", kFALSE));
  77   exec->AddMacro(new AliEveMacro(AliEveMacro::kRunLoader, "SIM Hits TPC", "tpc_hits.C",    "tpc_hits",    "", kFALSE));
  78   exec->AddMacro(new AliEveMacro(AliEveMacro::kRunLoader, "SIM Hits T0",  "t0_hits.C",     "t0_hits",     "", kFALSE));
  79   exec->AddMacro(new AliEveMacro(AliEveMacro::kRunLoader, "SIM Hits FMD", "fmd_hits.C",    "fmd_hits",    "", kFALSE));
  80   exec->AddMacro(new AliEveMacro(AliEveMacro::kRunLoader, "SIM Hits ACORDE", "acorde_hits.C",    "acorde_hits",    "", kFALSE));
  81   exec->AddMacro(new AliEveMacro(AliEveMacro::kRunLoader, "SIM Hits EMCAL", "emcal_hits.C",    "emcal_hits",    "", kFALSE));
  82   exec->AddMacro(new AliEveMacro(AliEveMacro::kRunLoader, "SIM Hits TOF",  "tof_hits.C",     "tof_hits",     "", kFALSE));
  83   exec->AddMacro(new AliEveMacro(AliEveMacro::kRunLoader, "SIM Hits TRD", "trd_hits.C",    "trd_hits",    "", kFALSE));
  84   exec->AddMacro(new AliEveMacro(AliEveMacro::kRunLoader, "SIM Hits VZERO", "vzero_hits.C",    "vzero_hits",    "", kFALSE));
  85
  86   exec->AddMacro(new AliEveMacro(AliEveMacro::kRunLoader, "DIG ITS",     "its_digits.C",  "its_digits",  "", kFALSE));
  87   exec->AddMacro(new AliEveMacro(AliEveMacro::kRunLoader, "DIG TPC",     "tpc_digits.C",  "tpc_digits",  "", kFALSE));
  88   exec->AddMacro(new AliEveMacro(AliEveMacro::kRunLoader, "DIG TOF",     "tof_digits.C",  "tof_digits",  "", kFALSE));
  89   exec->AddMacro(new AliEveMacro(AliEveMacro::kRunLoader, "DIG HMPID",   "hmpid_digits.C","hmpid_digits","", kFALSE));
  90   exec->AddMacro(new AliEveMacro(AliEveMacro::kRunLoader, "DIG FMD",     "fmd_digits.C",  "fmd_digits",  "", kFALSE));
  91
  92   exec->AddMacro(new AliEveMacro(AliEveMacro::kRawReader, "RAW ITS",     "its_raw.C",     "its_raw",     "", kFALSE));
  93   exec->AddMacro(new AliEveMacro(AliEveMacro::kRawReader, "RAW TPC",     "tpc_raw.C",     "tpc_raw",     "", kFALSE));
  94   exec->AddMacro(new AliEveMacro(AliEveMacro::kRawReader, "RAW TOF",     "tof_raw.C",     "tof_raw",     "", kFALSE));
  95   exec->AddMacro(new AliEveMacro(AliEveMacro::kRawReader, "RAW HMPID",   "hmpid_raw.C",   "hmpid_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 SPD",         "primary_vertex.C", "primary_vertex_spd",         "",                kTRUE));
 105   exec->AddMacro(new AliEveMacro(AliEveMacro::kESD, "REC PVTX Ellipse SPD", "primary_vertex.C", "primary_vertex_ellipse_spd", "",                kTRUE));
 106   exec->AddMacro(new AliEveMacro(AliEveMacro::kESD, "REC PVTX Box SPD",     "primary_vertex.C", "primary_vertex_box_spd",     "kFALSE, 3, 3, 3", kFALSE));
 107   exec->AddMacro(new AliEveMacro(AliEveMacro::kESD, "REC PVTX TPC",         "primary_vertex.C", "primary_vertex_tpc",         "",                kFALSE));
 108   exec->AddMacro(new AliEveMacro(AliEveMacro::kESD, "REC PVTX Ellipse TPC", "primary_vertex.C", "primary_vertex_ellipse_tpc", "",                kFALSE));
 109   exec->AddMacro(new AliEveMacro(AliEveMacro::kESD, "REC PVTX Box TPC",     "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 Tracks",              "esd_tracks.C", "esd_tracks",              "", kFALSE));
 120   exec->AddMacro(new AliEveMacro(AliEveMacro::kESD, "REC Tracks ITS standalone",          "esd_tracks.C", "esd_tracks_ITS_standalone",              "", kFALSE));
 121   exec->AddMacro(new AliEveMacro(AliEveMacro::kESD, "REC Tracks ITS",          "esd_tracks.C", "esd_tracks_ITS",              "", kFALSE));
 122   exec->AddMacro(new AliEveMacro(AliEveMacro::kESD, "REC Tracks TPC",           "esd_tracks.C", "esd_tracks_TPC",              "", kFALSE));
 123   exec->AddMacro(new AliEveMacro(AliEveMacro::kESD, "REC Tracks MI",           "esd_tracks.C", "esd_tracks_MI",           "", kFALSE));
 124   exec->AddMacro(new AliEveMacro(AliEveMacro::kESD, "REC Tracks by category",  "esd_tracks.C", "esd_tracks_by_category",  "", kTRUE));
 125   exec->AddMacro(new AliEveMacro(AliEveMacro::kESD, "REC Tracks by anal cuts", "esd_tracks.C", "esd_tracks_by_anal_cuts", "", kFALSE));
 126   exec->AddMacro(new AliEveMacro(AliEveMacro::kESD, "REC Tracks Lego", "lego.C", "lego", "", kFALSE));
 127   exec->AddMacro(new AliEveMacro(AliEveMacro::kESD, "REC Tracks Beams Info", "beams_info.C", "beams_info", "", kFALSE));
 128
 129   exec->AddMacro(new AliEveMacro(AliEveMacro::kESD, "REC Tracklets SPD", "esd_spd_tracklets.C", "esd_spd_tracklets", "", kTRUE));
 130
 131   exec->AddMacro(new AliEveMacro(AliEveMacro::kESD, "REC ZDC",      "esd_zdc.C", "esd_zdc", "", kFALSE));
 132
 133   exec->AddMacro(new AliEveMacro(AliEveMacro::kRunLoader, "REC Clusters",     "clusters.C",     "clusters", "", kFALSE));
 134   exec->AddMacro(new AliEveMacro(AliEveMacro::kRunLoader, "REC Clusters ITS", "its_clusters.C", "its_clusters"));
 135   exec->AddMacro(new AliEveMacro(AliEveMacro::kRunLoader, "REC Clusters TPC", "tpc_clusters.C", "tpc_clusters"));
 136   exec->AddMacro(new AliEveMacro(AliEveMacro::kRunLoader, "REC Clusters TRD", "trd_clusters.C", "trd_clusters"));
 137   exec->AddMacro(new AliEveMacro(AliEveMacro::kRunLoader, "REC Clusters TOF", "tof_clusters.C", "tof_clusters"));
 138   exec->AddMacro(new AliEveMacro(AliEveMacro::kRunLoader, "REC Clusters HMPID", "hmpid_clusters.C", "hmpid_clusters"));
 139   exec->AddMacro(new AliEveMacro(AliEveMacro::kRunLoader, "REC Clusters PHOS", "phos_clusters.C", "phos_clusters"));
 140
 141   exec->AddMacro(new AliEveMacro(AliEveMacro::kRunLoader, "REC Clusters TPC", "vplot_tpc.C",    "vplot_tpc", "", kFALSE));
 142
 143   exec->AddMacro(new AliEveMacro(AliEveMacro::kAOD, "ANA HF",   "aod_HF.C",   "aod_HF",   "", kFALSE));
 144   exec->AddMacro(new AliEveMacro(AliEveMacro::kAOD, "ANA Jets", "jetplane.C", "jetplane", "", kFALSE));
 145
 146   exec->AddMacro(new AliEveMacro(AliEveMacro::kESD, "DUMP VZERO",   "vzero_dump.C",   "vzero_dump",   "", kFALSE));
 147   /*
 148   if (showMuon)
 149   {
 150     exec->AddMacro(new AliEveMacro(AliEveMacro::kRunLoader, "SIM TrackRef MUON", "muon_trackRefs.C", "muon_trackRefs", "kTRUE", kFALSE));
 151     exec->AddMacro(new AliEveMacro(AliEveMacro::kRawReader, "RAW MUON", "muon_raw.C", "muon_raw", "", kFALSE));
 152     exec->AddMacro(new AliEveMacro(AliEveMacro::kRunLoader, "DIG MUON", "muon_digits.C", "muon_digits", "", kFALSE));
 153     exec->AddMacro(new AliEveMacro(AliEveMacro::kRunLoader, "REC Clusters MUON", "muon_clusters.C", "muon_clusters", "", kTRUE));
 154     exec->AddMacro(new AliEveMacro(AliEveMacro::kESD, "REC Tracks MUON", "esd_muon_tracks.C", "esd_muon_tracks", "kTRUE,kFALSE", kTRUE));
 155   }
 156   */
 157
 158   //==============================================================================
 159   // Additional GUI components
 160   //==============================================================================
 161
 162   // Macro / data selection
 163   slot = TEveWindow::CreateWindowInTab(browser->GetTabRight());
 164   slot->StartEmbedding();
 165   AliEveMacroExecutorWindow* exewin = new AliEveMacroExecutorWindow(exec);
 166   slot->StopEmbedding("DataSelection");
 167   exewin->PopulateMacros();
 168
 169   // Event selection tab
 170   slot = TEveWindow::CreateWindowInTab(browser->GetTabRight());
 171   slot->StartEmbedding();
 172   new AliEveEventSelectorWindow(gClient->GetRoot(), 600, 400, AliEveEventManager::GetMaster()->GetEventSelector());
 173   slot->StopEmbedding("Selections");
 174
 175   // QA viewer
 176 /*
 177   slot = TEveWindow::CreateWindowInTab(browser->GetTabRight());
 178   slot->StartEmbedding();
 179   new AliQAHistViewer(gClient->GetRoot(), 600, 400, kTRUE);
 180   slot->StopEmbedding("QA histograms");
 181
 182   browser->GetTabRight()->SetTab(1);
 183 */
 184   browser->StartEmbedding(TRootBrowser::kBottom);
 185   new AliEveEventManagerWindow(AliEveEventManager::GetMaster());
 186   browser->StopEmbedding("EventCtrl");
 187
 188   slot = TEveWindow::CreateWindowInTab(browser->GetTabRight());
 189   TEveWindowTab *store_tab = slot->MakeTab();
 190   store_tab->SetElementNameTitle("WindowStore",
 191     "Undocked windows whose previous container is not known\n"
 192     "are placed here when the main-frame is closed.");
 193   gEve->GetWindowManager()->SetDefaultContainer(store_tab);
 194
 195
 196   //==============================================================================
 197   // AliEve objects - global tools
 198   //==============================================================================
 199
 200   AliEveTrackFitter* fitter = new AliEveTrackFitter();
 201   gEve->AddToListTree(fitter, 1);
 202   gEve->AddElement(fitter, gEve->GetEventScene());
 203
 204   AliEveTrackCounter* g_trkcnt = new AliEveTrackCounter("Primary Counter");
 205   gEve->AddToListTree(g_trkcnt, kFALSE);
 206
 207
 208
 209
 210 printf("================================ Connecting to Server ...\n");
 211
 212 AliEveEventManager::GetMaster()->ConnectToServer("tcp://*",5024);
 213
 214 zmq::socket_t* subscriber = AliEveEventManager::GetMaster()->AssertSubscriber();
 215
 216 if(subscriber ==0) {
 217     printf("===================== Not connected! ====================\n");
 218 }
 219
 220
 221   //==============================================================================
 222   // Final stuff
 223   //==============================================================================
 224
 225   // A refresh to show proper window.
 226   gEve->GetViewers()->SwitchColorSet();
 227   gEve->Redraw3D(kTRUE);
 228   gSystem->ProcessEvents();
 229
 230   // Register command to call on each event.
 231   AliEveEventManager::GetMaster()->AddNewEventCommand("on_new_event();");
 232   AliEveEventManager::GetMaster()->GotoEvent(0);
 233
 234   gEve->EditElement(g_trkcnt);
 235   gEve->Redraw3D(kTRUE);
 236 }
 237
 238
 239 /******************************************************************************/
 240
 241 void on_new_event()
 242 {
 243   Double_t x[3] = { 0, 0, 0 };
 244
 245   if (AliEveEventManager::HasESD())
 246   {
 247     AliESDEvent* esd = AliEveEventManager::AssertESD();
 248     esd->GetPrimaryVertex()->GetXYZ(x);
 249
 250     TTimeStamp ts(esd->GetTimeStamp());
 251     TString win_title("Eve Main Window -- Timestamp: ");
 252     win_title += ts.AsString("s");
 253     win_title += "; Event # in ESD file: ";
 254     win_title += esd->GetEventNumberInFile();
 255     gEve->GetBrowser()->SetWindowName(win_title);
 256   }
 257
 258   TEveElement* top = gEve->GetCurrentEvent();
 259
 260   AliEveMultiView *mv = AliEveMultiView::Instance();
 261
 262   //mv->DestroyEventRPhi();
 263   if (gCenterProjectionsAtPrimaryVertex)
 264     mv->SetCenterRPhi(x[0], x[1], x[2]);
 265   mv->ImportEventRPhi(top);
 266
 267   //mv->DestroyEventRhoZ();
 268   if (gCenterProjectionsAtPrimaryVertex)
 269     mv->SetCenterRhoZ(x[0], x[1], x[2]);
 270   mv->ImportEventRhoZ(top);
 271 }
 272
 273
 274 Int_t      g_pic_id  = 0;
 275 Int_t      g_pic_max = 100;
 276 TTimeStamp g_pic_prev(0, 0);
 277
 278
 279 void alieve_init_import_macros()
 280 {
 281   // Put macros in the list of browsables, add a macro browser to
 282   // top-level GUI.
 283
 284   TString macdir("$(ALICE_ROOT)/EVE/alice-macros");
 285   gSystem->ExpandPathName(macdir);
 286
 287   TFolder* f = gEve->GetMacroFolder();
 288   void* dirhandle = gSystem->OpenDirectory(macdir.Data());
 289   if (dirhandle != 0)
 290   {
 291     char* filename;
 292     TPMERegexp re("\\.C$");
 293     TObjArray names;
 294     while ((filename = gSystem->GetDirEntry(dirhandle)) != 0)
 295     {
 296       if (re.Match(filename))
 297         names.AddLast(new TObjString(filename));
 298     }
 299     names.Sort();
 300
 301     for (Int_t ii=0; ii<names.GetEntries(); ++ii)
 302     {
 303       TObjString * si = (TObjString*) names.At(ii);
 304        f->Add(new TEveMacro(Form("%s/%s", macdir.Data(), (si->GetString()).Data())));
 305     }
 306   }
 307   gSystem->FreeDirectory(dirhandle);
 308
 309   gROOT->GetListOfBrowsables()->Add(new TSystemDirectory(macdir.Data(), macdir.Data()));
 310
 311   {
 312     TEveBrowser   *br = gEve->GetBrowser();
 313     TGFileBrowser *fb = 0;
 314     fb = br->GetFileBrowser();
 315     fb->GotoDir(macdir);
 316     {
 317       br->StartEmbedding(0);
 318       fb = br->MakeFileBrowser();
 319       fb->BrowseObj(f);
 320       fb->Show();
 321       br->StopEmbedding();
 322       br->SetTabTitle("Macros", 0);
 323       br->SetTab(0, 0);
 324     }
 325   }
 326 }