[u/mrichter/AliRoot.git] / PWG1 / TPC / AliTPCComparisonPID.h

#ifndef ALITPCOMPARISONPID_H
#define ALITPCOMPARISONPID_H

// ROOT includes
#include <TList.h>
#include <TH1.h>
#include <TH2.h>

// AliRoot includes
#include <AliAnalysisTask.h>
#include <AliESDEvent.h>
#include <AliESDfriend.h>
#include <AliESDtrack.h>
#include <AliESDfriendTrack.h>
#include <AliTPCseed.h>
#include <TString.h>
class AliGenInfoMaker;
class TTreeSRedirector;
class AliMCEventHadnler;
class TParticle;
class AliMCInfo;
class AliESDRecInfo;
class AliESDEvent;
class AliMCEvent;
class THnSparse;

class AliTPCComparisonPID : public AliAnalysisTask {
public:
  AliTPCComparisonPID();
  AliTPCComparisonPID(const char *name);
  AliTPCComparisonPID(const AliTPCComparisonPID& info);
  virtual ~AliTPCComparisonPID();  
  virtual void   ConnectInputData(Option_t *);
  virtual void   CreateOutputObjects();
  virtual void   Exec(Option_t *option);
  virtual void   Terminate(Option_t *);
  virtual void   FinishTaskOutput();
  void           SetDebugOuputhPath(const char * name){fDebugOutputPath=name;}
  void           Init();
  //
  void           ProcessMCInfo();
  //
  THnSparse * GetTPCsignal(){return fTPCsignal;}
  THnSparse * GetTPCsignalNorm(){return fTPCsignalNorm;}
  //
  // debug streamer part
  //
  TTreeSRedirector *GetDebugStreamer();
  void       SetStreamLevel(Int_t streamLevel){fStreamLevel=streamLevel;}
  void       SetDebugLevel(Int_t level) {fDebugLevel = level;}
  Int_t      GetStreamLevel() const {return fStreamLevel;}
  Int_t      GetDebugLevel() const {return fDebugLevel;}
  //
protected:
  void RegisterDebugOutput();
  AliTPCComparisonPID& operator=(const AliTPCComparisonPID& /*info*/) { return *this;}
  AliMCEvent  * fMCinfo;          //! MC event handler
  AliESDEvent * fESD;             //! current esd event
  //
  //
  //
  THnSparse * fTPCsignal;         //raw tpc signal - dEdx
  THnSparse * fTPCsignalNorm;     //normalized TPC signal
  //
  TTreeSRedirector *fDebugStreamer;     //! debug streamer
  Int_t  fStreamLevel;                  //  debug stream level 
  Int_t  fDebugLevel;                   //  debug level
  TString      fDebugOutputPath; // debug output path
  ClassDef(AliTPCComparisonPID, 1); // Analysis task base class for tracks
};

#endif
Commit	Line	Data
7cc34f08	1	#ifndef ALITPCOMPARISONPID_H
	2	#define ALITPCOMPARISONPID_H
	3
	4	// ROOT includes
	5	#include <TList.h>
	6	#include <TH1.h>
	7	#include <TH2.h>
	8
	9	// AliRoot includes
	10	#include <AliAnalysisTask.h>
	11	#include <AliESDEvent.h>
	12	#include <AliESDfriend.h>
	13	#include <AliESDtrack.h>
	14	#include <AliESDfriendTrack.h>
	15	#include <AliTPCseed.h>
	16	#include <TString.h>
	17	class AliGenInfoMaker;
	18	class TTreeSRedirector;
	19	class AliMCEventHadnler;
	20	class TParticle;
	21	class AliMCInfo;
	22	class AliESDRecInfo;
	23	class AliESDEvent;
	24	class AliMCEvent;
	25	class THnSparse;
	26
	27	class AliTPCComparisonPID : public AliAnalysisTask {
	28	public:
	29	AliTPCComparisonPID();
	30	AliTPCComparisonPID(const char *name);
	31	AliTPCComparisonPID(const AliTPCComparisonPID& info);
	32	virtual ~AliTPCComparisonPID();
	33	virtual void ConnectInputData(Option_t *);
	34	virtual void CreateOutputObjects();
	35	virtual void Exec(Option_t *option);
	36	virtual void Terminate(Option_t *);
	37	virtual void FinishTaskOutput();
	38	void SetDebugOuputhPath(const char * name){fDebugOutputPath=name;}
	39	void Init();
	40	//
	41	void ProcessMCInfo();
	42	//
	43	THnSparse * GetTPCsignal(){return fTPCsignal;}
	44	THnSparse * GetTPCsignalNorm(){return fTPCsignalNorm;}
	45	//
	46	// debug streamer part
	47	//
	48	TTreeSRedirector *GetDebugStreamer();
	49	void SetStreamLevel(Int_t streamLevel){fStreamLevel=streamLevel;}
	50	void SetDebugLevel(Int_t level) {fDebugLevel = level;}
	51	Int_t GetStreamLevel() const {return fStreamLevel;}
	52	Int_t GetDebugLevel() const {return fDebugLevel;}
	53	//
	54	protected:
	55	void RegisterDebugOutput();
	56	AliTPCComparisonPID& operator=(const AliTPCComparisonPID& /info/) { return *this;}
	57	AliMCEvent * fMCinfo; //! MC event handler
	58	AliESDEvent * fESD; //! current esd event
	59	//
	60	//
	61	//
	62	THnSparse * fTPCsignal; //raw tpc signal - dEdx
	63	THnSparse * fTPCsignalNorm; //normalized TPC signal
	64	//
65	TTreeSRedirector *fDebugStreamer; //! debug streamer
66	Int_t fStreamLevel; // debug stream level
67	Int_t fDebugLevel; // debug level
68	TString fDebugOutputPath; // debug output path
69	ClassDef(AliTPCComparisonPID, 1); // Analysis task base class for tracks
70	};
71
72	#endif