litex/litex_setup.py

144 lines
4.9 KiB
Python
Executable File

#!/usr/bin/env python3
import os
import sys
import subprocess
import shutil
from collections import OrderedDict
import urllib.request
current_path = os.path.dirname(os.path.realpath(__file__))
# name, (url, recursive clone, develop)
repos = [
# HDL
("migen", ("https://github.com/m-labs/", True, True)),
# LiteX SoC builder
("litex", ("https://github.com/enjoy-digital/", True, True)),
# LiteX cores ecosystem
("liteeth", ("https://github.com/enjoy-digital/", False, True)),
("litedram", ("https://github.com/enjoy-digital/", False, True)),
("litepcie", ("https://github.com/enjoy-digital/", False, True)),
("litesata", ("https://github.com/enjoy-digital/", False, True)),
("litesdcard", ("https://github.com/enjoy-digital/", False, True)),
("liteiclink", ("https://github.com/enjoy-digital/", False, True)),
("litevideo", ("https://github.com/enjoy-digital/", False, True)),
("litescope", ("https://github.com/enjoy-digital/", False, True)),
("litejesd204b", ("https://github.com/enjoy-digital/", False, True)),
("litespi", ("https://github.com/litex-hub/", False, True)),
# LiteX boards support
("litex-boards", ("https://github.com/litex-hub/", False, True)),
]
repos = OrderedDict(repos)
def sifive_riscv_download():
base_url = "https://static.dev.sifive.com/dev-tools/"
base_file = "riscv64-unknown-elf-gcc-8.3.0-2019.08.0-x86_64-"
is_windows = (
sys.platform.startswith('win') or sys.platform.startswith('cygwin'))
if is_windows:
end_file = 'w64-mingw32.zip'
elif sys.platform.startswith('linux'):
end_file = 'linux-ubuntu14.tar.gz'
elif sys.platform.startswith('darwin'):
end_file = 'apple-darwin.tar.gz'
else:
raise NotImplementedError(sys.platform)
fn = base_file + end_file
if not os.path.exists(fn):
url = base_url+fn
print("Downloading", url, "to", fn)
urllib.request.urlretrieve(url, fn)
else:
print("Using existing file", fn)
print("Extracting", fn)
shutil.unpack_archive(fn)
if "--user" in sys.argv[1:] and not is_windows:
print("Linking compiler into ~/.local/bin")
local_bin_dir = os.path.join(
base_file + end_file.split('.', 1)[0], "bin")
assert os.path.exists(local_bin_dir), local_bin_dir
user_bin_dir = os.path.abspath(os.path.expanduser("~/.local/bin"))
if not os.path.exists(user_bin_dir):
os.makedirs(user_bin_dir)
for f in os.listdir(local_bin_dir):
src = os.path.abspath(os.path.join(local_bin_dir, f))
dst = os.path.join(user_bin_dir, f)
assert os.path.exists(src), src
if os.path.exists(dst):
os.unlink(dst)
assert not os.path.exists(dst), dst
os.symlink(src, dst)
if os.environ.get('TRAVIS', '') == 'true':
# Ignore `ssl.SSLCertVerificationError` on CI.
import ssl
ssl._create_default_https_context = ssl._create_unverified_context
if len(sys.argv) < 2:
print("Available commands:")
print("- init")
print("- install (add --user to install to user directory)")
print("- update")
print("- gcc")
exit()
if "init" in sys.argv[1:]:
os.chdir(os.path.join(current_path))
for name in repos.keys():
url, need_recursive, need_develop = repos[name]
# clone repo (recursive if needed)
print("[cloning " + name + "]...")
full_url = url + name
opts = "--recursive" if need_recursive else ""
subprocess.check_call(
"git clone " + full_url + " " + opts,
shell=True)
if "install" in sys.argv[1:]:
for name in repos.keys():
url, need_recursive, need_develop = repos[name]
# develop if needed
print("[installing " + name + "]...")
if need_develop:
os.chdir(os.path.join(current_path, name))
if "--user" in sys.argv[1:]:
subprocess.check_call(
"python3 setup.py develop --user",
shell=True)
else:
subprocess.check_call(
"python3 setup.py develop",
shell=True)
os.chdir(os.path.join(current_path))
if "gcc" in sys.argv[1:]:
sifive_riscv_download()
if "update" in sys.argv[1:]:
for name in repos.keys():
# update
print("[updating " + name + "]...")
os.chdir(os.path.join(current_path, name))
subprocess.check_call(
"git pull",
shell=True)
os.chdir(os.path.join(current_path))
if "--user" in sys.argv[1:]:
if ".local/bin" not in os.environ.get("PATH", ""):
print("Make sure that ~/.local/bin is in your PATH")
print("export PATH=$PATH:~/.local/bin")
elif "gcc" in sys.argv[1:]:
print("Make sure that the downloaded RISC-V compiler is in your $PATH.")
print("export PATH=$PATH:$(echo $PWD/riscv64-*/bin/)")