aboutsummaryrefslogtreecommitdiffstats
path: root/src/VeriFuzz/Internal/Gen.hs
blob: d821cd79c3d5301fc637606f160a3e708d927aa9 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
{-|
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