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 TEveProjectionManager;
  12 class TEveGeoShape;
  13 class TEveUtil;
  14
  15
  16 Bool_t gShowTrd      = kTRUE;
  17 Bool_t gShowMuonRPhi = kFALSE;
  18 Bool_t gShowMuonRhoZ = kTRUE;
  19
  20 Bool_t gCenterProjectionsAtPrimaryVertex = kFALSE;
  21
  22
  23 void visscan_init(const TString& cdburi = "",
  24                   const TString& path   = ".", Bool_t show_extra_geo = kFALSE)
  25 {
  26   if (cdburi.IsNull() && ! AliCDBManager::Instance()->IsDefaultStorageSet())
  27   {
  28     gEnv->SetValue("Root.Stacktrace", "no");
  29     Fatal("visscan_init.C", "OCDB path MUST be specified as the first argument.");
  30   }
  31
  32   if (!show_extra_geo)
  33   {
  34     gShowTrd = gShowMuonRPhi = gShowMuonRhoZ = kFALSE;
  35   }
  36
  37   AliEveEventManager::AddAODfriend("AliAOD.VertexingHF.root");
  38
  39   TEveUtil::LoadMacro("alieve_init.C");
  40   alieve_init(cdburi, path, -1);
  41
  42   // TEveLine::SetDefaultSmooth(1);
  43
  44   TEveUtil::AssertMacro("VizDB_scan.C");
  45
  46   AliEveMacroExecutor *exec    = AliEveEventManager::GetMaster()->GetExecutor();
  47   TEveBrowser         *browser = gEve->GetBrowser();
  48   browser->ShowCloseTab(kFALSE);
  49
  50
  51   //==============================================================================
  52   // Geometry, scenes, projections and viewers
  53   //==============================================================================
  54
  55   AliEveMultiView *mv = new AliEveMultiView;
  56
  57   mv->SetDepth(-10);
  58
  59   TEveUtil::LoadMacro("geom_gentle.C");
  60   mv->InitGeomGentle(geom_gentle(), geom_gentle_rphi(), geom_gentle_rhoz());
  61
  62   if (gShowTrd) {
  63     TEveUtil::LoadMacro("geom_gentle_trd.C");
  64     mv->InitGeomGentleTrd(geom_gentle_trd());
  65   }
  66
  67   if (gShowMuonRPhi || gShowMuonRhoZ) {
  68     TEveUtil::LoadMacro("geom_gentle_muon.C");
  69     mv->InitGeomGentleMuon(geom_gentle_muon(kFALSE), gShowMuonRPhi, gShowMuonRhoZ);
  70   }
  71
  72   mv->SetDepth(0);
  73
  74   //==============================================================================
  75   // Registration of per-event macros
  76   //==============================================================================
  77
  78   exec->AddMacro(new AliEveMacro(AliEveMacro::kRunLoader, "SIM Track",   "kine_tracks.C+", "kine_tracks", "", kFALSE));
  79
  80   exec->AddMacro(new AliEveMacro(AliEveMacro::kRunLoader, "SIM Hits ITS", "its_hits.C",    "its_hits",    "", kFALSE));
  81   exec->AddMacro(new AliEveMacro(AliEveMacro::kRunLoader, "SIM Hits TPC", "tpc_hits.C",    "tpc_hits",    "", kFALSE));
  82   exec->AddMacro(new AliEveMacro(AliEveMacro::kRunLoader, "SIM Hits T0",  "t0_hits.C",     "t0_hits",     "", kFALSE));
  83   exec->AddMacro(new AliEveMacro(AliEveMacro::kRunLoader, "SIM Hits FMD", "fmd_hits.C",    "fmd_hits",    "", kFALSE));
  84   //  exec->AddMacro(new AliEveMacro(AliEveMacro::kRunLoader, "SIM Hits ACORDE", "acorde_hits.C",    "acorde_hits",    "", kFALSE));
  85   //  exec->AddMacro(new AliEveMacro(AliEveMacro::kRunLoader, "SIM Hits EMCAL", "emcal_hits.C",    "emcal_hits",    "", kFALSE));
  86   //  exec->AddMacro(new AliEveMacro(AliEveMacro::kRunLoader, "SIM Hits TOF",  "tof_hits.C",     "tof_hits",     "", kFALSE));
  87   //  exec->AddMacro(new AliEveMacro(AliEveMacro::kRunLoader, "SIM Hits TRD", "trd_hits.C",    "trd_hits",    "", kFALSE));
  88   //  exec->AddMacro(new AliEveMacro(AliEveMacro::kRunLoader, "SIM Hits VZERO", "vzero_hits.C",    "vzero_hits",    "", kFALSE));
  89
  90   exec->AddMacro(new AliEveMacro(AliEveMacro::kRunLoader, "DIG ITS",     "its_digits.C",  "its_digits",  "", kFALSE));
  91   exec->AddMacro(new AliEveMacro(AliEveMacro::kRunLoader, "DIG TPC",     "tpc_digits.C",  "tpc_digits",  "", kFALSE));
  92   exec->AddMacro(new AliEveMacro(AliEveMacro::kRunLoader, "DIG TOF",     "tof_digits.C",  "tof_digits",  "", kFALSE));
  93   exec->AddMacro(new AliEveMacro(AliEveMacro::kRunLoader, "DIG HMPID",   "hmpid_digits.C","hmpid_digits","", kFALSE));
  94   exec->AddMacro(new AliEveMacro(AliEveMacro::kRunLoader, "DIG FMD",     "fmd_digits.C",  "fmd_digits",  "", kFALSE));
  95
  96   exec->AddMacro(new AliEveMacro(AliEveMacro::kRawReader, "RAW ITS",     "its_raw.C",     "its_raw",     "", kFALSE));
  97   exec->AddMacro(new AliEveMacro(AliEveMacro::kRawReader, "RAW TPC",     "tpc_raw.C",     "tpc_raw",     "", kFALSE));
  98   exec->AddMacro(new AliEveMacro(AliEveMacro::kRawReader, "RAW TOF",     "tof_raw.C",     "tof_raw",     "", kFALSE));
  99   exec->AddMacro(new AliEveMacro(AliEveMacro::kRawReader, "RAW HMPID",   "hmpid_raw.C",   "hmpid_raw",   "", kFALSE));
 100   exec->AddMacro(new AliEveMacro(AliEveMacro::kRawReader, "RAW T0",      "t0_raw.C",      "t0_raw",      "", kFALSE));
 101   exec->AddMacro(new AliEveMacro(AliEveMacro::kRawReader, "RAW FMD",     "fmd_raw.C",     "fmd_raw",     "", kFALSE));
 102   exec->AddMacro(new AliEveMacro(AliEveMacro::kRawReader, "RAW VZERO",   "vzero_raw.C",   "vzero_raw",   "", kFALSE));
 103   exec->AddMacro(new AliEveMacro(AliEveMacro::kRawReader, "RAW ACORDE",  "acorde_raw.C",  "acorde_raw",  "", kFALSE));
 104
 105   exec->AddMacro(new AliEveMacro(AliEveMacro::kESD, "REC PVTX",             "primary_vertex.C+", "primary_vertex",             "",                kTRUE));
 106   exec->AddMacro(new AliEveMacro(AliEveMacro::kESD, "REC PVTX Ellipse",     "primary_vertex.C+", "primary_vertex_ellipse",     "",                kTRUE));
 107   exec->AddMacro(new AliEveMacro(AliEveMacro::kESD, "REC PVTX Box",         "primary_vertex.C+", "primary_vertex_box",         "kFALSE, 3, 3, 3", kFALSE));
 108   exec->AddMacro(new AliEveMacro(AliEveMacro::kESD, "REC PVTX SPD",         "primary_vertex.C+", "primary_vertex_spd",         "",                kTRUE));
 109   exec->AddMacro(new AliEveMacro(AliEveMacro::kESD, "REC PVTX Ellipse SPD", "primary_vertex.C+", "primary_vertex_ellipse_spd", "",                kTRUE));
 110   exec->AddMacro(new AliEveMacro(AliEveMacro::kESD, "REC PVTX Box SPD",     "primary_vertex.C+", "primary_vertex_box_spd",     "kFALSE, 3, 3, 3", kFALSE));
 111   exec->AddMacro(new AliEveMacro(AliEveMacro::kESD, "REC PVTX TPC",         "primary_vertex.C+", "primary_vertex_tpc",         "",                kFALSE));
 112   exec->AddMacro(new AliEveMacro(AliEveMacro::kESD, "REC PVTX Ellipse TPC", "primary_vertex.C+", "primary_vertex_ellipse_tpc", "",                kFALSE));
 113   exec->AddMacro(new AliEveMacro(AliEveMacro::kESD, "REC PVTX Box TPC",     "primary_vertex.C+", "primary_vertex_box_tpc",     "kFALSE, 3, 3, 3", kFALSE));
 114
 115   exec->AddMacro(new AliEveMacro(AliEveMacro::kESD, "REC V0",   "esd_V0_points.C",       "esd_V0_points_onfly"));
 116   exec->AddMacro(new AliEveMacro(AliEveMacro::kESD, "REC V0",   "esd_V0_points.C",       "esd_V0_points_offline"));
 117   exec->AddMacro(new AliEveMacro(AliEveMacro::kESD, "REC V0",   "esd_V0.C",              "esd_V0"));
 118   exec->AddMacro(new AliEveMacro(AliEveMacro::kESD, "REC CSCD", "esd_cascade_points.C",  "esd_cascade_points"));
 119   exec->AddMacro(new AliEveMacro(AliEveMacro::kESD, "REC CSCD", "esd_cascade.C",         "esd_cascade"));
 120   exec->AddMacro(new AliEveMacro(AliEveMacro::kESD, "REC KINK", "esd_kink_points.C",     "esd_kink_points"));
 121   exec->AddMacro(new AliEveMacro(AliEveMacro::kESD, "REC KINK", "esd_kink.C",            "esd_kink"));
 122
 123   exec->AddMacro(new AliEveMacro(AliEveMacro::kESD, "REC Tracks",              "esd_tracks.C+", "esd_tracks",              "", kFALSE));
 124   exec->AddMacro(new AliEveMacro(AliEveMacro::kESD, "REC Tracks ITS standalone",          "esd_tracks.C+", "esd_tracks_ITS_standalone",              "", kFALSE));
 125   exec->AddMacro(new AliEveMacro(AliEveMacro::kESD, "REC Tracks ITS",          "esd_tracks.C+", "esd_tracks_ITS",              "", kFALSE));
 126   exec->AddMacro(new AliEveMacro(AliEveMacro::kESD, "REC Tracks TPC",           "esd_tracks.C+", "esd_tracks_TPC",              "", kFALSE));
 127   exec->AddMacro(new AliEveMacro(AliEveMacro::kESD, "REC Tracks MI",           "esd_tracks.C+", "esd_tracks_MI",           "", kFALSE));
 128   exec->AddMacro(new AliEveMacro(AliEveMacro::kESD, "REC Tracks by category",  "esd_tracks.C+", "esd_tracks_by_category",  "", kTRUE));
 129   exec->AddMacro(new AliEveMacro(AliEveMacro::kESD, "REC Tracks by anal cuts", "esd_tracks.C+", "esd_tracks_by_anal_cuts", "", kFALSE));
 130
 131   exec->AddMacro(new AliEveMacro(AliEveMacro::kESD, "REC Tracklets SPD", "esd_spd_tracklets.C", "esd_spd_tracklets", "", kTRUE));
 132
 133   exec->AddMacro(new AliEveMacro(AliEveMacro::kESD, "REC ZDC",      "esd_zdc.C", "esd_zdc", "", kFALSE));
 134
 135   exec->AddMacro(new AliEveMacro(AliEveMacro::kRunLoader, "REC Clusters",     "clusters.C+",     "clusters", "", kFALSE));
 136   exec->AddMacro(new AliEveMacro(AliEveMacro::kRunLoader, "REC Clusters ITS", "its_clusters.C+", "its_clusters"));
 137   exec->AddMacro(new AliEveMacro(AliEveMacro::kRunLoader, "REC Clusters TPC", "tpc_clusters.C+", "tpc_clusters"));
 138   exec->AddMacro(new AliEveMacro(AliEveMacro::kRunLoader, "REC Clusters TRD", "trd_clusters.C+", "trd_clusters"));
 139   exec->AddMacro(new AliEveMacro(AliEveMacro::kRunLoader, "REC Clusters TOF", "tof_clusters.C+", "tof_clusters"));
 140   //  exec->AddMacro(new AliEveMacro(AliEveMacro::kRunLoader, "REC Clusters HMPID", "hmpid_clusters.C+", "hmpid_clusters"));
 141   //  exec->AddMacro(new AliEveMacro(AliEveMacro::kRunLoader, "REC Clusters PHOS", "phos_clusters.C+", "phos_clusters"));
 142
 143   exec->AddMacro(new AliEveMacro(AliEveMacro::kRunLoader, "REC Clusters TPC", "vplot_tpc.C+",    "vplot_tpc", "", kFALSE));
 144
 145   exec->AddMacro(new AliEveMacro(AliEveMacro::kAOD, "ANA HF",   "aod_HF.C",   "aod_HF",   "", kFALSE));
 146   exec->AddMacro(new AliEveMacro(AliEveMacro::kAOD, "ANA Jets", "jetplane.C", "jetplane", "", kFALSE));
 147
 148   exec->AddMacro(new AliEveMacro(AliEveMacro::kESD, "DUMP VZERO",   "vzero_dump.C",   "vzero_dump",   "", kFALSE));
 149
 150   //==============================================================================
 151   // Additional GUI components
 152   //==============================================================================
 153
 154   // Macro / data selection
 155   slot = TEveWindow::CreateWindowInTab(browser->GetTabRight());
 156   slot->StartEmbedding();
 157   AliEveMacroExecutorWindow* exewin = new AliEveMacroExecutorWindow(exec);
 158   slot->StopEmbedding("DataSelection");
 159   exewin->PopulateMacros();
 160
 161   // Event selection tab
 162   slot = TEveWindow::CreateWindowInTab(browser->GetTabRight());
 163   slot->StartEmbedding();
 164   new AliEveEventSelectorWindow(gClient->GetRoot(), 600, 400, AliEveEventManager::GetMaster()->GetEventSelector());
 165   slot->StopEmbedding("Selections");
 166
 167   // QA viewer
 168   slot = TEveWindow::CreateWindowInTab(browser->GetTabRight());
 169   slot->StartEmbedding();
 170   new AliQAHistViewer(gClient->GetRoot(), 600, 400, kTRUE);
 171   slot->StopEmbedding("QA histograms");
 172
 173   browser->GetTabRight()->SetTab(1);
 174
 175   browser->StartEmbedding(TRootBrowser::kBottom);
 176   new AliEveEventManagerWindow(AliEveEventManager::GetMaster());
 177   browser->StopEmbedding("EventCtrl");
 178
 179   slot = TEveWindow::CreateWindowInTab(browser->GetTabRight());
 180   TEveWindowTab *store_tab = slot->MakeTab();
 181   store_tab->SetElementNameTitle("WindowStore",
 182     "Undocked windows whose previous container is not known\n"
 183     "are placed here when the main-frame is closed.");
 184   gEve->GetWindowManager()->SetDefaultContainer(store_tab);
 185
 186
 187   //==============================================================================
 188   // AliEve objects - global tools
 189   //==============================================================================
 190
 191   AliEveTrackFitter* fitter = new AliEveTrackFitter();
 192   gEve->AddToListTree(fitter, 1);
 193   gEve->AddElement(fitter, gEve->GetEventScene());
 194
 195   AliEveTrackCounter* g_trkcnt = new AliEveTrackCounter("Primary Counter");
 196   gEve->AddToListTree(g_trkcnt, kFALSE);
 197
 198
 199   //==============================================================================
 200   // Final stuff
 201   //==============================================================================
 202
 203   // A refresh to show proper window.
 204   gEve->GetViewers()->SwitchColorSet();
 205   gEve->Redraw3D(kTRUE);
 206   gSystem->ProcessEvents();
 207
 208   // Register command to call on each event.
 209   AliEveEventManager::GetMaster()->AddNewEventCommand("on_new_event();");
 210   AliEveEventManager::GetMaster()->GotoEvent(0);
 211
 212   gEve->EditElement(g_trkcnt);
 213
 214   gEve->Redraw3D(kTRUE);
 215 }
 216
 217 /******************************************************************************/
 218
 219 void on_new_event()
 220 {
 221   Double_t x[3] = { 0, 0, 0 };
 222
 223   if (AliEveEventManager::HasESD())
 224   {
 225     AliESDEvent* esd = AliEveEventManager::AssertESD();
 226     esd->GetPrimaryVertex()->GetXYZ(x);
 227
 228     TTimeStamp ts(esd->GetTimeStamp());
 229     TString win_title("Eve Main Window -- Timestamp: ");
 230     win_title += ts.AsString("s");
 231     win_title += "; Event # in ESD file: ";
 232     win_title += esd->GetEventNumberInFile();
 233     gEve->GetBrowser()->SetWindowName(win_title);
 234   }
 235
 236   TEveElement* top = gEve->GetCurrentEvent();
 237
 238   AliEveMultiView *mv = AliEveMultiView::Instance();
 239
 240   mv->DestroyEventRPhi();
 241   if (gCenterProjectionsAtPrimaryVertex)
 242     mv->SetCenterRPhi(x[0], x[1], x[2]);
 243   mv->ImportEventRPhi(top);
 244
 245   mv->DestroyEventRhoZ();
 246   if (gCenterProjectionsAtPrimaryVertex)
 247     mv->SetCenterRhoZ(x[0], x[1], x[2]);
 248   mv->ImportEventRhoZ(top);
 249 }