tools/mpy-tool.py: Support freezing tuples and other consts.
This also simplifies how constants are frozen. Signed-off-by: Damien George <damien@micropython.org>
This commit is contained in:
Damien George
parent
2a075cc8a9
commit
68b3aeeb57
|
|
@ -600,30 +600,19 @@ class CompiledModule:
|
|||
print(" .rc = &raw_code_%s," % self.raw_code.escaped_name)
|
||||
print("};")
|
||||
|
||||
def freeze_constants(self):
|
||||
def freeze_constant_obj(self, obj_name, obj):
|
||||
global const_str_content, const_int_content, const_obj_content
|
||||
|
||||
if len(self.qstr_table):
|
||||
print(
|
||||
"static const qstr_short_t const_qstr_table_data_%s[%u] = {"
|
||||
% (self.escaped_name, len(self.qstr_table))
|
||||
)
|
||||
for q in self.qstr_table:
|
||||
print(" %s," % q.qstr_id)
|
||||
print("};")
|
||||
|
||||
if not len(self.obj_table):
|
||||
return
|
||||
|
||||
# generate constant objects
|
||||
print()
|
||||
print("// constants")
|
||||
for i, obj in enumerate(self.obj_table):
|
||||
obj_name = "const_obj_%s_%u" % (self.escaped_name, i)
|
||||
if isinstance(obj, MPFunTable):
|
||||
pass
|
||||
return "&mp_fun_table"
|
||||
elif obj is None:
|
||||
return "MP_ROM_NONE"
|
||||
elif obj is False:
|
||||
return "MP_ROM_FALSE"
|
||||
elif obj is True:
|
||||
return "MP_ROM_TRUE"
|
||||
elif obj is Ellipsis:
|
||||
print("#define %s mp_const_ellipsis_obj" % obj_name)
|
||||
return "MP_ROM_PTR(&mp_const_ellipsis_obj)"
|
||||
elif is_str_type(obj) or is_bytes_type(obj):
|
||||
if is_str_type(obj):
|
||||
obj = bytes_cons(obj, "utf8")
|
||||
|
|
@ -642,6 +631,7 @@ class CompiledModule:
|
|||
)
|
||||
const_str_content += len(obj)
|
||||
const_obj_content += 4 * 4
|
||||
return "MP_ROM_PTR(&%s)" % obj_name
|
||||
elif is_int_type(obj):
|
||||
if config.MICROPY_LONGINT_IMPL == config.MICROPY_LONGINT_IMPL_NONE:
|
||||
# TODO check if we can actually fit this long-int into a small-int
|
||||
|
|
@ -669,7 +659,9 @@ class CompiledModule:
|
|||
)
|
||||
const_int_content += (digs.count(",") + 1) * bits_per_dig // 8
|
||||
const_obj_content += 4 * 4
|
||||
return "MP_ROM_PTR(&%s)" % obj_name
|
||||
elif type(obj) is float:
|
||||
macro_name = "%s_macro" % obj_name
|
||||
print(
|
||||
"#if MICROPY_OBJ_REPR == MICROPY_OBJ_REPR_A || MICROPY_OBJ_REPR == MICROPY_OBJ_REPR_B"
|
||||
)
|
||||
|
|
@ -677,16 +669,64 @@ class CompiledModule:
|
|||
"static const mp_obj_float_t %s = {{&mp_type_float}, (mp_float_t)%.16g};"
|
||||
% (obj_name, obj)
|
||||
)
|
||||
print("#define %s MP_ROM_PTR(&%s)" % (macro_name, obj_name))
|
||||
print("#elif MICROPY_OBJ_REPR == MICROPY_OBJ_REPR_C")
|
||||
n = struct.unpack("<I", struct.pack("<f", obj))[0]
|
||||
n = ((n & ~0x3) | 2) + 0x80800000
|
||||
print("#define %s ((mp_rom_obj_t)(0x%08x))" % (macro_name, n))
|
||||
print("#elif MICROPY_OBJ_REPR == MICROPY_OBJ_REPR_D")
|
||||
n = struct.unpack("<Q", struct.pack("<d", obj))[0]
|
||||
n += 0x8004000000000000
|
||||
print("#define %s ((mp_rom_obj_t)(0x%016x))" % (macro_name, n))
|
||||
print("#endif")
|
||||
const_obj_content += 3 * 4
|
||||
return macro_name
|
||||
elif type(obj) is complex:
|
||||
print(
|
||||
"static const mp_obj_complex_t %s = {{&mp_type_complex}, (mp_float_t)%.16g, (mp_float_t)%.16g};"
|
||||
% (obj_name, obj.real, obj.imag)
|
||||
)
|
||||
return "MP_ROM_PTR(&%s)" % obj_name
|
||||
elif type(obj) is tuple:
|
||||
if len(obj) == 0:
|
||||
return "MP_ROM_PTR(&mp_const_empty_tuple_obj)"
|
||||
else:
|
||||
obj_refs = []
|
||||
for i, sub_obj in enumerate(obj):
|
||||
sub_obj_name = "%s_%u" % (obj_name, i)
|
||||
obj_refs.append(self.freeze_constant_obj(sub_obj_name, sub_obj))
|
||||
print(
|
||||
"static const mp_rom_obj_tuple_t %s = {{&mp_type_tuple}, %d, {"
|
||||
% (obj_name, len(obj))
|
||||
)
|
||||
for ref in obj_refs:
|
||||
print(" %s," % ref)
|
||||
print("}};")
|
||||
return "MP_ROM_PTR(&%s)" % obj_name
|
||||
else:
|
||||
raise FreezeError(self, "freezing of object %r is not implemented" % (obj,))
|
||||
|
||||
def freeze_constants(self):
|
||||
if len(self.qstr_table):
|
||||
print(
|
||||
"static const qstr_short_t const_qstr_table_data_%s[%u] = {"
|
||||
% (self.escaped_name, len(self.qstr_table))
|
||||
)
|
||||
for q in self.qstr_table:
|
||||
print(" %s," % q.qstr_id)
|
||||
print("};")
|
||||
|
||||
if not len(self.obj_table):
|
||||
return
|
||||
|
||||
# generate constant objects
|
||||
print()
|
||||
print("// constants")
|
||||
obj_refs = []
|
||||
for i, obj in enumerate(self.obj_table):
|
||||
obj_name = "const_obj_%s_%u" % (self.escaped_name, i)
|
||||
obj_refs.append(self.freeze_constant_obj(obj_name, obj))
|
||||
|
||||
# generate constant table
|
||||
print()
|
||||
print("// constant table")
|
||||
|
|
@ -694,25 +734,8 @@ class CompiledModule:
|
|||
"static const mp_rom_obj_t const_obj_table_data_%s[%u] = {"
|
||||
% (self.escaped_name, len(self.obj_table))
|
||||
)
|
||||
for i in range(len(self.obj_table)):
|
||||
if isinstance(self.obj_table[i], MPFunTable):
|
||||
print(" &mp_fun_table,")
|
||||
elif type(self.obj_table[i]) is float:
|
||||
print(
|
||||
"#if MICROPY_OBJ_REPR == MICROPY_OBJ_REPR_A || MICROPY_OBJ_REPR == MICROPY_OBJ_REPR_B"
|
||||
)
|
||||
print(" MP_ROM_PTR(&const_obj_%s_%u)," % (self.escaped_name, i))
|
||||
print("#elif MICROPY_OBJ_REPR == MICROPY_OBJ_REPR_C")
|
||||
n = struct.unpack("<I", struct.pack("<f", self.obj_table[i]))[0]
|
||||
n = ((n & ~0x3) | 2) + 0x80800000
|
||||
print(" (mp_rom_obj_t)(0x%08x)," % (n,))
|
||||
print("#elif MICROPY_OBJ_REPR == MICROPY_OBJ_REPR_D")
|
||||
n = struct.unpack("<Q", struct.pack("<d", self.obj_table[i]))[0]
|
||||
n += 0x8004000000000000
|
||||
print(" (mp_rom_obj_t)(0x%016x)," % (n,))
|
||||
print("#endif")
|
||||
else:
|
||||
print(" MP_ROM_PTR(&const_obj_%s_%u)," % (self.escaped_name, i))
|
||||
for ref in obj_refs:
|
||||
print(" %s," % ref)
|
||||
print("};")
|
||||
|
||||
global const_table_ptr_content
|
||||
|
|
|
|||
Loading…
Reference in New Issue