diff --git a/src/Search/Dot.hs b/src/Search/Dot.hs index 2cc3c01e3a3011f875cad95b4242d2863fb0e84f..96556303335260bffe165fa7ce43dfa7bd84fbbe 100644 --- a/src/Search/Dot.hs +++ b/src/Search/Dot.hs @@ -3,6 +3,7 @@ -- TODO Document module Search.Dot (graph2dot) where +import Compare.SSEGraph import Data.Monoid import qualified Data.Graph.Inductive as G @@ -36,4 +37,6 @@ dotEdges = foldMap formatEdge . G.labEdges . graph where formatEdge (from, to, label) = B.fromString (show from) <> " -> " <> B.fromString (show to) - <> " [label=\"" <> B.fromText label <> "\"];\n" + <> " [label=\"" <> B.fromText label <> "\", color=" <> color label <> "];\n" + color label = if isEpsilon $ SSEEdgeLabel label then "blue" else "black" +