function.hpp

Go to the documentation of this file.

 
#pragma once
 
#include <string>
#include <utility>
#include <vector>
 
#include "tcalc/ast/node.hpp"
#include "tcalc/common.hpp"
 
namespace tcalc::ast {
 
class FcallNode : public Node
{
private:
  std::string _name;
  std::vector<NodePtr<>> _args;
 
public:
  explicit FcallNode(std::string name)
    : FcallNode{ std::move(name), {} }
  {
  }
 
  FcallNode(std::string name, std::vector<NodePtr<>> args)
    : Node{ NodeType::FCALL }
    , _name{ std::move(name) }
    , _args{ std::move(args) }
  {
  }
 
  ~FcallNode() override = default;
 
  [[nodiscard]] TCALC_INLINE auto& args() const noexcept { return _args; }
 
  TCALC_INLINE auto& args() noexcept { return _args; }
 
  TCALC_INLINE void push_arg(const NodePtr<>& arg) { _args.push_back(arg); }
 
  [[nodiscard]] TCALC_INLINE auto& name() const noexcept { return _name; }
 
  TCALC_INLINE void name(std::string name) { _name = std::move(name); }
};
 
class FdefNode : public Node
{
private:
  std::string _name;
  std::vector<std::string> _args;
  NodePtr<> _body;
 
public:
  explicit FdefNode(std::string name)
    : FdefNode{ std::move(name), {}, {} }
  {
  }
 
  FdefNode(std::string name, std::vector<std::string> args, NodePtr<> body)
    : Node{ NodeType::FDEF }
    , _name{ std::move(name) }
    , _args{ std::move(args) }
    , _body{ std::move(body) }
  {
  }
 
  ~FdefNode() override = default;
 
  [[nodiscard]] TCALC_INLINE auto& args() const noexcept { return _args; }
 
  TCALC_INLINE auto& args() noexcept { return _args; }
 
  TCALC_INLINE void push_arg(const std::string& arg) { _args.push_back(arg); }
 
  [[nodiscard]] TCALC_INLINE auto& body() const noexcept { return _body; }
 
  TCALC_INLINE auto& body() noexcept { return _body; }
 
  TCALC_INLINE void body(NodePtr<> body) { _body = std::move(body); }
 
  [[nodiscard]] TCALC_INLINE auto& name() const noexcept { return _name; }
 
  TCALC_INLINE void name(std::string name) { _name = std::move(name); }
};
 
}