2020-09-06 12:45:31 -04:00
|
|
|
#include "goalc/compiler/Compiler.h"
|
2020-09-06 16:58:25 -04:00
|
|
|
#include "goalc/compiler/IR.h"
|
2020-11-24 20:48:38 -05:00
|
|
|
#include "common/goos/ParseHelpers.h"
|
2020-09-06 12:45:31 -04:00
|
|
|
|
|
|
|
goos::Arguments Compiler::get_va(const goos::Object& form, const goos::Object& rest) {
|
|
|
|
goos::Arguments args;
|
|
|
|
|
2020-11-24 20:48:38 -05:00
|
|
|
std::string err;
|
|
|
|
if (!goos::get_va(rest, &err, &args)) {
|
2020-12-01 21:39:46 -05:00
|
|
|
throw_compiler_error(form, err);
|
2020-09-06 12:45:31 -04:00
|
|
|
}
|
|
|
|
return args;
|
|
|
|
}
|
|
|
|
|
|
|
|
void Compiler::va_check(
|
|
|
|
const goos::Object& form,
|
|
|
|
const goos::Arguments& args,
|
2020-11-20 20:17:37 -05:00
|
|
|
const std::vector<std::optional<goos::ObjectType>>& unnamed,
|
|
|
|
const std::unordered_map<std::string, std::pair<bool, std::optional<goos::ObjectType>>>&
|
|
|
|
named) {
|
2020-11-24 20:48:38 -05:00
|
|
|
std::string err;
|
|
|
|
if (!goos::va_check(args, unnamed, named, &err)) {
|
2020-12-01 21:39:46 -05:00
|
|
|
throw_compiler_error(form, err);
|
2020-09-06 12:45:31 -04:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2020-09-06 16:58:25 -04:00
|
|
|
void Compiler::for_each_in_list(const goos::Object& list,
|
|
|
|
const std::function<void(const goos::Object&)>& f) {
|
2020-09-06 12:45:31 -04:00
|
|
|
const goos::Object* iter = &list;
|
2020-09-06 16:58:25 -04:00
|
|
|
while (iter->is_pair()) {
|
2020-09-06 12:45:31 -04:00
|
|
|
auto lap = iter->as_pair();
|
|
|
|
f(lap->car);
|
|
|
|
iter = &lap->cdr;
|
|
|
|
}
|
|
|
|
|
2020-09-06 16:58:25 -04:00
|
|
|
if (!iter->is_empty_list()) {
|
2020-12-01 21:39:46 -05:00
|
|
|
throw_compiler_error(list, "Invalid list: {}", list.print());
|
2020-09-06 12:45:31 -04:00
|
|
|
}
|
2020-09-07 13:28:16 -04:00
|
|
|
}
|
|
|
|
|
|
|
|
std::string Compiler::as_string(const goos::Object& o) {
|
|
|
|
return o.as_string()->data;
|
|
|
|
}
|
|
|
|
|
|
|
|
std::string Compiler::symbol_string(const goos::Object& o) {
|
|
|
|
return o.as_symbol()->name;
|
2020-09-07 19:17:48 -04:00
|
|
|
}
|
|
|
|
|
2020-09-19 13:22:14 -04:00
|
|
|
std::string Compiler::quoted_sym_as_string(const goos::Object& o) {
|
|
|
|
auto args = get_va(o, o);
|
|
|
|
va_check(o, args, {{goos::ObjectType::SYMBOL}, {goos::ObjectType::SYMBOL}}, {});
|
|
|
|
if (symbol_string(args.unnamed.at(0)) != "quote") {
|
2020-12-01 21:39:46 -05:00
|
|
|
throw_compiler_error(o, "Invalid quoted symbol: {}.", o.print());
|
2020-09-19 13:22:14 -04:00
|
|
|
}
|
|
|
|
return symbol_string(args.unnamed.at(1));
|
|
|
|
}
|
|
|
|
|
2020-10-16 17:08:26 -04:00
|
|
|
bool Compiler::is_quoted_sym(const goos::Object& o) {
|
|
|
|
if (o.is_pair()) {
|
|
|
|
auto car = pair_car(o);
|
|
|
|
auto cdr = pair_cdr(o);
|
|
|
|
if (car.is_symbol() && car.as_symbol()->name == "quote") {
|
|
|
|
if (cdr.is_pair()) {
|
|
|
|
auto thing = pair_car(cdr);
|
|
|
|
if (thing.is_symbol()) {
|
|
|
|
if (pair_cdr(cdr).is_empty_list()) {
|
|
|
|
return true;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
return false;
|
|
|
|
}
|
|
|
|
|
2020-09-07 19:17:48 -04:00
|
|
|
const goos::Object& Compiler::pair_car(const goos::Object& o) {
|
|
|
|
return o.as_pair()->car;
|
|
|
|
}
|
|
|
|
|
|
|
|
const goos::Object& Compiler::pair_cdr(const goos::Object& o) {
|
|
|
|
return o.as_pair()->cdr;
|
|
|
|
}
|
|
|
|
|
|
|
|
void Compiler::expect_empty_list(const goos::Object& o) {
|
|
|
|
if (!o.is_empty_list()) {
|
2020-12-01 21:39:46 -05:00
|
|
|
throw_compiler_error(o, "expected to be an empty list");
|
2020-09-07 19:17:48 -04:00
|
|
|
}
|
2020-09-12 13:11:42 -04:00
|
|
|
}
|
|
|
|
|
|
|
|
TypeSpec Compiler::parse_typespec(const goos::Object& src) {
|
|
|
|
if (src.is_symbol()) {
|
|
|
|
return m_ts.make_typespec(symbol_string(src));
|
|
|
|
} else if (src.is_pair()) {
|
|
|
|
TypeSpec ts = m_ts.make_typespec(symbol_string(pair_car(src)));
|
|
|
|
const auto& rest = pair_cdr(src);
|
|
|
|
|
|
|
|
for_each_in_list(rest, [&](const goos::Object& o) { ts.add_arg(parse_typespec(o)); });
|
|
|
|
|
|
|
|
return ts;
|
|
|
|
} else {
|
2020-12-01 21:39:46 -05:00
|
|
|
throw_compiler_error(src, "Invalid typespec.");
|
2020-09-12 13:11:42 -04:00
|
|
|
}
|
|
|
|
assert(false);
|
|
|
|
return {};
|
|
|
|
}
|
|
|
|
|
|
|
|
bool Compiler::is_local_symbol(const goos::Object& obj, Env* env) {
|
|
|
|
// check in the symbol macro env.
|
|
|
|
auto mlet_env = get_parent_env_of_type<SymbolMacroEnv>(env);
|
|
|
|
while (mlet_env) {
|
|
|
|
if (mlet_env->macros.find(obj.as_symbol()) != mlet_env->macros.end()) {
|
|
|
|
return true;
|
|
|
|
}
|
|
|
|
mlet_env = get_parent_env_of_type<SymbolMacroEnv>(mlet_env->parent());
|
|
|
|
}
|
|
|
|
|
|
|
|
// check lexical
|
|
|
|
if (env->lexical_lookup(obj)) {
|
|
|
|
return true;
|
|
|
|
}
|
|
|
|
|
|
|
|
// check global constants
|
|
|
|
if (m_global_constants.find(obj.as_symbol()) != m_global_constants.end()) {
|
|
|
|
return true;
|
|
|
|
}
|
|
|
|
|
|
|
|
return false;
|
|
|
|
}
|
|
|
|
|
|
|
|
emitter::RegKind Compiler::get_preferred_reg_kind(const TypeSpec& ts) {
|
|
|
|
switch (m_ts.lookup_type(ts)->get_preferred_reg_kind()) {
|
|
|
|
case RegKind::GPR_64:
|
|
|
|
return emitter::RegKind::GPR;
|
|
|
|
case RegKind::FLOAT:
|
|
|
|
return emitter::RegKind::XMM;
|
|
|
|
default:
|
2020-09-13 21:10:43 -04:00
|
|
|
throw std::runtime_error("Unknown preferred register kind");
|
2020-09-12 13:11:42 -04:00
|
|
|
}
|
2020-09-13 17:34:02 -04:00
|
|
|
}
|
|
|
|
|
|
|
|
bool Compiler::is_none(Val* in) {
|
|
|
|
return dynamic_cast<None*>(in);
|
2020-09-19 16:50:42 -04:00
|
|
|
}
|
|
|
|
|
|
|
|
bool Compiler::is_basic(const TypeSpec& ts) {
|
|
|
|
return m_ts.typecheck(m_ts.make_typespec("basic"), ts, "", false, false);
|
2020-10-14 13:42:14 -04:00
|
|
|
}
|
|
|
|
|
2020-10-16 17:08:26 -04:00
|
|
|
bool Compiler::is_structure(const TypeSpec& ts) {
|
|
|
|
return m_ts.typecheck(m_ts.make_typespec("structure"), ts, "", false, false);
|
|
|
|
}
|
|
|
|
|
2020-11-29 18:01:30 -05:00
|
|
|
bool Compiler::is_bitfield(const TypeSpec& ts) {
|
|
|
|
return m_ts.is_bitfield_type(ts.base_type());
|
|
|
|
}
|
|
|
|
|
2020-10-14 13:42:14 -04:00
|
|
|
bool Compiler::try_getting_constant_integer(const goos::Object& in, int64_t* out, Env* env) {
|
|
|
|
(void)env;
|
|
|
|
if (in.is_int()) {
|
|
|
|
*out = in.as_int();
|
|
|
|
return true;
|
|
|
|
}
|
|
|
|
|
2020-10-25 18:02:03 -04:00
|
|
|
// todo, try more things like constants before giving up.
|
|
|
|
return false;
|
|
|
|
}
|
|
|
|
|
|
|
|
float Compiler::try_getting_constant_float(const goos::Object& in, float* out, Env* env) {
|
|
|
|
(void)env;
|
|
|
|
if (in.is_float()) {
|
|
|
|
*out = in.as_float();
|
|
|
|
return true;
|
|
|
|
}
|
|
|
|
|
2020-10-16 17:08:26 -04:00
|
|
|
// todo, try more things like constants before giving up.
|
2020-10-14 13:42:14 -04:00
|
|
|
return false;
|
2020-09-06 12:45:31 -04:00
|
|
|
}
|