feat: add new widget Network Chart (#189) for Network module.

This commit is contained in:
Serhiy Mytrovtsiy
2021-01-21 21:56:06 +01:00
parent 38eeda6ac6
commit e524df7d5e
8 changed files with 250 additions and 30 deletions

View 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()
}
}

View File

@@ -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) {

View File

@@ -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