Files
fastfetch/src/detection/gpu/gpu_apple.m
T
2026-09-17 13:50:07 +08:00

80 lines
3.0 KiB
Objective-C

#include "gpu.h"
#import <Metal/MTLDevice.h>
#import <IOKit/kext/KextManager.h>
#ifndef MAC_OS_VERSION_26_0
#define MTLGPUFamilyMetal4 ((MTLGPUFamily) 5002)
#endif
#ifndef MAC_OS_VERSION_13_0
#define MTLGPUFamilyMetal3 ((MTLGPUFamily) 5001)
#endif
#ifndef MAC_OS_X_VERSION_10_15
#define MTLFeatureSet_macOS_GPUFamily1_v4 ((MTLFeatureSet) 10004)
#define MTLFeatureSet_macOS_GPUFamily2_v1 ((MTLFeatureSet) 10005)
#endif
const char* ffGpuDetectDriverVersion(FFlist* gpus) {
NSMutableArray* arr = NSMutableArray.new;
FF_LIST_FOR_EACH (FFGPUResult, x, *gpus) {
[arr addObject:@(x->driver.chars)];
}
NSDictionary* dict = CFBridgingRelease(KextManagerCopyLoadedKextInfo((__bridge CFArrayRef) arr, (__bridge CFArrayRef) @[ @"CFBundleVersion" ]));
FF_LIST_FOR_EACH (FFGPUResult, x, *gpus) {
NSString* version = dict[@(x->driver.chars)][@"CFBundleVersion"];
if (version) {
ffStrbufAppendC(&x->driver, ' ');
ffStrbufAppendS(&x->driver, version.UTF8String);
}
}
return nullptr;
}
const char* ffGpuDetectMetal(FFlist* gpus) {
// Metal.framework is statically linked into the application,
// so `MTLCopyAllDevices` is always available, or the application will not start.
for (id<MTLDevice> device in MTLCopyAllDevices()) {
FFGPUResult* gpu = nullptr;
FF_LIST_FOR_EACH (FFGPUResult, x, *gpus) {
if (x->deviceId == device.registryID) {
gpu = x;
break;
}
}
if (!gpu) {
continue;
}
#pragma clang diagnostic ignored "-Wdeprecated-declarations"
#ifndef MAC_OS_X_VERSION_10_15
if ([device supportsFeatureSet:MTLFeatureSet_macOS_GPUFamily2_v1]) {
ffStrbufSetStatic(&gpu->platformApi, "Metal Feature Set 2");
} else if ([device supportsFeatureSet:MTLFeatureSet_macOS_GPUFamily1_v1]) {
ffStrbufSetStatic(&gpu->platformApi, "Metal Feature Set 1");
}
#else // MAC_OS_X_VERSION_10_15
if ([device supportsFamily:MTLGPUFamilyMetal4]) {
ffStrbufSetStatic(&gpu->platformApi, "Metal 4");
} else if ([device supportsFamily:MTLGPUFamilyMetal3]) {
ffStrbufSetStatic(&gpu->platformApi, "Metal 3");
}
else if ([device supportsFamily:MTLGPUFamilyCommon3]) {
ffStrbufSetStatic(&gpu->platformApi, "Metal Common 3");
} else if ([device supportsFamily:MTLGPUFamilyCommon2]) {
ffStrbufSetStatic(&gpu->platformApi, "Metal Common 2");
} else if ([device supportsFamily:MTLGPUFamilyCommon1]) {
ffStrbufSetStatic(&gpu->platformApi, "Metal Common 1");
}
gpu->type = device.hasUnifiedMemory ? FF_GPU_TYPE_INTEGRATED : FF_GPU_TYPE_DISCRETE;
gpu->index = (uint32_t) device.locationNumber;
if (device.hasUnifiedMemory && device.recommendedMaxWorkingSetSize > 0) {
gpu->shared.total = device.recommendedMaxWorkingSetSize;
}
#endif
}
return nullptr;
}