feature: Support M5Capsule board

Signed-off-by: lbuque <[email protected]>

Author: lbuque

m5stack/Makefile

@@ -20,7 +20,8 @@ boards := \
 	M5STACK_StickC_PLUS:stickcplus   \
 	M5STACK_Fire:fire                \
 	M5STACK_Basic:basic              \
-	M5STACK_Basic_4MB:basic
+	M5STACK_Basic_4MB:basic          \
+	M5STACK_Capsule:capsule
 
 define find_board
 $(if $(filter $(1):%,$(boards)),$(word 2,$(subst :, ,$(filter $(1):%,$(boards)))),none)
@@ -38,7 +39,8 @@ BOARD_TYPE_DEF := \
 	stickcplus2 \
 	stickcplus  \
 	fire        \
-	basic
+	basic       \
+	capsule
 
 # Select the board type to build, default is None
 # This value affects which folder in the "./fs/system/" directory is pack into "fs-system.bin"

m5stack/boards/M5STACK_Capsule/board.json

@@ -0,0 +1,18 @@
+{
+    "deploy": [
+        "../deploy_s3.md"
+    ],
+    "docs": "",
+    "features": [
+        "BLE",
+        "WiFi"
+    ],
+    "images": [
+        "generic_s3.jpg"
+    ],
+    "mcu": "esp32s3",
+    "product": "M5Stack S3 Serials",
+    "thumbnail": "",
+    "url": "https://www.espressif.com/en/products/modules",
+    "vendor": "M5Stack"
+}

m5stack/boards/M5STACK_Capsule/manifest.py

@@ -0,0 +1 @@
+include("$(MPY_DIR)/../m5stack/modules/startup/manifest_capsule.py")

m5stack/boards/M5STACK_Capsule/mpconfigboard.cmake

@@ -0,0 +1,18 @@
+set(IDF_TARGET esp32s3)
+
+set(SDKCONFIG_DEFAULTS
+    ./boards/M5STACK_S3_8MB/sdkconfig.board
+    ./boards/sdkconfig.base
+    ./boards/sdkconfig.240mhz
+    ./boards/sdkconfig.disable_iram
+    ./boards/sdkconfig.ble
+    ./boards/sdkconfig.usb
+    ./boards/sdkconfig.flash_8mb
+)
+
+# If not enable LVGL, ignore this...
+set(LV_CFLAGS -DLV_COLOR_DEPTH=16 -DLV_COLOR_16_SWAP=0)
+
+if(NOT MICROPY_FROZEN_MANIFEST)
+    set(MICROPY_FROZEN_MANIFEST ${CMAKE_SOURCE_DIR}/boards/manifest.py)
+endif()

m5stack/boards/M5STACK_Capsule/mpconfigboard.h

@@ -0,0 +1,13 @@
+#define MICROPY_HW_BOARD_NAME               "M5STACK"
+#define MICROPY_HW_MCU_NAME                 "ESP32S3"
+
+#define MICROPY_PY_MACHINE_DAC              (0)
+
+// Enable UART REPL for modules that have an external USB-UART and don't use native USB.
+#define MICROPY_HW_ENABLE_UART_REPL         (1)
+
+#define MICROPY_HW_I2C0_SCL                 (9)
+#define MICROPY_HW_I2C0_SDA                 (8)
+
+// If not enable LVGL, ignore this...
+#include "./../mpconfiglvgl.h"

m5stack/boards/M5STACK_Capsule/sdkconfig.board

@@ -0,0 +1,9 @@
+CONFIG_FLASHMODE_DIO=y  # QIO mode has some problem when mount fs
+CONFIG_ESPTOOLPY_FLASHMODE_DIO=y
+CONFIG_ESPTOOLPY_FLASHMODE="dio"
+CONFIG_ESPTOOLPY_FLASHFREQ_80M=y
+CONFIG_ESPTOOLPY_FLASHSIZE_DETECT=y
+CONFIG_ESPTOOLPY_AFTER_NORESET=y
+
+CONFIG_SPIRAM_MEMTEST=
+CONFIG_FREERTOS_UNICORE=y

m5stack/boards/M5STACK_StampS3/manifest.py

@@ -0,0 +1 @@
+include("$(MPY_DIR)/../m5stack/modules/startup/manifest_stamps3.py")

m5stack/libs/hardware/rgb.py

@@ -28,13 +28,14 @@ def __new__(cls, **kwargs) -> NeoPixel:
                 return WS2812(io=cls._io, n=cls._n)  # default
 
         # If not provided any io, according to board type, initialize the built-in rgb
+        board_id = M5.getBoard()
         if cls._instance == None:
-            if M5.BOARD.M5AtomS3 == M5.getBoard():
+            if M5.BOARD.M5AtomS3 == board_id:
                 pass
-            elif M5.getBoard() in (M5.BOARD.M5AtomS3Lite, M5.BOARD.M5AtomS3U):
+            elif board_id in (M5.BOARD.M5AtomS3Lite, M5.BOARD.M5AtomS3U):
                 cls._instance = WS2812(io=35, n=1)
                 return cls._instance
-            elif M5.BOARD.M5StampS3 == M5.getBoard():
+            elif board_id in (M5.BOARD.M5StampS3, M5.BOARD.M5Capsule):
                 cls._instance = WS2812(io=21, n=1)
                 return cls._instance
             else:

m5stack/modules/startup/__init__.py

@@ -132,7 +132,7 @@ def startup(boot_opt, timeout: int = 60) -> None:
 
                 basic = Basic_Startup()
                 basic.startup(ssid, pswd, timeout)
-        elif board_id == M5.BOARD.M5Stack:
+        elif board_id == M5.BOARD.M5Capsule:
             from .capsule import Capsule_Startup
 
             capsule = Capsule_Startup()

m5stack/modules/startup/capsule.py

@@ -0,0 +1,58 @@
+from .stamps3 import StampS3_Startup
+import network
+import time
+import M5
+
+
+class Capsule_Startup(StampS3_Startup):
+    """AtomS3U startup menu"""
+
+    def __init__(self) -> None:
+        super().__init__()
+
+    def startup(self, ssid: str, pswd: str, timeout: int = 60) -> None:
+        self.show_mac()
+        M5.Speaker.setVolumePercentage(1.0)
+
+        if super().connect_network(ssid=ssid, pswd=pswd):
+            print("Connecting to " + ssid + " ", end="")
+            status = super().connect_status()
+            self.rgb.set_brightness(60)
+            start = time.time()
+            while True:
+                t = super().connect_status()
+                if t != network.STAT_GOT_IP:
+                    status = t
+                    M5.Speaker.tone(5000, 50)
+                    if t is network.STAT_NO_AP_FOUND:
+                        self.show_error(ssid, "NO AP FOUND")
+                        break
+                    elif t is network.STAT_WRONG_PASSWORD:
+                        self.show_error(ssid, "WRONG PASSWORD")
+                        break
+                    elif t is network.STAT_HANDSHAKE_TIMEOUT:
+                        self.show_error(ssid, "HANDSHAKE ERR")
+                        break
+                    elif t is network.STAT_CONNECTING:
+                        print(".", end="")
+                if t != status and t == network.STAT_GOT_IP:
+                    status = t
+                    print(" ")
+                    print("Local IP: " + super().local_ip())
+                    self.rgb.set_color(0, self.COLOR_GREEN)
+                    self.rgb.set_brightness(100)
+                    break
+                # connect to network timeout
+                if (time.time() - start) > timeout:
+                    self.show_error(ssid, "TIMEOUT")
+                    break
+                time.sleep_ms(300)
+            if status == network.STAT_GOT_IP:
+                M5.Speaker.tone(4500, 50)
+                time.sleep(0.1)
+                M5.Speaker.tone(4500, 50)
+        else:
+            self.rgb.set_color(0, self.COLOR_RED)
+            self.rgb.set_brightness(100)
+            self.show_error("Not Found", "Please use M5Burner setup :)")
+