// Plot implements the plot.Plotter interface. func (b *BarChart) Plot(c draw.Canvas, plt *plot.Plot) { trCat, trVal := plt.Transforms(&c) if b.Horizontal { trCat, trVal = trVal, trCat } for i, ht := range b.Values { catVal := b.XMin + float64(i) catMin := trCat(float64(catVal)) if !b.Horizontal { if !c.ContainsX(catMin) { continue } } else { if !c.ContainsY(catMin) { continue } } catMin = catMin - b.Width/2 + b.Offset catMax := catMin + b.Width bottom := b.stackedOn.BarHeight(i) valMin := trVal(bottom) valMax := trVal(bottom + ht) var pts []draw.Point var poly []draw.Point if !b.Horizontal { pts = []draw.Point{ {catMin, valMin}, {catMin, valMax}, {catMax, valMax}, {catMax, valMin}, } poly = c.ClipPolygonY(pts) } else { pts = []draw.Point{ {valMin, catMin}, {valMin, catMax}, {valMax, catMax}, {valMax, catMin}, } poly = c.ClipPolygonX(pts) } c.FillPolygon(b.Color, poly) var outline [][]draw.Point if !b.Horizontal { pts = append(pts, draw.Point{X: catMin, Y: valMin}) outline = c.ClipLinesY(pts) } else { pts = append(pts, draw.Point{X: valMin, Y: catMin}) outline = c.ClipLinesX(pts) } c.StrokeLines(b.LineStyle, outline...) } }
// Thumbnail fulfills the plot.Thumbnailer interface. func (b *BarChart) Thumbnail(c *draw.Canvas) { pts := []draw.Point{ {c.Min.X, c.Min.Y}, {c.Min.X, c.Max.Y}, {c.Max.X, c.Max.Y}, {c.Max.X, c.Min.Y}, } poly := c.ClipPolygonY(pts) c.FillPolygon(b.Color, poly) pts = append(pts, draw.Point{X: c.Min.X, Y: c.Min.Y}) outline := c.ClipLinesY(pts) c.StrokeLines(b.LineStyle, outline...) }
// Thumbnail the thumbnail for the Line, // implementing the plot.Thumbnailer interface. func (pts *Line) Thumbnail(c *draw.Canvas) { if pts.ShadeColor != nil { points := []draw.Point{ {c.Min.X, c.Min.Y}, {c.Min.X, c.Max.Y}, {c.Max.X, c.Max.Y}, {c.Max.X, c.Min.Y}, } poly := c.ClipPolygonY(points) c.FillPolygon(*pts.ShadeColor, poly) points = append(points, draw.Point{c.Min.X, c.Min.Y}) } else { y := c.Center().Y c.StrokeLine2(pts.LineStyle, c.Min.X, y, c.Max.X, y) } }
func (b *BoxPlotWithConfidenceIntervals) Plot(c draw.Canvas, plt *plot.Plot) { if b.Horizontal { panic("not handling horizontal BoxPlotWithConfidenceIntervals") } b.BoxPlot.Plot(c, plt) trX, trY := plt.Transforms(&c) x := trX(b.Location) if !c.ContainsX(x) { return } x += b.Offset mean, err, _ := meanAndConf95(b.Values) meanLength := trY(mean) lowLength := trY(mean - err) highLength := trY(mean + err) confidenceVertices := []draw.Point{ {x - b.Width/4, lowLength}, {x + b.Width/4, lowLength}, {x + b.Width/4, highLength}, {x - b.Width/4, highLength}, } confidencePoly := c.ClipPolygonY(confidenceVertices) gray := color.RGBA{200, 200, 200, 255} c.FillPolygon(gray, confidencePoly) meanLine := c.ClipLinesY([]draw.Point{ {x - b.Width/4, meanLength}, {x + b.Width/4, meanLength}, }) meanStyle := b.BoxStyle meanStyle.Color = color.RGBA{150, 150, 150, 255} c.StrokeLines(meanStyle, meanLine...) }