🎉 Initial commit

benchmark/style_theme_benchmark_test.go

@@ -0,0 +1,126 @@
+package benchmark
+
+import (
+	"testing"
+
+	"github.com/landaiqing/go-pixelnebula"
+	"github.com/landaiqing/go-pixelnebula/style"
+	"github.com/landaiqing/go-pixelnebula/theme"
+)
+
+// BenchmarkDifferentStyles 测试不同风格的生成性能
+func BenchmarkDifferentStyles(b *testing.B) {
+	styles := []struct {
+		name  string
+		style style.StyleType
+	}{
+		{"GirlStyle", style.GirlStyle},
+		{"AteamStyle", style.AteamStyle},
+		{"BlondStyle", style.BlondStyle},
+		{"FirehairStyle", style.FirehairStyle},
+	}
+
+	for _, s := range styles {
+		b.Run(s.name, func(b *testing.B) {
+			b.ResetTimer()
+
+			for i := 0; i < b.N; i++ {
+				pn := pixelnebula.NewPixelNebula()
+				pn.WithStyle(s.style)
+				pn.WithSize(231, 231)
+
+				_, err := pn.Generate("benchmark-style-"+s.name, false).ToSVG()
+				if err != nil {
+					b.Fatal(err)
+				}
+			}
+		})
+	}
+}
+
+// BenchmarkDifferentThemes 测试不同主题的生成性能
+func BenchmarkDifferentThemes(b *testing.B) {
+	// 假设有5个内置主题索引
+	themeCount := 5
+
+	for i := 0; i < themeCount; i++ {
+		b.Run("Theme_"+Itoa(i), func(b *testing.B) {
+			b.ResetTimer()
+
+			for j := 0; j < b.N; j++ {
+				pn := pixelnebula.NewPixelNebula()
+				pn.WithStyle(style.GirlStyle)
+				pn.WithSize(231, 231)
+
+				_, err := pn.Generate("benchmark-theme-"+Itoa(i), false).SetTheme(i).Build().ToSVG()
+				if err != nil {
+					b.Fatal(err)
+				}
+			}
+		})
+	}
+}
+
+// BenchmarkCustomTheme 测试自定义主题的性能
+func BenchmarkCustomTheme(b *testing.B) {
+	// 创建一个自定义主题
+	customTheme := []theme.Theme{
+		{
+			theme.ThemePart{
+				"env":   []string{"#f0f0f0", "#e0e0e0"},
+				"head":  []string{"#ffd699"},
+				"eyes":  []string{"#555555", "#ffffff"},
+				"mouth": []string{"#ff6b6b"},
+				"top":   []string{"#6b5b95", "#6b5b95"},
+				"clo":   []string{"#88b04b"},
+			},
+		},
+	}
+
+	b.ResetTimer()
+
+	for i := 0; i < b.N; i++ {
+		pn := pixelnebula.NewPixelNebula()
+		pn.WithStyle(style.GirlStyle)
+		pn.WithSize(231, 231)
+		pn.WithCustomizeTheme(customTheme)
+
+		_, err := pn.Generate("benchmark-custom-theme", false).ToSVG()
+		if err != nil {
+			b.Fatal(err)
+		}
+	}
+}
+
+// BenchmarkStyleThemeCombinations 测试不同风格和主题组合的性能
+func BenchmarkStyleThemeCombinations(b *testing.B) {
+	styles := []style.StyleType{style.GirlStyle, style.AsianStyle}
+	themes := []int{0, 1, 2}
+
+	for _, s := range styles {
+		for _, t := range themes {
+			styleName := "Unknown"
+			switch s {
+			case style.GirlStyle:
+				styleName = "Girl"
+			case style.AsianStyle:
+				styleName = "Asian"
+			}
+
+			b.Run(styleName+"_Theme"+Itoa(t), func(b *testing.B) {
+				b.ResetTimer()
+
+				for i := 0; i < b.N; i++ {
+					pn := pixelnebula.NewPixelNebula()
+					pn.WithStyle(s)
+					pn.WithSize(231, 231)
+
+					_, err := pn.Generate("benchmark-combo", false).SetTheme(t).Build().ToSVG()
+					if err != nil {
+						b.Fatal(err)
+					}
+				}
+			})
+		}
+	}
+}