98 lines
2.6 KiB
Go
98 lines
2.6 KiB
Go
// MIT License
|
|
|
|
// Copyright (c) 2021 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 (
|
|
"github.com/wcharczuk/go-chart/v2"
|
|
"github.com/wcharczuk/go-chart/v2/drawing"
|
|
)
|
|
|
|
type Series struct {
|
|
Type string
|
|
Name string
|
|
Data []float64
|
|
XValues []float64
|
|
}
|
|
|
|
const lineStrokeWidth = 2
|
|
const dotWith = 2
|
|
|
|
const (
|
|
SeriesBar = "bar"
|
|
SeriesLine = "line"
|
|
)
|
|
|
|
func getSeriesColor(theme string, index int) drawing.Color {
|
|
// TODO
|
|
if theme == ThemeDark {
|
|
}
|
|
return SeriesColorsLight[index%len(SeriesColorsLight)]
|
|
}
|
|
|
|
func GetSeries(series []Series, tickPosition chart.TickPosition, theme string) []chart.Series {
|
|
arr := make([]chart.Series, len(series))
|
|
barCount := 0
|
|
barIndex := 0
|
|
for _, item := range series {
|
|
if item.Type == SeriesBar {
|
|
barCount++
|
|
}
|
|
}
|
|
for index, item := range series {
|
|
style := chart.Style{
|
|
StrokeWidth: lineStrokeWidth,
|
|
StrokeColor: getSeriesColor(theme, index),
|
|
// TODO 调整为通过dot with color 生成
|
|
DotColor: getSeriesColor(theme, index),
|
|
DotWidth: dotWith,
|
|
}
|
|
// 如果居中，需要多增加一个点
|
|
if tickPosition == chart.TickPositionBetweenTicks {
|
|
item.Data = append([]float64{
|
|
0.0,
|
|
}, item.Data...)
|
|
}
|
|
baseSeries := BaseSeries{
|
|
Name: item.Name,
|
|
XValues: item.XValues,
|
|
Style: style,
|
|
YValues: item.Data,
|
|
TickPosition: tickPosition,
|
|
}
|
|
// TODO 判断类型
|
|
switch item.Type {
|
|
case SeriesBar:
|
|
arr[index] = BarSeries{
|
|
Count: barCount,
|
|
Index: barIndex,
|
|
BaseSeries: baseSeries,
|
|
}
|
|
barIndex++
|
|
default:
|
|
arr[index] = LineSeries{
|
|
BaseSeries: baseSeries,
|
|
}
|
|
}
|
|
}
|
|
return arr
|
|
}
|