refactor: reset
This commit is contained in:
vicanso
parent
7e80e9a848
commit
c363d1d5e3
50 changed files with 55 additions and 10282 deletions
472
axis.go
472
axis.go
|
|
@ -1,472 +0,0 @@
|
|||
// MIT License
|
||||
|
||||
// Copyright (c) 2022 Tree Xie
|
||||
|
||||
// Permission is hereby granted, free of charge, to any person obtaining a copy
|
||||
// of this software and associated documentation files (the "Software"), to deal
|
||||
// in the Software without restriction, including without limitation the rights
|
||||
// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
|
||||
// copies of the Software, and to permit persons to whom the Software is
|
||||
// furnished to do so, subject to the following conditions:
|
||||
|
||||
// The above copyright notice and this permission notice shall be included in all
|
||||
// copies or substantial portions of the Software.
|
||||
|
||||
// THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
|
||||
// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
|
||||
// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
|
||||
// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
|
||||
// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
|
||||
// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
|
||||
// SOFTWARE.
|
||||
|
||||
package charts
|
||||
|
||||
import (
|
||||
"math"
|
||||
|
||||
"github.com/golang/freetype/truetype"
|
||||
"github.com/wcharczuk/go-chart/v2"
|
||||
)
|
||||
|
||||
type AxisOption struct {
|
||||
// The boundary gap on both sides of a coordinate axis.
|
||||
// Nil or *true means the center part of two axis ticks
|
||||
BoundaryGap *bool
|
||||
// The flag for show axis, set this to *false will hide axis
|
||||
Show *bool
|
||||
// The position of axis, it can be 'left', 'top', 'right' or 'bottom'
|
||||
Position string
|
||||
// Number of segments that the axis is split into. Note that this number serves only as a recommendation.
|
||||
SplitNumber int
|
||||
ClassName string
|
||||
// The line color of axis
|
||||
StrokeColor Color
|
||||
// The line width
|
||||
StrokeWidth float64
|
||||
// The length of the axis tick
|
||||
TickLength int
|
||||
// The flag for show axis tick, set this to *false will hide axis tick
|
||||
TickShow *bool
|
||||
// The margin value of label
|
||||
LabelMargin int
|
||||
// The font size of label
|
||||
FontSize float64
|
||||
// The font of label
|
||||
Font *truetype.Font
|
||||
// The color of label
|
||||
FontColor Color
|
||||
// The flag for show axis split line, set this to true will show axis split line
|
||||
SplitLineShow bool
|
||||
// The color of split line
|
||||
SplitLineColor Color
|
||||
}
|
||||
|
||||
type axis struct {
|
||||
painter *Painter
|
||||
data *AxisDataList
|
||||
option *AxisOption
|
||||
}
|
||||
type axisMeasurement struct {
|
||||
Width int
|
||||
Height int
|
||||
}
|
||||
|
||||
// NewAxis creates a new axis with data and style options
|
||||
func NewAxis(p *Painter, data AxisDataList, option AxisOption) *axis {
|
||||
return &axis{
|
||||
painter: p,
|
||||
data: &data,
|
||||
option: &option,
|
||||
}
|
||||
|
||||
}
|
||||
|
||||
// GetLabelMargin returns the label margin value
|
||||
func (as *AxisOption) GetLabelMargin() int {
|
||||
return getDefaultInt(as.LabelMargin, 8)
|
||||
}
|
||||
|
||||
// GetTickLength returns the tick length value
|
||||
func (as *AxisOption) GetTickLength() int {
|
||||
return getDefaultInt(as.TickLength, 5)
|
||||
}
|
||||
|
||||
// Style returns the style of axis
|
||||
func (as *AxisOption) Style(f *truetype.Font) chart.Style {
|
||||
s := chart.Style{
|
||||
ClassName: as.ClassName,
|
||||
StrokeColor: as.StrokeColor,
|
||||
StrokeWidth: as.StrokeWidth,
|
||||
FontSize: as.FontSize,
|
||||
FontColor: as.FontColor,
|
||||
Font: as.Font,
|
||||
}
|
||||
if s.FontSize == 0 {
|
||||
s.FontSize = chart.DefaultFontSize
|
||||
}
|
||||
if s.Font == nil {
|
||||
s.Font = f
|
||||
}
|
||||
return s
|
||||
}
|
||||
|
||||
type AxisData struct {
|
||||
// The text value of axis
|
||||
Text string
|
||||
}
|
||||
type AxisDataList []AxisData
|
||||
|
||||
// TextList returns the text list of axis data
|
||||
func (l AxisDataList) TextList() []string {
|
||||
textList := make([]string, len(l))
|
||||
for index, item := range l {
|
||||
textList[index] = item.Text
|
||||
}
|
||||
return textList
|
||||
}
|
||||
|
||||
type axisRenderOption struct {
|
||||
textMaxWith int
|
||||
textMaxHeight int
|
||||
boundaryGap bool
|
||||
unitCount int
|
||||
modValue int
|
||||
}
|
||||
|
||||
// NewAxisDataListFromStringList creates a new axis data list from string list
|
||||
func NewAxisDataListFromStringList(textList []string) AxisDataList {
|
||||
list := make(AxisDataList, len(textList))
|
||||
for index, text := range textList {
|
||||
list[index] = AxisData{
|
||||
Text: text,
|
||||
}
|
||||
}
|
||||
return list
|
||||
}
|
||||
|
||||
func (a *axis) axisLabel(renderOpt *axisRenderOption) {
|
||||
option := a.option
|
||||
data := *a.data
|
||||
// d := a.d
|
||||
if option.FontColor.IsZero() || len(data) == 0 {
|
||||
return
|
||||
}
|
||||
// r := d.Render
|
||||
|
||||
// s.GetTextOptions().WriteTextOptionsToRenderer(r)
|
||||
p := a.painter
|
||||
s := option.Style(p.font)
|
||||
p.SetTextStyle(s)
|
||||
|
||||
width := p.Width()
|
||||
height := p.Height()
|
||||
textList := data.TextList()
|
||||
count := len(textList)
|
||||
|
||||
boundaryGap := renderOpt.boundaryGap
|
||||
if !boundaryGap {
|
||||
count--
|
||||
}
|
||||
|
||||
unitCount := renderOpt.unitCount
|
||||
modValue := renderOpt.modValue
|
||||
labelMargin := option.GetLabelMargin()
|
||||
|
||||
// 轴线
|
||||
labelHeight := labelMargin + renderOpt.textMaxHeight
|
||||
labelWidth := labelMargin + renderOpt.textMaxWith
|
||||
|
||||
// 坐标轴文本
|
||||
position := option.Position
|
||||
switch position {
|
||||
case PositionLeft:
|
||||
fallthrough
|
||||
case PositionRight:
|
||||
values := autoDivide(height, count)
|
||||
textList := data.TextList()
|
||||
// 由下往上
|
||||
reverseIntSlice(values)
|
||||
for index, text := range textList {
|
||||
y := values[index] - 2
|
||||
b := p.MeasureText(text)
|
||||
if boundaryGap {
|
||||
height := y - values[index+1]
|
||||
y -= (height - b.Height()) >> 1
|
||||
} else {
|
||||
y += b.Height() >> 1
|
||||
}
|
||||
// 左右位置的x不一样
|
||||
x := width - renderOpt.textMaxWith
|
||||
if position == PositionLeft {
|
||||
x = labelWidth - b.Width() - 1
|
||||
}
|
||||
p.Text(text, x, y)
|
||||
}
|
||||
default:
|
||||
// 定位bottom，重新计算y0的定位
|
||||
y0 := height - labelMargin
|
||||
if position == PositionTop {
|
||||
y0 = labelHeight - labelMargin
|
||||
}
|
||||
values := autoDivide(width, count)
|
||||
for index, text := range data.TextList() {
|
||||
if unitCount != 0 && index%unitCount != modValue {
|
||||
continue
|
||||
}
|
||||
x := values[index]
|
||||
leftOffset := 0
|
||||
b := p.MeasureText(text)
|
||||
if boundaryGap {
|
||||
width := values[index+1] - x
|
||||
leftOffset = (width - b.Width()) >> 1
|
||||
} else {
|
||||
// 左移文本长度
|
||||
leftOffset = -b.Width() >> 1
|
||||
}
|
||||
p.Text(text, x+leftOffset, y0)
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
func (a *axis) axisLine(renderOpt *axisRenderOption) {
|
||||
// d := a.d
|
||||
// r := d.Render
|
||||
p := a.painter
|
||||
option := a.option
|
||||
s := option.Style(p.font)
|
||||
p.SetDrawingStyle(s.GetStrokeOptions())
|
||||
|
||||
x0 := 0
|
||||
y0 := 0
|
||||
x1 := 0
|
||||
y1 := 0
|
||||
width := p.Width()
|
||||
height := p.Height()
|
||||
labelMargin := option.GetLabelMargin()
|
||||
|
||||
// 轴线
|
||||
labelHeight := labelMargin + renderOpt.textMaxHeight
|
||||
labelWidth := labelMargin + renderOpt.textMaxWith
|
||||
tickLength := option.GetTickLength()
|
||||
switch option.Position {
|
||||
case PositionLeft:
|
||||
x0 = tickLength + labelWidth
|
||||
x1 = x0
|
||||
y0 = 0
|
||||
y1 = height
|
||||
case PositionRight:
|
||||
x0 = width - labelWidth
|
||||
x1 = x0
|
||||
y0 = 0
|
||||
y1 = height
|
||||
case PositionTop:
|
||||
x0 = 0
|
||||
x1 = width
|
||||
y0 = labelHeight
|
||||
y1 = y0
|
||||
// bottom
|
||||
default:
|
||||
x0 = 0
|
||||
x1 = width
|
||||
y0 = height - tickLength - labelHeight
|
||||
y1 = y0
|
||||
}
|
||||
|
||||
p.MoveTo(x0, y0)
|
||||
p.LineTo(x1, y1)
|
||||
p.FillStroke()
|
||||
}
|
||||
|
||||
func (a *axis) axisTick(renderOpt *axisRenderOption) {
|
||||
// d := a.d
|
||||
// r := d.Render
|
||||
|
||||
p := a.painter
|
||||
option := a.option
|
||||
s := option.Style(p.font)
|
||||
p.SetDrawingStyle(s.GetStrokeOptions())
|
||||
|
||||
width := p.Width()
|
||||
height := p.Height()
|
||||
data := *a.data
|
||||
tickCount := len(data)
|
||||
if tickCount == 0 {
|
||||
return
|
||||
}
|
||||
if !renderOpt.boundaryGap {
|
||||
tickCount--
|
||||
}
|
||||
labelMargin := option.GetLabelMargin()
|
||||
tickShow := true
|
||||
if isFalse(option.TickShow) {
|
||||
tickShow = false
|
||||
}
|
||||
unitCount := renderOpt.unitCount
|
||||
|
||||
tickLengthValue := option.GetTickLength()
|
||||
labelHeight := labelMargin + renderOpt.textMaxHeight
|
||||
labelWidth := labelMargin + renderOpt.textMaxWith
|
||||
position := option.Position
|
||||
switch position {
|
||||
case PositionLeft:
|
||||
fallthrough
|
||||
case PositionRight:
|
||||
values := autoDivide(height, tickCount)
|
||||
// 左右仅是x0的位置不一样
|
||||
x0 := width - labelWidth
|
||||
if option.Position == PositionLeft {
|
||||
x0 = labelWidth
|
||||
}
|
||||
if tickShow {
|
||||
for _, v := range values {
|
||||
x := x0
|
||||
y := v
|
||||
p.MoveTo(x, y)
|
||||
p.LineTo(x+tickLengthValue, y)
|
||||
p.Stroke()
|
||||
}
|
||||
}
|
||||
// 辅助线
|
||||
if option.SplitLineShow && !option.SplitLineColor.IsZero() {
|
||||
p.SetStrokeColor(option.SplitLineColor)
|
||||
splitLineWidth := width - labelWidth - tickLengthValue
|
||||
x0 = labelWidth + tickLengthValue
|
||||
if position == PositionRight {
|
||||
x0 = 0
|
||||
splitLineWidth = width - labelWidth - 1
|
||||
}
|
||||
for _, v := range values[0 : len(values)-1] {
|
||||
x := x0
|
||||
y := v
|
||||
p.MoveTo(x, y)
|
||||
p.LineTo(x+splitLineWidth, y)
|
||||
p.Stroke()
|
||||
}
|
||||
}
|
||||
default:
|
||||
values := autoDivide(width, tickCount)
|
||||
// 上下仅是y0的位置不一样
|
||||
y0 := height - labelHeight
|
||||
if position == PositionTop {
|
||||
y0 = labelHeight
|
||||
}
|
||||
if tickShow {
|
||||
for index, v := range values {
|
||||
if index%unitCount != 0 {
|
||||
continue
|
||||
}
|
||||
x := v
|
||||
y := y0
|
||||
p.MoveTo(x, y-tickLengthValue)
|
||||
p.LineTo(x, y)
|
||||
p.Stroke()
|
||||
}
|
||||
}
|
||||
// 辅助线
|
||||
if option.SplitLineShow && !option.SplitLineColor.IsZero() {
|
||||
p.SetStrokeColor(option.SplitLineColor)
|
||||
y0 = 0
|
||||
splitLineHeight := height - labelHeight - tickLengthValue
|
||||
if position == PositionTop {
|
||||
y0 = labelHeight
|
||||
splitLineHeight = height - labelHeight
|
||||
}
|
||||
|
||||
for index, v := range values {
|
||||
if index%unitCount != 0 {
|
||||
continue
|
||||
}
|
||||
x := v
|
||||
y := y0
|
||||
|
||||
p.MoveTo(x, y)
|
||||
p.LineTo(x, y0+splitLineHeight)
|
||||
p.Stroke()
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
func (a *axis) measureTextMaxWidthHeight() (int, int) {
|
||||
// d := a.d
|
||||
// r := d.Render
|
||||
p := a.painter
|
||||
s := a.option.Style(p.font)
|
||||
data := a.data
|
||||
p.SetDrawingStyle(s.GetStrokeOptions())
|
||||
p.SetTextStyle(s.GetTextOptions())
|
||||
return measureTextMaxWidthHeight(data.TextList(), p)
|
||||
}
|
||||
|
||||
// measure returns the measurement of axis.
|
||||
// Width will be textMaxWidth + labelMargin + tickLength for position left or right.
|
||||
// Height will be textMaxHeight + labelMargin + tickLength for position top or bottom.
|
||||
func (a *axis) measure() axisMeasurement {
|
||||
option := a.option
|
||||
value := option.GetLabelMargin() + option.GetTickLength()
|
||||
textMaxWidth, textMaxHeight := a.measureTextMaxWidthHeight()
|
||||
info := axisMeasurement{}
|
||||
if option.Position == PositionLeft ||
|
||||
option.Position == PositionRight {
|
||||
info.Width = textMaxWidth + value
|
||||
} else {
|
||||
info.Height = textMaxHeight + value
|
||||
}
|
||||
return info
|
||||
}
|
||||
|
||||
// Render renders the axis for chart
|
||||
func (a *axis) Render() {
|
||||
option := a.option
|
||||
if isFalse(option.Show) {
|
||||
return
|
||||
}
|
||||
textMaxWidth, textMaxHeight := a.measureTextMaxWidthHeight()
|
||||
opt := &axisRenderOption{
|
||||
textMaxWith: textMaxWidth,
|
||||
textMaxHeight: textMaxHeight,
|
||||
boundaryGap: true,
|
||||
}
|
||||
if isFalse(option.BoundaryGap) {
|
||||
opt.boundaryGap = false
|
||||
}
|
||||
|
||||
unitCount := chart.MaxInt(option.SplitNumber, 1)
|
||||
width := a.painter.Width()
|
||||
textList := a.data.TextList()
|
||||
count := len(textList)
|
||||
|
||||
position := option.Position
|
||||
switch position {
|
||||
case PositionLeft:
|
||||
fallthrough
|
||||
case PositionRight:
|
||||
default:
|
||||
maxCount := width / (opt.textMaxWith + 10)
|
||||
// 可以显示所有
|
||||
if maxCount >= count {
|
||||
unitCount = 1
|
||||
} else if maxCount < count/unitCount {
|
||||
unitCount = int(math.Ceil(float64(count) / float64(maxCount)))
|
||||
}
|
||||
}
|
||||
|
||||
boundaryGap := opt.boundaryGap
|
||||
modValue := 0
|
||||
if boundaryGap && unitCount > 1 {
|
||||
// 如果是居中，unit count需要设置为奇数
|
||||
if unitCount%2 == 0 {
|
||||
unitCount++
|
||||
}
|
||||
modValue = unitCount / 2
|
||||
}
|
||||
opt.modValue = modValue
|
||||
opt.unitCount = unitCount
|
||||
|
||||
// 坐标轴线
|
||||
a.axisLine(opt)
|
||||
a.axisTick(opt)
|
||||
// 坐标文本
|
||||
a.axisLabel(opt)
|
||||
}
|
||||
Loading…
Add table
Add a link
Reference in a new issue