TPC/AliTPCkalmanFit.h

   1 #ifndef ALITPCKALMANFIT_H
   2 #define ALITPCKALMANFIT_H
   3
   4
   5
   6 #include "TNamed.h"
   7 #include "TMatrixD.h"
   8 #include "TVectorD.h"
   9 #include "TObjArray.h"
  10 class TTreeSRedirector;
  11 class AliTrackPointArray;
  12 class AliTrackPoint;
  13 class TFormula;
  14 class TBits;
  15 class THnSparse;
  16 //
  17
  18 class AliTPCkalmanFit: public TNamed{
  19 public:
  20   AliTPCkalmanFit();
  21   void Init();
  22   void InitTransformation();
  23   void Add(const AliTPCkalmanFit * kalman);
  24
  25   void  AddCalibration(AliTPCTransformation * calib);
  26   AliTPCTransformation * GetTransformation(Int_t i){return (fCalibration)? (AliTPCTransformation *)fCalibration->At(i):0;}
  27   //
  28   void SetStatus(const char * name, Bool_t setOn, Bool_t isOr=kTRUE);
  29   //
  30   void FitTrackLinear(AliTrackPointArray& points,  TTreeSRedirector *debug=0, Float_t scalingRMSY=1., Float_t scalingRMSZ=1.);
  31   void DumpTrackLinear(AliTrackPointArray& points, TTreeSRedirector *debug);
  32   void UpdateLinear(AliTrackPoint &point, TTreeSRedirector *debug=0);
  33
  34   void Propagate(TTreeSRedirector *debug=0);
  35   void PropagateTime(Int_t time);
  36   void AddCovariance(const char * varName, Double_t sigma);
  37   void Update(const AliTPCkalmanFit * kalman);
  38
  39   static AliTrackPointArray * SortPoints(AliTrackPointArray &points);
  40   static AliTrackPointArray * MakePointArrayLinear(Double_t alpha, Double_t y0, Double_t z0, Double_t ky, Double_t kz, Double_t err=0.02);
  41   void  ApplyCalibration(AliTrackPointArray *array, Double_t csign);
  42   Bool_t  CheckCovariance(TMatrixD &covar, Float_t maxEl);
  43
  44   Bool_t DumpCorelation(Double_t threshold, const char *mask0=0, const char *mask1=0);
  45   Bool_t DumpCalib(const char *mask=0, Float_t correlationCut=-1);
  46   //
  47   Double_t GetTPCDeltaXYZ(Int_t coord, Int_t volID, Double_t x, Double_t y, Double_t z);
  48   static Double_t SGetTPCDeltaXYZ(Int_t coord, Int_t volID, Double_t x, Double_t y, Double_t z);
  49   AliTPCkalmanFit *Test(Int_t ntracks);
  50  public:
  51   //
  52   // Calibration parameters
  53   //
  54   TObjArray *fCalibration;  // array of calibrations
  55   TMatrixD  *fCalibParam;   // calibration parameters
  56   TMatrixD  *fCalibCovar;   // calibration parameters
  57   //
  58   // Linear track
  59   //
  60   TMatrixD  *fLinearParam;      // linear parameters
  61   TMatrixD  *fLinearCovar;      // linear covariance
  62   THnSparse *fLinearTrackDelta[12];   // linear tracks matching residuals - delta
  63   THnSparse *fLinearTrackPull[12];    // linear tracks matching residuals  - pull
  64   //
  65   //
  66   //
  67   Int_t      fLastTimeStamp; // last time stamp - used for propagation of parameters
  68   //static AliTPCkalmanFit* Instance();
  69   void SetInstance(AliTPCkalmanFit*param){fgInstance = param;}
  70   static AliTPCkalmanFit*   fgInstance; //! Instance of this class (singleton implementation)
  71  private:
  72   Double_t   fCurrentAlpha; //! current rotation frame
  73   Double_t   fCA;           //! cosine of current angle
  74   Double_t   fSA;           //! sinus of current angle
  75   AliTPCkalmanFit&  operator=(const AliTPCkalmanFit&);// not implemented
  76   AliTPCkalmanFit(const AliTPCkalmanFit&):TNamed(),
  77                                   fCalibration(0),
  78                                   fCalibParam(0),
  79                                   fCalibCovar(0),
  80                                   fLinearParam(0),
  81                                   fLinearCovar(0),
  82                                   fLastTimeStamp(-1),
  83                                   fCurrentAlpha(0),  //! current rotation frame
  84                                   fCA(0),     //! cosine of current angle
  85                                   fSA(0)     //! sinus of current angle
  86
  87 {;} //not implemented
  88   ClassDef(AliTPCkalmanFit,3);
  89 };
  90
  91
  92
  93 #endif
  94