MaaFw 2.0.3__py3-none-win_amd64.whl → 2.1.0a2__py3-none-win_amd64.whl

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: MaaFw
-Version: 2.0.3
+Version: 2.1.0a2
 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.0a2.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=it3PSVO_NoJUHvoqZSbiQOjiMd-RqChTBrK5cl9CKWE,27383832
-maa/bin/MaaAdbControlUnit.dll,sha256=AQ8Sc6qlcTbQxHixhOiC3W5XJvT7xzUlqc285rXu5A0,768512
-maa/bin/MaaDbgControlUnit.dll,sha256=ki6M8910NsjcOKeRNFVyeTaPq6HtzCIHlCfqZBc36fY,424448
-maa/bin/MaaFramework.dll,sha256=_v-fUYTvFfY0vCFtfEGMupMlCaPEBAmki7aVs27L2lk,1352704
-maa/bin/MaaToolkit.dll,sha256=INZq-zZwz7F7wb6dip166Oh0RxHDF7gQEo0kXqMeMtc,689664
-maa/bin/MaaUtils.dll,sha256=SAC3itMYt95ZEWKJU19xq4H6M57LKpVBFxZlHWan6HU,536064
-maa/bin/MaaWin32ControlUnit.dll,sha256=I6xH4MhfVcjU_hNYjRK580BmE0N90pLFExI7pSwZ17k,367104
+maa/bin/MaaAdbControlUnit.dll,sha256=9JWVJBSWwzK2ya5h0fJxJjuiq6ZLEY4F7ig6gpVBomQ,768512
+maa/bin/MaaDbgControlUnit.dll,sha256=fEjoKtN-IfG8FNXnJsORuvGqN6s549lO8tWbTL_HmMU,424448
+maa/bin/MaaFramework.dll,sha256=SPkqpN4JlbToeMO54-RhHd0_w6DNqqGrk83n8aG9jUM,1370112
+maa/bin/MaaToolkit.dll,sha256=U0mNHNRszSk5PT-iRK9gCfjDjHrW_fE5ji5NgMopzUM,691712
+maa/bin/MaaUtils.dll,sha256=7TdVo110Ls6FNf9c1sDH8E0-eVSjA5c4Bh4-Q3NBKsk,536064
+maa/bin/MaaWin32ControlUnit.dll,sha256=3fkr5ykwH9m7xSWnNuNdHxUvI64sy2MvFKHJ0jKus6c,367104
 maa/bin/fastdeploy_ppocr_maa.dll,sha256=4ri3FE9bJemUgZu3bsvDSPbqsioIFki1J-Ifk3qkKvg,564736
 maa/bin/onnxruntime_maa.dll,sha256=cu4Z-EwQuqMuphxCt4Pho7XPlUcIEenPK9Tk90alBhc,14653440
 maa/bin/opencv_world4_maa.dll,sha256=NtWJEx6Dg29Qqk493eAAU8O8zZ7Ym5UV6RQ0N0RG9RU,18230272
-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=1lcNv5ya_rmBEsTRObX12WYzMCEm6jtegLvKau9-dKM,96
-MaaFw-2.0.3.dist-info/top_level.txt,sha256=G8r2L__aYaMq4S3R70G6NsfEROiVp08kUcWvATzk2J8,4
-MaaFw-2.0.3.dist-info/RECORD,,
+MaaFw-2.1.0a2.dist-info/LICENSE.md,sha256=RG51X65V_wNLuyG-RGcLXxFsKyZnlH5wNvK_5mMlOag,7560
+MaaFw-2.1.0a2.dist-info/METADATA,sha256=Z6g1mUwESLQ5vSOtdHnGraBTlgMV7nr8LzdvhjaCL_s,9430
+MaaFw-2.1.0a2.dist-info/WHEEL,sha256=1lcNv5ya_rmBEsTRObX12WYzMCEm6jtegLvKau9-dKM,96
+MaaFw-2.1.0a2.dist-info/top_level.txt,sha256=G8r2L__aYaMq4S3R70G6NsfEROiVp08kUcWvATzk2J8,4
+MaaFw-2.1.0a2.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,