///////////////////////////////////////////////////
// MT3Q7.cpp - Node Destruction template Policy  //
//                                               //
// Jim Fawcett, CSE687 - Midterm, Spring 2010    //
///////////////////////////////////////////////////

#include <vector>
#include <iostream>

template <typename T, typename DP>
class MNode
{
public:
  MNode(T t);
  ~MNode();
  void addChild(MNode<T,DP>* pNode);
  T& value();
private:
  // won't define copy or assignment so make private
  MNode(const MNode<T,DP>& node);
  MNode<T,DP>& operator=(const MNode<T,DP>& node); 
  std::vector<MNode<T,DP>*> children;
  T t_;
};

template <typename T, typename DP>
MNode<T,DP>::MNode(T t) : t_(t) {}

template <typename T, typename DP>
MNode<T,DP>::~MNode()
{
  std::cout << "\n  MNode destructor called";
  for(size_t i=0; i<children.size(); ++i)
    DP::GoodBye(children[i]);
}

template <typename T, typename DP>
void MNode<T,DP>::addChild(MNode<T,DP>* pNode)
{
  children.push_back(pNode);
}

template <typename T, typename DP>
T& MNode<T,DP>::value() { return t_; }

template <typename T>
class DeallocPolicy
{
public:
  static void GoodBye( MNode<T, DeallocPolicy<T> >* pNode)
  {
    std::cout << "\n  deleting MNode";
    delete pNode;
  }
};

template <typename T>
class NoDeallocPolicy
{
public:
  static void GoodBye(MNode<T,NoDeallocPolicy<T> >* pNode)
  {
    std::cout << "\n  not deleting MNode as it is on the stack";
  }
};

void main()
{
  std::cout << "\n  MT3Q7 - Node Destruction Template Policy";
  std::cout << "\n ==========================================";

  MNode< int, NoDeallocPolicy<int> > node1(1);
  MNode< int, NoDeallocPolicy<int> > node2(2);
  MNode< int, NoDeallocPolicy<int> > node3(3);
  MNode< int, NoDeallocPolicy<int> > node4(4);
  MNode< int, NoDeallocPolicy<int> > node5(5);

  node1.addChild(&node2);
  node1.addChild(&node3);
  node1.addChild(&node4);
  node1.addChild(&node5);

  MNode< int, DeallocPolicy<int> >* pNode1 = new MNode< int, DeallocPolicy<int> >(1);
  MNode< int, DeallocPolicy<int> >* pNode2 = new MNode< int, DeallocPolicy<int> >(2);
  MNode< int, DeallocPolicy<int> >* pNode3 = new MNode< int, DeallocPolicy<int> >(3);
  MNode< int, DeallocPolicy<int> >* pNode4 = new MNode< int, DeallocPolicy<int> >(4);
  MNode< int, DeallocPolicy<int> >* pNode5 = new MNode< int, DeallocPolicy<int> >(5);

  pNode1->addChild(pNode2);
  pNode1->addChild(pNode3);
  pNode1->addChild(pNode4);
  pNode1->addChild(pNode5);

  delete pNode1;
  std::cout << "\n";
}