[u/mrichter/AliRoot.git] / PWGCF / Correlations / DPhi / DiHadronPID / AliAnalysisTaskCompareAODTrackCuts.h

#ifndef ALIANALYSISTASKCOMPAREAODTRACKCUTS_H
#define ALIANALYSISTASKCOMPAREAODTRACKCUTS_H

#include "AliAODEventCutsDiHadronPID.h"
#include "AliAODTrackCutsDiHadronPID.h"
/* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. * 
* See cxx source for full Copyright notice */ 
/* $Id$ */

class AliAnalysisTaskCompareAODTrackCuts : public AliAnalysisTaskSE {

// Basic AnalysisTask Functions.
public:
	AliAnalysisTaskCompareAODTrackCuts();
	AliAnalysisTaskCompareAODTrackCuts(const char* name);
	virtual ~AliAnalysisTaskCompareAODTrackCuts();
	
private:
	AliAnalysisTaskCompareAODTrackCuts(const AliAnalysisTaskCompareAODTrackCuts&);
	AliAnalysisTaskCompareAODTrackCuts& operator=(const AliAnalysisTaskCompareAODTrackCuts&);		

public:
	virtual void UserCreateOutputObjects(); 
	virtual void UserExec(Option_t*);
	virtual void Terminate(Option_t*);	

// Mismatch related functions.
	Bool_t LoadExternalMismatchHistos();			// For each mismatch method, external histo's are needed.
	Double_t GenerateRandomHit(Double_t eta);		// Generates a random time for a certain eta.

// Task Settings.
	void SetMC(Bool_t isMC = kTRUE) {fIsMC = isMC;} 
	void SetVerbose(Bool_t verbose = kTRUE) {fVerbose = verbose;}
	void SetCalculateTOFMismatch(Bool_t calculatetofmismatch = kTRUE/* const Int_t method*/) {fCalculateTOFMismatch = calculatetofmismatch;}
	void SetUseMismatchFileFromGridHomeDir(Bool_t usefromhomedir = kTRUE) {fUseMismatchFileFromHomeDir = usefromhomedir;}
	void SetUseNSigmaOnPIDAxes(Bool_t UseNSigma = kTRUE);

// Managing Cuts.
    void SetEventCuts(AliAODEventCutsDiHadronPID* eventcuts);
    void AddTrackCuts(AliAODTrackCutsDiHadronPID* trackcuts);

// Override from AliAnalysisTaskSE.
	void SetDebugLevel(Int_t debuglvl);

private:
	void FillGlobalTracksArray();
	AliAODTrack* GetGlobalTrack(AliAODTrack* track);

private:

	// PID Response Object.
	AliPIDResponse*					fPIDResponse;				//! PID Response.

	// Output List.
	TList* 							fOutputList;				//! Output List.

	// Settings (streamed!).
	Bool_t							fIsMC;						// ran over MC or not.
	Bool_t							fVerbose;					// Verbose mode.
	Bool_t 							fCalculateTOFMismatch;		// Compute mismatch or not. (Needs input histograms!)
	Bool_t							fUseMismatchFileFromHomeDir;// Take TOFmistmachHistos.root from the home dir, or take the one copied when submitting jobs
	Bool_t							fUseNSigmaOnPIDAxes;		// Uses NSigma on the PID axes of all histograms.

	// Event Cut Object (streamed!).
	AliAODEventCutsDiHadronPID*		fEventCuts;					// Event Cuts.
	
	// Array of Track Cut Objects (streamed!).
	TObjArray*						fTrackCuts;					// TObjArray with all Track Cut Objects.

	// Inclusive track times.
	TH2F* 							fInclusiveTimes;			//!

	// TOF mismatch stuff.
	TH1F*							fT0Fill;					//
	TH2F*							fLvsEta;					//
	TH2F*							fGlobalPtvsTPCPt;			//
	TObjArray*						fLvsEtaProjections;			//	


	// Event and Track related objects.
	AliAODEvent*					fCurrentAODEvent;			//!
	AliAODTrack*					fCurrentAODTrack;			//!
	AliTrackDiHadronPID*			fCurrentDiHadronPIDTrack;	//!
	TObjArray*						fGlobalTracksArray;			//! Array of Global Tracks.

	ClassDef(AliAnalysisTaskCompareAODTrackCuts,1);

};

#endif
Commit	Line	Data
6214ec0c	1	#ifndef ALIANALYSISTASKCOMPAREAODTRACKCUTS_H
6214ec0c	2	#define ALIANALYSISTASKCOMPAREAODTRACKCUTS_H
12d5e167	3
	4	#include "AliAODEventCutsDiHadronPID.h"
	5	#include "AliAODTrackCutsDiHadronPID.h"
6214ec0c	6	/* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
	7	* See cxx source for full Copyright notice */
	8	/* $Id$ */
	9
	10	class AliAnalysisTaskCompareAODTrackCuts : public AliAnalysisTaskSE {
	11
	12	// Basic AnalysisTask Functions.
	13	public:
	14	AliAnalysisTaskCompareAODTrackCuts();
	15	AliAnalysisTaskCompareAODTrackCuts(const char* name);
	16	virtual ~AliAnalysisTaskCompareAODTrackCuts();
	17
	18	private:
	19	AliAnalysisTaskCompareAODTrackCuts(const AliAnalysisTaskCompareAODTrackCuts&);
	20	AliAnalysisTaskCompareAODTrackCuts& operator=(const AliAnalysisTaskCompareAODTrackCuts&);
	21
	22	public:
	23	virtual void UserCreateOutputObjects();
	24	virtual void UserExec(Option_t*);
	25	virtual void Terminate(Option_t*);
	26
	27	// Mismatch related functions.
a5422983	28	Bool_t LoadExternalMismatchHistos(); // For each mismatch method, external histo's are needed.
a5422983	29	Double_t GenerateRandomHit(Double_t eta); // Generates a random time for a certain eta.
6214ec0c	30
28161374	31	// Task Settings.
fe463f34	32	void SetMC(Bool_t isMC = kTRUE) {fIsMC = isMC;}
	33	void SetVerbose(Bool_t verbose = kTRUE) {fVerbose = verbose;}
	34	void SetCalculateTOFMismatch(Bool_t calculatetofmismatch = kTRUE/* const Int_t method*/) {fCalculateTOFMismatch = calculatetofmismatch;}
	35	void SetUseMismatchFileFromGridHomeDir(Bool_t usefromhomedir = kTRUE) {fUseMismatchFileFromHomeDir = usefromhomedir;}
	36	void SetUseNSigmaOnPIDAxes(Bool_t UseNSigma = kTRUE);
6214ec0c	37
28161374	38	// Managing Cuts.
	39	void SetEventCuts(AliAODEventCutsDiHadronPID* eventcuts);
	40	void AddTrackCuts(AliAODTrackCutsDiHadronPID* trackcuts);
6214ec0c	41
28161374	42	// Override from AliAnalysisTaskSE.
fe463f34	43	void SetDebugLevel(Int_t debuglvl);
a5422983	44
6214ec0c	45	private:
	46	void FillGlobalTracksArray();
	47	AliAODTrack* GetGlobalTrack(AliAODTrack* track);
	48
	49	private:
	50
	51	// PID Response Object.
	52	AliPIDResponse* fPIDResponse; //! PID Response.
	53
	54	// Output List.
	55	TList* fOutputList; //! Output List.
	56
	57	// Settings (streamed!).
	58	Bool_t fIsMC; // ran over MC or not.
	59	Bool_t fVerbose; // Verbose mode.
	60	Bool_t fCalculateTOFMismatch; // Compute mismatch or not. (Needs input histograms!)
a5422983	61	Bool_t fUseMismatchFileFromHomeDir;// Take TOFmistmachHistos.root from the home dir, or take the one copied when submitting jobs
28161374	62	Bool_t fUseNSigmaOnPIDAxes; // Uses NSigma on the PID axes of all histograms.
6214ec0c	63
	64	// Event Cut Object (streamed!).
	65	AliAODEventCutsDiHadronPID* fEventCuts; // Event Cuts.
	66
	67	// Array of Track Cut Objects (streamed!).
	68	TObjArray* fTrackCuts; // TObjArray with all Track Cut Objects.
	69
	70	// Inclusive track times.
	71	TH2F* fInclusiveTimes; //!
	72
69868b6b	73	// TOF mismatch stuff.
	74	TH1F* fT0Fill; //
	75	TH2F* fLvsEta; //
a5422983	76	TH2F* fGlobalPtvsTPCPt; //
69868b6b	77	TObjArray* fLvsEtaProjections; //
6214ec0c	78
	79
	80	// Event and Track related objects.
	81	AliAODEvent* fCurrentAODEvent; //!
	82	AliAODTrack* fCurrentAODTrack; //!
	83	AliTrackDiHadronPID* fCurrentDiHadronPIDTrack; //!
	84	TObjArray* fGlobalTracksArray; //! Array of Global Tracks.
	85
	86	ClassDef(AliAnalysisTaskCompareAODTrackCuts,1);
	87
	88	};
	89
01f5c9f4	90	#endif