MagneticFieldComponentConstant.h

/**************************************************************************
 * basf2 (Belle II Analysis Software Framework)                           *
 * Author: The Belle II Collaboration                                     *
 *                                                                        *
 * See git log for contributors and copyright holders.                    *
 * This file is licensed under LGPL-3.0, see LICENSE.md.                  *
 **************************************************************************/
 
#pragma once
 
#include <framework/dbobjects/MagneticFieldComponent.h>
 
namespace Belle2 {
  class MagneticFieldComponentConstant final: public MagneticFieldComponent {
  public:
    MagneticFieldComponentConstant(): MagneticFieldComponent(false) {};
    MagneticFieldComponentConstant(const ROOT::Math::XYZVector& field, float minR, float maxR, float minZ,
                                   float maxZ): MagneticFieldComponent(false),
      m_field(field), m_minR(minR), m_maxR(maxR), m_minZ(minZ), m_maxZ(maxZ) {}
    explicit MagneticFieldComponentConstant(const ROOT::Math::XYZVector& field): MagneticFieldComponentConstant(field, 0,
          std::numeric_limits<float>::infinity(), -std::numeric_limits<float>::infinity(),
          std::numeric_limits<float>::infinity()) {}
    bool inside(const ROOT::Math::XYZVector& pos) const override
    {
      const float r = pos.Rho();
      const float z = pos.Z();
      return (r >= m_minR && r <= m_maxR && z >= m_minZ && z <= m_maxZ);
    }
    ROOT::Math::XYZVector getField(__attribute((unused)) const ROOT::Math::XYZVector& pos) const override { return m_field; }
  private:
    ROOT::Math::XYZVector m_field;
    float m_minR{0};
    float m_maxR{0};
    float m_minZ{0};
    float m_maxZ{0};
    ClassDefOverride(MagneticFieldComponentConstant, 2);
  };
}; // Belle2 namespace