allocated_ptr.hpp

浏览该文件的文档.

#ifndef NEFORCE_CORE_MEMORY_ALLOCATED_PTR_HPP__
#define NEFORCE_CORE_MEMORY_ALLOCATED_PTR_HPP__

 
#include "NeForce/core/memory/allocator_traits.hpp"
NEFORCE_BEGIN_NAMESPACE__


template <typename Alloc>
struct allocated_ptr {
    using pointer = typename allocator_traits<Alloc>::pointer;       
    using value_type = typename allocator_traits<Alloc>::value_type; 
 
private:
    Alloc* alloc_; 
    pointer ptr_;  
 
public:
    allocated_ptr(Alloc& alloc, pointer ptr) noexcept :
    alloc_(_NEFORCE addressof(alloc)),
    ptr_(ptr) {}

    template <typename Ptr, typename = enable_if_t<is_same_v<Ptr, value_type*>>>
    allocated_ptr(Alloc& alloc, Ptr ptr) :
    alloc_(_NEFORCE addressof(alloc)),
    ptr_(pointer_traits<pointer>::pointer_to(*ptr)) {}

    allocated_ptr(allocated_ptr&& guard) noexcept :
    alloc_(guard.alloc_),
    ptr_(guard.ptr_) {
        guard.ptr_ = nullptr;
    }

    ~allocated_ptr() {
        if (ptr_ != nullptr) {
            _NEFORCE allocator_traits<Alloc>::deallocate(*alloc_, ptr_, 1);
        }
    }

    allocated_ptr& operator=(nullptr_t null) noexcept {
        ptr_ = nullptr;
        return *this;
    }

    value_type* get() { return _NEFORCE to_address(ptr_); }
};

template <typename Alloc>
allocated_ptr<Alloc> allocate_guarded(Alloc& alloc) {
    return allocated_ptr<Alloc>{alloc, _NEFORCE allocator_traits<Alloc>::allocate(alloc, 1)};
}
 // AllocatedPtr
 
NEFORCE_END_NAMESPACE__
#endif // NEFORCE_CORE_MEMORY_ALLOCATED_PTR_HPP__