go-chart/style.go

325 lines
8.3 KiB
Go
Raw Permalink 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
2016-08-01 01:54:09 +02:00
// StyleShow is a prebuilt style with the `Show` property set to true.
func StyleShow() Style {
return Style{
Show: true,
}
}
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-12 03:48:51 +02:00
StrokeWidth float64
StrokeColor drawing.Color
StrokeDashArray []float64
FillColor drawing.Color
FontSize float64
FontColor drawing.Color
Font *truetype.Font
2016-07-30 01:36:29 +02:00
2016-08-01 01:54:09 +02:00
TextHorizontalAlign TextHorizontalAlign
TextVerticalAlign TextVerticalAlign
TextWrap TextWrap
2016-07-30 03:24:25 +02:00
TextLineSpacing int
2016-07-07 03:54:00 +02:00
}
// IsZero returns if the object is set or not.
func (s Style) IsZero() bool {
2016-07-11 08:06:14 +02:00
return s.StrokeColor.IsZero() && s.FillColor.IsZero() && s.StrokeWidth == 0 && s.FontColor.IsZero() && s.FontSize == 0 && s.Font == nil
2016-07-07 03:54:00 +02:00
}
// String returns a text representation of the style.
2016-07-18 20:43:41 +02:00
func (s Style) String() string {
if s.IsZero() {
return "{}"
}
var output []string
if s.Show {
output = []string{"\"show\": true"}
} else {
output = []string{"\"show\": false"}
}
if !s.Padding.IsZero() {
output = append(output, fmt.Sprintf("\"padding\": %s", s.Padding.String()))
} else {
output = append(output, "\"padding\": null")
}
if s.StrokeWidth >= 0 {
output = append(output, fmt.Sprintf("\"stroke_width\": %0.2f", s.StrokeWidth))
} else {
output = append(output, "\"stroke_width\": null")
}
if !s.StrokeColor.IsZero() {
output = append(output, fmt.Sprintf("\"stroke_color\": %s", s.StrokeColor.String()))
} else {
output = append(output, "\"stroke_color\": null")
}
if len(s.StrokeDashArray) > 0 {
var elements []string
for _, v := range s.StrokeDashArray {
elements = append(elements, fmt.Sprintf("%.2f", v))
}
dashArray := strings.Join(elements, ", ")
output = append(output, fmt.Sprintf("\"stroke_dash_array\": [%s]", dashArray))
} else {
output = append(output, "\"stroke_dash_array\": null")
}
if !s.FillColor.IsZero() {
output = append(output, fmt.Sprintf("\"fill_color\": %s", s.FillColor.String()))
} else {
output = append(output, "\"fill_color\": null")
}
if s.FontSize != 0 {
output = append(output, fmt.Sprintf("\"font_size\": \"%0.2fpt\"", s.FontSize))
} else {
2016-07-28 11:34:44 +02:00
output = append(output, "\"font_size\": null")
2016-07-18 20:43:41 +02:00
}
2016-07-28 11:34:44 +02:00
if !s.FontColor.IsZero() {
output = append(output, fmt.Sprintf("\"font_color\": %s", s.FontColor.String()))
2016-07-18 20:43:41 +02:00
} else {
output = append(output, "\"font_color\": null")
}
if s.Font != nil {
output = append(output, fmt.Sprintf("\"font\": \"%s\"", s.Font.Name(truetype.NameIDFontFamily)))
} else {
output = append(output, "\"font_color\": null")
}
return "{" + strings.Join(output, ", ") + "}"
}
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
2016-07-12 03:48:51 +02:00
// GetStrokeDashArray returns the stroke dash array.
func (s Style) GetStrokeDashArray(defaults ...[]float64) []float64 {
if len(s.StrokeDashArray) == 0 {
if len(defaults) > 0 {
return defaults[0]
}
return nil
}
return s.StrokeDashArray
}
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
}
2016-07-30 01:36:29 +02:00
// GetTextHorizontalAlign returns the horizontal alignment.
2016-08-01 01:54:09 +02:00
func (s Style) GetTextHorizontalAlign(defaults ...TextHorizontalAlign) TextHorizontalAlign {
2016-07-30 01:36:29 +02:00
if s.TextHorizontalAlign == TextHorizontalAlignUnset {
if len(defaults) > 0 {
return defaults[0]
}
2016-07-30 03:40:43 +02:00
return TextHorizontalAlignUnset
2016-07-30 01:36:29 +02:00
}
return s.TextHorizontalAlign
}
// GetTextVerticalAlign returns the vertical alignment.
2016-08-01 01:54:09 +02:00
func (s Style) GetTextVerticalAlign(defaults ...TextVerticalAlign) TextVerticalAlign {
2016-07-30 01:36:29 +02:00
if s.TextVerticalAlign == TextVerticalAlignUnset {
if len(defaults) > 0 {
return defaults[0]
}
2016-07-30 03:40:43 +02:00
return TextVerticalAlignUnset
2016-07-30 01:36:29 +02:00
}
return s.TextVerticalAlign
}
// GetTextWrap returns the word wrap.
2016-08-01 01:54:09 +02:00
func (s Style) GetTextWrap(defaults ...TextWrap) TextWrap {
2016-07-30 01:36:29 +02:00
if s.TextWrap == TextWrapUnset {
if len(defaults) > 0 {
return defaults[0]
}
2016-07-30 03:40:43 +02:00
return TextWrapUnset
2016-07-30 01:36:29 +02:00
}
return s.TextWrap
}
2016-07-30 03:24:25 +02:00
// GetTextLineSpacing returns the spacing in pixels between lines of text (vertically).
func (s Style) GetTextLineSpacing(defaults ...int) int {
if s.TextLineSpacing == 0 {
if len(defaults) > 0 {
return defaults[0]
}
return DefaultLineSpacing
}
return s.TextLineSpacing
}
2016-07-30 01:36:29 +02:00
// WriteToRenderer passes the style's options to a renderer.
func (s Style) WriteToRenderer(r Renderer) {
r.SetStrokeColor(s.GetStrokeColor())
r.SetStrokeWidth(s.GetStrokeWidth())
r.SetStrokeDashArray(s.GetStrokeDashArray())
r.SetFillColor(s.GetFillColor())
r.SetFont(s.GetFont())
r.SetFontColor(s.GetFontColor())
r.SetFontSize(s.GetFontSize())
}
// WriteDrawingOptionsToRenderer passes just the drawing style options to a renderer.
func (s Style) WriteDrawingOptionsToRenderer(r Renderer) {
r.SetStrokeColor(s.GetStrokeColor())
r.SetStrokeWidth(s.GetStrokeWidth())
r.SetStrokeDashArray(s.GetStrokeDashArray())
2016-07-28 11:34:44 +02:00
r.SetFillColor(s.GetFillColor())
2016-07-30 01:36:29 +02:00
}
// WriteTextOptionsToRenderer passes just the text style options to a renderer.
func (s Style) WriteTextOptionsToRenderer(r Renderer) {
r.SetFont(s.GetFont())
r.SetFontColor(s.GetFontColor())
r.SetFontSize(s.GetFontSize())
}
// InheritFrom coalesces two styles into a new style.
func (s Style) InheritFrom(defaults Style) (final Style) {
2016-07-11 08:06:14 +02:00
final.StrokeColor = s.GetStrokeColor(defaults.StrokeColor)
final.StrokeWidth = s.GetStrokeWidth(defaults.StrokeWidth)
2016-07-14 18:27:23 +02:00
final.StrokeDashArray = s.GetStrokeDashArray(defaults.StrokeDashArray)
2016-07-10 10:11:47 +02:00
final.FillColor = s.GetFillColor(defaults.FillColor)
final.FontColor = s.GetFontColor(defaults.FontColor)
2016-07-15 04:45:23 +02:00
final.FontSize = s.GetFontSize(defaults.FontSize)
2016-07-10 10:11:47 +02:00
final.Font = s.GetFont(defaults.Font)
final.Padding = s.GetPadding(defaults.Padding)
2016-07-30 01:36:29 +02:00
final.TextHorizontalAlign = s.GetTextHorizontalAlign(defaults.TextHorizontalAlign)
final.TextVerticalAlign = s.GetTextVerticalAlign(defaults.TextVerticalAlign)
final.TextWrap = s.GetTextWrap(defaults.TextWrap)
2016-07-30 03:24:25 +02:00
final.TextLineSpacing = s.GetTextLineSpacing(defaults.TextLineSpacing)
2016-07-10 10:11:47 +02:00
return
}
2016-07-30 01:36:29 +02:00
// GetStrokeOptions returns the stroke components.
func (s Style) GetStrokeOptions() Style {
2016-07-11 03:09:41 +02:00
return Style{
StrokeDashArray: s.StrokeDashArray,
StrokeColor: s.StrokeColor,
StrokeWidth: s.StrokeWidth,
2016-07-11 03:09:41 +02:00
}
}
2016-07-30 01:36:29 +02:00
// GetFillOptions returns the fill components.
func (s Style) GetFillOptions() Style {
2016-07-11 03:09:41 +02:00
return Style{
FillColor: s.FillColor,
}
}
2016-07-30 01:36:29 +02:00
// GetFillAndStrokeOptions returns the fill and stroke components.
func (s Style) GetFillAndStrokeOptions() Style {
2016-07-11 03:09:41 +02:00
return Style{
StrokeDashArray: s.StrokeDashArray,
FillColor: s.FillColor,
StrokeColor: s.StrokeColor,
StrokeWidth: s.StrokeWidth,
2016-07-11 03:09:41 +02:00
}
}
2016-07-30 01:36:29 +02:00
// GetTextOptions returns just the text components of the style.
func (s Style) GetTextOptions() Style {
2016-07-11 03:09:41 +02:00
return Style{
2016-07-30 01:36:29 +02:00
FontColor: s.FontColor,
FontSize: s.FontSize,
Font: s.Font,
TextHorizontalAlign: s.TextHorizontalAlign,
TextVerticalAlign: s.TextVerticalAlign,
TextWrap: s.TextWrap,
2016-07-30 03:24:25 +02:00
TextLineSpacing: s.TextLineSpacing,
}
}