BFieldComponentQuad.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 <geometry/bfieldmap/BFieldComponentAbs.h>
 
#include <string>
#include <vector>
 
namespace Belle2 {
  class BFieldComponentQuad : public BFieldComponentAbs {
 
  public:
    struct range_t {
      double r0; 
      double r1; 
    };
    typedef std::vector<range_t> ranges_t;
 
    struct ApertPoint {
      double s{0}; 
      double r{0}; 
    };
 
    struct ParamPoint3 {
      double s{0};   
      double L{0};   
      double mxx{0}; 
      double mxy{0}; 
      double mx0{0}; 
      double myx{0}; 
      double myy{0}; 
      double my0{0}; 
      inline double getBx(double x, double y) const {return x * mxx + y * mxy + mx0;}
      inline double getBy(double x, double y) const {return x * myx + y * myy + my0;}
 
      ParamPoint3& operator +=(const ParamPoint3& t)
      {
        mxx += t.mxx;
        mxy += t.mxy;
        mx0 += t.mx0;
        myx += t.myx;
        myy += t.myy;
        my0 += t.my0;
        return *this;
      }
    };
 
    struct irange_t {
      short int i0{0}; 
      short int i1{0}; 
    };
 
    BFieldComponentQuad() = default;
 
    virtual ~BFieldComponentQuad() = default;
 
    virtual void initialize() override;
 
    virtual ROOT::Math::XYZVector calculate(const ROOT::Math::XYZVector& point) const override;
 
    double getApertureHER(double s) const;
 
    double getApertureLER(double s) const;
 
    void setMapFilename(const std::string& filenameHER, const std::string& filenameLER, const std::string& filenameHERleak) { m_mapFilenameHER = filenameHER; m_mapFilenameLER = filenameLER; m_mapFilenameHERleak = filenameHERleak; }
 
    void setApertFilename(const std::string& filenameHER, const std::string& filenameLER) { m_apertFilenameHER = filenameHER; m_apertFilenameLER = filenameLER; }
 
    void setMapSize(int sizeHER, int sizeLER, int sizeHERleak) { m_mapSizeHER = sizeHER; m_mapSizeLER = sizeLER; m_mapSizeHERleak = sizeHERleak;}
 
    void setApertSize(int sizeHER, int sizeLER) { m_apertSizeHER = sizeHER; m_apertSizeLER = sizeLER; }
 
  private:
    inline int getRange(double a, const ranges_t& b) const;
 
    inline double getAperture(double s, std::vector<ApertPoint>::const_iterator hint) const;
 
    std::string m_mapFilenameHER{""};
    std::string m_mapFilenameLER{""};
    std::string m_mapFilenameHERleak{""};
    std::string m_apertFilenameHER{""};
    std::string m_apertFilenameLER{""};
    int m_mapSizeHER{0};
    int m_mapSizeLER{0};
    int m_mapSizeHERleak{0};
    int m_apertSizeHER{0};
    int m_apertSizeLER{0};
    double m_maxr2{0};
 
    ranges_t m_ranges_her; 
    ranges_t m_ranges_ler; 
    std::vector<ApertPoint> m_ah;
    std::vector<ApertPoint> m_al;
 
    std::vector<ParamPoint3> m_h3;
    std::vector<ParamPoint3 >m_l3;
 
    std::vector<std::vector<ParamPoint3>::const_iterator> m_offset_pp_her;
    std::vector<std::vector<ParamPoint3>::const_iterator> m_offset_pp_ler;
    std::vector<std::vector<ApertPoint>::const_iterator> m_offset_ap_her;
    std::vector<std::vector<ApertPoint>::const_iterator> m_offset_ap_ler;
  };
 
} //end of namespace Belle2