paul@0 | 1 | #!/usr/bin/env python |
paul@0 | 2 | |
paul@0 | 3 | import micropython |
paul@149 | 4 | from micropython.graph import get_graph |
paul@90 | 5 | import rsvp |
paul@0 | 6 | import sys |
paul@94 | 7 | import os |
paul@4 | 8 | |
paul@90 | 9 | def show_code(code): |
paul@77 | 10 | for i, x in enumerate(code): |
paul@22 | 11 | print i, x |
paul@22 | 12 | |
paul@135 | 13 | def show_table_usage(raw_table, slice_size=100): |
paul@135 | 14 | for x in xrange(0, len(raw_table), slice_size): |
paul@135 | 15 | table_slice = raw_table[x:x+slice_size] |
paul@135 | 16 | print "%6d" % (len(table_slice) - table_slice.count(None)), \ |
paul@135 | 17 | "".join(entry and "#" or "_" for entry in table_slice) |
paul@135 | 18 | |
paul@52 | 19 | def attrs(obj): |
paul@52 | 20 | for name, attr in obj.items(): |
paul@52 | 21 | print name, attr |
paul@52 | 22 | |
paul@228 | 23 | # Command option processing. |
paul@94 | 24 | |
paul@228 | 25 | def parse_optimisations(args): |
paul@228 | 26 | |
paul@228 | 27 | "Parse 'args' for optimisation flags." |
paul@94 | 28 | |
paul@63 | 29 | if "-omax" in args: |
paul@175 | 30 | requested_optimisations = micropython.Program.supported_optimisations |
paul@63 | 31 | else: |
paul@63 | 32 | requested_optimisations = [] |
paul@63 | 33 | for arg in args: |
paul@63 | 34 | if arg.startswith("-o"): |
paul@93 | 35 | for arg_part in arg[2:].split(","): |
paul@93 | 36 | requested_optimisations.append(arg_part) |
paul@63 | 37 | |
paul@228 | 38 | return requested_optimisations |
paul@228 | 39 | |
paul@228 | 40 | # Main program. |
paul@228 | 41 | |
paul@228 | 42 | if __name__ == "__main__": |
paul@228 | 43 | args = sys.argv[2:] |
paul@228 | 44 | path = sys.path[:] |
paul@228 | 45 | |
paul@228 | 46 | if len(sys.argv) > 1: |
paul@228 | 47 | filename = os.path.abspath(sys.argv[1]) |
paul@228 | 48 | path.append(os.path.split(filename)[0]) |
paul@228 | 49 | else: |
paul@228 | 50 | filename = None |
paul@228 | 51 | |
paul@228 | 52 | requested_optimisations = parse_optimisations(args) |
paul@228 | 53 | |
paul@228 | 54 | # Make the program. |
paul@167 | 55 | |
paul@62 | 56 | try: |
paul@228 | 57 | p = micropython.program(filename, path, requested_optimisations, "-v" in args) |
paul@228 | 58 | if filename is None: |
paul@94 | 59 | print "Loading module micropython ..." |
paul@228 | 60 | m = p.get_importer().load("micropython") |
paul@62 | 61 | else: |
paul@228 | 62 | m = p.get_importer().get_module("__main__") |
paul@228 | 63 | |
paul@228 | 64 | # Report any errors. |
paul@228 | 65 | |
paul@62 | 66 | except micropython.ProcessingError, exc: |
paul@62 | 67 | print repr(exc) |
paul@99 | 68 | if "-tb" in args: |
paul@99 | 69 | raise |
paul@228 | 70 | |
paul@228 | 71 | # Provide information to any interactive session. |
paul@228 | 72 | |
paul@37 | 73 | else: |
paul@175 | 74 | ot = p.get_object_table() |
paul@175 | 75 | pt = p.get_parameter_table() |
paul@0 | 76 | |
paul@228 | 77 | if "-m" in args or "-t" in args: |
paul@228 | 78 | rm = rsvp.machine(p) |
paul@228 | 79 | if "-t" in args: |
paul@228 | 80 | success = rm.test(m) |
paul@228 | 81 | print "Test successful?", success |
paul@187 | 82 | |
paul@0 | 83 | # vim: tabstop=4 expandtab shiftwidth=4 |