[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 "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 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]); 
  //
 private:
  //
  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)

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