Golang WebP图像处理：编码、解码与元数据操作指南

go get github.com/chai2010/webp

package main

import (
    "bytes"
    "fmt"
    "image" // 用于处理图像数据
    "log"
    "os"    // 用于文件读写

    "github.com/chai2010/webp" // WebP库
)

// 假设 data 是从 WebP 文件中读取的字节切片
// data, err := os.ReadFile("./testdata/1_webp_ll.webp")
// if err != nil { log.Fatal(err) }

width, height, hasAlpha, err := webp.GetInfo(data)
if err != nil {
    log.Printf("获取WebP信息失败: %v\n", err)
} else {
    fmt.Printf("WebP图像信息: 宽度=%d, 高度=%d, 是否有Alpha通道=%t\n", width, height, hasAlpha)
}

// 假设 data 是从 WebP 文件中读取的字节切片
// data, err := os.ReadFile("./testdata/1_webp_ll.webp")
// if err != nil { log.Fatal(err) }

metadata, err := webp.GetMetadata(data, "ICCP") // 尝试获取ICC配置文件
if err != nil {
    fmt.Printf("获取ICC元数据失败: %v\n", err)
} else {
    fmt.Printf("ICC元数据: %s\n", string(metadata))
}

// 假设 data 是从 WebP 文件中读取的字节切片
// data, err := os.ReadFile("./testdata/1_webp_ll.webp")
// if err != nil { log.Fatal(err) }

m, err := webp.Decode(bytes.NewReader(data))
if err != nil {
    log.Printf("解码WebP图像失败: %v\n", err)
} else {
    fmt.Printf("图像解码成功，类型为: %T\n", m)
    // m 现在是一个 image.Image 对象，可以进行进一步处理
}

// 假设 m 是一个已解码的 image.Image 对象
// m, err := webp.Decode(bytes.NewReader(data))
// if err != nil { log.Fatal(err) }

var buf bytes.Buffer
// 无损编码示例
err = webp.Encode(&buf, m, &webp.Options{Lossless: true})
if err != nil {
    log.Printf("无损编码WebP图像失败: %v\n", err)
} else {
    fmt.Println("无损WebP图像编码成功。")
    // buf.Bytes() 包含了编码后的WebP数据
    // err = os.WriteFile("output_lossless.webp", buf.Bytes(), 0666)
    // if err != nil { log.Fatal(err) }
}

// 有损编码示例 (默认 quality 为 90)
// var bufLossy bytes.Buffer
// err = webp.Encode(&bufLossy, m, nil) // 或 &webp.Options{Quality: 75.0}
// if err != nil { log.Printf("有损编码WebP图像失败: %v\n", err) }
// else { fmt.Println("有损WebP图像编码成功。") }

package main

import (
    "bytes"
    "fmt"
    "log"
    "os" // 推荐使用 os 包进行文件读写，io/ioutil 已废弃

    "github.com/chai2010/webp"
)

func main() {
    var buf bytes.Buffer
    var width, height int
    var data []byte
    var err error

    // 1. 从文件加载WebP数据
    // 请确保在运行此代码前，在相同目录下创建一个名为 "testdata" 的文件夹，
    // 并在其中放置一个名为 "1_webp_ll.webp" 的WebP测试文件。
    // 如果没有测试文件，可以手动创建一个简单的图片并转换为WebP格式。
    webpFilePath := "./testdata/1_webp_ll.webp"
    data, err = os.ReadFile(webpFilePath)
    if err != nil {
        log.Fatalf("无法读取WebP文件 %s: %v", webpFilePath, err)
    }
    fmt.Printf("成功读取文件: %s\n", webpFilePath)

    // 2. 获取WebP图像信息
    width, height, hasAlpha, err := webp.GetInfo(data)
    if err != nil {
        log.Printf("获取WebP信息失败: %v\n", err)
    } else {
        fmt.Printf("WebP图像信息: 宽度=%d, 高度=%d, 是否有Alpha通道=%t\n", width, height, hasAlpha)
    }

    // 3. 获取WebP元数据 (例如 ICCP 配置文件)
    if metadata, err := webp.GetMetadata(data, "ICCP"); err != nil {
        fmt.Printf("获取ICC元数据失败: %v\n", err)
    } else {
        fmt.Printf("ICC元数据 (部分): %s...\n", string(metadata[:min(len(metadata), 50)])) // 只打印前50个字符
    }

    // 4. 解码WebP图像为 image.Image 对象
    m, err := webp.Decode(bytes.NewReader(data))
    if err != nil {
        log.Fatalf("解码WebP图像失败: %v", err)
    }
    fmt.Printf("图像解码成功，类型为: %T, 尺寸: %s\n", m, m.Bounds())

    // 5. 将 image.Image 对象编码为新的WebP文件 (无损编码)
    // 您可以通过修改 &webp.Options{} 来控制编码质量和类型
    // 例如：&webp.Options{Quality: 80.0} 用于有损压缩，质量为80
    if err = webp.Encode(&buf, m, &webp.Options{Lossless: true}); err != nil {
        log.Fatalf("编码WebP图像失败: %v", err)
    }
    fmt.Println("WebP图像编码成功。")

    // 6. 将编码后的WebP数据写入新文件
    outputFilePath := "output_lossless.webp"
    if err = os.WriteFile(outputFilePath, buf.Bytes(), 0666); err != nil {
        log.Fatalf("写入输出文件 %s 失败: %v", outputFilePath, err)
    }
    fmt.Printf("成功将编码后的WebP图像写入文件: %s\n", outputFilePath)
}

// 辅助函数，用于安全地获取切片的前N个字符
func min(a, b int) int {
    if a < b {
        return a
    }
    return b
}