Repository navigation
Expand file tree
/
Copy pathbargraph.go
More file actions
125 lines (99 loc) · 2.9 KB
/
Copy pathbargraph.go
File metadata and controls
125 lines (99 loc) · 2.9 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
package view
import (
"bytes"
"fmt"
"strings"
"github.com/rivo/tview"
"github.com/wtfutil/wtf/cfg"
"github.com/wtfutil/wtf/wtf"
)
// BarGraph defines the data required to make a bar graph
type BarGraph struct {
maxStars int
starChar string
*Base
*KeyboardWidget
View *tview.TextView
}
// Bar defines a single row in the bar graph
type Bar struct {
Label string
Percent int
ValueLabel string
LabelColor string
}
// NewBarGraph creates and returns an instance of BarGraph
func NewBarGraph(tviewApp *tview.Application, redrawChan chan bool, _ string, commonSettings *cfg.Common) BarGraph {
widget := BarGraph{
Base: NewBase(tviewApp, redrawChan, nil, commonSettings),
KeyboardWidget: NewKeyboardWidget(commonSettings),
maxStars: commonSettings.Config.UInt("graphStars", 20),
starChar: commonSettings.Config.UString("graphIcon", "|"),
}
widget.View = widget.createView(widget.bordered)
return widget
}
/* -------------------- Exported Functions -------------------- */
// BuildBars will build a string of * to represent your data of [time][value]
// time should be passed as a int64
func (widget *BarGraph) BuildBars(data []Bar) {
widget.View.SetText(BuildStars(data, widget.maxStars, widget.starChar))
widget.RedrawChan <- true
}
// BuildStars build the string to display
func BuildStars(data []Bar, maxStars int, starChar string) string {
var buffer bytes.Buffer
// the number of characters in the longest label
var longestLabel int
//just getting min and max values
for _, bar := range data {
if len(bar.Label) > longestLabel {
longestLabel = len(bar.Label)
}
}
// each number = how many stars?
var starRatio = float64(maxStars) / 100
//build the stars
for _, bar := range data {
//how many stars for this one?
var starCount = int(float64(bar.Percent) * starRatio)
label := bar.ValueLabel
if label == "" {
label = fmt.Sprint(bar.Percent)
}
labelColor := bar.LabelColor
if labelColor == "" {
labelColor = "default"
}
//write the line
_, err := fmt.Fprintf(
&buffer,
"%s%s[[%s]%s[default]%s] %s\n",
bar.Label,
strings.Repeat(" ", longestLabel-len(bar.Label)),
labelColor,
strings.Repeat(starChar, starCount),
strings.Repeat(" ", maxStars-starCount),
label,
)
if err != nil {
return ""
}
}
return buffer.String()
}
func (widget *BarGraph) TextView() *tview.TextView {
return widget.View
}
/* -------------------- Unexported Functions -------------------- */
func (widget *BarGraph) createView(bordered bool) *tview.TextView {
view := tview.NewTextView()
view.SetBackgroundColor(wtf.ColorFor(widget.commonSettings.Colors.Background))
view.SetBorder(bordered)
view.SetBorderColor(wtf.ColorFor(widget.BorderColor()))
view.SetDynamicColors(true)
view.SetTitle(widget.ContextualTitle(widget.CommonSettings().Title))
view.SetTitleColor(wtf.ColorFor(widget.commonSettings.Colors.Title))
view.SetWrap(false)
return view
}