Added DFU support to Butterstick

litex_boards/platforms/gsd_butterstick.py

@@ -8,6 +8,7 @@
 from litex.build.generic_platform import *
 from litex.build.lattice import LatticePlatform
 from litex.build.lattice.programmer import OpenOCDJTAGProgrammer
+from litex.build.dfu import DFUProg
 
 # IOs ----------------------------------------------------------------------------------------------
 
@@ -194,8 +195,13 @@ class Platform(LatticePlatform):
         connectors = {"1.0": _connectors_r1_0}[revision]
         LatticePlatform.__init__(self, f"LFE5UM5G-{device}-8BG381C", io, connectors, toolchain=toolchain, **kwargs)
 
-    def create_programmer(self):
+    def create_programmer(self, load):
+        if load == "jtag":
             return OpenOCDJTAGProgrammer("openocd_butterstick.cfg")
+        elif load == "dfu":
+            return DFUProg(vid="1209", pid="5af1", alt=0)
+        else:
+            print("Could not program board. "+load+" is not a valid argument. Please use 'jtag' or 'dfu'.")
 
     def do_finalize(self, fragment):
         LatticePlatform.do_finalize(self, fragment)

litex_boards/targets/gsd_butterstick.py

@@ -155,7 +155,7 @@ def main():
     parser = LiteXSoCArgumentParser(description="LiteX SoC on ButterStick")
     target_group = parser.add_argument_group(title="Target options")
     target_group.add_argument("--build",           action="store_true",    help="Build design.")
-    target_group.add_argument("--load",            action="store_true",    help="Load bitstream.")
+    target_group.add_argument("--load",            default="jtag",         help="Load bitstream (jtag or dfu).")
     target_group.add_argument("--toolchain",       default="trellis",      help="FPGA toolchain (trellis or diamond).")
     target_group.add_argument("--sys-clk-freq",    default=75e6,           help="System clock frequency.")
     target_group.add_argument("--revision",        default="1.0",          help="Board Revision (1.0).")
@@ -201,7 +201,7 @@ def main():
         builder.build(**builder_kargs)
 
     if args.load:
-        prog = soc.platform.create_programmer()
+        prog = soc.platform.create_programmer(args.load)
         prog.load_bitstream(builder.get_bitstream_filename(mode="sram"))
 
 if __name__ == "__main__":