sym_table.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 "sym_table.h"
 
#include <cstdlib>
#include <iostream>
 
#include "array.h"
 
using namespace GMAD;
 
Symtab::Symtab(std::string s):
  name(s),
  is_reserved(false),
  type(symtabtype::NUMBER),
  funcptr(nullptr),
  value(0.0)
{
}
 
void Symtab::Set(Array* a)
{
  array.clear();
  for(unsigned int i=0;i<a->data.size();i++)
    array.push_back(a->data[i]);
  type = symtabtype::ARRAY;
}
 
void Symtab::Set(std::string a)
{
  str = a;
  type = symtabtype::STRING;
}
 
 
void Symtab::Set(double a, bool reserved)
{
  value = a;
  is_reserved = reserved;
  type = symtabtype::NUMBER;
}
 
void Symtab::Set(Symtab::function a)
{
  funcptr = a;
  type = symtabtype::FUNCTION;
}
 
std::string Symtab::GetName()const
{
  return name;
}
 
Symtab::symtabtype Symtab::GetType()const
{
  return type;
}
 
bool Symtab::IsReserved()const
{
  return is_reserved;
}
 
double Symtab::GetNumber()const
{
  if (type!=symtabtype::NUMBER)
    {
      std::cout << "Symbol is not a number!" << std::endl;
      exit(1);
    }
  return value;
}
 
std::string Symtab::GetString()const
{
  if (type!=symtabtype::STRING)
    {
      std::cout << "Symbol is not a string!" << std::endl;
      exit(1);
    }
  return str;
}
 
std::list<double> Symtab::GetArray()const
{
  if (type!=symtabtype::ARRAY)
    {
      std::cout << "Symbol is not a array!" << std::endl;
      exit(1);
    }
  return array;
}
 
Symtab::function Symtab::GetFunction()const
{
  if (type!=symtabtype::FUNCTION)
    {
      std::cout << "Symbol is not a function!" << std::endl;
      exit(1);
    }
  return funcptr;
}
 
void Symtab::Print()
{
  std::cout << "\t" << name << " = ";
  std::list<double>::iterator it;
  switch(type) {
  case Symtab::symtabtype::NUMBER:
    std::cout << value << std::endl;
    break;
 
  case Symtab::symtabtype::STRING:
    std::cout << str << std::endl;
    break;
 
  case Symtab::symtabtype::ARRAY:
    std::cout << "{";
    for(it=array.begin();it!=array.end();++it)
      {std::cout << " " << (*it) << " ";}
    std::cout << "}" << std::endl;
    break;
    
  default:
    break;
  }
}