PX4-Autopilot/libuavcan/src/uc_dynamic_memory.cpp

/*
 * Copyright (C) 2014 Pavel Kirienko <pavel.kirienko@gmail.com>
 */

#include <uavcan/dynamic_memory.hpp>

namespace uavcan
{
/*
 * LimitedPoolAllocator
 */
void* LimitedPoolAllocator::allocate(std::size_t size)
{
    if (used_blocks_ < max_blocks_)
    {
        used_blocks_++;
        return allocator_.allocate(size);
    }
    else
    {
        return NULL;
    }
}

void LimitedPoolAllocator::deallocate(const void* ptr)
{
    allocator_.deallocate(ptr);

    assert(used_blocks_ > 0);
    if (used_blocks_ > 0)
    {
        used_blocks_--;
    }
}

bool LimitedPoolAllocator::isInPool(const void* ptr) const
{
    return allocator_.isInPool(ptr);
}

std::size_t LimitedPoolAllocator::getBlockSize() const
{
    return allocator_.getBlockSize();
}

std::size_t LimitedPoolAllocator::getNumBlocks() const
{
    return std::min(max_blocks_, allocator_.getNumBlocks());
}

}