[u/mrichter/AliRoot.git] / PWGPP / TRD / AliTRDrecoTask.h

#ifndef ALITRDRECOTASK_H
#define ALITRDRECOTASK_H

///////////////////////////////////////////////////////
//
// Basic class for Performance/Calibration TRD tasks
//
// Author: Alexandru Bercuci, 10/09/2008
//
//////////////////////////////////////////////////

#ifndef ALIANALYSISTASKSE_H
#include "AliAnalysisTaskSE.h"
#endif

#ifndef ALITRDTRACKINFO_H
#include "info/AliTRDtrackInfo.h"
#endif

#ifndef ALITRDEVENTINFO_H
#include "info/AliTRDeventInfo.h"
#endif

class TAxis;
class TH1;
class TH2;
class TH3;
class TF1;
class TList;
class TObjArray;
class TTreeSRedirector;
class AliTRDtrackV1;
class AliTRDrecoTask : public AliAnalysisTaskSE
{
friend class AliEveTRDTrackList;
public:
  enum AliTRDrecoSteeringBits{
     kMCdata      = BIT(18)
    ,kFriends     = BIT(19)
    ,kPostProcess = BIT(20)
    ,kHeavyIon    = BIT(21)
  };
  
  class AliTRDrecoProjection : public TNamed
  {
  public:
    AliTRDrecoProjection();
    virtual ~AliTRDrecoProjection();
    AliTRDrecoProjection& operator+=(const AliTRDrecoProjection& other);
    AliTRDrecoProjection& operator=(const AliTRDrecoProjection& other);
    void      Build(const Char_t *n, const Char_t *t, Int_t ix, Int_t iy, Int_t iz, TAxis *aa[]);
    Double_t  GetTrendValue(const Int_t mid=0, Double_t *m=NULL, Double_t *s=NULL, Double_t *se=NULL) const;
    TH3*      H() const { return fH;}
    void      Increment(Int_t bin[], Double_t v);
    TH2*      Projection2D(const Int_t nstat, const Int_t ncol, const Int_t mid=0, Bool_t del=kTRUE);
    TH2*      Projection2Dbin(Int_t ibin=-1, Bool_t mc=kFALSE);
    void      SetRebinStrategy(Int_t n, Int_t rebx[], Int_t reby[]);
    void      SetShowRange(Float_t zm, Float_t zM, Float_t em=0., Float_t eM=0.) {fRange[0] = zm; fRange[1] = zM; fRange[2] = em; fRange[3] = eM;}
  private:
    AliTRDrecoProjection(const AliTRDrecoProjection&);
  protected:
    TH3  *fH;          // data container
    Int_t fAx[3];      // projection axes
    Int_t fNrebin;     // no. of rebinning steps
    Int_t *fRebin[2];  //[fNrebin] rebinning of the X nd Y axes
    Float_t fRange[4]; //show range of the z processed

    ClassDef(AliTRDrecoProjection, 2)  // wrapper for a projection container THnSparse -> TH3
  };

  AliTRDrecoTask();
  AliTRDrecoTask(const char *name, const char *title);
  virtual ~AliTRDrecoTask();
  
  
  virtual void   UserCreateOutputObjects();
  virtual void   UserExec(Option_t *opt);
  virtual void   SetDebugLevel(Int_t level);
  
    
  static Float_t GetMeanStat(TH1 *h, Float_t cut=0., Int_t opt=0, Float_t *sigma=NULL);
  Int_t          GetNRefFigures() const; 
  const Char_t*  GetNameId() const       { return fNameId;}
  TList*         GetPlotFunctors() const { return fPlotFuncList;}
  Int_t          GetPtBin(Float_t pt);
  virtual Bool_t GetRefFigure(Int_t ifig);
  virtual void   MakeSummary();
  void           MakeDetectorPlot(Int_t ly=0, const Option_t *opt="");
  void           MakeDetectorPlotOLD(Int_t ly=0, const Option_t *opt="");
  Bool_t         IsHeavyIon() const      { return TestBit(kHeavyIon);};
  Bool_t         IsPP() const            { return !TestBit(kHeavyIon);};
  Bool_t         HasFriends() const      { return TestBit(kFriends);};
  Bool_t         HasMCdata() const       { return TestBit(kMCdata);};
  Bool_t         HasPostProcess() const  { return TestBit(kPostProcess);};
  Bool_t         HasRunTerminate() const { return fRunTerminate; }
  virtual TObjArray* Histos()            { return fContainer;}

  virtual Bool_t Load(const Char_t *file = "AnalysisResults.root", const Char_t *dir = "TRD_Performance");
  virtual Bool_t LoadDetectorMap(const Char_t *file = "AnalysisResults.root", const Char_t *dir = "TRD_Performance");
  virtual Bool_t Save(TObjArray * const res);
  virtual Bool_t PostProcess();
  virtual Bool_t PutTrendValue(const Char_t *name, Double_t val, Double_t err=0);
  static Int_t   Rebin(TH2 *h, Int_t n, Int_t rebinX[], Int_t rebinY[], Int_t nstat);
  virtual void   SetFriends(Bool_t fr = kTRUE) {SetBit(kFriends, fr);}
  virtual void   SetMCdata(Bool_t mc = kTRUE) {SetBit(kMCdata, mc);}
  virtual void   SetNameId(const Char_t *nid) {snprintf(fNameId, 10, "%s", nid);}
  virtual void   SetPostProcess(Bool_t pp = kTRUE) {SetBit(kPostProcess, pp);}
  static Float_t SetNormZ(TH2 *h2, Int_t bxmin=1, Int_t bxmax=-1, Int_t bymin=1, Int_t bymax=-1, Float_t thr=0.);
  static void    SetRangeZ(TH2 *h2, Float_t m, Float_t M, Float_t thr=0.);
  void           SetRunTerminate(Bool_t runTerminate = kTRUE) { fRunTerminate = runTerminate; }
  void           SetTriggerList(const Char_t *tl);
  virtual void   Terminate(Option_t *);

protected:
  static TTreeSRedirector* DebugStream() { return fgDebugStream;}
  virtual void   InitFunctorList();
  Bool_t         HasFunctorList() const { return fPlotFuncList != NULL; }
  Bool_t         MakeMomSegmentation();

  Char_t                fNameId[10];       // unique identifier of task particularity
  UChar_t               fNRefFigures;      // no of reference figures reported by task
  TObjArray             *fDets;            //! OLD container to store detector position and status support should be discontinued 
  TObjArray             *fDetsV;           //! NEW container to store detector position and status
  TObjArray             *fContainer;       //! container to store results
  AliTRDeventInfo       *fEvent;           //! Event Info
  TObjArray             *fTracks;          //! Array of tracks
  TObjArray             *fClusters;        //! Array of clusters
  const TObjArray       *fkClusters;       //! current detector clusters array
  const AliTRDtrackV1   *fkTrack;          //! current track
  const AliTRDtrackInfo::AliMCinfo  *fkMC; //! MC info
  const AliTRDtrackInfo::AliESDinfo *fkESD;//! ESD info
  Char_t                 fSpecies;         //! species index +1 with charge sign
  Char_t                 fTriggerSlot;     //! selected triggers map (if requested)
  Float_t                fPt;              //! p_t of the track being analyzed
  Float_t                fPhi;             //! phi of the track being analyzed
  Float_t                fEta;             //! eta of the track being analyzed
  Int_t                  fNpt;             // no of pt/p bins actually used
  TObjArray             *fTriggerList;     //! optional trigger list to be monitored

private:
  AliTRDrecoTask(const AliTRDrecoTask&);
  AliTRDrecoTask& operator=(const AliTRDrecoTask&);

  TList             *fPlotFuncList;        //! track functors list
  TList             *fDetFuncList;         //! detector functors list
  Bool_t             fRunTerminate;        // Switch for Terminate Function
  static TTreeSRedirector *fgDebugStream;  //! Debug stream
  static TH1        *fgProjector;          //! Projector histogram for Z statistics
protected:
  static const Int_t fgNPt = 25;           //! No of debug pt bins
  static Float_t         fgPt[fgNPt+1];    //! Array with limits for debug pt bins

  ClassDef(AliTRDrecoTask, 5) // base TRD reconstruction task
};

#endif
Commit	Line	Data
a65a7e70	1	#ifndef ALITRDRECOTASK_H
	2	#define ALITRDRECOTASK_H
	3
	4	///////////////////////////////////////////////////////
	5	//
	6	// Basic class for Performance/Calibration TRD tasks
	7	//
	8	// Author: Alexandru Bercuci, 10/09/2008
	9	//
	10	//////////////////////////////////////////////////
	11
	12	#ifndef ALIANALYSISTASKSE_H
	13	#include "AliAnalysisTaskSE.h"
	14	#endif
	15
	16	#ifndef ALITRDTRACKINFO_H
	17	#include "info/AliTRDtrackInfo.h"
	18	#endif
	19
	20	#ifndef ALITRDEVENTINFO_H
	21	#include "info/AliTRDeventInfo.h"
	22	#endif
	23
	24	class TAxis;
	25	class TH1;
	26	class TH2;
	27	class TH3;
	28	class TF1;
	29	class TList;
	30	class TObjArray;
	31	class TTreeSRedirector;
	32	class AliTRDtrackV1;
	33	class AliTRDrecoTask : public AliAnalysisTaskSE
	34	{
	35	friend class AliEveTRDTrackList;
	36	public:
	37	enum AliTRDrecoSteeringBits{
	38	kMCdata = BIT(18)
	39	,kFriends = BIT(19)
	40	,kPostProcess = BIT(20)
	41	,kHeavyIon = BIT(21)
	42	};
	43
	44	class AliTRDrecoProjection : public TNamed
	45	{
	46	public:
	47	AliTRDrecoProjection();
	48	virtual ~AliTRDrecoProjection();
	49	AliTRDrecoProjection& operator+=(const AliTRDrecoProjection& other);
	50	AliTRDrecoProjection& operator=(const AliTRDrecoProjection& other);
	51	void Build(const Char_t n, const Char_t t, Int_t ix, Int_t iy, Int_t iz, TAxis *aa[]);
	52	Double_t GetTrendValue(const Int_t mid=0, Double_t m=NULL, Double_t s=NULL, Double_t *se=NULL) const;
	53	TH3* H() const { return fH;}
	54	void Increment(Int_t bin[], Double_t v);
	55	TH2* Projection2D(const Int_t nstat, const Int_t ncol, const Int_t mid=0, Bool_t del=kTRUE);
	56	TH2* Projection2Dbin(Int_t ibin=-1, Bool_t mc=kFALSE);
	57	void SetRebinStrategy(Int_t n, Int_t rebx[], Int_t reby[]);
	58	void SetShowRange(Float_t zm, Float_t zM, Float_t em=0., Float_t eM=0.) {fRange[0] = zm; fRange[1] = zM; fRange[2] = em; fRange[3] = eM;}
	59	private:
	60	AliTRDrecoProjection(const AliTRDrecoProjection&);
	61	protected:
	62	TH3 *fH; // data container
	63	Int_t fAx[3]; // projection axes
	64	Int_t fNrebin; // no. of rebinning steps
65	Int_t *fRebin[2]; //[fNrebin] rebinning of the X nd Y axes
66	Float_t fRange[4]; //show range of the z processed
67
68	ClassDef(AliTRDrecoProjection, 2) // wrapper for a projection container THnSparse -> TH3
69	};
70
71	AliTRDrecoTask();
72	AliTRDrecoTask(const char name, const char title);
73	virtual ~AliTRDrecoTask();
74
75
76	virtual void UserCreateOutputObjects();
77	virtual void UserExec(Option_t *opt);
78	virtual void SetDebugLevel(Int_t level);
79
80
81	static Float_t GetMeanStat(TH1 h, Float_t cut=0., Int_t opt=0, Float_t sigma=NULL);
82	Int_t GetNRefFigures() const;
83	const Char_t* GetNameId() const { return fNameId;}
84	TList* GetPlotFunctors() const { return fPlotFuncList;}
85	Int_t GetPtBin(Float_t pt);
86	virtual Bool_t GetRefFigure(Int_t ifig);
87	virtual void MakeSummary();
88	void MakeDetectorPlot(Int_t ly=0, const Option_t *opt="");
89	void MakeDetectorPlotOLD(Int_t ly=0, const Option_t *opt="");
90	Bool_t IsHeavyIon() const { return TestBit(kHeavyIon);};
91	Bool_t IsPP() const { return !TestBit(kHeavyIon);};
92	Bool_t HasFriends() const { return TestBit(kFriends);};
93	Bool_t HasMCdata() const { return TestBit(kMCdata);};
94	Bool_t HasPostProcess() const { return TestBit(kPostProcess);};
95	Bool_t HasRunTerminate() const { return fRunTerminate; }
96	virtual TObjArray* Histos() { return fContainer;}
97
98	virtual Bool_t Load(const Char_t file = "AnalysisResults.root", const Char_t dir = "TRD_Performance");
99	virtual Bool_t LoadDetectorMap(const Char_t file = "AnalysisResults.root", const Char_t dir = "TRD_Performance");
100	virtual Bool_t Save(TObjArray * const res);
101	virtual Bool_t PostProcess();
102	virtual Bool_t PutTrendValue(const Char_t *name, Double_t val, Double_t err=0);
103	static Int_t Rebin(TH2 *h, Int_t n, Int_t rebinX[], Int_t rebinY[], Int_t nstat);
104	virtual void SetFriends(Bool_t fr = kTRUE) {SetBit(kFriends, fr);}
105	virtual void SetMCdata(Bool_t mc = kTRUE) {SetBit(kMCdata, mc);}
106	virtual void SetNameId(const Char_t *nid) {snprintf(fNameId, 10, "%s", nid);}
107	virtual void SetPostProcess(Bool_t pp = kTRUE) {SetBit(kPostProcess, pp);}
108	static Float_t SetNormZ(TH2 *h2, Int_t bxmin=1, Int_t bxmax=-1, Int_t bymin=1, Int_t bymax=-1, Float_t thr=0.);
109	static void SetRangeZ(TH2 *h2, Float_t m, Float_t M, Float_t thr=0.);
110	void SetRunTerminate(Bool_t runTerminate = kTRUE) { fRunTerminate = runTerminate; }
111	void SetTriggerList(const Char_t *tl);
112	virtual void Terminate(Option_t *);
113
114	protected:
115	static TTreeSRedirector* DebugStream() { return fgDebugStream;}
116	virtual void InitFunctorList();
117	Bool_t HasFunctorList() const { return fPlotFuncList != NULL; }
118	Bool_t MakeMomSegmentation();
119
120	Char_t fNameId[10]; // unique identifier of task particularity
121	UChar_t fNRefFigures; // no of reference figures reported by task
122	TObjArray *fDets; //! OLD container to store detector position and status support should be discontinued
123	TObjArray *fDetsV; //! NEW container to store detector position and status
124	TObjArray *fContainer; //! container to store results
125	AliTRDeventInfo *fEvent; //! Event Info
126	TObjArray *fTracks; //! Array of tracks
127	TObjArray *fClusters; //! Array of clusters
128	const TObjArray *fkClusters; //! current detector clusters array
129	const AliTRDtrackV1 *fkTrack; //! current track
130	const AliTRDtrackInfo::AliMCinfo *fkMC; //! MC info
131	const AliTRDtrackInfo::AliESDinfo *fkESD;//! ESD info
132	Char_t fSpecies; //! species index +1 with charge sign
133	Char_t fTriggerSlot; //! selected triggers map (if requested)
134	Float_t fPt; //! p_t of the track being analyzed
135	Float_t fPhi; //! phi of the track being analyzed
136	Float_t fEta; //! eta of the track being analyzed
137	Int_t fNpt; // no of pt/p bins actually used
138	TObjArray *fTriggerList; //! optional trigger list to be monitored
139
140	private:
141	AliTRDrecoTask(const AliTRDrecoTask&);
142	AliTRDrecoTask& operator=(const AliTRDrecoTask&);
143
144	TList *fPlotFuncList; //! track functors list
145	TList *fDetFuncList; //! detector functors list
146	Bool_t fRunTerminate; // Switch for Terminate Function
147	static TTreeSRedirector *fgDebugStream; //! Debug stream
148	static TH1 *fgProjector; //! Projector histogram for Z statistics
149	protected:
150	static const Int_t fgNPt = 25; //! No of debug pt bins
151	static Float_t fgPt[fgNPt+1]; //! Array with limits for debug pt bins
152
153	ClassDef(AliTRDrecoTask, 5) // base TRD reconstruction task
154	};
155
156	#endif
157