tree-node.h

Go to the documentation of this file.

// Copyright 2007, Massachusetts Institute of Technology.
// The use of this code is permitted for research only. There is
// absolutely no warranty for this software.
//
// Author: John Lee (jjl@mit.edu)
//

#ifndef TREE_TREE_NODE_H
#define TREE_TREE_NODE_H

#include <iostream>
#include <vector>

using namespace std;

namespace libpmk {


class TreeNode {
public:
  static const int kInvalidNodeID = -1;

  TreeNode();
  TreeNode(int node_id);

  TreeNode(const TreeNode& other);

  virtual ~TreeNode() { }

  int id() const { return id_; }
  int parent() const { return parent_; }
  int child(int child_index) const { return children_.at(child_index); }

  int child_size() const { return (int)children_.size(); }
  bool has_parent() const { return parent_ != kInvalidNodeID; }
  bool has_child() const { return child_size() > 0; }

  void set_id(int id) { id_ = id; }
  void set_parent(int id) { parent_ = id; }


  void add_child(int id) { children_.push_back(id); }

  void remove_child(int child_index);



  void ReadFromStream(istream& input_stream);


  void WriteToStream(ostream& output_stream) const;

  virtual void Combine(const TreeNode& other) = 0;

protected:
  virtual void ReadData(istream& input_stream) = 0;

  virtual void WriteData(ostream& output_stream) const = 0;

private:
  int id_;
  int parent_;
  vector<int> children_;
};
}  // namespace libpmk

#endif  // TREE_TREE_NODE_H

---