mirror of
https://github.com/zephyrproject-rtos/zephyr
synced 2025-08-17 12:16:08 +00:00
ddce583ca2
I've had some requests to be able to use code in the runners package without having west installed. It turns out to be pretty easy to make this happen, as west is currently only used for west.log and some trivial helper methods: - To replace west log, use the standard logging module - Add an appropriate handler for each runner's logger in run_common.py which delegates to west.log, to keep output working as expected. Signed-off-by: Marti Bolivar <marti.bolivar@nordicsemi.no>
102 lines
4.0 KiB
Python
102 lines
4.0 KiB
Python
# Copyright (c) 2017 Linaro Limited.
|
|
#
|
|
# SPDX-License-Identifier: Apache-2.0
|
|
|
|
'''Runner for flashing ESP32 devices with esptool/espidf.'''
|
|
|
|
from os import path
|
|
|
|
from runners.core import ZephyrBinaryRunner, RunnerCaps
|
|
|
|
|
|
class Esp32BinaryRunner(ZephyrBinaryRunner):
|
|
'''Runner front-end for espidf.'''
|
|
|
|
def __init__(self, cfg, device, baud=921600, flash_size='detect',
|
|
flash_freq='40m', flash_mode='dio', espidf='espidf',
|
|
bootloader_bin=None, partition_table_bin=None):
|
|
super(Esp32BinaryRunner, self).__init__(cfg)
|
|
self.elf = cfg.elf_file
|
|
self.device = device
|
|
self.baud = baud
|
|
self.flash_size = flash_size
|
|
self.flash_freq = flash_freq
|
|
self.flash_mode = flash_mode
|
|
self.espidf = espidf
|
|
self.bootloader_bin = bootloader_bin
|
|
self.partition_table_bin = partition_table_bin
|
|
|
|
@classmethod
|
|
def name(cls):
|
|
return 'esp32'
|
|
|
|
@classmethod
|
|
def capabilities(cls):
|
|
return RunnerCaps(commands={'flash'})
|
|
|
|
@classmethod
|
|
def do_add_parser(cls, parser):
|
|
# Required
|
|
parser.add_argument('--esp-idf-path', required=True,
|
|
help='path to ESP-IDF')
|
|
|
|
# Optional
|
|
parser.add_argument('--esp-device', default='/dev/ttyUSB0',
|
|
help='serial port to flash, default /dev/ttyUSB0')
|
|
parser.add_argument('--esp-baud-rate', default='921600',
|
|
help='serial baud rate, default 921600')
|
|
parser.add_argument('--esp-flash-size', default='detect',
|
|
help='flash size, default "detect"')
|
|
parser.add_argument('--esp-flash-freq', default='40m',
|
|
help='flash frequency, default "40m"')
|
|
parser.add_argument('--esp-flash-mode', default='dio',
|
|
help='flash mode, default "dio"')
|
|
parser.add_argument(
|
|
'--esp-tool',
|
|
help='''if given, complete path to espidf. default is to search for
|
|
it in [ESP_IDF_PATH]/components/esptool_py/esptool/esptool.py''')
|
|
parser.add_argument('--esp-flash-bootloader',
|
|
help='Bootloader image to flash')
|
|
parser.add_argument('--esp-flash-partition_table',
|
|
help='Partition table to flash')
|
|
|
|
@classmethod
|
|
def create(cls, cfg, args):
|
|
if args.esp_tool:
|
|
espidf = args.esp_tool
|
|
else:
|
|
espidf = path.join(args.esp_idf_path, 'components', 'esptool_py',
|
|
'esptool', 'esptool.py')
|
|
|
|
return Esp32BinaryRunner(
|
|
cfg, args.esp_device, baud=args.esp_baud_rate,
|
|
flash_size=args.esp_flash_size, flash_freq=args.esp_flash_freq,
|
|
flash_mode=args.esp_flash_mode, espidf=espidf,
|
|
bootloader_bin=args.esp_flash_bootloader,
|
|
partition_table_bin=args.esp_flash_partition_table)
|
|
|
|
def do_run(self, command, **kwargs):
|
|
self.require(self.espidf)
|
|
bin_name = path.splitext(self.elf)[0] + path.extsep + 'bin'
|
|
cmd_convert = [self.espidf, '--chip', 'esp32', 'elf2image', self.elf]
|
|
cmd_flash = [self.espidf, '--chip', 'esp32', '--port', self.device,
|
|
'--baud', self.baud, '--before', 'default_reset',
|
|
'--after', 'hard_reset', 'write_flash', '-u',
|
|
'--flash_mode', self.flash_mode,
|
|
'--flash_freq', self.flash_freq,
|
|
'--flash_size', self.flash_size]
|
|
|
|
if self.bootloader_bin:
|
|
cmd_flash.extend(['0x1000', self.bootloader_bin])
|
|
cmd_flash.extend(['0x8000', self.partition_table_bin])
|
|
cmd_flash.extend(['0x10000', bin_name])
|
|
else:
|
|
cmd_flash.extend(['0x1000', bin_name])
|
|
|
|
self.logger.info("Converting ELF to BIN")
|
|
self.check_call(cmd_convert)
|
|
|
|
self.logger.info("Flashing ESP32 on {} ({}bps)".
|
|
format(self.device, self.baud))
|
|
self.check_call(cmd_flash)
|