blob: ce49b3996ed52a30ac859fe51472dce58cd685e6 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
|
// Copyright 2019 the V8 project authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
#ifndef V8_BASE_ENUM_SET_H_
#define V8_BASE_ENUM_SET_H_
#include <type_traits>
#include "src/base/logging.h"
namespace v8 {
namespace base {
// A poor man's version of STL's bitset: A bit set of enums E (without explicit
// values), fitting into an integral type T.
template <class E, class T = int>
class EnumSet {
static_assert(std::is_enum<E>::value, "EnumSet can only be used with enums");
public:
constexpr EnumSet() = default;
explicit constexpr EnumSet(std::initializer_list<E> init) {
T bits = 0;
for (E e : init) bits |= Mask(e);
bits_ = bits;
}
constexpr bool empty() const { return bits_ == 0; }
constexpr bool contains(E element) const {
return (bits_ & Mask(element)) != 0;
}
constexpr bool contains_any(EnumSet set) const {
return (bits_ & set.bits_) != 0;
}
void Add(E element) { bits_ |= Mask(element); }
void Add(EnumSet set) { bits_ |= set.bits_; }
void Remove(E element) { bits_ &= ~Mask(element); }
void Remove(EnumSet set) { bits_ &= ~set.bits_; }
void RemoveAll() { bits_ = 0; }
void Intersect(EnumSet set) { bits_ &= set.bits_; }
constexpr T ToIntegral() const { return bits_; }
constexpr bool operator==(EnumSet set) const { return bits_ == set.bits_; }
constexpr bool operator!=(EnumSet set) const { return bits_ != set.bits_; }
constexpr EnumSet operator|(EnumSet set) const {
return EnumSet(bits_ | set.bits_);
}
constexpr EnumSet operator&(EnumSet set) const {
return EnumSet(bits_ & set.bits_);
}
constexpr EnumSet operator-(EnumSet set) const {
return EnumSet(bits_ & ~set.bits_);
}
EnumSet& operator|=(EnumSet set) { return *this = *this | set; }
EnumSet& operator&=(EnumSet set) { return *this = *this & set; }
EnumSet& operator-=(EnumSet set) { return *this = *this - set; }
constexpr EnumSet operator|(E element) const {
return EnumSet(bits_ | Mask(element));
}
constexpr EnumSet operator&(E element) const {
return EnumSet(bits_ & Mask(element));
}
constexpr EnumSet operator-(E element) const {
return EnumSet(bits_ & ~Mask(element));
}
EnumSet& operator|=(E element) { return *this = *this | element; }
EnumSet& operator&=(E element) { return *this = *this & element; }
EnumSet& operator-=(E element) { return *this = *this - element; }
static constexpr EnumSet FromIntegral(T bits) { return EnumSet{bits}; }
private:
explicit constexpr EnumSet(T bits) : bits_(bits) {}
static constexpr T Mask(E element) {
DCHECK_GT(sizeof(T) * 8, static_cast<size_t>(element));
return T{1} << static_cast<typename std::underlying_type<E>::type>(element);
}
T bits_ = 0;
};
} // namespace base
} // namespace v8
#endif // V8_BASE_ENUM_SET_H_
|
