[u/mrichter/AliRoot.git] / PWG1 / AliGenInfoTask.h

#ifndef ALIGENINFOTASK_H
#define ALIGENINFOTASK_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>
class AliGenInfoMaker;

class AliGenInfoTask : public AliAnalysisTask {
 public:
 AliGenInfoTask();
 AliGenInfoTask(const char *name);
  virtual ~AliGenInfoTask() {};
  
  virtual void   ConnectInputData(Option_t *);
  virtual void   CreateOutputObjects();
  virtual void   Exec(Option_t *option);
  virtual void   Terminate(Option_t *);
  virtual void   SetDebugLevel(Int_t level) {fDebug = level;}
  virtual void   SetMaxTracks(Int_t max=10) {fMaxTracks = max;}
  
 protected:
  AliGenInfoTask(const AliGenInfoTask& /*info*/);
  AliGenInfoTask& operator=(const AliGenInfoTask& /*info*/) { return *this;}

  virtual Int_t FillTrackHistograms(Int_t nTracks, AliESDtrack* track, 
				    AliESDfriendTrack* friendTrack, 
				    AliTPCseed* seed);
  AliGenInfoMaker *fGenMaker;    // gen Maker
  Int_t         fDebug;          //  Debug flag
  AliESDEvent*  fESD;            //! ESD
  AliESDfriend* fESDfriend;      //! ESD friend
  TList*        fListOfHists;    //! Output list of histograms
  
  Int_t         fMaxTracks;      // Max tracks in histogram
  TH1F*         hESDTracks;      //! N ESD tracks
  TH1F*         hGoodTracks;     //! GOOD tracks

  ClassDef(AliGenInfoTask, 1); // Analysis task base class for TPC tracks and clusters
};

#endif
Commit	Line	Data
cd875161	1	#ifndef ALIGENINFOTASK_H
	2	#define ALIGENINFOTASK_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	class AliGenInfoMaker;
	17
	18	class AliGenInfoTask : public AliAnalysisTask {
	19	public:
	20	AliGenInfoTask();
	21	AliGenInfoTask(const char *name);
	22	virtual ~AliGenInfoTask() {};
	23
	24	virtual void ConnectInputData(Option_t *);
	25	virtual void CreateOutputObjects();
	26	virtual void Exec(Option_t *option);
	27	virtual void Terminate(Option_t *);
	28	virtual void SetDebugLevel(Int_t level) {fDebug = level;}
	29	virtual void SetMaxTracks(Int_t max=10) {fMaxTracks = max;}
	30
	31	protected:
	32	AliGenInfoTask(const AliGenInfoTask& /info/);
	33	AliGenInfoTask& operator=(const AliGenInfoTask& /info/) { return *this;}
	34
	35	virtual Int_t FillTrackHistograms(Int_t nTracks, AliESDtrack* track,
	36	AliESDfriendTrack* friendTrack,
	37	AliTPCseed* seed);
	38	AliGenInfoMaker *fGenMaker; // gen Maker
	39	Int_t fDebug; // Debug flag
	40	AliESDEvent* fESD; //! ESD
	41	AliESDfriend* fESDfriend; //! ESD friend
	42	TList* fListOfHists; //! Output list of histograms
	43
	44	Int_t fMaxTracks; // Max tracks in histogram
	45	TH1F* hESDTracks; //! N ESD tracks
	46	TH1F* hGoodTracks; //! GOOD tracks
	47
	48	ClassDef(AliGenInfoTask, 1); // Analysis task base class for TPC tracks and clusters
	49	};
	50
	51	#endif