node.hpp

Go to the documentation of this file.

 
#pragma once
 
#include <cstddef>
#include <optional>
#include <string>
#include <vector>
 
#include "pulsatrix/op_type.hpp"
#include "pulsatrix/shape.hpp"
 
namespace pulsatrix {
 
using NodeId = size_t;
 
class Node {
public:
    Node(NodeId id, OpType op_type, Shape shape, std::optional<std::string> label = std::nullopt)
        : id_(id), op_type_(op_type), shape_(std::move(shape)), label_(std::move(label)) {}
 
    [[nodiscard]] NodeId id() const { return id_; }
    [[nodiscard]] OpType op_type() const { return op_type_; }
    [[nodiscard]] const Shape& shape() const { return shape_; }
    [[nodiscard]] const std::optional<std::string>& label() const { return label_; }
 
    [[nodiscard]] const std::vector<Node*>& parents() const { return parents_; }
 
    [[nodiscard]] const std::vector<Node*>& children() const { return children_; }
 
    void add_parent(Node* parent) { parents_.push_back(parent); }
 
    void add_child(Node* child) { children_.push_back(child); }
 
private:
    NodeId id_;
    OpType op_type_;
    Shape shape_;
    std::optional<std::string> label_;
    std::vector<Node*> parents_;
    std::vector<Node*> children_;
};
 
}  // namespace pulsatrix