1
1
mirror of https://github.com/wader/fq.git synced 2024-12-25 06:12:30 +03:00
fq/format/vpx/vp9_frame.go
Mattias Wadman 2fc0a71a47 decode: Refactor scalar usage
Move scalar into own package.
Split scalar code into decode related scalar code (that reads etc) and
scalar code that just transform the scalar value.
Use a scalar.Mapper interface instead of just a function.
Make mappers, assert and validat impement the interface.
2021-12-02 17:39:26 +01:00

143 lines
3.6 KiB
Go
Raw Blame History

This file contains ambiguous Unicode characters

This file contains Unicode characters that might be confused with other characters. If you think that this is intentional, you can safely ignore this warning. Use the Escape button to reveal them.

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"
"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: "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",
}
var vp9ProfilesMap = scalar.UToScalar{
0: {Description: "8 bit/sample, chroma subsampling: 4:2:0"},
1: {Description: "8 bit, chroma subsampling: 4:2:2, 4:4:0, 4:4:4"},
2: {Description: "1012 bit, chroma subsampling: 4:2:0"},
3: {Description: "1012 bit, chroma subsampling: 4:2:2, 4:4:0, 4:4:4"},
}
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", 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 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), 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
}