[u/mrichter/AliRoot.git] / PWG2 / RESONANCES / AliRsnAnalysisTaskEff.h

//
// Class AliRsnAnalysisTaskEff
//
// Base class for efficiency computation tasks
// which should be inherited by different efficiency computators
//
// author: Alberto Pulvirenti (alberto.pulvirenti@ct.infn.it)
//

#ifndef ALIRSNANALYSISTASKEFF_H
#define ALIRSNANALYSISTASKEFF_H

#include <TArrayI.h>
#include <TArrayD.h>
#include <TObjArray.h>
#include <TClonesArray.h>

#include "AliCFContainer.h"

#include "AliRsnValue.h"
#include "AliRsnCutSet.h"
#include "AliRsnVAnalysisTask.h"

class TList;
class AliVEvent;

class AliRsnAnalysisTaskEff : public AliRsnVAnalysisTask {

public:

   AliRsnAnalysisTaskEff(const char *name = "AliRsnAnalysisTasEff");
   AliRsnAnalysisTaskEff(const AliRsnAnalysisTaskEff& copy);
   AliRsnAnalysisTaskEff& operator=(const AliRsnAnalysisTaskEff& copy);
   virtual ~AliRsnAnalysisTaskEff() {;};
   
   // work-flow
   AliRsnCutSet*   GetEventCuts() {return &fEventCuts;}
   void            AddDef(TObject *def);
   void            AddAxis(AliRsnValue *axis);
   void            AddStepMC(TObject *set);
   void            AddStepRec(TObject *set);

   // inherited
   virtual void    RsnUserCreateOutputObjects();
   virtual void    RsnUserExec(Option_t*);
   virtual void    RsnTerminate(Option_t*);
   virtual Bool_t  RsnEventProcess();

protected:

   TArrayI         FindTracks(Int_t label, AliVEvent *esd);
   virtual void    ProcessEventESD();
   virtual void    ProcessEventAOD();
   virtual Int_t   NGoodSteps();
   virtual void    FillContainer(Bool_t mcList, TObject *def);
   
   TObjArray       fDefs;        //  list of definitions
   TObjArray       fStepsMC;     //  list of cuts for all steps with MC tracks
   TObjArray       fStepsRec;    //  list of cuts for all steps with reconstructed tracks
   TClonesArray    fAxes;        //  list of axes of efficiency plots
   
   TList          *fOutList;     //  global output list
   AliRsnCutSet    fEventCuts;   //  event cuts
   
   TArrayD         fVar;         //! list of variables of the container (temporary)

   ClassDef(AliRsnAnalysisTaskEff, 1)
};

#endif
Commit	Line	Data
99261e24	1	//
	2	// Class AliRsnAnalysisTaskEff
	3	//
	4	// Base class for efficiency computation tasks
	5	// which should be inherited by different efficiency computators
	6	//
	7	// author: Alberto Pulvirenti (alberto.pulvirenti@ct.infn.it)
	8	//
	9
	10	#ifndef ALIRSNANALYSISTASKEFF_H
	11	#define ALIRSNANALYSISTASKEFF_H
	12
	13	#include <TArrayI.h>
	14	#include <TArrayD.h>
	15	#include <TObjArray.h>
	16	#include <TClonesArray.h>
	17
	18	#include "AliCFContainer.h"
	19
	20	#include "AliRsnValue.h"
	21	#include "AliRsnCutSet.h"
	22	#include "AliRsnVAnalysisTask.h"
	23
	24	class TList;
	25	class AliVEvent;
	26
	27	class AliRsnAnalysisTaskEff : public AliRsnVAnalysisTask {
	28
	29	public:
	30
	31	AliRsnAnalysisTaskEff(const char *name = "AliRsnAnalysisTasEff");
	32	AliRsnAnalysisTaskEff(const AliRsnAnalysisTaskEff& copy);
	33	AliRsnAnalysisTaskEff& operator=(const AliRsnAnalysisTaskEff& copy);
	34	virtual ~AliRsnAnalysisTaskEff() {;};
	35
	36	// work-flow
	37	AliRsnCutSet* GetEventCuts() {return &fEventCuts;}
	38	void AddDef(TObject *def);
	39	void AddAxis(AliRsnValue *axis);
	40	void AddStepMC(TObject *set);
	41	void AddStepRec(TObject *set);
	42
	43	// inherited
	44	virtual void RsnUserCreateOutputObjects();
	45	virtual void RsnUserExec(Option_t*);
	46	virtual void RsnTerminate(Option_t*);
d7712d44	47	virtual Bool_t RsnEventProcess();
99261e24	48
	49	protected:
	50
	51	TArrayI FindTracks(Int_t label, AliVEvent *esd);
	52	virtual void ProcessEventESD();
	53	virtual void ProcessEventAOD();
	54	virtual Int_t NGoodSteps();
d7712d44	55	virtual void FillContainer(Bool_t mcList, TObject *def);
99261e24	56
	57	TObjArray fDefs; // list of definitions
	58	TObjArray fStepsMC; // list of cuts for all steps with MC tracks
	59	TObjArray fStepsRec; // list of cuts for all steps with reconstructed tracks
	60	TClonesArray fAxes; // list of axes of efficiency plots
	61
	62	TList *fOutList; // global output list
	63	AliRsnCutSet fEventCuts; // event cuts
	64
	65	TArrayD fVar; //! list of variables of the container (temporary)
	66
	67	ClassDef(AliRsnAnalysisTaskEff, 1)
	68	};
	69
	70	#endif