Class CachedWorkspaceMem

python/mod_cvcuda/WorkspaceCache.hpp:46–107

template<MemoryKind kind>
class CachedWorkspaceMem : public cvcuda::WorkspaceMem
{
public:
    CachedWorkspaceMem()
        : cvcuda::WorkspaceMem({})
    {
        assert(data == nullptr);
        assert(ready == nullptr);
    }

    CachedWorkspaceMem(const cvcuda::WorkspaceMem &mem, WorkspaceMemDestructor_t destructor)
        : cvcuda::WorkspaceMem(mem)
        , m_destructor(std::move(destructor))
    {
    }

    CachedWorkspaceMem(CachedWorkspaceMem &&mem)
    {
        *this = std::move(mem);
    }

    CachedWorkspaceMem &operator=(CachedWorkspaceMem &&mem)
    {
        std::swap(wsMem(), mem.wsMem());
        std::swap(m_destructor, mem.m_destructor);
        mem.reset();
        return *this;
    }

    ~CachedWorkspaceMem()
    {
        reset();
    }

    void reset()
    {
        if (m_destructor)
        {
            m_destructor(*this);
            m_destructor = {};
        }
        wsMem() = {};
    }

    explicit operator bool() const noexcept
    {
        return data != nullptr;
    }

private:
    cvcuda::WorkspaceMem &wsMem() &
    {
        return static_cast<cvcuda::WorkspaceMem &>(*this);
    }

    const cvcuda::WorkspaceMem &wsMem() const &
    {
        return static_cast<const cvcuda::WorkspaceMem &>(*this);