aboutsummaryrefslogtreecommitdiffstats
path: root/src/VeriFuzz/Gen.hs
blob: 56d48bf7bf0b4f2853b832d93ecf7c9df4585692 (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
35
36
37
38
39
40
41
42
43
44
45
46
{-|
Module      : VeriFuzz.Gen
Description : Various useful generators.
Copyright   : (c) 2019, Yann Herklotz Grave
License     : GPL-3
Maintainer  : ymherklotz [at] gmail [dot] com
Stability   : experimental
Portability : POSIX

Various useful generators.
-}

module VeriFuzz.Gen where

import           Control.Lens
import qualified Data.Text       as T
import           Test.QuickCheck (Gen)
import qualified Test.QuickCheck as QC
import           VeriFuzz.AST
import           VeriFuzz.ASTGen
import           VeriFuzz.Mutate
import           VeriFuzz.Random

random :: [Identifier] -> (Expr -> ContAssign) -> Gen ModItem
random ctx fun = do
  expr <- QC.sized (exprWithContext ctx)
  return . ModCA $ fun expr

randomAssigns :: [Identifier] -> [Gen ModItem]
randomAssigns ids = random ids . ContAssign <$> ids

randomMod :: Gen ModDecl
randomMod = do
  let ids = Identifier . ("w"<>) . T.pack . show <$> [1..100]
  _ <- sequence $ randomAssigns ids
  return $ ModDecl "" [] [] []

fromGraph :: Gen ModDecl
fromGraph = do
  gr <- rDupsCirc <$> QC.resize 100 randomCircuit
  return $ initMod
    .   head
    $   nestUpTo 5 (generateAST gr)
    ^.. getVerilogSrc
    .   traverse
    .   getDescription