golang图片处理库image基本操作

Fri, 11 Nov 2022 09:52:22 +0800

1. 基本操作

图片的基本读取与保存。

2. 读取

图片读取和文件读取类似，需要先获取流：

注册图片的解码器（如：jpg则import _ “image/jpeg”, png则import _ “image/png”）
通过os.open打开文件获取流；
通过image.Decode解码流，获取图片；

 1import _ "image/jpeg"
 2func readPic() image.Image {
 3    f, err := os.Open("C:\\hatAndSunglass.jpg")
 4    if err != nil {
 5        panic(err)
 6    }
 7    defer f.Close()
 8
 9    img, fmtName, err := image.Decode(f)
10    if err != nil {
11        panic(err)
12    }
13    fmt.Printf("Name: %v, Bounds: %+v, Color: %+v", fmtName, img.Bounds(), img.ColorModel())
14
15    return img
16}

解码后返回的第一个参数为Image接口：

1type Image interface {
2  ColorModel() color.Model // 返回图片的颜色模型
3  Bounds() Rectangle       // 返回图片外框
4  At(x, y int) color.Color // 返回(x,y)像素点的颜色
5}

3. 新建

新建一个图片非常简单，只需image.NewRGBA即可创建一个透明背景的图片了

1img := image.NewRGBA(image.Rect(0, 0, 300, 300))

4. 保存

保存图片也很简单，需要编码后，写入文件流即可：

注册图片的解码器
通过os.create创建文件；通
过png.Encode编码图片并写入文件；

 1func savePic(img *image.RGBA) {
 2    f, err := os.Create("C:\\tmp.jpg")
 3    if err != nil {
 4        panic(err)
 5    }
 6    defer f.Close()
 7    b := bufio.NewWriter(f)
 8    err = jpeg.Encode(b, img, nil)
 9    if err != nil {
10        panic(err)
11    }
12    b.Flush()
13}

5. 图片修改

很多操作都需要用到绘制图片：

1func Draw(dst Image, r image.Rectangle, src image.Image, sp image.Point, op Op)
2func DrawMask(dst Image, r image.Rectangle, src image.Image, sp image.Point, mask image.Image, mp image.Point, op Op)

主要参数说明：

dst：绘图的背景图
r：背景图的绘图区域
src：要绘制的图
sp：要绘制图src的开始点
op：组合方式
DrawMask多了一个遮罩蒙层参数，Draw为其一种特殊形式（遮罩相关参数为nil）。

6. 转换

读取的jpg图像不是RGBA格式的（为YCbCr格式）；在操作前需要先转换格式：

创建一个大小相同的RGBA图像；
把jpg画到新建的图像上去；

1func jpg2RGBA(img image.Image) *image.RGBA {
2    tmp := image.NewRGBA(img.Bounds())
3
4    draw.Draw(tmp, img.Bounds(), img, img.Bounds().Min, draw.Src)
5    return tmp
6}

7. 裁剪

通过subImage方法可方便地裁剪图片（需要为RGBA格式的）

1func subImg() {
2    pic := readPic()
3    fmt.Printf("Type: %T\n", pic)
4    img := jpg2RCBA(pic)
5
6    sub := img.SubImage(image.Rect(0, 0, pic.Bounds().Dx(), pic.Bounds().Dy()/2))
7    savePic(sub.(*image.RGBA))
8}

8. 缩放

图片缩放分为保持比例与不保持比例的缩放；保持比例时，要确定新图片的位置（是否居中），以及如何填充空白处。为了缩放，需要引入新的库golang.org/x/image/draw。

在保持比例缩放时，需要先计算缩放后的图片大小：

分别计算宽、高的缩放比例，以小者为准；
若是居中（否则靠左上）需要计算填充大小，然后据此计算位置；

 1func calcResizedRect(width int, src image.Rectangle, height int, centerAlign bool) image.Rectangle {
 2    var dst image.Rectangle
 3    if width*src.Dy() < height*src.Dx() { // width/src.width < height/src.height
 4        ratio := float64(width) / float64(src.Dx())
 5
 6        tH := int(float64(src.Dy()) * ratio)
 7        pad := 0
 8        if centerAlign {
 9            pad = (height - tH) / 2
10        }
11        dst = image.Rect(0, pad, width, pad+tH)
12    } else {
13        ratio := float64(height) / float64(src.Dy())
14        tW := int(float64(src.Dx()) * ratio)
15        pad := 0
16        if centerAlign {
17            pad = (width - tW) / 2
18        }
19        dst = image.Rect(pad, 0, pad+tW, height)
20    }
21
22    return dst
23}

有了缩放后的大小后，即可通过双线性插值bilinear的方式进行图片的缩放

img为要缩放的图片width、height为缩放后的大小
keepRatio为是否保持比例缩放
fill为填充的颜色（R、G、B都为fill）
centerAlign：保持比例缩放时，图片是否居中存放

 1import (
 2    "image"
 3    "image/color"
 4
 5    "golang.org/x/image/draw"
 6)
 7
 8func resizePic(img image.Image, width int, height int, keepRatio bool, fill int, centerAlign bool) image.Image {
 9    outImg := image.NewRGBA(image.Rect(0, 0, width, height))
10    if !keepRatio {
11        draw.BiLinear.Scale(outImg, outImg.Bounds(), img, img.Bounds(), draw.Over, nil)
12        return outImg
13    }
14
15    if fill != 0 {
16        fillColor := color.RGBA{R: uint8(fill), G: uint8(fill), B: uint8(fill), A: 255}
17        draw.Draw(outImg, outImg.Bounds(), &image.Uniform{C: fillColor}, image.Point{}, draw.Src)
18    }
19    dst := calcResizedRect(width, img.Bounds(), height, centerAlign)
20    draw.ApproxBiLinear.Scale(outImg, dst.Bounds(), img, img.Bounds(), draw.Over, nil)
21    return outImg
22}

到此这篇关于golang图片处理库image简介的文章就介绍到这了

Golang Image on ttz.pw