CbmLitFieldApproximationQa.h

Go to the documentation of this file.

 
#ifndef CBMLITFIELDAPPROXIMATIONQA_H_
#define CBMLITFIELDAPPROXIMATIONQA_H_
 
#include "FairTask.h"
#include <string>
#include <vector>
 
class FairField;
class CbmLitFieldFitter;
class CbmLitFieldGridCreator;
class CbmHistManager;
 
using std::string;
using std::vector;
 
class CbmLitFieldApproximationQa : public FairTask {
public:
  CbmLitFieldApproximationQa();
 
  virtual ~CbmLitFieldApproximationQa();
 
  virtual InitStatus Init();
 
  virtual void Exec(Option_t* opt);
 
  virtual void Finish();
 
  /* Setters */
  void SetSliceZPosition(const vector<Double_t>& zPos) {
    fZSlicePosition = zPos;
  }
  void SetPolynomDegrees(const vector<UInt_t>& degrees) {
    fPolynomDegrees = degrees;
  }
  void IsFixedBounds(Bool_t fixedBounds) { fFixedBounds = fixedBounds; }
  void SetAcceptanceAngleX(double xangle) { fAcceptanceAngleX = xangle; }
  void SetAcceptanceAngleY(double yangle) { fAcceptanceAngleY = yangle; }
  void SetNofBinsX(int nofBinsX) { fNofBinsX = nofBinsX; }
  void SetNofBinsY(int nofBinsY) { fNofBinsY = nofBinsY; }
  void SetUseEllipseAcc(bool useEllipseAcc) { fUseEllipseAcc = useEllipseAcc; }
  void SetOutputDir(const string& dir) { fOutputDir = dir; }
  void SetPolynomDegreeIndex(unsigned int degreeIndex) {
    fPolynomDegreeIndex = degreeIndex;
  }
 
private:
  void CreateHistos();
 
  void CreateFieldHistos();
 
  void CreateFitterHistos();
 
  void CreateGridHistos();
 
  void FillBHistos();
 
  void FillFieldApproximationHistos();
 
  void FillGridCreatorHistos();
 
  // Pointer to the magnetic field map
  FairField* fField;
 
  // Number of slices along Z for field approximation
  Int_t fNofSlices;
  // Slice is defined as rectangle at a certain Z position
  // with upper left corner coordinate equals to [-X, -Y]
  // and lower right corner coordinate equals to [X, Y].
  // Z, X, Y coordinates for each slice.
  // [slice number]
  vector<Double_t> fZSlicePosition;  // Z position of the slice
  vector<Double_t> fXSlicePosition;  // X coordinate
  vector<Double_t> fYSlicePosition;  // Y coordinate
 
  // Output directory for images
  string fOutputDir;
 
  //
  // Field approximation variables
  //
  // Acceptance angle for X
  Double_t fAcceptanceAngleX;
  // Acceptance angle for Y
  Double_t fAcceptanceAngleY;
  // Number of bins for X
  Int_t fNofBinsX;
  // Number of bins for Y
  Int_t fNofBinsY;
  // If true than only values inside an ellipse will be used for field approximation
  Bool_t fUseEllipseAcc;
  // Degree of the polynom
  UInt_t fPolynomDegreeIndex;
  // Number of polynoms for tests
  UInt_t fNofPolynoms;
  // Array with polynom degrees to be analyzed
  vector<UInt_t> fPolynomDegrees;
  // Field fitter tool for each polynom degree
  vector<CbmLitFieldFitter*> fFitter;
 
  //
  // Field grid
  //
  // Field grid creator tool
  CbmLitFieldGridCreator* fGridCreator;
 
  CbmHistManager* fHM;  // Histogram manager
 
  Bool_t fFixedBounds;  // Fixed bounds for error histograms
 
  CbmLitFieldApproximationQa(const CbmLitFieldApproximationQa&);
  CbmLitFieldApproximationQa& operator=(const CbmLitFieldApproximationQa&);
 
  ClassDef(CbmLitFieldApproximationQa, 1);
};
 
#endif /* CBMLITFIELDAPPROXIMATIONQA_H_ */