11a036e9cSMark de Wever //===----------------------------------------------------------------------===// 21a036e9cSMark de Wever // 31ee41b41SNikolas Klauser // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions. 41ee41b41SNikolas Klauser // See https://llvm.org/LICENSE.txt for license information. 51ee41b41SNikolas Klauser // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception 61a036e9cSMark de Wever // 71a036e9cSMark de Wever //===----------------------------------------------------------------------===// 81ee41b41SNikolas Klauser 91a036e9cSMark de Wever // UNSUPPORTED: c++03, c++11, c++14, c++17 101a036e9cSMark de Wever 111a036e9cSMark de Wever // template <class T> 121a036e9cSMark de Wever // constexpr bool has_single_bit(T x) noexcept; 131a036e9cSMark de Wever 14e130fbe2SArthur O'Dwyer // Constraints: T is an unsigned integer type 151a036e9cSMark de Wever 161a036e9cSMark de Wever #include <bit> 17e130fbe2SArthur O'Dwyer #include <cassert> 18*d6832a61SLouis Dionne #include <cstddef> 191a036e9cSMark de Wever #include <cstdint> 2086aac87fSNikolas Klauser #include <limits> 211a036e9cSMark de Wever #include <type_traits> 221a036e9cSMark de Wever 231a036e9cSMark de Wever #include "test_macros.h" 241a036e9cSMark de Wever 25e130fbe2SArthur O'Dwyer struct A {}; 261a036e9cSMark de Wever enum E1 : unsigned char { rEd }; 271a036e9cSMark de Wever enum class E2 : unsigned char { red }; 281a036e9cSMark de Wever 29e130fbe2SArthur O'Dwyer template <class T> 30e130fbe2SArthur O'Dwyer constexpr bool test() 311a036e9cSMark de Wever { 32e130fbe2SArthur O'Dwyer ASSERT_SAME_TYPE(decltype(std::has_single_bit(T())), bool); 33e130fbe2SArthur O'Dwyer ASSERT_NOEXCEPT(std::has_single_bit(T())); 34e130fbe2SArthur O'Dwyer T max = std::numeric_limits<T>::max(); 351a036e9cSMark de Wever 36e130fbe2SArthur O'Dwyer assert(!std::has_single_bit(T(0))); 37e130fbe2SArthur O'Dwyer assert( std::has_single_bit(T(1))); 38e130fbe2SArthur O'Dwyer assert( std::has_single_bit(T(2))); 39e130fbe2SArthur O'Dwyer assert(!std::has_single_bit(T(3))); 40e130fbe2SArthur O'Dwyer assert( std::has_single_bit(T(4))); 41e130fbe2SArthur O'Dwyer assert(!std::has_single_bit(T(5))); 42e130fbe2SArthur O'Dwyer assert(!std::has_single_bit(T(6))); 43e130fbe2SArthur O'Dwyer assert(!std::has_single_bit(T(7))); 44e130fbe2SArthur O'Dwyer assert( std::has_single_bit(T(8))); 45e130fbe2SArthur O'Dwyer assert(!std::has_single_bit(T(9))); 461a036e9cSMark de Wever assert(!std::has_single_bit(T(127))); 471a036e9cSMark de Wever assert( std::has_single_bit(T(128))); 481a036e9cSMark de Wever assert(!std::has_single_bit(T(129))); 49e130fbe2SArthur O'Dwyer assert(!std::has_single_bit(max)); 50e130fbe2SArthur O'Dwyer 518f972cb0SMark de Wever #ifndef TEST_HAS_NO_INT128 52e130fbe2SArthur O'Dwyer if constexpr (std::is_same_v<T, __uint128_t>) { 53e130fbe2SArthur O'Dwyer T val = T(1) << 32; 54e130fbe2SArthur O'Dwyer assert(!std::has_single_bit(val-1)); 55e130fbe2SArthur O'Dwyer assert( std::has_single_bit(val)); 56e130fbe2SArthur O'Dwyer assert(!std::has_single_bit(val+1)); 57e130fbe2SArthur O'Dwyer val <<= 60; 58e130fbe2SArthur O'Dwyer assert(!std::has_single_bit(val-1)); 59e130fbe2SArthur O'Dwyer assert( std::has_single_bit(val)); 60e130fbe2SArthur O'Dwyer assert(!std::has_single_bit(val+1)); 61e130fbe2SArthur O'Dwyer 62e130fbe2SArthur O'Dwyer T x = (T(1) << 63); 63e130fbe2SArthur O'Dwyer T y = (T(1) << 64); 64e130fbe2SArthur O'Dwyer assert( std::has_single_bit(x)); 65e130fbe2SArthur O'Dwyer assert( std::has_single_bit(y)); 66e130fbe2SArthur O'Dwyer assert(!std::has_single_bit(x + y)); 67e130fbe2SArthur O'Dwyer } 68e130fbe2SArthur O'Dwyer #endif 69e130fbe2SArthur O'Dwyer 70e130fbe2SArthur O'Dwyer return true; 711a036e9cSMark de Wever } 721a036e9cSMark de Wever 731a036e9cSMark de Wever int main(int, char**) 741a036e9cSMark de Wever { 751a036e9cSMark de Wever { 761a036e9cSMark de Wever auto lambda = [](auto x) -> decltype(std::has_single_bit(x)) {}; 771a036e9cSMark de Wever using L = decltype(lambda); 781a036e9cSMark de Wever 79e130fbe2SArthur O'Dwyer static_assert(!std::is_invocable_v<L, signed char>); 80e130fbe2SArthur O'Dwyer static_assert(!std::is_invocable_v<L, short>); 81e130fbe2SArthur O'Dwyer static_assert(!std::is_invocable_v<L, int>); 82e130fbe2SArthur O'Dwyer static_assert(!std::is_invocable_v<L, long>); 83e130fbe2SArthur O'Dwyer static_assert(!std::is_invocable_v<L, long long>); 848f972cb0SMark de Wever #ifndef TEST_HAS_NO_INT128 85e130fbe2SArthur O'Dwyer static_assert(!std::is_invocable_v<L, __int128_t>); 861a036e9cSMark de Wever #endif 871a036e9cSMark de Wever 88bd5d0feeSMark de Wever static_assert(!std::is_invocable_v<L, std::int8_t>); 89bd5d0feeSMark de Wever static_assert(!std::is_invocable_v<L, std::int16_t>); 90bd5d0feeSMark de Wever static_assert(!std::is_invocable_v<L, std::int32_t>); 91bd5d0feeSMark de Wever static_assert(!std::is_invocable_v<L, std::int64_t>); 9292fc93e1SMark de Wever static_assert(!std::is_invocable_v<L, std::intmax_t>); 93d59a43feSMark de Wever static_assert(!std::is_invocable_v<L, std::intptr_t>); 94d8681356SMark de Wever static_assert(!std::is_invocable_v<L, std::ptrdiff_t>); 95e130fbe2SArthur O'Dwyer 96e130fbe2SArthur O'Dwyer static_assert(!std::is_invocable_v<L, bool>); 97e130fbe2SArthur O'Dwyer static_assert(!std::is_invocable_v<L, char>); 98e130fbe2SArthur O'Dwyer static_assert(!std::is_invocable_v<L, wchar_t>); 999027887eSMark de Wever #ifndef TEST_HAS_NO_CHAR8_T 100e130fbe2SArthur O'Dwyer static_assert(!std::is_invocable_v<L, char8_t>); 101e130fbe2SArthur O'Dwyer #endif 102e130fbe2SArthur O'Dwyer static_assert(!std::is_invocable_v<L, char16_t>); 103e130fbe2SArthur O'Dwyer static_assert(!std::is_invocable_v<L, char32_t>); 104e130fbe2SArthur O'Dwyer 105e130fbe2SArthur O'Dwyer static_assert(!std::is_invocable_v<L, A>); 106e130fbe2SArthur O'Dwyer static_assert(!std::is_invocable_v<L, A*>); 107e130fbe2SArthur O'Dwyer static_assert(!std::is_invocable_v<L, E1>); 108e130fbe2SArthur O'Dwyer static_assert(!std::is_invocable_v<L, E2>); 1091a036e9cSMark de Wever } 1101a036e9cSMark de Wever 111e130fbe2SArthur O'Dwyer static_assert(test<unsigned char>()); 112e130fbe2SArthur O'Dwyer static_assert(test<unsigned short>()); 113e130fbe2SArthur O'Dwyer static_assert(test<unsigned int>()); 114e130fbe2SArthur O'Dwyer static_assert(test<unsigned long>()); 115e130fbe2SArthur O'Dwyer static_assert(test<unsigned long long>()); 1168f972cb0SMark de Wever #ifndef TEST_HAS_NO_INT128 117e130fbe2SArthur O'Dwyer static_assert(test<__uint128_t>()); 1181a036e9cSMark de Wever #endif 119bd5d0feeSMark de Wever static_assert(test<std::uint8_t>()); 120bd5d0feeSMark de Wever static_assert(test<std::uint16_t>()); 121da79d6e1SMark de Wever static_assert(test<std::uint32_t>()); 122bd5d0feeSMark de Wever static_assert(test<std::uint64_t>()); 12392fc93e1SMark de Wever static_assert(test<std::uintmax_t>()); 124d59a43feSMark de Wever static_assert(test<std::uintptr_t>()); 125fb855eb9SMark de Wever static_assert(test<std::size_t>()); 1261a036e9cSMark de Wever 127e130fbe2SArthur O'Dwyer test<unsigned char>(); 128e130fbe2SArthur O'Dwyer test<unsigned short>(); 129e130fbe2SArthur O'Dwyer test<unsigned int>(); 130e130fbe2SArthur O'Dwyer test<unsigned long>(); 131e130fbe2SArthur O'Dwyer test<unsigned long long>(); 1328f972cb0SMark de Wever #ifndef TEST_HAS_NO_INT128 133e130fbe2SArthur O'Dwyer test<__uint128_t>(); 1341a036e9cSMark de Wever #endif 135bd5d0feeSMark de Wever test<std::uint8_t>(); 136bd5d0feeSMark de Wever test<std::uint16_t>(); 137da79d6e1SMark de Wever test<std::uint32_t>(); 138bd5d0feeSMark de Wever test<std::uint64_t>(); 13992fc93e1SMark de Wever test<std::uintmax_t>(); 140d59a43feSMark de Wever test<std::uintptr_t>(); 141fb855eb9SMark de Wever test<std::size_t>(); 1421a036e9cSMark de Wever 1431a036e9cSMark de Wever return 0; 1441a036e9cSMark de Wever } 145