flash.py

#!/usr/bin/python3

# Copyright 2023 mjbots Robotic Systems, LLC.  info@mjbots.com
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
#     http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.

import argparse
import platform
import subprocess
import sys
import tempfile


BINPREFIX = '' if platform.machine().startswith('arm') else 'arm-none-eabi-'

OBJCOPY = BINPREFIX + 'objcopy'
OPENOCD = 'openocd -f interface/stlink.cfg -f target/stm32g4x.cfg '


def main():
    parser = argparse.ArgumentParser()
    parser.add_argument('--erase', action='store_true')
    parser.add_argument('elffile', nargs='?',
                        default='bazel-out/stm32g4-opt/bin/fw/pi3_hat.elf')

    args = parser.parse_args()

    tmpdir = tempfile.TemporaryDirectory()

    subprocess.check_call(
        f'{OBJCOPY} -Obinary {args.elffile} {tmpdir.name}/out.bin',
        shell=True)
    subprocess.check_call(
        f'{OPENOCD} -c "init" ' +
        f'-c "reset_config none separate; ' +
        f' halt; ' +
        (f' stm32l4x mass_erase 0; ' if args.erase else '') +
        f' program {tmpdir.name}/out.bin verify 0x8000000 verify ' +
        f' reset exit 0x8000000"',
        shell=True)


if __name__ == '__main__':
    main()