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 //
20 ///////////////////////////////////////////////////////////////////////////
23 #include "AliITSClusterFinderV2SPD.h"
24 #include "AliITSRecPoint.h"
25 #include "AliITSgeomTGeo.h"
26 #include "AliITSDetTypeRec.h"
27 #include "AliRawReader.h"
28 #include "AliITSRawStreamSPD.h"
29 #include <TClonesArray.h>
30 #include "AliITSdigitSPD.h"
32 ClassImp(AliITSClusterFinderV2SPD)
34 AliITSClusterFinderV2SPD::AliITSClusterFinderV2SPD(AliITSDetTypeRec* dettyp):AliITSClusterFinderV2(dettyp),
35 fLastSPD1(AliITSgeomTGeo::GetModuleIndex(2,1,1)-1),
46 fYSPD[0]=0.5*fYpitchSPD;
47 for (Int_t m=1; m<fNySPD; m++) fYSPD[m]=fYSPD[m-1]+fYpitchSPD;
49 for (Int_t m=1; m<fNzSPD; m++) {
50 Double_t dz=fZ1pitchSPD;
51 if (m==31 || m==32 || m==63 || m==64 || m==95 || m==96 ||
52 m==127 || m==128) dz=fZ2pitchSPD;
53 fZSPD[m]=fZSPD[m-1]+dz;
55 for (Int_t m=0; m<fNzSPD; m++) {
56 Double_t dz=0.5*fZ1pitchSPD;
57 if (m==31 || m==32 || m==63 || m==64 || m==95 || m==96 ||
58 m==127 || m==128) dz=0.5*fZ2pitchSPD;
65 void AliITSClusterFinderV2SPD::FindRawClusters(Int_t mod){
69 FindClustersSPD(fDigits);
73 void AliITSClusterFinderV2SPD::RawdataToClusters(AliRawReader* rawReader, TClonesArray** clusters){
74 //------------------------------------------------------------
75 // This function creates ITS clusters from raw data
76 //------------------------------------------------------------
78 AliITSRawStreamSPD inputSPD(rawReader);
79 FindClustersSPD(&inputSPD, clusters);
83 Int_t AliITSClusterFinderV2SPD::ClustersSPD(AliBin* bins, TClonesArray* digits,TClonesArray* clusters,Int_t maxBins,Int_t nzbins,Int_t iModule,Bool_t rawdata){
85 //Cluster finder for SPD (from digits and from rawdata)
88 for(Int_t iBin =0; iBin < maxBins;iBin++){
89 if(bins[iBin].IsUsed()) continue;
92 FindCluster(iBin,nzbins,bins,nBins,idxBins);
94 Error("ClustersSPD","SPD Too big cluster !\n");
98 for(Int_t ilab=0;ilab<10;ilab++){
102 milab[3]=fNdet[iModule];
104 Int_t ymin,ymax,zmin,zmax;
106 ymin = (idxBins[0] / nzbins) - 1;
108 zmin = (idxBins[0] % nzbins) - 1;
112 AliITSdigitSPD* dig = (AliITSdigitSPD*)digits->UncheckedAt(idxBins[0]);
113 ymin=dig->GetCoord2();
115 zmin=dig->GetCoord1();
118 for (Int_t idx = 0; idx < nBins; idx++) {
122 iy = (idxBins[idx] / nzbins) - 1;
123 iz = (idxBins[idx] % nzbins) - 1;
126 AliITSdigitSPD* dig = (AliITSdigitSPD*)digits->UncheckedAt(idxBins[idx]);
127 iy = dig->GetCoord2();
128 iz = dig->GetCoord1();
130 if (ymin > iy) ymin = iy;
131 if (ymax < iy) ymax = iy;
132 if (zmin > iz) zmin = iz;
133 if (zmax < iz) zmax = iz;
137 for(Int_t l=0;l<nBins;l++){
138 AliITSdigitSPD* dig = (AliITSdigitSPD*)digits->UncheckedAt(idxBins[l]);
139 for(Int_t dlab=0;dlab<10;dlab++){
140 Int_t digitlab = (dig->GetTracks())[dlab];
141 if(digitlab<0) continue;
142 AddLabel(milab,digitlab);
144 if (milab[9]>0) CheckLabels2(milab);
149 Int_t idy =0; //max 2 clusters
150 if((iModule <= fLastSPD1) &&idy<3) idy=3;
151 if((iModule > fLastSPD1) &&idy<4) idy=4;
153 for(Int_t iiz=zmin; iiz<=zmax;iiz+=idz){
154 for(Int_t iiy=ymin;iiy<=ymax;iiy+=idy){
157 Float_t y=0.,z=0.,q=0.;
158 for(Int_t idx=0;idx<nBins;idx++){
162 iy = (idxBins[idx] / nzbins)-1;
163 iz = (idxBins[idx] % nzbins)-1;
166 AliITSdigitSPD* dig = (AliITSdigitSPD*)digits->UncheckedAt(idxBins[idx]);
167 iy = dig->GetCoord2();
168 iz = dig->GetCoord1();
170 if(zmax-zmin>=idz || ymax-ymin>=idy){
171 if(TMath::Abs(iy-iiy)>0.75*idy) continue;
172 if(TMath::Abs(iz-iiz)>0.75*idz) continue;
176 if(rawdata) qBin = bins[idxBins[idx]].GetQ();
178 AliITSdigitSPD* dig = (AliITSdigitSPD*)digits->UncheckedAt(idxBins[idx]);
179 qBin = (Float_t)dig->GetSignal();
181 y+= qBin * fYSPD[iy];
182 z+= qBin * fZSPD[iz];
185 if(ndigits==0) continue;
190 Float_t hit[5]; //y,z,sigma(y)^2, sigma(z)^2, charge
192 if (iModule <= fLastSPD1) hit[0] = -y+fYshift[iModule];
193 else hit[0] = y+fYshift[iModule];
195 hit[1] = -z+fZshift[iModule];
196 hit[2] = fYpitchSPD*fYpitchSPD/12.;
197 hit[3] = fZ1pitchSPD*fZ1pitchSPD/12.;
199 if(!rawdata) milab[3]=fNdet[iModule];
200 Int_t info[3] = {ymax-ymin+1,zmax-zmin+1,fNlayer[iModule]};
202 AliITSRecPoint cl(milab,hit,info);
203 fDetTypeRec->AddRecPoint(cl);
206 Int_t label[4]={milab[0],milab[1],milab[2],milab[3]};
207 new (clusters->AddrAt(nclu))
208 AliITSRecPoint(label, hit,info);
221 void AliITSClusterFinderV2SPD::FindClustersSPD(AliITSRawStream* input,
222 TClonesArray** clusters)
224 //------------------------------------------------------------
225 // Actual SPD cluster finder for raw data
226 //------------------------------------------------------------
227 Int_t nClustersSPD = 0;
228 Int_t kNzBins = fNzSPD + 2;
229 Int_t kNyBins = fNySPD + 2;
230 Int_t kMaxBin = kNzBins * kNyBins;
231 AliBin *binsSPD = new AliBin[kMaxBin];
232 AliBin *binsSPDInit = new AliBin[kMaxBin];
235 // read raw data input stream
237 Bool_t next = input->Next();
238 if (!next || input->IsNewModule()) {
239 Int_t iModule = input->GetPrevModuleID();
241 // when all data from a module was read, search for clusters
243 clusters[iModule] = new TClonesArray("AliITSRecPoint");
244 Int_t nClusters = ClustersSPD(bins,0,clusters[iModule],kMaxBin,kNzBins,iModule,kTRUE);
245 nClustersSPD += nClusters;
251 memcpy(binsSPD,binsSPDInit,sizeof(AliBin)*kMaxBin);
254 // fill the current digit into the bins array
255 Int_t index = (input->GetCoord2()+1) * kNzBins + (input->GetCoord1()+1);
256 bins[index].SetIndex(index);
257 bins[index].SetMask(1);
261 delete [] binsSPDInit;
264 Info("FindClustersSPD", "found clusters in ITS SPD: %d", nClustersSPD);
269 void AliITSClusterFinderV2SPD::FindClustersSPD(TClonesArray *digits) {
270 //------------------------------------------------------------
271 // Actual SPD cluster finder
272 //------------------------------------------------------------
275 Int_t kNzBins = fNzSPD + 2;
276 const Int_t kMAXBIN=kNzBins*(fNySPD+2);
278 Int_t ndigits=digits->GetEntriesFast();
279 AliBin *bins=new AliBin[kMAXBIN];
283 for (k=0; k<ndigits; k++) {
284 d=(AliITSdigitSPD*)digits->UncheckedAt(k);
285 Int_t i=d->GetCoord2()+1; //y
286 Int_t j=d->GetCoord1()+1;
287 Int_t index=i*kNzBins+j;
288 bins[index].SetIndex(k);
289 bins[index].SetMask(1);
292 ClustersSPD(bins,digits,0,kMAXBIN,kNzBins,fModule,kFALSE);