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(Double_t dz = 40.0, Double_t dr=2.5){
  32     SetCoarseMaxRCut(dr);
  33     SetZCutDiamond(dz);
  34   }
  35   void SetNarrowFiducialRegion(Double_t dz = 0.5, Double_t dr=0.5){
  36     SetMaxRCut(dr);
  37     SetMaxZCut(dz);
  38   }
  39   void SetDeltaPhiCuts(Double_t dphiloose=0.5, Double_t dphitight=0.025){
  40     SetCoarseDiffPhiCut(dphiloose);
  41     SetDiffPhiMax(dphitight);
  42   }
  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;}
  54   void SetMeanPPtSelTracks(Double_t fieldTesla);
  55   void SetMinDCAforPileup(Double_t mindist=0.1) {fDCAforPileup=mindist;}
  56   void SetDeltaPhiforPileup(Double_t dphi=0.01) {fDiffPhiforPileup=dphi;}
  57   void SetPileupAlgo(UShort_t optalgo=0){fPileupAlgo=optalgo;}
  58   void SetBinSizeR(Double_t siz=0.1){fBinSizeR=siz;}
  59   void SetBinSizeZ(Double_t siz=0.8){fBinSizeZ=siz;}
  60
  61 protected:
  62   AliITSVertexer3D(const AliITSVertexer3D& vtxr);
  63   AliITSVertexer3D& operator=(const AliITSVertexer3D& /* vtxr */);
  64   Int_t FindTracklets(TTree *itsClusterTree, Int_t optCuts);
  65   Int_t Prepare3DVertex(Int_t optCuts);
  66   void ResetVert3D();
  67   void FindPeaks(TH3F* histo, Double_t *peak, Int_t &nOfTracklets, Int_t &nOfTimes);
  68   void PileupFromZ();
  69   void MarkUsedClusters();
  70   Int_t RemoveTracklets();
  71   void  FindOther3DVertices(TTree *itsClusterTree);
  72
  73   enum {kMaxCluPerMod=250};
  74
  75   TClonesArray fLines;      //! array of tracklets
  76   AliESDVertex fVert3D;        // 3D Vertex
  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
  88   TBits   fUsedCluster;  // flag for used clusters in vertex calculation
  89   TH1F *fZHisto;           //! histogram with coarse z distribution
  90   Double_t  fDCAforPileup;  // Minimum DCA to 1st vertex for pileup tracklets
  91   Double_t  fDiffPhiforPileup;  // Cut on delta phi for pileup
  92   Double_t  fBinSizeR;      // Histo3D bin size along radius
  93   Double_t  fBinSizeZ;      // Histo3D bin size along z
  94   UShort_t fPileupAlgo;    // Algo for pileup identification
  95                            // 0->VertexerZ pileup algo
  96                            // 1->Unused RecPoints algo
  97
  98   ClassDef(AliITSVertexer3D,11);
  99
 100 };
 101
 102 #endif