mirror of
https://github.com/wader/fq.git
synced 2024-11-24 11:16:09 +03:00
74 lines
1.6 KiB
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
|
|
}
|