bug fixes for writing raw data from simulation
[u/mrichter/AliRoot.git] / ITS / AliITSClusterFinderV2SDD.cxx
CommitLineData
04366a57 1/**************************************************************************
2 * Copyright(c) 1998-2003, ALICE Experiment at CERN, All rights reserved. *
3 * *
4 * Author: The ALICE Off-line Project. *
5 * Contributors are mentioned in the code where appropriate. *
6 * *
7 * Permission to use, copy, modify and distribute this software and its *
8 * documentation strictly for non-commercial purposes is hereby granted *
9 * without fee, provided that the above copyright notice appears in all *
10 * copies and that both the copyright notice and this permission notice *
11 * appear in the supporting documentation. The authors make no claims *
12 * about the suitability of this software for any purpose. It is *
13 * provided "as is" without express or implied warranty. *
14 **************************************************************************/
d76cb3ad 15
16/* $Id$*/
17
04366a57 18////////////////////////////////////////////////////////////////////////////
19// Implementation of the ITS clusterer V2 class //
20// //
21// Origin: Iouri Belikov, CERN, Jouri.Belikov@cern.ch //
22// //
23///////////////////////////////////////////////////////////////////////////
24
25
04366a57 26
ae485645 27#include <TClonesArray.h>
04366a57 28#include "AliITSClusterFinderV2SDD.h"
00a7cc50 29#include "AliITSRecPoint.h"
7d62fb64 30#include "AliITSDetTypeRec.h"
04366a57 31#include "AliRawReader.h"
0599a018 32#include "AliITSRawStreamSDD.h"
5bfe44ce 33#include "AliITSCalibrationSDD.h"
34#include "AliITSDetTypeRec.h"
35#include "AliITSsegmentationSDD.h"
04366a57 36#include "AliITSdigitSDD.h"
b4704be3 37#include "AliITSgeomTGeo.h"
04366a57 38ClassImp(AliITSClusterFinderV2SDD)
39
4952f440 40AliITSClusterFinderV2SDD::AliITSClusterFinderV2SDD(AliITSDetTypeRec* dettyp):AliITSClusterFinderV2(dettyp)
a66a0eb6 41{
04366a57 42
43 //Default constructor
44
04366a57 45}
46
47
48void AliITSClusterFinderV2SDD::FindRawClusters(Int_t mod){
49
50 //Find clusters V2
51 SetModule(mod);
52 FindClustersSDD(fDigits);
53
54}
55
56void AliITSClusterFinderV2SDD::FindClustersSDD(TClonesArray *digits) {
57 //------------------------------------------------------------
58 // Actual SDD cluster finder
59 //------------------------------------------------------------
4952f440 60 Int_t nAnodes = GetSeg()->NpzHalf();
61 Int_t nzBins = nAnodes+2;
62 Int_t nTimeBins = GetSeg()->Npx();
63 Int_t nxBins = nTimeBins+2;
64 const Int_t kMaxBin=nzBins*(nxBins+2);
04366a57 65
66 AliBin *bins[2];
4952f440 67 bins[0]=new AliBin[kMaxBin];
68 bins[1]=new AliBin[kMaxBin];
404c1c29 69 AliITSCalibrationSDD* cal = (AliITSCalibrationSDD*)GetResp(fModule);
0599a018 70 AliITSresponseSDD* res = (AliITSresponseSDD*)cal->GetResponse();
71 const char *option=res->ZeroSuppOption();
04366a57 72 AliITSdigitSDD *d=0;
73 Int_t i, ndigits=digits->GetEntriesFast();
74 for (i=0; i<ndigits; i++) {
75 d=(AliITSdigitSDD*)digits->UncheckedAt(i);
76 Int_t y=d->GetCoord2()+1; //y
77 Int_t z=d->GetCoord1()+1; //z
5683bd96 78 Float_t gain=cal->GetChannelGain(d->GetCoord1());
79 Float_t charge=d->GetSignal();
80
0599a018 81 if(!((strstr(option,"1D")) || (strstr(option,"2D")))){
82 Float_t baseline = cal->GetBaseline(d->GetCoord1());
5683bd96 83 if(charge>baseline) charge-=baseline;
84 else charge=0;
0599a018 85 }
5683bd96 86
87 if(gain>0) charge/=gain;
88 if(charge<cal->GetThresholdAnode(d->GetCoord1())) continue;
89 Int_t q=(Int_t)(charge+0.5);
4952f440 90 if (z <= nAnodes){
91 bins[0][y*nzBins+z].SetQ(q);
92 bins[0][y*nzBins+z].SetMask(1);
93 bins[0][y*nzBins+z].SetIndex(i);
04366a57 94 } else {
4952f440 95 z-=nAnodes;
96 bins[1][y*nzBins+z].SetQ(q);
97 bins[1][y*nzBins+z].SetMask(1);
98 bins[1][y*nzBins+z].SetIndex(i);
04366a57 99 }
100 }
101
4952f440 102 FindClustersSDD(bins, kMaxBin, nzBins, digits);
04366a57 103
104 delete[] bins[0];
105 delete[] bins[1];
106
107}
108
109void AliITSClusterFinderV2SDD::
110FindClustersSDD(AliBin* bins[2], Int_t nMaxBin, Int_t nzBins,
111 TClonesArray *digits, TClonesArray *clusters) {
112 //------------------------------------------------------------
113 // Actual SDD cluster finder
114 //------------------------------------------------------------
b4704be3 115
116 const TGeoHMatrix *mT2L=AliITSgeomTGeo::GetTracking2LocalMatrix(fModule);
404c1c29 117 AliITSCalibrationSDD* cal = (AliITSCalibrationSDD*)GetResp(fModule);
04366a57 118 Int_t ncl=0;
119 TClonesArray &cl=*clusters;
120 for (Int_t s=0; s<2; s++)
121 for (Int_t i=0; i<nMaxBin; i++) {
122 if (bins[s][i].IsUsed()) continue;
123 Int_t idx[32]; UInt_t msk[32]; Int_t npeaks=0;
124 FindPeaks(i, nzBins, bins[s], idx, msk, npeaks);
125
126 if (npeaks>30) continue;
127 if (npeaks==0) continue;
128
129 Int_t k,l;
130 for (k=0; k<npeaks-1; k++){//mark adjacent peaks
131 if (idx[k] < 0) continue; //this peak is already removed
132 for (l=k+1; l<npeaks; l++) {
133 if (idx[l] < 0) continue; //this peak is already removed
134 Int_t ki=idx[k]/nzBins, kj=idx[k] - ki*nzBins;
135 Int_t li=idx[l]/nzBins, lj=idx[l] - li*nzBins;
136 Int_t di=TMath::Abs(ki - li);
137 Int_t dj=TMath::Abs(kj - lj);
138 if (di>1 || dj>1) continue;
139 if (bins[s][idx[k]].GetQ() > bins[s][idx[l]].GetQ()) {
140 msk[l]=msk[k];
141 idx[l]*=-1;
142 } else {
143 msk[k]=msk[l];
144 idx[k]*=-1;
145 break;
146 }
147 }
148 }
149
150 for (k=0; k<npeaks; k++) {
151 MarkPeak(TMath::Abs(idx[k]), nzBins, bins[s], msk[k]);
152 }
153
154 for (k=0; k<npeaks; k++) {
155 if (idx[k] < 0) continue; //removed peak
75fb37cc 156 AliITSRecPoint c;
04366a57 157 MakeCluster(idx[k], nzBins, bins[s], msk[k], c);
158 //mi change
159 Int_t milab[10];
160 for (Int_t ilab=0;ilab<10;ilab++){
161 milab[ilab]=-2;
162 }
163 Int_t maxi=0,mini=0,maxj=0,minj=0;
164 //AliBin *bmax=&bins[s][idx[k]];
165 //Float_t max = TMath::Max(TMath::Abs(bmax->GetQ())/5.,3.);
24c98369 166
5dfa68c5 167 for (Int_t di=-2; di<=2;di++){
04366a57 168 for (Int_t dj=-3;dj<=3;dj++){
169 Int_t index = idx[k]+di+dj*nzBins;
170 if (index<0) continue;
171 if (index>=nMaxBin) continue;
172 AliBin *b=&bins[s][index];
5dfa68c5 173 Int_t nAnode=index%nzBins-1;
174 Int_t adcSignal=b->GetQ();
175 if(adcSignal>cal->GetThresholdAnode(nAnode)){
176 if (di>maxi) maxi=di;
177 if (di<mini) mini=di;
178 if (dj>maxj) maxj=dj;
179 if (dj<minj) minj=dj;
180 }
24c98369 181 //
182 if(digits) {
183 if (TMath::Abs(di)<2&&TMath::Abs(dj)<2){
184 AliITSdigitSDD* d=(AliITSdigitSDD*)digits->UncheckedAt(b->GetIndex());
185 for (Int_t itrack=0;itrack<10;itrack++){
186 Int_t track = (d->GetTracks())[itrack];
187 if (track>=0) {
188 AddLabel(milab, track);
04366a57 189 }
190 }
191 }
5dfa68c5 192 }
04366a57 193 }
5dfa68c5 194 }
d76cb3ad 195
04366a57 196
04366a57 197 Float_t y=c.GetY(),z=c.GetZ(), q=c.GetQ();
198 y/=q; z/=q;
d76cb3ad 199
404c1c29 200
4952f440 201 const Double_t kMicronTocm = 1.0e-4;
202 Float_t timeBinCenter = y-0.5;
203 Float_t zAnode=z-0.5;
204 Float_t zdet = (zAnode*GetSeg()->Dpz(0)-GetSeg()->Dz()/2.)*kMicronTocm;
205 Float_t driftTime = timeBinCenter*GetSeg()->Dpx(0) - cal->GetTimeOffset();
206 Float_t xdet = cal->GetDriftPath(driftTime,zAnode);
207 xdet=(xdet-GetSeg()->Dx())*kMicronTocm;
404c1c29 208 if (s) xdet=-xdet;
209
4952f440 210
404c1c29 211 CorrectPosition(zdet,xdet);
212
4952f440 213 Double_t loc[3]={xdet,0.,zdet},trk[3]={0.,0.,0.};
214 mT2L->MasterToLocal(loc,trk);
215 y=trk[1];
216 z=trk[2];
217
a66a0eb6 218 q/=cal->GetADC2keV(); //to have MPV 1 MIP = 86.4 KeV
ecb9da3c 219 Float_t hit[5] = {y, z, 0.0030*0.0030, 0.0020*0.0020, q};
220 Int_t info[3] = {maxj-minj+1, maxi-mini+1, fNlayer[fModule]};
04366a57 221 if (digits) {
222 // AliBin *b=&bins[s][idx[k]];
223 // AliITSdigitSDD* d=(AliITSdigitSDD*)digits->UncheckedAt(b->GetIndex());
224 {
225 //Int_t lab[3];
226 //lab[0]=(d->GetTracks())[0];
227 //lab[1]=(d->GetTracks())[1];
228 //lab[2]=(d->GetTracks())[2];
229 //CheckLabels(lab);
230 CheckLabels2(milab);
04366a57 231 }
232 }
ecb9da3c 233 milab[3]=fNdet[fModule];
234
235 AliITSRecPoint cc(milab,hit,info);
236 cc.SetType(npeaks);
237
238 if(clusters) new (cl[ncl]) AliITSRecPoint(cc);
04366a57 239 else {
ecb9da3c 240 fDetTypeRec->AddRecPoint(cc);
04366a57 241 }
242 ncl++;
243 }
244 }
24c98369 245
04366a57 246}
247
248
249
250void AliITSClusterFinderV2SDD::RawdataToClusters(AliRawReader* rawReader,TClonesArray** clusters){
251 //------------------------------------------------------------
252 // This function creates ITS clusters from raw data
253 //------------------------------------------------------------
254 rawReader->Reset();
0599a018 255 AliITSRawStreamSDD inputSDD(rawReader);
04366a57 256 FindClustersSDD(&inputSDD,clusters);
257
258}
259
260void AliITSClusterFinderV2SDD::FindClustersSDD(AliITSRawStream* input,
261 TClonesArray** clusters)
262{
263 //------------------------------------------------------------
264 // Actual SDD cluster finder for raw data
265 //------------------------------------------------------------
266 Int_t nClustersSDD = 0;
4952f440 267 Int_t nAnodes = GetSeg()->NpzHalf();
268 Int_t nzBins = nAnodes+2;
269 Int_t nTimeBins = GetSeg()->Npx();
270 Int_t nxBins = nTimeBins+2;
271 const Int_t kMaxBin=nzBins*(nxBins+2);
ae485645 272 AliBin *bins[2];
14dceddf 273 AliBin *ddlbins[24]; // 12 modules (=24 hybrids) of 1 DDL read "in parallel"
274 for(Int_t iHyb=0;iHyb<24;iHyb++) ddlbins[iHyb]=new AliBin[kMaxBin];
04366a57 275 // read raw data input stream
14dceddf 276 while (input->Next()) {
e4da2288 277 Int_t iModule = input->GetModuleID();
278 if(iModule<0) continue;
14dceddf 279 Int_t iCarlos =((AliITSRawStreamSDD*)input)->GetCarlosId();
280 Int_t iSide = ((AliITSRawStreamSDD*)input)->GetChannel();
281 Int_t iHybrid=iCarlos*2+iSide;
282 if (input->IsCompletedModule()) {
04366a57 283 // when all data from a module was read, search for clusters
ebbc4ece 284 clusters[iModule] = new TClonesArray("AliITSRecPoint");
285 fModule = iModule;
14dceddf 286 bins[0]=ddlbins[iCarlos*2]; // first hybrid of the completed module
287 bins[1]=ddlbins[iCarlos*2+1]; // second hybrid of the completed module
ebbc4ece 288 FindClustersSDD(bins, kMaxBin, nzBins, NULL, clusters[iModule]);
289 Int_t nClusters = clusters[iModule]->GetEntriesFast();
290 nClustersSDD += nClusters;
291 for(Int_t iBin=0;iBin<kMaxBin; iBin++){
14dceddf 292 ddlbins[iCarlos*2][iBin].Reset();
293 ddlbins[iCarlos*2+1][iBin].Reset();
ebbc4ece 294 }
5dfa68c5 295 }else{
04366a57 296 // fill the current digit into the bins array
e4da2288 297 AliITSCalibrationSDD* cal = (AliITSCalibrationSDD*)GetResp(iModule);
5dfa68c5 298 AliITSresponseSDD* res = (AliITSresponseSDD*)cal->GetResponse();
299 const char *option=res->ZeroSuppOption();
5683bd96 300 Float_t charge=input->GetSignal();
14dceddf 301 Int_t chan=input->GetCoord1()+nAnodes*iSide;
302 Float_t gain=cal->GetChannelGain(chan);
5dfa68c5 303 if(!((strstr(option,"1D")) || (strstr(option,"2D")))){
14dceddf 304 Float_t baseline = cal->GetBaseline(chan);
5683bd96 305 if(charge>baseline) charge-=baseline;
306 else charge=0;
5dfa68c5 307 }
5683bd96 308 if(gain>0) charge/=gain;
14dceddf 309 if(charge>=cal->GetThresholdAnode(chan)) {
310 Int_t q=(Int_t)(charge+0.5);
5dfa68c5 311 Int_t iz = input->GetCoord1()+1;
4952f440 312 Int_t index = (input->GetCoord2()+1) * nzBins + iz;
14dceddf 313 ddlbins[iHybrid][index].SetQ(q);
314 ddlbins[iHybrid][index].SetMask(1);
315 ddlbins[iHybrid][index].SetIndex(index);
5dfa68c5 316 }
04366a57 317 }
318 }
14dceddf 319 for(Int_t iHyb=0;iHyb<24;iHyb++) delete [] ddlbins[iHyb];
04366a57 320 Info("FindClustersSDD", "found clusters in ITS SDD: %d", nClustersSDD);
04366a57 321}
322
323
5bfe44ce 324//_________________________________________________________________________
325void AliITSClusterFinderV2SDD::CorrectPosition(Float_t &z, Float_t&y){
326
327 //correction of coordinates using the maps stored in the DB
328
329 AliITSCalibrationSDD* cal = (AliITSCalibrationSDD*)GetResp(fModule);
a97b3678 330 static const Int_t knbint = cal->GetMapTimeNBin();
331 static const Int_t knbina = cal->Chips()*cal->Channels();
4952f440 332 const Double_t kMicronTocm = 1.0e-4;
333 Float_t stepa = (GetSeg()->Dpz(0))*kMicronTocm; //anode pitch in cm
5bfe44ce 334 Float_t stept = (GetSeg()->Dx()/cal->GetMapTimeNBin()/2.)/10.;
404c1c29 335
336 Int_t bint = TMath::Abs((Int_t)(y/stept));
a97b3678 337 if(y>=0) bint+=(Int_t)(knbint/2.);
338 if(bint>knbint) AliError("Wrong bin number!");
5bfe44ce 339
404c1c29 340 Int_t bina = TMath::Abs((Int_t)(z/stepa));
a97b3678 341 if(z>=0) bina+=(Int_t)(knbina/2.);
342 if(bina>knbina) AliError("Wrong bin number!");
5bfe44ce 343
4952f440 344 Float_t devz = cal->GetMapACell(bina,bint)*kMicronTocm;
345 Float_t devx = cal->GetMapTCell(bina,bint)*kMicronTocm;
404c1c29 346 z+=devz;
347 y+=devx;
5bfe44ce 348
349
350}