mxdbg 0.2.2__py3-none-any.whl

mxdbg.toml

@@ -0,0 +1,197 @@
+[version]
+MAJOR = 1
+MINOR = 0
+
+[task_cmd]
+TASK_IDLE = 0
+TASK_I2C_WRITE_READ = 1
+TASK_I2C_CONFIG = 2
+TASK_GPIO_WRITE_READ = 3
+TASK_GPIO_CONFIG = 4
+TASK_SPI_WRITE_READ = 5
+TASK_SPI_CONFIG = 6
+TASK_PWM_RUN_STOP = 7
+TASK_PWM_CONFIG = 8
+TASK_SPI_READ_IMAGE = 9
+TASK_USB_CONFIG = 240
+TASK_RESET_DEVICE = 241
+
+[error_map.ESPRESSIF_OK]
+code = 0
+desc = "esp_err_t value indicating success (no error)"
+
+[error_map.ESPRESSIF_FAIL]
+code = -1
+desc = "Generic esp_err_t code indicating failure"
+
+[error_map.ESPRESSIF_ERR_NO_MEM]
+code = 257
+desc = "Out of memory"
+
+[error_map.ESPRESSIF_ERR_INVALID_ARG]
+code = 258
+desc = "Invalid argument"
+
+[error_map.ESPRESSIF_ERR_INVALID_STATE]
+code = 259
+desc = "Invalid state"
+
+[error_map.ESPRESSIF_ERR_INVALID_SIZE]
+code = 260
+desc = "Invalid size"
+
+[error_map.ESPRESSIF_ERR_NOT_FOUND]
+code = 261
+desc = "Requested resource not found"
+
+[error_map.ESPRESSIF_ERR_NOT_SUPPORTED]
+code = 262
+desc = "Operation or feature not supported"
+
+[error_map.ESPRESSIF_ERR_TIMEOUT]
+code = 263
+desc = "Operation timed out"
+
+[error_map.ESPRESSIF_ERR_INVALID_RESPONSE]
+code = 264
+desc = "Received response was invalid"
+
+[error_map.ESPRESSIF_ERR_INVALID_CRC]
+code = 265
+desc = "CRC or checksum was invalid"
+
+[error_map.ESPRESSIF_ERR_INVALID_VERSION]
+code = 266
+desc = "Version was invalid"
+
+[error_map.ESPRESSIF_ERR_INVALID_MAC]
+code = 267
+desc = "MAC address was invalid"
+
+[error_map.ESPRESSIF_ERR_NOT_FINISHED]
+code = 268
+desc = "There are items remained to retrieve"
+
+[error_map.ESPRESSIF_ERR_WIFI_BASE]
+code = 12288
+desc = "Starting number of WiFi error codes"
+
+[error_map.ESPRESSIF_ERR_MESH_BASE]
+code = 16384
+desc = "Starting number of MESH error codes"
+
+[error_map.ESPRESSIF_ERR_FLASH_BASE]
+code = 24576
+desc = "Starting number of flash error codes"
+
+[error_map.ESPRESSIF_ERR_HW_CRYPTO_BASE]
+code = 49152
+desc = "Starting number of HW cryptography module error codes"
+
+[error_map.ESPRESSIF_ERR_MEMPROT_BASE]
+code = 53248
+desc = "Starting number of Memory Protection API error codes"
+
+[error_map.MXDBG_ERR_UNKNOWN_TASK]
+code = 16842496
+desc = "Unknown task"
+
+[error_map.MXDBG_ERR_DATA_UNPACK_FAILED]
+code = 16834560
+desc = "Data unpack failed"
+
+[error_map.MXDBG_ERR_DATA_UNPACK_HEADER_ERROR]
+code = 16834561
+desc = "Data unpack header error"
+
+[error_map.MXDBG_ERR_DATA_UNPACK_SEPARATOR_ERROR]
+code = 16834562
+desc = "Data unpack separator error"
+
+[error_map.MXDBG_ERR_USB_RX_DATA_FAILED]
+code = 16834816
+desc = "USB RX data failed"
+
+[error_map.MXDBG_ERR_USB_RX_DATA_TIMEOUT]
+code = 16834818
+desc = "USB RX data timeout"
+
+[error_map.MXDBG_ERR_USB_RX_DATA_HEADER_ERROR]
+code = 16834819
+desc = "USB RX data header error"
+
+[error_map.MXDBG_ERR_PWM_RUN_ALREADY]
+code = 16779009
+desc = "PWM is running already."
+
+[error_map.MXDBG_ERR_PWM_STOP_ALREADY]
+code = 16779010
+desc = "PWM is stopped already."
+
+[error_map.MXDBG_ERR_PWM_CONFIG_FAILED]
+code = 16779264
+desc = "PWM config failed"
+
+[error_map.MXDBG_ERR_PWM_INVALID_CHANNEL]
+code = 16779265
+desc = "Invalid PWM channel"
+
+[error_map.MXDBG_ERR_PWM_INVALID_PARAMS]
+code = 16779266
+desc = "Invalid PWM parameters"
+
+[error_map.MXDBG_ERR_PWM_INVALID_DUTY_CYCLES]
+code = 16779267
+desc = "Invalid PWM duty cycles"
+
+[error_map.MXDBG_ERR_SPI_READ_IMAGE_FAILED]
+code = 16779520
+desc = "SPI read image failed"
+
+[error_map.MXDBG_ERR_I2C_INIT_FAILED]
+code = 16777728
+desc = "I2C init failed"
+
+[error_map.MXDBG_ERR_I2C_DEINIT_FAILED]
+code = 16777729
+desc = "I2C deinit failed"
+
+[error_map.MXDBG_ERR_I2C_INVALID_OPERATION]
+code = 16777474
+desc = "Invalid I2C operation"
+
+[error_map.MXDBG_ERR_I2C_WRITE_READ_FAILED]
+code = 16777475
+desc = "I2C write read failed"
+
+[error_map.MXDBG_ERR_I2C_WRITE_FAILED]
+code = 16777476
+desc = "I2C write failed"
+
+[error_map.MXDBG_ERR_I2C_READ_FAILED]
+code = 16777477
+desc = "I2C read failed"
+
+[error_map.MXDBG_ERR_SPI_INIT_FAILED]
+code = 16778752
+desc = "SPI init failed"
+
+[error_map.MXDBG_ERR_SPI_DEINIT_FAILED]
+code = 16778753
+desc = "SPI deinit failed"
+
+[error_map.MXDBG_ERR_SPI_INVALID_OPERATION]
+code = 16778498
+desc = "Invalid SPI operation"
+
+[error_map.MXDBG_ERR_SPI_READ_LEN_TOO_LONG]
+code = 16778499
+desc = "SPI read length too long"
+
+[error_map.MXDBG_ERR_SPI_TRANSACTION_FAILED]
+code = 16778500
+desc = "SPI transaction failed"
+
+[error_map.MXDBG_ERR_GPIO_CONFIG_FAILED]
+code = 16778240
+desc = "GPIO config failed"

mxdbg_ui.py

@@ -0,0 +1,262 @@
+import inspect
+import io
+import sys
+
+import streamlit as st
+from mxdbg import MXDBG
+
+# 页面配置必须在顶部
+st.set_page_config(page_title="MXDBG Web UI", layout="wide")
+
+
+# 定义全局 MXDBG 对象
+@st.cache_resource
+def get_mxdbg():
+    """创建并返回 MXDBG 对象（单例模式）"""
+    try:
+        mxdbg_instance = MXDBG()
+        if not hasattr(mxdbg_instance, "_client"):  # 确保 _client 属性存在
+            mxdbg_instance._client = None
+        return mxdbg_instance
+    except ValueError as e:
+        st.error(f"无法连接设备：{e}")
+        return None
+
+
+# 初始化 MXDBG 对象
+mxdbg = get_mxdbg()
+
+# 定义分类规则
+categories = {
+    "GPIO": ["gpio_"],
+    "PWM": ["pwm_"],
+    "SPI": ["spi_"],
+    "I2C": ["i2c_"],
+    "POWER": ["power_"],
+    "EXPAND IO": ["expand_io_"],
+    "OTHERS": [],
+}
+
+
+# 动态提取和分类接口
+def extract_and_categorize_methods(mxdbg):
+    """从 MXDBG 类中动态提取方法并分类"""
+    if not mxdbg:
+        return {key: [] for key in categories}
+
+    # 提取所有方法（过滤掉双下划线开头、私有方法和非方法的属性）
+    methods = [
+        method
+        for method in dir(mxdbg)
+        if callable(getattr(mxdbg, method))
+        and not method.startswith("__")
+        and not method.startswith("_")
+    ]
+
+    categorized_methods = {key: [] for key in categories}
+
+    for method in methods:
+        categorized = False
+        for category, prefixes in categories.items():
+            if any(method.lower().startswith(prefix) for prefix in prefixes):
+                categorized_methods[category].append(method)
+                categorized = True
+                break
+        if not categorized:
+            categorized_methods["OTHERS"].append(method)
+
+    return categorized_methods
+
+
+# 获取接口的签名和默认参数
+def get_method_signature(obj, method_name):
+    """获取方法签名和默认参数"""
+    method = getattr(obj, method_name, None)
+    if not method:
+        return f"{method_name}()"
+    try:
+        sig = inspect.signature(method)
+        params = [
+            f"{name}={param.default!r}" if param.default is not param.empty else name
+            for name, param in sig.parameters.items()
+        ]
+        return f"{method_name}({', '.join(params)})"
+    except Exception:
+        return f"{method_name}()"
+
+
+# 获取分类后的方法
+categorized_methods = extract_and_categorize_methods(mxdbg)
+
+# 页面布局
+st.title("MXDBG Web UI")
+
+col1, col2 = st.columns([1, 2])
+
+# 左侧功能选项
+with col1:
+    st.subheader("功能选项卡")
+    selected_tab = st.radio(
+        "选择功能类别", list(categorized_methods.keys()), horizontal=True
+    )
+
+    st.write(f"### {selected_tab} 的接口列表")
+    matching_interfaces = categorized_methods.get(selected_tab, [])
+    for interface in matching_interfaces:
+        if st.button(f"选择接口：{interface}", key=f"select_{interface}"):
+            # 自动填写接口默认参数
+            method_call = get_method_signature(mxdbg, interface)
+            if "program_box" not in st.session_state:
+                st.session_state["program_box"] = []
+                st.session_state["cursor_position"] = len(
+                    "\n".join(st.session_state["program_box"])
+                )  # 初始化光标位置
+
+            # 插入到光标位置
+            cursor_pos = st.session_state.get(
+                "cursor_position", len("\n".join(st.session_state["program_box"]))
+            )
+            current_code = "\n".join(st.session_state["program_box"])
+            new_code = (
+                current_code[:cursor_pos]
+                + f"\nmxdbg.{method_call}\n"
+                + current_code[cursor_pos:]
+            )
+
+            # 更新程序框内容
+            st.session_state["program_box"] = new_code.split("\n")
+
+# 右侧程序框和日志框
+with col2:
+    st.subheader("程序框")
+    if "program_box" not in st.session_state:
+        st.session_state["program_box"] = []
+        st.session_state["cursor_position"] = 0
+
+    # 显示程序框内容
+    program_code = "\n".join(st.session_state["program_box"])
+    new_code = st.text_area(
+        "程序框",
+        program_code,
+        height=300,
+        key="program_box_text_area",
+        on_change=lambda: st.session_state.update(
+            {"cursor_position": len(st.session_state["program_box_text_area"])}
+        ),
+    )
+
+    # 更新程序框内容
+    if new_code != program_code:
+        st.session_state["program_box"] = new_code.split("\n")
+
+    # 删除选定的接口
+    if st.button("清空程序框"):
+        st.session_state["program_box"] = []
+
+    # 执行按钮
+    st.subheader("执行代码")
+    result_output = st.empty()  # 显示执行结果
+    if st.button("执行程序框中的代码"):
+        if mxdbg:
+            try:
+                # 重定向 stdout 捕获执行结果
+                old_stdout = sys.stdout
+                redirected_output = io.StringIO()
+                sys.stdout = redirected_output
+
+                # 执行代码（仅允许公开 API 调用）
+                import ast
+
+                code = "\n".join(st.session_state["program_box"])
+                allowed_methods = {
+                    name for name in dir(MXDBG) if not name.startswith("_")
+                }
+
+                # 用 AST 解析，确保所有 mxdbg 方法调用都在 allowlist 中
+                try:
+                    tree = ast.parse(code)
+                except SyntaxError as e:
+                    raise ValueError(f"语法错误: {e}")
+
+                for node in ast.walk(tree):
+                    if isinstance(node, ast.Call):
+                        func = node.func
+                        if (
+                            isinstance(func, ast.Attribute)
+                            and isinstance(func.value, ast.Name)
+                            and func.value.id == "mxdbg"
+                        ):
+                            if func.attr not in allowed_methods:
+                                raise ValueError(f"不允许调用 mxdbg.{func.attr}")
+                        elif isinstance(func, ast.Name) and func.id in (
+                            "exec",
+                            "eval",
+                            "compile",
+                            "__import__",
+                            "open",
+                            "breakpoint",
+                            "help",
+                            "input",
+                            "getattr",
+                            "setattr",
+                            "delattr",
+                            "type",
+                        ):
+                            raise ValueError(f"不允许调用 {func.id}()")
+                        elif isinstance(func, ast.Attribute) and func.attr.startswith(
+                            "__"
+                        ):
+                            raise ValueError(f"不允许访问 dunder 属性: {func.attr}")
+                        elif isinstance(func, ast.Attribute):
+                            # Check for chained dangerous attributes
+                            parts = []
+                            cur = func
+                            while isinstance(cur, ast.Attribute):
+                                parts.append(cur.attr)
+                                cur = cur.value
+                            if any(
+                                p.startswith("__") or p in ("builtins",) for p in parts
+                            ):
+                                raise ValueError("不允许访问私有或 builtins 属性")
+                    if (
+                        isinstance(node, ast.Call)
+                        and isinstance(node.func, ast.Attribute)
+                        and (
+                            node.func.attr.startswith("_")
+                            or (
+                                isinstance(node.func.value, ast.Attribute)
+                                and node.func.value.attr.startswith("_")
+                            )
+                        )
+                    ):
+                        raise ValueError("不允许调用私有方法")
+                    if isinstance(node, (ast.Import, ast.ImportFrom)):
+                        raise ValueError("不允许导入模块")  # noqa: TRY004 -- 安全检查拒绝，非类型错误
+
+                exec(code, {"mxdbg": mxdbg})  # noqa: S102 -- 受上方 AST 白名单约束（禁 import/私有/builtins）
+
+                # 获取执行结果并显示
+                sys.stdout = old_stdout
+                result_output.success("代码执行成功！")
+                st.text_area("执行结果", redirected_output.getvalue(), height=200)
+            except Exception as e:
+                sys.stdout = old_stdout
+                result_output.error(f"执行失败：{e}")
+        else:
+            st.warning("设备未连接，无法执行代码")
+
+    # 日志框
+    st.subheader("日志框")
+    log_output = st.empty()
+
+    # 自动刷新日志
+    if st.button("刷新日志"):
+        if mxdbg and getattr(mxdbg, "_client", None):  # 确保 _client 存在且有效
+            log_data = (
+                mxdbg._client.read(mxdbg._client.in_waiting).decode("utf-8")
+                if mxdbg._client.in_waiting > 0
+                else "无日志"
+            )
+            log_output.text(log_data)
+        else:
+            st.warning("设备未连接或 _client 无效，无法获取日志")

paw3311dw.py

@@ -0,0 +1,110 @@
+import sys
+
+import numpy as np
+from mxdbg import MXDBG
+from PySide6.QtCore import QElapsedTimer, Qt, QTimer
+from PySide6.QtGui import QImage, QPixmap
+from PySide6.QtWidgets import QApplication, QLabel, QMainWindow, QVBoxLayout, QWidget
+
+
+class MainWindow(QMainWindow):
+    def __init__(self, dev=None):
+        super().__init__()
+
+        # Set up the main window layout and label to show the image
+        self.setWindowTitle("Live Image Display")
+
+        # Image display label
+        self.image_label = QLabel(self)
+        self.image_label.setAlignment(Qt.AlignCenter)
+
+        # Frame rate display label
+        self.fps_label = QLabel(self)
+        self.fps_label.setAlignment(Qt.AlignCenter)
+        self.fps_label.setStyleSheet("font-size: 20px; color: green;")
+
+        # Set up layout and central widget
+        layout = QVBoxLayout()
+        layout.addWidget(self.image_label)
+        layout.addWidget(self.fps_label)  # Add FPS label to layout
+        central_widget = QWidget(self)
+        central_widget.setLayout(layout)
+        self.setCentralWidget(central_widget)
+
+        # Initialize MXDBG device
+        self.dev = dev if dev is not None else MXDBG()
+        ret, data = self.dev.usb_config(crc_enable=False)
+        print(ret, data)
+
+        ret, data = self.dev.spi_config(
+            freq=6000000,
+            miso_io_num=-1,
+            cs_ena_pretrans=1,
+            cs_ena_posttrans=1,
+            device_interface_flags=(
+                self.dev.spi_device["SPI_DEVICE_HALFDUPLEX"]
+                | self.dev.spi_device["SPI_DEVICE_3WIRE"]
+            ),
+        )
+        print(ret, data)
+
+        # Set up the timer for continuous updates
+        self.timer = QTimer(self)
+        self.timer.timeout.connect(self.update_image)
+        self.timer.start(100)  # Update every 100ms
+
+        # Set up the elapsed timer to calculate FPS
+        self.elapsed_timer = QElapsedTimer()
+        self.elapsed_timer.start()
+        self.frames = 0
+
+    def update_image(self):
+        """Fetch new image data and update the display."""
+        _ret, data = self.dev.spi_read_image()
+
+        if len(data) == 900:
+            # Convert the data into a numpy array and reshape it to 30x30
+            data = np.array(data)
+            data = data.reshape(30, 30)
+
+            # Convert the numpy array to QImage
+            qimage = self.numpy_to_qimage(data)
+
+            # Resize image to 300x300 using SmoothTransformation (no loss)
+            scaled_qimage = qimage.scaled(
+                300, 300, Qt.KeepAspectRatio, Qt.SmoothTransformation
+            )
+
+            # Update the QLabel with the new image
+            pixmap = QPixmap.fromImage(scaled_qimage)
+            self.image_label.setPixmap(pixmap)
+
+            # Increment frame count and update FPS every 1 second
+            self.frames += 1
+            if self.elapsed_timer.elapsed() >= 1000:  # Every second
+                fps = self.frames
+                self.fps_label.setText(f"FPS: {fps}")
+                self.frames = 0  # Reset frame count
+                self.elapsed_timer.restart()  # Restart the timer
+
+    def keyPressEvent(self, event):
+        if event.key() == Qt.Key_Q:
+            self.close()
+        super().keyPressEvent(event)
+
+    def numpy_to_qimage(self, np_array):
+        """Convert a numpy array to a QImage."""
+        height, width = np_array.shape
+        bytes_per_line = width
+        return QImage(
+            np_array.data, width, height, bytes_per_line, QImage.Format_Grayscale8
+        )
+
+
+if __name__ == "__main__":
+    # Set up the application and the main window
+    app = QApplication(sys.argv)
+    window = MainWindow()
+    window.show()
+
+    sys.exit(app.exec())