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