ppad-bech32-0.1.0: bench/Main.hs

{-# OPTIONS_GHC -fno-warn-orphans #-}
{-# LANGUAGE DeriveGeneric #-}
{-# LANGUAGE OverloadedStrings #-}
{-# LANGUAGE StandaloneDeriving #-}

module Main where

import Criterion.Main
import qualified Data.ByteString as BS
import qualified Data.ByteString.Bech32 as Bech32
import GHC.Generics
import qualified Reference.Bech32 as R
import Control.DeepSeq

deriving instance Generic R.Word5
instance NFData R.Word5

main :: IO ()
main = defaultMain [
    suite
  ]

suite :: Benchmark
suite = env setup $ \ ~(a, b, c) -> bgroup "benchmarks" [
      bgroup "ppad-bech32" [
        bgroup "bech32" [
            bench "120b" $ whnf (Bech32.encode "bc")
              "jtobin was here"
          , bench "128b (non 40-bit multiple length)" $ whnf (Bech32.encode "bc")
              "jtobin was here!"
          , bench "240b" $ whnf (Bech32.encode "bc")
              "jtobin was herejtobin was here"
        ]
      ]
    , bgroup "reference" [
        bgroup "bech32" [
            bench "120b" $ whnf (R.bech32Encode "bc") a
          , bench "128b (non 40-bit multiple length)" $
              whnf (R.bech32Encode "bc") b
          , bench "240b" $ whnf (R.bech32Encode "bc") c
        ]
      ]
    ]
  where
    setup = do
      let a = R.toBase32 (BS.unpack "jtobin was here")
          b = R.toBase32 (BS.unpack "jtobin was here!")
          c = R.toBase32 (BS.unpack "jtobin was herejtobin was here")
      pure (a, b, c)