BDSFieldFormat.cc

/* 
Beam Delivery Simulation (BDSIM) Copyright (C) Royal Holloway, 
University of London 2001 - 2022.
 
This file is part of BDSIM.
 
BDSIM is free software: you can redistribute it and/or modify 
it under the terms of the GNU General Public License as published 
by the Free Software Foundation version 3 of the License.
 
BDSIM is distributed in the hope that it will be useful, but 
WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
GNU General Public License for more details.
 
You should have received a copy of the GNU General Public License
along with BDSIM.  If not, see <http://www.gnu.org/licenses/>.
*/
#include "BDSFieldFormat.hh"
#include "BDSDebug.hh"
#include "BDSException.hh"
#include "BDSUtilities.hh"
 
#include "globals.hh"
#include "G4String.hh"
 
#include <map>
#include <string>
 
// dictionary for BDSFieldFormat for reflexivity
template<>
std::map<BDSFieldFormat, std::string>* BDSFieldFormat::dictionary =
  new std::map<BDSFieldFormat, std::string> ({
      {BDSFieldFormat::none,      "none"},
      {BDSFieldFormat::bdsim1d,   "bdsim1d"},
      {BDSFieldFormat::bdsim2d,   "bdsim2d"},
      {BDSFieldFormat::bdsim3d,   "bdsim3d"},
      {BDSFieldFormat::bdsim4d,   "bdsim4d"},
      {BDSFieldFormat::poisson2d, "poisson2d"},
      {BDSFieldFormat::poisson2dquad, "poisson2dquad"},
      {BDSFieldFormat::poisson2ddipole, "poisson2ddipole"}
});     
 
BDSFieldFormat BDS::DetermineFieldFormat(G4String bFormat)
{
  std::map<G4String, BDSFieldFormat> formats;
  formats["none"]          = BDSFieldFormat::none;
  formats["bdsim1d"]       = BDSFieldFormat::bdsim1d;
  formats["bdsim2d"]       = BDSFieldFormat::bdsim2d;
  formats["bdsim3d"]       = BDSFieldFormat::bdsim3d;
  formats["bdsim4d"]       = BDSFieldFormat::bdsim4d;
  formats["poisson2d"]     = BDSFieldFormat::poisson2d;
  formats["poisson2dquad"] = BDSFieldFormat::poisson2dquad;
  formats["poisson2ddipole"] = BDSFieldFormat::poisson2ddipole;
 
  bFormat = BDS::LowerCase(bFormat);
 
  auto result = formats.find(bFormat);
  if (result == formats.end())
    {
      // it's not a valid key
      G4cerr << __METHOD_NAME__ << "\"" << bFormat
             << "\" is not a valid field format" << G4endl;
      G4cout << "Available field formats are:" << G4endl;
      for (auto it : formats)
        {G4cout << "\"" << it.first << "\"" << G4endl;}
      throw BDSException(__METHOD_NAME__, "can't determine field type");
    }
 
#ifdef BDSDEBUG
  G4cout << __METHOD_NAME__ << "determined field format to be " << result->second << G4endl;
#endif
  return result->second;
}
 
G4int BDS::NDimensionsOfFieldFormat(const BDSFieldFormat& ff)
{
  G4int result = 0;
  switch (ff.underlying())
    {
      case BDSFieldFormat::bdsim1d:
        {result = 1; break;}
      case BDSFieldFormat::bdsim2d:
      case BDSFieldFormat::poisson2d:
      case BDSFieldFormat::poisson2dquad:
      case BDSFieldFormat::poisson2ddipole:
        {result = 2; break;}
      case BDSFieldFormat::bdsim3d:
        {result = 3; break;}
      case BDSFieldFormat::bdsim4d:
        {result = 4; break;}
      case BDSFieldFormat::none:
      default:
        {result = 0; break;}
    }
  return result;
}