daShangDao_utils/imglib/watermark_test.go
Cai1Cai1 a92f8e2c10 feat: 初始版本 v1.0.0
- 新增 imglib 包,支持 FilePath/URL/Base64 三种图片输入方式
- 纯白占比检测、白底居中合成、等比缩放、去白边、裁切
- 二维码生成与识别、条形码生成(Code128/EAN13/Code39)
- 中文文字图片、书籍信息水印、通用水印叠加
- 输出辅助:EncodeToBytes/EncodeToBase64/SaveToFile/SaveJPEG/SavePNG
- 字体缓存,避免重复加载
- 完整测试覆盖(23个测试用例)
2026-06-30 11:56:28 +08:00

56 lines
1.2 KiB
Go

package imglib
import (
"image"
"image/color"
"testing"
"github.com/golang/freetype"
"golang.org/x/image/draw"
)
// TestAddChineseWatermark 给桌面 kbk.png 添加中文文字水印
func TestAddChineseWatermark(t *testing.T) {
srcPath := `C:\Users\Administrator\Desktop\kbk.png`
srcImg, _, err := NewImageInputFromFile(srcPath).Load()
if err != nil {
t.Fatalf("加载源图片失败: %v", err)
}
bounds := srcImg.Bounds()
dst := image.NewRGBA(bounds)
draw.Draw(dst, bounds, srcImg, bounds.Min, draw.Src)
font, err := GetFont()
if err != nil {
t.Skipf("获取字体失败: %v", err)
}
c := freetype.NewContext()
c.SetDPI(72)
c.SetFont(font)
c.SetClip(dst.Bounds())
c.SetDst(dst)
fontSize := 36.0
c.SetFontSize(fontSize)
text := "这是一个水印"
x := bounds.Dx() - 280
y := bounds.Dy() - 60
c.SetSrc(image.NewUniform(color.RGBA{0, 0, 0, 180}))
pt := freetype.Pt(x, y)
if _, err := c.DrawString(text, pt); err != nil {
t.Fatalf("绘制水印失败: %v", err)
}
outPath := "d:\\source\\daShangDao_utils\\kbk_watermarked.png"
err = SavePNG(dst, outPath)
if err != nil {
t.Fatalf("保存图片失败: %v", err)
}
t.Logf("水印图片已保存到: %s", outPath)
}