mirror of
https://github.com/wader/fq.git
synced 2024-11-23 00:57:15 +03:00
135 lines
3.2 KiB
Go
135 lines
3.2 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/format/registry"
|
|
"github.com/wader/fq/pkg/decode"
|
|
)
|
|
|
|
// TODO: vpx frame?
|
|
|
|
const (
|
|
vp9FeatureProfile = 1
|
|
vp9FeatureLevel = 2
|
|
vp9FeatureBitDepth = 3
|
|
vp9FeatureChromaSubsampling = 4
|
|
)
|
|
|
|
var vp9FeatureIDNames = decode.UToStr{
|
|
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 = decode.UToStr{
|
|
CS_UNKNOWN: "CS_UNKNOWN",
|
|
CS_BT_601: "CS_BT_601",
|
|
CS_BT_709: "CS_BT_709",
|
|
CS_SMPTE_170: "CS_SMPTE_170",
|
|
CS_SMPTE_240: "CS_SMPTE_240",
|
|
CS_BT_2020: "CS_BT_2020",
|
|
CS_RESERVED: "CS_RESERVED",
|
|
CS_RGB: "CS_RGB",
|
|
}
|
|
|
|
func init() {
|
|
registry.MustRegister(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", d.MapUToStrSym(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_zero")
|
|
} 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_zero")
|
|
}
|
|
}
|
|
}
|
|
|
|
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 interface{}) interface{} {
|
|
|
|
// 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))
|
|
if profile == 3 {
|
|
d.FieldU1("reserved_zero")
|
|
}
|
|
showExistingFrame := d.FieldBool("show_existing_frame")
|
|
if showExistingFrame {
|
|
d.FieldU2("frame_to_show_map_idx")
|
|
return nil
|
|
}
|
|
|
|
frameType := d.FieldBool("frame_type", d.MapBoolToStrSym(decode.BoolToStr{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
|
|
}
|