BFieldMap.h

/**************************************************************************
 * BASF2 (Belle Analysis Framework 2)                                     *
 * Copyright(C) 2010-2011  Belle II Collaboration                         *
 *                                                                        *
 * Author: The Belle II Collaboration                                     *
 * Contributors: Andreas Moll                                             *
 *                                                                        *
 * This software is provided "as is" without any warranty.                *
 **************************************************************************/
 
#ifndef BFIELDMAP_H_
#define BFIELDMAP_H_
 
#include <geometry/bfieldmap/BFieldComponentAbs.h>
 
#include <framework/geometry/B2Vector3.h>
#include <framework/logging/Logger.h>
 
#include <list>
 
namespace Belle2 {
  class BFieldMap {
 
  public:
 
    static BFieldMap& Instance();
 
  private:
    B2Vector3D getBField(const B2Vector3D& point) const;
  public:
 
    template<class BFIELDCOMP> BFIELDCOMP& addBFieldComponent();
 
    void initialize();
 
    void clear();
 
  protected:
 
    std::list<BFieldComponentAbs*> m_components; 
  private:
 
    bool m_isMapInitialized; 
    BFieldMap();
 
    BFieldMap(const BFieldMap&);
 
    BFieldMap& operator=(const BFieldMap&);
 
    virtual ~BFieldMap();
 
    friend struct std::default_delete<BFieldMap>;
    friend class BFieldFrameworkInterface;
  };
 
 
  //------------------------------------------------------
  //       Implementation of template based methods
  //------------------------------------------------------
  template<class BFIELDCOMP> BFIELDCOMP&  BFieldMap::addBFieldComponent()
  {
    BFIELDCOMP* newComponent = new BFIELDCOMP;
    m_components.push_back(newComponent);
    return *newComponent;
  }
 
 
  inline B2Vector3D BFieldMap::getBField(const B2Vector3D& point) const
  {
    B2Vector3D magFieldVec(0.0, 0.0, 0.0);
    //Check that the point makes sense
    if (std::isnan(point.X()) || std::isnan(point.Y()) || std::isnan(point.Z())) {
      B2ERROR("Bfield requested for a position containing NaN, returning field 0");
      return magFieldVec;
    }
    //Loop over all magnetic field components and add their magnetic field vectors
    for (const BFieldComponentAbs* comp : m_components) {
      magFieldVec += comp->calculate(point);
    }
    return magFieldVec;
  }
 
 
} //end of namespace Belle2
 
#endif /* BFIELDMAP_H_ */