python.cc

/* 
Beam Delivery Simulation (BDSIM) Copyright (C) Royal Holloway, 
University of London 2001 - 2023.
 
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 "python.h"
 
#include "parser.h"
 
#include <cstdio>
#include <list>
 
int GMAD::GmadParser_c(char *name)
{
  Parser::Instance(std::string(name));
  return 0;
}
 
int GMAD::GetNElements()
{
  return Parser::Instance()->GetBeamline().size();
}
 
const char* GMAD::GetName(int i)
{
  std::list<Element>::const_iterator it = Parser::Instance()->GetBeamline().begin();
  std::advance(it, i);
  return (it->name).c_str();
}
 
int GMAD::GetType(int i)
{
  std::list<Element>::const_iterator it = Parser::Instance()->GetBeamline().begin();
  std::advance(it, i);
  return static_cast<int>(it->type);
}
 
double GMAD::GetLength(int i)
{
  std::list<Element>::const_iterator it = Parser::Instance()->GetBeamline().begin();
  std::advance(it, i);
  return it->l;
}
 
double GMAD::GetAngle(int i)
{
  std::list<Element>::const_iterator it = Parser::Instance()->GetBeamline().begin();
  std::advance(it, i);
  return it->angle;
}
 
double* GMAD::GetKs(int i)
{
  std::list<Element>::const_iterator it = Parser::Instance()->GetBeamline().begin();
  std::advance(it, i);
  double* result = new double[5];
  result[0] = it->ks;
  result[1] = it->k1;
  result[2] = it->k2;
  result[3] = it->k3;
  result[4] = it->k4;
  return result;
}
 
double GMAD::GetAper1(int i)
{
  std::list<Element>::const_iterator it = Parser::Instance()->GetBeamline().begin();
  std::advance(it, i);
  return it->aper1;
}
 
double GMAD::GetAper2(int i)
{
  std::list<Element>::const_iterator it = Parser::Instance()->GetBeamline().begin();
  std::advance(it, i);
  return it->aper2;
}
 
double GMAD::GetAper3(int i)
{
  std::list<Element>::const_iterator it = Parser::Instance()->GetBeamline().begin();
  std::advance(it, i);
  return it->aper3;
}
 
double GMAD::GetAper4(int i)
{
  std::list<Element>::const_iterator it = Parser::Instance()->GetBeamline().begin();
  std::advance(it, i);
  return it->aper4;
}
 
const char* GMAD::GetApertureType(int i)
{
  std::list<Element>::const_iterator it = Parser::Instance()->GetBeamline().begin();
  std::advance(it, i);
  return (it->apertureType).c_str();
}
 
double GMAD::GetBeampipeThickness(int i)
{
  std::list<Element>::const_iterator it = Parser::Instance()->GetBeamline().begin();
  std::advance(it, i);
  return it->beampipeThickness;
}