diff --git a/src/Search/Search.hs b/src/Search/Search.hs index 9fcbf3af0c97a2bf2cfa888e3ead2b371642c17d..9b1e012ebafc1ec6e27f39c4e3c7efa413f54156 100644 --- a/src/Search/Search.hs +++ b/src/Search/Search.hs @@ -16,6 +16,7 @@ import Control.Monad import System.Exit import System.Process import System.IO +import Data.Foldable -- | (Syscall, Statehash) type Vertex = (Text, Text) @@ -156,8 +157,9 @@ graph2Dot graph = T.drop (T.length "OSEKOS_") txt dotNodes :: Graph -> Text -dotNodes = foldMap formatNode . M.keys +dotNodes = foldMap formatNode . allNotes where + allNotes m = M.keysSet m `S.union` (fold m) formatNode (call, state) = nodeId (call,state) <> " [label=\"" <> "State: " <> state <> "\\n"