Subversion Repositories Tronxy-X3A-Marlin

#

#  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 subprocess

import os

import sys

from SCons.Script import DefaultEnvironment

import platform

current_OS = platform.system()

env = DefaultEnvironment()

build_type = os.environ.get("BUILD_TYPE", 'Not Set')

if not(build_type == 'upload' or build_type == 'traceback' or build_type == 'Not Set') :

  env.Replace(UPLOAD_PROTOCOL = 'teensy-gui')  # run normal Teensy2 scripts

else:

  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'

      while 0 <= com_CDC.find('\n'):

        com_CDC = com_CDC.replace('\n', '')

      while 0 <= com_CDC.find('\r'):

        com_CDC = com_CDC.replace('\r', '')

      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'

      avrdude_conf_path =  'buildroot\\share\\atom\\avrdude.conf'

      avrdude_exe_path =  'buildroot\\share\\atom\\avrdude_5.10.exe'

  #    source_path = env.get("PROJECTBUILD_DIR") + '\\' + env.get("PIOENV") + '\\firmware.hex'

      source_path =  '.pioenvs\\' + env.get("PIOENV") + '\\firmware.hex'

      upload_string = avrdude_exe_path + ' -p usb1286 -c avr109 -P ' + com_CDC + ' -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'

      avrdude_conf_path =  'buildroot/share/atom/avrdude_macOS.conf'

      avrdude_exe_path =  'buildroot/share/atom/avrdude_5.10_macOS'

#      source_path = env.get("PROJECTBUILD_DIR") + '/' + env.get("PIOENV") + '/firmware.hex'

      source_path = '.pioenvs/' + env.get("PIOENV") + '/firmware.hex'

#      upload_string = 'avrdude -p usb1286 -c avr109 -P ' + com_CDC + ' -U flash:w:' + source_path + ':i'

      upload_string = avrdude_exe_path + ' -p usb1286 -c avr109 -P ' + com_CDC + ' -C ' + avrdude_conf_path  + ' -U flash:w:' + source_path + ':i'

      print 'upload_string: ', upload_string

  if current_OS == 'Linux':

      get_com_port('/dev/tty', 'Description:', 13)

#      avrdude_conf_path =  env.get("PIOHOME_DIR") + '/packages/toolchain-atmelavr/etc/avrdude.conf'

      avrdude_conf_path =  'buildroot/share/atom/avrdude_linux.conf'

      avrdude_exe_path =  'buildroot/share/atom/avrdude_5.10_linux'

#      source_path = env.get("PROJECTBUILD_DIR") + '/' + env.get("PIOENV") + '/firmware.hex'

      source_path = '.pioenvs/' + env.get("PIOENV") + '/firmware.hex'

#      upload_string = 'avrdude -p usb1286 -c avr109 -P ' + com_CDC + ' -U flash:w:' + source_path + ':i'

      upload_string = avrdude_exe_path + ' -p usb1286 -c avr109 -P ' + com_CDC + ' -C ' + avrdude_conf_path  + ' -U flash:w:' + source_path + ':i'

  env.Replace(

       UPLOADCMD = upload_string,

       MAXIMUM_RAM_SIZE = 8192,

       MAXIMUM_SIZE = 130048

  )