[u/mrichter/AliRoot.git] / PWGHF / correlationHF / AliAnalysisTaskSED0Correlations.h

#ifndef ALIANALYSISTASKSED0CORRELATIONS_H
#define ALIANALYSISTASKSED0CORRELATIONS_H

/* Copyright(c) 1998-2012, ALICE Experiment at CERN, All rights reserved. *
 * See cxx source for full Copyright notice                               */

/* $Id$ */

//*************************************************************************
// Class AliAnalysisTaskSED0Correlations
// AliAnalysisTaskSE for D0 candidates (2Prongs) and hadrons correlations
// Authors:
// C.Bianchin, chiara.bianchin@pd.infn.it
// F.Colamaria, fabio.colamaria@ba.infn.it
//*************************************************************************

#include <TROOT.h>
#include <TSystem.h>
#include <TNtuple.h>
#include <TTree.h>
#include <TH1F.h>
#include <THnSparse.h>

#include "AliAnalysisTaskSE.h"
#include "AliRDHFCutsD0toKpi.h"
#include "AliHFAssociatedTrackCuts.h"
#include "AliHFCorrelator.h"
#include "AliNormalizationCounter.h"

using std::vector;

class AliAODEvent;

class AliAnalysisTaskSED0Correlations : public AliAnalysisTaskSE
{
 public:

  AliAnalysisTaskSED0Correlations();
  AliAnalysisTaskSED0Correlations(const char *name,AliRDHFCutsD0toKpi* cutsD0, AliHFAssociatedTrackCuts* cutsTrk);
  virtual ~AliAnalysisTaskSED0Correlations();

  // Implementation of interface methods
  virtual void UserCreateOutputObjects();
  virtual void Init();
  virtual void LocalInit() {Init();}
  virtual void UserExec(Option_t *option);
  virtual void Terminate(Option_t *option);

  void SetReadMC(Bool_t readMC=kFALSE){fReadMC=readMC;}
  void SetMCReconstructedTracks(Bool_t recoTrMC=kTRUE){fRecoTr=recoTrMC;}
  void SetMCReconstructedD0(Bool_t recoD0MC=kTRUE){fRecoD0=recoD0MC;}
  void SetMCSelEventType(Bool_t sel=kFALSE){fSelEvType=sel;}
  void SetFillOnlyD0D0bar(Int_t flagfill){fFillOnlyD0D0bar=flagfill;}
  void SetSystem(Int_t sys){fSys=sys;}
  void SetRejectSDDClusters(Bool_t flag) {fIsRejectSDDClusters=flag; }
  void SetFillGlobalPlots(Bool_t fill=kTRUE){fFillGlobal=fill;}
  void SetSoftPiFlag(Bool_t piflag) {fSoftPiCut=piflag;}
  void SetMEAxisThresh(Bool_t methresh) {fMEAxisThresh=methresh;}
  void SetKaonCorrelations(Bool_t kaonCorr) {fKaonCorr=kaonCorr;}

  Int_t  GetReadMC() const {return fReadMC;}
  Int_t  GetMCReconstructedTracks() const {return fRecoTr;}
  Int_t  GetMCReconstructedD0() const {return fRecoD0;}
  Int_t  GetMCSelEventType() const {return fSelEvType;}
  Int_t  GetFillOnlyD0D0bar() const {return fFillOnlyD0D0bar;}
  Int_t  GetSystem() const {return fSys;}
  Bool_t GetRejectSDDClusters() const {return fIsRejectSDDClusters;}
  Bool_t GetFillGlobalPlots() const {return fFillGlobal;}
  Double_t GetEtaForCorrel() {return fEtaForCorrel;}
  Double_t GetMultEv() {return fMultEv;}
  Bool_t GetSoftPiFlag() const {return fSoftPiCut;}
  Bool_t GetMEAxisThresh() const {return fMEAxisThresh;}
  Bool_t GetKaonCorrelations() const {return fKaonCorr;}

  //correlations setters/printers
  void SetNPtBinsCorr(Int_t nbins) {fNPtBinsCorr = nbins;}
  void SetPtBinsLimsCorr(Double_t* ptlims) {for(int i=0;i<=fNPtBinsCorr;i++) {fBinLimsCorr.push_back(ptlims[i]);}}
  void SetPtBinsLimsCorr(Float_t* ptlims) {for(int i=0;i<=fNPtBinsCorr;i++) {fBinLimsCorr.push_back((Double_t)ptlims[i]);}}
  void SetPtTreshLow(Double_t* pttreshlow) {for(int i=0;i<fNPtBinsCorr;i++) {fPtThreshLow.push_back(pttreshlow[i]);}}
  void SetPtTreshUp(Double_t* pttreshup) {for(int i=0;i<fNPtBinsCorr;i++) {fPtThreshUp.push_back(pttreshup[i]);}}

  void PrintBinsAndLimits();
  Int_t PtBinCorr(Double_t pt) const;
  void SetEvMixing(Bool_t mix) {fMixing=mix;}
  void SetEtaForCorrel(Double_t etacorr) {fEtaForCorrel=etacorr;}

  enum PartType {kTrack,kKCharg,kK0};
  enum FillType {kSE, kME}; //for single event or event mixing histos fill

 private:

  AliAnalysisTaskSED0Correlations(const AliAnalysisTaskSED0Correlations &source);
  AliAnalysisTaskSED0Correlations& operator=(const AliAnalysisTaskSED0Correlations& source); 
  void FillMassHists(AliAODRecoDecayHF2Prong *part, TClonesArray *arrMC, AliRDHFCutsD0toKpi *cuts, TList *listout);
  Int_t CheckD0Origin(TClonesArray* arrayMC, AliAODMCParticle *mcPartCandidate) const;
  //correlation methods
  void CreateCorrelationsObjs();
  void CalculateCorrelations(AliAODRecoDecayHF2Prong* d, Int_t labD0=-1, TClonesArray* mcArray=0x0);
  void CalculateCorrelationsMCKine(AliAODMCParticle* d, TClonesArray* mcArray=0x0);
  void FillSparsePlots(TClonesArray* arrayMC, Double_t mInv[], Int_t origD0, Int_t PdgD0, AliReducedParticle* track, Int_t ptbin, Int_t type, Double_t wg=1.);
  Int_t CheckTrackOrigin(TClonesArray* arrayMC, AliAODMCParticle *mcPartCandidate) const;
  Bool_t IsDDaughter(AliAODMCParticle* d, AliAODMCParticle* track, TClonesArray* mcArray) const;
  Bool_t SelectV0(AliAODv0* v0, AliAODVertex *vtx, Int_t option, Int_t idArrayV0[][2]) const;
  Bool_t IsSoftPion_MCKine(AliAODMCParticle* d, AliAODMCParticle* track, TClonesArray* arrayMC) const;

  Int_t             fNPtBinsCorr;        // number of pt bins per correlations
  std::vector<Double_t>  fBinLimsCorr;        // limits of pt bins per correlations
  std::vector<Double_t>  fPtThreshLow;        // pT treshold of hadrons - low
  std::vector<Double_t>  fPtThreshUp;         // pT treshold of hadrons - up

  Int_t     fEvents;		  	// EventCounter
  Bool_t    fAlreadyFilled;	  	// D0 in an event already analyzed (for track distribution plots)
  TList    *fOutputMass;          	//!list send on output slot 1
  TList    *fOutputCorr;	  	//!list of correlation histos, output slot 5
  TList    *fOutputStudy;	  	//!list of histos with MC distributions, output slot 6
  TH1F     *fNentries;            	//!histogram with number of events on output slot 2
  AliRDHFCutsD0toKpi *fCutsD0;    	// Cuts for D0, output 3
  AliHFAssociatedTrackCuts *fCutsTracks;// Cuts for tracks and K0, output 7
  AliHFCorrelator* fCorrelatorTr;	// Correlator for tracks
  AliHFCorrelator* fCorrelatorKc;	// Correlator for charged K
  AliHFCorrelator* fCorrelatorK0;	// Correlator for K0
  Bool_t    fReadMC;              	// flag for MC array: kTRUE = read it, kFALSE = do not read it
  Bool_t    fRecoTr;   		       	// flag for using MC reconstructed (kTRUE) or pure kinematic MC (kFALSE) - Associated tracks
  Bool_t    fRecoD0;   		       	// flag for using MC reconstructed (kTRUE) or pure kinematic MC (kFALSE) - D0
  Bool_t    fSelEvType;		       	// flag for enabling selection of event tpye (PP, GS, FE, ...) on MC analysis
  Bool_t    fMixing;			// flag to enable also event mixing
  AliNormalizationCounter *fCounter;	//!AliNormalizationCounter on output slot 4
  Int_t     fNPtBins;             	// Number of pt bins
  Int_t     fFillOnlyD0D0bar;     	// flag to fill mass histogram with D0/D0bar only (0 = fill with both, 1 = fill with D0 only, 2 = fill with D0bar only)
  Int_t     fIsSelectedCandidate; 	// selection outcome
  Int_t     fSys;                 	// fSys=0 -> p-p; fSys=1 ->PbPb
  Double_t  fEtaForCorrel;		// cut for D0 eta to enable correlation with associated particles
  Bool_t    fIsRejectSDDClusters; 	// flag to reject events with SDD clusters
  Bool_t    fFillGlobal;          	// flag to fill global plots (in loops on tracks and V0 for each event)
  Double_t  fMultEv;			// event multiplicity (for trigger eff)
  Bool_t    fSoftPiCut;			// flag to activate soft pion cut on Data
  Bool_t    fMEAxisThresh;		// flag to fill threshold axis in ME plots
  Bool_t    fKaonCorr;			// enables correlations of D0-Kcharg and D0-K0

  ClassDef(AliAnalysisTaskSED0Correlations,5); // AliAnalysisTaskSE for D0->Kpi
};

#endif
Commit	Line	Data
a2ad7da1	1	#ifndef ALIANALYSISTASKSED0CORRELATIONS_H
	2	#define ALIANALYSISTASKSED0CORRELATIONS_H
	3
	4	/* Copyright(c) 1998-2012, ALICE Experiment at CERN, All rights reserved. *
	5	* See cxx source for full Copyright notice */
	6
241994ab	7	/* $Id$ */
a2ad7da1	8
	9	//*************************************************************************
	10	// Class AliAnalysisTaskSED0Correlations
	11	// AliAnalysisTaskSE for D0 candidates (2Prongs) and hadrons correlations
	12	// Authors:
	13	// C.Bianchin, chiara.bianchin@pd.infn.it
	14	// F.Colamaria, fabio.colamaria@ba.infn.it
	15	//*************************************************************************
	16
	17	#include <TROOT.h>
	18	#include <TSystem.h>
	19	#include <TNtuple.h>
	20	#include <TTree.h>
	21	#include <TH1F.h>
	22	#include <THnSparse.h>
	23
	24	#include "AliAnalysisTaskSE.h"
	25	#include "AliRDHFCutsD0toKpi.h"
	26	#include "AliHFAssociatedTrackCuts.h"
bce70c96	27	#include "AliHFCorrelator.h"
a2ad7da1	28	#include "AliNormalizationCounter.h"
a2ad7da1	29
e61afb80	30	using std::vector;
e61afb80	31
a2ad7da1	32	class AliAODEvent;
	33
	34	class AliAnalysisTaskSED0Correlations : public AliAnalysisTaskSE
	35	{
	36	public:
	37
	38	AliAnalysisTaskSED0Correlations();
	39	AliAnalysisTaskSED0Correlations(const char name,AliRDHFCutsD0toKpi cutsD0, AliHFAssociatedTrackCuts* cutsTrk);
	40	virtual ~AliAnalysisTaskSED0Correlations();
	41
	42	// Implementation of interface methods
	43	virtual void UserCreateOutputObjects();
	44	virtual void Init();
	45	virtual void LocalInit() {Init();}
	46	virtual void UserExec(Option_t *option);
	47	virtual void Terminate(Option_t *option);
	48
	49	void SetReadMC(Bool_t readMC=kFALSE){fReadMC=readMC;}
4d978a4a	50	void SetMCReconstructedTracks(Bool_t recoTrMC=kTRUE){fRecoTr=recoTrMC;}
	51	void SetMCReconstructedD0(Bool_t recoD0MC=kTRUE){fRecoD0=recoD0MC;}
	52	void SetMCSelEventType(Bool_t sel=kFALSE){fSelEvType=sel;}
a2ad7da1	53	void SetFillOnlyD0D0bar(Int_t flagfill){fFillOnlyD0D0bar=flagfill;}
	54	void SetSystem(Int_t sys){fSys=sys;}
	55	void SetRejectSDDClusters(Bool_t flag) {fIsRejectSDDClusters=flag; }
	56	void SetFillGlobalPlots(Bool_t fill=kTRUE){fFillGlobal=fill;}
241994ab	57	void SetSoftPiFlag(Bool_t piflag) {fSoftPiCut=piflag;}
241994ab	58	void SetMEAxisThresh(Bool_t methresh) {fMEAxisThresh=methresh;}
dea90a65	59	void SetKaonCorrelations(Bool_t kaonCorr) {fKaonCorr=kaonCorr;}
a2ad7da1	60
a2ad7da1	61	Int_t GetReadMC() const {return fReadMC;}
4d978a4a	62	Int_t GetMCReconstructedTracks() const {return fRecoTr;}
	63	Int_t GetMCReconstructedD0() const {return fRecoD0;}
	64	Int_t GetMCSelEventType() const {return fSelEvType;}
a2ad7da1	65	Int_t GetFillOnlyD0D0bar() const {return fFillOnlyD0D0bar;}
	66	Int_t GetSystem() const {return fSys;}
	67	Bool_t GetRejectSDDClusters() const {return fIsRejectSDDClusters;}
	68	Bool_t GetFillGlobalPlots() const {return fFillGlobal;}
8c2d7467	69	Double_t GetEtaForCorrel() {return fEtaForCorrel;}
241994ab	70	Double_t GetMultEv() {return fMultEv;}
	71	Bool_t GetSoftPiFlag() const {return fSoftPiCut;}
	72	Bool_t GetMEAxisThresh() const {return fMEAxisThresh;}
dea90a65	73	Bool_t GetKaonCorrelations() const {return fKaonCorr;}
a2ad7da1	74
	75	//correlations setters/printers
	76	void SetNPtBinsCorr(Int_t nbins) {fNPtBinsCorr = nbins;}
	77	void SetPtBinsLimsCorr(Double_t* ptlims) {for(int i=0;i<=fNPtBinsCorr;i++) {fBinLimsCorr.push_back(ptlims[i]);}}
	78	void SetPtBinsLimsCorr(Float_t* ptlims) {for(int i=0;i<=fNPtBinsCorr;i++) {fBinLimsCorr.push_back((Double_t)ptlims[i]);}}
	79	void SetPtTreshLow(Double_t* pttreshlow) {for(int i=0;i<fNPtBinsCorr;i++) {fPtThreshLow.push_back(pttreshlow[i]);}}
	80	void SetPtTreshUp(Double_t* pttreshup) {for(int i=0;i<fNPtBinsCorr;i++) {fPtThreshUp.push_back(pttreshup[i]);}}
7f221b36	81
a2ad7da1	82	void PrintBinsAndLimits();
a2ad7da1	83	Int_t PtBinCorr(Double_t pt) const;
bce70c96	84	void SetEvMixing(Bool_t mix) {fMixing=mix;}
8c2d7467	85	void SetEtaForCorrel(Double_t etacorr) {fEtaForCorrel=etacorr;}
bce70c96	86
	87	enum PartType {kTrack,kKCharg,kK0};
	88	enum FillType {kSE, kME}; //for single event or event mixing histos fill
a2ad7da1	89
	90	private:
	91
	92	AliAnalysisTaskSED0Correlations(const AliAnalysisTaskSED0Correlations &source);
	93	AliAnalysisTaskSED0Correlations& operator=(const AliAnalysisTaskSED0Correlations& source);
	94	void FillMassHists(AliAODRecoDecayHF2Prong part, TClonesArray arrMC, AliRDHFCutsD0toKpi cuts, TList listout);
a2ad7da1	95	Int_t CheckD0Origin(TClonesArray* arrayMC, AliAODMCParticle *mcPartCandidate) const;
	96	//correlation methods
	97	void CreateCorrelationsObjs();
bce70c96	98	void CalculateCorrelations(AliAODRecoDecayHF2Prong* d, Int_t labD0=-1, TClonesArray* mcArray=0x0);
4d978a4a	99	void CalculateCorrelationsMCKine(AliAODMCParticle* d, TClonesArray* mcArray=0x0);
4d978a4a	100	void FillSparsePlots(TClonesArray* arrayMC, Double_t mInv[], Int_t origD0, Int_t PdgD0, AliReducedParticle* track, Int_t ptbin, Int_t type, Double_t wg=1.);
a2ad7da1	101	Int_t CheckTrackOrigin(TClonesArray* arrayMC, AliAODMCParticle *mcPartCandidate) const;
4d978a4a	102	Bool_t IsDDaughter(AliAODMCParticle* d, AliAODMCParticle* track, TClonesArray* mcArray) const;
a2ad7da1	103	Bool_t SelectV0(AliAODv0* v0, AliAODVertex *vtx, Int_t option, Int_t idArrayV0[][2]) const;
241994ab	104	Bool_t IsSoftPion_MCKine(AliAODMCParticle* d, AliAODMCParticle* track, TClonesArray* arrayMC) const;
a2ad7da1	105
a2ad7da1	106	Int_t fNPtBinsCorr; // number of pt bins per correlations
2ad98fc5	107	std::vector<Double_t> fBinLimsCorr; // limits of pt bins per correlations
	108	std::vector<Double_t> fPtThreshLow; // pT treshold of hadrons - low
	109	std::vector<Double_t> fPtThreshUp; // pT treshold of hadrons - up
a2ad7da1	110
	111	Int_t fEvents; // EventCounter
	112	Bool_t fAlreadyFilled; // D0 in an event already analyzed (for track distribution plots)
	113	TList *fOutputMass; //!list send on output slot 1
	114	TList *fOutputCorr; //!list of correlation histos, output slot 5
	115	TList *fOutputStudy; //!list of histos with MC distributions, output slot 6
	116	TH1F *fNentries; //!histogram with number of events on output slot 2
	117	AliRDHFCutsD0toKpi *fCutsD0; // Cuts for D0, output 3
	118	AliHFAssociatedTrackCuts *fCutsTracks;// Cuts for tracks and K0, output 7
bce70c96	119	AliHFCorrelator* fCorrelatorTr; // Correlator for tracks
	120	AliHFCorrelator* fCorrelatorKc; // Correlator for charged K
	121	AliHFCorrelator* fCorrelatorK0; // Correlator for K0
a2ad7da1	122	Bool_t fReadMC; // flag for MC array: kTRUE = read it, kFALSE = do not read it
4d978a4a	123	Bool_t fRecoTr; // flag for using MC reconstructed (kTRUE) or pure kinematic MC (kFALSE) - Associated tracks
	124	Bool_t fRecoD0; // flag for using MC reconstructed (kTRUE) or pure kinematic MC (kFALSE) - D0
	125	Bool_t fSelEvType; // flag for enabling selection of event tpye (PP, GS, FE, ...) on MC analysis
bce70c96	126	Bool_t fMixing; // flag to enable also event mixing
a2ad7da1	127	AliNormalizationCounter *fCounter; //!AliNormalizationCounter on output slot 4
	128	Int_t fNPtBins; // Number of pt bins
	129	Int_t fFillOnlyD0D0bar; // flag to fill mass histogram with D0/D0bar only (0 = fill with both, 1 = fill with D0 only, 2 = fill with D0bar only)
	130	Int_t fIsSelectedCandidate; // selection outcome
	131	Int_t fSys; // fSys=0 -> p-p; fSys=1 ->PbPb
8c2d7467	132	Double_t fEtaForCorrel; // cut for D0 eta to enable correlation with associated particles
a2ad7da1	133	Bool_t fIsRejectSDDClusters; // flag to reject events with SDD clusters
a2ad7da1	134	Bool_t fFillGlobal; // flag to fill global plots (in loops on tracks and V0 for each event)
241994ab	135	Double_t fMultEv; // event multiplicity (for trigger eff)
	136	Bool_t fSoftPiCut; // flag to activate soft pion cut on Data
	137	Bool_t fMEAxisThresh; // flag to fill threshold axis in ME plots
dea90a65	138	Bool_t fKaonCorr; // enables correlations of D0-Kcharg and D0-K0
a2ad7da1	139
dea90a65	140	ClassDef(AliAnalysisTaskSED0Correlations,5); // AliAnalysisTaskSE for D0->Kpi
a2ad7da1	141	};
	142
	143	#endif
	144