AST: Add Atom data type
This commit is contained in:
@@ -80,6 +80,13 @@ Lambda::Lambda(std::vector<std::string> bindings, ASTNodePtr body, EnvironmentPt
|
|||||||
{
|
{
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// -----------------------------------------
|
||||||
|
|
||||||
|
Atom::Atom(ASTNodePtr pointer)
|
||||||
|
: m_value(pointer)
|
||||||
|
{
|
||||||
|
}
|
||||||
|
|
||||||
} // namespace blaze
|
} // namespace blaze
|
||||||
|
|
||||||
// -----------------------------------------
|
// -----------------------------------------
|
||||||
|
|||||||
@@ -43,6 +43,7 @@ public:
|
|||||||
virtual bool isSymbol() const { return false; }
|
virtual bool isSymbol() const { return false; }
|
||||||
virtual bool isFunction() const { return false; }
|
virtual bool isFunction() const { return false; }
|
||||||
virtual bool isLambda() const { return false; }
|
virtual bool isLambda() const { return false; }
|
||||||
|
virtual bool isAtom() const { return false; }
|
||||||
|
|
||||||
protected:
|
protected:
|
||||||
ASTNode() {}
|
ASTNode() {}
|
||||||
@@ -238,6 +239,23 @@ private:
|
|||||||
|
|
||||||
// -----------------------------------------
|
// -----------------------------------------
|
||||||
|
|
||||||
|
class Atom final : public ASTNode {
|
||||||
|
public:
|
||||||
|
Atom() = default;
|
||||||
|
Atom(ASTNodePtr pointer);
|
||||||
|
virtual ~Atom() = default;
|
||||||
|
|
||||||
|
ASTNodePtr reset(ASTNodePtr value) { return m_value = value; }
|
||||||
|
ASTNodePtr deref() const { return m_value; }
|
||||||
|
|
||||||
|
private:
|
||||||
|
virtual bool isAtom() const override { return true; }
|
||||||
|
|
||||||
|
ASTNodePtr m_value;
|
||||||
|
};
|
||||||
|
|
||||||
|
// -----------------------------------------
|
||||||
|
|
||||||
template<typename T, typename... Args>
|
template<typename T, typename... Args>
|
||||||
std::shared_ptr<T> makePtr(Args&&... args)
|
std::shared_ptr<T> makePtr(Args&&... args)
|
||||||
{
|
{
|
||||||
@@ -279,6 +297,9 @@ inline bool ASTNode::fastIs<Function>() const { return isFunction(); }
|
|||||||
|
|
||||||
template<>
|
template<>
|
||||||
inline bool ASTNode::fastIs<Lambda>() const { return isLambda(); }
|
inline bool ASTNode::fastIs<Lambda>() const { return isLambda(); }
|
||||||
|
|
||||||
|
template<>
|
||||||
|
inline bool ASTNode::fastIs<Atom>() const { return isAtom(); }
|
||||||
// clang-format on
|
// clang-format on
|
||||||
|
|
||||||
} // namespace blaze
|
} // namespace blaze
|
||||||
|
|||||||
@@ -141,6 +141,12 @@ void Printer::printImpl(ASTNodePtr node, bool print_readably)
|
|||||||
printSpacing();
|
printSpacing();
|
||||||
m_print += format("#<user-function>({:p})", node_raw_ptr);
|
m_print += format("#<user-function>({:p})", node_raw_ptr);
|
||||||
}
|
}
|
||||||
|
else if (is<Atom>(node_raw_ptr)) {
|
||||||
|
printSpacing();
|
||||||
|
m_print += "(atom ";
|
||||||
|
printImpl(std::static_pointer_cast<Atom>(node)->deref(), print_readably);
|
||||||
|
m_print += ")";
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
void Printer::printError()
|
void Printer::printError()
|
||||||
|
|||||||
Reference in New Issue
Block a user