零:背景
这是我工作中实际碰到的后端生成图片拼接和文字贴图需求。特此总结下来,方便后人。文中代码都是我们生产环境使用的。
一:图片拼接
go标准库的image包本身就能实现拼接,因此还是比较简单的
直接上代码
1.1 图片拼接代码
//图片拼接
func MergeImageNew(base image.Image, mask image.Image, paddingX int, paddingY int) (*image.RGBA, error) {
baseSrcBounds := base.Bounds().Max
maskSrcBounds := mask.Bounds().Max
newWidth := baseSrcBounds.X
newHeight := baseSrcBounds.Y
maskWidth := maskSrcBounds.X
maskHeight := maskSrcBounds.Y
des := image.NewRGBA(image.Rect(0, 0, newWidth, newHeight)) // 底板
//首先将一个图片信息存入jpg
draw.Draw(des, des.Bounds(), base, base.Bounds().Min, draw.Over)
//将另外一张图片信息存入jpg
draw.Draw(des, image.Rect(paddingX, newHeight-paddingY-maskHeight, (paddingX+maskWidth), (newHeight-paddingY)), mask, image.ZP, draw.Over)
return des, nil
}
核心就是使用image>newRGBA新建一个空白底图,让后将背景图,拼接图使用draw.Draw画上去就好了。
1.2 从本地、网络读取图片
从本地读取
func GetImageFromFile(filePath string) (img image.Image, err error) {
f1Src, err := os.Open(filePath)
if err != nil {
return nil, err
}
defer f1Src.Close()
buff := make([]byte, 512) // why 512 bytes ? see http://golang.org/pkg/net/http/#DetectContentType
_, err = f1Src.Read(buff)
if err != nil {
return nil, err
}
filetype := http.DetectContentType(buff)
fmt.Println(filetype)
fSrc, err := os.Open(filePath)
defer fSrc.Close()
switch filetype {
case "image/jpeg", "image/jpg":
img, err = jpeg.Decode(fSrc)
if err != nil {
fmt.Println("jpeg error")
return nil, err
}
case "image/gif":
img, err = gif.Decode(fSrc)
if err != nil {
return nil, err
}
case "image/png":
img, err = png.Decode(fSrc)
if err != nil {
return nil, err
}
default:
return nil, err
}
return img, nil
}
从网络中读取
func GetImageFromNet(url string) (image.Image, error) {
res, err := http.Get(url)
if err != nil || res.StatusCode != 200 {
return nil, err
}
defer res.Body.Close()
m, _, err := image.Decode(res.Body)
return m, err
}
保存图片
func SaveImage(targetPath string, m image.Image) error {
fSave, err := os.Create(targetPath)
if err != nil {
return err
}
defer fSave.Close()
err = jpeg.Encode(fSave, m, nil)
if err != nil {
return err
}
return nil
}
二:文字书写
图片书写文字是基于 github.com/golang/freetype 这个库实现的
import (
"github.com/golang/freetype"
"github.com/golang/freetype/truetype"
"golang.org/x/image/font"
"image"
"io/ioutil"
)
//字体相关
type TextBrush struct {
FontType *truetype.Font
FontSize float64
FontColor *image.Uniform
TextWidth int
}
func NewTextBrush(FontFilePath string, FontSize float64, FontColor *image.Uniform, textWidth int) (*TextBrush, error) {
fontFile, err := ioutil.ReadFile(FontFilePath)
if err != nil {
return nil, err
}
fontType, err := truetype.Parse(fontFile)
if err != nil {
return nil, err
}
if textWidth <= 0 {
textWidth = 20
}
return &TextBrush{FontType: fontType, FontSize: FontSize, FontColor: FontColor, TextWidth: textWidth}, nil
}
// 图片插入文字
func (fb *TextBrush) DrawFontOnRGBA(rgba *image.RGBA, pt image.Point, content string) {
c := freetype.NewContext()
c.SetDPI(72)
c.SetFont(fb.FontType)
c.SetHinting(font.HintingFull)
c.SetFontSize(fb.FontSize)
c.SetClip(rgba.Bounds())
c.SetDst(rgba)
c.SetSrc(fb.FontColor)
c.DrawString(content, freetype.Pt(pt.X, pt.Y))
}
func Image2RGBA(img image.Image) *image.RGBA {
baseSrcBounds := img.Bounds().Max
newWidth := baseSrcBounds.X
newHeight := baseSrcBounds.Y
des := image.NewRGBA(image.Rect(0, 0, newWidth, newHeight)) // 底板
//首先将一个图片信息存入jpg
draw.Draw(des, des.Bounds(), img, img.Bounds().Min, draw.Over)
return des
}
使用example
func TestTextBrush_DrawFontOnRGBA(t *testing.T) {
textBrush, err := NewTextBrush("字体库ttf位置", 20, image.Black, 20)
if err != nil {
t.Log(err)
}
backgroud, err := GetImageFromFile("./resource/backgroud.jpg")
if err != nil {
t.Log(err)
}
des := Image2RGBA(backgroud)
textBrush.DrawFontOnRGBA(des, image.Pt(10, 50), "世界你好")
//调整颜色
textBrush.FontColor = image.NewUniform(color.RGBA{
R: 0x8E,
G: 0xE5,
B: 0xEE,
A: 255,
})
textBrush.DrawFontOnRGBA(des, image.Pt(10, 80), "我是用Go拼上的文字")
if err := SaveImage("./resource/text.png", des); err != nil {
t.Log(err)
}
}
先使用NewTextBrush第一个参数是字体库文件位置。这里使用的ttf格式的字体库,网上应该有免费的字体库。
参考我的example中的代码就可以直接使用。
希望能帮到各位读者。