on master help sign in
.gitignore+1
since splitsnapshots
unifiedsplit
@@ -1,3 +1,4 @@
1.cabal-sandbox
1.cabal-sandbox
2cabal.sandbox.config
2cabal.sandbox.config
3dist/
3dist/
4+.stack-work
BitsQC.hs+65−32
@@ -9,6 +9,7 @@ import Data.Binary.Put ( runPut )
9import Data.Binary.Bits
9import Data.Binary.Bits
10import Data.Binary.Bits.Get
10import Data.Binary.Bits.Get
11import Data.Binary.Bits.Put
11import Data.Binary.Bits.Put
12+import Data.Binary.Bits.BitOrder
12
13
13import qualified Data.ByteString as B
14import qualified Data.ByteString as B
14import qualified Data.ByteString.Lazy as L
15import qualified Data.ByteString.Lazy as L
@@ -17,7 +18,8 @@ import Control.Applicative
17import Data.Bits
18import Data.Bits
18import Data.Word
19import Data.Word
19import Foreign.Storable
20import Foreign.Storable
20−import System.Random
21+import Data.Traversable (traverse)
22+import Data.Foldable (traverse_)
21
23
22import Test.Framework.Providers.QuickCheck2 ( testProperty )
24import Test.Framework.Providers.QuickCheck2 ( testProperty )
23import Test.Framework.Runners.Console ( defaultMain )
25import Test.Framework.Runners.Console ( defaultMain )
@@ -36,7 +38,10 @@ tests =
36 [ testProperty "prop_composite_case" prop_composite_case ]
38 [ testProperty "prop_composite_case" prop_composite_case ]
37
39
38 , testGroup "getByteString"
40 , testGroup "getByteString"
39− [ testProperty "prop_getByteString_negative" prop_getByteString_negative ]
41+ [ testProperty "prop_getByteString_negative" prop_getByteString_negative
42+ , testProperty "prop_putByteString_getByteString" (prop_putByteString_getByteString :: BitOrder -> B.ByteString -> Property)
43+ , testProperty "prop_putByteString_getByteString_many" (prop_putByteString_getByteString_many :: BitOrder -> [B.ByteString] -> Property)
44+ ]
40
45
41 , testGroup "getLazyByteString"
46 , testGroup "getLazyByteString"
42 [ testProperty "getLazyByteString == getByteString"
47 [ testProperty "getLazyByteString == getByteString"
@@ -84,10 +89,10 @@ tests =
84 ]
89 ]
85
90
86 , testGroup "prop_put_with_bitreq"
91 , testGroup "prop_put_with_bitreq"
87− [ testProperty "Word8" (prop_putget_with_bitreq :: W Word8 -> Property)
92+ [ testProperty "Word8" (prop_putget_with_bitreq :: BitOrder -> W Word8 -> Property)
88− , testProperty "Word16" (prop_putget_with_bitreq :: W Word16 -> Property)
93+ , testProperty "Word16" (prop_putget_with_bitreq :: BitOrder -> W Word16 -> Property)
89− , testProperty "Word32" (prop_putget_with_bitreq :: W Word32 -> Property)
94+ , testProperty "Word32" (prop_putget_with_bitreq :: BitOrder -> W Word32 -> Property)
90− , testProperty "Word64" (prop_putget_with_bitreq :: W Word64 -> Property)
95+ , testProperty "Word64" (prop_putget_with_bitreq :: BitOrder -> W Word64 -> Property)
91 ]
96 ]
92
97
93 , testGroup "prop_putget_list_simple"
98 , testGroup "prop_putget_list_simple"
@@ -112,10 +117,10 @@ tests =
112 , testProperty "Word64" (prop_putget_list_with_bitreq :: W [Word64] -> Property)
117 , testProperty "Word64" (prop_putget_list_with_bitreq :: W [Word64] -> Property)
113 ]
118 ]
114 , testGroup "prop_bitget_bytestring_interspersed"
119 , testGroup "prop_bitget_bytestring_interspersed"
115− [ testProperty "Word8" (prop_bitget_bytestring_interspersed :: W Word8 -> [B.ByteString] -> Property)
120+ [ testProperty "Word8" (prop_bitget_bytestring_interspersed :: BitOrder -> W Word8 -> [B.ByteString] -> Property)
116− , testProperty "Word16" (prop_bitget_bytestring_interspersed :: W Word16 -> [B.ByteString] -> Property)
121+ , testProperty "Word16" (prop_bitget_bytestring_interspersed :: BitOrder -> W Word16 -> [B.ByteString] -> Property)
117− , testProperty "Word32" (prop_bitget_bytestring_interspersed :: W Word32 -> [B.ByteString] -> Property)
122+ , testProperty "Word32" (prop_bitget_bytestring_interspersed :: BitOrder -> W Word32 -> [B.ByteString] -> Property)
118− , testProperty "Word64" (prop_bitget_bytestring_interspersed :: W Word64 -> [B.ByteString] -> Property)
123+ , testProperty "Word64" (prop_bitget_bytestring_interspersed :: BitOrder -> W Word64 -> [B.ByteString] -> Property)
119 ]
124 ]
120 , testGroup "Simulate programs"
125 , testGroup "Simulate programs"
121 [ testProperty "primitive" prop_primitive
126 [ testProperty "primitive" prop_primitive
@@ -158,13 +163,28 @@ prop_getByteString_negative n =
158 n < 1 ==>
163 n < 1 ==>
159 runGet (runBitGet (getByteString n)) L.empty == B.empty
164 runGet (runBitGet (getByteString n)) L.empty == B.empty
160
165
166+prop_putByteString_getByteString :: BitOrder -> B.ByteString -> Property
167+prop_putByteString_getByteString bo bs = property $ bs' == bs
168+ where
169+ n = B.length bs
170+ w = runPut (runBitPut (withBitOrder bo (putByteString bs)))
171+ bs' = runGet (runBitGet (withBitOrder bo (getByteString n))) w
172+
173+prop_putByteString_getByteString_many :: BitOrder -> [B.ByteString] -> Property
174+prop_putByteString_getByteString_many bo bs = property $ bs' == bs
175+ where
176+ n = fmap B.length bs
177+ w = runPut (runBitPut (withBitOrder bo (traverse_ putByteString bs)))
178+ bs' = runGet (runBitGet (withBitOrder bo (traverse getByteString n))) w
179+
180+
161−prop_putget_with_bitreq :: (BinaryBit a, Num a, Bits a, Ord a) => W a -> Property
181+prop_putget_with_bitreq :: (BinaryBit a, Num a, Bits a, Ord a) => BitOrder -> W a -> Property
162−prop_putget_with_bitreq (W w) = property $
182+prop_putget_with_bitreq bo (W w) = property $
163 -- write all words with as many bits as it's required
183 -- write all words with as many bits as it's required
164 let p = putBits (bitreq w) w
184 let p = putBits (bitreq w) w
165 g = getBits (bitreq w)
185 g = getBits (bitreq w)
166− lbs = runPut (runBitPut p)
186+ lbs = runPut (runBitPut (withBitOrder bo p))
167− w' = runGet (runBitGet g) lbs
187+ w' = runGet (runBitGet (withBitOrder bo g)) lbs
168 in w == w'
188 in w == w'
169
189
170-- | Write a list of items. Each item is written with the maximum amount of
190-- | Write a list of items. Each item is written with the maximum amount of
@@ -226,12 +246,12 @@ prop_bitget_with_put_from_binary (W ws) = property $
226 in ws == ws'
246 in ws == ws'
227
247
228-- | Write each 'ByteString' with a variable sized value as a separator.
248-- | Write each 'ByteString' with a variable sized value as a separator.
229−prop_bitget_bytestring_interspersed :: (BinaryBit a, Binary a, Num a, Ord a, Bits a) => W a -> [B.ByteString] -> Property
249+prop_bitget_bytestring_interspersed :: (BinaryBit a, Binary a, Num a, Ord a, Bits a) => BitOrder -> W a -> [B.ByteString] -> Property
230−prop_bitget_bytestring_interspersed (W ws) bss = property $
250+prop_bitget_bytestring_interspersed bo (W ws) bss = property $
231 let p = mapM_ (\bs -> putBits (bitreq ws) ws >> putByteString bs) bss
251 let p = mapM_ (\bs -> putBits (bitreq ws) ws >> putByteString bs) bss
232 g = mapM (\bs -> (,) <$> getBits (bitreq ws) <*> getByteString (B.length bs)) bss
252 g = mapM (\bs -> (,) <$> getBits (bitreq ws) <*> getByteString (B.length bs)) bss
233− lbs = runPut (runBitPut p)
253+ lbs = runPut (runBitPut (withBitOrder bo p))
234− r = runGet (runBitGet g) lbs
254+ r = runGet (runBitGet (withBitOrder bo g)) lbs
235 in map (ws,) bss == r
255 in map (ws,) bss == r
236
256
237-- | Test failing.
257-- | Test failing.
@@ -241,8 +261,8 @@ prop_fail lbs errMsg0 = forAll (choose (0, 8 * L.length lbs)) $ \len ->
241 expectedBytesConsumed
261 expectedBytesConsumed
242 | bits == 0 = bytes
262 | bits == 0 = bytes
243 | otherwise = bytes + 1
263 | otherwise = bytes + 1
244− p = do getByteString (fromIntegral bytes)
264+ p = do _ <- getByteString (fromIntegral bytes)
245− getBits (fromIntegral bits) :: BitGet Word8
265+ _ <- getBits (fromIntegral bits) :: BitGet Word8
246 fail errMsg0
266 fail errMsg0
247 r = runGetIncremental (runBitGet p) `pushChunks` lbs
267 r = runGetIncremental (runBitGet p) `pushChunks` lbs
248 in case r of
268 in case r of
@@ -374,11 +394,6 @@ instance (Arbitrary (W a), Arbitrary (W b), Arbitrary (W c)) => Arbitrary (W (a,
374 arbitrary = ((W .) .) . (,,) <$> arbitraryW <*> arbitraryW <*> arbitraryW
394 arbitrary = ((W .) .) . (,,) <$> arbitraryW <*> arbitraryW <*> arbitraryW
375 shrink (W (a,b,c)) = ((W .) .) . (,,) <$> shrinkW a <*> shrinkW b <*> shrinkW c
395 shrink (W (a,b,c)) = ((W .) .) . (,,) <$> shrinkW a <*> shrinkW b <*> shrinkW c
376
396
377−integralRandomR :: (Integral a, RandomGen g) => (a,a) -> g -> (a,g)
378−integralRandomR (a,b) g = case randomR (fromIntegral a :: Integer,
379− fromIntegral b :: Integer) g of
380− (x,g) -> (fromIntegral x, g)
381−
382data Primitive
397data Primitive
383 = Bool Bool
398 = Bool Bool
384 | W8 Int Word8
399 | W8 Int Word8
@@ -387,6 +402,7 @@ data Primitive
387 | W64 Int Word64
402 | W64 Int Word64
388 | BS Int B.ByteString
403 | BS Int B.ByteString
389 | LBS Int L.ByteString
404 | LBS Int L.ByteString
405+ | Skip Int
390 | IsEmpty
406 | IsEmpty
391 deriving (Eq, Show)
407 deriving (Eq, Show)
392
408
@@ -405,6 +421,7 @@ instance Arbitrary Primitive where
405 , gen W16
421 , gen W16
406 , gen W32
422 , gen W32
407 , gen W64
423 , gen W64
424+ , Skip <$> choose (0, 3000)
408 , do n <- choose (0,10)
425 , do n <- choose (0,10)
409 cs <- vector n
426 cs <- vector n
410 return (BS n (B.pack cs))
427 return (BS n (B.pack cs))
@@ -421,10 +438,19 @@ instance Arbitrary Primitive where
421 W16 _ x -> snk W16 x
438 W16 _ x -> snk W16 x
422 W32 _ x -> snk W32 x
439 W32 _ x -> snk W32 x
423 W64 _ x -> snk W64 x
440 W64 _ x -> snk W64 x
441+ Skip x -> Skip <$> shrink x
424 BS _ bs -> let ws = B.unpack bs in map (\ws' -> BS (length ws') (B.pack ws')) (shrink ws)
442 BS _ bs -> let ws = B.unpack bs in map (\ws' -> BS (length ws') (B.pack ws')) (shrink ws)
425 LBS _ lbs -> let ws = L.unpack lbs in map (\ws' -> LBS (length ws') (L.pack ws')) (shrink ws)
443 LBS _ lbs -> let ws = L.unpack lbs in map (\ws' -> LBS (length ws') (L.pack ws')) (shrink ws)
426 IsEmpty -> []
444 IsEmpty -> []
427
445
446+instance Arbitrary BitOrder where
447+ arbitrary = elements [BB, LB, LL, BL]
448+ shrink LL = [BB,LB,BL]
449+ shrink BL = [BB,LB]
450+ shrink LB = [BB]
451+ shrink BB = []
452+
453+
428prop_primitive :: Primitive -> Property
454prop_primitive :: Primitive -> Property
429prop_primitive prim = property $
455prop_primitive prim = property $
430 let p = putPrimitive prim
456 let p = putPrimitive prim
@@ -449,6 +475,7 @@ putPrimitive p =
449 W16 n x -> putWord16 n x
475 W16 n x -> putWord16 n x
450 W32 n x -> putWord32 n x
476 W32 n x -> putWord32 n x
451 W64 n x -> putWord64 n x
477 W64 n x -> putWord64 n x
478+ Skip n -> skipBits n
452 BS _ bs -> putByteString bs
479 BS _ bs -> putByteString bs
453 LBS _ lbs -> mapM_ putByteString (L.toChunks lbs)
480 LBS _ lbs -> mapM_ putByteString (L.toChunks lbs)
454 IsEmpty -> return ()
481 IsEmpty -> return ()
@@ -461,10 +488,22 @@ getPrimitive p =
461 W16 n _ -> W16 n <$> getWord16 n
488 W16 n _ -> W16 n <$> getWord16 n
462 W32 n _ -> W32 n <$> getWord32 n
489 W32 n _ -> W32 n <$> getWord32 n
463 W64 n _ -> W64 n <$> getWord64 n
490 W64 n _ -> W64 n <$> getWord64 n
491+ Skip n -> skipBits n >> return (Skip n)
464 BS n _ -> BS n <$> getByteString n
492 BS n _ -> BS n <$> getByteString n
465 LBS n _ -> LBS n <$> getLazyByteString n
493 LBS n _ -> LBS n <$> getLazyByteString n
466 IsEmpty -> isEmpty >> return IsEmpty
494 IsEmpty -> isEmpty >> return IsEmpty
467
495
496+getPrimitiveSize :: Primitive -> Int
497+getPrimitiveSize p = case p of
498+ Bool _ -> 1
499+ W8 n _ -> n
500+ W16 n _ -> n
501+ W32 n _ -> n
502+ W64 n _ -> n
503+ Skip n -> n
504+ BS n _ -> n*8
505+ LBS n _ -> n*8
506+ IsEmpty -> 0
468
507
469verifyProgram :: Int -> Program -> BitGet Bool
508verifyProgram :: Int -> Program -> BitGet Bool
470verifyProgram totalLength ps0 = go 0 ps0
509verifyProgram totalLength ps0 = go 0 ps0
@@ -472,13 +511,6 @@ verifyProgram totalLength ps0 = go 0 ps0
472 go _ [] = return True
511 go _ [] = return True
473 go pos (p:ps) =
512 go pos (p:ps) =
474 case p of
513 case p of
475− Bool x -> check x getBool >> go (pos+1) ps
476− W8 n x -> check x (getWord8 n) >> go (pos+n) ps
477− W16 n x -> check x (getWord16 n) >> go (pos+n) ps
478− W32 n x -> check x (getWord32 n) >> go (pos+n) ps
479− W64 n x -> check x (getWord64 n) >> go (pos+n) ps
480− BS n x -> check x (getByteString n) >> go (pos+(8*n)) ps
481− LBS n x -> check x (getLazyByteString n) >> go (pos+(8*n)) ps
482 IsEmpty -> do
514 IsEmpty -> do
483 let expected = pos == totalLength
515 let expected = pos == totalLength
484 actual <- isEmpty
516 actual <- isEmpty
@@ -486,6 +518,7 @@ verifyProgram totalLength ps0 = go 0 ps0
486 then go pos ps
518 then go pos ps
487 else error $ "isEmpty returned wrong value, expected "
519 else error $ "isEmpty returned wrong value, expected "
488 ++ show expected ++ " but got " ++ show actual
520 ++ show expected ++ " but got " ++ show actual
521+ _ -> check p (getPrimitive p) >> go (pos + getPrimitiveSize p) ps
489 check x g = do
522 check x g = do
490 y <- g
523 y <- g
491 if x == y
524 if x == y
Data/Binary/Bits/Alignment.hsnew file+22
@@ -0,0 +1,22 @@
1+-----------------------------------------------------------------------------
2+-- |
3+-- Module : Data.Binary.Bits.Get
4+-- Copyright : (c) Lennart Kolmodin 2010-2011
5+-- (c) Sylvain Henry 2015
6+-- License : BSD3-style (see LICENSE)
7+--
8+-- Maintainer : kolmodin@gmail.com
9+-- Stability : experimental
10+-- Portability : portable (should run where the package binary runs)
11+
12+module Data.Binary.Bits.Alignment
13+ ( Alignable(..)
14+ )
15+where
16+
17+class Monad m => Alignable m where
18+ -- | Skip the given number of bits
19+ skipBits :: Int -> m ()
20+
21+ -- | Skip bits if necessary to align to the next byte
22+ alignByte :: m ()
Data/Binary/Bits/BitOrder.hs+2−2
@@ -19,9 +19,9 @@ where
19--
19--
20-- E.g. two words of 5 bits: ABCDE, VWXYZ
20-- E.g. two words of 5 bits: ABCDE, VWXYZ
21-- - BB: ABCDEVWX YZxxxxxx
21-- - BB: ABCDEVWX YZxxxxxx
22−-- - LB: XYZABCDE xxxxxxVW
22+-- - LL: XYZABCDE xxxxxxVW
23-- - BL: EDCBAZYX WVxxxxxx
23-- - BL: EDCBAZYX WVxxxxxx
24−-- - LL: XWVEDCBA xxxxxxZY
24+-- - LB: XWVEDCBA xxxxxxZY
25data BitOrder
25data BitOrder
26 = BB -- ^ Big-endian bytes and bits
26 = BB -- ^ Big-endian bytes and bits
27 | LB -- ^ Little-endian bytes, big-endian bits
27 | LB -- ^ Little-endian bytes, big-endian bits
Data/Binary/Bits/Get.hs+54−30
@@ -60,6 +60,7 @@ module Data.Binary.Bits.Get
60
60
61 -- ** Get bytes
61 -- ** Get bytes
62 , getBool
62 , getBool
63+ , getBool#
63 , getWord8
64 , getWord8
64 , getWord16
65 , getWord16
65 , getWord32
66 , getWord32
@@ -98,6 +99,7 @@ import Data.Binary.Get as B ( Get, getLazyByteString, isEmpty )
98import Data.Binary.Get.Internal as B ( get, put, ensureN )
99import Data.Binary.Get.Internal as B ( get, put, ensureN )
99import Data.Binary.Bits.BitOrder
100import Data.Binary.Bits.BitOrder
100import Data.Binary.Bits.Internal
101import Data.Binary.Bits.Internal
102+import Data.Binary.Bits.Alignment
101
103
102import Data.ByteString as B
104import Data.ByteString as B
103import qualified Data.ByteString.Lazy as L
105import qualified Data.ByteString.Lazy as L
@@ -113,6 +115,8 @@ import Control.Applicative
113import Control.Monad (when,foldM_)
115import Control.Monad (when,foldM_)
114
116
115import Prelude as P
117import Prelude as P
118+import GHC.Word
119+import GHC.Base (uncheckedShiftLWord8#, wordToWord8#, Int (..), andWord8#, Word8#, (-#), Int#, xorI#, eqWord8#, TYPE)
116
120
117-- $bitget
121-- $bitget
118-- Parse bits using a monad.
122-- Parse bits using a monad.
@@ -178,8 +182,20 @@ readBool :: S -> Bool
178readBool (S bs o bo) = case bo of
182readBool (S bs o bo) = case bo of
179 BB -> testBit (unsafeHead bs) (7-o)
183 BB -> testBit (unsafeHead bs) (7-o)
180 BL -> testBit (unsafeHead bs) (7-o)
184 BL -> testBit (unsafeHead bs) (7-o)
181− LB -> testBit (unsafeHead bs) o
182 LL -> testBit (unsafeHead bs) o
185 LL -> testBit (unsafeHead bs) o
186+ LB -> testBit (unsafeHead bs) o
187+
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+
183
199
184-- | Extract a range of bits from (ws :: ByteString)
200-- | Extract a range of bits from (ws :: ByteString)
185--
201--
@@ -202,17 +218,17 @@ extract bo bs o n
202
218
203 -- shift offset depending on the byte position (0..B.length-1)
219 -- shift offset depending on the byte position (0..B.length-1)
204 off i = case bo of
220 off i = case bo of
205− LB -> 8*i - o
206 LL -> 8*i - o
221 LL -> 8*i - o
222+ LB -> 8*i - o
207 BB -> (B.length bs -1 - i) * 8 - r
223 BB -> (B.length bs -1 - i) * 8 - r
208 BL -> (B.length bs -1 - i) * 8 - r
224 BL -> (B.length bs -1 - i) * 8 - r
209
225
210 -- reverse bits if necessary
226 -- reverse bits if necessary
211 rev = case bo of
227 rev = case bo of
212− LL -> reverseBits n
228+ LB -> reverseBits n
213 BL -> reverseBits n
229 BL -> reverseBits n
214 BB -> id
230 BB -> id
215− LB -> id
231+ LL -> id
216
232
217
233
218-- | Generic readWord
234-- | Generic readWord
@@ -233,22 +249,23 @@ readWordChecked m n s
233--
249--
234-- Examples:
250-- Examples:
235-- BB: xxxABCDE FGHIJKLM NOPxxxxx -> ABCDEFGH IJKLMNOP
251-- BB: xxxABCDE FGHIJKLM NOPxxxxx -> ABCDEFGH IJKLMNOP
236−-- LB: LMNOPxxx DEFGHIJK xxxxxABC -> ABCDEFGH IJKLMNOP
252+-- LL: LMNOPxxx DEFGHIJK xxxxxABC -> ABCDEFGH IJKLMNOP
237-- BL: xxxPONML KJIHGFED CBAxxxxx -> ABCDEFGH IJKLMNOP
253-- BL: xxxPONML KJIHGFED CBAxxxxx -> ABCDEFGH IJKLMNOP
238−-- LL: EDCBAxxx MLKJIHGF xxxxxPON -> ABCDEFGH IJKLMNOP
254+-- LB: EDCBAxxx MLKJIHGF xxxxxPON -> ABCDEFGH IJKLMNOP
239readByteString :: Int -> S -> ByteString
255readByteString :: Int -> S -> ByteString
240readByteString n (S bs o bo) =
256readByteString n (S bs o bo) =
241 let
257 let
242− bs' = unsafeTake (n+1) bs
258+ bs' = unsafeTake (n+1) bs
259+ bs'' = unsafeTake n bs
243− rev = B.map (reverseBits 8)
260+ rev = B.map (reverseBits 8)
244 in case (o,bo) of
261 in case (o,bo) of
245− (0,BB) -> unsafeTake n bs
262+ (0,BB) -> bs''
246− (0,LB) -> B.reverse (unsafeTake n bs)
263+ (0,LL) -> B.reverse bs''
247− (0,BL) -> rev (unsafeTake n bs)
264+ (0,LB) -> rev bs''
248− (0,LL) -> rev (B.reverse (unsafeTake n bs))
265+ (0,BL) -> rev . B.reverse $ bs''
249− (_,BL) -> rev (readByteString n (S (B.reverse bs') o LB))
266+ (_,LL) -> readByteString n (S (B.reverse bs') (8-o) BB)
250− (_,LL) -> rev (readByteString n (S (B.reverse bs') o BB))
267+ (_,BL) -> rev . B.reverse $ readByteString n (S bs' o BB)
251− (_,LB) -> readByteString n (S (B.reverse bs') o BB)
268+ (_,LB) -> rev . B.reverse $ readByteString n (S bs' o LL)
252 (_,BB) -> unsafePerformIO $ do
269 (_,BB) -> unsafePerformIO $ do
253 let len = n+1
270 let len = n+1
254 ptr <- mallocBytes len
271 ptr <- mallocBytes len
@@ -270,7 +287,6 @@ newtype BitGet a = B { runState :: S -> Get (S,a) }
270
287
271instance Monad BitGet where
288instance Monad BitGet where
272 return x = B $ \s -> return (s,x)
289 return x = B $ \s -> return (s,x)
273− fail str = B $ \s -> putBackState s >> fail str
274 (B f) >>= g = B $ \s -> do (s',a) <- f s
290 (B f) >>= g = B $ \s -> do (s',a) <- f s
275 runState (g a) s'
291 runState (g a) s'
276
292
@@ -290,6 +306,20 @@ instance BitOrderable BitGet where
290 (S _ _ bo) <- getState
306 (S _ _ bo) <- getState
291 return bo
307 return bo
292
308
309+instance Alignable BitGet where
310+ -- | Skip the given number of bits
311+ skipBits n = do
312+ ensureBits n
313+ withState (incS n)
314+
315+ -- | Skip bits if necessary to align to the next byte
316+ alignByte = do
317+ (S _ o _) <- getState
318+ when (o /= 0) $
319+ skipBits (8-o)
320+
321+
322+
293-- | Run a 'BitGet' within the Binary packages 'Get' monad. If a byte has
323-- | Run a 'BitGet' within the Binary packages 'Get' monad. If a byte has
294-- been partially consumed it will be discarded once 'runBitGet' is finished.
324-- been partially consumed it will be discarded once 'runBitGet' is finished.
295runBitGet :: BitGet a -> Get a
325runBitGet :: BitGet a -> Get a
@@ -337,20 +367,6 @@ ensureBits n = do
337 put B.empty
367 put B.empty
338 return (S (bs`append`bs') o bo, ())
368 return (S (bs`append`bs') o bo, ())
339
369
340−-- | Skip the given number of bits
341−skipBits :: Int -> BitGet ()
342−skipBits n = do
343− ensureBits n
344− withState (incS n)
345−
346−-- | Skip bits if necessary to align to the next byte
347−alignByte :: BitGet ()
348−alignByte = do
349− (S _ o _) <- getState
350− when (o /= 0) $
351− skipBits (8-o)
352−
353−
354-- | Test whether all input has been consumed, i.e. there are no remaining
370-- | Test whether all input has been consumed, i.e. there are no remaining
355-- undecoded bytes.
371-- undecoded bytes.
356isEmpty :: BitGet Bool
372isEmpty :: BitGet Bool
@@ -362,6 +378,10 @@ isEmpty = B $ \ (S bs o bo) -> if B.null bs
362getBool :: BitGet Bool
378getBool :: BitGet Bool
363getBool = block bool
379getBool = block bool
364
380
381+-- | Get 1 bit as a 'Bool'.
382+getBool# :: BitGet Int
383+getBool# = block bool#
384+
365-- | Get @n@ bits as a 'Word8'. @n@ must be within @[0..8]@.
385-- | Get @n@ bits as a 'Word8'. @n@ must be within @[0..8]@.
366getWord8 :: Int -> BitGet Word8
386getWord8 :: Int -> BitGet Word8
367getWord8 n = block (word8 n)
387getWord8 n = block (word8 n)
@@ -437,6 +457,10 @@ block (Block i p) = do
437bool :: Block Bool
457bool :: Block Bool
438bool = Block 1 readBool
458bool = Block 1 readBool
439
459
460+-- | Read a 1 bit 'Bool'.
461+bool# :: Block Int
462+bool# = Block 1 readBool#
463+
440-- | Read @n@ bits as a 'Word8'. @n@ must be within @[0..8]@.
464-- | Read @n@ bits as a 'Word8'. @n@ must be within @[0..8]@.
441word8 :: Int -> Block Word8
465word8 :: Int -> Block Word8
442word8 n = Block n (readWordChecked 8 n)
466word8 n = Block n (readWordChecked 8 n)
Data/Binary/Bits/Internal.hs+27−6
@@ -12,10 +12,13 @@
12
12
13module Data.Binary.Bits.Internal
13module Data.Binary.Bits.Internal
14 ( make_mask
14 ( make_mask
15+ , make_mask#
15 , mask
16 , mask
17+ , mask#
16 , bit_offset
18 , bit_offset
17 , byte_offset
19 , byte_offset
18 , reverseBits
20 , reverseBits
21+ , reverseBits#
19 , FastBits(..)
22 , FastBits(..)
20 )
23 )
21where
24where
@@ -38,11 +41,18 @@ make_mask n = (1 `shiftL` fromIntegral n) - 1
38{-# SPECIALIZE make_mask :: Int -> Word32 #-}
41{-# SPECIALIZE make_mask :: Int -> Word32 #-}
39{-# SPECIALIZE make_mask :: Int -> Word64 #-}
42{-# SPECIALIZE make_mask :: Int -> Word64 #-}
40
43
44+make_mask# :: Int# -> Word#
45+make_mask# n = (1## `shiftL#` n) `minusWord#` 1##
46+
41-- | Keep only the n least-significant bits of the given value
47-- | Keep only the n least-significant bits of the given value
42mask :: (Bits a, Num a) => Int -> a -> a
48mask :: (Bits a, Num a) => Int -> a -> a
43mask n v = v .&. make_mask n
49mask n v = v .&. make_mask n
44{-# INLINE mask #-}
50{-# INLINE mask #-}
45
51
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+
46-- | Compute bit offset (equivalent to x `mod` 8 but faster)
56-- | Compute bit offset (equivalent to x `mod` 8 but faster)
47bit_offset :: Int -> Int
57bit_offset :: Int -> Int
48bit_offset n = make_mask 3 .&. n
58bit_offset n = make_mask 3 .&. n
@@ -65,6 +75,17 @@ reverseBits n value = rec value n 0
65 rec 0 i r = r `fastShiftL` i
75 rec 0 i r = r `fastShiftL` i
66 rec v i r = rec (v `fastShiftR` 1) (i-1) ((r `fastShiftL` 1) .|. (v .&. 0x1))
76 rec v i r = rec (v `fastShiftR` 1) (i-1) ((r `fastShiftL` 1) .|. (v .&. 0x1))
67
77
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+
68
89
69---------------------------------------------------------------------
90---------------------------------------------------------------------
70-- Unchecked shifts, from the "binary" package
91-- Unchecked shifts, from the "binary" package
@@ -116,13 +137,13 @@ shiftr_w32 :: Word32 -> Int -> Word32
116shiftr_w64 :: Word64 -> Int -> Word64
137shiftr_w64 :: Word64 -> Int -> Word64
117
138
118#if defined(__GLASGOW_HASKELL__) && !defined(__HADDOCK__)
139#if defined(__GLASGOW_HASKELL__) && !defined(__HADDOCK__)
119−shiftl_w8 (W8# w) (I# i) = W8# (w `uncheckedShiftL#` i)
140+shiftl_w8 w i = w `unsafeShiftL` i
120−shiftl_w16 (W16# w) (I# i) = W16# (w `uncheckedShiftL#` i)
141+shiftl_w16 w i = w `unsafeShiftL` i
121−shiftl_w32 (W32# w) (I# i) = W32# (w `uncheckedShiftL#` i)
142+shiftl_w32 w i = w `unsafeShiftL` i
122
143
123−shiftr_w8 (W8# w) (I# i) = W8# (w `uncheckedShiftRL#` i)
144+shiftr_w8 w i = w `unsafeShiftR` i
124−shiftr_w16 (W16# w) (I# i) = W16# (w `uncheckedShiftRL#` i)
145+shiftr_w16 w i = w `unsafeShiftR` i
125−shiftr_w32 (W32# w) (I# i) = W32# (w `uncheckedShiftRL#` i)
146+shiftr_w32 w i = w `unsafeShiftR` i
126
147
127
148
128#if WORD_SIZE_IN_BITS < 64
149#if WORD_SIZE_IN_BITS < 64
Data/Binary/Bits/Put.hs+96−22
@@ -1,3 +1,5 @@
1+{-# LANGUAGE MagicHash #-}
2+
1-----------------------------------------------------------------------------
3-----------------------------------------------------------------------------
2-- |
4-- |
3-- Module : Data.Binary.Bits.Put
5-- Module : Data.Binary.Bits.Put
@@ -20,8 +22,10 @@ module Data.Binary.Bits.Put
20 -- * Data types
22 -- * Data types
21 -- ** Bool
23 -- ** Bool
22 , putBool
24 , putBool
25+ , putBool#
23
26
24 -- ** Words
27 -- ** Words
28+ , putWord#
25 , putWord8
29 , putWord8
26 , putWord16
30 , putWord16
27 , putWord32
31 , putWord32
@@ -41,14 +45,16 @@ import qualified Data.Binary.Put as Put
41import Data.Binary.Put ( Put )
45import Data.Binary.Put ( Put )
42import Data.Binary.Bits.Internal
46import Data.Binary.Bits.Internal
43import Data.Binary.Bits.BitOrder
47import Data.Binary.Bits.BitOrder
48+import Data.Binary.Bits.Alignment
44
49
45import Data.ByteString as BS
50import Data.ByteString as BS
46import Data.ByteString.Unsafe as BS
51import Data.ByteString.Unsafe as BS
47
52
48−import Control.Applicative
53+import Control.Monad (when)
49import Data.Bits
54import Data.Bits
50−import Data.Monoid
51import Data.Word
55import Data.Word
56+import GHC.Base (Word#, shiftRL#, Int (..), Int#, (-#), (<#), isTrue#, (+#), shiftL#, wordToWord8#, orWord8#)
57+import GHC.Word (Word8(..))
52
58
53data BitPut a = BitPut { run :: (S -> PairS a) }
59data BitPut a = BitPut { run :: (S -> PairS a) }
54
60
@@ -60,11 +66,18 @@ data S = S !Builder !Word8 !Int !BitOrder
60putBool :: Bool -> BitPut ()
66putBool :: Bool -> BitPut ()
61putBool b = putWord8 1 (if b then 0xff else 0x00)
67putBool b = putWord8 1 (if b then 0xff else 0x00)
62
68
69+-- | Put a 1 bit 'Bool'.
70+putBool# :: Word# -> BitPut ()
71+putBool# b = putWord# 1# b
72+
63
73
64-- | Generic putWord
74-- | Generic putWord
65putWord :: (Num a, FastBits a, Integral a) => Int -> a -> BitPut ()
75putWord :: (Num a, FastBits a, Integral a) => Int -> a -> BitPut ()
66putWord n w = BitPut $ \s -> PairS () (putWordS n w s)
76putWord n w = BitPut $ \s -> PairS () (putWordS n w s)
67
77
78+putWord# :: Int# -> Word# -> BitPut ()
79+putWord# n w = BitPut $ \s -> PairS () (putWordS# n w s)
80+
68putWordS :: (Num a, FastBits a, Integral a) => Int -> a -> S -> S
81putWordS :: (Num a, FastBits a, Integral a) => Int -> a -> S -> S
69putWordS n w s@(S builder b o bo) = s'
82putWordS n w s@(S builder b o bo) = s'
70 where
83 where
@@ -82,9 +95,9 @@ putWordS n w s@(S builder b o bo) = s'
82 -- Word containing the remaining (n-cn) bits to store in its LSB
95 -- Word containing the remaining (n-cn) bits to store in its LSB
83 w' = case bo of
96 w' = case bo of
84 BB -> w
97 BB -> w
85− BL -> w
86− LB -> w `fastShiftR` cn
98+ BL -> w `fastShiftR` cn
87 LL -> w `fastShiftR` cn
99 LL -> w `fastShiftR` cn
100+ LB -> w
88
101
89 -- Select bits to store in the current byte.
102 -- Select bits to store in the current byte.
90 -- Put them in the correct order and return them in the least-significant
103 -- Put them in the correct order and return them in the least-significant
@@ -92,17 +105,64 @@ putWordS n w s@(S builder b o bo) = s'
92 selectBits :: (Num a, FastBits a, Integral a) => a -> Word8
105 selectBits :: (Num a, FastBits a, Integral a) => a -> Word8
93 selectBits x = fromIntegral $ case bo of
106 selectBits x = fromIntegral $ case bo of
94 BB -> mask cn $ x `fastShiftR` (n-cn)
107 BB -> mask cn $ x `fastShiftR` (n-cn)
95− BL -> reverseBits cn $ mask cn $ x `fastShiftR` (n-cn)
108+ LB -> reverseBits cn $ mask cn $ x `fastShiftR` (n-cn)
96− LB -> mask cn x
109+ LL -> mask cn x
97− LL -> reverseBits cn $ mask cn x
110+ BL -> reverseBits cn $ mask cn x
98
111
99 -- shift left at the correct position
112 -- shift left at the correct position
100 shl :: Word8 -> Word8
113 shl :: Word8 -> Word8
101 shl x = case bo of
114 shl x = case bo of
102 BB -> x `fastShiftL` (8-o-cn)
115 BB -> x `fastShiftL` (8-o-cn)
103 BL -> x `fastShiftL` (8-o-cn)
116 BL -> x `fastShiftL` (8-o-cn)
104− LB -> x `fastShiftL` o
105 LL -> x `fastShiftL` o
117 LL -> x `fastShiftL` o
118+ LB -> x `fastShiftL` o
119+
120+ flush s2@(S b2 w2 o2 bo2)
121+ | o2 == 8 = S (b2 `mappend` B.singleton w2) 0 0 bo2
122+ | otherwise = s2
123+
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
106
166
107 flush s2@(S b2 w2 o2 bo2)
167 flush s2@(S b2 w2 o2 bo2)
108 | o2 == 8 = S (b2 `mappend` B.singleton w2) 0 0 bo2
168 | o2 == 8 = S (b2 `mappend` B.singleton w2) 0 0 bo2
@@ -141,27 +201,26 @@ putWord64be = putWord
141--
201--
142-- Examples: 3 bits are already written in the current byte
202-- Examples: 3 bits are already written in the current byte
143-- BB: ABCDEFGH IJKLMNOP -> xxxABCDE FGHIJKLM NOPxxxxx
203-- BB: ABCDEFGH IJKLMNOP -> xxxABCDE FGHIJKLM NOPxxxxx
144−-- LB: ABCDEFGH IJKLMNOP -> LMNOPxxx DEFGHIJK xxxxxABC
204+-- LL: ABCDEFGH IJKLMNOP -> LMNOPxxx DEFGHIJK xxxxxABC
145-- BL: ABCDEFGH IJKLMNOP -> xxxPONML KJIHGFED CBAxxxxx
205-- BL: ABCDEFGH IJKLMNOP -> xxxPONML KJIHGFED CBAxxxxx
146−-- LL: ABCDEFGH IJKLMNOP -> EDCBAxxx MLKJIHGF xxxxxPON
206+-- LB: ABCDEFGH IJKLMNOP -> EDCBAxxx MLKJIHGF xxxxxPON
147putByteString :: ByteString -> BitPut ()
207putByteString :: ByteString -> BitPut ()
148putByteString bs = BitPut $ \s -> PairS () (putByteStringS bs s)
208putByteString bs = BitPut $ \s -> PairS () (putByteStringS bs s)
149
209
150putByteStringS :: ByteString -> S -> S
210putByteStringS :: ByteString -> S -> S
151−putByteStringS bs (S builder b 0 BB) = S (builder `mappend` B.fromByteString bs) b 0 BB
211+putByteStringS bs s
152−putByteStringS bs s@(S _ _ _ BB)
153 | BS.null bs = s
212 | BS.null bs = s
213+ | otherwise = case s of
214+ (S builder b 0 BB) -> S (builder `mappend` B.fromByteString bs) b 0 BB
215+ (S builder b 0 LL) -> S (builder `mappend` B.fromByteString (BS.reverse bs)) b 0 LL
216+ (S builder b 0 LB) -> S (builder `mappend` B.fromByteString (rev bs)) b 0 LB
217+ (S builder b 0 BL) -> S (builder `mappend` B.fromByteString (rev (BS.reverse bs))) b 0 BL
154− | otherwise = putByteStringS (BS.unsafeTail bs) (putWordS 8 (BS.unsafeHead bs) s)
218+ (S _ _ _ BB) -> putByteStringS (BS.unsafeTail bs) (putWordS 8 (BS.unsafeHead bs) s)
155−putByteStringS bs (S builder b o bo) = putByteStringS bs' (S builder b o BB)
219+ (S _ _ _ LL) -> putByteStringS (BS.unsafeInit bs) (putWordS 8 (BS.unsafeLast bs) s)
220+ (S _ _ _ BL) -> putByteStringS (BS.unsafeInit bs) (putWordS 8 (BS.unsafeLast bs) s)
221+ (S _ _ _ LB) -> putByteStringS (BS.unsafeTail bs) (putWordS 8 (BS.unsafeHead bs) s)
156 where
222 where
157− rev = BS.map (reverseBits 8)
223+ rev = BS.map (reverseBits 8)
158−
159− bs' = case bo of
160− BL -> BS.reverse (rev bs)
161− LB -> BS.reverse bs
162− LL -> rev bs
163− BB -> bs -- should not occur, already matched but the compiler doesn't detect it
164−
165
224
166
225
167-- | Run a 'Put' inside 'BitPut'. Any partially written bytes will be flushed
226-- | Run a 'Put' inside 'BitPut'. Any partially written bytes will be flushed
@@ -179,6 +238,9 @@ flushIncomplete s@(S b w o bo)
179 | o == 0 = s
238 | o == 0 = s
180 | otherwise = (S (b `mappend` B.singleton w) 0 0 bo)
239 | otherwise = (S (b `mappend` B.singleton w) 0 0 bo)
181
240
241+getOffset :: BitPut Int
242+getOffset = BitPut $ \s@(S _ _ o _) -> PairS o s
243+
182-- | Run the 'BitPut' monad inside 'Put'.
244-- | Run the 'BitPut' monad inside 'Put'.
183runBitPut :: BitPut () -> Put.Put
245runBitPut :: BitPut () -> Put.Put
184runBitPut m = Put.putBuilder b
246runBitPut m = Put.putBuilder b
@@ -209,3 +271,15 @@ instance BitOrderable BitPut where
209 setBitOrder bo = BitPut $ \(S bu b o _) -> PairS () (S bu b o bo)
271 setBitOrder bo = BitPut $ \(S bu b o _) -> PairS () (S bu b o bo)
210
272
211 getBitOrder = BitPut $ \s@(S _ _ _ bo) -> PairS bo s
273 getBitOrder = BitPut $ \s@(S _ _ _ bo) -> PairS bo s
274+
275+instance Alignable BitPut where
276+ -- | Skip the given number of bits
277+ skipBits n
278+ | n <= 64 = putWord64 n 0
279+ | otherwise = putWord64 64 0 >> skipBits (n-64)
280+
281+ -- | Skip bits if necessary to align to the next byte
282+ alignByte = do
283+ o <- getOffset
284+ when (o /= 0) $
285+ skipBits (8-o)
binary-bits.cabal+3−1
@@ -25,7 +25,8 @@ library
25 exposed-modules: Data.Binary.Bits ,
25 exposed-modules: Data.Binary.Bits ,
26 Data.Binary.Bits.Put ,
26 Data.Binary.Bits.Put ,
27 Data.Binary.Bits.Get ,
27 Data.Binary.Bits.Get ,
28− Data.Binary.Bits.BitOrder
28+ Data.Binary.Bits.BitOrder ,
29+ Data.Binary.Bits.Alignment
29 other-modules: Data.Binary.Bits.Internal
30 other-modules: Data.Binary.Bits.Internal
30
31
31 default-language: Haskell98
32 default-language: Haskell98
@@ -36,6 +37,7 @@ test-suite qc
36 type: exitcode-stdio-1.0
37 type: exitcode-stdio-1.0
37 main-is: BitsQC.hs
38 main-is: BitsQC.hs
38 default-language: Haskell98
39 default-language: Haskell98
40+ --ghc-options: -O2 -Wall
39
41
40 build-depends: base==4.*, binary >= 0.6.0.0, bytestring,
42 build-depends: base==4.*, binary >= 0.6.0.0, bytestring,
41 QuickCheck>=2, random,
43 QuickCheck>=2, random,
flake.locknew file+59
@@ -0,0 +1,59 @@
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.nixnew file+24
@@ -0,0 +1,24 @@
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.yamlnew file+66
@@ -0,0 +1,66 @@
1+# This file was automatically generated by 'stack init'
2+#
3+# Some commonly used options have been documented as comments in this file.
4+# For advanced use and comprehensive documentation of the format, please see:
5+# https://docs.haskellstack.org/en/stable/yaml_configuration/
6+
7+# Resolver to choose a 'specific' stackage snapshot or a compiler version.
8+# A snapshot resolver dictates the compiler version and the set of packages
9+# to be used for project dependencies. For example:
10+#
11+# resolver: lts-3.5
12+# resolver: nightly-2015-09-21
13+# resolver: ghc-7.10.2
14+#
15+# The location of a snapshot can be provided as a file or url. Stack assumes
16+# a snapshot provided as a file might change, whereas a url resource does not.
17+#
18+# resolver: ./custom-snapshot.yaml
19+# resolver: https://example.com/snapshots/2018-01-01.yaml
20+resolver: lts-22.43
21+
22+# User packages to be built.
23+# Various formats can be used as shown in the example below.
24+#
25+# packages:
26+# - some-directory
27+# - https://example.com/foo/bar/baz-0.0.2.tar.gz
28+# subdirs:
29+# - auto-update
30+# - wai
31+packages:
32+- .
33+# Dependency packages to be pulled from upstream that are not in the resolver.
34+# These entries can reference officially published versions as well as
35+# forks / in-progress versions pinned to a git hash. For example:
36+#
37+# extra-deps:
38+# - acme-missiles-0.3
39+# - git: https://github.com/commercialhaskell/stack.git
40+# commit: e7b331f14bcffb8367cd58fbfc8b40ec7642100a
41+#
42+# extra-deps: []
43+
44+# Override default flag values for local packages and extra-deps
45+# flags: {}
46+
47+# Extra package databases containing global packages
48+# extra-package-dbs: []
49+
50+# Control whether we use the GHC we find on the path
51+# system-ghc: true
52+#
53+# Require a specific version of stack, using version ranges
54+# require-stack-version: -any # Default
55+# require-stack-version: ">=2.9"
56+#
57+# Override the architecture used by stack, especially useful on Windows
58+# arch: i386
59+# arch: x86_64
60+#
61+# Extra directories used by stack for building
62+# extra-include-dirs: [/path/to/dir]
63+# extra-lib-dirs: [/path/to/dir]
64+#
65+# Allow a newer minor version of GHC than the snapshot specifies
66+# compiler-check: newer-minor
stack.yaml.locknew file+12
@@ -0,0 +1,12 @@
1+# This file was autogenerated by Stack.
2+# You should not edit this file by hand.
3+# For more information, please see the documentation at:
4+# https://docs.haskellstack.org/en/stable/lock_files
5+
6+packages: []
7+snapshots:
8+- completed:
9+ sha256: 08bd13ce621b41a8f5e51456b38d5b46d7783ce114a50ab604d6bbab0d002146
10+ size: 720271
11+ url: https://raw.githubusercontent.com/commercialhaskell/stackage-snapshots/master/lts/22/43.yaml
12+ original: lts-22.43