mirror of https://bitbucket.org/ausocean/av.git
codec/h264/h264dec: moved CABAC encoding related stuff to cabacenc.go and cabacenc_test.go and corrected function names
This commit is contained in:
Saxon
parent
7ab8fd9e87
commit
d4fe1498f5
|
|
@ -28,8 +28,6 @@ LICENSE
|
||||||
package h264dec
|
package h264dec
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"math"
|
|
||||||
|
|
||||||
"bitbucket.org/ausocean/av/codec/h264/h264dec/bits"
|
"bitbucket.org/ausocean/av/codec/h264/h264dec/bits"
|
||||||
"github.com/pkg/errors"
|
"github.com/pkg/errors"
|
||||||
)
|
)
|
||||||
|
|
@ -306,80 +304,6 @@ var (
|
||||||
}
|
}
|
||||||
)
|
)
|
||||||
|
|
||||||
// Errors used by mbTypeBinarization.
|
|
||||||
var (
|
|
||||||
errBadMbType = errors.New("macroblock type outside of valid range")
|
|
||||||
errBadMbSliceType = errors.New("bad slice type for macroblock")
|
|
||||||
)
|
|
||||||
|
|
||||||
// mbTypeBinarization returns the macroblock type binarization for the given
|
|
||||||
// macroblock type value and slice type using the process defined in section
|
|
||||||
// 9.3.2.5 of the specifications.
|
|
||||||
func mbTypeBinarization(v, slice int) ([]int, error) {
|
|
||||||
switch slice {
|
|
||||||
case sliceTypeI:
|
|
||||||
if v < minIMbType || v > maxIMbType {
|
|
||||||
return nil, errBadMbType
|
|
||||||
}
|
|
||||||
return binOfIMBTypes[v], nil
|
|
||||||
|
|
||||||
case sliceTypeSI:
|
|
||||||
if v < minSIMbType || v > maxSIMbType {
|
|
||||||
return nil, errBadMbType
|
|
||||||
}
|
|
||||||
if v == sliceTypeSI {
|
|
||||||
return []int{0}, nil
|
|
||||||
}
|
|
||||||
return append([]int{1}, binOfIMBTypes[v-1]...), nil
|
|
||||||
|
|
||||||
case sliceTypeP, sliceTypeSP:
|
|
||||||
if v < minPOrSPMbType || v > maxPOrSPMbType || v == P8x8ref0 {
|
|
||||||
return nil, errBadMbType
|
|
||||||
}
|
|
||||||
if v < 5 {
|
|
||||||
return binOfPOrSPMBTypes[v], nil
|
|
||||||
}
|
|
||||||
return append([]int{1}, binOfIMBTypes[v-5]...), nil
|
|
||||||
|
|
||||||
case sliceTypeB:
|
|
||||||
if v < minBMbType || v > maxBMbType {
|
|
||||||
return nil, errBadMbType
|
|
||||||
}
|
|
||||||
if v < 23 {
|
|
||||||
return binOfBMBTypes[v], nil
|
|
||||||
}
|
|
||||||
return append([]int{1, 1, 1, 1, 0, 1}, binOfIMBTypes[v-23]...), nil
|
|
||||||
|
|
||||||
default:
|
|
||||||
return nil, errBadMbSliceType
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
// Error used by subMbTypeBinarization.
|
|
||||||
var errBadSubMbSliceType = errors.New("bad slice type for sub-macroblock")
|
|
||||||
|
|
||||||
// subMbTypeBinarization returns the binarization of a sub-macroblock type
|
|
||||||
// given the slice in which it is in using the process defined in section
|
|
||||||
// 9.3.2.5 of the specifications.
|
|
||||||
func subMbTypeBinarization(v, slice int) ([]int, error) {
|
|
||||||
switch slice {
|
|
||||||
case sliceTypeP, sliceTypeSP:
|
|
||||||
if v < minPOrSPSubMbType || v > maxPOrSPSubMbType {
|
|
||||||
return nil, errBadMbType
|
|
||||||
}
|
|
||||||
return binOfPOrSPSubMBTypes[v], nil
|
|
||||||
|
|
||||||
case sliceTypeB:
|
|
||||||
if v < minBSubMbType || v > maxBSubMbType {
|
|
||||||
return nil, errBadMbType
|
|
||||||
}
|
|
||||||
return binOfBSubMBTypes[v], nil
|
|
||||||
|
|
||||||
default:
|
|
||||||
return nil, errBadSubMbSliceType
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
// Table 9-34
|
// Table 9-34
|
||||||
type MaxBinIdxCtx struct {
|
type MaxBinIdxCtx struct {
|
||||||
// When false, Prefix is the MaxBinIdxCtx
|
// When false, Prefix is the MaxBinIdxCtx
|
||||||
|
|
@ -762,119 +686,3 @@ func CtxIdx(binIdx, maxBinIdxCtx, ctxIdxOffset int) int {
|
||||||
|
|
||||||
return ctxIdx
|
return ctxIdx
|
||||||
}
|
}
|
||||||
|
|
||||||
// Error used by unaryBinarization.
|
|
||||||
var errNegativeSyntaxVal = errors.New("cannot get unary binarization of negative value")
|
|
||||||
|
|
||||||
// unaryBinarization returns the unary binarization of a syntax element having
|
|
||||||
// value v, as specified by setion 9.3.2.1 in the specifications.
|
|
||||||
func unaryBinarization(v int) ([]int, error) {
|
|
||||||
if v < 0 {
|
|
||||||
return nil, errNegativeSyntaxVal
|
|
||||||
}
|
|
||||||
r := make([]int, v+1)
|
|
||||||
for i := 0; i <= v; i++ {
|
|
||||||
if i < v {
|
|
||||||
r[i] = 1
|
|
||||||
}
|
|
||||||
}
|
|
||||||
return r, nil
|
|
||||||
}
|
|
||||||
|
|
||||||
// Error used by truncUnaryBinarization.
|
|
||||||
var errInvalidSyntaxVal = errors.New("syntax value cannot be greater than cMax")
|
|
||||||
|
|
||||||
// truncUnaryBinarization returns the truncated unary binarization of a syntax
|
|
||||||
// element v given a cMax as specified in section 9.3.2.2 of the specifications.
|
|
||||||
func truncUnaryBinarization(v, cMax int) ([]int, error) {
|
|
||||||
if v < 0 {
|
|
||||||
return nil, errNegativeSyntaxVal
|
|
||||||
}
|
|
||||||
|
|
||||||
if v > cMax {
|
|
||||||
return nil, errInvalidSyntaxVal
|
|
||||||
}
|
|
||||||
|
|
||||||
if v == cMax {
|
|
||||||
b, _ := unaryBinarization(v)
|
|
||||||
return b[:len(b)-1], nil
|
|
||||||
}
|
|
||||||
return unaryBinarization(v)
|
|
||||||
}
|
|
||||||
|
|
||||||
// Error used by unaryExpGolombBinarization.
|
|
||||||
var errInvalidUCoff = errors.New("uCoff cannot be less than or equal to zero")
|
|
||||||
|
|
||||||
// unaryExpGolombBinarization returns the concatendated unary/k-th order
|
|
||||||
// Exp-Golomb (UEGk) binarization of a syntax element using the process defined
|
|
||||||
// in section 9.3.2.3 of the specifications.
|
|
||||||
func unaryExpGolombBinarization(v, uCoff, k int, signedValFlag bool) ([]int, error) {
|
|
||||||
if uCoff <= 0 {
|
|
||||||
return nil, errInvalidUCoff
|
|
||||||
}
|
|
||||||
|
|
||||||
prefix, err := truncUnaryBinarization(mini(uCoff, absi(v)), uCoff)
|
|
||||||
if err != nil {
|
|
||||||
return nil, err
|
|
||||||
}
|
|
||||||
|
|
||||||
return append(prefix, suffix(v, uCoff, k, signedValFlag)...), nil
|
|
||||||
}
|
|
||||||
|
|
||||||
// suffix returns the suffix part of a unary k-th Exp-Golomb Binarization
|
|
||||||
// using the the algorithm as described by pseudo code 9-6 in section 9.3.2.3.
|
|
||||||
// TODO: could probably reduce allocations.
|
|
||||||
func suffix(v, uCoff, k int, signedValFlag bool) []int {
|
|
||||||
var s []int
|
|
||||||
|
|
||||||
if absi(v) >= uCoff {
|
|
||||||
sufS := absi(v) - uCoff
|
|
||||||
var stop bool
|
|
||||||
|
|
||||||
for {
|
|
||||||
if sufS >= (1 << uint(k)) {
|
|
||||||
s = append(s, 1)
|
|
||||||
sufS = sufS - (1 << uint(k))
|
|
||||||
k++
|
|
||||||
} else {
|
|
||||||
s = append(s, 0)
|
|
||||||
for k = k - 1; k >= 0; k-- {
|
|
||||||
s = append(s, (sufS>>uint(k))&1)
|
|
||||||
}
|
|
||||||
stop = true
|
|
||||||
}
|
|
||||||
if stop {
|
|
||||||
break
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
if signedValFlag && v != 0 {
|
|
||||||
if v > 0 {
|
|
||||||
s = append(s, 0)
|
|
||||||
} else {
|
|
||||||
s = append(s, 1)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
return s
|
|
||||||
}
|
|
||||||
|
|
||||||
// Error used by fixedLenBinariztion.
|
|
||||||
var errNegativeValue = errors.New("cannot get fixed length binarization of negative value")
|
|
||||||
|
|
||||||
// fixedLenBinarization returns the fixed-length (FL) binarization of the syntax
|
|
||||||
// element v, given cMax to determine bin length, as specified by section 9.3.2.4
|
|
||||||
// of the specifications.
|
|
||||||
func fixedLenBinarization(v, cMax int) ([]int, error) {
|
|
||||||
if v < 0 {
|
|
||||||
return nil, errNegativeValue
|
|
||||||
}
|
|
||||||
l := int(math.Ceil(math.Log2(float64(cMax + 1))))
|
|
||||||
r := make([]int, l)
|
|
||||||
for i := l - 1; i >= 0; i-- {
|
|
||||||
r[i] = v % 2
|
|
||||||
v = v / 2
|
|
||||||
}
|
|
||||||
return r, nil
|
|
||||||
}
|
|
||||||
|
|
|
||||||
|
|
@ -26,74 +26,9 @@ LICENSE
|
||||||
package h264dec
|
package h264dec
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"reflect"
|
|
||||||
"testing"
|
"testing"
|
||||||
)
|
)
|
||||||
|
|
||||||
func TestMbTypeBinarization(t *testing.T) {
|
|
||||||
tests := []struct {
|
|
||||||
v, slice int
|
|
||||||
want []int
|
|
||||||
err error
|
|
||||||
}{
|
|
||||||
{v: 6, slice: sliceTypeI, want: []int{1, 0, 0, 1, 0, 0, 1}},
|
|
||||||
{v: 26, slice: sliceTypeI, err: errBadMbType},
|
|
||||||
{v: -1, slice: sliceTypeI, err: errBadMbType},
|
|
||||||
{v: 4, slice: sliceTypeSI, want: []int{0}},
|
|
||||||
{v: 6, slice: sliceTypeSI, want: []int{1, 1, 0, 0, 1, 0, 0, 0}},
|
|
||||||
{v: 0, slice: sliceTypeSI, err: errBadMbType},
|
|
||||||
{v: 27, slice: sliceTypeSI, err: errBadMbType},
|
|
||||||
{v: 2, slice: sliceTypeP, want: []int{0, 1, 0}},
|
|
||||||
{v: 3, slice: sliceTypeSP, want: []int{0, 0, 1}},
|
|
||||||
{v: 7, slice: sliceTypeP, want: []int{1, 1, 0, 0, 0, 0, 1}},
|
|
||||||
{v: 7, slice: sliceTypeSP, want: []int{1, 1, 0, 0, 0, 0, 1}},
|
|
||||||
{v: -1, slice: sliceTypeP, err: errBadMbType},
|
|
||||||
{v: 31, slice: sliceTypeP, err: errBadMbType},
|
|
||||||
{v: 8, slice: sliceTypeB, want: []int{1, 1, 0, 1, 0, 1}},
|
|
||||||
{v: 30, slice: sliceTypeB, want: []int{1, 1, 1, 1, 0, 1, 1, 0, 0, 1, 0, 1, 0}},
|
|
||||||
{v: -1, slice: sliceTypeB, err: errBadMbType},
|
|
||||||
{v: 49, slice: sliceTypeB, err: errBadMbType},
|
|
||||||
{v: 6, slice: 20, err: errBadMbSliceType},
|
|
||||||
}
|
|
||||||
|
|
||||||
for i, test := range tests {
|
|
||||||
got, err := mbTypeBinarization(test.v, test.slice)
|
|
||||||
if err != test.err {
|
|
||||||
t.Errorf("did not get expected error for test %d\nGot: %v\nWant: %v", i, err, test.err)
|
|
||||||
}
|
|
||||||
|
|
||||||
if !reflect.DeepEqual(got, test.want) {
|
|
||||||
t.Errorf("did not get expected result for test %d\nGot: %v\nWant: %v", i, got, test.want)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestSubMbTypeBinarization(t *testing.T) {
|
|
||||||
tests := []struct {
|
|
||||||
v, slice int
|
|
||||||
want []int
|
|
||||||
err error
|
|
||||||
}{
|
|
||||||
{v: 2, slice: sliceTypeP, want: []int{0, 1, 1}},
|
|
||||||
{v: 2, slice: sliceTypeSP, want: []int{0, 1, 1}},
|
|
||||||
{v: -1, slice: sliceTypeSP, err: errBadMbType},
|
|
||||||
{v: 4, slice: sliceTypeSP, err: errBadMbType},
|
|
||||||
{v: 9, slice: sliceTypeB, want: []int{1, 1, 1, 0, 1, 0}},
|
|
||||||
{v: 9, slice: 40, err: errBadSubMbSliceType},
|
|
||||||
}
|
|
||||||
|
|
||||||
for i, test := range tests {
|
|
||||||
got, err := subMbTypeBinarization(test.v, test.slice)
|
|
||||||
if err != test.err {
|
|
||||||
t.Errorf("did not get expected error for test %d\nGot: %v\nWant: %v", i, err, test.err)
|
|
||||||
}
|
|
||||||
|
|
||||||
if !reflect.DeepEqual(got, test.want) {
|
|
||||||
t.Errorf("did not get expected result for test %d\nGot: %v\nWant: %v", i, got, test.want)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
var ctxIdxTests = []struct {
|
var ctxIdxTests = []struct {
|
||||||
binIdx int
|
binIdx int
|
||||||
maxBinIdxCtx int
|
maxBinIdxCtx int
|
||||||
|
|
@ -211,144 +146,3 @@ func TestCtxIdx(t *testing.T) {
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
func TestUnaryBinarization(t *testing.T) {
|
|
||||||
// Test data has been extracted from table 9-35 of the specifications.
|
|
||||||
tests := []struct {
|
|
||||||
in int
|
|
||||||
want []int
|
|
||||||
err error
|
|
||||||
}{
|
|
||||||
{in: 0, want: []int{0}, err: nil},
|
|
||||||
{in: 1, want: []int{1, 0}, err: nil},
|
|
||||||
{in: 2, want: []int{1, 1, 0}, err: nil},
|
|
||||||
{in: 3, want: []int{1, 1, 1, 0}, err: nil},
|
|
||||||
{in: 4, want: []int{1, 1, 1, 1, 0}, err: nil},
|
|
||||||
{in: 5, want: []int{1, 1, 1, 1, 1, 0}, err: nil},
|
|
||||||
{in: -3, want: nil, err: errNegativeSyntaxVal},
|
|
||||||
}
|
|
||||||
|
|
||||||
for i, test := range tests {
|
|
||||||
got, err := unaryBinarization(test.in)
|
|
||||||
if err != test.err {
|
|
||||||
t.Errorf("did not get expected error for test %d\nGot: %v\nWant: %v", i, err, test.err)
|
|
||||||
}
|
|
||||||
|
|
||||||
if !reflect.DeepEqual(test.want, got) {
|
|
||||||
t.Errorf("did not get expected result for test %d\nGot: %v\nWant: %v\n", i, got, test.want)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestFixedLengthBinarization(t *testing.T) {
|
|
||||||
tests := []struct {
|
|
||||||
v int
|
|
||||||
cMax int
|
|
||||||
want []int
|
|
||||||
err error
|
|
||||||
}{
|
|
||||||
{v: 0, cMax: 7, want: []int{0, 0, 0}},
|
|
||||||
{v: 1, cMax: 7, want: []int{0, 0, 1}},
|
|
||||||
{v: 2, cMax: 7, want: []int{0, 1, 0}},
|
|
||||||
{v: 3, cMax: 7, want: []int{0, 1, 1}},
|
|
||||||
{v: 4, cMax: 7, want: []int{1, 0, 0}},
|
|
||||||
{v: 5, cMax: 7, want: []int{1, 0, 1}},
|
|
||||||
{v: 6, cMax: 7, want: []int{1, 1, 0}},
|
|
||||||
{v: 7, cMax: 7, want: []int{1, 1, 1}},
|
|
||||||
{v: -1, cMax: 7, want: nil, err: errNegativeValue},
|
|
||||||
}
|
|
||||||
|
|
||||||
for i, test := range tests {
|
|
||||||
got, err := fixedLenBinarization(test.v, test.cMax)
|
|
||||||
if err != test.err {
|
|
||||||
t.Errorf("did not get expected error for test %d\nGot: %v\nWant: %v\n", i, err, test.err)
|
|
||||||
}
|
|
||||||
|
|
||||||
if !reflect.DeepEqual(test.want, got) {
|
|
||||||
t.Errorf("did not get expected result for test %d\nGot: %v\nWant: %v\n", i, got, test.want)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestTruncUnaryBinarization(t *testing.T) {
|
|
||||||
tests := []struct {
|
|
||||||
v int
|
|
||||||
cMax int
|
|
||||||
want []int
|
|
||||||
err error
|
|
||||||
}{
|
|
||||||
{v: 0, cMax: 10, want: []int{0}, err: nil},
|
|
||||||
{v: 1, cMax: 10, want: []int{1, 0}, err: nil},
|
|
||||||
{v: 2, cMax: 10, want: []int{1, 1, 0}, err: nil},
|
|
||||||
{v: 0, cMax: 0, want: []int{}, err: nil},
|
|
||||||
{v: 4, cMax: 4, want: []int{1, 1, 1, 1}, err: nil},
|
|
||||||
{v: 1, cMax: 10, want: []int{1, 0}, err: nil},
|
|
||||||
{v: 2, cMax: 10, want: []int{1, 1, 0}, err: nil},
|
|
||||||
{v: -3, cMax: 10, want: nil, err: errNegativeSyntaxVal},
|
|
||||||
{v: 5, cMax: 4, want: nil, err: errInvalidSyntaxVal},
|
|
||||||
}
|
|
||||||
|
|
||||||
for i, test := range tests {
|
|
||||||
got, err := truncUnaryBinarization(test.v, test.cMax)
|
|
||||||
if err != test.err {
|
|
||||||
t.Errorf("did not get expected error for test %d\nGot: %v\nWant: %v", i, err, test.err)
|
|
||||||
}
|
|
||||||
|
|
||||||
if !reflect.DeepEqual(test.want, got) {
|
|
||||||
t.Errorf("did not get expected result for test %d\nGot: %v\nWant: %v\n", i, got, test.want)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestUEGkSuffix(t *testing.T) {
|
|
||||||
// Data from https://patents.google.com/patent/US20070092150
|
|
||||||
tests := []struct {
|
|
||||||
v, uCoff, k int
|
|
||||||
signedValFlag bool
|
|
||||||
want []int
|
|
||||||
}{
|
|
||||||
0: {v: 14, uCoff: 14, want: []int{0}},
|
|
||||||
1: {v: 15, uCoff: 14, want: []int{1, 0, 0}},
|
|
||||||
2: {v: 16, uCoff: 14, want: []int{1, 0, 1}},
|
|
||||||
3: {v: 17, uCoff: 14, want: []int{1, 1, 0, 0, 0}},
|
|
||||||
4: {v: 18, uCoff: 14, want: []int{1, 1, 0, 0, 1}},
|
|
||||||
5: {v: 19, uCoff: 14, want: []int{1, 1, 0, 1, 0}},
|
|
||||||
6: {v: 20, uCoff: 14, want: []int{1, 1, 0, 1, 1}},
|
|
||||||
7: {v: 21, uCoff: 14, want: []int{1, 1, 1, 0, 0, 0, 0}},
|
|
||||||
8: {v: 22, uCoff: 14, want: []int{1, 1, 1, 0, 0, 0, 1}},
|
|
||||||
9: {v: 23, uCoff: 14, want: []int{1, 1, 1, 0, 0, 1, 0}},
|
|
||||||
10: {v: 24, uCoff: 14, want: []int{1, 1, 1, 0, 0, 1, 1}},
|
|
||||||
11: {v: 25, uCoff: 14, want: []int{1, 1, 1, 0, 1, 0, 0}},
|
|
||||||
}
|
|
||||||
|
|
||||||
for i, test := range tests {
|
|
||||||
got := suffix(test.v, test.uCoff, test.k, test.signedValFlag)
|
|
||||||
if !reflect.DeepEqual(got, test.want) {
|
|
||||||
t.Errorf("did not get expected result for test %d\nGot: %v\nWant: %v\n", i, got, test.want)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestUnaryExpGolombBinarization(t *testing.T) {
|
|
||||||
tests := []struct {
|
|
||||||
v, uCoff, k int
|
|
||||||
signedValFlag bool
|
|
||||||
want []int
|
|
||||||
}{
|
|
||||||
0: {v: 7, uCoff: 14, want: []int{1, 1, 1, 1, 1, 1, 1, 0}},
|
|
||||||
1: {v: 17, uCoff: 14, want: []int{1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 0, 0, 0}},
|
|
||||||
2: {v: 15, uCoff: 14, signedValFlag: true, want: []int{1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 0, 0, 0}},
|
|
||||||
3: {v: -15, uCoff: 14, signedValFlag: true, want: []int{1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 0, 0, 1}},
|
|
||||||
}
|
|
||||||
|
|
||||||
for i, test := range tests {
|
|
||||||
got, err := unaryExpGolombBinarization(test.v, test.uCoff, test.k, test.signedValFlag)
|
|
||||||
if err != nil {
|
|
||||||
t.Errorf("did not expect error %v for test %d", err, i)
|
|
||||||
}
|
|
||||||
|
|
||||||
if !reflect.DeepEqual(got, test.want) {
|
|
||||||
t.Errorf("did not get expected result for test %d\nGot: %v\nWant: %v\n", i, got, test.want)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
|
||||||
|
|
@ -0,0 +1,222 @@
|
||||||
|
/*
|
||||||
|
TODO: this file should really be in a 'h264enc' package.
|
||||||
|
|
||||||
|
DESCRIPTION
|
||||||
|
cabacenc.go provides functionality for CABAC encoding.
|
||||||
|
|
||||||
|
AUTHORS
|
||||||
|
Saxon A. Nelson-Milton <saxon@ausocean.org>
|
||||||
|
|
||||||
|
LICENSE
|
||||||
|
Copyright (C) 2019 the Australian Ocean Lab (AusOcean).
|
||||||
|
|
||||||
|
It is free software: you can redistribute it and/or modify them
|
||||||
|
under the terms of the GNU General Public License as published by the
|
||||||
|
Free Software Foundation, either version 3 of the License, or (at your
|
||||||
|
option) any later version.
|
||||||
|
|
||||||
|
It is distributed in the hope that it will be useful, but WITHOUT
|
||||||
|
ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
|
||||||
|
FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License
|
||||||
|
for more details.
|
||||||
|
|
||||||
|
You should have received a copy of the GNU General Public License
|
||||||
|
in gpl.txt. If not, see http://www.gnu.org/licenses.
|
||||||
|
*/
|
||||||
|
|
||||||
|
package h264dec
|
||||||
|
|
||||||
|
import (
|
||||||
|
"errors"
|
||||||
|
"math"
|
||||||
|
)
|
||||||
|
|
||||||
|
// Error used by unaryBinString.
|
||||||
|
var errNegativeSyntaxVal = errors.New("cannot get unary binary string of negative value")
|
||||||
|
|
||||||
|
// unaryBinString returns the unary binary string of a syntax element having
|
||||||
|
// value v, as specified by setion 9.3.2.1 in the specifications.
|
||||||
|
func unaryBinString(v int) ([]int, error) {
|
||||||
|
if v < 0 {
|
||||||
|
return nil, errNegativeSyntaxVal
|
||||||
|
}
|
||||||
|
r := make([]int, v+1)
|
||||||
|
for i := 0; i <= v; i++ {
|
||||||
|
if i < v {
|
||||||
|
r[i] = 1
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return r, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// Error used by truncUnaryBinString.
|
||||||
|
var errInvalidSyntaxVal = errors.New("syntax value cannot be greater than cMax")
|
||||||
|
|
||||||
|
// truncUnaryBinString returns the truncated unary binary string of a syntax
|
||||||
|
// element v given a cMax as specified in section 9.3.2.2 of the specifications.
|
||||||
|
func truncUnaryBinString(v, cMax int) ([]int, error) {
|
||||||
|
if v < 0 {
|
||||||
|
return nil, errNegativeSyntaxVal
|
||||||
|
}
|
||||||
|
|
||||||
|
if v > cMax {
|
||||||
|
return nil, errInvalidSyntaxVal
|
||||||
|
}
|
||||||
|
|
||||||
|
if v == cMax {
|
||||||
|
b, _ := unaryBinString(v)
|
||||||
|
return b[:len(b)-1], nil
|
||||||
|
}
|
||||||
|
return unaryBinString(v)
|
||||||
|
}
|
||||||
|
|
||||||
|
// Error used by unaryExpGolombBinString.
|
||||||
|
var errInvalidUCoff = errors.New("uCoff cannot be less than or equal to zero")
|
||||||
|
|
||||||
|
// unaryExpGolombBinString returns the concatendated unary/k-th order
|
||||||
|
// Exp-Golomb (UEGk) binary string of a syntax element using the process defined
|
||||||
|
// in section 9.3.2.3 of the specifications.
|
||||||
|
func unaryExpGolombBinString(v, uCoff, k int, signedValFlag bool) ([]int, error) {
|
||||||
|
if uCoff <= 0 {
|
||||||
|
return nil, errInvalidUCoff
|
||||||
|
}
|
||||||
|
|
||||||
|
prefix, err := truncUnaryBinString(mini(uCoff, absi(v)), uCoff)
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
|
||||||
|
return append(prefix, suffix(v, uCoff, k, signedValFlag)...), nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// suffix returns the suffix part of a unary k-th Exp-Golomb binar string
|
||||||
|
// using the the algorithm as described by pseudo code 9-6 in section 9.3.2.3.
|
||||||
|
// TODO: could probably reduce allocations.
|
||||||
|
func suffix(v, uCoff, k int, signedValFlag bool) []int {
|
||||||
|
var s []int
|
||||||
|
|
||||||
|
if absi(v) >= uCoff {
|
||||||
|
sufS := absi(v) - uCoff
|
||||||
|
var stop bool
|
||||||
|
|
||||||
|
for {
|
||||||
|
if sufS >= (1 << uint(k)) {
|
||||||
|
s = append(s, 1)
|
||||||
|
sufS = sufS - (1 << uint(k))
|
||||||
|
k++
|
||||||
|
} else {
|
||||||
|
s = append(s, 0)
|
||||||
|
for k = k - 1; k >= 0; k-- {
|
||||||
|
s = append(s, (sufS>>uint(k))&1)
|
||||||
|
}
|
||||||
|
stop = true
|
||||||
|
}
|
||||||
|
if stop {
|
||||||
|
break
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
if signedValFlag && v != 0 {
|
||||||
|
if v > 0 {
|
||||||
|
s = append(s, 0)
|
||||||
|
} else {
|
||||||
|
s = append(s, 1)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
return s
|
||||||
|
}
|
||||||
|
|
||||||
|
// Error used by fixedLenBinString.
|
||||||
|
var errNegativeValue = errors.New("cannot get fixed length binary string of negative value")
|
||||||
|
|
||||||
|
// fixedLenBinString returns the fixed-length (FL) binary string of the syntax
|
||||||
|
// element v, given cMax to determine bin length, as specified by section 9.3.2.4
|
||||||
|
// of the specifications.
|
||||||
|
func fixedLenBinString(v, cMax int) ([]int, error) {
|
||||||
|
if v < 0 {
|
||||||
|
return nil, errNegativeValue
|
||||||
|
}
|
||||||
|
l := int(math.Ceil(math.Log2(float64(cMax + 1))))
|
||||||
|
r := make([]int, l)
|
||||||
|
for i := l - 1; i >= 0; i-- {
|
||||||
|
r[i] = v % 2
|
||||||
|
v = v / 2
|
||||||
|
}
|
||||||
|
return r, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// Errors used by mbTypeBinString.
|
||||||
|
var (
|
||||||
|
errBadMbType = errors.New("macroblock type outside of valid range")
|
||||||
|
errBadMbSliceType = errors.New("bad slice type for macroblock")
|
||||||
|
)
|
||||||
|
|
||||||
|
// mbTypeBinString returns the macroblock type binary string for the given
|
||||||
|
// macroblock type value and slice type using the process defined in section
|
||||||
|
// 9.3.2.5 of the specifications.
|
||||||
|
func mbTypeBinString(v, slice int) ([]int, error) {
|
||||||
|
switch slice {
|
||||||
|
case sliceTypeI:
|
||||||
|
if v < minIMbType || v > maxIMbType {
|
||||||
|
return nil, errBadMbType
|
||||||
|
}
|
||||||
|
return binOfIMBTypes[v], nil
|
||||||
|
|
||||||
|
case sliceTypeSI:
|
||||||
|
if v < minSIMbType || v > maxSIMbType {
|
||||||
|
return nil, errBadMbType
|
||||||
|
}
|
||||||
|
if v == sliceTypeSI {
|
||||||
|
return []int{0}, nil
|
||||||
|
}
|
||||||
|
return append([]int{1}, binOfIMBTypes[v-1]...), nil
|
||||||
|
|
||||||
|
case sliceTypeP, sliceTypeSP:
|
||||||
|
if v < minPOrSPMbType || v > maxPOrSPMbType || v == P8x8ref0 {
|
||||||
|
return nil, errBadMbType
|
||||||
|
}
|
||||||
|
if v < 5 {
|
||||||
|
return binOfPOrSPMBTypes[v], nil
|
||||||
|
}
|
||||||
|
return append([]int{1}, binOfIMBTypes[v-5]...), nil
|
||||||
|
|
||||||
|
case sliceTypeB:
|
||||||
|
if v < minBMbType || v > maxBMbType {
|
||||||
|
return nil, errBadMbType
|
||||||
|
}
|
||||||
|
if v < 23 {
|
||||||
|
return binOfBMBTypes[v], nil
|
||||||
|
}
|
||||||
|
return append([]int{1, 1, 1, 1, 0, 1}, binOfIMBTypes[v-23]...), nil
|
||||||
|
|
||||||
|
default:
|
||||||
|
return nil, errBadMbSliceType
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// Error used by subMbTypeBinString.
|
||||||
|
var errBadSubMbSliceType = errors.New("bad slice type for sub-macroblock")
|
||||||
|
|
||||||
|
// subMbTypeBinString returns the binary string of a sub-macroblock type
|
||||||
|
// given the slice in which it is in using the process defined in section
|
||||||
|
// 9.3.2.5 of the specifications.
|
||||||
|
func subMbTypeBinString(v, slice int) ([]int, error) {
|
||||||
|
switch slice {
|
||||||
|
case sliceTypeP, sliceTypeSP:
|
||||||
|
if v < minPOrSPSubMbType || v > maxPOrSPSubMbType {
|
||||||
|
return nil, errBadMbType
|
||||||
|
}
|
||||||
|
return binOfPOrSPSubMBTypes[v], nil
|
||||||
|
|
||||||
|
case sliceTypeB:
|
||||||
|
if v < minBSubMbType || v > maxBSubMbType {
|
||||||
|
return nil, errBadMbType
|
||||||
|
}
|
||||||
|
return binOfBSubMBTypes[v], nil
|
||||||
|
|
||||||
|
default:
|
||||||
|
return nil, errBadSubMbSliceType
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
@ -0,0 +1,237 @@
|
||||||
|
/*
|
||||||
|
TODO: this file should really be in a 'h264enc' package.
|
||||||
|
|
||||||
|
DESCRIPTION
|
||||||
|
cabacenc_test.go provides testing for functionality found in cabacenc.go.
|
||||||
|
|
||||||
|
AUTHORS
|
||||||
|
Saxon A. Nelson-Milton <saxon@ausocean.org>
|
||||||
|
|
||||||
|
LICENSE
|
||||||
|
Copyright (C) 2019 the Australian Ocean Lab (AusOcean).
|
||||||
|
|
||||||
|
It is free software: you can redistribute it and/or modify them
|
||||||
|
under the terms of the GNU General Public License as published by the
|
||||||
|
Free Software Foundation, either version 3 of the License, or (at your
|
||||||
|
option) any later version.
|
||||||
|
|
||||||
|
It is distributed in the hope that it will be useful, but WITHOUT
|
||||||
|
ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
|
||||||
|
FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License
|
||||||
|
for more details.
|
||||||
|
|
||||||
|
You should have received a copy of the GNU General Public License
|
||||||
|
in gpl.txt. If not, see http://www.gnu.org/licenses.
|
||||||
|
*/
|
||||||
|
|
||||||
|
package h264dec
|
||||||
|
|
||||||
|
import (
|
||||||
|
"reflect"
|
||||||
|
"testing"
|
||||||
|
)
|
||||||
|
|
||||||
|
func TestMbTypeBinString(t *testing.T) {
|
||||||
|
tests := []struct {
|
||||||
|
v, slice int
|
||||||
|
want []int
|
||||||
|
err error
|
||||||
|
}{
|
||||||
|
{v: 6, slice: sliceTypeI, want: []int{1, 0, 0, 1, 0, 0, 1}},
|
||||||
|
{v: 26, slice: sliceTypeI, err: errBadMbType},
|
||||||
|
{v: -1, slice: sliceTypeI, err: errBadMbType},
|
||||||
|
{v: 4, slice: sliceTypeSI, want: []int{0}},
|
||||||
|
{v: 6, slice: sliceTypeSI, want: []int{1, 1, 0, 0, 1, 0, 0, 0}},
|
||||||
|
{v: 0, slice: sliceTypeSI, err: errBadMbType},
|
||||||
|
{v: 27, slice: sliceTypeSI, err: errBadMbType},
|
||||||
|
{v: 2, slice: sliceTypeP, want: []int{0, 1, 0}},
|
||||||
|
{v: 3, slice: sliceTypeSP, want: []int{0, 0, 1}},
|
||||||
|
{v: 7, slice: sliceTypeP, want: []int{1, 1, 0, 0, 0, 0, 1}},
|
||||||
|
{v: 7, slice: sliceTypeSP, want: []int{1, 1, 0, 0, 0, 0, 1}},
|
||||||
|
{v: -1, slice: sliceTypeP, err: errBadMbType},
|
||||||
|
{v: 31, slice: sliceTypeP, err: errBadMbType},
|
||||||
|
{v: 8, slice: sliceTypeB, want: []int{1, 1, 0, 1, 0, 1}},
|
||||||
|
{v: 30, slice: sliceTypeB, want: []int{1, 1, 1, 1, 0, 1, 1, 0, 0, 1, 0, 1, 0}},
|
||||||
|
{v: -1, slice: sliceTypeB, err: errBadMbType},
|
||||||
|
{v: 49, slice: sliceTypeB, err: errBadMbType},
|
||||||
|
{v: 6, slice: 20, err: errBadMbSliceType},
|
||||||
|
}
|
||||||
|
|
||||||
|
for i, test := range tests {
|
||||||
|
got, err := mbTypeBinString(test.v, test.slice)
|
||||||
|
if err != test.err {
|
||||||
|
t.Errorf("did not get expected error for test %d\nGot: %v\nWant: %v", i, err, test.err)
|
||||||
|
}
|
||||||
|
|
||||||
|
if !reflect.DeepEqual(got, test.want) {
|
||||||
|
t.Errorf("did not get expected result for test %d\nGot: %v\nWant: %v", i, got, test.want)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestSubMbTypeBinString(t *testing.T) {
|
||||||
|
tests := []struct {
|
||||||
|
v, slice int
|
||||||
|
want []int
|
||||||
|
err error
|
||||||
|
}{
|
||||||
|
{v: 2, slice: sliceTypeP, want: []int{0, 1, 1}},
|
||||||
|
{v: 2, slice: sliceTypeSP, want: []int{0, 1, 1}},
|
||||||
|
{v: -1, slice: sliceTypeSP, err: errBadMbType},
|
||||||
|
{v: 4, slice: sliceTypeSP, err: errBadMbType},
|
||||||
|
{v: 9, slice: sliceTypeB, want: []int{1, 1, 1, 0, 1, 0}},
|
||||||
|
{v: 9, slice: 40, err: errBadSubMbSliceType},
|
||||||
|
}
|
||||||
|
|
||||||
|
for i, test := range tests {
|
||||||
|
got, err := subMbTypeBinString(test.v, test.slice)
|
||||||
|
if err != test.err {
|
||||||
|
t.Errorf("did not get expected error for test %d\nGot: %v\nWant: %v", i, err, test.err)
|
||||||
|
}
|
||||||
|
|
||||||
|
if !reflect.DeepEqual(got, test.want) {
|
||||||
|
t.Errorf("did not get expected result for test %d\nGot: %v\nWant: %v", i, got, test.want)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestUnaryBinString(t *testing.T) {
|
||||||
|
// Test data has been extracted from table 9-35 of the specifications.
|
||||||
|
tests := []struct {
|
||||||
|
in int
|
||||||
|
want []int
|
||||||
|
err error
|
||||||
|
}{
|
||||||
|
{in: 0, want: []int{0}, err: nil},
|
||||||
|
{in: 1, want: []int{1, 0}, err: nil},
|
||||||
|
{in: 2, want: []int{1, 1, 0}, err: nil},
|
||||||
|
{in: 3, want: []int{1, 1, 1, 0}, err: nil},
|
||||||
|
{in: 4, want: []int{1, 1, 1, 1, 0}, err: nil},
|
||||||
|
{in: 5, want: []int{1, 1, 1, 1, 1, 0}, err: nil},
|
||||||
|
{in: -3, want: nil, err: errNegativeSyntaxVal},
|
||||||
|
}
|
||||||
|
|
||||||
|
for i, test := range tests {
|
||||||
|
got, err := unaryBinString(test.in)
|
||||||
|
if err != test.err {
|
||||||
|
t.Errorf("did not get expected error for test %d\nGot: %v\nWant: %v", i, err, test.err)
|
||||||
|
}
|
||||||
|
|
||||||
|
if !reflect.DeepEqual(test.want, got) {
|
||||||
|
t.Errorf("did not get expected result for test %d\nGot: %v\nWant: %v\n", i, got, test.want)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestFixedLengthBinString(t *testing.T) {
|
||||||
|
tests := []struct {
|
||||||
|
v int
|
||||||
|
cMax int
|
||||||
|
want []int
|
||||||
|
err error
|
||||||
|
}{
|
||||||
|
{v: 0, cMax: 7, want: []int{0, 0, 0}},
|
||||||
|
{v: 1, cMax: 7, want: []int{0, 0, 1}},
|
||||||
|
{v: 2, cMax: 7, want: []int{0, 1, 0}},
|
||||||
|
{v: 3, cMax: 7, want: []int{0, 1, 1}},
|
||||||
|
{v: 4, cMax: 7, want: []int{1, 0, 0}},
|
||||||
|
{v: 5, cMax: 7, want: []int{1, 0, 1}},
|
||||||
|
{v: 6, cMax: 7, want: []int{1, 1, 0}},
|
||||||
|
{v: 7, cMax: 7, want: []int{1, 1, 1}},
|
||||||
|
{v: -1, cMax: 7, want: nil, err: errNegativeValue},
|
||||||
|
}
|
||||||
|
|
||||||
|
for i, test := range tests {
|
||||||
|
got, err := fixedLenBinString(test.v, test.cMax)
|
||||||
|
if err != test.err {
|
||||||
|
t.Errorf("did not get expected error for test %d\nGot: %v\nWant: %v\n", i, err, test.err)
|
||||||
|
}
|
||||||
|
|
||||||
|
if !reflect.DeepEqual(test.want, got) {
|
||||||
|
t.Errorf("did not get expected result for test %d\nGot: %v\nWant: %v\n", i, got, test.want)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestTruncUnaryBinString(t *testing.T) {
|
||||||
|
tests := []struct {
|
||||||
|
v int
|
||||||
|
cMax int
|
||||||
|
want []int
|
||||||
|
err error
|
||||||
|
}{
|
||||||
|
{v: 0, cMax: 10, want: []int{0}, err: nil},
|
||||||
|
{v: 1, cMax: 10, want: []int{1, 0}, err: nil},
|
||||||
|
{v: 2, cMax: 10, want: []int{1, 1, 0}, err: nil},
|
||||||
|
{v: 0, cMax: 0, want: []int{}, err: nil},
|
||||||
|
{v: 4, cMax: 4, want: []int{1, 1, 1, 1}, err: nil},
|
||||||
|
{v: 1, cMax: 10, want: []int{1, 0}, err: nil},
|
||||||
|
{v: 2, cMax: 10, want: []int{1, 1, 0}, err: nil},
|
||||||
|
{v: -3, cMax: 10, want: nil, err: errNegativeSyntaxVal},
|
||||||
|
{v: 5, cMax: 4, want: nil, err: errInvalidSyntaxVal},
|
||||||
|
}
|
||||||
|
|
||||||
|
for i, test := range tests {
|
||||||
|
got, err := truncUnaryBinString(test.v, test.cMax)
|
||||||
|
if err != test.err {
|
||||||
|
t.Errorf("did not get expected error for test %d\nGot: %v\nWant: %v", i, err, test.err)
|
||||||
|
}
|
||||||
|
|
||||||
|
if !reflect.DeepEqual(test.want, got) {
|
||||||
|
t.Errorf("did not get expected result for test %d\nGot: %v\nWant: %v\n", i, got, test.want)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestUEGkSuffix(t *testing.T) {
|
||||||
|
// Data from https://patents.google.com/patent/US20070092150
|
||||||
|
tests := []struct {
|
||||||
|
v, uCoff, k int
|
||||||
|
signedValFlag bool
|
||||||
|
want []int
|
||||||
|
}{
|
||||||
|
0: {v: 14, uCoff: 14, want: []int{0}},
|
||||||
|
1: {v: 15, uCoff: 14, want: []int{1, 0, 0}},
|
||||||
|
2: {v: 16, uCoff: 14, want: []int{1, 0, 1}},
|
||||||
|
3: {v: 17, uCoff: 14, want: []int{1, 1, 0, 0, 0}},
|
||||||
|
4: {v: 18, uCoff: 14, want: []int{1, 1, 0, 0, 1}},
|
||||||
|
5: {v: 19, uCoff: 14, want: []int{1, 1, 0, 1, 0}},
|
||||||
|
6: {v: 20, uCoff: 14, want: []int{1, 1, 0, 1, 1}},
|
||||||
|
7: {v: 21, uCoff: 14, want: []int{1, 1, 1, 0, 0, 0, 0}},
|
||||||
|
8: {v: 22, uCoff: 14, want: []int{1, 1, 1, 0, 0, 0, 1}},
|
||||||
|
9: {v: 23, uCoff: 14, want: []int{1, 1, 1, 0, 0, 1, 0}},
|
||||||
|
10: {v: 24, uCoff: 14, want: []int{1, 1, 1, 0, 0, 1, 1}},
|
||||||
|
11: {v: 25, uCoff: 14, want: []int{1, 1, 1, 0, 1, 0, 0}},
|
||||||
|
}
|
||||||
|
|
||||||
|
for i, test := range tests {
|
||||||
|
got := suffix(test.v, test.uCoff, test.k, test.signedValFlag)
|
||||||
|
if !reflect.DeepEqual(got, test.want) {
|
||||||
|
t.Errorf("did not get expected result for test %d\nGot: %v\nWant: %v\n", i, got, test.want)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestUnaryExpGolombBinString(t *testing.T) {
|
||||||
|
tests := []struct {
|
||||||
|
v, uCoff, k int
|
||||||
|
signedValFlag bool
|
||||||
|
want []int
|
||||||
|
}{
|
||||||
|
0: {v: 7, uCoff: 14, want: []int{1, 1, 1, 1, 1, 1, 1, 0}},
|
||||||
|
1: {v: 17, uCoff: 14, want: []int{1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 0, 0, 0}},
|
||||||
|
2: {v: 15, uCoff: 14, signedValFlag: true, want: []int{1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 0, 0, 0}},
|
||||||
|
3: {v: -15, uCoff: 14, signedValFlag: true, want: []int{1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 0, 0, 1}},
|
||||||
|
}
|
||||||
|
|
||||||
|
for i, test := range tests {
|
||||||
|
got, err := unaryExpGolombBinString(test.v, test.uCoff, test.k, test.signedValFlag)
|
||||||
|
if err != nil {
|
||||||
|
t.Errorf("did not expect error %v for test %d", err, i)
|
||||||
|
}
|
||||||
|
|
||||||
|
if !reflect.DeepEqual(got, test.want) {
|
||||||
|
t.Errorf("did not get expected result for test %d\nGot: %v\nWant: %v\n", i, got, test.want)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
Loading…
Reference in New Issue