mirror of
https://github.com/TinyTapeout/tt-support-tools.git
synced 2026-11-03 12:46:08 +01:00
116 lines
3.7 KiB
Python
116 lines
3.7 KiB
Python
import yaml
|
|
import logging
|
|
import requests
|
|
import os
|
|
import subprocess
|
|
|
|
|
|
def load_yaml(args):
|
|
with open(args.yaml, "r") as stream:
|
|
return (yaml.safe_load(stream))
|
|
|
|
|
|
def print_wokwi_id(yaml):
|
|
print(yaml['project']['wokwi_id'])
|
|
|
|
|
|
def create_user_config(yaml):
|
|
logging.info("creating include file")
|
|
sources = get_project_source(yaml)
|
|
top_module = get_top_module(yaml)
|
|
|
|
if top_module == 'top':
|
|
logging.error("top module cannot be called top - prepend your repo name to make it unique")
|
|
exit(1)
|
|
|
|
filename = 'user_config.tcl'
|
|
with open(os.path.join('src', filename), 'w') as fh:
|
|
fh.write("set ::env(DESIGN_NAME) {}\n".format(top_module))
|
|
fh.write('set ::env(VERILOG_FILES) "\\\n')
|
|
for line, source in enumerate(sources):
|
|
fh.write(" $::env(DESIGN_DIR)/" + source)
|
|
if line != len(sources) - 1:
|
|
fh.write(' \\\n')
|
|
fh.write('"\n')
|
|
|
|
|
|
def fetch_file(url, filename):
|
|
logging.info("trying to download {}".format(url))
|
|
r = requests.get(url)
|
|
if r.status_code != 200:
|
|
logging.warning("couldn't download {}".format(url))
|
|
exit(1)
|
|
|
|
with open(filename, 'wb') as fh:
|
|
logging.info("written to {}".format(filename))
|
|
fh.write(r.content)
|
|
|
|
|
|
def get_project_source(yaml):
|
|
# wokwi_id must be an int or 0
|
|
try:
|
|
wokwi_id = int(yaml['project']['wokwi_id'])
|
|
except ValueError:
|
|
logging.error("wokwi id must be an integer")
|
|
exit(1)
|
|
|
|
# it's a wokwi project
|
|
if wokwi_id != 0:
|
|
url = "https://wokwi.com/api/projects/{}/verilog".format(wokwi_id)
|
|
src_file = "user_module_{}.v".format(wokwi_id)
|
|
fetch_file(url, os.path.join("src", src_file))
|
|
|
|
# also fetch the wokwi diagram
|
|
url = "https://wokwi.com/api/projects/{}/diagram.json".format(wokwi_id)
|
|
diagram_file = "wokwi_diagram.json"
|
|
fetch_file(url, os.path.join("src", diagram_file))
|
|
|
|
return [src_file, 'cells.v']
|
|
|
|
# else it's HDL, so check source files
|
|
else:
|
|
if 'source_files' not in yaml['project']:
|
|
logging.error("source files must be provided if wokwi_id is set to 0")
|
|
exit(1)
|
|
|
|
source_files = yaml['project']['source_files']
|
|
if source_files is None:
|
|
logging.error("must be more than 1 source file")
|
|
exit(1)
|
|
|
|
if len(source_files) == 0:
|
|
logging.error("must be more than 1 source file")
|
|
exit(1)
|
|
|
|
if 'top_module' not in yaml['project']:
|
|
logging.error("must provide a top module name")
|
|
exit(1)
|
|
|
|
for filename in source_files:
|
|
if '*' in filename:
|
|
logging.error("* not allowed, please specify each file")
|
|
exit(1)
|
|
if not os.path.exists(os.path.join('src', filename)):
|
|
logging.error(f"{filename} doesn't exist in the repo")
|
|
exit(1)
|
|
|
|
return source_files
|
|
|
|
|
|
def get_top_module(yaml):
|
|
wokwi_id = int(yaml['project']['wokwi_id'])
|
|
if wokwi_id != 0:
|
|
return "user_module_{}".format(wokwi_id)
|
|
else:
|
|
return yaml['project']['top_module']
|
|
|
|
|
|
def harden():
|
|
# requires PDK, PDK_ROOT, OPENLANE_ROOT & OPENLANE_IMAGE_NAME to be set in local environment
|
|
harden_cmd = 'docker run --rm -v $OPENLANE_ROOT:/openlane -v $PDK_ROOT:$PDK_ROOT -v $(pwd):/work -e PDK=$PDK -e PDK_ROOT=$PDK_ROOT -u $(id -u $USER):$(id -g $USER) $OPENLANE_IMAGE_NAME /bin/bash -c "./flow.tcl -overwrite -design /work/src -run_path /work/runs -tag wokwi"'
|
|
env = os.environ.copy()
|
|
p = subprocess.run(harden_cmd, shell=True, env=env)
|
|
if p.returncode != 0:
|
|
logging.error("harden failed")
|
|
exit(1)
|