{-# LANGUAGE CPP #-}
{-# LANGUAGE BangPatterns #-}
{-# LANGUAGE StandaloneDeriving, DeriveGeneric #-}
{-# OPTIONS_GHC -fno-warn-orphans #-}

-- | Benchmark speed.

module Main where

import           Control.DeepSeq
import           Criterion
import           Criterion.Main
import           Data.ByteString (ByteString)
import qualified Data.ByteString as S
import qualified Data.ByteString.Lazy as L
import           GHC.Generics
import qualified Text.XML.Expat.SAX as Hexpat
import qualified Text.XML.Expat.Tree as HexpatTree
import qualified Text.XML.Hexml as Hexml
import           Text.XML.Light as XML
import qualified Xeno.SAX
import qualified Xeno.Types
import qualified Xeno.DOM
import qualified Xeno.DOM.Robust
#ifdef LIBXML2
import qualified Text.XML.LibXML.Parser as Libxml2
#endif


readFileZ :: FilePath -> IO (ByteString, Xeno.Types.ByteStringZeroTerminated)
readFileZ fn = do
    !s <- S.readFile fn
    let !sz = Xeno.Types.BSZT (s `S.snoc` 0)
    return (s,  sz)


main :: IO ()
main = defaultMain $
    (flip map) [ ("4KB", "data/books-4kb.xml")
               , ("31KB", "data/text-31kb.xml")
               , ("211KB", "data/fabricated-211kb.xml")
               ]
    $ \(group, fn) ->
        env (readFileZ fn)
            (\ ~(!input, !inputz) -> bgroup group
                 [ bench "hexml-dom" (whnf Hexml.parse input)
                 , bench "xeno-sax" (whnf Xeno.SAX.validate input)
                 , bench "xeno-sax-z" (whnf Xeno.SAX.validate inputz)
                 , bench "xeno-sax-ex" (whnf Xeno.SAX.validateEx input)
                 , bench "xeno-sax-ex-z" (whnf Xeno.SAX.validateEx inputz)
                 , bench "xeno-dom" (whnf Xeno.DOM.parse input)
                 , bench "xeno-dom-with-recovery" (whnf Xeno.DOM.Robust.parse input)
                 , bench
                     "hexpat-sax"
                     (whnf
                        ((Hexpat.parseThrowing Hexpat.defaultParseOptions :: L.ByteString -> [Hexpat.SAXEvent ByteString ByteString]) .
                         L.fromStrict)
                        input)
                 , bench
                     "hexpat-dom"
                     (whnf
                        ((HexpatTree.parse' HexpatTree.defaultParseOptions :: ByteString -> Either HexpatTree.XMLParseError (HexpatTree.Node ByteString ByteString)))
                        input)
                 , bench "xml-dom" (nf XML.parseXMLDoc input)
#ifdef LIBXML2
                 , bench "libxml2-dom" (whnfIO (Libxml2.parseMemory input))
#endif
                 ])


deriving instance Generic Content
deriving instance Generic Element
deriving instance Generic CData
deriving instance Generic CDataKind
deriving instance Generic QName
deriving instance Generic Attr
instance NFData Content
instance NFData Element
instance NFData CData
instance NFData CDataKind
instance NFData QName
instance NFData Attr