[u/mrichter/AliRoot.git] / PWGHF / vertexingHF / AliCFVertexingHF.h

#ifndef ALICFVERTEXINGHF_H
#define ALICFVERTEXINGHF_H


/**************************************************************************
 * Copyright(c) 1998-2009, ALICE Experiment at CERN, All rights reserved. *
 *                                                                        *
 * Author: The ALICE Off-line Project.                                    *
 * Contributors are mentioned in the code where appropriate.              *
 *                                                                        *
 * Permission to use, copy, modify and distribute this software and its   *
 * documentation strictly for non-commercial purposes is hereby granted   *
 * without fee, provided that the above copyright notice appears in all   *
 * copies and that both the copyright notice and this permission notice   *
 * appear in the supporting documentation. The authors make no claims     *
 * about the suitability of this software for any purpose. It is          *
 * provided "as is" without express or implied warranty.                  *
 **************************************************************************/

//-----------------------------------------------------------------------
// Class for HF corrections as a function of many variables and step 
// Author : C. Zampolli, CERN
// D. Caffarri, Univ & INFN Padova caffarri@pd.infn.it
// Base class for HF Unfolding - agrelli@uu.nl
//-----------------------------------------------------------------------

#include "AliCFContainer.h"
#include "AliAODRecoDecayHF.h"

class TH1I;
class TParticle ;
class TFile ;
class TClonesArray ;
class AliAODMCParticle;
class AliAODMCHeader;
class AliAODEvent;
class THnSparse;
class TClonesArray;
class AliESDtrackCuts;


class AliCFVertexingHF : public TObject {
	public:

  enum DecayChannel{kD0toKpi = 2, kDStartoKpipi = 21, kLctoV0bachelor = 22, kDplustoKpipi = 31, kLctopKpi = 32, kDstoKKpi = 33, kD0toKpipipi = 4};

	AliCFVertexingHF() ;
	AliCFVertexingHF(TClonesArray *mcArray, UShort_t originDselection);
	AliCFVertexingHF(const AliCFVertexingHF& c);
	AliCFVertexingHF& operator= (const AliCFVertexingHF& c);

	virtual ~AliCFVertexingHF();
		
	virtual Bool_t GetGeneratedValuesFromMCParticle(Double_t* /*vectorMC*/) {return kFALSE;} 
	virtual Bool_t GetRecoValuesFromCandidate(Double_t* /*vectorReco*/) const {return kFALSE;}
	virtual Bool_t CheckMCChannelDecay() const {return kFALSE;}
	virtual Bool_t SetRecoCandidateParam(AliAODRecoDecayHF* /*recoCand*/) {return kFALSE;}

        virtual void SetDecay3Prong(Int_t /*decay*/){};
	
	void   SetFillFromGenerated(Bool_t flag) {fFillFromGenerated = flag;}
	Bool_t GetFillFromGenerated() const {return fFillFromGenerated;}
		
	void  SetNVar(Int_t nVar) {fNVar = nVar;}  

	void  SetRecoPrimVertex (Double_t zPrimVertex) {fzPrimVertex = zPrimVertex;}
	void  SetMCPrimaryVertex (Double_t zMCVertex){fzMCVertex = zMCVertex;}
	void  SetMCLabel (Int_t mcLabel) {fmcLabel = mcLabel;}
	Int_t GetMCLabel () const {return  fmcLabel;}
		
	void   SetMCCandidateParam(Int_t label);

	Int_t  MCcquarkCounting(AliAODMCParticle* mcPart) const; 
	Bool_t CheckMCPartFamily(AliAODMCParticle */*mcPart*/, TClonesArray */*mcArray*/) const;
	//	Int_t  CheckOrigin(AliAODMCParticle* mcPart) const;
	Int_t  CheckOrigin() const;
  	Bool_t CheckMCDaughters() const;
	Bool_t FillMCContainer(Double_t *containerInputMC); 
	Bool_t FillRecoContainer(Double_t *containerInput); 
	Bool_t MCAcceptanceStep() const;
	Bool_t MCRefitStep(AliAODEvent *aodEvent, AliESDtrackCuts **trackCuts) const;
	Bool_t RecoStep();

	virtual Double_t GetEtaProng(Int_t iProng) const;
	virtual Double_t GetPtProng(Int_t iProng) const;

	Double_t GetPtCand() const {return fRecoCandidate->Pt();}
	Double_t GetYCand() const {return fRecoCandidate->Y();}

	Bool_t RecoAcceptStep(AliESDtrackCuts **trackCuts) const;
	
	Bool_t FillUnfoldingMatrix(Double_t fill[4]) const;
	
	void SetNProngs(Int_t nProngs){fProngs = nProngs;}
	Int_t GetNProngs() const {return fProngs;}
	void SetDselection(UShort_t originDselection); 
	UShort_t GetDselection() {return fOriginDselection;}; 
	Int_t CheckReflexion(Char_t isSign);
	Bool_t SetLabelArray();

	void SetCentralityValue(Float_t centValue) {fCentValue = centValue;}

	virtual void SetPtAccCut(Float_t* ptAccCut);
	virtual void SetEtaAccCut(Float_t* etaAccCut);
	virtual void SetAccCut(Float_t* ptAccCut, Float_t* etaAccCut);
	virtual void SetAccCut();
	virtual void SetGeneratedDsOption(Int_t /*opt*/){
	  AliError("Method implemented in derived class");
	}

	void SetFakeSelection(Int_t fakeSel){fFakeSelection = fakeSel;}
	Int_t GetFakeSelection(){return fFakeSelection;}

	void SetRejectCandidateIfNotFromQuark(Bool_t opt){fRejectIfNoQuark=opt;}

	void SetMultiplicity(Double_t multiplicity) {fMultiplicity = multiplicity;}
	void SetConfiguration(Int_t configuration) {fConfiguration = configuration;}

	protected:
	
	TClonesArray      *fmcArray;               //mcArray candidate
	AliAODRecoDecayHF *fRecoCandidate;         // Reconstructed HF candidate 
	AliAODMCParticle  *fmcPartCandidate;
	Int_t fNDaughters;
	Int_t fNVar;                // get Number of variables for the container from the channel decay
	Double_t fzPrimVertex;       //Reco z primary vertex	
	Double_t fzMCVertex;         //MC z primary vertex
	
	Bool_t fFillFromGenerated;   //  flag to indicate whether data container should be filled  
	UShort_t fOriginDselection;      // flag to select D0 origins. 0 Only from charm 1 only from beauty 2 both from charm and beauty
	
	Bool_t fKeepDfromB;           //flag for the feed down from b quark decay. 			
	Bool_t fKeepDfromBOnly;       // flag to keep only the charm particles that comes from beauty decays
	Int_t fmcLabel;              // results of the MatchToMC()
	Int_t fProngs;               // n. of prongs	
	Int_t* fLabelArray;          //[fProngs]  array of labels

	Float_t fCentValue;         // centrality value
	Float_t* fPtAccCut;         //[fProngs] array of pt cut values for the Acceptance (MC+Rec) steps 
	Float_t* fEtaAccCut;          //[fProngs] array of eta (absolute value) cut values for the Acceptance (MC+Rec) steps 
	Int_t fFakeSelection; // fakes selection: 0 --> all, 1 --> non-fake, 2 --> fake
	Float_t fFake;              // variable to indicate whether the D0 was a fake or not: 0 --> fake, 1 --> MC, 2 --> non-fake
	Bool_t fRejectIfNoQuark;  // flag to remove events not geenrated with PYTHIA
	Double_t fMultiplicity;    // multiplicity of the event
	Int_t fConfiguration;    // configuration (slow / fast) of the CF --> different variables will be allocated (all / reduced number)

	ClassDef(AliCFVertexingHF, 7);
	
};

#endif
Commit	Line	Data
379592af	1	#ifndef ALICFVERTEXINGHF_H
	2	#define ALICFVERTEXINGHF_H
	3
2bf2e62b	4
379592af	5	/**************************************************************************
	6	* Copyright(c) 1998-2009, ALICE Experiment at CERN, All rights reserved. *
	7	* *
	8	* Author: The ALICE Off-line Project. *
	9	* Contributors are mentioned in the code where appropriate. *
	10	* *
	11	* Permission to use, copy, modify and distribute this software and its *
	12	* documentation strictly for non-commercial purposes is hereby granted *
	13	* without fee, provided that the above copyright notice appears in all *
	14	* copies and that both the copyright notice and this permission notice *
	15	* appear in the supporting documentation. The authors make no claims *
	16	* about the suitability of this software for any purpose. It is *
	17	* provided "as is" without express or implied warranty. *
	18	**************************************************************************/
	19
	20	//-----------------------------------------------------------------------
	21	// Class for HF corrections as a function of many variables and step
	22	// Author : C. Zampolli, CERN
	23	// D. Caffarri, Univ & INFN Padova caffarri@pd.infn.it
	24	// Base class for HF Unfolding - agrelli@uu.nl
	25	//-----------------------------------------------------------------------
	26
	27	#include "AliCFContainer.h"
	28	#include "AliAODRecoDecayHF.h"
	29
	30	class TH1I;
	31	class TParticle ;
	32	class TFile ;
	33	class TClonesArray ;
	34	class AliAODMCParticle;
	35	class AliAODMCHeader;
	36	class AliAODEvent;
	37	class THnSparse;
	38	class TClonesArray;
	39	class AliESDtrackCuts;
	40
	41
	42
	43	class AliCFVertexingHF : public TObject {
	44	public:
f2703bd2	45
5cd139bc	46	enum DecayChannel{kD0toKpi = 2, kDStartoKpipi = 21, kLctoV0bachelor = 22, kDplustoKpipi = 31, kLctopKpi = 32, kDstoKKpi = 33, kD0toKpipipi = 4};
f2703bd2	47
379592af	48	AliCFVertexingHF() ;
	49	AliCFVertexingHF(TClonesArray *mcArray, UShort_t originDselection);
	50	AliCFVertexingHF(const AliCFVertexingHF& c);
	51	AliCFVertexingHF& operator= (const AliCFVertexingHF& c);
	52
	53	virtual ~AliCFVertexingHF();
	54
	55	virtual Bool_t GetGeneratedValuesFromMCParticle(Double_t* /vectorMC/) {return kFALSE;}
	56	virtual Bool_t GetRecoValuesFromCandidate(Double_t* /vectorReco/) const {return kFALSE;}
	57	virtual Bool_t CheckMCChannelDecay() const {return kFALSE;}
f2703bd2	58	virtual Bool_t SetRecoCandidateParam(AliAODRecoDecayHF* /recoCand/) {return kFALSE;}
	59
	60	virtual void SetDecay3Prong(Int_t /decay/){};
379592af	61
	62	void SetFillFromGenerated(Bool_t flag) {fFillFromGenerated = flag;}
	63	Bool_t GetFillFromGenerated() const {return fFillFromGenerated;}
	64
	65	void SetNVar(Int_t nVar) {fNVar = nVar;}
	66
	67	void SetRecoPrimVertex (Double_t zPrimVertex) {fzPrimVertex = zPrimVertex;}
	68	void SetMCPrimaryVertex (Double_t zMCVertex){fzMCVertex = zMCVertex;}
	69	void SetMCLabel (Int_t mcLabel) {fmcLabel = mcLabel;}
	70	Int_t GetMCLabel () const {return fmcLabel;}
	71
	72	void SetMCCandidateParam(Int_t label);
	73
379592af	74	Int_t MCcquarkCounting(AliAODMCParticle* mcPart) const;
	75	Bool_t CheckMCPartFamily(AliAODMCParticle /mcPart/, TClonesArray /mcArray/) const;
	76	// Int_t CheckOrigin(AliAODMCParticle* mcPart) const;
	77	Int_t CheckOrigin() const;
	78	Bool_t CheckMCDaughters() const;
	79	Bool_t FillMCContainer(Double_t *containerInputMC);
	80	Bool_t FillRecoContainer(Double_t *containerInput);
	81	Bool_t MCAcceptanceStep() const;
2bf2e62b	82	Bool_t MCRefitStep(AliAODEvent aodEvent, AliESDtrackCuts *trackCuts) const;
379592af	83	Bool_t RecoStep();
379592af	84
367e9aa3	85	virtual Double_t GetEtaProng(Int_t iProng) const;
367e9aa3	86	virtual Double_t GetPtProng(Int_t iProng) const;
379592af	87
	88	Double_t GetPtCand() const {return fRecoCandidate->Pt();}
	89	Double_t GetYCand() const {return fRecoCandidate->Y();}
	90
2bf2e62b	91	Bool_t RecoAcceptStep(AliESDtrackCuts **trackCuts) const;
379592af	92
e11ae259	93	Bool_t FillUnfoldingMatrix(Double_t fill[4]) const;
379592af	94
379592af	95	void SetNProngs(Int_t nProngs){fProngs = nProngs;}
661d6c29	96	Int_t GetNProngs() const {return fProngs;}
379592af	97	void SetDselection(UShort_t originDselection);
6e2e4f50	98	UShort_t GetDselection() {return fOriginDselection;};
3ee5eb83	99	Int_t CheckReflexion(Char_t isSign);
6e2e4f50	100	Bool_t SetLabelArray();
379592af	101
b7af2639	102	void SetCentralityValue(Float_t centValue) {fCentValue = centValue;}
b7af2639	103
2bf2e62b	104	virtual void SetPtAccCut(Float_t* ptAccCut);
	105	virtual void SetEtaAccCut(Float_t* etaAccCut);
	106	virtual void SetAccCut(Float_t* ptAccCut, Float_t* etaAccCut);
	107	virtual void SetAccCut();
c83eda41	108	virtual void SetGeneratedDsOption(Int_t /opt/){
	109	AliError("Method implemented in derived class");
	110	}
2bf2e62b	111
fbec9fa9	112	void SetFakeSelection(Int_t fakeSel){fFakeSelection = fakeSel;}
1f780958	113	Int_t GetFakeSelection(){return fFakeSelection;}
fbec9fa9	114
0ada222f	115	void SetRejectCandidateIfNotFromQuark(Bool_t opt){fRejectIfNoQuark=opt;}
0ada222f	116
ec5288c3	117	void SetMultiplicity(Double_t multiplicity) {fMultiplicity = multiplicity;}
	118	void SetConfiguration(Int_t configuration) {fConfiguration = configuration;}
	119
379592af	120	protected:
	121
	122	TClonesArray *fmcArray; //mcArray candidate
	123	AliAODRecoDecayHF *fRecoCandidate; // Reconstructed HF candidate
	124	AliAODMCParticle *fmcPartCandidate;
	125	Int_t fNDaughters;
	126	Int_t fNVar; // get Number of variables for the container from the channel decay
	127	Double_t fzPrimVertex; //Reco z primary vertex
	128	Double_t fzMCVertex; //MC z primary vertex
	129
	130	Bool_t fFillFromGenerated; // flag to indicate whether data container should be filled
	131	UShort_t fOriginDselection; // flag to select D0 origins. 0 Only from charm 1 only from beauty 2 both from charm and beauty
	132
	133	Bool_t fKeepDfromB; //flag for the feed down from b quark decay.
	134	Bool_t fKeepDfromBOnly; // flag to keep only the charm particles that comes from beauty decays
	135	Int_t fmcLabel; // results of the MatchToMC()
	136	Int_t fProngs; // n. of prongs
6e2e4f50	137	Int_t* fLabelArray; //[fProngs] array of labels
6e2e4f50	138
b7af2639	139	Float_t fCentValue; // centrality value
2bf2e62b	140	Float_t* fPtAccCut; //[fProngs] array of pt cut values for the Acceptance (MC+Rec) steps
2bf2e62b	141	Float_t* fEtaAccCut; //[fProngs] array of eta (absolute value) cut values for the Acceptance (MC+Rec) steps
1f780958	142	Int_t fFakeSelection; // fakes selection: 0 --> all, 1 --> non-fake, 2 --> fake
1f780958	143	Float_t fFake; // variable to indicate whether the D0 was a fake or not: 0 --> fake, 1 --> MC, 2 --> non-fake
0ada222f	144	Bool_t fRejectIfNoQuark; // flag to remove events not geenrated with PYTHIA
ec5288c3	145	Double_t fMultiplicity; // multiplicity of the event
ec5288c3	146	Int_t fConfiguration; // configuration (slow / fast) of the CF --> different variables will be allocated (all / reduced number)
b7af2639	147
ec5288c3	148	ClassDef(AliCFVertexingHF, 7);
379592af	149
	150	};
	151
	152	#endif