add F66 benchs

Author: raehik

bench/Bench.hs

@@ -7,7 +7,9 @@ import           Language.Fortran.Version
 import           Language.Fortran.AST
 
 import qualified Language.Fortran.Parser.Fortran90 as F90
+import qualified Language.Fortran.Parser.Fortran66 as F66
 import qualified Language.Fortran.Lexer.FreeForm   as LexFree
+import qualified Language.Fortran.Lexer.FixedForm  as LexFixed
 
 import           Data.ByteString (ByteString)
 import qualified Data.Text as Text
@@ -18,32 +20,52 @@ main :: IO ()
 main = defaultMain
   [ bgroup "parser"
       [ bgroup "Fortran 90"
-          [ bench "statement (lone)" $ whnf pF90Stmt snippetFreeStmt
-          , bench "function  (lone)" $ whnf pF90Func snippetFreeFuncComplex
+          [ bench "statement (expr)" $ whnf pF90Stmt snippetFreeStmtExpr
+          , bench "statement (assign)" $ whnf pF90Stmt snippetFreeStmtDeclAssign
+          , bench "function" $ whnf pF90Func snippetFreeFunc
+          ]
+      , bgroup "Fortran 66"
+          [ bench "statement (expr)" $ whnf pF66Stmt snippetFixedStmt
+          , bench "function" $ whnf pF90Func snippetFixedFunc
           ]
       ]
   ]
 
+--------------------------------------------------------------------------------
+
 pF90Stmt :: ByteString -> Statement A0
 pF90Stmt = parseFree Fortran90 F90.statementParser
 
 pF90Func :: ByteString -> ProgramUnit A0
 pF90Func = parseFree Fortran90 F90.functionParser
 
+pF66Stmt :: ByteString -> Statement A0
+pF66Stmt = parseFixed Fortran66 F66.statementParser
+
 parseFree :: FortranVersion
           -> Parse LexFree.AlexInput LexFree.Token a -> ByteString -> a
 parseFree ver parser src = evalParse parser parserState
   where parserState = LexFree.initParseState src ver "<unknown>"
 
+parseFixed :: FortranVersion
+           -> Parse LexFixed.AlexInput LexFixed.Token a -> ByteString -> a
+parseFixed ver parser src = evalParse parser parserState
+  where parserState = LexFixed.initParseState src ver "<unknown>"
+
 --------------------------------------------------------------------------------
 
-snippetFreeStmt :: ByteString
-snippetFreeStmt = programListing $
+snippetFreeStmtExpr :: ByteString
+snippetFreeStmtExpr = programListing $
+  [ "x = y*2 + z - 1"
+  ]
+
+snippetFreeStmtDeclAssign :: ByteString
+snippetFreeStmtDeclAssign = programListing $
   [ "character(5) :: assign_in_decl*5 = \"test!\""
   ]
 
-snippetFreeFuncComplex :: ByteString
-snippetFreeFuncComplex = programListing $
+snippetFreeFunc :: ByteString
+snippetFreeFunc = programListing $
   [ "integer function f(x, y, z) result(i)"
   , "  print *, i"
   , "  i = (i - 1)"
@@ -52,6 +74,21 @@ snippetFreeFuncComplex = programListing $
 
 --------------------------------------------------------------------------------
 
+snippetFixedStmt :: ByteString
+snippetFixedStmt = programListing $
+  [ "      x = y*2 + z - 1"
+  ]
+
+snippetFixedFunc :: ByteString
+snippetFixedFunc = programListing $
+  [ "      subroutine f(x, y, z)"
+  , "      print *, i"
+  , "      i = (i - 1)"
+  , "      end"
+  ]
+
+--------------------------------------------------------------------------------
+
 -- | unlines but without the trailing newline
 programListing :: [String] -> ByteString
 programListing = strToByteString . intercalate "\n"