更新UCD-API库及文档

UniTAP/dev/dev_4xx_roles.py

@@ -1,6 +1,7 @@
 from UniTAP.dev.ports import DPRX4xx, DPTX4xx, HDRX4xx, HDTX4xx, PDC424
 from .modules import MemoryManager, Capturer, DUTTests
-from UniTAP.libs.lib_tsi.tsi_io import DeviceIO, PortProtocol
+from UniTAP.libs.lib_tsi.tsi_io import PortProtocol
+from .dev_base_roles import DevBase, DeviceIO
 
 
 class UCD400:
@@ -8,7 +9,7 @@ class UCD400:
     Class `UCD400` describes of device UCD-400. Device has one possible role:
     - 'DisplayPort Source and Sink' `DPSourceDPSink`
     """
-    class DPSourceDPSink:
+    class DPSourceDPSink(DevBase):
         """
         Class `DPSourceDPSink` contains information of available functionality modules for role DP Sink
         (RX - receiver) and DP Source (TX - transmitter) roles:
@@ -17,7 +18,7 @@ class UCD400:
         - `DUTTests`.
         """
         def __init__(self, dev_io: DeviceIO, memory_manager: MemoryManager, capturer: Capturer):
-            super().__init__()
+            super().__init__(dev_io)
             self.__dprx = DPRX4xx(dev_io.create_port_io(0, PortProtocol.DisplayPort), memory_manager, capturer)
             self.__dptx = DPTX4xx(dev_io.create_port_io(1, PortProtocol.DisplayPort), memory_manager, capturer)
             self.__dut_tests = DUTTests(dev_io)
@@ -72,7 +73,7 @@ class UCD422:
     Class `UCD422` describes of device UCD-422. Device has one possible role:
     - 'HDMI Source and Sink' `HDMISourceHDMISink`
     """
-    class HDMISourceHDMISink:
+    class HDMISourceHDMISink(DevBase):
         """
         Class `DPSourceDPSink` contains information of available functionality modules for role HDMI Sink
         (RX - receiver) and HDMI Source (TX - transmitter) roles:
@@ -81,7 +82,7 @@ class UCD422:
         - `DUTTests`.
         """
         def __init__(self, dev_io: DeviceIO, memory_manager: MemoryManager, capturer: Capturer):
-            super().__init__()
+            super().__init__(dev_io)
             self.__hdrx = HDRX4xx(dev_io.create_port_io(0, PortProtocol.HDMI), memory_manager, capturer)
             self.__hdtx = HDTX4xx(dev_io.create_port_io(1, PortProtocol.HDMI), memory_manager, capturer)
             self.__dut_tests = DUTTests(dev_io)
@@ -136,7 +137,7 @@ class UCD424:
     Class `UCD424` describes of device UCD-424. Device has one possible role:
     - 'USB-C Source and Sink' `USBCSourceUSBCSink`.
     """
-    class USBCSourceUSBCSink:
+    class USBCSourceUSBCSink(DevBase):
         """
         Class `DPSourceDPSink` contains information of available functionality modules for role HDMI Sink
         (RX - receiver) and HDMI Source (TX - transmitter) roles:
@@ -146,7 +147,7 @@ class UCD424:
         - `PDC`.
         """
         def __init__(self, dev_io: DeviceIO, memory_manager: MemoryManager, capturer: Capturer):
-            super().__init__()
+            super().__init__(dev_io)
             self.__port_rx = dev_io.create_port_io(0, PortProtocol.DisplayPortThrowUSBC)
             self.__port_tx = dev_io.create_port_io(1, PortProtocol.DisplayPortThrowUSBC)
             self.__dprx = DPRX4xx(self.__port_rx, memory_manager, capturer)
@@ -218,3 +219,237 @@ class UCD424:
     ROLE_DICT = {
         "USB-C Source and Sink": USBCSourceUSBCSink
     }
+
+
+class UCD451:
+    """
+    Class `UCD451` describes of device UCD-451. Device has one possible role:
+    - 'DP Source' `DPSource`
+    """
+
+    class DPSource(DevBase):
+        """
+        Class `DPSourceDPSink` contains information of available functionality modules for role
+        DP Source (TX - transmitter) roles:
+        - `DPTX4xx`.
+        - `DUTTests`.
+        """
+
+        def __init__(self, dev_io: DeviceIO, memory_manager: MemoryManager, capturer: Capturer):
+            super().__init__(dev_io)
+            self.__dptx = DPTX4xx(dev_io.create_port_io(0, PortProtocol.DisplayPort), memory_manager, capturer)
+            self.__dut_tests = DUTTests(dev_io)
+
+        @property
+        def dptx(self) -> DPTX4xx:
+            """
+            Returns DP Source (TX - transmitter) role.
+
+            Returns:
+                object of DPTX4xx 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(UCD451.ROLE_DICT.keys())[list(UCD451.ROLE_DICT.values()).index(type(self))]
+
+    ROLE_DICT = {
+        "DisplayPort Source": DPSource
+    }
+
+
+class UCD461:
+    """
+    Class `UCD461` describes of device UCD-461. Device has one possible role:
+    - 'DisplayPort Source and Sink' `DPSourceDPSink`
+    """
+    class DPSourceDPSink(DevBase):
+        """
+        Class `DPSourceDPSink` contains information of available functionality modules for role DP Sink
+        (RX - receiver) and DP Source (TX - transmitter) roles:
+        - `DPRX4xx`.
+        - `DPTX4xx`.
+        - `DUTTests`.
+        """
+        def __init__(self, dev_io: DeviceIO, memory_manager: MemoryManager, capturer: Capturer):
+            super().__init__(dev_io)
+            self.__dprx = DPRX4xx(dev_io.create_port_io(0, PortProtocol.DisplayPort), memory_manager, capturer)
+            self.__dptx = DPTX4xx(dev_io.create_port_io(1, PortProtocol.DisplayPort), memory_manager, capturer)
+            self.__dut_tests = DUTTests(dev_io)
+
+        @property
+        def dprx(self) -> DPRX4xx:
+            """
+            Returns DP Sink (RX - receiver) role.
+
+            Returns:
+                object of DPRX4xx type.
+            """
+            return self.__dprx
+
+        @property
+        def dptx(self) -> DPTX4xx:
+            """
+            Returns DP Source (TX - transmitter) role.
+
+            Returns:
+                object of DPTX4xx 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(UCD461.ROLE_DICT.keys())[list(UCD461.ROLE_DICT.values()).index(type(self))]
+
+    ROLE_DICT = {
+        "DisplayPort Source and Sink": DPSourceDPSink
+    }
+
+
+class UCD452:
+    """
+    Class `UCD452` describes of device UCD-452. Device has one possible role:
+    - 'HDMI Source' `HDMISource`
+    """
+    class HDMISource(DevBase):
+        """
+        Class `DPSourceDPSink` contains information of available functionality modules for role
+        HDMI Source (TX - transmitter) roles:
+        - `HDTX4xx`.
+        - `DUTTests`.
+        """
+        def __init__(self, dev_io: DeviceIO, memory_manager: MemoryManager, capturer: Capturer):
+            super().__init__(dev_io)
+            self.__hdtx = HDTX4xx(dev_io.create_port_io(0, PortProtocol.HDMI), memory_manager, capturer)
+            self.__dut_tests = DUTTests(dev_io)
+
+        @property
+        def hdtx(self) -> HDTX4xx:
+            """
+            Returns HDMI Source (TX - transmitter) role.
+
+            Returns:
+                object of HDTX4xx type.
+            """
+            return self.__hdtx
+
+        @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(UCD452.ROLE_DICT.keys())[list(UCD452.ROLE_DICT.values()).index(type(self))]
+
+    ROLE_DICT = {
+        "HDMI Source": HDMISource
+    }
+
+
+class UCD462:
+    """
+    Class `UCD462` describes of device UCD-462. Device has one possible role:
+    - 'HDMI Source and Sink' `HDMISourceHDMISink`
+    """
+    class HDMISourceHDMISink(DevBase):
+        """
+        Class `DPSourceDPSink` contains information of available functionality modules for role HDMI Sink
+        (RX - receiver) and HDMI Source (TX - transmitter) roles:
+        - `HDRX4xx`.
+        - `HDTX4xx`.
+        - `DUTTests`.
+        """
+        def __init__(self, dev_io: DeviceIO, memory_manager: MemoryManager, capturer: Capturer):
+            super().__init__(dev_io)
+            self.__hdrx = HDRX4xx(dev_io.create_port_io(0, PortProtocol.HDMI), memory_manager, capturer)
+            self.__hdtx = HDTX4xx(dev_io.create_port_io(1, PortProtocol.HDMI), memory_manager, capturer)
+            self.__dut_tests = DUTTests(dev_io)
+
+        @property
+        def hdtx(self) -> HDTX4xx:
+            """
+            Returns HDMI Source (TX - transmitter) role.
+
+            Returns:
+                object of HDTX4xx type.
+            """
+            return self.__hdtx
+
+        @property
+        def hdrx(self) -> HDRX4xx:
+            """
+            Returns HDMI Sink (RX - receiver) role.
+
+            Returns:
+                object of HDRX4xx type.
+            """
+            return self.__hdrx
+
+        @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(UCD462.ROLE_DICT.keys())[list(UCD462.ROLE_DICT.values()).index(type(self))]
+
+    ROLE_DICT = {
+        "HDMI Source and Sink": HDMISourceHDMISink
+    }