mirror of
https://github.com/langhuihui/monibuca.git
synced 2026-05-08 01:21:04 +08:00
langhuihui
86 lines
2.1 KiB
Go
86 lines
2.1 KiB
Go
package gb28181
|
|
|
|
import (
|
|
"io"
|
|
"m7s.live/v5/pkg"
|
|
"m7s.live/v5/pkg/codec"
|
|
"m7s.live/v5/pkg/util"
|
|
mpegts "m7s.live/v5/plugin/hls/pkg/ts"
|
|
"time"
|
|
)
|
|
|
|
type PSVideo struct {
|
|
PSAudio
|
|
}
|
|
|
|
func (es *PSVideo) parsePESPacket(payload util.Memory) (result *pkg.AnnexB, err error) {
|
|
if payload.Size < 4 {
|
|
err = io.ErrShortBuffer
|
|
return
|
|
}
|
|
var flag, pesHeaderDataLen byte
|
|
reader := payload.NewReader()
|
|
reader.Skip(1)
|
|
//data_alignment_indicator := (payload[0]&0b0001_0000)>>4 == 1
|
|
err = reader.ReadByteTo(&flag, &pesHeaderDataLen)
|
|
if err != nil {
|
|
return
|
|
}
|
|
ptsFlag := flag>>7 == 1
|
|
dtsFlag := (flag&0b0100_0000)>>6 == 1
|
|
if payload.Size < int(pesHeaderDataLen) {
|
|
err = io.ErrShortBuffer
|
|
return
|
|
}
|
|
var extraData []byte
|
|
extraData, err = reader.ReadBytes(int(pesHeaderDataLen))
|
|
pts, dts := es.PTS, es.DTS
|
|
if ptsFlag && len(extraData) > 4 {
|
|
pts = uint32(extraData[0]&0b0000_1110) << 29
|
|
pts |= uint32(extraData[1]) << 22
|
|
pts |= uint32(extraData[2]&0b1111_1110) << 14
|
|
pts |= uint32(extraData[3]) << 7
|
|
pts |= uint32(extraData[4]) >> 1
|
|
if dtsFlag && len(extraData) > 9 {
|
|
dts = uint32(extraData[5]&0b0000_1110) << 29
|
|
dts |= uint32(extraData[6]) << 22
|
|
dts |= uint32(extraData[7]&0b1111_1110) << 14
|
|
dts |= uint32(extraData[8]) << 7
|
|
dts |= uint32(extraData[9]) >> 1
|
|
} else {
|
|
dts = pts
|
|
}
|
|
}
|
|
if pts != es.PTS && es.Memory.Size > 0 {
|
|
result = &pkg.AnnexB{
|
|
PTS: time.Duration(es.PTS),
|
|
DTS: time.Duration(es.DTS),
|
|
}
|
|
switch es.streamType {
|
|
case 0:
|
|
//推测编码类型
|
|
switch codec.ParseH264NALUType(es.Memory.Buffers[0][4]) {
|
|
case codec.NALU_Non_IDR_Picture,
|
|
codec.NALU_IDR_Picture,
|
|
codec.NALU_SEI,
|
|
codec.NALU_SPS,
|
|
codec.NALU_PPS,
|
|
codec.NALU_Access_Unit_Delimiter:
|
|
default:
|
|
result.Hevc = true
|
|
}
|
|
case mpegts.STREAM_TYPE_H265:
|
|
result.Hevc = true
|
|
}
|
|
result.Memory.CopyFrom(&es.Memory)
|
|
// fmt.Println("clone", es.PTS, es.Buffer[4]&0x0f)
|
|
es.Recycle()
|
|
es.Memory = util.Memory{}
|
|
}
|
|
es.PTS, es.DTS = pts, dts
|
|
// fmt.Println("append", es.PTS, payload[pesHeaderDataLen+4]&0x0f)
|
|
reader.Range(es.AppendOne)
|
|
// es.Buffer = append(es.Buffer, payload[pesHeaderDataLen:]...)
|
|
return
|
|
}
|