2 files changed, 51 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
diff --git a/src/VeriFuzz/Internal/Shared.hs b/src/VeriFuzz/Internal/Shared.hs
new file mode 100644
index 0000000..c7d2760
--- /dev/null
+++ b/src/VeriFuzz/Internal/Shared.hs
@@ -0,0 +1,18 @@
+{-|
+Module      : VeriFuzz.Internal.Shared
+Description : Shared high level code used in the other modules internally.
+Copyright   : (c) 2018-2019, Yann Herklotz Grave
+License     : BSD-3
+Maintainer  : ymherklotz [at] gmail [dot] com
+Stability   : experimental
+Portability : POSIX
+
+Shared high level code used in the other modules internally.
+-}
+
+module VeriFuzz.Internal.Shared where
+
+-- | Converts unsafe list functions in the Prelude to a safe version.
+safe :: ([a] -> b) -> [a] -> Maybe b
+safe _ [] = Nothing
+safe f l  = Just $ f l