From 7e05be18f7f38d1d210d1bd6e02fc6997a44e4bd Mon Sep 17 00:00:00 2001
From: Matevz Tadel <matevz.tadel@cern.ch>
Date: Fri, 20 Nov 2009 20:44:22 +0000
Subject: [PATCH] * alice_vsd.C New tutorial -- showing VSD (Visualization
Summar File) from ALICE. For now shows clusters and reconstructed tracks
but adding hits and kinematics is now easy!
* MultiView.C
Add function SetDepth() to make it easy to set current z-depth of
all projection managers at the same time.
git-svn-id: http://root.cern.ch/svn/root/trunk@31350 27541ba8-7e3a-0410-8455-c3a389f83636
---
tutorials/eve/MultiView.C | 16 +-
tutorials/eve/alice_vsd.C | 379 ++++++++++++++++++++++++++++++++++++++
2 files changed, 394 insertions(+), 1 deletion(-)
create mode 100644 tutorials/eve/alice_vsd.C
diff --git a/tutorials/eve/MultiView.C b/tutorials/eve/MultiView.C
index 98d0d6e845d..a759816bf24 100644
--- a/tutorials/eve/MultiView.C
+++ b/tutorials/eve/MultiView.C
@@ -35,7 +35,7 @@ struct MultiView
TEveScene *fRPhiEventScene;
TEveScene *fRhoZEventScene;
- //-------------------------------------------------------------------------
+ //---------------------------------------------------------------------------
MultiView()
{
@@ -116,6 +116,18 @@ struct MultiView
fRhoZView->AddScene(fRhoZEventScene);
}
+ //---------------------------------------------------------------------------
+
+ void SetDepth(Float_t d)
+ {
+ // Set current depth on all projection managers.
+
+ fRPhiMgr->SetCurrentDepth(d);
+ fRhoZMgr->SetCurrentDepth(d);
+ }
+
+ //---------------------------------------------------------------------------
+
void ImportGeomRPhi(TEveElement* el)
{
fRPhiMgr->ImportElements(el, fRPhiGeomScene);
@@ -136,6 +148,8 @@ struct MultiView
fRhoZMgr->ImportElements(el, fRhoZEventScene);
}
+ //---------------------------------------------------------------------------
+
void DestroyEventRPhi()
{
fRPhiEventScene->DestroyElements();
diff --git a/tutorials/eve/alice_vsd.C b/tutorials/eve/alice_vsd.C
new file mode 100644
index 00000000000..9cd82ff406e
--- /dev/null
+++ b/tutorials/eve/alice_vsd.C
@@ -0,0 +1,379 @@
+// @(#)root/eve:$Id$
+// Author: Matevz Tadel
+
+// Complex example showing ALICE VSD visualization.
+
+/*
+ alice_vsd.C - a simple event-display for ALICE
+
+ ------------------------------------------------------------------------
+ ------------------------------------------------------------------------
+
+ Only standard ROOT is used to process the ALICE VSD files.
+
+ No ALICE code is needed -- the VSD file is exported from AliRoot into
+ VSD format -- see TEveVSDStructs.h and TEveVSD.h.
+
+ A simple geometry of 10KB, extracted from the full TGeo-geometry, is
+ used to outline the central detectors of ALICE.
+
+ All files are access from the web by using the "CACHEREAD" option.
+
+*/
+
+#include <TEveManager.h>
+#include <TEveEventManager.h>
+#include <TEveVSD.h>
+#include <TEveVSDStructs.h>
+
+#include <TEveTrack.h>
+#include <TEveTrackPropagator.h>
+#include <TEveGeoShape.h>
+
+#include <TGTab.h>
+#include <TGButton.h>
+
+#include <TFile.h>
+#include <TKey.h>
+#include <TSystem.h>
+#include <TPRegexp.h>
+
+
+// Include componets -- compile time link :)
+
+#include "MultiView.C"
+MultiView* gMultiView = 0;
+
+
+class TVSDReader
+{
+public:
+ // ----------------------------------------------------------
+ // File / Event Data
+ // ----------------------------------------------------------
+
+ TFile *fFile;
+ TDirectory *fDirectory;
+
+ TObjArray *fEvDirKeys;
+
+ TEveVSD *fVSD;
+
+ Int_t fMaxEv, fCurEv;
+
+ // ----------------------------------------------------------
+ // Event visualization structures
+ // ----------------------------------------------------------
+
+ TEveTrackList *fTrackList;
+ TEvePointSet *fITSClusters;
+ TEvePointSet *fTPCClusters;
+ TEvePointSet *fTRDClusters;
+ TEvePointSet *fTOFClusters;
+
+public:
+ TVSDReader(const char* file_name) :
+ fFile(0), fDirectory(0), fEvDirKeys(0),
+ fVSD(0),
+
+ fMaxEv(-1), fCurEv(-1),
+
+ fTrackList(0),
+ fITSClusters(0), fTPCClusters(0), fTRDClusters(0), fTOFClusters(0)
+ {
+ fFile = TFile::Open(file_name);
+ if (!fFile)
+ {
+ Error("VSD_Reader", "Can not open file '%s' ... terminating.", file_name);
+ gSystem->Exit(1);
+ }
+
+ fEvDirKeys = new TObjArray;
+ TPMERegexp name_re("Event\\d+");
+ TObjLink* lnk = fFile->GetListOfKeys()->FirstLink();
+ while (lnk)
+ {
+ if (name_re.Match(lnk->GetObject()->GetName()))
+ {
+ fEvDirKeys->Add(lnk->GetObject());
+ }
+ lnk = lnk->Next();
+ }
+
+ fMaxEv = fEvDirKeys->GetEntriesFast();
+ if (fMaxEv == 0)
+ {
+ Error("VSD_Reader", "No events to show ... terminating.");
+ gSystem->Exit(1);
+ }
+
+ fVSD = new TEveVSD;
+ }
+
+
+ //---------------------------------------------------------------------------
+ // Event navigation
+ //---------------------------------------------------------------------------
+
+ void NextEvent()
+ {
+ GotoEvent(fCurEv + 1);
+ }
+
+ void PrevEvent()
+ {
+ GotoEvent(fCurEv - 1);
+ }
+
+ Bool_t GotoEvent(Int_t ev)
+ {
+ if (ev < 0 || ev >= fMaxEv)
+ {
+ Warning("GotoEvent", "Invalid event id %d.", ev);
+ return kFALSE;
+ }
+
+ // Drop old visualization structures.
+
+ gEve->GetViewers()->DeleteAnnotations();
+ gEve->GetCurrentEvent()->DestroyElements();
+
+ // Drop old event-data.
+
+ fVSD->DeleteTrees();
+ delete fDirectory;
+
+ // Connect to new event-data.
+
+ fCurEv = ev;
+ fDirectory = (TDirectory*) ((TKey*) fEvDirKeys->At(fCurEv))->ReadObj();
+
+ fVSD->SetDirectory(fDirectory);
+ fVSD->LoadTrees();
+ fVSD->SetBranchAddresses();
+
+ // Load event data into visualization structures.
+
+ LoadClusters(fITSClusters, "ITS", 0);
+ LoadClusters(fTPCClusters, "TPC", 1);
+ LoadClusters(fTRDClusters, "TRD", 2);
+ LoadClusters(fTOFClusters, "TOF", 3);
+
+ LoadEsdTracks();
+
+ // Fill projected views.
+
+ TEveElement* top = gEve->GetCurrentEvent();
+
+ gMultiView->DestroyEventRPhi();
+ gMultiView->ImportEventRPhi(top);
+
+ gMultiView->DestroyEventRhoZ();
+ gMultiView->ImportEventRhoZ(top);
+
+ gEve->Redraw3D(kFALSE, kTRUE);
+
+ return kTRUE;
+ }
+
+
+ //---------------------------------------------------------------------------
+ // Cluster loading
+ //---------------------------------------------------------------------------
+
+ void LoadClusters(TEvePointSet*& ps, const TString& det_name, Int_t det_id)
+ {
+ if (ps == 0)
+ {
+ ps = new TEvePointSet(det_name);
+ ps->SetMainColor((Color_t)(det_id + 2));
+ ps->SetMarkerSize(0.5);
+ ps->SetMarkerStyle(2);
+ ps->IncDenyDestroy();
+ }
+ else
+ {
+ ps->Reset();
+ }
+
+ TEvePointSelector ss(fVSD->fTreeC, ps, "fV.fX:fV.fY:fV.fZ",
+ TString::Format("fDetId==%d", det_id));
+ ss.Select();
+ ps->SetTitle(TString::Format("N=%d", ps->Size()));
+
+ gEve->AddElement(ps);
+ }
+
+
+ //---------------------------------------------------------------------------
+ // Track loading
+ //---------------------------------------------------------------------------
+
+ enum ESDTrackFlags
+ {
+ kITSin=0x0001,kITSout=0x0002,kITSrefit=0x0004,kITSpid=0x0008,
+ kTPCin=0x0010,kTPCout=0x0020,kTPCrefit=0x0040,kTPCpid=0x0080,
+ kTRDin=0x0100,kTRDout=0x0200,kTRDrefit=0x0400,kTRDpid=0x0800,
+ kTOFin=0x1000,kTOFout=0x2000,kTOFrefit=0x4000,kTOFpid=0x8000,
+ kHMPIDpid=0x20000,
+ kEMCALmatch=0x40000,
+ kTRDbackup=0x80000,
+ kTRDStop=0x20000000,
+ kESDpid=0x40000000,
+ kTIME=0x80000000
+ };
+
+ Bool_t trackIsOn(TEveTrack* t, Int_t mask)
+ {
+ // Check is track-flag specified by mask are set.
+
+ return (t->fStatus & mask) > 0;
+ }
+
+ void LoadEsdTracks()
+ {
+ // Read reconstructed tracks from current event.
+
+ if (fTrackList == 0)
+ {
+ fTrackList = new TEveTrackList("ESD Tracks");
+ fTrackList->SetMainColor(6);
+ fTrackList->SetMarkerColor(kYellow);
+ fTrackList->SetMarkerStyle(4);
+ fTrackList->SetMarkerSize(0.5);
+
+ fTrackList->IncDenyDestroy();
+ }
+ else
+ {
+ fTrackList->DestroyElements();
+ }
+
+ TEveTrackPropagator* trkProp = fTrackList->GetPropagator();
+ // !!!! Need to store field on file !!!!
+ // Can store TEveMagField ?
+ trkProp->SetMagField(0.5);
+
+ Int_t nTracks = fVSD->fTreeR->GetEntries();
+ for (Int_t n = 0; n < nTracks; ++n)
+ {
+ fVSD->fTreeR->GetEntry(n);
+
+ TEveTrack* track = new TEveTrack(&fVSD->fR, trkProp);
+ track->SetName(Form("ESD Track %d", fVSD->fR.fIndex));
+ track->SetStdTitle();
+ track->SetAttLineAttMarker(fTrackList);
+ fTrackList->AddElement(track);
+ }
+
+ fTrackList->MakeTracks();
+
+ gEve->AddElement(fTrackList);
+ }
+
+ ClassDef(TVSDReader, 0);
+};
+
+TVSDReader* gVSDReader = 0;
+
+
+// Forward declaration.
+void make_gui();
+
+//______________________________________________________________________________
+void alice_vsd(const char* vsd_file_name = "http://mtadel.home.cern.ch/mtadel/root/AliVSD.root")
+{
+ // Main function, initializes the application.
+ //
+ // 1. Load the auto-generated library holding ESD classes and ESD dictionaries.
+ // 2. Open ESD data-files.
+ // 3. Load cartoon geometry.
+ // 4. Spawn simple GUI.
+ // 5. Load first event.
+
+ TFile::SetCacheFileDir(".");
+
+ TEveVSD::DisableTObjectStreamersForVSDStruct();
+
+ gVSDReader = new TVSDReader(vsd_file_name);
+
+ TEveManager::Create();
+
+ TEveGeoShape *gentle_geom = 0;
+
+ { // Simple geometry
+ TFile* geom = TFile::Open("http://mtadel.home.cern.ch/mtadel/root/alice_mini_geom.root", "CACHEREAD");
+ if (!geom)
+ return;
+ TEveGeoShapeExtract* gse = (TEveGeoShapeExtract*) geom->Get("Gentle");
+ gentle_geom = TEveGeoShape::ImportShapeExtract(gse, 0);
+ geom->Close();
+ delete geom;
+ gEve->AddGlobalElement(gentle_geom);
+ }
+
+
+ // Standard multi-view
+ //=====================
+
+ gMultiView = new MultiView;
+ gMultiView->f3DView->GetGLViewer()->SetStyle(TGLRnrCtx::kOutline);
+
+ gMultiView->SetDepth(-10);
+ gMultiView->ImportGeomRPhi(gentle_geom);
+ gMultiView->ImportGeomRhoZ(gentle_geom);
+ gMultiView->SetDepth(0);
+
+
+ // Final stuff
+ //=============
+
+ gEve->GetViewers()->SwitchColorSet();
+ gEve->GetDefaultGLViewer()->SetStyle(TGLRnrCtx::kOutline);
+
+ gEve->GetBrowser()->GetTabRight()->SetTab(1);
+
+ make_gui();
+
+ gEve->AddEvent(new TEveEventManager("Event", "ALICE VSD Event"));
+
+ gVSDReader->GotoEvent(0);
+
+ gEve->Redraw3D(kTRUE); // Reset camera after the first event has been shown.
+}
+
+
+//______________________________________________________________________________
+void make_gui()
+{
+ // Create minimal GUI for event navigation.
+
+ TEveBrowser* browser = gEve->GetBrowser();
+ browser->StartEmbedding(TRootBrowser::kLeft);
+
+ TGMainFrame* frmMain = new TGMainFrame(gClient->GetRoot(), 1000, 600);
+ frmMain->SetWindowName("XX GUI");
+ frmMain->SetCleanup(kDeepCleanup);
+
+ TGHorizontalFrame* hf = new TGHorizontalFrame(frmMain);
+ {
+ TString icondir(TString::Format("%s/icons/", gSystem->Getenv("ROOTSYS")) );
+ TGPictureButton* b = 0;
+
+ b = new TGPictureButton(hf, gClient->GetPicture(icondir + "GoBack.gif"));
+ hf->AddFrame(b);
+ b->Connect("Clicked()", "TVSDReader", gVSDReader, "PrevEvent()");
+
+ b = new TGPictureButton(hf, gClient->GetPicture(icondir + "GoForward.gif"));
+ hf->AddFrame(b);
+ b->Connect("Clicked()", "TVSDReader", gVSDReader, "NextEvent()");
+ }
+ frmMain->AddFrame(hf);
+
+ frmMain->MapSubwindows();
+ frmMain->Resize();
+ frmMain->MapWindow();
+
+ browser->StopEmbedding();
+ browser->SetTabTitle("Event Control", 0);
+}
--
GitLab