149 lines
5.0 KiB
C++
149 lines
5.0 KiB
C++
// Copyright 2017 The Dawn Authors
|
|
//
|
|
// Licensed under the Apache License, Version 2.0 (the "License");
|
|
// you may not use this file except in compliance with the License.
|
|
// You may obtain a copy of the License at
|
|
//
|
|
// http://www.apache.org/licenses/LICENSE-2.0
|
|
//
|
|
// Unless required by applicable law or agreed to in writing, software
|
|
// distributed under the License is distributed on an "AS IS" BASIS,
|
|
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
// See the License for the specific language governing permissions and
|
|
// limitations under the License.
|
|
|
|
#ifndef INCLUDE_DAWN_ENUMCLASSBITMASKS_H_
|
|
#define INCLUDE_DAWN_ENUMCLASSBITMASKS_H_
|
|
|
|
#include <type_traits>
|
|
|
|
// The operators in dawn:: namespace need be introduced into other namespaces with
|
|
// using-declarations for C++ Argument Dependent Lookup to work.
|
|
#define DAWN_IMPORT_BITMASK_OPERATORS \
|
|
using dawn::operator|; \
|
|
using dawn::operator&; \
|
|
using dawn::operator^; \
|
|
using dawn::operator~; \
|
|
using dawn::operator&=; \
|
|
using dawn::operator|=; \
|
|
using dawn::operator^=; \
|
|
using dawn::HasZeroOrOneBits;
|
|
|
|
namespace dawn {
|
|
|
|
template <typename T>
|
|
struct IsDawnBitmask {
|
|
static constexpr bool enable = false;
|
|
};
|
|
|
|
template <typename T, typename Enable = void>
|
|
struct LowerBitmask {
|
|
static constexpr bool enable = false;
|
|
};
|
|
|
|
template <typename T>
|
|
struct LowerBitmask<T, typename std::enable_if<IsDawnBitmask<T>::enable>::type> {
|
|
static constexpr bool enable = true;
|
|
using type = T;
|
|
constexpr static T Lower(T t) { return t; }
|
|
};
|
|
|
|
template <typename T>
|
|
struct BoolConvertible {
|
|
using Integral = typename std::underlying_type<T>::type;
|
|
|
|
// NOLINTNEXTLINE(runtime/explicit)
|
|
explicit constexpr BoolConvertible(Integral value) : value(value) {}
|
|
constexpr operator bool() const { return value != 0; }
|
|
constexpr operator T() const { return static_cast<T>(value); }
|
|
|
|
Integral value;
|
|
};
|
|
|
|
template <typename T>
|
|
struct LowerBitmask<BoolConvertible<T>> {
|
|
static constexpr bool enable = true;
|
|
using type = T;
|
|
static constexpr type Lower(BoolConvertible<T> t) { return t; }
|
|
};
|
|
|
|
template <
|
|
typename T1,
|
|
typename T2,
|
|
typename = typename std::enable_if<LowerBitmask<T1>::enable && LowerBitmask<T2>::enable>::type>
|
|
constexpr BoolConvertible<typename LowerBitmask<T1>::type> operator|(T1 left, T2 right) {
|
|
using T = typename LowerBitmask<T1>::type;
|
|
using Integral = typename std::underlying_type<T>::type;
|
|
return BoolConvertible<T>(static_cast<Integral>(LowerBitmask<T1>::Lower(left)) |
|
|
static_cast<Integral>(LowerBitmask<T2>::Lower(right)));
|
|
}
|
|
|
|
template <
|
|
typename T1,
|
|
typename T2,
|
|
typename = typename std::enable_if<LowerBitmask<T1>::enable && LowerBitmask<T2>::enable>::type>
|
|
constexpr BoolConvertible<typename LowerBitmask<T1>::type> operator&(T1 left, T2 right) {
|
|
using T = typename LowerBitmask<T1>::type;
|
|
using Integral = typename std::underlying_type<T>::type;
|
|
return BoolConvertible<T>(static_cast<Integral>(LowerBitmask<T1>::Lower(left)) &
|
|
static_cast<Integral>(LowerBitmask<T2>::Lower(right)));
|
|
}
|
|
|
|
template <
|
|
typename T1,
|
|
typename T2,
|
|
typename = typename std::enable_if<LowerBitmask<T1>::enable && LowerBitmask<T2>::enable>::type>
|
|
constexpr BoolConvertible<typename LowerBitmask<T1>::type> operator^(T1 left, T2 right) {
|
|
using T = typename LowerBitmask<T1>::type;
|
|
using Integral = typename std::underlying_type<T>::type;
|
|
return BoolConvertible<T>(static_cast<Integral>(LowerBitmask<T1>::Lower(left)) ^
|
|
static_cast<Integral>(LowerBitmask<T2>::Lower(right)));
|
|
}
|
|
|
|
template <typename T1>
|
|
constexpr BoolConvertible<typename LowerBitmask<T1>::type> operator~(T1 t) {
|
|
using T = typename LowerBitmask<T1>::type;
|
|
using Integral = typename std::underlying_type<T>::type;
|
|
return BoolConvertible<T>(~static_cast<Integral>(LowerBitmask<T1>::Lower(t)));
|
|
}
|
|
|
|
template <
|
|
typename T,
|
|
typename T2,
|
|
typename = typename std::enable_if<IsDawnBitmask<T>::enable && LowerBitmask<T2>::enable>::type>
|
|
constexpr T& operator&=(T& l, T2 right) {
|
|
T r = LowerBitmask<T2>::Lower(right);
|
|
l = l & r;
|
|
return l;
|
|
}
|
|
|
|
template <
|
|
typename T,
|
|
typename T2,
|
|
typename = typename std::enable_if<IsDawnBitmask<T>::enable && LowerBitmask<T2>::enable>::type>
|
|
constexpr T& operator|=(T& l, T2 right) {
|
|
T r = LowerBitmask<T2>::Lower(right);
|
|
l = l | r;
|
|
return l;
|
|
}
|
|
|
|
template <
|
|
typename T,
|
|
typename T2,
|
|
typename = typename std::enable_if<IsDawnBitmask<T>::enable && LowerBitmask<T2>::enable>::type>
|
|
constexpr T& operator^=(T& l, T2 right) {
|
|
T r = LowerBitmask<T2>::Lower(right);
|
|
l = l ^ r;
|
|
return l;
|
|
}
|
|
|
|
template <typename T>
|
|
constexpr bool HasZeroOrOneBits(T value) {
|
|
using Integral = typename std::underlying_type<T>::type;
|
|
return (static_cast<Integral>(value) & (static_cast<Integral>(value) - 1)) == 0;
|
|
}
|
|
|
|
} // namespace dawn
|
|
|
|
#endif // INCLUDE_DAWN_ENUMCLASSBITMASKS_H_
|