pool.hpp

Go to the documentation of this file.

// =======================================
// pool.hpp - Fast aligned memory pool
// =======================================
 
 
 
#pragma once
 
#include <cstddef>
#include <cstdlib>
#include <new>
#include <mutex>
#include <cassert>

struct PoolAllocator {
    char* memory;                     
    std::size_t capacity;            
    std::size_t offset; 
 
public:
    explicit PoolAllocator(size_t bytes) {
        memory = static_cast<char*>(std::aligned_alloc(64, bytes));
        capacity = bytes;
        offset = 0;
        assert(memory && "Failed to allocate aligned memory");
    }
    
    ~PoolAllocator() {
        std::free(memory);
    }

    template<typename T>
    T* allocate(std::size_t align = alignof(T)) {
        std::size_t current = offset;
        offset += sizeof(T);  // 64-byte spacing to avoid false sharing
        
        if (current + sizeof(T) > capacity) return nullptr;
 
        char* ptr = memory + current;
        std::uintptr_t aligned = (reinterpret_cast<std::uintptr_t>(ptr) + (align - 1)) & ~(align - 1);
        
        return reinterpret_cast<T*>(aligned);
    }

    void reset() {
        offset = 0;
    }
};