47 lines
1.1 KiB
C++
47 lines
1.1 KiB
C++
|
// This file is part of the mcl project.
|
||
|
|
// Copyright (c) 2022 merryhime
|
||
|
|
// SPDX-License-Identifier: MIT
|
||
|
|
|
||
|
|
#pragma once
|
||
|
|
|
||
|
|
#include <type_traits>
|
||
|
|
|
||
|
|
#ifdef _MSC_VER
|
||
|
|
# include <malloc.h>
|
||
|
|
#else
|
||
|
|
# include <cstdlib>
|
||
|
|
#endif
|
||
|
|
|
||
|
|
namespace mcl {
|
||
|
|
|
||
|
|
namespace detail {
|
||
|
|
struct aligned_alloc_deleter {
|
||
|
|
template<typename T>
|
||
|
|
void operator()(T* p) const
|
||
|
|
{
|
||
|
|
#ifdef _MSC_VER
|
||
|
|
_aligned_free(const_cast<std::remove_const_t<T>*>(p));
|
||
|
|
#else
|
||
|
|
std::free(const_cast<std::remove_const_t<T>*>(p));
|
||
|
|
#endif
|
||
|
|
}
|
||
|
|
};
|
||
|
|
} // namespace detail
|
||
|
|
|
||
|
|
template<size_t, typename T>
|
||
|
|
using overaligned_unique_ptr = std::unique_ptr<T, detail::aligned_alloc_deleter>;
|
||
|
|
|
||
|
|
template<size_t alignment, typename T>
|
||
|
|
auto make_overaligned_unique_ptr_array(size_t element_count)
|
||
|
|
{
|
||
|
|
const size_t min_size = element_count * sizeof(T);
|
||
|
|
const size_t alloc_size = (min_size + alignment - 1) / alignment * alignment;
|
||
|
|
#ifdef _MSC_VER
|
||
|
|
return overaligned_unique_ptr<alignment, T[]>{static_cast<T*>(_aligned_malloc(alloc_size, alignment))};
|
||
|
|
#else
|
||
|
|
return overaligned_unique_ptr<alignment, T[]>{static_cast<T*>(std::aligned_alloc(alignment, alloc_size))};
|
||
|
|
#endif
|
||
|
|
}
|
||
|
|
|
||
|
|
} // namespace mcl
|