123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115 |
- #
- # Builds custom upload command
- # 1) Run platformio as a subprocess to find a COM port
- # 2) Build the upload command
- # 3) Exit and let upload tool do the work
- #
- # This script runs between completion of the library/dependencies installation and compilation.
- #
- # Will continue on if a COM port isn't found so that the compilation can be done.
- #
-
- import sys
-
- import subprocess
-
-
- import platform
- current_OS = platform.system()
-
- from SCons.Script import DefaultEnvironment
-
- env = DefaultEnvironment()
-
- com_first = ''
- com_last = ''
- com_CDC = ''
- description_first = ''
- description_last = ''
- description_CDC = ''
-
- #
- # grab the first com port that pops up unless we find one we know for sure
- # is a CDC device
- #
- def get_com_port(com_search_text, descr_search_text, start):
-
- global com_first
- global com_last
- global com_CDC
- global description_first
- global description_last
- global description_CDC
-
-
- print '\nLooking for Serial Port\n'
-
- # stream output from subprocess and split it into lines
- pio_subprocess = subprocess.Popen(['platformio', 'device', 'list'], stdout=subprocess.PIPE, stderr=subprocess.STDOUT)
-
- looking_for_description = False
- for line in iter(pio_subprocess.stdout.readline, ''):
- if 0 <= line.find(com_search_text):
- looking_for_description = True
- com_last = line.replace('\n', '')
- if com_first == '':
- com_first = com_last
- if 0 <= line.find(descr_search_text) and looking_for_description:
- looking_for_description = False
- description_last = line[ start : ]
- if description_first == '':
- description_first = description_last
- if 0 <= description_last.find('CDC'):
- com_CDC = com_last
- description_CDC = description_last
-
- if com_CDC == '' and not(com_first == ''):
- com_CDC = com_first
- description_CDC = description_first
- elif com_CDC == '':
- com_CDC = 'COM_PORT_NOT_FOUND'
-
- if com_CDC == 'COM_PORT_NOT_FOUND':
- print com_CDC, '\n'
- else:
- print 'FOUND: ' ,com_CDC
- print 'DESCRIPTION: ', description_CDC , '\n'
-
- if current_OS == 'Windows':
-
- get_com_port('COM', 'Hardware ID:', 13)
-
- avrdude_conf_path = env.get("PIOHOME_DIR") + '\\packages\\toolchain-atmelavr\\etc\\avrdude.conf'
-
- source_path = env.get("PROJECTBUILD_DIR") + '\\' + env.get("PIOENV") + '\\firmware.hex'
-
- upload_string = 'avrdude -p usb1286 -c avr109 -P ' + com_CDC + ' -C ' + avrdude_conf_path + ' -U flash:w:' + source_path + ':i'
-
-
- if current_OS == 'Darwin': # MAC
-
- get_com_port('usbmodem', 'Description:', 13)
-
- avrdude_conf_path = env.get("PIOHOME_DIR") + '/packages/toolchain-atmelavr/etc/avrdude.conf'
-
- source_path = env.get("PROJECTBUILD_DIR") + '/' + env.get("PIOENV") + '/firmware.hex'
-
- upload_string = 'avrdude -p usb1286 -c avr109 -P ' + com_CDC + ' -U flash:w:' + source_path + ':i'
-
-
- if current_OS == 'Linux':
-
- get_com_port('/dev/tty', 'Description:', 13)
-
- avrdude_conf_path = env.get("PIOHOME_DIR") + '/packages/toolchain-atmelavr/etc/avrdude.conf'
-
- source_path = env.get("PROJECTBUILD_DIR") + '/' + env.get("PIOENV") + '/firmware.hex'
-
- upload_string = 'avrdude -p usb1286 -c avr109 -P ' + com_CDC + ' -U flash:w:' + source_path + ':i'
-
-
- env.Replace(
- UPLOADCMD = upload_string,
- MAXIMUM_RAM_SIZE = 8192,
- MAXIMUM_SIZE = 130048
- )
|