mower-ng/mower-ng
4
0
Fork 0
Code Issues 51 Pull requests Wiki Activity

fix:mumuipc内存泄漏，提高内存利用率

Browse source
This commit is contained in:
Elaina 2024-12-14 00:09:23 +08:00
parent 13e84a09f0
commit 17d1fd919f
1 changed files with 7 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

11
mower/utils/device/method/mumu_ipc.py
View file

@ -48,6 +48,7 @@ class MuMu12IPC:
self.instanse_index = int(config.conf.emulator.index)
self.connection = 0
self.display_id = -1
self.pixels = (ctypes.c_ubyte * 8294400)()
# 加载动态链接库
dll_path = os.path.join(
self.emulator_folder, "sdk", "external_renderer_ipc.dll"
@ -170,22 +171,24 @@ class MuMu12IPC:
def capture_display(self) -> np.ndarray:
self.check_status()
pixels = (ctypes.c_ubyte * 8294400)()
result = self.external_renderer.nemu_capture_display(
self.connection,
self.display_id,
8294400,
ctypes.byref(ctypes.c_int(1920)),
ctypes.byref(ctypes.c_int(1080)),
pixels,
self.pixels,
)
if result != 0:
logger.error(f"获取截图失败: {result}")
self.connection = 0
self.display_id = -1
config.device.exit() # 可能是游戏卡死
return self.capture_display()
return np.zeros((1080, 1920, 3), dtype=np.uint8)
image = np.frombuffer(pixels, dtype=np.uint8).reshape((1080, 1920, 4))[:, :, :3]
image = np.frombuffer(self.pixels, dtype=np.uint8).reshape((1080, 1920, 4))[
:, :, :3
]
image = np.flipud(image) # 翻转
return image