base.hpp

Go to the documentation of this file.

 
#pragma once
 
#include <memory>
 
#include "tcalc/ast/binaryop.hpp"
#include "tcalc/ast/control_flow.hpp"
#include "tcalc/ast/function.hpp"
#include "tcalc/ast/node.hpp"
#include "tcalc/ast/number.hpp"
#include "tcalc/ast/program.hpp"
#include "tcalc/ast/unaryop.hpp"
#include "tcalc/ast/variable.hpp"
#include "tcalc/error.hpp"
 
#define __check_node_type(node, type, entry)                                   \
  if (auto node_type = std::dynamic_pointer_cast<type>(node)) {                \
    return entry(node_type);                                                   \
  }
 
#define VISIT_DEFAULT(node)                                                    \
  {                                                                            \
    (void)node;                                                                \
    return error::ok<RT>();                                                    \
  }
 
namespace tcalc::ast {
 
template<typename RT>
class BaseVisitor
{
public:
  virtual ~BaseVisitor() = default;
 
  virtual error::Result<RT> visit(NodePtr<>& node)
  {
    __check_node_type(node, BinaryOpNode, visit_bin_op);
    __check_node_type(node, UnaryOpNode, visit_unary_op);
    __check_node_type(node, NumberNode, visit_number);
    __check_node_type(node, VarRefNode, visit_varref);
    __check_node_type(node, VarAssignNode, visit_varassign);
    __check_node_type(node, FcallNode, visit_fcall);
    __check_node_type(node, FdefNode, visit_fdef);
    __check_node_type(node, IfNode, visit_if);
    __check_node_type(node, ProgramNode, visit_program);
    __check_node_type(node, ProgramImportNode, visit_import);
 
    return error::ok<RT>();
  }
 
  virtual error::Result<RT> visit_bin_op(NodePtr<BinaryOpNode>& node)
    VISIT_DEFAULT(node);
 
  virtual error::Result<RT> visit_unary_op(NodePtr<UnaryOpNode>& node)
    VISIT_DEFAULT(node);
 
  virtual error::Result<RT> visit_number(NodePtr<NumberNode>& node)
    VISIT_DEFAULT(node);
 
  virtual error::Result<RT> visit_varref(NodePtr<VarRefNode>& node)
    VISIT_DEFAULT(node);
 
  virtual error::Result<RT> visit_varassign(NodePtr<VarAssignNode>& node)
    VISIT_DEFAULT(node);
 
  virtual error::Result<RT> visit_fcall(NodePtr<FcallNode>& node)
    VISIT_DEFAULT(node);
 
  virtual error::Result<RT> visit_fdef(NodePtr<FdefNode>& node)
    VISIT_DEFAULT(node);
 
  virtual error::Result<RT> visit_if(NodePtr<IfNode>& node) VISIT_DEFAULT(node);
 
  virtual error::Result<RT> visit_program(NodePtr<ProgramNode>& node)
    VISIT_DEFAULT(node);
 
  virtual error::Result<RT> visit_import(NodePtr<ProgramImportNode>& node)
    VISIT_DEFAULT(node);
};
 
}