module Main (main) where

import Control.DeepSeq
import Control.Monad
import Data.Text (Text)
import Data.Text.Metrics
import Weigh
import qualified Data.Text as T

main :: IO ()
main = mainWith $ do
  setColumns [Case, Allocated, GCs, Max]
  bmetric "levenshtein"            levenshtein
  bmetric "levenshteinNorm"        levenshteinNorm
  bmetric "damerauLevenshtein"     damerauLevenshtein
  bmetric "damerauLevenshteinNorm" damerauLevenshteinNorm
  bmetric "overlap"                overlap
  bmetric "jaccard"                jaccard
  bmetric "hamming"                hamming
  bmetric "jaro"                   jaro
  bmetric "jaroWinkler"            jaroWinkler

-- | Perform a series to measurements with the same metric function.

bmetric :: NFData a
  => String            -- ^ Name of the benchmark group
  -> (Text -> Text -> a) -- ^ The function to benchmark
  -> Weigh ()
bmetric name f = forM_ stdSeries $ \n ->
  func (name ++ "/" ++ show n) (uncurry f) (testData n, testData n)

-- | The series of lengths to try with every function as part of 'btmetric'.

stdSeries :: [Int]
stdSeries = [5,10,20,40,80,160]

testData :: Int -> Text
testData n = T.pack . take n . drop (n `mod` 4) . cycle $ ['a'..'z']