#pragma once

#ifndef TSMARTPOINTER_INCLUDED

#define TSMARTPOINTER_INCLUDED

#include "tutil.h"

#include "tatomicvar.h"

#undef DVAPI

#undef DVVAR

#ifdef TNZCORE_EXPORTS

#define DVAPI DV_EXPORT_API

#define DVVAR DV_EXPORT_VAR

#else

#define DVAPI DV_IMPORT_API

#define DVVAR DV_IMPORT_VAR

#endif

//=========================================================

#ifndef NDEBUG

#define INSTANCE_COUNT_ENABLED

#endif

//=========================================================

class DVAPI TSmartObject {

  TAtomicVar m_refCount;

#ifdef INSTANCE_COUNT_ENABLED

  const TINT32 m_classCodeRef;

  static const TINT32 m_unknownClassCode;

#endif

public:

  typedef short ClassCode;

  TSmartObject(ClassCode

#ifdef INSTANCE_COUNT_ENABLED

                   classCode

#endif

               )

      : m_refCount()

#ifdef INSTANCE_COUNT_ENABLED

      , m_classCodeRef(classCode)

#endif

  {

#ifdef INSTANCE_COUNT_ENABLED

    incrementInstanceCount();

#endif

  }

  TSmartObject()

      : m_refCount()

#ifdef INSTANCE_COUNT_ENABLED

      , m_classCodeRef(m_unknownClassCode)

#endif

  {

#ifdef INSTANCE_COUNT_ENABLED

    incrementInstanceCount();

#endif

  }

  virtual ~TSmartObject() {

    assert(m_refCount == 0);

#ifdef INSTANCE_COUNT_ENABLED

    decrementInstanceCount();

#endif

  }

  inline void addRef() { ++m_refCount; }

  inline void release() {

    if ((--m_refCount) <= 0) delete this;

  };

  inline TINT32 getRefCount() const { return m_refCount; }

  static TINT32 getInstanceCount(ClassCode code);

private:

  void incrementInstanceCount();

  void decrementInstanceCount();

private:

  // not implemented

  TSmartObject(const TSmartObject &);

  TSmartObject &operator=(const TSmartObject &);

};

#define DECLARE_CLASS_CODE                                                     \

private:                                                                       \

  static const TSmartObject::ClassCode m_classCode;                            \

                                                                               \

public:                                                                        \

  inline static TINT32 getInstanceCount() {                                    \

    return TSmartObject::getInstanceCount(m_classCode);                        \

  }

#define DEFINE_CLASS_CODE(T, ID)                                               \

  const TSmartObject::ClassCode T::m_classCode = ID;

//=========================================================

template <class t=""></class>

class DVAPI TSmartPointerT {

protected:

  T *m_pointer;

public:

  TSmartPointerT() : m_pointer(0) {}

  TSmartPointerT(const TSmartPointerT &src) : m_pointer(src.m_pointer) {

    if (m_pointer) m_pointer->addRef();

  }

  TSmartPointerT(T *pointer) : m_pointer(pointer) {

    if (m_pointer) m_pointer->addRef();

  }

  virtual ~TSmartPointerT() {

    if (m_pointer) {

      m_pointer->release();

      m_pointer = 0;

    }

  }

  TSmartPointerT &operator=(const TSmartPointerT &src) {

    // prima addRef  e poi release per evitare brutti scherzi

    // in caso di parentela

    T *old    = m_pointer;

    m_pointer = src.m_pointer;

    if (m_pointer) m_pointer->addRef();

    if (old) old->release();

    return *this;

  }

  T *operator->() const {

    assert(m_pointer);

    return m_pointer;

  }

  T &operator*() const {

    assert(m_pointer);

    return *m_pointer;

  }

  T *getPointer() const { return m_pointer; }

  bool operator!() const { return m_pointer == 0; }

  operator bool() const { return m_pointer != 0; }

  bool operator==(const TSmartPointerT &p) const {

    return m_pointer == p.m_pointer;

  }

  bool operator!=(const TSmartPointerT &p) const {

    return m_pointer != p.m_pointer;

  }

  bool operator<(const TSmartPointerT &p) const {

    return m_pointer < p.m_pointer;

  }

  bool operator>(const TSmartPointerT &p) const {

    return m_pointer > p.m_pointer;

  }

};

//=========================================================

template <class base="" class="" derived,=""></class>

class DVAPI TDerivedSmartPointerT : public TSmartPointerT<derived> {</derived>

public:

  typedef TDerivedSmartPointerT<derived, base=""> DerivedSmartPointer;</derived,>

  TDerivedSmartPointerT(){};

  TDerivedSmartPointerT(DERIVED *pointer) : TSmartPointerT<derived>(pointer) {}</derived>

  TDerivedSmartPointerT(const TSmartPointerT<base> &p) {

    TSmartPointerT<derived>::m_pointer =</derived>

        dynamic_cast<derived *="">(p.getPointer());</derived>

    if (TSmartPointerT<derived>::m_pointer)</derived>

      TSmartPointerT<derived>::m_pointer->addRef();</derived>

  }

};

#endif