[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>
#include <TBits.h>

class AliITSVertexer3D : public AliITSVertexer {

 public:

  AliITSVertexer3D();
  virtual ~AliITSVertexer3D();
  virtual AliESDVertex* FindVertexForCurrentEvent(TTree *itsClusterTree);
  void FindVertex3DIterative();
  void FindVertex3D(TTree *itsClusterTree);
  AliESDVertex GetVertex3D() const {return fVert3D;}
  virtual void PrintStatus() const;
  static Bool_t DistBetweenVertices(AliESDVertex &a, AliESDVertex &b, Double_t test, Double_t &dist);
  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.025){
    SetCoarseDiffPhiCut(dphiloose);
    SetDiffPhiMax(dphitight);
  }
  void SetCoarseDiffPhiCut(Float_t dphi = 0.5){fCoarseDiffPhiCut=dphi;}
  void SetFineDiffPhiCut(Float_t dphi = 0.05){fFineDiffPhiCut=dphi;}
  void SetCutOnPairs(Float_t cp = 0.1){fCutOnPairs = cp;}
  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.025){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);
  void SetMinDCAforPileup(Float_t mindist=0.1) {fDCAforPileup=mindist;}
  void SetPileupAlgo(UShort_t optalgo=0){fPileupAlgo=optalgo;}
  void SetBinSizeR(Float_t siz=0.1){fBinSizeR=siz;}
  void SetBinSizeZ(Float_t siz=0.8){fBinSizeZ=siz;}

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);
  void PileupFromZ();
  void MarkUsedClusters();
  Int_t RemoveTracklets();
  void  FindOther3DVertices(TTree *itsClusterTree);

  enum {kMaxCluPerMod=250};

  TClonesArray fLines;      //! array of tracklets
  AliESDVertex fVert3D;        // 3D Vertex
  Float_t fCoarseDiffPhiCut; // loose cut on DeltaPhi for RecPoint matching
  Float_t fFineDiffPhiCut; // tight value of DeltaPhi for RP matching (2nd method) 
  Float_t fCutOnPairs; //cut on distance between pairs of tracklets 
  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
  TBits   fUsedCluster;  // flag for used clusters in vertex calculation
  TH1F *fZHisto;           //! histogram with coarse z distribution
  Float_t  fDCAforPileup;  // Minimum DCA to 1st vertex for pileup tracklets 
  Float_t  fBinSizeR;      // Histo3D bin size along radius
  Float_t  fBinSizeZ;      // Histo3D bin size along z
  UShort_t fPileupAlgo;    // Algo for pileup identification
                           // 0->VertexerZ pileup algo
                           // 1->Unused RecPoints algo

  ClassDef(AliITSVertexer3D,9);

};

#endif
Commit	Line	Data
70c95f95	1	#ifndef ALIITSVERTEXER3D_H
	2	#define ALIITSVERTEXER3D_H
	3
27167524	4	#include<AliITSVertexer.h>
70c95f95	5
	6	///////////////////////////////////////////////////////////////////
	7	// //
	8	// Class for primary vertex finding (3D reconstruction) //
	9	// //
	10	///////////////////////////////////////////////////////////////////
	11
6a0d56b8	12	/* $Id$ */
6a0d56b8	13
f5f6da22	14	#include <TClonesArray.h>
6a0d56b8	15	#include <AliESDVertex.h>
7340864d	16	#include <TH3F.h>
cd1c2af1	17	#include <TBits.h>
70c95f95	18
	19	class AliITSVertexer3D : public AliITSVertexer {
	20
	21	public:
	22
	23	AliITSVertexer3D();
70c95f95	24	virtual ~AliITSVertexer3D();
308c2f7c	25	virtual AliESDVertex* FindVertexForCurrentEvent(TTree *itsClusterTree);
6b4d9537	26	void FindVertex3DIterative();
6b4d9537	27	void FindVertex3D(TTree *itsClusterTree);
6a0d56b8	28	AliESDVertex GetVertex3D() const {return fVert3D;}
70c95f95	29	virtual void PrintStatus() const;
6b4d9537	30	static Bool_t DistBetweenVertices(AliESDVertex &a, AliESDVertex &b, Double_t test, Double_t &dist);
d8ba9ddf	31	void SetWideFiducialRegion(Float_t dz = 20.0, Float_t dr=2.5){
7203e11a	32	SetCoarseMaxRCut(dr);
	33	SetZCutDiamond(dz);
	34	}
	35	void SetNarrowFiducialRegion(Float_t dz = 0.5, Float_t dr=0.5){
	36	SetMaxRCut(dr);
	37	SetMaxZCut(dz);
	38	}
2ecf7a6f	39	void SetDeltaPhiCuts(Float_t dphiloose=0.5, Float_t dphitight=0.025){
7203e11a	40	SetCoarseDiffPhiCut(dphiloose);
	41	SetDiffPhiMax(dphitight);
	42	}
70c95f95	43	void SetCoarseDiffPhiCut(Float_t dphi = 0.5){fCoarseDiffPhiCut=dphi;}
6b4d9537	44	void SetFineDiffPhiCut(Float_t dphi = 0.05){fFineDiffPhiCut=dphi;}
6b4d9537	45	void SetCutOnPairs(Float_t cp = 0.1){fCutOnPairs = cp;}
05d1294c	46	void SetCoarseMaxRCut(Float_t rad = 2.5){fCoarseMaxRCut=rad;}
05d1294c	47	void SetMaxRCut(Float_t rad = 0.5){fMaxRCut=rad;}
d8ba9ddf	48	void SetZCutDiamond(Float_t zcut = 20.0){fZCutDiamond=zcut;}
05d1294c	49	void SetMaxZCut(Float_t dz = 0.5){fMaxZCut=dz;}
7203e11a	50	void SetDCACut(Float_t dca=0.1){fDCAcut=dca;}
2ecf7a6f	51	void SetDiffPhiMax(Float_t pm = 0.025){fDiffPhiMax = pm;}
6a0d56b8	52	void SetMeanPSelTracks(Float_t pGeV=0.875){fMeanPSelTrk = pGeV;}
	53	void SetMeanPtSelTracks(Float_t ptGeV=0.630){fMeanPtSelTrk = ptGeV;}
	54	void SetMeanPPtSelTracks(Float_t fieldTesla);
cd1c2af1	55	void SetMinDCAforPileup(Float_t mindist=0.1) {fDCAforPileup=mindist;}
cd1c2af1	56	void SetPileupAlgo(UShort_t optalgo=0){fPileupAlgo=optalgo;}
7d766abb	57	void SetBinSizeR(Float_t siz=0.1){fBinSizeR=siz;}
7d766abb	58	void SetBinSizeZ(Float_t siz=0.8){fBinSizeZ=siz;}
70c95f95	59
	60	protected:
	61	AliITSVertexer3D(const AliITSVertexer3D& vtxr);
	62	AliITSVertexer3D& operator=(const AliITSVertexer3D& /* vtxr */);
308c2f7c	63	Int_t FindTracklets(TTree *itsClusterTree, Int_t optCuts);
05d1294c	64	Int_t Prepare3DVertex(Int_t optCuts);
27167524	65	void ResetVert3D();
7340864d	66	void FindPeaks(TH3F* histo, Double_t *peak, Int_t &nOfTracklets, Int_t &nOfTimes);
cd1c2af1	67	void PileupFromZ();
	68	void MarkUsedClusters();
	69	Int_t RemoveTracklets();
	70	void FindOther3DVertices(TTree *itsClusterTree);
70c95f95	71
cd1c2af1	72	enum {kMaxCluPerMod=250};
6a0d56b8	73
f5f6da22	74	TClonesArray fLines; //! array of tracklets
6a0d56b8	75	AliESDVertex fVert3D; // 3D Vertex
6b4d9537	76	Float_t fCoarseDiffPhiCut; // loose cut on DeltaPhi for RecPoint matching
	77	Float_t fFineDiffPhiCut; // tight value of DeltaPhi for RP matching (2nd method)
	78	Float_t fCutOnPairs; //cut on distance between pairs of tracklets
05d1294c	79	Float_t fCoarseMaxRCut; // cut on tracklet DCA to Z axis
05d1294c	80	Float_t fMaxRCut; // cut on tracklet DCA to beam axis
70c95f95	81	Float_t fZCutDiamond; // cut on +-Z of the diamond
05d1294c	82	Float_t fMaxZCut; // cut on Z distance from estimated vertex
70c95f95	83	Float_t fDCAcut; // cut on tracklet to tracklet and tracklet to vertex DCA
70c95f95	84	Float_t fDiffPhiMax; // Maximum delta phi allowed among corr. pixels
6a0d56b8	85	Float_t fMeanPSelTrk; // GeV, mean P for tracks with dphi<0.01 rad
6a0d56b8	86	Float_t fMeanPtSelTrk; // GeV, mean Pt for tracks with dphi<0.01 rad
cd1c2af1	87	TBits fUsedCluster; // flag for used clusters in vertex calculation
	88	TH1F *fZHisto; //! histogram with coarse z distribution
	89	Float_t fDCAforPileup; // Minimum DCA to 1st vertex for pileup tracklets
7d766abb	90	Float_t fBinSizeR; // Histo3D bin size along radius
7d766abb	91	Float_t fBinSizeZ; // Histo3D bin size along z
cd1c2af1	92	UShort_t fPileupAlgo; // Algo for pileup identification
	93	// 0->VertexerZ pileup algo
	94	// 1->Unused RecPoints algo
6a0d56b8	95
7d766abb	96	ClassDef(AliITSVertexer3D,9);
70c95f95	97
	98	};
	99
	100	#endif