tree_node.h

Go to the documentation of this file.

 
#ifndef TREE_NODE_H
#define TREE_NODE_H
 
#include <string>
#include <vector>
#include <memory>
#include <cstddef>
 
#include "frame_info.h"
#include "signal_info.h"
 
enum class NodeType {
  Root,
  Frame,
  Signal
};
 
class TreeNode {
  public:
    TreeNode();
 
    explicit TreeNode (const FrameInfo &frame);
 
    explicit TreeNode (const SignalInfo &signal);
 
    ~TreeNode() = default;
 
    TreeNode (const TreeNode &) = delete;
    TreeNode &operator= (const TreeNode &) = delete;
 
    TreeNode (TreeNode &&) = default;
    TreeNode &operator= (TreeNode &&) = default;
 
    void AddChild (std::unique_ptr<TreeNode> child);
 
    std::size_t GetChildCount() const;
 
    const TreeNode *GetChild (std::size_t index) const;
 
    TreeNode *GetChild (std::size_t index);
 
    NodeType GetType() const;
 
    const std::string &GetName() const;
 
    const FrameInfo *GetFrame() const;
 
    const SignalInfo *GetSignal() const;
 
  private:
    NodeType m_type;
    std::string m_name;
    std::vector<std::unique_ptr<TreeNode> > m_children;
    std::unique_ptr<FrameInfo> m_frame;
    std::unique_ptr<SignalInfo> m_signal;
};
 
#endif /* TREE_NODE_H */