1.1.0版本

UniTAP/dev/ports/modules/pdc/pdc_utils.py

@@ -0,0 +1,72 @@
+import warnings
+import os
+import json
+
+from UniTAP.dev.ports.modules.pdc.pdo import Pdo
+from UniTAP.dev.ports.modules.pdc.pdo_utils import get_pdo_value
+from UniTAP.dev.ports.modules.pdc.pdo_types import *
+from UniTAP.dev.ports.modules.pdc.pdo_private_types import PdoUnion
+
+
+def support_or_not(value) -> str:
+    return "Yes" if bool(value) else "No"
+
+
+def save_pdo(path: str, pdo_list: list, source: bool):
+    filename, file_extension = os.path.splitext(path)
+    if file_extension.find('.txt') != -1:
+        file = open(path, 'w')
+        file.write("TSI_USBC_PWR_LOCAL_SOURCE_PDO\n" if source else "TSI_USBC_PWR_LOCAL_SINK_PDO\n")
+        file.write("1\n")
+        file.write(str(len(pdo_list) * 4))
+        for pdo in pdo_list:
+            file.write(str(get_pdo_value(pdo.get_pdo_object())))
+        file.close()
+    elif file_extension.find('.json') != -1:
+        values = {}
+        for index, pdo in enumerate(pdo_list):
+            values.update({f"EPR{index}": get_pdo_value(pdo.get_pdo_object())})
+        with open(path, "w") as outfile:
+            json.dump(values, outfile)
+    else:
+        warnings.warn(f"Unsupported file format: {file_extension}. "
+                      f"Please, select file format from available variants: 'txt', 'json'")
+
+
+def load_pdo(path: str, source: bool) -> list:
+    if os.path.exists(path):
+        raise ValueError(f"Incorrect path {path} to file. Path does not exist.")
+    filename, file_extension = os.path.splitext(path)
+    pdo_list = []
+    if file_extension.find('.txt') != -1:
+        file = open(path, 'r')
+        data = file.read().split("\n")
+        for index, line in enumerate(data):
+            if index >= 3:
+                fill_list(int(line), pdo_list, source)
+        file.close()
+    elif file_extension.find('.json') != -1:
+        with open(path, "r", encoding='utf-8') as read_file:
+            data = json.load(read_file)
+        keys = list(data.keys())
+        for index, key in enumerate(keys):
+            fill_list(int(data.get(key)), pdo_list, source)
+    else:
+        raise ValueError(f"Unsupported file format: {file_extension}. "
+                         f"Please, select file format from available variants: 'txt', 'json'")
+    return pdo_list
+
+
+def fill_list(data, pdo_list, source):
+    pdo_data = PdoUnion(data)
+    pdo_type = PdoTypeEnum(pdo_data.brief_info.pdo_type)
+    if pdo_type == PdoTypeEnum.Fixed:
+        pdo_list.append(Pdo(pdo=FixedPdoSource(pdo_data.fixed_pdo_source) if source else
+                            FixedPdoSink(pdo_data.fixed_pdo_sink),
+                            side=PdoSide.Source if source else PdoSide.Sink))
+    elif pdo_type == PdoTypeEnum.Battery:
+        pdo_list.append(Pdo(pdo=BatteryPdo(pdo_data.battery_pdo),
+                            side=PdoSide.Source if source else PdoSide.Sink))
+    elif pdo_type == PdoTypeEnum.Variable:
+        pdo_list.append(Pdo(pdo=VariablePdo(pdo_data.variable_pdo),
+                            side=PdoSide.Source if source else PdoSide.Sink))