Add VHDL support via GHDL

Author: nobodywasishere

sphinxcontrib_hdl_diagrams/__init__.py

@@ -165,7 +165,8 @@ class HDLDiagram(Directive):
         "hdl_diagram_output_format": ["svg", "png"],
         "hdl_diagram_skin": ["default"],  # or path
         "hdl_diagram_yosys_script": ["default"],  # or path
-        "hdl_diagram_yosys": ["yowasp", "system"]  # or path
+        "hdl_diagram_yosys": ["yowasp", "system"],  # or path
+        "hdl_diagram_ghdl": ["module", "system"]  # or path
     }
 
     def run(self):
@@ -237,18 +238,20 @@ def run(self):
         return [node]
 
 
-def run_yosys(src, cmd, yosys='yowasp'):
+def run_yosys(src, cmd, yosys='yowasp', opt=''):
     if yosys == 'yowasp':
         import yowasp_yosys
         ycmd = ["-q", "-p", "{}".format(cmd), src]
+        if opt != '':
+            ycmd.insert(0, opt)
         print("Running YoWASP yosys: {}".format(ycmd))
         yowasp_yosys.run_yosys(ycmd)
     elif yosys == 'system':
-        ycmd = "yosys -p '{cmd}' {src}".format(src=src, cmd=cmd)
+        ycmd = "yosys {opt} -p '{cmd}' {src}".format(opt=opt, src=src, cmd=cmd)
         print("Running yosys: {}".format(ycmd))
         subprocess.check_output(ycmd, shell=True)
     else:
-        ycmd = "{yosys} -p '{cmd}' {src}".format(yosys=yosys, src=src, cmd=cmd)
+        ycmd = "{yosys} {opt} -p '{cmd}' {src}".format(opt=opt, yosys=yosys, src=src, cmd=cmd)
         print("Running yosys: {}".format(ycmd))
         subprocess.check_output(ycmd, shell=True)
 
@@ -372,6 +375,29 @@ def nmigen_to_rtlil(fname, oname):
     cmd = "{python} {script} > {output}".format(python=sys.executable, script=fname, output=oname)
     subprocess.run(cmd, shell=True, check=True)
 
+def vhdl_to_verilog(fname, oname, module, ghdl, yosys):
+    assert os.path.exists(fname)
+
+    if ghdl == "module":
+        yosys_opt = "-m ghdl "
+    elif ghdl == "prebuilt":
+        yosys_opt = ""
+    elif os.path.exists(ghdl):
+        yosys_opt = "-m '{}'".format(ghdl)
+    else:
+        raise HDLDiagramError("hdl_diagram_ghdl can only be \"module\", \"prebuilt\", or "
+                              "a path to a ghdl-yosys-plugin shared library, not '{}'".format(ghdl))
+
+    assert yosys != "yowasp", HDLDiagramError('Cannot use YoWASP for VHDL - GHDL is not compatible with YoWASP')
+
+    output_dir = os.path.dirname(oname)
+    os.makedirs(output_dir, exist_ok=True)
+    cmd = "ghdl {input} -e {module}; write_verilog {output}".format(
+        module=module,
+        input=fname,
+        output=oname
+    )
+    run_yosys('', cmd, yosys, opt=yosys_opt)
 
 def render_diagram(self, code, options, format, skin, yosys_script):
     # type: (nodes.NodeVisitor, unicode, Dict, unicode, unicode) -> Tuple[unicode, unicode]
@@ -384,13 +410,20 @@ def render_diagram(self, code, options, format, skin, yosys_script):
     relfn = posixpath.join(self.builder.imgpath, fname)
     outfn = path.join(self.builder.outdir, self.builder.imagedir, fname)
 
+    yosys = self.builder.config.hdl_diagram_yosys
+
     if source_ext == '.py':
         module = 'top'
         ilfn = path.join(self.builder.outdir, self.builder.imagedir, options['outname'] + '.il')
         nmigen_to_rtlil(source_path, ilfn)
         source_path = ilfn
     elif source_ext == '.il' or source_ext == '.v':
         module = options['module']
+    elif source_ext == '.vhd' or source_ext == '.vhdl':
+        module = options['module']
+        ilfn = path.join(self.builder.outdir, self.builder.imagedir, options['outname'] + '.v')
+        vhdl_to_verilog(source_path, ilfn, module, self.builder.config.hdl_diagram_ghdl, yosys)
+        source_path = ilfn
     else:
         raise HDLDiagramError("hdl_diagram_code file extension must be one of '.v', "
                               "'.il', or '.py', but is %r" % source_ext)
@@ -404,7 +437,6 @@ def render_diagram(self, code, options, format, skin, yosys_script):
     yosys_script = options['yosys_script'] if options['yosys_script'] is not None else yosys_script
     skin = options['skin'] if options['skin'] is not None else skin
 
-    yosys = self.builder.config.hdl_diagram_yosys
     yosys_options = HDLDiagram.global_variable_options["hdl_diagram_yosys"]
     if yosys not in yosys_options and not os.path.exists(yosys):
         raise HDLDiagramError("Yosys not found!")
@@ -568,4 +600,5 @@ def setup(app):
     app.add_config_value('hdl_diagram_skin', 'default', 'html')
     app.add_config_value('hdl_diagram_yosys_script', 'default', 'html')
     app.add_config_value('hdl_diagram_yosys', 'yowasp', 'html')
+    app.add_config_value('hdl_diagram_ghdl', 'module', 'html')
     return {'version': '1.0', 'parallel_read_safe': True}