mirror of
https://github.com/livekit/livekit.git
synced 2026-04-25 09:02:11 +00:00
57 lines
1.2 KiB
Go
57 lines
1.2 KiB
Go
package rtc
|
|
|
|
import (
|
|
"github.com/pion/ion-sfu/pkg/sfu"
|
|
"github.com/pion/webrtc/v3"
|
|
|
|
"github.com/livekit/livekit-server/pkg/utils"
|
|
"github.com/livekit/livekit-server/proto/livekit"
|
|
)
|
|
|
|
type SubscribedTrack struct {
|
|
dt *sfu.DownTrack
|
|
subMuted utils.AtomicFlag
|
|
pubMuted utils.AtomicFlag
|
|
}
|
|
|
|
func NewSubscribedTrack(dt *sfu.DownTrack) *SubscribedTrack {
|
|
return &SubscribedTrack{
|
|
dt: dt,
|
|
}
|
|
}
|
|
|
|
func (t *SubscribedTrack) ID() string {
|
|
return t.dt.ID()
|
|
}
|
|
|
|
func (t *SubscribedTrack) DownTrack() *sfu.DownTrack {
|
|
return t.dt
|
|
}
|
|
|
|
// has subscriber indicated it wants to mute this track
|
|
func (t *SubscribedTrack) IsMuted() bool {
|
|
return t.subMuted.Get()
|
|
}
|
|
|
|
// set subscriber mute preference
|
|
func (t *SubscribedTrack) SetMuted(muted bool) {
|
|
t.subMuted.TrySet(muted)
|
|
t.updateDownTrackMute()
|
|
}
|
|
|
|
func (t *SubscribedTrack) SetPublisherMuted(muted bool) {
|
|
t.pubMuted.TrySet(muted)
|
|
t.updateDownTrackMute()
|
|
}
|
|
|
|
func (t *SubscribedTrack) SetVideoQuality(quality livekit.VideoQuality) {
|
|
if t.dt.Kind() == webrtc.RTPCodecTypeVideo {
|
|
t.dt.SwitchSpatialLayer(int64(quality), true)
|
|
}
|
|
}
|
|
|
|
func (t *SubscribedTrack) updateDownTrackMute() {
|
|
muted := t.subMuted.Get() || t.pubMuted.Get()
|
|
t.dt.Mute(muted)
|
|
}
|