@@ -16,6 +16,10 @@ along with this program; if not, write to the Free Software
1616Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
1717-}
1818import Text.Pandoc
19+import Text.Pandoc.Class hiding (getCurrentTime)
20+import Data.Time (getCurrentTime)
21+import qualified Data.ByteString as B
22+import qualified Data.Map as Map
1923import Criterion.Main
2024import Criterion.Types (Config(..))
2125import Data.Maybe (mapMaybe)
@@ -30,7 +34,7 @@ readerBench doc (name, reader) =
3034 let inp = either (error . show) id $ runPure
3135 $ writer def{ writerWrapText = WrapAuto} doc
3236 in return $ bench (name ++ " reader") $ nf
33− (reader def{ readerSmart = True }) inp
37+ (reader def) inp
3438 _ -> trace ("\nCould not find writer for " ++ name ++ "\n") Nothing
3539
3640writerBench :: Pandoc
@@ -42,13 +46,23 @@ writerBench doc (name, writer) = bench (name ++ " writer") $ nf
4246main :: IO ()
4347main = do
4448 inp <- readFile "tests/testsuite.txt"
45− let opts = def{ readerSmart = True }
49+ lalune <- B.readFile "tests/lalune.jpg"
50+ movie <- B.readFile "tests/movie.jpg"
51+ time <- getCurrentTime
52+ let setupFakeFiles = modifyPureState $ \st -> st{ stFiles =
53+ FileTree $ Map.fromList [
54+ ("lalune.jpg", FileInfo time lalune),
55+ ("movie.jpg", FileInfo time movie)
56+ ]}
57+ let opts = def
4658 let doc = either (error . show) id $ runPure $ readMarkdown opts inp
47− let readers' = [(n, \o -> either (error . show) id . runPure . r o)
59+ let readers' = [(n, \o d ->
60+ either (error . show) id $ runPure $ r o d)
4861 | (n, StringReader r) <- readers]
4962 let readerBs = mapMaybe (readerBench doc)
5063 $ filter (\(n,_) -> n /="haddock") readers'
51− let writers' = [(n, \o -> either (error . show) id . runPure . w o)
64+ let writers' = [(n, \o d ->
65+ either (error . show) id $ runPure $ setupFakeFiles >> w o d)
5266 | (n, StringWriter w) <- writers]
5367 let writerBs = map (writerBench doc)
5468 $ writers'