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

//
// Class AliRsnAnalysisEffSE
//
// Virtual Class derivated from AliRsnVAnalysisTaskSE which will be base class
// for all RSN SE tasks
//
// authors: Martin Vala (martin.vala@cern.ch)
//          Alberto Pulvirenti (alberto.pulvirenti@ct.infn.it)
//
#ifndef ALIRSNANALYSISEFFSE_H
#define ALIRSNANALYSISEFFSE_H

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

#include "AliRsnVAnalysisTaskSE.h"
#include "AliRsnEvent.h"
#include "AliRsnMother.h"

class AliPID;

class AliCFContainer;

class AliRsnPairDef;
class AliRsnPIDIndex;
class AliRsnPIDDefESD;
class AliRsnCutSet;
class AliRsnCutManager;
class AliRsnValue;

class AliRsnAnalysisManager;
class AliRsnAnalysisEffSE : public AliRsnVAnalysisTaskSE
{

  public:
    AliRsnAnalysisEffSE(const char *name = "AliRsnAnalysisTaskEffSE");
    AliRsnAnalysisEffSE(const AliRsnAnalysisEffSE& copy);
    virtual ~AliRsnAnalysisEffSE() {;};

    // Implement this
    virtual void    RsnUserCreateOutputObjects();
    virtual void    RsnUserExec(Option_t*);
    virtual void    RsnTerminate(Option_t*);

    // settings
    AliRsnCutSet*   GetEventCuts() {return &fEventCuts;}
    void            AddPairDef(AliRsnPairDef *pairDef);
    void            AddStepMC(AliRsnCutManager *mgr);
    void            AddStepESD(AliRsnCutManager *mgr);
    void            AddAxis(AliRsnValue *axis);

  private:

    AliRsnAnalysisEffSE& operator=(const AliRsnAnalysisEffSE& /*copy*/) {return *this;}
    void                 ProcessEvent(AliRsnPairDef *pairDef);
    void                 ProcessEventMC(AliRsnPairDef *pairDef);
    void                 ProcessEventESD(AliRsnPairDef *pairDef);
    void                 FillContainer(AliCFContainer *cont, const TObjArray *stepList, AliRsnPairDef *pd, Int_t firstOutStep);
    Int_t                FindESDtrack(Int_t label, AliESDEvent *esd, Bool_t rejectFakes);
    TArrayI              FindESDtracks(Int_t label, AliESDEvent *esd);

    Bool_t                fUseITSSA;                // switch to use ITS standalone tracks
    Bool_t                fUseGlobal;               // switch to use global tracks
    TObjArray             fStepListMC;              // list of cut managers for all steps with MC
    TObjArray             fStepListESD;             // list of cut managers for all steps with ESD
    TClonesArray          fAxisList;                // list of axes of efficiency plots
    TObjArray             fPairDefList;             // decay channels
    TList                *fContainerList;           // list of CF containers
    TList                *fOutList;                 // global output list
    TArrayD               fVar;                     // list of variables of the container
    AliRsnMother          fPair;                    // interface to pair
    AliRsnDaughter        fDaughter[2];             // interface to tracks
    AliRsnCutSet          fEventCuts;               // event cuts

    ClassDef(AliRsnAnalysisEffSE, 1)
};

#endif
Commit	Line	Data
4fbb2459	1	//
	2	// Class AliRsnAnalysisEffSE
	3	//
	4	// Virtual Class derivated from AliRsnVAnalysisTaskSE which will be base class
	5	// for all RSN SE tasks
	6	//
	7	// authors: Martin Vala (martin.vala@cern.ch)
	8	// Alberto Pulvirenti (alberto.pulvirenti@ct.infn.it)
	9	//
	10	#ifndef ALIRSNANALYSISEFFSE_H
	11	#define ALIRSNANALYSISEFFSE_H
	12
	13	#include <TArrayD.h>
c18b1218	14	#include <TArrayI.h>
c18b1218	15	#include <TClonesArray.h>
4fbb2459	16
	17	#include "AliRsnVAnalysisTaskSE.h"
	18	#include "AliRsnEvent.h"
2dab9030	19	#include "AliRsnMother.h"
4fbb2459	20
	21	class AliPID;
	22
	23	class AliCFContainer;
	24
	25	class AliRsnPairDef;
	26	class AliRsnPIDIndex;
	27	class AliRsnPIDDefESD;
	28	class AliRsnCutSet;
2dab9030	29	class AliRsnCutManager;
2dab9030	30	class AliRsnValue;
4fbb2459	31
	32	class AliRsnAnalysisManager;
	33	class AliRsnAnalysisEffSE : public AliRsnVAnalysisTaskSE
	34	{
	35
	36	public:
	37	AliRsnAnalysisEffSE(const char *name = "AliRsnAnalysisTaskEffSE");
	38	AliRsnAnalysisEffSE(const AliRsnAnalysisEffSE& copy);
	39	virtual ~AliRsnAnalysisEffSE() {;};
	40
	41	// Implement this
	42	virtual void RsnUserCreateOutputObjects();
	43	virtual void RsnUserExec(Option_t*);
	44	virtual void RsnTerminate(Option_t*);
	45
	46	// settings
c18b1218	47	AliRsnCutSet* GetEventCuts() {return &fEventCuts;}
4fbb2459	48	void AddPairDef(AliRsnPairDef *pairDef);
c18b1218	49	void AddStepMC(AliRsnCutManager *mgr);
	50	void AddStepESD(AliRsnCutManager *mgr);
	51	void AddAxis(AliRsnValue *axis);
4fbb2459	52
	53	private:
	54
	55	AliRsnAnalysisEffSE& operator=(const AliRsnAnalysisEffSE& /copy/) {return *this;}
fecaa02e	56	void ProcessEvent(AliRsnPairDef *pairDef);
4fbb2459	57	void ProcessEventMC(AliRsnPairDef *pairDef);
	58	void ProcessEventESD(AliRsnPairDef *pairDef);
	59	void FillContainer(AliCFContainer cont, const TObjArray stepList, AliRsnPairDef *pd, Int_t firstOutStep);
fecaa02e	60	Int_t FindESDtrack(Int_t label, AliESDEvent *esd, Bool_t rejectFakes);
c18b1218	61	TArrayI FindESDtracks(Int_t label, AliESDEvent *esd);
4fbb2459	62
fecaa02e	63	Bool_t fUseITSSA; // switch to use ITS standalone tracks
fecaa02e	64	Bool_t fUseGlobal; // switch to use global tracks
4fbb2459	65	TObjArray fStepListMC; // list of cut managers for all steps with MC
4fbb2459	66	TObjArray fStepListESD; // list of cut managers for all steps with ESD
c18b1218	67	TClonesArray fAxisList; // list of axes of efficiency plots
4fbb2459	68	TObjArray fPairDefList; // decay channels
4fbb2459	69	TList *fContainerList; // list of CF containers
fecaa02e	70	TList *fOutList; // global output list
4fbb2459	71	TArrayD fVar; // list of variables of the container
2dab9030	72	AliRsnMother fPair; // interface to pair
4fbb2459	73	AliRsnDaughter fDaughter[2]; // interface to tracks
c18b1218	74	AliRsnCutSet fEventCuts; // event cuts
4fbb2459	75
	76	ClassDef(AliRsnAnalysisEffSE, 1)
	77	};
	78
	79	#endif