-
Notifications
You must be signed in to change notification settings - Fork 0
/
aligned_buf_impl.hpp
82 lines (72 loc) · 1.31 KB
/
aligned_buf_impl.hpp
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
// SPDX-License-Identifier: GPL-2.0
#include <cstdlib>
template <typename T>
T* assume_aligned(T *d)
{
return static_cast<T*>(__builtin_assume_aligned(d, AlignedBuf<T>::align));
}
template <typename T>
AlignedBuf<T>::AlignedBuf()
: buf(nullptr)
{
}
template<typename T>
AlignedBuf<T>::AlignedBuf(size_t n)
// Windows doesn't have the standard(!) aligned_alloc() function.
// Instead it provides an _aligned_malloc() with reversed arguments.
#ifdef _WIN32
: buf(reinterpret_cast<T*>(_aligned_malloc(n * sizeof(T), align)))
#else
: buf(reinterpret_cast<T*>(std::aligned_alloc(align, n * sizeof(T))))
#endif
{
}
template<typename T>
AlignedBuf<T>::AlignedBuf(AlignedBuf &&b)
: buf(b.buf)
{
b.buf = nullptr;
}
template<typename T>
AlignedBuf<T> &AlignedBuf<T>::operator=(AlignedBuf &&b)
{
if (buf)
#ifdef _WIN32
_aligned_free(buf);
#else
free(buf);
#endif
buf = b.buf;
b.buf = nullptr;
return *this;
}
template<typename T>
AlignedBuf<T>::~AlignedBuf()
{
if (buf)
#ifdef _WIN32
_aligned_free(buf);
#else
free(buf);
#endif
}
template<typename T>
bool AlignedBuf<T>::operator!() const
{
return !buf;
}
template<typename T>
AlignedBuf<T>::operator bool() const
{
return !!buf;
}
template<typename T>
T *AlignedBuf<T>::get()
{
return buf;
}
template<typename T>
const T *AlignedBuf<T>::get() const
{
return buf;
}