MaaFw 2.0.3__py3-none-win_arm64.whl → 2.1.0a1__py3-none-win_arm64.whl

{MaaFw-2.0.3.dist-info → MaaFw-2.1.0a1.dist-info}/METADATA

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

{MaaFw-2.0.3.dist-info → MaaFw-2.1.0a1.dist-info}/RECORD

@@ -4,25 +4,25 @@ maa/context.py,sha256=Y71T92tIAGq6x8bJfYX8emUPEmPUA7nhnWgqOHh390E,5722
 maa/controller.py,sha256=F7BV221e0lPCElRD89mBJGJvXhNFe_QhvhV3yWSnKoY,22167
 maa/custom_action.py,sha256=pqPjDK7nVkGL8fceJadwb7IkeGjYc2X8RaXZOnVZXQs,2527
 maa/custom_recognition.py,sha256=Y3Asdgb19YZioOVZZVs_hLSW6uBg7WDZDQPR2XfZpFU,3087
-maa/define.py,sha256=cCqrFo4I4Ije1Ebs2ylz_esa2BQVPXz3mkKnzX6Ou0A,10468
+maa/define.py,sha256=pYgO5_djp0TBP2sq9SUYb_I-sNjJ2Uv6dfdgVxq2hy0,10745
 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=SL0P0A4mmV8IbTMH36tP_AwjzGlCG8-SwLI4NOe30Ao,7204
+maa/resource.py,sha256=9BHZ6XCLTomfvQAO6xv272DH364r3sl8zMGDXJOQ_HU,7913
 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=rrWqJzzW_PIMM-s6-_p_Knr4SPyrAXsGGM_7veQH5c8,861184
-maa/bin/MaaDbgControlUnit.dll,sha256=hJii-1SMH6mggM80CkEH_LNKEnbdptpl70Lp-wwtHB0,444416
-maa/bin/MaaFramework.dll,sha256=uK1YPOBMSixg1lbo_-M7bID2WFVXR7hHgSEm5OIQyUM,1594880
-maa/bin/MaaToolkit.dll,sha256=1-eoi26a9eRJNg3RJ7V5eJEl-MqPVBeuyMYhdt-tAaI,728576
-maa/bin/MaaUtils.dll,sha256=2S3B5wIA0Rtoj-vwU8Pe5e7pWYACZSAKBQM47AIABJg,555008
-maa/bin/MaaWin32ControlUnit.dll,sha256=CgNMSApI1fSJ2I_QDdLaJ2PdaEokEr1jR8w8-jH8rb8,399872
+maa/bin/MaaAdbControlUnit.dll,sha256=yD0BhlZytANATMisJY3Jtjr0GJmnknZaoVkVK5b7M0k,861184
+maa/bin/MaaDbgControlUnit.dll,sha256=bBxq1QUgbOEIi99NjEw1s8SSaHVt1_lHGjx_cDDZSOA,444416
+maa/bin/MaaFramework.dll,sha256=goTM2R-gB9UWv0NoPxx091o9ZsYa0St3dySDxr7qLp0,1618944
+maa/bin/MaaToolkit.dll,sha256=rGkO0Bj4vVtZNSnjdgLb721_l4OlIZEnVA5Vdet9zco,730624
+maa/bin/MaaUtils.dll,sha256=UUapPYAuhCSTMTL65uYidQt7pb7cAoXXh6k-X768k7w,555008
+maa/bin/MaaWin32ControlUnit.dll,sha256=5LGIyNZHNNVpDkXe0bPSzNT9F_FlwqWo6sVsw3-LjeM,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.0.3.dist-info/LICENSE.md,sha256=RG51X65V_wNLuyG-RGcLXxFsKyZnlH5wNvK_5mMlOag,7560
-MaaFw-2.0.3.dist-info/METADATA,sha256=y81HFpAvDLTZT0u0cVph_Eow21A84EMgVOmHNc6PdfE,9428
-MaaFw-2.0.3.dist-info/WHEEL,sha256=Duyuyl5UQT6_Wtko3k3gVoeKjXVfTFS3DpEnxomAcIY,95
-MaaFw-2.0.3.dist-info/top_level.txt,sha256=G8r2L__aYaMq4S3R70G6NsfEROiVp08kUcWvATzk2J8,4
-MaaFw-2.0.3.dist-info/RECORD,,
+MaaFw-2.1.0a1.dist-info/LICENSE.md,sha256=RG51X65V_wNLuyG-RGcLXxFsKyZnlH5wNvK_5mMlOag,7560
+MaaFw-2.1.0a1.dist-info/METADATA,sha256=Q6B5oFjQjZ28El9MPhOoLrWnycfOihUHcWUfXz4RXLk,9430
+MaaFw-2.1.0a1.dist-info/WHEEL,sha256=Duyuyl5UQT6_Wtko3k3gVoeKjXVfTFS3DpEnxomAcIY,95
+MaaFw-2.1.0a1.dist-info/top_level.txt,sha256=G8r2L__aYaMq4S3R70G6NsfEROiVp08kUcWvATzk2J8,4
+MaaFw-2.1.0a1.dist-info/RECORD,,

maa/define.py

@@ -49,6 +49,7 @@ MaaOptionValue = ctypes.c_void_p
 MaaOption = ctypes.c_int32
 MaaGlobalOption = MaaOption
 MaaCtrlOption = MaaOption
+MaaResOption = MaaOption
 
 
 class MaaGlobalOptionEnum:
@@ -104,6 +105,15 @@ class MaaCtrlOptionEnum:
     Recording = 5
 
 
+class MaaResOptionEnum:
+    Invalid = 0
+
+    # Use the specified inference device, the default is INT32_MAX, which means CPU.
+    # Please set this option before loading the model.
+    # value: int32_t, eg: 0; val_size: sizeof(int32_t)
+    GpuId = 1
+
+
 MaaAdbScreencapMethod = ctypes.c_uint64
 
 

maa/resource.py

@@ -60,6 +60,21 @@ 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_cpu(self) -> bool:
+        INT32_MAX = 2147483647  # means CPU
+        return self.set_gpu(INT32_MAX)
+
     def register_custom_recognition(
         self, name: str, recognition: "CustomRecognition"  # type: ignore
     ) -> bool:
@@ -189,6 +204,14 @@ class Resource:
             MaaStringBufferHandle,
         ]
 
+        Library.framework.MaaResourceSetOption.restype = MaaBool
+        Library.framework.MaaResourceSetOption.argtypes = [
+            MaaResourceHandle,
+            MaaResOption,
+            MaaOptionValue,
+            MaaOptionValueSize,
+        ]
+
         Library.framework.MaaResourceRegisterCustomRecognition.restype = MaaBool
         Library.framework.MaaResourceRegisterCustomRecognition.argtypes = [
             MaaResourceHandle,