]> git.uio.no Git - u/mrichter/AliRoot.git/blob - ITS/AliITSClusterFinderV2SDD.cxx
Possibility of selecting one subdetector when making recpoints starting from raw...
[u/mrichter/AliRoot.git] / ITS / AliITSClusterFinderV2SDD.cxx
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  **************************************************************************/
15
16 /* $Id$*/
17
18 ////////////////////////////////////////////////////////////////////////////
19 //            Implementation of the ITS clusterer V2 class                //
20 //                                                                        //
21 //          Origin: Iouri Belikov, CERN, Jouri.Belikov@cern.ch            //
22 //                                                                        //
23 ///////////////////////////////////////////////////////////////////////////
24
25
26
27 #include "AliITSClusterFinderV2SDD.h"
28 #include "AliITSRecPoint.h"
29 #include "AliITSDetTypeRec.h"
30 #include "AliRawReader.h"
31 #include "AliITSRawStreamSDD.h"
32 #include "AliITSCalibrationSDD.h"
33 #include "AliITSDetTypeRec.h"
34 #include "AliITSsegmentationSDD.h"
35 #include <TClonesArray.h>
36 #include "AliITSdigitSDD.h"
37 #include "AliITSgeomTGeo.h"
38 ClassImp(AliITSClusterFinderV2SDD)
39
40 AliITSClusterFinderV2SDD::AliITSClusterFinderV2SDD(AliITSDetTypeRec* dettyp):AliITSClusterFinderV2(dettyp),
41 fNySDD(256),
42 fNzSDD(256),
43 fZpitchSDD(0.02940),
44 fHwSDD(3.5085),
45 fHlSDD(3.7632),
46 fTimeOffsetSDD(0.){
47
48   //Default constructor
49
50   SetTimeOffset();
51 }
52  
53
54 void AliITSClusterFinderV2SDD::FindRawClusters(Int_t mod){
55
56   //Find clusters V2
57   SetModule(mod);
58   FindClustersSDD(fDigits);
59
60 }
61
62 void AliITSClusterFinderV2SDD::FindClustersSDD(TClonesArray *digits) {
63   //------------------------------------------------------------
64   // Actual SDD cluster finder
65   //------------------------------------------------------------
66   Int_t kNzBins = fNzSDD + 2;
67   const Int_t kMAXBIN=kNzBins*(fNySDD+2);
68
69   AliBin *bins[2];
70   bins[0]=new AliBin[kMAXBIN];
71   bins[1]=new AliBin[kMAXBIN];
72   AliITSCalibrationSDD* cal = (AliITSCalibrationSDD*)GetResp(fModule);
73   AliITSresponseSDD* res  = (AliITSresponseSDD*)cal->GetResponse();
74   const char *option=res->ZeroSuppOption();
75   AliITSdigitSDD *d=0;
76   Int_t i, ndigits=digits->GetEntriesFast();
77   for (i=0; i<ndigits; i++) {
78      d=(AliITSdigitSDD*)digits->UncheckedAt(i);
79      Int_t y=d->GetCoord2()+1;   //y
80      Int_t z=d->GetCoord1()+1;   //z
81      Int_t q=d->GetSignal();
82      if(!((strstr(option,"1D")) || (strstr(option,"2D")))){
83        Float_t baseline = cal->GetBaseline(d->GetCoord1());
84        if(q>baseline) q-=(Int_t)baseline;
85        else q=0;
86      }
87      if(q<cal->GetThresholdAnode(d->GetCoord1())) continue;
88      
89      //if (q<3) continue;
90
91      if (z <= fNzSDD) {
92        bins[0][y*kNzBins+z].SetQ(q);
93        bins[0][y*kNzBins+z].SetMask(1);
94        bins[0][y*kNzBins+z].SetIndex(i);
95      } else {
96        z-=fNzSDD; 
97        bins[1][y*kNzBins+z].SetQ(q);
98        bins[1][y*kNzBins+z].SetMask(1);
99        bins[1][y*kNzBins+z].SetIndex(i);
100      }
101   }
102   
103   FindClustersSDD(bins, kMAXBIN, kNzBins, digits);
104
105   delete[] bins[0];
106   delete[] bins[1];
107
108 }
109
110 void AliITSClusterFinderV2SDD::
111 FindClustersSDD(AliBin* bins[2], Int_t nMaxBin, Int_t nzBins, 
112                 TClonesArray *digits, TClonesArray *clusters) {
113   //------------------------------------------------------------
114   // Actual SDD cluster finder
115   //------------------------------------------------------------
116
117   const TGeoHMatrix *mT2L=AliITSgeomTGeo::GetTracking2LocalMatrix(fModule);
118   AliITSCalibrationSDD* cal = (AliITSCalibrationSDD*)GetResp(fModule);
119   Int_t ncl=0; 
120   TClonesArray &cl=*clusters;
121   for (Int_t s=0; s<2; s++)
122     for (Int_t i=0; i<nMaxBin; i++) {
123       if (bins[s][i].IsUsed()) continue;
124       Int_t idx[32]; UInt_t msk[32]; Int_t npeaks=0;
125       FindPeaks(i, nzBins, bins[s], idx, msk, npeaks);
126
127       if (npeaks>30) continue;
128       if (npeaks==0) continue;
129
130       Int_t k,l;
131       for (k=0; k<npeaks-1; k++){//mark adjacent peaks
132         if (idx[k] < 0) continue; //this peak is already removed
133         for (l=k+1; l<npeaks; l++) {
134            if (idx[l] < 0) continue; //this peak is already removed
135            Int_t ki=idx[k]/nzBins, kj=idx[k] - ki*nzBins;
136            Int_t li=idx[l]/nzBins, lj=idx[l] - li*nzBins;
137            Int_t di=TMath::Abs(ki - li);
138            Int_t dj=TMath::Abs(kj - lj);
139            if (di>1 || dj>1) continue;
140            if (bins[s][idx[k]].GetQ() > bins[s][idx[l]].GetQ()) {
141               msk[l]=msk[k];
142               idx[l]*=-1;
143            } else {
144               msk[k]=msk[l];
145               idx[k]*=-1;
146               break;
147            } 
148         }
149       }
150
151       for (k=0; k<npeaks; k++) {
152         MarkPeak(TMath::Abs(idx[k]), nzBins, bins[s], msk[k]);
153       }
154         
155       for (k=0; k<npeaks; k++) {
156          if (idx[k] < 0) continue; //removed peak
157          AliITSRecPoint c;
158          MakeCluster(idx[k], nzBins, bins[s], msk[k], c);
159          //mi change
160          Int_t milab[10];
161          for (Int_t ilab=0;ilab<10;ilab++){
162            milab[ilab]=-2;
163          }
164          Int_t maxi=0,mini=0,maxj=0,minj=0;
165          //AliBin *bmax=&bins[s][idx[k]];
166          //Float_t max = TMath::Max(TMath::Abs(bmax->GetQ())/5.,3.);
167     
168          for (Int_t di=-2; di<=2;di++)
169            for (Int_t dj=-3;dj<=3;dj++){
170              Int_t index = idx[k]+di+dj*nzBins;
171              if (index<0) continue;
172              if (index>=nMaxBin) continue;
173              AliBin *b=&bins[s][index];
174              if (di>maxi) maxi=di;
175              if (di<mini) mini=di;
176              if (dj>maxj) maxj=dj;
177              if (dj<minj) minj=dj;
178              //
179              if(digits) {
180                if (TMath::Abs(di)<2&&TMath::Abs(dj)<2){
181                  AliITSdigitSDD* d=(AliITSdigitSDD*)digits->UncheckedAt(b->GetIndex());
182                  for (Int_t itrack=0;itrack<10;itrack++){
183                    Int_t track = (d->GetTracks())[itrack];
184                    if (track>=0) {
185                      AddLabel(milab, track); 
186                    }
187                  }
188                }
189              }   
190            }
191          
192
193
194          Float_t y=c.GetY(),z=c.GetZ(), q=c.GetQ();
195          y/=q; z/=q;
196
197
198          Float_t yyyy = y;
199
200          z=(z-0.5)*fZpitchSDD;
201          z-=fHlSDD;
202          Float_t zdet = z;
203          Float_t timebin = GetSeg()->Dpx(0);
204          Float_t driftTime = (yyyy-0.5)*timebin - fTimeOffsetSDD;
205          Float_t xdet = cal->GetDriftPath(driftTime,0);
206          xdet=xdet/10000.-fHwSDD;
207          if (s) xdet=-xdet;
208          
209          CorrectPosition(zdet,xdet);
210
211          {
212          Double_t loc[3]={xdet,0.,zdet},trk[3]={0.,0.,0.};
213          mT2L->MasterToLocal(loc,trk);
214          y=trk[1];
215          z=trk[2]; 
216          }
217          q/=5.243;  //to have MPV 1 MIP = 86.4 KeV --> this must go to calibr.
218          Float_t hit[5] = {y, z, 0.0030*0.0030, 0.0020*0.0020, q};
219          Int_t  info[3] = {maxj-minj+1, maxi-mini+1, fNlayer[fModule]};
220          
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); 
231            }
232          }
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); 
239          else {
240            fDetTypeRec->AddRecPoint(cc);
241          }
242          ncl++;
243       }
244     }
245   
246 }
247
248
249
250 void AliITSClusterFinderV2SDD::RawdataToClusters(AliRawReader* rawReader,TClonesArray** clusters){
251     //------------------------------------------------------------
252   // This function creates ITS clusters from raw data
253   //------------------------------------------------------------
254   rawReader->Reset();
255   AliITSRawStreamSDD inputSDD(rawReader);
256   FindClustersSDD(&inputSDD,clusters);
257
258 }
259
260 void AliITSClusterFinderV2SDD::FindClustersSDD(AliITSRawStream* input, 
261                                         TClonesArray** clusters) 
262 {
263   //------------------------------------------------------------
264   // Actual SDD cluster finder for raw data
265   //------------------------------------------------------------
266   Int_t nClustersSDD = 0;
267   Int_t kNzBins = fNzSDD + 2;
268   Int_t kMaxBin = kNzBins * (fNySDD+2);
269   AliBin *binsSDDInit = new AliBin[kMaxBin];
270   AliBin *binsSDD1 = new AliBin[kMaxBin];
271   AliBin *binsSDD2 = new AliBin[kMaxBin];
272   AliBin* bins[2] = {NULL, NULL};
273
274   // read raw data input stream
275   while (kTRUE) {
276     Bool_t next = input->Next();
277     if (!next || input->IsNewModule()) {
278       Int_t iModule = input->GetPrevModuleID();
279
280       // when all data from a module was read, search for clusters
281       if (bins[0]) { 
282         clusters[iModule] = new TClonesArray("AliITSRecPoint");
283         fModule = iModule;
284         FindClustersSDD(bins, kMaxBin, kNzBins, NULL, clusters[iModule]);
285         Int_t nClusters = clusters[iModule]->GetEntriesFast();
286         nClustersSDD += nClusters;
287         bins[0] = bins[1] = NULL;       
288       }
289
290       if (!next) break;
291       bins[0]=binsSDD1;
292       bins[1]=binsSDD2;
293       memcpy(binsSDD1,binsSDDInit,sizeof(AliBin)*kMaxBin);
294       memcpy(binsSDD2,binsSDDInit,sizeof(AliBin)*kMaxBin);
295
296     }
297
298     // fill the current digit into the bins array
299     AliITSCalibrationSDD* cal = (AliITSCalibrationSDD*)GetResp(input->GetModuleID());
300     AliITSresponseSDD* res  = (AliITSresponseSDD*)cal->GetResponse();
301     const char *option=res->ZeroSuppOption();
302     Int_t q=input->GetSignal();
303     if(!((strstr(option,"1D")) || (strstr(option,"2D")))){
304       Float_t baseline = cal->GetBaseline(input->GetCoord1());
305        if(q>baseline) q-=(Int_t)baseline;
306        else q=0;
307      }
308     if(q>=cal->GetThresholdAnode(input->GetCoord1())) {
309       Int_t iz = input->GetCoord1()+1;
310       //Int_t side = ((iz <= fNzSDD) ? 0 : 1);
311       Int_t side = ((AliITSRawStreamSDD*)input)->GetChannel();
312       //  iz -= side*fNzSDD;
313       Int_t index = (input->GetCoord2()+1) * kNzBins + iz;
314       bins[side][index].SetQ(q);
315       bins[side][index].SetMask(1);
316       bins[side][index].SetIndex(index);
317     }
318   }
319   delete[] binsSDD1;
320   delete[] binsSDD2;
321   delete[] binsSDDInit;
322
323   Info("FindClustersSDD", "found clusters in ITS SDD: %d", nClustersSDD);
324 }
325
326
327 //_________________________________________________________________________
328 void AliITSClusterFinderV2SDD::CorrectPosition(Float_t &z, Float_t&y){
329
330   //correction of coordinates using the maps stored in the DB
331
332   AliITSCalibrationSDD* cal = (AliITSCalibrationSDD*)GetResp(fModule);
333   static const Int_t knbint = cal->GetMapTimeNBin();
334   static const Int_t knbina = cal->Chips()*cal->Channels();
335   Float_t stepa = (GetSeg()->Dpz(0))/10000.; //anode pitch in cm
336   Float_t stept = (GetSeg()->Dx()/cal->GetMapTimeNBin()/2.)/10.;
337   
338   Int_t bint = TMath::Abs((Int_t)(y/stept));
339   if(y>=0) bint+=(Int_t)(knbint/2.);
340   if(bint>knbint) AliError("Wrong bin number!");
341
342   Int_t bina = TMath::Abs((Int_t)(z/stepa));
343   if(z>=0) bina+=(Int_t)(knbina/2.);
344   if(bina>knbina) AliError("Wrong bin number!");
345
346   Float_t devz = cal->GetMapACell(bina,bint)/10000.;
347   Float_t devx = cal->GetMapTCell(bina,bint)/10000.;
348   z+=devz;
349   y+=devx;
350
351
352 }