EVE/macros/alieve_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 class TEveProjectionManager;
   8 class TEveGeoShape;
   9 class TEveUtil;
  10
  11 Bool_t gCenterProjectionsAtPrimaryVertex = kFALSE;
  12
  13
  14 void alieve_online_init()
  15 {
  16
  17   if (gSystem->Getenv("ALICE_ROOT") != 0)
  18   {
  19     gInterpreter->AddIncludePath(Form("%s/MUON", gSystem->Getenv("ALICE_ROOT")));
  20     gInterpreter->AddIncludePath(Form("%s/MUON/mapping", gSystem->Getenv("ALICE_ROOT")));
  21   }
  22
  23   TEveUtil::AssertMacro("VizDB_scan.C");
  24
  25   TEveBrowser *browser = gEve->GetBrowser();
  26   browser->ShowCloseTab(kFALSE);
  27
  28   // Gentle-geom loading changes gGeoManager.
  29   TEveGeoManagerHolder mgrRestore;
  30
  31   AliEveMultiView *multiView = new AliEveMultiView;
  32
  33   TEveUtil::LoadMacro("geom_gentle.C");
  34   multiView->InitGeomGentle(geom_gentle(),
  35                              geom_gentle_rphi(),
  36                              geom_gentle_rhoz());
  37
  38   TEveUtil::LoadMacro("geom_gentle_trd.C");
  39   multiView->InitGeomGentleTrd(geom_gentle_trd());
  40
  41   TEveUtil::LoadMacro("geom_gentle_muon.C");
  42   multiView->InitGeomGentleMuon(geom_gentle_muon(), kFALSE, kTRUE);
  43
  44   //============================================================================
  45   // Standard macros to execute -- not all are enabled by default.
  46   //============================================================================
  47
  48   AliEveMacroExecutor *exec = AliEveEventManager::GetMaster()->GetExecutor();
  49
  50   exec->AddMacro(new AliEveMacro(AliEveMacro::kESD, "REC PVTX",         "primary_vertex.C", "primary_vertex",             "",                kTRUE));
  51   exec->AddMacro(new AliEveMacro(AliEveMacro::kESD, "REC PVTX Ellipse", "primary_vertex.C", "primary_vertex_ellipse",     "",                kTRUE));
  52   exec->AddMacro(new AliEveMacro(AliEveMacro::kESD, "REC PVTX Box",     "primary_vertex.C", "primary_vertex_box",         "kFALSE, 3, 3, 3", kFALSE));
  53   exec->AddMacro(new AliEveMacro(AliEveMacro::kESD, "REC PVTX",         "primary_vertex.C", "primary_vertex_spd",         "",                kTRUE));
  54   exec->AddMacro(new AliEveMacro(AliEveMacro::kESD, "REC PVTX Ellipse", "primary_vertex.C", "primary_vertex_ellipse_spd", "",                kTRUE));
  55   exec->AddMacro(new AliEveMacro(AliEveMacro::kESD, "REC PVTX Box",     "primary_vertex.C", "primary_vertex_box_spd",     "kFALSE, 3, 3, 3", kFALSE));
  56   exec->AddMacro(new AliEveMacro(AliEveMacro::kESD, "REC PVTX",         "primary_vertex.C", "primary_vertex_tpc",         "",                kFALSE));
  57   exec->AddMacro(new AliEveMacro(AliEveMacro::kESD, "REC PVTX Ellipse", "primary_vertex.C", "primary_vertex_ellipse_tpc", "",                kFALSE));
  58   exec->AddMacro(new AliEveMacro(AliEveMacro::kESD, "REC PVTX Box",     "primary_vertex.C", "primary_vertex_box_tpc",     "kFALSE, 3, 3, 3", kFALSE));
  59
  60   exec->AddMacro(new AliEveMacro(AliEveMacro::kRunLoader, "REC Clus ITS",   "its_clusters.C++",   "its_clusters"));
  61   exec->AddMacro(new AliEveMacro(AliEveMacro::kRunLoader, "REC Clus TPC",   "tpc_clusters.C++",   "tpc_clusters"));
  62   exec->AddMacro(new AliEveMacro(AliEveMacro::kRunLoader, "REC Clus TRD",   "trd_clusters.C++",   "trd_clusters"));
  63   exec->AddMacro(new AliEveMacro(AliEveMacro::kRunLoader, "REC Clus TOF",   "tof_clusters.C++",   "tof_clusters"));
  64   exec->AddMacro(new AliEveMacro(AliEveMacro::kRunLoader, "REC Clus HMPID", "hmpid_clusters.C++", "hmpid_clusters"));
  65   exec->AddMacro(new AliEveMacro(AliEveMacro::kRunLoader, "REC Clus MUON",  "muon_clusters.C++",  "muon_clusters"));
  66
  67   exec->AddMacro(new AliEveMacro(AliEveMacro::kRunLoader, "REC Clus TOF",   "emcal_digits.C++",   "emcal_digits"));
  68
  69   exec->AddMacro(new AliEveMacro(AliEveMacro::kRawReader, "RAW ITS",     "its_raw.C",     "its_raw"));
  70   exec->AddMacro(new AliEveMacro(AliEveMacro::kRawReader, "RAW TPC",     "tpc_raw.C",     "tpc_raw"));
  71   exec->AddMacro(new AliEveMacro(AliEveMacro::kRawReader, "RAW TOF",     "tof_raw.C",     "tof_raw"));
  72   exec->AddMacro(new AliEveMacro(AliEveMacro::kRawReader, "RAW VZERO",   "vzero_raw.C",   "vzero_raw"));
  73   exec->AddMacro(new AliEveMacro(AliEveMacro::kRawReader, "RAW ACORDE",  "acorde_raw.C",  "acorde_raw"));
  74   exec->AddMacro(new AliEveMacro(AliEveMacro::kRawReader, "RAW MUON",    "muon_raw.C++",  "muon_raw", "", kFALSE));
  75
  76   exec->AddMacro(new AliEveMacro(AliEveMacro::kESD, "REC Track", "esd_tracks.C", "esd_tracks",             "", kFALSE));
  77   exec->AddMacro(new AliEveMacro(AliEveMacro::kESD, "REC Track", "esd_tracks.C", "esd_tracks_MI",          "", kFALSE));
  78   exec->AddMacro(new AliEveMacro(AliEveMacro::kESD, "REC Track", "esd_tracks.C", "esd_tracks_by_category", "", kTRUE));
  79   exec->AddMacro(new AliEveMacro(AliEveMacro::kESD, "REC Track MUON", "esd_muon_tracks.C++", "esd_muon_tracks", "kTRUE,kFALSE", kTRUE));
  80
  81   // ???
  82   exec->AddMacro(new AliEveMacro(AliEveMacro::kESD, "REC TRD", "trd_detectors.C++", "trd_detectors",         "", kFALSE));
  83   // trd_tracks disabled due to memory leaks
  84
  85   //----------------------------------------------------------------------------
  86
  87   slot = TEveWindow::CreateWindowInTab(browser->GetTabRight());
  88   slot->StartEmbedding();
  89   AliEveMacroExecutorWindow* exewin = new AliEveMacroExecutorWindow(exec);
  90   slot->StopEmbedding("DataSelection");
  91   exewin->PopulateMacros();
  92
  93   //============================================================================
  94   // Final GUI setup
  95   //============================================================================
  96
  97   browser->GetTabRight()->SetTab(1);
  98
  99   browser->StartEmbedding(TRootBrowser::kBottom);
 100   new AliEveEventManagerWindow(AliEveEventManager::GetMaster());
 101   browser->StopEmbedding("EventCtrl");
 102
 103   browser->MoveResize(0, 0, gClient->GetDisplayWidth(),
 104                       gClient->GetDisplayHeight() - 32);
 105
 106   gEve->GetViewers()->SwitchColorSet();
 107
 108   TString autoRun(gSystem->Getenv("ONLINERECO_AUTORUN"));
 109   if (autoRun == "1" || autoRun.CompareTo("true", TString::kIgnoreCase) == 0)
 110   {
 111     AliEveEventManager::GetMaster()->SetAutoLoad(kTRUE);
 112   }
 113
 114   {
 115     TGTab *tab = gEve->GetBrowser()->GetTab(2);
 116
 117     TGHorizontalFrame *hf = (TGHorizontalFrame*) tab->GetParent();
 118     TGVerticalFrame   *vf = (TGVerticalFrame*)   hf ->GetParent();
 119
 120     hf->Resize(hf->GetWidth(), hf->GetHeight() + 80);
 121     vf->Layout();
 122   }
 123
 124   gEve->FullRedraw3D(kTRUE);
 125
 126   TGLViewer *glv = multiView->Get3DView()->GetGLViewer();
 127   glv->CurrentCamera().RotateRad(-0.4, 1);
 128   glv->DoDraw();
 129 }
 130
 131
 132 Int_t      g_pic_id  = 0;
 133 Int_t      g_pic_max = 10;
 134 TTimeStamp g_pic_prev(0, 0);
 135
 136 void alieve_online_on_new_event()
 137 {
 138   AliESDEvent* esd = AliEveEventManager::AssertESD();
 139   Double_t x[3];
 140   esd->GetPrimaryVertex()->GetXYZ(x);
 141
 142   TEveElement* top = gEve->GetCurrentEvent();
 143
 144   AliEveMultiView *multiView = AliEveMultiView::Instance();
 145
 146   multiView->DestroyEventRPhi();
 147   if (gCenterProjectionsAtPrimaryVertex)
 148     multiView->SetCenterRPhi(x[0], x[1], x[2]);
 149   multiView->ImportEventRPhi(top);
 150
 151   multiView->DestroyEventRhoZ();
 152   if (gCenterProjectionsAtPrimaryVertex)
 153     multiView->SetCenterRhoZ(x[0], x[1], x[2]);
 154   multiView->ImportEventRhoZ(top);
 155
 156   // Register image to amore.
 157   const TString pichost("aldaqacrs3");
 158   TTimeStamp now;
 159   Double_t delta = now.AsDouble() - g_pic_prev.AsDouble();
 160
 161   printf("Pre image dump: host='%s', delta=%f.\n",
 162          gSystem->HostName(), delta);
 163
 164   if (pichost == gSystem->HostName() && delta >= 30)
 165   {
 166     TString id;      id.Form("online-viz-%03d", g_pic_id);
 167     TString pic(id); pic += ".png";
 168
 169     printf("In image dump: file='%s'.\n", pic.Data());
 170
 171     gEve->GetBrowser()->RaiseWindow();
 172     gEve->FullRedraw3D();
 173     gSystem->ProcessEvents();
 174
 175     Int_t status;
 176
 177     status = gSystem->Exec(TString::Format("xwd -id %u | convert - %s",
 178                            gEve->GetBrowser()->GetId(), pic.Data()));
 179
 180     printf("Post capture -- status=%d.\n", status);
 181
 182     status = gSystem->Exec(TString::Format("SendImageToAmore %s %s %d",
 183                           id.Data(), pic.Data(),
 184                           AliEveEventManager::AssertRawReader()->GetRunNumber()));
 185
 186     printf("Post AMORE reg -- status=%d, run=%d.\n", status,
 187            AliEveEventManager::AssertRawReader()->GetRunNumber());
 188
 189     if (++g_pic_id >= g_pic_max)
 190       g_pic_id = 0;
 191     g_pic_prev.Set();
 192   }
 193 }