[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 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 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
  //==============================================================================

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

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

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

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