diff options
| -rw-r--r-- | src/Test/VeriFuzz/Internal/Gen.hs | 34 | ||||
| -rw-r--r-- | verifuzz.cabal | 1 |
2 files changed, 35 insertions, 0 deletions
diff --git a/src/Test/VeriFuzz/Internal/Gen.hs b/src/Test/VeriFuzz/Internal/Gen.hs new file mode 100644 index 0000000..a8bd57a --- /dev/null +++ b/src/Test/VeriFuzz/Internal/Gen.hs @@ -0,0 +1,34 @@ +{-| +Module : Test.VeriFuzz.Internal.Gen +Description : Internal helpers for generation. +Copyright : (c) Yann Herklotz Grave 2018 +License : GPL-3 +Maintainer : ymherklotz@gmail.com +Stability : experimental +Portability : POSIX + +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 + +fromNode :: Node -> T.Text +fromNode node = Identifier . 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 + +inputs :: (Graph gr) => gr n e -> [Node] +inputs graph = filterGr graph $ only G.indeg G.outdeg + +outputs :: (Graph gr) => gr n e -> [Node] +outputs graph = filterGr graph $ only G.outdeg G.indeg diff --git a/verifuzz.cabal b/verifuzz.cabal index fcd5015..17bea7f 100644 --- a/verifuzz.cabal +++ b/verifuzz.cabal @@ -17,6 +17,7 @@ library hs-source-dirs: src default-language: Haskell2010 other-modules: Test.VeriFuzz.Internal.Shared + , Test.VeriFuzz.Internal.Gen exposed-modules: Test.VeriFuzz , Test.VeriFuzz.Circuit , Test.VeriFuzz.CodeGen |