重构设备类定义, 使用继承简化代码;

source/post_work.py

@@ -1,4 +1,5 @@
 from pathlib import Path
+from device.LaminaAdapter import LaminaAdapter
 from device.LaminaAdapter import GenerateImage_SLCP001_p4a0, GeneratePackage_SLCP001_p4a0
 from device.LaminaAdapter import GenerateImage_SLCP101_p460, GeneratePackage_SLCP101_p460
 from device.LaminaAdapter import GenerateImage_SLCP102_p460, GeneratePackage_SLCP102_p460
@@ -198,6 +199,51 @@ def Process2():
     file_image3.write_bytes(data_image)
 
 
+def Process(type_dev: str, path_boot: Path, path_project: Path):
+    """ 镜像生成流程 """
+    root_boot = path_boot
+    root_main = path_project
+    result = root_main
+
+    # 正常启动镜像
+    hex_boot = root_boot / "bootloader.hex"
+    hex_main = root_main / "lamina_adapter.hex"
+    hex_back = root_main / "lamina_adapter_back.hex"
+    hex_update = hex_main
+    if (not hex_boot.exists()) or (not hex_main.exists()) or (not hex_back.exists()):
+        raise Exception("缺失必要程序文件")
+
+    file_image = result / f'{hex_main.stem}_ROM.bin'
+    file_package = result / f'{hex_update.stem}.dat'
+
+    dev_lamina = LaminaAdapter(None, type_dev=type_dev)
+
+    data_package = dev_lamina.make_package(hex_update)
+    data_image = dev_lamina.make_image(hex_boot, hex_main, hex_back, output_header=True, output_bin=True)
+
+    file_image.write_bytes(data_image)
+    file_package.write_bytes(data_package)
+
+    # 异常镜像-主分区md5错误
+    file_image1 = result / f'{file_image.stem}_b1.bin'
+    data_image_copy = data_image
+    data_image_copy[0x054018: 0x05401A] = [0x00, 0x01]
+    file_image1.write_bytes(data_image_copy)
+
+    # 异常镜像-备份分区md5错误
+    file_image2 = result / f'{file_image.stem}_b2.bin'
+    data_image_copy = data_image
+    data_image_copy[0x056018: 0x05601A] = [0x00, 0x01]
+    file_image2.write_bytes(data_image_copy)
+
+    # 异常镜像-双分区md5错误
+    file_image3 = result / f'{file_image.stem}_b3.bin'
+    data_image_copy = data_image
+    data_image_copy[0x054018: 0x05401A] = [0x00, 0x01]
+    data_image_copy[0x056018: 0x05601A] = [0x00, 0x01]
+    file_image3.write_bytes(data_image_copy)
+
+
 if __name__ == "__main__":
     path_boot1 =  Path(r"D:\WorkingProject\LightStackAdapter\software\umon\4A0-PROJ_STACKLIGHT_PARALLEL_ADAPTOR")
     path_boot2 =  Path(r"D:\WorkingProject\LightStackAdapter\software\umon\460-PROJ_STACKLIGHT_PARALLEL_ADAPTOR")
@@ -207,3 +253,8 @@ if __name__ == "__main__":
     Process1(path_boot2, path_main)         # 适配器SLCP101
     # Process1_v2(path_boot2, path_main)      # 适配器SLCP102
     # Process2()
+
+    path_boot1 = Path(r"resource/460-PROJ_STACKLIGHT_OPTIMIZER")
+    path_boot2 = Path(r"resource/460-PROJ_STACKLIGHT_PARALLEL_ADAPTOR")
+    path_project = Path(r"resource\SLCP101")
+    Process('SLCP101', path_boot2, path_project)