samplerplacement.h

/* 
Beam Delivery Simulation (BDSIM) Copyright (C) BDSIM Collaboration, 2001 - 2026.
 
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/>.
*/
#ifndef SAMPLERPLACEMENT_H
#define SAMPLERPLACEMENT_H
 
#include <iomanip>
#include <iostream>
#include <list>
#include <string>
#if __cplusplus >= 201703L
#include <variant>
#endif
 
#include "published.h"
 
namespace GMAD
{
  class SamplerPlacement : public Published<SamplerPlacement>
  {
  public:
    std::string name;         
    std::string samplerType;  
    std::string referenceElement; 
    int         referenceElementNumber; 
    double s; 
    double x; 
    double y; 
    double z; 
    double phi;
    double theta;
    double psi;
    double axisX;
    double axisY;
    double axisZ;
    double angle;
    bool   axisAngle;     
 
    std::string apertureModel;
    std::string shape;
    double aper1;
    double aper2;
    double aper3;
    double aper4;
    
    double startAnglePhi;
    double sweepAnglePhi;
    double startAngleTheta;
    double sweepAngleTheta;
    
    // This should be std::list<int> but the parser only supports double for numerical types in a list.
    std::list<double> partID;
    int partIDSetID; 
    
    SamplerPlacement();
    void clear();
    void print()const;
    template <typename T>
    void set_value(std::string property, T value, bool bExit = true);
    void set_value(const std::string& property, Array* value, bool bExit = true);
#if __cplusplus >= 201703L
    std::list<std::variant<bool, int, double, std::string>> get_value_array(const std::string &);
#endif
  private:
    void PublishMembers();
  };
  
  template <typename T>
  void SamplerPlacement::set_value(std::string property, T value, bool bExit)
  {
#ifdef BDSDEBUG
    std::cout << "samplerplacement> Setting value " << std::setw(25) << std::left
              << property << value << std::endl;
#endif
    // member method can throw runtime_error, catch and exit gracefully
    try
      {
        set(this, property, value);
      }
    catch (const std::runtime_error &)
      {
        std::cerr << "Error: samplerplacement> unknown option \"" << property
                  << "\" with value \"" << value << "\"" << std::endl;
        if (property == "partID")
          {
            std::cerr << "Note \"partID\" should be a list {int,int...}" << std::endl;
            if (bExit)
              {exit(1);}
            else
              {std::rethrow_exception(std::current_exception());} // to be caught by python
          }
      }
  }
}
 
#endif