import Data.BinaryList (BinList)
import qualified Data.BinaryList as BL

import Control.Applicative
import qualified Data.Foldable as F

-- criterion
import Criterion.Main

list1024 :: [Int]
list1024 = [1..1024] -- 2^10 = 1024

list513 :: [Int]
list513 = [1..513]

blist1024 :: BinList Int
blist1024 = BL.generate 10 id

blist1024d :: BinList (Int,Int)
blist1024d = BL.zip blist1024 (BL.reverse blist1024)

main :: IO ()
main = defaultMain
  [ bgroup "1024"
      [ bench "fromList" $ nf (\i -> const BL.fromList i $ list1024) 0
      , bench "fromListWithDefault" $ nf (\i -> BL.fromListWithDefault i list513) 0
      , bench "fromListSplit" $ nf (\i -> BL.fromListSplit i 10 list513) 0
      , bench "generate" $ nf (\i -> BL.generate i id) 10
      , bench "replicate" $ nf (\i -> BL.replicate i (0 :: Int)) 10
      , bench "toListSegment" $ nf (\e -> BL.toListSegment 256 e blist1024) 768
      , bench "split" $ nf BL.split blist1024
      , bench "take" $ nf (BL.take 5) blist1024
      , bench "takeEnd" $ nf (BL.takeEnd 5) blist1024
      , bench "reverse" $ nf BL.reverse blist1024
      , bench "joinPairs" $ nf BL.joinPairs blist1024d
      , bench "disjoinPairs" $ nf BL.disjoinPairs blist1024
      , bench "zip" $ nf (\(xs,ys) -> BL.zip xs ys) (blist1024, blist1024)
      , bench "unzip" $ nf BL.unzip blist1024d
        ]
    ]