PrimeWorldEditor/src/Core/Resource/TResPtr.h

#ifndef TRESPTR_H
#define TRESPTR_H

#include "CResource.h"

template <typename ResType>
class TResPtr
{
    ResType *mpRes;

public:
    TResPtr()
        : mpRes(nullptr) {}

    TResPtr(void *pPtr)
        : mpRes(nullptr)
    {
        *this = pPtr;
    }

    TResPtr(const TResPtr<ResType>& rkSource)
        : mpRes(nullptr)
    {
        *this = rkSource;
    }

    ~TResPtr()
    {
        if (mpRes)
            mpRes->Release();
    }

    inline void Serialize(IArchive& rArc)
    {
        CAssetID ID = (mpRes && !rArc.IsReader() ? mpRes->ID() : CAssetID::InvalidID(rArc.Game()));
        rArc.SerializePrimitive(ID);

        if (rArc.IsReader())
        {
            CResourceEntry *pEntry = gpResourceStore->FindEntry(ID);
            *this = (pEntry ? pEntry->Load() : nullptr);
        }
    }

    inline void Delete()
    {
        // use with caution! this function exists because not all resources are cached currently
        delete mpRes;
        mpRes = nullptr;
    }

    inline ResType* RawPointer() const
    {
        return mpRes;
    }

    inline operator ResType*() const
    {
        return mpRes;
    }

    inline ResType& operator*() const
    {
        return *mpRes;
    }

    inline ResType* operator->() const
    {
        return mpRes;
    }

    TResPtr<ResType>& operator=(void *pPtr)
    {
        // todo: this probably crashes if you try to pass a non-CResource-derived pointer. is there a safer way?
        // dynamic_cast seems like the best option for type-checking here. other methods don't catch inheritance
        // eg if you have a TResPtr<CResource> dynamic_cast is the only method that will allow derived resource types
        if (mpRes)
            mpRes->Release();

        CResource *pRes = (CResource*) pPtr;
        mpRes = dynamic_cast<ResType*>(pRes);

        if (mpRes)
            mpRes->Lock();

        return *this;
    }

    TResPtr<ResType>& operator=(const TResPtr<ResType>& rkRight)
    {
        if (mpRes)
            mpRes->Release();

        mpRes = rkRight.mpRes;

        if (mpRes)
            mpRes->Lock();

        return *this;
    }
};

#endif // TRESPTR_H