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
|
module Main where
import Criterion.Main
import Criterion.Main.Options
import Options.Applicative
import Prelude ()
import Prelude.Compat
data CustomArgs = CustomArgs
{ -- This data type adds two new arguments, listed below.
customArg1 :: Int
, customArg2 :: String
-- The remaining arguments come from criterion itself.
, criterionArgs :: Mode
}
customParser :: Parser CustomArgs
customParser = CustomArgs
<$> option auto
( long "custom-arg1"
<> value 42
<> metavar "INT"
<> help "Custom argument 1" )
<*> strOption
( long "custom-arg2"
<> value "Benchmark name"
<> metavar "STR"
<> help "Custom argument 2" )
<*> parseWith defaultConfig
main :: IO ()
main = do
args <- execParser $ describeWith customParser
putStrLn $ "custom-arg1: " ++ show (customArg1 args)
putStrLn $ "custom-arg2: " ++ customArg2 args
runMode (criterionArgs args)
[ bench (customArg2 args) $ whnf id $ customArg1 args ]
|
