diff options
Diffstat (limited to 'src/VeriFuzz/Internal/Gen.hs')
-rw-r--r-- | src/VeriFuzz/Internal/Gen.hs | 33 |
1 files changed, 33 insertions, 0 deletions
diff --git a/src/VeriFuzz/Internal/Gen.hs b/src/VeriFuzz/Internal/Gen.hs new file mode 100644 index 0000000..be275dd --- /dev/null +++ b/src/VeriFuzz/Internal/Gen.hs @@ -0,0 +1,33 @@ +{-| +Module : VeriFuzz.Internal.Gen +Description : Internal helpers for generation. +Copyright : (c) 2018-2019, Yann Herklotz Grave +License : BSD-3 +Maintainer : ymherklotz [at] gmail [dot] com +Stability : experimental +Portability : POSIX + +Internal helpers for generation. +-} + +module VeriFuzz.Internal.Gen where + +import Data.Graph.Inductive (Graph, Node) +import qualified Data.Graph.Inductive as G +import qualified Data.Text as T + +fromNode :: Int -> T.Text +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 -> (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 graph G.indeg G.outdeg + +outputs :: (Graph gr) => gr n e -> [Node] +outputs graph = filterGr graph $ only graph G.outdeg G.indeg |