时间:2022-10-27 10:47:05 | 栏目:Golang | 点击:次
这是我工作中实际碰到的后端生成图片拼接和文字贴图需求。特此总结下来,方便后人。文中代码都是我们生产环境使用的。
go标准库的image包本身就能实现拼接,因此还是比较简单的
直接上代码
//图片拼接 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画上去就好了。
从本地读取
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中的代码就可以直接使用。