Fast generator for prompt photons (Serguei Kiselev)

[u/mrichter/AliRoot.git] / ITS / AliITSVertexer3D.h

diff --git a/ITS/AliITSVertexer3D.h b/ITS/AliITSVertexer3D.h
index 879e8a97727f93b2636a527c91a4ae71a80e0886..7fdb0b0b350c78222e75496654a52b8eb4481d6b 100644 (file)
--- a/ITS/AliITSVertexer3D.h
+++ b/ITS/AliITSVertexer3D.h
@@ -1,7 +1,7 @@
 #ifndef ALIITSVERTEXER3D_H
 #define ALIITSVERTEXER3D_H
 
-#include<AliITSVertexerZ.h>
+#include<AliITSVertexer.h>
 
 ///////////////////////////////////////////////////////////////////
 //                                                               //
@@ -9,39 +9,55 @@
 //                                                               //
 ///////////////////////////////////////////////////////////////////
 
-class AliVertex;
-class TClonesArray;
+/* $Id$ */
+
+#include <TClonesArray.h>
+#include <AliESDVertex.h>
+#include <TH3F.h>
 
 class AliITSVertexer3D : public AliITSVertexer {
 
  public:
 
   AliITSVertexer3D();
-  AliITSVertexer3D(TString filename);
   virtual ~AliITSVertexer3D();
-  virtual AliESDVertex* FindVertexForCurrentEvent(Int_t evnumb);
-  virtual void FindVertices();
-  AliVertex *GetVertex3D() const {return fVert3D;}
-  virtual void MakeTracklet(Double_t *pA, Double_t *pB, Int_t &nolines);
-  virtual void MakeTracklet(Float_t *pA, Float_t *pB, Int_t &nolines);
+  virtual AliESDVertex* FindVertexForCurrentEvent(TTree *itsClusterTree);
+  AliESDVertex GetVertex3D() const {return fVert3D;}
   virtual void PrintStatus() const;
+  void SetWideFiducialRegion(Float_t dz = 20.0, Float_t dr=2.5){
+    SetCoarseMaxRCut(dr);
+    SetZCutDiamond(dz);
+  }
+  void SetNarrowFiducialRegion(Float_t dz = 0.5, Float_t dr=0.5){
+    SetMaxRCut(dr);
+    SetMaxZCut(dz);
+  }
+  void SetDeltaPhiCuts(Float_t dphiloose=0.5, Float_t dphitight=0.01){
+    SetCoarseDiffPhiCut(dphiloose);
+    SetDiffPhiMax(dphitight);
+  }
   void SetCoarseDiffPhiCut(Float_t dphi = 0.5){fCoarseDiffPhiCut=dphi;}
   void SetCoarseMaxRCut(Float_t rad = 2.5){fCoarseMaxRCut=rad;}
   void SetMaxRCut(Float_t rad = 0.5){fMaxRCut=rad;}
-  void SetZCutDiamond(Float_t zcut = 5.3){fZCutDiamond=zcut;}
+  void SetZCutDiamond(Float_t zcut = 20.0){fZCutDiamond=zcut;}
   void SetMaxZCut(Float_t dz = 0.5){fMaxZCut=dz;}
-  void SetDCAcut(Float_t dca=0.1){fDCAcut=dca;} 
+  void SetDCACut(Float_t dca=0.1){fDCAcut=dca;} 
   void SetDiffPhiMax(Float_t pm = 0.01){fDiffPhiMax = pm;}
+  void SetMeanPSelTracks(Float_t pGeV=0.875){fMeanPSelTrk = pGeV;}
+  void SetMeanPtSelTracks(Float_t ptGeV=0.630){fMeanPtSelTrk = ptGeV;}
+  void SetMeanPPtSelTracks(Float_t fieldTesla);
 
 protected:
   AliITSVertexer3D(const AliITSVertexer3D& vtxr);
   AliITSVertexer3D& operator=(const AliITSVertexer3D& /* vtxr */);
-  Int_t FindTracklets(Int_t evnumber, Int_t optCuts);
-  void Find3DVertex();
+  Int_t FindTracklets(TTree *itsClusterTree, Int_t optCuts);
   Int_t Prepare3DVertex(Int_t optCuts);
+  void ResetVert3D();
+  void FindPeaks(TH3F* histo, Double_t *peak, Int_t &nOfTracklets, Int_t &nOfTimes);
+
 
-  TClonesArray *fLines;      //! array of tracklets
-  AliVertex *fVert3D;        // 3D Vertex
+  TClonesArray fLines;      //! array of tracklets
+  AliESDVertex fVert3D;        // 3D Vertex
   Float_t fCoarseDiffPhiCut; // loose cut on DeltaPhi for RecPoint matching 
   Float_t fCoarseMaxRCut; // cut on tracklet DCA to Z axis
   Float_t fMaxRCut; // cut on tracklet DCA to beam axis
@@ -49,7 +65,10 @@ protected:
   Float_t fMaxZCut;   // cut on Z distance from estimated vertex
   Float_t fDCAcut; // cut on tracklet to tracklet and tracklet to vertex DCA
   Float_t fDiffPhiMax;     // Maximum delta phi allowed among corr. pixels
-  ClassDef(AliITSVertexer3D,2);
+  Float_t fMeanPSelTrk; // GeV, mean P for tracks with dphi<0.01 rad
+  Float_t fMeanPtSelTrk; // GeV, mean Pt for tracks with dphi<0.01 rad
+
+  ClassDef(AliITSVertexer3D,6);
 
 };