mirror of https://github.com/PentHertz/srsLTE.git
79 lines
2.2 KiB
C
79 lines
2.2 KiB
C
|
/**
|
||
|
*
|
||
|
* \section COPYRIGHT
|
||
|
*
|
||
|
* Copyright 2013-2021 Software Radio Systems Limited
|
||
|
*
|
||
|
* By using this file, you agree to the terms and conditions set
|
||
|
* forth in the LICENSE file which can be found at the top level of
|
||
|
* the distribution.
|
||
|
*
|
||
|
*/
|
||
|
|
||
|
#ifndef SRSRAN_LINEAR_ALLOCATOR_H
|
||
|
#define SRSRAN_LINEAR_ALLOCATOR_H
|
||
|
|
||
|
#include "pool_utils.h"
|
||
|
#include "srsran/common/srsran_assert.h"
|
||
|
|
||
|
namespace srsran {
|
||
|
|
||
|
class linear_allocator
|
||
|
{
|
||
|
public:
|
||
|
linear_allocator() = default;
|
||
|
linear_allocator(void* start_, void* end_) :
|
||
|
start(static_cast<uint8_t*>(start_)), end(static_cast<uint8_t*>(end_)), cur(start)
|
||
|
{}
|
||
|
linear_allocator(void* start_, size_t sz) : start(static_cast<uint8_t*>(start_)), end(start + sz), cur(start) {}
|
||
|
linear_allocator(const linear_allocator& other) = delete;
|
||
|
linear_allocator(linear_allocator&& other) noexcept : start(other.start), end(other.end), cur(other.cur)
|
||
|
{
|
||
|
other.clear();
|
||
|
}
|
||
|
linear_allocator& operator=(const linear_allocator& other) = delete;
|
||
|
linear_allocator& operator =(linear_allocator&& other) noexcept
|
||
|
{
|
||
|
start = other.start;
|
||
|
end = other.end;
|
||
|
cur = other.cur;
|
||
|
other.clear();
|
||
|
return *this;
|
||
|
}
|
||
|
|
||
|
void* allocate(size_t sz, size_t alignment)
|
||
|
{
|
||
|
void* alloc_start = align_to(cur, alignment);
|
||
|
cur = static_cast<uint8_t*>(alloc_start) + sz;
|
||
|
srsran_assert(cur <= end,
|
||
|
"Linear Allocator buffer of size=%zd was exceeded (current size=%zd)",
|
||
|
size(),
|
||
|
nof_bytes_allocated());
|
||
|
return alloc_start;
|
||
|
}
|
||
|
|
||
|
void deallocate(void* p) { srsran_assert(p >= start and p < end, "pointer does not belong to pool"); }
|
||
|
|
||
|
size_t nof_bytes_allocated() const { return cur - start; }
|
||
|
size_t nof_bytes_left() const { return end - cur; }
|
||
|
size_t size() const { return end - start; }
|
||
|
bool is_init() const { return start != end; }
|
||
|
void* memblock_ptr() { return static_cast<void*>(start); }
|
||
|
|
||
|
void clear()
|
||
|
{
|
||
|
start = nullptr;
|
||
|
cur = nullptr;
|
||
|
end = nullptr;
|
||
|
}
|
||
|
|
||
|
protected:
|
||
|
uint8_t* start = nullptr;
|
||
|
uint8_t* end = nullptr;
|
||
|
uint8_t* cur = nullptr;
|
||
|
};
|
||
|
|
||
|
} // namespace srsran
|
||
|
|
||
|
#endif // SRSRAN_LINEAR_ALLOCATOR_H
|