forked from PuqiAR/Fig-TreeWalker
98 lines
3.1 KiB
C++
98 lines
3.1 KiB
C++
#pragma once
|
|
|
|
#include <Value/BaseValue.hpp>
|
|
#include <Ast/functionParameters.hpp>
|
|
#include <context_forward.hpp>
|
|
|
|
#include <atomic>
|
|
|
|
namespace Fig
|
|
{
|
|
class Value;
|
|
|
|
/* complex objects */
|
|
struct FunctionStruct
|
|
{
|
|
std::size_t id;
|
|
Ast::FunctionParameters paras;
|
|
TypeInfo retType;
|
|
Ast::BlockStatement body;
|
|
|
|
bool isBuiltin = false;
|
|
std::function<Value(const std::vector<Value> &)> builtin;
|
|
int builtinParamCount = -1;
|
|
|
|
std::shared_ptr<Context> closureContext;
|
|
|
|
FunctionStruct() = default;
|
|
|
|
FunctionStruct(Ast::FunctionParameters _paras, TypeInfo _retType, Ast::BlockStatement _body, ContextPtr _closureContext) :
|
|
id(nextId()), // 分配唯一 ID
|
|
paras(std::move(_paras)),
|
|
retType(std::move(_retType)),
|
|
body(std::move(_body)),
|
|
closureContext(std::move(_closureContext))
|
|
{
|
|
}
|
|
|
|
FunctionStruct(std::function<Value(const std::vector<Value> &)> fn, int argc);
|
|
|
|
FunctionStruct(const FunctionStruct &other) :
|
|
id(other.id),
|
|
paras(other.paras),
|
|
retType(other.retType),
|
|
body(other.body),
|
|
isBuiltin(other.isBuiltin),
|
|
builtin(other.builtin),
|
|
builtinParamCount(other.builtinParamCount),
|
|
closureContext(other.closureContext) {}
|
|
|
|
FunctionStruct &operator=(const FunctionStruct &other) = default;
|
|
FunctionStruct(FunctionStruct &&) noexcept = default;
|
|
FunctionStruct &operator=(FunctionStruct &&) noexcept = default;
|
|
|
|
bool operator==(const FunctionStruct &other) const noexcept
|
|
{
|
|
return id == other.id;
|
|
}
|
|
bool operator!=(const FunctionStruct &other) const noexcept
|
|
{
|
|
return !(*this == other);
|
|
}
|
|
|
|
private:
|
|
static std::size_t nextId()
|
|
{
|
|
static std::atomic<std::size_t> counter{1};
|
|
return counter++;
|
|
}
|
|
};
|
|
|
|
class Function final : public __ValueWrapper<FunctionStruct>
|
|
{
|
|
public:
|
|
Function(const FunctionStruct &x) :
|
|
__ValueWrapper(ValueType::Function)
|
|
{
|
|
data = std::make_unique<FunctionStruct>(x);
|
|
}
|
|
Function(Ast::FunctionParameters paras, TypeInfo ret, Ast::BlockStatement body, ContextPtr closureContext) :
|
|
__ValueWrapper(ValueType::Function)
|
|
{
|
|
data = std::make_unique<FunctionStruct>(
|
|
std::move(paras), std::move(ret), std::move(body), std::move(closureContext));
|
|
}
|
|
Function(std::function<Value(const std::vector<Value> &)> fn, int argc);
|
|
|
|
bool operator==(const Function &other) const noexcept
|
|
{
|
|
if (!data || !other.data) return false;
|
|
return *data == *other.data; // call -> FunctionStruct::operator== (based on ID comparing)
|
|
}
|
|
Function(const Function &) = default;
|
|
Function(Function &&) noexcept = default;
|
|
Function &operator=(const Function &) = default;
|
|
Function &operator=(Function &&) noexcept = default;
|
|
};
|
|
} // namespace Fig
|