From 30dbb0cd1ba86b3ec3d30ae9ee3d90b4096d8c4c Mon Sep 17 00:00:00 2001
From: htfab <mpw@htamas.net>
Date: Sun, 22 Sep 2024 23:58:24 +0200
Subject: [PATCH] feat: merge VHDL support branch

os.makedirs() line contributed by @gfcwfzkm
---
 project.py | 24 +++++++++++++++++++++++-
 1 file changed, 23 insertions(+), 1 deletion(-)

diff --git a/project.py b/project.py
index 1815616..7c6603a 100644
--- a/project.py
+++ b/project.py
@@ -125,6 +125,8 @@ class Project:
             if self.is_wokwi():
                 self.top_verilog_filename = self.sources[0]
             else:
+                if any(s.endswith(".vhdl") for s in self.sources):
+                    self.transpile_vhdl()
                 self.check_sources()
                 self.top_verilog_filename = self.find_top_verilog()
         else:
@@ -303,7 +305,7 @@ class Project:
         rgx_mod = re.compile(r"(?:^|[\s])module[\s]{1,}([\w]+)")
         top_verilog: typing.List[str] = []
         top_module = self.info.top_module
-        for src in self.info.source_files:
+        for src in self.sources:
             with open(os.path.join(self.src_dir, src)) as fh:
                 for line in fh.readlines():
                     for match in rgx_mod.finditer(line):
@@ -489,6 +491,26 @@ class Project:
                     outf.write(customizedContents)
                     outf.close()
 
+    def transpile_vhdl(self):
+        # logging.info("transpiling vhdl sources")
+        new_sources = []
+        for s in self.sources:
+            if s.endswith(".vhdl"):
+                # logging.info(f"running ghdl on {s}")
+                t = "generated/" + s.replace(".vhdl", ".v")
+                s_full = os.path.join(self.src_dir, s)
+                t_full = os.path.join(self.src_dir, t)
+                os.makedirs(os.path.dirname(t_full), exist_ok=True)
+                transpile_cmd = f"ghdl synth -Wno-binding --std=08 --out=verilog {s_full} -e > {t_full}"
+                p = subprocess.run(transpile_cmd, shell=True)
+                if p.returncode != 0:
+                    logging.error("transpile failed")
+                    exit(1)
+                new_sources.append(t)
+            else:
+                new_sources.append(s)
+        self.sources = new_sources
+
     def create_user_config(self):
         if self.is_wokwi():
             self.fetch_wokwi_files()