blob: 561eef1a4e6c6da5be045070b09e61f9a6415114 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
|
{-# LANGUAGE DataKinds #-}
{-# LANGUAGE OverloadedStrings #-}
{-# LANGUAGE QuasiQuotes #-}
{-# LANGUAGE ScopedTypeVariables #-}
module Main where
import Control.Monad
import Criterion
import Criterion.Main
import qualified Data.ByteString.Char8 as B
import Data.FastQ
import Data.Function
import qualified Streamly.Data.Fold as F
import qualified Streamly.Data.Stream as S
import qualified Streamly.Data.Unfold as U
import Test.QuickCheck.Arbitrary
import Test.QuickCheck.Gen
import Prelude hiding (Read)
import System.IO.Temp
import Options.Generic
import qualified Dedumi as D
import Data.Maybe
import qualified Data.Text as T
import Debug.Trace
instance Arbitrary Read where
arbitrary = Read <$> genStr <*> genStr <*> genStr
where
genBS = fmap B.pack . listOf1 . elements
genStr = genBS $ [' ' .. '~']
--sizes = [2 ^ i | i <- [11 .. 19]]
sizes = [2*8192]
setupEnv root size = do
let fileroot = root <> "/" <> show size
a = fileroot <> "a"
b = fileroot <> "b"
rp <- replicateM size $ generate $ arbitrary
S.unfold U.fromList rp & unparse a b
pure (a, b)
main :: IO ()
main = withSystemTempDirectory "dedumi-bench" $ \tmpdir ->
defaultMain
[ bgroup "main" $ map (\s -> env (setupEnv tmpdir s) $ \ ~(a,b) -> bench (show s) (run (T.pack a) (T.pack b))) sizes
]
run fq1 fq2 = nfAppIO D.go $ fromJust $ unwrapRecordPure [fq1, fq2, "/dev/null", "/dev/null"]
|
