pmonitor 1.3.6__py3-none-any.whl → 1.3.7__py3-none-any.whl

monitor/mac_gpu.py

@@ -0,0 +1,53 @@
+import objc
+import Foundation
+
+IOKit = Foundation.NSBundle.bundleWithIdentifier_("com.apple.framework.IOKit")
+
+functions = [
+    ("IOServiceGetMatchingService", b"II@"),
+    ("IOServiceMatching", b"@*"),
+    ("IORegistryEntryCreateCFProperties", b"IIo^@@I"),
+]
+
+objc.loadBundleFunctions(IOKit, globals(), functions)
+
+
+def accelerator_performance_statistics() -> dict[str, int]:
+    accelerator_info = IOServiceGetMatchingService(
+        0, IOServiceMatching(b"IOAccelerator")
+    )
+    if accelerator_info == 0:
+        raise RuntimeError("IOAccelerator not found")
+    # else...
+    err, props = IORegistryEntryCreateCFProperties(accelerator_info, None, None, 0)
+    if err != 0:
+        raise RuntimeError("IOAccelerator properties not found")
+    # else...
+    return dict(props)
+
+
+def get_gpu_memory():
+    try:
+        gpu_info = accelerator_performance_statistics()
+        use_gpu_memory = 0
+        device_usage = 0
+        print(f"gpu_info: {gpu_info['PerformanceStatistics']}")
+        for key, value in gpu_info["PerformanceStatistics"].items():
+            if key == "In use system memory":
+                use_gpu_memory = int(value) / 1024 / 1024
+                print(f"{key}: {value}")
+            elif key == "inUseSysMemoryBytes":
+                use_gpu_memory = int(value) / 1024 / 1024
+                print(f"{key}: {value}")
+            elif key == "Device Utilization %":
+                device_usage = int(value)
+                print('device usage', device_usage)
+        # use_gpu_memory = gpu_info["PerformanceStatistics"]["In use system memory"]
+        # total_gpu_memory = gpu_info["PerformanceStatistics"]["Alloc system memory"]
+    except Exception as e:
+        print(f"Mac端在arm64平台通过接口获取Gpu Memory失败：{e}")
+        use_gpu_memory = 0
+    return use_gpu_memory, device_usage
+
+
+print(get_gpu_memory())

monitor/monitor_pids.py

@@ -10,6 +10,7 @@ import os
 import stat
 import subprocess
 import re
+import mac_gpu
 
 
 class PidsPerf:
@@ -21,14 +22,6 @@ class PidsPerf:
 
     def get_mac_perf(self):
         current_pid = self.processUtil.find_main_process_pid(self.process_name)
-        gpu_tool = os.path.join(os.path.dirname(__file__), 'exec', 'GpuMonitor')
-        if os.path.exists(gpu_tool):
-            if not os.access(gpu_tool, os.X_OK):
-                try:
-                    current_permissions = os.stat(gpu_tool).st_mode
-                    os.chmod(gpu_tool, current_permissions | stat.S_IEXEC | stat.S_IXGRP | stat.S_IXOTH)
-                except Exception as e:
-                    raise e
         while True:
             minor_cpu_sum = 0
             minor_real_mem_sum = 0
@@ -42,19 +35,7 @@ class PidsPerf:
             current_pids.add(current_pid)  # 将主进程及子进程添加到list中
             pids_process = self.processUtil.get_pids_process(current_pids)  # 获取所有进程
 
-            result = subprocess.run([gpu_tool], stdout=subprocess.PIPE, text=True)
-            for line in result.stdout.split('\n'):
-                pattern = r"GPU Name:(.*?),Device Percent:(.*?),Dedicated Usage:(\d+)"
-                matches = re.findall(pattern, line)
-                for match in matches:
-                    gpu_name, device_percent, dedicated_usage = match
-                    if 'driver' in gpu_name or 'Apple' in gpu_name:
-                        gpu_memory_usage_0 = round(float(dedicated_usage) / 1024 / 1024, 2)
-                        gpu_memory_percent_0 = round(float(device_percent), 2)
-                    else:
-                        gpu_memory_usage_1 = round(float(dedicated_usage) / 1024 / 1024, 2)
-                        gpu_memory_percent_1 = round(float(device_percent), 2)
-
+            gpu_memory_usage_0, gpu_memory_percent_0 = mac_gpu.get_gpu_memory()
             for process in pids_process:
                 try:
                     cpu_percent = process.cpu_percent()

{pmonitor-1.3.6.dist-info → pmonitor-1.3.7.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: pmonitor
-Version: 1.3.6
+Version: 1.3.7
 Summary: pc monitor
 Home-page: UNKNOWN
 Author: cfr

{pmonitor-1.3.6.dist-info → pmonitor-1.3.7.dist-info}/RECORD

@@ -1,8 +1,9 @@
 monitor/WinPidUtil.py,sha256=0vu2cqcQRrBTb5t_9HOmUoOfbe9KRrI-cyDlJX9bcrw,8753
 monitor/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 monitor/get_process_name.py,sha256=HxePo2gdrTo2Ukg4DTnYzUMdczVWywCeLCXQ-RRWyu8,3913
+monitor/mac_gpu.py,sha256=0gR5MUlx0wXkkETDHcJCsH_R_pul9mBF2bK_RIyAKyU,1861
 monitor/monitor_mac.py,sha256=OoptTTXTZQupzBwgGOfKNjyYAKBxDy-wI3l6T7XVE8s,4651
-monitor/monitor_pids.py,sha256=Fsh0GZKW1GJpWd98hTkalRA5JJ6_XbC_0Iy9fx4iYh4,15700
+monitor/monitor_pids.py,sha256=YeW0BtT6toY7ut5LFk1a0gKrKdCGnAbipS6gROB67IQ,14549
 monitor/monitor_win.py,sha256=xs5nzqqEPoDmJTegh3lQhVjjpPcOWnruWKK65ttqnTo,6161
 monitor/run_monitor.py,sha256=JKjb0X2tjPECFg_k7oby7Yg5Xhs8UNmbjCBP_6oXqMk,1579
 monitor/sys_info.py,sha256=aNultuRoQuRYPkYo397xAXVDXP07Qx5JOHtYzNmEvuc,3208
@@ -22,10 +23,9 @@ monitor/DLL/sechost.dll,sha256=1mXCrgNDRBiZ2XZk7eTpDm4oPG-x9BW4uYUVkrfvBkk,69986
 monitor/DLL/ucrtbased.dll,sha256=vtmKFPEHyr2OXkrUOu3Qs1dlbKG1dxZ8ItKCkTTU5S4,2238056
 monitor/DLL/vcruntime140.dll,sha256=AsaqDm5iRBGp8ZsDYKeGWrFZCOJgJFEOXDipwINiw1o,119888
 monitor/DLL/vcruntime140_1.dll,sha256=fdmqAuJxxoym1fGNZR0joV1yWXFa9DMmV4993ifzdjc,49640
-monitor/exec/GpuMonitor,sha256=C2I_EYswJ8m6LODAJc6Rqpnc4hM-eTmN9q0WVNLTE8w,111568
 monitor/exec/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-pmonitor-1.3.6.dist-info/METADATA,sha256=lBCe_9MJ5oi5e9dPZC38QWoplGJ6YvizQ1PNnmH8JtA,215
-pmonitor-1.3.6.dist-info/WHEEL,sha256=GJ7t_kWBFywbagK5eo9IoUwLW6oyOeTKmQ-9iHFVNxQ,92
-pmonitor-1.3.6.dist-info/entry_points.txt,sha256=vmlLEANgf2fZar9BeXYiKdF6GMJbVXip-SIZx5yPXDo,55
-pmonitor-1.3.6.dist-info/top_level.txt,sha256=tGkQDkVeyKgP5Rr7acpp0df83NBAnI8up0sGwRxuuQ4,8
-pmonitor-1.3.6.dist-info/RECORD,,
+pmonitor-1.3.7.dist-info/METADATA,sha256=ZvRXJSu78CzsXM8mZhHRw3GtREilcFJJe5XLR4uo0zY,215
+pmonitor-1.3.7.dist-info/WHEEL,sha256=GJ7t_kWBFywbagK5eo9IoUwLW6oyOeTKmQ-9iHFVNxQ,92
+pmonitor-1.3.7.dist-info/entry_points.txt,sha256=vmlLEANgf2fZar9BeXYiKdF6GMJbVXip-SIZx5yPXDo,55
+pmonitor-1.3.7.dist-info/top_level.txt,sha256=tGkQDkVeyKgP5Rr7acpp0df83NBAnI8up0sGwRxuuQ4,8
+pmonitor-1.3.7.dist-info/RECORD,,