feat: support line chart draw function
This commit is contained in:
vicanso
parent
4ac419fce9
commit
ccdaf70dcb
34 changed files with 1780 additions and 4672 deletions
477
draw.go
477
draw.go
|
|
@ -24,6 +24,7 @@ package charts
|
|||
|
||||
import (
|
||||
"bytes"
|
||||
"errors"
|
||||
|
||||
"github.com/golang/freetype/truetype"
|
||||
"github.com/wcharczuk/go-chart/v2"
|
||||
|
|
@ -37,152 +38,78 @@ const (
|
|||
PositionBottom = "bottom"
|
||||
)
|
||||
|
||||
type draw struct {
|
||||
type Draw struct {
|
||||
Render chart.Renderer
|
||||
Box chart.Box
|
||||
parent *draw
|
||||
Font *truetype.Font
|
||||
parent *Draw
|
||||
}
|
||||
|
||||
type Point struct {
|
||||
X int
|
||||
Y int
|
||||
type DrawOption struct {
|
||||
Type string
|
||||
Parent *Draw
|
||||
Width int
|
||||
Height int
|
||||
}
|
||||
|
||||
type LineStyle struct {
|
||||
ClassName string
|
||||
StrokeDashArray []float64
|
||||
StrokeColor drawing.Color
|
||||
StrokeWidth float64
|
||||
FillColor drawing.Color
|
||||
DotWidth float64
|
||||
DotColor drawing.Color
|
||||
DotFillColor drawing.Color
|
||||
}
|
||||
type Option func(*Draw) error
|
||||
|
||||
func (ls *LineStyle) Style() chart.Style {
|
||||
return chart.Style{
|
||||
ClassName: ls.ClassName,
|
||||
StrokeDashArray: ls.StrokeDashArray,
|
||||
StrokeColor: ls.StrokeColor,
|
||||
StrokeWidth: ls.StrokeWidth,
|
||||
FillColor: ls.FillColor,
|
||||
DotWidth: ls.DotWidth,
|
||||
DotColor: ls.DotColor,
|
||||
func PaddingOption(padding chart.Box) Option {
|
||||
return func(d *Draw) error {
|
||||
d.Box.Left += padding.Left
|
||||
d.Box.Top += padding.Top
|
||||
d.Box.Right -= padding.Right
|
||||
d.Box.Bottom -= padding.Bottom
|
||||
return nil
|
||||
}
|
||||
}
|
||||
|
||||
type BarStyle struct {
|
||||
ClassName string
|
||||
StrokeDashArray []float64
|
||||
FillColor drawing.Color
|
||||
}
|
||||
|
||||
func (bs *BarStyle) Style() chart.Style {
|
||||
return chart.Style{
|
||||
ClassName: bs.ClassName,
|
||||
StrokeDashArray: bs.StrokeDashArray,
|
||||
StrokeColor: bs.FillColor,
|
||||
StrokeWidth: 1,
|
||||
FillColor: bs.FillColor,
|
||||
func NewDraw(opt DrawOption, opts ...Option) (*Draw, error) {
|
||||
if opt.Parent == nil && (opt.Width <= 0 || opt.Height <= 0) {
|
||||
return nil, errors.New("parent and width/height can not be nil")
|
||||
}
|
||||
}
|
||||
|
||||
type AxisStyle struct {
|
||||
BoundaryGap *bool
|
||||
Show *bool
|
||||
Position string
|
||||
Offset int
|
||||
ClassName string
|
||||
StrokeColor drawing.Color
|
||||
StrokeWidth float64
|
||||
TickLength int
|
||||
TickShow *bool
|
||||
LabelMargin int
|
||||
FontSize float64
|
||||
Font *truetype.Font
|
||||
FontColor drawing.Color
|
||||
}
|
||||
|
||||
func (as *AxisStyle) GetLabelMargin() int {
|
||||
return getDefaultInt(as.LabelMargin, 8)
|
||||
}
|
||||
|
||||
func (as *AxisStyle) GetTickLength() int {
|
||||
return getDefaultInt(as.TickLength, 5)
|
||||
}
|
||||
|
||||
func (as *AxisStyle) Style() chart.Style {
|
||||
s := chart.Style{
|
||||
ClassName: as.ClassName,
|
||||
StrokeColor: as.StrokeColor,
|
||||
StrokeWidth: as.StrokeWidth,
|
||||
FontSize: as.FontSize,
|
||||
FontColor: as.FontColor,
|
||||
Font: as.Font,
|
||||
font, _ := chart.GetDefaultFont()
|
||||
d := &Draw{
|
||||
Font: font,
|
||||
}
|
||||
if s.FontSize == 0 {
|
||||
s.FontSize = chart.DefaultFontSize
|
||||
width := opt.Width
|
||||
height := opt.Height
|
||||
if opt.Parent != nil {
|
||||
d.parent = opt.Parent
|
||||
d.Render = d.parent.Render
|
||||
d.Box = opt.Parent.Box.Clone()
|
||||
}
|
||||
if s.Font == nil {
|
||||
s.Font, _ = chart.GetDefaultFont()
|
||||
if width != 0 && height != 0 {
|
||||
d.Box.Right = width + d.Box.Left
|
||||
d.Box.Bottom = height + d.Box.Top
|
||||
}
|
||||
return s
|
||||
}
|
||||
|
||||
type AxisData struct {
|
||||
Text string
|
||||
}
|
||||
type AxisDataList []AxisData
|
||||
|
||||
func (l AxisDataList) TextList() []string {
|
||||
textList := make([]string, len(l))
|
||||
for index, item := range l {
|
||||
textList[index] = item.Text
|
||||
// 创建render
|
||||
if d.parent == nil {
|
||||
fn := chart.SVG
|
||||
if opt.Type == "png" {
|
||||
fn = chart.PNG
|
||||
}
|
||||
r, err := fn(d.Box.Right, d.Box.Bottom)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
d.Render = r
|
||||
}
|
||||
return textList
|
||||
}
|
||||
|
||||
type axisOption struct {
|
||||
data *AxisDataList
|
||||
style *AxisStyle
|
||||
textMaxWith int
|
||||
textMaxHeight int
|
||||
}
|
||||
|
||||
func NewAxisDataListFromStringList(textList []string) AxisDataList {
|
||||
list := make(AxisDataList, len(textList))
|
||||
for index, text := range textList {
|
||||
list[index] = AxisData{
|
||||
Text: text,
|
||||
for _, o := range opts {
|
||||
err := o(d)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
}
|
||||
return list
|
||||
return d, nil
|
||||
}
|
||||
|
||||
type Option func(*draw)
|
||||
|
||||
func ParentOption(p *draw) Option {
|
||||
return func(d *draw) {
|
||||
d.parent = p
|
||||
}
|
||||
}
|
||||
|
||||
func NewDraw(r chart.Renderer, box chart.Box, opts ...Option) *draw {
|
||||
d := &draw{
|
||||
Render: r,
|
||||
Box: box,
|
||||
}
|
||||
for _, opt := range opts {
|
||||
opt(d)
|
||||
}
|
||||
return d
|
||||
}
|
||||
|
||||
func (d *draw) Parent() *draw {
|
||||
func (d *Draw) Parent() *Draw {
|
||||
return d.parent
|
||||
}
|
||||
|
||||
func (d *draw) Top() *draw {
|
||||
func (d *Draw) Top() *Draw {
|
||||
if d.parent == nil {
|
||||
return nil
|
||||
}
|
||||
|
|
@ -197,7 +124,7 @@ func (d *draw) Top() *draw {
|
|||
return t
|
||||
}
|
||||
|
||||
func (d *draw) Bytes() ([]byte, error) {
|
||||
func (d *Draw) Bytes() ([]byte, error) {
|
||||
buffer := bytes.Buffer{}
|
||||
err := d.Render.Save(&buffer)
|
||||
if err != nil {
|
||||
|
|
@ -206,23 +133,23 @@ func (d *draw) Bytes() ([]byte, error) {
|
|||
return buffer.Bytes(), err
|
||||
}
|
||||
|
||||
func (d *draw) moveTo(x, y int) {
|
||||
func (d *Draw) moveTo(x, y int) {
|
||||
d.Render.MoveTo(x+d.Box.Left, y+d.Box.Top)
|
||||
}
|
||||
|
||||
func (d *draw) lineTo(x, y int) {
|
||||
func (d *Draw) lineTo(x, y int) {
|
||||
d.Render.LineTo(x+d.Box.Left, y+d.Box.Top)
|
||||
}
|
||||
|
||||
func (d *draw) circle(radius float64, x, y int) {
|
||||
func (d *Draw) circle(radius float64, x, y int) {
|
||||
d.Render.Circle(radius, x+d.Box.Left, y+d.Box.Top)
|
||||
}
|
||||
|
||||
func (d *draw) text(body string, x, y int) {
|
||||
func (d *Draw) text(body string, x, y int) {
|
||||
d.Render.Text(body, x+d.Box.Left, y+d.Box.Top)
|
||||
}
|
||||
|
||||
func (d *draw) lineStroke(points []Point, style LineStyle) {
|
||||
func (d *Draw) lineStroke(points []Point, style LineStyle) {
|
||||
s := style.Style()
|
||||
if !s.ShouldDrawStroke() {
|
||||
return
|
||||
|
|
@ -241,290 +168,16 @@ func (d *draw) lineStroke(points []Point, style LineStyle) {
|
|||
r.Stroke()
|
||||
}
|
||||
|
||||
func (d *draw) lineFill(points []Point, style LineStyle) {
|
||||
s := style.Style()
|
||||
if !(s.ShouldDrawStroke() && s.ShouldDrawFill()) {
|
||||
return
|
||||
}
|
||||
func (d *Draw) setBackground(width, height int, color drawing.Color) {
|
||||
r := d.Render
|
||||
var x, y int
|
||||
s.GetFillOptions().WriteDrawingOptionsToRenderer(r)
|
||||
for index, point := range points {
|
||||
x = point.X
|
||||
y = point.Y
|
||||
if index == 0 {
|
||||
d.moveTo(x, y)
|
||||
} else {
|
||||
d.lineTo(x, y)
|
||||
}
|
||||
s := chart.Style{
|
||||
FillColor: color,
|
||||
}
|
||||
height := d.Box.Height()
|
||||
d.lineTo(x, height)
|
||||
x0 := points[0].X
|
||||
y0 := points[0].Y
|
||||
d.lineTo(x0, height)
|
||||
d.lineTo(x0, y0)
|
||||
r.Fill()
|
||||
}
|
||||
|
||||
func (d *draw) lineDot(points []Point, style LineStyle) {
|
||||
s := style.Style()
|
||||
if !s.ShouldDrawDot() {
|
||||
return
|
||||
}
|
||||
r := d.Render
|
||||
dotWith := s.GetDotWidth()
|
||||
|
||||
s.GetDotOptions().WriteDrawingOptionsToRenderer(r)
|
||||
for _, point := range points {
|
||||
if !style.DotFillColor.IsZero() {
|
||||
r.SetFillColor(style.DotFillColor)
|
||||
}
|
||||
r.SetStrokeColor(s.DotColor)
|
||||
d.circle(dotWith, point.X, point.Y)
|
||||
r.FillStroke()
|
||||
}
|
||||
}
|
||||
|
||||
func (d *draw) Line(points []Point, style LineStyle) {
|
||||
if len(points) == 0 {
|
||||
return
|
||||
}
|
||||
d.lineFill(points, style)
|
||||
d.lineStroke(points, style)
|
||||
d.lineDot(points, style)
|
||||
}
|
||||
|
||||
func (d *draw) Bar(b chart.Box, style BarStyle) {
|
||||
s := style.Style()
|
||||
|
||||
r := d.Render
|
||||
s.GetFillAndStrokeOptions().WriteToRenderer(r)
|
||||
|
||||
d.moveTo(b.Left, b.Top)
|
||||
d.lineTo(b.Right, b.Top)
|
||||
d.lineTo(b.Right, b.Bottom)
|
||||
d.lineTo(b.Left, b.Bottom)
|
||||
d.lineTo(b.Left, b.Top)
|
||||
s.WriteToRenderer(r)
|
||||
r.MoveTo(0, 0)
|
||||
r.LineTo(width, 0)
|
||||
r.LineTo(width, height)
|
||||
r.LineTo(0, height)
|
||||
r.LineTo(0, 0)
|
||||
r.FillStroke()
|
||||
}
|
||||
|
||||
func (d *draw) axisLabel(opt *axisOption) {
|
||||
style := opt.style
|
||||
data := *opt.data
|
||||
if style.FontColor.IsZero() || len(data) == 0 {
|
||||
return
|
||||
}
|
||||
r := d.Render
|
||||
|
||||
s := style.Style()
|
||||
s.GetTextOptions().WriteTextOptionsToRenderer(r)
|
||||
|
||||
width := d.Box.Width()
|
||||
height := d.Box.Height()
|
||||
textList := data.TextList()
|
||||
count := len(textList)
|
||||
x0 := 0
|
||||
y0 := 0
|
||||
tickLength := style.GetTickLength()
|
||||
|
||||
// 坐标轴文本
|
||||
switch style.Position {
|
||||
case PositionLeft:
|
||||
values := autoDivide(height, count)
|
||||
textList := data.TextList()
|
||||
// 由下往上
|
||||
reverseIntSlice(values)
|
||||
for index, text := range textList {
|
||||
y := values[index]
|
||||
height := y - values[index+1]
|
||||
b := r.MeasureText(text)
|
||||
y -= (height - b.Height()) >> 1
|
||||
x := x0 + opt.textMaxWith - (b.Width())
|
||||
d.text(text, x, y)
|
||||
}
|
||||
case PositionRight:
|
||||
values := autoDivide(height, count)
|
||||
textList := data.TextList()
|
||||
// 由下往上
|
||||
reverseIntSlice(values)
|
||||
for index, text := range textList {
|
||||
y := values[index]
|
||||
height := y - values[index+1]
|
||||
b := r.MeasureText(text)
|
||||
y -= (height - b.Height()) >> 1
|
||||
x := width - opt.textMaxWith
|
||||
d.text(text, x, y)
|
||||
}
|
||||
case PositionTop:
|
||||
y0 = tickLength + style.Offset
|
||||
values := autoDivide(width, count)
|
||||
maxIndex := len(values) - 2
|
||||
for index, text := range data.TextList() {
|
||||
if index > maxIndex {
|
||||
break
|
||||
}
|
||||
x := values[index]
|
||||
width := values[index+1] - x
|
||||
b := r.MeasureText(text)
|
||||
leftOffset := (width - b.Width()) >> 1
|
||||
d.text(text, x+leftOffset, y0)
|
||||
}
|
||||
default:
|
||||
// 定位bottom，重新计算y0的定位
|
||||
y0 = height - tickLength + style.Offset
|
||||
values := autoDivide(width, count)
|
||||
maxIndex := len(values) - 2
|
||||
for index, text := range data.TextList() {
|
||||
if index > maxIndex {
|
||||
break
|
||||
}
|
||||
x := values[index]
|
||||
width := values[index+1] - x
|
||||
b := r.MeasureText(text)
|
||||
leftOffset := (width - b.Width()) >> 1
|
||||
d.text(text, x+leftOffset, y0)
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
func (d *draw) axisLine(opt *axisOption) {
|
||||
|
||||
r := d.Render
|
||||
style := opt.style
|
||||
s := style.Style()
|
||||
s.GetStrokeOptions().WriteDrawingOptionsToRenderer(r)
|
||||
|
||||
x0 := 0
|
||||
y0 := 0
|
||||
x1 := 0
|
||||
y1 := 0
|
||||
width := d.Box.Width()
|
||||
height := d.Box.Height()
|
||||
labelMargin := style.GetLabelMargin()
|
||||
|
||||
// 轴线
|
||||
labelHeight := labelMargin + opt.textMaxHeight
|
||||
labelWidth := labelMargin + opt.textMaxWith
|
||||
tickLength := style.GetTickLength()
|
||||
switch style.Position {
|
||||
case PositionLeft:
|
||||
x0 = tickLength + style.Offset + labelWidth
|
||||
x1 = x0
|
||||
y0 = 0
|
||||
y1 = height
|
||||
case PositionRight:
|
||||
x0 = width + style.Offset - labelWidth
|
||||
x1 = x0
|
||||
y0 = 0
|
||||
y1 = height
|
||||
case PositionTop:
|
||||
x0 = 0
|
||||
x1 = width
|
||||
y0 = style.Offset + labelHeight
|
||||
y1 = y0
|
||||
// bottom
|
||||
default:
|
||||
x0 = 0
|
||||
x1 = width
|
||||
y0 = height - tickLength + style.Offset - labelHeight
|
||||
y1 = y0
|
||||
}
|
||||
|
||||
d.moveTo(x0, y0)
|
||||
d.lineTo(x1, y1)
|
||||
r.FillStroke()
|
||||
}
|
||||
|
||||
func (d *draw) axisTick(opt *axisOption) {
|
||||
r := d.Render
|
||||
|
||||
style := opt.style
|
||||
s := style.Style()
|
||||
s.GetStrokeOptions().WriteDrawingOptionsToRenderer(r)
|
||||
|
||||
x0 := 0
|
||||
y0 := 0
|
||||
width := d.Box.Width()
|
||||
height := d.Box.Height()
|
||||
data := *opt.data
|
||||
tickCount := len(data)
|
||||
labelMargin := style.GetLabelMargin()
|
||||
|
||||
tickLengthValue := style.GetTickLength()
|
||||
labelHeight := labelMargin + opt.textMaxHeight
|
||||
labelWidth := labelMargin + opt.textMaxWith
|
||||
switch style.Position {
|
||||
case PositionLeft:
|
||||
x0 += labelWidth
|
||||
values := autoDivide(height, tickCount)
|
||||
for _, v := range values {
|
||||
x := x0
|
||||
y := v
|
||||
d.moveTo(x, y)
|
||||
d.lineTo(x+tickLengthValue, y)
|
||||
r.Stroke()
|
||||
}
|
||||
case PositionRight:
|
||||
values := autoDivide(height, tickCount)
|
||||
x0 = width - labelWidth
|
||||
for _, v := range values {
|
||||
x := x0
|
||||
y := v
|
||||
d.moveTo(x, y)
|
||||
d.lineTo(x+tickLengthValue, y)
|
||||
r.Stroke()
|
||||
}
|
||||
case PositionTop:
|
||||
values := autoDivide(width, tickCount)
|
||||
y0 = style.Offset + labelHeight
|
||||
for _, v := range values {
|
||||
x := v
|
||||
y := y0
|
||||
d.moveTo(x, y-tickLengthValue)
|
||||
d.lineTo(x, y)
|
||||
r.Stroke()
|
||||
}
|
||||
|
||||
default:
|
||||
values := autoDivide(width, tickCount)
|
||||
y0 = height + style.Offset - labelHeight
|
||||
for _, v := range values {
|
||||
x := v
|
||||
y := y0
|
||||
d.moveTo(x, y-tickLengthValue)
|
||||
d.lineTo(x, y)
|
||||
r.Stroke()
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
func (d *draw) axisMeasureTextMaxWidthHeight(data AxisDataList, style AxisStyle) (int, int) {
|
||||
r := d.Render
|
||||
s := style.Style()
|
||||
s.GetStrokeOptions().WriteDrawingOptionsToRenderer(r)
|
||||
s.GetTextOptions().WriteTextOptionsToRenderer(r)
|
||||
return measureTextMaxWidthHeight(data.TextList(), r)
|
||||
}
|
||||
|
||||
func (d *draw) Axis(data AxisDataList, style AxisStyle) {
|
||||
if style.Show != nil && !*style.Show {
|
||||
return
|
||||
}
|
||||
r := d.Render
|
||||
s := style.Style()
|
||||
s.GetTextOptions().WriteTextOptionsToRenderer(r)
|
||||
textMaxWidth, textMaxHeight := d.axisMeasureTextMaxWidthHeight(data, style)
|
||||
opt := &axisOption{
|
||||
data: &data,
|
||||
style: &style,
|
||||
textMaxWith: textMaxWidth,
|
||||
textMaxHeight: textMaxHeight,
|
||||
}
|
||||
|
||||
// 坐标轴线
|
||||
d.axisLine(opt)
|
||||
d.axisTick(opt)
|
||||
// 坐标文本
|
||||
d.axisLabel(opt)
|
||||
}
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue