[u/mrichter/AliRoot.git] / PWG1 / TRD / AliTRDcheckESD.h

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

/* $Id: AliTRDcheckESD.h 27496 2008-07-22 08:35:45Z cblume $ */

/////////////////////////////////////////////////////
//
// Check basic detector results at ESD level
//
// Author
//   Alex Bercuci <A.Bercuci@gsi.de>
//
//////////////////////////////////////////////////////

#ifndef ALIANALYSISTASK_H
#include "AliAnalysisTask.h"
#endif

class AliESDEvent;
class AliMCEvent;
class TH1;
class TObjArray;
class TGraph;
class TGraphErrors;
class AliTRDcheckESD : public AliAnalysisTask {
public:
  enum ETRDcheckESDstatus {
     kMC   = BIT(0)  // use MC info
    ,kLoad = BIT(1)  // container from file
  };
  enum ETRDcheckESDhistos {
    kNCl  = 0    // number of clusters per track
   ,kTRDstat     // TRD tracks status
   ,kTRDmom      // TRD track momentum
   ,kPtRes       // Pt resolution @ vertex for TRD
   ,kNhistos = 4 // number of histograms
   ,kNgraphs = 6 // number of graphs
  };
  enum ETRDcheckESDbits {
    kTPCout = 1 // track left TPC
   ,kTRDin      // track reach TRD fiducial volume
   ,kTRDout     // track reconstructed in TRD
   ,kTRDpid     // PID calculated in TRD
   ,kTRDref     // track refitted in TRD
  };
  AliTRDcheckESD();
  virtual ~AliTRDcheckESD();
  
  void          ConnectInputData(Option_t *);
  void          CreateOutputObjects();
  TGraph*       GetGraph(Int_t id, Option_t *opt="bc");
  void          Exec(Option_t *);

  Bool_t        HasMC() const { return TESTBIT(fStatus, kMC);}
  Bool_t        IsLoad() const { return TESTBIT(fStatus, kLoad);}
  TObjArray*    Histos();
  Bool_t        Load(const Char_t *fn, const Char_t *name=0x0);
  void          SetMC(Bool_t mc = kTRUE) { mc ? SETBIT(fStatus, kMC) : CLRBIT(fStatus, kMC);}
  Bool_t        PutTrendValue(const Char_t *name, Double_t val);
  void          Terminate(Option_t *);

private:
  static const Float_t fgkxTPC; // end radial position of TPC
  static const Float_t fgkxTOF; // start radial position of TOF

  AliTRDcheckESD(const AliTRDcheckESD&);
  AliTRDcheckESD& operator=(const AliTRDcheckESD&);
  Int_t         Pdg2Idx(Int_t pdg);
  void          Process(TH1 **h, TGraphErrors *g);
  void          PrintStatus(ULong_t s);

  Int_t            fStatus;            // bit mask for controlling the task
  AliESDEvent      *fESD;              //! ESD event
  AliMCEvent       *fMC;               //! MC event
  TObjArray        *fHistos;           //! QA histos
  TObjArray        *fResults;          // QA graphs
  static FILE *fgFile;                 //! trend file streamer
  ClassDef(AliTRDcheckESD, 3)          // user oriented TRD analysis based on ESD-MC data
};
#endif
Commit	Line	Data
1ee39b3a	1	#ifndef ALITRDCHECKESD_H
	2	#define ALITRDCHECKESD_H
	3	/* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
	4	* See cxx source for full Copyright notice */
	5
	6	/* $Id: AliTRDcheckESD.h 27496 2008-07-22 08:35:45Z cblume $ */
	7
	8	/////////////////////////////////////////////////////
	9	//
	10	// Check basic detector results at ESD level
	11	//
	12	// Author
	13	// Alex Bercuci <A.Bercuci@gsi.de>
	14	//
	15	//////////////////////////////////////////////////////
	16
	17	#ifndef ALIANALYSISTASK_H
	18	#include "AliAnalysisTask.h"
	19	#endif
	20
	21	class AliESDEvent;
	22	class AliMCEvent;
	23	class TH1;
	24	class TObjArray;
	25	class TGraph;
	26	class TGraphErrors;
	27	class AliTRDcheckESD : public AliAnalysisTask {
	28	public:
	29	enum ETRDcheckESDstatus {
a96ac33a	30	kMC = BIT(0) // use MC info
a96ac33a	31	,kLoad = BIT(1) // container from file
1ee39b3a	32	};
	33	enum ETRDcheckESDhistos {
	34	kNCl = 0 // number of clusters per track
	35	,kTRDstat // TRD tracks status
	36	,kTRDmom // TRD track momentum
fb09668f	37	,kPtRes // Pt resolution @ vertex for TRD
fb09668f	38	,kNhistos = 4 // number of histograms
1ee39b3a	39	,kNgraphs = 6 // number of graphs
	40	};
	41	enum ETRDcheckESDbits {
	42	kTPCout = 1 // track left TPC
	43	,kTRDin // track reach TRD fiducial volume
	44	,kTRDout // track reconstructed in TRD
	45	,kTRDpid // PID calculated in TRD
	46	,kTRDref // track refitted in TRD
1ee39b3a	47	};
	48	AliTRDcheckESD();
	49	virtual ~AliTRDcheckESD();
	50
	51	void ConnectInputData(Option_t *);
	52	void CreateOutputObjects();
	53	TGraph* GetGraph(Int_t id, Option_t *opt="bc");
	54	void Exec(Option_t *);
	55
	56	Bool_t HasMC() const { return TESTBIT(fStatus, kMC);}
a96ac33a	57	Bool_t IsLoad() const { return TESTBIT(fStatus, kLoad);}
1ee39b3a	58	TObjArray* Histos();
	59	Bool_t Load(const Char_t fn, const Char_t name=0x0);
	60	void SetMC(Bool_t mc = kTRUE) { mc ? SETBIT(fStatus, kMC) : CLRBIT(fStatus, kMC);}
	61	Bool_t PutTrendValue(const Char_t *name, Double_t val);
	62	void Terminate(Option_t *);
	63
	64	private:
	65	static const Float_t fgkxTPC; // end radial position of TPC
	66	static const Float_t fgkxTOF; // start radial position of TOF
	67
	68	AliTRDcheckESD(const AliTRDcheckESD&);
	69	AliTRDcheckESD& operator=(const AliTRDcheckESD&);
fb09668f	70	Int_t Pdg2Idx(Int_t pdg);
1ee39b3a	71	void Process(TH1 *h, TGraphErrors g);
	72	void PrintStatus(ULong_t s);
	73
	74	Int_t fStatus; // bit mask for controlling the task
a96ac33a	75	AliESDEvent *fESD; //! ESD event
	76	AliMCEvent *fMC; //! MC event
	77	TObjArray *fHistos; //! QA histos
1ee39b3a	78	TObjArray *fResults; // QA graphs
	79	static FILE *fgFile; //! trend file streamer
	80	ClassDef(AliTRDcheckESD, 3) // user oriented TRD analysis based on ESD-MC data
	81	};
	82	#endif