go-chart/draw.go

261 lines
6.5 KiB
Go
Raw Permalink Normal View History

2016-07-10 10:11:47 +02:00
package chart
2016-07-30 01:36:29 +02:00
import "math"
2016-07-17 10:42:31 +02:00
2016-07-30 01:36:29 +02:00
var (
// Draw contains helpers for drawing common objects.
Draw = &draw{}
2016-07-17 10:42:31 +02:00
)
2016-07-30 01:36:29 +02:00
type draw struct{}
// LineSeries draws a line series with a renderer.
2016-07-30 03:24:25 +02:00
func (d draw) LineSeries(r Renderer, canvasBox Box, xrange, yrange Range, style Style, vs ValueProvider) {
2016-07-10 10:11:47 +02:00
if vs.Len() == 0 {
return
}
cb := canvasBox.Bottom
2016-07-12 03:48:51 +02:00
cl := canvasBox.Left
2016-07-10 10:11:47 +02:00
v0x, v0y := vs.GetValue(0)
2016-07-12 03:48:51 +02:00
x0 := cl + xrange.Translate(v0x)
y0 := cb - yrange.Translate(v0y)
2016-07-10 10:11:47 +02:00
yv0 := yrange.Translate(0)
2016-07-10 10:11:47 +02:00
var vx, vy float64
var x, y int
2016-07-30 03:24:25 +02:00
fill := style.GetFillColor()
2016-07-10 10:11:47 +02:00
if !fill.IsZero() {
style.GetFillOptions().WriteToRenderer(r)
2016-07-10 19:43:04 +02:00
r.MoveTo(x0, y0)
2016-07-10 10:11:47 +02:00
for i := 1; i < vs.Len(); i++ {
vx, vy = vs.GetValue(i)
2016-07-12 03:48:51 +02:00
x = cl + xrange.Translate(vx)
y = cb - yrange.Translate(vy)
2016-07-10 19:43:04 +02:00
r.LineTo(x, y)
2016-07-10 10:11:47 +02:00
}
r.LineTo(x, Math.MinInt(cb, cb-yv0))
r.LineTo(x0, Math.MinInt(cb, cb-yv0))
r.LineTo(x0, y0)
2016-07-10 10:11:47 +02:00
r.Fill()
}
style.GetStrokeOptions().WriteToRenderer(r)
2016-07-10 10:11:47 +02:00
2016-07-10 19:43:04 +02:00
r.MoveTo(x0, y0)
2016-07-10 10:11:47 +02:00
for i := 1; i < vs.Len(); i++ {
vx, vy = vs.GetValue(i)
2016-07-12 03:48:51 +02:00
x = cl + xrange.Translate(vx)
y = cb - yrange.Translate(vy)
2016-07-10 19:43:04 +02:00
r.LineTo(x, y)
2016-07-10 10:11:47 +02:00
}
r.Stroke()
}
2016-07-30 01:36:29 +02:00
// BoundedSeries draws a series that implements BoundedValueProvider.
2016-07-30 03:24:25 +02:00
func (d draw) BoundedSeries(r Renderer, canvasBox Box, xrange, yrange Range, style Style, bbs BoundedValueProvider, drawOffsetIndexes ...int) {
drawOffsetIndex := 0
if len(drawOffsetIndexes) > 0 {
drawOffsetIndex = drawOffsetIndexes[0]
}
cb := canvasBox.Bottom
cl := canvasBox.Left
v0x, v0y1, v0y2 := bbs.GetBoundedValue(0)
x0 := cl + xrange.Translate(v0x)
y0 := cb - yrange.Translate(v0y1)
var vx, vy1, vy2 float64
var x, y int
xvalues := make([]float64, bbs.Len())
xvalues[0] = v0x
y2values := make([]float64, bbs.Len())
y2values[0] = v0y2
2016-07-30 18:12:03 +02:00
style.GetFillAndStrokeOptions().WriteToRenderer(r)
r.MoveTo(x0, y0)
for i := 1; i < bbs.Len(); i++ {
vx, vy1, vy2 = bbs.GetBoundedValue(i)
xvalues[i] = vx
y2values[i] = vy2
x = cl + xrange.Translate(vx)
y = cb - yrange.Translate(vy1)
if i > drawOffsetIndex {
r.LineTo(x, y)
} else {
r.MoveTo(x, y)
}
}
y = cb - yrange.Translate(vy2)
r.LineTo(x, y)
for i := bbs.Len() - 1; i >= drawOffsetIndex; i-- {
vx, vy2 = xvalues[i], y2values[i]
x = cl + xrange.Translate(vx)
y = cb - yrange.Translate(vy2)
r.LineTo(x, y)
}
r.Close()
r.FillStroke()
}
2016-07-30 01:36:29 +02:00
// HistogramSeries draws a value provider as boxes from 0.
2016-07-30 03:24:25 +02:00
func (d draw) HistogramSeries(r Renderer, canvasBox Box, xrange, yrange Range, style Style, vs ValueProvider, barWidths ...int) {
2016-07-17 10:42:31 +02:00
if vs.Len() == 0 {
return
}
//calculate bar width?
seriesLength := vs.Len()
barWidth := int(math.Floor(float64(xrange.GetDomain()) / float64(seriesLength)))
2016-07-17 10:42:31 +02:00
if len(barWidths) > 0 {
barWidth = barWidths[0]
}
cb := canvasBox.Bottom
cl := canvasBox.Left
//foreach datapoint, draw a box.
for index := 0; index < seriesLength; index++ {
vx, vy := vs.GetValue(index)
y0 := yrange.Translate(0)
x := cl + xrange.Translate(vx)
y := yrange.Translate(vy)
2016-07-30 01:36:29 +02:00
d.Box(r, Box{
2016-07-17 10:42:31 +02:00
Top: cb - y0,
Left: x - (barWidth >> 1),
Right: x + (barWidth >> 1),
Bottom: cb - y,
2016-07-30 03:24:25 +02:00
}, style)
2016-07-17 10:42:31 +02:00
}
}
// MeasureAnnotation measures how big an annotation would be.
2016-07-30 03:24:25 +02:00
func (d draw) MeasureAnnotation(r Renderer, canvasBox Box, style Style, lx, ly int, label string) Box {
style.WriteToRenderer(r)
2016-07-13 05:34:59 +02:00
2016-07-12 03:48:51 +02:00
textBox := r.MeasureText(label)
textWidth := textBox.Width()
textHeight := textBox.Height()
halfTextHeight := textHeight >> 1
2016-07-30 03:24:25 +02:00
pt := style.Padding.GetTop(DefaultAnnotationPadding.Top)
pl := style.Padding.GetLeft(DefaultAnnotationPadding.Left)
pr := style.Padding.GetRight(DefaultAnnotationPadding.Right)
pb := style.Padding.GetBottom(DefaultAnnotationPadding.Bottom)
2016-07-30 03:24:25 +02:00
strokeWidth := style.GetStrokeWidth()
2016-07-11 20:03:07 +02:00
2016-07-11 09:18:03 +02:00
top := ly - (pt + halfTextHeight)
right := lx + pl + pr + textWidth + DefaultAnnotationDeltaWidth + int(strokeWidth)
2016-07-11 09:18:03 +02:00
bottom := ly + (pb + halfTextHeight)
return Box{
2016-07-11 09:18:03 +02:00
Top: top,
Left: lx,
2016-07-11 09:18:03 +02:00
Right: right,
Bottom: bottom,
}
}
2016-07-30 01:36:29 +02:00
// Annotation draws an anotation with a renderer.
func (d draw) Annotation(r Renderer, canvasBox Box, style Style, lx, ly int, label string) {
style.GetTextOptions().WriteToRenderer(r)
2016-07-12 03:48:51 +02:00
textBox := r.MeasureText(label)
textWidth := textBox.Width()
halfTextHeight := textBox.Height() >> 1
2016-07-10 10:11:47 +02:00
style.GetFillAndStrokeOptions().WriteToRenderer(r)
2016-07-18 20:43:41 +02:00
pt := style.Padding.GetTop(DefaultAnnotationPadding.Top)
pl := style.Padding.GetLeft(DefaultAnnotationPadding.Left)
pr := style.Padding.GetRight(DefaultAnnotationPadding.Right)
pb := style.Padding.GetBottom(DefaultAnnotationPadding.Bottom)
2016-07-10 10:11:47 +02:00
textX := lx + pl + DefaultAnnotationDeltaWidth
textY := ly + halfTextHeight
2016-07-11 09:22:24 +02:00
ltx := lx + DefaultAnnotationDeltaWidth
2016-07-11 09:18:03 +02:00
lty := ly - (pt + halfTextHeight)
2016-07-10 10:11:47 +02:00
rtx := lx + pl + pr + textWidth + DefaultAnnotationDeltaWidth
2016-07-11 09:18:03 +02:00
rty := ly - (pt + halfTextHeight)
2016-07-10 10:11:47 +02:00
rbx := lx + pl + pr + textWidth + DefaultAnnotationDeltaWidth
2016-07-11 09:18:03 +02:00
rby := ly + (pb + halfTextHeight)
2016-07-10 10:11:47 +02:00
2016-07-11 09:22:24 +02:00
lbx := lx + DefaultAnnotationDeltaWidth
2016-07-11 09:18:03 +02:00
lby := ly + (pb + halfTextHeight)
2016-07-10 10:11:47 +02:00
r.MoveTo(lx, ly)
2016-07-11 09:18:03 +02:00
r.LineTo(ltx, lty)
r.LineTo(rtx, rty)
r.LineTo(rbx, rby)
r.LineTo(lbx, lby)
2016-07-10 10:11:47 +02:00
r.LineTo(lx, ly)
r.Close()
r.FillStroke()
style.GetTextOptions().WriteToRenderer(r)
2016-07-10 10:11:47 +02:00
r.Text(label, textX, textY)
}
2016-07-12 03:48:51 +02:00
2016-07-30 01:36:29 +02:00
// Box draws a box with a given style.
func (d draw) Box(r Renderer, b Box, s Style) {
2016-07-30 03:24:25 +02:00
s.WriteToRenderer(r)
2016-07-13 05:34:59 +02:00
2016-07-12 03:48:51 +02:00
r.MoveTo(b.Left, b.Top)
r.LineTo(b.Right, b.Top)
r.LineTo(b.Right, b.Bottom)
r.LineTo(b.Left, b.Bottom)
r.LineTo(b.Left, b.Top)
r.FillStroke()
}
// DrawText draws text with a given style.
2016-07-30 03:24:25 +02:00
func (d draw) Text(r Renderer, text string, x, y int, style Style) {
style.GetTextOptions().WriteToRenderer(r)
r.Text(text, x, y)
}
2016-07-30 01:36:29 +02:00
// TextWithin draws the text within a given box.
2016-07-30 03:24:25 +02:00
func (d draw) TextWithin(r Renderer, text string, box Box, style Style) {
lines := Text.WrapFit(r, text, box.Width(), style)
linesBox := Text.MeasureLines(r, lines, style)
style.GetTextOptions().WriteToRenderer(r)
2016-07-30 03:24:25 +02:00
y := box.Top
switch style.GetTextVerticalAlign() {
case TextVerticalAlignBottom, TextVerticalAlignBaseline: // i have to build better baseline handling into measure text
y = y - linesBox.Height()
case TextVerticalAlignMiddle, TextVerticalAlignMiddleBaseline:
y = (y - linesBox.Height()) >> 1
}
var tx, ty int
for _, line := range lines {
lineBox := r.MeasureText(line)
switch style.GetTextHorizontalAlign() {
case TextHorizontalAlignCenter:
2016-07-30 03:40:43 +02:00
tx = box.Left + ((box.Width() - lineBox.Width()) >> 1)
2016-07-30 03:24:25 +02:00
case TextHorizontalAlignRight:
tx = box.Right - lineBox.Width()
default:
tx = box.Left
}
ty = y + lineBox.Height()
d.Text(r, line, tx, ty, style)
y += lineBox.Height() + style.GetTextLineSpacing()
}
}