CbmLitFieldFitter.h

Go to the documentation of this file.

 
#ifndef CBMLITFIELDFITTER_H_
#define CBMLITFIELDFITTER_H_
 
#include "../../parallel/LitFieldSlice.h"
#include <vector>
class FairField;
 
class CbmLitPolynom {
public:
  CbmLitPolynom() {}
 
  virtual ~CbmLitPolynom() {}
 
  virtual double Calculate(double x, double y, double c[]) const = 0;
 
  virtual unsigned int GetNofCoefficients() const = 0;
};
 
class CbmLitFieldFitter {
public:
  CbmLitFieldFitter(unsigned int polynomDegree);
 
  virtual ~CbmLitFieldFitter();
 
  template<class T>
  void FitSlice(float Z, lit::parallel::LitFieldSlice<T>& slice);
 
  void FitSliceScal(float Z, lit::parallel::LitFieldSlice<fscal>& slice);
 
  void FitSliceVec(float Z, lit::parallel::LitFieldSlice<fvec>& slice);
 
  /* Setters */
  void SetXangle(double xangle) { fXangle = xangle; }
  void SetYangle(double yangle) { fYangle = yangle; }
  void SetNofBinsX(int nofBinsX) { fNofBinsX = nofBinsX; }
  void SetNofBinsY(int nofBinsY) { fNofBinsY = nofBinsY; }
  void SetUseEllipseAcc(bool useEllipseAcc) { fUseEllipseAcc = useEllipseAcc; }
 
private:
  void FitSlice(double Z,
                std::vector<double>& parBx,
                std::vector<double>& parBy,
                std::vector<double>& parBz);
 
  void FitSlice(const std::vector<double>& x,
                const std::vector<double>& y,
                const std::vector<double>& z,
                std::vector<double>& par);
 
  void FitSliceMy(double Z,
                  std::vector<double>& parBx,
                  std::vector<double>& parBy,
                  std::vector<double>& parBz);
 
  const CbmLitPolynom* GetPolynom() const { return fPolynom; }
 
private:
  FairField* fField;  // Magnetic field
 
  double fXangle;  // Acceptance angle for X [grad]
  double fYangle;  // Acceptance angle for Y [grad]
 
  int fNofBinsX;  // Number of bins for X [cm]
  int fNofBinsY;  // Number of bins for Y [cm]
 
  bool
    fUseEllipseAcc;  // If true than only values inside an ellipse will be fitted
 
  unsigned int
    fPolynomDegree;  // Degree of the polynomial to be used for the field approximation
  CbmLitPolynom* fPolynom;  // Polynomial to be used for field approximation
};
 
#endif /* CBMLITFIELDFITTER_H_ */