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