mirror of
https://github.com/livekit/livekit.git
synced 2026-05-25 12:04:47 +00:00
David Zhao
96 lines
2.4 KiB
Go
96 lines
2.4 KiB
Go
package rtc
|
|
|
|
import (
|
|
"time"
|
|
|
|
"github.com/bep/debounce"
|
|
livekit "github.com/livekit/protocol/proto"
|
|
"github.com/livekit/protocol/utils"
|
|
"github.com/pion/ion-sfu/pkg/sfu"
|
|
"github.com/pion/webrtc/v3"
|
|
)
|
|
|
|
const (
|
|
subscriptionDebounceInterval = 100 * time.Millisecond
|
|
)
|
|
|
|
type SubscribedTrack struct {
|
|
dt *sfu.DownTrack
|
|
publisherIdentity string
|
|
subMuted utils.AtomicFlag
|
|
pubMuted utils.AtomicFlag
|
|
|
|
debouncer func(func())
|
|
}
|
|
|
|
func NewSubscribedTrack(publisherIdentity string, dt *sfu.DownTrack) *SubscribedTrack {
|
|
return &SubscribedTrack{
|
|
publisherIdentity: publisherIdentity,
|
|
dt: dt,
|
|
debouncer: debounce.New(subscriptionDebounceInterval),
|
|
}
|
|
}
|
|
|
|
func (t *SubscribedTrack) ID() string {
|
|
return t.dt.ID()
|
|
}
|
|
|
|
func (t *SubscribedTrack) PublisherIdentity() string {
|
|
return t.publisherIdentity
|
|
}
|
|
|
|
func (t *SubscribedTrack) DownTrack() *sfu.DownTrack {
|
|
return t.dt
|
|
}
|
|
|
|
func (t *SubscribedTrack) SubscribeLossPercentage() uint32 {
|
|
return FixedPointToPercent(t.DownTrack().CurrentMaxLossFraction())
|
|
}
|
|
|
|
// 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(enabled bool, quality livekit.VideoQuality) {
|
|
t.debouncer(func() {
|
|
t.subMuted.TrySet(!enabled)
|
|
t.updateDownTrackMute()
|
|
if enabled && t.dt.Kind() == webrtc.RTPCodecTypeVideo {
|
|
err := t.dt.SwitchSpatialLayer(spatialLayerForQuality(quality), true)
|
|
if err == sfu.ErrSpatialLayerNotFound && quality != livekit.VideoQuality_MEDIUM {
|
|
// try to switch to middle layer
|
|
_ = t.dt.SwitchSpatialLayer(spatialLayerForQuality(livekit.VideoQuality_MEDIUM), true)
|
|
}
|
|
}
|
|
})
|
|
}
|
|
|
|
func (t *SubscribedTrack) updateDownTrackMute() {
|
|
muted := t.subMuted.Get() || t.pubMuted.Get()
|
|
t.dt.Mute(muted)
|
|
}
|
|
|
|
// GetQualityForDimension finds the closest quality to use for desired dimensions
|
|
// affords a 10% tolerance on dimension
|
|
func GetQualityForDimension(width, height uint32) livekit.VideoQuality {
|
|
// currently the layers are set to 180p/360p/original res, we should re
|
|
return livekit.VideoQuality_HIGH
|
|
}
|
|
|
|
func spatialLayerForQuality(quality livekit.VideoQuality) int32 {
|
|
switch quality {
|
|
case livekit.VideoQuality_LOW:
|
|
return 0
|
|
case livekit.VideoQuality_MEDIUM:
|
|
return 1
|
|
default:
|
|
return 2
|
|
}
|
|
}
|