更改包名

imglib/watermark_test.go

@@ -1,15 +1,13 @@
 package imglib
 
 import (
-	"image"
+	"os"
-	"image/color"
 	"testing"
 
-	"github.com/golang/freetype"
+	"github.com/fogleman/gg"
-	"golang.org/x/image/draw"
 )
 
-// TestAddChineseWatermark 给桌面 kbk.png 添加中文文字水印
+// TestAddChineseWatermark 给桌面 kbk.png 添加豆包AI风格斜纹平铺水印
 func TestAddChineseWatermark(t *testing.T) {
 	srcPath := `C:\Users\Administrator\Desktop\kbk.png`
 
@@ -19,35 +17,42 @@ func TestAddChineseWatermark(t *testing.T) {
 	}
 
 	bounds := srcImg.Bounds()
-	dst := image.NewRGBA(bounds)
+	dc := gg.NewContext(bounds.Dx(), bounds.Dy())
-	draw.Draw(dst, bounds, srcImg, bounds.Min, draw.Src)
+	dc.DrawImage(srcImg, 0, 0)
 
-	font, err := GetFont()
+	fontPath := getDefaultFontPath()
-	if err != nil {
+	if fontPath == "" {
-		t.Skipf("获取字体失败: %v", err)
+		t.Skip("未找到中文字体")
 	}
 
-	c := freetype.NewContext()
+	if err := dc.LoadFontFace(fontPath, 48); err != nil {
-	c.SetDPI(72)
+		t.Fatalf("加载字体失败: %v", err)
-	c.SetFont(font)
+	}
-	c.SetClip(dst.Bounds())
-	c.SetDst(dst)
 
-	fontSize := 36.0
+	// 斜纹平铺水印
-	c.SetFontSize(fontSize)
+	tileW, tileH := 500, 400
-
+	for y := -tileH; y < bounds.Dy()+tileH; y += tileH {
-	text := "这是一个水印"
+		for x := -tileW; x < bounds.Dx()+tileW; x += tileW {
-	x := bounds.Dx() - 280
+			cx := float64(x) + 250
-	y := bounds.Dy() - 60
+			cy := float64(y) + 200
-
+			dc.Push()
-	c.SetSrc(image.NewUniform(color.RGBA{0, 0, 0, 180}))
+			dc.RotateAbout(-0.35, cx, cy)
-	pt := freetype.Pt(x, y)
+			dc.SetRGB(1, 1, 1)
-	if _, err := c.DrawString(text, pt); err != nil {
+			dc.SetAlpha(0.25)
-		t.Fatalf("绘制水印失败: %v", err)
+			dc.DrawStringAnchored("这是一个水印", cx, cy, 0.5, 0.5)
+			dc.SetRGBA(1, 1, 1, 0.15)
+			dc.DrawStringAnchored("这是一个水印", float64(x)+250, float64(y)+200, 0.5, 0.5)
+			dc.Pop()
+		}
 	}
 
 	outPath := "d:\\source\\daShangDao_utils\\kbk_watermarked.png"
-	err = SavePNG(dst, outPath)
+	f, err := os.Create(outPath)
+	if err != nil {
+		t.Fatalf("创建文件失败: %v", err)
+	}
+	defer f.Close()
+	err = SavePNG(dc.Image(), outPath)
 	if err != nil {
 		t.Fatalf("保存图片失败: %v", err)
 	}