litex/programmer.py

77 lines
2.2 KiB
Python

import subprocess
import os
class Programmer:
def __init__(self, platform_name, flash_proxy_dir):
self.platform_name = platform_name
self.flash_proxy_dir = flash_proxy_dir
def find_flash_proxy(self, name):
if self.flash_proxy_dir is not None:
search_dirs = [self.flash_proxy_dir]
else:
search_dirs = ["~/.mlabs", "/usr/local/share/mlabs", "/usr/share/mlabs"]
for d in search_dirs:
fulldir = os.path.abspath(os.path.expanduser(d))
fullname = os.path.join(fulldir, name)
if os.path.exists(fullname):
return fullname
raise OSError("Failed to find flash proxy bitstream")
def _run_urjtag(cmds):
with subprocess.Popen("jtag", stdin=subprocess.PIPE) as process:
process.stdin.write(cmds.encode("ASCII"))
process.communicate()
class UrJTAG(Programmer):
needs_bitreverse = True
def load_bitstream(self, bitstream_file):
cmds = """cable milkymist
detect
pld load {bitstream}
quit
""".format(bitstream=bitstream_file)
_run_urjtag(cmds)
def flash(self, address, data_file):
flash_proxy = self.find_flash_proxy("fjmem-" + self.platform_name + ".bit")
cmds = """cable milkymist
detect
pld load "{flash_proxy}"
initbus fjmem opcode=000010
frequency 6000000
detectflash 0
endian big
flashmem "{address}" "{data_file}" noverify
""".format(flash_proxy=flash_proxy, address=address, data_file=data_file)
_run_urjtag(cmds)
class XC3SProg(Programmer):
needs_bitreverse = False
def load_bitstream(self, bitstream_file):
subprocess.call(["xc3sprog", "-c", "papilio", bitstream_file])
def flash(self, address, data_file):
flash_proxy = self.find_flash_proxy("bscan_spi_lx9_papilio.bit")
subprocess.call(["xc3sprog", "-c", "papilio", "-I"+flash_proxy, "{}:w:0x{:x}:BIN".format(data_file, address)])
class USBBlaster(Programmer):
needs_bitreverse = False
def load_bitstream(self, bitstream_file, port=0):
usb_port = "[USB-"+str(port)+"]"
subprocess.call(["quartus_pgm", "-m", "jtag", "-c", "USB-Blaster"+usb_port, "-o", "p;"+bitstream_file])
def flash(self, address, data_file):
raise NotImplementedError
def create_programmer(platform_name, *args, **kwargs):
return {
"mixxeo": UrJTAG,
"m1": UrJTAG,
"papilio_pro": XC3SProg,
"de0nano": USBBlaster
}[platform_name](platform_name, *args, **kwargs)