diff options
Diffstat (limited to 'src/Test/VeriFuzz/Internal/Gen.hs')
| -rw-r--r-- | src/Test/VeriFuzz/Internal/Gen.hs | 17 |
1 files changed, 8 insertions, 9 deletions
diff --git a/src/Test/VeriFuzz/Internal/Gen.hs b/src/Test/VeriFuzz/Internal/Gen.hs index a8bd57a..2eba531 100644 --- a/src/Test/VeriFuzz/Internal/Gen.hs +++ b/src/Test/VeriFuzz/Internal/Gen.hs @@ -12,23 +12,22 @@ Internal helpers for generation. module Test.VeriFuzz.Internal.Gen where -import Data.Graph.Inductive (Graph, Node) -import qualified Data.Graph.Inductive as G -import qualified Data.Text as T -import Test.VeriFuzz.VerilogAST +import Data.Graph.Inductive (Graph, Node) +import qualified Data.Graph.Inductive as G +import qualified Data.Text as T fromNode :: Node -> T.Text -fromNode node = Identifier . T.pack $ "w" <> show node +fromNode node = T.pack $ "w" <> show node filterGr :: (Graph gr) => gr n e -> (Node -> Bool) -> [Node] filterGr graph f = filter f $ G.nodes graph -only :: (Graph gr) => (gr n e -> Node -> Int) -> (gr n e -> Node -> Int) -> gr n e -> Node -> [Node] -only fun1 fun2 n = fun1 graph n == 0 && fun2 graph n /= 0 +only :: (Graph gr) => gr n e -> (gr n e -> Node -> Int) -> (gr n e -> Node -> Int) -> Node -> Bool +only graph fun1 fun2 n = fun1 graph n == 0 && fun2 graph n /= 0 inputs :: (Graph gr) => gr n e -> [Node] -inputs graph = filterGr graph $ only G.indeg G.outdeg +inputs graph = filterGr graph $ only graph G.indeg G.outdeg outputs :: (Graph gr) => gr n e -> [Node] -outputs graph = filterGr graph $ only G.outdeg G.indeg +outputs graph = filterGr graph $ only graph G.outdeg G.indeg |