From 7b35da52453bee336c83fdaa859fbcd3bf8f9440 Mon Sep 17 00:00:00 2001
From: Kevin Kim <kevindkim723@gmail.com>
Date: Wed, 18 Oct 2023 14:29:46 -0700
Subject: [PATCH] wrapper bug fix

---
 synthDC/scripts/wrapperGen.py | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/synthDC/scripts/wrapperGen.py b/synthDC/scripts/wrapperGen.py
index aacdb0634..e931d0a65 100755
--- a/synthDC/scripts/wrapperGen.py
+++ b/synthDC/scripts/wrapperGen.py
@@ -43,7 +43,7 @@ index=0
 
 # string copy logic
 for l in lines:
-    if l.find("module") == 0:
+    if l.lstrip().find("module") == 0:
         lineModuleStart = index
         moduleName = l.split()[1]
         writeBuf = True
@@ -51,7 +51,7 @@ for l in lines:
         continue
     if (writeBuf):
         buf += l
-    if l.find (");") == 0:
+    if l.lstrip().find (");") == 0:
         lineModuleEnd = index
         break
     index+=1
@@ -64,6 +64,7 @@ wrapperPath = f"{os.getenv('WALLY')}/synthDC/wrappers/{moduleName}wrapper.sv"
 
 # clear wrappers directory 
 os.system(f"rm {os.getenv('WALLY')}/synthDC/wrappers/*")
+os.system(f"mkdir {os.getenv('WALLY')}/synthDC/wrappers")
 
 fout = open(wrapperPath, "w")