go-chart/style.go

184 lines
3.9 KiB
Go
Raw Normal View History

2016-07-07 03:54:00 +02:00
package chart
2016-07-08 07:18:53 +02:00
import (
"fmt"
"strings"
2016-07-09 19:27:47 +02:00
2016-07-10 10:11:47 +02:00
"github.com/golang/freetype/truetype"
2016-07-09 19:27:47 +02:00
"github.com/wcharczuk/go-chart/drawing"
2016-07-08 07:18:53 +02:00
)
2016-07-07 03:54:00 +02:00
// Style is a simple style set.
type Style struct {
2016-07-10 10:11:47 +02:00
Show bool
Padding Box
2016-07-07 23:44:03 +02:00
StrokeWidth float64
2016-07-10 10:11:47 +02:00
StrokeColor drawing.Color
FillColor drawing.Color
FontSize float64
FontColor drawing.Color
Font *truetype.Font
2016-07-07 03:54:00 +02:00
}
// IsZero returns if the object is set or not.
func (s Style) IsZero() bool {
2016-07-10 10:11:47 +02:00
return s.StrokeColor.IsZero() && s.FillColor.IsZero() && s.StrokeWidth == 0 && s.FontSize == 0 && s.Font == nil
2016-07-07 03:54:00 +02:00
}
// GetStrokeColor returns the stroke color.
2016-07-09 20:23:35 +02:00
func (s Style) GetStrokeColor(defaults ...drawing.Color) drawing.Color {
if s.StrokeColor.IsZero() {
2016-07-07 23:44:03 +02:00
if len(defaults) > 0 {
return defaults[0]
}
2016-07-09 20:23:35 +02:00
return drawing.ColorTransparent
2016-07-07 03:54:00 +02:00
}
return s.StrokeColor
}
// GetFillColor returns the fill color.
2016-07-09 20:23:35 +02:00
func (s Style) GetFillColor(defaults ...drawing.Color) drawing.Color {
if s.FillColor.IsZero() {
2016-07-07 23:44:03 +02:00
if len(defaults) > 0 {
return defaults[0]
}
2016-07-09 20:23:35 +02:00
return drawing.ColorTransparent
2016-07-07 03:54:00 +02:00
}
return s.FillColor
}
// GetStrokeWidth returns the stroke width.
2016-07-07 23:44:03 +02:00
func (s Style) GetStrokeWidth(defaults ...float64) float64 {
2016-07-07 03:54:00 +02:00
if s.StrokeWidth == 0 {
2016-07-07 23:44:03 +02:00
if len(defaults) > 0 {
return defaults[0]
}
2016-07-08 02:50:16 +02:00
return DefaultStrokeWidth
2016-07-07 03:54:00 +02:00
}
return s.StrokeWidth
}
2016-07-07 23:44:03 +02:00
// GetFontSize gets the font size.
func (s Style) GetFontSize(defaults ...float64) float64 {
if s.FontSize == 0 {
if len(defaults) > 0 {
return defaults[0]
}
return DefaultFontSize
}
return s.FontSize
}
// GetFontColor gets the font size.
2016-07-09 20:23:35 +02:00
func (s Style) GetFontColor(defaults ...drawing.Color) drawing.Color {
if s.FontColor.IsZero() {
2016-07-07 23:44:03 +02:00
if len(defaults) > 0 {
return defaults[0]
}
2016-07-09 20:23:35 +02:00
return drawing.ColorTransparent
2016-07-07 23:44:03 +02:00
}
return s.FontColor
}
2016-07-08 07:18:53 +02:00
2016-07-10 10:11:47 +02:00
// GetFont returns the font face.
func (s Style) GetFont(defaults ...*truetype.Font) *truetype.Font {
if s.Font == nil {
if len(defaults) > 0 {
return defaults[0]
}
return nil
}
return s.Font
}
// GetPadding returns the padding.
func (s Style) GetPadding(defaults ...Box) Box {
if s.Padding.IsZero() {
if len(defaults) > 0 {
return defaults[0]
}
return Box{}
}
return s.Padding
}
// WithDefaultsFrom coalesces two styles into a new style.
func (s Style) WithDefaultsFrom(defaults Style) (final Style) {
final.FillColor = s.GetFillColor(defaults.FillColor)
final.FontColor = s.GetFontColor(defaults.FontColor)
final.Font = s.GetFont(defaults.Font)
final.Padding = s.GetPadding(defaults.Padding)
final.StrokeColor = s.GetStrokeColor(defaults.StrokeColor)
final.StrokeWidth = s.GetStrokeWidth(defaults.StrokeWidth)
return
}
2016-07-08 07:18:53 +02:00
// SVG returns the style as a svg style string.
2016-07-09 19:27:47 +02:00
func (s Style) SVG(dpi float64) string {
2016-07-08 07:18:53 +02:00
sw := s.StrokeWidth
sc := s.StrokeColor
fc := s.FillColor
fs := s.FontSize
fnc := s.FontColor
strokeWidthText := "stroke-width:0"
if sw != 0 {
strokeWidthText = "stroke-width:" + fmt.Sprintf("%d", int(sw))
}
strokeText := "stroke:none"
2016-07-09 20:23:35 +02:00
if !sc.IsZero() {
strokeText = "stroke:" + sc.String()
2016-07-08 07:18:53 +02:00
}
fillText := "fill:none"
2016-07-09 20:23:35 +02:00
if !fc.IsZero() {
fillText = "fill:" + fc.String()
2016-07-08 07:18:53 +02:00
}
fontSizeText := ""
if fs != 0 {
2016-07-09 19:27:47 +02:00
fontSizeText = "font-size:" + fmt.Sprintf("%.1fpx", drawing.PointsToPixels(dpi, fs))
2016-07-08 07:18:53 +02:00
}
2016-07-09 20:23:35 +02:00
if !fnc.IsZero() {
fillText = "fill:" + fnc.String()
2016-07-08 07:18:53 +02:00
}
return strings.Join([]string{strokeWidthText, strokeText, fillText, fontSizeText}, ";")
}
2016-07-11 03:09:41 +02:00
// SVGStroke returns the stroke components.
func (s Style) SVGStroke() Style {
return Style{
StrokeColor: s.StrokeColor,
StrokeWidth: s.StrokeWidth,
}
}
// SVGFill returns the fill components.
func (s Style) SVGFill() Style {
return Style{
FillColor: s.FillColor,
}
}
// SVGFillAndStroke returns the fill and stroke components.
func (s Style) SVGFillAndStroke() Style {
return Style{
FillColor: s.FillColor,
StrokeColor: s.StrokeColor,
StrokeWidth: s.StrokeWidth,
}
}
// SVGText returns just the text components of the style.
func (s Style) SVGText() Style {
return Style{
FontColor: s.FontColor,
FontSize: s.FontSize,
}
}