BDSFieldLoaderPoisson.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 "BDSArray2DCoords.hh"
#include "BDSDebug.hh"
#include "BDSException.hh"
#include "BDSFieldLoaderPoisson.hh"
#include "BDSFieldValue.hh"
 
#include "globals.hh" // geant4 types / globals
 
#include "CLHEP/Units/SystemOfUnits.h"
 
#include <algorithm>
#include <fstream>
#include <iostream>
#include <regex>
#include <sstream>
#include <string>
#include <vector>
 
#ifdef USE_GZSTREAM
#include "src-external/gzstream/gzstream.h"
#endif
 
template <class T>
BDSFieldLoaderPoisson<T>::BDSFieldLoaderPoisson()
{;}
 
template <class T>
BDSFieldLoaderPoisson<T>::~BDSFieldLoaderPoisson()
{;}
 
template <class T>
BDSArray2DCoords* BDSFieldLoaderPoisson<T>::LoadMag2D(G4String fileName)
{
  file.open(fileName);
 
  // test if file is valid
#ifdef USE_GZSTREAM
  bool validFile = file.rdbuf()->is_open();
#else
  bool validFile = file.is_open();
#endif
  if (!validFile)
    {throw BDSException(__METHOD_NAME__, "Invalid file name or no such file named \"" + fileName + "\"");}
  else
    {G4cout << "Loading \"" << fileName << "\"" << G4endl;}
 
  // Pointer to where result will be stored.  Can't be constructed until we know
  // the dimensions.
  BDSArray2DCoords* result = nullptr;
  G4int       nX = 1;
  G4int       nY = 1;
  G4double  xMin = 0;
  G4double  yMin = 0;
  G4double  xMax = 0;
  G4double  yMax = 0;
  G4int nColumns = 0;
  std::vector<G4String> keys;
  std::vector<G4String> units;
 
  // temporary variables
  G4bool intoData     = false;
  G4bool pastNStep    = false;
  G4bool dataFinished = false;
  std::string line;
  G4int indX = 0;
  G4int indY = 0;
  
  while (std::getline(file, line))
    {// read a line only if it's not a blank one
 
      // Skip a line if it's only whitespace
      if (std::all_of(line.begin(), line.end(), isspace))
        {continue;}
 
      if (intoData)
        {
          if (dataFinished)
            {continue;} // just keep skipping lines after the appropriate amount of data is loaded
          
          // General data entry
          std::istringstream liness(line);
          G4double value = 0;
          std::vector<G4float> lineData(nColumns); // reserve to avoid copying as it expands
          for (G4int i = 0; i < nColumns; ++i)
            {
              liness >> value;
              lineData[i] = value;
            }
          // Construct field value
          BDSFieldValue fv = BDSFieldValue(lineData[2], lineData[3], 0);
          // we could use the gradients here, but we don't
          
          // Copy into array.
          (*result)(indX,indY) = fv;
 
          indX++;
          if (indX == nX)
            {// we've completed one set of x
              indX = 0;
              indY++;
            }
          if (indY == nY)
            {// we've completed one set of y
              indY = 0;
              dataFinished = true;
            }
 
          continue;
        }
 
      // else try and parser what must be header bits.
      
      // Min
      std::regex min("^.*?\\(Xmin,Ymin\\)\\s*=\\s*", std::regex_constants::icase);
      if (std::regex_search(line, min))
        {// found line beginning with '(Xmin,YMin)'
          std::regex minValues("\\(([0-9eE.+-]+),([0-9eE.+-]+)\\)");
          std::smatch match;
          std::regex_search(line, match, minValues);
          xMin = std::stod(match[1])*CLHEP::cm;
          yMin = std::stod(match[2])*CLHEP::cm;
          continue;
        }
 
      // Max
      std::regex max("^.*?\\(Xmax,Ymax\\)\\s*=\\s*", std::regex_constants::icase);
      if (std::regex_search(line, max))
        {// found line beginning with '(Xmax,YMax)'
          std::regex maxValues("\\(([0-9eE.+-]+),([0-9eE.+-]+)\\)");
          std::smatch match;
          std::regex_search(line, match, maxValues);
          xMax = std::stod(match[1])*CLHEP::cm;
          yMax = std::stod(match[2])*CLHEP::cm;
          continue;
        }
 
      // Steps
      std::regex incr("\\s*X\\s*and\\s*Y", std::regex_constants::icase);
      if (std::regex_search(line, incr))
        {//found line for number of increments
          std::regex nStepValues(":\\s*([0-9eE.+-]+)\\s*([0-9eE.+-]+)");
          std::smatch match;
          std::regex_search(line, match, nStepValues);
          nX = std::stoi(match[1]) + 1;
          nY = std::stoi(match[2]) + 1; // number of increments, so number of points is +1
          pastNStep = true;
          continue;
        }
      
      // Columns
      std::regex noNumbers("^(\\D)*$");
      G4bool noNumbersR = std::regex_search(line, noNumbers);
      G4bool hasParenthesis = (line.find("(") != std::string::npos) || (line.find(")") != std::string::npos);
      G4bool hasNoParenthesis = !hasParenthesis;
      if (pastNStep && noNumbersR && hasNoParenthesis)
        {
          std::istringstream liness(line);
          for (G4String key; liness >> key;)
            {keys.push_back(key);}
          nColumns = G4int(std::distance(keys.begin(), keys.end()));
          continue;
        }
      
      // Units
      if (pastNStep && noNumbersR && hasParenthesis)
        {
          // match text inside (XXX) parenthesis, ie strip them off
          std::regex unitsRE("[\\(](\\w+\\-*\\/*\\w*)[\\)]");
          std::smatch match;
          std::regex_search(line, match, unitsRE);
          for (const auto& u : match)
            {units.push_back(G4String(u));}
          result = new BDSArray2DCoords(nX, nY, xMin, xMax, yMin, yMax);
          intoData = true;
        }
      //nothing heap allocated so safe to let all go out of scope
    }
 
  file.close();
 
  if (!result)
    {throw BDSException(__METHOD_NAME__, "Invalid file contents - no column header found");}
  
  return result;
}
 
 
template class BDSFieldLoaderPoisson<std::ifstream>;
 
#ifdef USE_GZSTREAM
template class BDSFieldLoaderPoisson<igzstream>;
#endif