libs/module: Add Module ASR support.

Signed-off-by: tinyu <tinyu.zhao@gmail.com>

Author: TinyuZhao

m5stack/libs/module/__init__.py

@@ -6,6 +6,7 @@
 
 _attrs = {
     "AIN4Module": "ain4",
+    "ASRModule": "asr",
     "AudioModule": "audio",
     "Bala2Module": "bala2",
     "CC1101Module": "cc1101",

m5stack/libs/module/asr.py

@@ -0,0 +1,28 @@
+# SPDX-FileCopyrightText: 2024 M5Stack Technology CO LTD
+#
+# SPDX-License-Identifier: MIT
+from unit.asr import ASRUnit
+
+
+class ASRModule(ASRUnit):
+    """Voice recognition hardware module.
+
+    :param int id: UART port ID for communication. Default is 1.
+    :param list|tuple port: Tuple containing TX and RX pin numbers.
+
+    UiFlow2 Code Block:
+
+        |init.png|
+
+    MicroPython Code Block:
+
+        .. code-block:: python
+
+            from unit import ASRUnit
+
+            # Initialize with UART1, TX on pin 2, RX on pin 1
+            asr = ASRUnit(id=1, port=(1, 2))
+    """
+
+    def __init__(self, id: Literal[0, 1, 2] = 1, tx=-1, rx=-1, verbose: bool = False):
+        super().__init__(id, port=(rx, tx), verbose=verbose)

m5stack/libs/module/manifest.py

@@ -7,6 +7,7 @@
     (
         "__init__.py",
         "ain4.py",
+        "asr.py",
         "audio.py",
         "bala2.py",
         "cc1101.py",

m5stack/libs/unit/asr.py

@@ -71,6 +71,17 @@ class ASRUnit:
         0x43: ["medium volume", None],
         0x44: ["minimum volume", None],
         0x45: ["check firmware version", None],
+        0x50: ["PA2 high level", None],
+        0x51: ["PA2 low level", None],
+        0x52: ["PA3 high level", None],
+        0x53: ["PA3 low level", None],
+        0x54: ["PA4 high level", None],
+        0x55: ["PA4 low level", None],
+        0x56: ["PA5 high level", None],
+        0x57: ["PA5 low level", None],
+        0x58: ["PC4 high level", None],
+        0x59: ["PC4 low level", None],
+        0x5A: ["inversion level", None],
         0xFE: ["Announce", None],
         0xFF: ["Hi,M Five", None],
     }
@@ -90,20 +101,48 @@ def _debug_print(self, *args, **kwargs):
             print(*args, **kwargs)
 
     def _handler(self, uart) -> None:
+        if uart.any() < 5:
+            return
+
         data = uart.read()
-        if data is not None and len(data) >= 5:
-            self._debug_print(("Received data: ", data))
-
-            if data[0] == 0xAA and data[1] == 0x55 and data[-2] == 0x55 and data[-1] == 0xAA:
-                self.is_recieved = True
-                self.raw_message = " ".join(f"0x{byte:02X}" for byte in data)
-                self._debug_print(("Parsed message:", self.raw_message.split()))
-
-                self.command_num = data[2]
-                self.check_tick_callback()
-            else:
-                self._debug_print("Invalid frame received: header/footer mismatch")
-                uart.read()
+        if data is None or len(data) < 5:
+            return
+
+        self._debug_print(("Received data: ", data))
+
+        # 检查5字节格式: AA 55 CMD 55 AA
+        if (
+            len(data) >= 5
+            and data[0] == 0xAA
+            and data[1] == 0x55
+            and data[3] == 0x55
+            and data[4] == 0xAA
+        ):
+            self.is_recieved = True
+            self.command_num = data[2]
+            self.msg = 0  # 5字节格式没有msg字段
+            self.raw_message = " ".join(f"0x{byte:02X}" for byte in data[:5])
+            self._debug_print(("Parsed 5-byte message:", self.raw_message))
+            self.check_tick_callback()
+
+        # 检查6字节格式: AA 55 CMD MSG 55 AA
+        elif (
+            len(data) >= 6
+            and data[0] == 0xAA
+            and data[1] == 0x55
+            and data[4] == 0x55
+            and data[5] == 0xAA
+        ):
+            self.is_recieved = True
+            self.command_num = data[2]
+            self.msg = data[3]  # 6字节格式包含msg字段
+            self.raw_message = " ".join(f"0x{byte:02X}" for byte in data[:6])
+            self._debug_print(("Parsed 6-byte message:", self.raw_message))
+            self.check_tick_callback()
+
+        else:
+            self._debug_print("Invalid frame received: header/footer mismatch")
+            uart.read()
 
     def get_received_status(self) -> bool:
         """Get message reception status.