bitwise-0.1.1: hugs98/Compat.hs

module Compat
  ( unsafeIOToST
  , uncons
  , packCStringLen
  , unsafeUseAsCStringLen
  , popCount
  ) where

import Control.Monad.ST (unsafeIOToST)
import Data.Bits ((.&,), shiftR)
import qualified Data.ByteString as BS
import Data.Word (Word8)
import Foreign.C.String (CStringLen)

uncons :: BS.ByteString -> Maybe (Word8, BS.ByteString)
uncons s = if BS.null s then Nothing else Just (BS.head s, BS.tail s)

unsafeUseAsCStringLen :: BS.ByteString -> (CStringLen -> IO a) -> IO a
unsafeUseAsCStringLen = BS.useAsCStringLen

packCStringLen :: CStringLen -> IO BS.ByteString
packCStringLen = return . BS.packCStringLen

popCount :: Word64 -> Int
popCount = fromIntegral . sum . map (.&. 1) . take 64 . iterate (`shiftR` 1)