EL4121 Embedded System / mbed-os

mbed-os

Dependents:   cobaLCDJoyMotor_Thread odometry_omni_3roda_v3 odometry_omni_3roda_v1 odometry_omni_3roda_v2 ... more

tools/flash_algo/extract.py@0:b74591d5ab33, 2017-12-11 (annotated)

Committer:
be_bryan
Date:
Mon Dec 11 17:54:04 2017 +0000
Revision:
0:b74591d5ab33
motor ++

Who changed what in which revision?

UserRevisionLine numberNew contents of line
be_bryan 0:b74591d5ab33 1 #!/usr/bin/env python
be_bryan 0:b74591d5ab33 2 """
be_bryan 0:b74591d5ab33 3 mbed
be_bryan 0:b74591d5ab33 4 Copyright (c) 2017-2017 ARM Limited
be_bryan 0:b74591d5ab33 5
be_bryan 0:b74591d5ab33 6 Licensed under the Apache License, Version 2.0 (the "License");
be_bryan 0:b74591d5ab33 7 you may not use this file except in compliance with the License.
be_bryan 0:b74591d5ab33 8 You may obtain a copy of the License at
be_bryan 0:b74591d5ab33 9
be_bryan 0:b74591d5ab33 10 http://www.apache.org/licenses/LICENSE-2.0
be_bryan 0:b74591d5ab33 11
be_bryan 0:b74591d5ab33 12 Unless required by applicable law or agreed to in writing, software
be_bryan 0:b74591d5ab33 13 distributed under the License is distributed on an "AS IS" BASIS,
be_bryan 0:b74591d5ab33 14 WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
be_bryan 0:b74591d5ab33 15 See the License for the specific language governing permissions and
be_bryan 0:b74591d5ab33 16 limitations under the License.
be_bryan 0:b74591d5ab33 17 """
be_bryan 0:b74591d5ab33 18
be_bryan 0:b74591d5ab33 19 from __future__ import print_function
be_bryan 0:b74591d5ab33 20 import sys
be_bryan 0:b74591d5ab33 21 import os
be_bryan 0:b74591d5ab33 22 import argparse
be_bryan 0:b74591d5ab33 23 from os.path import join, abspath, dirname
be_bryan 0:b74591d5ab33 24 from flash_algo import PackFlashAlgo
be_bryan 0:b74591d5ab33 25
be_bryan 0:b74591d5ab33 26 # Be sure that the tools directory is in the search path
be_bryan 0:b74591d5ab33 27 ROOT = abspath(join(dirname(__file__), "..", ".."))
be_bryan 0:b74591d5ab33 28 sys.path.insert(0, ROOT)
be_bryan 0:b74591d5ab33 29
be_bryan 0:b74591d5ab33 30 from tools.targets import TARGETS
be_bryan 0:b74591d5ab33 31 from tools.arm_pack_manager import Cache
be_bryan 0:b74591d5ab33 32
be_bryan 0:b74591d5ab33 33 TEMPLATE_PATH = "c_blob_mbed.tmpl"
be_bryan 0:b74591d5ab33 34
be_bryan 0:b74591d5ab33 35
be_bryan 0:b74591d5ab33 36 def main():
be_bryan 0:b74591d5ab33 37 """Generate flash algorithms"""
be_bryan 0:b74591d5ab33 38 parser = argparse.ArgumentParser(description='Flash generator')
be_bryan 0:b74591d5ab33 39 parser.add_argument("--rebuild_all", action="store_true",
be_bryan 0:b74591d5ab33 40 help="Rebuild entire cache")
be_bryan 0:b74591d5ab33 41 parser.add_argument("--rebuild_descriptors", action="store_true",
be_bryan 0:b74591d5ab33 42 help="Rebuild descriptors")
be_bryan 0:b74591d5ab33 43 parser.add_argument("--target", default=None,
be_bryan 0:b74591d5ab33 44 help="Name of target to generate algo for")
be_bryan 0:b74591d5ab33 45 parser.add_argument("--all", action="store_true",
be_bryan 0:b74591d5ab33 46 help="Build all flash algos for devcies")
be_bryan 0:b74591d5ab33 47 args = parser.parse_args()
be_bryan 0:b74591d5ab33 48
be_bryan 0:b74591d5ab33 49 cache = Cache(True, True)
be_bryan 0:b74591d5ab33 50 if args.rebuild_all:
be_bryan 0:b74591d5ab33 51 cache.cache_everything()
be_bryan 0:b74591d5ab33 52 print("Cache rebuilt")
be_bryan 0:b74591d5ab33 53 return
be_bryan 0:b74591d5ab33 54
be_bryan 0:b74591d5ab33 55 if args.rebuild_descriptors:
be_bryan 0:b74591d5ab33 56 cache.cache_descriptors()
be_bryan 0:b74591d5ab33 57 print("Descriptors rebuilt")
be_bryan 0:b74591d5ab33 58 return
be_bryan 0:b74591d5ab33 59
be_bryan 0:b74591d5ab33 60 if args.target is None:
be_bryan 0:b74591d5ab33 61 device_and_filenames = [(target.device_name, target.name) for target
be_bryan 0:b74591d5ab33 62 in TARGETS if hasattr(target, "device_name")]
be_bryan 0:b74591d5ab33 63 else:
be_bryan 0:b74591d5ab33 64 device_and_filenames = [(args.target, args.target.replace("/", "-"))]
be_bryan 0:b74591d5ab33 65
be_bryan 0:b74591d5ab33 66 try:
be_bryan 0:b74591d5ab33 67 os.mkdir("output")
be_bryan 0:b74591d5ab33 68 except OSError:
be_bryan 0:b74591d5ab33 69 # Directory already exists
be_bryan 0:b74591d5ab33 70 pass
be_bryan 0:b74591d5ab33 71
be_bryan 0:b74591d5ab33 72 for device, filename in device_and_filenames:
be_bryan 0:b74591d5ab33 73 dev = cache.index[device]
be_bryan 0:b74591d5ab33 74 binaries = cache.get_flash_algorthim_binary(device, all=True)
be_bryan 0:b74591d5ab33 75 algos = [PackFlashAlgo(binary.read()) for binary in binaries]
be_bryan 0:b74591d5ab33 76 filtered_algos = algos if args.all else filter_algos(dev, algos)
be_bryan 0:b74591d5ab33 77 for idx, algo in enumerate(filtered_algos):
be_bryan 0:b74591d5ab33 78 file_name = ("%s_%i.c" % (filename, idx)
be_bryan 0:b74591d5ab33 79 if args.all or len(filtered_algos) != 1
be_bryan 0:b74591d5ab33 80 else "%s.c" % filename)
be_bryan 0:b74591d5ab33 81 output_path = join("output", file_name)
be_bryan 0:b74591d5ab33 82 algo.process_template(TEMPLATE_PATH, output_path)
be_bryan 0:b74591d5ab33 83 print("%s: %s \r" % (device, filename))
be_bryan 0:b74591d5ab33 84
be_bryan 0:b74591d5ab33 85
be_bryan 0:b74591d5ab33 86 def filter_algos(dev, algos):
be_bryan 0:b74591d5ab33 87 if "memory" not in dev:
be_bryan 0:b74591d5ab33 88 return algos
be_bryan 0:b74591d5ab33 89 if "IROM1" not in dev["memory"]:
be_bryan 0:b74591d5ab33 90 return algos
be_bryan 0:b74591d5ab33 91 if "IROM2" in dev["memory"]:
be_bryan 0:b74591d5ab33 92 return algos
be_bryan 0:b74591d5ab33 93
be_bryan 0:b74591d5ab33 94 rom_rgn = dev["memory"]["IROM1"]
be_bryan 0:b74591d5ab33 95 try:
be_bryan 0:b74591d5ab33 96 start = int(rom_rgn["start"], 0)
be_bryan 0:b74591d5ab33 97 size = int(rom_rgn["size"], 0)
be_bryan 0:b74591d5ab33 98 except ValueError:
be_bryan 0:b74591d5ab33 99 return algos
be_bryan 0:b74591d5ab33 100
be_bryan 0:b74591d5ab33 101 matching_algos = [algo for algo in algos if
be_bryan 0:b74591d5ab33 102 algo.flash_start == start and algo.flash_size == size]
be_bryan 0:b74591d5ab33 103 return matching_algos if len(matching_algos) == 1 else algos
be_bryan 0:b74591d5ab33 104
be_bryan 0:b74591d5ab33 105
be_bryan 0:b74591d5ab33 106 if __name__ == '__main__':
be_bryan 0:b74591d5ab33 107 main()