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

#ifndef ALIANALYSISTASKEMCALJETV0CF_H
#define ALIANALYSISTASKEMCALJETV0CF_H

#include <TH1D.h>
#include "AliAnalysisTaskEmcalJet.h"
//=============================================================================

class TClonesArray;
class TVector3;

class AliAODEvent;
class AliESDEvent;
class AliCentrality;

class AliParticleContainer;
class AliClusterContainer;
class AliJetContainer;
//=============================================================================

class AliAnalysisTaskEmcalJetV0CF : public AliAnalysisTaskEmcalJet {

 public :

  AliAnalysisTaskEmcalJetV0CF();
  AliAnalysisTaskEmcalJetV0CF(const char *name, const Bool_t bHistos=kTRUE);
  virtual ~AliAnalysisTaskEmcalJetV0CF();

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

  void SetKaCutNS(Double_t d) { fKaCutNS = d; }
  void SetLaCutNS(Double_t d) { fLaCutNS = d; }
  void SetHistoKshortInvM(TH1D const *h) { fHistoKshortInvM = new TH1D(*h); }
  void SetHistoLambdaInvM(TH1D const *h) { fHistoLambdaInvM = new TH1D(*h); }
  void SetHistoAntiLaInvM(TH1D const *h) { fHistoAntiLaInvM = new TH1D(*h); }
  void SetV0EtaRange(Double_t dMin, Double_t dMax) { fV0CutMinEta = dMin, fV0CutMaxEta = dMax; }
//=============================================================================

 protected :

  virtual void   ExecOnce();
  virtual Bool_t FillGeneralHistograms();
  virtual Bool_t FillHistograms();
  virtual Bool_t IsEventSelected();
  virtual Bool_t RetrieveEventObjects();
  virtual Bool_t Run();
//=============================================================================

 private :

  AliAnalysisTaskEmcalJetV0CF(const AliAnalysisTaskEmcalJetV0CF &);
  AliAnalysisTaskEmcalJetV0CF& operator=(const AliAnalysisTaskEmcalJetV0CF &);

  Bool_t FillRecoInfo();
  Bool_t FillKineInfo();
  void CreateUserOutputHistograms();

  Bool_t IsV0InJet(TVector3 vV0, Double_t dJetPtMin);

  Double_t fKaCutNS; //
  Double_t fLaCutNS; //

  Double_t fV0CutMinEta; //
  Double_t fV0CutMaxEta; //

  AliAODEvent   *fEventAOD;  //!
  AliESDEvent   *fEventESD;  //!
  AliCentrality *fCentInfo;  //!

  AliJetContainer      *fJetsContRD;          //!
  AliParticleContainer *fTracksContRD;        //!
  AliClusterContainer  *fCaloClustersContRD;  //!

  AliJetContainer      *fJetsContMC;    //!
  AliParticleContainer *fTracksContMC;  //!

  TClonesArray *fV0s;  //!

  TH1D *fHistoKshortInvM;  //!
  TH1D *fHistoLambdaInvM;  //!
  TH1D *fHistoAntiLaInvM;  //!

  TList *fListUserOutputs; //!

  ClassDef(AliAnalysisTaskEmcalJetV0CF, 1);
};

#endif
Commit	Line	Data
b254f323	1	#ifndef ALIANALYSISTASKEMCALJETV0CF_H
	2	#define ALIANALYSISTASKEMCALJETV0CF_H
	3
	4	#include <TH1D.h>
	5	#include "AliAnalysisTaskEmcalJet.h"
	6	//=============================================================================
	7
	8	class TClonesArray;
	9	class TVector3;
	10
	11	class AliAODEvent;
	12	class AliESDEvent;
	13	class AliCentrality;
	14
	15	class AliParticleContainer;
	16	class AliClusterContainer;
	17	class AliJetContainer;
	18	//=============================================================================
	19
	20	class AliAnalysisTaskEmcalJetV0CF : public AliAnalysisTaskEmcalJet {
	21
	22	public :
	23
	24	AliAnalysisTaskEmcalJetV0CF();
	25	AliAnalysisTaskEmcalJetV0CF(const char *name, const Bool_t bHistos=kTRUE);
	26	virtual ~AliAnalysisTaskEmcalJetV0CF();
	27
	28	virtual void Init();
	29	virtual void LocalInit() { Init(); }
	30	virtual void UserCreateOutputObjects();
	31	virtual void Terminate(Option_t *opt);
	32
	33	void SetKaCutNS(Double_t d) { fKaCutNS = d; }
	34	void SetLaCutNS(Double_t d) { fLaCutNS = d; }
	35	void SetHistoKshortInvM(TH1D const h) { fHistoKshortInvM = new TH1D(h); }
	36	void SetHistoLambdaInvM(TH1D const h) { fHistoLambdaInvM = new TH1D(h); }
	37	void SetHistoAntiLaInvM(TH1D const h) { fHistoAntiLaInvM = new TH1D(h); }
	38	void SetV0EtaRange(Double_t dMin, Double_t dMax) { fV0CutMinEta = dMin, fV0CutMaxEta = dMax; }
	39	//=============================================================================
	40
	41	protected :
	42
	43	virtual void ExecOnce();
	44	virtual Bool_t FillGeneralHistograms();
	45	virtual Bool_t FillHistograms();
	46	virtual Bool_t IsEventSelected();
	47	virtual Bool_t RetrieveEventObjects();
	48	virtual Bool_t Run();
	49	//=============================================================================
	50
	51	private :
	52
	53	AliAnalysisTaskEmcalJetV0CF(const AliAnalysisTaskEmcalJetV0CF &);
	54	AliAnalysisTaskEmcalJetV0CF& operator=(const AliAnalysisTaskEmcalJetV0CF &);
	55
	56	Bool_t FillRecoInfo();
	57	Bool_t FillKineInfo();
	58	void CreateUserOutputHistograms();
	59
	60	Bool_t IsV0InJet(TVector3 vV0, Double_t dJetPtMin);
	61
	62	Double_t fKaCutNS; //
	63	Double_t fLaCutNS; //
	64
65	Double_t fV0CutMinEta; //
66	Double_t fV0CutMaxEta; //
67
68	AliAODEvent *fEventAOD; //!
69	AliESDEvent *fEventESD; //!
70	AliCentrality *fCentInfo; //!
71
72	AliJetContainer *fJetsContRD; //!
73	AliParticleContainer *fTracksContRD; //!
74	AliClusterContainer *fCaloClustersContRD; //!
75
76	AliJetContainer *fJetsContMC; //!
77	AliParticleContainer *fTracksContMC; //!
78
79	TClonesArray *fV0s; //!
80
81	TH1D *fHistoKshortInvM; //!
82	TH1D *fHistoLambdaInvM; //!
83	TH1D *fHistoAntiLaInvM; //!
84
85	TList *fListUserOutputs; //!
86
87	ClassDef(AliAnalysisTaskEmcalJetV0CF, 1);
88	};
89
90	#endif