Implemented Binary and Display traits for FixedBitSet.

src/lib.rs

@@ -28,6 +28,10 @@ use core as std;
 
 mod range;
 
+#[cfg(feature = "std")]
+use std::fmt::{Display, Error, Formatter, Binary};
+use std::ops::{Add};
+
 use std::ops::{BitAnd, BitAndAssign, BitOr, BitOrAssign, BitXor, BitXorAssign, Index};
 use std::cmp::{Ord, Ordering};
 use std::iter::{Chain, FromIterator};
@@ -356,6 +360,30 @@ impl FixedBitSet
     }
 }
 
+#[cfg(feature = "std")]
+impl Binary for FixedBitSet {
+    fn fmt(&self, f: &mut Formatter<'_>) -> Result<(), Error> {
+        let mut display_data = if f.alternate() {
+            String::from("0b")
+        } else {
+            String::new()
+        };
+
+        for block in self.data.iter().rev() {
+            display_data = display_data.add(&format!("{:b}", block));
+        };
+
+        write!(f, "{}", display_data)
+    }
+}
+
+#[cfg(feature = "std")]
+impl Display for FixedBitSet {
+    fn fmt(&self, f: &mut Formatter<'_>) -> Result<(), Error> {
+        Binary::fmt(&self, f)
+    }
+}
+
 /// An iterator producing elements in the difference of two sets.
 ///
 /// This struct is created by the [`FixedBitSet::difference`] method.
@@ -1373,3 +1401,26 @@ fn from_iterator_ones() {
     assert_eq!(fb.len(), dup.len());
     assert_eq!(fb.ones().collect::<Vec<usize>>(), dup.ones().collect::<Vec<usize>>());
 }
+
+#[cfg(feature = "std")]
+#[test]
+fn binary_trait() {
+    let items: Vec<usize> = vec![1, 5, 7, 10, 14, 15];
+    let fb = items.iter().cloned().collect::<FixedBitSet>();
+
+    assert_eq!(format!("{:b}", fb), "1100010010100010");
+    assert_eq!(format!("{:#b}", fb), "0b1100010010100010");
+}
+
+#[cfg(feature = "std")]
+#[test]
+fn display_trait() {
+    let len = 8;
+    let mut fb = FixedBitSet::with_capacity(len);
+
+    fb.put(4);
+    fb.put(2);
+
+    assert_eq!(format!("{}", fb), "10100");
+    assert_eq!(format!("{:#}", fb), "0b10100");
+}