91 lines
2.7 KiB
C++
91 lines
2.7 KiB
C++
#include "Environment.h"
|
|
#include "Parser.h"
|
|
|
|
#include <stdexcept>
|
|
|
|
#include <locale>
|
|
#include <codecvt>
|
|
|
|
namespace {
|
|
std::wstring_convert<std::codecvt_utf8_utf16<wchar_t>> converter;
|
|
}
|
|
|
|
namespace NVL::Environment {
|
|
Environment ENVIRONMENT;
|
|
Variable::Variable() : type(Type::Nil), length(0) { }
|
|
Variable::Variable(Type t) : type(t), length(0) {
|
|
if (t != Type::Nil)
|
|
throw std::runtime_error("Cannot create non-nil object with type alone");
|
|
}
|
|
Variable::Variable(const Number& v) : type(Type::Number), value(v), length(1) {}
|
|
Variable::Variable(const std::string& v) : type(Type::String), value(converter.from_bytes(v)), length(1) {}
|
|
Variable::Variable(const std::vector<Variable>& v) : type(Type::Array), value(v), length(v.size()) {}
|
|
Variable::Variable(const std::function < Variable(std::vector<Variable>)>& v, u32 l) : type(Type::Procedure), value(v), length(l) {}
|
|
|
|
bool Variable::operator==(const Variable& other) const {
|
|
if (other.type != type)
|
|
return false;
|
|
switch (type) {
|
|
case Type::Number:
|
|
return std::get<Number>(other.value) == std::get<Number>(value);
|
|
case Type::String:
|
|
return std::get<NVL::String>(other.value) == std::get<NVL::String>(value);
|
|
default:
|
|
return false;
|
|
}
|
|
}
|
|
|
|
void Environment::enter(const std::string& name, Variable p) {
|
|
if (env.find(name) != env.end())
|
|
throw std::runtime_error("Redefinition of symbol in environment");
|
|
env.emplace(name, p);
|
|
}
|
|
|
|
void Environment::set(const std::string& name, Variable p) {
|
|
if (env.find(name) == env.end())
|
|
throw std::runtime_error("Attempting to set undefined variable");
|
|
env[name] = p;
|
|
}
|
|
|
|
Variable& Environment::get(const std::string& name) {
|
|
return env.at(name);
|
|
}
|
|
|
|
bool Environment::exists(const std::string& name) {
|
|
return env.find(name) != env.end();
|
|
}
|
|
|
|
Variable Eval(const Parse::Object& obj) {
|
|
switch (obj.type) {
|
|
case Parse::Type::Symbol:
|
|
return std::get<std::string>(obj.value);
|
|
case Parse::Type::Number:
|
|
return Variable(std::get<Number>(obj.value));
|
|
case Parse::Type::String:
|
|
return Variable(std::get<std::string>(obj.value));
|
|
case Parse::Type::Array:
|
|
{
|
|
std::vector<Variable> v{};
|
|
for (const auto& x : std::get<std::vector<Parse::Object>>(obj.value))
|
|
v.push_back(Eval(x));
|
|
return Variable(v);
|
|
}
|
|
case Parse::Type::Subexpression:
|
|
{
|
|
return Apply(std::get<std::vector<Parse::Object>>(obj.value));
|
|
}
|
|
}
|
|
}
|
|
|
|
Variable Apply(const Parse::Command& c) {
|
|
std::vector<Variable> args{};
|
|
for (u32 i = 1; i < c.size(); i++)
|
|
args.push_back(Eval(c[i]));
|
|
return std::get<std::function<Variable(std::vector<Variable>)>>(ENVIRONMENT.get(std::get<std::string>(c[0].value)).value)(args);
|
|
}
|
|
|
|
void EvalScene(const Parse::Scene& s) {
|
|
for (const auto& c : s.get())
|
|
Apply(c);
|
|
}
|
|
}
|