ResourcePointer.h

#pragma once
 
#include "Utilities.h"
 
namespace Cogs
{
  template<class T>
  struct ResourcePointer
  {
    ResourcePointer() : resource(nullptr) {}
    ResourcePointer(T * resource) : resource(resource) {}
 
    ResourcePointer(const ResourcePointer & resourcePointer)
    {
     this->resource = resourcePointer.resource;
 
     if (this->resource) {
       this->resource->AddRef();
     }
    }
 
    ResourcePointer(ResourcePointer && resourcePointer)
    {
      this->resource = resourcePointer.resource;
      resourcePointer.resource = nullptr;
    }
 
    ~ResourcePointer() { this->release(); }
 
    void release()
    {
      T * t = this->resource;
 
      if (t) {
        this->resource = nullptr;
        if (!t->Release()) {
          UnregisterGPUResource(t);
        }
      }
    }
 
    ResourcePointer & operator=(const ResourcePointer & resourcePointer)
    {
     this->release();
 
     this->resource = resourcePointer.resource;
 
     if (this->resource) {
       this->resource->AddRef();
     }
 
     return *this;
    }
 
    ResourcePointer & operator=(ResourcePointer && resourcePointer) noexcept
    {
      this->release();
 
      this->resource = resourcePointer.resource;
      resourcePointer.resource = nullptr;
 
      return *this;
    }
 
    T **      internalPointer()       { this->release(); return &resource; }
    void **   internalVoidPointer()   { this->release(); return reinterpret_cast<void**>(&resource); }
 
    T * const * operator &() const { return &resource; }
 
    operator T *() const { return resource; }
 
    T * operator->() { return resource; }
 
    const T * operator->() const { return resource; }
 
    T & operator*() { return *resource; }
 
    const T & operator*() const { return *resource; }
 
    template<typename U>
    U * get()
    {
      return static_cast<U *>(resource);
    }
 
  private:
    T * resource;
  };
}