libc/src/__support/CPP/cstddef.h

   1 //===-- A self contained equivalent of cstddef ------------------*- C++ -*-===//
   2 //
   3 // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
   4 // See https://llvm.org/LICENSE.txt for license information.
   5 // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
   6 //
   7 //===----------------------------------------------------------------------===//
   8
   9 #ifndef LLVM_LIBC_SRC_SUPPORT_CPP_BYTE_H
  10 #define LLVM_LIBC_SRC_SUPPORT_CPP_BYTE_H
  11
  12 #include "type_traits.h" // For enable_if_t, is_integral_v.
  13
  14 namespace __llvm_libc::cpp {
  15
  16 enum class byte : unsigned char {};
  17
  18 template <class IntegerType>
  19 constexpr enable_if_t<is_integral_v<IntegerType>, byte>
  20 operator>>(byte b, IntegerType shift) noexcept {
  21   return static_cast<byte>(static_cast<unsigned char>(b) >> shift);
  22 }
  23 template <class IntegerType>
  24 constexpr enable_if_t<is_integral_v<IntegerType>, byte &>
  25 operator>>=(byte &b, IntegerType shift) noexcept {
  26   return b = b >> shift;
  27 }
  28 template <class IntegerType>
  29 constexpr enable_if_t<is_integral_v<IntegerType>, byte>
  30 operator<<(byte b, IntegerType shift) noexcept {
  31   return static_cast<byte>(static_cast<unsigned char>(b) << shift);
  32 }
  33 template <class IntegerType>
  34 constexpr enable_if_t<is_integral_v<IntegerType>, byte &>
  35 operator<<=(byte &b, IntegerType shift) noexcept {
  36   return b = b << shift;
  37 }
  38 constexpr byte operator|(byte l, byte r) noexcept {
  39   return static_cast<byte>(static_cast<unsigned char>(l) |
  40                            static_cast<unsigned char>(r));
  41 }
  42 constexpr byte &operator|=(byte &l, byte r) noexcept { return l = l | r; }
  43 constexpr byte operator&(byte l, byte r) noexcept {
  44   return static_cast<byte>(static_cast<unsigned char>(l) &
  45                            static_cast<unsigned char>(r));
  46 }
  47 constexpr byte &operator&=(byte &l, byte r) noexcept { return l = l & r; }
  48 constexpr byte operator^(byte l, byte r) noexcept {
  49   return static_cast<byte>(static_cast<unsigned char>(l) ^
  50                            static_cast<unsigned char>(r));
  51 }
  52 constexpr byte &operator^=(byte &l, byte r) noexcept { return l = l ^ r; }
  53 constexpr byte operator~(byte b) noexcept {
  54   return static_cast<byte>(~static_cast<unsigned char>(b));
  55 }
  56 template <typename IntegerType>
  57 constexpr enable_if_t<is_integral_v<IntegerType>, IntegerType>
  58 to_integer(byte b) noexcept {
  59   return static_cast<IntegerType>(b);
  60 }
  61
  62 } // namespace __llvm_libc::cpp
  63
  64 #endif // LLVM_LIBC_SRC_SUPPORT_CPP_BYTE_H