master..f58ae47 · binary-bits

git.iko.soy/binary-bits

on f58ae47 help sign in

Data/Binary/Bits/Get.hs−23

since splitsnapshots

unifiedsplit

@@ -60,7 +60,6 @@ module Data.Binary.Bits.Get

61 -- ** Get bytes

62 , getBool

63− , getBool#

64 , getWord8

63 , getWord8

65 , getWord16

64 , getWord16

66 , getWord32

65 , getWord32

@@ -115,8 +114,6 @@ import Control.Applicative

115import Control.Monad (when,foldM_)

114import Control.Monad (when,foldM_)

116

115

117import Prelude as P

116import Prelude as P

118−import GHC.Word

119−import GHC.Base (uncheckedShiftLWord8#, wordToWord8#, Int (..), andWord8#, Word8#, (-#), Int#, xorI#, eqWord8#, TYPE)

120

117

121-- $bitget

118-- $bitget

122-- Parse bits using a monad.

119-- Parse bits using a monad.

@@ -185,18 +182,6 @@ readBool (S bs o bo) = case bo of

185 LL -> testBit (unsafeHead bs) o

182 LL -> testBit (unsafeHead bs) o

186 LB -> testBit (unsafeHead bs) o

183 LB -> testBit (unsafeHead bs) o

187

184

188−-- | Read a single bit

189−readBool# :: S -> Int

190−readBool# (S bs (I# o) bo) = case bo of

191− BB -> testBit# (7# -# o)

192− BL -> testBit# (7# -# o)

193− LL -> testBit# o

194− LB -> testBit# o

195− where

196− !(W8# w) = unsafeHead bs

197− testBit# n = I# (((w `andWord8#` (wordToWord8# 1## `uncheckedShiftLWord8#` n)) `eqWord8#` wordToWord8# 0##) `xorI#` 1#)

198−

199−

200-- | Extract a range of bits from (ws :: ByteString)

185-- | Extract a range of bits from (ws :: ByteString)

201--

186--

202-- Constraint: 8 * (length ws -1 ) < o+n <= 8 * length ws

187-- Constraint: 8 * (length ws -1 ) < o+n <= 8 * length ws

@@ -378,10 +363,6 @@ isEmpty = B $ \ (S bs o bo) -> if B.null bs

378getBool :: BitGet Bool

363getBool :: BitGet Bool

379getBool = block bool

364getBool = block bool

380

365

381−-- | Get 1 bit as a 'Bool'.

382−getBool# :: BitGet Int

383−getBool# = block bool#

384−

385-- | Get @n@ bits as a 'Word8'. @n@ must be within @[0..8]@.

366-- | Get @n@ bits as a 'Word8'. @n@ must be within @[0..8]@.

386getWord8 :: Int -> BitGet Word8

367getWord8 :: Int -> BitGet Word8

387getWord8 n = block (word8 n)

368getWord8 n = block (word8 n)

@@ -457,10 +438,6 @@ block (Block i p) = do

457bool :: Block Bool

438bool :: Block Bool

458bool = Block 1 readBool

439bool = Block 1 readBool

459

440

460−-- | Read a 1 bit 'Bool'.

461−bool# :: Block Int

462−bool# = Block 1 readBool#

463−

464-- | Read @n@ bits as a 'Word8'. @n@ must be within @[0..8]@.

441-- | Read @n@ bits as a 'Word8'. @n@ must be within @[0..8]@.

465word8 :: Int -> Block Word8

442word8 :: Int -> Block Word8

466word8 n = Block n (readWordChecked 8 n)

443word8 n = Block n (readWordChecked 8 n)

Data/Binary/Bits/Internal.hs−21

@@ -12,13 +12,10 @@

13module Data.Binary.Bits.Internal

14 ( make_mask

15− , make_mask#

16 , mask

15 , mask

17− , mask#

18 , bit_offset

16 , bit_offset

19 , byte_offset

17 , byte_offset

20 , reverseBits

18 , reverseBits

21− , reverseBits#

22 , FastBits(..)

19 , FastBits(..)

23 )

20 )

24where

21where

@@ -41,18 +38,11 @@ make_mask n = (1 `shiftL` fromIntegral n) - 1

41{-# SPECIALIZE make_mask :: Int -> Word32 #-}

38{-# SPECIALIZE make_mask :: Int -> Word32 #-}

42{-# SPECIALIZE make_mask :: Int -> Word64 #-}

39{-# SPECIALIZE make_mask :: Int -> Word64 #-}

44−make_mask# :: Int# -> Word#

45−make_mask# n = (1## `shiftL#` n) `minusWord#` 1##

46−

47-- | Keep only the n least-significant bits of the given value

41-- | Keep only the n least-significant bits of the given value

48mask :: (Bits a, Num a) => Int -> a -> a

42mask :: (Bits a, Num a) => Int -> a -> a

49mask n v = v .&. make_mask n

43mask n v = v .&. make_mask n

50{-# INLINE mask #-}

44{-# INLINE mask #-}

52−-- | Keep only the n least-significant bits of the given value

53−mask# :: Int# -> Word# -> Word#

54−mask# n v = v `and#` make_mask# n

55−

56-- | Compute bit offset (equivalent to x `mod` 8 but faster)

46-- | Compute bit offset (equivalent to x `mod` 8 but faster)

57bit_offset :: Int -> Int

47bit_offset :: Int -> Int

58bit_offset n = make_mask 3 .&. n

48bit_offset n = make_mask 3 .&. n

@@ -75,17 +65,6 @@ reverseBits n value = rec value n 0

75 rec 0 i r = r `fastShiftL` i

65 rec 0 i r = r `fastShiftL` i

76 rec v i r = rec (v `fastShiftR` 1) (i-1) ((r `fastShiftL` 1) .|. (v .&. 0x1))

66 rec v i r = rec (v `fastShiftR` 1) (i-1) ((r `fastShiftL` 1) .|. (v .&. 0x1))

78−reverseBits# :: Int# -> Word# -> Word#

79−reverseBits# n value = rec value n 0##

80− where

81− -- rec v i r, where

82− -- v is orginal value shifted

83− -- i is the remaining number of bits

84− -- r is current value

85− rec 0## 0# r = r

86− rec 0## i r = r `shiftL#` i

87− rec v i r = rec (v `shiftRL#` 1#) (i -# 1#) ((r `shiftL#` 1#) `or#` (v `and#` 0x1##))

88−

90---------------------------------------------------------------------

69---------------------------------------------------------------------

91-- Unchecked shifts, from the "binary" package

70-- Unchecked shifts, from the "binary" package

Data/Binary/Bits/Put.hs+2−60

@@ -1,5 +1,3 @@

1−{-# LANGUAGE MagicHash #-}

2−

3-----------------------------------------------------------------------------

1-----------------------------------------------------------------------------

4-- |

2-- |

5-- Module : Data.Binary.Bits.Put

3-- Module : Data.Binary.Bits.Put

@@ -22,10 +20,8 @@ module Data.Binary.Bits.Put

22 -- * Data types

20 -- * Data types

23 -- ** Bool

21 -- ** Bool

24 , putBool

22 , putBool

25− , putBool#

27 -- ** Words

24 -- ** Words

28− , putWord#

29 , putWord8

25 , putWord8

30 , putWord16

26 , putWord16

31 , putWord32

27 , putWord32

@@ -50,11 +46,11 @@ import Data.Binary.Bits.Alignment

50import Data.ByteString as BS

46import Data.ByteString as BS

51import Data.ByteString.Unsafe as BS

47import Data.ByteString.Unsafe as BS

49+import Control.Applicative

53import Control.Monad (when)

50import Control.Monad (when)

54import Data.Bits

51import Data.Bits

52+import Data.Monoid

55import Data.Word

53import Data.Word

56−import GHC.Base (Word#, shiftRL#, Int (..), Int#, (-#), (<#), isTrue#, (+#), shiftL#, wordToWord8#, orWord8#)

57−import GHC.Word (Word8(..))

59data BitPut a = BitPut { run :: (S -> PairS a) }

55data BitPut a = BitPut { run :: (S -> PairS a) }

@@ -66,18 +62,11 @@ data S = S !Builder !Word8 !Int !BitOrder

66putBool :: Bool -> BitPut ()

62putBool :: Bool -> BitPut ()

67putBool b = putWord8 1 (if b then 0xff else 0x00)

63putBool b = putWord8 1 (if b then 0xff else 0x00)

69−-- | Put a 1 bit 'Bool'.

70−putBool# :: Word# -> BitPut ()

71−putBool# b = putWord# 1# b

72−

74-- | Generic putWord

66-- | Generic putWord

75putWord :: (Num a, FastBits a, Integral a) => Int -> a -> BitPut ()

67putWord :: (Num a, FastBits a, Integral a) => Int -> a -> BitPut ()

76putWord n w = BitPut $ \s -> PairS () (putWordS n w s)

68putWord n w = BitPut $ \s -> PairS () (putWordS n w s)

78−putWord# :: Int# -> Word# -> BitPut ()

79−putWord# n w = BitPut $ \s -> PairS () (putWordS# n w s)

80−

81putWordS :: (Num a, FastBits a, Integral a) => Int -> a -> S -> S

70putWordS :: (Num a, FastBits a, Integral a) => Int -> a -> S -> S

82putWordS n w s@(S builder b o bo) = s'

71putWordS n w s@(S builder b o bo) = s'

83 where

72 where

@@ -121,53 +110,6 @@ putWordS n w s@(S builder b o bo) = s'

121 | o2 == 8 = S (b2 `mappend` B.singleton w2) 0 0 bo2

110 | o2 == 8 = S (b2 `mappend` B.singleton w2) 0 0 bo2

122 | otherwise = s2

111 | otherwise = s2

123

112

124−putWordS# :: Int# -> Word# -> S -> S

125−putWordS# n w s@(S builder (W8# b) (I# o) bo) = s'

126− where

127− min# :: Int# -> Int# -> Int#

128− min# x y | isTrue# (x <# y) = x

129− min# _ y = y

130−

131− -- number of bits that will be stored in the current byte

132− cn = min# (8# -# o) n

133−

134− -- new state

135− s' = case n of

136− 0# -> s

137− _ -> putWordS# (n -# cn) w' (flush (S builder (W8# b') (I# (o +# cn)) bo))

138−

139− -- new current byte

140− b' = wordToWord8# (shl (selectBits w)) `orWord8#` b

141−

142− -- Word containing the remaining (n-cn) bits to store in its LSB

143− w' = case bo of

144− BB -> w

145− BL -> w `shiftRL#` cn

146− LL -> w `shiftRL#` cn

147− LB -> w

148−

149− -- Select bits to store in the current byte.

150− -- Put them in the correct order and return them in the least-significant

151− -- bits of the returned value

152− selectBits :: Word# -> Word#

153− selectBits x = case bo of

154− BB -> mask# cn (x `shiftRL#` (n -# cn))

155− LB -> reverseBits# cn (mask# cn (x `shiftRL#` (n -# cn)))

156− LL -> mask# cn x

157− BL -> reverseBits# cn (mask# cn x)

158−

159− -- shift left at the correct position

160− shl :: Word# -> Word#

161− shl x = case bo of

162− BB -> x `shiftL#` (8# -# o -# cn)

163− BL -> x `shiftL#` (8# -# o -# cn)

164− LL -> x `shiftL#` o

165− LB -> x `shiftL#` o

166−

167− flush s2@(S b2 w2 o2 bo2)

168− | o2 == 8 = S (b2 `mappend` B.singleton w2) 0 0 bo2

169− | otherwise = s2

170−

171

113

172-- | Put the @n@ lower bits of a 'Word8'.

114-- | Put the @n@ lower bits of a 'Word8'.

173putWord8 :: Int -> Word8 -> BitPut ()

115putWord8 :: Int -> Word8 -> BitPut ()

flake.lockdeleted−59

@@ -1,59 +0,0 @@

1−{

2− "nodes": {

3− "flake-utils": {

4− "inputs": {

5− "systems": "systems"

6− },

7− "locked": {

8− "lastModified": 1731533236,

9− "narHash": "sha256-l0KFg5HjrsfsO/JpG+r7fRrqm12kzFHyUHqHCVpMMbI=",

10− "owner": "numtide",

11− "repo": "flake-utils",

12− "rev": "11707dc2f618dd54ca8739b309ec4fc024de578b",

13− "type": "github"

14− },

15− "original": {

16− "id": "flake-utils",

17− "type": "indirect"

18− }

19− },

20− "nixpkgs": {

21− "locked": {

22− "lastModified": 1737469691,

23− "narHash": "sha256-nmKOgAU48S41dTPIXAq0AHZSehWUn6ZPrUKijHAMmIk=",

24− "owner": "NixOS",

25− "repo": "nixpkgs",

26− "rev": "9e4d5190a9482a1fb9d18adf0bdb83c6e506eaab",

27− "type": "github"

28− },

29− "original": {

30− "id": "nixpkgs",

31− "ref": "nixos-unstable",

32− "type": "indirect"

33− }

34− },

35− "root": {

36− "inputs": {

37− "flake-utils": "flake-utils",

38− "nixpkgs": "nixpkgs"

39− }

40− },

41− "systems": {

42− "locked": {

43− "lastModified": 1681028828,

44− "narHash": "sha256-Vy1rq5AaRuLzOxct8nz4T6wlgyUR7zLU309k9mBC768=",

45− "owner": "nix-systems",

46− "repo": "default",

47− "rev": "da67096a3b9bf56a91d16901293e51ba5b49a27e",

48− "type": "github"

49− },

50− "original": {

51− "owner": "nix-systems",

52− "repo": "default",

53− "type": "github"

54− }

55− }

56− },

57− "root": "root",

58− "version": 7

59−}

flake.nixdeleted−24

@@ -1,24 +0,0 @@

1−{

2− inputs = {

3− nixpkgs.url = "nixpkgs/nixos-unstable";

4− flake-utils.url = "flake-utils";

5− };

6−

7− outputs = { self, flake-utils, nixpkgs }:

8− let

9− supportedSystems =

10− [ "x86_64-linux" "x86_64-darwin" "aarch64-darwin" "aarch64-linux" ];

11− compilerVersion = "966";

12− in flake-utils.lib.eachSystem supportedSystems (system:

13− let pkgs = import nixpkgs { inherit system; };

14− in {

15− devShells.default = pkgs.mkShell {

16− buildInputs = with pkgs; [

17− haskell.compiler."ghc${compilerVersion}"

18− (pkgs.haskell-language-server.override {

19− supportedGhcVersions = [ compilerVersion ];

20− })

21− ];

22− };

23− });

24−}

stack.yaml+2−1

@@ -17,7 +17,8 @@

17#

18# resolver: ./custom-snapshot.yaml

19# resolver: https://example.com/snapshots/2018-01-01.yaml

20−resolver: lts-22.43

20+resolver:

21+ url: https://raw.githubusercontent.com/commercialhaskell/stackage-snapshots/master/lts/21/6.yaml

22# User packages to be built.

23# User packages to be built.

23# Various formats can be used as shown in the example below.

24# Various formats can be used as shown in the example below.

stack.yaml.lock+5−4

@@ -6,7 +6,8 @@

6packages: []

7snapshots:

8- completed:

9− sha256: 08bd13ce621b41a8f5e51456b38d5b46d7783ce114a50ab604d6bbab0d002146

9+ sha256: 2e7d4a730d8eb5373b2d383fac84efcf7c81e3b7a5fce71b4c2e19a1768f25a6

10− size: 720271

10+ size: 640239

11− url: https://raw.githubusercontent.com/commercialhaskell/stackage-snapshots/master/lts/22/43.yaml

11+ url: https://raw.githubusercontent.com/commercialhaskell/stackage-snapshots/master/lts/21/6.yaml

12− original: lts-22.43

12+ original:

13+ url: https://raw.githubusercontent.com/commercialhaskell/stackage-snapshots/master/lts/21/6.yaml