MaaFw 2.1.0b1__py3-none-win_arm64.whl → 2.1.0b2__py3-none-win_arm64.whl

{MaaFw-2.1.0b1.dist-info → MaaFw-2.1.0b2.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: MaaFw
-Version: 2.1.0b1
+Version: 2.1.0b2
 Summary: An automation black-box testing framework based on image recognition
 Author: MaaXYZ
 Project-URL: Homepage, https://github.com/MaaXYZ/MaaFramework

{MaaFw-2.1.0b1.dist-info → MaaFw-2.1.0b2.dist-info}/RECORD

@@ -4,25 +4,25 @@ maa/context.py,sha256=Y71T92tIAGq6x8bJfYX8emUPEmPUA7nhnWgqOHh390E,5722
 maa/controller.py,sha256=BpAotY8qlxLSKU0ZO249YDhJWU35B5avnwCTE5PxItQ,22160
 maa/custom_action.py,sha256=pqPjDK7nVkGL8fceJadwb7IkeGjYc2X8RaXZOnVZXQs,2527
 maa/custom_recognition.py,sha256=Y3Asdgb19YZioOVZZVs_hLSW6uBg7WDZDQPR2XfZpFU,3087
-maa/define.py,sha256=pYgO5_djp0TBP2sq9SUYb_I-sNjJ2Uv6dfdgVxq2hy0,10745
+maa/define.py,sha256=nFPEOa7asyf4UG9kkeQrQaarXWbN0w-tSHoQWdZi_4s,11032
 maa/job.py,sha256=Vj6q3sFPlmo260mxJWBnPuO2oT2LdZ-92vdDvrqBZYU,1777
 maa/library.py,sha256=nDTBUFC6_91gmyy8SfkTTccMZET9XGU8IxvuVUEwQ6E,2099
 maa/notification_handler.py,sha256=HtGmTFgTj7ibTdAIVNr5-LBn38trv_j6beHvYXUVFeg,5172
-maa/resource.py,sha256=9BHZ6XCLTomfvQAO6xv272DH364r3sl8zMGDXJOQ_HU,7913
+maa/resource.py,sha256=bvS02sCCS9DRw_zJZeyCk8_DrZry6KBlUIvcfpKoUIg,8243
 maa/tasker.py,sha256=6FOzU-PxWSA0dK8vEmiPVaJThvUMwNDi6EC6V-TcQmM,14758
 maa/toolkit.py,sha256=TlF8MoaGcWLklpISIlQNvc_rpTFBFrpDbLgctfQ_zqo,11325
 maa/bin/DirectML.dll,sha256=OZzXLdjebjwHxY63t7obvESKdwuE0E_D8DHGZC2wPp4,27307440
-maa/bin/MaaAdbControlUnit.dll,sha256=kzhqDzPmOzKJ_Inxw99QCmNmQEyGUYOP3m_ZDVst76w,861184
-maa/bin/MaaDbgControlUnit.dll,sha256=vKY9OgUaR2c54KsFu_Xxqh-DE7eWo9LP0KN0HCUmA28,444416
-maa/bin/MaaFramework.dll,sha256=u6Mfv7ZTkPApb0H5sQPVMaUNhEZFi7ql7NTHZyDnScQ,1626112
-maa/bin/MaaToolkit.dll,sha256=ySPBkeuw8x-vFuiR15Jxsj24GjwpuBZweHyZcPXKNuE,730624
-maa/bin/MaaUtils.dll,sha256=gEgc5Dbq4D3XNM_LfQ86ff3WwCNJKDxQ0iDXmXfd_fA,555008
-maa/bin/MaaWin32ControlUnit.dll,sha256=4HSJ8R0tb1jCfHrYjBangNM-y-ihhMgjij4gk7T5ywA,399872
+maa/bin/MaaAdbControlUnit.dll,sha256=XhUBk2vejFgXdSEyYMhgJuIa5QFOiB3tswh6Wzkf2Xw,871936
+maa/bin/MaaDbgControlUnit.dll,sha256=wdEghIjW-352DUo4OjChDrFby0nScO3AkSRXvE-osrM,444416
+maa/bin/MaaFramework.dll,sha256=RwPUQEJJPivmeh91VoGYnLckLgdkCKRO9rT3BE4rWvI,1626624
+maa/bin/MaaToolkit.dll,sha256=zrqofz1ubmgw9A1OyO2O_eJ5u3ph2cBjdSHeVSxkb14,730624
+maa/bin/MaaUtils.dll,sha256=wJhFEakokVyKACIU2ULrdD8o3mkLSeP_mxvZ2v5ZsDA,555008
+maa/bin/MaaWin32ControlUnit.dll,sha256=FamIbnQjtxcvi_zmdoBYROnEADHSUBbtWTVtSWqKBrU,399872
 maa/bin/fastdeploy_ppocr_maa.dll,sha256=-noAb-Sf_YeQW9P9X-oWvD7FeySSCiEJi98X3uqkR3E,554496
 maa/bin/onnxruntime_maa.dll,sha256=z1h6QyRTL5EtFS1Rc4TGBJGXUW_6LaVQ3yVhpz2B4mg,14457856
 maa/bin/opencv_world4_maa.dll,sha256=n56ncTibRBFGMSZMXyEKXZ_aAkMiTAhGl8wjQnU4CJY,14959104
-MaaFw-2.1.0b1.dist-info/LICENSE.md,sha256=RG51X65V_wNLuyG-RGcLXxFsKyZnlH5wNvK_5mMlOag,7560
-MaaFw-2.1.0b1.dist-info/METADATA,sha256=7zB3Lo59K4fjZDHekp7Oe9PBijw7RcnUD3QY2YqqGqE,9430
-MaaFw-2.1.0b1.dist-info/WHEEL,sha256=Duyuyl5UQT6_Wtko3k3gVoeKjXVfTFS3DpEnxomAcIY,95
-MaaFw-2.1.0b1.dist-info/top_level.txt,sha256=G8r2L__aYaMq4S3R70G6NsfEROiVp08kUcWvATzk2J8,4
-MaaFw-2.1.0b1.dist-info/RECORD,,
+MaaFw-2.1.0b2.dist-info/LICENSE.md,sha256=RG51X65V_wNLuyG-RGcLXxFsKyZnlH5wNvK_5mMlOag,7560
+MaaFw-2.1.0b2.dist-info/METADATA,sha256=Mjl4tQaNLNTD7mh1GiJaYHE_MgUcejnuzuutaN8tUNs,9430
+MaaFw-2.1.0b2.dist-info/WHEEL,sha256=Duyuyl5UQT6_Wtko3k3gVoeKjXVfTFS3DpEnxomAcIY,95
+MaaFw-2.1.0b2.dist-info/top_level.txt,sha256=G8r2L__aYaMq4S3R70G6NsfEROiVp08kUcWvATzk2J8,4
+MaaFw-2.1.0b2.dist-info/RECORD,,

maa/define.py

@@ -105,13 +105,25 @@ class MaaCtrlOptionEnum:
     Recording = 5
 
 
+# enum MaaInferenceDeviceEnum
+# {
+#     MaaInferenceDevice_CPU = -2,
+#     MaaInferenceDevice_Auto = -1,
+#     MaaInferenceDevice_GPU0 = 0,
+#     MaaInferenceDevice_GPU1 = 1,
+#     // and more gpu id...
+# };
+MaaInferenceDevice = ctypes.c_int32
+
 class MaaResOptionEnum:
     Invalid = 0
 
-    # Use the specified inference device, the default is INT32_MAX, which means CPU.
+    # Use the specified inference device.
     # Please set this option before loading the model.
-    # value: int32_t, eg: 0; val_size: sizeof(int32_t)
-    GpuId = 1
+    #
+    # value: MaaInferenceDevice, eg: 0; val_size: sizeof(MaaInferenceDevice)
+    # default value is MaaInferenceDevice_Auto
+    InferenceDevice = 1
 
 
 MaaAdbScreencapMethod = ctypes.c_uint64

maa/resource.py

@@ -60,20 +60,18 @@ class Resource:
     def clear(self) -> bool:
         return bool(Library.framework.MaaResourceClear(self._handle))
 
-    def set_gpu(self, device_id: int) -> bool:
-        cint = ctypes.c_int32(device_id)
-        return bool(
-            Library.framework.MaaResourceSetOption(
-                self._handle,
-                MaaResOptionEnum.GpuId,
-                ctypes.pointer(cint),
-                ctypes.sizeof(ctypes.c_int32),
-            )
-        )
+    def set_gpu(self, gpu_id: int) -> bool:
+        if gpu_id < 0:
+            return False
+        return self.set_inference_device(gpu_id)
 
     def set_cpu(self) -> bool:
-        INT32_MAX = 2147483647  # means CPU
-        return self.set_gpu(INT32_MAX)
+        MaaInferenceDevice_CPU = -2
+        return self.set_inference_device(MaaInferenceDevice_CPU)
+    
+    def set_auto_device(self) -> bool:
+        MaaInferenceDevice_Auto = -1
+        return self.set_inference_device(MaaInferenceDevice_Auto)
 
     def register_custom_recognition(
         self, name: str, recognition: "CustomRecognition"  # type: ignore
@@ -151,6 +149,17 @@ class Resource:
 
     ### private ###
 
+    def set_inference_device(self, device_id: int) -> bool:
+        cint = ctypes.c_int32(device_id)
+        return bool(
+            Library.framework.MaaResourceSetOption(
+                self._handle,
+                MaaResOptionEnum.InferenceDevice,
+                ctypes.pointer(cint),
+                ctypes.sizeof(ctypes.c_int32),
+            )
+        )
+
     def _status(self, id: int) -> ctypes.c_int32:
         return Library.framework.MaaResourceStatus(self._handle, id)