~ Minor changes.

1 files changed, 10 insertions, 2 deletions

diff --git a/src/Network/BitTorrent/PeerWire/Bitfield.hs b/src/Network/BitTorrent/PeerWire/Bitfield.hs
index 2baeb516..9d88e784 100644
--- a/src/Network/BitTorrent/PeerWire/Bitfield.hs
+++ b/src/Network/BitTorrent/PeerWire/Bitfield.hs
@@ -18,6 +18,7 @@ module Network.BitTorrent.PeerWire.Bitfield
 
          -- * Construction
        , empty, full
+       , toList
        , fromByteString, toByteString
 
          -- * Query
@@ -62,6 +63,13 @@ full :: Int -> Bitfield
 full n = MkBitfield $ B.replicate (sizeInBase n 8)  (complement 0)
 {-# INLINE full #-}
 
+toList :: Bitfield -> [Bool]
+toList (MkBitfield bs) = concatMap unpkg (B.unpack bs)
+  where
+    unpkg :: Word8 -> [Bool]
+    unpkg byte = L.map (testBit byte) [0..bitSize (undefined :: Word8) - 1]
+{-# INLINE toList #-}
+
 fromByteString :: ByteString -> Bitfield
 fromByteString = MkBitfield
 {-# INLINE fromByteString #-}
@@ -278,5 +286,5 @@ findMax (MkBitfield b) = do
 
 {-# INLINE findMax #-}
 
-frequencies :: [Bitfield] -> UArray PieceIx Int
-frequencies = undefined
+frequencies :: [Bitfield] -> [Int]
+frequencies xs = foldr1 (zipWith (+)) $ map (map fromEnum . toList) xs