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

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

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

////////////////////////////////////////////////////////////////////////////
//                                                                        //
//  TRD Resolution performance                                            //
//                                                                        //
//  Authors:                                                              //
//    Alexandru Bercuci <A.Bercuci@gsi.de>                                //
//    Markus Fasel <M.Fasel@gsi.de>                                       //
//                                                                        //
////////////////////////////////////////////////////////////////////////////

#ifndef ALITRDRECOTASK_H
#include "AliTRDrecoTask.h"
#endif

class TH1;
class TH2;
class TF1;
class TGraphErrors;
class TObjArray;
class TDatabasePDG;
class AliTRDReconstructor;
class AliTRDgeometry;
class AliTRDrecoParam;
class AliTRDseedV1;
class AliTRDtrackInfo;
class AliTRDresolution : public AliTRDrecoTask
{
public:
  enum ETRDresolutionPlot {
     kCharge     =  0 // charge resolution
    ,kCluster    =  1 // cluster - track
    ,kTrack      =  2 // tracklet - track residuals/pulls
    ,kTrackIn    =  3 // tracklet - track residuals/pulls at lower TRD entrance 
    ,kTrackOut   =  4 // tracklet - track residuals/pulls at lower TRD entrance during refit
    ,kMCcluster  =  5 // cluster-mc resolution/pulls
    ,kMCtracklet =  6 // tracklet-mc resolution/pulls
    ,kMCtrackIn  =  7 // TPC track monitor
    ,kMCtrackOut =  8 // TOF/HMPID track monitor
    ,kMCtrack    =  9 // TRD track monitor
    ,kNviews     = 10 // total number of resolution views
    ,kNprojs     = 70 // total number of projections for all views
  };
  enum ETRDresolutionSteer {
    kVerbose  = BIT(18)
    ,kVisual  = BIT(19)
    ,kGeom    = BIT(20)
  };
  enum ETRDresolutionOutSlots {
     kClToTrk    = 2
    ,kClToMC     = 3
    ,kTrkltToTrk = 4
    ,kTrkltToMC  = 5
    ,kNOutSlots  = 4
  };

  AliTRDresolution();
  AliTRDresolution(char* name);
  virtual ~AliTRDresolution();
  
  void    UserCreateOutputObjects();
  Float_t GetPtThreshold() const {return fPtThreshold;}
  Float_t GetSegmentationLevel() {return fSegmentLevel;}
  Bool_t  GetRefFigure(Int_t ifig);
  TObjArray*  Histos(); 
  Bool_t  Load(const Char_t *file = "AnalysisResults.root", const Char_t *dir="TRD_Performance");
  void    MakeSummary();

  TObjArray*  Results(Int_t i=0) const {return i ? fGraphS : fGraphM;} 
  void    UserExec(Option_t * opt);
  void    InitExchangeContainers();
  Bool_t  IsInitGeom() const {return TestBit(kGeom);}
  Bool_t  IsVerbose() const {return TestBit(kVerbose);}
  Bool_t  IsVisual() const {return TestBit(kVisual);}
  Bool_t  PostProcess();

  TH1*    PlotCharge(const AliTRDtrackV1 *t=NULL);
  TH1*    PlotCluster(const AliTRDtrackV1 *t=NULL);
  TH1*    PlotTracklet(const AliTRDtrackV1 *t=NULL);
  TH1*    PlotTrackIn(const AliTRDtrackV1 *t=NULL);
  TH1*    PlotTrackOut(const AliTRDtrackV1 *t=NULL);
  TH1*    PlotMC(const AliTRDtrackV1 *t=NULL);

  void    SetSegmentationLevel(Int_t l=0);
  void    SetPtThreshold(Float_t pt) {fPtThreshold = pt;}
  void    SetRecoParam(AliTRDrecoParam *r);
  void    SetInitGeom(Bool_t set = kTRUE) {SetBit(kGeom, set);}
  void    SetVerbose(Bool_t v = kTRUE) {SetBit(kVerbose, v);}
  void    SetVisual(Bool_t v = kTRUE) {SetBit(kVisual, v);}

  void    Terminate(Option_t * opt);
  Bool_t  GetGraph(Float_t *bb, ETRDresolutionPlot ip, Int_t idx=-1, Bool_t kLEG=kTRUE, const Char_t *explain=NULL);
  Bool_t  GetGraphArray(Float_t *bb, ETRDresolutionPlot ip, Int_t idx, Bool_t kLEG=kTRUE, Int_t n=0, Int_t *sel=NULL, const Char_t *explain=NULL);
  
private:
  AliTRDresolution(const AliTRDresolution&);
  AliTRDresolution& operator=(const AliTRDresolution&);

  void    AdjustF1(TH1 *h, TF1 *f);
  TObjArray*  BuildMonitorContainerCluster(const char* name, Bool_t expand=kFALSE);
  TObjArray*  BuildMonitorContainerTracklet(const char* name, Bool_t expand=kFALSE);
  TObjArray*  BuildMonitorContainerTrack(const char* name);
  void    GetLandauMpvFwhm(TF1 * const f, Float_t &mpv, Float_t &xm, Float_t &xM);
  void    GetRange(TH2 *h2, Char_t mod, Float_t *range);
  void    MakeSummaryPlot(TObjArray *a, TH2 *h2);
  Bool_t  Process(TH2* const h2, TF1 *f, Float_t k, TGraphErrors **g);
  Bool_t  Process2D(ETRDresolutionPlot ip, Int_t idx=-1, TF1 *f=NULL,  Float_t scale=1., Int_t gidx=-1);
  Bool_t  Process2Darray(ETRDresolutionPlot ip, Int_t idx=-1, TF1 *f=NULL,  Float_t scale=1.);
  Bool_t  Process3D(ETRDresolutionPlot ip, Int_t idx=-1, TF1 *f=NULL,  Float_t scale=1.);
  Bool_t  Process3Dlinked(ETRDresolutionPlot ip, Int_t idx=-1, TF1 *f=NULL,  Float_t scale=1.);
  Bool_t  Process3DL(ETRDresolutionPlot ip, Int_t idx=-1, TF1 *f=NULL,  Float_t scale=1.);
  Bool_t  Process3Darray(ETRDresolutionPlot ip, Int_t idx=-1, TF1 *f=NULL,  Float_t scale=1.);
  Bool_t  Process3DlinkedArray(ETRDresolutionPlot ip, Int_t idx=-1, TF1 *f=NULL,  Float_t scale=1.);
  Bool_t  Pulls(Double_t dyz[2], Double_t cc[3], Double_t tilt);

  UChar_t             fSegmentLevel;    // segmentation level [sector/stack/chamber]
  UShort_t            fIdxPlot;         // plot counter (internal)
  UShort_t            fIdxFrame;        // frame counter (internal)
  UShort_t            fNcomp[kNprojs];  // number of projections per task
  Char_t              *fAxTitle[kNprojs][4]; //! Title for all ref histos
  Float_t             fPtThreshold;     // pt threshold for some performance plots
  static Char_t const *fgPerformanceName[kNviews]; //! name of performance plot
  static UChar_t const fgNproj[kNviews]; //! number of projections per task
  static Int_t const  fgkNresYsegm[3];  //! number of segments for saving y resolution
  static Char_t const *fgkResYsegmName[3];//! name of segment for saving y resolution
  AliTRDReconstructor *fReconstructor;  //! local reconstructor
  AliTRDgeometry      *fGeo;            //! TRD geometry
  TDatabasePDG        *fDBPDG;          // PDG database
  TObjArray           *fGraphS;         //! result holder - sigma values
  TObjArray           *fGraphM;         //! result holder - mean values

  // calibration containers
  TObjArray           *fCl;     //! cluster2track calib
  TObjArray           *fMCcl;   //! cluster2mc calib
/*  TObjArray           *fTrklt;  //! tracklet2track calib
  TObjArray           *fMCtrklt;//! tracklet2mc calib*/
  
  ClassDef(AliTRDresolution, 7) // TRD tracking resolution task
};
#endif
Commit	Line	Data
1ee39b3a	1	#ifndef ALITRDRESOLUTION_H
	2	#define ALITRDRESOLUTION_H
	3	/* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
	4	* See cxx source for full Copyright notice */
	5
	6	/* $Id: AliTRDresolution.h 27496 2008-07-22 08:35:45Z cblume $ */
	7
	8	////////////////////////////////////////////////////////////////////////////
	9	// //
	10	// TRD Resolution performance //
	11	// //
	12	// Authors: //
	13	// Alexandru Bercuci <A.Bercuci@gsi.de> //
	14	// Markus Fasel <M.Fasel@gsi.de> //
	15	// //
	16	////////////////////////////////////////////////////////////////////////////
	17
	18	#ifndef ALITRDRECOTASK_H
	19	#include "AliTRDrecoTask.h"
	20	#endif
	21
	22	class TH1;
	23	class TH2;
	24	class TF1;
	25	class TGraphErrors;
	26	class TObjArray;
92c40e64	27	class TDatabasePDG;
1ee39b3a	28	class AliTRDReconstructor;
	29	class AliTRDgeometry;
	30	class AliTRDrecoParam;
	31	class AliTRDseedV1;
	32	class AliTRDtrackInfo;
	33	class AliTRDresolution : public AliTRDrecoTask
	34	{
	35	public:
	36	enum ETRDresolutionPlot {
	37	kCharge = 0 // charge resolution
	38	,kCluster = 1 // cluster - track
92d6d80c	39	,kTrack = 2 // tracklet - track residuals/pulls
92d6d80c	40	,kTrackIn = 3 // tracklet - track residuals/pulls at lower TRD entrance
a310e49b	41	,kTrackOut = 4 // tracklet - track residuals/pulls at lower TRD entrance during refit
	42	,kMCcluster = 5 // cluster-mc resolution/pulls
	43	,kMCtracklet = 6 // tracklet-mc resolution/pulls
	44	,kMCtrackIn = 7 // TPC track monitor
	45	,kMCtrackOut = 8 // TOF/HMPID track monitor
	46	,kMCtrack = 9 // TRD track monitor
	47	,kNviews = 10 // total number of resolution views
	48	,kNprojs = 70 // total number of projections for all views
1ee39b3a	49	};
1ee39b3a	50	enum ETRDresolutionSteer {
801d4d50	51	kVerbose = BIT(18)
	52	,kVisual = BIT(19)
	53	,kGeom = BIT(20)
1ee39b3a	54	};
705f8b0a	55	enum ETRDresolutionOutSlots {
705f8b0a	56	kClToTrk = 2
83b44483	57	,kClToMC = 3
83b44483	58	,kTrkltToTrk = 4
705f8b0a	59	,kTrkltToMC = 5
	60	,kNOutSlots = 4
	61	};
1ee39b3a	62
1ee39b3a	63	AliTRDresolution();
f8f46e4d	64	AliTRDresolution(char* name);
1ee39b3a	65	virtual ~AliTRDresolution();
1ee39b3a	66
f8f46e4d	67	void UserCreateOutputObjects();
3ba3e0a4	68	Float_t GetPtThreshold() const {return fPtThreshold;}
2589cf64	69	Float_t GetSegmentationLevel() {return fSegmentLevel;}
1ee39b3a	70	Bool_t GetRefFigure(Int_t ifig);
1ee39b3a	71	TObjArray* Histos();
fe1d1beb	72	Bool_t Load(const Char_t file = "AnalysisResults.root", const Char_t dir="TRD_Performance");
00a3f7a4	73	void MakeSummary();
2589cf64	74
1ee39b3a	75	TObjArray* Results(Int_t i=0) const {return i ? fGraphS : fGraphM;}
f8f46e4d	76	void UserExec(Option_t * opt);
83b44483	77	void InitExchangeContainers();
801d4d50	78	Bool_t IsInitGeom() const {return TestBit(kGeom);}
	79	Bool_t IsVerbose() const {return TestBit(kVerbose);}
	80	Bool_t IsVisual() const {return TestBit(kVisual);}
1ee39b3a	81	Bool_t PostProcess();
1ee39b3a	82
a310e49b	83	TH1* PlotCharge(const AliTRDtrackV1 *t=NULL);
	84	TH1* PlotCluster(const AliTRDtrackV1 *t=NULL);
	85	TH1* PlotTracklet(const AliTRDtrackV1 *t=NULL);
	86	TH1* PlotTrackIn(const AliTRDtrackV1 *t=NULL);
	87	TH1* PlotTrackOut(const AliTRDtrackV1 *t=NULL);
	88	TH1* PlotMC(const AliTRDtrackV1 *t=NULL);
1ee39b3a	89
2589cf64	90	void SetSegmentationLevel(Int_t l=0);
3ba3e0a4	91	void SetPtThreshold(Float_t pt) {fPtThreshold = pt;}
1ee39b3a	92	void SetRecoParam(AliTRDrecoParam *r);
801d4d50	93	void SetInitGeom(Bool_t set = kTRUE) {SetBit(kGeom, set);}
	94	void SetVerbose(Bool_t v = kTRUE) {SetBit(kVerbose, v);}
	95	void SetVisual(Bool_t v = kTRUE) {SetBit(kVisual, v);}
1ee39b3a	96
1ee39b3a	97	void Terminate(Option_t * opt);
a310e49b	98	Bool_t GetGraph(Float_t bb, ETRDresolutionPlot ip, Int_t idx=-1, Bool_t kLEG=kTRUE, const Char_t explain=NULL);
a310e49b	99	Bool_t GetGraphArray(Float_t bb, ETRDresolutionPlot ip, Int_t idx, Bool_t kLEG=kTRUE, Int_t n=0, Int_t sel=NULL, const Char_t *explain=NULL);
1ee39b3a	100
	101	private:
	102	AliTRDresolution(const AliTRDresolution&);
	103	AliTRDresolution& operator=(const AliTRDresolution&);
afca20ef	104
1ee39b3a	105	void AdjustF1(TH1 h, TF1 f);
2589cf64	106	TObjArray* BuildMonitorContainerCluster(const char* name, Bool_t expand=kFALSE);
2589cf64	107	TObjArray* BuildMonitorContainerTracklet(const char* name, Bool_t expand=kFALSE);
3d2a3dff	108	TObjArray* BuildMonitorContainerTrack(const char* name);
1ee39b3a	109	void GetLandauMpvFwhm(TF1 * const f, Float_t &mpv, Float_t &xm, Float_t &xM);
068e2c00	110	void GetRange(TH2 h2, Char_t mod, Float_t range);
068e2c00	111	void MakeSummaryPlot(TObjArray a, TH2 h2);
1ee39b3a	112	Bool_t Process(TH2* const h2, TF1 f, Float_t k, TGraphErrors *g);
a310e49b	113	Bool_t Process2D(ETRDresolutionPlot ip, Int_t idx=-1, TF1 *f=NULL, Float_t scale=1., Int_t gidx=-1);
	114	Bool_t Process2Darray(ETRDresolutionPlot ip, Int_t idx=-1, TF1 *f=NULL, Float_t scale=1.);
	115	Bool_t Process3D(ETRDresolutionPlot ip, Int_t idx=-1, TF1 *f=NULL, Float_t scale=1.);
81979445	116	Bool_t Process3Dlinked(ETRDresolutionPlot ip, Int_t idx=-1, TF1 *f=NULL, Float_t scale=1.);
a310e49b	117	Bool_t Process3DL(ETRDresolutionPlot ip, Int_t idx=-1, TF1 *f=NULL, Float_t scale=1.);
a310e49b	118	Bool_t Process3Darray(ETRDresolutionPlot ip, Int_t idx=-1, TF1 *f=NULL, Float_t scale=1.);
81979445	119	Bool_t Process3DlinkedArray(ETRDresolutionPlot ip, Int_t idx=-1, TF1 *f=NULL, Float_t scale=1.);
553528eb	120	Bool_t Pulls(Double_t dyz[2], Double_t cc[3], Double_t tilt);
1ee39b3a	121
801d4d50	122	UChar_t fSegmentLevel; // segmentation level [sector/stack/chamber]
5935a6da	123	UShort_t fIdxPlot; // plot counter (internal)
	124	UShort_t fIdxFrame; // frame counter (internal)
	125	UShort_t fNcomp[kNprojs]; // number of projections per task
9653acd3	126	Char_t *fAxTitle[kNprojs][4]; //! Title for all ref histos
5935a6da	127	Float_t fPtThreshold; // pt threshold for some performance plots
2589cf64	128	static Char_t const *fgPerformanceName[kNviews]; //! name of performance plot
2589cf64	129	static UChar_t const fgNproj[kNviews]; //! number of projections per task
3ba3e0a4	130	static Int_t const fgkNresYsegm[3]; //! number of segments for saving y resolution
3ba3e0a4	131	static Char_t const *fgkResYsegmName[3];//! name of segment for saving y resolution
fe1d1beb	132	AliTRDReconstructor *fReconstructor; //! local reconstructor
fe1d1beb	133	AliTRDgeometry *fGeo; //! TRD geometry
5935a6da	134	TDatabasePDG *fDBPDG; // PDG database
1ee39b3a	135	TObjArray *fGraphS; //! result holder - sigma values
	136	TObjArray *fGraphM; //! result holder - mean values
	137
	138	// calibration containers
	139	TObjArray *fCl; //! cluster2track calib
1ee39b3a	140	TObjArray *fMCcl; //! cluster2mc calib
83b44483	141	/* TObjArray *fTrklt; //! tracklet2track calib
83b44483	142	TObjArray fMCtrklt;//! tracklet2mc calib/
1ee39b3a	143
83b44483	144	ClassDef(AliTRDresolution, 7) // TRD tracking resolution task
1ee39b3a	145	};
1ee39b3a	146	#endif