Removing obsolete macros which used the AliTPCtracker
[u/mrichter/AliRoot.git] / EVE / alice-macros / clusters.C
1 #ifndef __CINT__
2 #include <TEveManager.h>
3 #include <TEvePointSet.h>
4 #include <TGeoManager.h>
5
6 #include <AliRunLoader.h>
7 #include <AliCluster.h>
8 #include <AliTracker.h>
9 #include <AliReconstruction.h>
10 #include <AliESDEvent.h>
11 #include <AliESDtrack.h>
12 #include <AliESDfriend.h>
13 #include <AliITSRecoParam.h>
14 #include <AliEveEventManager.h>
15 #endif
16
17 void clusters()
18 {
19   AliEveEventManager::AssertGeometry();
20
21   AliRunLoader *rl        = AliEveEventManager::AssertRunLoader();
22   AliESDEvent  *esd       = AliEveEventManager::AssertESD();
23   AliEveEventManager::AssertESDfriend();
24   AliEveEventManager::AssertMagField();
25
26   const char* detNames[] = { "ITS", "TPC", /*"TRD",*/ "TOF", "HMPID" };
27   const Int_t detIds[]   = {   0,     1,   /*  2,  */   3,     5   };
28   const Int_t detN       = sizeof(detNames)/sizeof(char*);
29
30   // Hack - AliReconstruction does wonders with gGeoManager.
31   TGeoManager* xxx = gGeoManager; gGeoManager = 0;
32   AliReconstruction* reco = new AliReconstruction;
33   gGeoManager = xxx;
34
35   // Hack for ITS - it requires RecoParams outside event-loop!
36   reco->SetRecoParam("ITS", AliITSRecoParam::GetLowFluxParam());
37
38   reco->ImportRunLoader(rl);
39   {
40     TString alldets;
41     for (Int_t i = 0; i < detN; ++i) {
42       alldets += detNames[i];
43       alldets += " ";
44     }
45     reco->CreateTrackers(alldets);
46   }
47
48   TObjArray* clarr = new TObjArray();
49
50   // Load clusters, fill them into clarr.
51
52   for (Int_t i = 0; i < detN; ++i)
53   {
54     Int_t det = detIds[i];
55     rl->LoadRecPoints(detNames[i]);
56
57     TTree *cTree = rl->GetTreeR(detNames[i], false);
58     if (cTree == 0)
59       continue;
60
61     AliTracker* tracker = reco->GetTracker(det);
62     if (tracker == 0) continue;
63     tracker->LoadClusters(cTree);
64     tracker->FillClusterArray(clarr);
65   }
66
67   // Loop over tracks and friends, tag used clusters.
68
69   for (Int_t n = 0; n < esd->GetNumberOfTracks(); ++n)
70   {
71     AliESDtrack* at = esd->GetTrack(n);
72
73     Int_t idx[200];
74     for (Int_t i = 0; i < detN; ++i)
75     {
76       Int_t det = detIds[i];
77       AliTracker* tracker = reco->GetTracker(det);
78       if (tracker == 0) continue;
79       Int_t nclusters = at->GetClusters(det, idx);
80       Int_t p=0;
81       for (Int_t c = 0; c < nclusters; )
82       {
83         Int_t index = idx[p++];
84         if (index < 0) continue;
85         c++;
86         AliCluster* cluster = tracker->GetCluster(index);
87         if (cluster) cluster->IncreaseClusterUsage();
88         //else printf("Zero cluster pointer for detector: %s\n",detNames[i]);
89       }
90     }
91   }
92
93   for (Int_t i = 0; i < detN; ++i)
94     rl->UnloadRecPoints(detNames[i]);
95
96   // Fill visualization structs
97
98   TEveElementList* list = new TEveElementList("Clusters");
99   gEve->AddElement(list);
100
101   TEvePointSet* shared = new TEvePointSet("Shared Clusters");
102   shared->SetMainColor(2);
103   shared->SetMarkerSize(0.4);
104   shared->SetMarkerStyle(2);
105   list->AddElement(shared);
106
107   TEvePointSet* used = new TEvePointSet("Single-used Clusters");
108   used->SetMainColor(3);
109   used->SetMarkerSize(0.4);
110   used->SetMarkerStyle(2);
111   list->AddElement(used);
112
113   TEvePointSet* nonused = new TEvePointSet("Not-used Clusters");
114   nonused->SetMainColor(4);
115   nonused->SetMarkerSize(0.4);
116   nonused->SetMarkerStyle(2);
117   list->AddElement(nonused);
118
119   // Loop over all clusters, fill appropriate container based
120   // on shared/used information.
121
122   Int_t ncl = clarr->GetEntriesFast();
123   for (Int_t i = 0; i < ncl; ++i)
124   {
125     AliCluster   *cluster = (AliCluster*) clarr->UncheckedAt(i);
126     TEvePointSet *dest    = 0;
127     if (cluster->IsClusterShared())
128       dest = shared;
129     else if (cluster->IsClusterUsed())
130       dest = used;
131     else
132       dest = nonused;
133
134     Float_t g[3]; //global coordinates
135     cluster->GetGlobalXYZ(g);
136     dest->SetNextPoint(g[0], g[1], g[2]);
137     dest->SetPointId(cluster);
138   }
139
140   delete clarr;
141
142   // ??? What to do with trackers ???
143   // I'd propose: have global reconstruction that owns them.
144   // AliEveEventManager::AssertAliReconstruction();
145   // do we have bit-field for detectors, like
146   // enum AliDetectors_e {
147   //    kITS = BIT(0),
148   //    kTPC = BIT(1),
149   //    ...
150   //    kCentralTracking = kITS | kTPC | kTRD | kTOF,
151   //    ...
152   //  };
153
154   gEve->Redraw3D();
155 }