OLD | NEW |
1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. |
2 // Use of this source code is governed by a BSD-style license that can be | 2 // Use of this source code is governed by a BSD-style license that can be |
3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
4 | 4 |
5 #ifndef COMPONENTS_SYNC_BASE_ENUM_SET_H_ | 5 #ifndef COMPONENTS_SYNC_BASE_ENUM_SET_H_ |
6 #define COMPONENTS_SYNC_BASE_ENUM_SET_H_ | 6 #define COMPONENTS_SYNC_BASE_ENUM_SET_H_ |
7 | 7 |
8 #include <bitset> | 8 #include <bitset> |
9 #include <cstddef> | 9 #include <cstddef> |
10 #include <string> | 10 #include <string> |
(...skipping 101 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
112 } | 112 } |
113 return i; | 113 return i; |
114 } | 114 } |
115 | 115 |
116 const EnumBitSet* enums_; | 116 const EnumBitSet* enums_; |
117 size_t i_; | 117 size_t i_; |
118 }; | 118 }; |
119 | 119 |
120 EnumSet() {} | 120 EnumSet() {} |
121 | 121 |
122 // Recursively chain constructors. Base case is the empty pack. | 122 ~EnumSet() = default; |
123 template <class... T> | 123 |
124 EnumSet(E head, T... tail) : EnumSet(tail...) { | 124 static constexpr uint64_t single_val_bitstring(E val) { |
125 Put(head); | 125 return 1ULL << (ToIndex(val)); |
126 } | 126 } |
127 | 127 |
| 128 // Base case for recursive packing of a list of enum values. The uint64_t |
| 129 // corresponding to an empty list is 0. |
| 130 static constexpr uint64_t bitstring() { return 0ULL; } |
| 131 |
| 132 // As of writing, constexpr expressions can't contain anything other than a |
| 133 // return statement (and static asserts). To pack a variable number of enum |
| 134 // value arguments into a bitstring, we use template varargs with a recursive |
| 135 // constructor. Each recursive call packs one more enum into the bitstring, |
| 136 // and the individual results are combined with bitwise or. |
| 137 template <class... T> |
| 138 static constexpr uint64_t bitstring(E head, T... tail) { |
| 139 return (single_val_bitstring(head)) | bitstring(tail...); |
| 140 } |
| 141 |
| 142 template <class... T> |
| 143 constexpr EnumSet(E head, T... tail) |
| 144 : EnumSet(EnumBitSet(bitstring(head, tail...))) {} |
| 145 |
128 // Returns an EnumSet with all possible values. | 146 // Returns an EnumSet with all possible values. |
129 static EnumSet All() { | 147 static constexpr EnumSet All() { |
130 EnumBitSet enums; | 148 return EnumSet(EnumBitSet((1ULL << kValueCount) - 1)); |
131 enums.set(); | |
132 return EnumSet(enums); | |
133 } | 149 } |
134 | 150 |
135 // Returns an EnumSet with all the values from start to end, inclusive. | 151 // Returns an EnumSet with all the values from start to end, inclusive. |
136 static EnumSet FromRange(E start, E end) { | 152 static constexpr EnumSet FromRange(E start, E end) { |
137 EnumSet set; | 153 return EnumSet(EnumBitSet( |
138 set.PutRange(start, end); | 154 ((single_val_bitstring(end)) - (single_val_bitstring(start))) | |
139 return set; | 155 (single_val_bitstring(end)))); |
140 } | 156 } |
141 | 157 |
142 ~EnumSet() {} | |
143 | |
144 // Copy constructor and assignment welcome. | 158 // Copy constructor and assignment welcome. |
145 | 159 |
146 // Set operations. Put, Retain, and Remove are basically | 160 // Set operations. Put, Retain, and Remove are basically |
147 // self-mutating versions of Union, Intersection, and Difference | 161 // self-mutating versions of Union, Intersection, and Difference |
148 // (defined below). | 162 // (defined below). |
149 | 163 |
150 // Adds the given value (which must be in range) to our set. | 164 // Adds the given value (which must be in range) to our set. |
151 void Put(E value) { enums_.set(ToIndex(value)); } | 165 void Put(E value) { enums_.set(ToIndex(value)); } |
152 | 166 |
153 // Adds all values in the given set to our set. | 167 // Adds all values in the given set to our set. |
(...skipping 21 matching lines...) Expand all Loading... |
175 } | 189 } |
176 } | 190 } |
177 | 191 |
178 // Removes all values in the given set from our set. | 192 // Removes all values in the given set from our set. |
179 void RemoveAll(EnumSet other) { enums_ &= ~other.enums_; } | 193 void RemoveAll(EnumSet other) { enums_ &= ~other.enums_; } |
180 | 194 |
181 // Removes all values from our set. | 195 // Removes all values from our set. |
182 void Clear() { enums_.reset(); } | 196 void Clear() { enums_.reset(); } |
183 | 197 |
184 // Returns true iff the given value is in range and a member of our set. | 198 // Returns true iff the given value is in range and a member of our set. |
185 bool Has(E value) const { | 199 constexpr bool Has(E value) const { |
186 return InRange(value) && enums_.test(ToIndex(value)); | 200 return InRange(value) && enums_[ToIndex(value)]; |
187 } | 201 } |
188 | 202 |
189 // Returns true iff the given set is a subset of our set. | 203 // Returns true iff the given set is a subset of our set. |
190 bool HasAll(EnumSet other) const { | 204 bool HasAll(EnumSet other) const { |
191 return (enums_ & other.enums_) == other.enums_; | 205 return (enums_ & other.enums_) == other.enums_; |
192 } | 206 } |
193 | 207 |
194 // Returns true iff our set is empty. | 208 // Returns true iff our set is empty. |
195 bool Empty() const { return !enums_.any(); } | 209 bool Empty() const { return !enums_.any(); } |
196 | 210 |
(...skipping 11 matching lines...) Expand all Loading... |
208 bool operator!=(const EnumSet& other) const { return enums_ != other.enums_; } | 222 bool operator!=(const EnumSet& other) const { return enums_ != other.enums_; } |
209 | 223 |
210 private: | 224 private: |
211 friend EnumSet Union<E, MinEnumValue, MaxEnumValue>(EnumSet set1, | 225 friend EnumSet Union<E, MinEnumValue, MaxEnumValue>(EnumSet set1, |
212 EnumSet set2); | 226 EnumSet set2); |
213 friend EnumSet Intersection<E, MinEnumValue, MaxEnumValue>(EnumSet set1, | 227 friend EnumSet Intersection<E, MinEnumValue, MaxEnumValue>(EnumSet set1, |
214 EnumSet set2); | 228 EnumSet set2); |
215 friend EnumSet Difference<E, MinEnumValue, MaxEnumValue>(EnumSet set1, | 229 friend EnumSet Difference<E, MinEnumValue, MaxEnumValue>(EnumSet set1, |
216 EnumSet set2); | 230 EnumSet set2); |
217 | 231 |
218 explicit EnumSet(EnumBitSet enums) : enums_(enums) {} | 232 // A bitset can't be constexpr constructed if it has size > 64, since the |
| 233 // constexpr constructor uses a uint64_t. If your EnumSet has > 64 values, you |
| 234 // can safely remove the constepxr qualifiers from this file, at the cost of |
| 235 // some minor optimizations. |
| 236 explicit constexpr EnumSet(EnumBitSet enums) : enums_(enums) { |
| 237 static_assert(kValueCount < 64, |
| 238 "Max number of enum values is 64 for constexpr "); |
| 239 } |
219 | 240 |
220 static bool InRange(E value) { | 241 static constexpr bool InRange(E value) { |
221 return (value >= MinEnumValue) && (value <= MaxEnumValue); | 242 return (value >= MinEnumValue) && (value <= MaxEnumValue); |
222 } | 243 } |
223 | 244 |
224 // Converts a value to/from an index into |enums_|. | 245 // Converts a value to/from an index into |enums_|. |
225 | 246 |
226 static size_t ToIndex(E value) { | 247 static constexpr size_t ToIndex(E value) { return value - MinEnumValue; } |
227 DCHECK_GE(value, MinEnumValue); | |
228 DCHECK_LE(value, MaxEnumValue); | |
229 return value - MinEnumValue; | |
230 } | |
231 | 248 |
232 static E FromIndex(size_t i) { | 249 static E FromIndex(size_t i) { |
233 DCHECK_LT(i, kValueCount); | 250 DCHECK_LT(i, kValueCount); |
234 return static_cast<E>(MinEnumValue + i); | 251 return static_cast<E>(MinEnumValue + i); |
235 } | 252 } |
236 | 253 |
237 EnumBitSet enums_; | 254 EnumBitSet enums_; |
238 }; | 255 }; |
239 | 256 |
240 template <typename E, E MinEnumValue, E MaxEnumValue> | 257 template <typename E, E MinEnumValue, E MaxEnumValue> |
(...skipping 21 matching lines...) Expand all Loading... |
262 | 279 |
263 template <typename E, E Min, E Max> | 280 template <typename E, E Min, E Max> |
264 EnumSet<E, Min, Max> Difference(EnumSet<E, Min, Max> set1, | 281 EnumSet<E, Min, Max> Difference(EnumSet<E, Min, Max> set1, |
265 EnumSet<E, Min, Max> set2) { | 282 EnumSet<E, Min, Max> set2) { |
266 return EnumSet<E, Min, Max>(set1.enums_ & ~set2.enums_); | 283 return EnumSet<E, Min, Max>(set1.enums_ & ~set2.enums_); |
267 } | 284 } |
268 | 285 |
269 } // namespace syncer | 286 } // namespace syncer |
270 | 287 |
271 #endif // COMPONENTS_SYNC_BASE_ENUM_SET_H_ | 288 #endif // COMPONENTS_SYNC_BASE_ENUM_SET_H_ |
OLD | NEW |