Line data    Source code

       1             : /*
       2             :  *  Copyright (c) 2016 The WebRTC project authors. All Rights Reserved.
       3             :  *
       4             :  *  Use of this source code is governed by a BSD-style license
       5             :  *  that can be found in the LICENSE file in the root of the source
       6             :  *  tree. An additional intellectual property rights grant can be found
       7             :  *  in the file PATENTS.  All contributing project authors may
       8             :  *  be found in the AUTHORS file in the root of the source tree.
       9             :  */
      10             : 
      11             : #ifndef WEBRTC_MODULES_VIDEO_CODING_UTILITY_SIMULCAST_RATE_ALLOCATOR_H_
      12             : #define WEBRTC_MODULES_VIDEO_CODING_UTILITY_SIMULCAST_RATE_ALLOCATOR_H_
      13             : 
      14             : #include <stdint.h>
      15             : 
      16             : #include <map>
      17             : #include <memory>
      18             : 
      19             : #include "webrtc/base/constructormagic.h"
      20             : #include "webrtc/common_video/include/video_bitrate_allocator.h"
      21             : #include "webrtc/modules/video_coding/codecs/vp8/temporal_layers.h"
      22             : #include "webrtc/video_encoder.h"
      23             : 
      24             : namespace webrtc {
      25             : 
      26           0 : class SimulcastRateAllocator : public VideoBitrateAllocator,
      27             :                                public TemporalLayersListener {
      28             :  public:
      29             :   explicit SimulcastRateAllocator(
      30             :       const VideoCodec& codec,
      31             :       std::unique_ptr<TemporalLayersFactory> tl_factory);
      32             : 
      33             :   void OnTemporalLayersCreated(int simulcast_id,
      34             :                                TemporalLayers* layers) override;
      35             : 
      36             :   BitrateAllocation GetAllocation(uint32_t total_bitrate_bps,
      37             :                                   uint32_t framerate) override;
      38             :   uint32_t GetPreferredBitrateBps(uint32_t framerate) override;
      39             :   const VideoCodec& GetCodec() const;
      40             : 
      41             :  private:
      42             :   const VideoCodec codec_;
      43             :   std::map<uint32_t, TemporalLayers*> temporal_layers_;
      44             :   std::unique_ptr<TemporalLayersFactory> tl_factory_;
      45             : 
      46             :   RTC_DISALLOW_COPY_AND_ASSIGN(SimulcastRateAllocator);
      47             : };
      48             : 
      49             : }  // namespace webrtc
      50             : 
      51             : #endif  // WEBRTC_MODULES_VIDEO_CODING_UTILITY_SIMULCAST_RATE_ALLOCATOR_H_