mirror of https://github.com/hb9fxq/gr-digitalhf
44 lines
1.0 KiB
C++
44 lines
1.0 KiB
C++
// -*- c++ -*-
|
|
|
|
#ifndef _LIB_VOLK_ALLOCATOR_HPP_
|
|
#define _LIB_VOLK_ALLOCATOR_HPP_
|
|
|
|
#include <cstdlib>
|
|
#include <limits>
|
|
#include <volk/volk.h>
|
|
|
|
namespace gr {
|
|
namespace digitalhf {
|
|
|
|
// see https://en.cppreference.com/w/cpp/named_req/Allocator
|
|
template <class T>
|
|
struct volk_allocator {
|
|
typedef T value_type;
|
|
|
|
volk_allocator() = default;
|
|
|
|
template <class U> constexpr volk_allocator(const volk_allocator<U>&) noexcept {}
|
|
|
|
T* allocate(std::size_t n) {
|
|
if (n > std::numeric_limits<std::size_t>::max() / sizeof(T)) throw std::bad_alloc();
|
|
|
|
if (auto p = static_cast<T*>(volk_malloc(n*sizeof(T), volk_get_alignment())))
|
|
return p;
|
|
|
|
throw std::bad_alloc();
|
|
}
|
|
|
|
void deallocate(T* p, std::size_t) noexcept { volk_free(p); }
|
|
|
|
} ;
|
|
|
|
template <class T, class U>
|
|
bool operator==(const volk_allocator<T>&, const volk_allocator<U>&) { return true; }
|
|
|
|
template <class T, class U>
|
|
bool operator!=(const volk_allocator<T>&, const volk_allocator<U>&) { return false; }
|
|
|
|
} // namespace digitalhf
|
|
} // namespace gr
|
|
#endif // _LIB_VOLK_ALLOCATOR_HPP_
|