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

#ifndef ALIANALYSISTASKDIELECTRONFILTER_H
#define ALIANALYSISTASKDIELECTRONFILTER_H
/* Copyright(c) 1998-2009, ALICE Experiment at CERN, All rights reserved. *
 * See cxx source for full Copyright notice                               */

//#####################################################
//#                                                   # 
//#        Dielectron even filter task                #
//#                                                   #
//#                                                   #
//#  by WooJin J. Park, GSI / W.J.Park@gsi.de         #
//#     Ionut C. Arsene, GSI / I.C.Arsene@gsi.de      #
//#     Magnus Mager, CERN / Magnus.Mager@cern.ch     #
//#     Jens Wiechula, Uni HD / Jens.Wiechula@cern.ch #
//#                                                   #
//#####################################################
/*
Filter Event based on cuts provided in the AliDielectron class.

Write an AOD file containing events with Dielectron candidates.
Add a sattelite AOD with the array of candidates.
*/


#include "AliAnalysisTaskSE.h"

class AliDielectron;
class TH1D;

class AliAnalysisTaskDielectronFilter : public AliAnalysisTaskSE {
  
public:
  AliAnalysisTaskDielectronFilter();
  AliAnalysisTaskDielectronFilter(const char *name);
  virtual ~AliAnalysisTaskDielectronFilter(){}

  virtual void UserExec(Option_t *option);
  virtual void Init();
  virtual void UserCreateOutputObjects();
  virtual void LocalInit() {Init();}

  void UsePhysicsSelection(Bool_t phy=kTRUE) {fSelectPhysics=phy;}
  void SetTriggerMask(UInt_t mask) {fTriggerMask=mask;}
  UInt_t GetTriggerMask() const { return fTriggerMask; }
  
  void SetDielectron(AliDielectron * const die) { fDielectron = die; }
  
private:
  
  AliDielectron *fDielectron;             // J/psi framework object

  Bool_t fSelectPhysics;                  // Whether to use physics selection
  UInt_t fTriggerMask;               // Event trigger mask

  TH1D *fEventStat;                  //! Histogram with event statistics
  
  AliAnalysisTaskDielectronFilter(const AliAnalysisTaskDielectronFilter &c);
  AliAnalysisTaskDielectronFilter& operator= (const AliAnalysisTaskDielectronFilter &c);
  
  ClassDef(AliAnalysisTaskDielectronFilter, 1);
};
#endif
Commit	Line	Data
b2a297fa	1	#ifndef ALIANALYSISTASKDIELECTRONFILTER_H
	2	#define ALIANALYSISTASKDIELECTRONFILTER_H
	3	/* Copyright(c) 1998-2009, ALICE Experiment at CERN, All rights reserved. *
	4	* See cxx source for full Copyright notice */
	5
	6	//#####################################################
	7	//# #
	8	//# Dielectron even filter task #
	9	//# #
	10	//# #
	11	//# by WooJin J. Park, GSI / W.J.Park@gsi.de #
	12	//# Ionut C. Arsene, GSI / I.C.Arsene@gsi.de #
	13	//# Magnus Mager, CERN / Magnus.Mager@cern.ch #
	14	//# Jens Wiechula, Uni HD / Jens.Wiechula@cern.ch #
	15	//# #
	16	//#####################################################
	17	/*
	18	Filter Event based on cuts provided in the AliDielectron class.
	19
	20	Write an AOD file containing events with Dielectron candidates.
	21	Add a sattelite AOD with the array of candidates.
	22	*/
	23
	24
	25
	26	#include "AliAnalysisTaskSE.h"
	27
	28	class AliDielectron;
61d106d3	29	class TH1D;
b2a297fa	30
	31	class AliAnalysisTaskDielectronFilter : public AliAnalysisTaskSE {
	32
	33	public:
	34	AliAnalysisTaskDielectronFilter();
	35	AliAnalysisTaskDielectronFilter(const char *name);
	36	virtual ~AliAnalysisTaskDielectronFilter(){}
	37
	38	virtual void UserExec(Option_t *option);
	39	virtual void Init();
61d106d3	40	virtual void UserCreateOutputObjects();
b2a297fa	41	virtual void LocalInit() {Init();}
8df8e382	42
8df8e382	43	void UsePhysicsSelection(Bool_t phy=kTRUE) {fSelectPhysics=phy;}
61d106d3	44	void SetTriggerMask(UInt_t mask) {fTriggerMask=mask;}
61d106d3	45	UInt_t GetTriggerMask() const { return fTriggerMask; }
b2a297fa	46
	47	void SetDielectron(AliDielectron * const die) { fDielectron = die; }
	48
	49	private:
	50
	51	AliDielectron *fDielectron; // J/psi framework object
	52
8df8e382	53	Bool_t fSelectPhysics; // Whether to use physics selection
61d106d3	54	UInt_t fTriggerMask; // Event trigger mask
	55
	56	TH1D *fEventStat; //! Histogram with event statistics
8df8e382	57
b2a297fa	58	AliAnalysisTaskDielectronFilter(const AliAnalysisTaskDielectronFilter &c);
	59	AliAnalysisTaskDielectronFilter& operator= (const AliAnalysisTaskDielectronFilter &c);
	60
	61	ClassDef(AliAnalysisTaskDielectronFilter, 1);
	62	};
	63	#endif