[u/mrichter/AliRoot.git] / PWGPP / 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

#ifndef Root_TNamed
#include "TNamed.h"
#endif

class TH1;
class TH2;
class TF1;
class TGraphErrors;
class TObjArray;
class TDatabasePDG;
class AliTRDrecoParam;
class AliTRDseedV1;
class AliTRDtrackInfo;
class AliTrackPoint;
class AliTRDresolution : public AliTRDrecoTask
{
public:
  enum ETRDresolutionSteer {
     kVerbose    = BIT(18) // switch verbosity
    ,kVisual     = BIT(19) // show partial results during processing
    ,kTrackRefit = BIT(20) // steer track refit
    ,kTrackSelect= BIT(21) // steer track selection
    ,kXchange    = BIT(22) // use exchange containers
  };
  enum ETRDresolutionSlots {
     kClToTrk    = 2
    ,kClToMC
    ,kTrkltToTrk
    ,kTrkltToMC
    ,kNSlots     = 4
  };
  enum ETRDresolutionClass {
     kDetector=0      // cluster - detector
    ,kCluster         // cluster - track
    ,kTracklet        // tracklet - track residuals/pulls
    ,kTrackIn         // tracklet - track residuals/pulls at lower TRD entrance
    ,kMCcluster       // cluster-mc resolution/pulls
    ,kMCtracklet      // tracklet-mc resolution/pulls
    ,kMCtrackIn       // TPC track monitor
    ,kMCtrack         // TRD track monitor
/*    ,kTrackOut        // tracklet - track residuals/pulls at lower TRD entrance during refit
    ,kMCtrackOut      // TOF/HMPID track monitor*/
    ,kNclasses        // total number of resolution classes
    ,kV0TrackIn       // tracklet - track residuals/pulls at lower TRD entrance for V0 tagged tracks
  };
  enum ETRDresolutionClassProjs {
    kDetNproj=460      // detector projections
    ,kClNproj=2800     // cluster projections
    ,kTrkltNproj=8000  // tracklet projections
    ,kTrkInNproj=1440  // trackIn projections
    ,kTrkNproj=2500    // track projections
    ,kMCTrkInNproj=1260// trackIn projections
  };
  enum ETRDresolutionProjs {
    kBC    = 0 // bunch cross
    ,kPhi
    ,kEta
    ,kYrez
    ,kPrez
    ,kZrez
    ,kSpeciesChgRC
    ,kPt
    ,kNdim  // no of dimensions in the THnSparse
    ,kNdimDet     = 4
    ,kNdimCl      = 4
    ,kNdimTrklt   = 4
    ,kNdimTrkIn   = 7
    ,kNbunchCross = 3  // no of classes for bunch crossing
    ,kNspc        = 3  // no of species e, mu+pi, K+p
    ,kNcharge     = 2  // no of charges
    ,kNpads       = 4  // no of charges
  };

  AliTRDresolution();
  AliTRDresolution(char* name, Bool_t xchange=kTRUE);
  AliTRDresolution(const AliTRDresolution&);
  AliTRDresolution& operator=(const AliTRDresolution&);
  virtual ~AliTRDresolution();
  
  static Bool_t   FitTrack(const Int_t np, AliTrackPoint *points, Float_t params[10]);
  static Bool_t   FitTracklet(const Int_t ly, const Int_t np, const AliTrackPoint *points, const Float_t trackPars[10], Float_t trackletPars[3]);
  void            UserCreateOutputObjects();
//  Float_t GetCorrectionX(Int_t det, Int_t tb) const {return fXcorr[det][tb];}
  static void     GetRangeZ(TH2 *h2, Float_t &m, Float_t &M);
  Float_t         GetPtThreshold() const {return fPtThreshold;}
  Bool_t          GetRefFigure(Int_t ifig);

  virtual TObjArray*  Histos(); 
//  Bool_t  Load(const Char_t *file = "AnalysisResults.root", const Char_t *dir="TRD_Performance");
//  Bool_t  LoadCorrection(const Char_t *file=NULL);
  void            MakeSummary();

  TObjArray*      Results(ETRDresolutionClass c) const  { if(!fProj) return NULL; return (TObjArray*)fProj->At(c);}
  void            UserExec(Option_t * opt);
  void            InitExchangeContainers();
/*  Bool_t          HasTrackRefit() const                 { return TestBit(kTrackRefit);}
  Bool_t          HasTrackSelection() const             { return TestBit(kTrackSelect);}*/
  Bool_t          IsVerbose() const                     { return TestBit(kVerbose);}
  Bool_t          IsVisual() const                      { return TestBit(kVisual);}
  Bool_t          UseBCselectTOF() const                { return fBCbinTOF>0;}
  Bool_t          UseBCselectFill() const               { return fBCbinFill>0;}
  Bool_t          UseLYselectTrklt() const              { return fLYselect>=0;}
  Bool_t          UseExchangeContainers() const         { return TestBit(kXchange);}
  Bool_t          PostProcess();

  TH1*            DetCluster(const TObjArray *cl=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);

  static Bool_t   Process(TH2* const /*h2*/, TGraphErrors **/*g*/, Int_t stat=100){ return Bool_t(stat);}
  void            SetPtThreshold(Float_t pt)            { fPtThreshold = pt;}
  void            SetBCselectTOF(Int_t b=0)             { fBCbinTOF = b;}
  void            SetBCselectFill(Int_t b=0)            { fBCbinFill = b<0||b>3499?1:b+1;}
  void            SetBsign(Int_t b=0)                   { fBsign = Bool_t(b);}
  void            SetLYselect(Int_t ly=0)               { fLYselect = ly;}
  void            SetProcesses(Bool_t det, Bool_t cl, Bool_t trklt, Bool_t trkin);
  void            SetDump3D(Bool_t det, Bool_t cl, Bool_t trklt, Bool_t trkin);
  void            SetVerbose(Bool_t v = kTRUE)          { SetBit(kVerbose, v);}
  void            SetVisual(Bool_t v = kTRUE)           { SetBit(kVisual, v);}
/*  void            SetTrackRefit(Bool_t v = kTRUE)       { SetBit(kTrackRefit, v);}
  void            SetTrackSelection(Bool_t v = kTRUE)   { SetBit(kTrackSelect, v);}*/
  void            SetUseExchangeContainers(Bool_t v = kTRUE) { SetBit(kXchange, v);}

  void            Terminate(Option_t * opt);
  static Bool_t   UseTrack(const Int_t np, const AliTrackPoint *points, Float_t params[10]);

  void        AdjustF1(TH1 *h, TF1 *f);
  TH2*        DrawSigma(TH2 *h2, const Char_t *t, Float_t m=0., Float_t M=-1., Float_t scale=1);
  void        GetLandauMpvFwhm(TF1 * const f, Float_t &mpv, Float_t &xm, Float_t &xM);
  void        GetRange(TH2 *h2, Char_t mod, Float_t *range);

protected:
  Bool_t      HasDump3DFor(ETRDresolutionClass cls) const { return TESTBIT(fSteer, 4+cls);}
  Bool_t      HasProcess(ETRDresolutionClass cls) const   { return TESTBIT(fSteer, cls);}
  Bool_t      MakeProjectionDetector();
  Bool_t      MakeProjectionCluster(Bool_t mc=kFALSE);
  Bool_t      MakeProjectionTracklet(Bool_t mc=kFALSE);
  Bool_t      MakeProjectionTrackIn(Bool_t mc=kFALSE, Bool_t v0=kFALSE);
  Bool_t      MakeProjectionTrack();
  Bool_t      Process(TH2* const /*h2*/, TF1 */*f*/, Float_t /*k*/, TGraphErrors **/*g*/) { return kTRUE;}
  Bool_t      Pulls(Double_t dyz[2], Double_t cc[3], Double_t tilt) const;

  UShort_t              fSteer;           // bit map to steer internal behaviour of class
                                          // MakeProjection [kTrackIn kTracklet kCluster kDetector]
                                          // Dump3D [4+kTrackIn 4+kTracklet 4+kCluster 4+kDetector]
  Float_t               fPtThreshold;     // pt threshold for some performance plots
  Int_t                 fBCbinTOF;        // set/select by TOF BC index
  Int_t                 fBCbinFill;       // set/select by Bunch Fill index
  Int_t                 fLYselect;        // select layer for Tracklet projections (in debug mode)
  Bool_t                fBsign;           // sign of magnetic field (kFALSE[-] kTRUE[+])
  static Char_t const  *fgPerformanceName[kNclasses]; //! name of performance plot
  static Int_t const    fgkNbins[kNdim];  //! no of bins/projection
  static Double_t const fgkMin[kNdim];    //! low limits for projections
  static Double_t const fgkMax[kNdim];    //! high limits for projections
  static Char_t const  *fgkTitle[kNdim];  //! title of projection
  TObjArray            *fProj;            //! result holder - sigma values
  TDatabasePDG         *fDBPDG;           //! PDG database

  // calibration containers
  TObjArray            *fCl;              //! cluster2track calib
  TObjArray            *fMCcl;            //! cluster2mc calib
  
  ClassDef(AliTRDresolution, 11) // 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
01ccc21a	22	#ifndef Root_TNamed
	23	#include "TNamed.h"
	24	#endif
	25
1ee39b3a	26	class TH1;
	27	class TH2;
	28	class TF1;
	29	class TGraphErrors;
	30	class TObjArray;
92c40e64	31	class TDatabasePDG;
1ee39b3a	32	class AliTRDrecoParam;
	33	class AliTRDseedV1;
	34	class AliTRDtrackInfo;
08bdd9d1	35	class AliTrackPoint;
1ee39b3a	36	class AliTRDresolution : public AliTRDrecoTask
	37	{
	38	public:
1ee39b3a	39	enum ETRDresolutionSteer {
08bdd9d1	40	kVerbose = BIT(18) // switch verbosity
	41	,kVisual = BIT(19) // show partial results during processing
	42	,kTrackRefit = BIT(20) // steer track refit
00a56108	43	,kTrackSelect= BIT(21) // steer track selection
3ed01fbe	44	,kXchange = BIT(22) // use exchange containers
1ee39b3a	45	};
3ceb45ae	46	enum ETRDresolutionSlots {
705f8b0a	47	kClToTrk = 2
3ceb45ae	48	,kClToMC
	49	,kTrkltToTrk
	50	,kTrkltToMC
	51	,kNSlots = 4
	52	};
	53	enum ETRDresolutionClass {
f073d500	54	kDetector=0 // cluster - detector
f073d500	55	,kCluster // cluster - track
3ceb45ae	56	,kTracklet // tracklet - track residuals/pulls
3ceb45ae	57	,kTrackIn // tracklet - track residuals/pulls at lower TRD entrance
3ceb45ae	58	,kMCcluster // cluster-mc resolution/pulls
	59	,kMCtracklet // tracklet-mc resolution/pulls
	60	,kMCtrackIn // TPC track monitor
3ceb45ae	61	,kMCtrack // TRD track monitor
f429b017	62	/* ,kTrackOut // tracklet - track residuals/pulls at lower TRD entrance during refit
f429b017	63	,kMCtrackOut // TOF/HMPID track monitor*/
3ceb45ae	64	,kNclasses // total number of resolution classes
95d47440	65	,kV0TrackIn // tracklet - track residuals/pulls at lower TRD entrance for V0 tagged tracks
3ceb45ae	66	};
0b366e97	67	enum ETRDresolutionClassProjs {
b9058d72	68	kDetNproj=460 // detector projections
b9058d72	69	,kClNproj=2800 // cluster projections
8892a7f3	70	,kTrkltNproj=8000 // tracklet projections
95d47440	71	,kTrkInNproj=1440 // trackIn projections
808d178e	72	,kTrkNproj=2500 // track projections
95d47440	73	,kMCTrkInNproj=1260// trackIn projections
0b366e97	74	};
3ceb45ae	75	enum ETRDresolutionProjs {
	76	kBC = 0 // bunch cross
	77	,kPhi
	78	,kEta
3ceb45ae	79	,kYrez
3ceb45ae	80	,kPrez
3ed01fbe	81	,kZrez
	82	,kSpeciesChgRC
	83	,kPt
3ceb45ae	84	,kNdim // no of dimensions in the THnSparse
56838f9d	85	,kNdimDet = 4
f073d500	86	,kNdimCl = 4
566c3d46	87	,kNdimTrklt = 4
566c3d46	88	,kNdimTrkIn = 7
3ceb45ae	89	,kNbunchCross = 3 // no of classes for bunch crossing
cc98ff07	90	,kNspc = 3 // no of species e, mu+pi, K+p
b9058d72	91	,kNcharge = 2 // no of charges
b9058d72	92	,kNpads = 4 // no of charges
9dcc64cc	93	};
1ee39b3a	94
1ee39b3a	95	AliTRDresolution();
3ed01fbe	96	AliTRDresolution(char* name, Bool_t xchange=kTRUE);
eb05d549	97	AliTRDresolution(const AliTRDresolution&);
eb05d549	98	AliTRDresolution& operator=(const AliTRDresolution&);
1ee39b3a	99	virtual ~AliTRDresolution();
1ee39b3a	100
3ed01fbe	101	static Bool_t FitTrack(const Int_t np, AliTrackPoint *points, Float_t params[10]);
	102	static Bool_t FitTracklet(const Int_t ly, const Int_t np, const AliTrackPoint *points, const Float_t trackPars[10], Float_t trackletPars[3]);
	103	void UserCreateOutputObjects();
3ceb45ae	104	// Float_t GetCorrectionX(Int_t det, Int_t tb) const {return fXcorr[det][tb];}
b351a191	105	static void GetRangeZ(TH2 *h2, Float_t &m, Float_t &M);
3ed01fbe	106	Float_t GetPtThreshold() const {return fPtThreshold;}
3ed01fbe	107	Bool_t GetRefFigure(Int_t ifig);
3ceb45ae	108
566c3d46	109	virtual TObjArray* Histos();
3ceb45ae	110	// Bool_t Load(const Char_t file = "AnalysisResults.root", const Char_t dir="TRD_Performance");
3ceb45ae	111	// Bool_t LoadCorrection(const Char_t *file=NULL);
3ed01fbe	112	void MakeSummary();
2589cf64	113
566c3d46	114	TObjArray* Results(ETRDresolutionClass c) const { if(!fProj) return NULL; return (TObjArray*)fProj->At(c);}
3ed01fbe	115	void UserExec(Option_t * opt);
3ed01fbe	116	void InitExchangeContainers();
95d47440	117	/* Bool_t HasTrackRefit() const { return TestBit(kTrackRefit);}
95d47440	118	Bool_t HasTrackSelection() const { return TestBit(kTrackSelect);}*/
566c3d46	119	Bool_t IsVerbose() const { return TestBit(kVerbose);}
	120	Bool_t IsVisual() const { return TestBit(kVisual);}
	121	Bool_t UseBCselectTOF() const { return fBCbinTOF>0;}
	122	Bool_t UseBCselectFill() const { return fBCbinFill>0;}
95d47440	123	Bool_t UseLYselectTrklt() const { return fLYselect>=0;}
566c3d46	124	Bool_t UseExchangeContainers() const { return TestBit(kXchange);}
3ed01fbe	125	Bool_t PostProcess();
3ed01fbe	126
f073d500	127	TH1* DetCluster(const TObjArray *cl=NULL);
3ed01fbe	128	TH1* PlotCluster(const AliTRDtrackV1 *t=NULL);
	129	TH1* PlotTracklet(const AliTRDtrackV1 *t=NULL);
	130	TH1* PlotTrackIn(const AliTRDtrackV1 *t=NULL);
f429b017	131	// TH1* PlotTrackOut(const AliTRDtrackV1 *t=NULL);
3ed01fbe	132	TH1* PlotMC(const AliTRDtrackV1 *t=NULL);
3ed01fbe	133
b9058d72	134	static Bool_t Process(TH2* const /h2/, TGraphErrors */g*/, Int_t stat=100){ return Bool_t(stat);}
566c3d46	135	void SetPtThreshold(Float_t pt) { fPtThreshold = pt;}
dffc3357	136	void SetBCselectTOF(Int_t b=0) { fBCbinTOF = b;}
566c3d46	137	void SetBCselectFill(Int_t b=0) { fBCbinFill = b<0\|\|b>3499?1:b+1;}
56838f9d	138	void SetBsign(Int_t b=0) { fBsign = Bool_t(b);}
95d47440	139	void SetLYselect(Int_t ly=0) { fLYselect = ly;}
b9058d72	140	void SetProcesses(Bool_t det, Bool_t cl, Bool_t trklt, Bool_t trkin);
33056e04	141	void SetDump3D(Bool_t det, Bool_t cl, Bool_t trklt, Bool_t trkin);
566c3d46	142	void SetVerbose(Bool_t v = kTRUE) { SetBit(kVerbose, v);}
566c3d46	143	void SetVisual(Bool_t v = kTRUE) { SetBit(kVisual, v);}
95d47440	144	/* void SetTrackRefit(Bool_t v = kTRUE) { SetBit(kTrackRefit, v);}
95d47440	145	void SetTrackSelection(Bool_t v = kTRUE) { SetBit(kTrackSelect, v);}*/
566c3d46	146	void SetUseExchangeContainers(Bool_t v = kTRUE) { SetBit(kXchange, v);}
1ee39b3a	147
3ed01fbe	148	void Terminate(Option_t * opt);
3ed01fbe	149	static Bool_t UseTrack(const Int_t np, const AliTrackPoint *points, Float_t params[10]);
3ceb45ae	150
3ed01fbe	151	void AdjustF1(TH1 h, TF1 f);
95d47440	152	TH2* DrawSigma(TH2 h2, const Char_t t, Float_t m=0., Float_t M=-1., Float_t scale=1);
3ed01fbe	153	void GetLandauMpvFwhm(TF1 * const f, Float_t &mpv, Float_t &xm, Float_t &xM);
	154	void GetRange(TH2 h2, Char_t mod, Float_t range);
	155
	156	protected:
33056e04	157	Bool_t HasDump3DFor(ETRDresolutionClass cls) const { return TESTBIT(fSteer, 4+cls);}
33056e04	158	Bool_t HasProcess(ETRDresolutionClass cls) const { return TESTBIT(fSteer, cls);}
f073d500	159	Bool_t MakeProjectionDetector();
f429b017	160	Bool_t MakeProjectionCluster(Bool_t mc=kFALSE);
f429b017	161	Bool_t MakeProjectionTracklet(Bool_t mc=kFALSE);
95d47440	162	Bool_t MakeProjectionTrackIn(Bool_t mc=kFALSE, Bool_t v0=kFALSE);
f429b017	163	Bool_t MakeProjectionTrack();
b9058d72	164	Bool_t Process(TH2* const /h2/, TF1 /f/, Float_t /k/, TGraphErrors /g*/) { return kTRUE;}
3ed01fbe	165	Bool_t Pulls(Double_t dyz[2], Double_t cc[3], Double_t tilt) const;
1ee39b3a	166
b9058d72	167	UShort_t fSteer; // bit map to steer internal behaviour of class
33056e04	168	// MakeProjection [kTrackIn kTracklet kCluster kDetector]
33056e04	169	// Dump3D [4+kTrackIn 4+kTracklet 4+kCluster 4+kDetector]
3ceb45ae	170	Float_t fPtThreshold; // pt threshold for some performance plots
566c3d46	171	Int_t fBCbinTOF; // set/select by TOF BC index
566c3d46	172	Int_t fBCbinFill; // set/select by Bunch Fill index
95d47440	173	Int_t fLYselect; // select layer for Tracklet projections (in debug mode)
56838f9d	174	Bool_t fBsign; // sign of magnetic field (kFALSE[-] kTRUE[+])
3ceb45ae	175	static Char_t const *fgPerformanceName[kNclasses]; //! name of performance plot
3ceb45ae	176	static Int_t const fgkNbins[kNdim]; //! no of bins/projection
	177	static Double_t const fgkMin[kNdim]; //! low limits for projections
	178	static Double_t const fgkMax[kNdim]; //! high limits for projections
eb05d549	179	static Char_t const *fgkTitle[kNdim]; //! title of projection
3ceb45ae	180	TObjArray *fProj; //! result holder - sigma values
3ceb45ae	181	TDatabasePDG *fDBPDG; //! PDG database
1ee39b3a	182
1ee39b3a	183	// calibration containers
3ed01fbe	184	TObjArray *fCl; //! cluster2track calib
3ed01fbe	185	TObjArray *fMCcl; //! cluster2mc calib
1ee39b3a	186
dffc3357	187	ClassDef(AliTRDresolution, 11) // TRD tracking resolution task
1ee39b3a	188	};
1ee39b3a	189	#endif