126 lines
3.4 KiB
Go
126 lines
3.4 KiB
Go
package animation
|
||
|
||
import (
|
||
"strings"
|
||
)
|
||
|
||
// AnimationType 表示动画类型
|
||
type AnimationType string
|
||
|
||
// 预定义动画类型常量
|
||
const (
|
||
Rotate AnimationType = "rotate" // 旋转动画
|
||
Gradient AnimationType = "gradient" // 渐变动画
|
||
Transform AnimationType = "transform" // 变换动画
|
||
Fade AnimationType = "fade" // 淡入淡出动画
|
||
Path AnimationType = "path" // 路径动画
|
||
Color AnimationType = "color" // 颜色变换动画
|
||
Bounce AnimationType = "bounce" // 弹跳动画
|
||
Wave AnimationType = "wave" // 波浪动画
|
||
Blink AnimationType = "blink" // 闪烁动画
|
||
)
|
||
|
||
// Animation 表示一个SVG动画接口
|
||
type Animation interface {
|
||
// GenerateSVG 生成动画的SVG代码
|
||
GenerateSVG() string
|
||
// GetTargetID 获取目标元素ID
|
||
GetTargetID() string
|
||
}
|
||
|
||
// BaseAnimation 基础动画结构，包含所有动画共有的属性
|
||
type BaseAnimation struct {
|
||
Type AnimationType // 动画类型
|
||
Duration float64 // 动画持续时间（秒）
|
||
RepeatCount int // 重复次数，-1表示无限重复
|
||
Delay float64 // 延迟时间（秒）
|
||
TargetID string // 目标元素ID
|
||
Attributes map[string]string // 动画属性
|
||
}
|
||
|
||
// GetTargetID 获取目标元素ID
|
||
func (a *BaseAnimation) GetTargetID() string {
|
||
return a.TargetID
|
||
}
|
||
|
||
// Manager 动画管理器，负责管理所有动画
|
||
type Manager struct {
|
||
animations []Animation
|
||
}
|
||
|
||
// GetAnimations 获取所有动画
|
||
func (m *Manager) GetAnimations() []Animation {
|
||
return m.animations
|
||
}
|
||
|
||
// NewAnimationManager 创建一个新的动画管理器
|
||
func NewAnimationManager() *Manager {
|
||
return &Manager{
|
||
animations: make([]Animation, 0),
|
||
}
|
||
}
|
||
|
||
// AddAnimation 添加一个动画
|
||
func (m *Manager) AddAnimation(animation Animation) {
|
||
m.animations = append(m.animations, animation)
|
||
}
|
||
|
||
// GenerateSVGAnimations 生成SVG动画代码
|
||
func (m *Manager) GenerateSVGAnimations() string {
|
||
if len(m.animations) == 0 {
|
||
return ""
|
||
}
|
||
|
||
var sb strings.Builder
|
||
|
||
// 添加SVG命名空间声明
|
||
sb.WriteString("<defs>\n")
|
||
|
||
// 用于存储需要放在defs中的定义
|
||
var defsContent strings.Builder
|
||
// 用于存储需要直接添加到SVG中的动画元素
|
||
var animationsContent strings.Builder
|
||
// 用于存储旋转动画的映射，键为目标元素ID
|
||
rotateAnimations := make(map[string]string)
|
||
|
||
// 处理所有动画
|
||
for _, anim := range m.animations {
|
||
svgCode := anim.GenerateSVG()
|
||
if svgCode == "" {
|
||
continue
|
||
}
|
||
|
||
// 根据动画类型决定放置位置
|
||
switch a := anim.(type) {
|
||
case *GradientAnimation:
|
||
// 渐变定义需要放在defs中
|
||
defsContent.WriteString(svgCode)
|
||
case *RotateAnimation:
|
||
// 旋转动画需要包裹目标元素，先存储起来
|
||
// 提取animateTransform标签
|
||
if start := strings.Index(svgCode, "<animateTransform"); start != -1 {
|
||
if end := strings.Index(svgCode[start:], "/>"); end != -1 {
|
||
rotateAnimations[a.GetTargetID()] = svgCode[start : start+end+2]
|
||
}
|
||
}
|
||
default:
|
||
// 其他动画元素直接添加到SVG中
|
||
animationsContent.WriteString(svgCode)
|
||
}
|
||
}
|
||
|
||
// 只有当存在需要放在defs中的内容时才添加defs标签
|
||
if defsContent.Len() > 0 {
|
||
sb.WriteString(defsContent.String())
|
||
sb.WriteString("</defs>\n")
|
||
} else {
|
||
// 如果没有需要放在defs中的内容，则不添加defs标签
|
||
sb.Reset()
|
||
}
|
||
|
||
// 添加直接放置的动画元素
|
||
sb.WriteString(animationsContent.String())
|
||
|
||
return sb.String()
|
||
}
|