]>
Commit | Line | Data |
---|---|---|
1 | #ifndef ALIITSVERTEXER3D_H | |
2 | #define ALIITSVERTEXER3D_H | |
3 | ||
4 | #include<AliITSVertexer.h> | |
5 | ||
6 | /////////////////////////////////////////////////////////////////// | |
7 | // // | |
8 | // Class for primary vertex finding (3D reconstruction) // | |
9 | // // | |
10 | /////////////////////////////////////////////////////////////////// | |
11 | ||
12 | /* $Id$ */ | |
13 | ||
14 | #include <TClonesArray.h> | |
15 | #include <TRandom3.h> | |
16 | #include <AliESDVertex.h> | |
17 | #include <TH3F.h> | |
18 | #include <TBits.h> | |
19 | ||
20 | class AliITSVertexer3D : public AliITSVertexer { | |
21 | ||
22 | public: | |
23 | ||
24 | AliITSVertexer3D(Double_t zcut=0.); | |
25 | AliITSVertexer3D(TRootIOCtor* ); | |
26 | virtual ~AliITSVertexer3D(); | |
27 | virtual AliESDVertex* FindVertexForCurrentEvent(TTree *itsClusterTree); | |
28 | void FindVertex3DIterative(); | |
29 | void FindVertex3DIterativeMM(); | |
30 | void FindVertex3D(TTree *itsClusterTree); | |
31 | AliESDVertex GetVertex3D() const {return fVert3D;} | |
32 | // Double_t *Get3DPeak() {return f3DPeak;} | |
33 | virtual void PrintStatus() const; | |
34 | static Bool_t DistBetweenVertices(AliESDVertex &a, AliESDVertex &b, Double_t test, Double_t &dist); | |
35 | void SetWideFiducialRegion(Double_t dz = 40.0, Double_t dr=2.5){ | |
36 | SetCoarseMaxRCut(dr); | |
37 | SetZCutDiamond(dz); | |
38 | } | |
39 | void SetNarrowFiducialRegion(Double_t dz = 0.5, Double_t dr=0.5){ | |
40 | SetMaxRCut(dr); | |
41 | SetMaxZCut(dz); | |
42 | } | |
43 | void SetDeltaPhiCuts(Double_t dphiloose=0.5, Double_t dphitight=0.025){ | |
44 | SetCoarseDiffPhiCut(dphiloose); | |
45 | SetDiffPhiMax(dphitight); | |
46 | } | |
47 | void SetCoarseDiffPhiCut(Double_t dphi = 0.5){fCoarseDiffPhiCut=dphi;} | |
48 | void SetFineDiffPhiCut(Double_t dphi = 0.05){fFineDiffPhiCut=dphi;} | |
49 | void SetCutOnPairs(Double_t cp = 0.15){fCutOnPairs = cp;} | |
50 | void SetCoarseMaxRCut(Double_t rad = 2.5){fCoarseMaxRCut=rad;} | |
51 | void SetMaxRCut(Double_t rad = 0.5){fMaxRCut=rad;} | |
52 | void SetMaxRCutAlgo2(Double_t rad = 0.2){fMaxRCut2=rad;} | |
53 | void SetZCutDiamond(Double_t zcut = 40.0); | |
54 | void SetMaxZCut(Double_t dz = 0.5){fMaxZCut=dz;} | |
55 | void SetDCACut(Double_t dca=0.1){fDCAcut=dca;} | |
56 | void SetDiffPhiMax(Double_t pm = 0.025){fDiffPhiMax = pm;} | |
57 | void SetMeanPSelTracks(Double_t pGeV=0.875){fMeanPSelTrk = pGeV;} | |
58 | void SetMeanPtSelTracks(Double_t ptGeV=0.630){fMeanPtSelTrk = ptGeV;} | |
59 | void SetMeanPPtSelTracks(Double_t fieldTesla); | |
60 | void SetMinDCAforPileup(Double_t mindist=0.1) {fDCAforPileup=mindist;} | |
61 | void SetDeltaPhiforPileup(Double_t dphi=0.01) {fDiffPhiforPileup=dphi;} | |
62 | void SetPileupAlgo(UShort_t optalgo=1){fPileupAlgo=optalgo;} | |
63 | void SetBinSizeR(Double_t siz=0.1){fBinSizeR=siz;} | |
64 | void SetBinSizeZ(Double_t siz=0.8){fBinSizeZ=siz;} | |
65 | void SetHighMultAlgo(UChar_t n){ | |
66 | if(n<2) fHighMultAlgo=n; | |
67 | else AliError("Only algos 0 and 1 implemented"); | |
68 | } | |
69 | void SetHighMultDownscalingAlgo(){fHighMultAlgo=0;} | |
70 | void SetHighMultTracesAlgo(){fHighMultAlgo=1;} | |
71 | ||
72 | void SetMaxNumOfClustersForHighMult(Int_t ncl){fMaxNumOfCl=ncl;} | |
73 | void SetMaxNumOfClustersForDownScale(Int_t ncl){fMaxNumOfClForDownScale=ncl;} | |
74 | void SetMaxNumOfClustersForRebin(Int_t ncl){fMaxNumOfClForRebin=ncl;} | |
75 | Int_t GetMaxNumOfClustersForHighMult() const {return fMaxNumOfCl;} | |
76 | Int_t GetMaxNumOfClustersForDownScale() const {return fMaxNumOfClForDownScale;} | |
77 | Int_t GetMaxNumOfClustersForRebin() const {return fMaxNumOfClForRebin;} | |
78 | ||
79 | protected: | |
80 | AliITSVertexer3D(const AliITSVertexer3D& vtxr); | |
81 | AliITSVertexer3D& operator=(const AliITSVertexer3D& /* vtxr */); | |
82 | Int_t FindTracklets(TTree *itsClusterTree, Int_t optCuts); | |
83 | Int_t Prepare3DVertex(Int_t optCuts); | |
84 | Int_t Prepare3DVertexPbPb(); | |
85 | void ResetVert3D(); | |
86 | void FindPeaks(TH3F* histo, Double_t *peak, Int_t &nOfTracklets, Int_t &nOfTimes); | |
87 | void PileupFromZ(); | |
88 | void MarkUsedClusters(); | |
89 | Int_t RemoveTracklets(); | |
90 | void FindOther3DVertices(TTree *itsClusterTree); | |
91 | Double_t GetFraction(Int_t itr) const; | |
92 | ||
93 | enum {kMaxCluPerMod=250}; | |
94 | enum {kMaxPileupVertices=10}; | |
95 | TClonesArray fLines; //! array of tracklets | |
96 | AliESDVertex fVert3D; // 3D Vertex | |
97 | Double_t fCoarseDiffPhiCut; // loose cut on DeltaPhi for RecPoint matching | |
98 | Double_t fFineDiffPhiCut; // tight value of DeltaPhi for RP matching (2nd method) | |
99 | Double_t fCutOnPairs; //cut on distance between pairs of tracklets | |
100 | Double_t fCoarseMaxRCut; // cut on tracklet DCA to Z axis | |
101 | Double_t fMaxRCut; // cut on tracklet DCA to beam axis | |
102 | Double_t fMaxRCut2; // cut on tracklet DCA to beam axis - algo2 | |
103 | Double_t fZCutDiamond; // cut on +-Z of the diamond | |
104 | Double_t fMaxZCut; // cut on Z distance from estimated vertex | |
105 | Double_t fDCAcut; // cut on tracklet to tracklet and tracklet to vertex DCA | |
106 | Double_t fDiffPhiMax; // Maximum delta phi allowed among corr. pixels | |
107 | Double_t fMeanPSelTrk; // GeV, mean P for tracks with dphi<0.01 rad | |
108 | Double_t fMeanPtSelTrk; // GeV, mean Pt for tracks with dphi<0.01 rad | |
109 | TBits fUsedCluster; // flag for used clusters in vertex calculation | |
110 | TH1F *fZHisto; //! histogram with coarse z distribution | |
111 | Double_t fDCAforPileup; // Minimum DCA to 1st vertex for pileup tracklets | |
112 | Double_t fDiffPhiforPileup; // Cut on delta phi for pileup | |
113 | Double_t fBinSizeR; // Histo3D bin size along radius | |
114 | Double_t fBinSizeZ; // Histo3D bin size along z | |
115 | UShort_t fPileupAlgo; // Algo for pileup identification | |
116 | // 0->VertexerZ pileup algo | |
117 | // 1->Unused RecPoints algo | |
118 | Int_t fMaxNumOfCl; // max n. of clusters on L1 or L2 for high mult definition | |
119 | Int_t fMaxNumOfClForRebin; // max n. of clusters on L1 or L2 for rebin | |
120 | Int_t fMaxNumOfClForDownScale; // max n. of clusters on L1 or L2 for downscale | |
121 | Int_t fNRecPLay1; // number of rec ponts on SPD layer 1 | |
122 | Int_t fNRecPLay2; // number of rec ponts on SPD layer 2 | |
123 | Float_t f3DBinSize; // Size of the 3D bins | |
124 | Bool_t fDoDownScale; // Control downscaling of tracklets in high mult | |
125 | TRandom3 *fGenerForDownScale; // randomnumber generator fordownscaling | |
126 | Double_t f3DPeak[3]; // TH3F peak coords | |
127 | UChar_t fHighMultAlgo; // algorithm used for high mult. events | |
128 | Bool_t fSwitchAlgorithm; // Switch between two algoritms in testing phase | |
129 | ||
130 | static const Int_t fgkMaxNumOfClDefault; // Default max n. of clusters for downscale | |
131 | static const Int_t fgkMaxNumOfClRebinDefault; // Default max n. of clusters for rebin | |
132 | static const Int_t fgkMaxNumOfClDownscaleDefault; // Default max n. of clusters for rebin | |
133 | static const Float_t fgk3DBinSizeDefault; // Default 3D bins size | |
134 | ||
135 | ClassDef(AliITSVertexer3D,15); | |
136 | ||
137 | }; | |
138 | ||
139 | #endif |