implement frozen module changes from support #829

pybricks/support#829
Novakasa · Jan 13, 2023 · f29a248 · f29a248
1 parent 1410c40
commit f29a248
Show file tree

Hide file tree

Showing 4 changed files with 19 additions and 0 deletions.
diff --git a/bricks/_common_stm32/make.mk b/bricks/_common_stm32/make.mk
@@ -57,6 +57,9 @@ QSTR_GLOBAL_DEPENDENCIES = ../_common/mpconfigport.h ../_common_stm32/mpconfigpo
 # MicroPython feature configurations
 MICROPY_ROM_TEXT_COMPRESSION ?= 1
 
+# Frozen Python code
+FROZEN_MANIFEST ?= ../_common_stm32/manifest.py
+
 # include py core make definitions
 include $(TOP)/py/py.mk
 
@@ -353,6 +356,17 @@ SRC_QSTR += $(PY_EXTRA_SRC_C) $(PY_STM32_SRC_C) $(PYBRICKS_PYBRICKS_SRC_C)
 # Append any auto-generated sources that are needed by sources listed in SRC_QSTR
 SRC_QSTR_AUTO_DEPS +=
 
+ifneq ($(FROZEN_MANIFEST),)
+CFLAGS += -DMICROPY_QSTR_EXTRA_POOL=mp_qstr_frozen_const_pool
+CFLAGS += -DMICROPY_MODULE_FROZEN_MPY
+CFLAGS += -DMPZ_DIG_SIZE=16
+MPY_TOOL_FLAGS += -mlongint-impl none
+endif
+
+ifneq ($(FROZEN_MANIFEST),)
+CFLAGS += -DMICROPY_MODULE_FROZEN_STR
+endif
+
 # Main firmware build targets
 TARGETS := $(BUILD)/firmware.zip
 

diff --git a/bricks/_common_stm32/manifest.py b/bricks/_common_stm32/manifest.py
@@ -0,0 +1 @@
+freeze_as_mpy("modules")
diff --git a/bricks/_common_stm32/modules/my_code.py b/bricks/_common_stm32/modules/my_code.py
@@ -0,0 +1,2 @@
+print("hello, data!")
+data = 123
diff --git a/pybricks/pybricks.c b/pybricks/pybricks.c
@@ -2,6 +2,7 @@
 // Copyright (c) 2018-2022 The Pybricks Authors
 
 #include "py/mpconfig.h"
+#include "shared/runtime/pyexec.h"
 #include "py/obj.h"
 #include "py/objmodule.h"
 #include "py/objstr.h"
@@ -105,6 +106,7 @@ void pb_package_pybricks_init(bool import_all) {
         if (import_all) {
             pb_package_import_all();
         }
+        pyexec_file_if_exists("testmod.py");
         nlr_pop();
     } else {
         // Print initialization or import exception.