diff options
Diffstat (limited to 'src/Fun/Compiler.hs')
| -rw-r--r-- | src/Fun/Compiler.hs | 19 |
1 files changed, 15 insertions, 4 deletions
diff --git a/src/Fun/Compiler.hs b/src/Fun/Compiler.hs index a8134ed..4d9a67a 100644 --- a/src/Fun/Compiler.hs +++ b/src/Fun/Compiler.hs @@ -6,10 +6,21 @@ import Fun.Tree import System.Exit import System.IO -genTrace :: [String] -> String -genTrace xs = - "Trace of expectance: " - ++ foldr (\a b -> a ++ if b == "" then b else " -> " ++ b) "" xs +traceBranch :: Int -> Trace -> String +traceBranch c (StringTrace t) = "\n" ++ (replicate c ' ') ++ t +traceBranch c (OrTrace t1 t2) = + "\n" + ++ (replicate c ' ') + ++ "<either>" + ++ (traceTree (c + 2) t1) + ++ (traceTree (c + 2) t2) + +traceTree :: Int -> [Trace] -> String -- TODO: Indent/arrow first map +traceTree c ts = foldr join "" (map (traceBranch c) ts) + where join = (\a b -> a ++ if b == "" then b else " " ++ b) + +genTrace :: [Trace] -> String +genTrace ts = "Trace of expectance:\n" ++ traceTree 0 ts parse :: String -> Either String Program -- TODO: Should be tree parse file = case program file of |