#ifndef __RAK_SMART_PTR_H
#define __RAK_SMART_PTR_H

#include "RakSmartObj.h"

template <class Base> class RakSmartPtr
{
public:
	RakSmartPtr(Base* input = 0);
	RakSmartPtr(const RakSmartPtr& ptr);
	~RakSmartPtr();
	RakSmartPtr& operator=(const RakSmartPtr& ptr);
	RakSmartPtr& operator=(Base* right);
	bool operator==(Base* right) const;
	bool operator!=(Base* right) const;
	bool operator==(const RakSmartPtr& ptr) const;
	bool operator!=(const RakSmartPtr& ptr) const;
	operator Base*() const;
	Base& operator*() const;
	Base* operator->() const;

protected:
	Base* object;
};

//--------------------------------------------------------------------------
template <class Base>
RakSmartPtr<Base>::RakSmartPtr(Base* input)
{
	object = input;
	if (object)
		object->IncRefCount();
}
//--------------------------------------------------------------------------
template <class Base>
RakSmartPtr<Base>::RakSmartPtr(const RakSmartPtr& ptr)
{
	object = ptr.object;
	// If you see
	// Error	1	error C2039: 'IncRefCount' : is not a member of 'A'	c:\rak3d\source\raksmartptr.h	40
	// You forgot to derive from RakSmartObj for the class that uses RakSmartPtr
	if (object)
		object->IncRefCount();
}
//--------------------------------------------------------------------------
template <class Base>
RakSmartPtr<Base>::~RakSmartPtr()
{
	if (object)
		object->DecRefCount();
}

//--------------------------------------------------------------------------
template <class Base>
RakSmartPtr<Base>& RakSmartPtr<Base>::operator=(const RakSmartPtr& ptr)
{
	if (object != ptr.object)
	{
		if (object)
			object->DecRefCount();
		object = ptr.object;
		if (object)
			object->IncRefCount();
	}
	return *this;
}
//--------------------------------------------------------------------------
template <class Base>
RakSmartPtr<Base>& RakSmartPtr<Base>::operator=(Base* right)
{
	if (object != right)
	{
		if (object)
			object->DecRefCount();
		object = right;
		if (object)
			object->IncRefCount();
	}
	return *this;
}
//--------------------------------------------------------------------------
template <class Base>
bool RakSmartPtr<Base>::operator==(Base* right) const
{
	return (object == right);
}
//--------------------------------------------------------------------------
template <class Base>
bool RakSmartPtr<Base>::operator!=(Base* right) const
{
	return (object != right);
}
//--------------------------------------------------------------------------
template <class Base>
bool RakSmartPtr<Base>::operator==(const RakSmartPtr& ptr) const
{
	return (object == ptr.object);
}
//--------------------------------------------------------------------------
template <class Base>
bool RakSmartPtr<Base>::operator!=(const RakSmartPtr& ptr) const
{
	return (object != ptr.object);
}
//--------------------------------------------------------------------------
template <class Base>
RakSmartPtr<Base>::operator Base*() const
{
	return object;
}
//--------------------------------------------------------------------------
template <class Base>
Base& RakSmartPtr<Base>::operator*() const
{
	return *object;
}
//--------------------------------------------------------------------------
template <class Base>
Base* RakSmartPtr<Base>::operator->() const
{
	return object;
}


#endif