golang中实现给gif、png、jpeg图片添加文字水印

时间:2022-08-26 08:52:31

添加水印示例

添加main文件

“watermark/main.go”

?
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
package main
import (
 "fmt"
 "watermark/textwatermark"
)
func main() {
 SavePath := "./kaf"
 str := textwatermark.FontInfo{18, "努力向上", textwatermark.TopLeft, 20, 20, 255, 255, 0, 255}
 arr := make([]textwatermark.FontInfo, 0)
 arr = append(arr, str)
 str2 := textwatermark.FontInfo{Size: 24, Message: "努力向上,涨工资", Position: textwatermark.TopLeft, Dx: 20, Dy: 40, R: 255, G: 255, B: 0, A: 255}
 arr = append(arr, str2)
 //加水印图片路径
 // fileName := "123123.jpg"
 fileName := "17.gif"
 w := new(textwatermark.Water)
 w.Pattern = "2006/01/02"
 textwatermark.Ttf = "./wrzh.ttf" //字体路径
 err := w.New(SavePath, fileName, arr)
 if err != nil {
  fmt.Println(err)
 }
}

golang添加水印包文件

"watermark/textwatermark.go"

?
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
package textwatermark
import (
    "errors"
    "fmt"
    "image"
    "image/color"
    "image/draw"
    "image/gif"
    "image/jpeg"
    "image/png"
    "io/ioutil"
    "math/rand"
    "os"
    "time"
    "github.com/golang/freetype"
)
// 水印的位置
const (
    TopLeft int = iota
    TopRight
    BottomLeft
    BottomRight
    Center
)
//字体路径
var Ttf string
type Water struct {
    Pattern string //增加按时间划分的子目录:默认没有时间划分的子目录
}
func (w *Water) New(SavePath, fileName string, typeface []FontInfo) error {
    var subPath string
    subPath = w.Pattern
    dirs, err := createDir(SavePath, subPath)
    if err != nil {
        return err
    }
    imgfile, _ := os.Open(fileName)
    defer imgfile.Close()
    _, str, err := image.DecodeConfig(imgfile)
    if err != nil {
        return err
    }
    newName := fmt.Sprintf("%s%s.%s", dirs, getRandomString(10), str)
    if str == "gif" {
        err = gifFontWater(fileName, newName, typeface)
    } else {
        err = staticFontWater(fileName, newName, str, typeface)
    }
    return err
}
//gif图片水印
func gifFontWater(file, name string, typeface []FontInfo) (err error) {
    imgfile, _ := os.Open(file)
    defer imgfile.Close()
    var err2 error
    gifimg2, _ := gif.DecodeAll(imgfile)
    gifs := make([]*image.Paletted, 0)
    x0 := 0
    y0 := 0
    yuan := 0
    for k, gifimg := range gifimg2.Image {
        img := image.NewNRGBA(gifimg.Bounds())
        if k == 0 {
            x0 = img.Bounds().Dx()
            y0 = img.Bounds().Dy()
        }
        fmt.Printf("%v, %v\n", img.Bounds().Dx(), img.Bounds().Dy())
        if k == 0 && gifimg2.Image[k+1].Bounds().Dx() > x0 && gifimg2.Image[k+1].Bounds().Dy() > y0 {
            yuan = 1
            break
        }
        if x0 == img.Bounds().Dx() && y0 == img.Bounds().Dy() {
            for y := 0; y < img.Bounds().Dy(); y++ {
                for x := 0; x < img.Bounds().Dx(); x++ {
                    img.Set(x, y, gifimg.At(x, y))
                }
            }
            img, err2 = common(img, typeface) //添加文字水印
            if err2 != nil {
                break
            }
            //定义一个新的图片调色板img.Bounds():使用原图的颜色域,gifimg.Palette:使用原图的调色板
            p1 := image.NewPaletted(gifimg.Bounds(), gifimg.Palette)
            //把绘制过文字的图片添加到新的图片调色板上
            draw.Draw(p1, gifimg.Bounds(), img, image.ZP, draw.Src)
            //把添加过文字的新调色板放入调色板slice
            gifs = append(gifs, p1)
        } else {
            gifs = append(gifs, gifimg)
        }
    }
    if yuan == 1 {
        return errors.New("gif: image block is out of bounds")
    } else {
        if err2 != nil {
            return err2
        }
        //保存到新文件中
        newfile, err := os.Create(name)
        if err != nil {
            return err
        }
        defer newfile.Close()
        g1 := &gif.GIF{
            Image:     gifs,
            Delay:     gifimg2.Delay,
            LoopCount: gifimg2.LoopCount,
        }
        err = gif.EncodeAll(newfile, g1)
        return err
    }
}
//png,jpeg图片水印
func staticFontWater(file, name, status string, typeface []FontInfo) (err error) {
    //需要加水印的图片
    imgfile, _ := os.Open(file)
    defer imgfile.Close()
    var staticImg image.Image
    if status == "png" {
        staticImg, _ = png.Decode(imgfile)
    } else {
        staticImg, _ = jpeg.Decode(imgfile)
    }
    img := image.NewNRGBA(staticImg.Bounds())
    for y := 0; y < img.Bounds().Dy(); y++ {
        for x := 0; x < img.Bounds().Dx(); x++ {
            img.Set(x, y, staticImg.At(x, y))
        }
    }
    img, err = common(img, typeface) //添加文字水印
    if err != nil {
        return err
    }
    //保存到新文件中
    newfile, err := os.Create(name)
    if err != nil {
        return err
    }
    defer newfile.Close()
    if status == "png" {
        err = png.Encode(newfile, img)
    } else {
        err = jpeg.Encode(newfile, img, &jpeg.Options{100})
    }
    return err
}
//添加文字水印函数
func common(img *image.NRGBA, typeface []FontInfo) (*image.NRGBA, error) {
    var err2 error
    //拷贝一个字体文件到运行目录
    fontBytes, err := ioutil.ReadFile(Ttf)
    if err != nil {
        err2 = err
        return nil, err2
    }
    font, err := freetype.ParseFont(fontBytes)
    if err != nil {
        err2 = err
        return nil, err2
    }
    errNum := 1
Loop:
    for _, t := range typeface {
        info := t.Message
        f := freetype.NewContext()
        f.SetDPI(108)
        f.SetFont(font)
        f.SetFontSize(t.Size)
        f.SetClip(img.Bounds())
        f.SetDst(img)
        f.SetSrc(image.NewUniform(color.RGBA{R: t.R, G: t.G, B: t.B, A: t.A}))
        //第一行的文字
        // pt := freetype.Pt(img.Bounds().Dx()-len(info)*4-20, img.Bounds().Dy()-100)
        first := 0
        two := 0
        switch int(t.Position) {
        case 0:
            first = t.Dx
            two = t.Dy + int(f.PointToFixed(t.Size)>>6)
        case 1:
            first = img.Bounds().Dx() - len(info)*4 - t.Dx
            two = t.Dy + int(f.PointToFixed(t.Size)>>6)
        case 2:
            first = t.Dx
            two = img.Bounds().Dy() - t.Dy
        case 3:
            first = img.Bounds().Dx() - len(info)*4 - t.Dx
            two = img.Bounds().Dy() - t.Dy
        case 4:
            first = (img.Bounds().Dx() - len(info)*4) / 2
            two = (img.Bounds().Dy() - t.Dy) / 2
        default:
            errNum = 0
            break Loop
        }
        // fmt.Printf("%v, %v, %v\n", first, two, info)
        pt := freetype.Pt(first, two)
        _, err = f.DrawString(info, pt)
        if err != nil {
            err2 = err
            break
        }
    }
    if errNum == 0 {
        err2 = errors.New("坐标值不对")
    }
    return img, err2
}
//定义添加的文字信息
type FontInfo struct {
    Size     float64 //文字大小
    Message  string  //文字内容
    Position int     //文字存放位置
    Dx       int     //文字x轴留白距离
    Dy       int     //文字y轴留白距离
    R        uint8   //文字颜色值RGBA中的R值
    G        uint8   //文字颜色值RGBA中的G值
    B        uint8   //文字颜色值RGBA中的B值
    A        uint8   //文字颜色值RGBA中的A值
}
//生成图片名字
func getRandomString(lenght int) string {
    str := "0123456789abcdefghijklmnopqrstuvwxyz"
    bytes := []byte(str)
    bytesLen := len(bytes)
    result := []byte{}
    r := rand.New(rand.NewSource(time.Now().UnixNano()))
    for i := 0; i < lenght; i++ {
        result = append(result, bytes[r.Intn(bytesLen)])
    }
    return string(result)
}
//检查并生成存放图片的目录
func createDir(SavePath, subPath string) (string, error) {
    var dirs string
    if subPath == "" {
        dirs = fmt.Sprintf("%s/", SavePath)
    } else {
        dirs = fmt.Sprintf("%s/%s/", SavePath, time.Now().Format(subPath))
    }
    _, err := os.Stat(dirs)
    if err != nil {
        err = os.MkdirAll(dirs, os.ModePerm)
        if err != nil {
            return "", err
        }
    }
    return dirs, nil
}

补充:golang基础--image/draw渲染图片、利用golang/freetype库在图片上生成文字

需求

在一张A4纸上,利用image/draw标准库生成4张二维码,和该二维码的客户信息

1、二维码生成利用到的库就是image/draw,通过draw.Draw进行写入

2、然后字体渲染利用了golang/freetype开源库

https://github.com/golang/freetype/blob/master/example/freetype/main.go

安装依赖

?
1
2
"github.com/golang/freetype"
"golang.org/x/image/font"

以上的golang.org/x/image/font需要*,如果不能*利用如下方法也可以:

golang中实现给gif、png、jpeg图片添加文字水印

逻辑

1、通过os.Create("dst.jpg")生成一个最终的图片,该图片上画了4个二维码,和头部文字渲染

2、通过os.Open("/Users/zhiliao/zhiliao/gopro/go_safly/src/qr.png")去获取本地的一个二维码图片路径,然后通过png.Decode(file1)生成图片

3、修改二维码图片的尺寸resize.Resize(314, 314, img, resize.Lanczos3)

4、通过image.NewRGBA(image.Rect(0, 0, 827, 1169))生成一个RGBA结构体的矩形框,就好比是画布的概念

5、选渲染头部的客户信息字体,需要一个中文字体库,这个可以用Mac系统库的中文字体,或者自行下载ttf字体库,然后加载该字体

6、draw.Draw(jpg, jpg.Bounds(), bg, image.ZP, draw.Src)是进行设置渲染的参数,参数是画布、渲染开始的地方、图片来源、图片参数、渲染模式

7、然后就是设置baseline,我这里去掉了,然后就是划线,最后就是渲染字体,通过c.DrawString(s, pt)

8、接下来就是画4个二维码

?
1
2
3
4
draw.Draw(jpg, img.Bounds().Add(image.Pt(60, 150)), img, img.Bounds().Min, draw.Src) //截取图片的一部分
    draw.Draw(jpg, img.Bounds().Add(image.Pt(435, 150)), img, img.Bounds().Min, draw.Src) //截取图片的一部分
    draw.Draw(jpg, img.Bounds().Add(image.Pt(60, 610)), img, img.Bounds().Min, draw.Src) //截取图片的一部分
    draw.Draw(jpg, img.Bounds().Add(image.Pt(435, 610)), img, img.Bounds().Min, draw.Src) //截取图片的一部分

9、最后通过png.Encode(file, jpg)输出到我们最终生成的图片

效果图

golang中实现给gif、png、jpeg图片添加文字水印

实例

?
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
package main
import (
    "flag"
    "fmt"
    "github.com/gin-gonic/gin"
    "github.com/golang/freetype"
    "github.com/nfnt/resize"
    "golang.org/x/image/font"
    "image"
    "image/draw"
    "image/png"
    "io/ioutil"
    "log"
    "net/http"
    "os"
    "strings"
)
var (
    dpi      = flag.Float64("dpi", 72, "screen resolution in Dots Per Inch")
    fontfile = flag.String("fontfile", "/Users/zhiliao/Downloads/ffffonts/simsun.ttf", "filename of the ttf font")
    hinting  = flag.String("hinting", "none", "none | full")
    size     = flag.Float64("size", 30, "font size in points")
    spacing  = flag.Float64("spacing", 1.5, "line spacing (e.g. 2 means double spaced)")
    wonb     = flag.Bool("whiteonblack", false, "white text on a black background")
)
var text = []string{
    "地支:沈阳市某区某镇某街道某楼某",
    "姓名:王永飞",
    "电话:1232131231232",
}
func main() {
    file, err := os.Create("dst.jpg")
    if err != nil {
        fmt.Println(err)
    }
    defer file.Close()
    file1, err := os.Open("/Users/zhiliao/zhiliao/gopro/go_safly/src/qr.png")
    if err != nil {
        fmt.Println(err)
    }
    defer file1.Close()
    img, _ := png.Decode(file1)
    //尺寸
    img = resize.Resize(314, 314, img, resize.Lanczos3)
    jpg := image.NewRGBA(image.Rect(0, 0, 827, 1169))
    fontRender(jpg)
    draw.Draw(jpg, img.Bounds().Add(image.Pt(60, 150)), img, img.Bounds().Min, draw.Src) //截取图片的一部分
    draw.Draw(jpg, img.Bounds().Add(image.Pt(435, 150)), img, img.Bounds().Min, draw.Src) //截取图片的一部分
    draw.Draw(jpg, img.Bounds().Add(image.Pt(60, 610)), img, img.Bounds().Min, draw.Src) //截取图片的一部分
    draw.Draw(jpg, img.Bounds().Add(image.Pt(435, 610)), img, img.Bounds().Min, draw.Src) //截取图片的一部分
    png.Encode(file, jpg)
}
func fontRender(jpg *image.RGBA)  {
    flag.Parse()
    fontBytes, err := ioutil.ReadFile(*fontfile)
    if err != nil {
        log.Println(err)
        return
    }
    f, err := freetype.ParseFont(fontBytes)
    if err != nil {
        log.Println(err)
        return
    }
    fg, bg := image.Black, image.White
    //ruler := color.RGBA{0xdd, 0xdd, 0xdd, 0xff}
    //if *wonb {
    //  fg, bg = image.White, image.Black
    //  ruler = color.RGBA{0x22, 0x22, 0x22, 0xff}
    //}
    draw.Draw(jpg, jpg.Bounds(), bg, image.ZP, draw.Src)
    c := freetype.NewContext()
    c.SetDPI(*dpi)
    c.SetFont(f)
    c.SetFontSize(*size)
    c.SetClip(jpg.Bounds())
    c.SetDst(jpg)
    c.SetSrc(fg)
    switch *hinting {
    default:
        c.SetHinting(font.HintingNone)
    case "full":
        c.SetHinting(font.HintingFull)
    }
    //Draw the guidelines.
    //for i := 0; i < 200; i++ {
    //  jpg.Set(10, 10+i, ruler)
    //  jpg.Set(10+i, 10, ruler)
    //}
    // Draw the text.
    pt := freetype.Pt(200, 10+int(c.PointToFixed(*size)>>6))
    for _, s := range text {
        _, err = c.DrawString(s, pt)
        if err != nil {
            log.Println(err)
            return
        }
        pt.Y += c.PointToFixed(*size * *spacing)
    }
}
func Cors() gin.HandlerFunc {
    return func(c *gin.Context) {
        method := c.Request.Method      //请求方法
        origin := c.Request.Header.Get("Origin")        //请求头部
        var headerKeys []string                             // 声明请求头keys
        for k, _ := range c.Request.Header {
            headerKeys = append(headerKeys, k)
        }
        headerStr := strings.Join(headerKeys, ", ")
        if headerStr != "" {
            headerStr = fmt.Sprintf("access-control-allow-origin, access-control-allow-headers, %s", headerStr)
        } else {
            headerStr = "access-control-allow-origin, access-control-allow-headers"
        }
        if origin != "" {
            c.Writer.Header().Set("Access-Control-Allow-Origin", "*")
            c.Header("Access-Control-Allow-Origin", "*")        // 这是允许访问所有域
            c.Header("Access-Control-Allow-Methods", "POST, GET, OPTIONS, PUT, DELETE,UPDATE")      //服务器支持的所有跨域请求的方法,为了避免浏览次请求的多次'预检'请求
            //  header的类型
            c.Header("Access-Control-Allow-Headers", "Authorization, Content-Length, X-CSRF-Token, Token,session,X_Requested_With,Accept, Origin, Host, Connection, Accept-Encoding, Accept-Language,DNT, X-CustomHeader, Keep-Alive, User-Agent, X-Requested-With, If-Modified-Since, Cache-Control, Content-Type, Pragma")
            //              允许跨域设置                                                                                                      可以返回其他子段
            c.Header("Access-Control-Expose-Headers", "Content-Length, Access-Control-Allow-Origin, Access-Control-Allow-Headers,Cache-Control,Content-Language,Content-Type,Expires,Last-Modified,Pragma,FooBar")      // 跨域关键设置 让浏览器可以解析
            c.Header("Access-Control-Max-Age", "172800")        // 缓存请求信息 单位为秒
            c.Header("Access-Control-Allow-Credentials", "false")       //  跨域请求是否需要带cookie信息 默认设置为true
            c.Set("content-type", "application/json")       // 设置返回格式是json
        }
        //放行所有OPTIONS方法
        if method == "OPTIONS" {
            c.JSON(http.StatusOK, "Options Request!")
        }
        // 处理请求
        c.Next()        //  处理请求
    }
}

以上为个人经验,希望能给大家一个参考,也希望大家多多支持服务器之家。如有错误或未考虑完全的地方,望不吝赐教。

原文链接:https://blog.csdn.net/huangfenhu/article/details/105143490