[u/mrichter/AliRoot.git] / PWG3 / dielectron / AliAnalysisTaskDielectronEfficiency.h

#ifndef ALIANALYSISTASKDIELECTRONEFFICIENCY_H
#define ALIANALYSISTASKDIELECTRONEFFICIENCY_H

/* $Id$ */ 

//#####################################################
//#                                                   # 
//#  Analysis Task for Event Mixing for dielectron    #
//#                                                   #
//#     J.Wiechula (Jens.Wiechula@cern.ch)            #
//#                                                   #
//#####################################################

#include <AliAnalysisTask.h>

class TDatabasePDG;

class AliESDtrackCuts;
class AliKineTrackCuts;
class AliDielectronHistos;
class AliVEvent;
class AliStack;

class AliAnalysisTaskDielectronEfficiency : public AliAnalysisTask {

public:
  AliAnalysisTaskDielectronEfficiency();
  AliAnalysisTaskDielectronEfficiency(const char *name);
  virtual ~AliAnalysisTaskDielectronEfficiency();
  
  void SetupDefaultCuts(Int_t type=0);
  
	virtual void ConnectInputData(Option_t *);
	virtual void CreateOutputObjects();
  virtual void Exec(Option_t *option);
  virtual void Terminate(Option_t *);

//   virtual Int_t Merge(TCollection *list);
  //getters
  AliESDtrackCuts  *GetESDTrackCuts()   const {return fESDtrackCuts;}
  AliKineTrackCuts *GetKineCutsLeg()    const {return fKineCutsLegs;}
  AliKineTrackCuts *GetKineCutsMother() const {return fKineCutsMother;}
  //
  Int_t GetIdMother() const {return fIdMCMother;}

  //setters
  void SetIdMother(Int_t id) {fIdMCMother=id;}
  void SetIdDaughters(Int_t idPositive, Int_t idNegative) {fIdMCDaughterP=idPositive; fIdMCDaughterN=idNegative;}
  
private:

  AliVEvent           *fInputEvent;     //! Input event
  AliDielectronHistos *fHist;           //! Histogram container
  //cut objects
  AliESDtrackCuts     *fESDtrackCuts;   //  ESD track cuts
  AliKineTrackCuts    *fKineCutsLegs;   //  MC cuts on Legs
  AliKineTrackCuts    *fKineCutsMother; //  MC cuts on Mother (Id see below)
  
  Int_t                fIdMCMother;   //  MC Id of mother particle of interest (eg. Jpsi=443)
  Int_t                fIdMCDaughterP; //  MC Id of legs without sign
  Int_t                fIdMCDaughterN; //  MC Id of legs without sign
  //
  TDatabasePDG        *fPDG;           //! PDG database

  void FillPlots(AliVEvent *event);
  void FillMCInfo(AliStack * const pStack);

  AliAnalysisTaskDielectronEfficiency(const AliAnalysisTaskDielectronEfficiency &c);
  AliAnalysisTaskDielectronEfficiency& operator= (const AliAnalysisTaskDielectronEfficiency &c);
  
	ClassDef(AliAnalysisTaskDielectronEfficiency, 1);
};
#endif
Commit	Line	Data
b2a297fa	1	#ifndef ALIANALYSISTASKDIELECTRONEFFICIENCY_H
b2a297fa	2	#define ALIANALYSISTASKDIELECTRONEFFICIENCY_H
27de2dfb	3
	4	/* $Id$ */
	5
b2a297fa	6	//#####################################################
	7	//# #
	8	//# Analysis Task for Event Mixing for dielectron #
	9	//# #
	10	//# J.Wiechula (Jens.Wiechula@cern.ch) #
	11	//# #
	12	//#####################################################
	13
	14	#include <AliAnalysisTask.h>
	15
	16	class TDatabasePDG;
	17
	18	class AliESDtrackCuts;
	19	class AliKineTrackCuts;
	20	class AliDielectronHistos;
	21	class AliVEvent;
	22	class AliStack;
	23
	24	class AliAnalysisTaskDielectronEfficiency : public AliAnalysisTask {
	25
	26	public:
	27	AliAnalysisTaskDielectronEfficiency();
	28	AliAnalysisTaskDielectronEfficiency(const char *name);
	29	virtual ~AliAnalysisTaskDielectronEfficiency();
	30
	31	void SetupDefaultCuts(Int_t type=0);
	32
	33	virtual void ConnectInputData(Option_t *);
	34	virtual void CreateOutputObjects();
	35	virtual void Exec(Option_t *option);
	36	virtual void Terminate(Option_t *);
	37
	38	// virtual Int_t Merge(TCollection *list);
	39	//getters
	40	AliESDtrackCuts *GetESDTrackCuts() const {return fESDtrackCuts;}
	41	AliKineTrackCuts *GetKineCutsLeg() const {return fKineCutsLegs;}
	42	AliKineTrackCuts *GetKineCutsMother() const {return fKineCutsMother;}
	43	//
	44	Int_t GetIdMother() const {return fIdMCMother;}
	45
	46	//setters
	47	void SetIdMother(Int_t id) {fIdMCMother=id;}
	48	void SetIdDaughters(Int_t idPositive, Int_t idNegative) {fIdMCDaughterP=idPositive; fIdMCDaughterN=idNegative;}
	49
	50	private:
	51
	52	AliVEvent *fInputEvent; //! Input event
	53	AliDielectronHistos *fHist; //! Histogram container
	54	//cut objects
	55	AliESDtrackCuts *fESDtrackCuts; // ESD track cuts
	56	AliKineTrackCuts *fKineCutsLegs; // MC cuts on Legs
	57	AliKineTrackCuts *fKineCutsMother; // MC cuts on Mother (Id see below)
	58
	59	Int_t fIdMCMother; // MC Id of mother particle of interest (eg. Jpsi=443)
	60	Int_t fIdMCDaughterP; // MC Id of legs without sign
	61	Int_t fIdMCDaughterN; // MC Id of legs without sign
	62	//
	63	TDatabasePDG *fPDG; //! PDG database
	64
	65	void FillPlots(AliVEvent *event);
e123f993	66	void FillMCInfo(AliStack * const pStack);
b2a297fa	67
	68	AliAnalysisTaskDielectronEfficiency(const AliAnalysisTaskDielectronEfficiency &c);
	69	AliAnalysisTaskDielectronEfficiency& operator= (const AliAnalysisTaskDielectronEfficiency &c);
	70
	71	ClassDef(AliAnalysisTaskDielectronEfficiency, 1);
	72	};
	73	#endif