MagneticField.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>
#include <framework/gearbox/Unit.h>
 
namespace Belle2 {
  class MagneticField: public TObject {
  public:
    MagneticField(): TObject() {}
    ~MagneticField()
    {
      for (auto c : m_components) delete c;
      m_components.clear();
    }

    MagneticField(const MagneticField&) = delete;
    MagneticField(MagneticField&& other) noexcept: m_components(std::move(other.m_components)) {}
    ROOT::Math::XYZVector getField(const ROOT::Math::XYZVector& pos) const;
    ROOT::Math::XYZVector getFieldInTesla(const ROOT::Math::XYZVector& pos) const { return getField(pos) / Unit::T; }
    void addComponent(MagneticFieldComponent* component) { m_components.emplace_back(component); }
  private:
    std::vector<MagneticFieldComponent*> m_components;
    ClassDef(MagneticField, 1);
  };
 
  inline ROOT::Math::XYZVector MagneticField::getField(const ROOT::Math::XYZVector& pos) const
  {
    ROOT::Math::XYZVector field;
    for (auto c : m_components) {
      if (c->inside(pos)) {
        // is it an exclusive component? If so return the field
        if (c->isExclusive()) return c->getField(pos);
        // else add it to what we already have
        field += c->getField(pos);
      }
    }
    return field;
  }

} //Belle2 namespace