| 1 | // Copyright (c) 2015-2016 The Khronos Group Inc. | 
| 2 | // | 
| 3 | // Licensed under the Apache License, Version 2.0 (the "License"); | 
| 4 | // you may not use this file except in compliance with the License. | 
| 5 | // You may obtain a copy of the License at | 
| 6 | // | 
| 7 | //     http://www.apache.org/licenses/LICENSE-2.0 | 
| 8 | // | 
| 9 | // Unless required by applicable law or agreed to in writing, software | 
| 10 | // distributed under the License is distributed on an "AS IS" BASIS, | 
| 11 | // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. | 
| 12 | // See the License for the specific language governing permissions and | 
| 13 | // limitations under the License. | 
| 14 |  | 
| 15 | #ifndef LIBSPIRV_UTIL_BITUTILS_H_ | 
| 16 | #define LIBSPIRV_UTIL_BITUTILS_H_ | 
| 17 |  | 
| 18 | #include <cstdint> | 
| 19 | #include <cstring> | 
| 20 |  | 
| 21 | namespace spvutils { | 
| 22 |  | 
| 23 | // Performs a bitwise copy of source to the destination type Dest. | 
| 24 | template <typename Dest, typename Src> | 
| 25 | Dest BitwiseCast(Src source) { | 
| 26 |   Dest dest; | 
| 27 |   static_assert(sizeof(source) == sizeof(dest), | 
| 28 |                 "BitwiseCast: Source and destination must have the same size" ); | 
| 29 |   std::memcpy(dest: static_cast<void*>(&dest), src: &source, n: sizeof(dest)); | 
| 30 |   return dest; | 
| 31 | } | 
| 32 |  | 
| 33 | // SetBits<T, First, Num> returns an integer of type <T> with bits set | 
| 34 | // for position <First> through <First + Num - 1>, counting from the least | 
| 35 | // significant bit. In particular when Num == 0, no positions are set to 1. | 
| 36 | // A static assert will be triggered if First + Num > sizeof(T) * 8, that is, | 
| 37 | // a bit that will not fit in the underlying type is set. | 
| 38 | template <typename T, size_t First = 0, size_t Num = 0> | 
| 39 | struct SetBits { | 
| 40 |   static_assert(First < sizeof(T) * 8, | 
| 41 |                 "Tried to set a bit that is shifted too far." ); | 
| 42 |   const static T get = (T(1) << First) | SetBits<T, First + 1, Num - 1>::get; | 
| 43 | }; | 
| 44 |  | 
| 45 | template <typename T, size_t Last> | 
| 46 | struct SetBits<T, Last, 0> { | 
| 47 |   const static T get = T(0); | 
| 48 | }; | 
| 49 |  | 
| 50 | // This is all compile-time so we can put our tests right here. | 
| 51 | static_assert(SetBits<uint32_t, 0, 0>::get == uint32_t(0x00000000), | 
| 52 |               "SetBits failed" ); | 
| 53 | static_assert(SetBits<uint32_t, 0, 1>::get == uint32_t(0x00000001), | 
| 54 |               "SetBits failed" ); | 
| 55 | static_assert(SetBits<uint32_t, 31, 1>::get == uint32_t(0x80000000), | 
| 56 |               "SetBits failed" ); | 
| 57 | static_assert(SetBits<uint32_t, 1, 2>::get == uint32_t(0x00000006), | 
| 58 |               "SetBits failed" ); | 
| 59 | static_assert(SetBits<uint32_t, 30, 2>::get == uint32_t(0xc0000000), | 
| 60 |               "SetBits failed" ); | 
| 61 | static_assert(SetBits<uint32_t, 0, 31>::get == uint32_t(0x7FFFFFFF), | 
| 62 |               "SetBits failed" ); | 
| 63 | static_assert(SetBits<uint32_t, 0, 32>::get == uint32_t(0xFFFFFFFF), | 
| 64 |               "SetBits failed" ); | 
| 65 | static_assert(SetBits<uint32_t, 16, 16>::get == uint32_t(0xFFFF0000), | 
| 66 |               "SetBits failed" ); | 
| 67 |  | 
| 68 | static_assert(SetBits<uint64_t, 0, 1>::get == uint64_t(0x0000000000000001LL), | 
| 69 |               "SetBits failed" ); | 
| 70 | static_assert(SetBits<uint64_t, 63, 1>::get == uint64_t(0x8000000000000000LL), | 
| 71 |               "SetBits failed" ); | 
| 72 | static_assert(SetBits<uint64_t, 62, 2>::get == uint64_t(0xc000000000000000LL), | 
| 73 |               "SetBits failed" ); | 
| 74 | static_assert(SetBits<uint64_t, 31, 1>::get == uint64_t(0x0000000080000000LL), | 
| 75 |               "SetBits failed" ); | 
| 76 | static_assert(SetBits<uint64_t, 16, 16>::get == uint64_t(0x00000000FFFF0000LL), | 
| 77 |               "SetBits failed" ); | 
| 78 |  | 
| 79 | }  // namespace spvutils | 
| 80 |  | 
| 81 | #endif  // LIBSPIRV_UTIL_BITUTILS_H_ | 
| 82 |  |