57 lines
1.3 KiB
Go
57 lines
1.3 KiB
Go
package imglib
|
|
|
|
import (
|
|
"image"
|
|
"image/color"
|
|
"os"
|
|
"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)
|
|
|
|
// 大号红色文字（中间偏上）
|
|
c.SetFontSize(120)
|
|
c.SetSrc(image.NewUniform(color.RGBA{255, 0, 0, 255}))
|
|
pt := freetype.Pt(300, 800)
|
|
if _, err := c.DrawString("这是一个水印", pt); err != nil {
|
|
t.Fatalf("绘制水印失败: %v", err)
|
|
}
|
|
|
|
outPath := "d:\\source\\daShangDao_utils\\kbk_watermarked.png"
|
|
f, err := os.Create(outPath)
|
|
if err != nil {
|
|
t.Fatalf("创建文件失败: %v", err)
|
|
}
|
|
defer f.Close()
|
|
err = SavePNG(dst, outPath)
|
|
if err != nil {
|
|
t.Fatalf("保存图片失败: %v", err)
|
|
}
|
|
t.Logf("水印图片已保存到: %s", outPath)
|
|
}
|