#ifndef TREE_REFERENCE_H
#define TREE_REFERENCE_H

#include "tree_pool.h"

#include <stdio.h>

template <class T>
class TreeReference {
public:
  TreeReference(const TreeReference & tr) {
    int trNodeIdentifier = tr.m_identifier;
    printf("TreeReference copy of %d\n", trNodeIdentifier);
    TreeNode * nodeCopy = TreePool::sharedPool()->deepCopy(TreePool::sharedPool()->node(trNodeIdentifier));
    m_identifier = nodeCopy->identifier();
  }

  ~TreeReference() {
    printf("TreeRef destroy of %d\n", m_identifier);
    assert(node());
    assert(node()->identifier() == m_identifier);
    node()->release();
  }

  operator TreeReference<TreeNode>() const {
    // TODO: make sure this is kosher
    // static_assert(sizeof(ExpressionReference<T>) == sizeof(ExpressionReference<ExpressionNode>), "All ExpressionReference are supposed to have the same size");
    return *(reinterpret_cast<const TreeReference<TreeNode> *>(this));
  }

  int numberOfChildren() const {
    return node()->numberOfChildren();
  }

  TreeReference<T> childAtIndex(int i) const {
    return TreeReference(node()->childAtIndex(i));
  }

  void addChild(TreeReference<TreeNode> t) {
    TreeNode * deepCopy = TreePool::sharedPool()->deepCopy(t.node());
    TreePool::sharedPool()->move(
      deepCopy,
      node()->next()
    );
  }

  void removeChild(TreeReference<TreeNode> t) {
    TreePool::sharedPool()->move(
        t.node(),
        TreePool::sharedPool()->last()
    );
    t.node()->release();
  }

  T * node() const {
    // TODO: Here, assert that the node type is indeed T
    return static_cast<T*>(TreePool::sharedPool()->node(m_identifier));
  }

  int identifier() const { return m_identifier; }

protected:
  TreeReference() {
    TreeNode * node = TreePool::sharedPool()->createTreeNode<T>();
    m_identifier = node->identifier();
    printf("Creating TreeRef of new node %d\n", m_identifier);
  }

private:
  TreeReference(TreeNode * node) :
    m_identifier(node->identifier())
  {
    printf("Creating TreeRef of existing node %d\n", m_identifier);
    node->retain();
  }

  int m_identifier;
};

#endif