image - Golang - 使用 Image 和 Image/PNG 交换图片的 PNG channel

Question

我正在尝试写一个简短的，它将读取一个 PNG 文件，并将一个 channel 与另一个 channel (R,G,B) 交换作为可能的选择。

但是，我无法找到如何从 image.At(x,y) 返回的 color.Color 对象中提取整数。一旦我可以用交换的 channel 构造新的 RGBA 颜色，使用 image.Set(x,y,color) 写回它可能会更容易。

我现在在这里(你几乎可以跳到最后一个循环):

package main

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

type Choice struct {
value string
valid bool
}

func (c *Choice) validate() {
goodchoices := []string{"R", "G", "B"}
for _, v := range goodchoices {
    if c.value == v {
        c.valid = true
    }
}
}

func main() {

var fname string
var c1 Choice
var c2 Choice

flag.StringVar(&c1.value, "c1", "", "The color channel to swap - R or G or B ")
flag.StringVar(&c2.value, "c2", "", "The color channel to swap with - R or G or B ")
flag.StringVar(&fname, "f", "", "A .png image (normal map)")
flag.Parse()

c1.validate()
c2.validate()

if c1.valid == true && c2.valid == true {
    fmt.Println("We could proceed..")
    fmt.Println("Swapping channels:", c1.value, "<->", c2.value, "In", fname) //for testing
} else {
    fmt.Println("Invalid channel... Please use R, G or B.")
    return
}

file, err := os.Open(fname)
if err != nil {
    fmt.Println(err)
    return
}
defer file.Close()

pic, err := png.Decode(file)
if err != nil {
    fmt.Fprintf(os.Stderr, "%s: %v\n", fname, err)
    return
}

b := pic.Bounds()

for y := b.Min.Y; y < b.Max.Y; y++ {
    for x := b.Min.X; x < b.Max.X; x++ {
        col := pic.At(x, y)
        ???? How do I swap the channels in col ???? 
    }
}
}

我对 Go 和一般编程真的很陌生，所以请在您的回答中考虑一下。谢谢。

Answer 1

嗯，这比我想象的要难 - 我想知道是否有人能想出更好的主意!

问题是您不知道 png.Decode 返回的具体类型 - 它可能返回任何图像类型。您只有一个没有 Set 方法的 image.Image 接口(interface)。

为了解决这个问题，首先定义一个接口(interface)，所有可以设置像素的图像类型都满足

type ImageSet interface {
     Set(x, y int, c color.Color)
}

接下来看看 pic 是否实现了那个接口(interface)(如果没有实现，go 会 panic - 使用 picSet，如果这让你感到困扰，就可以了)

// Get an interface which can set pixels
picSet := pic.(ImageSet)

现在你的循环看起来像这样 - 我只交换了红色和绿色所以你可以看到这个想法。

for y := b.Min.Y; y < b.Max.Y; y++ {
    for x := b.Min.X; x < b.Max.X; x++ {
        col := pic.At(x, y)
        r, g, b, a := col.RGBA()
        // Swap green and red
        newCol := color.RGBA{uint8(g>>8), uint8(r>>8), uint8(b>>8), uint8(a>>8)}
        picSet.Set(x, y, newCol)
    }
}

我怀疑这个的高性能版本必须使用类型开关来确定它是哪种图像类型，然后为每个图像类型定制代码，其中 uint8s 用于 24 位图像，uint16s 用于 48 位图像等。

Here is the complete working例如，如果你想试一试。但它在 playground 中不起作用 - 你必须下载它。

---

更新:刚刚注意到您的评论。如果您知道自己有 RGBA 图像，那么您可以使用类型断言来获取底层图像，这会让事情变得容易得多。

// Get an image.RGBA if it is one
rgba, ok := pic.(*image.RGBA)
if !ok {
    fmt.Println("That wasn't an RGBA!")
    return
}

for y := b.Min.Y; y < b.Max.Y; y++ {
    for x := b.Min.X; x < b.Max.X; x++ {
        // Note type assertion to get a color.RGBA
        col := rgba.At(x, y).(color.RGBA)
        // Swap green and red
        col.G, col.R = col.R, col.G
        rgba.Set(x, y, col)
    }
}