[u/mrichter/AliRoot.git] / ANALYSIS / AliAnalysisTaskPIDResponse.h

#ifndef ALIANALYSISTASKPIDRESPONSE_H
#define ALIANALYSISTASKPIDRESPONSE_H

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

/* $Id: AliAnalysisTaskPIDResponse.h 43642 2010-09-17 15:50:04Z wiechula $ */
// Author: Jens Wiechula, 24/02/2011

//==============================================================================
//
//
//
//
//==============================================================================

#include <TVectorDfwd.h>
#include <TString.h>

#ifndef ALIANALYSISTASKSE_H
#include "AliAnalysisTaskSE.h"
#endif

class AliPIDResponse;
class AliVEvent;

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

  void SetIsMC(Bool_t isMC=kTRUE)   { fIsMC=isMC; }
  void SetCachePID(Bool_t cachePID) { fCachePID=cachePID; }
  Bool_t GetCachePID() const { return fCachePID; }
  
  virtual void UserCreateOutputObjects();
  
  virtual void UserExec(Option_t */*option*/);

  void SetOADBPath(const char* path) {fOADBPath=path;}
  const char* GetOADBPath() const { return fOADBPath.Data(); }
  void SetTuneOnData(Bool_t flag,Int_t recopass){fIsTunedOnData=flag;fRecoPassTuned=recopass;};
  Bool_t GetTunedOnData() const { return fIsTunedOnData; };
  void SetTuneOnDataMask(Int_t mask){fTunedOnDataMask=mask;};
  
  void SetUseTPCEtaCorrection(Bool_t useTPCEtaCorrection) { fUseTPCEtaCorrection = useTPCEtaCorrection; };
  Bool_t UseTPCEtaCorrection() const { return fUseTPCEtaCorrection; };
  
  void SetUseTPCMultiplicityCorrection(Bool_t useMultiplicityCorrection = kTRUE) { fUseTPCMultiplicityCorrection = useMultiplicityCorrection; };
  Bool_t UseTPCMultiplicityCorrection() const { return fUseTPCMultiplicityCorrection; };

  void SetSpecialDetectorResponse(const char* det) { fSpecialDetResponse=det; }
  void SetUserDataRecoPass(Int_t pass){fUserDataRecoPass=pass;};


private:
  Bool_t fIsMC;                        // If we run on MC data
  Bool_t fCachePID;                    // Cache PID values in transient object
  TString fOADBPath;                   // OADB path to use
  TString fSpecialDetResponse;         // Special detector response files for debugging
  
  AliPIDResponse *fPIDResponse;        //! PID response Handler
  Int_t   fRun;                        //! current run number
  Int_t   fOldRun;                     //! current run number
  Int_t   fRecoPass;                   //! reconstruction pass

  Bool_t  fIsTunedOnData;              // flag to tune MC on data (dE/dx)
  Int_t   fTunedOnDataMask;            // mask to activate tuning on data on specific detectors
  Int_t   fRecoPassTuned;              // Reco pass tuned on data for MC
  
  Bool_t fUseTPCEtaCorrection;          // Use TPC eta correction
  Bool_t fUseTPCMultiplicityCorrection; // Use TPC multiplicity correction
  Int_t  fUserDataRecoPass;            // forced DATA reco pass
  
  //
  void SetRecoInfo();
    
  AliAnalysisTaskPIDResponse(const AliAnalysisTaskPIDResponse &other);
  AliAnalysisTaskPIDResponse& operator=(const AliAnalysisTaskPIDResponse &other);
  
  ClassDef(AliAnalysisTaskPIDResponse,7)  // Task to properly set the PID response functions of all detectors
};
#endif
Commit	Line	Data
	1	#ifndef ALIANALYSISTASKPIDRESPONSE_H
	2	#define ALIANALYSISTASKPIDRESPONSE_H
	3
	4	/* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
	5	* See cxx source for full Copyright notice */
	6
	7	/* $Id: AliAnalysisTaskPIDResponse.h 43642 2010-09-17 15:50:04Z wiechula $ */
	8	// Author: Jens Wiechula, 24/02/2011
	9
	10	//==============================================================================
	11	//
	12	//
	13	//
	14	//
	15	//==============================================================================
	16
	17	#include <TVectorDfwd.h>
	18	#include <TString.h>
	19
	20	#ifndef ALIANALYSISTASKSE_H
	21	#include "AliAnalysisTaskSE.h"
	22	#endif
	23
	24	class AliPIDResponse;
	25	class AliVEvent;
	26
	27	class AliAnalysisTaskPIDResponse : public AliAnalysisTaskSE {
	28
	29
	30	public:
	31	AliAnalysisTaskPIDResponse();
	32	AliAnalysisTaskPIDResponse(const char *name);
	33	virtual ~AliAnalysisTaskPIDResponse();
	34
	35	void SetIsMC(Bool_t isMC=kTRUE) { fIsMC=isMC; }
	36	void SetCachePID(Bool_t cachePID) { fCachePID=cachePID; }
	37	Bool_t GetCachePID() const { return fCachePID; }
	38
	39	virtual void UserCreateOutputObjects();
	40
	41	virtual void UserExec(Option_t /option*/);
	42
	43	void SetOADBPath(const char* path) {fOADBPath=path;}
	44	const char* GetOADBPath() const { return fOADBPath.Data(); }
	45	void SetTuneOnData(Bool_t flag,Int_t recopass){fIsTunedOnData=flag;fRecoPassTuned=recopass;};
	46	Bool_t GetTunedOnData() const { return fIsTunedOnData; };
	47	void SetTuneOnDataMask(Int_t mask){fTunedOnDataMask=mask;};
	48
	49	void SetUseTPCEtaCorrection(Bool_t useTPCEtaCorrection) { fUseTPCEtaCorrection = useTPCEtaCorrection; };
	50	Bool_t UseTPCEtaCorrection() const { return fUseTPCEtaCorrection; };
	51
	52	void SetUseTPCMultiplicityCorrection(Bool_t useMultiplicityCorrection = kTRUE) { fUseTPCMultiplicityCorrection = useMultiplicityCorrection; };
	53	Bool_t UseTPCMultiplicityCorrection() const { return fUseTPCMultiplicityCorrection; };
	54
	55	void SetSpecialDetectorResponse(const char* det) { fSpecialDetResponse=det; }
	56	void SetUserDataRecoPass(Int_t pass){fUserDataRecoPass=pass;};
	57
	58
	59	private:
	60	Bool_t fIsMC; // If we run on MC data
	61	Bool_t fCachePID; // Cache PID values in transient object
	62	TString fOADBPath; // OADB path to use
	63	TString fSpecialDetResponse; // Special detector response files for debugging
	64
	65	AliPIDResponse *fPIDResponse; //! PID response Handler
	66	Int_t fRun; //! current run number
	67	Int_t fOldRun; //! current run number
	68	Int_t fRecoPass; //! reconstruction pass
	69
	70	Bool_t fIsTunedOnData; // flag to tune MC on data (dE/dx)
	71	Int_t fTunedOnDataMask; // mask to activate tuning on data on specific detectors
	72	Int_t fRecoPassTuned; // Reco pass tuned on data for MC
	73
	74	Bool_t fUseTPCEtaCorrection; // Use TPC eta correction
	75	Bool_t fUseTPCMultiplicityCorrection; // Use TPC multiplicity correction
	76	Int_t fUserDataRecoPass; // forced DATA reco pass
	77
	78	//
	79	void SetRecoInfo();
	80
	81	AliAnalysisTaskPIDResponse(const AliAnalysisTaskPIDResponse &other);
	82	AliAnalysisTaskPIDResponse& operator=(const AliAnalysisTaskPIDResponse &other);
	83
	84	ClassDef(AliAnalysisTaskPIDResponse,7) // Task to properly set the PID response functions of all detectors
	85	};
	86	#endif