]>
Commit | Line | Data |
---|---|---|
f37d206b | 1 | //#ifndef __CINT__ |
2 | #include "alles.h" | |
3 | #include "AliTPCtracker.h" | |
4 | #include "TView.h" | |
5 | #include "TPolyMarker3D.h" | |
6 | #include "AliSimDigits.h" | |
7 | #include "AliTPCParam.h" | |
8 | #include "AliRunLoader.h" | |
9 | #include "AliTPCclusterMI.h" | |
10 | //#endif | |
11 | ||
12 | /* | |
13 | Author: marian.ivanov@cern.ch | |
14 | ||
15 | How to use ? | |
16 | .L AliTPCDisplayClustersMI.C+ | |
17 | AliTPCDisplayClusters disp; | |
18 | disp.Init(0,0); //specify event number, and threshold for the noise | |
19 | disp.DisplayClusters(); | |
20 | ||
21 | */ | |
22 | ||
23 | ||
24 | ||
25 | ||
26 | class AliTPCDisplayClusters{ | |
27 | public: | |
28 | AliTPCDisplayClusters(); | |
29 | void SetIO(Int_t event); | |
30 | void LoadClusters(Int_t noiseth); | |
31 | void DisplayClusters(Int_t first=0, Int_t last=-1); | |
32 | void Init(Int_t event, Int_t noiseth){SetIO(event); LoadClusters(noiseth);} | |
33 | TObjArray * fArray; | |
34 | AliTPCParam * fParam; | |
35 | TTree * fTree; | |
36 | TGeometry * fGeom; | |
37 | }; | |
38 | ||
39 | ||
40 | //---------------------------------------------------------------------- | |
41 | AliTPCDisplayClusters::AliTPCDisplayClusters() | |
42 | { | |
43 | fArray = 0; | |
44 | fParam = 0; | |
45 | fTree = 0; | |
46 | fGeom = 0; | |
47 | } | |
48 | //---------------------------------------------------------------------- | |
49 | void AliTPCDisplayClusters::SetIO(Int_t event) | |
50 | { | |
51 | AliRunLoader* rl = AliRunLoader::Open(); | |
52 | rl->GetEvent(event); | |
53 | AliLoader* tpcl = (AliTPCLoader*)rl->GetLoader("TPCLoader"); | |
54 | if (tpcl == 0x0) | |
55 | { | |
56 | cerr<<"Can not get TPC Loader"<<endl; | |
57 | return; | |
58 | } | |
59 | rl->CdGAFile(); | |
60 | fParam=(AliTPCParam *)gDirectory->Get("75x40_100x60_150x60"); | |
61 | fGeom=(TGeometry*)gDirectory->Get("AliceGeom"); | |
62 | if(!fParam){ | |
63 | fParam = new AliTPCParamSR(); | |
64 | fParam->Update(); | |
65 | } | |
66 | ||
67 | if (!fParam) {cerr<<"TPC parameters have not been found !\n"; return ;} | |
68 | tpcl->LoadRecPoints(); | |
69 | fTree = tpcl->TreeR(); | |
70 | ||
71 | } | |
72 | //---------------------------------------------------------------------- | |
73 | void AliTPCDisplayClusters::LoadClusters(Int_t noiseth) | |
74 | { | |
75 | // | |
76 | // load all clusters to memory | |
77 | if (fArray) { | |
78 | fArray->Delete(); | |
79 | delete fArray; | |
80 | } | |
81 | fArray = new TObjArray(fParam->GetNSegmentsTotal()); | |
82 | // | |
83 | AliTPCClustersRow * pclrow = new AliTPCClustersRow; | |
84 | pclrow->SetClass("AliTPCclusterMI"); | |
85 | pclrow->SetArray(0); | |
86 | TBranch *br = fTree->GetBranch("Segment"); | |
87 | br->SetAddress(&pclrow); | |
88 | // | |
89 | // | |
90 | Int_t nrows=Int_t(fTree->GetEntries()); | |
91 | for (Int_t n=0; n<nrows; n++) { | |
92 | // | |
93 | pclrow = new AliTPCClustersRow; | |
94 | pclrow->SetClass("AliTPCclusterMI"); | |
95 | pclrow->SetArray(0); | |
96 | br->SetAddress(&pclrow); | |
97 | // | |
98 | br->GetEntry(n); | |
99 | AliTPCClustersRow &clrow = *pclrow; | |
100 | Int_t ncl=clrow.GetArray()->GetEntriesFast(); | |
101 | TObjArray * arrrow = new TObjArray(0); | |
102 | fArray->AddAt(arrrow,pclrow->GetID()); | |
103 | ||
104 | // printf("segment\t%d\trow\t%d\tclusters\t%d",n,pclrow->GetID(),ncl); | |
105 | while (ncl--) { | |
106 | AliTPCclusterMI *cl=(AliTPCclusterMI*)clrow[ncl]; | |
107 | if (cl->GetQ()>noiseth){ | |
108 | arrrow->AddLast(cl); | |
109 | } | |
110 | } | |
111 | // printf("over\t%d\n",arrrow->GetEntries()); | |
112 | ||
113 | } | |
114 | } | |
115 | ||
116 | ||
117 | void AliTPCDisplayClusters::DisplayClusters(Int_t first, Int_t last) | |
118 | { | |
119 | Int_t nrows = fParam->GetNSegmentsTotal(); | |
120 | // some "constants" | |
121 | Int_t markerColorSignal = 5; | |
122 | Int_t markerColorBgr = 2; | |
123 | Int_t MASK = 10000000; | |
124 | ||
125 | TCanvas *c1=new TCanvas("cdisplay", "Cluster display",0,0,700,730); | |
126 | TView *v=new TView(1); | |
127 | v->SetRange(-330,-360,-330,360,360,1710); | |
128 | c1->Clear(); | |
129 | c1->SetFillColor(1); | |
130 | c1->SetTheta(90.); | |
131 | c1->SetPhi(0.); | |
132 | ||
133 | for (Int_t irow=0; irow<nrows; irow++) { | |
134 | ||
135 | TObjArray * arr = (TObjArray*)fArray->At(irow); | |
136 | if (!arr) continue; | |
137 | ||
138 | Int_t sec,row; | |
139 | fParam->AdjustSectorRow(irow,sec,row); | |
140 | Int_t ncl=arr->GetEntriesFast(); | |
141 | ||
142 | TPolyMarker3D *pm=new TPolyMarker3D(ncl); | |
143 | TPolyMarker3D *pmSignal=new TPolyMarker3D(ncl); // polymarker for signal | |
144 | Int_t imarBgr=0; | |
145 | Int_t imarSignal=0; | |
146 | ||
147 | while (ncl--) { | |
148 | AliTPCclusterMI *cl=(AliTPCclusterMI*)arr->At(ncl); | |
149 | if (cl){ | |
150 | // | |
151 | Double_t x=fParam->GetPadRowRadii(sec,row), y=cl->GetY(), z=cl->GetZ(); | |
152 | Float_t cs, sn, tmp; | |
153 | fParam->AdjustCosSin(sec,cs,sn); | |
154 | tmp = x*cs-y*sn; y= x*sn+y*cs; x=tmp; | |
155 | Int_t trackId = cl->GetLabel(0); | |
156 | if ( (last>0) &&trackId>last) continue; | |
157 | if (trackId<first) continue; | |
158 | if (trackId < MASK-1) { | |
159 | pmSignal->SetPoint(imarSignal,x,y,z); | |
160 | imarSignal++; | |
161 | } else { | |
162 | pm->SetPoint(imarBgr,x,y,z); | |
163 | imarBgr++; | |
164 | } | |
165 | } | |
166 | } | |
167 | ||
168 | // change color for signal | |
169 | pm->SetMarkerSize(1); | |
170 | pm->SetMarkerColor(markerColorBgr); | |
171 | pm->SetMarkerStyle(1); | |
172 | pm->Draw(); | |
173 | ||
174 | pmSignal->SetMarkerSize(1); | |
175 | pmSignal->SetMarkerColor(markerColorSignal); | |
176 | pmSignal->SetMarkerStyle(1); | |
177 | pmSignal->Draw(); | |
178 | } | |
179 | ||
180 | ||
181 | TNode * main = (TNode*)((fGeom->GetListOfNodes())->First()); | |
182 | TIter next(main->GetListOfNodes()); | |
183 | TNode *module=0; | |
184 | while((module = (TNode*)next())) { | |
185 | char ch[100]; | |
186 | sprintf(ch,"%s\n",module->GetTitle()); | |
187 | //printf("%s\n",module->GetTitle()); | |
188 | if (ch[0]=='T'&&ch[1]=='P' && ch[2]=='C') //if TPC draw | |
189 | module->SetVisibility(3); | |
190 | else | |
191 | module->SetVisibility(-1); | |
192 | } | |
193 | ||
194 | ||
195 | fGeom->Draw("same"); | |
196 | c1->Modified(); c1->Update(); | |
197 | ||
198 | ||
199 | ||
200 | } |