mirror of
https://github.com/AlexxIT/go2rtc.git
synced 2025-09-26 20:31:11 +08:00
90 lines
2.3 KiB
Go
90 lines
2.3 KiB
Go
package webrtc
|
|
|
|
import (
|
|
"errors"
|
|
|
|
"github.com/AlexxIT/go2rtc/pkg/core"
|
|
"github.com/AlexxIT/go2rtc/pkg/h264"
|
|
"github.com/AlexxIT/go2rtc/pkg/h265"
|
|
"github.com/AlexxIT/go2rtc/pkg/pcm"
|
|
"github.com/pion/rtp"
|
|
)
|
|
|
|
func (c *Conn) GetMedias() []*core.Media {
|
|
return WithResampling(c.Medias)
|
|
}
|
|
|
|
func (c *Conn) AddTrack(media *core.Media, codec *core.Codec, track *core.Receiver) error {
|
|
core.Assert(media.Direction == core.DirectionSendonly)
|
|
|
|
for _, sender := range c.Senders {
|
|
if sender.Codec == codec {
|
|
sender.Bind(track)
|
|
return nil
|
|
}
|
|
}
|
|
|
|
switch c.Mode {
|
|
case core.ModePassiveConsumer: // video/audio for browser
|
|
case core.ModeActiveProducer: // go2rtc as WebRTC client (backchannel)
|
|
case core.ModePassiveProducer: // WebRTC/WHIP
|
|
default:
|
|
panic(core.Caller())
|
|
}
|
|
|
|
localTrack := c.getSenderTrack(media.ID)
|
|
if localTrack == nil {
|
|
return errors.New("webrtc: can't get track")
|
|
}
|
|
|
|
payloadType := codec.PayloadType
|
|
|
|
sender := core.NewSender(media, codec)
|
|
sender.Handler = func(packet *rtp.Packet) {
|
|
c.Send += packet.MarshalSize()
|
|
//important to send with remote PayloadType
|
|
_ = localTrack.WriteRTP(payloadType, packet)
|
|
}
|
|
|
|
switch track.Codec.Name {
|
|
case core.CodecH264:
|
|
sender.Handler = h264.RTPPay(1200, sender.Handler)
|
|
if track.Codec.IsRTP() {
|
|
sender.Handler = h264.RTPDepay(track.Codec, sender.Handler)
|
|
} else {
|
|
sender.Handler = h264.RepairAVCC(track.Codec, sender.Handler)
|
|
}
|
|
|
|
case core.CodecH265:
|
|
// SafariPay because it is the only browser in the world
|
|
// that supports WebRTC + H265
|
|
sender.Handler = h265.SafariPay(1200, sender.Handler)
|
|
if track.Codec.IsRTP() {
|
|
sender.Handler = h265.RTPDepay(track.Codec, sender.Handler)
|
|
}
|
|
|
|
case core.CodecPCMA, core.CodecPCMU, core.CodecPCM, core.CodecPCML:
|
|
if codec.ClockRate == 0 {
|
|
if codec.Name == core.CodecPCM || codec.Name == core.CodecPCML {
|
|
codec.Name = core.CodecPCMA
|
|
}
|
|
codec.ClockRate = 8000
|
|
sender.Handler = pcm.ResampleToG711(track.Codec, 8000, sender.Handler)
|
|
}
|
|
|
|
// Fix audio quality https://github.com/AlexxIT/WebRTC/issues/500
|
|
sender.Handler = pcm.RepackG711(false, sender.Handler)
|
|
}
|
|
|
|
// TODO: rewrite this dirty logic
|
|
// maybe not best solution, but ActiveProducer connected before AddTrack
|
|
if c.Mode != core.ModeActiveProducer {
|
|
sender.Bind(track)
|
|
} else {
|
|
sender.HandleRTP(track)
|
|
}
|
|
|
|
c.Senders = append(c.Senders, sender)
|
|
return nil
|
|
}
|