Adds benchmarking code.

Author: angerman

bench/src/Bench.hs

@@ -0,0 +1,25 @@
+module Bench where
+
+import Prelude hiding (writeFile)
+import Data.Binary (decodeFile)
+import Data.BitCode
+import Data.BitCode.Writer             (emitTopLevel)
+import Data.BitCode.Writer.Monad       (writeFile)
+import Data.BitCode.Writer.Combinators (withHeader)
+
+import           Criterion.Main
+
+--------------------------------------------------------------------------------
+-- Reading a serialized module.
+readModule :: FilePath -> IO [BitCode]
+readModule = decodeFile
+
+writeModule :: FilePath -> [BitCode] -> IO ()
+writeModule f = writeFile f . withHeader True . emitTopLevel
+
+main :: IO ()
+main = defaultMain [
+  bgroup "writer"
+  [ bench "HelloWorld" $ nfIO (writeModule "HelloWorld.bc" =<< readModule "bench/data/HelloWorld.mod")
+  , bench "HelloWorld2" $ nfIO (writeModule "HelloWorld2.bc" =<< readModule "bench/data/HelloWorld2.mod")
+  ]]

data-bitcode.cabal

@@ -45,3 +45,21 @@ library
     , binary >= 0.8
     , base16-bytestring
   default-language: Haskell2010
+
+benchmark writer
+  type: exitcode-stdio-1.0
+  main-is: Bench.hs
+  hs-source-dirs:
+      bench/src
+  ghc-options: -main-is Bench
+  build-depends:
+      base >= 4.7 && < 5
+    , pretty >= 1.1
+    , bytestring >= 0.10
+    , binary >= 0.8
+    , base16-bytestring
+    , criterion
+    , data-bitcode
+  other-modules:
+      Bench
+  default-language: Haskell2010

package.yaml

@@ -18,3 +18,11 @@ dependencies:
 
 library:
   source-dirs: src
+
+benchmarks:
+  writer:
+    dependencies:
+      - criterion
+      - data-bitcode
+    source-dirs: bench/src
+    main: Bench

src/Data/BitCode.hs

@@ -9,7 +9,7 @@ import Data.Maybe (catMaybes)
 import Data.Bits (FiniteBits, finiteBitSize, countLeadingZeros)
 
 import GHC.Generics                      (Generic)
-import Data.Binary                       (Binary)
+import Data.Binary                       (Binary, decodeFile)
 
 --- Bit Codes ------------------------------------------------------------------
 -- see BitCodes.h (e.g. http://llvm.org/docs/doxygen/html/BitCodes_8h_source.html)