[u/mrichter/AliRoot.git] / PWG3 / vertexingHF / 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
//  Contact: andrea.dainese@lnl.infn.it
//-------------------------------------------------------------------------

#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 FindCandidatesESDtoAOD(AliESDEvent *esd,
			      TClonesArray *aodVerticesHFTClArr,
			      TClonesArray *aodD0toKpiTClArr,
			      TClonesArray *aodJPSItoEleTClArr,
			      TClonesArray *aodCharm3ProngTClArr,
			      TClonesArray *aodCharm4ProngTClArr);
  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 SetUseTRef() { fUseTRef=kTRUE; }
  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; }
  Bool_t GetD0toKpi() { return fD0toKpi; }
  Bool_t GetJPSItoEle() { return fJPSItoEle; }
  Bool_t Get3Prong() { return f3Prong; }
  Bool_t Get4Prong() { return f4Prong; }
  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 fUseTRef;      // use TRef to store in the AOD 

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