@@ -19,6 +19,7 @@ Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
1919-}
2020import Prelude
2121import Text.Pandoc
22+import Text.Pandoc.MIME
2223import Text.Pandoc.Error (PandocError(..))
2324import Control.Monad.Except (throwError)
2425import qualified Text.Pandoc.UTF8 as UTF8
@@ -28,6 +29,7 @@ import Criterion.Types (Config(..))
2829import Data.List (intersect)
2930import Data.Maybe (mapMaybe)
3031import System.Environment (getArgs)
32+import qualified Data.ByteString.Lazy as BL
3133
3234readerBench :: Pandoc
3335 -> String
@@ -43,21 +45,34 @@ readerBench doc name =
4345 case (getReader name, getWriter name) of
4446 (Right (TextReader r, rexts),
4547 Right (TextWriter w, wexts)) -> do
48+ setResourcePath ["../test"]
4649 inp <- w def{ writerWrapText = WrapAuto
4750 , writerExtensions = wexts } doc
4851 return $ (r def{ readerExtensions = rexts }, inp)
4952 _ -> throwError $ PandocSomeError
5053 $ "could not get text reader and writer for " ++ name
5154
55+getImages :: IO [(FilePath, MimeType, BL.ByteString)]
56+getImages = do
57+ ll <- BL.readFile "test/lalune.jpg"
58+ mv <- BL.readFile "test/movie.jpg"
59+ return [("lalune.jpg", "image/jpg", ll)
60+ ,("movie.jpg", "image/jpg", mv)]
61+
5262writerBench :: Pandoc
5363 -> String
5464 -> Maybe Benchmark
5565writerBench doc name =
5666 case res of
5767 Right writerFun ->
58− Just $ bench (name ++ " writer")
68+ Just $ env getImages $ \imgs ->
69+ bench (name ++ " writer")
5970 $ nf (\d -> either (error . show) id $
60− runPure (writerFun d)) doc
71+ runPure (do mapM_
72+ (\(fp, mt, bs) ->
73+ insertMedia fp (Just mt) bs)
74+ imgs
75+ writerFun d)) doc
6176 Left _ -> Nothing
6277 where res = runPure $ do
6378 case (getWriter name) of