module Copilot.Compile.C99.Test.CheckSpec (checkSpec) where
import Copilot.Core (Spec)
import Copilot.Core.Interpret.Eval (eval, ExtEnv(..))
import Copilot.Compile.C99 (compile)
import Copilot.Compile.C99.Params (Params (..), defaultParams)
import Copilot.Compile.C99.Test.Driver (driver)
import Copilot.Compile.C99.Test.Iteration (Iteration(..), execTraceToIterations)
import Copilot.Compile.C99.Test.ReadCSV (iterationsFromCSV)
import Copilot.Core.Type.Show (ShowType(..))
import Data.ByteString (ByteString)
import qualified Data.ByteString.Char8 as B
import qualified Data.Map as M
import qualified Data.Text.IO as TIO
import System.Directory (removeFile)
import System.Process (system, readProcess)
import Control.Monad (when, unless)
import Text.PrettyPrint (text, (<+>), ($$), render, vcat, hang)
checkSpec :: Int -> Spec -> IO Bool
checkSpec numIterations spec = do
genCFiles numIterations spec
compileCFiles
csv <- execute numIterations
let is1 = iterationsFromCSV csv
let is2 = interp numIterations spec
let eq = is1 == is2
unless eq (putStrLn $ showCompare is1 is2)
when eq cleanUp
return eq
showCompare :: [Iteration] -> [Iteration] -> String
showCompare s1 s2 =
render $ text "From C:" <+> text "---" <+> text "From Interpreter:" $$
(vcat $ map (\(a,b) -> hang a 10 b) zipped)
where
zipped = zip (toDoc s1) (toDoc s2)
toDoc = map (text . show)
genCFiles :: Int -> Spec -> IO ()
genCFiles numIterations spec = do
compile (defaultParams { prefix = Nothing, verbose = False }) spec
TIO.writeFile "driver.c" (driver M.empty numIterations spec)
return ()
compileCFiles :: IO ()
compileCFiles = do
_ <- system $ "gcc copilot.c driver.c -o _test"
return ()
execute :: Int -> IO ByteString
execute _ = do
fmap B.pack (readProcess "./_test" [] "")
interp :: Int -> Spec -> [Iteration]
interp numIterations =
execTraceToIterations . eval C numIterations (ExtEnv [] [] [])
cleanUp :: IO ()
cleanUp = do
removeFile "copilot.c"
removeFile "copilot.h"
removeFile "driver.c"
removeFile "_test"
return ()