项目文件夹

文件
2026-07-13 13:12:56 +08:00

122 行
3.5 KiB
Go

package svg //nolint:revive
import (
"fmt"
"strconv"
"strings"
"github.com/beevik/etree"
)
// AddShadow adds a definition of a shadow to the <defs> with the given id.
func AddShadow(element *etree.Element, id string, x, y, blur float64) {
f := etree.NewElement("filter")
f.CreateAttr("id", id)
f.CreateAttr("filterUnits", "userSpaceOnUse")
b := etree.NewElement("feGaussianBlur")
b.CreateAttr("in", "SourceAlpha")
b.CreateAttr("stdDeviation", fmt.Sprintf("%.2f", blur))
o := etree.NewElement("feOffset")
o.CreateAttr("result", "offsetblur")
o.CreateAttr("dx", fmt.Sprintf("%.2f", x))
o.CreateAttr("dy", fmt.Sprintf("%.2f", y))
m := etree.NewElement("feMerge")
mn1 := etree.NewElement("feMergeNode")
mn2 := etree.NewElement("feMergeNode")
mn2.CreateAttr("in", "SourceGraphic")
m.AddChild(mn1)
m.AddChild(mn2)
f.AddChild(b)
f.AddChild(o)
f.AddChild(m)
defs := etree.NewElement("defs")
defs.AddChild(f)
element.AddChild(defs)
}
// AddClipPath adds a definition of a clip path to the <defs> with the given id.
func AddClipPath(element *etree.Element, id string, x, y, w, h float64) {
p := etree.NewElement("clipPath")
p.CreateAttr("id", id)
rect := etree.NewElement("rect")
rect.CreateAttr("x", fmt.Sprintf("%.2f", x))
rect.CreateAttr("y", fmt.Sprintf("%.2f", y))
rect.CreateAttr("width", fmt.Sprintf("%.2f", w))
rect.CreateAttr("height", fmt.Sprintf("%.2f", h))
p.AddChild(rect)
defs := etree.NewElement("defs")
defs.AddChild(p)
element.AddChild(defs)
}
// AddCornerRadius adds corner radius to an element.
func AddCornerRadius(e *etree.Element, radius float64) {
e.CreateAttr("rx", fmt.Sprintf("%.2f", radius))
e.CreateAttr("ry", fmt.Sprintf("%.2f", radius))
}
// Move moves the given element to the (x, y) position.
func Move(e *etree.Element, x, y float64) {
e.CreateAttr("x", fmt.Sprintf("%.2fpx", x))
e.CreateAttr("y", fmt.Sprintf("%.2fpx", y))
}
// AddOutline adds an outline to the given element.
func AddOutline(e *etree.Element, width float64, color string) {
e.CreateAttr("stroke", color)
e.CreateAttr("stroke-width", fmt.Sprintf("%.2f", width))
}
const (
red string = "#FF5A54"
yellow string = "#E6BF29"
green string = "#52C12B"
)
// NewWindowControls returns a colorful window bar element.
func NewWindowControls(r float64, x, y float64) *etree.Element {
bar := etree.NewElement("svg")
for i, color := range []string{red, yellow, green} {
circle := etree.NewElement("circle")
circle.CreateAttr("cx", fmt.Sprintf("%.2f", float64(i+1)*float64(x)-float64(r)))
circle.CreateAttr("cy", fmt.Sprintf("%.2f", y))
circle.CreateAttr("r", fmt.Sprintf("%.2f", r))
circle.CreateAttr("fill", color)
bar.AddChild(circle)
}
return bar
}
// SetDimensions sets the width and height of the given element.
func SetDimensions(element *etree.Element, width, height float64) {
widthAttr := element.SelectAttr("width")
heightAttr := element.SelectAttr("height")
heightAttr.Value = fmt.Sprintf("%.2f", height)
widthAttr.Value = fmt.Sprintf("%.2f", width)
}
// GetDimensions returns the width and height of the element.
func GetDimensions(element *etree.Element) (int, int) {
widthValue := element.SelectAttrValue("width", "0px")
heightValue := element.SelectAttrValue("height", "0px")
width := dimensionToInt(widthValue)
height := dimensionToInt(heightValue)
return width, height
}
// dimensionToInt takes a string and returns the integer value.
// e.g. "500px" -> 500.
func dimensionToInt(px string) int {
d := strings.TrimSuffix(px, "px")
v, _ := strconv.ParseInt(d, 10, 64)
return int(v)
}