From 42381c9f461da5c248c0504173012ce386010711 Mon Sep 17 00:00:00 2001
From: Yann Herklotz <ymherklotz@gmail.com>
Date: Sun, 23 Dec 2018 11:25:40 +0000
Subject: Start implementing the nesting functionality

---
 src/Test/VeriFuzz/Mutate.hs | 25 +++++++++++++++++++++++--
 1 file changed, 23 insertions(+), 2 deletions(-)

(limited to 'src')

diff --git a/src/Test/VeriFuzz/Mutate.hs b/src/Test/VeriFuzz/Mutate.hs
index 1c8f44a..dd391fb 100644
--- a/src/Test/VeriFuzz/Mutate.hs
+++ b/src/Test/VeriFuzz/Mutate.hs
@@ -13,7 +13,28 @@ more random patterns, such as nesting wires instead of creating new ones.
 
 module Test.VeriFuzz.Mutate where
 
+import           Control.Lens
+import           Data.Maybe                    (catMaybes)
+import           Test.VeriFuzz.Internal.Shared
 import           Test.VeriFuzz.VerilogAST
 
-nestId :: ModuleDecl -> Identifier -> ModuleDecl
-nestId mod id = (error "FIXME: nestId")
+-- | Return if the 'Identifier' is in a 'ModuleDecl'.
+inPort :: Identifier -> ModuleDecl -> Bool
+inPort id mod = any (\a -> a ^. portName == id) $ mod ^. modPorts
+
+-- | Find the last assignment of a specific wire/reg to an expression, and
+-- returns that expression.
+findAssign :: Identifier -> [ModuleItem] -> Maybe Expression
+findAssign id items =
+  safe last . catMaybes $ isAssign <$> items
+  where
+    isAssign (Assign ca)
+      | ca ^. contAssignNetLVal == id = Just $ ca ^. contAssignExpr
+      | otherwise = Nothing
+
+-- | Nest expressions for a specific 'Identifier'. If the 'Identifier' is not found,
+-- the AST is not changed.
+nestId :: Identifier -> ModuleDecl -> ModuleDecl
+nestId id mod
+  | not $ inPort id mod = mod
+  | otherwise = mod
-- 
cgit