Handle missing rec-point tree in a uniform way: return null pointer to TEvePointSet.
[u/mrichter/AliRoot.git] / EVE / alice-macros / its_clusters.C
CommitLineData
d810d0de 1// $Id$
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 *
51346b82 7 * full copyright notice. *
d810d0de 8 **************************************************************************/
32e219c2 9#ifdef __CINT__
d5e9edb3 10
84aff7a4 11class TEveElement;
12class TEvePointSet;
32e219c2 13
14#else
15
84aff7a4 16#include <TEveManager.h>
17#include <TEvePointSet.h>
ea5d9491 18#include <EveBase/AliEveEventManager.h>
32e219c2 19
20#include <AliRunLoader.h>
21#include <AliCluster.h>
22
23#include <TClonesArray.h>
24
25#endif
26
84aff7a4 27TEvePointSet* its_clusters(TEveElement* cont=0, Float_t maxR=50)
d5e9edb3 28{
d810d0de 29 AliEveEventManager::AssertGeometry();
d5e9edb3 30
d810d0de 31 AliRunLoader* rl = AliEveEventManager::AssertRunLoader();
d5e9edb3 32 rl->LoadRecPoints("ITS");
33
34 TTree *cTree = rl->GetTreeR("ITS", false);
8ebd7df4 35 if (cTree == 0)
36 return 0;
d5e9edb3 37
32e219c2 38 TClonesArray *cl = NULL;
39 TBranch *branch = cTree->GetBranch("ITSRecPoints");
d5e9edb3 40 branch->SetAddress(&cl);
41
8ebd7df4 42 TEvePointSet* clusters = new TEvePointSet(10000);
43 clusters->SetOwnIds(kTRUE);
44
d5e9edb3 45 Int_t nentr=(Int_t)cTree->GetEntries();
46 for (Int_t i=0; i<nentr; i++) {
47 if (!cTree->GetEvent(i)) continue;
48
49 Int_t ncl=cl->GetEntriesFast();
50
ec357b01 51 Float_t maxRsqr = maxR*maxR;
d5e9edb3 52 while (ncl--) {
53 AliCluster *c=(AliCluster*)cl->UncheckedAt(ncl);
54 Float_t g[3]; //global coordinates
55 c->GetGlobalXYZ(g);
ec357b01 56 if (g[0]*g[0]+g[1]*g[1] < maxRsqr)
57 {
58 clusters->SetNextPoint(g[0], g[1], g[2]);
59 AliCluster *atp = new AliCluster(*c);
60 clusters->SetPointId(atp);
61 }
d5e9edb3 62 }
63 }
64
84aff7a4 65 if (clusters->Size() == 0 && gEve->GetKeepEmptyCont() == kFALSE) {
8ebd7df4 66 Warning("its_clusters.C", "No ITS clusters");
d5e9edb3 67 delete clusters;
68 return 0;
69 }
70
71 clusters->SetMarkerStyle(2);
0ccc545c 72 clusters->SetMarkerSize(0.2);
d5e9edb3 73 clusters->SetMarkerColor(4);
74
75 char form[1000];
76 sprintf(form,"ITS Clusters");
77 clusters->SetName(form);
78
79 char tip[1000];
80 sprintf(tip,"N=%d", clusters->Size());
81 clusters->SetTitle(tip);
84aff7a4 82 gEve->AddElement(clusters, cont);
83 gEve->Redraw3D();
d5e9edb3 84
85 return clusters;
86}