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 TPC",     "tpc_digits.C",  "tpc_digits",  "", kFALSE));
  93   exec->AddMacro(new AliEveMacro(AliEveMacro::kRunLoader, "DIG TOF",     "tof_digits.C",  "tof_digits",  "", kFALSE));
  94   exec->AddMacro(new AliEveMacro(AliEveMacro::kRunLoader, "DIG HMPID",   "hmpid_digits.C","hmpid_digits","", kFALSE));
  95   exec->AddMacro(new AliEveMacro(AliEveMacro::kRunLoader, "DIG FMD",     "fmd_digits.C",  "fmd_digits",  "", kFALSE));
  96
  97   exec->AddMacro(new AliEveMacro(AliEveMacro::kRawReader, "RAW ITS",     "its_raw.C",     "its_raw",     "", kFALSE));
  98   exec->AddMacro(new AliEveMacro(AliEveMacro::kRawReader, "RAW TPC",     "tpc_raw.C",     "tpc_raw",     "", kFALSE));
  99   exec->AddMacro(new AliEveMacro(AliEveMacro::kRawReader, "RAW TOF",     "tof_raw.C",     "tof_raw",     "", kFALSE));
 100   exec->AddMacro(new AliEveMacro(AliEveMacro::kRawReader, "RAW HMPID",   "hmpid_raw.C",   "hmpid_raw",   "", kFALSE));
 101   exec->AddMacro(new AliEveMacro(AliEveMacro::kRawReader, "RAW T0",      "t0_raw.C",      "t0_raw",      "", kFALSE));
 102   exec->AddMacro(new AliEveMacro(AliEveMacro::kRawReader, "RAW FMD",     "fmd_raw.C",     "fmd_raw",     "", kFALSE));
 103   exec->AddMacro(new AliEveMacro(AliEveMacro::kRawReader, "RAW VZERO",   "vzero_raw.C",   "vzero_raw",   "", kFALSE));
 104   exec->AddMacro(new AliEveMacro(AliEveMacro::kRawReader, "RAW ACORDE",  "acorde_raw.C",  "acorde_raw",  "", kFALSE));
 105
 106   exec->AddMacro(new AliEveMacro(AliEveMacro::kESD, "REC PVTX",         "primary_vertex.C", "primary_vertex",             "",                kTRUE));
 107   exec->AddMacro(new AliEveMacro(AliEveMacro::kESD, "REC PVTX Ellipse", "primary_vertex.C", "primary_vertex_ellipse",     "",                kTRUE));
 108   exec->AddMacro(new AliEveMacro(AliEveMacro::kESD, "REC PVTX Box",     "primary_vertex.C", "primary_vertex_box",         "kFALSE, 3, 3, 3", kFALSE));
 109   exec->AddMacro(new AliEveMacro(AliEveMacro::kESD, "REC PVTX",         "primary_vertex.C", "primary_vertex_spd",         "",                kTRUE));
 110   exec->AddMacro(new AliEveMacro(AliEveMacro::kESD, "REC PVTX Ellipse", "primary_vertex.C", "primary_vertex_ellipse_spd", "",                kTRUE));
 111   exec->AddMacro(new AliEveMacro(AliEveMacro::kESD, "REC PVTX Box",     "primary_vertex.C", "primary_vertex_box_spd",     "kFALSE, 3, 3, 3", kFALSE));
 112   exec->AddMacro(new AliEveMacro(AliEveMacro::kESD, "REC PVTX",         "primary_vertex.C", "primary_vertex_tpc",         "",                kFALSE));
 113   exec->AddMacro(new AliEveMacro(AliEveMacro::kESD, "REC PVTX Ellipse", "primary_vertex.C", "primary_vertex_ellipse_tpc", "",                kFALSE));
 114   exec->AddMacro(new AliEveMacro(AliEveMacro::kESD, "REC PVTX Box",     "primary_vertex.C", "primary_vertex_box_tpc",     "kFALSE, 3, 3, 3", kFALSE));
 115
 116   exec->AddMacro(new AliEveMacro(AliEveMacro::kESD, "REC V0",   "esd_V0_points.C",       "esd_V0_points_onfly"));
 117   exec->AddMacro(new AliEveMacro(AliEveMacro::kESD, "REC V0",   "esd_V0_points.C",       "esd_V0_points_offline"));
 118   exec->AddMacro(new AliEveMacro(AliEveMacro::kESD, "REC V0",   "esd_V0.C",              "esd_V0"));
 119   exec->AddMacro(new AliEveMacro(AliEveMacro::kESD, "REC CSCD", "esd_cascade_points.C",  "esd_cascade_points"));
 120   exec->AddMacro(new AliEveMacro(AliEveMacro::kESD, "REC CSCD", "esd_cascade.C",         "esd_cascade"));
 121   exec->AddMacro(new AliEveMacro(AliEveMacro::kESD, "REC KINK", "esd_kink_points.C",     "esd_kink_points"));
 122   exec->AddMacro(new AliEveMacro(AliEveMacro::kESD, "REC KINK", "esd_kink.C",            "esd_kink"));
 123
 124   exec->AddMacro(new AliEveMacro(AliEveMacro::kESD, "REC Track", "esd_tracks.C", "esd_tracks",             "", kFALSE));
 125   exec->AddMacro(new AliEveMacro(AliEveMacro::kESD, "REC Track", "esd_tracks.C", "esd_tracks_MI",          "", kFALSE));
 126   exec->AddMacro(new AliEveMacro(AliEveMacro::kESD, "REC Track", "esd_tracks.C", "esd_tracks_by_category", "", kTRUE));
 127
 128   exec->AddMacro(new AliEveMacro(AliEveMacro::kESD, "REC Tracklet", "esd_spd_tracklets.C", "esd_spd_tracklets", "", kTRUE));
 129
 130   exec->AddMacro(new AliEveMacro(AliEveMacro::kESD, "REC ZDC",      "esd_zdc.C", "esd_zdc", "", kFALSE));
 131
 132   exec->AddMacro(new AliEveMacro(AliEveMacro::kRunLoader, "REC Clus",     "clusters.C+",     "clusters", "", kFALSE));
 133   exec->AddMacro(new AliEveMacro(AliEveMacro::kRunLoader, "REC Clus ITS", "its_clusters.C+", "its_clusters"));
 134   exec->AddMacro(new AliEveMacro(AliEveMacro::kRunLoader, "REC Clus TPC", "tpc_clusters.C+", "tpc_clusters"));
 135   exec->AddMacro(new AliEveMacro(AliEveMacro::kRunLoader, "REC Clus TRD", "trd_clusters.C+", "trd_clusters"));
 136   exec->AddMacro(new AliEveMacro(AliEveMacro::kRunLoader, "REC Clus TOF", "tof_clusters.C+", "tof_clusters"));
 137
 138   exec->AddMacro(new AliEveMacro(AliEveMacro::kRunLoader, "REC Clus TPC", "vplot_tpc.C+",    "vplot_tpc", "", kFALSE));
 139
 140   exec->AddMacro(new AliEveMacro(AliEveMacro::kAOD, "ANA HF",   "aod_HF.C",   "aod_HF",   "", kFALSE));
 141   exec->AddMacro(new AliEveMacro(AliEveMacro::kAOD, "ANA Jets", "jetplane.C", "jetplane", "", kFALSE));
 142
 143
 144   //==============================================================================
 145   // Additional GUI components
 146   //==============================================================================
 147
 148   // Macro / data selection
 149   slot = TEveWindow::CreateWindowInTab(browser->GetTabRight());
 150   slot->StartEmbedding();
 151   AliEveMacroExecutorWindow* exewin = new AliEveMacroExecutorWindow(exec);
 152   slot->StopEmbedding("DataSelection");
 153   exewin->PopulateMacros();
 154
 155   // Event selection tab
 156   slot = TEveWindow::CreateWindowInTab(browser->GetTabRight());
 157   slot->StartEmbedding();
 158   new AliEveEventSelectorWindow(gClient->GetRoot(), 600, 400, AliEveEventManager::GetMaster()->GetEventSelector());
 159   slot->StopEmbedding("Selections");
 160
 161   // QA viewer
 162   slot = TEveWindow::CreateWindowInTab(browser->GetTabRight());
 163   slot->StartEmbedding();
 164   new AliQAHistViewer(gClient->GetRoot(), 600, 400, kTRUE);
 165   slot->StopEmbedding("QA histograms");
 166
 167   browser->GetTabRight()->SetTab(1);
 168
 169   browser->StartEmbedding(TRootBrowser::kBottom);
 170   new AliEveEventManagerWindow(AliEveEventManager::GetMaster());
 171   browser->StopEmbedding("EventCtrl");
 172
 173   slot = TEveWindow::CreateWindowInTab(browser->GetTabRight());
 174   TEveWindowTab *store_tab = slot->MakeTab();
 175   store_tab->SetElementNameTitle("WindowStore",
 176     "Undocked windows whose previous container is not known\n"
 177     "are placed here when the main-frame is closed.");
 178   gEve->GetWindowManager()->SetDefaultContainer(store_tab);
 179
 180
 181   //==============================================================================
 182   // AliEve objects - global tools
 183   //==============================================================================
 184
 185   AliEveTrackFitter* fitter = new AliEveTrackFitter();
 186   gEve->AddToListTree(fitter, 1);
 187   gEve->AddElement(fitter, gEve->GetEventScene());
 188
 189   AliEveTrackCounter* g_trkcnt = new AliEveTrackCounter("Primary Counter");
 190   gEve->AddToListTree(g_trkcnt, kFALSE);
 191
 192
 193   //==============================================================================
 194   // Final stuff
 195   //==============================================================================
 196
 197   // A refresh to show proper window.
 198   gEve->Redraw3D(kTRUE);
 199   gSystem->ProcessEvents();
 200
 201   // Register command to call on each event.
 202   AliEveEventManager::GetMaster()->AddNewEventCommand("on_new_event();");
 203   AliEveEventManager::GetMaster()->GotoEvent(0);
 204
 205   gEve->EditElement(g_trkcnt);
 206
 207   gEve->Redraw3D(kTRUE);
 208 }
 209
 210 /******************************************************************************/
 211
 212 void on_new_event()
 213 {
 214   AliEveTrackCounter* g_trkcnt = AliEveTrackCounter::fgInstance;
 215   g_trkcnt->Reset();
 216   g_trkcnt->SetEventId(AliEveEventManager::GetMaster()->GetEventId());
 217
 218   if (g_esd_tracks_by_category_container != 0)
 219   {
 220     TEveElementList* cont = g_esd_tracks_by_category_container;
 221
 222     // Here we expect several TEveTrackList containers.
 223     // First two have reasonable primaries (sigma-to-prim-vertex < 5).
 224     // Others are almost certainly secondaries.
 225     Int_t count = 1;
 226     TEveElement::List_i i = cont->BeginChildren();
 227     while (i != cont->EndChildren())
 228     {
 229       TEveTrackList* l = dynamic_cast<TEveTrackList*>(*i);
 230       if (l != 0)
 231       {
 232         g_trkcnt->RegisterTracks(l, (count <= 2));
 233         ++count;
 234       }
 235       ++i;
 236     }
 237
 238     // Set it to zero, so that we do not reuse an old one.
 239     g_esd_tracks_by_category_container = 0;
 240   }
 241   else
 242   {
 243     Warning("on_new_event", "g_esd_tracks_by_category_container not initialized.");
 244   }
 245
 246   TEveTrackList *tracklets = dynamic_cast<TEveTrackList*>(gEve->GetCurrentEvent()->FindChild("SPD Tracklets"));
 247   if (tracklets)
 248   {
 249     g_trkcnt->RegisterTracklets(tracklets, kTRUE);
 250   }
 251   else
 252   {
 253     Warning("on_new_event", "'SPD Tracklets' not found.");
 254   }
 255
 256   Double_t x[3] = { 0, 0, 0 };
 257
 258   if (AliEveEventManager::HasESD())
 259   {
 260     AliESDEvent* esd = AliEveEventManager::AssertESD();
 261     esd->GetPrimaryVertex()->GetXYZ(x);
 262
 263     TTimeStamp ts(esd->GetTimeStamp());
 264     TString win_title("Eve Main Window -- Timestamp: ");
 265     win_title += ts.AsString("s");
 266     win_title += "; Event # in ESD file: ";
 267     win_title += esd->GetEventNumberInFile();
 268     gEve->GetBrowser()->SetWindowName(win_title);
 269   }
 270
 271   TEveElement* top = gEve->GetCurrentEvent();
 272
 273   gMultiView->DestroyEventRPhi();
 274   if (gCenterProjectionsAtPrimaryVertex)
 275     gMultiView->SetCenterRPhi(x[0], x[1], x[2]);
 276   gMultiView->ImportEventRPhi(top);
 277
 278   gMultiView->DestroyEventRhoZ();
 279   if (gCenterProjectionsAtPrimaryVertex)
 280     gMultiView->SetCenterRhoZ(x[0], x[1], x[2]);
 281   gMultiView->ImportEventRhoZ(top);
 282 }