Composites and Visitors - a Templatized Approach

#pragma once

#include "GenericTree.h"
#include "Visitor.h"

#include <string>

class CNodeBase :
	public Loki::BaseVisitable<>,
	public Types::CGenericTree<CNodeBase>
{
public:
	DEFINE_VISITABLE()
	CNodeBase(void) : m_id( m_nextId++ ) {}
	CNodeBase( const CNodeBase &src ) : Types::CGenericTree<CNodeBase>( src ), m_id( m_nextId++ ) {}
	virtual ~CNodeBase(void) {}
	CNodeBase &operator=( const CNodeBase &rhs ) { if ( this != &rhs ) Types::CGenericTree<CNodeBase>::operator=( rhs ); return *this; }
	virtual CNodeBase *Clone() { return new CNodeBase( *this ); }

	unsigned long GetId() { return m_id; }
	virtual std::string GetType() { return std::string( "CNodeBase" ); }

private:
	unsigned long m_id;
	static unsigned long m_nextId;
};

class CNodeDerived1 :
	public CNodeBase
{
public:
	DEFINE_VISITABLE()
	CNodeDerived1(void) {}
	CNodeDerived1( const CNodeDerived1 &src ) : CNodeBase( src ) {}
	virtual ~CNodeDerived1(void) {}
	CNodeDerived1 &operator=( const CNodeDerived1 &rhs ) { if ( this != &rhs ) CNodeBase::operator=( rhs ); return *this; }
	virtual CNodeDerived1 *Clone() { return new CNodeDerived1( *this ); }

	virtual std::string GetType() { return std::string( "CNodeDerived1" ); }

};

class CNodeDerived2 :
	public CNodeBase
{
public:
	DEFINE_VISITABLE()
	CNodeDerived2(void) {}
	CNodeDerived2( const CNodeDerived2 &src ) : CNodeBase( src ) {}
	virtual ~CNodeDerived2(void) {}
	CNodeDerived2 &operator=( const CNodeDerived2 &rhs ) { if ( this != &rhs ) CNodeBase::operator=( rhs ); return *this; }
	virtual CNodeDerived2 *Clone() { return new CNodeDerived2( *this ); }

	virtual std::string GetType() { return std::string( "CNodeDerived2" ); }

};

unsigned long CNodeBase::m_nextId = 0;