148 lines
4.0 KiB
Go
148 lines
4.0 KiB
Go
package chart
|
|
|
|
import (
|
|
"math"
|
|
"sort"
|
|
)
|
|
|
|
// XAxis represents the horizontal axis.
|
|
type XAxis struct {
|
|
Name string
|
|
Style Style
|
|
ValueFormatter ValueFormatter
|
|
Range Range
|
|
Ticks []Tick
|
|
|
|
GridLines []GridLine
|
|
GridMajorStyle Style
|
|
GridMinorStyle Style
|
|
}
|
|
|
|
// GetName returns the name.
|
|
func (xa XAxis) GetName() string {
|
|
return xa.Name
|
|
}
|
|
|
|
// GetStyle returns the style.
|
|
func (xa XAxis) GetStyle() Style {
|
|
return xa.Style
|
|
}
|
|
|
|
// GetTicks returns the ticks for a series. It coalesces between user provided ticks and
|
|
// generated ticks.
|
|
func (xa XAxis) GetTicks(r Renderer, ra Range, defaults Style, vf ValueFormatter) []Tick {
|
|
if len(xa.Ticks) > 0 {
|
|
return xa.Ticks
|
|
}
|
|
return xa.generateTicks(r, ra, defaults, vf)
|
|
}
|
|
|
|
func (xa XAxis) generateTicks(r Renderer, ra Range, defaults Style, vf ValueFormatter) []Tick {
|
|
step := xa.getTickStep(r, ra, defaults, vf)
|
|
return GenerateTicksWithStep(ra, step, vf)
|
|
}
|
|
|
|
func (xa XAxis) getTickStep(r Renderer, ra Range, defaults Style, vf ValueFormatter) float64 {
|
|
tickCount := xa.getTickCount(r, ra, defaults, vf)
|
|
step := ra.Delta() / float64(tickCount)
|
|
return step
|
|
}
|
|
|
|
func (xa XAxis) getTickCount(r Renderer, ra Range, defaults Style, vf ValueFormatter) int {
|
|
r.SetFont(xa.Style.GetFont(defaults.GetFont()))
|
|
r.SetFontSize(xa.Style.GetFontSize(defaults.GetFontSize(DefaultFontSize)))
|
|
|
|
// take a cut at determining the 'widest' value.
|
|
l0 := vf(ra.Min)
|
|
ln := vf(ra.Max)
|
|
ll := l0
|
|
if len(ln) > len(l0) {
|
|
ll = ln
|
|
}
|
|
llb := r.MeasureText(ll)
|
|
textWidth := llb.Width()
|
|
width := textWidth + DefaultMinimumTickHorizontalSpacing
|
|
count := int(math.Ceil(float64(ra.Domain) / float64(width)))
|
|
return count
|
|
}
|
|
|
|
// GetGridLines returns the gridlines for the axis.
|
|
func (xa XAxis) GetGridLines(ticks []Tick) []GridLine {
|
|
if len(xa.GridLines) > 0 {
|
|
return xa.GridLines
|
|
}
|
|
return GenerateGridLines(ticks, true)
|
|
}
|
|
|
|
// Measure returns the bounds of the axis.
|
|
func (xa XAxis) Measure(r Renderer, canvasBox Box, ra Range, defaults Style, ticks []Tick) Box {
|
|
r.SetStrokeColor(xa.Style.GetStrokeColor(defaults.StrokeColor))
|
|
r.SetStrokeWidth(xa.Style.GetStrokeWidth(defaults.StrokeWidth))
|
|
r.SetStrokeDashArray(xa.Style.GetStrokeDashArray())
|
|
r.SetFont(xa.Style.GetFont(defaults.GetFont()))
|
|
r.SetFontColor(xa.Style.GetFontColor(DefaultAxisColor))
|
|
r.SetFontSize(xa.Style.GetFontSize(defaults.GetFontSize()))
|
|
|
|
sort.Sort(Ticks(ticks))
|
|
|
|
var left, right, top, bottom = math.MaxInt32, 0, math.MaxInt32, 0
|
|
for _, t := range ticks {
|
|
v := t.Value
|
|
lx := ra.Translate(v)
|
|
tb := r.MeasureText(t.Label)
|
|
|
|
tx := canvasBox.Left + lx
|
|
ty := canvasBox.Bottom + DefaultXAxisMargin + tb.Height()
|
|
|
|
top = MinInt(top, canvasBox.Bottom)
|
|
left = MinInt(left, tx-(tb.Width()>>1))
|
|
right = MaxInt(right, tx+(tb.Width()>>1))
|
|
bottom = MaxInt(bottom, ty)
|
|
}
|
|
|
|
return Box{
|
|
Top: top,
|
|
Left: left,
|
|
Right: right,
|
|
Bottom: bottom,
|
|
}
|
|
}
|
|
|
|
// Render renders the axis
|
|
func (xa XAxis) Render(r Renderer, canvasBox Box, ra Range, defaults Style, ticks []Tick) {
|
|
r.SetStrokeColor(xa.Style.GetStrokeColor(defaults.StrokeColor))
|
|
r.SetStrokeWidth(xa.Style.GetStrokeWidth(defaults.StrokeWidth))
|
|
r.SetStrokeDashArray(xa.Style.GetStrokeDashArray())
|
|
r.SetFont(xa.Style.GetFont(defaults.GetFont()))
|
|
r.SetFontColor(xa.Style.GetFontColor(DefaultAxisColor))
|
|
r.SetFontSize(xa.Style.GetFontSize(defaults.GetFontSize()))
|
|
|
|
r.MoveTo(canvasBox.Left, canvasBox.Bottom)
|
|
r.LineTo(canvasBox.Right, canvasBox.Bottom)
|
|
r.Stroke()
|
|
|
|
sort.Sort(Ticks(ticks))
|
|
|
|
for _, t := range ticks {
|
|
v := t.Value
|
|
lx := ra.Translate(v)
|
|
tb := r.MeasureText(t.Label)
|
|
tx := canvasBox.Left + lx
|
|
ty := canvasBox.Bottom + DefaultXAxisMargin + tb.Height()
|
|
r.Text(t.Label, tx-tb.Width()>>1, ty)
|
|
|
|
r.MoveTo(tx, canvasBox.Bottom)
|
|
r.LineTo(tx, canvasBox.Bottom+DefaultVerticalTickHeight)
|
|
r.Stroke()
|
|
}
|
|
|
|
if xa.GridMajorStyle.Show || xa.GridMinorStyle.Show {
|
|
for _, gl := range xa.GetGridLines(ticks) {
|
|
if (gl.IsMinor && xa.GridMinorStyle.Show) ||
|
|
(!gl.IsMinor && xa.GridMajorStyle.Show) {
|
|
gl.Render(r, canvasBox, ra)
|
|
}
|
|
}
|
|
}
|
|
}
|