import subprocess import os class Programmer: def __init__(self, flash_proxy_basename=None): self.flash_proxy_basename = flash_proxy_basename self.flash_proxy_dirs = ["~/.mlabs", "/usr/local/share/mlabs", "/usr/share/mlabs"] def set_flash_proxy_dir(self, flash_proxy_dir): if flash_proxy_dir is not None: self.flash_proxy_dirs = [flash_proxy_dir] def find_flash_proxy(self): for d in self.flash_proxy_dirs: fulldir = os.path.abspath(os.path.expanduser(d)) fullname = os.path.join(fulldir, self.flash_proxy_basename) 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 needs_flash_proxy = 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() 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 needs_flash_proxy = True def __init__(self, cable, flash_proxy_basename=None): Programmer.__init__(self, flash_proxy_basename) self.cable = cable def load_bitstream(self, bitstream_file): subprocess.call(["xc3sprog", "-v", "-c", self.cable, bitstream_file]) def flash(self, address, data_file): flash_proxy = self.find_flash_proxy() subprocess.call(["xc3sprog", "-v", "-c", self.cable, "-I"+flash_proxy, "{}:w:0x{:x}:BIN".format(data_file, address)]) class USBBlaster(Programmer): needs_bitreverse = False needs_flash_proxy = 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