Two methods for pile-up detection (F. Prino)
[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   AliESDVertex GetVertex3D() const {return fVert3D;}
27   virtual void PrintStatus() const;
28   void SetWideFiducialRegion(Float_t dz = 20.0, Float_t dr=2.5){
29     SetCoarseMaxRCut(dr);
30     SetZCutDiamond(dz);
31   }
32   void SetNarrowFiducialRegion(Float_t dz = 0.5, Float_t dr=0.5){
33     SetMaxRCut(dr);
34     SetMaxZCut(dz);
35   }
36   void SetDeltaPhiCuts(Float_t dphiloose=0.5, Float_t dphitight=0.01){
37     SetCoarseDiffPhiCut(dphiloose);
38     SetDiffPhiMax(dphitight);
39   }
40   void SetCoarseDiffPhiCut(Float_t dphi = 0.5){fCoarseDiffPhiCut=dphi;}
41   void SetCoarseMaxRCut(Float_t rad = 2.5){fCoarseMaxRCut=rad;}
42   void SetMaxRCut(Float_t rad = 0.5){fMaxRCut=rad;}
43   void SetZCutDiamond(Float_t zcut = 20.0){fZCutDiamond=zcut;}
44   void SetMaxZCut(Float_t dz = 0.5){fMaxZCut=dz;}
45   void SetDCACut(Float_t dca=0.1){fDCAcut=dca;} 
46   void SetDiffPhiMax(Float_t pm = 0.01){fDiffPhiMax = pm;}
47   void SetMeanPSelTracks(Float_t pGeV=0.875){fMeanPSelTrk = pGeV;}
48   void SetMeanPtSelTracks(Float_t ptGeV=0.630){fMeanPtSelTrk = ptGeV;}
49   void SetMeanPPtSelTracks(Float_t fieldTesla);
50   void SetMinDCAforPileup(Float_t mindist=0.1) {fDCAforPileup=mindist;}
51   void SetPileupAlgo(UShort_t optalgo=0){fPileupAlgo=optalgo;}
52
53 protected:
54   AliITSVertexer3D(const AliITSVertexer3D& vtxr);
55   AliITSVertexer3D& operator=(const AliITSVertexer3D& /* vtxr */);
56   Int_t FindTracklets(TTree *itsClusterTree, Int_t optCuts);
57   Int_t Prepare3DVertex(Int_t optCuts);
58   void ResetVert3D();
59   void FindPeaks(TH3F* histo, Double_t *peak, Int_t &nOfTracklets, Int_t &nOfTimes);
60   void PileupFromZ();
61   void MarkUsedClusters();
62   Int_t RemoveTracklets();
63   void  FindOther3DVertices(TTree *itsClusterTree);
64
65   enum {kMaxCluPerMod=250};
66
67   TClonesArray fLines;      //! array of tracklets
68   AliESDVertex fVert3D;        // 3D Vertex
69   Float_t fCoarseDiffPhiCut; // loose cut on DeltaPhi for RecPoint matching 
70   Float_t fCoarseMaxRCut; // cut on tracklet DCA to Z axis
71   Float_t fMaxRCut; // cut on tracklet DCA to beam axis
72   Float_t fZCutDiamond;   // cut on +-Z of the diamond
73   Float_t fMaxZCut;   // cut on Z distance from estimated vertex
74   Float_t fDCAcut; // cut on tracklet to tracklet and tracklet to vertex DCA
75   Float_t fDiffPhiMax;     // Maximum delta phi allowed among corr. pixels
76   Float_t fMeanPSelTrk; // GeV, mean P for tracks with dphi<0.01 rad
77   Float_t fMeanPtSelTrk; // GeV, mean Pt for tracks with dphi<0.01 rad
78   TBits   fUsedCluster;  // flag for used clusters in vertex calculation
79   TH1F *fZHisto;           //! histogram with coarse z distribution
80   Float_t  fDCAforPileup;  // Minimum DCA to 1st vertex for pileup tracklets 
81   UShort_t fPileupAlgo;    // Algo for pileup identification
82                            // 0->VertexerZ pileup algo
83                            // 1->Unused RecPoints algo
84
85   ClassDef(AliITSVertexer3D,7);
86
87 };
88
89 #endif