djsundog/circuitpython
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

tools/mpy-tool.py: Optimise freezing of str when str data is a qstr.

Browse Source 
Signed-off-by: Damien George <damien@micropython.org>
This commit is contained in:
Damien George 2022-04-08 14:05:23 +10:00
parent e647966fc9
commit 40d431d1bb
1 changed files with 9 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
tools/mpy-tool.py
View File

@ -478,6 +478,12 @@ class GlobalQStrList:
def get_by_index(self, i): def get_by_index(self, i):
return self.qstrs[i] return self.qstrs[i]
def find_by_str(self, s):
for q in self.qstrs:
if q is not None and q.str == s:
return q
return None
class MPFunTable: class MPFunTable:
def __repr__(self): def __repr__(self):
@ -633,6 +639,9 @@ class CompiledModule:
else: else:
return "MP_ROM_PTR(&mp_const_empty_bytes_obj)" return "MP_ROM_PTR(&mp_const_empty_bytes_obj)"
if is_str_type(obj): if is_str_type(obj):
q = global_qstrs.find_by_str(obj)
if q:
return "MP_ROM_QSTR(%s)" % q.qstr_id
obj = bytes_cons(obj, "utf8") obj = bytes_cons(obj, "utf8")
obj_type = "mp_type_str" obj_type = "mp_type_str"
else: else: