[u/mrichter/AliRoot.git] / PWGJE / FlavourJetTasks / AliAnalysisTaskFlavourJetCorrelations.h

#ifndef ALIANALYSISTASKFLAVOURJETCORRELATIONS_H
#define ALIANALYSISTASKFLAVOURJETCORRELATIONS_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.                  *
 **************************************************************************/

//-----------------------------------------------------------------------
// Author : A. Grelli,  Utrecht University
//          C. Bianchin, Utrecht University
//          X. Zhang, LBNL
//-----------------------------------------------------------------------


#include <TH2F.h>
#include "AliAODEvent.h"
#include "AliPicoTrack.h"
#include "AliAnalysisTaskEmcalJet.h"

class TH3F;
class TParticle ;
class TClonesArray ;
class AliMCParticle;
class AliAODMCParticle;
class AliRDHFCuts;
class AliEmcalJet;
class AliAODRecoDecayHF;
class AliAODRecoCascadeHF;
class AliAODEvent;

class AliAnalysisTaskFlavourJetCorrelations : public AliAnalysisTaskEmcalJet 
{

 public:

  enum ECandidateType{ kD0toKpi, kDstartoKpipi };

  AliAnalysisTaskFlavourJetCorrelations();
  AliAnalysisTaskFlavourJetCorrelations(const Char_t* name,AliRDHFCuts* cuts, ECandidateType candtype);
  virtual ~AliAnalysisTaskFlavourJetCorrelations();

  virtual void     UserCreateOutputObjects();
  virtual Bool_t   Run();
  virtual void     Terminate(Option_t *);
  virtual void     Init();
  virtual void     LocalInit() {Init();}

  // inizializations
  Bool_t DefineHistoForAnalysis();  

  // set MC usage
  void   SetMC(Bool_t theMCon) {fUseMCInfo = theMCon;}
  Bool_t GetMC() const {return fUseMCInfo;}
  // set usage of reconstructed tracks
  void   SetUseReco(Bool_t reco) {fUseReco=reco;}
  Bool_t GetUseReco() {return fUseReco;}
  
  
  void SetMassLimits(Double_t range, Int_t pdg);
  void SetMassLimits(Double_t lowlimit, Double_t uplimit);

  //jet reconstruction algorithm
  void SetJetArrayName(TString jetArrName) {fJetArrName=jetArrName;};
  TString GetJetArrayName() const {return fJetArrName;};

  // trigger on jet events
  void SetTriggerOnLeadingJet(Bool_t triggerOnLeadingJet) {fLeadingJetOnly=triggerOnLeadingJet;};
  Bool_t GetTriggerOnLeadingJet() const {return fLeadingJetOnly;}


  // Array of D0 width for the Dstar
  Bool_t SetD0WidthForDStar(Int_t nptbins,Float_t* width);

  //Bool_t   FillMCDJetInfo(AliPicoTrack *jetTrk,AliEmcalJet* jet, TClonesArray *mcArray,Double_t ptjet);
  void FillHistogramsRecoJetCorr(AliVParticle* candidate, AliEmcalJet *jet, AliAODEvent* aodEvent);
  void FillHistogramsD0JetCorr(AliAODRecoDecayHF* candidate, Double_t dPhi, Double_t z, Double_t ptD, Double_t ptj, Double_t deltaR, AliAODEvent* aodEvent);

  void FillHistogramsDstarJetCorr(AliAODRecoCascadeHF* dstar, Double_t dPhi, Double_t z, Double_t ptD, Double_t ptj,Double_t deltaR);
  void FillHistogramsMCGenDJetCorr(Double_t dPhi, Double_t z,Double_t ptD,Double_t ptjet,Double_t deltaR);
  void SideBandBackground(AliAODRecoCascadeHF *candDstar, AliEmcalJet *jet);
  void MCBackground(AliAODRecoDecayHF *candbg, AliEmcalJet *jet);
  void FillMassHistograms(Double_t mass,Double_t ptD, Double_t deltaR);
  void FlagFlavour(AliVParticle* charm, AliEmcalJet* jet);
  Int_t IsDzeroSideBand(AliAODRecoCascadeHF *candDstar);

 private:
  
  AliAnalysisTaskFlavourJetCorrelations(const AliAnalysisTaskFlavourJetCorrelations &source);
  AliAnalysisTaskFlavourJetCorrelations& operator=(const AliAnalysisTaskFlavourJetCorrelations& source); 

  Double_t Z(AliVParticle* part,AliEmcalJet* jet) const;
  Float_t DeltaR(AliVParticle *p1, AliVParticle *p2) const;


  Bool_t fUseMCInfo;             //  Use MC info
  Bool_t fUseReco;               // use reconstructed tracks when running on MC
  Int_t  fCandidateType;         // Dstar or D0
  Int_t  fPDGmother;             // PDG code of D meson
  Int_t  fNProngs;               // number of prong of the decay channel  
  Int_t  fPDGdaughters[4];       // PDG codes of daughters
  Float_t fSigmaD0[30];          //
  TString fBranchName;           // AOD branch name
  TList *fmyOutput;                //! user output
  AliRDHFCuts *fCuts;            // Cuts 

  Double_t fMinMass;             // mass lower limit histogram
  Double_t fMaxMass;             // mass upper limit histogram

  TString  fJetArrName;          // name of the jet array, taken from the task running the jet finder
  TString fCandArrName;          // string which correspond to the candidate type
  Bool_t fLeadingJetOnly;        // use only the leading jet in the event to make the correlations
  Double_t fJetRadius;           // jet radius (filled from the JetContainer)
  TClonesArray *fCandidateArray;   //! contains candidates selected by AliRDHFCuts
  TClonesArray *fSideBandArray;    //! contains candidates selected by AliRDHFCuts::IsSelected(kTracks), to be used for side bands (DStar case only!!)

  ClassDef(AliAnalysisTaskFlavourJetCorrelations,2); // class for charm-jet CorrelationsExch
};

#endif
Commit	Line	Data
c683985a	1	#ifndef ALIANALYSISTASKFLAVOURJETCORRELATIONS_H
	2	#define ALIANALYSISTASKFLAVOURJETCORRELATIONS_H
	3	/**************************************************************************
	4	* Copyright(c) 1998-2009, ALICE Experiment at CERN, All rights reserved. *
	5	* *
	6	* Author: The ALICE Off-line Project. *
	7	* Contributors are mentioned in the code where appropriate. *
	8	* *
	9	* Permission to use, copy, modify and distribute this software and its *
	10	* documentation strictly for non-commercial purposes is hereby granted *
	11	* without fee, provided that the above copyright notice appears in all *
	12	* copies and that both the copyright notice and this permission notice *
	13	* appear in the supporting documentation. The authors make no claims *
	14	* about the suitability of this software for any purpose. It is *
	15	* provided "as is" without express or implied warranty. *
	16	**************************************************************************/
	17
	18	//-----------------------------------------------------------------------
	19	// Author : A. Grelli, Utrecht University
	20	// C. Bianchin, Utrecht University
	21	// X. Zhang, LBNL
	22	//-----------------------------------------------------------------------
	23
	24
	25	#include <TH2F.h>
	26	#include "AliAODEvent.h"
	27	#include "AliPicoTrack.h"
	28	#include "AliAnalysisTaskEmcalJet.h"
	29
	30	class TH3F;
	31	class TParticle ;
	32	class TClonesArray ;
	33	class AliMCParticle;
	34	class AliAODMCParticle;
	35	class AliRDHFCuts;
	36	class AliEmcalJet;
	37	class AliAODRecoDecayHF;
	38	class AliAODRecoCascadeHF;
	39	class AliAODEvent;
	40
	41	class AliAnalysisTaskFlavourJetCorrelations : public AliAnalysisTaskEmcalJet
	42	{
	43
	44	public:
	45
	46	enum ECandidateType{ kD0toKpi, kDstartoKpipi };
	47
	48	AliAnalysisTaskFlavourJetCorrelations();
	49	AliAnalysisTaskFlavourJetCorrelations(const Char_t* name,AliRDHFCuts* cuts, ECandidateType candtype);
	50	virtual ~AliAnalysisTaskFlavourJetCorrelations();
	51
	52	virtual void UserCreateOutputObjects();
b5d0ba13	53	virtual Bool_t Run();
c683985a	54	virtual void Terminate(Option_t *);
	55	virtual void Init();
	56	virtual void LocalInit() {Init();}
	57
	58	// inizializations
	59	Bool_t DefineHistoForAnalysis();
	60
	61	// set MC usage
	62	void SetMC(Bool_t theMCon) {fUseMCInfo = theMCon;}
	63	Bool_t GetMC() const {return fUseMCInfo;}
	64	// set usage of reconstructed tracks
	65	void SetUseReco(Bool_t reco) {fUseReco=reco;}
	66	Bool_t GetUseReco() {return fUseReco;}
	67
	68
	69	void SetMassLimits(Double_t range, Int_t pdg);
	70	void SetMassLimits(Double_t lowlimit, Double_t uplimit);
	71
	72	//jet reconstruction algorithm
	73	void SetJetArrayName(TString jetArrName) {fJetArrName=jetArrName;};
	74	TString GetJetArrayName() const {return fJetArrName;};
	75
	76	// trigger on jet events
	77	void SetTriggerOnLeadingJet(Bool_t triggerOnLeadingJet) {fLeadingJetOnly=triggerOnLeadingJet;};
	78	Bool_t GetTriggerOnLeadingJet() const {return fLeadingJetOnly;}
	79
	80
	81	// Array of D0 width for the Dstar
	82	Bool_t SetD0WidthForDStar(Int_t nptbins,Float_t* width);
	83
	84	//Bool_t FillMCDJetInfo(AliPicoTrack jetTrk,AliEmcalJet jet, TClonesArray *mcArray,Double_t ptjet);
8577f7bc	85	void FillHistogramsRecoJetCorr(AliVParticle* candidate, AliEmcalJet jet, AliAODEvent aodEvent);
c683985a	86	void FillHistogramsD0JetCorr(AliAODRecoDecayHF* candidate, Double_t dPhi, Double_t z, Double_t ptD, Double_t ptj, Double_t deltaR, AliAODEvent* aodEvent);
	87
	88	void FillHistogramsDstarJetCorr(AliAODRecoCascadeHF* dstar, Double_t dPhi, Double_t z, Double_t ptD, Double_t ptj,Double_t deltaR);
	89	void FillHistogramsMCGenDJetCorr(Double_t dPhi, Double_t z,Double_t ptD,Double_t ptjet,Double_t deltaR);
	90	void SideBandBackground(AliAODRecoCascadeHF candDstar, AliEmcalJet jet);
	91	void MCBackground(AliAODRecoDecayHF candbg, AliEmcalJet jet);
	92	void FillMassHistograms(Double_t mass,Double_t ptD, Double_t deltaR);
	93	void FlagFlavour(AliVParticle* charm, AliEmcalJet* jet);
25094cd7	94	Int_t IsDzeroSideBand(AliAODRecoCascadeHF *candDstar);
c683985a	95
	96	private:
	97
	98	AliAnalysisTaskFlavourJetCorrelations(const AliAnalysisTaskFlavourJetCorrelations &source);
	99	AliAnalysisTaskFlavourJetCorrelations& operator=(const AliAnalysisTaskFlavourJetCorrelations& source);
	100
	101	Double_t Z(AliVParticle* part,AliEmcalJet* jet) const;
	102	Float_t DeltaR(AliVParticle p1, AliVParticle p2) const;
	103
	104
	105	Bool_t fUseMCInfo; // Use MC info
	106	Bool_t fUseReco; // use reconstructed tracks when running on MC
	107	Int_t fCandidateType; // Dstar or D0
	108	Int_t fPDGmother; // PDG code of D meson
	109	Int_t fNProngs; // number of prong of the decay channel
	110	Int_t fPDGdaughters[4]; // PDG codes of daughters
	111	Float_t fSigmaD0[30]; //
	112	TString fBranchName; // AOD branch name
	113	TList *fmyOutput; //! user output
	114	AliRDHFCuts *fCuts; // Cuts
	115
	116	Double_t fMinMass; // mass lower limit histogram
	117	Double_t fMaxMass; // mass upper limit histogram
	118
	119	TString fJetArrName; // name of the jet array, taken from the task running the jet finder
	120	TString fCandArrName; // string which correspond to the candidate type
	121	Bool_t fLeadingJetOnly; // use only the leading jet in the event to make the correlations
	122	Double_t fJetRadius; // jet radius (filled from the JetContainer)
bbb94467	123	TClonesArray *fCandidateArray; //! contains candidates selected by AliRDHFCuts
bbb94467	124	TClonesArray *fSideBandArray; //! contains candidates selected by AliRDHFCuts::IsSelected(kTracks), to be used for side bands (DStar case only!!)
c683985a	125
bbb94467	126	ClassDef(AliAnalysisTaskFlavourJetCorrelations,2); // class for charm-jet CorrelationsExch
c683985a	127	};
	128
	129	#endif