go-chart/style.go

package chart

import (
	"fmt"
	"image/color"
	"strings"

	"github.com/wcharczuk/go-chart/drawing"
)

// Style is a simple style set.
type Style struct {
	Show        bool
	StrokeColor color.RGBA
	FillColor   color.RGBA
	StrokeWidth float64
	FontSize    float64
	FontColor   color.RGBA
	Padding     Box
}

// IsZero returns if the object is set or not.
func (s Style) IsZero() bool {
	return ColorIsZero(s.StrokeColor) && ColorIsZero(s.FillColor) && s.StrokeWidth == 0 && s.FontSize == 0
}

// GetStrokeColor returns the stroke color.
func (s Style) GetStrokeColor(defaults ...color.RGBA) color.RGBA {
	if ColorIsZero(s.StrokeColor) {
		if len(defaults) > 0 {
			return defaults[0]
		}
		return color.RGBA{}
	}
	return s.StrokeColor
}

// GetFillColor returns the fill color.
func (s Style) GetFillColor(defaults ...color.RGBA) color.RGBA {
	if ColorIsZero(s.FillColor) {
		if len(defaults) > 0 {
			return defaults[0]
		}
		return color.RGBA{}
	}
	return s.FillColor
}

// GetStrokeWidth returns the stroke width.
func (s Style) GetStrokeWidth(defaults ...float64) float64 {
	if s.StrokeWidth == 0 {
		if len(defaults) > 0 {
			return defaults[0]
		}
		return DefaultStrokeWidth
	}
	return s.StrokeWidth
}

// 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.
func (s Style) GetFontColor(defaults ...color.RGBA) color.RGBA {
	if ColorIsZero(s.FontColor) {
		if len(defaults) > 0 {
			return defaults[0]
		}
		return color.RGBA{}
	}
	return s.FontColor
}

// SVG returns the style as a svg style string.
func (s Style) SVG(dpi float64) string {
	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"
	if !ColorIsZero(sc) {
		strokeText = "stroke:" + ColorAsString(sc)
	}

	fillText := "fill:none"
	if !ColorIsZero(fc) {
		fillText = "fill:" + ColorAsString(fc)
	}

	fontSizeText := ""
	if fs != 0 {
		fontSizeText = "font-size:" + fmt.Sprintf("%.1fpx", drawing.PointsToPixels(dpi, fs))
	}

	if !ColorIsZero(fnc) {
		fillText = "fill:" + ColorAsString(fnc)
	}
	return strings.Join([]string{strokeWidthText, strokeText, fillText, fontSizeText}, ";")
}