BFieldComponentKlm1.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>
 
namespace Belle2 {

  class BFieldComponentKlm1 : public BFieldComponentAbs {
 
  public:

    struct BFieldPoint {
      double r;
      double z;
    };

    BFieldComponentKlm1() = default;

    virtual ~BFieldComponentKlm1() override = default;

    virtual void initialize() override;

    virtual ROOT::Math::XYZVector calculate(const ROOT::Math::XYZVector& point) const override;
    void setMapFilename(const std::string& filename) { m_mapFilename = filename; };

    void setNLayers(int b, int e) {m_nBarrelLayers = b; m_nEndcapLayers = e;}

    void setBarrelRegion(double minR, double maxZ, double offset) { m_barrelRMin = minR; m_barrelZMax = maxZ; m_mapOffset = offset; }

    void setEndcapRegion(double minR, double minZ) { m_endcapRMin = minR; m_endcapZMin = minZ;}

    void setLayerParam(double bgapl0, double bironth, double egap, double dl)
    {
      m_barrelGapHeightLayer0 = bgapl0;  m_barrelIronThickness = bironth;
      m_endcapGapHeight = egap; m_dLayer = dl;
    }
 
  private:
    double m_cospi8{cos(M_PI / 8)};
    double m_cos3pi8{cos(3 * M_PI / 8)};
    double m_cospi4{cos(M_PI / 4)};

    std::string m_mapFilename{""};
    double m_mapOffset{0};

    double m_barrelRMin{0};
    double m_barrelZMax{0};
    double m_endcapRMin{0};
    double m_endcapZMin{0};

    int m_nBarrelLayers{0};
    int m_nEndcapLayers{0};

    double m_barrelGapHeightLayer0{0};
    double m_endcapGapHeight{0};
    double m_dLayer{0};
    double m_barrelIronThickness{0};

    double m_barrelZBreakpoint[15] {0};
    double m_barrelRBreakpoint[15] {0};
    double m_barrelFieldZSlope1[15] {0};
    double m_barrelFieldZIntercept1[15] {0};
    double m_barrelFieldZSlope2[15] {0};
    double m_barrelFieldZIntercept2[15] {0};
    double m_barrelFieldRSlope1[15] {0};
    double m_barrelFieldRIntercept1[15] {0};
    double m_barrelFieldRSlope2[15] {0};
    double m_barrelFieldRIntercept2[15] {0};

    double m_endcapZBreakpoint[2][15][5] {{{0}}};
    double m_endcapRBreakpoint[2][15][5] {{{0}}};

    double m_endcapFieldZSlope[2][15][5] {{{0}}};
    double m_endcapFieldZIntercept[2][15][5] {{{0}}};
    double m_endcapFieldRSlope[2][15][5] {{{0}}};
    double m_endcapFieldRIntercept[2][15][5] {{{0}}};
  };

} //end of namespace Belle2