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