From de249084d343a1503400112580a28fe5b038d4f6 Mon Sep 17 00:00:00 2001
From: Marvin Borner
Date: Wed, 23 Feb 2022 20:42:03 +0100
Subject: Cool trees

---
 src/Fun/Compiler.hs | 19 +++++++++++++++----
 1 file changed, 15 insertions(+), 4 deletions(-)

(limited to 'src/Fun/Compiler.hs')

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
-- 
cgit v1.2.3