mirror of
https://github.com/Monibuca/plugin-rtsp.git
synced 2025-09-26 19:51:14 +08:00
104 lines
2.7 KiB
Go
104 lines
2.7 KiB
Go
package rtsp
|
||
|
||
import (
|
||
"github.com/bluenviron/mediacommon/pkg/codecs/mpeg4audio"
|
||
"github.com/bluenviron/gortsplib/v3/pkg/formats"
|
||
"github.com/bluenviron/gortsplib/v3/pkg/media"
|
||
"github.com/pion/rtp"
|
||
"go.uber.org/zap"
|
||
. "m7s.live/engine/v4"
|
||
"m7s.live/engine/v4/common"
|
||
. "m7s.live/engine/v4/track"
|
||
"m7s.live/engine/v4/util"
|
||
)
|
||
|
||
type RTSPPublisher struct {
|
||
Publisher
|
||
Tracks map[*media.Media]common.AVTrack `json:"-" yaml:"-"`
|
||
RTSPIO
|
||
}
|
||
|
||
func (p *RTSPPublisher) SetTracks() error {
|
||
p.Tracks = make(map[*media.Media]common.AVTrack, len(p.tracks))
|
||
defer func() {
|
||
for _, track := range p.Tracks {
|
||
p.Info("set track", zap.String("name", track.GetBase().Name))
|
||
}
|
||
}()
|
||
for _, track := range p.tracks {
|
||
for _, forma := range track.Formats {
|
||
switch f := forma.(type) {
|
||
case *formats.H264:
|
||
vt := p.VideoTrack
|
||
if vt == nil {
|
||
vt = NewH264(p.Stream, f.PayloadType(), make(util.BytesPool, 17))
|
||
p.VideoTrack = vt
|
||
}
|
||
p.Tracks[track] = p.VideoTrack
|
||
if len(f.SPS) > 0 {
|
||
vt.WriteSliceBytes(f.SPS)
|
||
}
|
||
if len(f.PPS) > 0 {
|
||
vt.WriteSliceBytes(f.PPS)
|
||
}
|
||
case *formats.H265:
|
||
vt := p.VideoTrack
|
||
if vt == nil {
|
||
vt = NewH265(p.Stream, f.PayloadType(), make(util.BytesPool, 17))
|
||
p.VideoTrack = vt
|
||
}
|
||
p.Tracks[track] = p.VideoTrack
|
||
if len(f.VPS) > 0 {
|
||
vt.WriteSliceBytes(f.VPS)
|
||
}
|
||
if len(f.SPS) > 0 {
|
||
vt.WriteSliceBytes(f.SPS)
|
||
}
|
||
if len(f.PPS) > 0 {
|
||
vt.WriteSliceBytes(f.PPS)
|
||
}
|
||
case *formats.MPEG4Audio:
|
||
at := p.AudioTrack
|
||
if at == nil {
|
||
at := NewAAC(p.Stream, f.PayloadType(), uint32(f.Config.SampleRate), make(util.BytesPool, 17))
|
||
at.IndexDeltaLength = f.IndexDeltaLength
|
||
at.IndexLength = f.IndexLength
|
||
at.SizeLength = f.SizeLength
|
||
if f.Config.Type == mpeg4audio.ObjectTypeAACLC {
|
||
at.Mode = 1
|
||
}
|
||
at.Channels = uint8(f.Config.ChannelCount)
|
||
asc, _ := f.Config.Marshal()
|
||
// 复用AVCC写入逻辑，解析出AAC的配置信息
|
||
at.WriteSequenceHead(append([]byte{0xAF, 0x00}, asc...))
|
||
p.AudioTrack = at
|
||
}
|
||
p.Tracks[track] = p.AudioTrack
|
||
case *formats.G711:
|
||
at := p.AudioTrack
|
||
if at == nil {
|
||
at := NewG711(p.Stream, !f.MULaw, f.PayloadType(), uint32(f.ClockRate()), make(util.BytesPool, 17))
|
||
at.AVCCHead = []byte{(byte(at.CodecID) << 4) | (1 << 1)}
|
||
p.AudioTrack = at
|
||
}
|
||
p.Tracks[track] = p.AudioTrack
|
||
}
|
||
}
|
||
}
|
||
if p.VideoTrack == nil {
|
||
p.Config.PubVideo = false
|
||
p.Info("no video track")
|
||
}
|
||
if p.AudioTrack == nil {
|
||
p.Config.PubAudio = false
|
||
p.Info("no audio track")
|
||
}
|
||
return nil
|
||
}
|
||
|
||
func (p *RTSPPublisher) OnPacket(m *media.Media, f formats.Format, pack *rtp.Packet) {
|
||
if t, ok := p.Tracks[m]; ok {
|
||
t.WriteRTPPack(pack)
|
||
}
|
||
}
|