[u/mrichter/AliRoot.git] / TRD / AliTRDCalibraVdriftLinearFit.h

#ifndef ALITRDCALIBRAVDRIFTLINEARFIT_H
#define ALITRDCALIBRAVDRIFTLINEARFIT_H

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

/* $Id$ */

///////////////////////////////////////////////////////////////////////////////
//                                                                           //
//  TRD calibration class for online calibration                             //
//                                                                           //
///////////////////////////////////////////////////////////////////////////////

//#include "TObjArray.h"
#include "TObject.h"

//class TVectorD;
class TObjArray;
class TH2S;
class TH2;
class TTreeSRedirector;
class TString;
class TGraphErrors;
class TLinearFitter;
class TTreeSRedirector;

class AliTRDCalibraVdriftLinearFit : public TObject {

 public:

  AliTRDCalibraVdriftLinearFit();
  AliTRDCalibraVdriftLinearFit(const AliTRDCalibraVdriftLinearFit &ped);
  AliTRDCalibraVdriftLinearFit(const TObjArray &obja);
  virtual ~AliTRDCalibraVdriftLinearFit();
  virtual Long64_t Merge(const TCollection* list);
  virtual void Copy(TObject &c) const;

  AliTRDCalibraVdriftLinearFit& operator = (const  AliTRDCalibraVdriftLinearFit &source);

  void            Update(Int_t detector, Float_t tnp, Float_t pars1);
  void            FillPEArray();
  void            FillPEArray2();
  void            SetNameCalibUsed(const char*name)          { fNameCalibUsed = name;};
  const char*     GetNameCalibUsed() const                   { return fNameCalibUsed;};
  void            Add(const AliTRDCalibraVdriftLinearFit *ped);
  TH2S           *GetLinearFitterHisto(Int_t detector, Bool_t force=kFALSE);
  TH2S           *GetLinearFitterHistoForce(Int_t detector);
  TH2S           *GetLinearFitterHistoNoForce(Int_t detector) const   { return (TH2S*)fLinearFitterHistoArray.UncheckedAt(detector);};
  Bool_t          GetParam(Int_t detector, TVectorD *param);
  Bool_t          GetError(Int_t detector, TVectorD *error);

  TObjArray      *GetPArray()                    { return &fLinearFitterPArray;       };
  TObjArray      *GetEArray()                    { return &fLinearFitterEArray;       };
  TObjArray       GetHistoArray() const          { return fLinearFitterHistoArray;    };
  void            SetRobustFit(Bool_t robustFit) { fRobustFit = robustFit;            };

  // Debug
  void     SetDebugLevel(Short_t level)          { fDebugLevel = level;               };
  void     SetSeeDetector(Int_t seeDetector)     { fSeeDetector = seeDetector;        };

private:
  TGraphErrors   *DrawMS(const TH2 *const h2, Int_t &nEntries);

  Int_t           fVersion;                 // Version of the object
  TString         fNameCalibUsed;           // Info of the version, subversion, firstrun of the calib used

  TObjArray       fLinearFitterHistoArray;  // TObjArray of histo2D for debugging Linear Fitters
  TObjArray       fLinearFitterPArray;      // Array of result parameters from linear fitters for the detectors
  TObjArray       fLinearFitterEArray;      // Array of result errors from linear fitters for the detectors
  Bool_t          fRobustFit;               // Robust fit or not
  
 //For debugging
  TTreeSRedirector *fDebugStreamer;         //!Debug streamer
  Short_t           fDebugLevel;            // Flag for debugging
  Int_t             fSeeDetector;           // Detector to see
 
  ClassDef(AliTRDCalibraVdriftLinearFit,3)  // Online Vdrift calibration

};


#endif
Commit	Line	Data
3a0f6479	1	#ifndef ALITRDCALIBRAVDRIFTLINEARFIT_H
	2	#define ALITRDCALIBRAVDRIFTLINEARFIT_H
	3
	4	/* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
	5	* See cxx source for full Copyright notice */
	6
	7	/* $Id$ */
	8
	9	///////////////////////////////////////////////////////////////////////////////
	10	// //
	11	// TRD calibration class for online calibration //
	12	// //
	13	///////////////////////////////////////////////////////////////////////////////
	14
36dc3337	15	//#include "TObjArray.h"
36dc3337	16	#include "TObject.h"
d0569428	17
36dc3337	18	//class TVectorD;
3a0f6479	19	class TObjArray;
0ca1720e	20	class TH2S;
661c7be6	21	class TH2;
3a0f6479	22	class TTreeSRedirector;
b2277aa2	23	class TString;
661c7be6	24	class TGraphErrors;
	25	class TLinearFitter;
	26	class TTreeSRedirector;
3a0f6479	27
	28	class AliTRDCalibraVdriftLinearFit : public TObject {
	29
0bc7827a	30	public:
3a0f6479	31
	32	AliTRDCalibraVdriftLinearFit();
	33	AliTRDCalibraVdriftLinearFit(const AliTRDCalibraVdriftLinearFit &ped);
d0569428	34	AliTRDCalibraVdriftLinearFit(const TObjArray &obja);
3a0f6479	35	virtual ~AliTRDCalibraVdriftLinearFit();
6bbdc11a	36	virtual Long64_t Merge(const TCollection* list);
64942b85	37	virtual void Copy(TObject &c) const;
3a0f6479	38
	39	AliTRDCalibraVdriftLinearFit& operator = (const AliTRDCalibraVdriftLinearFit &source);
	40
	41	void Update(Int_t detector, Float_t tnp, Float_t pars1);
	42	void FillPEArray();
661c7be6	43	void FillPEArray2();
b2277aa2	44	void SetNameCalibUsed(const char*name) { fNameCalibUsed = name;};
b2277aa2	45	const char* GetNameCalibUsed() const { return fNameCalibUsed;};
fe4ee353	46	void Add(const AliTRDCalibraVdriftLinearFit *ped);
0ca1720e	47	TH2S *GetLinearFitterHisto(Int_t detector, Bool_t force=kFALSE);
fe4ee353	48	TH2S *GetLinearFitterHistoForce(Int_t detector);
fe4ee353	49	TH2S GetLinearFitterHistoNoForce(Int_t detector) const { return (TH2S)fLinearFitterHistoArray.UncheckedAt(detector);};
3a0f6479	50	Bool_t GetParam(Int_t detector, TVectorD *param);
	51	Bool_t GetError(Int_t detector, TVectorD *error);
	52
0bc7827a	53	TObjArray *GetPArray() { return &fLinearFitterPArray; };
	54	TObjArray *GetEArray() { return &fLinearFitterEArray; };
	55	TObjArray GetHistoArray() const { return fLinearFitterHistoArray; };
2a1a7b36	56	void SetRobustFit(Bool_t robustFit) { fRobustFit = robustFit; };
3a0f6479	57
661c7be6	58	// Debug
	59	void SetDebugLevel(Short_t level) { fDebugLevel = level; };
	60	void SetSeeDetector(Int_t seeDetector) { fSeeDetector = seeDetector; };
	61
	62	private:
	63	TGraphErrors DrawMS(const TH2 const h2, Int_t &nEntries);
	64
3a0f6479	65	Int_t fVersion; // Version of the object
b2277aa2	66	TString fNameCalibUsed; // Info of the version, subversion, firstrun of the calib used
3a0f6479	67
	68	TObjArray fLinearFitterHistoArray; // TObjArray of histo2D for debugging Linear Fitters
	69	TObjArray fLinearFitterPArray; // Array of result parameters from linear fitters for the detectors
	70	TObjArray fLinearFitterEArray; // Array of result errors from linear fitters for the detectors
2a1a7b36	71	Bool_t fRobustFit; // Robust fit or not
ba94744a	72
661c7be6	73	//For debugging
	74	TTreeSRedirector *fDebugStreamer; //!Debug streamer
	75	Short_t fDebugLevel; // Flag for debugging
	76	Int_t fSeeDetector; // Detector to see
	77
2a1a7b36	78	ClassDef(AliTRDCalibraVdriftLinearFit,3) // Online Vdrift calibration
3a0f6479	79
	80	};
	81
	82
	83
	84	#endif
a5dcf618	85