termui/barchart.go

150 lines
3.4 KiB
Go
Raw Normal View History

2017-01-13 23:07:43 -07:00
// Copyright 2017 Zack Guo <zack.y.guo@gmail.com>. All rights reserved.
2015-03-20 14:21:50 -06:00
// Use of this source code is governed by a MIT license that can
// be found in the LICENSE file.
2015-02-07 18:19:16 -07:00
package termui
import "fmt"
2015-03-24 15:16:43 -06:00
// BarChart creates multiple bars in a widget:
/*
bc := termui.NewBarChart()
data := []int{3, 2, 5, 3, 9, 5}
bclabels := []string{"S0", "S1", "S2", "S3", "S4", "S5"}
2016-10-14 05:48:50 -06:00
bc.BorderLabel = "Bar Chart"
2015-03-24 15:16:43 -06:00
bc.Data = data
bc.Width = 26
bc.Height = 10
bc.DataLabels = bclabels
bc.TextColor = termui.ColorGreen
bc.BarColor = termui.ColorRed
bc.NumColor = termui.ColorYellow
*/
2015-02-07 18:19:16 -07:00
type BarChart struct {
Block
BarColor Attribute
TextColor Attribute
NumColor Attribute
Data []int
DataLabels []string
BarWidth int
BarGap int
CellChar rune
2015-02-07 18:19:16 -07:00
labels [][]rune
dataNum [][]rune
numBar int
scale float64
max int
}
2015-03-24 15:16:43 -06:00
// NewBarChart returns a new *BarChart with current theme.
2015-02-07 18:19:16 -07:00
func NewBarChart() *BarChart {
bc := &BarChart{Block: *NewBlock()}
2015-10-07 12:25:59 -06:00
bc.BarColor = ThemeAttr("barchart.bar.bg")
bc.NumColor = ThemeAttr("barchart.num.fg")
bc.TextColor = ThemeAttr("barchart.text.fg")
2015-02-07 18:19:16 -07:00
bc.BarGap = 1
bc.BarWidth = 3
bc.CellChar = ' '
2015-02-07 18:19:16 -07:00
return bc
}
func (bc *BarChart) layout() {
2015-10-07 12:25:59 -06:00
bc.numBar = bc.innerArea.Dx() / (bc.BarGap + bc.BarWidth)
2015-02-07 18:19:16 -07:00
bc.labels = make([][]rune, bc.numBar)
bc.dataNum = make([][]rune, len(bc.Data))
for i := 0; i < bc.numBar && i < len(bc.DataLabels) && i < len(bc.Data); i++ {
bc.labels[i] = trimStr2Runes(bc.DataLabels[i], bc.BarWidth)
n := bc.Data[i]
s := fmt.Sprint(n)
bc.dataNum[i] = trimStr2Runes(s, bc.BarWidth)
}
//bc.max = bc.Data[0] // what if Data is nil? Sometimes when bar graph is nill it produces panic with panic: runtime error: index out of range
2017-04-03 23:47:18 -06:00
// Assign a negative value to get maxvalue auto-populates
if bc.max == 0 {
bc.max = -1
}
2015-02-07 18:19:16 -07:00
for i := 0; i < len(bc.Data); i++ {
if bc.max < bc.Data[i] {
bc.max = bc.Data[i]
}
}
2015-10-07 12:25:59 -06:00
bc.scale = float64(bc.max) / float64(bc.innerArea.Dy()-1)
2015-02-07 18:19:16 -07:00
}
func (bc *BarChart) SetMax(max int) {
if max > 0 {
bc.max = max
}
}
2015-03-24 15:16:43 -06:00
// Buffer implements Bufferer interface.
2015-10-07 12:25:59 -06:00
func (bc *BarChart) Buffer() Buffer {
buf := bc.Block.Buffer()
2015-02-07 18:19:16 -07:00
bc.layout()
for i := 0; i < bc.numBar && i < len(bc.Data) && i < len(bc.DataLabels); i++ {
h := int(float64(bc.Data[i]) / bc.scale)
oftX := i * (bc.BarWidth + bc.BarGap)
2016-08-13 20:34:50 -06:00
barBg := bc.Bg
barFg := bc.BarColor
if bc.CellChar == ' ' {
barBg = bc.BarColor
barFg = ColorDefault
if bc.BarColor == ColorDefault { // the same as above
barBg |= AttrReverse
}
}
2015-02-07 18:19:16 -07:00
// plot bar
for j := 0; j < bc.BarWidth; j++ {
for k := 0; k < h; k++ {
2015-10-07 12:25:59 -06:00
c := Cell{
Ch: bc.CellChar,
2016-08-13 20:34:50 -06:00
Bg: barBg,
Fg: barFg,
}
2016-08-13 20:34:50 -06:00
2015-10-07 12:25:59 -06:00
x := bc.innerArea.Min.X + i*(bc.BarWidth+bc.BarGap) + j
y := bc.innerArea.Min.Y + bc.innerArea.Dy() - 2 - k
buf.Set(x, y, c)
2015-02-07 18:19:16 -07:00
}
}
// plot text
for j, k := 0, 0; j < len(bc.labels[i]); j++ {
w := charWidth(bc.labels[i][j])
2015-10-08 20:11:26 -06:00
c := Cell{
Ch: bc.labels[i][j],
Bg: bc.Bg,
Fg: bc.TextColor,
}
y := bc.innerArea.Min.Y + bc.innerArea.Dy() - 1
x := bc.innerArea.Min.X + oftX + k
buf.Set(x, y, c)
k += w
2015-02-07 18:19:16 -07:00
}
// plot num
for j := 0; j < len(bc.dataNum[i]); j++ {
2015-10-07 12:25:59 -06:00
c := Cell{
Ch: bc.dataNum[i][j],
Fg: bc.NumColor,
2016-08-13 20:34:50 -06:00
Bg: barBg,
}
2016-08-13 20:34:50 -06:00
2015-02-07 18:19:16 -07:00
if h == 0 {
2015-10-08 20:11:26 -06:00
c.Bg = bc.Bg
2015-02-07 18:19:16 -07:00
}
2015-10-07 12:25:59 -06:00
x := bc.innerArea.Min.X + oftX + (bc.BarWidth-len(bc.dataNum[i]))/2 + j
y := bc.innerArea.Min.Y + bc.innerArea.Dy() - 2
buf.Set(x, y, c)
2015-02-07 18:19:16 -07:00
}
}
2015-10-07 12:25:59 -06:00
return buf
2015-02-07 18:19:16 -07:00
}