1 //===----------------------------------------------------------------------===//
2 //
3 //                     The LLVM Compiler Infrastructure
4 //
5 // This file is dual licensed under the MIT and the University of Illinois Open
6 // Source Licenses. See LICENSE.TXT for details.
7 //
8 //===----------------------------------------------------------------------===//
9 // UNSUPPORTED: c++03, c++11, c++14, c++17
10 
11 // template <class T>
12 //   constexpr int rotr(T x, unsigned int s) noexcept;
13 
14 // Constraints: T is an unsigned integer type
15 
16 #include <bit>
17 #include <cassert>
18 #include <cstdint>
19 #include <type_traits>
20 
21 #include "test_macros.h"
22 
23 struct A {};
24 enum       E1 : unsigned char { rEd };
25 enum class E2 : unsigned char { red };
26 
27 template <class T>
28 constexpr bool test()
29 {
30     ASSERT_SAME_TYPE(decltype(std::rotr(T(), 0)), T);
31     ASSERT_NOEXCEPT(std::rotr(T(), 0));
32     T max = std::numeric_limits<T>::max();
33     T highbit = std::rotr(T(1), 1);
34 
35     assert(std::rotr(T(max - 1), 0) == T(max - 1));
36     assert(std::rotr(T(max - 1), 1) == T(max - highbit));
37     assert(std::rotr(T(max - 1), 2) == T(max - (highbit >> 1)));
38     assert(std::rotr(T(max - 1), 3) == T(max - (highbit >> 2)));
39     assert(std::rotr(T(max - 1), 4) == T(max - (highbit >> 3)));
40     assert(std::rotr(T(max - 1), 5) == T(max - (highbit >> 4)));
41     assert(std::rotr(T(max - 1), 6) == T(max - (highbit >> 5)));
42     assert(std::rotr(T(max - 1), 7) == T(max - (highbit >> 6)));
43 
44     assert(std::rotr(T(128), 0) == T(128));
45     assert(std::rotr(T(128), 1) == T(64));
46     assert(std::rotr(T(128), 2) == T(32));
47     assert(std::rotr(T(128), 3) == T(16));
48     assert(std::rotr(T(128), 4) == T(8));
49     assert(std::rotr(T(128), 5) == T(4));
50     assert(std::rotr(T(128), 6) == T(2));
51     assert(std::rotr(T(128), 7) == T(1));
52 
53 #ifndef _LIBCPP_HAS_NO_INT128
54     if constexpr (std::is_same_v<T, __uint128_t>) {
55         T val = (T(1) << 63) | (T(1) << 64);
56         assert(std::rotr(val, 0) == val);
57         assert(std::rotr(val, 128) == val);
58         assert(std::rotr(val, 256) == val);
59         assert(std::rotr(val, 1) == val >> 1);
60         assert(std::rotr(val, 127) == val << 1);
61         assert(std::rotr(T(3), 1) == ((T(1) << 127) | T(1)));
62     }
63 #endif
64 
65     return true;
66 }
67 
68 int main(int, char**)
69 {
70     {
71     auto lambda = [](auto x) -> decltype(std::rotl(x, 1U)) {};
72     using L = decltype(lambda);
73 
74     static_assert(!std::is_invocable_v<L, signed char>);
75     static_assert(!std::is_invocable_v<L, short>);
76     static_assert(!std::is_invocable_v<L, int>);
77     static_assert(!std::is_invocable_v<L, long>);
78     static_assert(!std::is_invocable_v<L, long long>);
79 #ifndef _LIBCPP_HAS_NO_INT128
80     static_assert(!std::is_invocable_v<L, __int128_t>);
81 #endif
82 
83     static_assert(!std::is_invocable_v<L, int8_t>);
84     static_assert(!std::is_invocable_v<L, int16_t>);
85     static_assert(!std::is_invocable_v<L, int32_t>);
86     static_assert(!std::is_invocable_v<L, int64_t>);
87     static_assert(!std::is_invocable_v<L, intmax_t>);
88     static_assert(!std::is_invocable_v<L, intptr_t>);
89     static_assert(!std::is_invocable_v<L, ptrdiff_t>);
90 
91     static_assert(!std::is_invocable_v<L, bool>);
92     static_assert(!std::is_invocable_v<L, char>);
93     static_assert(!std::is_invocable_v<L, wchar_t>);
94 #ifndef _LIBCPP_HAS_NO_CHAR8_T
95     static_assert(!std::is_invocable_v<L, char8_t>);
96 #endif
97     static_assert(!std::is_invocable_v<L, char16_t>);
98     static_assert(!std::is_invocable_v<L, char32_t>);
99 
100     static_assert(!std::is_invocable_v<L, A>);
101     static_assert(!std::is_invocable_v<L, A*>);
102     static_assert(!std::is_invocable_v<L, E1>);
103     static_assert(!std::is_invocable_v<L, E2>);
104     }
105 
106     static_assert(test<unsigned char>());
107     static_assert(test<unsigned short>());
108     static_assert(test<unsigned int>());
109     static_assert(test<unsigned long>());
110     static_assert(test<unsigned long long>());
111 #ifndef _LIBCPP_HAS_NO_INT128
112     static_assert(test<__uint128_t>());
113 #endif
114 
115     static_assert(test<uint8_t>());
116     static_assert(test<uint16_t>());
117     static_assert(test<uint32_t>());
118     static_assert(test<uint64_t>());
119     static_assert(test<uintmax_t>());
120     static_assert(test<uintptr_t>());
121     static_assert(test<size_t>());
122 
123     test<unsigned char>();
124     test<unsigned short>();
125     test<unsigned int>();
126     test<unsigned long>();
127     test<unsigned long long>();
128 #ifndef _LIBCPP_HAS_NO_INT128
129     test<__uint128_t>();
130 #endif
131 
132     test<uint8_t>();
133     test<uint16_t>();
134     test<uint32_t>();
135     test<uint64_t>();
136     test<uintmax_t>();
137     test<uintptr_t>();
138     test<size_t>();
139 
140     return 0;
141 }
142