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

// $Id: NLT_trackcount_init.C 24927 2008-04-04 13:46:04Z mtadel $
// 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 TEveProjectionManager;
class TEveGeoShape;
class TEveUtil;

R__EXTERN TEveProjectionManager *gRPhiMgr;
R__EXTERN TEveProjectionManager *gRhoZMgr;

TEveGeoShape *gGeomGentle     = 0;
TEveGeoShape *gGeomGentleRPhi = 0;
TEveGeoShape *gGeomGentleRhoZ = 0;
TEveGeoShape *gGeomGentleTRD  = 0;

Bool_t gShowTRD = kFALSE;

void anyscan_init()
{
  TEveUtil::LoadMacro("alieve_init.C");
  alieve_init(".", -1);

  // TEveLine::SetDefaultSmooth(1);

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


  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);


  gROOT->ProcessLine(".L SplitGLView.C+");
  TEveBrowser* browser = gEve->GetBrowser();
  browser->ExecPlugin("SplitGLView", 0, "new SplitGLView(gClient->GetRoot(), 600, 450, kTRUE)");

  if (gRPhiMgr) {
    TEveProjectionAxes* a = new TEveProjectionAxes(gRPhiMgr);
    a->SetNumTickMarks(3);
    a->SetText("R-Phi");
    a->SetFontFile("comicbd");
    a->SetFontSize(10);
    gEve->GetScenes()->FindChild("R-Phi Projection")->AddElement(a);
  }
  if (gRhoZMgr) {
    TEveProjectionAxes* a = new TEveProjectionAxes(gRhoZMgr);
    a->SetNumTickMarks(3);
    a->SetText("Rho-Z");
    a->SetFontFile("comicbd");
    a->SetFontSize(10);
    gEve->GetScenes()->FindChild("Rho-Z Projection")->AddElement(a);
  }

  // geometry
  TEveUtil::LoadMacro("geom_gentle.C");
  gGeomGentle = geom_gentle();
  gGeomGentleRPhi = geom_gentle_rphi(); gGeomGentleRPhi->IncDenyDestroy();
  gGeomGentleRhoZ = geom_gentle_rhoz(); gGeomGentleRhoZ->IncDenyDestroy();
  if (gShowTRD) {
    TEveUtil::LoadMacro("geom_gentle_trd.C");
    gGeomGentleTRD = geom_gentle_trd();
  }


  AliEveMacroExecutor *exec = AliEveEventManager::GetMaster()->GetExecutor();

  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::kRawReader, "RAW TPC",     "tpc_raw.C",     "tpc_raw",     "", kFALSE));
  exec->AddMacro(new AliEveMacro(AliEveMacro::kRawReader, "RAW FMD",     "fmd_raw.C",     "fmd_raw",     "", kFALSE));
  exec->AddMacro(new AliEveMacro(AliEveMacro::kRunLoader, "DIG FMD",     "fmd_digits.C",  "fmd_digits",  "", kFALSE));
  exec->AddMacro(new AliEveMacro(AliEveMacro::kRunLoader, "SIM Hit FMD", "fmd_hits.C",    "fmd_hits",    "", kFALSE));
  exec->AddMacro(new AliEveMacro(AliEveMacro::kRawReader, "RAW VZERO",   "vzero_raw.C",   "vzero_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"));
  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 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", "", 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"));

  TEveBrowser* browser = gEve->GetBrowser();

  browser->StartEmbedding(TRootBrowser::kRight);
  AliEveMacroExecutorWindow* exewin = new AliEveMacroExecutorWindow(exec);
  browser->StopEmbedding("DataSelection");
  exewin->PopulateMacros();

  browser->StartEmbedding();
  new AliQAHistViewer(gClient->GetRoot(), 600, 400, kTRUE);
  browser->StopEmbedding("QA histograms");

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

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

  // 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.");
  }

  AliESDEvent* esd = AliEveEventManager::AssertESD();
  {
    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);
  }
  Double_t x[3];
  esd->GetPrimaryVertex()->GetXYZ(x);

  TEveElement* top = gEve->GetCurrentEvent();

  if (gRPhiMgr && top)
  {
    gRPhiMgr->DestroyElements();
    gRPhiMgr->SetCenter(x[0], x[1], x[2]);
    gRPhiMgr->ImportElements(gGeomGentleRPhi);
    if (gShowTRD) gRPhiMgr->ImportElements(gGeomGentleTRD);
    gRPhiMgr->ImportElements(top);
  }
  if (gRhoZMgr && top)
  {
    gRhoZMgr->DestroyElements();
    gRhoZMgr->SetCenter(x[0], x[1], x[2]);
    gRhoZMgr->ImportElements(gGeomGentleRhoZ);
    if (gShowTRD) gRhoZMgr->ImportElements(gGeomGentleTRD);
    gRhoZMgr->ImportElements(top);
  }

  SplitGLView::UpdateSummary();
}
Commit	Line	Data
f6afd0e1	1	// $Id: NLT_trackcount_init.C 24927 2008-04-04 13:46:04Z mtadel $
	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
	12	class TEveProjectionManager;
	13	class TEveGeoShape;
	14	class TEveUtil;
	15
	16	R__EXTERN TEveProjectionManager *gRPhiMgr;
	17	R__EXTERN TEveProjectionManager *gRhoZMgr;
	18
ad5abc55	19	TEveGeoShape *gGeomGentle = 0;
	20	TEveGeoShape *gGeomGentleRPhi = 0;
	21	TEveGeoShape *gGeomGentleRhoZ = 0;
	22	TEveGeoShape *gGeomGentleTRD = 0;
f6afd0e1	23
	24	Bool_t gShowTRD = kFALSE;
	25
	26	void anyscan_init()
	27	{
	28	TEveUtil::LoadMacro("alieve_init.C");
	29	alieve_init(".", -1);
	30
df56539f	31	// TEveLine::SetDefaultSmooth(1);
f6afd0e1	32
	33	TEveUtil::AssertMacro("VizDB_scan.C");
	34
	35
	36	AliEveTrackFitter* fitter = new AliEveTrackFitter();
	37	gEve->AddToListTree(fitter, 1);
	38	gEve->AddElement(fitter, gEve->GetEventScene());
	39
	40	AliEveTrackCounter* g_trkcnt = new AliEveTrackCounter("Primary Counter");
	41	gEve->AddToListTree(g_trkcnt, kFALSE);
	42
	43
	44	gROOT->ProcessLine(".L SplitGLView.C+");
	45	TEveBrowser* browser = gEve->GetBrowser();
	46	browser->ExecPlugin("SplitGLView", 0, "new SplitGLView(gClient->GetRoot(), 600, 450, kTRUE)");
	47
	48	if (gRPhiMgr) {
	49	TEveProjectionAxes* a = new TEveProjectionAxes(gRPhiMgr);
	50	a->SetNumTickMarks(3);
	51	a->SetText("R-Phi");
	52	a->SetFontFile("comicbd");
	53	a->SetFontSize(10);
	54	gEve->GetScenes()->FindChild("R-Phi Projection")->AddElement(a);
	55	}
	56	if (gRhoZMgr) {
	57	TEveProjectionAxes* a = new TEveProjectionAxes(gRhoZMgr);
	58	a->SetNumTickMarks(3);
	59	a->SetText("Rho-Z");
	60	a->SetFontFile("comicbd");
	61	a->SetFontSize(10);
	62	gEve->GetScenes()->FindChild("Rho-Z Projection")->AddElement(a);
	63	}
	64
	65	// geometry
	66	TEveUtil::LoadMacro("geom_gentle.C");
	67	gGeomGentle = geom_gentle();
ad5abc55	68	gGeomGentleRPhi = geom_gentle_rphi(); gGeomGentleRPhi->IncDenyDestroy();
ad5abc55	69	gGeomGentleRhoZ = geom_gentle_rhoz(); gGeomGentleRhoZ->IncDenyDestroy();
f6afd0e1	70	if (gShowTRD) {
	71	TEveUtil::LoadMacro("geom_gentle_trd.C");
	72	gGeomGentleTRD = geom_gentle_trd();
	73	}
	74
	75
4d62585e	76	AliEveMacroExecutor *exec = AliEveEventManager::GetMaster()->GetExecutor();
f6afd0e1	77
7b2d546e	78	exec->AddMacro(new AliEveMacro(AliEveMacro::kRunLoader, "SIM Track", "kine_tracks.C", "kine_tracks", "", kFALSE));
	79	exec->AddMacro(new AliEveMacro(AliEveMacro::kRunLoader, "SIM Hit ITS", "its_hits.C", "its_hits", "", kFALSE));
	80	exec->AddMacro(new AliEveMacro(AliEveMacro::kRunLoader, "SIM Hit TPC", "tpc_hits.C", "tpc_hits", "", kFALSE));
29207369	81	exec->AddMacro(new AliEveMacro(AliEveMacro::kRawReader, "RAW TPC", "tpc_raw.C", "tpc_raw", "", kFALSE));
	82	exec->AddMacro(new AliEveMacro(AliEveMacro::kRawReader, "RAW FMD", "fmd_raw.C", "fmd_raw", "", kFALSE));
	83	exec->AddMacro(new AliEveMacro(AliEveMacro::kRunLoader, "DIG FMD", "fmd_digits.C", "fmd_digits", "", kFALSE));
	84	exec->AddMacro(new AliEveMacro(AliEveMacro::kRunLoader, "SIM Hit FMD", "fmd_hits.C", "fmd_hits", "", kFALSE));
	85	exec->AddMacro(new AliEveMacro(AliEveMacro::kRawReader, "RAW VZERO", "vzero_raw.C", "vzero_raw", "", kFALSE));
7b2d546e	86
ad5abc55	87	exec->AddMacro(new AliEveMacro(AliEveMacro::kESD, "REC PVTX", "primary_vertex.C", "primary_vertex", "", kTRUE));
	88	exec->AddMacro(new AliEveMacro(AliEveMacro::kESD, "REC PVTX Ellipse", "primary_vertex.C", "primary_vertex_ellipse", "", kTRUE));
	89	exec->AddMacro(new AliEveMacro(AliEveMacro::kESD, "REC PVTX Box", "primary_vertex.C", "primary_vertex_box", "kFALSE, 3, 3, 3", kFALSE));
	90	exec->AddMacro(new AliEveMacro(AliEveMacro::kESD, "REC PVTX", "primary_vertex.C", "primary_vertex_spd", "", kTRUE));
	91	exec->AddMacro(new AliEveMacro(AliEveMacro::kESD, "REC PVTX Ellipse", "primary_vertex.C", "primary_vertex_ellipse_spd", "", kTRUE));
	92	exec->AddMacro(new AliEveMacro(AliEveMacro::kESD, "REC PVTX Box", "primary_vertex.C", "primary_vertex_box_spd", "kFALSE, 3, 3, 3", kFALSE));
	93	exec->AddMacro(new AliEveMacro(AliEveMacro::kESD, "REC PVTX", "primary_vertex.C", "primary_vertex_tpc", "", kFALSE));
	94	exec->AddMacro(new AliEveMacro(AliEveMacro::kESD, "REC PVTX Ellipse", "primary_vertex.C", "primary_vertex_ellipse_tpc", "", kFALSE));
	95	exec->AddMacro(new AliEveMacro(AliEveMacro::kESD, "REC PVTX Box", "primary_vertex.C", "primary_vertex_box_tpc", "kFALSE, 3, 3, 3", kFALSE));
	96
75b17362	97	exec->AddMacro(new AliEveMacro(AliEveMacro::kESD, "REC V0", "esd_V0_points.C", "esd_V0_points"));
	98	exec->AddMacro(new AliEveMacro(AliEveMacro::kESD, "REC V0", "esd_V0.C", "esd_V0"));
	99	exec->AddMacro(new AliEveMacro(AliEveMacro::kESD, "REC CSCD", "esd_cascade_points.C", "esd_cascade_points"));
f6afd0e1	100
29207369	101	exec->AddMacro(new AliEveMacro(AliEveMacro::kESD, "REC Track", "esd_tracks.C", "esd_tracks", "", kFALSE));
	102	exec->AddMacro(new AliEveMacro(AliEveMacro::kESD, "REC Track", "esd_tracks.C", "esd_tracks_MI", "", kFALSE));
	103	exec->AddMacro(new AliEveMacro(AliEveMacro::kESD, "REC Track", "esd_tracks.C", "esd_tracks_by_category", "", kTRUE));
	104
	105	exec->AddMacro(new AliEveMacro(AliEveMacro::kESD, "REC Tracklet", "esd_spd_tracklets.C", "esd_spd_tracklets", "", kFALSE));
ad5abc55	106
7b2d546e	107	exec->AddMacro(new AliEveMacro(AliEveMacro::kRunLoader, "REC Clus ITS", "its_clusters.C+", "its_clusters"));
	108	exec->AddMacro(new AliEveMacro(AliEveMacro::kRunLoader, "REC Clus TPC", "tpc_clusters.C+", "tpc_clusters"));
	109	exec->AddMacro(new AliEveMacro(AliEveMacro::kRunLoader, "REC Clus TRD", "trd_clusters.C+", "trd_clusters"));
	110	exec->AddMacro(new AliEveMacro(AliEveMacro::kRunLoader, "REC Clus TOF", "tof_clusters.C+", "tof_clusters"));
f6afd0e1	111
	112	TEveBrowser* browser = gEve->GetBrowser();
	113
	114	browser->StartEmbedding(TRootBrowser::kRight);
	115	AliEveMacroExecutorWindow* exewin = new AliEveMacroExecutorWindow(exec);
	116	browser->StopEmbedding("DataSelection");
	117	exewin->PopulateMacros();
	118
12365217	119	browser->StartEmbedding();
	120	new AliQAHistViewer(gClient->GetRoot(), 600, 400, kTRUE);
	121	browser->StopEmbedding("QA histograms");
	122
f6afd0e1	123	browser->GetTabRight()->SetTab(1);
	124
	125	browser->StartEmbedding(TRootBrowser::kBottom);
4d62585e	126	new AliEveEventManagerWindow(AliEveEventManager::GetMaster());
f6afd0e1	127	browser->StopEmbedding("EventCtrl");
	128
	129	// event
4d62585e	130	AliEveEventManager::GetMaster()->AddNewEventCommand("on_new_event();");
4d62585e	131	AliEveEventManager::GetMaster()->GotoEvent(0);
f6afd0e1	132
	133	gEve->EditElement(g_trkcnt);
	134
	135	gEve->Redraw3D(kTRUE);
	136	}
	137
	138	/******************************************************************************/
	139
	140	void on_new_event()
	141	{
f6afd0e1	142	AliEveTrackCounter* g_trkcnt = AliEveTrackCounter::fgInstance;
f6afd0e1	143	g_trkcnt->Reset();
4d62585e	144	g_trkcnt->SetEventId(AliEveEventManager::GetMaster()->GetEventId());
29207369	145
	146	if (g_esd_tracks_by_category_container != 0)
	147	{
	148	TEveElementList* cont = g_esd_tracks_by_category_container;
	149
	150	// Here we expect several TEveTrackList containers.
	151	// First two have reasonable primaries (sigma-to-prim-vertex < 5).
	152	// Others are almost certainly secondaries.
	153	Int_t count = 1;
	154	TEveElement::List_i i = cont->BeginChildren();
	155	while (i != cont->EndChildren())
	156	{
	157	TEveTrackList* l = dynamic_cast<TEveTrackList>(i);
	158	if (l != 0)
	159	{
	160	g_trkcnt->RegisterTracks(l, (count <= 2));
	161	++count;
	162	}
	163	++i;
f6afd0e1	164	}
29207369	165
	166	// Set it to zero, so that we do not reuse an old one.
	167	g_esd_tracks_by_category_container = 0;
	168	}
	169	else
	170	{
	171	Warning("on_new_event", "g_esd_tracks_by_category_container not initialized.");
f6afd0e1	172	}
	173
	174	AliESDEvent* esd = AliEveEventManager::AssertESD();
	175	{
	176	TTimeStamp ts(esd->GetTimeStamp());
	177	TString win_title("Eve Main Window -- Timestamp: ");
	178	win_title += ts.AsString("s");
	179	win_title += "; Event # in ESD file: ";
	180	win_title += esd->GetEventNumberInFile();
	181	gEve->GetBrowser()->SetWindowName(win_title);
	182	}
	183	Double_t x[3];
	184	esd->GetPrimaryVertex()->GetXYZ(x);
	185
	186	TEveElement* top = gEve->GetCurrentEvent();
	187
29207369	188	if (gRPhiMgr && top)
29207369	189	{
f6afd0e1	190	gRPhiMgr->DestroyElements();
f6afd0e1	191	gRPhiMgr->SetCenter(x[0], x[1], x[2]);
ad5abc55	192	gRPhiMgr->ImportElements(gGeomGentleRPhi);
f6afd0e1	193	if (gShowTRD) gRPhiMgr->ImportElements(gGeomGentleTRD);
	194	gRPhiMgr->ImportElements(top);
	195	}
29207369	196	if (gRhoZMgr && top)
29207369	197	{
f6afd0e1	198	gRhoZMgr->DestroyElements();
f6afd0e1	199	gRhoZMgr->SetCenter(x[0], x[1], x[2]);
ad5abc55	200	gRhoZMgr->ImportElements(gGeomGentleRhoZ);
f6afd0e1	201	if (gShowTRD) gRhoZMgr->ImportElements(gGeomGentleTRD);
	202	gRhoZMgr->ImportElements(top);
	203	}
	204
29207369	205	SplitGLView::UpdateSummary();
f6afd0e1	206	}