[u/mrichter/AliRoot.git] / PWG4 / JetTasks / AliAnalysisTaskJetServices.h

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

// **************************************
// Task used for the correction of determiantion of reconstructed jet spectra
// Compares input (gen) and output (rec) jets   
// *******************************************

#include  "AliAnalysisTaskSE.h"
#include  "THnSparse.h" // cannot forward declare ThnSparseF

////////////////
class AliJetHeader;
class AliESDEvent;
class AliESDVertex;
class AliAODEvent;
class AliAODVertex;
class AliAODJet;
class AliGenPythiaEventHeader;
class AliCFManager;
class AliTriggerAnalysis;

class TList;
class TChain;
class TH1F;
class TH2F;
class TH3F;
class TProfile;


class AliAnalysisTaskJetServices : public AliAnalysisTaskSE
{
 public:
    AliAnalysisTaskJetServices();
    AliAnalysisTaskJetServices(const char* name);
    virtual ~AliAnalysisTaskJetServices() {;}
    // Implementation of interface methods
    virtual void UserCreateOutputObjects();
    virtual void Init();
    virtual void LocalInit() { Init(); }
    virtual void UserExec(Option_t *option);
    virtual void Terminate(Option_t *option);
    virtual void SetZVertexCut(Float_t f){fVtxZCut = f;}
    virtual void SetPtMinCosmic(Float_t ptMin) {fPtMinCosmic = ptMin;}
    virtual void SetRMinCosmic(Float_t rMin) {fRIsolMinCosmic = rMin;}
    virtual void SetMaxCosmicAngle(Float_t angle)   {fMaxCosmicAngle = angle;}
    virtual Bool_t Notify();

    virtual void SetAODInput(Bool_t b){fUseAODInput = b;}
    virtual void SetRunRange(Float_t fLo,Float_t fUp){fRunRange[0] = fLo;fRunRange[1] = fUp;}
    virtual void SetMCData(Bool_t b){fMC = b;}
    virtual void SetUsePhysicsSelection(Bool_t b){fUsePhysicsSelection = b;}
    virtual void SetPhysicsSelectionFlag(Int_t i){fPhysicsSelectionFlag = i;}
    virtual void SetFilterAODCollisions(Bool_t b){fFilterAODCollisions = b;}

    virtual void SetNonStdFile(char *c){fNonStdFile = c;}
    Bool_t IsEventSelected(const AliESDEvent* esd);
    Bool_t IsEventSelected(const AliAODEvent* aod) const;

    Bool_t IsEventPileUp(const AliESDEvent* esd) const;

    Bool_t IsEventCosmic(const AliESDEvent* esd) const;

    Bool_t IsVertexValid(const AliESDVertex *vtx);
    Bool_t IsVertexValid(const AliAODVertex *vtx) const;

    Bool_t IsVertexIn(const AliESDVertex *vtx);
    Bool_t IsVertexIn(const AliAODVertex *vtx) const;
    Int_t GetEventClass(AliESDEvent *esd);

    enum { kAllTriggered = 0,kTriggeredVertex,kTriggeredVertexIn,kSelectedALICE,kSelectedALICEVertexValid,kSelectedALICEVertexIn,kSelected,kConstraints};

    enum { kNoEventCut=1<<0,
	   kPhysicsSelectionCut=1<<1,
	   kContributorsCut1=1<<2,
	   kContributorsCut2=1<<3,
	   kContributorsCut3=1<<4,
	   kVertexTPC=1<<5,
	   kVertexSPD=1<<6,
	   kVertexGlobal=1<<7,
	   kSPDDispersionCut=1<<8,
	   kVertexZCut=1<<9,
	   kVertexRCut=1<<10,
	   kTotalEventCuts=(1<<11)-1};

 private:

    AliAnalysisTaskJetServices(const AliAnalysisTaskJetServices&);
    AliAnalysisTaskJetServices& operator=(const AliAnalysisTaskJetServices&);

    Bool_t        fUseAODInput;        // take jet from input AOD not from ouptu AOD
    Bool_t        fUsePhysicsSelection;// decide wether we take into account physicsselction task
    Bool_t        fMC;                 // true for MC data to allow correct trigger slection
    Bool_t        fFilterAODCollisions; // filter out collision canditates to the AOD
    UInt_t        fPhysicsSelectionFlag; // defines the glag for acceptance of events from physics selection
    UInt_t        fSelectionInfoESD;   // slection info bit mask
    UInt_t        fEventCutInfoESD;   // event selection info of what is cutted after physics selection
    Float_t       fAvgTrials;          // Average number of trials
    Float_t       fVtxXMean;           // mean x for cuts
    Float_t       fVtxYMean;           // mean y for cuts
    Float_t       fVtxZMean;           // mean z for cuts
    Float_t       fVtxRCut;            // vtx cut in R
    Float_t       fVtxZCut;            // vtzx cut in Z
    Float_t       fPtMinCosmic;        // Minimum pT to be considered as cosmic candidate
    Float_t       fRIsolMinCosmic;     // Minimum R = sqrt{deltaPhi^2 + deltaEta^2} to be considered as cosmic candidate
    Float_t       fMaxCosmicAngle;     // Max deviation from pi (angle between two tracks) in case of cosmic candidate
    Float_t       fRunRange[2];        // only important for real data for 
    TString       fNonStdFile;         // outputName for replication
    TProfile*     fh1Xsec;             //! pythia cross section and trials
    TH1F*         fh1Trials;           //! trials are added
    TH1F*         fh1PtHard;           //! Pt har of the event...       
    TH1F*         fh1PtHardTrials;     //! Number of trials 
    TH1F*         fh1SelectionInfoESD; //! Masks that satisfy fSelectionInfo
    TH1F*         fh1EventCutInfoESD;  //! Masks that satisfy fSelectionInfo
    TH2F*         fh2TriggerCount;     //! number of fire triggers in each case
    TH2F*         fh2ESDTriggerCount;  //! number of fire triggers in each case
    TH2F*         fh2TriggerVtx;       //! vtx. position vs. trigger decision
    TH2F*         fh2ESDTriggerVtx;    //! vtx. position vs. trigger decision 
    TH2F*         fh2ESDTriggerRun;    //! fired triggers vs. run number
    TH2F*         fh2VtxXY;            //! XY position of VTX were available
    TH1F*         fh1NCosmicsPerEvent;  //! Number of coscmic candidates found in event
    AliTriggerAnalysis *fTriggerAnalysis; //! Trigger Analysis to get the background rates etc.
    TList *fHistList; //! Output list

        // Provisions for replication
    static AliAODHeader*    fgAODHeader;        //! Header for replication

    ClassDef(AliAnalysisTaskJetServices,11)
};
 
#endif
Commit	Line	Data
	1	#ifndef ALIANALYSISTASKJETSERVICES_H
	2	#define ALIANALYSISTASKJETSERVICES_H
	3
	4	/* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
	5	* See cxx source for full Copyright notice */
	6
	7	// **************************************
	8	// Task used for the correction of determiantion of reconstructed jet spectra
	9	// Compares input (gen) and output (rec) jets
	10	// *******************************************
	11
	12	#include "AliAnalysisTaskSE.h"
	13	#include "THnSparse.h" // cannot forward declare ThnSparseF
	14
	15	////////////////
	16	class AliJetHeader;
	17	class AliESDEvent;
	18	class AliESDVertex;
	19	class AliAODEvent;
	20	class AliAODVertex;
	21	class AliAODJet;
	22	class AliGenPythiaEventHeader;
	23	class AliCFManager;
	24	class AliTriggerAnalysis;
	25
	26	class TList;
	27	class TChain;
	28	class TH1F;
	29	class TH2F;
	30	class TH3F;
	31	class TProfile;
	32
	33
	34
	35	class AliAnalysisTaskJetServices : public AliAnalysisTaskSE
	36	{
	37	public:
	38	AliAnalysisTaskJetServices();
	39	AliAnalysisTaskJetServices(const char* name);
	40	virtual ~AliAnalysisTaskJetServices() {;}
	41	// Implementation of interface methods
	42	virtual void UserCreateOutputObjects();
	43	virtual void Init();
	44	virtual void LocalInit() { Init(); }
	45	virtual void UserExec(Option_t *option);
	46	virtual void Terminate(Option_t *option);
	47	virtual void SetZVertexCut(Float_t f){fVtxZCut = f;}
	48	virtual void SetPtMinCosmic(Float_t ptMin) {fPtMinCosmic = ptMin;}
	49	virtual void SetRMinCosmic(Float_t rMin) {fRIsolMinCosmic = rMin;}
	50	virtual void SetMaxCosmicAngle(Float_t angle) {fMaxCosmicAngle = angle;}
	51	virtual Bool_t Notify();
	52
	53	virtual void SetAODInput(Bool_t b){fUseAODInput = b;}
	54	virtual void SetRunRange(Float_t fLo,Float_t fUp){fRunRange[0] = fLo;fRunRange[1] = fUp;}
	55	virtual void SetMCData(Bool_t b){fMC = b;}
	56	virtual void SetUsePhysicsSelection(Bool_t b){fUsePhysicsSelection = b;}
	57	virtual void SetPhysicsSelectionFlag(Int_t i){fPhysicsSelectionFlag = i;}
	58	virtual void SetFilterAODCollisions(Bool_t b){fFilterAODCollisions = b;}
	59
	60	virtual void SetNonStdFile(char *c){fNonStdFile = c;}
	61	Bool_t IsEventSelected(const AliESDEvent* esd);
	62	Bool_t IsEventSelected(const AliAODEvent* aod) const;
	63
	64	Bool_t IsEventPileUp(const AliESDEvent* esd) const;
	65
	66	Bool_t IsEventCosmic(const AliESDEvent* esd) const;
	67
	68	Bool_t IsVertexValid(const AliESDVertex *vtx);
	69	Bool_t IsVertexValid(const AliAODVertex *vtx) const;
	70
	71	Bool_t IsVertexIn(const AliESDVertex *vtx);
	72	Bool_t IsVertexIn(const AliAODVertex *vtx) const;
	73	Int_t GetEventClass(AliESDEvent *esd);
	74
	75	enum { kAllTriggered = 0,kTriggeredVertex,kTriggeredVertexIn,kSelectedALICE,kSelectedALICEVertexValid,kSelectedALICEVertexIn,kSelected,kConstraints};
	76
	77	enum { kNoEventCut=1<<0,
	78	kPhysicsSelectionCut=1<<1,
	79	kContributorsCut1=1<<2,
	80	kContributorsCut2=1<<3,
	81	kContributorsCut3=1<<4,
	82	kVertexTPC=1<<5,
	83	kVertexSPD=1<<6,
	84	kVertexGlobal=1<<7,
	85	kSPDDispersionCut=1<<8,
	86	kVertexZCut=1<<9,
	87	kVertexRCut=1<<10,
	88	kTotalEventCuts=(1<<11)-1};
	89
	90	private:
	91
	92	AliAnalysisTaskJetServices(const AliAnalysisTaskJetServices&);
	93	AliAnalysisTaskJetServices& operator=(const AliAnalysisTaskJetServices&);
	94
	95	Bool_t fUseAODInput; // take jet from input AOD not from ouptu AOD
	96	Bool_t fUsePhysicsSelection;// decide wether we take into account physicsselction task
	97	Bool_t fMC; // true for MC data to allow correct trigger slection
	98	Bool_t fFilterAODCollisions; // filter out collision canditates to the AOD
	99	UInt_t fPhysicsSelectionFlag; // defines the glag for acceptance of events from physics selection
	100	UInt_t fSelectionInfoESD; // slection info bit mask
	101	UInt_t fEventCutInfoESD; // event selection info of what is cutted after physics selection
	102	Float_t fAvgTrials; // Average number of trials
	103	Float_t fVtxXMean; // mean x for cuts
	104	Float_t fVtxYMean; // mean y for cuts
	105	Float_t fVtxZMean; // mean z for cuts
	106	Float_t fVtxRCut; // vtx cut in R
	107	Float_t fVtxZCut; // vtzx cut in Z
	108	Float_t fPtMinCosmic; // Minimum pT to be considered as cosmic candidate
	109	Float_t fRIsolMinCosmic; // Minimum R = sqrt{deltaPhi^2 + deltaEta^2} to be considered as cosmic candidate
	110	Float_t fMaxCosmicAngle; // Max deviation from pi (angle between two tracks) in case of cosmic candidate
	111	Float_t fRunRange[2]; // only important for real data for
	112	TString fNonStdFile; // outputName for replication
	113	TProfile* fh1Xsec; //! pythia cross section and trials
	114	TH1F* fh1Trials; //! trials are added
	115	TH1F* fh1PtHard; //! Pt har of the event...
	116	TH1F* fh1PtHardTrials; //! Number of trials
	117	TH1F* fh1SelectionInfoESD; //! Masks that satisfy fSelectionInfo
	118	TH1F* fh1EventCutInfoESD; //! Masks that satisfy fSelectionInfo
	119	TH2F* fh2TriggerCount; //! number of fire triggers in each case
	120	TH2F* fh2ESDTriggerCount; //! number of fire triggers in each case
	121	TH2F* fh2TriggerVtx; //! vtx. position vs. trigger decision
	122	TH2F* fh2ESDTriggerVtx; //! vtx. position vs. trigger decision
	123	TH2F* fh2ESDTriggerRun; //! fired triggers vs. run number
	124	TH2F* fh2VtxXY; //! XY position of VTX were available
	125	TH1F* fh1NCosmicsPerEvent; //! Number of coscmic candidates found in event
	126	AliTriggerAnalysis *fTriggerAnalysis; //! Trigger Analysis to get the background rates etc.
	127	TList *fHistList; //! Output list
	128
	129	// Provisions for replication
	130	static AliAODHeader* fgAODHeader; //! Header for replication
	131
	132	ClassDef(AliAnalysisTaskJetServices,11)
	133	};
	134
	135	#endif