mirror of
https://github.com/wader/fq.git
synced 2024-11-23 00:57:15 +03:00
1ddea1ada3
Move registry to interp and add support for functions and filesystems. This will be used later for allow formats to add own functions and fq code. Add gojqextra function helpers to have more comfortable API to add functions. Takes care of argument type casting and JQValue:s and some more things. Refactor interp package to use new function helper and registry. Probably fixes a bunch of JQValue bugs and other type errors. Refactor out some mpeg nal things to mpeg format. Refactor interp jq code into display.q and init.jq. Remove undocumented aes_ctr funciton, was a test. Hopefully will add more crypto things laster.
143 lines
3.5 KiB
Go
143 lines
3.5 KiB
Go
package vpx
|
||
|
||
// https://storage.googleapis.com/downloads.webmproject.org/docs/vp9/vp9-bitstream-specification-v0.6-20160331-draft.pdf
|
||
|
||
import (
|
||
"github.com/wader/fq/format"
|
||
"github.com/wader/fq/pkg/decode"
|
||
"github.com/wader/fq/pkg/interp"
|
||
"github.com/wader/fq/pkg/scalar"
|
||
)
|
||
|
||
// TODO: vpx frame?
|
||
|
||
const (
|
||
vp9FeatureProfile = 1
|
||
vp9FeatureLevel = 2
|
||
vp9FeatureBitDepth = 3
|
||
vp9FeatureChromaSubsampling = 4
|
||
)
|
||
|
||
var vp9FeatureIDNames = scalar.UToSymStr{
|
||
vp9FeatureProfile: "Profile",
|
||
vp9FeatureLevel: "Level",
|
||
vp9FeatureBitDepth: "Bit Depth",
|
||
vp9FeatureChromaSubsampling: "Chroma Subsampling",
|
||
}
|
||
|
||
//nolint:revive
|
||
const (
|
||
CS_UNKNOWN = 0
|
||
CS_BT_601 = 1
|
||
CS_BT_709 = 2
|
||
CS_SMPTE_170 = 3
|
||
CS_SMPTE_240 = 4
|
||
CS_BT_2020 = 5
|
||
CS_RESERVED = 6
|
||
CS_RGB = 7
|
||
)
|
||
|
||
var vp9ColorSpaceNames = scalar.UToSymStr{
|
||
CS_UNKNOWN: "unknown",
|
||
CS_BT_601: "bt_601",
|
||
CS_BT_709: "bt_709",
|
||
CS_SMPTE_170: "smpte_170",
|
||
CS_SMPTE_240: "smpte_240",
|
||
CS_BT_2020: "bt_2020",
|
||
CS_RESERVED: "reserved",
|
||
CS_RGB: "rgb",
|
||
}
|
||
|
||
var vp9ProfilesMap = scalar.UToDescription{
|
||
0: "8 bit/sample, chroma subsampling: 4:2:0",
|
||
1: "8 bit, chroma subsampling: 4:2:2, 4:4:0, 4:4:4",
|
||
2: "10–12 bit, chroma subsampling: 4:2:0",
|
||
3: "10–12 bit, chroma subsampling: 4:2:2, 4:4:0, 4:4:4",
|
||
}
|
||
|
||
func init() {
|
||
interp.RegisterFormat(decode.Format{
|
||
Name: format.VP9_FRAME,
|
||
Description: "VP9 frame",
|
||
DecodeFn: vp9Decode,
|
||
})
|
||
}
|
||
|
||
func vp9DecodeFrameSyncCode(d *decode.D) {
|
||
d.FieldU8("frame_sync_byte_0")
|
||
d.FieldU8("frame_sync_byte_1")
|
||
d.FieldU8("frame_sync_byte_2")
|
||
}
|
||
|
||
func vp9DecodeColorConfig(d *decode.D, profile int) {
|
||
bitDepth := 8
|
||
if profile >= 2 {
|
||
tenOrTwelveBit := d.FieldBool("ten_or_twelve_bit")
|
||
if tenOrTwelveBit {
|
||
bitDepth = 12
|
||
} else {
|
||
bitDepth = 10
|
||
}
|
||
}
|
||
d.FieldValueU("bit_depth", uint64(bitDepth))
|
||
colorSpace := d.FieldU3("color_space", vp9ColorSpaceNames)
|
||
_, colorSpaceOk := vp9ColorSpaceNames[colorSpace]
|
||
if !colorSpaceOk || colorSpace != CS_RGB {
|
||
d.FieldU1("color_range")
|
||
if profile == 1 || profile == 3 {
|
||
d.FieldU1("subsampling_x")
|
||
d.FieldU1("subsampling_y")
|
||
d.FieldU1("reserved_zero1")
|
||
} else {
|
||
d.FieldValueU("subsampling_x", 1)
|
||
d.FieldValueU("subsampling_y", 1)
|
||
}
|
||
} else {
|
||
d.FieldValueU("color_range", 1)
|
||
if profile == 1 || profile == 3 {
|
||
d.FieldValueU("subsampling_x", 0)
|
||
d.FieldValueU("subsampling_y", 0)
|
||
d.FieldU1("reserved_zero2")
|
||
}
|
||
}
|
||
}
|
||
|
||
func vp9DecodeFrameSize(d *decode.D) {
|
||
d.FieldUFn("frame_width", func(d *decode.D) uint64 { return d.U16() + 1 })
|
||
d.FieldUFn("frame_height", func(d *decode.D) uint64 { return d.U16() + 1 })
|
||
}
|
||
|
||
func vp9Decode(d *decode.D, in any) any {
|
||
|
||
// TODO: header_size at end? even for show_existing_frame?
|
||
|
||
d.FieldU2("frame_marker")
|
||
profileLowBit := d.FieldU1("profile_low_bit")
|
||
profileHighBit := d.FieldU1("profile_high_bit")
|
||
profile := int(profileHighBit<<1 + profileLowBit)
|
||
d.FieldValueU("profile", uint64(profile), vp9ProfilesMap)
|
||
if profile == 3 {
|
||
d.FieldU1("reserved_zero0")
|
||
}
|
||
showExistingFrame := d.FieldBool("show_existing_frame")
|
||
if showExistingFrame {
|
||
d.FieldU2("frame_to_show_map_idx")
|
||
return nil
|
||
}
|
||
|
||
frameType := d.FieldBool("frame_type", scalar.BoolToSymStr{true: "non_key_frame", false: "key_frame"})
|
||
d.FieldU1("show_frame")
|
||
d.FieldU1("error_resilient_mode")
|
||
|
||
if !frameType {
|
||
// is key frame
|
||
vp9DecodeFrameSyncCode(d)
|
||
vp9DecodeColorConfig(d, profile)
|
||
vp9DecodeFrameSize(d)
|
||
}
|
||
|
||
d.FieldRawLen("data", d.BitsLeft())
|
||
|
||
return nil
|
||
}
|