golang 生成个性二维码

前言

好久没发文章了,最近登录看到有朋友评论了golang生成二维码的文章,想了解下生成个性二维码的解决方案!

实现思路 ,网上搜相关资料几乎没有, 没有那就就自己想吧,就去看了下草料二维码;

然后突然就有感觉了 不就是把像素的颜色替换嘛 这还不简单

1 打开二维码图片
2 打开要填充的图片
3 将要填充的图片剪裁成和二维码一样大小
4 分析二维码的每个像素 替换成图片的像素就行了

以下是生成的例子:

例子

代码

package main
import (
 "fmt"
 "github.com/nfnt/resize"
 "image"
 "image/color"
 "image/draw"
 "log"
 "os"
 "time"
 _ "image/jpeg"
 _ "image/gif"
 "image/png"
)
func main() {
 // 打开二维码图片
 qrFile, err := os.Open("img/qr.png")
 if err != nil {
 log.Fatalf("打开二维码失败 %s", err.Error())
 return
 }
 defer qrFile.Close()
 // 将二维码文件接码成图片
 img, _, err := image.Decode(qrFile)
 if err != nil {
 log.Fatalf("image decode error %s", err.Error())
 return
 }
 // 获取二维码的宽高
 width, height := img.Bounds().Max.X, img.Bounds().Max.Y
 // 打开要填充的图片
 bgFile, err := os.Open("bg/7.jpg")
 if err != nil {
 log.Fatalf("打开填充图失败 %s", err.Error())
 return
 }
 defer bgFile.Close()
 // 将填充图解码成png图片
 bgImg, _, err := image.Decode(bgFile)
 if err != nil {
 log.Fatalf("填充图解码失败 %s", err.Error())
 return
 }
 // 获取填充图的宽高
 bgWidth, bgHeight := bgImg.Bounds().Max.X, bgImg.Bounds().Max.Y
 // 检测二维码和填充图宽高是否一致
 if width != bgWidth || height != bgHeight {
 // 如果不一致将填充图剪裁
 bgImg = ImageResize(bgImg, width, height)
 }
 // 开始填充二维码
 for y := 0; y < img.Bounds().Max.X; y++ {
 for x := 0; x < img.Bounds().Max.X; x++ {
 qrImgColor := img.At(x, y)
 // 检测图片颜色 如果rgb值是 255 255 255 255 则像素点为白色 跳过
 // 如果rgba值是 0 0 0 0 则为透明色 跳过
 switch img.(type) {
 case *image.NRGBA:
 c := qrImgColor.(color.NRGBA)
 if (c.R == 0 && c.G == 0 && c.B == 0 && c.A == 0) || (c.R == 255 && c.G == 255 && c.B == 255 && c.A == 255) {
 continue
 }
 case *image.RGBA:
 c := qrImgColor.(color.RGBA)
 if (c.R == 0 && c.G == 0 && c.B == 0 && c.A == 0) || (c.R == 255 && c.G == 255 && c.B == 255 && c.A == 255) {
 continue
 }
 }
 // 获取要填充的图片的颜色
 bgImgColor := bgImg.At(x, y)
 
 // 填充颜色
 switch bgImg.(type) {
 case *image.RGBA64:
 c := bgImgColor.(color.RGBA64)
 img.(draw.Image).Set(x, y, color.RGBA64{R: c.R, G: c.G, B: c.B, A: c.A})
 case *image.NRGBA:
 c := bgImgColor.(color.NRGBA)
 img.(draw.Image).Set(x, y, color.NRGBA{R: c.R, G: c.G, B: c.B, A: c.A})
 case *image.RGBA:
 c := bgImgColor.(color.RGBA)
 img.(draw.Image).Set(x, y, color.RGBA{R: c.R, G: c.G, B: c.B, A: c.A})
 case *image.YCbCr:
 c := bgImgColor.(color.YCbCr)
 img.(draw.Image).Set(x, y, color.YCbCr{Y: c.Y, Cb: c.Cb, Cr: c.Cr})
 default:
 fmt.Println("error")
 }
 }
 }
 filename := fmt.Sprintf("%s.png", time.Now().Format("20060102150405"))
 // 写入文件
 outFile, err := os.Create(filename)
 if err != nil {
 log.Fatal(err)
 }
 defer outFile.Close()
 _ = png.Encode(outFile, img)
}
func ImageResize(src image.Image, w, h int) image.Image {
 return resize.Resize(uint(w), uint(h), src, resize.Lanczos3)
}

注意

1 尽量用png的二维码和填充图
2 案例中png和jpg gif图片都可以填充 其他格式的图片自己加就行了

代码地址

https://github.com/lihaotian0607/qr

其他

另外附上封装好的
https://github.com/lihaotian0607/qrcode

生成普通二维码

var qr, err = qrcode.New("https://www.xxx.com/", qrcode.Highest)

设置头像

 qr.SetAvatar(&qrcode.Avatar{
 Src: "../static/1.jpg",
 Width: 60,
 Height: 60,
 Round: 10,
 })

设置背景图

qr.SetBackground(&qrcode.Background{
 Src: "../static/3.png",
 X: 70,
 Y: 55,
 Width: 270,
 Height: 270,
 })

设置前景图

qr.SetForeground("../static/2.png")

写入文件

img, err := qr.Generate(300)
 if err != nil {
 log.Fatal(err.Error())
 return
 }
 file, err := os.Create("out.png")
 if err != nil {
 log.Fatalf("创建文件失败 %s", err.Error())
 return
 }
 err = png.Encode(file, img)
 if err != nil {
 log.Fatalf("文件写入失败 %s", err.Error())
 }

本文来自:简书

感谢作者:hi李昊天

查看原文:golang 生成个性二维码

• 编辑
• 预览