module Main where

import           Criterion.Main
import           Numeric.Combinatorics
import           Numeric.Integer
import           Numeric.NumberTheory
import           Numeric.Pure

main :: IO ()
main =
    defaultMain [ bgroup "primality check"
                      [ bench "isPrime" $ nf isPrime 2017
                      , bench "hsIsPrime" $ nf hsIsPrime (2017 :: Int)
                      ]
                , bgroup "φ"
                      [ bench "totient" $ nf totient 2016
                      , bench "hsTotient" $ nf hsTotient (2016 :: Int)
                      ]
                , bgroup "τ"
                      [ bench "tau" $ nf tau 3018
                      , bench "hsTau" $ nf hsTau (3018 :: Int)
                      ]
                , bgroup "ω"
                      [ bench "littleOmega" $ nf littleOmega 91
                      , bench "hsLittleOmega" $ nf hsLittleOmega (91 :: Int)
                      ]
                , bgroup "perfection check"
                      [ bench "isPerfect" $ nf isPerfect 318
                      , bench "hsIsPerfect" $ nf hsIsPerfect (318 :: Int)
                      ]
                , bgroup "factorial"
                      [ bench "factorial" $ nfIO (factorial 160)
                      , bench "hsFactorial" $ nf hsFactorial (160 :: Integer)
                      ]
                , bgroup "doubleFactorial"
                      [ bench "doubleFactorial" $ nfIO (doubleFactorial 79)
                      , bench "hsDoubleFactorial" $ nf hsDoubleFactorial (79 :: Integer)
                      ]
                , bgroup "choose"
                      [ bench "choose" $ nfIO (choose 322 16)
                      , bench "hsChoose" $ nf (hsChoose 322) (16 :: Integer)
                      ]
                , bgroup "catalan"
                      [ bench "catalan" $ nfIO (catalan 300)
                      , bench "hsCatalan" $ nf hsCatalan (300 :: Integer)
                      ]
                , bgroup "jacobi"
                      [ bench "jacobi" $ nf (jacobi 80) 111
                      , bench "hsJacobi" $ nf (hsJacobi 80) (111 :: Int)
                      ]
                , bgroup "fibonacci"
                      [ bench "fibonacci" $ nfIO (fibonacci 200)
                      , bench "hsFibonacci" $ nf (hsFibonacci :: Int -> Integer) 200
                      ]
                ]