variable.hpp

Go to the documentation of this file.

 
#pragma once
 
#include <string>
#include <utility>
 
#include "tcalc/ast/node.hpp"
#include "tcalc/common.hpp"
 
namespace tcalc::ast {
 
class VarRefNode : public Node
{
private:
  std::string _name;
 
public:
  explicit VarRefNode(std::string name)
    : Node{ NodeType::VARREF }
    , _name{ std::move(name) }
  {
  }
 
  ~VarRefNode() override = default;
 
  [[nodiscard]] TCALC_INLINE auto& name() const noexcept { return _name; }
 
  TCALC_INLINE void name(std::string name) noexcept { _name = std::move(name); }
};
 
class VarAssignNode : public Node
{
private:
  std::string _name;
  NodePtr<> _body;
 
public:
  explicit VarAssignNode(std::string name)
    : VarAssignNode{ std::move(name), nullptr }
  {
  }
 
  VarAssignNode(std::string name, NodePtr<> body)
    : Node{ NodeType::VARASSIGN }
    , _name{ std::move(name) }
    , _body{ std::move(body) }
  {
  }
 
  ~VarAssignNode() override = default;
 
  [[nodiscard]] TCALC_INLINE auto& name() const noexcept { return _name; }
 
  TCALC_INLINE void name(std::string name) noexcept { _name = std::move(name); }
 
  [[nodiscard]] TCALC_INLINE auto& body() const noexcept { return _body; }
 
  TCALC_INLINE auto& body() noexcept { return _body; }
 
  TCALC_INLINE void body(NodePtr<> body) noexcept { _body = std::move(body); }
};
 
}