[u/mrichter/AliRoot.git] / EVE / macros / visscan_init.C

// $Id$
// Main authors: Matevz Tadel & Alja Mrak-Tadel: 2006, 2007

/**************************************************************************
 * Copyright(c) 1998-2008, ALICE Experiment at CERN, all rights reserved. *
 * See http://aliceinfo.cern.ch/Offline/AliRoot/License.html for          *
 * full copyright notice.                                                 *
 **************************************************************************/

class AliEveMacroExecutor;
class MultiView;
class TEveProjectionManager;
class TEveGeoShape;
class TEveUtil;


Bool_t gShowTrd      = kTRUE;
Bool_t gShowMuonRPhi = kFALSE;
Bool_t gShowMuonRhoZ = kTRUE;

Bool_t gCenterProjectionsAtPrimaryVertex = kFALSE;


void visscan_init(const TString& cdburi = "",
		  const TString& path   = ".", Bool_t show_extra_geo = kFALSE)
{
  if (cdburi.IsNull() && ! AliCDBManager::Instance()->IsDefaultStorageSet())
  {
    gEnv->SetValue("Root.Stacktrace", "no");
    Fatal("visscan_init.C", "OCDB path MUST be specified as the first argument.");
  }

  if (gROOT->LoadMacro("MultiView.C+") != 0)
  {
    gEnv->SetValue("Root.Stacktrace", "no");
    Fatal("visscan_init.C", "Failed loading MultiView.C in compiled mode.");
  }

  if (!show_extra_geo)
  {
    gShowTrd = gShowMuonRPhi = gShowMuonRhoZ = kFALSE;
  }

  AliEveEventManager::AddAODfriend("AliAOD.VertexingHF.root");

  TEveUtil::LoadMacro("alieve_init.C");
  alieve_init(cdburi, path, -1);

  // TEveLine::SetDefaultSmooth(1);

  TEveUtil::AssertMacro("VizDB_scan.C");

  AliEveMacroExecutor *exec    = AliEveEventManager::GetMaster()->GetExecutor();
  TEveBrowser         *browser = gEve->GetBrowser();
  browser->ShowCloseTab(kFALSE);


  //==============================================================================
  // Geometry, scenes, projections and viewers
  //==============================================================================

  gMultiView = new MultiView;

  TEveUtil::LoadMacro("geom_gentle.C");
  gMultiView->InitGeomGentle(geom_gentle(),
                             geom_gentle_rphi(), 
                             geom_gentle_rhoz());

  if (gShowTrd) {
    TEveUtil::LoadMacro("geom_gentle_trd.C");
    gMultiView->InitGeomGentleTrd(geom_gentle_trd());
  }

  if (gShowMuonRPhi || gShowMuonRhoZ) {
    TEveUtil::LoadMacro("geom_gentle_muon.C");
    gMultiView->InitGeomGentleMuon(geom_gentle_muon(kFALSE), gShowMuonRPhi, gShowMuonRhoZ);
  }


  //==============================================================================
  // Registration of per-event macros
  //==============================================================================

  exec->AddMacro(new AliEveMacro(AliEveMacro::kRunLoader, "SIM Track",   "kine_tracks.C", "kine_tracks", "", kFALSE));

  exec->AddMacro(new AliEveMacro(AliEveMacro::kRunLoader, "SIM Hit ITS", "its_hits.C",    "its_hits",    "", kFALSE));
  exec->AddMacro(new AliEveMacro(AliEveMacro::kRunLoader, "SIM Hit TPC", "tpc_hits.C",    "tpc_hits",    "", kFALSE));
  exec->AddMacro(new AliEveMacro(AliEveMacro::kRunLoader, "SIM Hit T0",  "t0_hits.C",     "t0_hits",     "", kFALSE));
  exec->AddMacro(new AliEveMacro(AliEveMacro::kRunLoader, "SIM Hit FMD", "fmd_hits.C",    "fmd_hits",    "", kFALSE));

  exec->AddMacro(new AliEveMacro(AliEveMacro::kRunLoader, "DIG ITS",     "its_digits.C",  "its_digits",  "", kFALSE));
  exec->AddMacro(new AliEveMacro(AliEveMacro::kRunLoader, "DIG TPC",     "tpc_digits.C",  "tpc_digits",  "", kFALSE));
  exec->AddMacro(new AliEveMacro(AliEveMacro::kRunLoader, "DIG TOF",     "tof_digits.C",  "tof_digits",  "", kFALSE));
  exec->AddMacro(new AliEveMacro(AliEveMacro::kRunLoader, "DIG HMPID",   "hmpid_digits.C","hmpid_digits","", kFALSE));
  exec->AddMacro(new AliEveMacro(AliEveMacro::kRunLoader, "DIG FMD",     "fmd_digits.C",  "fmd_digits",  "", kFALSE));

  exec->AddMacro(new AliEveMacro(AliEveMacro::kRawReader, "RAW ITS",     "its_raw.C",     "its_raw",     "", kFALSE));
  exec->AddMacro(new AliEveMacro(AliEveMacro::kRawReader, "RAW TPC",     "tpc_raw.C",     "tpc_raw",     "", kFALSE));
  exec->AddMacro(new AliEveMacro(AliEveMacro::kRawReader, "RAW TOF",     "tof_raw.C",     "tof_raw",     "", kFALSE));
  exec->AddMacro(new AliEveMacro(AliEveMacro::kRawReader, "RAW HMPID",   "hmpid_raw.C",   "hmpid_raw",   "", kFALSE));
  exec->AddMacro(new AliEveMacro(AliEveMacro::kRawReader, "RAW T0",      "t0_raw.C",      "t0_raw",      "", kFALSE));
  exec->AddMacro(new AliEveMacro(AliEveMacro::kRawReader, "RAW FMD",     "fmd_raw.C",     "fmd_raw",     "", kFALSE));
  exec->AddMacro(new AliEveMacro(AliEveMacro::kRawReader, "RAW VZERO",   "vzero_raw.C",   "vzero_raw",   "", kFALSE));
  exec->AddMacro(new AliEveMacro(AliEveMacro::kRawReader, "RAW ACORDE",  "acorde_raw.C",  "acorde_raw",  "", kFALSE));

  exec->AddMacro(new AliEveMacro(AliEveMacro::kESD, "REC PVTX",         "primary_vertex.C", "primary_vertex",             "",                kTRUE));
  exec->AddMacro(new AliEveMacro(AliEveMacro::kESD, "REC PVTX Ellipse", "primary_vertex.C", "primary_vertex_ellipse",     "",                kTRUE));
  exec->AddMacro(new AliEveMacro(AliEveMacro::kESD, "REC PVTX Box",     "primary_vertex.C", "primary_vertex_box",         "kFALSE, 3, 3, 3", kFALSE));
  exec->AddMacro(new AliEveMacro(AliEveMacro::kESD, "REC PVTX",         "primary_vertex.C", "primary_vertex_spd",         "",                kTRUE));
  exec->AddMacro(new AliEveMacro(AliEveMacro::kESD, "REC PVTX Ellipse", "primary_vertex.C", "primary_vertex_ellipse_spd", "",                kTRUE));
  exec->AddMacro(new AliEveMacro(AliEveMacro::kESD, "REC PVTX Box",     "primary_vertex.C", "primary_vertex_box_spd",     "kFALSE, 3, 3, 3", kFALSE));
  exec->AddMacro(new AliEveMacro(AliEveMacro::kESD, "REC PVTX",         "primary_vertex.C", "primary_vertex_tpc",         "",                kFALSE));
  exec->AddMacro(new AliEveMacro(AliEveMacro::kESD, "REC PVTX Ellipse", "primary_vertex.C", "primary_vertex_ellipse_tpc", "",                kFALSE));
  exec->AddMacro(new AliEveMacro(AliEveMacro::kESD, "REC PVTX Box",     "primary_vertex.C", "primary_vertex_box_tpc",     "kFALSE, 3, 3, 3", kFALSE));

  exec->AddMacro(new AliEveMacro(AliEveMacro::kESD, "REC V0",   "esd_V0_points.C",       "esd_V0_points_onfly"));
  exec->AddMacro(new AliEveMacro(AliEveMacro::kESD, "REC V0",   "esd_V0_points.C",       "esd_V0_points_offline"));
  exec->AddMacro(new AliEveMacro(AliEveMacro::kESD, "REC V0",   "esd_V0.C",              "esd_V0"));
  exec->AddMacro(new AliEveMacro(AliEveMacro::kESD, "REC CSCD", "esd_cascade_points.C",  "esd_cascade_points"));
  exec->AddMacro(new AliEveMacro(AliEveMacro::kESD, "REC CSCD", "esd_cascade.C",         "esd_cascade"));
  exec->AddMacro(new AliEveMacro(AliEveMacro::kESD, "REC KINK", "esd_kink_points.C",     "esd_kink_points"));
  exec->AddMacro(new AliEveMacro(AliEveMacro::kESD, "REC KINK", "esd_kink.C",            "esd_kink"));

  exec->AddMacro(new AliEveMacro(AliEveMacro::kESD, "REC Track", "esd_tracks.C", "esd_tracks",             "", kFALSE));
  exec->AddMacro(new AliEveMacro(AliEveMacro::kESD, "REC Track", "esd_tracks.C", "esd_tracks_MI",          "", kFALSE));
  exec->AddMacro(new AliEveMacro(AliEveMacro::kESD, "REC Track", "esd_tracks.C", "esd_tracks_by_category", "", kTRUE));

  exec->AddMacro(new AliEveMacro(AliEveMacro::kESD, "REC Tracklet", "esd_spd_tracklets.C", "esd_spd_tracklets", "", kTRUE));

  exec->AddMacro(new AliEveMacro(AliEveMacro::kESD, "REC ZDC",      "esd_zdc.C", "esd_zdc", "", kFALSE));

  exec->AddMacro(new AliEveMacro(AliEveMacro::kRunLoader, "REC Clus",     "clusters.C+",     "clusters", "", kFALSE));
  exec->AddMacro(new AliEveMacro(AliEveMacro::kRunLoader, "REC Clus ITS", "its_clusters.C+", "its_clusters"));
  exec->AddMacro(new AliEveMacro(AliEveMacro::kRunLoader, "REC Clus TPC", "tpc_clusters.C+", "tpc_clusters"));
  exec->AddMacro(new AliEveMacro(AliEveMacro::kRunLoader, "REC Clus TRD", "trd_clusters.C+", "trd_clusters"));
  exec->AddMacro(new AliEveMacro(AliEveMacro::kRunLoader, "REC Clus TOF", "tof_clusters.C+", "tof_clusters"));

  exec->AddMacro(new AliEveMacro(AliEveMacro::kRunLoader, "REC Clus TPC", "vplot_tpc.C+",    "vplot_tpc", "", kFALSE));

  exec->AddMacro(new AliEveMacro(AliEveMacro::kAOD, "ANA HF",   "aod_HF.C",   "aod_HF",   "", kFALSE));
  exec->AddMacro(new AliEveMacro(AliEveMacro::kAOD, "ANA Jets", "jetplane.C", "jetplane", "", kFALSE));


  //==============================================================================
  // Additional GUI components
  //==============================================================================

  // Macro / data selection
  slot = TEveWindow::CreateWindowInTab(browser->GetTabRight());
  slot->StartEmbedding();
  AliEveMacroExecutorWindow* exewin = new AliEveMacroExecutorWindow(exec);
  slot->StopEmbedding("DataSelection");
  exewin->PopulateMacros();

  // Event selection tab
  slot = TEveWindow::CreateWindowInTab(browser->GetTabRight());
  slot->StartEmbedding();
  new AliEveEventSelectorWindow(gClient->GetRoot(), 600, 400, AliEveEventManager::GetMaster()->GetEventSelector());
  slot->StopEmbedding("Selections");

  // QA viewer
  slot = TEveWindow::CreateWindowInTab(browser->GetTabRight());
  slot->StartEmbedding();
  new AliQAHistViewer(gClient->GetRoot(), 600, 400, kTRUE);
  slot->StopEmbedding("QA histograms");

  browser->GetTabRight()->SetTab(1);

  browser->StartEmbedding(TRootBrowser::kBottom);
  new AliEveEventManagerWindow(AliEveEventManager::GetMaster());
  browser->StopEmbedding("EventCtrl");

  slot = TEveWindow::CreateWindowInTab(browser->GetTabRight());
  TEveWindowTab *store_tab = slot->MakeTab();
  store_tab->SetElementNameTitle("WindowStore",
    "Undocked windows whose previous container is not known\n"
    "are placed here when the main-frame is closed.");
  gEve->GetWindowManager()->SetDefaultContainer(store_tab);


  //==============================================================================
  // AliEve objects - global tools
  //==============================================================================

  AliEveTrackFitter* fitter = new AliEveTrackFitter();
  gEve->AddToListTree(fitter, 1);
  gEve->AddElement(fitter, gEve->GetEventScene());

  AliEveTrackCounter* g_trkcnt = new AliEveTrackCounter("Primary Counter");
  gEve->AddToListTree(g_trkcnt, kFALSE);


  //==============================================================================
  // Final stuff
  //==============================================================================

  // A refresh to show proper window.
  gEve->Redraw3D(kTRUE);
  gSystem->ProcessEvents();

  // Register command to call on each event.
  AliEveEventManager::GetMaster()->AddNewEventCommand("on_new_event();");
  AliEveEventManager::GetMaster()->GotoEvent(0);

  gEve->EditElement(g_trkcnt);

  gEve->Redraw3D(kTRUE);
}

/******************************************************************************/

void on_new_event()
{
  AliEveTrackCounter* g_trkcnt = AliEveTrackCounter::fgInstance;
  g_trkcnt->Reset();
  g_trkcnt->SetEventId(AliEveEventManager::GetMaster()->GetEventId());

  if (g_esd_tracks_by_category_container != 0)
  {
    TEveElementList* cont = g_esd_tracks_by_category_container;

    // Here we expect several TEveTrackList containers.
    // First two have reasonable primaries (sigma-to-prim-vertex < 5).
    // Others are almost certainly secondaries.
    Int_t count = 1;
    TEveElement::List_i i = cont->BeginChildren();
    while (i != cont->EndChildren())
    {
      TEveTrackList* l = dynamic_cast<TEveTrackList*>(*i);
      if (l != 0)
      {
	g_trkcnt->RegisterTracks(l, (count <= 2));
	++count;
      }
      ++i;
    }

    // Set it to zero, so that we do not reuse an old one.
    g_esd_tracks_by_category_container = 0;
  }
  else
  {
    Warning("on_new_event", "g_esd_tracks_by_category_container not initialized.");
  }

  TEveTrackList *tracklets = dynamic_cast<TEveTrackList*>(gEve->GetCurrentEvent()->FindChild("SPD Tracklets"));
  if (tracklets)
  {
    g_trkcnt->RegisterTracklets(tracklets, kTRUE);
  }
  else
  {
    Warning("on_new_event", "'SPD Tracklets' not found.");
  }

  Double_t x[3] = { 0, 0, 0 };

  if (AliEveEventManager::HasESD())
  {
    AliESDEvent* esd = AliEveEventManager::AssertESD();
    esd->GetPrimaryVertex()->GetXYZ(x);

    TTimeStamp ts(esd->GetTimeStamp());
    TString win_title("Eve Main Window -- Timestamp: ");
    win_title += ts.AsString("s");
    win_title += "; Event # in ESD file: ";
    win_title += esd->GetEventNumberInFile();
    gEve->GetBrowser()->SetWindowName(win_title);
  }

  TEveElement* top = gEve->GetCurrentEvent();

  gMultiView->DestroyEventRPhi();
  if (gCenterProjectionsAtPrimaryVertex)
    gMultiView->SetCenterRPhi(x[0], x[1], x[2]);
  gMultiView->ImportEventRPhi(top);

  gMultiView->DestroyEventRhoZ();
  if (gCenterProjectionsAtPrimaryVertex)
    gMultiView->SetCenterRhoZ(x[0], x[1], x[2]);
  gMultiView->ImportEventRhoZ(top);
}
Commit	Line	Data
20dae051	1	// $Id$
e9b9f7d2	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
20dae051	10	class AliEveMacroExecutor;
3545fba9	11	class MultiView;
e9b9f7d2	12	class TEveProjectionManager;
	13	class TEveGeoShape;
	14	class TEveUtil;
	15
20dae051	16
3545fba9	17	Bool_t gShowTrd = kTRUE;
	18	Bool_t gShowMuonRPhi = kFALSE;
	19	Bool_t gShowMuonRhoZ = kTRUE;
20dae051	20
20dae051	21	Bool_t gCenterProjectionsAtPrimaryVertex = kFALSE;
e9b9f7d2	22
c63f2997	23
	24	void visscan_init(const TString& cdburi = "",
	25	const TString& path = ".", Bool_t show_extra_geo = kFALSE)
e9b9f7d2	26	{
c63f2997	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
3545fba9	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
115b6655	39	if (!show_extra_geo)
115b6655	40	{
3545fba9	41	gShowTrd = gShowMuonRPhi = gShowMuonRhoZ = kFALSE;
115b6655	42	}
115b6655	43
f8d5a389	44	AliEveEventManager::AddAODfriend("AliAOD.VertexingHF.root");
f8d5a389	45
e9b9f7d2	46	TEveUtil::LoadMacro("alieve_init.C");
c63f2997	47	alieve_init(cdburi, path, -1);
e9b9f7d2	48
20dae051	49	// TEveLine::SetDefaultSmooth(1);
e9b9f7d2	50
20dae051	51	TEveUtil::AssertMacro("VizDB_scan.C");
	52
	53	AliEveMacroExecutor *exec = AliEveEventManager::GetMaster()->GetExecutor();
	54	TEveBrowser *browser = gEve->GetBrowser();
3545fba9	55	browser->ShowCloseTab(kFALSE);
3545fba9	56
20dae051	57
	58	//==============================================================================
	59	// Geometry, scenes, projections and viewers
	60	//==============================================================================
	61
3545fba9	62	gMultiView = new MultiView;
20dae051	63
f6afd0e1	64	TEveUtil::LoadMacro("geom_gentle.C");
3545fba9	65	gMultiView->InitGeomGentle(geom_gentle(),
	66	geom_gentle_rphi(),
	67	geom_gentle_rhoz());
	68
	69	if (gShowTrd) {
f6afd0e1	70	TEveUtil::LoadMacro("geom_gentle_trd.C");
3545fba9	71	gMultiView->InitGeomGentleTrd(geom_gentle_trd());
f6afd0e1	72	}
f6afd0e1	73
3545fba9	74	if (gShowMuonRPhi \|\| gShowMuonRhoZ) {
	75	TEveUtil::LoadMacro("geom_gentle_muon.C");
	76	gMultiView->InitGeomGentleMuon(geom_gentle_muon(kFALSE), gShowMuonRPhi, gShowMuonRhoZ);
68ca2fe7	77	}
20dae051	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
1eda7a67	91	exec->AddMacro(new AliEveMacro(AliEveMacro::kRunLoader, "DIG ITS", "its_digits.C", "its_digits", "", kFALSE));
391fa967	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));
20dae051	95	exec->AddMacro(new AliEveMacro(AliEveMacro::kRunLoader, "DIG FMD", "fmd_digits.C", "fmd_digits", "", kFALSE));
20dae051	96
1eda7a67	97	exec->AddMacro(new AliEveMacro(AliEveMacro::kRawReader, "RAW ITS", "its_raw.C", "its_raw", "", kFALSE));
20dae051	98	exec->AddMacro(new AliEveMacro(AliEveMacro::kRawReader, "RAW TPC", "tpc_raw.C", "tpc_raw", "", kFALSE));
391fa967	99	exec->AddMacro(new AliEveMacro(AliEveMacro::kRawReader, "RAW TOF", "tof_raw.C", "tof_raw", "", kFALSE));
391fa967	100	exec->AddMacro(new AliEveMacro(AliEveMacro::kRawReader, "RAW HMPID", "hmpid_raw.C", "hmpid_raw", "", kFALSE));
20dae051	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"));
4711ad43	121	exec->AddMacro(new AliEveMacro(AliEveMacro::kESD, "REC KINK", "esd_kink_points.C", "esd_kink_points"));
4711ad43	122	exec->AddMacro(new AliEveMacro(AliEveMacro::kESD, "REC KINK", "esd_kink.C", "esd_kink"));
20dae051	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
3a20f984	128	exec->AddMacro(new AliEveMacro(AliEveMacro::kESD, "REC Tracklet", "esd_spd_tracklets.C", "esd_spd_tracklets", "", kTRUE));
20dae051	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
08b0f222	138	exec->AddMacro(new AliEveMacro(AliEveMacro::kRunLoader, "REC Clus TPC", "vplot_tpc.C+", "vplot_tpc", "", kFALSE));
08b0f222	139
f8d5a389	140	exec->AddMacro(new AliEveMacro(AliEveMacro::kAOD, "ANA HF", "aod_HF.C", "aod_HF", "", kFALSE));
08b0f222	141	exec->AddMacro(new AliEveMacro(AliEveMacro::kAOD, "ANA Jets", "jetplane.C", "jetplane", "", kFALSE));
20dae051	142
3545fba9	143
20dae051	144	//==============================================================================
	145	// Additional GUI components
	146	//==============================================================================
	147
51158ea3	148	// Macro / data selection
20dae051	149	slot = TEveWindow::CreateWindowInTab(browser->GetTabRight());
	150	slot->StartEmbedding();
	151	AliEveMacroExecutorWindow* exewin = new AliEveMacroExecutorWindow(exec);
	152	slot->StopEmbedding("DataSelection");
	153	exewin->PopulateMacros();
	154
51158ea3	155	// Event selection tab
20dae051	156	slot = TEveWindow::CreateWindowInTab(browser->GetTabRight());
20dae051	157	slot->StartEmbedding();
51158ea3	158	new AliEveEventSelectorWindow(gClient->GetRoot(), 600, 400, AliEveEventManager::GetMaster()->GetEventSelector());
51158ea3	159	slot->StopEmbedding("Selections");
20dae051	160
51158ea3	161	// QA viewer
008ac94c	162	slot = TEveWindow::CreateWindowInTab(browser->GetTabRight());
008ac94c	163	slot->StartEmbedding();
51158ea3	164	new AliQAHistViewer(gClient->GetRoot(), 600, 400, kTRUE);
51158ea3	165	slot->StopEmbedding("QA histograms");
008ac94c	166
20dae051	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);
68ca2fe7	179
3545fba9	180
20dae051	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.
4d62585e	202	AliEveEventManager::GetMaster()->AddNewEventCommand("on_new_event();");
4d62585e	203	AliEveEventManager::GetMaster()->GotoEvent(0);
e9b9f7d2	204
ca49b003	205	gEve->EditElement(g_trkcnt);
ca49b003	206
e9b9f7d2	207	gEve->Redraw3D(kTRUE);
	208	}
	209
	210	/******************************************************************************/
	211
	212	void on_new_event()
	213	{
f76c9e9b	214	AliEveTrackCounter* g_trkcnt = AliEveTrackCounter::fgInstance;
e9b9f7d2	215	g_trkcnt->Reset();
4d62585e	216	g_trkcnt->SetEventId(AliEveEventManager::GetMaster()->GetEventId());
29207369	217
20dae051	218	if (g_esd_tracks_by_category_container != 0)
df56539f	219	{
20dae051	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())
29207369	228	{
20dae051	229	TEveTrackList* l = dynamic_cast<TEveTrackList>(i);
	230	if (l != 0)
	231	{
	232	g_trkcnt->RegisterTracks(l, (count <= 2));
	233	++count;
	234	}
	235	++i;
e9b9f7d2	236	}
20dae051	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.");
e9b9f7d2	244	}
e9b9f7d2	245
3a20f984	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
20dae051	256	Double_t x[3] = { 0, 0, 0 };
	257
	258	if (AliEveEventManager::HasESD())
a27140cb	259	{
20dae051	260	AliESDEvent* esd = AliEveEventManager::AssertESD();
	261	esd->GetPrimaryVertex()->GetXYZ(x);
	262
a27140cb	263	TTimeStamp ts(esd->GetTimeStamp());
	264	TString win_title("Eve Main Window -- Timestamp: ");
	265	win_title += ts.AsString("s");
5858bcbe	266	win_title += "; Event # in ESD file: ";
a27140cb	267	win_title += esd->GetEventNumberInFile();
	268	gEve->GetBrowser()->SetWindowName(win_title);
	269	}
e9b9f7d2	270
	271	TEveElement* top = gEve->GetCurrentEvent();
	272
3545fba9	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);
e9b9f7d2	282	}