barchart sample consuming and default options

This commit is contained in:
sqshq 2019-02-19 23:03:19 -05:00
parent 56fbf0cc0d
commit 56e9550611
3 changed files with 70 additions and 29 deletions

View File

@ -47,6 +47,14 @@ runcharts:
script: mongo --quiet --host=localhost blog --eval "db.getCollection('posts').find({status:'INACTIVE'}).itcount()"
barcharts:
- title: ANCHOR CONTEXT EVENTS
refresh-rate-ms: 300
position:
w: 30
h: 10
size:
w: 16
h: 4
items:
- label: ACTIVE
script: mongo --quiet --host=localhost blog --eval "db.getCollection('posts').find({status:'ACTIVE'}).itcount()"
- label: INACTIVE

View File

@ -40,6 +40,18 @@ func (c *Config) setDefaultValues() {
c.RunCharts[i] = chart
}
for i, chart := range c.BarCharts {
if chart.RefreshRateMs == nil {
r := defaultRefreshRateMs
chart.RefreshRateMs = &r
}
if chart.Scale == nil {
p := defaultScale
chart.Scale = &p
}
c.BarCharts[i] = chart
}
for i, box := range c.AsciiBoxes {
if box.RefreshRateMs == nil {
r := defaultRefreshRateMs
@ -78,4 +90,13 @@ func (c *Config) setDefaultColors() {
}
}
}
for _, chart := range c.BarCharts {
for j, item := range chart.Items {
if item.Color == nil {
item.Color = &palette.Colors[j%colorsCount]
chart.Items[j] = item
}
}
}
}

View File

@ -1,10 +1,13 @@
package barchart
import (
"fmt"
rw "github.com/mattn/go-runewidth"
"github.com/sqshq/sampler/console"
"github.com/sqshq/sampler/data"
ui "github.com/sqshq/termui"
"image"
"math"
"strconv"
)
@ -33,7 +36,7 @@ func NewBarChart(title string, scale int) *BarChart {
Block: block,
bars: []Bar{},
scale: scale,
maxValue: 0,
maxValue: -math.MaxFloat64,
}
}
@ -56,6 +59,10 @@ func (b *BarChart) ConsumeSample(sample data.Sample) {
}
}
if float > b.maxValue {
b.maxValue = float
}
bar := b.bars[index]
bar.value = float
b.bars[index] = bar
@ -64,46 +71,51 @@ func (b *BarChart) ConsumeSample(sample data.Sample) {
func (b *BarChart) Draw(buf *ui.Buffer) {
b.Block.Draw(buf)
maxVal := b.maxValue
barWidth := b.Inner.Dx() / len(b.bars)
barXCoordinate := b.Inner.Min.X
for i, data := range b.Data {
labelStyle := ui.NewStyle(console.ColorWhite)
for _, bar := range b.bars {
// draw bar
height := int((data / maxVal) * float64(b.Inner.Dy()-1))
for x := barXCoordinate; x < ui.MinInt(barXCoordinate+b.BarWidth, b.Inner.Max.X); x++ {
height := int((bar.value / b.maxValue) * float64(b.Inner.Dy()-1))
for x := barXCoordinate; x < ui.MinInt(barXCoordinate+barWidth, b.Inner.Max.X); x++ {
for y := b.Inner.Max.Y - 2; y > (b.Inner.Max.Y-2)-height; y-- {
c := ui.NewCell(barSymbol, ui.NewStyle(ui.SelectColor(b.BarColors, i)))
c := ui.NewCell(barSymbol, ui.NewStyle(bar.color))
buf.SetCell(c, image.Pt(x, y))
}
}
// draw label
if i < len(b.Labels) {
labelXCoordinate := barXCoordinate +
int((float64(b.BarWidth) / 2)) -
int((float64(rw.StringWidth(b.Labels[i])) / 2))
int(float64(barWidth)/2) -
int(float64(rw.StringWidth(bar.label))/2)
buf.SetString(
b.Labels[i],
ui.SelectStyle(b.LabelStyles, i),
bar.label,
labelStyle,
image.Pt(labelXCoordinate, b.Inner.Max.Y-1),
)
}
// draw value
numberXCoordinate := barXCoordinate + int((float64(b.BarWidth) / 2))
numberXCoordinate := barXCoordinate + int(float64(barWidth)/2)
if numberXCoordinate <= b.Inner.Max.X {
buf.SetString(
b.NumFmt(data),
ui.NewStyle(
ui.SelectStyle(b.NumStyles, i+1).Fg,
ui.SelectColor(b.BarColors, i),
ui.SelectStyle(b.NumStyles, i+1).Modifier,
),
formatValue(bar.value, b.scale),
labelStyle,
image.Pt(numberXCoordinate, b.Inner.Max.Y-2),
)
}
barXCoordinate += (b.BarWidth + b.BarGap)
barXCoordinate += barWidth + barIndent
}
}
// TODO extract to utils
func formatValue(value float64, scale int) string {
if math.Abs(value) == math.MaxFloat64 {
return "Inf"
} else {
format := "%." + strconv.Itoa(scale) + "f"
return fmt.Sprintf(format, value)
}
}