mirror of
https://github.com/livekit/livekit.git
synced 2026-04-26 17:27:33 +00:00
42a9b6657d
* Scoped speaker update Include only participants a participant is subscribed to. NOTE: Not doing this for active speaker changes for Protocol < 3. * correct comment spelling
134 lines
3.0 KiB
Go
134 lines
3.0 KiB
Go
package rtc
|
|
|
|
import (
|
|
"sync/atomic"
|
|
"time"
|
|
|
|
"github.com/bep/debounce"
|
|
"github.com/livekit/protocol/livekit"
|
|
"github.com/livekit/protocol/utils"
|
|
"github.com/pion/webrtc/v3"
|
|
|
|
"github.com/livekit/livekit-server/pkg/rtc/types"
|
|
"github.com/livekit/livekit-server/pkg/sfu"
|
|
)
|
|
|
|
const (
|
|
subscriptionDebounceInterval = 100 * time.Millisecond
|
|
)
|
|
|
|
type SubscribedTrackParams struct {
|
|
PublisherID string
|
|
PublisherIdentity string
|
|
SubscriberID string
|
|
MediaTrack types.MediaTrack
|
|
DownTrack *sfu.DownTrack
|
|
}
|
|
type SubscribedTrack struct {
|
|
params SubscribedTrackParams
|
|
subMuted utils.AtomicFlag
|
|
pubMuted utils.AtomicFlag
|
|
settings atomic.Value // *livekit.UpdateTrackSettings
|
|
|
|
onBind func()
|
|
|
|
debouncer func(func())
|
|
}
|
|
|
|
func NewSubscribedTrack(params SubscribedTrackParams) *SubscribedTrack {
|
|
return &SubscribedTrack{
|
|
params: params,
|
|
debouncer: debounce.New(subscriptionDebounceInterval),
|
|
}
|
|
}
|
|
|
|
func (t *SubscribedTrack) OnBind(f func()) {
|
|
t.onBind = f
|
|
}
|
|
|
|
func (t *SubscribedTrack) Bound() {
|
|
if t.onBind != nil {
|
|
t.onBind()
|
|
}
|
|
}
|
|
|
|
func (t *SubscribedTrack) ID() string {
|
|
return t.params.DownTrack.ID()
|
|
}
|
|
|
|
func (t *SubscribedTrack) PublisherID() string {
|
|
return t.params.PublisherID
|
|
}
|
|
|
|
func (t *SubscribedTrack) PublisherIdentity() string {
|
|
return t.params.PublisherIdentity
|
|
}
|
|
|
|
func (t *SubscribedTrack) DownTrack() *sfu.DownTrack {
|
|
return t.params.DownTrack
|
|
}
|
|
|
|
func (t *SubscribedTrack) MediaTrack() types.MediaTrack {
|
|
return t.params.MediaTrack
|
|
}
|
|
|
|
// has subscriber indicated it wants to mute this track
|
|
func (t *SubscribedTrack) IsMuted() bool {
|
|
return t.subMuted.Get()
|
|
}
|
|
|
|
func (t *SubscribedTrack) SetPublisherMuted(muted bool) {
|
|
t.pubMuted.TrySet(muted)
|
|
t.updateDownTrackMute()
|
|
}
|
|
|
|
func (t *SubscribedTrack) UpdateSubscriberSettings(settings *livekit.UpdateTrackSettings) {
|
|
visibilityChanged := t.subMuted.TrySet(settings.Disabled)
|
|
t.settings.Store(settings)
|
|
// avoid frequent changes to mute & video layers, unless it became visible
|
|
if visibilityChanged && !settings.Disabled {
|
|
t.UpdateVideoLayer()
|
|
} else {
|
|
t.debouncer(t.UpdateVideoLayer)
|
|
}
|
|
}
|
|
|
|
func (t *SubscribedTrack) UpdateVideoLayer() {
|
|
t.updateDownTrackMute()
|
|
if t.DownTrack().Kind() != webrtc.RTPCodecTypeVideo {
|
|
return
|
|
}
|
|
if t.subMuted.Get() {
|
|
t.MediaTrack().NotifySubscriberMute(t.params.SubscriberID)
|
|
return
|
|
}
|
|
settings, ok := t.settings.Load().(*livekit.UpdateTrackSettings)
|
|
if !ok {
|
|
return
|
|
}
|
|
|
|
quality := settings.Quality
|
|
if settings.Width > 0 {
|
|
quality = t.MediaTrack().GetQualityForDimension(settings.Width, settings.Height)
|
|
}
|
|
t.DownTrack().SetMaxSpatialLayer(spatialLayerForQuality(quality))
|
|
|
|
t.MediaTrack().NotifySubscriberMaxQuality(t.params.SubscriberID, quality)
|
|
}
|
|
|
|
func (t *SubscribedTrack) updateDownTrackMute() {
|
|
muted := t.subMuted.Get() || t.pubMuted.Get()
|
|
t.DownTrack().Mute(muted)
|
|
}
|
|
|
|
func spatialLayerForQuality(quality livekit.VideoQuality) int32 {
|
|
switch quality {
|
|
case livekit.VideoQuality_LOW:
|
|
return 0
|
|
case livekit.VideoQuality_MEDIUM:
|
|
return 1
|
|
default:
|
|
return 2
|
|
}
|
|
}
|