1
1
mirror of https://github.com/wader/fq.git synced 2024-11-25 23:13:19 +03:00
fq/format/mpeg/hevc_dcr.go

67 lines
2.0 KiB
Go

package mpeg
import (
"github.com/wader/fq/format"
"github.com/wader/fq/format/registry"
"github.com/wader/fq/pkg/decode"
)
var hevcDCRNALFormat []*decode.Format
func init() {
registry.MustRegister(&decode.Format{
Name: format.HEVC_DCR,
Description: "H.265/HEVC Decoder Configuration Record",
DecodeFn: hevcDcrDecode,
Dependencies: []decode.Dependency{
{Names: []string{format.HEVC_NALU}, Formats: &hevcDCRNALFormat},
},
})
}
func hevcDcrDecode(d *decode.D, in interface{}) interface{} {
d.FieldU8("configuration_version")
d.FieldU2("general_profile_space")
d.FieldU1("general_tier_flag")
d.FieldU5("general_profile_idc")
d.FieldU32("general_profile_compatibility_flags")
d.FieldU48("general_constraint_indicator_flags")
d.FieldU8("general_level_idc")
d.FieldU4("reserved0")
d.FieldU12("min_spatial_segmentation_idc")
d.FieldU6("reserved1")
d.FieldU2("parallelism_type")
d.FieldU6("reserved2")
d.FieldU2("chroma_format_idc")
d.FieldU5("reserved3")
d.FieldU3("bit_depth_luma_minus8")
d.FieldU5("reserved4")
d.FieldU3("bit_depth_chroma_minus8")
d.FieldU16("avg_frame_rate")
d.FieldU2("constant_frame_rate")
d.FieldU3("num_temporal_layers")
d.FieldU1("temporal_id_nested")
lengthSizeMinusOne := d.FieldU2("length_size_minus_one")
numArrays := d.FieldU8("num_of_arrays")
d.FieldArrayFn("arrays", func(d *decode.D) {
for i := uint64(0); i < numArrays; i++ {
d.FieldStructFn("array", func(d *decode.D) {
d.FieldU1("array_completeness")
d.FieldU1("reserved0")
d.FieldStringMapFn("nal_unit_type", hevcNALNames, "Unknown", d.U6, decode.NumberDecimal)
numNals := d.FieldU16("num_nalus")
d.FieldArrayFn("nals", func(d *decode.D) {
for i := uint64(0); i < numNals; i++ {
d.FieldStructFn("nal", func(d *decode.D) {
nalUnitLength := int64(d.FieldU16("nal_unit_length"))
d.FieldFormatLen("nal", nalUnitLength*8, hevcDCRNALFormat, nil)
})
}
})
})
}
})
return format.HevcDcrOut{LengthSize: lengthSizeMinusOne + 1}
}