1 //-------------------------------------------------------------------------
2 // Implementation of the ITS clusterer V2 class
4 // Origin: Iouri Belikov, CERN, Jouri.Belikov@cern.ch
5 //-------------------------------------------------------------------------
6 //uncomment the line below for running with V1 cluster finder classes
11 #include "AliITSclustererV2.h"
12 #include "AliITSclusterV2.h"
13 #include "AliRawReader.h"
14 #include "AliITSRawStreamSPD.h"
15 #include "AliITSRawStreamSDD.h"
16 #include "AliITSRawStreamSSD.h"
20 #include <TClonesArray.h>
21 #include "AliITSgeom.h"
22 #include "AliITSdigit.h"
24 ClassImp(AliITSclustererV2)
26 extern AliRun *gAlice;
28 AliITSclustererV2::AliITSclustererV2(const AliITSgeom *geom) {
29 //------------------------------------------------------------
30 // Standard constructor
31 //------------------------------------------------------------
32 AliITSgeom *g=(AliITSgeom*)geom;
37 Int_t mmax=geom->GetIndexMax();
39 Fatal("AliITSclustererV2","Too many ITS subdetectors !");
42 for (m=0; m<mmax; m++) {
43 Int_t lay,lad,det; g->GetModuleId(m,lay,lad,det);
44 Float_t x,y,z; g->GetTrans(lay,lad,det,x,y,z);
45 Double_t rot[9]; g->GetRotMatrix(lay,lad,det,rot);
46 fYshift[m] = x*rot[0] + y*rot[1];
47 fZshift[m] = (Double_t)z;
48 fNdet[m] = (lad-1)*g->GetNdetectors(lay) + (det-1);
50 fNModules = g->GetIndexMax();
53 fLastSPD1=g->GetModuleIndex(2,1,1)-1;
54 fNySPD=256; fNzSPD=160;
56 fZ1pitchSPD=0.0425; fZ2pitchSPD=0.0625;
57 fHwSPD=0.64; fHlSPD=3.48;
58 fYSPD[0]=0.5*fYpitchSPD;
59 for (m=1; m<fNySPD; m++) fYSPD[m]=fYSPD[m-1]+fYpitchSPD;
61 for (m=1; m<fNzSPD; m++) {
62 Double_t dz=fZ1pitchSPD;
63 if (m==31 || m==32 || m==63 || m==64 || m==95 || m==96 ||
64 m==127 || m==128) dz=fZ2pitchSPD;
65 fZSPD[m]=fZSPD[m-1]+dz;
67 for (m=0; m<fNzSPD; m++) {
68 Double_t dz=0.5*fZ1pitchSPD;
69 if (m==31 || m==32 || m==63 || m==64 || m==95 || m==96 ||
70 m==127 || m==128) dz=0.5*fZ2pitchSPD;
75 fNySDD=256; fNzSDD=256;
78 fHwSDD=3.5085; fHlSDD=3.7632;
82 fLastSSD1=g->GetModuleIndex(6,1,1)-1;
90 Int_t AliITSclustererV2::Digits2Clusters(TTree *dTree, TTree *cTree) {
91 //------------------------------------------------------------
92 // This function creates ITS clusters
93 //------------------------------------------------------------
97 Error("Digits2Clusters","Can't get the tree with digits !");
101 TClonesArray *digitsSPD=new TClonesArray("AliITSdigitSPD",3000);
102 dTree->SetBranchAddress("ITSDigitsSPD",&digitsSPD);
103 TClonesArray *digitsSDD=new TClonesArray("AliITSdigitSDD",3000);
104 dTree->SetBranchAddress("ITSDigitsSDD",&digitsSDD);
105 TClonesArray *digitsSSD=new TClonesArray("AliITSdigitSSD",3000);
106 dTree->SetBranchAddress("ITSDigitsSSD",&digitsSSD);
108 TClonesArray *clusters=new TClonesArray("AliITSclusterV2",1000);
109 TBranch *branch=cTree->GetBranch("Clusters");
110 if (!branch) cTree->Branch("Clusters",&clusters);
111 else branch->SetAddress(&clusters);
113 Int_t mmax=(Int_t)dTree->GetEntries();
115 for (fI=0; fI<mmax; fI++) {
118 if (digitsSPD->GetEntriesFast()!=0)
119 FindClustersSPD(digitsSPD,clusters);
120 else if(digitsSDD->GetEntriesFast()!=0)
121 FindClustersSDD(digitsSDD,clusters);
122 else if(digitsSSD->GetEntriesFast()!=0)
123 FindClustersSSD(digitsSSD,clusters);
125 ncl+=clusters->GetEntriesFast();
145 Info("Digits2Clusters","Number of found clusters : %d",ncl);
150 void AliITSclustererV2::Digits2Clusters(AliRawReader* rawReader) {
151 //------------------------------------------------------------
152 // This function creates ITS clusters from raw data
153 //------------------------------------------------------------
154 AliRunLoader* runLoader = AliRunLoader::GetRunLoader();
156 Error("Digits2Clusters", "no run loader found");
159 AliLoader* itsLoader = runLoader->GetLoader("ITSLoader");
161 Error("Digits2Clusters", "no loader for ITS found");
164 if (!itsLoader->TreeR()) itsLoader->MakeTree("R");
165 TTree* cTree = itsLoader->TreeR();
167 TClonesArray *array=new TClonesArray("AliITSclusterV2",1000);
168 cTree->Branch("Clusters",&array);
171 TClonesArray** clusters = new TClonesArray*[fNModules];
172 // one TClonesArray per module
175 AliITSRawStreamSPD inputSPD(rawReader);
176 FindClustersSPD(&inputSPD, clusters);
179 AliITSRawStreamSDD inputSDD(rawReader);
180 FindClustersSDD(&inputSDD, clusters);
183 AliITSRawStreamSSD inputSSD(rawReader);
184 FindClustersSSD(&inputSSD, clusters);
186 // write all clusters to the tree
188 for (Int_t iModule = 0; iModule < fNModules; iModule++) {
189 TClonesArray* array = clusters[iModule];
191 Error("Digits2Clusters", "data for module %d missing!", iModule);
192 array = new TClonesArray("AliITSclusterV2");
194 cTree->SetBranchAddress("Clusters", &array);
196 nClusters += array->GetEntriesFast();
199 itsLoader->WriteRecPoints("OVERWRITE");
201 Info("Digits2Clusters", "total number of found clusters in ITS: %d\n",
205 //**** Fast clusters *******************************
206 #include "TParticle.h"
209 #include "AliITSmodule.h"
210 #include "AliITSRecPoint.h"
211 #include "AliITSsimulationFastPoints.h"
212 #include "AliITSRecPoint.h"
214 static void CheckLabels(Int_t lab[3]) {
215 //------------------------------------------------------------
216 // Tries to find mother's labels
217 //------------------------------------------------------------
220 TParticle *part=(TParticle*)gAlice->Particle(label);
222 while (part->P() < 0.005) {
223 Int_t m=part->GetFirstMother();
225 Info("CheckLabels","Primary momentum: %f",part->P());
228 if (part->GetStatusCode()>0) {
229 Info("CheckLabels","Primary momentum: %f",part->P());
233 part=(TParticle*)gAlice->Particle(label);
235 if (lab[1]<0) lab[1]=label;
236 else if (lab[2]<0) lab[2]=label;
237 else ;//cerr<<"CheckLabels : No empty labels !\n";
241 void AliITSclustererV2::RecPoints2Clusters
242 (const TClonesArray *points, Int_t idx, TClonesArray *clusters) {
243 //------------------------------------------------------------
244 // Conversion AliITSRecPoint -> AliITSclusterV2 for the ITS
245 // subdetector indexed by idx
246 //------------------------------------------------------------
247 TClonesArray &cl=*clusters;
248 Int_t ncl=points->GetEntriesFast();
249 for (Int_t i=0; i<ncl; i++) {
250 AliITSRecPoint *p = (AliITSRecPoint *)points->UncheckedAt(i);
252 lp[0]=-p->GetX()-fYshift[idx]; if (idx<=fLastSPD1) lp[0]*=-1; //SPD1
253 lp[1]=p->GetZ()+fZshift[idx];
254 lp[2]=p->GetSigmaX2();
255 lp[3]=p->GetSigmaZ2();
258 lab[0]=p->GetLabel(0); lab[1]=p->GetLabel(1); lab[2]=p->GetLabel(2);
261 new (cl[i]) AliITSclusterV2(lab,lp);
265 Int_t AliITSclustererV2::Hits2Clusters(TTree *hTree, TTree *cTree) {
266 //------------------------------------------------------------
267 // This function creates ITS clusters
268 //------------------------------------------------------------
270 Error("Hits2Clusters","gAlice==0 !");
274 AliITS *its = (AliITS*)gAlice->GetModule("ITS");
276 Error("Hits2Clusters","Can't find the ITS !");
279 AliITSgeom *geom=its->GetITSgeom();
280 Int_t mmax=geom->GetIndexMax();
282 its->InitModules(-1,mmax);
283 its->FillModules(hTree,0);
285 TClonesArray *clusters=new TClonesArray("AliITSclusterV2",1000);
286 TBranch *branch=cTree->GetBranch("Clusters");
287 if (!branch) cTree->Branch("Clusters",&clusters);
288 else branch->SetAddress(&clusters);
290 static TClonesArray *points=its->RecPoints();
291 AliITSsimulationFastPoints sim;
293 for (Int_t m=0; m<mmax; m++) {
294 AliITSmodule *mod=its->GetModule(m);
295 sim.CreateFastRecPoints(mod,m,gRandom);
297 RecPoints2Clusters(points, m, clusters);
298 its->ResetRecPoints();
300 ncl+=clusters->GetEntriesFast();
305 Info("Hits2Clusters","Number of found fast clusters : %d",ncl);
314 //***********************************
318 void AliITSclustererV2::
319 FindCluster(Int_t k,Int_t maxz,AliBin *bins,Int_t &n,Int_t *idx) {
320 //------------------------------------------------------------
321 // returns an array of indices of digits belonging to the cluster
322 // (needed when the segmentation is not regular)
323 //------------------------------------------------------------
324 if (n<200) idx[n++]=bins[k].GetIndex();
327 if (bins[k-maxz].IsNotUsed()) FindCluster(k-maxz,maxz,bins,n,idx);
328 if (bins[k-1 ].IsNotUsed()) FindCluster(k-1 ,maxz,bins,n,idx);
329 if (bins[k+maxz].IsNotUsed()) FindCluster(k+maxz,maxz,bins,n,idx);
330 if (bins[k+1 ].IsNotUsed()) FindCluster(k+1 ,maxz,bins,n,idx);
332 if (bins[k-maxz-1].IsNotUsed()) FindCluster(k-maxz-1,maxz,bins,n,idx);
333 if (bins[k-maxz+1].IsNotUsed()) FindCluster(k-maxz+1,maxz,bins,n,idx);
334 if (bins[k+maxz-1].IsNotUsed()) FindCluster(k+maxz-1,maxz,bins,n,idx);
335 if (bins[k+maxz+1].IsNotUsed()) FindCluster(k+maxz+1,maxz,bins,n,idx);
339 void AliITSclustererV2::
340 FindClustersSPD(const TClonesArray *digits, TClonesArray *clusters) {
341 //------------------------------------------------------------
342 // Actual SPD cluster finder
343 //------------------------------------------------------------
344 Int_t kNzBins = fNzSPD + 2;
345 const Int_t kMAXBIN=kNzBins*(fNySPD+2);
347 Int_t ndigits=digits->GetEntriesFast();
348 AliBin *bins=new AliBin[kMAXBIN];
352 for (k=0; k<ndigits; k++) {
353 d=(AliITSdigitSPD*)digits->UncheckedAt(k);
354 Int_t i=d->GetCoord2()+1; //y
355 Int_t j=d->GetCoord1()+1;
356 bins[i*kNzBins+j].SetIndex(k);
357 bins[i*kNzBins+j].SetMask(1);
360 Int_t n=0; TClonesArray &cl=*clusters;
361 for (k=0; k<kMAXBIN; k++) {
362 if (!bins[k].IsNotUsed()) continue;
363 Int_t ni=0, idx[200];
364 FindCluster(k,kNzBins,bins,ni,idx);
366 Info("FindClustersSPD","Too big cluster !");
375 d=(AliITSdigitSPD*)digits->UncheckedAt(idx[0]);
376 Int_t ymin=d->GetCoord2(),ymax=ymin;
377 Int_t zmin=d->GetCoord1(),zmax=zmin;
378 Float_t y=0.,z=0.,q=0.;
379 for (Int_t l=0; l<ni; l++) {
380 d=(AliITSdigitSPD*)digits->UncheckedAt(idx[l]);
382 if (ymin > d->GetCoord2()) ymin=d->GetCoord2();
383 if (ymax < d->GetCoord2()) ymax=d->GetCoord2();
384 if (zmin > d->GetCoord1()) zmin=d->GetCoord1();
385 if (zmax < d->GetCoord1()) zmax=d->GetCoord1();
387 Int_t lab0=(d->GetTracks())[0];
391 } else if (lab[1]<0) {
392 if (lab0!=lab[0]) lab[1]=lab0;
393 } else if (lab[2]<0) {
395 if (lab0!=lab[1]) lab[2]=lab0;
398 Float_t qq=d->GetSignal();
399 y+=qq*fYSPD[d->GetCoord2()]; z+=qq*fZSPD[d->GetCoord1()]; q+=qq;
402 y-=fHwSPD; z-=fHlSPD;
405 lp[0]=-y-fYshift[fI]; if (fI<=fLastSPD1) lp[0]=-lp[0];
406 lp[1]= z+fZshift[fI];
407 lp[2]= fYpitchSPD*fYpitchSPD/12.;
408 lp[3]= fZ1pitchSPD*fZ1pitchSPD/12.;
410 lp[4]= (zmax-zmin+1)*100 + (ymax-ymin+1);
413 d=(AliITSdigitSPD*)digits->UncheckedAt(idx[0]);
414 new (cl[n]) AliITSclusterV2(lab,lp); n++;
420 void AliITSclustererV2::FindClustersSPD(AliITSRawStream* input,
421 TClonesArray** clusters)
423 //------------------------------------------------------------
424 // Actual SPD cluster finder for raw data
425 //------------------------------------------------------------
427 Int_t nClustersSPD = 0;
428 Int_t kNzBins = fNzSPD + 2;
429 Int_t kNyBins = fNySPD + 2;
430 Int_t kMaxBin = kNzBins * kNyBins;
433 // read raw data input stream
435 Bool_t next = input->Next();
436 if (!next || input->IsNewModule()) {
437 Int_t iModule = input->GetPrevModuleID();
439 // when all data from a module was read, search for clusters
441 clusters[iModule] = new TClonesArray("AliITSclusterV2");
444 for (Int_t iBin = 0; iBin < kMaxBin; iBin++) {
445 if (bins[iBin].IsUsed()) continue;
448 FindCluster(iBin, kNzBins, bins, nBins, idxBins);
450 Error("FindClustersSPD", "SPD: Too big cluster !\n");
458 // label[3] = iModule;
459 label[3] = fNdet[iModule];
461 Int_t ymin = (idxBins[0] / kNzBins) - 1;
463 Int_t zmin = (idxBins[0] % kNzBins) - 1;
468 for (Int_t idx = 0; idx < nBins; idx++) {
469 Int_t iy = (idxBins[idx] / kNzBins) - 1;
470 Int_t iz = (idxBins[idx] % kNzBins) - 1;
471 if (ymin > iy) ymin = iy;
472 if (ymax < iy) ymax = iy;
473 if (zmin > iz) zmin = iz;
474 if (zmax < iz) zmax = iz;
476 Float_t qBin = bins[idxBins[idx]].GetQ();
477 y += qBin * fYSPD[iy];
478 z += qBin * fZSPD[iz];
486 Float_t hit[5]; // y, z, sigma(y)^2, sigma(z)^2, charge
487 hit[0] = -y-fYshift[iModule];
488 if (iModule <= fLastSPD1) hit[0] = -hit[0];
489 hit[1] = z+fZshift[iModule];
490 hit[2] = fYpitchSPD*fYpitchSPD/12.;
491 hit[3] = fZ1pitchSPD*fZ1pitchSPD/12.;
493 hit[4] = (zmax-zmin+1)*100 + (ymax-ymin+1);
495 //CheckLabels(label);
496 new (clusters[iModule]->AddrAt(nClusters))
497 AliITSclusterV2(label, hit);
501 nClustersSPD += nClusters;
506 bins = new AliBin[kMaxBin];
509 // fill the current digit into the bins array
510 Int_t index = (input->GetCoord2()+1) * kNzBins + (input->GetCoord1()+1);
511 bins[index].SetIndex(index);
512 bins[index].SetMask(1);
516 Info("FindClustersSPD", "found clusters in ITS SPD: %d", nClustersSPD);
520 Bool_t AliITSclustererV2::IsMaximum(Int_t k,Int_t max,const AliBin *bins) {
521 //------------------------------------------------------------
522 //is this a local maximum ?
523 //------------------------------------------------------------
524 UShort_t q=bins[k].GetQ();
525 if (q==1023) return kFALSE;
526 if (bins[k-max].GetQ() > q) return kFALSE;
527 if (bins[k-1 ].GetQ() > q) return kFALSE;
528 if (bins[k+max].GetQ() > q) return kFALSE;
529 if (bins[k+1 ].GetQ() > q) return kFALSE;
530 if (bins[k-max-1].GetQ() > q) return kFALSE;
531 if (bins[k+max-1].GetQ() > q) return kFALSE;
532 if (bins[k+max+1].GetQ() > q) return kFALSE;
533 if (bins[k-max+1].GetQ() > q) return kFALSE;
537 void AliITSclustererV2::
538 FindPeaks(Int_t k,Int_t max,AliBin *b,Int_t *idx,UInt_t *msk,Int_t& n) {
539 //------------------------------------------------------------
541 //------------------------------------------------------------
543 if (IsMaximum(k,max,b)) {
544 idx[n]=k; msk[n]=(2<<n);
548 if (b[k-max].GetMask()&1) FindPeaks(k-max,max,b,idx,msk,n);
549 if (b[k-1 ].GetMask()&1) FindPeaks(k-1 ,max,b,idx,msk,n);
550 if (b[k+max].GetMask()&1) FindPeaks(k+max,max,b,idx,msk,n);
551 if (b[k+1 ].GetMask()&1) FindPeaks(k+1 ,max,b,idx,msk,n);
554 void AliITSclustererV2::
555 MarkPeak(Int_t k, Int_t max, AliBin *bins, UInt_t m) {
556 //------------------------------------------------------------
558 //------------------------------------------------------------
559 UShort_t q=bins[k].GetQ();
561 bins[k].SetMask(bins[k].GetMask()|m);
563 if (bins[k-max].GetQ() <= q)
564 if ((bins[k-max].GetMask()&m) == 0) MarkPeak(k-max,max,bins,m);
565 if (bins[k-1 ].GetQ() <= q)
566 if ((bins[k-1 ].GetMask()&m) == 0) MarkPeak(k-1 ,max,bins,m);
567 if (bins[k+max].GetQ() <= q)
568 if ((bins[k+max].GetMask()&m) == 0) MarkPeak(k+max,max,bins,m);
569 if (bins[k+1 ].GetQ() <= q)
570 if ((bins[k+1 ].GetMask()&m) == 0) MarkPeak(k+1 ,max,bins,m);
573 void AliITSclustererV2::
574 MakeCluster(Int_t k,Int_t max,AliBin *bins,UInt_t m,AliITSclusterV2 &c) {
575 //------------------------------------------------------------
576 //make cluster using digits of this peak
577 //------------------------------------------------------------
578 Float_t q=(Float_t)bins[k].GetQ();
579 Int_t i=k/max, j=k-i*max;
582 c.SetY(c.GetY()+i*q);
583 c.SetZ(c.GetZ()+j*q);
584 c.SetSigmaY2(c.GetSigmaY2()+i*i*q);
585 c.SetSigmaZ2(c.GetSigmaZ2()+j*j*q);
587 bins[k].SetMask(0xFFFFFFFE);
589 if (bins[k-max].GetMask() == m) MakeCluster(k-max,max,bins,m,c);
590 if (bins[k-1 ].GetMask() == m) MakeCluster(k-1 ,max,bins,m,c);
591 if (bins[k+max].GetMask() == m) MakeCluster(k+max,max,bins,m,c);
592 if (bins[k+1 ].GetMask() == m) MakeCluster(k+1 ,max,bins,m,c);
595 void AliITSclustererV2::
596 FindClustersSDD(AliBin* bins[2], Int_t nMaxBin, Int_t nzBins,
597 const TClonesArray *digits, TClonesArray *clusters) {
598 //------------------------------------------------------------
599 // Actual SDD cluster finder
600 //------------------------------------------------------------
601 Int_t ncl=0; TClonesArray &cl=*clusters;
602 for (Int_t s=0; s<2; s++)
603 for (Int_t i=0; i<nMaxBin; i++) {
604 if (bins[s][i].IsUsed()) continue;
605 Int_t idx[32]; UInt_t msk[32]; Int_t npeaks=0;
606 FindPeaks(i, nzBins, bins[s], idx, msk, npeaks);
608 if (npeaks>30) continue;
611 for (k=0; k<npeaks-1; k++){//mark adjacent peaks
612 if (idx[k] < 0) continue; //this peak is already removed
613 for (l=k+1; l<npeaks; l++) {
614 if (idx[l] < 0) continue; //this peak is already removed
615 Int_t ki=idx[k]/nzBins, kj=idx[k] - ki*nzBins;
616 Int_t li=idx[l]/nzBins, lj=idx[l] - li*nzBins;
617 Int_t di=TMath::Abs(ki - li);
618 Int_t dj=TMath::Abs(kj - lj);
619 if (di>1 || dj>1) continue;
620 if (bins[s][idx[k]].GetQ() > bins[s][idx[l]].GetQ()) {
631 for (k=0; k<npeaks; k++) {
632 MarkPeak(TMath::Abs(idx[k]), nzBins, bins[s], msk[k]);
635 for (k=0; k<npeaks; k++) {
636 if (idx[k] < 0) continue; //removed peak
638 MakeCluster(idx[k], nzBins, bins[s], msk[k], c);
640 //if (c.GetQ() < 200) continue; //noise cluster
643 Float_t s2 = c.GetSigmaY2()/c.GetQ() - c.GetY()*c.GetY();
644 Float_t w=par->GetPadPitchWidth(sec);
645 c.SetSigmaY2((s2 + 1./12.)*w*w);
647 c.SetSigmaY2(c.GetSigmaY2()*0.108);
648 if (sec<par->GetNInnerSector()) c.SetSigmaY2(c.GetSigmaY2()*2.07);
651 s2 = c.GetSigmaZ2()/c.GetQ() - c.GetZ()*c.GetZ();
653 c.SetSigmaZ2((s2 + 1./12.)*w*w);
655 c.SetSigmaZ2(c.GetSigmaZ2()*0.169);
656 if (sec<par->GetNInnerSector()) c.SetSigmaZ2(c.GetSigmaZ2()*1.77);
660 c.SetSigmaY2(0.0030*0.0030);
661 c.SetSigmaZ2(0.0020*0.0020);
662 c.SetDetectorIndex(fNdet[fI]);
664 Float_t y=c.GetY(),z=c.GetZ(), q=c.GetQ();
667 y=(y-0.5)*fYpitchSDD;
669 y-=fYoffSDD; //delay ?
672 z=(z-0.5)*fZpitchSDD;
680 c.SetQ(q/20.); //to be consistent with the SSD charges
683 AliBin *b=&bins[s][idx[k]];
684 AliITSdigitSDD* d=(AliITSdigitSDD*)digits->UncheckedAt(b->GetIndex());
685 Int_t l0=(d->GetTracks())[0];
687 b=&bins[s][idx[k]-1];
689 d=(AliITSdigitSDD*)digits->UncheckedAt(b->GetIndex());
690 l0=(d->GetTracks())[0];
694 b=&bins[s][idx[k]+1];
696 d=(AliITSdigitSDD*)digits->UncheckedAt(b->GetIndex());
697 l0=(d->GetTracks())[0];
701 b=&bins[s][idx[k]-nzBins];
703 d=(AliITSdigitSDD*)digits->UncheckedAt(b->GetIndex());
704 l0=(d->GetTracks())[0];
708 b=&bins[s][idx[k]+nzBins];
710 d=(AliITSdigitSDD*)digits->UncheckedAt(b->GetIndex());
711 l0=(d->GetTracks())[0];
716 b=&bins[s][idx[k]+nzBins+1];
718 d=(AliITSdigitSDD*)digits->UncheckedAt(b->GetIndex());
719 l0=(d->GetTracks())[0];
723 b=&bins[s][idx[k]+nzBins-1];
725 d=(AliITSdigitSDD*)digits->UncheckedAt(b->GetIndex());
726 l0=(d->GetTracks())[0];
730 b=&bins[s][idx[k]-nzBins+1];
732 d=(AliITSdigitSDD*)digits->UncheckedAt(b->GetIndex());
733 l0=(d->GetTracks())[0];
737 b=&bins[s][idx[k]-nzBins-1];
739 d=(AliITSdigitSDD*)digits->UncheckedAt(b->GetIndex());
740 l0=(d->GetTracks())[0];
746 lab[0]=(d->GetTracks())[0];
747 lab[1]=(d->GetTracks())[1];
748 lab[2]=(d->GetTracks())[2];
750 c.SetLabel(lab[0],0);
751 c.SetLabel(lab[1],1);
752 c.SetLabel(lab[2],2);
756 new (cl[ncl]) AliITSclusterV2(c); ncl++;
761 void AliITSclustererV2::
762 FindClustersSDD(const TClonesArray *digits, TClonesArray *clusters) {
763 //------------------------------------------------------------
764 // Actual SDD cluster finder
765 //------------------------------------------------------------
766 Int_t kNzBins = fNzSDD + 2;
767 const Int_t kMAXBIN=kNzBins*(fNySDD+2);
770 bins[0]=new AliBin[kMAXBIN];
771 bins[1]=new AliBin[kMAXBIN];
774 Int_t i, ndigits=digits->GetEntriesFast();
775 for (i=0; i<ndigits; i++) {
776 d=(AliITSdigitSDD*)digits->UncheckedAt(i);
777 Int_t y=d->GetCoord2()+1; //y
778 Int_t z=d->GetCoord1()+1; //z
779 Int_t q=d->GetSignal();
781 bins[0][y*kNzBins+z].SetQ(q);
782 bins[0][y*kNzBins+z].SetMask(1);
783 bins[0][y*kNzBins+z].SetIndex(i);
786 bins[1][y*kNzBins+z].SetQ(q);
787 bins[1][y*kNzBins+z].SetMask(1);
788 bins[1][y*kNzBins+z].SetIndex(i);
792 FindClustersSDD(bins, kMAXBIN, kNzBins, digits, clusters);
798 void AliITSclustererV2::FindClustersSDD(AliITSRawStream* input,
799 TClonesArray** clusters)
801 //------------------------------------------------------------
802 // Actual SDD cluster finder for raw data
803 //------------------------------------------------------------
804 Int_t nClustersSDD = 0;
805 Int_t kNzBins = fNzSDD + 2;
806 Int_t kMaxBin = kNzBins * (fNySDD+2);
807 AliBin* bins[2] = {NULL, NULL};
809 // read raw data input stream
811 Bool_t next = input->Next();
812 if (!next || input->IsNewModule()) {
813 Int_t iModule = input->GetPrevModuleID();
815 // when all data from a module was read, search for clusters
817 clusters[iModule] = new TClonesArray("AliITSclusterV2");
819 FindClustersSDD(bins, kMaxBin, kNzBins, NULL, clusters[iModule]);
820 Int_t nClusters = clusters[iModule]->GetEntriesFast();
821 nClustersSDD += nClusters;
827 bins[0] = new AliBin[kMaxBin];
828 bins[1] = new AliBin[kMaxBin];
831 // fill the current digit into the bins array
832 Int_t iz = input->GetCoord1()+1;
833 Int_t side = ((iz <= fNzSDD) ? 0 : 1);
835 Int_t index = (input->GetCoord2()+1) * kNzBins + iz;
836 bins[side][index].SetQ(input->GetSignal());
837 bins[side][index].SetMask(1);
838 bins[side][index].SetIndex(index);
841 Info("FindClustersSDD", "found clusters in ITS SDD: %d", nClustersSDD);
844 void AliITSclustererV2::
845 FindClustersSSD(Ali1Dcluster* neg, Int_t nn,
846 Ali1Dcluster* pos, Int_t np,
847 TClonesArray *clusters) {
848 //------------------------------------------------------------
849 // Actual SSD cluster finder
850 //------------------------------------------------------------
851 TClonesArray &cl=*clusters;
853 Int_t lab[4]={-2,-2,-2,-2};
854 Float_t tanp=fTanP, tann=fTanN;
855 if (fI>fLastSSD1) {tann=fTanP; tanp=fTanN;}
857 Int_t idet=fNdet[fI];
859 for (Int_t i=0; i<np; i++) {
860 //Float_t dq_min=1.e+33;
861 Float_t ybest=1000,zbest=1000,qbest=0;
862 Float_t yp=pos[i].GetY()*fYpitchSSD;
863 for (Int_t j=0; j<nn; j++) {
864 //if (pos[i].fTracks[0] != neg[j].fTracks[0]) continue;
865 Float_t yn=neg[j].GetY()*fYpitchSSD;
866 Float_t zt=(2*fHlSSD*tanp + yp - yn)/(tann+tanp);
867 Float_t yt=yn + tann*zt;
868 zt-=fHlSSD; yt-=fHwSSD;
869 if (TMath::Abs(yt)<fHwSSD+0.01)
870 if (TMath::Abs(zt)<fHlSSD+0.01) {
871 //if (TMath::Abs(pos[i].GetQ()-neg[j].GetQ())<dq_min) {
872 //dq_min=TMath::Abs(pos[i].GetQ()-neg[j].GetQ());
874 qbest=0.5*(pos[i].GetQ()+neg[j].GetQ());
876 lab[0]=pos[i].GetLabel(0);
877 lab[1]=pos[i].GetLabel(1);
878 lab[2]=neg[i].GetLabel(0);
879 lab[3]=(((i<<10) + j)<<10) + idet; // pos|neg|det
881 lp[0]=-ybest-fYshift[fI];
882 lp[1]= zbest+fZshift[fI];
883 lp[2]=0.0025*0.0025; //SigmaY2
884 lp[3]=0.110*0.110; //SigmaZ2
885 if (pos[i].GetNd()+neg[j].GetNd() > 4) {
892 new (cl[ncl]) AliITSclusterV2(lab,lp); ncl++;
899 lp[0]=-ybest-fYshift[fI];
900 lp[1]= zbest+fZshift[fI];
901 lp[2]=0.002*0.002; //SigmaY2
902 lp[3]=0.080*0.080; //SigmaZ2
905 new (cl[ncl]) AliITSclusterV2(lab,lp); ncl++;
911 void AliITSclustererV2::
912 FindClustersSSD(const TClonesArray *digits, TClonesArray *clusters) {
913 //------------------------------------------------------------
914 // Actual SSD cluster finder
915 //------------------------------------------------------------
916 Int_t smax=digits->GetEntriesFast();
919 const Int_t MAX=1000;
921 Ali1Dcluster pos[MAX], neg[MAX];
922 Float_t y=0., q=0., qmax=0.;
923 Int_t lab[4]={-2,-2,-2,-2};
925 AliITSdigitSSD *d=(AliITSdigitSSD*)digits->UncheckedAt(0);
927 y += d->GetCoord2()*d->GetSignal();
929 lab[0]=d->GetTrack(0); lab[1]=d->GetTrack(1); lab[2]=d->GetTrack(2);
930 Int_t curr=d->GetCoord2();
931 Int_t flag=d->GetCoord1();
935 for (Int_t s=1; s<smax; s++) {
936 d=(AliITSdigitSSD*)digits->UncheckedAt(s);
937 Int_t strip=d->GetCoord2();
938 if ((strip-curr) > 1 || flag!=d->GetCoord1()) {
942 c[*n].SetLabels(lab);
943 //Split suspiciously big cluster
945 c[*n].SetY(y/q-0.5*nd);
949 Error("FindClustersSSD","Too many 1D clusters !");
952 c[*n].SetY(y/q+0.5*nd);
955 c[*n].SetLabels(lab);
959 Error("FindClustersSSD","Too many 1D clusters !");
964 lab[0]=lab[1]=lab[2]=-2;
965 if (flag!=d->GetCoord1()) { n=&np; c=pos; }
969 y += d->GetCoord2()*d->GetSignal();
971 if (d->GetSignal()>qmax) {
973 lab[0]=d->GetTrack(0); lab[1]=d->GetTrack(1); lab[2]=d->GetTrack(2);
980 c[*n].SetLabels(lab);
981 //Split suspiciously big cluster
983 c[*n].SetY(y/q-0.5*nd);
987 Error("FindClustersSSD","Too many 1D clusters !");
990 c[*n].SetY(y/q+0.5*nd);
993 c[*n].SetLabels(lab);
997 Error("FindClustersSSD","Too many 1D clusters !");
1001 FindClustersSSD(neg, nn, pos, np, clusters);
1004 void AliITSclustererV2::FindClustersSSD(AliITSRawStream* input,
1005 TClonesArray** clusters)
1007 //------------------------------------------------------------
1008 // Actual SSD cluster finder for raw data
1009 //------------------------------------------------------------
1010 Int_t nClustersSSD = 0;
1011 const Int_t MAX = 1000;
1012 Ali1Dcluster clusters1D[2][MAX];
1013 Int_t nClusters[2] = {0, 0};
1017 Int_t prevStrip = -1;
1018 Int_t prevFlag = -1;
1020 // read raw data input stream
1022 Bool_t next = input->Next();
1024 // check if a new cluster starts
1025 Int_t strip = input->GetCoord2();
1026 Int_t flag = input->GetCoord1();
1027 if ((!next || input->IsNewModule() ||
1028 (strip-prevStrip > 1) || (flag != prevFlag)) &&
1030 if (nClusters[prevFlag] == MAX) {
1031 Error("FindClustersSSD", "Too many 1D clusters !");
1034 Ali1Dcluster& cluster = clusters1D[prevFlag][nClusters[prevFlag]++];
1037 cluster.SetNd(nDigits);
1039 //Split suspiciously big cluster
1041 cluster.SetY(y/q - 0.5*nDigits);
1042 cluster.SetQ(0.5*q);
1043 if (nClusters[prevFlag] == MAX) {
1044 Error("FindClustersSSD", "Too many 1D clusters !");
1047 Ali1Dcluster& cluster2 = clusters1D[prevFlag][nClusters[prevFlag]++];
1048 cluster2.SetY(y/q + 0.5*nDigits);
1049 cluster2.SetQ(0.5*q);
1050 cluster2.SetNd(nDigits);
1056 if (!next || input->IsNewModule()) {
1057 Int_t iModule = input->GetPrevModuleID();
1059 // when all data from a module was read, search for clusters
1060 if (prevFlag >= 0) {
1061 clusters[iModule] = new TClonesArray("AliITSclusterV2");
1063 FindClustersSSD(&clusters1D[0][0], nClusters[0],
1064 &clusters1D[1][0], nClusters[1], clusters[iModule]);
1065 Int_t nClusters = clusters[iModule]->GetEntriesFast();
1066 nClustersSSD += nClusters;
1070 nClusters[0] = nClusters[1] = 0;
1075 // add digit to current cluster
1076 q += input->GetSignal();
1077 y += strip * input->GetSignal();
1083 Info("FindClustersSSD", "found clusters in ITS SSD: %d", nClustersSSD);
1088 #include "AliITSDetType.h"
1090 #include "AliITSsegmentationSPD.h"
1091 #include "AliITSClusterFinderSPD.h"
1093 #include "AliITSresponseSDD.h"
1094 #include "AliITSsegmentationSDD.h"
1095 #include "AliITSClusterFinderSDD.h"
1097 #include "AliITSsegmentationSSD.h"
1098 #include "AliITSClusterFinderSSD.h"
1101 void AliITSclustererV2::
1102 FindClustersSPD(const TClonesArray *digits, TClonesArray *clusters) {
1103 //------------------------------------------------------------
1104 // Actual SPD cluster finding based on AliITSClusterFinderSPD
1105 //------------------------------------------------------------
1106 static AliITS *its=(AliITS*)gAlice->GetModule("ITS");
1107 static TClonesArray *points=its->RecPoints();
1108 static AliITSsegmentationSPD *seg=
1109 (AliITSsegmentationSPD *)its->DetType(0)->GetSegmentationModel();
1110 static AliITSClusterFinderSPD cf(seg, (TClonesArray*)digits, points);
1112 cf.FindRawClusters(fI);
1113 RecPoints2Clusters(points, fI, clusters);
1114 its->ResetRecPoints();
1118 void AliITSclustererV2::
1119 FindClustersSDD(const TClonesArray *digits, TClonesArray *clusters) {
1120 //------------------------------------------------------------
1121 // Actual SDD cluster finding based on AliITSClusterFinderSDD
1122 //------------------------------------------------------------
1123 static AliITS *its=(AliITS*)gAlice->GetModule("ITS");
1124 static TClonesArray *points=its->RecPoints();
1125 static AliITSresponseSDD *resp=
1126 (AliITSresponseSDD *)its->DetType(1)->GetResponseModel();
1127 static AliITSsegmentationSDD *seg=
1128 (AliITSsegmentationSDD *)its->DetType(1)->GetSegmentationModel();
1129 static AliITSClusterFinderSDD
1130 cf(seg,resp,(TClonesArray*)digits,its->ClustersAddress(1));
1132 cf.FindRawClusters(fI);
1133 Int_t nc=points->GetEntriesFast();
1134 while (nc--) { //To be consistent with the SSD cluster charges
1135 AliITSRecPoint *p=(AliITSRecPoint*)points->UncheckedAt(nc);
1136 p->SetQ(p->GetQ/12.);
1138 RecPoints2Clusters(points, fI, clusters);
1139 its->ResetClusters(1);
1140 its->ResetRecPoints();
1144 void AliITSclustererV2::
1145 FindClustersSSD(const TClonesArray *digits, TClonesArray *clusters) {
1146 //------------------------------------------------------------
1147 // Actual SSD cluster finding based on AliITSClusterFinderSSD
1148 //------------------------------------------------------------
1149 static AliITS *its=(AliITS*)gAlice->GetModule("ITS");
1150 static TClonesArray *points=its->RecPoints();
1151 static AliITSsegmentationSSD *seg=
1152 (AliITSsegmentationSSD *)its->DetType(2)->GetSegmentationModel();
1153 static AliITSClusterFinderSSD cf(seg,(TClonesArray*)digits);
1155 cf.FindRawClusters(fI);
1156 RecPoints2Clusters(points, fI, clusters);
1157 its->ResetRecPoints();