#pragma once
#include <wtf/CagedPtr.h>
namespace WTF {
template<Gigacage::Kind kind, typename T, bool shouldTag = false>
class CagedUniquePtr : public CagedPtr<kind, T, shouldTag> {
static_assert(std::is_trivially_destructible<T>::value, "We expect the contents of a caged pointer to be trivially destructable.");
public:
using Base = CagedPtr<kind, T, shouldTag>;
CagedUniquePtr() = default;
CagedUniquePtr(T* ptr, unsigned size)
: Base(ptr, size)
{ }
CagedUniquePtr(CagedUniquePtr&& ptr)
: Base(std::forward<CagedUniquePtr&&>(ptr))
{ }
CagedUniquePtr(const CagedUniquePtr&) = delete;
template<typename... Arguments>
static CagedUniquePtr create(unsigned length, Arguments&&... arguments)
{
T* result = static_cast<T*>(Gigacage::malloc(kind, sizeof(T) * length));
while (length--)
new (result + length) T(arguments...);
return CagedUniquePtr(result, length);
}
template<typename... Arguments>
static CagedUniquePtr tryCreate(unsigned length, Arguments&&... arguments)
{
T* result = static_cast<T*>(Gigacage::tryMalloc(kind, sizeof(T) * length));
if (!result)
return { };
while (length--)
new (result + length) T(arguments...);
return CagedUniquePtr(result, length);
}
CagedUniquePtr& operator=(CagedUniquePtr&& ptr)
{
destroy();
this->m_ptr = ptr.m_ptr;
ptr.m_ptr = nullptr;
return *this;
}
CagedUniquePtr& operator=(const CagedUniquePtr&) = delete;
~CagedUniquePtr()
{
destroy();
}
private:
void destroy()
{
T* ptr = Base::getUnsafe();
if (!ptr)
return;
ptr->~T();
Gigacage::free(kind, ptr);
}
};
}
using WTF::CagedUniquePtr;