TPC/Base/AliTPCCalROC.h

   1 #ifndef ALITPCCALROC_H
   2 #define ALITPCCALROC_H
   3 /* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
   4  * See cxx source for full Copyright notice                               */
   5
   6 /* $Id: AliTPCCalROC.h -1   */
   7
   8 //////////////////////////////////////////////////
   9 //                                              //
  10 //  TPC calibration base class for one ROC      //
  11 //                                              //
  12 //////////////////////////////////////////////////
  13 #include <TObject.h>
  14 #include <TMath.h>
  15 #include <AliTPCROC.h>
  16 #include "TLinearFitter.h"
  17
  18 class TH1F;
  19 class TH2F;
  20 class TArrayI;
  21 //_____________________________________________________________________________
  22 class AliTPCCalROC : public TNamed {
  23
  24  public:
  25
  26   AliTPCCalROC();
  27   AliTPCCalROC(UInt_t sector);
  28   AliTPCCalROC(const AliTPCCalROC &c);
  29  AliTPCCalROC &operator = (const AliTPCCalROC & param);
  30   virtual           ~AliTPCCalROC();
  31   UInt_t        GetSector() const { return fSector;}
  32   UInt_t        GetNrows() const               { return fNRows;};
  33   UInt_t        GetNchannels()       const     { return fNChannels;};
  34   Bool_t        IsInRange(UInt_t row, UInt_t pad) { return (row<fNRows)? (fkIndexes[row]+pad)<fNChannels:kFALSE;}
  35   UInt_t        GetNPads(UInt_t row)  const     { return (row<fNRows)? AliTPCROC::Instance()->GetNPads(fSector,row):0;};
  36   Float_t      GetValue(UInt_t row, UInt_t pad) const { return ( (row<fNRows) && (fkIndexes[row]+pad)<fNChannels)? fData[fkIndexes[row]+pad]: 0; };
  37   Float_t      GetValue(UInt_t channel) const { return  fData[channel]; };
  38   void         SetValue(UInt_t row, UInt_t pad, Float_t vd) { if ( row<fNRows && (fkIndexes[row]+pad)<fNChannels)fData[fkIndexes[row]+pad]= vd; };
  39   void         SetValue(UInt_t channel, Float_t vd) {fData[channel]= vd; };
  40   virtual void Draw(Option_t* option = "");
  41   //
  42   Bool_t MedianFilter(Int_t deltaRow, Int_t deltaPad, AliTPCCalROC*outlierROC=0, Bool_t doEdge=kTRUE);
  43   Bool_t LTMFilter(Int_t deltaRow, Int_t deltaPad, Float_t fraction, Int_t type,  AliTPCCalROC*outlierROC=0, Bool_t doEdge=kTRUE);
  44   //
  45   // algebra
  46   void Add(Float_t c1); // add c1 to each channel of the ROC
  47   void Multiply(Float_t c1); // multiply each channel of the ROC with c1
  48   void Add(const AliTPCCalROC * roc, Double_t c1 = 1);  // multiply AliTPCCalROC roc by c1 and add each channel to the coresponing channel in the ROC
  49   void Multiply(const AliTPCCalROC * roc);   // multiply each channel of the ROC with the coresponding channel of 'roc'
  50   void Divide(const AliTPCCalROC * roc);   // divide each channel of the ROC by the coresponding value of 'roc'
  51   // statistic
  52   //
  53   Double_t GetMean(AliTPCCalROC *const outlierROC = 0) const;
  54   Double_t GetRMS(AliTPCCalROC *const outlierROC = 0) const;
  55   Double_t GetMedian(AliTPCCalROC *const outlierROC = 0) const;
  56   Double_t GetLTM(Double_t *const sigma=0, Double_t fraction=0.9, AliTPCCalROC *const outlierROC = 0);
  57   TH1F * MakeHisto1D(Float_t min=4, Float_t max=-4, Int_t type=0);
  58   TH2F * MakeHisto2D(Float_t min=4, Float_t max=-4, Int_t type=0);
  59   TH2F * MakeHistoOutliers(Float_t delta=4, Float_t fraction=0.7, Int_t mode=0);
  60
  61   AliTPCCalROC * LocalFit(Int_t rowRadius, Int_t padRadius, AliTPCCalROC* ROCoutliers = 0, Bool_t robust = kFALSE, Double_t chi2Threshold = 5, Double_t robustFraction = 0.7);
  62   void GlobalFit(const AliTPCCalROC* ROCoutliers, Bool_t robust, TVectorD &fitParam, TMatrixD &covMatrix, Float_t & chi2, Int_t fitType = 1, Double_t chi2Threshold = 5, Double_t robustFraction = 0.7, Double_t err=1);
  63
  64   static AliTPCCalROC* CreateGlobalFitCalROC(TVectorD &fitParam, Int_t sector);
  65
  66   static void Test();
  67  protected:
  68
  69   Double_t GetNeighbourhoodValue(TLinearFitter* fitterQ, Int_t row, Int_t pad, Int_t rRadius, Int_t pRadius, AliTPCCalROC *const ROCoutliers, Bool_t robust, Double_t chi2Threshold, Double_t robustFraction);
  70   void GetNeighbourhood(TArrayI* &rowArray, TArrayI* &padArray, Int_t row, Int_t pad, Int_t rRadius, Int_t pRadius);
  71
  72   UInt_t     fSector;          // sector number
  73   UInt_t     fNChannels;       // number of channels
  74   UInt_t     fNRows;           // number of rows
  75   const UInt_t* fkIndexes;      //!indexes
  76   Float_t  *fData;            //[fNChannels] Data
  77   ClassDef(AliTPCCalROC,2)    //  TPC ROC calibration class
  78
  79 };
  80
  81 #endif