Parameterization.h

Go to the documentation of this file.

/*
Copyright (C) 2002-2021 CERN for the benefit of the ATLAS collaboration
*/
 
 
#ifndef AFP_GLOBRECO_PARAMETRIZATION_H
#define AFP_GLOBRECO_PARAMETRIZATION_H
 
// STL include(s):
#include <memory>
#include <array>
#include <string>
#include <iostream>
#include <fstream>
#include <sstream>
 
// ROOT include(s):
#include <TF1.h>
 
namespace AFP {
 
  class Parameterization {
 
    class Equation {
      public:
        Equation(double energy, std::vector<std::unique_ptr<TF1>> polynomial) : m_energy(energy), m_polynomial(std::move(polynomial)) {}
 
        double evaluate(double x, double y, double z, double sx, double sy, double E) const;
 
        const TF1* getPolynomial(int iPoly) const { return m_polynomial.at(iPoly).get(); };
 
      private:
        double m_energy;
 
        std::vector<std::unique_ptr<TF1>> m_polynomial;
    };
 
    public:
      Parameterization(std::string_view filename);
 
      double x(double x0, double y0, double z0, double sx0, double sy0, double E) const;
 
      double y(double x0, double y0, double z0, double sx0, double sy0, double E) const;
 
      double sx(double x0, double y0, double z0, double sx0, double sy0, double E) const;
 
      double sy(double x0, double y0, double z0, double sx0, double sy0, double E) const;
 
      double parametrizationPosition() const { return m_parametrizationPosition; }
 
      double energy() const { return m_energy; }
 
      const Equation* getEquation(int iEqu) const { return m_equation.at(iEqu).get(); }
 
      bool yIsUsed() const { return m_yIsUsed; }
 
    private:
      static constexpr int m_numberOfPolynomials = 8;
 
      static constexpr int m_numberOfEquations = 4;
 
      std::array<double, 2> m_slopeInitIP;
 
      std::array<double, 3> m_positionInitIP;
 
      double m_energy;
 
      double m_parametrizationPosition;
 
      std::vector<std::unique_ptr<Equation>> m_equation;
 
      bool m_yIsUsed;
  };
}
 
#endif // AFP_GLOBRECO_PARAMETRIZATION_H