mirror of
https://github.com/langhuihui/monibuca.git
synced 2025-12-24 13:48:04 +08:00
61 lines
1.6 KiB
Go
61 lines
1.6 KiB
Go
package srt
|
|
|
|
import (
|
|
srt "github.com/datarhei/gosrt"
|
|
"github.com/langhuihui/gomem"
|
|
task "github.com/langhuihui/gotask"
|
|
"m7s.live/v5"
|
|
"m7s.live/v5/pkg/codec"
|
|
"m7s.live/v5/pkg/format"
|
|
mpegts "m7s.live/v5/pkg/format/ts"
|
|
hls "m7s.live/v5/plugin/hls/pkg"
|
|
)
|
|
|
|
type Sender struct {
|
|
task.Task
|
|
hls.TsInMemory
|
|
srt.Conn
|
|
Subscriber *m7s.Subscriber
|
|
}
|
|
|
|
func (s *Sender) Start() error {
|
|
var audioCodec, videoCodec codec.FourCC
|
|
if s.Subscriber.Publisher.HasAudioTrack() {
|
|
audioCodec = s.Subscriber.Publisher.AudioTrack.FourCC()
|
|
}
|
|
if s.Subscriber.Publisher.HasVideoTrack() {
|
|
videoCodec = s.Subscriber.Publisher.VideoTrack.FourCC()
|
|
}
|
|
s.SetAllocator(gomem.NewScalableMemoryAllocator(1 << gomem.MinPowerOf2))
|
|
s.Using(s.GetAllocator(), s.Subscriber)
|
|
s.OnStop(s.Conn.Close)
|
|
s.WritePMTPacket(audioCodec, videoCodec)
|
|
s.Write(mpegts.DefaultPATPacket)
|
|
s.Write(s.PMT)
|
|
return nil
|
|
}
|
|
|
|
func (s *Sender) Run() error {
|
|
pesAudio, pesVideo := mpegts.CreatePESWriters()
|
|
return m7s.PlayBlock(s.Subscriber, func(audio *format.Mpeg2Audio) (err error) {
|
|
pesAudio.Pts = uint64(s.Subscriber.AudioReader.AbsTime) * 90
|
|
err = pesAudio.WritePESPacket(audio.Memory, &s.RecyclableMemory)
|
|
if err == nil {
|
|
s.RecyclableMemory.WriteTo(s)
|
|
s.RecyclableMemory.Recycle()
|
|
}
|
|
return
|
|
}, func(video *mpegts.VideoFrame) (err error) {
|
|
vr := s.Subscriber.VideoReader
|
|
pesVideo.IsKeyFrame = video.IDR
|
|
pesVideo.Pts = uint64(vr.AbsTime+video.GetCTS32()) * 90
|
|
pesVideo.Dts = uint64(vr.AbsTime) * 90
|
|
err = pesVideo.WritePESPacket(video.Memory, &s.RecyclableMemory)
|
|
if err == nil {
|
|
s.RecyclableMemory.WriteTo(s)
|
|
s.RecyclableMemory.Recycle()
|
|
}
|
|
return
|
|
})
|
|
}
|