the other jimmy / mbed-sdk-tools

Fork of mbed-sdk-tools by mbed official

Committer:
screamer
Date:
Sat Jul 16 22:51:17 2016 +0100
Revision:
23:fbae331171fa
Parent:
22:9e85236d8716
Child:
24:25bff2709c20
Ported fixes from mbedmicro/mbed

Who changed what in which revision?

UserRevisionLine numberNew contents of line
screamer 0:66f3b5499f7f 1 import sys
screamer 0:66f3b5499f7f 2 from os.path import join, abspath, dirname, exists, basename
screamer 0:66f3b5499f7f 3 ROOT = abspath(join(dirname(__file__), ".."))
screamer 0:66f3b5499f7f 4 sys.path.insert(0, ROOT)
screamer 0:66f3b5499f7f 5
screamer 0:66f3b5499f7f 6 from shutil import move, rmtree
screamer 22:9e85236d8716 7 from argparse import ArgumentParser
screamer 0:66f3b5499f7f 8 from os import path
screamer 0:66f3b5499f7f 9
screamer 23:fbae331171fa 10 from tools.paths import EXPORT_DIR
screamer 23:fbae331171fa 11 from tools.export import export, EXPORTERS, mcu_ide_matrix
screamer 23:fbae331171fa 12 from tools.tests import TESTS, TEST_MAP
screamer 22:9e85236d8716 13 from tools.tests import test_known, test_name_known
screamer 0:66f3b5499f7f 14 from tools.targets import TARGET_NAMES
screamer 23:fbae331171fa 15 from utils import argparse_filestring_type, argparse_many
screamer 22:9e85236d8716 16 from utils import argparse_force_lowercase_type, argparse_force_uppercase_type
screamer 23:fbae331171fa 17 from project_api import setup_project, perform_export, print_results, get_lib_symbols
screamer 0:66f3b5499f7f 18
screamer 0:66f3b5499f7f 19
screamer 0:66f3b5499f7f 20
screamer 0:66f3b5499f7f 21 if __name__ == '__main__':
screamer 0:66f3b5499f7f 22 # Parse Options
screamer 22:9e85236d8716 23 parser = ArgumentParser()
screamer 0:66f3b5499f7f 24
screamer 0:66f3b5499f7f 25 targetnames = TARGET_NAMES
screamer 0:66f3b5499f7f 26 targetnames.sort()
screamer 0:66f3b5499f7f 27 toolchainlist = EXPORTERS.keys()
screamer 0:66f3b5499f7f 28 toolchainlist.sort()
screamer 0:66f3b5499f7f 29
screamer 22:9e85236d8716 30 parser.add_argument("-m", "--mcu",
screamer 0:66f3b5499f7f 31 metavar="MCU",
screamer 0:66f3b5499f7f 32 default='LPC1768',
screamer 22:9e85236d8716 33 type=argparse_many(argparse_force_uppercase_type(targetnames, "MCU")),
screamer 0:66f3b5499f7f 34 help="generate project for the given MCU (%s)"% ', '.join(targetnames))
screamer 0:66f3b5499f7f 35
screamer 22:9e85236d8716 36 parser.add_argument("-i",
screamer 0:66f3b5499f7f 37 dest="ide",
screamer 0:66f3b5499f7f 38 default='uvision',
screamer 22:9e85236d8716 39 type=argparse_force_lowercase_type(toolchainlist, "toolchain"),
screamer 0:66f3b5499f7f 40 help="The target IDE: %s"% str(toolchainlist))
screamer 0:66f3b5499f7f 41
screamer 22:9e85236d8716 42 parser.add_argument("-c", "--clean",
screamer 0:66f3b5499f7f 43 action="store_true",
screamer 0:66f3b5499f7f 44 default=False,
screamer 0:66f3b5499f7f 45 help="clean the export directory")
screamer 0:66f3b5499f7f 46
screamer 22:9e85236d8716 47 group = parser.add_mutually_exclusive_group(required=False)
screamer 22:9e85236d8716 48 group.add_argument("-p",
screamer 22:9e85236d8716 49 type=test_known,
screamer 0:66f3b5499f7f 50 dest="program",
screamer 0:66f3b5499f7f 51 help="The index of the desired test program: [0-%d]"% (len(TESTS)-1))
screamer 0:66f3b5499f7f 52
screamer 22:9e85236d8716 53 group.add_argument("-n",
screamer 22:9e85236d8716 54 type=test_name_known,
screamer 22:9e85236d8716 55 dest="program",
screamer 0:66f3b5499f7f 56 help="The name of the desired test program")
screamer 0:66f3b5499f7f 57
screamer 22:9e85236d8716 58 parser.add_argument("-b",
screamer 0:66f3b5499f7f 59 dest="build",
screamer 0:66f3b5499f7f 60 action="store_true",
screamer 0:66f3b5499f7f 61 default=False,
screamer 0:66f3b5499f7f 62 help="use the mbed library build, instead of the sources")
screamer 0:66f3b5499f7f 63
screamer 22:9e85236d8716 64 group.add_argument("-L", "--list-tests",
screamer 0:66f3b5499f7f 65 action="store_true",
screamer 0:66f3b5499f7f 66 dest="list_tests",
screamer 0:66f3b5499f7f 67 default=False,
screamer 0:66f3b5499f7f 68 help="list available programs in order and exit")
screamer 0:66f3b5499f7f 69
screamer 22:9e85236d8716 70 group.add_argument("-S", "--list-matrix",
screamer 0:66f3b5499f7f 71 action="store_true",
screamer 0:66f3b5499f7f 72 dest="supported_ides",
screamer 0:66f3b5499f7f 73 default=False,
screamer 0:66f3b5499f7f 74 help="displays supported matrix of MCUs and IDEs")
screamer 0:66f3b5499f7f 75
screamer 22:9e85236d8716 76 parser.add_argument("-E",
screamer 0:66f3b5499f7f 77 action="store_true",
screamer 0:66f3b5499f7f 78 dest="supported_ides_html",
screamer 0:66f3b5499f7f 79 default=False,
screamer 0:66f3b5499f7f 80 help="writes tools/export/README.md")
screamer 0:66f3b5499f7f 81
screamer 22:9e85236d8716 82 parser.add_argument("--source",
screamer 13:ab47a20b66f0 83 action="append",
screamer 22:9e85236d8716 84 type=argparse_filestring_type,
screamer 0:66f3b5499f7f 85 dest="source_dir",
screamer 22:9e85236d8716 86 default=[],
screamer 0:66f3b5499f7f 87 help="The source (input) directory")
screamer 0:66f3b5499f7f 88
screamer 22:9e85236d8716 89 parser.add_argument("-D",
screamer 0:66f3b5499f7f 90 action="append",
screamer 0:66f3b5499f7f 91 dest="macros",
screamer 0:66f3b5499f7f 92 help="Add a macro definition")
screamer 0:66f3b5499f7f 93
screamer 22:9e85236d8716 94 options = parser.parse_args()
screamer 0:66f3b5499f7f 95
screamer 0:66f3b5499f7f 96 # Print available tests in order and exit
screamer 0:66f3b5499f7f 97 if options.list_tests is True:
screamer 0:66f3b5499f7f 98 print '\n'.join(map(str, sorted(TEST_MAP.values())))
screamer 0:66f3b5499f7f 99 sys.exit()
screamer 0:66f3b5499f7f 100
screamer 0:66f3b5499f7f 101 # Only prints matrix of supported IDEs
screamer 0:66f3b5499f7f 102 if options.supported_ides:
screamer 0:66f3b5499f7f 103 print mcu_ide_matrix()
screamer 0:66f3b5499f7f 104 exit(0)
screamer 0:66f3b5499f7f 105
screamer 0:66f3b5499f7f 106 # Only prints matrix of supported IDEs
screamer 0:66f3b5499f7f 107 if options.supported_ides_html:
screamer 0:66f3b5499f7f 108 html = mcu_ide_matrix(verbose_html=True)
screamer 0:66f3b5499f7f 109 try:
screamer 0:66f3b5499f7f 110 with open("./export/README.md","w") as f:
screamer 0:66f3b5499f7f 111 f.write("Exporter IDE/Platform Support\n")
screamer 0:66f3b5499f7f 112 f.write("-----------------------------------\n")
screamer 0:66f3b5499f7f 113 f.write("\n")
screamer 0:66f3b5499f7f 114 f.write(html)
screamer 0:66f3b5499f7f 115 except IOError as e:
screamer 0:66f3b5499f7f 116 print "I/O error({0}): {1}".format(e.errno, e.strerror)
screamer 0:66f3b5499f7f 117 except:
screamer 0:66f3b5499f7f 118 print "Unexpected error:", sys.exc_info()[0]
screamer 0:66f3b5499f7f 119 raise
screamer 0:66f3b5499f7f 120 exit(0)
screamer 0:66f3b5499f7f 121
screamer 0:66f3b5499f7f 122 # Clean Export Directory
screamer 0:66f3b5499f7f 123 if options.clean:
screamer 0:66f3b5499f7f 124 if exists(EXPORT_DIR):
screamer 0:66f3b5499f7f 125 rmtree(EXPORT_DIR)
screamer 0:66f3b5499f7f 126
screamer 0:66f3b5499f7f 127 # Export results
screamer 0:66f3b5499f7f 128 successes = []
screamer 0:66f3b5499f7f 129 failures = []
screamer 0:66f3b5499f7f 130
screamer 0:66f3b5499f7f 131 # source_dir = use relative paths, otherwise sources are copied
screamer 0:66f3b5499f7f 132 sources_relative = True if options.source_dir else False
screamer 0:66f3b5499f7f 133
screamer 22:9e85236d8716 134 for mcu in options.mcu:
screamer 0:66f3b5499f7f 135 # Program Number or name
screamer 22:9e85236d8716 136 p, src, ide = options.program, options.source_dir, options.ide
screamer 23:fbae331171fa 137 project_dir, project_name, project_temp = setup_project(mcu, ide, p, src, options.build)
screamer 0:66f3b5499f7f 138
screamer 23:fbae331171fa 139 zip = src is [] # create zip when no src_dir provided
screamer 23:fbae331171fa 140 clean = src is [] # don't clean when source is provided, use acrual source tree for IDE files
screamer 0:66f3b5499f7f 141
screamer 0:66f3b5499f7f 142 # Export to selected toolchain
screamer 23:fbae331171fa 143 lib_symbols = get_lib_symbols(options.macros, src, p)
screamer 13:ab47a20b66f0 144 tmp_path, report = export(project_dir, project_name, ide, mcu, project_dir[0], project_temp, clean=clean, make_zip=zip, extra_symbols=lib_symbols, sources_relative=sources_relative)
screamer 0:66f3b5499f7f 145 if report['success']:
screamer 13:ab47a20b66f0 146 if not zip:
screamer 13:ab47a20b66f0 147 zip_path = join(project_temp, project_name)
screamer 13:ab47a20b66f0 148 else:
screamer 13:ab47a20b66f0 149 zip_path = join(EXPORT_DIR, "%s_%s_%s.zip" % (project_name, ide, mcu))
screamer 0:66f3b5499f7f 150 move(tmp_path, zip_path)
screamer 0:66f3b5499f7f 151 successes.append("%s::%s\t%s"% (mcu, ide, zip_path))
screamer 0:66f3b5499f7f 152 else:
screamer 0:66f3b5499f7f 153 failures.append("%s::%s\t%s"% (mcu, ide, report['errormsg']))
screamer 0:66f3b5499f7f 154
screamer 0:66f3b5499f7f 155 # Prints export results
screamer 23:fbae331171fa 156 print_results(successes, failures)