Files
macos-stats/Modules/CPU/popup.swift
2022-10-05 20:55:10 +02:00

497 lines
20 KiB
Swift

//
// popup.swift
// CPU
//
// Created by Serhiy Mytrovtsiy on 15/04/2020.
// Using Swift 5.0.
// Running on macOS 10.15.
//
// Copyright © 2020 Serhiy Mytrovtsiy. All rights reserved.
//
import Cocoa
import Kit
internal class Popup: NSView, Popup_p {
private var title: String
private var grid: NSGridView? = nil
private let dashboardHeight: CGFloat = 90
private let chartHeight: CGFloat = 90 + Constants.Popup.separatorHeight
private var detailsHeight: CGFloat {
get {
var count: CGFloat = 5
if isARM {
count = 3
}
if SystemKit.shared.device.info.cpu?.eCores != nil {
count += 1
}
if SystemKit.shared.device.info.cpu?.pCores != nil {
count += 1
}
return (22*count) + Constants.Popup.separatorHeight
}
}
private let averageHeight: CGFloat = (22*3) + Constants.Popup.separatorHeight
private let processHeight: CGFloat = 22
private var systemField: NSTextField? = nil
private var userField: NSTextField? = nil
private var idleField: NSTextField? = nil
private var shedulerLimitField: NSTextField? = nil
private var speedLimitField: NSTextField? = nil
private var eCoresField: NSTextField? = nil
private var pCoresField: NSTextField? = nil
private var average1Field: NSTextField? = nil
private var average5Field: NSTextField? = nil
private var average15Field: NSTextField? = nil
private var systemColorView: NSView? = nil
private var userColorView: NSView? = nil
private var idleColorView: NSView? = nil
private var chart: LineChartView? = nil
private var circle: PieChartView? = nil
private var temperatureCircle: HalfCircleGraphView? = nil
private var frequencyCircle: HalfCircleGraphView? = nil
private var initialized: Bool = false
private var initializedTemperature: Bool = false
private var initializedFrequency: Bool = false
private var initializedProcesses: Bool = false
private var initializedLimits: Bool = false
private var initializedAverage: Bool = false
private var processes: [ProcessView] = []
private var maxFreq: Double = 0
private var systemColorState: Color = .secondRed
private var systemColor: NSColor {
var value = NSColor.systemRed
if let color = self.systemColorState.additional as? NSColor {
value = color
}
return value
}
private var userColorState: Color = .secondBlue
private var userColor: NSColor {
var value = NSColor.systemBlue
if let color = self.userColorState.additional as? NSColor {
value = color
}
return value
}
private var idleColorState: Color = .lightGray
private var idleColor: NSColor {
var value = NSColor.lightGray
if let color = self.idleColorState.additional as? NSColor {
value = color
}
return value
}
private var chartColorState: Color = .systemAccent
private var chartColor: NSColor {
var value = NSColor.systemBlue
if let color = self.chartColorState.additional as? NSColor {
value = color
}
return value
}
public var sizeCallback: ((NSSize) -> Void)? = nil
private var numberOfProcesses: Int {
get {
return Store.shared.int(key: "\(self.title)_processes", defaultValue: 8)
}
}
private var processesHeight: CGFloat {
get {
let num = self.numberOfProcesses
return (self.processHeight*CGFloat(num)) + (num == 0 ? 0 : Constants.Popup.separatorHeight)
}
}
public init(_ title: String) {
self.title = title
super.init(frame: NSRect(
x: 0,
y: 0,
width: Constants.Popup.width,
height: self.dashboardHeight + self.chartHeight + self.averageHeight
))
self.setFrameSize(NSSize(width: self.frame.width, height: self.frame.height + self.detailsHeight + self.processesHeight))
self.systemColorState = Color.fromString(Store.shared.string(key: "\(self.title)_systemColor", defaultValue: self.systemColorState.key))
self.userColorState = Color.fromString(Store.shared.string(key: "\(self.title)_userColor", defaultValue: self.userColorState.key))
self.idleColorState = Color.fromString(Store.shared.string(key: "\(self.title)_idleColor", defaultValue: self.idleColorState.key))
self.chartColorState = Color.fromString(Store.shared.string(key: "\(self.title)_chartColor", defaultValue: self.chartColorState.key))
let gridView: NSGridView = NSGridView(frame: NSRect(x: 0, y: 0, width: self.frame.width, height: self.frame.height))
gridView.rowSpacing = 0
gridView.yPlacement = .fill
gridView.addRow(with: [self.initDashboard()])
gridView.addRow(with: [self.initChart()])
gridView.addRow(with: [self.initDetails()])
gridView.addRow(with: [self.initAverage()])
gridView.addRow(with: [self.initProcesses()])
gridView.row(at: 0).height = self.dashboardHeight
gridView.row(at: 1).height = self.chartHeight
gridView.row(at: 2).height = self.detailsHeight
gridView.row(at: 3).height = self.averageHeight
self.addSubview(gridView)
self.grid = gridView
}
required init?(coder: NSCoder) {
fatalError("init(coder:) has not been implemented")
}
public override func updateLayer() {
self.chart?.display()
}
public func numberOfProcessesUpdated() {
if self.processes.count == self.numberOfProcesses {
return
}
DispatchQueue.main.async(execute: {
self.processes = []
let h: CGFloat = self.dashboardHeight + self.chartHeight + self.detailsHeight + self.averageHeight + self.processesHeight
self.setFrameSize(NSSize(width: self.frame.width, height: h))
self.grid?.setFrameSize(NSSize(width: self.frame.width, height: h))
self.grid?.row(at: 4).cell(at: 0).contentView?.removeFromSuperview()
self.grid?.removeRow(at: 4)
self.grid?.addRow(with: [self.initProcesses()])
self.initializedProcesses = false
self.sizeCallback?(self.frame.size)
})
}
private func initDashboard() -> NSView {
let view: NSView = NSView(frame: NSRect(x: 0, y: 0, width: self.frame.width, height: self.dashboardHeight))
let container: NSView = NSView(frame: NSRect(x: 0, y: 10, width: view.frame.width, height: self.dashboardHeight-20))
self.circle = PieChartView(frame: NSRect(
x: (container.frame.width - container.frame.height)/2,
y: 0,
width: container.frame.height,
height: container.frame.height
), segments: [], drawValue: true)
self.circle!.toolTip = localizedString("CPU usage")
container.addSubview(self.circle!)
let centralWidth: CGFloat = self.dashboardHeight-20
let sideWidth: CGFloat = (view.frame.width - centralWidth - (Constants.Popup.margins*2))/2
self.temperatureCircle = HalfCircleGraphView(frame: NSRect(x: (sideWidth - 60)/2, y: 10, width: 60, height: 50))
self.temperatureCircle!.toolTip = localizedString("CPU temperature")
(self.temperatureCircle! as NSView).isHidden = true
self.frequencyCircle = HalfCircleGraphView(frame: NSRect(x: view.frame.width - 60 - Constants.Popup.margins*2, y: 10, width: 60, height: 50))
self.frequencyCircle!.toolTip = localizedString("CPU frequency")
(self.frequencyCircle! as NSView).isHidden = true
view.addSubview(self.temperatureCircle!)
view.addSubview(container)
view.addSubview(self.frequencyCircle!)
return view
}
private func initChart() -> NSView {
let view: NSView = NSView(frame: NSRect(x: 0, y: 0, width: self.frame.width, height: self.chartHeight))
let separator = separatorView(localizedString("Usage history"), origin: NSPoint(x: 0, y: self.chartHeight-Constants.Popup.separatorHeight), width: self.frame.width)
let container: NSView = NSView(frame: NSRect(x: 0, y: 0, width: self.frame.width, height: separator.frame.origin.y))
container.wantsLayer = true
container.layer?.backgroundColor = NSColor.lightGray.withAlphaComponent(0.1).cgColor
container.layer?.cornerRadius = 3
self.chart = LineChartView(frame: NSRect(x: 1, y: 0, width: view.frame.width, height: container.frame.height), num: 120)
self.chart?.color = self.chartColor
container.addSubview(self.chart!)
view.addSubview(separator)
view.addSubview(container)
return view
}
private func initDetails() -> NSView {
let view: NSView = NSView(frame: NSRect(x: 0, y: 0, width: self.frame.width, height: self.detailsHeight))
let separator = separatorView(localizedString("Details"), origin: NSPoint(
x: 0,
y: self.detailsHeight-Constants.Popup.separatorHeight
), width: self.frame.width)
let container: NSStackView = NSStackView(frame: NSRect(x: 0, y: 0, width: view.frame.width, height: separator.frame.origin.y))
container.orientation = .vertical
container.spacing = 0
(self.systemColorView, self.systemField) = popupWithColorRow(container, color: self.systemColor, n: 4, title: "\(localizedString("System")):", value: "")
(self.userColorView, self.userField) = popupWithColorRow(container, color: self.userColor, n: 3, title: "\(localizedString("User")):", value: "")
(self.idleColorView, self.idleField) = popupWithColorRow(container, color: self.idleColor.withAlphaComponent(0.5), n: 2, title: "\(localizedString("Idle")):", value: "")
if !isARM {
self.shedulerLimitField = popupRow(container, n: 1, title: "\(localizedString("Scheduler limit")):", value: "").1
self.speedLimitField = popupRow(container, n: 0, title: "\(localizedString("Speed limit")):", value: "").1
}
if SystemKit.shared.device.info.cpu?.eCores != nil {
self.eCoresField = popupRow(container, n: 0, title: "\(localizedString("Efficiency cores")):", value: "").1
}
if SystemKit.shared.device.info.cpu?.pCores != nil {
self.pCoresField = popupRow(container, n: 0, title: "\(localizedString("Performance cores")):", value: "").1
}
view.addSubview(separator)
view.addSubview(container)
return view
}
private func initAverage() -> NSView {
let view: NSView = NSView(frame: NSRect(x: 0, y: 0, width: self.frame.width, height: self.averageHeight))
let separator = separatorView(localizedString("Average load"), origin: NSPoint(x: 0, y: self.averageHeight-Constants.Popup.separatorHeight), width: self.frame.width)
let container: NSView = NSView(frame: NSRect(x: 0, y: 0, width: self.frame.width, height: separator.frame.origin.y))
self.average1Field = popupRow(container, n: 2, title: "\(localizedString("1 minute")):", value: "").1
self.average5Field = popupRow(container, n: 1, title: "\(localizedString("5 minutes")):", value: "").1
self.average15Field = popupRow(container, n: 0, title: "\(localizedString("15 minutes")):", value: "").1
view.addSubview(separator)
view.addSubview(container)
return view
}
private func initProcesses() -> NSView {
let view: NSView = NSView(frame: NSRect(x: 0, y: 0, width: self.frame.width, height: self.processesHeight))
let separator = separatorView(localizedString("Top processes"), origin: NSPoint(x: 0, y: self.processesHeight-Constants.Popup.separatorHeight), width: self.frame.width)
let container: NSStackView = NSStackView(frame: NSRect(x: 0, y: 0, width: self.frame.width, height: separator.frame.origin.y))
container.orientation = .vertical
container.spacing = 0
for _ in 0..<self.numberOfProcesses {
let processView = ProcessView()
self.processes.append(processView)
container.addArrangedSubview(processView)
}
view.addSubview(separator)
view.addSubview(container)
return view
}
public func loadCallback(_ value: CPU_Load) {
DispatchQueue.main.async(execute: {
if (self.window?.isVisible ?? false) || !self.initialized {
self.systemField?.stringValue = "\(Int(value.systemLoad.rounded(toPlaces: 2) * 100))%"
self.userField?.stringValue = "\(Int(value.userLoad.rounded(toPlaces: 2) * 100))%"
self.idleField?.stringValue = "\(Int(value.idleLoad.rounded(toPlaces: 2) * 100))%"
self.circle?.setValue(value.totalUsage)
self.circle?.setSegments([
circle_segment(value: value.systemLoad, color: self.systemColor),
circle_segment(value: value.userLoad, color: self.userColor)
])
self.circle?.setNonActiveSegmentColor(self.idleColor)
if let field = self.eCoresField, let usage = value.usageECores {
field.stringValue = "\(Int(usage * 100))%"
}
if let field = self.pCoresField, let usage = value.usagePCores {
field.stringValue = "\(Int(usage * 100))%"
}
self.initialized = true
}
self.chart?.addValue(value.totalUsage)
})
}
public func temperatureCallback(_ value: Double) {
DispatchQueue.main.async(execute: {
if (self.window?.isVisible ?? false) || !self.initializedTemperature {
if let view = self.temperatureCircle, (view as NSView).isHidden {
view.isHidden = false
}
self.temperatureCircle?.setValue(value)
self.temperatureCircle?.setText(Temperature(value))
self.initializedTemperature = true
}
})
}
public func frequencyCallback(_ value: Double) {
DispatchQueue.main.async(execute: {
if let view = self.frequencyCircle, (view as NSView).isHidden {
view.isHidden = false
}
if (self.window?.isVisible ?? false) || !self.initializedFrequency {
if value > self.maxFreq {
self.maxFreq = value
}
if let freqCircle = self.frequencyCircle {
freqCircle.setValue((100*value)/self.maxFreq)
freqCircle.setText("\((value/1000).rounded(toPlaces: 2))")
}
self.initializedFrequency = true
}
})
}
public func processCallback(_ list: [TopProcess]) {
DispatchQueue.main.async(execute: {
if !(self.window?.isVisible ?? false) && self.initializedProcesses {
return
}
if list.count != self.processes.count {
self.processes.forEach { processView in
processView.clear()
}
}
for i in 0..<list.count {
self.processes[i].set(list[i], "\(list[i].usage)%")
}
self.initializedProcesses = true
})
}
public func limitCallback(_ value: CPU_Limit) {
DispatchQueue.main.async(execute: {
if !(self.window?.isVisible ?? false) && self.initializedLimits {
return
}
self.shedulerLimitField?.stringValue = "\(value.scheduler)%"
self.speedLimitField?.stringValue = "\(value.speed)%"
self.initializedLimits = true
})
}
public func averageCallback(_ value: [Double]) {
guard value.count == 3 else {
return
}
DispatchQueue.main.async(execute: {
if !(self.window?.isVisible ?? false) && self.initializedAverage {
return
}
self.average1Field?.stringValue = "\(value[0])"
self.average5Field?.stringValue = "\(value[1])"
self.average15Field?.stringValue = "\(value[2])"
self.initializedAverage = true
})
}
public func toggleFrequency(state: Bool) {
DispatchQueue.main.async(execute: {
if let view = self.frequencyCircle {
view.isHidden = !state
}
self.initializedFrequency = false
})
}
// MARK: - Settings
public func settings() -> NSView? {
let view = SettingsContainerView()
view.addArrangedSubview(selectSettingsRow(
title: localizedString("System color"),
action: #selector(toggleSystemColor),
items: Color.allColors,
selected: self.systemColorState.key
))
view.addArrangedSubview(selectSettingsRow(
title: localizedString("User color"),
action: #selector(toggleUserColor),
items: Color.allColors,
selected: self.userColorState.key
))
view.addArrangedSubview(selectSettingsRow(
title: localizedString("Idle color"),
action: #selector(toggleIdleColor),
items: Color.allColors,
selected: self.idleColorState.key
))
view.addArrangedSubview(selectSettingsRow(
title: localizedString("Chart color"),
action: #selector(toggleChartColor),
items: Color.allColors,
selected: self.chartColorState.key
))
return view
}
@objc private func toggleSystemColor(_ sender: NSMenuItem) {
guard let key = sender.representedObject as? String,
let newValue = Color.allColors.first(where: { $0.key == key }) else {
return
}
self.systemColorState = newValue
Store.shared.set(key: "\(self.title)_systemColor", value: key)
if let color = newValue.additional as? NSColor {
self.systemColorView?.layer?.backgroundColor = color.cgColor
}
}
@objc private func toggleUserColor(_ sender: NSMenuItem) {
guard let key = sender.representedObject as? String,
let newValue = Color.allColors.first(where: { $0.key == key }) else {
return
}
self.userColorState = newValue
Store.shared.set(key: "\(self.title)_userColor", value: key)
if let color = newValue.additional as? NSColor {
self.userColorView?.layer?.backgroundColor = color.cgColor
}
}
@objc private func toggleIdleColor(_ sender: NSMenuItem) {
guard let key = sender.representedObject as? String,
let newValue = Color.allColors.first(where: { $0.key == key }) else {
return
}
self.idleColorState = newValue
Store.shared.set(key: "\(self.title)_idleColor", value: key)
if let color = newValue.additional as? NSColor {
self.idleColorView?.layer?.backgroundColor = color.cgColor
}
}
@objc private func toggleChartColor(_ sender: NSMenuItem) {
guard let key = sender.representedObject as? String,
let newValue = Color.allColors.first(where: { $0.key == key }) else {
return
}
self.chartColorState = newValue
Store.shared.set(key: "\(self.title)_chartColor", value: key)
if let color = newValue.additional as? NSColor {
self.chart?.color = color
}
}
}