首頁 > 軟體

golang圖片處理庫image基本操作

2022-07-26 18:03:04

基本操作

圖片的基本讀取與儲存。

讀取

圖片讀取和檔案讀取類似,需要先獲取流:

  • 註冊圖片的解碼器(如:jpg則import _ "image/jpeg", png則import _ "image/png"
  • 通過os.open開啟檔案獲取流;
  • 通過image.Decode解碼流,獲取圖片;
import _ "image/jpeg"
func readPic() image.Image {
    f, err := os.Open("C:\hatAndSunglass.jpg")
    if err != nil {
        panic(err)
    }
    defer f.Close()

    img, fmtName, err := image.Decode(f)
    if err != nil {
        panic(err)
    }
    fmt.Printf("Name: %v, Bounds: %+v, Color: %+v", fmtName, img.Bounds(), img.ColorModel())

    return img
}

解碼後返回的第一個引數為Image介面:

type Image interface {
  ColorModel() color.Model // 返回圖片的顏色模型
  Bounds() Rectangle       // 返回圖片外框
  At(x, y int) color.Color // 返回(x,y)畫素點的顏色
}

新建

新建一個圖片非常簡單,只需image.NewRGBA即可建立一個透明背景的圖片了

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

儲存

儲存圖片也很簡單,需要編碼後,寫入檔案流即可:

  • 註冊圖片的解碼器
  • 通過os.create建立檔案;通
  • png.Encode編碼圖片並寫入檔案;
func savePic(img *image.RGBA) {
    f, err := os.Create("C:\tmp.jpg")
    if err != nil {
        panic(err)
    }
    defer f.Close()
    b := bufio.NewWriter(f)
    err = jpeg.Encode(b, img, nil)
    if err != nil {
        panic(err)
    }
    b.Flush()
}

圖片修改

很多操作都需要用到繪製圖片:

func Draw(dst Image, r image.Rectangle, src image.Image, sp image.Point, op Op)
func 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)。

轉換

讀取的jpg影象不是RGBA格式的(為YCbCr格式);在操作前需要先轉換格式:

  • 建立一個大小相同的RGBA影象;
  • 把jpg畫到新建的影象上去;
func jpg2RGBA(img image.Image) *image.RGBA {
    tmp := image.NewRGBA(img.Bounds())

    draw.Draw(tmp, img.Bounds(), img, img.Bounds().Min, draw.Src)
    return tmp
}

裁剪

通過subImage方法可方便地裁剪圖片(需要為RGBA格式的)

func subImg() {
    pic := readPic()
    fmt.Printf("Type: %Tn", pic)
    img := jpg2RCBA(pic)

    sub := img.SubImage(image.Rect(0, 0, pic.Bounds().Dx(), pic.Bounds().Dy()/2))
    savePic(sub.(*image.RGBA))
}

縮放

圖片縮放分為保持比例與不保持比例的縮放;保持比例時,要確定新圖片的位置(是否居中),以及如何填充空白處。
為了縮放,需要引入新的庫golang.org/x/image/draw

在保持比例縮放時,需要先計算縮放後的圖片大小:

  • 分別計算寬、高的縮放比例,以小者為準;
  • 若是居中(否則靠左上)需要計算填充大小,然後據此計算位置;
func calcResizedRect(width int, src image.Rectangle, height int, centerAlign bool) image.Rectangle {
    var dst image.Rectangle
    if width*src.Dy() < height*src.Dx() { // width/src.width < height/src.height
        ratio := float64(width) / float64(src.Dx())

        tH := int(float64(src.Dy()) * ratio)
        pad := 0
        if centerAlign {
            pad = (height - tH) / 2
        }
        dst = image.Rect(0, pad, width, pad+tH)
    } else {
        ratio := float64(height) / float64(src.Dy())
        tW := int(float64(src.Dx()) * ratio)
        pad := 0
        if centerAlign {
            pad = (width - tW) / 2
        }
        dst = image.Rect(pad, 0, pad+tW, height)
    }

    return dst
}

有了縮放後的大小後,即可通過雙線性插值bilinear的方式進行圖片的縮放

  • img為要縮放的圖片width、height為縮放後的大小
  • keepRatio為是否保持比例縮放
  • fill為填充的顏色(R、G、B都為fill)
  • centerAlign:保持比例縮放時,圖片是否居中存放
import (
    "image"
    "image/color"

    "golang.org/x/image/draw"
)

func resizePic(img image.Image, width int, height int, keepRatio bool, fill int, centerAlign bool) image.Image {
    outImg := image.NewRGBA(image.Rect(0, 0, width, height))
    if !keepRatio {
        draw.BiLinear.Scale(outImg, outImg.Bounds(), img, img.Bounds(), draw.Over, nil)
        return outImg
    }

    if fill != 0 {
        fillColor := color.RGBA{R: uint8(fill), G: uint8(fill), B: uint8(fill), A: 255}
        draw.Draw(outImg, outImg.Bounds(), &image.Uniform{C: fillColor}, image.Point{}, draw.Src)
    }
    dst := calcResizedRect(width, img.Bounds(), height, centerAlign)
    draw.ApproxBiLinear.Scale(outImg, dst.Bounds(), img, img.Bounds(), draw.Over, nil)
    return outImg
}

到此這篇關於golang圖片處理庫image簡介的文章就介紹到這了,更多相關golang圖片處理庫image內容請搜尋it145.com以前的文章或繼續瀏覽下面的相關文章希望大家以後多多支援it145.com!


IT145.com E-mail:sddin#qq.com