1
1
mirror of https://github.com/wader/fq.git synced 2024-12-24 22:05:31 +03:00
fq/format/webp/webp.go

74 lines
1.6 KiB
Go

package webp
// https://developers.google.com/speed/webp/docs/riff_container
import (
"bytes"
"github.com/wader/fq/format"
"github.com/wader/fq/pkg/decode"
"github.com/wader/fq/pkg/interp"
"github.com/wader/fq/pkg/scalar"
)
var vp8Frame decode.Group
func init() {
interp.RegisterFormat(decode.Format{
Name: format.WEBP,
Description: "WebP image",
Groups: []string{format.PROBE, format.IMAGE},
DecodeFn: webpDecode,
Dependencies: []decode.Dependency{
{Names: []string{format.VP8_FRAME}, Group: &vp8Frame},
},
})
}
func decodeChunk(d *decode.D, expectedChunkID string, fn func(d *decode.D)) {
trimChunkID := d.FieldUTF8("id", 4, scalar.ActualTrimSpace)
if expectedChunkID != "" && trimChunkID != expectedChunkID {
return
}
chunkLen := int64(d.FieldU32("size"))
if fn != nil {
d.FramedFn(chunkLen*8, fn)
} else {
d.FieldRawLen("data", chunkLen*8)
}
}
func webpDecode(d *decode.D, _ any) any {
d.Endian = decode.LittleEndian
d.FieldUTF8("riff_id", 4, d.AssertStr("RIFF"))
riffLength := d.FieldU32("riff_length")
d.FieldUTF8("webp_id", 4, d.AssertStr("WEBP"))
d.FramedFn(int64(riffLength-4)*8, func(d *decode.D) {
p := d.PeekBytes(4)
// TODO: VP8X
switch {
case bytes.Equal(p, []byte("VP8 ")):
d.FieldStruct("image", func(d *decode.D) {
decodeChunk(d, "VP8", func(d *decode.D) {
d.Format(vp8Frame, nil)
})
})
case bytes.Equal(p, []byte("VP8L")):
d.FieldStruct("image", func(d *decode.D) {
decodeChunk(d, "VP8L", func(d *decode.D) {
// TODO
})
})
default:
d.Fatalf("could not find VP8 or VP8L chunk")
}
})
return nil
}