CbmLitFieldFitter.h

Go to the documentation of this file.

/* Copyright (C) 2009-2012 GSI/JINR-LIT, Darmstadt/Dubna
   SPDX-License-Identifier: GPL-3.0-only
   Authors: Andrey Lebedev [committer] */
 
#ifndef CBMLITFIELDFITTER_H_
#define CBMLITFIELDFITTER_H_
 
#include "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)
  {
    std::vector<double> aparBx, aparBy, aparBz;
    FitSlice(Z, aparBx, aparBy, aparBz);
 
    std::vector<T> aparBxT, aparByT, aparBzT;
    aparBxT.assign(aparBx.begin(), aparBx.end());
    aparByT.assign(aparBy.begin(), aparBy.end());
    aparBzT.assign(aparBz.begin(), aparBz.end());
 
    slice.SetZ(Z);
    slice.SetCoefficients(aparBxT, aparByT, aparBzT);
  }
 
  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_ */