|
1 | 1 | #!/usr/bin/env python3 |
2 | 2 |
|
3 | 3 | import sys |
| 4 | +import os |
4 | 5 | import time |
5 | 6 | import serial |
6 | 7 | import binascii |
| 8 | +import argparse |
7 | 9 |
|
8 | 10 |
|
9 | | -dev = '/dev/ttyAMA0' |
10 | | -baud = 921600 |
11 | 11 | boot_id = 'bbctrl ' |
12 | | -verbose = False |
13 | 12 |
|
14 | 13 |
|
15 | 14 | def crc16(data): |
@@ -70,41 +69,57 @@ def read_intel_hex(f): |
70 | 69 |
|
71 | 70 |
|
72 | 71 | def send(data): |
73 | | - if verbose: print('Sending:', data) |
| 72 | + if args.verbose: print('Sending:', data) |
74 | 73 | sp.write(bytes(data, 'utf8')) |
75 | 74 |
|
76 | 75 |
|
77 | 76 | def send_int(x, size): |
78 | | - if verbose: print('Sending: %d', x) |
| 77 | + if args.verbose: print('Sending: %d', x) |
79 | 78 | sp.write((x).to_bytes(size, byteorder = 'big')) |
80 | 79 |
|
81 | 80 |
|
82 | 81 | def recv(count): |
83 | 82 | data = sp.read(count).decode('utf8') |
84 | | - if count and verbose: print('Received:', data) |
| 83 | + if count and args.verbose: print('Received:', data) |
85 | 84 | return data |
86 | 85 |
|
87 | 86 |
|
88 | 87 | def recv_int(size): |
89 | 88 | x = int.from_bytes(sp.read(size), byteorder = 'big') |
90 | | - if verbose: print('Received:', x) |
| 89 | + if args.verbose: print('Received:', x) |
91 | 90 | return x |
92 | 91 |
|
| 92 | +# Parse arguments |
| 93 | +parser = argparse.ArgumentParser(description = 'Program AVR over serial port ' |
| 94 | + 'using the AVR109 bootloader protocol.') |
| 95 | +parser.add_argument('-b', '--baud', default = 921600, type = int, |
| 96 | + help = 'Set baud rate.') |
| 97 | +parser.add_argument('-t', '--timeout', default = 10, type = int, |
| 98 | + help = 'Timeout in seconds.') |
| 99 | +parser.add_argument('-d', '--device', default = '/dev/ttyAMA0', |
| 100 | + help = 'Serial device.') |
| 101 | +parser.add_argument('-r', '--reset', default = 27, type = int, |
| 102 | + help = 'Reset line GPIO.') |
| 103 | +parser.add_argument('-v', '--verbose', help = 'Enable verbose logging', |
| 104 | + action = 'count', default = 0) |
| 105 | +parser.add_argument('file', help = 'The Intel hex file to program.') |
| 106 | +args = parser.parse_args() |
93 | 107 |
|
94 | 108 | # Read firmware hex file |
95 | | -data = list(read_intel_hex(open(sys.argv[1], 'r'))) |
| 109 | +data = list(read_intel_hex(open(args.file, 'r'))) |
96 | 110 |
|
97 | 111 | # Open serial port |
98 | | -sp = serial.Serial(dev, baud, timeout = 10) |
| 112 | +sp = serial.Serial(args.device, args.baud, timeout = args.timeout) |
99 | 113 |
|
100 | 114 | # Reset AVR |
101 | | -import RPi.GPIO as gpio |
102 | | -gpio.setwarnings(False) |
103 | | -gpio.setmode(gpio.BCM) |
104 | | -gpio.setup(27, gpio.OUT) |
105 | | -gpio.output(27, 0) |
106 | | -gpio.output(27, 1) |
107 | | -gpio.setup(27, gpio.IN, pull_up_down = gpio.PUD_UP) |
| 115 | +reset_gpio_path = '/sys/class/gpio/gpio%s' % args.reset |
| 116 | +if not os.path.exists(reset_gpio_path): |
| 117 | + with open('/sys/class/gpio/export', 'w') as f: |
| 118 | + f.write(str(args.reset)) |
| 119 | + |
| 120 | +with open(reset_gpio_path + '/direction', 'w') as f: f.write('out') |
| 121 | +with open(reset_gpio_path + '/value', 'w') as f: f.write('0') |
| 122 | +with open(reset_gpio_path + '/value', 'w') as f: f.write('1') |
108 | 123 | time.sleep(0.1) |
109 | 124 |
|
110 | 125 | # Sync |
|
0 commit comments