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

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

/* $Id$ */

///////////////////////////////////////////////////////////////////////////////
//                                                                           //
//  TRD calibration class                                                    //
//                                                                           //
///////////////////////////////////////////////////////////////////////////////

#ifndef ROOT_TObject
#  include <TObject.h>
#endif

class TGraphErrors;
class TH1F;
class TObjArray;

class TArrayF;
class TArrayI;

class AliTRDCalibraVector : public TObject {

 public: 

  AliTRDCalibraVector();
  AliTRDCalibraVector(const AliTRDCalibraVector &c);
  virtual ~AliTRDCalibraVector();

  AliTRDCalibraVector& operator = (const  AliTRDCalibraVector &source);

  // Fill
  Int_t          SearchBin(Float_t value, Int_t i) const;  
  Bool_t         UpdateVectorCH(Int_t det, Int_t group, Float_t value);
  Bool_t         UpdateVectorPRF(Int_t det, Int_t group, Float_t x, Float_t y);
  Bool_t         UpdateVectorPH(Int_t det, Int_t group, Int_t time, Float_t value);
 

  // Add
  Bool_t         Add(AliTRDCalibraVector *calvector);
  
  // Fit
  TGraphErrors  *ConvertVectorPHTGraphErrors(Int_t det, Int_t group, const Char_t *name);
  TGraphErrors  *ConvertVectorPRFTGraphErrors(Int_t det, Int_t group, const Char_t *name);
  TH1F          *ConvertVectorCHHisto(Int_t det, Int_t group, const Char_t *name);

  //
  // Set and Get methods
  //

  void           SetNumberBinCharge(Short_t numberbincharge)   { fNumberBinCharge = numberbincharge;            } 
  void           SetNumberBinPRF(Short_t numberbinprf)         { fNumberBinPRF    = numberbinprf;               } 
  void           SetTimeMax(Int_t timemax)                     { fTimeMax         = timemax;                    } 
  void           SetPRFRange(Float_t prfrange)                 { fPRFRange        = prfrange;                   }  
  void           SetDetCha0(Int_t i, Short_t total)            { fDetCha0[i]      = total;                      } 
  void           SetDetCha2(Int_t i, Short_t total)            { fDetCha2[i]      = total;                      } 
  void           SetNamePH(const char* name)                   { fNamePH = name;  } 
  void           SetNamePRF(const char* name)                  { fNamePRF = name; } 
  void           SetNameCH(const char* name)                   { fNameCH = name;  } 

  Short_t        GetNumberBinCharge()const                     { return fNumberBinCharge;                       }
  Short_t        GetNumberBinPRF()const                        { return fNumberBinPRF;                          }
  Int_t          GetTimeMax()const                             { return fTimeMax;                               } 
  Float_t        GetPRFRange()const                            { return fPRFRange;                              } 
  Short_t        GetDetCha0(Int_t i) const                     { return fDetCha0[i];                            }
  Short_t        GetDetCha2(Int_t i) const                     { return fDetCha2[i];                            }
  const char*    GetNamePH()                                   { return fNamePH;             }
  const char*    GetNamePRF()                                  { return fNamePRF;            }
  const char*    GetNameCH()                                   { return fNameCH;             }


  TArrayI  *GetPHEntries(Int_t det,Bool_t force = kFALSE);
  TArrayF  *GetPHMean(Int_t det,Bool_t force = kFALSE);
  TArrayF  *GetPHSquares(Int_t det,Bool_t force = kFALSE);

  TArrayI  *GetPRFEntries(Int_t det,Bool_t force = kFALSE);
  TArrayF  *GetPRFMean(Int_t det,Bool_t force = kFALSE);
  TArrayF  *GetPRFSquares(Int_t det,Bool_t force = kFALSE);

  TArrayI  *GetCHEntries(Int_t det,Bool_t force = kFALSE);

 protected:

  // Current data

         TArrayI     *fPHEntries[540];                //  Current AliTRDArrayI PH entries
         TArrayF     *fPHMean[540];                   //  Current AliTRDArrayF PH Mean
         TArrayF     *fPHSquares[540];                //  Current AliTRDArrayF PH Squares

	 TArrayI     *fPRFEntries[540];               //  Current AliTRDArrayI PH entries
         TArrayF     *fPRFMean[540];                  //  Current AliTRDArrayF PH Mean
         TArrayF     *fPRFSquares[540];               //  Current AliTRDArrayF PH Squares

	 TArrayI     *fCHEntries[540];                //  Current AliTRDArrayI PH entries

	 const char *fNameCH;                         //  Name for calibration mode
	 const char *fNamePH;                         //  Name for calibration mode
	 const char *fNamePRF;                        //  Name for calibration mode

	 Int_t            fDetectorPH;                //  Current detector
	 Int_t            fDetectorCH;                //  Current detector
	 Int_t            fDetectorPRF;               //  Current detector
       	 
  // Size of the infos

	  Short_t          fNumberBinCharge;          // Number of bins for the gain factor
	  Short_t          fNumberBinPRF;             // Number of bin for the PRF
	  Int_t            fTimeMax;                  // Number of time bins
	  Float_t          fPRFRange;                 // Range PRF
	  Short_t          fDetCha0[3];               // Number of XBins for chamber != 2
          Short_t          fDetCha2[3];               // Number of Xbins for chamber 2

  // Some functions


	  TArrayI  *GetEntriesPH(Int_t det, TArrayI** array, Bool_t force);
	  TArrayF  *GetMeanSquaresPH(Int_t det, TArrayF** array, Bool_t force);

	  TArrayI  *GetEntriesPRF(Int_t det, TArrayI** array, Bool_t force);
	  TArrayF  *GetMeanSquaresPRF(Int_t det, TArrayF** array, Bool_t force);
	  
	  TArrayI  *GetEntriesCH(Int_t det, TArrayI** array, Bool_t force);

  // Some basic geometry function
          virtual Int_t    GetStack(Int_t d) const;
  
  ClassDef(AliTRDCalibraVector,1)                   // TRD Calibration class

};
  
#endif
Commit	Line	Data
55a288e5	1	#ifndef ALITRDCALIBRAVECTOR_H
	2	#define ALITRDCALIBRAVECTOR_H
	3	/* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
	4	* See cxx source for full Copyright notice */
	5
	6	/* $Id$ */
	7
	8	///////////////////////////////////////////////////////////////////////////////
	9	// //
	10	// TRD calibration class //
	11	// //
	12	///////////////////////////////////////////////////////////////////////////////
	13
	14	#ifndef ROOT_TObject
	15	# include <TObject.h>
	16	#endif
	17
	18	class TGraphErrors;
	19	class TH1F;
55a288e5	20	class TObjArray;
55a288e5	21
00a3834d	22	class TArrayF;
00a3834d	23	class TArrayI;
3a0f6479	24
55a288e5	25	class AliTRDCalibraVector : public TObject {
	26
	27	public:
	28
	29	AliTRDCalibraVector();
	30	AliTRDCalibraVector(const AliTRDCalibraVector &c);
	31	virtual ~AliTRDCalibraVector();
55a288e5	32
3a0f6479	33	AliTRDCalibraVector& operator = (const AliTRDCalibraVector &source);
	34
	35	// Fill
55a288e5	36	Int_t SearchBin(Float_t value, Int_t i) const;
3a0f6479	37	Bool_t UpdateVectorCH(Int_t det, Int_t group, Float_t value);
	38	Bool_t UpdateVectorPRF(Int_t det, Int_t group, Float_t x, Float_t y);
	39	Bool_t UpdateVectorPH(Int_t det, Int_t group, Int_t time, Float_t value);
00a3834d	40
00a3834d	41
3a0f6479	42
	43	// Add
	44	Bool_t Add(AliTRDCalibraVector *calvector);
	45
	46	// Fit
	47	TGraphErrors ConvertVectorPHTGraphErrors(Int_t det, Int_t group, const Char_t name);
	48	TGraphErrors ConvertVectorPRFTGraphErrors(Int_t det, Int_t group, const Char_t name);
	49	TH1F ConvertVectorCHHisto(Int_t det, Int_t group, const Char_t name);
55a288e5	50
	51	//
	52	// Set and Get methods
	53	//
	54
	55	void SetNumberBinCharge(Short_t numberbincharge) { fNumberBinCharge = numberbincharge; }
	56	void SetNumberBinPRF(Short_t numberbinprf) { fNumberBinPRF = numberbinprf; }
170c35f1	57	void SetTimeMax(Int_t timemax) { fTimeMax = timemax; }
170c35f1	58	void SetPRFRange(Float_t prfrange) { fPRFRange = prfrange; }
3a0f6479	59	void SetDetCha0(Int_t i, Short_t total) { fDetCha0[i] = total; }
3a0f6479	60	void SetDetCha2(Int_t i, Short_t total) { fDetCha2[i] = total; }
00a3834d	61	void SetNamePH(const char* name) { fNamePH = name; }
	62	void SetNamePRF(const char* name) { fNamePRF = name; }
	63	void SetNameCH(const char* name) { fNameCH = name; }
55a288e5	64
	65	Short_t GetNumberBinCharge()const { return fNumberBinCharge; }
	66	Short_t GetNumberBinPRF()const { return fNumberBinPRF; }
	67	Int_t GetTimeMax()const { return fTimeMax; }
170c35f1	68	Float_t GetPRFRange()const { return fPRFRange; }
0bc7827a	69	Short_t GetDetCha0(Int_t i) const { return fDetCha0[i]; }
0bc7827a	70	Short_t GetDetCha2(Int_t i) const { return fDetCha2[i]; }
00a3834d	71	const char* GetNamePH() { return fNamePH; }
	72	const char* GetNamePRF() { return fNamePRF; }
	73	const char* GetNameCH() { return fNameCH; }
	74
3a0f6479	75
00a3834d	76
	77	TArrayI *GetPHEntries(Int_t det,Bool_t force = kFALSE);
	78	TArrayF *GetPHMean(Int_t det,Bool_t force = kFALSE);
	79	TArrayF *GetPHSquares(Int_t det,Bool_t force = kFALSE);
3a0f6479	80
00a3834d	81	TArrayI *GetPRFEntries(Int_t det,Bool_t force = kFALSE);
	82	TArrayF *GetPRFMean(Int_t det,Bool_t force = kFALSE);
	83	TArrayF *GetPRFSquares(Int_t det,Bool_t force = kFALSE);
3a0f6479	84
00a3834d	85	TArrayI *GetCHEntries(Int_t det,Bool_t force = kFALSE);
3a0f6479	86
55a288e5	87	protected:
55a288e5	88
3a0f6479	89	// Current data
3a0f6479	90
00a3834d	91	TArrayI *fPHEntries[540]; // Current AliTRDArrayI PH entries
	92	TArrayF *fPHMean[540]; // Current AliTRDArrayF PH Mean
	93	TArrayF *fPHSquares[540]; // Current AliTRDArrayF PH Squares
3a0f6479	94
00a3834d	95	TArrayI *fPRFEntries[540]; // Current AliTRDArrayI PH entries
	96	TArrayF *fPRFMean[540]; // Current AliTRDArrayF PH Mean
	97	TArrayF *fPRFSquares[540]; // Current AliTRDArrayF PH Squares
55a288e5	98
00a3834d	99	TArrayI *fCHEntries[540]; // Current AliTRDArrayI PH entries
	100
	101	const char *fNameCH; // Name for calibration mode
	102	const char *fNamePH; // Name for calibration mode
	103	const char *fNamePRF; // Name for calibration mode
3a0f6479	104
	105	Int_t fDetectorPH; // Current detector
	106	Int_t fDetectorCH; // Current detector
	107	Int_t fDetectorPRF; // Current detector
	108
55a288e5	109	// Size of the infos
55a288e5	110
3a0f6479	111	Short_t fNumberBinCharge; // Number of bins for the gain factor
	112	Short_t fNumberBinPRF; // Number of bin for the PRF
	113	Int_t fTimeMax; // Number of time bins
	114	Float_t fPRFRange; // Range PRF
	115	Short_t fDetCha0[3]; // Number of XBins for chamber != 2
	116	Short_t fDetCha2[3]; // Number of Xbins for chamber 2
55a288e5	117
	118	// Some functions
	119
55a288e5	120
00a3834d	121	TArrayI GetEntriesPH(Int_t det, TArrayI* array, Bool_t force);
00a3834d	122	TArrayF GetMeanSquaresPH(Int_t det, TArrayF* array, Bool_t force);
3a0f6479	123
00a3834d	124	TArrayI GetEntriesPRF(Int_t det, TArrayI* array, Bool_t force);
00a3834d	125	TArrayF GetMeanSquaresPRF(Int_t det, TArrayF* array, Bool_t force);
3a0f6479	126
00a3834d	127	TArrayI GetEntriesCH(Int_t det, TArrayI* array, Bool_t force);
3a0f6479	128
3a0f6479	129	// Some basic geometry function
053767a4	130	virtual Int_t GetStack(Int_t d) const;
3a0f6479	131
55a288e5	132	ClassDef(AliTRDCalibraVector,1) // TRD Calibration class
	133
	134	};
	135
	136	#endif
	137
	138