[u/mrichter/AliRoot.git] / PWGPP / 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 TH2;
class TMap;
class AliESDHeader;
class AliExternalTrackParam;
class AliTRDcluster;
class AliTRDseedV1;
class AliTRDgeometry;
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,
    kNclustersLayer     = 19
  };
  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);}

private:
  enum{
    kUseClustersOutsideChamber = 0 
  };
  AliTRDcheckDET(const AliTRDcheckDET &);
  AliTRDcheckDET& operator=(const AliTRDcheckDET &);
  void GetDistanceToTracklet(Double_t *dist, AliTRDseedV1 * const tracklet, AliTRDcluster * const c);
  //----------------------------------------------------
  // Functions creating the reference figures
  TH1* MakePlotChi2() const;
  TH1* MakePlotNTracklets();
  Bool_t MakePlotPulseHeight();
  void MakePlotnTrackletsVsP();
  void MakePlotMeanClustersLayer();
  void MakePlotNclustersTrack();
  void MakePlotNclustersTracklet();
  void MakePlotTrackletCharge();
  Bool_t MakeBarPlot(TH1 *histo, Int_t Color);
  //----------------------------------------------------
  void GetEtaPhiAt(const AliExternalTrackParam *track, Double_t x, Double_t &eta, Double_t &phi);
  TH1 *ProjectCentrality(TH2 *h2d, Int_t centralityBin = -1);

  Int_t fCentralityClass;              // Centrality Class
  TMap *fTriggerNames;                 //! Containing trigger class names
  UChar_t fFlags;                      // Flags for setting

  static const Color_t fkColorsCentrality[AliTRDeventInfo::kCentralityClasses];		// Colors for the different centrality classes in the Ref Figures
  ClassDef(AliTRDcheckDET, 2)
};
#endif
Commit	Line	Data
	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 TH2;
	20	class TMap;
	21	class AliESDHeader;
	22	class AliExternalTrackParam;
	23	class AliTRDcluster;
	24	class AliTRDseedV1;
	25	class AliTRDgeometry;
	26	class AliTRDcheckDET : public AliTRDrecoTask{
	27	public:
	28	// The Histogram number
	29	enum HistType_t {
	30	kNclustersTrack = 0,
	31	kNclustersTracklet = 1,
	32	kNtrackletsTrack = 2,
	33	kNtrackletsSTA = 3,
	34	kNtrackletsBAR = 4,
	35	kNtrackletsCross = 5,
	36	kNtrackletsFindable = 6,
	37	kNtracksEvent = 7,
	38	kNtracksSector = 8,
	39	kPH = 9,
	40	kChi2 = 10,
	41	kChargeCluster = 11,
	42	kChargeTracklet = 12,
	43	kNeventsTrigger = 13,
	44	kNeventsTriggerTracks=14,
	45	kTriggerPurity = 15,
	46	kTrackStatus = 16,
	47	kTrackletStatus = 17,
	48	kNTrackletsP = 18,
	49	kNclustersLayer = 19
	50	};
	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
	71	AliTRDcheckDET();
	72	AliTRDcheckDET(char* name);
	73	virtual ~AliTRDcheckDET();
	74
	75	virtual void UserCreateOutputObjects();
	76	virtual void UserExec(Option_t *opt);
	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);
	96	virtual void MakeSummary();
	97
	98	Bool_t IsUsingClustersOutsideChamber() const {return TESTBIT(fFlags, kUseClustersOutsideChamber);}
	99	void UseClustersOutsideChamber(Bool_t b = kTRUE) {if(b) SETBIT(fFlags, kUseClustersOutsideChamber); else CLRBIT(fFlags, kUseClustersOutsideChamber);}
	100
	101	private:
	102	enum{
	103	kUseClustersOutsideChamber = 0
	104	};
	105	AliTRDcheckDET(const AliTRDcheckDET &);
	106	AliTRDcheckDET& operator=(const AliTRDcheckDET &);
	107	void GetDistanceToTracklet(Double_t dist, AliTRDseedV1 const tracklet, AliTRDcluster * const c);
	108	//----------------------------------------------------
	109	// Functions creating the reference figures
	110	TH1* MakePlotChi2() const;
	111	TH1* MakePlotNTracklets();
	112	Bool_t MakePlotPulseHeight();
	113	void MakePlotnTrackletsVsP();
	114	void MakePlotMeanClustersLayer();
	115	void MakePlotNclustersTrack();
	116	void MakePlotNclustersTracklet();
	117	void MakePlotTrackletCharge();
	118	Bool_t MakeBarPlot(TH1 *histo, Int_t Color);
	119	//----------------------------------------------------
	120	void GetEtaPhiAt(const AliExternalTrackParam *track, Double_t x, Double_t &eta, Double_t &phi);
	121	TH1 ProjectCentrality(TH2 h2d, Int_t centralityBin = -1);
	122
	123	Int_t fCentralityClass; // Centrality Class
	124	TMap *fTriggerNames; //! Containing trigger class names
	125	UChar_t fFlags; // Flags for setting
	126
	127	static const Color_t fkColorsCentrality[AliTRDeventInfo::kCentralityClasses]; // Colors for the different centrality classes in the Ref Figures
	128	ClassDef(AliTRDcheckDET, 2)
	129	};
	130	#endif