333 lines
8.5 KiB
Go
333 lines
8.5 KiB
Go
package chart
|
|
|
|
import (
|
|
"github.com/wcharczuk/go-chart/drawing"
|
|
"github.com/wcharczuk/go-chart/util"
|
|
)
|
|
|
|
// Legend returns a legend renderable function.
|
|
func Legend(c *Chart, userDefaults ...Style) Renderable {
|
|
return func(r Renderer, cb Box, chartDefaults Style) {
|
|
legendDefaults := Style{
|
|
FillColor: drawing.ColorWhite,
|
|
FontColor: DefaultTextColor,
|
|
FontSize: 8.0,
|
|
StrokeColor: DefaultAxisColor,
|
|
StrokeWidth: DefaultAxisLineWidth,
|
|
}
|
|
|
|
var legendStyle Style
|
|
if len(userDefaults) > 0 {
|
|
legendStyle = userDefaults[0].InheritFrom(chartDefaults.InheritFrom(legendDefaults))
|
|
} else {
|
|
legendStyle = chartDefaults.InheritFrom(legendDefaults)
|
|
}
|
|
|
|
// DEFAULTS
|
|
legendPadding := Box{
|
|
Top: 5,
|
|
Left: 5,
|
|
Right: 5,
|
|
Bottom: 5,
|
|
}
|
|
lineTextGap := 5
|
|
lineLengthMinimum := 25
|
|
|
|
var labels []string
|
|
var lines []Style
|
|
for index, s := range c.Series {
|
|
if s.GetStyle().IsZero() || s.GetStyle().Show {
|
|
if _, isAnnotationSeries := s.(AnnotationSeries); !isAnnotationSeries {
|
|
labels = append(labels, s.GetName())
|
|
lines = append(lines, s.GetStyle().InheritFrom(c.styleDefaultsSeries(index)))
|
|
}
|
|
}
|
|
}
|
|
|
|
legend := Box{
|
|
Top: cb.Top,
|
|
Left: cb.Left,
|
|
// bottom and right will be sized by the legend content + relevant padding.
|
|
}
|
|
|
|
legendContent := Box{
|
|
Top: legend.Top + legendPadding.Top,
|
|
Left: legend.Left + legendPadding.Left,
|
|
Right: legend.Left + legendPadding.Left,
|
|
Bottom: legend.Top + legendPadding.Top,
|
|
}
|
|
|
|
legendStyle.GetTextOptions().WriteToRenderer(r)
|
|
|
|
// measure
|
|
labelCount := 0
|
|
for x := 0; x < len(labels); x++ {
|
|
if len(labels[x]) > 0 {
|
|
tb := r.MeasureText(labels[x])
|
|
if labelCount > 0 {
|
|
legendContent.Bottom += DefaultMinimumTickVerticalSpacing
|
|
}
|
|
legendContent.Bottom += int(tb.Height())
|
|
right := legendContent.Left + int(tb.Width()) + lineTextGap + lineLengthMinimum
|
|
legendContent.Right = util.Math.MaxInt(legendContent.Right, right)
|
|
labelCount++
|
|
}
|
|
}
|
|
|
|
legend = legend.Grow(legendContent)
|
|
legend.Right = legendContent.Right + legendPadding.Right
|
|
legend.Bottom = legendContent.Bottom + legendPadding.Bottom
|
|
|
|
Draw.Box(r, legend, legendStyle)
|
|
|
|
legendStyle.GetTextOptions().WriteToRenderer(r)
|
|
|
|
ycursor := legendContent.Top
|
|
tx := legendContent.Left
|
|
legendCount := 0
|
|
var label string
|
|
for x := 0; x < len(labels); x++ {
|
|
label = labels[x]
|
|
if len(label) > 0 {
|
|
if legendCount > 0 {
|
|
ycursor += DefaultMinimumTickVerticalSpacing
|
|
}
|
|
|
|
tb := r.MeasureText(label)
|
|
|
|
ty := ycursor + int(tb.Height())
|
|
r.Text(label, tx, ty)
|
|
|
|
th2 := int(tb.Height()) >> 1
|
|
|
|
lx := tx + int(tb.Width()) + lineTextGap
|
|
ly := ty - th2
|
|
lx2 := legendContent.Right - legendPadding.Right
|
|
|
|
r.SetStrokeColor(lines[x].GetStrokeColor())
|
|
r.SetStrokeWidth(lines[x].GetStrokeWidth())
|
|
r.SetStrokeDashArray(lines[x].GetStrokeDashArray())
|
|
|
|
r.MoveTo(lx, ly)
|
|
r.LineTo(lx2, ly)
|
|
r.Stroke()
|
|
|
|
ycursor += int(tb.Height())
|
|
legendCount++
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
// LegendThin is a legend that doesn't obscure the chart area.
|
|
func LegendThin(c *Chart, userDefaults ...Style) Renderable {
|
|
return func(r Renderer, cb Box, chartDefaults Style) {
|
|
legendDefaults := Style{
|
|
FillColor: drawing.ColorWhite,
|
|
FontColor: DefaultTextColor,
|
|
FontSize: 8.0,
|
|
StrokeColor: DefaultAxisColor,
|
|
StrokeWidth: DefaultAxisLineWidth,
|
|
Padding: Box{
|
|
Top: 2,
|
|
Left: 7,
|
|
Right: 7,
|
|
Bottom: 5,
|
|
},
|
|
}
|
|
|
|
var legendStyle Style
|
|
if len(userDefaults) > 0 {
|
|
legendStyle = userDefaults[0].InheritFrom(chartDefaults.InheritFrom(legendDefaults))
|
|
} else {
|
|
legendStyle = chartDefaults.InheritFrom(legendDefaults)
|
|
}
|
|
|
|
r.SetFont(legendStyle.GetFont())
|
|
r.SetFontColor(legendStyle.GetFontColor())
|
|
r.SetFontSize(legendStyle.GetFontSize())
|
|
|
|
var labels []string
|
|
var lines []Style
|
|
for index, s := range c.Series {
|
|
if s.GetStyle().IsZero() || s.GetStyle().Show {
|
|
if _, isAnnotationSeries := s.(AnnotationSeries); !isAnnotationSeries {
|
|
labels = append(labels, s.GetName())
|
|
lines = append(lines, s.GetStyle().InheritFrom(c.styleDefaultsSeries(index)))
|
|
}
|
|
}
|
|
}
|
|
|
|
var textHeight int
|
|
var textWidth int
|
|
var textBox Box2d
|
|
for x := 0; x < len(labels); x++ {
|
|
if len(labels[x]) > 0 {
|
|
textBox = r.MeasureText(labels[x])
|
|
textHeight = util.Math.MaxInt(int(textBox.Height()), textHeight)
|
|
textWidth = util.Math.MaxInt(int(textBox.Width()), textWidth)
|
|
}
|
|
}
|
|
|
|
legendBoxHeight := textHeight + legendStyle.Padding.Top + legendStyle.Padding.Bottom
|
|
chartPadding := cb.Top
|
|
legendYMargin := (chartPadding - legendBoxHeight) >> 1
|
|
|
|
legendBox := Box{
|
|
Left: cb.Left,
|
|
Right: cb.Right,
|
|
Top: legendYMargin,
|
|
Bottom: legendYMargin + legendBoxHeight,
|
|
}
|
|
|
|
Draw.Box(r, legendBox, legendDefaults)
|
|
|
|
r.SetFont(legendStyle.GetFont())
|
|
r.SetFontColor(legendStyle.GetFontColor())
|
|
r.SetFontSize(legendStyle.GetFontSize())
|
|
|
|
lineTextGap := 5
|
|
lineLengthMinimum := 25
|
|
|
|
tx := legendBox.Left + legendStyle.Padding.Left
|
|
ty := legendYMargin + legendStyle.Padding.Top + textHeight
|
|
var label string
|
|
var lx, ly int
|
|
th2 := textHeight >> 1
|
|
for index := range labels {
|
|
label = labels[index]
|
|
if len(label) > 0 {
|
|
textBox = r.MeasureText(label)
|
|
r.Text(label, tx, ty)
|
|
|
|
lx = tx + int(textBox.Width()) + lineTextGap
|
|
ly = ty - th2
|
|
|
|
r.SetStrokeColor(lines[index].GetStrokeColor())
|
|
r.SetStrokeWidth(lines[index].GetStrokeWidth())
|
|
r.SetStrokeDashArray(lines[index].GetStrokeDashArray())
|
|
|
|
r.MoveTo(lx, ly)
|
|
r.LineTo(lx+lineLengthMinimum, ly)
|
|
r.Stroke()
|
|
|
|
tx += int(textBox.Width()) + DefaultMinimumTickHorizontalSpacing + lineTextGap + lineLengthMinimum
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
// LegendLeft is a legend that is designed for longer series lists.
|
|
func LegendLeft(c *Chart, userDefaults ...Style) Renderable {
|
|
return func(r Renderer, cb Box, chartDefaults Style) {
|
|
legendDefaults := Style{
|
|
FillColor: drawing.ColorWhite,
|
|
FontColor: DefaultTextColor,
|
|
FontSize: 8.0,
|
|
StrokeColor: DefaultAxisColor,
|
|
StrokeWidth: DefaultAxisLineWidth,
|
|
}
|
|
|
|
var legendStyle Style
|
|
if len(userDefaults) > 0 {
|
|
legendStyle = userDefaults[0].InheritFrom(chartDefaults.InheritFrom(legendDefaults))
|
|
} else {
|
|
legendStyle = chartDefaults.InheritFrom(legendDefaults)
|
|
}
|
|
|
|
// DEFAULTS
|
|
legendPadding := Box{
|
|
Top: 5,
|
|
Left: 5,
|
|
Right: 5,
|
|
Bottom: 5,
|
|
}
|
|
lineTextGap := 5
|
|
lineLengthMinimum := 25
|
|
|
|
var labels []string
|
|
var lines []Style
|
|
for index, s := range c.Series {
|
|
if s.GetStyle().IsZero() || s.GetStyle().Show {
|
|
if _, isAnnotationSeries := s.(AnnotationSeries); !isAnnotationSeries {
|
|
labels = append(labels, s.GetName())
|
|
lines = append(lines, s.GetStyle().InheritFrom(c.styleDefaultsSeries(index)))
|
|
}
|
|
}
|
|
}
|
|
|
|
legend := Box{
|
|
Top: 5,
|
|
Left: 5,
|
|
// bottom and right will be sized by the legend content + relevant padding.
|
|
}
|
|
|
|
legendContent := Box{
|
|
Top: legend.Top + legendPadding.Top,
|
|
Left: legend.Left + legendPadding.Left,
|
|
Right: legend.Left + legendPadding.Left,
|
|
Bottom: legend.Top + legendPadding.Top,
|
|
}
|
|
|
|
legendStyle.GetTextOptions().WriteToRenderer(r)
|
|
|
|
// measure
|
|
labelCount := 0
|
|
for x := 0; x < len(labels); x++ {
|
|
if len(labels[x]) > 0 {
|
|
tb := r.MeasureText(labels[x])
|
|
if labelCount > 0 {
|
|
legendContent.Bottom += DefaultMinimumTickVerticalSpacing
|
|
}
|
|
legendContent.Bottom += int(tb.Height())
|
|
right := legendContent.Left + int(tb.Width()) + lineTextGap + lineLengthMinimum
|
|
legendContent.Right = util.Math.MaxInt(legendContent.Right, right)
|
|
labelCount++
|
|
}
|
|
}
|
|
|
|
legend = legend.Grow(legendContent)
|
|
legend.Right = legendContent.Right + legendPadding.Right
|
|
legend.Bottom = legendContent.Bottom + legendPadding.Bottom
|
|
|
|
Draw.Box(r, legend, legendStyle)
|
|
|
|
legendStyle.GetTextOptions().WriteToRenderer(r)
|
|
|
|
ycursor := legendContent.Top
|
|
tx := legendContent.Left
|
|
legendCount := 0
|
|
var label string
|
|
for x := 0; x < len(labels); x++ {
|
|
label = labels[x]
|
|
if len(label) > 0 {
|
|
if legendCount > 0 {
|
|
ycursor += DefaultMinimumTickVerticalSpacing
|
|
}
|
|
|
|
tb := r.MeasureText(label)
|
|
|
|
ty := ycursor + int(tb.Height())
|
|
r.Text(label, tx, ty)
|
|
|
|
th2 := int(tb.Height()) >> 1
|
|
|
|
lx := tx + int(tb.Width()) + lineTextGap
|
|
ly := ty - th2
|
|
lx2 := legendContent.Right - legendPadding.Right
|
|
|
|
r.SetStrokeColor(lines[x].GetStrokeColor())
|
|
r.SetStrokeWidth(lines[x].GetStrokeWidth())
|
|
r.SetStrokeDashArray(lines[x].GetStrokeDashArray())
|
|
|
|
r.MoveTo(lx, ly)
|
|
r.LineTo(lx2, ly)
|
|
r.Stroke()
|
|
|
|
ycursor += int(tb.Height())
|
|
legendCount++
|
|
}
|
|
}
|
|
}
|
|
}
|