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 class AliTriggerAnalysis;
  11
  12 TH2D* V0StateHistogram;
  13
  14 Bool_t gCenterProjectionsAtPrimaryVertex = kFALSE;
  15
  16 void alieve_online_init()
  17 {
  18
  19   if (gSystem->Getenv("ALICE_ROOT") != 0)
  20   {
  21     gInterpreter->AddIncludePath(Form("%s/MUON", gSystem->Getenv("ALICE_ROOT")));
  22     gInterpreter->AddIncludePath(Form("%s/MUON/mapping", gSystem->Getenv("ALICE_ROOT")));
  23   }
  24
  25   TEveUtil::AssertMacro("VizDB_scan.C");
  26
  27   TEveBrowser *browser = gEve->GetBrowser();
  28   browser->ShowCloseTab(kFALSE);
  29
  30   // Gentle-geom loading changes gGeoManager.
  31   TEveGeoManagerHolder mgrRestore;
  32
  33   AliEveMultiView *multiView = new AliEveMultiView(kTRUE);
  34
  35   TEveUtil::LoadMacro("geom_gentle.C");
  36   multiView->InitGeomGentle(geom_gentle(),
  37                              geom_gentle_rphi(),
  38                              geom_gentle_rhoz(),
  39                              geom_gentle_rhoz());
  40
  41   TEveUtil::LoadMacro("geom_gentle_trd.C");
  42   multiView->InitGeomGentleTrd(geom_gentle_trd());
  43
  44   TEveUtil::LoadMacro("geom_gentle_muon.C");
  45   multiView->InitGeomGentleMuon(geom_gentle_muon(), kFALSE, kFALSE, kTRUE);
  46
  47   //============================================================================
  48   // Standard macros to execute -- not all are enabled by default.
  49   //============================================================================
  50
  51   AliEveMacroExecutor *exec = AliEveEventManager::GetMaster()->GetExecutor();
  52
  53   exec->AddMacro(new AliEveMacro(AliEveMacro::kESD, "REC PVTX",         "primary_vertex.C", "primary_vertex",             "",                kTRUE));
  54   exec->AddMacro(new AliEveMacro(AliEveMacro::kESD, "REC PVTX Ellipse", "primary_vertex.C", "primary_vertex_ellipse",     "",                kTRUE));
  55   exec->AddMacro(new AliEveMacro(AliEveMacro::kESD, "REC PVTX Box",     "primary_vertex.C", "primary_vertex_box",         "kFALSE, 3, 3, 3", kFALSE));
  56   exec->AddMacro(new AliEveMacro(AliEveMacro::kESD, "REC PVTX",         "primary_vertex.C", "primary_vertex_spd",         "",                kTRUE));
  57   exec->AddMacro(new AliEveMacro(AliEveMacro::kESD, "REC PVTX Ellipse", "primary_vertex.C", "primary_vertex_ellipse_spd", "",                kTRUE));
  58   exec->AddMacro(new AliEveMacro(AliEveMacro::kESD, "REC PVTX Box",     "primary_vertex.C", "primary_vertex_box_spd",     "kFALSE, 3, 3, 3", kFALSE));
  59   exec->AddMacro(new AliEveMacro(AliEveMacro::kESD, "REC PVTX",         "primary_vertex.C", "primary_vertex_tpc",         "",                kFALSE));
  60   exec->AddMacro(new AliEveMacro(AliEveMacro::kESD, "REC PVTX Ellipse", "primary_vertex.C", "primary_vertex_ellipse_tpc", "",                kFALSE));
  61   exec->AddMacro(new AliEveMacro(AliEveMacro::kESD, "REC PVTX Box",     "primary_vertex.C", "primary_vertex_box_tpc",     "kFALSE, 3, 3, 3", kFALSE));
  62
  63   exec->AddMacro(new AliEveMacro(AliEveMacro::kRunLoader, "REC Clus ITS",   "its_clusters.C++",   "its_clusters"));
  64   exec->AddMacro(new AliEveMacro(AliEveMacro::kRunLoader, "REC Clus TPC",   "tpc_clusters.C++",   "tpc_clusters"));
  65   exec->AddMacro(new AliEveMacro(AliEveMacro::kRunLoader, "REC Clus TRD",   "trd_clusters.C++",   "trd_clusters"));
  66   exec->AddMacro(new AliEveMacro(AliEveMacro::kRunLoader, "REC Clus TOF",   "tof_clusters.C++",   "tof_clusters"));
  67   exec->AddMacro(new AliEveMacro(AliEveMacro::kRunLoader, "REC Clus HMPID", "hmpid_clusters.C++", "hmpid_clusters"));
  68   exec->AddMacro(new AliEveMacro(AliEveMacro::kRunLoader, "REC Clus MUON",  "muon_clusters.C++",  "muon_clusters"));
  69
  70   exec->AddMacro(new AliEveMacro(AliEveMacro::kRunLoader, "REC Clus TOF",   "emcal_digits.C++",   "emcal_digits"));
  71
  72   exec->AddMacro(new AliEveMacro(AliEveMacro::kRawReader, "RAW ITS",     "its_raw.C",     "its_raw"));
  73   exec->AddMacro(new AliEveMacro(AliEveMacro::kRawReader, "RAW TPC",     "tpc_raw.C",     "tpc_raw"));
  74   exec->AddMacro(new AliEveMacro(AliEveMacro::kRawReader, "RAW TOF",     "tof_raw.C",     "tof_raw"));
  75   exec->AddMacro(new AliEveMacro(AliEveMacro::kRawReader, "RAW VZERO",   "vzero_raw.C",   "vzero_raw"));
  76   exec->AddMacro(new AliEveMacro(AliEveMacro::kRawReader, "RAW ACORDE",  "acorde_raw.C",  "acorde_raw"));
  77   exec->AddMacro(new AliEveMacro(AliEveMacro::kRawReader, "RAW MUON",    "muon_raw.C++",  "muon_raw"));
  78   exec->AddMacro(new AliEveMacro(AliEveMacro::kRawReader, "RAW FMD",     "fmd_raw.C",     "fmd_raw"));
  79
  80   exec->AddMacro(new AliEveMacro(AliEveMacro::kESD, "REC Track",      "esd_tracks.C",        "esd_tracks",             "", kFALSE));
  81   exec->AddMacro(new AliEveMacro(AliEveMacro::kESD, "REC Track",      "esd_tracks.C",        "esd_tracks_MI",          "", kFALSE));
  82   exec->AddMacro(new AliEveMacro(AliEveMacro::kESD, "REC Track",      "esd_tracks.C",        "esd_tracks_by_category", "", kTRUE));
  83   exec->AddMacro(new AliEveMacro(AliEveMacro::kESD, "REC Track MUON", "esd_muon_tracks.C++", "esd_muon_tracks",        "kTRUE,kFALSE", kTRUE));
  84   exec->AddMacro(new AliEveMacro(AliEveMacro::kESD, "REC FMD",        "fmd_esd.C",           "fmd_esd",                "", kTRUE));
  85
  86   // ???
  87   // exec->AddMacro(new AliEveMacro(AliEveMacro::kESD, "REC TRD", "trd_detectors.C++", "trd_detectors",         "", kFALSE));
  88   // trd_tracks disabled due to memory leaks
  89
  90   //----------------------------------------------------------------------------
  91
  92   slot = TEveWindow::CreateWindowInTab(browser->GetTabRight());
  93   slot->StartEmbedding();
  94   AliEveMacroExecutorWindow* exewin = new AliEveMacroExecutorWindow(exec);
  95   slot->StopEmbedding("DataSelection");
  96   exewin->PopulateMacros();
  97
  98   //============================================================================
  99   // VZERO state histogram
 100   //============================================================================
 101
 102   slot = TEveWindow::CreateWindowInTab(browser->GetTabRight());
 103   slot->StartEmbedding();
 104   TCanvas* pad = new TCanvas();
 105
 106   gStyle->SetCanvasColor(0);
 107
 108   V0StateHistogram = new TH2D("V0 Histogram","V0 Trigger State", 4, 0, 4, 4, 0, 4);
 109   V0StateHistogram->Draw("colz");
 110
 111   V0StateHistogram->GetXaxis()->SetBinLabel(1,"V0A Invalid");
 112   V0StateHistogram->GetXaxis()->SetBinLabel(2,"V0A Empty");
 113   V0StateHistogram->GetXaxis()->SetBinLabel(3,"V0A BB");
 114   V0StateHistogram->GetXaxis()->SetBinLabel(4,"V0A BG");
 115
 116   V0StateHistogram->GetYaxis()->SetBinLabel(1,"V0C Invalid");
 117   V0StateHistogram->GetYaxis()->SetBinLabel(2,"V0C Empty");
 118   V0StateHistogram->GetYaxis()->SetBinLabel(3,"V0C BB");
 119   V0StateHistogram->GetYaxis()->SetBinLabel(4,"V0C BG");
 120
 121   slot->StopEmbedding("V0 Trigger State");
 122
 123   //============================================================================
 124   // Final GUI setup
 125   //============================================================================
 126
 127   browser->GetTabRight()->SetTab(1);
 128
 129   browser->StartEmbedding(TRootBrowser::kBottom);
 130   new AliEveEventManagerWindow(AliEveEventManager::GetMaster());
 131   browser->StopEmbedding("EventCtrl");
 132
 133   browser->MoveResize(0, 0, gClient->GetDisplayWidth(),
 134                       gClient->GetDisplayHeight() - 32);
 135
 136   gEve->GetViewers()->SwitchColorSet();
 137
 138   TString autoRun(gSystem->Getenv("ONLINERECO_AUTORUN"));
 139   if (autoRun == "1" || autoRun.CompareTo("true", TString::kIgnoreCase) == 0)
 140   {
 141     AliEveEventManager::GetMaster()->SetAutoLoad(kTRUE);
 142   }
 143
 144   {
 145     TGTab *tab = gEve->GetBrowser()->GetTab(2);
 146
 147     TGHorizontalFrame *hf = (TGHorizontalFrame*) tab->GetParent();
 148     TGVerticalFrame   *vf = (TGVerticalFrame*)   hf ->GetParent();
 149
 150     hf->Resize(hf->GetWidth(), hf->GetHeight() + 80);
 151     vf->Layout();
 152   }
 153
 154   gEve->GetWindowManager()->HideAllEveDecorations();
 155
 156   gEve->FullRedraw3D(kTRUE);
 157
 158   TGLViewer *glv = multiView->Get3DView()->GetGLViewer();
 159   gEve->FullRedraw3D(kTRUE);
 160   glv->CurrentCamera().RotateRad(-0.4, -1.8);
 161
 162 }
 163
 164 //   multiView->Get3DView()->GetGLViewer()->CurrentCamera().RotateRad(-1, 1)
 165
 166 Int_t      g_pic_id  = 0;
 167 Int_t      g_pic_max = 100;
 168 TTimeStamp g_pic_prev(0, 0);
 169
 170 void alieve_online_on_new_event()
 171 {
 172
 173   using namespace AliTriggerAnalysis;
 174
 175   AliESDEvent* esd = AliEveEventManager::AssertESD();
 176   Double_t x[3] = { 0, 0, 0 };
 177   esd->GetPrimaryVertex()->GetXYZ(x);
 178
 179   TEveElement* top = gEve->GetCurrentEvent();
 180
 181   AliEveMultiView *multiView = AliEveMultiView::Instance();
 182
 183 /*
 184   TGLViewer *glv = (dynamic_cast<TEveViewer*>(gEve->GetViewers()->FindChild("3D View")))->GetGLViewer();
 185
 186   if(gEve->GetScenes()->FirstChild()->FindChild("Gentle MUON"))
 187   {
 188     if (esd->GetNumberOfMuonTracks() == 0 && !gEve->GetKeepEmptyCont())
 189     {
 190       gEve->GetScenes()->FirstChild()->FindChild("Gentle MUON")->SetRnrChildren(kFALSE);
 191
 192       if(gEve->GetEventScene()->FirstChild()->FindChild("MUON Clusters"))
 193         gEve->GetEventScene()->FirstChild()->FindChild("MUON Clusters")->SetRnrSelf(kFALSE);
 194       if(gEve->GetEventScene()->FirstChild()->FindChild("MUON Raw digits"))
 195         gEve->GetEventScene()->FirstChild()->FindChild("MUON Raw digits")->SetRnrChildren(kFALSE);
 196
 197       gEve->FullRedraw3D(kTRUE);
 198       glv->CurrentCamera().RotateRad(-0.4, -1.8);
 199     }
 200     else
 201     {
 202       gEve->GetScenes()->FirstChild()->FindChild("Gentle MUON")->SetRnrChildren(kTRUE);
 203
 204       if(gEve->GetEventScene()->FirstChild()->FindChild("MUON Clusters"))
 205         gEve->GetEventScene()->FirstChild()->FindChild("MUON Clusters")->SetRnrSelf(kTRUE);
 206       if(gEve->GetEventScene()->FirstChild()->FindChild("MUON Raw digits"))
 207         gEve->GetEventScene()->FirstChild()->FindChild("MUON Raw digits")->SetRnrChildren(kTRUE);
 208
 209       gEve->FullRedraw3D(kTRUE);
 210       glv->CurrentCamera().RotateRad(-0.4, 1);
 211     }
 212   }
 213
 214   glv->DoDraw();
 215 */
 216
 217   V0Decision decisionV0a = V0Trigger(esd, kASide, kFALSE);
 218   V0Decision decisionV0c = V0Trigger(esd, kCSide, kFALSE);
 219
 220   Double_t a = 0;
 221   Double_t c = 0;
 222
 223   if( decisionV0a == kV0Invalid ) a = 0.5;
 224   if( decisionV0a == kV0Empty ) a = 1.5;
 225   if( decisionV0a == kV0BB ) a = 2.5;
 226   if( decisionV0a == kV0BG ) a = 3.5;
 227
 228   if( decisionV0c == kV0Invalid ) c = 0.5;
 229   if( decisionV0c == kV0Empty ) c = 1.5;
 230   if( decisionV0c == kV0BB ) c = 2.5;
 231   if( decisionV0c == kV0BG ) c = 3.5;
 232
 233   V0StateHistogram->Fill(a,c);
 234
 235   TGLViewer *glv = (dynamic_cast<TEveViewer*>(gEve->GetViewers()->FindChild("3D View")))->GetGLViewer();
 236   TGLViewer *glv1 = (dynamic_cast<TEveViewer*>(gEve->GetViewers()->FindChild("RPhi View")))->GetGLViewer();
 237   TGLViewer *glv2 = (dynamic_cast<TEveViewer*>(gEve->GetViewers()->FindChild("RhoZ View")))->GetGLViewer();
 238
 239   Double_t RPhiCameraFrustrumCenter = TMath::Sqrt(glv1->CurrentCamera().FrustumCenter().X()*glv1->CurrentCamera().FrustumCenter().X() + glv1->CurrentCamera().FrustumCenter().Y()*glv1->CurrentCamera().FrustumCenter().Y());
 240
 241   Double_t RhoZCameraFrustrumCenter = TMath::Sqrt(glv2->CurrentCamera().FrustumCenter().X()*glv2->CurrentCamera().FrustumCenter().X() + glv2->CurrentCamera().FrustumCenter().Y()*glv2->CurrentCamera().FrustumCenter().Y());
 242
 243   if(RPhiCameraFrustrumCenter > 500 || RhoZCameraFrustrumCenter > 500)
 244   {
 245
 246     glv->ResetCurrentCamera();
 247     glv->CurrentCamera().RotateRad(-0.3, -1.8);
 248     glv->CurrentCamera().Dolly(250, kFALSE, kFALSE);
 249
 250     glv1->ResetCurrentCamera();
 251     glv2->ResetCurrentCamera();
 252
 253     gEve->FullRedraw3D();
 254
 255   }
 256
 257   multiView->DestroyEventRPhi();
 258   if (gCenterProjectionsAtPrimaryVertex)
 259     multiView->SetCenterRPhi(x[0], x[1], x[2]);
 260   multiView->ImportEventRPhi(top);
 261
 262   multiView->DestroyEventRhoZ();
 263   if (gCenterProjectionsAtPrimaryVertex)
 264     multiView->SetCenterRhoZ(x[0], x[1], x[2]);
 265   multiView->ImportEventRhoZ(top);
 266
 267   if(multiView->IsMuonView()) multiView->ImportEventMuon(top);
 268
 269   // Register image to amore.
 270   const TString pichost("aldaqacrs3");
 271   TTimeStamp now;
 272   Double_t delta = now.AsDouble() - g_pic_prev.AsDouble();
 273
 274   printf("Pre image dump: host='%s', delta=%f.\n",
 275          gSystem->HostName(), delta);
 276
 277   if (pichost == gSystem->HostName() && delta >= 30)
 278   {
 279     TString id;      id.Form("online-viz-%03d", g_pic_id);
 280     TString pic(id); pic += ".png";
 281
 282     printf("In image dump: file='%s'.\n", pic.Data());
 283
 284     gEve->GetBrowser()->RaiseWindow();
 285     gEve->FullRedraw3D();
 286     gSystem->ProcessEvents();
 287
 288     Int_t status;
 289
 290     status = gSystem->Exec(TString::Format("xwd -id %u | convert - %s",
 291                            gEve->GetBrowser()->GetId(), pic.Data()));
 292
 293     printf("Post capture -- status=%d.\n", status);
 294
 295     status = gSystem->Exec(TString::Format("SendImageToAmore %s %s %d",
 296                           id.Data(), pic.Data(),
 297                           AliEveEventManager::AssertRawReader()->GetRunNumber()));
 298
 299     printf("Post AMORE reg -- status=%d, run=%d.\n", status,
 300            AliEveEventManager::AssertRawReader()->GetRunNumber());
 301
 302     if (++g_pic_id >= g_pic_max)
 303       g_pic_id = 0;
 304     g_pic_prev.Set();
 305   }
 306 }