File: annexb.go

package info (click to toggle)
fq 0.9.0-2
  • links: PTS, VCS
  • area: main
  • in suites: forky, sid, trixie
  • size: 106,624 kB
  • sloc: xml: 2,835; makefile: 250; sh: 241; exp: 57; ansic: 21
file content (47 lines) | stat: -rw-r--r-- 1,022 bytes parent folder | download
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
package mpeg

import (
	"github.com/wader/fq/pkg/decode"
)

func annexBFindStartCode(d *decode.D) (int64, int64, error) {
	offset, v, err := d.TryPeekFind(32, 8, -1, func(v uint64) bool {
		return annexBDecodeStartCodeLen(v) > 0
	})
	return offset, annexBDecodeStartCodeLen(v), err
}

func annexBDecodeStartCodeLen(v uint64) int64 {
	switch {
	case v == 0x00_00_00_01:
		return 4 * 8
	case v&0xff_ff_ff_00 == 0x00_00_01_00:
		return 3 * 8
	default:
		return 0
	}
}

func annexBDecode(d *decode.D, format decode.Group) any {
	currentOffset, currentPrefixLen, err := annexBFindStartCode(d)
	// TODO: really restrict to 0?
	if err != nil || currentOffset != 0 {
		d.Errorf("could not find start code (first)")
	}

	for d.NotEnd() {
		d.FieldRawLen("start_code", currentPrefixLen)

		nextOffset, nextPrefixLen, err := annexBFindStartCode(d)
		if err != nil {
			nextOffset = d.Len() - d.Pos()
		}

		naluLen := nextOffset
		d.FieldFormatLen("nalu", naluLen, &format, nil)

		currentPrefixLen = nextPrefixLen
	}

	return nil
}