Niels Möller | 0327c2d | 2018-05-21 14:09:31 +0200 | [diff] [blame] | 1 | /* |
| 2 | * Copyright (c) 2018 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 API_VIDEO_VIDEO_STREAM_ENCODER_INTERFACE_H_ |
| 12 | #define API_VIDEO_VIDEO_STREAM_ENCODER_INTERFACE_H_ |
| 13 | |
| 14 | #include <vector> |
| 15 | |
| 16 | #include "api/rtpparameters.h" // For DegradationPreference. |
| 17 | #include "api/video/video_sink_interface.h" |
| 18 | #include "api/video/video_source_interface.h" |
| 19 | #include "api/video_codecs/video_encoder.h" |
| 20 | #include "api/video_codecs/video_encoder_config.h" |
| 21 | |
| 22 | namespace webrtc { |
| 23 | |
| 24 | // TODO(nisse): Move full declaration to api/. |
| 25 | class VideoBitrateAllocationObserver; |
| 26 | |
| 27 | // This interface represents a class responsible for creating and driving the |
| 28 | // encoder(s) for a single video stream. It is also responsible for adaptation |
| 29 | // decisions related to video quality, requesting reduced frame rate or |
| 30 | // resolution from the VideoSource when needed. |
| 31 | // TODO(bugs.webrtc.org/8830): This interface is under development. Changes |
| 32 | // under consideration include: |
| 33 | // |
| 34 | // 1. Taking out responsibility for adaptation decisions, instead only reporting |
| 35 | // per-frame measurements to the decision maker. |
| 36 | // |
| 37 | // 2. Moving responsibility for simulcast and for software fallback into this |
| 38 | // class. |
| 39 | class VideoStreamEncoderInterface : public rtc::VideoSinkInterface<VideoFrame> { |
| 40 | public: |
| 41 | // Interface for receiving encoded video frames and notifications about |
| 42 | // configuration changes. |
| 43 | class EncoderSink : public EncodedImageCallback { |
| 44 | public: |
| 45 | virtual void OnEncoderConfigurationChanged( |
| 46 | std::vector<VideoStream> streams, |
| 47 | int min_transmit_bitrate_bps) = 0; |
| 48 | }; |
| 49 | |
Niels Möller | 0327c2d | 2018-05-21 14:09:31 +0200 | [diff] [blame] | 50 | // Sets the source that will provide video frames to the VideoStreamEncoder's |
| 51 | // OnFrame method. |degradation_preference| control whether or not resolution |
| 52 | // or frame rate may be reduced. The VideoStreamEncoder registers itself with |
| 53 | // |source|, and signals adaptation decisions to the source in the form of |
| 54 | // VideoSinkWants. |
| 55 | // TODO(nisse): When adaptation logic is extracted from this class, |
| 56 | // it no longer needs to know the source. |
| 57 | virtual void SetSource( |
| 58 | rtc::VideoSourceInterface<VideoFrame>* source, |
| 59 | const DegradationPreference& degradation_preference) = 0; |
| 60 | |
| 61 | // Sets the |sink| that gets the encoded frames. |rotation_applied| means |
| 62 | // that the source must support rotation. Only set |rotation_applied| if the |
| 63 | // remote side does not support the rotation extension. |
| 64 | virtual void SetSink(EncoderSink* sink, bool rotation_applied) = 0; |
| 65 | |
| 66 | // Sets an initial bitrate, later overriden by OnBitrateUpdated. Mainly |
| 67 | // affects the resolution of the initial key frame: If incoming frames are |
| 68 | // larger than reasonable for the start bitrate, and scaling is enabled, |
| 69 | // VideoStreamEncoder asks the source to scale down and drops a few initial |
| 70 | // frames. |
| 71 | // TODO(nisse): This is a poor interface, and mixes bandwidth estimation and |
| 72 | // codec configuration in an undesired way. For the actual send bandwidth, we |
| 73 | // should always be somewhat conservative, but we may nevertheless want to let |
| 74 | // the application configure a more optimistic quality for the initial |
| 75 | // resolution. Should be replaced by a construction time setting. |
| 76 | virtual void SetStartBitrate(int start_bitrate_bps) = 0; |
| 77 | |
| 78 | // Request a key frame. Used for signalling from the remote receiver. |
| 79 | virtual void SendKeyFrame() = 0; |
| 80 | |
| 81 | // Set the currently estimated network properties. A |bitrate_bps| |
| 82 | // of zero pauses the encoder. |
| 83 | virtual void OnBitrateUpdated(uint32_t bitrate_bps, |
| 84 | uint8_t fraction_lost, |
| 85 | int64_t round_trip_time_ms) = 0; |
| 86 | |
| 87 | // Register observer for the bitrate allocation between the temporal |
| 88 | // and spatial layers. |
| 89 | virtual void SetBitrateAllocationObserver( |
| 90 | VideoBitrateAllocationObserver* bitrate_observer) = 0; |
| 91 | |
| 92 | // Creates and configures an encoder with the given |config|. The |
| 93 | // |max_data_payload_length| is used to support single NAL unit |
| 94 | // packetization for H.264. |
| 95 | virtual void ConfigureEncoder(VideoEncoderConfig config, |
| 96 | size_t max_data_payload_length) = 0; |
| 97 | |
| 98 | // Permanently stop encoding. After this method has returned, it is |
| 99 | // guaranteed that no encoded frames will be delivered to the sink. |
| 100 | virtual void Stop() = 0; |
| 101 | }; |
| 102 | |
| 103 | } // namespace webrtc |
| 104 | |
| 105 | #endif // API_VIDEO_VIDEO_STREAM_ENCODER_INTERFACE_H_ |