allocator.hpp

Go to the documentation of this file.

// SPDX-FileCopyrightText: 2025 - 2026 NeoN authors
//
// SPDX-License-Identifier: MIT
 
#pragma once
 
#include "NeoN/core/error.hpp"
#include "NeoN/core/primitives/label.hpp"
 
#include <memory>
 
 
namespace NeoN
{
 
 
enum class MemorySpace
{
    CPU,
    GPU
};
 
 
class AllocatorStrategy
{
 
protected:
 
    // TODO check if enforcing via Ctor is an option
    MemorySpace memSpace_ = MemorySpace::CPU;
 
public:

    void setMemSpace(MemorySpace memSpace) { memSpace_ = memSpace; }

    virtual void* alloc(sizeType size) = 0;

    virtual void* realloc(void* ptr, sizeType size) = 0;

    virtual void free(void* ptr) = 0;
 
    virtual ~AllocatorStrategy() {}
};
 
 
class AllocatorContext
{
 
    std::unique_ptr<AllocatorStrategy> strategy {};
 
public:
 
    void setStrategy(std::unique_ptr<AllocatorStrategy> strategyIn)
    {
        strategy = std::move(strategyIn);
    }

    template<typename T>
    T* alloc(sizeType elements) const
    {
        if (!strategy)
        {
            NF_ERROR_EXIT("No allocator set");
        }
        return reinterpret_cast<T*>(strategy->alloc(sizeof(T) * elements));
    }

    template<typename T>
    T* realloc(void* ptr, sizeType elements) const
    {
        if (!strategy)
        {
            NF_ERROR_EXIT("No allocator set");
        }
        return reinterpret_cast<T*>(strategy->realloc(ptr, sizeof(T) * elements));
    }
 
    void free(void* ptr) const
    {
        if (!strategy)
        {
            NF_ERROR_EXIT("No allocator set");
        }
        return strategy->free(ptr);
    }
};
 
 
}