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

#ifndef ALICOMPARISONRESTASK_H
#define ALICOMPARISONRESTASK_H

//------------------------------------------------------------------------------
// Class to compare properties of reconstructed and MC particle tracks. 
// 
// Author: J.Otwinowski 04/02/2008 
//------------------------------------------------------------------------------

class AliComparisonRes;
class AliComparisonEff;
class AliComparisonDEdx;
class AliComparisonDCA;
class AliMagFMaps;
class TList;

#include "AliAnalysisTask.h"

class AliComparisonTask : public AliAnalysisTask {
 public:
  AliComparisonTask(const char *name = "AliComparisonTask");
  virtual ~AliComparisonTask();
  
  virtual void   ConnectInputData(Option_t *);
  virtual void   CreateOutputObjects();
  virtual void   Exec(Option_t *option);
  virtual void   Terminate(Option_t *);

  // Read TTree entry (event by event)
  Bool_t  ReadEntry(Int_t evt);

  // Set comparison objects
  void SetAliComparisonRes(AliComparisonRes* comp) {fCompRes = comp;}
  void SetAliComparisonEff(AliComparisonEff* comp) {fCompEff = comp;}
  void SetAliComparisonDEdx(AliComparisonDEdx* comp) {fCompDEdx = comp;}
  void SetAliComparisonDCA(AliComparisonDCA* comp) {fCompDCA = comp;}

  void SetMagField(Int_t mag = 2) {fMagField = mag;}
  void SetGeometry(char* geom = "/d/alice12/jacek/sim/v4-10-Release/pp/0/geometry.root")  {fGeom = geom;}

 private:
  TTree* fTree;                   //! input tree
  AliMCInfo *fInfoMC;             //! AliMCInfo object
  AliESDRecInfo *fInfoRC;         //! AliESDRecInfo object
  AliComparisonRes* fCompRes;     // TPC resolution comparison object
  AliComparisonEff* fCompEff;     // TPC efficiency comparison object
  AliComparisonDEdx* fCompDEdx;   // TPC DEdx comparison object
  AliComparisonDCA* fCompDCA;     // TPC DCA comparison object

  TList* fOutput;                 //! list send on output slot 0
  static Int_t evtNumber;         //! event number
  Int_t  fMagField;               //! mag. field (0 - 0.2 T, 1 - 0.4 T, 2 - 0.5 T) 
  AliMagFMaps *fMagFMap;          //! mag. field map 
  const char *fGeom;              //! ROOT file with detector geometry

  AliComparisonTask(const AliComparisonTask&); // not implemented
  AliComparisonTask& operator=(const AliComparisonTask&); // not implemented
  
  ClassDef(AliComparisonTask, 1); // example of analysis
};

#endif
Commit	Line	Data
09b20ad1	1	#ifndef ALICOMPARISONRESTASK_H
	2	#define ALICOMPARISONRESTASK_H
	3
	4	//------------------------------------------------------------------------------
	5	// Class to compare properties of reconstructed and MC particle tracks.
	6	//
	7	// Author: J.Otwinowski 04/02/2008
	8	//------------------------------------------------------------------------------
	9
	10	class AliComparisonRes;
	11	class AliComparisonEff;
	12	class AliComparisonDEdx;
	13	class AliComparisonDCA;
	14	class AliMagFMaps;
	15	class TList;
	16
	17	#include "AliAnalysisTask.h"
	18
	19	class AliComparisonTask : public AliAnalysisTask {
	20	public:
	21	AliComparisonTask(const char *name = "AliComparisonTask");
	22	virtual ~AliComparisonTask();
	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
	29	// Read TTree entry (event by event)
	30	Bool_t ReadEntry(Int_t evt);
	31
	32	// Set comparison objects
	33	void SetAliComparisonRes(AliComparisonRes* comp) {fCompRes = comp;}
	34	void SetAliComparisonEff(AliComparisonEff* comp) {fCompEff = comp;}
	35	void SetAliComparisonDEdx(AliComparisonDEdx* comp) {fCompDEdx = comp;}
	36	void SetAliComparisonDCA(AliComparisonDCA* comp) {fCompDCA = comp;}
	37
	38	void SetMagField(Int_t mag = 2) {fMagField = mag;}
	39	void SetGeometry(char* geom = "/d/alice12/jacek/sim/v4-10-Release/pp/0/geometry.root") {fGeom = geom;}
	40
	41	private:
	42	TTree* fTree; //! input tree
	43	AliMCInfo *fInfoMC; //! AliMCInfo object
	44	AliESDRecInfo *fInfoRC; //! AliESDRecInfo object
	45	AliComparisonRes* fCompRes; // TPC resolution comparison object
	46	AliComparisonEff* fCompEff; // TPC efficiency comparison object
	47	AliComparisonDEdx* fCompDEdx; // TPC DEdx comparison object
	48	AliComparisonDCA* fCompDCA; // TPC DCA comparison object
	49
	50	TList* fOutput; //! list send on output slot 0
	51	static Int_t evtNumber; //! event number
	52	Int_t fMagField; //! mag. field (0 - 0.2 T, 1 - 0.4 T, 2 - 0.5 T)
	53	AliMagFMaps *fMagFMap; //! mag. field map
	54	const char *fGeom; //! ROOT file with detector geometry
	55
	56	AliComparisonTask(const AliComparisonTask&); // not implemented
	57	AliComparisonTask& operator=(const AliComparisonTask&); // not implemented
	58
	59	ClassDef(AliComparisonTask, 1); // example of analysis
	60	};
	61
	62	#endif