Added two missing includes to allow macro compilation (thanks to Laurent for remarkin...
[u/mrichter/AliRoot.git] / ITS / AliITSVertexer3D.h
CommitLineData
70c95f95 1#ifndef ALIITSVERTEXER3D_H
2#define ALIITSVERTEXER3D_H
3
27167524 4#include<AliITSVertexer.h>
70c95f95 5
6///////////////////////////////////////////////////////////////////
7// //
8// Class for primary vertex finding (3D reconstruction) //
9// //
10///////////////////////////////////////////////////////////////////
11
6a0d56b8 12/* $Id$ */
13
f5f6da22 14#include <TClonesArray.h>
6a0d56b8 15#include <AliESDVertex.h>
7340864d 16#include <TH3F.h>
cd1c2af1 17#include <TBits.h>
70c95f95 18
19class AliITSVertexer3D : public AliITSVertexer {
20
21 public:
22
23 AliITSVertexer3D();
70c95f95 24 virtual ~AliITSVertexer3D();
308c2f7c 25 virtual AliESDVertex* FindVertexForCurrentEvent(TTree *itsClusterTree);
6b4d9537 26 void FindVertex3DIterative();
27 void FindVertex3D(TTree *itsClusterTree);
6a0d56b8 28 AliESDVertex GetVertex3D() const {return fVert3D;}
70c95f95 29 virtual void PrintStatus() const;
6b4d9537 30 static Bool_t DistBetweenVertices(AliESDVertex &a, AliESDVertex &b, Double_t test, Double_t &dist);
242c8025 31 void SetWideFiducialRegion(Double_t dz = 40.0, Double_t dr=2.5){
7203e11a 32 SetCoarseMaxRCut(dr);
33 SetZCutDiamond(dz);
34 }
242c8025 35 void SetNarrowFiducialRegion(Double_t dz = 0.5, Double_t dr=0.5){
7203e11a 36 SetMaxRCut(dr);
37 SetMaxZCut(dz);
38 }
242c8025 39 void SetDeltaPhiCuts(Double_t dphiloose=0.5, Double_t dphitight=0.025){
7203e11a 40 SetCoarseDiffPhiCut(dphiloose);
41 SetDiffPhiMax(dphitight);
42 }
242c8025 43 void SetCoarseDiffPhiCut(Double_t dphi = 0.5){fCoarseDiffPhiCut=dphi;}
44 void SetFineDiffPhiCut(Double_t dphi = 0.05){fFineDiffPhiCut=dphi;}
45 void SetCutOnPairs(Double_t cp = 0.1){fCutOnPairs = cp;}
46 void SetCoarseMaxRCut(Double_t rad = 2.5){fCoarseMaxRCut=rad;}
47 void SetMaxRCut(Double_t rad = 0.5){fMaxRCut=rad;}
48 void SetZCutDiamond(Double_t zcut = 20.0){fZCutDiamond=zcut;}
49 void SetMaxZCut(Double_t dz = 0.5){fMaxZCut=dz;}
50 void SetDCACut(Double_t dca=0.1){fDCAcut=dca;}
51 void SetDiffPhiMax(Double_t pm = 0.025){fDiffPhiMax = pm;}
52 void SetMeanPSelTracks(Double_t pGeV=0.875){fMeanPSelTrk = pGeV;}
53 void SetMeanPtSelTracks(Double_t ptGeV=0.630){fMeanPtSelTrk = ptGeV;}
0496a544 54 void SetMeanPPtSelTracks(Double_t fieldTesla);
242c8025 55 void SetMinDCAforPileup(Double_t mindist=0.1) {fDCAforPileup=mindist;}
c238ba33 56 void SetDeltaPhiforPileup(Double_t dphi=0.01) {fDiffPhiforPileup=dphi;}
cd1c2af1 57 void SetPileupAlgo(UShort_t optalgo=0){fPileupAlgo=optalgo;}
242c8025 58 void SetBinSizeR(Double_t siz=0.1){fBinSizeR=siz;}
59 void SetBinSizeZ(Double_t siz=0.8){fBinSizeZ=siz;}
70c95f95 60
61protected:
62 AliITSVertexer3D(const AliITSVertexer3D& vtxr);
63 AliITSVertexer3D& operator=(const AliITSVertexer3D& /* vtxr */);
308c2f7c 64 Int_t FindTracklets(TTree *itsClusterTree, Int_t optCuts);
05d1294c 65 Int_t Prepare3DVertex(Int_t optCuts);
27167524 66 void ResetVert3D();
7340864d 67 void FindPeaks(TH3F* histo, Double_t *peak, Int_t &nOfTracklets, Int_t &nOfTimes);
cd1c2af1 68 void PileupFromZ();
69 void MarkUsedClusters();
70 Int_t RemoveTracklets();
71 void FindOther3DVertices(TTree *itsClusterTree);
70c95f95 72
cd1c2af1 73 enum {kMaxCluPerMod=250};
6a0d56b8 74
f5f6da22 75 TClonesArray fLines; //! array of tracklets
6a0d56b8 76 AliESDVertex fVert3D; // 3D Vertex
242c8025 77 Double_t fCoarseDiffPhiCut; // loose cut on DeltaPhi for RecPoint matching
78 Double_t fFineDiffPhiCut; // tight value of DeltaPhi for RP matching (2nd method)
79 Double_t fCutOnPairs; //cut on distance between pairs of tracklets
80 Double_t fCoarseMaxRCut; // cut on tracklet DCA to Z axis
81 Double_t fMaxRCut; // cut on tracklet DCA to beam axis
82 Double_t fZCutDiamond; // cut on +-Z of the diamond
83 Double_t fMaxZCut; // cut on Z distance from estimated vertex
84 Double_t fDCAcut; // cut on tracklet to tracklet and tracklet to vertex DCA
85 Double_t fDiffPhiMax; // Maximum delta phi allowed among corr. pixels
86 Double_t fMeanPSelTrk; // GeV, mean P for tracks with dphi<0.01 rad
87 Double_t fMeanPtSelTrk; // GeV, mean Pt for tracks with dphi<0.01 rad
cd1c2af1 88 TBits fUsedCluster; // flag for used clusters in vertex calculation
89 TH1F *fZHisto; //! histogram with coarse z distribution
242c8025 90 Double_t fDCAforPileup; // Minimum DCA to 1st vertex for pileup tracklets
c238ba33 91 Double_t fDiffPhiforPileup; // Cut on delta phi for pileup
242c8025 92 Double_t fBinSizeR; // Histo3D bin size along radius
93 Double_t fBinSizeZ; // Histo3D bin size along z
cd1c2af1 94 UShort_t fPileupAlgo; // Algo for pileup identification
95 // 0->VertexerZ pileup algo
96 // 1->Unused RecPoints algo
6a0d56b8 97
c238ba33 98 ClassDef(AliITSVertexer3D,11);
70c95f95 99
100};
101
102#endif