Add set API to BitFlags

meithecatte · Sep 9, 2023 · dc3c613 · dc3c613
1 parent bf161cf
commit dc3c613
Showing 1 changed file with 11 additions and 3 deletions.
diff --git a/src/lib.rs b/src/lib.rs
@@ -655,12 +655,20 @@ where
         *self &= !other.into();
     }
 
+    /// Inserts if `cond` holds, else removes
+    #[inline(always)]
+    pub fn set<B: Into<BitFlags<T>>>(&mut self, other: B, cond: bool) {
+        if cond {
+            self.insert(other);
+        } else {
+            self.remove(other);
+        }
+    }
+
     /// Returns an iterator that yields each set flag
     #[inline]
     pub fn iter(self) -> Iter<T> {
-        Iter {
-            rest: self,
-        }
+        Iter { rest: self }
     }
 }