pbio/sys/hmi_mpy_env: Compile script with Pybricksdev.

Drop-in for HMI module that simulates sending a multi-mpy file to the hub and then starts it. Used in the simulated hub to test different programs. The script to start is set with an environment variable and compiled with Pybricksdev.

Author: laurensvalk

.vscode/launch.json

@@ -97,16 +97,17 @@
             "type": "cppdbg",
             "request": "launch",
             "program": "${workspaceFolder}/bricks/simhub/debug/firmware.elf",
-            "args": [
-                "${workspaceFolder}/tests/virtualhub/motor/car.py"
-            ],
             "stopAtEntry": false,
             "cwd": "${workspaceFolder}",
             "environment": [
                 {
                     "name": "PBIO_TEST_CONNECT_SOCKET",
                     "value": "true"
                 },
+                {
+                    "name": "TEST_SCRIPT",
+                    "value": "${workspaceFolder}/tests/virtualhub/basics/hello.py"
+                },
             ],
             "externalConsole": false,
             "MIMode": "gdb",

bricks/_common/common.mk

@@ -706,9 +706,5 @@ deploy-openocd: $(BUILD)/firmware-base.bin
 # Run emulation build on a POSIX system using normal stdio
 run: $(BUILD)/firmware.elf
 	@$(BUILD)/firmware.elf
-	@echo "Exit status: $$?"
-
-test: $(BUILD)/firmware.elf
-	$(Q)/usr/bin/printf "print('hello world!', list(x+1 for x in range(10)), end='eol\\\\n')\\r\\n\\004" | $(BUILD)/firmware.elf | tail -n2 | grep "^hello world! \\[1, 2, 3, 4, 5, 6, 7, 8, 9, 10\\]eol"
 
 include $(TOP)/py/mkrules.mk

bricks/_common/sources.mk

@@ -237,6 +237,7 @@ PBIO_SRC_C = $(addprefix lib/pbio/,\
 	sys/battery.c \
 	sys/command.c \
 	sys/core.c \
+	sys/hmi_env_mpy.c \
 	sys/hmi_lcd.c \
 	sys/hmi_pup.c \
 	sys/hmi_none.c \

bricks/simhub/make_mpy.py

@@ -0,0 +1,15 @@
+from pybricksdev.compile import compile_multi_file
+import asyncio
+import argparse
+import sys
+
+# Set up argument parser
+parser = argparse.ArgumentParser(description="Compile to multi-mpy using pybricksdev.")
+parser.add_argument("file", help="The path to the Python file to compile.")
+args = parser.parse_args()
+
+# Compile the file
+result = asyncio.run(compile_multi_file(args.file, (6, 1)))
+
+# Write the result to stdout
+sys.stdout.buffer.write(result)

lib/pbio/platform/sim_hub/pbdrvconfig.h

@@ -5,7 +5,7 @@
 #define PBDRV_CONFIG_BATTERY_TEST                           (1)
 
 #define PBDRV_CONFIG_BLOCK_DEVICE                           (1)
-#define PBDRV_CONFIG_BLOCK_DEVICE_RAM_SIZE                  (10 * 1024)
+#define PBDRV_CONFIG_BLOCK_DEVICE_RAM_SIZE                  (50 * 1024)
 #define PBDRV_CONFIG_BLOCK_DEVICE_TEST                      (1)
 
 #define PBDRV_CONFIG_BLUETOOTH                              (1)

lib/pbio/platform/sim_hub/pbsysconfig.h

@@ -13,7 +13,7 @@
 #define PBSYS_CONFIG_HOST_STDIN_BUF_SIZE            (21)
 #define PBSYS_CONFIG_HMI                            (1)
 #define PBSYS_CONFIG_HMI_STOP_BUTTON                (1 << 5) // center
-#define PBSYS_CONFIG_HMI_NONE                       (1)
+#define PBSYS_CONFIG_HMI_ENV_MPY                    (1)
 #define PBSYS_CONFIG_HMI_NUM_SLOTS                  (0)
 #define PBSYS_CONFIG_HUB_LIGHT_MATRIX               (0)
 #define PBSYS_CONFIG_MAIN                           (1)

lib/pbio/sys/hmi_env_mpy.c

@@ -0,0 +1,79 @@
+// SPDX-License-Identifier: MIT
+// Copyright (c) 2018-2025 The Pybricks Authors
+
+// Drop-in for HMI module that simulates sending a multi-mpy file to the hub
+// and then starts it. Used in the simulated hub to test different programs.
+// The script to start is set with an environment variable.
+
+#include <pbsys/config.h>
+
+#if PBSYS_CONFIG_HMI_ENV_MPY
+
+#include <pbio/button.h>
+#include <pbio/os.h>
+#include <pbsys/command.h>
+#include <pbsys/main.h>
+#include <pbsys/status.h>
+#include "storage.h"
+
+#include <errno.h>
+#include <signal.h>
+#include <stdbool.h>
+#include <stddef.h>
+#include <stdint.h>
+#include <stdio.h>
+#include <stdlib.h>
+#include <sys/stat.h>
+
+void pbsys_hmi_init(void) {
+}
+
+void pbsys_hmi_deinit(void) {
+}
+
+pbio_error_t pbsys_hmi_await_program_selection(void) {
+
+    pbio_os_run_processes_and_wait_for_event();
+
+    // With this HMI, we run a script once and then exit.
+    static bool ran_once = false;
+    if (ran_once) {
+        return PBIO_ERROR_CANCELED;
+    }
+    ran_once = true;
+
+    // Test if script is provided via environment.
+    const char *script_path = getenv("TEST_SCRIPT");
+    if (!script_path) {
+        // No script given, just start REPL
+        return pbsys_main_program_request_start(PBIO_PYBRICKS_USER_PROGRAM_ID_REPL, PBSYS_MAIN_PROGRAM_START_REQUEST_TYPE_BOOT);
+    }
+
+    // Pybricksdev helper script, pipes multi-mpy to us.
+    char command[512];
+    snprintf(command, sizeof(command), "python ./bricks/simhub/make_mpy.py %s", script_path);
+    FILE *pipe = popen(command, "r");
+    if (!pipe) {
+        perror("Failed to compile program with Pybricksdev\n");
+        return PBIO_ERROR_CANCELED;
+    }
+
+    // Read the multi-mpy file from pipe.
+    uint8_t program_buf[PBDRV_CONFIG_BLOCK_DEVICE_RAM_SIZE];
+    size_t read_size = fread(program_buf, 1, sizeof(program_buf), pipe);
+    pclose(pipe);
+
+    if (read_size == 0) {
+        perror("Error reading from pipe");
+        return PBIO_ERROR_CANCELED;
+    }
+
+    // Load the program in storage, as if receiving it.
+    pbsys_storage_set_program_size(0);
+    pbsys_storage_set_program_data(0, program_buf, read_size);
+    pbsys_storage_set_program_size(read_size);
+
+    return pbsys_main_program_request_start(PBIO_PYBRICKS_USER_PROGRAM_ID_FIRST_SLOT, PBSYS_MAIN_PROGRAM_START_REQUEST_TYPE_BOOT);
+}
+
+#endif // PBSYS_CONFIG_HMI_ENV_MPY

tests/virtualhub/basics/hello.py

@@ -0,0 +1,5 @@
+from pybricks.tools import wait
+
+print("\nHello")
+wait(1000)
+print("World\n")

tests/virtualhub/basics/hello.py.exp

@@ -0,0 +1,4 @@
+
+Hello
+World
+