Golang 繪圖基礎- 不同的輸出源

先看一個簡單代碼， 它執行后會產生下面的300*500的png圖片文件：

代碼：

package main

import (
    "fmt"
    "image"
    "image/color"
    "image/png"
    "log"
    "os"
)

func main() {
    const (
        dx = 300
        dy = 500
    )

    // 需要保存的文件
    imgcounter := 123
    imgfile, _ := os.Create(fmt.Sprintf("%03d.png", imgcounter))
    defer imgfile.Close()

    // 新建一個 指定大小的 RGBA位圖
    img := image.NewNRGBA(image.Rect(0, 0, dx, dy))

    for y := 0; y < dy; y++ {
        for x := 0; x < dx; x++ {
            // 設置某個點的顏色，依次是 RGBA
            img.Set(x, y, color.RGBA{uint8(x % 256), uint8(y % 256), 0, 255})
        }
    }

    // 以PNG格式保存文件
    err := png.Encode(imgfile, img)
    if err != nil {
        log.Fatal(err)
    }

}

我們再看一個代碼，以http文件流的方式展示圖片，效果如下:

代碼如下：

package main

import (
    "image"
    "image/color"
    "image/png"
    "net/http"
    "time"
)

func pic(w http.ResponseWriter, req *http.Request) {
    const (
        dx = 300
        dy = 500
    )

    // 新建一個 指定大小的 RGBA位圖
    img := image.NewNRGBA(image.Rect(0, 0, dx, dy))

    for y := 0; y < dy; y++ {
        for x := 0; x < dx; x++ {
            // 設置某個點的顏色，依次是 RGBA
            img.Set(x, y, color.RGBA{uint8(x % 256), uint8(y % 256), 0, 255})
        }
    }

    // 圖片流方式輸出
    w.Header().Set("Content-Type", "image/png")
    png.Encode(w, img)
}

func main() {
    http.HandleFunc("/", pic)
    s := &http.Server{
        Addr:           ":82",
        ReadTimeout:    30 * time.Second,
        WriteTimeout:   30 * time.Second,
        MaxHeaderBytes: 1 << 20}
    s.ListenAndServe()

}

通過兩種方式來展示圖片,我們可以看到,上面兩個代碼最大的區別就是輸出源不一樣, 一個是到文件,一個是到http流. 而我們的代碼也很清晰的展示了這兩種方式只需要修改對應的實現即可.其他代碼是沒有變化的.

這里代碼中的 image.NewNRGBA 返回的 image對象是一個內存中的圖片。