1.1.0版本

2026-04-16 16:51:05 +08:00
commit c157e774e5
333 changed files with 70759 additions and 0 deletions
--- a/UniTAP/dev/dev_5xx_roles.py
+++ b/UniTAP/dev/dev_5xx_roles.py
@@ -0,0 +1,258 @@
+from UniTAP.dev.ports import DPRX5xx, DPTX5xx, PDC500
+from .modules import MemoryManager, Capturer, DUTTests
+from UniTAP.libs.lib_tsi.tsi_io import DeviceIO, PortProtocol
+
+
+class UCD500:
+    """
+    Class `UCD500` describes of device UCD-500. Device has one possible role:
+    - 'DisplayPort Source and Sink' `DPSourceDPSink`.
+    - 'DisplayPort Source and USB-C, DP Alt Mode Sink' `DPSourceUSBCSink`.
+    - 'DisplayPort Sink and USB-C, DP Alt Mode Source' `USBCSourceDPSink`.
+    - 'USB-C, DP Alt Mode Source and Sink' `USBCSourceUSBCSink`.
+    """
+    class DPSourceDPSink:
+        """
+        Class `DPSourceDPSink` contains information of available functionality modules for role DP Sink
+        (RX - receiver) and DP Source (TX - transmitter) roles:
+        - `DPRX5xx`.
+        - `DPTX5xx`.
+        - `DUTTests`.
+        """
+        def __init__(self, dev_io: DeviceIO, memory_manager: MemoryManager, capturer: Capturer):
+            self.__dprx = DPRX5xx(dev_io.create_port_io(0, PortProtocol.DisplayPort), memory_manager, capturer)
+            self.__dptx = DPTX5xx(dev_io.create_port_io(1, PortProtocol.DisplayPort), memory_manager, capturer)
+            self.__dut_tests = DUTTests(dev_io)
+
+        @property
+        def dprx(self) -> DPRX5xx:
+            """
+            Returns DP Sink (RX - receiver) role.
+
+            Returns:
+                object of DPRX5xx type.
+            """
+            return self.__dprx
+
+        @property
+        def dptx(self) -> DPTX5xx:
+            """
+            Returns DP Source (TX - transmitter) role.
+
+            Returns:
+                object of DPTX5xx type.
+            """
+            return self.__dptx
+
+        @property
+        def dut_tests(self) -> DUTTests:
+            """
+            Returns DUT Test module.
+
+            Returns:
+                object of DUTTests type.
+            """
+            return self.__dut_tests
+
+        @property
+        def name(self) -> str:
+            """
+            Returns name of role.
+
+            Returns:
+                object of str type.
+            """
+            return list(UCD500.ROLE_DICT.keys())[list(UCD500.ROLE_DICT.values()).index(type(self))]
+
+    class DPSourceUSBCSink:
+        def __init__(self, dev_io: DeviceIO, memory_manager: MemoryManager, capturer: Capturer):
+            self.__port_rx = dev_io.create_port_io(0, PortProtocol.DisplayPortThrowUSBC)
+            self.__dptx = DPTX5xx(dev_io.create_port_io(1, PortProtocol.DisplayPort), memory_manager, capturer)
+            self.__dprx = DPRX5xx(self.__port_rx, memory_manager, capturer)
+            self.__pdcrx = PDC500(self.__port_rx)
+            self.__dut_tests = DUTTests(dev_io)
+
+        @property
+        def dprx(self) -> DPRX5xx:
+            """
+            Returns DP Sink (RX - receiver) role.
+
+            Returns:
+                object of DPRX5xx type.
+            """
+            return self.__dprx
+
+        @property
+        def dptx(self) -> DPTX5xx:
+            """
+            Returns DP Source (TX - transmitter) role.
+
+            Returns:
+                object of DPTX5xx type.
+            """
+            return self.__dptx
+
+        @property
+        def pdcrx(self) -> PDC500:
+            """
+            Returns PDC Sink (RX - receiver) role.
+
+            Returns:
+                object of `PDC500` type.
+            """
+            return self.__pdcrx
+
+        @property
+        def dut_tests(self) -> DUTTests:
+            """
+            Returns DUT Test module.
+
+            Returns:
+                object of DUTTests type.
+            """
+            return self.__dut_tests
+
+        @property
+        def name(self) -> str:
+            """
+            Returns name of role.
+
+            Returns:
+                object of str type.
+            """
+            return list(UCD500.ROLE_DICT.keys())[list(UCD500.ROLE_DICT.values()).index(type(self))]
+
+    class USBCSourceDPSink:
+        def __init__(self, dev_io: DeviceIO, memory_manager: MemoryManager, capturer: Capturer):
+            self.__port_tx = dev_io.create_port_io(1, PortProtocol.DisplayPortThrowUSBC)
+            self.__dptx = DPTX5xx(self.__port_tx, memory_manager, capturer)
+            self.__dprx = DPRX5xx(dev_io.create_port_io(0, PortProtocol.DisplayPort), memory_manager, capturer)
+            self.__pdctx = PDC500(self.__port_tx)
+            self.__dut_tests = DUTTests(dev_io)
+
+        @property
+        def dprx(self) -> DPRX5xx:
+            """
+            Returns DP Sink (RX - receiver) role.
+
+            Returns:
+                object of DPRX5xx type.
+            """
+            return self.__dprx
+
+        @property
+        def dptx(self) -> DPTX5xx:
+            """
+            Returns DP Source (TX - transmitter) role.
+
+            Returns:
+                object of DPTX5xx type.
+            """
+            return self.__dptx
+
+        @property
+        def pdctx(self) -> PDC500:
+            """
+            Returns PDC source (TX - transmitter) role.
+
+            Returns:
+                object of `PDC500` type.
+            """
+            return self.__pdctx
+
+        @property
+        def dut_tests(self) -> DUTTests:
+            """
+            Returns DUT Test module.
+
+            Returns:
+                object of DUTTests type.
+            """
+            return self.__dut_tests
+
+        @property
+        def name(self) -> str:
+            """
+            Returns name of role.
+
+            Returns:
+                object of str type.
+            """
+            return list(UCD500.ROLE_DICT.keys())[list(UCD500.ROLE_DICT.values()).index(type(self))]
+
+    class USBCSourceUSBCSink:
+        def __init__(self, dev_io: DeviceIO, memory_manager: MemoryManager, capturer: Capturer):
+            self.__port_rx = dev_io.create_port_io(0, PortProtocol.DisplayPortThrowUSBC)
+            self.__port_tx = dev_io.create_port_io(1, PortProtocol.DisplayPortThrowUSBC)
+            self.__dprx = DPRX5xx(self.__port_rx, memory_manager, capturer)
+            self.__dptx = DPTX5xx(self.__port_tx, memory_manager, capturer)
+            self.__pdcrx = PDC500(self.__port_rx)
+            self.__pdctx = PDC500(self.__port_tx)
+            self.__dut_tests = DUTTests(dev_io)
+
+        @property
+        def dprx(self) -> DPRX5xx:
+            """
+            Returns DP Sink (RX - receiver) role.
+
+            Returns:
+                object of DPRX5xx type.
+            """
+            return self.__dprx
+
+        @property
+        def dptx(self) -> DPTX5xx:
+            """
+            Returns DP Source (TX - transmitter) role.
+
+            Returns:
+                object of DPTX5xx type.
+            """
+            return self.__dptx
+
+        @property
+        def pdcrx(self) -> PDC500:
+            """
+            Returns PDC Sink (RX - receiver) role.
+
+            Returns:
+                object of `PDC500` type.
+            """
+            return self.__pdcrx
+
+        @property
+        def pdctx(self) -> PDC500:
+            """
+            Returns PDC source (TX - transmitter) role.
+
+            Returns:
+                object of `PDC500` type.
+            """
+            return self.__pdctx
+
+        @property
+        def dut_tests(self) -> DUTTests:
+            """
+            Returns DUT Test module.
+
+            Returns:
+                object of DUTTests type.
+            """
+            return self.__dut_tests
+
+        @property
+        def name(self) -> str:
+            """
+            Returns name of role.
+
+            Returns:
+                object of str type.
+            """
+            return list(UCD500.ROLE_DICT.keys())[list(UCD500.ROLE_DICT.values()).index(type(self))]
+
+    ROLE_DICT = {
+        "DisplayPort Source and Sink": DPSourceDPSink,
+        "DisplayPort Source and USB-C, DP Alt Mode Sink": DPSourceUSBCSink,
+        "DisplayPort Sink and USB-C, DP Alt Mode Source": USBCSourceDPSink,
+        "USB-C, DP Alt Mode Source and Sink": USBCSourceUSBCSink
+    }