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