Correct treatment of tracks with pT<pTmin
[u/mrichter/AliRoot.git] / ITS / AliITSclustererV2.h
CommitLineData
7f4044f0 1#ifndef ALIITSCLUSTERERV2_H
2#define ALIITSCLUSTERERV2_H
3//--------------------------------------------------------------
4// ITS clusterer V2
5//
6// This can be a "wrapping" for the V1 cluster finding classes
7// if compiled with uncommented "#define V1" line
8// in the AliITSclustererV2.cxx file.
9//
10// Origin: Iouri Belikov, CERN, Jouri.Belikov@cern.ch
11//--------------------------------------------------------------
c630aafd 12#include <TObject.h>
7f4044f0 13
14class TFile;
c630aafd 15class TTree;
7f4044f0 16class TClonesArray;
17
7f4044f0 18class AliITSclusterV2;
7941072e 19class AliRawReader;
c391f9d9 20class AliITSRawStream;
7f4044f0 21
c630aafd 22class AliITSclustererV2 : public TObject {
7f4044f0 23public:
e56160b8 24 AliITSclustererV2();
79c75366 25 AliITSclustererV2(const Char_t *geom);
7f4044f0 26
27 void SetEvent(Int_t event) { fEvent=event; }
c630aafd 28 Int_t Digits2Clusters(TTree *in, TTree *out);
7941072e 29 void Digits2Clusters(AliRawReader* rawReader);
7f4044f0 30 void FindClustersSPD(const TClonesArray *dig, TClonesArray *cls);
c391f9d9 31 void FindClustersSPD(AliITSRawStream* input, TClonesArray** clusters);
7f4044f0 32 void FindClustersSDD(const TClonesArray *dig, TClonesArray *cls);
c391f9d9 33 void FindClustersSDD(AliITSRawStream* input, TClonesArray** clusters);
7f4044f0 34 void FindClustersSSD(const TClonesArray *dig, TClonesArray *cls);
c391f9d9 35 void FindClustersSSD(AliITSRawStream* input, TClonesArray** clusters);
7f4044f0 36
1cca57bf 37 void RecPoints2Clusters(const TClonesArray *p, Int_t idx, TClonesArray *c);
7f4044f0 38
7f4044f0 39 class Ali1Dcluster {
7f4044f0 40 public:
41 void SetY(Float_t y) {fY=y;}
42 void SetQ(Float_t q) {fQ=q;}
43 void SetNd(Int_t n) {fNd=n;}
44 void SetLabels(Int_t *lab) {fLab[0]=lab[0];fLab[1]=lab[1];fLab[2]=lab[2];}
45 Float_t GetY() const {return fY;}
46 Float_t GetQ() const {return fQ;}
47 Int_t GetNd()const {return fNd;}
48 Int_t GetLabel(Int_t lab) const { return fLab[lab]; }
42283fd4 49 private:
50 Float_t fY; //cluster position
51 Float_t fQ; //cluster charge
52 Int_t fNd; //number of digits
53 Int_t fLab[3]; //track label
7f4044f0 54 };
55 class AliBin {
56 public:
e56160b8 57 AliBin():fIndex(0),fMask(0xFFFFFFFE),fQ(0) {}
7f4044f0 58 void SetIndex(UInt_t idx) {fIndex=idx;}
59 void SetQ(UShort_t q) {fQ=q;}
60 void SetMask(UInt_t m) {fMask=m;}
4b59acab 61 void Reset() {fIndex=0; fMask=0xFFFFFFFE; fQ=0;}
7f4044f0 62
63 void Use() {fMask&=0xFFFFFFFE;}
64 Bool_t IsNotUsed() const {return (fMask&1);}
65 Bool_t IsUsed() const {return !(IsNotUsed());}
66
67 UInt_t GetIndex() const {return fIndex;}
68 UShort_t GetQ() const {return fQ;}
69 UInt_t GetMask() const {return fMask;}
70 private:
71 UInt_t fIndex; //digit index
72 UInt_t fMask; //peak mask
73 UShort_t fQ; //signal
74 };
75 static Bool_t IsMaximum(Int_t k, Int_t max, const AliBin *bins);
76 static void FindPeaks(Int_t k,Int_t m,AliBin*b,Int_t*idx,UInt_t*msk,Int_t&n);
77 static void MarkPeak(Int_t k, Int_t max, AliBin *bins, UInt_t m);
78 static void MakeCluster(Int_t k,Int_t max,AliBin *bins,UInt_t m,
79 AliITSclusterV2 &c);
c391f9d9 80 void FindClustersSDD(AliBin* bins[2], Int_t nMaxBin, Int_t nMaxZ,
81 const TClonesArray *dig, TClonesArray *cls);
82 void FindClustersSSD(Ali1Dcluster* neg, Int_t nn,
83 Ali1Dcluster* pos, Int_t np,
84 TClonesArray *clusters);
85
7f4044f0 86
87 static void FindCluster(Int_t k,Int_t maxz,AliBin *bins,Int_t &n,Int_t *idx);
88
0e986ef9 89protected:
90 Int_t fNModules; // total number of modules
91
7f4044f0 92private:
93 Int_t fEvent; //event number
94
95 Int_t fI; //index of the current subdetector
96 Float_t fYshift[2200]; //y-shifts of detector local coor. systems
97 Float_t fZshift[2200]; //z-shifts of detector local coor. systems
98 Int_t fNdet[2200]; //detector index
e43c066c 99 Int_t fNlayer[2200]; //detector layer
7f4044f0 100
101 //SPD related values:
102 Int_t fLastSPD1; //index of the last SPD1 detector
103 Int_t fNySPD; //number of pixels in Y
104 Int_t fNzSPD; //number of pixels in Z
105 Float_t fYpitchSPD; //pixel size in Y
106 Float_t fZ1pitchSPD,fZ2pitchSPD; //pixel sizes in Z
107 Float_t fHwSPD; //half width of the SPD detector
108 Float_t fHlSPD; //half length of the SPD detector
109 Float_t fYSPD[260]; //Y-coordinates of pixel centers
110 Float_t fZSPD[170]; //Z-coordinates of pixel centers
111
112 //SDD related values:
113 Int_t fNySDD; //number of "pixels" in Y
114 Int_t fNzSDD; //number of "pixels" in Z
115 Float_t fYpitchSDD; //"pixel size" in Y (drift direction)
116 Float_t fZpitchSDD; //"pixel sizes" in Z
117 Float_t fHwSDD; //half width of the SDD detector
118 Float_t fHlSDD; //half length of the SDD detector
119 Float_t fYoffSDD; //some delay in the drift channel
120
121 //SSD related values:
122 Int_t fLastSSD1; //index of the last SSD1 detector
123 Float_t fYpitchSSD; //strip pitch (cm)
124 Float_t fHwSSD; //half-width of an SSD detector (cm)
125 Float_t fHlSSD; //half-length of an SSD detector (cm)
126 Float_t fTanP; //tangent of the stereo angle on the P side
127 Float_t fTanN; //tangent of the stereo angle on the N side
128
129 ClassDef(AliITSclustererV2,1) // ITS cluster finder V2
130};
131
132#endif