3bd15c08fd
git-subtree-dir: ts git-subtree-mainline: 92528599badd9da7f5b10d64f10186a1646bb520 git-subtree-split: 7530d353fdefcc59d83c3c8ebea0a147dcbab234
123 lines
2.1 KiB
Go
123 lines
2.1 KiB
Go
package ts
|
|
|
|
import (
|
|
"fmt"
|
|
"os"
|
|
)
|
|
|
|
const (
|
|
ElementaryStreamTypeH264 = 0x1B
|
|
ElementaryStreamTypeAdtsAAC = 0x0F
|
|
)
|
|
|
|
type TSHeader struct {
|
|
PID uint
|
|
PCR uint64
|
|
OPCR uint64
|
|
ContinuityCounter uint
|
|
PayloadUnitStart bool
|
|
DiscontinuityIndicator bool
|
|
RandomAccessIndicator bool
|
|
HeaderLength uint
|
|
}
|
|
|
|
type PATEntry struct {
|
|
ProgramNumber uint
|
|
NetworkPID uint
|
|
ProgramMapPID uint
|
|
}
|
|
|
|
type PAT struct {
|
|
Entries []PATEntry
|
|
}
|
|
|
|
type PMT struct {
|
|
PCRPID uint
|
|
ProgramDescriptors []Descriptor
|
|
ElementaryStreamInfos []ElementaryStreamInfo
|
|
}
|
|
|
|
type Descriptor struct {
|
|
Tag uint
|
|
Data []byte
|
|
}
|
|
|
|
type ElementaryStreamInfo struct {
|
|
StreamType uint
|
|
ElementaryPID uint
|
|
Descriptors []Descriptor
|
|
}
|
|
|
|
type PSI struct {
|
|
TableIdExtension uint
|
|
TableId uint
|
|
SecNum uint
|
|
LastSecNum uint
|
|
}
|
|
|
|
const (
|
|
StreamIdH264 = 0xe0
|
|
StreamIdAAC = 0xc0
|
|
)
|
|
|
|
type PESHeader struct {
|
|
StreamId uint // H264=0xe0 AAC=0xc0
|
|
DataLength uint
|
|
PTS uint64
|
|
DTS uint64
|
|
ESCR uint64
|
|
}
|
|
|
|
func PESUIntToTs(v uint64) (ts uint64) {
|
|
// 0010 PTS 32..30 1 PTS 29..15 1 PTS 14..00 1
|
|
return (((v >> 33) & 0x7) << 30) | (((v >> 17) & 0x7fff) << 15) | ((v >> 1) & 0x7fff)
|
|
}
|
|
|
|
func PESTsToUInt(ts uint64) (v uint64) {
|
|
// 0010 PTS 32..30 1 PTS 29..15 1 PTS 14..00 1
|
|
return ((ts>>30)&0x7)<<33 | ((ts>>15)&0x7fff)<<17 | (ts&0x7fff)<<1 | 0x100010001
|
|
}
|
|
|
|
const (
|
|
PTS_HZ = 90000
|
|
PCR_HZ = 27000000
|
|
)
|
|
|
|
func UIntToPCR(v uint64) uint64 {
|
|
// base(33)+resverd(6)+ext(9)
|
|
base := v >> 15
|
|
ext := v & 0x1ff
|
|
return base*300 + ext
|
|
}
|
|
|
|
func PCRToUInt(pcr uint64) uint64 {
|
|
base := pcr / 300
|
|
ext := pcr % 300
|
|
return base<<15 | 0x3f<<9 | ext
|
|
}
|
|
|
|
var DebugOutput = os.Stdout
|
|
var DebugReader = false
|
|
var DebugWriter = false
|
|
|
|
type FieldsDumper struct {
|
|
Fields []struct {
|
|
Length int
|
|
Desc string
|
|
}
|
|
Val uint
|
|
Length uint
|
|
}
|
|
|
|
func (self FieldsDumper) String() (res string) {
|
|
pos := uint(self.Length)
|
|
for _, field := range self.Fields {
|
|
pos -= uint(field.Length)
|
|
val := (self.Val >> pos) & (1<<uint(field.Length) - 1)
|
|
if val != 0 {
|
|
res += fmt.Sprintf("%s=%x ", field.Desc, val)
|
|
}
|
|
}
|
|
return
|
|
}
|