[u/mrichter/AliRoot.git] / PWG3 / AliAnalysisVertexingHF.h

#ifndef AliAnalysisVertexingHF_H
#define AliAnalysisVertexingHF_H
/* Copyright(c) 1998-2007, ALICE Experiment at CERN, All rights reserved. *
 * See cxx source for full Copyright notice                               */

//-------------------------------------------------------------------------
//                      Class AliAnalysisVertexingHF
//            Reconstruction of heavy-flavour decay candidates
//      
//  Origin: E.Bruna, G.E.Bruno, A.Dainese, F.Prino, R.Romita
//-------------------------------------------------------------------------

#include <TNamed.h>
#include <TTree.h>
#include "AliESDEvent.h"
#include "AliESDVertex.h"
#include "AliAODVertex.h"
#include "AliAODRecoDecayHF.h"
#include "AliAODRecoDecayHF2Prong.h"
#include "AliAODRecoDecayHF3Prong.h"
#include "AliAODRecoDecayHF4Prong.h"

//-----------------------------------------------------------------------------
class AliAnalysisVertexingHF : public TNamed {
 public:
  //
  AliAnalysisVertexingHF();
  AliAnalysisVertexingHF(const AliAnalysisVertexingHF& source);
  AliAnalysisVertexingHF& operator=(const AliAnalysisVertexingHF& source); 
  virtual ~AliAnalysisVertexingHF();

  void FindCandidates(AliESDEvent *esd,TTree treeout[]);
  AliAODRecoDecayHF2Prong* Make2Prong(TObjArray *twoTrackArray1,AliESDEvent *esd,
				     AliESDVertex *vertexp1n1,Double_t dcap1n1,
				     Bool_t &okD0,Bool_t &okJPSI) const;
  AliAODRecoDecayHF3Prong* Make3Prong(TObjArray *threeTrackArray,AliESDEvent *esd,
				     AliESDVertex *vertexp1n1,
				     AliESDVertex *vertexp2n1,
				     Double_t dcap1n1,Double_t dcap2n1,Double_t dcap1p2,
				     Bool_t &ok3Prong) const;
  AliAODRecoDecayHF4Prong* Make4Prong(TObjArray *fourTrackArray,AliESDEvent *esd,
			       AliESDVertex *vertexp1n1,
			       AliESDVertex *vertexp2n1,
			       Double_t dcap1n1,Double_t dcap1n2,
			       Double_t dcap2n1,
			       Bool_t &ok4Prong) const;


  void SetDebug(Int_t debug=0) {fDebug=debug;}
  void PrintStatus() const;
  void SetSecVtxWithKF() { fSecVtxWithKF=kTRUE; }
  void SetD0toKpiOn() { fD0toKpi=kTRUE; }
  void SetD0toKpiOff() { fD0toKpi=kFALSE; }
  void SetJPSItoEleOn() { fJPSItoEle=kTRUE; }
  void SetJPSItoEleOff() { fJPSItoEle=kFALSE; }
  void Set3ProngOn() { f3Prong=kTRUE; }
  void Set3ProngOff() { f3Prong=kFALSE; }
  void Set4ProngOn() { f4Prong=kTRUE; }
  void Set4ProngOff() { f4Prong=kFALSE; }
  void SetRecoPrimVtxSkippingTrks() 
    { fRecoPrimVtxSkippingTrks=kTRUE; fRmTrksFromPrimVtx=kFALSE;}
  void SetRmTrksFromPrimVtx() 
    {fRmTrksFromPrimVtx=kTRUE; fRecoPrimVtxSkippingTrks=kFALSE; }
  void SetITSrefitRequired() { fITSrefit=kTRUE; }
  void SetITSrefitNotRequired() { fITSrefit=kFALSE; }
  void SetBothSPDRequired() { fBothSPD=kTRUE; }
  void SetBothSPDNotRequired() {fBothSPD=kFALSE;}
  void SetMinITSCls(Int_t n=6) { fMinITSCls=n; }
  void SetMinPtCut(Double_t pt=0.) { fMinPtCut=pt; }
  void SetMind0Cut(Double_t d0=0.) { fMind0rphiCut=d0; } 
  void SetD0toKpiCuts(Double_t cut0=1000.,Double_t cut1=100000.,
		      Double_t cut2=1.1,Double_t cut3=0.,Double_t cut4=0.,
		      Double_t cut5=100000.,Double_t cut6=100000.,
		      Double_t cut7=100000000.,Double_t cut8=-1.1); 
  void SetD0toKpiCuts(const Double_t cuts[9]); 
  void SetBtoJPSICuts(Double_t cut0=1000.,Double_t cut1=100000.,
		      Double_t cut2=1.1,Double_t cut3=0.,Double_t cut4=0.,
		      Double_t cut5=100000.,Double_t cut6=100000.,
		      Double_t cut7=100000000.,Double_t cut8=-1.1); 
  void SetBtoJPSICuts(const Double_t cuts[9]); 
  void SetDplusCuts(Double_t cut0=1000.,Double_t cut1=0.,
		    Double_t cut2=0.,Double_t cut3=0.,Double_t cut4=0.,
		    Double_t cut5=0.,Double_t cut6=10000000000.,
		    Double_t cut7=0.,Double_t cut8=0.,
		    Double_t cut9=-1.1,Double_t cut10=0.,
		    Double_t cut11=0.); 
  void SetDplusCuts(const Double_t cuts[12]); 
  void SetDsCuts(Double_t cut0=0.); 
  void SetDsCuts(const Double_t cuts[1]); 
  void SetLcCuts(Double_t cut0=0.); 
  void SetLcCuts(const Double_t cuts[1]); 
  //
 private:
  //
  Double_t fBzkG; // z componenent of field in kG

  Bool_t fSecVtxWithKF; // if kTRUE use KF vertexer, else AliVertexerTracks

  Bool_t fRecoPrimVtxSkippingTrks; // flag for primary vertex reco on the fly
                                   // for each candidate, w/o its daughters
  Bool_t fRmTrksFromPrimVtx; // flag for fast removal of daughters from 
                             // the primary vertex

  AliESDVertex *fV1; // primary vertex

  Int_t  fDebug; // enable verbose mode

  // flag to enable candidates production
  Bool_t fD0toKpi; 
  Bool_t fJPSItoEle;
  Bool_t f3Prong;
  Bool_t f4Prong;

  // single-track cuts
  Bool_t   fITSrefit;   // require kITSrefit
  Bool_t   fBothSPD;    // require both SPD layers
  Int_t    fMinITSCls;  // minimum number of ITS clusters
  Double_t fMinPtCut;   // minimum track pt [GeV/c]
  Double_t fMind0rphiCut;  // minimum track |rphi impact parameter| [cm] 
  // candidates cuts
  Double_t fD0toKpiCuts[9]; // cuts on D0->Kpi candidates
                  // (to be passed to AliAODRecoDecayHF2Prong::SelectD0())
                          // 0 = inv. mass half width [GeV]   
                          // 1 = dca [cm]
                          // 2 = cosThetaStar 
                          // 3 = pTK [GeV/c]
                          // 4 = pTPi [GeV/c]
                          // 5 = d0K [cm]   upper limit!
                          // 6 = d0Pi [cm]  upper limit!
                          // 7 = d0d0 [cm^2]
                          // 8 = cosThetaPoint
  Double_t fBtoJPSICuts[9]; // cuts on JPSI candidates
                  // (to be passed to AliAODRecoDecayHF2Prong::SelectBtoJPSI())
                          // 0 = inv. mass half width [GeV]   
                          // 1 = dca [cm]
                          // 2 = cosThetaStar (negative electron)
                          // 3 = pTP [GeV/c]
                          // 4 = pTN [GeV/c]
                          // 5 = d0O [cm]   upper limit!
                          // 6 = d0N [cm]  upper limit!
                          // 7 = d0d0 [cm^2]
                          // 8 = cosThetaPoint
  Double_t fDplusCuts[12]; // cuts on Dplus candidates
                  // (to be passed to AliAODRecoDecayHF2Prong::SelectDplus())
                          // 0 = inv. mass half width [GeV]   
                          // 1 = pTK [GeV/c]
                          // 2 = pTPi [GeV/c]
                          // 3 = d0K [cm]   lower limit!
                          // 4 = d0Pi [cm]  lower limit!
                          // 5 = dist12 (cm)
                          // 6 = sigmavert (cm)
                          // 7 = dist prim-sec (cm)
                          // 8 = pM=Max{pT1,pT2,pT3} (GeV/c)
                          // 9 = cosThetaPoint
                          // 10 = Sum d0^2 (cm^2)
                          // 11 = dca cut (cm)
  Double_t fDsCuts[1]; // cuts on Ds candidates
                       // (to be passed to AliAODRecoDecayHF2Prong::SelectDs())
                       // 0 = inv. mass half width [GeV]   
  Double_t fLcCuts[1]; // cuts on Lambdac candidates
                       // (to be passed to AliAODRecoDecayHF2Prong::SelectLc())
                       // 0 = inv. mass half width [GeV]   

  //
  AliESDVertex* OwnPrimaryVertex(Int_t ntrks,TObjArray *trkArray,AliESDEvent *esd) const;
  Bool_t SelectInvMass(Int_t decay,Int_t nprongs,
		       Double_t *px,Double_t *py,Double_t *pz) const;
  void SelectTracks(AliESDEvent *esd,
		    TObjArray &trksP,Int_t &nTrksP,
		    TObjArray &trksN,Int_t &nTrksN) const;
  void SetPrimaryVertex(AliESDVertex* v1) { fV1 = v1; }
  Bool_t SingleTrkCuts(AliESDtrack& trk) const;
  AliESDVertex* ReconstructSecondaryVertex(TObjArray *trkArray) const;
  //
  ClassDef(AliAnalysisVertexingHF,2)  // Reconstruction of HF decay candidates
};


#endif
Commit	Line	Data
6a213b59	1	#ifndef AliAnalysisVertexingHF_H
	2	#define AliAnalysisVertexingHF_H
	3	/* Copyright(c) 1998-2007, ALICE Experiment at CERN, All rights reserved. *
	4	* See cxx source for full Copyright notice */
	5
	6	//-------------------------------------------------------------------------
	7	// Class AliAnalysisVertexingHF
	8	// Reconstruction of heavy-flavour decay candidates
	9	//
	10	// Origin: E.Bruna, G.E.Bruno, A.Dainese, F.Prino, R.Romita
	11	//-------------------------------------------------------------------------
	12
	13	#include <TNamed.h>
	14	#include <TTree.h>
5e6f195c	15	#include "AliESDEvent.h"
6a213b59	16	#include "AliESDVertex.h"
b82f6d67	17	#include "AliAODVertex.h"
6a213b59	18	#include "AliAODRecoDecayHF.h"
	19	#include "AliAODRecoDecayHF2Prong.h"
	20	#include "AliAODRecoDecayHF3Prong.h"
	21	#include "AliAODRecoDecayHF4Prong.h"
	22
	23	//-----------------------------------------------------------------------------
	24	class AliAnalysisVertexingHF : public TNamed {
	25	public:
	26	//
	27	AliAnalysisVertexingHF();
13977a79	28	AliAnalysisVertexingHF(const AliAnalysisVertexingHF& source);
13977a79	29	AliAnalysisVertexingHF& operator=(const AliAnalysisVertexingHF& source);
6a213b59	30	virtual ~AliAnalysisVertexingHF();
6a213b59	31
5e6f195c	32	void FindCandidates(AliESDEvent *esd,TTree treeout[]);
5e6f195c	33	AliAODRecoDecayHF2Prong* Make2Prong(TObjArray twoTrackArray1,AliESDEvent esd,
6a213b59	34	AliESDVertex *vertexp1n1,Double_t dcap1n1,
6a213b59	35	Bool_t &okD0,Bool_t &okJPSI) const;
5e6f195c	36	AliAODRecoDecayHF3Prong* Make3Prong(TObjArray threeTrackArray,AliESDEvent esd,
6a213b59	37	AliESDVertex *vertexp1n1,
	38	AliESDVertex *vertexp2n1,
	39	Double_t dcap1n1,Double_t dcap2n1,Double_t dcap1p2,
	40	Bool_t &ok3Prong) const;
5e6f195c	41	AliAODRecoDecayHF4Prong* Make4Prong(TObjArray fourTrackArray,AliESDEvent esd,
6a213b59	42	AliESDVertex *vertexp1n1,
	43	AliESDVertex *vertexp2n1,
	44	Double_t dcap1n1,Double_t dcap1n2,
	45	Double_t dcap2n1,
	46	Bool_t &ok4Prong) const;
	47
	48
	49	void SetDebug(Int_t debug=0) {fDebug=debug;}
	50	void PrintStatus() const;
b82f6d67	51	void SetSecVtxWithKF() { fSecVtxWithKF=kTRUE; }
6a213b59	52	void SetD0toKpiOn() { fD0toKpi=kTRUE; }
	53	void SetD0toKpiOff() { fD0toKpi=kFALSE; }
	54	void SetJPSItoEleOn() { fJPSItoEle=kTRUE; }
	55	void SetJPSItoEleOff() { fJPSItoEle=kFALSE; }
	56	void Set3ProngOn() { f3Prong=kTRUE; }
	57	void Set3ProngOff() { f3Prong=kFALSE; }
	58	void Set4ProngOn() { f4Prong=kTRUE; }
	59	void Set4ProngOff() { f4Prong=kFALSE; }
	60	void SetRecoPrimVtxSkippingTrks()
	61	{ fRecoPrimVtxSkippingTrks=kTRUE; fRmTrksFromPrimVtx=kFALSE;}
	62	void SetRmTrksFromPrimVtx()
	63	{fRmTrksFromPrimVtx=kTRUE; fRecoPrimVtxSkippingTrks=kFALSE; }
	64	void SetITSrefitRequired() { fITSrefit=kTRUE; }
	65	void SetITSrefitNotRequired() { fITSrefit=kFALSE; }
	66	void SetBothSPDRequired() { fBothSPD=kTRUE; }
	67	void SetBothSPDNotRequired() {fBothSPD=kFALSE;}
	68	void SetMinITSCls(Int_t n=6) { fMinITSCls=n; }
	69	void SetMinPtCut(Double_t pt=0.) { fMinPtCut=pt; }
	70	void SetMind0Cut(Double_t d0=0.) { fMind0rphiCut=d0; }
	71	void SetD0toKpiCuts(Double_t cut0=1000.,Double_t cut1=100000.,
	72	Double_t cut2=1.1,Double_t cut3=0.,Double_t cut4=0.,
	73	Double_t cut5=100000.,Double_t cut6=100000.,
	74	Double_t cut7=100000000.,Double_t cut8=-1.1);
	75	void SetD0toKpiCuts(const Double_t cuts[9]);
	76	void SetBtoJPSICuts(Double_t cut0=1000.,Double_t cut1=100000.,
	77	Double_t cut2=1.1,Double_t cut3=0.,Double_t cut4=0.,
	78	Double_t cut5=100000.,Double_t cut6=100000.,
	79	Double_t cut7=100000000.,Double_t cut8=-1.1);
	80	void SetBtoJPSICuts(const Double_t cuts[9]);
	81	void SetDplusCuts(Double_t cut0=1000.,Double_t cut1=0.,
	82	Double_t cut2=0.,Double_t cut3=0.,Double_t cut4=0.,
	83	Double_t cut5=0.,Double_t cut6=10000000000.,
	84	Double_t cut7=0.,Double_t cut8=0.,
	85	Double_t cut9=-1.1,Double_t cut10=0.,
	86	Double_t cut11=0.);
	87	void SetDplusCuts(const Double_t cuts[12]);
b82f6d67	88	void SetDsCuts(Double_t cut0=0.);
	89	void SetDsCuts(const Double_t cuts[1]);
	90	void SetLcCuts(Double_t cut0=0.);
	91	void SetLcCuts(const Double_t cuts[1]);
6a213b59	92	//
	93	private:
	94	//
b82f6d67	95	Double_t fBzkG; // z componenent of field in kG
	96
	97	Bool_t fSecVtxWithKF; // if kTRUE use KF vertexer, else AliVertexerTracks
	98
6a213b59	99	Bool_t fRecoPrimVtxSkippingTrks; // flag for primary vertex reco on the fly
	100	// for each candidate, w/o its daughters
	101	Bool_t fRmTrksFromPrimVtx; // flag for fast removal of daughters from
	102	// the primary vertex
	103
	104	AliESDVertex *fV1; // primary vertex
	105
	106	Int_t fDebug; // enable verbose mode
	107
	108	// flag to enable candidates production
	109	Bool_t fD0toKpi;
	110	Bool_t fJPSItoEle;
	111	Bool_t f3Prong;
	112	Bool_t f4Prong;
	113
	114	// single-track cuts
	115	Bool_t fITSrefit; // require kITSrefit
	116	Bool_t fBothSPD; // require both SPD layers
	117	Int_t fMinITSCls; // minimum number of ITS clusters
	118	Double_t fMinPtCut; // minimum track pt [GeV/c]
	119	Double_t fMind0rphiCut; // minimum track \|rphi impact parameter\| [cm]
	120	// candidates cuts
	121	Double_t fD0toKpiCuts[9]; // cuts on D0->Kpi candidates
	122	// (to be passed to AliAODRecoDecayHF2Prong::SelectD0())
	123	// 0 = inv. mass half width [GeV]
	124	// 1 = dca [cm]
	125	// 2 = cosThetaStar
	126	// 3 = pTK [GeV/c]
	127	// 4 = pTPi [GeV/c]
	128	// 5 = d0K [cm] upper limit!
	129	// 6 = d0Pi [cm] upper limit!
	130	// 7 = d0d0 [cm^2]
	131	// 8 = cosThetaPoint
	132	Double_t fBtoJPSICuts[9]; // cuts on JPSI candidates
	133	// (to be passed to AliAODRecoDecayHF2Prong::SelectBtoJPSI())
	134	// 0 = inv. mass half width [GeV]
	135	// 1 = dca [cm]
	136	// 2 = cosThetaStar (negative electron)
	137	// 3 = pTP [GeV/c]
	138	// 4 = pTN [GeV/c]
	139	// 5 = d0O [cm] upper limit!
	140	// 6 = d0N [cm] upper limit!
	141	// 7 = d0d0 [cm^2]
	142	// 8 = cosThetaPoint
	143	Double_t fDplusCuts[12]; // cuts on Dplus candidates
	144	// (to be passed to AliAODRecoDecayHF2Prong::SelectDplus())
	145	// 0 = inv. mass half width [GeV]
	146	// 1 = pTK [GeV/c]
	147	// 2 = pTPi [GeV/c]
	148	// 3 = d0K [cm] lower limit!
	149	// 4 = d0Pi [cm] lower limit!
	150	// 5 = dist12 (cm)
	151	// 6 = sigmavert (cm)
	152	// 7 = dist prim-sec (cm)
	153	// 8 = pM=Max{pT1,pT2,pT3} (GeV/c)
	154	// 9 = cosThetaPoint
	155	// 10 = Sum d0^2 (cm^2)
	156	// 11 = dca cut (cm)
b82f6d67	157	Double_t fDsCuts[1]; // cuts on Ds candidates
	158	// (to be passed to AliAODRecoDecayHF2Prong::SelectDs())
	159	// 0 = inv. mass half width [GeV]
	160	Double_t fLcCuts[1]; // cuts on Lambdac candidates
	161	// (to be passed to AliAODRecoDecayHF2Prong::SelectLc())
	162	// 0 = inv. mass half width [GeV]
6a213b59	163
6a213b59	164	//
5e6f195c	165	AliESDVertex* OwnPrimaryVertex(Int_t ntrks,TObjArray trkArray,AliESDEvent esd) const;
6a213b59	166	Bool_t SelectInvMass(Int_t decay,Int_t nprongs,
6a213b59	167	Double_t px,Double_t py,Double_t *pz) const;
5e6f195c	168	void SelectTracks(AliESDEvent *esd,
6a213b59	169	TObjArray &trksP,Int_t &nTrksP,
	170	TObjArray &trksN,Int_t &nTrksN) const;
	171	void SetPrimaryVertex(AliESDVertex* v1) { fV1 = v1; }
b82f6d67	172	Bool_t SingleTrkCuts(AliESDtrack& trk) const;
b82f6d67	173	AliESDVertex* ReconstructSecondaryVertex(TObjArray *trkArray) const;
6a213b59	174	//
b82f6d67	175	ClassDef(AliAnalysisVertexingHF,2) // Reconstruction of HF decay candidates
6a213b59	176	};
	177
	178
	179	#endif
	180
	181
	182
	183
	184
	185
	186
	187