104 lines
2.3 KiB
Go
104 lines
2.3 KiB
Go
package runchart
|
|
|
|
import (
|
|
"fmt"
|
|
ui "github.com/sqshq/termui"
|
|
"image"
|
|
"math"
|
|
)
|
|
|
|
const (
|
|
xAxisLegendIndent = 10
|
|
yAxisLegendIndent = 1
|
|
heightWithLabelOnly = 3
|
|
heightWithDetails = 7
|
|
)
|
|
|
|
type Legend struct {
|
|
Enabled bool
|
|
Details bool
|
|
}
|
|
|
|
func (c *RunChart) renderLegend(buffer *ui.Buffer, rectangle image.Rectangle) {
|
|
|
|
if !c.legend.Enabled {
|
|
return
|
|
}
|
|
|
|
height := heightWithLabelOnly
|
|
if c.legend.Details {
|
|
height = heightWithDetails
|
|
}
|
|
|
|
rowCount := (c.Dx() - yAxisLegendIndent) / (height + yAxisLegendIndent)
|
|
columnCount := int(math.Ceil(float64(len(c.lines)) / float64(rowCount)))
|
|
columnWidth := getColumnWidth(c.lines, c.precision)
|
|
|
|
for col := 0; col < columnCount; col++ {
|
|
for row := 0; row < rowCount; row++ {
|
|
|
|
lineIndex := row + rowCount*col
|
|
if len(c.lines) <= lineIndex {
|
|
break
|
|
}
|
|
|
|
line := c.lines[row+rowCount*col]
|
|
extrema := getLineValueExtrema(line.points)
|
|
x := c.Inner.Max.X - (columnWidth+xAxisLegendIndent)*(col+1)
|
|
y := c.Inner.Min.Y + yAxisLegendIndent + row*(height)
|
|
|
|
titleStyle := ui.NewStyle(line.color)
|
|
detailsStyle := ui.NewStyle(ui.ColorWhite)
|
|
|
|
buffer.SetString(string(ui.DOT), titleStyle, image.Pt(x-2, y))
|
|
buffer.SetString(line.label, titleStyle, image.Pt(x, y))
|
|
|
|
if !c.legend.Details {
|
|
continue
|
|
}
|
|
|
|
details := [4]string{
|
|
fmt.Sprintf("cur %s", formatValue(line.points[len(line.points)-1].value, c.precision)),
|
|
fmt.Sprintf("max %s", formatValue(extrema.max, c.precision)),
|
|
fmt.Sprintf("min %s", formatValue(extrema.min, c.precision)),
|
|
fmt.Sprintf("dif %s", formatValue(1, c.precision)),
|
|
}
|
|
|
|
for i, detail := range details {
|
|
buffer.SetString(detail, detailsStyle, image.Pt(x, y+i+yAxisLegendIndent))
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
func getColumnWidth(lines []TimeLine, precision int) int {
|
|
width := len(formatValue(0, precision))
|
|
for _, line := range lines {
|
|
if len(line.label) > width {
|
|
width = len(line.label)
|
|
}
|
|
}
|
|
return width
|
|
}
|
|
|
|
// TODO remove and use the one from line
|
|
func getLineValueExtrema(points []TimePoint) ValueExtrema {
|
|
|
|
if len(points) == 0 {
|
|
return ValueExtrema{0, 0}
|
|
}
|
|
|
|
var max, min = -math.MaxFloat64, math.MaxFloat64
|
|
|
|
for _, point := range points {
|
|
if point.value > max {
|
|
max = point.value
|
|
}
|
|
if point.value < min {
|
|
min = point.value
|
|
}
|
|
}
|
|
|
|
return ValueExtrema{max: max, min: min}
|
|
}
|