[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 TTreeSRedirector;
class AliMCEventHadnler;
class TParticle;
class AliMCInfo;
class AliESDRecInfo;
class AliESDEvent;
class AliMCEvent;
class AliComparisonObject;

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 *);
  //
  //
  void ProcessMCInfo();
  void ProcessESDInfo();
  void ProcessComparison();
  void DumpInfo();
  //
  //
  // 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;}
  //
  Bool_t     AcceptParticle(TParticle *part);  
  AliMCInfo *GetTrack(Int_t index, Bool_t force=kFALSE);
  AliESDRecInfo *GetRecTrack(Int_t index, Bool_t force=kFALSE);
  Bool_t     AddComparisonObject(AliComparisonObject *pObj);
 protected:
  AliGenInfoTask(const AliGenInfoTask& /*info*/);
  AliGenInfoTask& operator=(const AliGenInfoTask& /*info*/) { return *this;}
  AliMCEvent  * fMCinfo;     //! MC event handler
  AliESDEvent * fESD;             //! current esd event
  //
  TObjArray   *fCompList;        // comparison object list
  //
  TClonesArray *fGenTracksArray;  //clones array with filtered particles
  TClonesArray *fGenKinkArray;    //clones array with filtered Kinks
  TClonesArray *fGenV0Array;      //clones array with filtered V0s
  //
  TClonesArray *fRecTracksArray;  //clones array with filtered tracks 
  //
  //
  TTreeSRedirector *fDebugStreamer;     //! debug streamer
  Int_t  fStreamLevel;                  //  debug stream level 
  Int_t  fDebugLevel;                   //  debug level
  ClassDef(AliGenInfoTask, 1); // Analysis task base class for tracks
};

#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;
9c3fd353	17	class TTreeSRedirector;
	18	class AliMCEventHadnler;
	19	class TParticle;
	20	class AliMCInfo;
	21	class AliESDRecInfo;
	22	class AliESDEvent;
	23	class AliMCEvent;
	24	class AliComparisonObject;
cd875161	25
	26	class AliGenInfoTask : public AliAnalysisTask {
	27	public:
	28	AliGenInfoTask();
	29	AliGenInfoTask(const char *name);
9c3fd353	30	virtual ~AliGenInfoTask();
cd875161	31
	32	virtual void ConnectInputData(Option_t *);
	33	virtual void CreateOutputObjects();
	34	virtual void Exec(Option_t *option);
	35	virtual void Terminate(Option_t *);
9c3fd353	36	//
	37	//
	38	void ProcessMCInfo();
	39	void ProcessESDInfo();
	40	void ProcessComparison();
	41	void DumpInfo();
	42	//
	43	//
	44	// debug streamer part
	45	//
	46	TTreeSRedirector *GetDebugStreamer();
	47	void SetStreamLevel(Int_t streamLevel){fStreamLevel=streamLevel;}
	48	void SetDebugLevel(Int_t level) {fDebugLevel = level;}
	49	Int_t GetStreamLevel() const {return fStreamLevel;}
	50	Int_t GetDebugLevel() const {return fDebugLevel;}
	51	//
	52	Bool_t AcceptParticle(TParticle *part);
	53	AliMCInfo *GetTrack(Int_t index, Bool_t force=kFALSE);
	54	AliESDRecInfo *GetRecTrack(Int_t index, Bool_t force=kFALSE);
	55	Bool_t AddComparisonObject(AliComparisonObject *pObj);
cd875161	56	protected:
	57	AliGenInfoTask(const AliGenInfoTask& /info/);
	58	AliGenInfoTask& operator=(const AliGenInfoTask& /info/) { return *this;}
9c3fd353	59	AliMCEvent * fMCinfo; //! MC event handler
	60	AliESDEvent * fESD; //! current esd event
	61	//
	62	TObjArray *fCompList; // comparison object list
	63	//
	64	TClonesArray *fGenTracksArray; //clones array with filtered particles
	65	TClonesArray *fGenKinkArray; //clones array with filtered Kinks
	66	TClonesArray *fGenV0Array; //clones array with filtered V0s
	67	//
	68	TClonesArray *fRecTracksArray; //clones array with filtered tracks
	69	//
	70	//
	71	TTreeSRedirector *fDebugStreamer; //! debug streamer
	72	Int_t fStreamLevel; // debug stream level
	73	Int_t fDebugLevel; // debug level
	74	ClassDef(AliGenInfoTask, 1); // Analysis task base class for tracks
cd875161	75	};
	76
	77	#endif