-
Notifications
You must be signed in to change notification settings - Fork 4
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
- Loading branch information
Showing
4 changed files
with
147 additions
and
44 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,109 @@ | ||
| #!/usr/bin/python3 | ||
|
|
||
| import sh | ||
| import time | ||
| import sys | ||
| import argparse | ||
| from pyftdi.ftdi import Ftdi | ||
|
|
||
| # CBUS0 - BOOT0 | ||
| # CBUS1 - RST | ||
|
|
||
| class FirmwareFlasher: | ||
| def __init__(self, binary_file, port): | ||
|
|
||
| # ftdi.show_devices() | ||
| # self.ftdi = Ftdi.create_from_url('ftdi://ftdi:ft-x:DK0AM0V0/1') | ||
| self.device = 'ftdi://ftdi:ft-x:/1' | ||
| self.ftdi = Ftdi() | ||
|
|
||
| self.binary_file = binary_file | ||
| self.max_approach_no = 3 | ||
| self.port = port | ||
|
|
||
| def enter_bootloader_mode(self): | ||
|
|
||
| self.ftdi.open_from_url(url=self.device) | ||
| self.ftdi.set_cbus_direction(0b11,0b11) # set CBUS0 and CBUS1 to output | ||
| time.sleep(0.1) | ||
| self.ftdi.set_cbus_gpio(0b11) #set CBUS0 to 1 and RST to 1 | ||
| time.sleep(0.1) | ||
| self.ftdi.set_cbus_gpio(0b01) #set CBUS0 to 1 and RST to 0 | ||
| time.sleep(0.1) | ||
| # self.ftdi.set_cbus_direction(0b11,0b00) # set CBUS0 and CBUS1 to input | ||
| time.sleep(0.1) | ||
| self.ftdi.close() | ||
|
|
||
| def exit_bootloader_mode(self): | ||
|
|
||
| self.ftdi.open_from_url(url=self.device) | ||
| self.ftdi.set_cbus_direction(0b11,0b11) # set CBUS0 and CBUS1 to output | ||
| time.sleep(0.1) | ||
| self.ftdi.set_cbus_gpio(0b10) #set CBUS0 to 1 and RST to 1 | ||
| time.sleep(0.1) | ||
| self.ftdi.set_cbus_gpio(0b00) #set CBUS0 to 1 and RST to 0 | ||
| time.sleep(0.1) | ||
| # self.ftdi.set_cbus_direction(0b11,0b00) # set CBUS0 and CBUS1 to input | ||
| time.sleep(0.1) | ||
| self.ftdi.close() | ||
|
|
||
| def try_flash_operation(self, operation_name, flash_command, flash_args): | ||
| for i in range(self.max_approach_no): | ||
| try: | ||
| self.enter_bootloader_mode() | ||
| sh.usbreset("0403:6015") | ||
| flash_command(self.port, *flash_args, _out=sys.stdout) | ||
| self.exit_bootloader_mode() | ||
| time.sleep(0.2) | ||
| break | ||
| except Exception as e: | ||
| print(f"{operation_name} error! Trying again.") | ||
| print(f"Error: {e}") | ||
| print("---------------------------------------") | ||
| else: | ||
| print(f"WARNING! {operation_name} went wrong.") | ||
|
|
||
| def flash_firmware(self): | ||
| # Disable the flash write-protection | ||
| self.try_flash_operation("Write-UnProtection", sh.stm32flash, ["-u"]) | ||
|
|
||
| # Disable the flash read-protection | ||
| self.try_flash_operation("Read-UnProtection", sh.stm32flash, ["-k"]) | ||
|
|
||
| # Flashing the firmware | ||
| # /usr/bin/stm32flash /dev/ttyUSB0 -v -w /root/firmware.bin -b 115200 | ||
| flash_args = ["-v", "-w", self.binary_file, "-b", "115200"] | ||
| self.try_flash_operation("Flashing", sh.stm32flash, flash_args) | ||
|
|
||
|
|
||
| sh.usbreset("0403:6015") | ||
|
|
||
|
|
||
| def main(): | ||
|
|
||
| parser = argparse.ArgumentParser( | ||
| description='Flashing the firmware on STM32 microcontroller in ROSbot XL') | ||
|
|
||
| parser.add_argument( | ||
| "-f", | ||
| "--file", | ||
| nargs='?', | ||
| default="/root/firmware.bin", | ||
| help="Path to a firmware file. Default: /root/firmware.bin") | ||
| parser.add_argument( | ||
| "-p", | ||
| "--port", | ||
| nargs='?', | ||
| default="/dev/ttyUSB0", | ||
| help="Path to serial connection. Default: /dev/ttyUSB0") | ||
|
|
||
| binary_file = parser.parse_args().file | ||
| port = parser.parse_args().port | ||
|
|
||
| flasher = FirmwareFlasher(binary_file, port) | ||
| flasher.flash_firmware() | ||
| print("Done.") | ||
|
|
||
|
|
||
| if __name__ == "__main__": | ||
| main() |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters