mirror of
https://github.com/Monibuca/plugin-rtsp.git
synced 2025-11-01 03:12:45 +08:00
101 lines
2.4 KiB
Go
101 lines
2.4 KiB
Go
package rtsp
|
|
|
|
import (
|
|
"github.com/bluenviron/gortsplib/v3"
|
|
"github.com/bluenviron/mediacommon/pkg/codecs/mpeg4audio"
|
|
"github.com/bluenviron/gortsplib/v3/pkg/formats"
|
|
"github.com/bluenviron/gortsplib/v3/pkg/media"
|
|
. "m7s.live/engine/v4"
|
|
"m7s.live/engine/v4/codec"
|
|
"m7s.live/engine/v4/track"
|
|
)
|
|
|
|
type RTSPSubscriber struct {
|
|
Subscriber
|
|
RTSPIO
|
|
}
|
|
|
|
func (s *RTSPSubscriber) OnEvent(event any) {
|
|
switch v := event.(type) {
|
|
case *track.Video:
|
|
if s.Video != nil {
|
|
return
|
|
}
|
|
switch v.CodecID {
|
|
case codec.CodecID_H264:
|
|
video := &media.Media{
|
|
Type: media.TypeVideo,
|
|
Formats: []formats.Format{&formats.H264{
|
|
PacketizationMode: 1,
|
|
PayloadTyp: v.PayloadType,
|
|
SPS: v.ParamaterSets[0],
|
|
PPS: v.ParamaterSets[1],
|
|
}},
|
|
}
|
|
s.videoTrack = video
|
|
s.tracks = append(s.tracks, video)
|
|
case codec.CodecID_H265:
|
|
video := &media.Media{
|
|
Type: media.TypeVideo,
|
|
Formats: []formats.Format{&formats.H265{
|
|
PayloadTyp: v.PayloadType,
|
|
VPS: v.ParamaterSets[0],
|
|
SPS: v.ParamaterSets[1],
|
|
PPS: v.ParamaterSets[2],
|
|
}},
|
|
}
|
|
s.videoTrack = video
|
|
s.tracks = append(s.tracks, video)
|
|
}
|
|
s.AddTrack(v)
|
|
case *track.Audio:
|
|
if s.Audio != nil {
|
|
return
|
|
}
|
|
switch v.CodecID {
|
|
case codec.CodecID_AAC:
|
|
audio := &media.Media{
|
|
Type: media.TypeAudio,
|
|
Formats: []formats.Format{&formats.MPEG4Audio{
|
|
PayloadTyp: v.PayloadType,
|
|
Config: &mpeg4audio.Config{
|
|
Type: mpeg4audio.ObjectTypeAACLC,
|
|
SampleRate: int(v.SampleRate),
|
|
ChannelCount: int(v.Channels),
|
|
},
|
|
SizeLength: v.SizeLength,
|
|
IndexLength: v.IndexLength,
|
|
IndexDeltaLength: v.IndexDeltaLength,
|
|
}},
|
|
}
|
|
s.audioTrack = audio
|
|
s.tracks = append(s.tracks, audio)
|
|
case codec.CodecID_PCMA:
|
|
audio := &media.Media{
|
|
Type: media.TypeAudio,
|
|
Formats: []formats.Format{&formats.G711{}},
|
|
}
|
|
s.audioTrack = audio
|
|
s.tracks = append(s.tracks, audio)
|
|
case codec.CodecID_PCMU:
|
|
audio := &media.Media{
|
|
Type: media.TypeAudio,
|
|
Formats: []formats.Format{&formats.G711{
|
|
MULaw: true,
|
|
}},
|
|
}
|
|
s.audioTrack = audio
|
|
s.tracks = append(s.tracks, audio)
|
|
}
|
|
s.AddTrack(v)
|
|
case ISubscriber:
|
|
s.stream = gortsplib.NewServerStream(s.tracks)
|
|
case VideoRTP:
|
|
s.stream.WritePacketRTP(s.videoTrack, v.Packet)
|
|
case AudioRTP:
|
|
s.stream.WritePacketRTP(s.audioTrack, v.Packet)
|
|
default:
|
|
s.Subscriber.OnEvent(event)
|
|
}
|
|
}
|