]>
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(); | |
25 | virtual ~AliITSVertexer3D(); | |
26 | virtual AliESDVertex* FindVertexForCurrentEvent(TTree *itsClusterTree); | |
27 | void FindVertex3DIterative(); | |
28 | void FindVertex3DIterativeMM(); | |
29 | void FindVertex3D(TTree *itsClusterTree); | |
30 | AliESDVertex GetVertex3D() const {return fVert3D;} | |
31 | virtual void PrintStatus() const; | |
32 | static Bool_t DistBetweenVertices(AliESDVertex &a, AliESDVertex &b, Double_t test, Double_t &dist); | |
33 | void SetWideFiducialRegion(Double_t dz = 40.0, Double_t dr=2.5){ | |
34 | SetCoarseMaxRCut(dr); | |
35 | SetZCutDiamond(dz); | |
36 | } | |
37 | void SetNarrowFiducialRegion(Double_t dz = 0.5, Double_t dr=0.5){ | |
38 | SetMaxRCut(dr); | |
39 | SetMaxZCut(dz); | |
40 | } | |
41 | void SetDeltaPhiCuts(Double_t dphiloose=0.5, Double_t dphitight=0.025){ | |
42 | SetCoarseDiffPhiCut(dphiloose); | |
43 | SetDiffPhiMax(dphitight); | |
44 | } | |
45 | void SetCoarseDiffPhiCut(Double_t dphi = 0.5){fCoarseDiffPhiCut=dphi;} | |
46 | void SetFineDiffPhiCut(Double_t dphi = 0.05){fFineDiffPhiCut=dphi;} | |
47 | void SetCutOnPairs(Double_t cp = 0.15){fCutOnPairs = cp;} | |
48 | void SetCoarseMaxRCut(Double_t rad = 2.5){fCoarseMaxRCut=rad;} | |
49 | void SetMaxRCut(Double_t rad = 0.5){fMaxRCut=rad;} | |
50 | void SetMaxRCutAlgo2(Double_t rad = 0.2){fMaxRCut2=rad;} | |
51 | void SetZCutDiamond(Double_t zcut = 40.0){fZCutDiamond=zcut;} | |
52 | void SetMaxZCut(Double_t dz = 0.5){fMaxZCut=dz;} | |
53 | void SetDCACut(Double_t dca=0.1){fDCAcut=dca;} | |
54 | void SetDiffPhiMax(Double_t pm = 0.025){fDiffPhiMax = pm;} | |
55 | void SetMeanPSelTracks(Double_t pGeV=0.875){fMeanPSelTrk = pGeV;} | |
56 | void SetMeanPtSelTracks(Double_t ptGeV=0.630){fMeanPtSelTrk = ptGeV;} | |
57 | void SetMeanPPtSelTracks(Double_t fieldTesla); | |
58 | void SetMinDCAforPileup(Double_t mindist=0.1) {fDCAforPileup=mindist;} | |
59 | void SetDeltaPhiforPileup(Double_t dphi=0.01) {fDiffPhiforPileup=dphi;} | |
60 | void SetPileupAlgo(UShort_t optalgo=1){fPileupAlgo=optalgo;} | |
61 | void SetBinSizeR(Double_t siz=0.1){fBinSizeR=siz;} | |
62 | void SetBinSizeZ(Double_t siz=0.8){fBinSizeZ=siz;} | |
63 | void SetMaxNumOfClusters(Int_t ncl){fMaxNumOfCl=ncl;} | |
64 | Int_t GetMaxNumOfClusters() const {return fMaxNumOfCl;} | |
65 | ||
66 | protected: | |
67 | AliITSVertexer3D(const AliITSVertexer3D& vtxr); | |
68 | AliITSVertexer3D& operator=(const AliITSVertexer3D& /* vtxr */); | |
69 | Int_t FindTracklets(TTree *itsClusterTree, Int_t optCuts); | |
70 | Int_t Prepare3DVertex(Int_t optCuts); | |
71 | void ResetVert3D(); | |
72 | void FindPeaks(TH3F* histo, Double_t *peak, Int_t &nOfTracklets, Int_t &nOfTimes); | |
73 | void PileupFromZ(); | |
74 | void MarkUsedClusters(); | |
75 | Int_t RemoveTracklets(); | |
76 | void FindOther3DVertices(TTree *itsClusterTree); | |
77 | ||
78 | enum {kMaxCluPerMod=250}; | |
79 | ||
80 | TClonesArray fLines; //! array of tracklets | |
81 | AliESDVertex fVert3D; // 3D Vertex | |
82 | Double_t fCoarseDiffPhiCut; // loose cut on DeltaPhi for RecPoint matching | |
83 | Double_t fFineDiffPhiCut; // tight value of DeltaPhi for RP matching (2nd method) | |
84 | Double_t fCutOnPairs; //cut on distance between pairs of tracklets | |
85 | Double_t fCoarseMaxRCut; // cut on tracklet DCA to Z axis | |
86 | Double_t fMaxRCut; // cut on tracklet DCA to beam axis | |
87 | Double_t fMaxRCut2; // cut on tracklet DCA to beam axis - algo2 | |
88 | Double_t fZCutDiamond; // cut on +-Z of the diamond | |
89 | Double_t fMaxZCut; // cut on Z distance from estimated vertex | |
90 | Double_t fDCAcut; // cut on tracklet to tracklet and tracklet to vertex DCA | |
91 | Double_t fDiffPhiMax; // Maximum delta phi allowed among corr. pixels | |
92 | Double_t fMeanPSelTrk; // GeV, mean P for tracks with dphi<0.01 rad | |
93 | Double_t fMeanPtSelTrk; // GeV, mean Pt for tracks with dphi<0.01 rad | |
94 | TBits fUsedCluster; // flag for used clusters in vertex calculation | |
95 | TH1F *fZHisto; //! histogram with coarse z distribution | |
96 | Double_t fDCAforPileup; // Minimum DCA to 1st vertex for pileup tracklets | |
97 | Double_t fDiffPhiforPileup; // Cut on delta phi for pileup | |
98 | Double_t fBinSizeR; // Histo3D bin size along radius | |
99 | Double_t fBinSizeZ; // Histo3D bin size along z | |
100 | UShort_t fPileupAlgo; // Algo for pileup identification | |
101 | // 0->VertexerZ pileup algo | |
102 | // 1->Unused RecPoints algo | |
103 | Int_t fMaxNumOfCl; // max number of clusters on L1 or L2 | |
104 | Bool_t fDoDownScale; // Control downscaling of tracklets in high mult | |
105 | TRandom3 *fGenerForDownScale; // randomnumber generator fordownscaling | |
106 | ||
107 | static const Int_t fgkMaxNumOfClDefault; // Default max number of clusters | |
108 | ||
109 | ClassDef(AliITSVertexer3D,14); | |
110 | ||
111 | }; | |
112 | ||
113 | #endif |