MaaFw 2.0.3__py3-none-macosx_13_0_arm64.whl → 2.1.0a2__py3-none-macosx_13_0_arm64.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,23 +4,23 @@ 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/libMaaAdbControlUnit.dylib,sha256=CHvWXzv3bjVpyeI63XJwL7ksl5mryEz0O6DV7A-bk2A,703848
-maa/bin/libMaaDbgControlUnit.dylib,sha256=DqYr3Q5zybAL0vw2Wsdu273ntvmhe1Mu3y8_BygtVZ4,346488
-maa/bin/libMaaFramework.dylib,sha256=g2BkCeqUVDhGO8wOHcAlGDCyfQOBD-37012fc16TMoU,2065160
-maa/bin/libMaaToolkit.dylib,sha256=O0qbOx2gEBv5awNiyYrxyDiRx7p4MOCBBOq7SHrbDFU,760032
-maa/bin/libMaaUtils.dylib,sha256=zQtDZ4mXs_PqemHlImnLUtofa-eK4BvYJUToA6RZrnc,485856
+maa/bin/libMaaAdbControlUnit.dylib,sha256=h5XHjITq22AzWcU19r0FXfuLeW9l4eUPP9vtoR2bqk8,703848
+maa/bin/libMaaDbgControlUnit.dylib,sha256=PFp7fMap5LtOH7wkfZpBdmGiQzYk2MtyDWrSTEZSoLc,346488
+maa/bin/libMaaFramework.dylib,sha256=o4Uvr65QFtFmZsjs8ZpwoNOzVHrV7aKYXarBgevHikA,2088056
+maa/bin/libMaaToolkit.dylib,sha256=tU5ozFvcttTCd_tsACL7OVcH_PS9A1sFUVx8WsaLNuw,760912
+maa/bin/libMaaUtils.dylib,sha256=SCrEJCohp6-pHySRZudJGgITDd5A08AIe7bqukXjII8,485616
 maa/bin/libfastdeploy_ppocr.dylib,sha256=DlJtGHtCQwaLUqvMbQi7G3AcIAevy93pGyzGL5euZFU,4111976
 maa/bin/libonnxruntime.1.18.0.dylib,sha256=lZICOr6fl9QLl7UIrrWzpXQhoXwMz8Ew16Ju5WkAD78,13542696
 maa/bin/libopencv_world4.4.8.0.dylib,sha256=eLk14hRsABuJGcsxJmwV6ALe0I066pG0shREyVHk9LM,13176832
-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=X6qrXI7-5LbyjxgkaRzOPjEeLC4duk_uRWVzV3ptlj8,103
-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=X6qrXI7-5LbyjxgkaRzOPjEeLC4duk_uRWVzV3ptlj8,103
+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,