]>
Commit | Line | Data |
---|---|---|
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 | |
24 | class TEvePointSet; | |
25 | class TEveElement; | |
af13c843 | 26 | #endif |
27 | ||
503bfbc8 | 28 | TEvePointSet* 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 | } |