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

#ifndef ALITRDCHECKDET_H
#define ALITRDCHECKDET_H

#ifndef ALITRDRECOTASK_H
#include "AliTRDrecoTask.h"
#endif
////////////////////////////////////////////////////////////////////////////
//  Basic checks for tracking and detector performance                    //
//                                                                        //
//  Authors:                                                              //
//    Anton Andronic <A.Andronic@gsi.de>                                  //
//    Alexandru Bercuci <A.Bercuci@gsi.de>                                //
//    Markus Fasel <M.Fasel@gsi.de>                                       //
////////////////////////////////////////////////////////////////////////////


class TObjArray;
class TH1;
class TMap;
class AliESDHeader;
class AliTRDcluster;
class AliTRDseedV1;
class AliTRDgeometry;
class AliTRDReconstructor;
class AliTRDrecoParam;
class AliTRDeventInfo;
class AliTRDcheckDET : public AliTRDrecoTask{
public:
  // The Histogram number
  enum  HistType_t {
    kNclustersTrack     = 0,
    kNclustersTracklet  = 1,
    kNtrackletsTrack    = 2,
    kNtrackletsSTA      = 3,
    kNtrackletsBAR      = 4,
    kNtrackletsCross    = 5,
    kNtrackletsFindable = 6,
    kNtracksEvent       = 7,
    kNtracksSector      = 8,
    kPH                 = 9,
    kChi2               = 10,
    kChargeCluster      = 11,
    kChargeTracklet     = 12,
    kNeventsTrigger     = 13,
    kNeventsTriggerTracks=14,
    kTriggerPurity      = 15,
    kTrackStatus        = 16,
    kTrackletStatus     = 17,
    kNTrackletsP        = 18
  };
  enum FigureType_t{
    kFigNclustersTrack,
    kFigNclustersTracklet,
    kFigNtrackletsTrack,
    kFigNTrackletsP,
    kFigNtrackletsCross,
    kFigNtrackletsFindable,
    kFigNtracksEvent,
    kFigNtracksSector,
    kFigTrackStatus,
    kFigTrackletStatus,
    kFigChi2,
    kFigPH,
    kFigChargeCluster,
    kFigChargeTracklet,
    kFigNeventsTrigger,
    kFigNeventsTriggerTracks,
    kFigTriggerPurity
  };
 
  AliTRDcheckDET();
  AliTRDcheckDET(char* name);
  virtual ~AliTRDcheckDET();

  virtual void UserCreateOutputObjects();
  virtual void UserExec(Option_t *opt);
  virtual TObjArray *Histos();

  // Plotting Functions:
  TH1 *PlotTrackStatus(const AliTRDtrackV1 *track = 0x0);
  TH1 *PlotTrackletStatus(const AliTRDtrackV1 *track = 0x0);
  TH1 *PlotNClustersTracklet(const AliTRDtrackV1 *t = 0x0);
  TH1 *PlotNClustersTrack(const AliTRDtrackV1 *t = 0x0);
  TH1 *PlotNTrackletsTrack(const AliTRDtrackV1 *t = 0x0);
  TH1 *PlotNTrackletsRowCross(const AliTRDtrackV1 *t = 0x0);
  TH1 *PlotFindableTracklets(const AliTRDtrackV1 *track = 0x0);
  TH1 *PlotNTracksSector(const AliTRDtrackV1 *t = 0x0);
  TH1 *PlotPHt(const AliTRDtrackV1 *t = 0x0);
  TH1 *PlotPHx(const AliTRDtrackV1 *track = 0x0);
  TH1 *PlotChi2(const AliTRDtrackV1 *t = 0x0);
  TH1 *PlotChargeCluster(const AliTRDtrackV1 *t = 0x0);
  TH1 *PlotChargeTracklet(const AliTRDtrackV1 *t = 0x0);

  virtual Bool_t PostProcess();
  virtual Bool_t GetRefFigure(Int_t ifig);
  virtual void MakeSummary();
  
  Bool_t IsUsingClustersOutsideChamber() const {return TESTBIT(fFlags, kUseClustersOutsideChamber);}
  void UseClustersOutsideChamber(Bool_t b = kTRUE) {if(b) SETBIT(fFlags, kUseClustersOutsideChamber); else CLRBIT(fFlags, kUseClustersOutsideChamber);}
  void SetRecoParam(AliTRDrecoParam *r);

private:
  enum{
    kUseClustersOutsideChamber = 0 
  };
  AliTRDcheckDET(const AliTRDcheckDET &);
  AliTRDcheckDET& operator=(const AliTRDcheckDET &);
  void GetDistanceToTracklet(Double_t *dist, AliTRDseedV1 * const tracklet, AliTRDcluster * const c);
  TH1* MakePlotChi2();
  TH1* MakePlotNTracklets();
  Bool_t MakePlotPulseHeight();
  void MakePlotnTrackletsVsP();
  Bool_t MakeBarPlot(TH1 *histo, Int_t Color);

  AliTRDeventInfo *fEventInfo;         //! ESD Header
  TMap *fTriggerNames;                 //! Containing trigger class names
  AliTRDReconstructor *fReconstructor; // TRD Reconstructor
  AliTRDgeometry *fGeo;                // TRD Geometry object
  UChar_t fFlags;                      // Flags for setting
    
  ClassDef(AliTRDcheckDET, 1)
};
#endif
Commit	Line	Data
1ee39b3a	1	#ifndef ALITRDCHECKDET_H
	2	#define ALITRDCHECKDET_H
	3
	4	#ifndef ALITRDRECOTASK_H
	5	#include "AliTRDrecoTask.h"
	6	#endif
	7	////////////////////////////////////////////////////////////////////////////
	8	// Basic checks for tracking and detector performance //
	9	// //
	10	// Authors: //
	11	// Anton Andronic <A.Andronic@gsi.de> //
	12	// Alexandru Bercuci <A.Bercuci@gsi.de> //
	13	// Markus Fasel <M.Fasel@gsi.de> //
	14	////////////////////////////////////////////////////////////////////////////
	15
	16
	17	class TObjArray;
	18	class TH1;
	19	class TMap;
	20	class AliESDHeader;
	21	class AliTRDcluster;
	22	class AliTRDseedV1;
	23	class AliTRDgeometry;
	24	class AliTRDReconstructor;
	25	class AliTRDrecoParam;
	26	class AliTRDeventInfo;
	27	class AliTRDcheckDET : public AliTRDrecoTask{
	28	public:
	29	// The Histogram number
	30	enum HistType_t {
	31	kNclustersTrack = 0,
	32	kNclustersTracklet = 1,
	33	kNtrackletsTrack = 2,
	34	kNtrackletsSTA = 3,
	35	kNtrackletsBAR = 4,
	36	kNtrackletsCross = 5,
	37	kNtrackletsFindable = 6,
	38	kNtracksEvent = 7,
	39	kNtracksSector = 8,
	40	kPH = 9,
	41	kChi2 = 10,
	42	kChargeCluster = 11,
	43	kChargeTracklet = 12,
	44	kNeventsTrigger = 13,
	45	kNeventsTriggerTracks=14,
	46	kTriggerPurity = 15,
	47	kTrackStatus = 16,
0c76cfa4	48	kTrackletStatus = 17,
0c76cfa4	49	kNTrackletsP = 18
1ee39b3a	50	};
b795b480	51	enum FigureType_t{
	52	kFigNclustersTrack,
	53	kFigNclustersTracklet,
	54	kFigNtrackletsTrack,
	55	kFigNTrackletsP,
	56	kFigNtrackletsCross,
	57	kFigNtrackletsFindable,
	58	kFigNtracksEvent,
	59	kFigNtracksSector,
	60	kFigTrackStatus,
	61	kFigTrackletStatus,
	62	kFigChi2,
	63	kFigPH,
	64	kFigChargeCluster,
	65	kFigChargeTracklet,
	66	kFigNeventsTrigger,
	67	kFigNeventsTriggerTracks,
	68	kFigTriggerPurity
	69	};
	70
1ee39b3a	71	AliTRDcheckDET();
f8f46e4d	72	AliTRDcheckDET(char* name);
1ee39b3a	73	virtual ~AliTRDcheckDET();
1ee39b3a	74
f8f46e4d	75	virtual void UserCreateOutputObjects();
f8f46e4d	76	virtual void UserExec(Option_t *opt);
1ee39b3a	77	virtual TObjArray *Histos();
	78
	79	// Plotting Functions:
	80	TH1 PlotTrackStatus(const AliTRDtrackV1 track = 0x0);
	81	TH1 PlotTrackletStatus(const AliTRDtrackV1 track = 0x0);
	82	TH1 PlotNClustersTracklet(const AliTRDtrackV1 t = 0x0);
	83	TH1 PlotNClustersTrack(const AliTRDtrackV1 t = 0x0);
	84	TH1 PlotNTrackletsTrack(const AliTRDtrackV1 t = 0x0);
	85	TH1 PlotNTrackletsRowCross(const AliTRDtrackV1 t = 0x0);
	86	TH1 PlotFindableTracklets(const AliTRDtrackV1 track = 0x0);
	87	TH1 PlotNTracksSector(const AliTRDtrackV1 t = 0x0);
	88	TH1 PlotPHt(const AliTRDtrackV1 t = 0x0);
	89	TH1 PlotPHx(const AliTRDtrackV1 track = 0x0);
	90	TH1 PlotChi2(const AliTRDtrackV1 t = 0x0);
	91	TH1 PlotChargeCluster(const AliTRDtrackV1 t = 0x0);
	92	TH1 PlotChargeTracklet(const AliTRDtrackV1 t = 0x0);
	93
	94	virtual Bool_t PostProcess();
	95	virtual Bool_t GetRefFigure(Int_t ifig);
b795b480	96	virtual void MakeSummary();
1ee39b3a	97
c732f879	98	Bool_t IsUsingClustersOutsideChamber() const {return TESTBIT(fFlags, kUseClustersOutsideChamber);}
c732f879	99	void UseClustersOutsideChamber(Bool_t b = kTRUE) {if(b) SETBIT(fFlags, kUseClustersOutsideChamber); else CLRBIT(fFlags, kUseClustersOutsideChamber);}
1ee39b3a	100	void SetRecoParam(AliTRDrecoParam *r);
	101
	102	private:
c732f879	103	enum{
c0a6ac06	104	kUseClustersOutsideChamber = 0
c732f879	105	};
1ee39b3a	106	AliTRDcheckDET(const AliTRDcheckDET &);
	107	AliTRDcheckDET& operator=(const AliTRDcheckDET &);
	108	void GetDistanceToTracklet(Double_t dist, AliTRDseedV1 const tracklet, AliTRDcluster * const c);
	109	TH1* MakePlotChi2();
	110	TH1* MakePlotNTracklets();
c0a6ac06	111	Bool_t MakePlotPulseHeight();
0c76cfa4	112	void MakePlotnTrackletsVsP();
1ee39b3a	113	Bool_t MakeBarPlot(TH1 *histo, Int_t Color);
	114
	115	AliTRDeventInfo *fEventInfo; //! ESD Header
	116	TMap *fTriggerNames; //! Containing trigger class names
	117	AliTRDReconstructor *fReconstructor; // TRD Reconstructor
	118	AliTRDgeometry *fGeo; // TRD Geometry object
c732f879	119	UChar_t fFlags; // Flags for setting
1ee39b3a	120
	121	ClassDef(AliTRDcheckDET, 1)
	122	};
	123	#endif