1 /**************************************************************************
2 * Copyright(c) 1998-2003, ALICE Experiment at CERN, All rights reserved. *
4 * Author: The ALICE Off-line Project. *
5 * Contributors are mentioned in the code where appropriate. *
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 // Implementation of the ITS clusterer V2 class //
18 // Origin: Iouri Belikov, CERN, Jouri.Belikov@cern.ch //
19 // Unfolding switch from AliITSRecoParam: D. Elia, INFN Bari //
21 ////////////////////////////////////////////////////////////////////////////
24 #include "AliITSCalibrationSPD.h"
25 #include "AliITSClusterFinderV2SPD.h"
26 #include "AliITSRecPoint.h"
27 #include "AliITSgeomTGeo.h"
28 #include "AliITSDetTypeRec.h"
29 #include "AliITSReconstructor.h"
30 #include "AliRawReader.h"
31 #include "AliITSRawStreamSPD.h"
32 #include <TClonesArray.h>
33 #include "AliITSdigitSPD.h"
35 ClassImp(AliITSClusterFinderV2SPD)
37 AliITSClusterFinderV2SPD::AliITSClusterFinderV2SPD(AliITSDetTypeRec* dettyp):AliITSClusterFinderV2(dettyp),
38 fLastSPD1(AliITSgeomTGeo::GetModuleIndex(2,1,1)-1),
49 fYSPD[0]=0.5*fYpitchSPD;
50 for (Int_t m=1; m<fNySPD; m++) fYSPD[m]=fYSPD[m-1]+fYpitchSPD;
52 for (Int_t m=1; m<fNzSPD; m++) {
53 Double_t dz=fZ1pitchSPD;
54 if (m==31 || m==32 || m==63 || m==64 || m==95 || m==96 ||
55 m==127 || m==128) dz=fZ2pitchSPD;
56 fZSPD[m]=fZSPD[m-1]+dz;
58 for (Int_t m=0; m<fNzSPD; m++) {
59 Double_t dz=0.5*fZ1pitchSPD;
60 if (m==31 || m==32 || m==63 || m==64 || m==95 || m==96 ||
61 m==127 || m==128) dz=0.5*fZ2pitchSPD;
68 void AliITSClusterFinderV2SPD::FindRawClusters(Int_t mod){
72 FindClustersSPD(fDigits);
76 void AliITSClusterFinderV2SPD::RawdataToClusters(AliRawReader* rawReader, TClonesArray** clusters){
77 //------------------------------------------------------------
78 // This function creates ITS clusters from raw data
79 //------------------------------------------------------------
81 AliITSRawStreamSPD inputSPD(rawReader);
82 FindClustersSPD(&inputSPD, clusters);
86 Int_t AliITSClusterFinderV2SPD::ClustersSPD(AliBin* bins, TClonesArray* digits,TClonesArray* clusters,Int_t maxBins,Int_t nzbins,Int_t iModule,Bool_t rawdata){
88 //Cluster finder for SPD (from digits and from rawdata)
90 static AliITSRecoParam *repa = NULL;
92 repa = (AliITSRecoParam*) AliITSReconstructor::GetRecoParam();
94 repa = AliITSRecoParam::GetHighFluxParam();
95 AliWarning("Using default AliITSRecoParam class");
98 const TGeoHMatrix *mT2L=AliITSgeomTGeo::GetTracking2LocalMatrix(iModule);
100 AliITSCalibrationSPD *cal =
101 (AliITSCalibrationSPD*) fDetTypeRec->GetCalibrationModel(iModule);
102 // Loop on bad pixels and reset them
103 for(Int_t ipix = 0; ipix<cal->GetNrBad(); ipix++){
105 cal->GetBadPixel(ipix,row,col);
106 Int_t index = (row+1) * nzbins + (col+1);
108 bins[index].SetMask(0xFFFFFFFE);
111 for(Int_t iBin =0; iBin < maxBins;iBin++){
112 if(bins[iBin].IsUsed()) continue;
115 FindCluster(iBin,nzbins,bins,nBins,idxBins);
117 Error("ClustersSPD","SPD Too big cluster !\n");
121 for(Int_t ilab=0;ilab<10;ilab++){
125 milab[3]=fNdet[iModule];
127 Int_t ymin,ymax,zmin,zmax;
129 ymin = (idxBins[0] / nzbins) - 1;
131 zmin = (idxBins[0] % nzbins) - 1;
135 AliITSdigitSPD* dig = (AliITSdigitSPD*)digits->UncheckedAt(idxBins[0]);
136 ymin=dig->GetCoord2();
138 zmin=dig->GetCoord1();
141 for (Int_t idx = 0; idx < nBins; idx++) {
145 iy = (idxBins[idx] / nzbins) - 1;
146 iz = (idxBins[idx] % nzbins) - 1;
149 AliITSdigitSPD* dig = (AliITSdigitSPD*)digits->UncheckedAt(idxBins[idx]);
150 iy = dig->GetCoord2();
151 iz = dig->GetCoord1();
153 if (ymin > iy) ymin = iy;
154 if (ymax < iy) ymax = iy;
155 if (zmin > iz) zmin = iz;
156 if (zmax < iz) zmax = iz;
160 for(Int_t l=0;l<nBins;l++){
161 AliITSdigitSPD* dig = (AliITSdigitSPD*)digits->UncheckedAt(idxBins[l]);
162 for(Int_t dlab=0;dlab<10;dlab++){
163 Int_t digitlab = (dig->GetTracks())[dlab];
164 if(digitlab<0) continue;
165 AddLabel(milab,digitlab);
167 if (milab[9]>0) CheckLabels2(milab);
172 Int_t idy =0; //max 2 clusters
173 if((iModule <= fLastSPD1) &&idy<3) idy=3;
174 if((iModule > fLastSPD1) &&idy<4) idy=4;
177 // Switch the unfolding OFF/ON
178 if(!repa->GetUseUnfoldingInClusterFinderSPD()) {
183 for(Int_t iiz=zmin; iiz<=zmax;iiz+=idz){
184 for(Int_t iiy=ymin;iiy<=ymax;iiy+=idy){
187 Float_t y=0.,z=0.,q=0.;
188 for(Int_t idx=0;idx<nBins;idx++){
192 iy = (idxBins[idx] / nzbins)-1;
193 iz = (idxBins[idx] % nzbins)-1;
196 AliITSdigitSPD* dig = (AliITSdigitSPD*)digits->UncheckedAt(idxBins[idx]);
197 iy = dig->GetCoord2();
198 iz = dig->GetCoord1();
200 if(zmax-zmin>=idz || ymax-ymin>=idy){
201 if(TMath::Abs(iy-iiy)>0.75*idy) continue;
202 if(TMath::Abs(iz-iiz)>0.75*idz) continue;
206 if(rawdata) qBin = bins[idxBins[idx]].GetQ();
208 AliITSdigitSPD* dig = (AliITSdigitSPD*)digits->UncheckedAt(idxBins[idx]);
209 qBin = (Float_t)dig->GetSignal();
211 y+= qBin * fYSPD[iy];
212 z+= qBin * fZSPD[iz];
215 if(ndigits==0) continue;
221 Float_t hit[5]; //y,z,sigma(y)^2, sigma(z)^2, charge
223 Double_t loc[3]={y,0.,z},trk[3]={0.,0.,0.};
224 mT2L->MasterToLocal(loc,trk);
228 hit[2] = fYpitchSPD*fYpitchSPD/12.;
229 hit[3] = fZ1pitchSPD*fZ1pitchSPD/12.;
232 if(!rawdata) milab[3]=fNdet[iModule];
233 Int_t info[3] = {ymax-ymin+1,zmax-zmin+1,fNlayer[iModule]};
235 AliITSRecPoint cl(milab,hit,info);
236 fDetTypeRec->AddRecPoint(cl);
239 Int_t label[4]={milab[0],milab[1],milab[2],milab[3]};
240 new (clusters->AddrAt(nclu))
241 AliITSRecPoint(label, hit,info);
254 void AliITSClusterFinderV2SPD::FindClustersSPD(AliITSRawStream* input,
255 TClonesArray** clusters)
257 //------------------------------------------------------------
258 // Actual SPD cluster finder for raw data
259 //------------------------------------------------------------
260 Int_t nClustersSPD = 0;
261 Int_t kNzBins = fNzSPD + 2;
262 Int_t kNyBins = fNySPD + 2;
263 Int_t kMaxBin = kNzBins * kNyBins;
264 AliBin *binsSPD = new AliBin[kMaxBin];
265 AliBin *binsSPDInit = new AliBin[kMaxBin];
268 // read raw data input stream
270 Bool_t next = input->Next();
271 if (!next || input->IsNewModule()) {
272 Int_t iModule = input->GetPrevModuleID();
274 // when all data from a module was read, search for clusters
276 clusters[iModule] = new TClonesArray("AliITSRecPoint");
277 Int_t nClusters = ClustersSPD(bins,0,clusters[iModule],kMaxBin,kNzBins,iModule,kTRUE);
278 nClustersSPD += nClusters;
284 memcpy(binsSPD,binsSPDInit,sizeof(AliBin)*kMaxBin);
287 // fill the current digit into the bins array
288 Int_t index = (input->GetCoord2()+1) * kNzBins + (input->GetCoord1()+1);
289 bins[index].SetIndex(index);
290 bins[index].SetMask(1);
294 delete [] binsSPDInit;
297 AliDebug(1,Form("found clusters in ITS SPD: %d", nClustersSPD));
302 void AliITSClusterFinderV2SPD::FindClustersSPD(TClonesArray *digits) {
303 //------------------------------------------------------------
304 // Actual SPD cluster finder
305 //------------------------------------------------------------
308 Int_t kNzBins = fNzSPD + 2;
309 const Int_t kMAXBIN=kNzBins*(fNySPD+2);
311 Int_t ndigits=digits->GetEntriesFast();
312 AliBin *bins=new AliBin[kMAXBIN];
316 for (k=0; k<ndigits; k++) {
317 d=(AliITSdigitSPD*)digits->UncheckedAt(k);
318 Int_t i=d->GetCoord2()+1; //y
319 Int_t j=d->GetCoord1()+1;
320 Int_t index=i*kNzBins+j;
321 bins[index].SetIndex(k);
322 bins[index].SetMask(1);
325 Int_t nClustersSPD = ClustersSPD(bins,digits,0,kMAXBIN,kNzBins,fModule,kFALSE);
328 AliDebug(1,Form("found clusters in ITS SPD: %d", nClustersSPD));