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

#ifndef ALIITSVERTEXER3D_H
#define ALIITSVERTEXER3D_H

#include<AliITSVertexer.h>

///////////////////////////////////////////////////////////////////
//                                                               //
// Class for primary vertex finding  (3D reconstruction)         //
//                                                               //
///////////////////////////////////////////////////////////////////

/* $Id$ */

#include <TClonesArray.h>
#include <AliESDVertex.h>
#include <TH3F.h>

class AliITSVertexer3D : public AliITSVertexer {

 public:

  AliITSVertexer3D();
  virtual ~AliITSVertexer3D();
  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 = 20.0){fZCutDiamond=zcut;}
  void SetMaxZCut(Float_t dz = 0.5){fMaxZCut=dz;}
  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(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
  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
  Float_t fZCutDiamond;   // cut on +-Z of the diamond
  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
  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);

};

#endif
Commit	Line	Data
	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
	18	class AliITSVertexer3D : public AliITSVertexer {
	19
	20	public:
	21
	22	AliITSVertexer3D();
	23	virtual ~AliITSVertexer3D();
	24	virtual AliESDVertex* FindVertexForCurrentEvent(TTree *itsClusterTree);
	25	AliESDVertex GetVertex3D() const {return fVert3D;}
	26	virtual void PrintStatus() const;
	27	void SetWideFiducialRegion(Float_t dz = 20.0, Float_t dr=2.5){
	28	SetCoarseMaxRCut(dr);
	29	SetZCutDiamond(dz);
	30	}
	31	void SetNarrowFiducialRegion(Float_t dz = 0.5, Float_t dr=0.5){
	32	SetMaxRCut(dr);
	33	SetMaxZCut(dz);
	34	}
	35	void SetDeltaPhiCuts(Float_t dphiloose=0.5, Float_t dphitight=0.01){
	36	SetCoarseDiffPhiCut(dphiloose);
	37	SetDiffPhiMax(dphitight);
	38	}
	39	void SetCoarseDiffPhiCut(Float_t dphi = 0.5){fCoarseDiffPhiCut=dphi;}
	40	void SetCoarseMaxRCut(Float_t rad = 2.5){fCoarseMaxRCut=rad;}
	41	void SetMaxRCut(Float_t rad = 0.5){fMaxRCut=rad;}
	42	void SetZCutDiamond(Float_t zcut = 20.0){fZCutDiamond=zcut;}
	43	void SetMaxZCut(Float_t dz = 0.5){fMaxZCut=dz;}
	44	void SetDCACut(Float_t dca=0.1){fDCAcut=dca;}
	45	void SetDiffPhiMax(Float_t pm = 0.01){fDiffPhiMax = pm;}
	46	void SetMeanPSelTracks(Float_t pGeV=0.875){fMeanPSelTrk = pGeV;}
	47	void SetMeanPtSelTracks(Float_t ptGeV=0.630){fMeanPtSelTrk = ptGeV;}
	48	void SetMeanPPtSelTracks(Float_t fieldTesla);
	49
	50	protected:
	51	AliITSVertexer3D(const AliITSVertexer3D& vtxr);
	52	AliITSVertexer3D& operator=(const AliITSVertexer3D& /* vtxr */);
	53	Int_t FindTracklets(TTree *itsClusterTree, Int_t optCuts);
	54	Int_t Prepare3DVertex(Int_t optCuts);
	55	void ResetVert3D();
	56	void FindPeaks(TH3F* histo, Double_t *peak, Int_t &nOfTracklets, Int_t &nOfTimes);
	57
	58
	59	TClonesArray fLines; //! array of tracklets
	60	AliESDVertex fVert3D; // 3D Vertex
	61	Float_t fCoarseDiffPhiCut; // loose cut on DeltaPhi for RecPoint matching
	62	Float_t fCoarseMaxRCut; // cut on tracklet DCA to Z axis
	63	Float_t fMaxRCut; // cut on tracklet DCA to beam axis
	64	Float_t fZCutDiamond; // cut on +-Z of the diamond
	65	Float_t fMaxZCut; // cut on Z distance from estimated vertex
	66	Float_t fDCAcut; // cut on tracklet to tracklet and tracklet to vertex DCA
	67	Float_t fDiffPhiMax; // Maximum delta phi allowed among corr. pixels
	68	Float_t fMeanPSelTrk; // GeV, mean P for tracks with dphi<0.01 rad
	69	Float_t fMeanPtSelTrk; // GeV, mean Pt for tracks with dphi<0.01 rad
	70
	71	ClassDef(AliITSVertexer3D,6);
	72
	73	};
	74
	75	#endif