mirror of
https://github.com/morgan9e/macos-stats
synced 2026-04-14 08:14:19 +09:00
feat: add new widget Network Chart (#189) for Network module.
This commit is contained in:
191
ModuleKit/Widgets/NetworkChart.swift
Normal file
191
ModuleKit/Widgets/NetworkChart.swift
Normal file
@@ -0,0 +1,191 @@
|
||||
//
|
||||
// NetworkChart.swift
|
||||
// ModuleKit
|
||||
//
|
||||
// Created by Serhiy Mytrovtsiy on 19/01/2021.
|
||||
// Using Swift 5.0.
|
||||
// Running on macOS 11.1.
|
||||
//
|
||||
// Copyright © 2021 Serhiy Mytrovtsiy. All rights reserved.
|
||||
//
|
||||
|
||||
import Cocoa
|
||||
import StatsKit
|
||||
|
||||
public class NetworkChart: Widget {
|
||||
private var boxState: Bool = false
|
||||
private var frameState: Bool = false
|
||||
|
||||
private let store: UnsafePointer<Store>?
|
||||
private var chart: NetworkChartView = NetworkChartView(
|
||||
frame: NSRect(
|
||||
x: 0,
|
||||
y: 0,
|
||||
width: 34,
|
||||
height: Constants.Widget.height - (2*Constants.Widget.margin.y)
|
||||
),
|
||||
num: 60, minMax: false
|
||||
)
|
||||
private let width: CGFloat = 34
|
||||
|
||||
private var boxSettingsView: NSView? = nil
|
||||
private var frameSettingsView: NSView? = nil
|
||||
|
||||
public init(preview: Bool, title: String, config: NSDictionary?, store: UnsafePointer<Store>?) {
|
||||
var widgetTitle: String = title
|
||||
self.store = store
|
||||
if config != nil {
|
||||
if let titleFromConfig = config!["Title"] as? String {
|
||||
widgetTitle = titleFromConfig
|
||||
}
|
||||
}
|
||||
|
||||
super.init(frame: CGRect(
|
||||
x: Constants.Widget.margin.x,
|
||||
y: Constants.Widget.margin.y,
|
||||
width: self.width + (2*Constants.Widget.margin.x),
|
||||
height: Constants.Widget.height - (2*Constants.Widget.margin.y)
|
||||
))
|
||||
|
||||
self.preview = preview
|
||||
self.title = widgetTitle
|
||||
self.type = .networkChart
|
||||
self.wantsLayer = true
|
||||
self.canDrawConcurrently = true
|
||||
|
||||
if self.store != nil && !preview {
|
||||
self.boxState = store!.pointee.bool(key: "\(self.title)_\(self.type.rawValue)_box", defaultValue: self.boxState)
|
||||
self.frameState = store!.pointee.bool(key: "\(self.title)_\(self.type.rawValue)_frame", defaultValue: self.frameState)
|
||||
}
|
||||
|
||||
if preview {
|
||||
var list: [(Double, Double)] = []
|
||||
for _ in 0..<60 {
|
||||
list.append((Double.random(in: 0..<23), Double.random(in: 0..<23)))
|
||||
}
|
||||
self.chart.points = list
|
||||
}
|
||||
}
|
||||
|
||||
required init?(coder: NSCoder) {
|
||||
fatalError("init(coder:) has not been implemented")
|
||||
}
|
||||
|
||||
public override func draw(_ dirtyRect: NSRect) {
|
||||
super.draw(dirtyRect)
|
||||
|
||||
guard let context = NSGraphicsContext.current?.cgContext else { return }
|
||||
|
||||
let lineWidth = 1 / (NSScreen.main?.backingScaleFactor ?? 1)
|
||||
let offset = lineWidth / 2
|
||||
let boxSize: CGSize = CGSize(width: self.width - (Constants.Widget.margin.x*2), height: self.frame.size.height)
|
||||
|
||||
let box = NSBezierPath(roundedRect: NSRect(
|
||||
x: offset,
|
||||
y: offset,
|
||||
width: boxSize.width - (offset*2),
|
||||
height: boxSize.height - (offset*2)
|
||||
), xRadius: 2, yRadius: 2)
|
||||
|
||||
if self.boxState {
|
||||
(isDarkMode ? NSColor.white : NSColor.black).set()
|
||||
box.stroke()
|
||||
box.fill()
|
||||
}
|
||||
|
||||
context.saveGState()
|
||||
|
||||
let chartFrame = NSRect(
|
||||
x: offset,
|
||||
y: 1,
|
||||
width: box.bounds.width,
|
||||
height: box.bounds.height-1
|
||||
)
|
||||
self.chart.setFrameSize(NSSize(width: chartFrame.width, height: chartFrame.height))
|
||||
self.chart.draw(chartFrame)
|
||||
|
||||
context.restoreGState()
|
||||
|
||||
if self.boxState || self.frameState {
|
||||
(isDarkMode ? NSColor.white : NSColor.black).set()
|
||||
box.lineWidth = lineWidth
|
||||
box.stroke()
|
||||
}
|
||||
|
||||
self.setWidth(width)
|
||||
}
|
||||
|
||||
public func setValue(upload: Double, download: Double) {
|
||||
DispatchQueue.main.async(execute: {
|
||||
self.chart.addValue(upload: upload, download: download)
|
||||
self.display()
|
||||
})
|
||||
}
|
||||
|
||||
// MARK: - Settings
|
||||
|
||||
public override func settings(superview: NSView) {
|
||||
let rowHeight: CGFloat = 30
|
||||
let settingsNumber: CGFloat = 2
|
||||
let height: CGFloat = ((rowHeight + Constants.Settings.margin) * settingsNumber) + Constants.Settings.margin
|
||||
superview.setFrameSize(NSSize(width: superview.frame.width, height: height))
|
||||
|
||||
let view: NSView = NSView(frame: NSRect(x: Constants.Settings.margin, y: Constants.Settings.margin, width: superview.frame.width - (Constants.Settings.margin*2), height: superview.frame.height - (Constants.Settings.margin*2)))
|
||||
|
||||
self.boxSettingsView = ToggleTitleRow(
|
||||
frame: NSRect(x: 0, y: (rowHeight + Constants.Settings.margin) * 1, width: view.frame.width, height: rowHeight),
|
||||
title: LocalizedString("Box"),
|
||||
action: #selector(toggleBox),
|
||||
state: self.boxState
|
||||
)
|
||||
view.addSubview(self.boxSettingsView!)
|
||||
|
||||
self.frameSettingsView = ToggleTitleRow(
|
||||
frame: NSRect(x: 0, y: (rowHeight + Constants.Settings.margin) * 0, width: view.frame.width, height: rowHeight),
|
||||
title: LocalizedString("Frame"),
|
||||
action: #selector(toggleFrame),
|
||||
state: self.frameState
|
||||
)
|
||||
view.addSubview(self.frameSettingsView!)
|
||||
|
||||
superview.addSubview(view)
|
||||
}
|
||||
|
||||
@objc private func toggleBox(_ sender: NSControl) {
|
||||
var state: NSControl.StateValue? = nil
|
||||
if #available(OSX 10.15, *) {
|
||||
state = sender is NSSwitch ? (sender as! NSSwitch).state: nil
|
||||
} else {
|
||||
state = sender is NSButton ? (sender as! NSButton).state: nil
|
||||
}
|
||||
self.boxState = state! == .on ? true : false
|
||||
self.store?.pointee.set(key: "\(self.title)_\(self.type.rawValue)_box", value: self.boxState)
|
||||
|
||||
if self.frameState {
|
||||
FindAndToggleNSControlState(self.frameSettingsView, state: .off)
|
||||
self.frameState = false
|
||||
self.store?.pointee.set(key: "\(self.title)_\(self.type.rawValue)_frame", value: self.frameState)
|
||||
}
|
||||
|
||||
self.display()
|
||||
}
|
||||
|
||||
@objc private func toggleFrame(_ sender: NSControl) {
|
||||
var state: NSControl.StateValue? = nil
|
||||
if #available(OSX 10.15, *) {
|
||||
state = sender is NSSwitch ? (sender as! NSSwitch).state: nil
|
||||
} else {
|
||||
state = sender is NSButton ? (sender as! NSButton).state: nil
|
||||
}
|
||||
self.frameState = state! == .on ? true : false
|
||||
self.store?.pointee.set(key: "\(self.title)_\(self.type.rawValue)_frame", value: self.frameState)
|
||||
|
||||
if self.boxState {
|
||||
FindAndToggleNSControlState(self.boxSettingsView, state: .off)
|
||||
self.boxState = false
|
||||
self.store?.pointee.set(key: "\(self.title)_\(self.type.rawValue)_box", value: self.boxState)
|
||||
}
|
||||
|
||||
self.display()
|
||||
}
|
||||
}
|
||||
@@ -62,6 +62,7 @@ open class Reader<T>: ReaderInternal_p {
|
||||
private var nilCallbackCounter: Int = 0
|
||||
private var ready: Bool = false
|
||||
private var locked: Bool = true
|
||||
private var initlizalized: Bool = false
|
||||
public var active: Bool = false
|
||||
|
||||
private var history: [T]? = []
|
||||
@@ -124,7 +125,7 @@ open class Reader<T>: ReaderInternal_p {
|
||||
open func start() {
|
||||
if self.popup && self.locked {
|
||||
if !self.ready {
|
||||
DispatchQueue.global().async {
|
||||
DispatchQueue.global(qos: .background).async {
|
||||
self.read()
|
||||
}
|
||||
}
|
||||
@@ -140,9 +141,12 @@ open class Reader<T>: ReaderInternal_p {
|
||||
self.read()
|
||||
})
|
||||
}
|
||||
|
||||
DispatchQueue.global().async {
|
||||
self.read()
|
||||
|
||||
if !self.initlizalized {
|
||||
DispatchQueue.global(qos: .background).async {
|
||||
self.read()
|
||||
}
|
||||
self.initlizalized = true
|
||||
}
|
||||
self.repeatTask?.start()
|
||||
self.active = true
|
||||
@@ -154,11 +158,10 @@ open class Reader<T>: ReaderInternal_p {
|
||||
}
|
||||
|
||||
open func stop() {
|
||||
if let repeater = self.repeatTask {
|
||||
repeater.removeAllObservers(thenStop: true)
|
||||
}
|
||||
self.repeatTask?.removeAllObservers(thenStop: true)
|
||||
self.repeatTask = nil
|
||||
self.active = false
|
||||
self.initlizalized = false
|
||||
}
|
||||
|
||||
public func setInterval(_ value: Int) {
|
||||
|
||||
@@ -58,6 +58,7 @@ public enum widget_t: String {
|
||||
case lineChart = "line_chart"
|
||||
case barChart = "bar_chart"
|
||||
case pieChart = "pie_chart"
|
||||
case networkChart = "network_chart"
|
||||
case speed = "speed"
|
||||
case battery = "battery"
|
||||
case sensors = "sensors"
|
||||
@@ -85,6 +86,7 @@ open class Widget: NSView, Widget_p {
|
||||
case .lineChart: return "Line chart"
|
||||
case .barChart: return "Bar chart"
|
||||
case .pieChart: return "Pie chart"
|
||||
case .networkChart: return "Network chart"
|
||||
case .speed: return "Speed"
|
||||
case .battery: return "Battery"
|
||||
case .sensors: return "Text"
|
||||
@@ -146,6 +148,9 @@ func LoadWidget(_ type: widget_t, preview: Bool, name: String, config: NSDiction
|
||||
case .pieChart:
|
||||
widget = PieChart(preview: preview, title: name, config: widgetConfig, store: store)
|
||||
break
|
||||
case .networkChart:
|
||||
widget = NetworkChart(preview: preview, title: name, config: widgetConfig, store: store)
|
||||
break
|
||||
case .speed:
|
||||
widget = SpeedWidget(preview: preview, title: name, config: widgetConfig, store: store)
|
||||
break
|
||||
|
||||
Reference in New Issue
Block a user