[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 "AliESD.h"
#include "AliESDVertex.h"
#include "AliAODRecoDecayHF.h"
#include "AliAODRecoDecayHF2Prong.h"
#include "AliAODRecoDecayHF3Prong.h"
#include "AliAODRecoDecayHF4Prong.h"

//-----------------------------------------------------------------------------
class AliAnalysisVertexingHF : public TNamed {
 public:
  //
  AliAnalysisVertexingHF();
  virtual ~AliAnalysisVertexingHF();

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