Toggle estimator config (#32)

* add estimator to conifg.

* ensure video auto is false when estimator is disabled.

internal/config/webrtc.go

@@ -26,6 +26,9 @@ type WebRTC struct {
 
 	NAT1To1IPs     []string
 	IpRetrievalUrl string
+
+	EstimatorEnabled        bool
+	EstimatorInitialBitrate int
 }
 
 func (WebRTC) Init(cmd *cobra.Command) error {
@@ -69,6 +72,18 @@ func (WebRTC) Init(cmd *cobra.Command) error {
 		return err
 	}
 
+	// bandwidth estimator
+
+	cmd.PersistentFlags().Bool("webrtc.estimator.enabled", false, "enables the bandwidth estimator")
+	if err := viper.BindPFlag("webrtc.estimator.enabled", cmd.PersistentFlags().Lookup("webrtc.estimator.enabled")); err != nil {
+		return err
+	}
+
+	cmd.PersistentFlags().Int("webrtc.estimator.initial_bitrate", 1_000_000, "initial bitrate for the bandwidth estimator")
+	if err := viper.BindPFlag("webrtc.estimator.initial_bitrate", cmd.PersistentFlags().Lookup("webrtc.estimator.initial_bitrate")); err != nil {
+		return err
+	}
+
 	return nil
 }
 
@@ -135,4 +150,9 @@ func (s *WebRTC) Set() {
 			log.Warn().Err(err).Msgf("IP retrieval failed")
 		}
 	}
+
+	// bandwidth estimator
+
+	s.EstimatorEnabled = viper.GetBool("webrtc.estimator.enabled")
+	s.EstimatorInitialBitrate = viper.GetInt("webrtc.estimator.initial_bitrate")
 }

internal/webrtc/manager.go

@@ -211,9 +211,12 @@ func (manager *WebRTCManagerCtx) newPeerConnection(bitrate int, codecs []codec.R
 	// create interceptor registry
 	registry := &interceptor.Registry{}
 
+	// create bandwidth estimator
+	estimatorChan := make(chan cc.BandwidthEstimator, 1)
+	if manager.config.EstimatorEnabled {
 		congestionController, err := cc.NewInterceptor(func() (cc.BandwidthEstimator, error) {
 			if bitrate == 0 {
-			bitrate = 1_000_000
+				bitrate = manager.config.EstimatorInitialBitrate
 			}
 
 			return gcc.NewSendSideBWE(
@@ -225,7 +228,6 @@ func (manager *WebRTCManagerCtx) newPeerConnection(bitrate int, codecs []codec.R
 			return nil, nil, err
 		}
 
-	estimatorChan := make(chan cc.BandwidthEstimator, 1)
 		congestionController.OnNewPeerConnection(func(id string, estimator cc.BandwidthEstimator) {
 			estimatorChan <- estimator
 		})
@@ -234,6 +236,10 @@ func (manager *WebRTCManagerCtx) newPeerConnection(bitrate int, codecs []codec.R
 		if err = webrtc.ConfigureTWCCHeaderExtensionSender(engine, registry); err != nil {
 			return nil, nil, err
 		}
+	} else {
+		// no estimator, send nil
+		estimatorChan <- nil
+	}
 
 	if err := webrtc.RegisterDefaultInterceptors(engine, registry); err != nil {
 		return nil, nil, err
@@ -276,7 +282,8 @@ func (manager *WebRTCManagerCtx) CreatePeer(session types.Session, bitrate int,
 		return nil, err
 	}
 
-	if bitrate == 0 {
+	// if bitrate is 0, and estimator is enabled, use estimator bitrate
+	if bitrate == 0 && estimator != nil {
 		bitrate = estimator.GetTargetBitrate()
 	}
 
@@ -309,6 +316,11 @@ func (manager *WebRTCManagerCtx) CreatePeer(session types.Session, bitrate int,
 		return nil, err
 	}
 
+	// if estimator is disabled, disable video auto
+	if !manager.config.EstimatorEnabled {
+		videoAuto = false
+	}
+
 	// video track
 	videoTrack, err := NewTrack(logger, videoCodec, connection, WithVideoAuto(videoAuto))
 	if err != nil {
@@ -321,7 +333,7 @@ func (manager *WebRTCManagerCtx) CreatePeer(session types.Session, bitrate int,
 	changeVideoFromBitrate := func(peerBitrate int) {
 		// when switching from manual to auto bitrate estimation, in case the estimator is
 		// idle (lastBitrate > maxBitrate), we want to go back to the previous estimated bitrate
-		if peerBitrate == 0 {
+		if peerBitrate == 0 && estimator != nil {
 			peerBitrate = estimator.GetTargetBitrate()
 			manager.logger.Debug().
 				Int("peer_bitrate", peerBitrate).
@@ -397,8 +409,10 @@ func (manager *WebRTCManagerCtx) CreatePeer(session types.Session, bitrate int,
 	// set initial video bitrate
 	changeVideoFromBitrate(bitrate)
 
-	// use a ticker to get current client target bitrate
+	// if estimator is enabled, use it to change video stream
+	if estimator != nil {
 		go func() {
+			// use a ticker to get current client target bitrate
 			ticker := time.NewTicker(bitrateCheckInterval)
 			defer ticker.Stop()
 
@@ -415,6 +429,7 @@ func (manager *WebRTCManagerCtx) CreatePeer(session types.Session, bitrate int,
 				changeVideoFromBitrate(targetBitrate)
 			}
 		}()
+	}
 
 	// data channel
 
@@ -436,7 +451,14 @@ func (manager *WebRTCManagerCtx) CreatePeer(session types.Session, bitrate int,
 			audioTrack.SetPaused(isPaused)
 		},
 		iceTrickle: manager.config.ICETrickle,
-		setVideoAuto: videoTrack.SetVideoAuto,
+		setVideoAuto: func(videoAuto bool) {
+			if manager.config.EstimatorEnabled {
+				videoTrack.SetVideoAuto(videoAuto)
+			} else {
+				logger.Warn().Msg("estimator is disabled, cannot change video auto")
+				videoTrack.SetVideoAuto(false) // ensure video auto is disabled
+			}
+		},
 		getVideoAuto: videoTrack.VideoAuto,
 	}
 

internal/websocket/handler/signal.go

@@ -40,6 +40,7 @@ func (h *MessageHandlerCtx) signalRequest(session types.Session, payload *messag
 		}
 
 		payload.Video = webrtcPeer.GetVideoID()
+		payload.VideoAuto = webrtcPeer.VideoAuto()
 	}
 
 	session.Send(