BFieldComponent3d.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 BFieldComponent3d : public BFieldComponentAbs {
 
  public:

    BFieldComponent3d() = default;

    virtual ~BFieldComponent3d() 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 setMapSize(int sizeR, int sizePhi, int sizeZ) { m_mapSize[0] = sizeR; m_mapSize[1] = sizePhi; m_mapSize[2] = sizeZ; }

    void setMapRegionZ(double minZ, double maxZ, double offset) { m_mapRegionZ[0] = minZ; m_mapRegionZ[1] = maxZ; m_mapOffset = offset; }

    void setMapRegionR(double minR, double maxR) { m_mapRegionR[0] = minR; m_mapRegionR[1] = maxR; }

    void setGridPitch(double pitchR, double pitchPhi, double pitchZ) { m_gridPitch[0] = pitchR; m_gridPitch[1] = pitchPhi; m_gridPitch[2] = pitchZ; }

    void setExcludeRegionZ(double minZ, double maxZ) { m_exRegionZ[0] = minZ + m_mapOffset; m_exRegionZ[1] = maxZ + m_mapOffset; }

    void setExcludeRegionR(double minR, double maxR) { m_exRegionR[0] = minR; m_exRegionR[1] = maxR; }

    void setErrorRegionR(double minR = -1., double maxR = -1., double errBr = 0.0, double errBphi = 0.0, double errBz = 0.0)
    { m_errRegionR[0] = minR; m_errRegionR[1] = maxR; m_errB[0] = errBr; m_errB[1] = errBphi; m_errB[2] = errBz; }

    void mirrorPhi(bool mirrorPhi = 0.) { m_mirrorPhi = mirrorPhi; }

    void doInterpolation(bool interpolate = true) { m_interpolate = interpolate; }

    void enableCoordinate(const std::string& mapEnable = "rphiz") { m_mapEnable = mapEnable; }
  protected:
 
  private:

    ROOT::Math::XYZVector interpolate(unsigned int ir, unsigned int iphi, unsigned int iz, double wr, double wphi, double wz) const;

    std::string m_mapFilename{""};
    std::vector<ROOT::Math::XYZVector> m_bmap;
    std::string m_mapEnable{"rphiz"};
    bool m_interpolate{true};
    int m_mapSize[3] = {0, 0, 0};
    double m_mapRegionZ[2] {0};
    double m_mapOffset{0};
    double m_mapRegionR[2] {0};
    double m_gridPitch[3] {0};
    double m_igridPitch[3] {0};
    double m_exRegionZ[2] {0};
    double m_exRegionR[2] {0};
    bool m_exRegion{true};
    bool m_mirrorPhi{true};
    double m_errRegionR[2] {0};
    double m_errB[3] {0};
  };

} //end of namespace Belle2