From efed9c072304cdde0a041d6f1363a5d1fb6fdb7f Mon Sep 17 00:00:00 2001 From: ShikiSuen Date: Fri, 18 Nov 2022 14:30:46 +0800 Subject: [PATCH] CocoaExtension // Add memory footprint calculator. --- .../CocoaExtension/CocoaExtension_Misc.swift | 24 +++++++++++++++++++ 1 file changed, 24 insertions(+) diff --git a/Packages/vChewing_CocoaExtension/Sources/CocoaExtension/CocoaExtension_Misc.swift b/Packages/vChewing_CocoaExtension/Sources/CocoaExtension/CocoaExtension_Misc.swift index bd0921d1..3278d304 100644 --- a/Packages/vChewing_CocoaExtension/Sources/CocoaExtension/CocoaExtension_Misc.swift +++ b/Packages/vChewing_CocoaExtension/Sources/CocoaExtension/CocoaExtension_Misc.swift @@ -94,3 +94,27 @@ extension FileManager { return true } } + +// MARK: - Memory Footprint Calculator + +// Ref: https://developer.apple.com/forums/thread/105088?answerId=357415022#357415022 +extension NSApplication { + /// The memory footprint of the current application in bytes. + public static var memoryFootprint: UInt64? { + // The `TASK_VM_INFO_COUNT` and `TASK_VM_INFO_REV1_COUNT` macros are too + // complex for the Swift C importer, so we have to define them ourselves. + let tskVMInfoCount = mach_msg_type_number_t( + MemoryLayout.size / MemoryLayout.size) + let tskVMInfoRev1Count = mach_msg_type_number_t( + MemoryLayout.offset(of: \task_vm_info_data_t.min_address)! / MemoryLayout.size) + var info = task_vm_info_data_t() + var count = tskVMInfoCount + let kr = withUnsafeMutablePointer(to: &info) { infoPtr in + infoPtr.withMemoryRebound(to: integer_t.self, capacity: Int(count)) { intPtr in + task_info(mach_task_self_, task_flavor_t(TASK_VM_INFO), intPtr, &count) + } + } + guard kr == KERN_SUCCESS, count >= tskVMInfoRev1Count else { return nil } + return info.phys_footprint as UInt64 + } +}