141 lines
4.5 KiB
Python
141 lines
4.5 KiB
Python
|
#!/usr/bin/python
|
||
|
|
|
||
|
|
import sys, time, os
|
||
|
|
import subprocess
|
||
|
|
sys.path.append("./tools")
|
||
|
|
|
||
|
|
def ERROR(msg=''):
|
||
|
|
enablePrint()
|
||
|
|
sys.stdout.write("\033[1;31m")
|
||
|
|
print('ERROR ' + msg)
|
||
|
|
sys.stdout.write("\033[0;0m")
|
||
|
|
if not verbose: blockPrint()
|
||
|
|
sys.exit()
|
||
|
|
|
||
|
|
def OK(msg=''):
|
||
|
|
enablePrint()
|
||
|
|
sys.stdout.write("\033[1;32m")
|
||
|
|
print('OK ' + msg)
|
||
|
|
sys.stdout.write("\033[0;0m")
|
||
|
|
if not verbose: blockPrint()
|
||
|
|
|
||
|
|
def oneLine(msg):
|
||
|
|
enablePrint()
|
||
|
|
sys.stdout.write(msg)
|
||
|
|
sys.stdout.flush()
|
||
|
|
if not verbose: blockPrint()
|
||
|
|
|
||
|
|
def blockPrint():
|
||
|
|
sys.stdout = open(os.devnull, 'w')
|
||
|
|
|
||
|
|
def enablePrint():
|
||
|
|
sys.stdout = sys.__stdout__
|
||
|
|
|
||
|
|
if '-h' in sys.argv or '--help' in sys.argv or '-help' in sys.argv or len(sys.argv) < 2 or (not 'build' in sys.argv and not 'flash' in sys.argv and not 'boot' in sys.argv):
|
||
|
|
print('USAGE:\n\n\tpython make.py [options] action[s] target[s]')
|
||
|
|
print('\nOptions:')
|
||
|
|
print('\t-v: verbose')
|
||
|
|
print('\t-k: keep configuration')
|
||
|
|
print('\t-p port: specify a port instead of scanning')
|
||
|
|
print('\t-f: force flashing even if no SCK is found, (port must be specified)')
|
||
|
|
print('\nActions:\n\tboot: flash SAM bootloader (Extra hardware is needed)\n\tbuild: build firmware\n\tflash: upload compiled code')
|
||
|
|
print('\nTargets:\n\tsam: SAMD21 chip\n\tesp: ESP8266 (WiFi) chip')
|
||
|
|
sys.exit()
|
||
|
|
|
||
|
|
verbose = False
|
||
|
|
blockPrint()
|
||
|
|
if '-v' in sys.argv:
|
||
|
|
verbose = True
|
||
|
|
enablePrint()
|
||
|
|
|
||
|
|
import sck
|
||
|
|
kit = sck.sck()
|
||
|
|
|
||
|
|
if 'flash' in sys.argv:
|
||
|
|
force = False
|
||
|
|
port = None
|
||
|
|
if '-p' in sys.argv:
|
||
|
|
port = sys.argv[sys.argv.index('-p')+1]
|
||
|
|
if '-f' in sys.argv: force = True
|
||
|
|
elif '-f' in sys.argv: ERROR('Port must be specified to force flashing'); sys.exit()
|
||
|
|
if not kit.begin(port=port, force=force): sys.exit()
|
||
|
|
if '-k' in sys.argv:
|
||
|
|
kit.getConfig()
|
||
|
|
if kit.mode == 'network':
|
||
|
|
print('Current mode: ' + kit.mode + ', Wifi: ' + kit.wifi_ssid + ' - ' + kit.wifi_pass + ', Token: ' + kit.token)
|
||
|
|
elif kit.mode == 'sdcard':
|
||
|
|
ww = ""
|
||
|
|
if len(kit.wifi_ssid) > 0: ww = ', Wifi: ' + kit.wifi_ssid + ' - ' + kit.wifi_pass
|
||
|
|
print('Current mode: ' + kit.mode + ww)
|
||
|
|
else:
|
||
|
|
print('Kit is not configured')
|
||
|
|
|
||
|
|
if 'erase' in sys.argv:
|
||
|
|
print('Erasing ESP flash...')
|
||
|
|
kit.eraseESP()
|
||
|
|
|
||
|
|
if 'boot' in sys.argv:
|
||
|
|
oneLine('Flashing SAM bootloader...')
|
||
|
|
os.chdir('bootloader/uf2-samdx1')
|
||
|
|
make_bootloader = subprocess.call(['make'], stdout=sys.stdout, stderr=subprocess.STDOUT)
|
||
|
|
if make_bootloader == 0:
|
||
|
|
openocd = subprocess.call(['openocd', '-f', 'openocd_sck2_SAMICE.cfg'], stdout=sys.stdout, stderr=subprocess.STDOUT)
|
||
|
|
if openocd != 0:
|
||
|
|
ERROR("Failed flashing SCK bootloader!!!\nDo you have openocd executable in your path???");
|
||
|
|
else:
|
||
|
|
OK()
|
||
|
|
else:
|
||
|
|
ERROR('Failed building SCK bootloader!!!')
|
||
|
|
os.chdir('../..')
|
||
|
|
|
||
|
|
if 'build' in sys.argv:
|
||
|
|
if 'sam' in sys.argv or 'all' in sys.argv:
|
||
|
|
oneLine('Building SAM firmware... ')
|
||
|
|
if kit.buildSAM(sys.stdout): OK()
|
||
|
|
else: ERROR()
|
||
|
|
|
||
|
|
if 'esp' in sys.argv or 'all' in sys.argv:
|
||
|
|
oneLine('Building ESP firmware... ')
|
||
|
|
if kit.buildESP(sys.stdout): OK()
|
||
|
|
else: ERROR()
|
||
|
|
|
||
|
|
if 'flash' in sys.argv:
|
||
|
|
if not 'build' in sys.argv:
|
||
|
|
if not verbose: enablePrint()
|
||
|
|
print('\nWARNING: No build instruction received, trying to flash previous built firmware...')
|
||
|
|
if not verbose: blockPrint()
|
||
|
|
|
||
|
|
if 'sam' in sys.argv or 'all' in sys.argv:
|
||
|
|
time.sleep(1)
|
||
|
|
oneLine('Flashing SAM firmware...')
|
||
|
|
if kit.flashSAM(sys.stdout): OK()
|
||
|
|
else: ERROR()
|
||
|
|
|
||
|
|
if 'esp' in sys.argv or 'all' in sys.argv:
|
||
|
|
time.sleep(0.5)
|
||
|
|
oneLine('Flashing ESP firmware')
|
||
|
|
for i in range(4):
|
||
|
|
mySpeed = 115200 / pow(2, i)
|
||
|
|
oneLine(' at ' + str(mySpeed) + '...')
|
||
|
|
time.sleep(1)
|
||
|
|
if kit.flashESP(mySpeed, sys.stdout):
|
||
|
|
OK()
|
||
|
|
break;
|
||
|
|
else:
|
||
|
|
if i == 3: ERROR()
|
||
|
|
else:
|
||
|
|
time.sleep(3)
|
||
|
|
if i == 0: oneLine(' Retry')
|
||
|
|
oneLine(' ' + str(i+1))
|
||
|
|
|
||
|
|
if '-k' in sys.argv and len(kit.mode) > 0:
|
||
|
|
oneLine("Reconfiguring kit...")
|
||
|
|
time.sleep(1)
|
||
|
|
|
||
|
|
if 'network' in kit.mode:
|
||
|
|
if kit.netConfig(): OK()
|
||
|
|
else: ERROR()
|
||
|
|
elif 'sdcard' in kit.mode:
|
||
|
|
if kit.sdConfig(): OK()
|
||
|
|
else: ERROR()
|