Added support to display the HLT ESD Tree. See the comment in visscan_local.C for How
[u/mrichter/AliRoot.git] / EVE / alice-macros / trd_clusters.C
CommitLineData
20dae051 1// $Id$
af13c843 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
9cd50616 10#if !defined(__CINT__) || defined(__MAKECINT__)
11#include <TObjArray.h>
12#include <TTree.h>
af13c843 13#include <TEveManager.h>
9cd50616 14#include <TEveElement.h>
af13c843 15#include <TEvePointSet.h>
97587798 16#include <TGeoMatrix.h>
af13c843 17
9cd50616 18#include <AliCluster.h>
97587798 19#include <AliGeomManager.h>
9cd50616 20#include <AliRunLoader.h>
21#include <AliTRDcluster.h>
22#include <AliEveEventManager.h>
23#else
24class TEvePointSet;
25class TEveElement;
af13c843 26#endif
27
503bfbc8 28TEvePointSet* trd_clusters(TEveElement *cont = 0)
29{
30 const Int_t kMaxClusters = 18 * 6 * 24 *10;
af13c843 31
8ebd7df4 32 AliEveEventManager::AssertGeometry();
af13c843 33
503bfbc8 34 AliRunLoader *rl = AliEveEventManager::AssertRunLoader();
35 rl->LoadRecPoints("TRD");
af13c843 36
503bfbc8 37 TTree *recPoints = rl->GetTreeR("TRD", kFALSE);
8ebd7df4 38 if (recPoints == 0)
39 return 0;
40
41 TObjArray *TRDcluster = 0x0;
503bfbc8 42 recPoints->SetBranchAddress("TRDcluster", &TRDcluster);
af13c843 43
8ebd7df4 44 TEvePointSet *clusters = new TEvePointSet(kMaxClusters);
45 clusters->SetOwnIds(kTRUE);
46
af13c843 47 Int_t nentr=(Int_t)recPoints->GetEntries();
48 for (Int_t i=0; i<nentr; i++) {
49 if (!recPoints->GetEvent(i)) continue;
50
51 Int_t ncl=TRDcluster->GetEntriesFast();
52
53 while (ncl--) {
54 AliTRDcluster *c = (AliTRDcluster*)TRDcluster->UncheckedAt(ncl);
d05259ab 55 //Float_t g[3]; //global coordinates
56 //c->GetGlobalXYZ(g);
57
58 Int_t fVolumeId = c->GetVolumeId();
59 const TGeoHMatrix *mt =AliGeomManager::GetTracking2LocalMatrix(fVolumeId);;
60 Double_t txyz[3] = {c->GetX(), c->GetY(), c->GetZ()};
61 Double_t lxyz[3] = {0, 0, 0};
62 mt->LocalToMaster(txyz,lxyz);
63
64 TGeoHMatrix *mlIdeal = AliGeomManager::GetOrigGlobalMatrix(fVolumeId);
65 Double_t gxyzIdeal[3] = {0, 0, 0};
66 mlIdeal->LocalToMaster(lxyz,gxyzIdeal);
67
68 clusters->SetNextPoint(gxyzIdeal[0], gxyzIdeal[1], gxyzIdeal[2]);
69
503bfbc8 70 AliCluster *atp = new AliCluster(*c);
71 clusters->SetPointId(atp);
af13c843 72 }
73 TRDcluster->Clear();
74 }
75
9fb2a3ed 76 rl->UnloadRecPoints("TRD");
77
af13c843 78 if(clusters->Size() == 0 && gEve->GetKeepEmptyCont() == kFALSE) {
8ebd7df4 79 Warning("trd_clusters.C", "No TRD clusters");
af13c843 80 delete clusters;
81 return 0;
82 }
83
30650838 84 clusters->SetName("TRD Clusters");
af13c843 85
9dcd42ea 86 clusters->SetTitle(Form("N=%d", clusters->Size()));
f6afd0e1 87
9dcd42ea 88 const TString viz_tag("REC Clusters TRD");
30650838 89
68ca2fe7 90 clusters->ApplyVizTag(viz_tag, "Clusters");
f6afd0e1 91
af13c843 92 gEve->AddElement(clusters, cont);
f6afd0e1 93
af13c843 94 gEve->Redraw3D();
95
96 return clusters;
97}