harukasan
11 years ago
commit
5af6d93b3b
@ -0,0 +1,3 @@
|
||||
.DS_Store
|
||||
|
||||
/examples/out/*
|
||||
@ -0,0 +1,23 @@
|
||||
Copyright (c) 2015, Shunsuke MICHII
|
||||
All rights reserved.
|
||||
|
||||
Redistribution and use in source and binary forms, with or without modification,
|
||||
are permitted provided that the following conditions are met:
|
||||
|
||||
* Redistributions of source code must retain the above copyright notice,
|
||||
this list of conditions and the following disclaimer.
|
||||
* Redistributions in binary form must reproduce the above copyright notice,
|
||||
this list of conditions and the following disclaimer in the documentation
|
||||
and/or other materials provided with the distribution.
|
||||
|
||||
THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
|
||||
"AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
|
||||
LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
|
||||
A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER
|
||||
OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
|
||||
EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
|
||||
PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR
|
||||
PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF
|
||||
LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING
|
||||
NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
|
||||
SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
|
||||
@ -0,0 +1,88 @@
|
||||
go-libwebp
|
||||
==========
|
||||
|
||||
A implementation of Go binding for [libwebp](https://developers.google.com/speed/webp/docs/api) written with cgo.
|
||||
|
||||
## Usage
|
||||
|
||||
The [examples](./examples) directory contains example codes and images.
|
||||
|
||||
### Decoding WebP into image.RGBA
|
||||
|
||||
```
|
||||
import (
|
||||
"io/ioutil"
|
||||
|
||||
"github.com/harukasan/go-libwebp/examples/util"
|
||||
"github.com/harukasan/go-libwebp/webp"
|
||||
)
|
||||
|
||||
func main() {
|
||||
var err error
|
||||
|
||||
// Read binary data
|
||||
data, err := ioutil.ReadFile("examples/cosmos.webp")
|
||||
if err != nil {
|
||||
panic(err)
|
||||
}
|
||||
|
||||
// Decode
|
||||
options := &webp.DecoderOptions{}
|
||||
img, err := webp.DecodeRGBA(data, options)
|
||||
if err != nil {
|
||||
panic(err)
|
||||
}
|
||||
|
||||
err = util.WritePNG(img, "out/encoded_cosmos.png")
|
||||
if err != nil {
|
||||
panic(err)
|
||||
}
|
||||
}
|
||||
```
|
||||
|
||||
You can set more decoding options such as cropping, flipping and scaling.
|
||||
|
||||
### Encoding WebP from image.RGBA
|
||||
|
||||
```
|
||||
package main
|
||||
|
||||
import (
|
||||
"bufio"
|
||||
"image"
|
||||
|
||||
"github.com/harukasan/go-libwebp/examples/util"
|
||||
"github.com/harukasan/go-libwebp/webp"
|
||||
)
|
||||
|
||||
func main() {
|
||||
err := util.ReadPNG("examples/cosmos.png")
|
||||
if err != nil {
|
||||
panic()
|
||||
}
|
||||
|
||||
// Encode to WebP
|
||||
io := ("out.webp")
|
||||
w := bufio.NewWriter(f)
|
||||
defer func() {
|
||||
w.Flush()
|
||||
f.Close()
|
||||
}()
|
||||
|
||||
if err := webp.EncodeRGBA(w, img.(*image.RGBA), config); err != nil {
|
||||
panic(err)
|
||||
}
|
||||
}
|
||||
```
|
||||
|
||||
## TODO
|
||||
|
||||
- Incremental decoding API
|
||||
- Container API (Animation)
|
||||
|
||||
## License
|
||||
|
||||
Copyright (c) 2014 MICHII Shunsuke. All rights reserved.
|
||||
|
||||
This library is released under The BSD 2-Clause License.
|
||||
See [LICENSE](./LICENSE).
|
||||
@ -0,0 +1,5 @@
|
||||
Example Codes and Images
|
||||
========================
|
||||
|
||||
- [decode/decode.go](./decode/decode.go) -- a example code to decoding WebP image into image.RGBA.
|
||||
- [encode/encode.go](./encode/encode.go) -- a example code to encoding and writing image.RGBA into WebP file.
|
||||
@ -0,0 +1,22 @@
|
||||
package main
|
||||
|
||||
import (
|
||||
"github.com/harukasan/go-libwebp/examples/util"
|
||||
"github.com/harukasan/go-libwebp/webp"
|
||||
)
|
||||
|
||||
func main() {
|
||||
var err error
|
||||
|
||||
// Read binary data
|
||||
data := util.ReadFile("cosmos.webp")
|
||||
|
||||
// Decode
|
||||
options := &webp.DecoderOptions{}
|
||||
img, err := webp.DecodeRGBA(data, options)
|
||||
if err != nil {
|
||||
panic(err)
|
||||
}
|
||||
|
||||
util.WritePNG(img, "encoded_cosmos.png")
|
||||
}
|
||||
@ -0,0 +1,32 @@
|
||||
package main
|
||||
|
||||
import (
|
||||
"bufio"
|
||||
"image"
|
||||
|
||||
"github.com/harukasan/go-libwebp/examples/util"
|
||||
"github.com/harukasan/go-libwebp/webp"
|
||||
)
|
||||
|
||||
func main() {
|
||||
img := util.ReadPNG("cosmos.png")
|
||||
|
||||
// Create file and buffered writer
|
||||
io := util.CreateFile("encoded_cosmos.webp")
|
||||
w := bufio.NewWriter(io)
|
||||
defer func() {
|
||||
w.Flush()
|
||||
io.Close()
|
||||
}()
|
||||
|
||||
config := webp.Config{
|
||||
Preset: webp.PresetDefault,
|
||||
Quality: 90,
|
||||
Method: 6,
|
||||
}
|
||||
|
||||
// Encode into WebP
|
||||
if err := webp.EncodeRGBA(w, img.(*image.RGBA), config); err != nil {
|
||||
panic(err)
|
||||
}
|
||||
}
|
||||
@ -0,0 +1,111 @@
|
||||
Example Images
|
||||
==============
|
||||
|
||||
This directory contains example WebP encoded files and PNG source files.
|
||||
|
||||
## Image Credits
|
||||
|
||||
### Photos by Author
|
||||
|
||||
These images are taken by author.
|
||||
These photos are licensed under the Creative Commons Attribution 3.0.
|
||||
You can also use these images under the same as [go-libwebp's license](../LICENSE).
|
||||
|
||||
#### cosmos.png
|
||||
|
||||
The cosmos taken in Nokono-shima (Nokono Island), Fukuoka, JAPAN.
|
||||
|
||||
|
||||
|
||||
WebP file is generated by following command:
|
||||
|
||||
```sh
|
||||
$ cwebp -q 90 cosmos.png -o cosmos.webp
|
||||
Saving file 'cosmos.webp'
|
||||
File: cosmos.png
|
||||
Dimension: 1024 x 768
|
||||
Output: 76954 bytes Y-U-V-All-PSNR 45.87 47.63 48.10 46.44 dB
|
||||
block count: intra4: 2972
|
||||
intra16: 100 (-> 3.26%)
|
||||
skipped block: 1 (0.03%)
|
||||
bytes used: header: 249 (0.3%)
|
||||
mode-partition: 15161 (19.7%)
|
||||
Residuals bytes |segment 1|segment 2|segment 3|segment 4| total
|
||||
macroblocks: | 0%| 11%| 33%| 54%| 3072
|
||||
quantizer: | 12 | 11 | 9 | 8 |
|
||||
filter level: | 4 | 2 | 2 | 4 |
|
||||
```
|
||||
|
||||
#### butterfly.png
|
||||
|
||||
The butterfly taken in Ishigaki-jima (Ishigaki Island) in Okinawa, JAPAN.
|
||||
|
||||
|
||||
|
||||
WebP file is generated by following command:
|
||||
|
||||
```sh
|
||||
$ cwebp -q 90 butterfly.png -o butterfly.webp
|
||||
Saving file 'butterfly.webp'
|
||||
File: butterfly.png
|
||||
Dimension: 1024 x 768
|
||||
Output: 79198 bytes Y-U-V-All-PSNR 45.50 48.71 49.90 46.44 dB
|
||||
block count: intra4: 2775
|
||||
intra16: 297 (-> 9.67%)
|
||||
skipped block: 0 (0.00%)
|
||||
bytes used: header: 383 (0.5%)
|
||||
mode-partition: 13227 (16.7%)
|
||||
Residuals bytes |segment 1|segment 2|segment 3|segment 4| total
|
||||
macroblocks: | 1%| 7%| 14%| 75%| 3072
|
||||
quantizer: | 12 | 12 | 10 | 8 |
|
||||
filter level: | 4 | 3 | 2 | 6 |
|
||||
```
|
||||
|
||||
#### kinkaku.png
|
||||
|
||||
Kinkaku taken in Kyoto, JAPAN.
|
||||
|
||||
|
||||
|
||||
WebP file is generated by following command:
|
||||
|
||||
```sh
|
||||
$ cwebp -q 90 kinkaku.png -o kinkaku.webp
|
||||
Saving file 'kinkaku.webp'
|
||||
File: kinkaku.png
|
||||
Dimension: 1024 x 768
|
||||
Output: 186300 bytes Y-U-V-All-PSNR 43.86 47.25 48.26 44.81 dB
|
||||
block count: intra4: 2775
|
||||
intra16: 297 (-> 9.67%)
|
||||
skipped block: 243 (7.91%)
|
||||
bytes used: header: 425 (0.2%)
|
||||
mode-partition: 16737 (9.0%)
|
||||
Residuals bytes |segment 1|segment 2|segment 3|segment 4| total
|
||||
macroblocks: | 2%| 28%| 40%| 28%| 3072
|
||||
quantizer: | 12 | 11 | 9 | 6 |
|
||||
filter level: | 4 | 2 | 2 | 1 |
|
||||
```
|
||||
|
||||
### From Google WebP Gallery
|
||||
|
||||
These images picked up from [WebP Gallery](https://developers.google.com/speed/webp/gallery) in Google Developers.
|
||||
|
||||
#### yellow-rose-3.png
|
||||
|
||||
"Free Stock Photo in High Resolution - Yellow Rose 3 - Flowers" by Jon Sullivan
|
||||
|
||||
This image is clipped by Google and licensed under [Creative Commons Attribution 3.0](https://creativecommons.org/licenses/by/3.0/). The Source of this image is in the public domain.
|
||||
|
||||
<div style="background:url('./checkerboard.png')">
|
||||
<img title="yellow-rose-3.png" src="yellow-rose-3.png">
|
||||
</div>
|
||||
|
||||
#### fizyplankton.png
|
||||
|
||||
"baby tux for my user page" by Fizyplankton.
|
||||
|
||||
This image file is in the public domain.
|
||||
|
||||
<div style="background:url('./checkerboard.png')">
|
||||
<img title="fizyplankton.png" src="fizyplankton.png">
|
||||
</div>
|
||||
|
After Width: | Height: | Size: 997 KiB |
|
After Width: | Height: | Size: 77 KiB |
|
After Width: | Height: | Size: 1.1 KiB |
|
After Width: | Height: | Size: 964 KiB |
|
After Width: | Height: | Size: 75 KiB |
|
After Width: | Height: | Size: 44 KiB |
|
After Width: | Height: | Size: 15 KiB |
|
After Width: | Height: | Size: 1.2 MiB |
|
After Width: | Height: | Size: 182 KiB |
|
After Width: | Height: | Size: 118 KiB |
|
After Width: | Height: | Size: 20 KiB |
@ -0,0 +1,80 @@
|
||||
// Package util contains utility code for demosntration of go-libwebp.
|
||||
package util
|
||||
|
||||
import (
|
||||
"bufio"
|
||||
"image"
|
||||
"image/png"
|
||||
"io"
|
||||
"io/ioutil"
|
||||
"os"
|
||||
"path/filepath"
|
||||
)
|
||||
|
||||
// GetExFilePath returns the path of specified example file.
|
||||
func GetExFilePath(name string) string {
|
||||
return filepath.Join(os.Getenv("GOPATH"), "src/github.com/harukasan/go-libwebp/examples/images", name)
|
||||
}
|
||||
|
||||
// GetOutFilePath returns the path of specified out file.
|
||||
func GetOutFilePath(name string) string {
|
||||
return filepath.Join(os.Getenv("GOPATH"), "src/github.com/harukasan/go-libwebp/examples/out", name)
|
||||
}
|
||||
|
||||
// OpenFile opens specified example file
|
||||
func OpenFile(name string) (io io.Reader) {
|
||||
io, err := os.Open(GetExFilePath(name))
|
||||
if err != nil {
|
||||
panic(err)
|
||||
}
|
||||
return
|
||||
}
|
||||
|
||||
// ReadFile reads and returns data bytes of specified example file.
|
||||
func ReadFile(name string) (data []byte) {
|
||||
data, err := ioutil.ReadFile(GetExFilePath(name))
|
||||
if err != nil {
|
||||
panic(err)
|
||||
}
|
||||
return
|
||||
}
|
||||
|
||||
// CreateFile opens specified example file
|
||||
func CreateFile(name string) (f *os.File) {
|
||||
f, err := os.Create(GetOutFilePath(name))
|
||||
if err != nil {
|
||||
panic(err)
|
||||
}
|
||||
return
|
||||
}
|
||||
|
||||
// WritePNG encodes and writes image into PNG file.
|
||||
func WritePNG(img image.Image, name string) {
|
||||
f, err := os.Create(GetOutFilePath(name))
|
||||
if err != nil {
|
||||
panic(err)
|
||||
}
|
||||
b := bufio.NewWriter(f)
|
||||
defer func() {
|
||||
b.Flush()
|
||||
f.Close()
|
||||
}()
|
||||
|
||||
if err := png.Encode(b, img); err != nil {
|
||||
panic(err)
|
||||
}
|
||||
return
|
||||
}
|
||||
|
||||
// ReadPNG reads and decodes png data into image.Image
|
||||
func ReadPNG(name string) (img image.Image) {
|
||||
io, err := os.Open(GetExFilePath(name))
|
||||
if err != nil {
|
||||
panic(err)
|
||||
}
|
||||
img, err = png.Decode(io)
|
||||
if err != nil {
|
||||
panic(err)
|
||||
}
|
||||
return
|
||||
}
|
||||
@ -0,0 +1,80 @@
|
||||
mode: count
|
||||
github.com/harukasan/go-libwebp/webp/decode.go:41.34,43.2 1 1
|
||||
github.com/harukasan/go-libwebp/webp/decode.go:46.47,50.2 3 1
|
||||
github.com/harukasan/go-libwebp/webp/decode.go:53.65,57.37 3 1
|
||||
github.com/harukasan/go-libwebp/webp/decode.go:61.2,69.8 2 1
|
||||
github.com/harukasan/go-libwebp/webp/decode.go:57.37,59.13 1 0
|
||||
github.com/harukasan/go-libwebp/webp/decode.go:73.83,75.16 2 2
|
||||
github.com/harukasan/go-libwebp/webp/decode.go:79.2,83.107 3 2
|
||||
github.com/harukasan/go-libwebp/webp/decode.go:87.2,96.42 6 2
|
||||
github.com/harukasan/go-libwebp/webp/decode.go:100.2,114.32 14 2
|
||||
github.com/harukasan/go-libwebp/webp/decode.go:120.2,120.95 1 2
|
||||
github.com/harukasan/go-libwebp/webp/decode.go:124.2,124.8 1 2
|
||||
github.com/harukasan/go-libwebp/webp/decode.go:75.16,77.3 1 0
|
||||
github.com/harukasan/go-libwebp/webp/decode.go:83.107,85.13 1 0
|
||||
github.com/harukasan/go-libwebp/webp/decode.go:96.42,98.3 1 0
|
||||
github.com/harukasan/go-libwebp/webp/decode.go:114.32,118.3 3 0
|
||||
github.com/harukasan/go-libwebp/webp/decode.go:120.95,122.13 1 0
|
||||
github.com/harukasan/go-libwebp/webp/decode.go:128.84,130.16 2 3
|
||||
github.com/harukasan/go-libwebp/webp/decode.go:134.2,138.107 3 3
|
||||
github.com/harukasan/go-libwebp/webp/decode.go:143.2,157.95 9 3
|
||||
github.com/harukasan/go-libwebp/webp/decode.go:161.2,161.8 1 3
|
||||
github.com/harukasan/go-libwebp/webp/decode.go:130.16,132.3 1 0
|
||||
github.com/harukasan/go-libwebp/webp/decode.go:138.107,140.13 1 0
|
||||
github.com/harukasan/go-libwebp/webp/decode.go:157.95,159.13 1 0
|
||||
github.com/harukasan/go-libwebp/webp/decode.go:165.55,166.16 1 0
|
||||
github.com/harukasan/go-libwebp/webp/decode.go:184.2,184.33 1 0
|
||||
github.com/harukasan/go-libwebp/webp/decode.go:167.2,168.25 1 0
|
||||
github.com/harukasan/go-libwebp/webp/decode.go:169.2,170.36 1 0
|
||||
github.com/harukasan/go-libwebp/webp/decode.go:171.2,172.36 1 0
|
||||
github.com/harukasan/go-libwebp/webp/decode.go:173.2,174.38 1 0
|
||||
github.com/harukasan/go-libwebp/webp/decode.go:175.2,176.42 1 0
|
||||
github.com/harukasan/go-libwebp/webp/decode.go:177.2,178.32 1 0
|
||||
github.com/harukasan/go-libwebp/webp/decode.go:179.2,180.33 1 0
|
||||
github.com/harukasan/go-libwebp/webp/decode.go:181.2,182.38 1 0
|
||||
github.com/harukasan/go-libwebp/webp/decode.go:188.102,191.57 2 5
|
||||
github.com/harukasan/go-libwebp/webp/decode.go:196.2,196.29 1 5
|
||||
github.com/harukasan/go-libwebp/webp/decode.go:199.2,199.31 1 5
|
||||
github.com/harukasan/go-libwebp/webp/decode.go:202.2,202.54 1 5
|
||||
github.com/harukasan/go-libwebp/webp/decode.go:209.2,209.56 1 5
|
||||
github.com/harukasan/go-libwebp/webp/decode.go:214.2,214.24 1 5
|
||||
github.com/harukasan/go-libwebp/webp/decode.go:217.12,219.8 2 5
|
||||
github.com/harukasan/go-libwebp/webp/decode.go:191.57,193.3 1 0
|
||||
github.com/harukasan/go-libwebp/webp/decode.go:196.29,198.3 1 0
|
||||
github.com/harukasan/go-libwebp/webp/decode.go:199.31,201.3 1 0
|
||||
github.com/harukasan/go-libwebp/webp/decode.go:202.54,208.3 5 1
|
||||
github.com/harukasan/go-libwebp/webp/decode.go:209.56,213.3 3 1
|
||||
github.com/harukasan/go-libwebp/webp/decode.go:214.24,216.3 1 0
|
||||
github.com/harukasan/go-libwebp/webp/decode.go:223.82,225.29 2 5
|
||||
github.com/harukasan/go-libwebp/webp/decode.go:230.2,230.37 1 4
|
||||
github.com/harukasan/go-libwebp/webp/decode.go:236.2,238.18 3 3
|
||||
github.com/harukasan/go-libwebp/webp/decode.go:225.29,229.3 3 1
|
||||
github.com/harukasan/go-libwebp/webp/decode.go:230.37,234.3 3 1
|
||||
github.com/harukasan/go-libwebp/webp/encode.go:42.101,46.16 4 11
|
||||
github.com/harukasan/go-libwebp/webp/encode.go:49.2,49.10 1 11
|
||||
github.com/harukasan/go-libwebp/webp/encode.go:46.16,48.3 1 0
|
||||
github.com/harukasan/go-libwebp/webp/encode.go:53.69,55.16 2 1
|
||||
github.com/harukasan/go-libwebp/webp/encode.go:59.2,60.49 2 1
|
||||
github.com/harukasan/go-libwebp/webp/encode.go:63.12,73.46 8 1
|
||||
github.com/harukasan/go-libwebp/webp/encode.go:77.2,77.8 1 1
|
||||
github.com/harukasan/go-libwebp/webp/encode.go:55.16,57.3 1 0
|
||||
github.com/harukasan/go-libwebp/webp/encode.go:60.49,62.3 1 0
|
||||
github.com/harukasan/go-libwebp/webp/encode.go:73.46,75.13 1 0
|
||||
github.com/harukasan/go-libwebp/webp/encode.go:81.68,83.16 2 1
|
||||
github.com/harukasan/go-libwebp/webp/encode.go:87.2,88.49 2 1
|
||||
github.com/harukasan/go-libwebp/webp/encode.go:91.12,101.31 10 1
|
||||
github.com/harukasan/go-libwebp/webp/encode.go:106.2,109.46 3 1
|
||||
github.com/harukasan/go-libwebp/webp/encode.go:113.2,113.8 1 1
|
||||
github.com/harukasan/go-libwebp/webp/encode.go:83.16,85.3 1 0
|
||||
github.com/harukasan/go-libwebp/webp/encode.go:88.49,90.3 1 0
|
||||
github.com/harukasan/go-libwebp/webp/encode.go:101.31,104.3 2 0
|
||||
github.com/harukasan/go-libwebp/webp/encode.go:109.46,111.13 1 0
|
||||
github.com/harukasan/go-libwebp/webp/encode.go:117.73,119.96 2 2
|
||||
github.com/harukasan/go-libwebp/webp/encode.go:122.12,125.44 3 2
|
||||
github.com/harukasan/go-libwebp/webp/encode.go:128.2,128.8 1 2
|
||||
github.com/harukasan/go-libwebp/webp/encode.go:119.96,121.3 1 0
|
||||
github.com/harukasan/go-libwebp/webp/encode.go:125.44,127.3 1 0
|
||||
github.com/harukasan/go-libwebp/webp/yuva_image.go:21.71,25.11 3 2
|
||||
github.com/harukasan/go-libwebp/webp/yuva_image.go:55.2,55.8 1 2
|
||||
github.com/harukasan/go-libwebp/webp/yuva_image.go:26.2,38.4 2 2
|
||||
github.com/harukasan/go-libwebp/webp/yuva_image.go:40.2,52.4 2 0
|
||||
@ -0,0 +1,242 @@
|
||||
package webp
|
||||
|
||||
/*
|
||||
#cgo LDFLAGS: -lwebp
|
||||
#include <stdlib.h>
|
||||
#include <webp/decode.h>
|
||||
|
||||
static VP8StatusCode CheckDecBuffer(const WebPDecBuffer* const buffer);
|
||||
|
||||
*/
|
||||
import "C"
|
||||
|
||||
import (
|
||||
"errors"
|
||||
"fmt"
|
||||
"image"
|
||||
"unsafe"
|
||||
)
|
||||
|
||||
// DecoderOptions specifies decoding options
|
||||
type DecoderOptions struct {
|
||||
BypassFiltering bool
|
||||
NoFancyUpsampling bool
|
||||
Crop image.Rectangle
|
||||
Scale image.Rectangle
|
||||
UseThreads bool
|
||||
DitheringStrength int
|
||||
Flip bool
|
||||
AlphaDitheringStrength int
|
||||
}
|
||||
|
||||
// BitstreamFeatures retrived from data stream.
|
||||
type BitstreamFeatures struct {
|
||||
Width int // Image width in pixels
|
||||
Height int // Image height in pixles
|
||||
HasAlpha bool // True if data stream contains a alpha channel.
|
||||
HasAnimation bool // True if data stream is an animation
|
||||
Format int // Image compression format
|
||||
NoIncrementalDecoding bool // True if incremental decording is not using
|
||||
}
|
||||
|
||||
// GetDecoderVersion returns decoder's version number, packed in hexadecimal.
|
||||
// e.g; v0.4.2 is 0x000402
|
||||
func GetDecoderVersion() (v int) {
|
||||
return int(C.WebPGetDecoderVersion())
|
||||
}
|
||||
|
||||
// GetInfo retrives width/height from data bytes.
|
||||
func GetInfo(data []byte) (width, height int) {
|
||||
var w, h C.int
|
||||
C.WebPGetInfo((*C.uint8_t)(&data[0]), (C.size_t)(len(data)), &w, &h)
|
||||
return int(w), int(h)
|
||||
}
|
||||
|
||||
// GetFeatures returns features retrived from data stream.
|
||||
func GetFeatures(data []byte) (f *BitstreamFeatures, err error) {
|
||||
var cf C.WebPBitstreamFeatures
|
||||
status := C.WebPGetFeatures((*C.uint8_t)(&data[0]), (C.size_t)(len(data)), &cf)
|
||||
|
||||
if status != C.VP8_STATUS_OK {
|
||||
return nil, fmt.Errorf("WebPGetFeatures returns unexpected status: %s", statusString(status))
|
||||
}
|
||||
|
||||
f = &BitstreamFeatures{
|
||||
Width: int(cf.width), // TODO: use Rectangle instaed?
|
||||
Height: int(cf.height),
|
||||
HasAlpha: cf.has_alpha > 0,
|
||||
HasAnimation: cf.has_animation > 0,
|
||||
Format: int(cf.format),
|
||||
NoIncrementalDecoding: cf.no_incremental_decoding == 1,
|
||||
}
|
||||
return
|
||||
}
|
||||
|
||||
// DecodeYUVA decodes WebP image into YUV image with alpha channel.
|
||||
func DecodeYUVA(data []byte, options *DecoderOptions) (img *YUVAImage, err error) {
|
||||
config, err := initDecoderConfig(options)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
cDataPtr := (*C.uint8_t)(&data[0])
|
||||
cDataSize := (C.size_t)(len(data))
|
||||
|
||||
// Retrive WebP features from data stream
|
||||
if status := C.WebPGetFeatures(cDataPtr, cDataSize, &config.input); status != C.VP8_STATUS_OK {
|
||||
return nil, fmt.Errorf("Could not get features from the data stream, return %s", statusString(status))
|
||||
}
|
||||
|
||||
outWidth, outHeight := calcOutputSize(config)
|
||||
buf := (*C.WebPYUVABuffer)(unsafe.Pointer(&config.output.u[0]))
|
||||
|
||||
// Set up output configurations
|
||||
colorSpace := YUV420
|
||||
config.output.colorspace = C.MODE_YUV
|
||||
if config.input.has_alpha > 0 {
|
||||
colorSpace = YUV420A
|
||||
config.output.colorspace = C.MODE_YUVA
|
||||
}
|
||||
config.output.is_external_memory = 1
|
||||
|
||||
// Allocate image and fill into buffer
|
||||
img = NewYUVAImage(image.Rect(0, 0, outWidth, outHeight), colorSpace)
|
||||
buf.y = (*C.uint8_t)(&img.Y[0])
|
||||
buf.u = (*C.uint8_t)(&img.Cb[0])
|
||||
buf.v = (*C.uint8_t)(&img.Cr[0])
|
||||
buf.a = nil
|
||||
buf.y_stride = C.int(img.YStride)
|
||||
buf.u_stride = C.int(img.CStride)
|
||||
buf.v_stride = C.int(img.CStride)
|
||||
buf.a_stride = 0
|
||||
buf.y_size = C.size_t(len(img.Y))
|
||||
buf.u_size = C.size_t(len(img.Cb))
|
||||
buf.v_size = C.size_t(len(img.Cr))
|
||||
buf.a_size = 0
|
||||
|
||||
if config.input.has_alpha > 0 {
|
||||
buf.a = (*C.uint8_t)(&img.A[0])
|
||||
buf.a_stride = C.int(img.AStride)
|
||||
buf.a_size = C.size_t(len(img.A))
|
||||
}
|
||||
|
||||
if status := C.WebPDecode(cDataPtr, cDataSize, config); status != C.VP8_STATUS_OK {
|
||||
return nil, fmt.Errorf("Could not decode data stream, return %s", statusString(status))
|
||||
}
|
||||
|
||||
return
|
||||
}
|
||||
|
||||
// DecodeRGBA decodes WebP image into RGBA image.
|
||||
func DecodeRGBA(data []byte, options *DecoderOptions) (img *image.RGBA, err error) {
|
||||
config, err := initDecoderConfig(options)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
cDataPtr := (*C.uint8_t)(&data[0])
|
||||
cDataSize := (C.size_t)(len(data))
|
||||
|
||||
// Retrive WebP features
|
||||
if status := C.WebPGetFeatures(cDataPtr, cDataSize, &config.input); status != C.VP8_STATUS_OK {
|
||||
return nil, fmt.Errorf("Could not get features from the data stream, return %s", statusString(status))
|
||||
}
|
||||
|
||||
// Allocate output image
|
||||
outWidth, outHeight := calcOutputSize(config)
|
||||
img = image.NewRGBA(image.Rect(0, 0, outWidth, outHeight))
|
||||
|
||||
// Set up output configurations
|
||||
config.output.colorspace = C.MODE_RGBA
|
||||
config.output.is_external_memory = 1
|
||||
|
||||
// Allocate WebPRGBABuffer and fill in the pointers to output image
|
||||
buf := (*C.WebPRGBABuffer)(unsafe.Pointer(&config.output.u[0]))
|
||||
buf.rgba = (*C.uint8_t)(&img.Pix[0])
|
||||
buf.stride = C.int(img.Stride)
|
||||
buf.size = (C.size_t)(len(img.Pix))
|
||||
|
||||
// Decode
|
||||
if status := C.WebPDecode(cDataPtr, cDataSize, config); status != C.VP8_STATUS_OK {
|
||||
return nil, fmt.Errorf("Could not decode data stream, return %s", statusString(status))
|
||||
}
|
||||
|
||||
return
|
||||
}
|
||||
|
||||
// sattusString convert the VP8StatsCode to string.
|
||||
func statusString(status C.VP8StatusCode) string {
|
||||
switch status {
|
||||
case C.VP8_STATUS_OK:
|
||||
return "VP8_STATUS_OK"
|
||||
case C.VP8_STATUS_OUT_OF_MEMORY:
|
||||
return "VP8_STATUS_OUT_OF_MEMORY"
|
||||
case C.VP8_STATUS_INVALID_PARAM:
|
||||
return "VP8_STATUS_INVALID_PARAM"
|
||||
case C.VP8_STATUS_BITSTREAM_ERROR:
|
||||
return "VP8_STATUS_BITSTREAM_ERROR"
|
||||
case C.VP8_STATUS_UNSUPPORTED_FEATURE:
|
||||
return "VP8_STATUS_UNSUPPORTED_FEATURE"
|
||||
case C.VP8_STATUS_SUSPENDED:
|
||||
return "VP8_STATUS_SUSPENDED"
|
||||
case C.VP8_STATUS_USER_ABORT:
|
||||
return "VP8_STATUS_USER_ABORT"
|
||||
case C.VP8_STATUS_NOT_ENOUGH_DATA:
|
||||
return "VP8_STATUS_NOT_ENOUGH_DATA"
|
||||
}
|
||||
return "Unexpected Status Code"
|
||||
}
|
||||
|
||||
// initDecoderConfing initializes a decoder configration and sets up the options.
|
||||
func initDecoderConfig(options *DecoderOptions) (config *C.WebPDecoderConfig, err error) {
|
||||
// Initialize decoder config
|
||||
config = &C.WebPDecoderConfig{}
|
||||
if C.WebPInitDecoderConfig(config) == 0 {
|
||||
return nil, errors.New("Could not initialize decoder config")
|
||||
}
|
||||
|
||||
// Set up decoder options
|
||||
if options.BypassFiltering {
|
||||
config.options.bypass_filtering = 1
|
||||
}
|
||||
if options.NoFancyUpsampling {
|
||||
config.options.no_fancy_upsampling = 1
|
||||
}
|
||||
if options.Crop.Max.X > 0 && options.Crop.Max.Y > 0 {
|
||||
config.options.use_cropping = 1
|
||||
config.options.crop_left = C.int(options.Crop.Min.X)
|
||||
config.options.crop_top = C.int(options.Crop.Min.Y)
|
||||
config.options.crop_width = C.int(options.Crop.Dx())
|
||||
config.options.crop_height = C.int(options.Crop.Dy())
|
||||
}
|
||||
if options.Scale.Max.X > 0 && options.Scale.Max.Y > 0 {
|
||||
config.options.use_scaling = 1
|
||||
config.options.scaled_width = C.int(options.Scale.Max.X)
|
||||
config.options.scaled_height = C.int(options.Scale.Max.Y)
|
||||
}
|
||||
if options.UseThreads {
|
||||
config.options.use_threads = 1
|
||||
}
|
||||
config.options.dithering_strength = C.int(options.DitheringStrength)
|
||||
|
||||
return
|
||||
}
|
||||
|
||||
// calcOutputSize retrives width and height of output image from the decoder configuration.
|
||||
func calcOutputSize(config *C.WebPDecoderConfig) (width, height int) {
|
||||
options := config.options
|
||||
if options.use_scaling > 0 {
|
||||
width = int(config.options.scaled_width)
|
||||
height = int(config.options.scaled_height)
|
||||
return
|
||||
}
|
||||
if config.options.use_cropping > 0 {
|
||||
width = int(config.options.crop_width)
|
||||
height = int(config.options.crop_height)
|
||||
return
|
||||
}
|
||||
|
||||
width = int(config.input.width)
|
||||
height = int(config.input.height)
|
||||
return
|
||||
}
|
||||
@ -0,0 +1,139 @@
|
||||
package webp
|
||||
|
||||
/*
|
||||
#cgo LDFLAGS: -lwebp
|
||||
|
||||
#include <stdlib.h>
|
||||
#include <webp/encode.h>
|
||||
|
||||
int writeWebP(uint8_t*, size_t, struct WebPPicture*);
|
||||
|
||||
*/
|
||||
import "C"
|
||||
import (
|
||||
"errors"
|
||||
"fmt"
|
||||
"image"
|
||||
"io"
|
||||
"unsafe"
|
||||
)
|
||||
|
||||
// Config specifies WebP encoding configuration.
|
||||
type Config struct {
|
||||
Preset Preset // Parameters Preset
|
||||
Lossless bool // True if use lossless encoding
|
||||
Quality float32 // WebP quality factor, 0-100
|
||||
Method int // Quality/Speed trade-off factor, 0=faster / 6=slower-better
|
||||
TargetSize int // Target size of encoded file in bytes
|
||||
TargetPSNR float32 // Target PSNR, takes precedence over TargetSize
|
||||
Segments int // Maximum number of segments to use, 1..4
|
||||
SNSStrength int // Strength of spartial noise shaping, 0..100=maximum
|
||||
FilterStrength int // Strength of filter, 0..100=strength
|
||||
FilterSharpness int // Sharpness of filter, 0..7=sharpness
|
||||
FilterType FilterType // Filtering type
|
||||
Pass int // Number of entropy-analysis passes, 0..100
|
||||
}
|
||||
|
||||
type destinationManager struct {
|
||||
writer io.Writer
|
||||
}
|
||||
|
||||
//export writeWebP
|
||||
func writeWebP(data *C.uint8_t, size C.size_t, pic *C.WebPPicture) C.int {
|
||||
mgr := (*destinationManager)(unsafe.Pointer(pic.custom_ptr))
|
||||
bytes := C.GoBytes(unsafe.Pointer(data), C.int(size))
|
||||
_, err := mgr.writer.Write(bytes)
|
||||
if err != nil {
|
||||
return 0 // TODO: can't pass error message
|
||||
}
|
||||
return 1
|
||||
}
|
||||
|
||||
// EncodeRGBA encodes and writes image.Image into the writer as WebP.
|
||||
// Now supports image.RGBA or image.NRGBA.
|
||||
func EncodeRGBA(w io.Writer, img image.Image, c Config) (err error) {
|
||||
webpConfig, err := initConfig(c)
|
||||
if err != nil {
|
||||
return
|
||||
}
|
||||
|
||||
var pic C.WebPPicture
|
||||
if C.WebPPictureInit(&pic) == 0 {
|
||||
return errors.New("Could not initialize webp picture")
|
||||
}
|
||||
pic.use_argb = 1
|
||||
|
||||
pic.width = C.int(img.Bounds().Dx())
|
||||
pic.height = C.int(img.Bounds().Dy())
|
||||
|
||||
pic.writer = C.WebPWriterFunction(C.writeWebP)
|
||||
pic.custom_ptr = unsafe.Pointer(&destinationManager{writer: w})
|
||||
|
||||
switch p := img.(type) {
|
||||
case *image.RGBA:
|
||||
C.WebPPictureImportRGBA(&pic, (*C.uint8_t)(&p.Pix[0]), C.int(p.Stride))
|
||||
case *image.NRGBA:
|
||||
C.WebPPictureImportRGBA(&pic, (*C.uint8_t)(&p.Pix[0]), C.int(p.Stride))
|
||||
default:
|
||||
return errors.New("unsupported image type")
|
||||
}
|
||||
|
||||
defer C.WebPPictureFree(&pic)
|
||||
|
||||
if C.WebPEncode(webpConfig, &pic) == 0 {
|
||||
return fmt.Errorf("Encoding error: %d", pic.error_code)
|
||||
}
|
||||
|
||||
return
|
||||
}
|
||||
|
||||
// EncodeYUVA encodes and writes YUVA Image data into the writer as WebP.
|
||||
func EncodeYUVA(w io.Writer, img *YUVAImage, c Config) (err error) {
|
||||
webpConfig, err := initConfig(c)
|
||||
if err != nil {
|
||||
return
|
||||
}
|
||||
|
||||
var pic C.WebPPicture
|
||||
if C.WebPPictureInit(&pic) == 0 {
|
||||
return errors.New("Could not initialize webp picture")
|
||||
}
|
||||
pic.use_argb = 0
|
||||
pic.colorspace = C.WebPEncCSP(img.ColorSpace)
|
||||
pic.width = C.int(img.Rect.Dx())
|
||||
pic.height = C.int(img.Rect.Dy())
|
||||
pic.y = (*C.uint8_t)(&img.Y[0])
|
||||
pic.u = (*C.uint8_t)(&img.Cb[0])
|
||||
pic.v = (*C.uint8_t)(&img.Cr[0])
|
||||
pic.y_stride = C.int(img.YStride)
|
||||
pic.uv_stride = C.int(img.CStride)
|
||||
|
||||
if img.ColorSpace == YUV420A {
|
||||
pic.a = (*C.uint8_t)(&img.A[0])
|
||||
pic.a_stride = C.int(img.AStride)
|
||||
}
|
||||
|
||||
pic.writer = C.WebPWriterFunction(C.writeWebP)
|
||||
pic.custom_ptr = unsafe.Pointer(&destinationManager{writer: w})
|
||||
|
||||
if C.WebPEncode(webpConfig, &pic) == 0 {
|
||||
return fmt.Errorf("Encoding error: %d", pic.error_code)
|
||||
}
|
||||
|
||||
return
|
||||
}
|
||||
|
||||
// initConfig initializes C.WebPConfig with encoding parameters.
|
||||
func initConfig(c Config) (config *C.WebPConfig, err error) {
|
||||
config = &C.WebPConfig{}
|
||||
if C.WebPConfigPreset(config, C.WebPPreset(c.Preset), C.float(c.Quality)) == 0 {
|
||||
return nil, errors.New("Could not initialize configuration with preset")
|
||||
}
|
||||
config.target_size = C.int(c.TargetSize)
|
||||
config.target_PSNR = C.float(c.TargetPSNR)
|
||||
|
||||
if C.WebPValidateConfig(config) == 0 {
|
||||
return nil, errors.New("Invalid configuration")
|
||||
}
|
||||
return
|
||||
}
|
||||
@ -0,0 +1,48 @@
|
||||
package webp
|
||||
|
||||
/*
|
||||
#cgo LDFLAGS: -lwebp
|
||||
|
||||
#include <stdlib.h>
|
||||
#include <webp/encode.h>
|
||||
|
||||
*/
|
||||
import "C"
|
||||
|
||||
// ColorSpace represents encoding color space in WebP
|
||||
type ColorSpace int
|
||||
|
||||
const (
|
||||
// YUV420 specifies YUV4:2:0
|
||||
YUV420 ColorSpace = C.WEBP_YUV420
|
||||
// YUV420A specifies YUV4:2:0 with alpha channel
|
||||
YUV420A ColorSpace = C.WEBP_YUV420A
|
||||
)
|
||||
|
||||
// Preset corresponds to C.WebPPreset
|
||||
type Preset int
|
||||
|
||||
const (
|
||||
// PresetDefault corresponds to WEBP_PRESET_DEFAULT, for default preset.
|
||||
PresetDefault Preset = C.WEBP_PRESET_DEFAULT
|
||||
// PresetPicture corresponds to WEBP_PRESET_PICTURE, for digital picture, like portrait, inner shot
|
||||
PresetPicture Preset = C.WEBP_PRESET_PICTURE
|
||||
// PresetPhoto corresponds to WEBP_PRESET_PHOTO, for outdoor photograph, with natural lighting
|
||||
PresetPhoto Preset = C.WEBP_PRESET_PHOTO
|
||||
// PresetDrawing corresponds to WEBP_PRESET_DRAWING, for hand or line drawing, with high-contrast details
|
||||
PresetDrawing Preset = C.WEBP_PRESET_DRAWING
|
||||
// PresetIcon corresponds to WEBP_PRESET_ICON, for small-sized colorful images
|
||||
PresetIcon Preset = C.WEBP_PRESET_ICON
|
||||
// PresetText corresponds to WEBP_PRESET_TEXT, for text-like
|
||||
PresetText Preset = C.WEBP_PRESET_TEXT
|
||||
)
|
||||
|
||||
// FilterType corresponds to filter types in compression parameters.
|
||||
type FilterType int
|
||||
|
||||
const (
|
||||
// SimpleFilter (=0, default)
|
||||
SimpleFilter FilterType = iota
|
||||
// StrongFilter (=1)
|
||||
StrongFilter
|
||||
)
|
||||
@ -0,0 +1,236 @@
|
||||
package webp_test
|
||||
|
||||
import (
|
||||
"bufio"
|
||||
"image"
|
||||
"image/png"
|
||||
"io"
|
||||
"io/ioutil"
|
||||
"os"
|
||||
"path/filepath"
|
||||
"testing"
|
||||
|
||||
"github.com/harukasan/go-libwebp/webp"
|
||||
)
|
||||
|
||||
//
|
||||
// Utitlities of input/output example images
|
||||
//
|
||||
|
||||
// GetExFilePath returns the path of specified example file.
|
||||
func GetExFilePath(name string) string {
|
||||
return filepath.Join(os.Getenv("GOPATH"), "src/github.com/harukasan/go-libwebp/examples/images", name)
|
||||
}
|
||||
|
||||
// OpenExFile opens specified example file
|
||||
func OpenExFile(name string) (io io.Reader) {
|
||||
io, err := os.Open(GetExFilePath(name))
|
||||
if err != nil {
|
||||
panic(err)
|
||||
}
|
||||
return
|
||||
}
|
||||
|
||||
// ReadExFile reads and returns data bytes of specified example file.
|
||||
func ReadExFile(name string) (data []byte) {
|
||||
data, err := ioutil.ReadFile(GetExFilePath(name))
|
||||
if err != nil {
|
||||
panic(err)
|
||||
}
|
||||
return
|
||||
}
|
||||
|
||||
// CreateExOutFile creates output file into example/out directory.
|
||||
func CreateExOutFile(name string) (file *os.File) {
|
||||
// Make output directory
|
||||
dir := filepath.Join(os.Getenv("GOPATH"), "src/github.com/harukasan/go-libwebp/examples/out")
|
||||
if err := os.Mkdir(dir, 0755); err != nil && os.IsNotExist(err) {
|
||||
panic(err)
|
||||
}
|
||||
|
||||
// Create example output file
|
||||
file, err := os.Create(filepath.Join(dir, name))
|
||||
if err != nil {
|
||||
panic(err)
|
||||
}
|
||||
return
|
||||
}
|
||||
|
||||
//
|
||||
// Decode
|
||||
//
|
||||
|
||||
// Test Get Decoder Version
|
||||
func TestGetDecoderVersion(t *testing.T) {
|
||||
v := webp.GetDecoderVersion()
|
||||
if v < 0 {
|
||||
t.Errorf("GetDecoderVersion should returns positive version number, got %v\n", v)
|
||||
}
|
||||
}
|
||||
|
||||
func TestGetInfo(t *testing.T) {
|
||||
data := ReadExFile("cosmos.webp")
|
||||
width, height := webp.GetInfo(data)
|
||||
|
||||
if width != 1024 {
|
||||
t.Errorf("Expected width: %d, but got %d", 1024, width)
|
||||
}
|
||||
if height != 768 {
|
||||
t.Errorf("Expected height: %d, but got %d", 768, height)
|
||||
}
|
||||
}
|
||||
|
||||
func TestGetFeatures(t *testing.T) {
|
||||
data := ReadExFile("cosmos.webp")
|
||||
f, err := webp.GetFeatures(data)
|
||||
if err != nil {
|
||||
t.Errorf("Got Error: %v", err)
|
||||
return
|
||||
}
|
||||
if got := f.Width; got != 1024 {
|
||||
t.Errorf("Expected Width: %v, but got %v", 1024, got)
|
||||
}
|
||||
if got := f.Height; got != 768 {
|
||||
t.Errorf("Expected Width: %v, but got %v", 768, got)
|
||||
}
|
||||
if got := f.HasAlpha; got != false {
|
||||
t.Errorf("Expected HasAlpha: %v, but got %v", false, got)
|
||||
}
|
||||
if got := f.HasAnimation; got != false {
|
||||
t.Errorf("Expected HasAlpha: %v, but got %v", false, got)
|
||||
}
|
||||
if got := f.Format; got != 1 {
|
||||
t.Errorf("Expected Format: %v, but got %v", 1, got)
|
||||
}
|
||||
}
|
||||
|
||||
func TestDecodeYUV(t *testing.T) {
|
||||
files := []string{
|
||||
"cosmos.webp",
|
||||
"butterfly.webp",
|
||||
"kinkaku.webp",
|
||||
"yellow-rose-3.webp",
|
||||
}
|
||||
|
||||
for _, file := range files {
|
||||
data := ReadExFile(file)
|
||||
options := &webp.DecoderOptions{}
|
||||
|
||||
_, err := webp.DecodeYUVA(data, options)
|
||||
if err != nil {
|
||||
t.Errorf("Got Error: %v", err)
|
||||
return
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
func TestDecodeRGBA(t *testing.T) {
|
||||
files := []string{
|
||||
"cosmos.webp",
|
||||
"butterfly.webp",
|
||||
"kinkaku.webp",
|
||||
"yellow-rose-3.webp",
|
||||
}
|
||||
|
||||
for _, file := range files {
|
||||
data := ReadExFile(file)
|
||||
options := &webp.DecoderOptions{}
|
||||
|
||||
_, err := webp.DecodeRGBA(data, options)
|
||||
if err != nil {
|
||||
t.Errorf("Got Error: %v", err)
|
||||
return
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
func TestDecodeRGBAWithCropping(t *testing.T) {
|
||||
data := ReadExFile("cosmos.webp")
|
||||
crop := image.Rect(100, 100, 300, 200)
|
||||
|
||||
options := &webp.DecoderOptions{
|
||||
Crop: crop,
|
||||
}
|
||||
|
||||
img, err := webp.DecodeRGBA(data, options)
|
||||
if err != nil {
|
||||
t.Errorf("Got Error: %v", err)
|
||||
return
|
||||
}
|
||||
if img.Rect.Dx() != crop.Dx() || img.Rect.Dy() != crop.Dy() {
|
||||
t.Errorf("Decoded image should cropped to %v, but got %v", crop, img.Rect)
|
||||
}
|
||||
}
|
||||
|
||||
func TestDecodeRGBAWithScaling(t *testing.T) {
|
||||
data := ReadExFile("cosmos.webp")
|
||||
scale := image.Rect(0, 0, 640, 480)
|
||||
|
||||
options := &webp.DecoderOptions{
|
||||
Scale: scale,
|
||||
}
|
||||
|
||||
img, err := webp.DecodeRGBA(data, options)
|
||||
if err != nil {
|
||||
t.Errorf("Got Error: %v", err)
|
||||
return
|
||||
}
|
||||
if img.Rect.Dx() != scale.Dx() || img.Rect.Dy() != scale.Dy() {
|
||||
t.Errorf("Decoded image should scaled to %v, but got %v", scale, img.Rect)
|
||||
}
|
||||
}
|
||||
|
||||
//
|
||||
// Encoding
|
||||
//
|
||||
|
||||
func TestEncodeRGBA(t *testing.T) {
|
||||
img, _ := png.Decode(OpenExFile("yellow-rose-3.png"))
|
||||
|
||||
config := webp.Config{
|
||||
Preset: webp.PresetDefault,
|
||||
Quality: 100,
|
||||
Method: 6,
|
||||
}
|
||||
|
||||
f := CreateExOutFile("TestEncodeRGBA.webp")
|
||||
w := bufio.NewWriter(f)
|
||||
defer func() {
|
||||
w.Flush()
|
||||
f.Close()
|
||||
}()
|
||||
|
||||
if err := webp.EncodeRGBA(w, img, config); err != nil {
|
||||
t.Errorf("Got Error: %v", err)
|
||||
return
|
||||
}
|
||||
}
|
||||
|
||||
func TestEncodeYUVA(t *testing.T) {
|
||||
data := ReadExFile("cosmos.webp")
|
||||
options := &webp.DecoderOptions{}
|
||||
|
||||
img, err := webp.DecodeYUVA(data, options)
|
||||
if err != nil {
|
||||
t.Errorf("Got Error: %v in decoding", err)
|
||||
return
|
||||
}
|
||||
|
||||
f := CreateExOutFile("TestEncodeYUVA.webp")
|
||||
w := bufio.NewWriter(f)
|
||||
defer func() {
|
||||
w.Flush()
|
||||
f.Close()
|
||||
}()
|
||||
|
||||
config := webp.Config{
|
||||
Preset: webp.PresetDefault,
|
||||
Quality: 100,
|
||||
Method: 6,
|
||||
}
|
||||
|
||||
if err := webp.EncodeYUVA(w, img, config); err != nil {
|
||||
t.Errorf("Got Error: %v", err)
|
||||
return
|
||||
}
|
||||
}
|
||||
@ -0,0 +1,56 @@
|
||||
package webp
|
||||
|
||||
import "image"
|
||||
|
||||
// YUVAImage represents a image of YUV colors with alpha channel image.
|
||||
//
|
||||
// We can not insert WebP planer image into image.YCbCr,
|
||||
// because image.YCbCr is not compatible with ITU-R BT.601, but JFIF/JPEG one.
|
||||
//
|
||||
// See: http://en.wikipedia.org/wiki/YCbCr
|
||||
type YUVAImage struct {
|
||||
Y, Cb, Cr, A []uint8
|
||||
YStride int
|
||||
CStride int
|
||||
AStride int
|
||||
ColorSpace ColorSpace
|
||||
Rect image.Rectangle
|
||||
}
|
||||
|
||||
// NewYUVAImage creates and allocates image buffer.
|
||||
func NewYUVAImage(r image.Rectangle, c ColorSpace) (image *YUVAImage) {
|
||||
yw, yh := r.Dx(), r.Dx()
|
||||
cw, ch := ((r.Max.X+1)/2 - r.Min.X/2), ((r.Max.Y+1)/2 - r.Min.Y/2)
|
||||
|
||||
switch c {
|
||||
case YUV420:
|
||||
b := make([]byte, yw*yh+2*cw*ch)
|
||||
image = &YUVAImage{
|
||||
Y: b[:yw*yh],
|
||||
Cb: b[yw*yh+0*cw*ch : yw*yh+1*cw*ch],
|
||||
Cr: b[yw*yh+1*cw*ch : yw*yh+2*cw*ch],
|
||||
A: nil,
|
||||
YStride: yw,
|
||||
CStride: cw,
|
||||
AStride: 0,
|
||||
ColorSpace: c,
|
||||
Rect: r,
|
||||
}
|
||||
|
||||
case YUV420A:
|
||||
b := make([]byte, 2*yw*yh+2*cw*ch)
|
||||
image = &YUVAImage{
|
||||
Y: b[:yw*yh],
|
||||
Cb: b[yw*yh+0*cw*ch : yw*yh+1*cw*ch],
|
||||
Cr: b[yw*yh+1*cw*ch : yw*yh+2*cw*ch],
|
||||
A: b[yw*yh+2*cw*ch:],
|
||||
YStride: yw,
|
||||
CStride: cw,
|
||||
AStride: yw,
|
||||
ColorSpace: c,
|
||||
Rect: r,
|
||||
}
|
||||
}
|
||||
|
||||
return
|
||||
}
|
||||
Loading…
Reference in new issue