Ilya Nikolaevskiy | b0588e6 | 2018-08-27 14:12:27 +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 | #include "test/fake_vp8_encoder.h" |
| 12 | |
Erik Språng | 4529fbc | 2018-10-12 10:30:31 +0200 | [diff] [blame] | 13 | #include "api/video_codecs/create_vp8_temporal_layers.h" |
| 14 | #include "api/video_codecs/vp8_temporal_layers.h" |
Ilya Nikolaevskiy | b0588e6 | 2018-08-27 14:12:27 +0200 | [diff] [blame] | 15 | #include "common_types.h" // NOLINT(build/include) |
Ilya Nikolaevskiy | b0588e6 | 2018-08-27 14:12:27 +0200 | [diff] [blame] | 16 | #include "modules/video_coding/include/video_codec_interface.h" |
| 17 | #include "modules/video_coding/include/video_error_codes.h" |
| 18 | #include "modules/video_coding/utility/simulcast_utility.h" |
| 19 | #include "rtc_base/checks.h" |
| 20 | #include "rtc_base/logging.h" |
| 21 | #include "rtc_base/random.h" |
| 22 | #include "rtc_base/timeutils.h" |
| 23 | |
Per Kjellander | 841c912 | 2018-10-04 18:40:28 +0200 | [diff] [blame] | 24 | namespace { |
| 25 | |
| 26 | // Write width and height to the payload the same way as the real encoder does. |
| 27 | // It requires that |payload| has a size of at least kMinPayLoadHeaderLength. |
| 28 | void WriteFakeVp8(unsigned char* payload, |
| 29 | int width, |
| 30 | int height, |
| 31 | bool key_frame) { |
| 32 | payload[0] = key_frame ? 0 : 0x01; |
| 33 | |
| 34 | if (key_frame) { |
| 35 | payload[9] = (height & 0x3F00) >> 8; |
| 36 | payload[8] = (height & 0x00FF); |
| 37 | |
| 38 | payload[7] = (width & 0x3F00) >> 8; |
| 39 | payload[6] = (width & 0x00FF); |
| 40 | } |
| 41 | } |
| 42 | } // namespace |
| 43 | |
Ilya Nikolaevskiy | b0588e6 | 2018-08-27 14:12:27 +0200 | [diff] [blame] | 44 | namespace webrtc { |
| 45 | |
| 46 | namespace test { |
| 47 | |
| 48 | FakeVP8Encoder::FakeVP8Encoder(Clock* clock) |
| 49 | : FakeEncoder(clock), callback_(nullptr) { |
| 50 | FakeEncoder::RegisterEncodeCompleteCallback(this); |
| 51 | sequence_checker_.Detach(); |
| 52 | } |
| 53 | |
| 54 | int32_t FakeVP8Encoder::RegisterEncodeCompleteCallback( |
| 55 | EncodedImageCallback* callback) { |
| 56 | RTC_DCHECK_CALLED_SEQUENTIALLY(&sequence_checker_); |
| 57 | callback_ = callback; |
| 58 | return 0; |
| 59 | } |
| 60 | |
| 61 | int32_t FakeVP8Encoder::InitEncode(const VideoCodec* config, |
| 62 | int32_t number_of_cores, |
| 63 | size_t max_payload_size) { |
| 64 | RTC_DCHECK_CALLED_SEQUENTIALLY(&sequence_checker_); |
| 65 | auto result = |
| 66 | FakeEncoder::InitEncode(config, number_of_cores, max_payload_size); |
| 67 | if (result != WEBRTC_VIDEO_CODEC_OK) { |
| 68 | return result; |
| 69 | } |
| 70 | |
Erik Språng | 8abd56c | 2018-10-01 18:47:03 +0200 | [diff] [blame] | 71 | SetupTemporalLayers(*config); |
Ilya Nikolaevskiy | b0588e6 | 2018-08-27 14:12:27 +0200 | [diff] [blame] | 72 | |
| 73 | return WEBRTC_VIDEO_CODEC_OK; |
| 74 | } |
| 75 | |
| 76 | int32_t FakeVP8Encoder::Release() { |
| 77 | auto result = FakeEncoder::Release(); |
| 78 | sequence_checker_.Detach(); |
| 79 | return result; |
| 80 | } |
| 81 | |
Erik Språng | 8abd56c | 2018-10-01 18:47:03 +0200 | [diff] [blame] | 82 | void FakeVP8Encoder::SetupTemporalLayers(const VideoCodec& codec) { |
Ilya Nikolaevskiy | b0588e6 | 2018-08-27 14:12:27 +0200 | [diff] [blame] | 83 | RTC_DCHECK_CALLED_SEQUENTIALLY(&sequence_checker_); |
| 84 | |
Erik Språng | 8abd56c | 2018-10-01 18:47:03 +0200 | [diff] [blame] | 85 | int num_streams = SimulcastUtility::NumberOfSimulcastStreams(codec); |
Ilya Nikolaevskiy | b0588e6 | 2018-08-27 14:12:27 +0200 | [diff] [blame] | 86 | for (int i = 0; i < num_streams; ++i) { |
Erik Språng | 4529fbc | 2018-10-12 10:30:31 +0200 | [diff] [blame] | 87 | Vp8TemporalLayersType type; |
Erik Språng | 8abd56c | 2018-10-01 18:47:03 +0200 | [diff] [blame] | 88 | int num_temporal_layers = |
| 89 | SimulcastUtility::NumberOfTemporalLayers(codec, i); |
| 90 | if (SimulcastUtility::IsConferenceModeScreenshare(codec) && i == 0) { |
Erik Språng | 4529fbc | 2018-10-12 10:30:31 +0200 | [diff] [blame] | 91 | type = Vp8TemporalLayersType::kBitrateDynamic; |
Erik Språng | 8abd56c | 2018-10-01 18:47:03 +0200 | [diff] [blame] | 92 | // Legacy screenshare layers supports max 2 layers. |
| 93 | num_temporal_layers = std::max<int>(2, num_temporal_layers); |
| 94 | } else { |
Erik Språng | 4529fbc | 2018-10-12 10:30:31 +0200 | [diff] [blame] | 95 | type = Vp8TemporalLayersType::kFixedPattern; |
Erik Språng | 8abd56c | 2018-10-01 18:47:03 +0200 | [diff] [blame] | 96 | } |
Ilya Nikolaevskiy | b0588e6 | 2018-08-27 14:12:27 +0200 | [diff] [blame] | 97 | temporal_layers_.emplace_back( |
Erik Språng | 4529fbc | 2018-10-12 10:30:31 +0200 | [diff] [blame] | 98 | CreateVp8TemporalLayers(type, num_temporal_layers)); |
Ilya Nikolaevskiy | b0588e6 | 2018-08-27 14:12:27 +0200 | [diff] [blame] | 99 | } |
| 100 | } |
| 101 | |
Erik Språng | 59021ba | 2018-10-03 11:05:16 +0200 | [diff] [blame] | 102 | void FakeVP8Encoder::PopulateCodecSpecific(CodecSpecificInfo* codec_specific, |
| 103 | size_t size_bytes, |
| 104 | FrameType frame_type, |
| 105 | int stream_idx, |
| 106 | uint32_t timestamp) { |
Ilya Nikolaevskiy | b0588e6 | 2018-08-27 14:12:27 +0200 | [diff] [blame] | 107 | RTC_DCHECK_CALLED_SEQUENTIALLY(&sequence_checker_); |
| 108 | codec_specific->codecType = kVideoCodecVP8; |
Ilya Nikolaevskiy | b0588e6 | 2018-08-27 14:12:27 +0200 | [diff] [blame] | 109 | CodecSpecificInfoVP8* vp8Info = &(codec_specific->codecSpecific.VP8); |
Ilya Nikolaevskiy | b0588e6 | 2018-08-27 14:12:27 +0200 | [diff] [blame] | 110 | vp8Info->keyIdx = kNoKeyIdx; |
| 111 | vp8Info->nonReference = false; |
Erik Språng | 59021ba | 2018-10-03 11:05:16 +0200 | [diff] [blame] | 112 | temporal_layers_[stream_idx]->OnEncodeDone( |
| 113 | timestamp, size_bytes, frame_type == kVideoFrameKey, -1, vp8Info); |
Ilya Nikolaevskiy | b0588e6 | 2018-08-27 14:12:27 +0200 | [diff] [blame] | 114 | } |
| 115 | |
| 116 | EncodedImageCallback::Result FakeVP8Encoder::OnEncodedImage( |
| 117 | const EncodedImage& encoded_image, |
| 118 | const CodecSpecificInfo* codec_specific_info, |
| 119 | const RTPFragmentationHeader* fragments) { |
| 120 | RTC_DCHECK_CALLED_SEQUENTIALLY(&sequence_checker_); |
Niels Möller | d3b8c63 | 2018-08-27 15:33:42 +0200 | [diff] [blame] | 121 | uint8_t stream_idx = encoded_image.SpatialIndex().value_or(0); |
Ilya Nikolaevskiy | b0588e6 | 2018-08-27 14:12:27 +0200 | [diff] [blame] | 122 | CodecSpecificInfo overrided_specific_info; |
Erik Språng | 59021ba | 2018-10-03 11:05:16 +0200 | [diff] [blame] | 123 | temporal_layers_[stream_idx]->UpdateLayerConfig(encoded_image.Timestamp()); |
| 124 | PopulateCodecSpecific(&overrided_specific_info, encoded_image._length, |
Ilya Nikolaevskiy | b0588e6 | 2018-08-27 14:12:27 +0200 | [diff] [blame] | 125 | encoded_image._frameType, stream_idx, |
Niels Möller | 72bc8d6 | 2018-09-12 10:03:51 +0200 | [diff] [blame] | 126 | encoded_image.Timestamp()); |
Ilya Nikolaevskiy | b0588e6 | 2018-08-27 14:12:27 +0200 | [diff] [blame] | 127 | |
Per Kjellander | 841c912 | 2018-10-04 18:40:28 +0200 | [diff] [blame] | 128 | // Write width and height to the payload the same way as the real encoder |
| 129 | // does. |
| 130 | WriteFakeVp8(encoded_image._buffer, encoded_image._encodedWidth, |
| 131 | encoded_image._encodedHeight, |
| 132 | encoded_image._frameType == kVideoFrameKey); |
Ilya Nikolaevskiy | b0588e6 | 2018-08-27 14:12:27 +0200 | [diff] [blame] | 133 | return callback_->OnEncodedImage(encoded_image, &overrided_specific_info, |
| 134 | fragments); |
| 135 | } |
| 136 | |
Erik Språng | 86336a5 | 2018-11-15 15:38:05 +0100 | [diff] [blame^] | 137 | VideoEncoder::EncoderInfo FakeVP8Encoder::GetEncoderInfo() const { |
| 138 | EncoderInfo info; |
| 139 | info.implementation_name = "FakeVp8Encoder"; |
| 140 | return info; |
| 141 | } |
| 142 | |
Ilya Nikolaevskiy | b0588e6 | 2018-08-27 14:12:27 +0200 | [diff] [blame] | 143 | } // namespace test |
| 144 | } // namespace webrtc |