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 | |
Yves Gerey | 3e70781 | 2018-11-28 16:47:49 +0100 | [diff] [blame] | 13 | #include <algorithm> |
| 14 | |
| 15 | #include "absl/types/optional.h" |
Elad Alon | 370f93a | 2019-06-11 14:57:57 +0200 | [diff] [blame] | 16 | #include "api/video_codecs/video_encoder.h" |
Erik Språng | 4529fbc | 2018-10-12 10:30:31 +0200 | [diff] [blame] | 17 | #include "api/video_codecs/vp8_temporal_layers.h" |
Elad Alon | cde8ab2 | 2019-03-20 11:56:20 +0100 | [diff] [blame] | 18 | #include "api/video_codecs/vp8_temporal_layers_factory.h" |
Yves Gerey | 3e70781 | 2018-11-28 16:47:49 +0100 | [diff] [blame] | 19 | #include "modules/video_coding/codecs/interface/common_constants.h" |
Ilya Nikolaevskiy | b0588e6 | 2018-08-27 14:12:27 +0200 | [diff] [blame] | 20 | #include "modules/video_coding/include/video_codec_interface.h" |
| 21 | #include "modules/video_coding/include/video_error_codes.h" |
| 22 | #include "modules/video_coding/utility/simulcast_utility.h" |
Ilya Nikolaevskiy | b0588e6 | 2018-08-27 14:12:27 +0200 | [diff] [blame] | 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 | |
Niels Möller | d738071 | 2019-03-06 10:09:47 +0100 | [diff] [blame] | 48 | FakeVP8Encoder::FakeVP8Encoder(Clock* clock) : FakeEncoder(clock) { |
Ilya Nikolaevskiy | b0588e6 | 2018-08-27 14:12:27 +0200 | [diff] [blame] | 49 | sequence_checker_.Detach(); |
| 50 | } |
| 51 | |
Ilya Nikolaevskiy | b0588e6 | 2018-08-27 14:12:27 +0200 | [diff] [blame] | 52 | int32_t FakeVP8Encoder::InitEncode(const VideoCodec* config, |
Elad Alon | 370f93a | 2019-06-11 14:57:57 +0200 | [diff] [blame] | 53 | const Settings& settings) { |
Sebastian Jansson | b55015e | 2019-04-09 13:44:04 +0200 | [diff] [blame] | 54 | RTC_DCHECK_RUN_ON(&sequence_checker_); |
Elad Alon | 370f93a | 2019-06-11 14:57:57 +0200 | [diff] [blame] | 55 | auto result = FakeEncoder::InitEncode(config, settings); |
Ilya Nikolaevskiy | b0588e6 | 2018-08-27 14:12:27 +0200 | [diff] [blame] | 56 | if (result != WEBRTC_VIDEO_CODEC_OK) { |
| 57 | return result; |
| 58 | } |
| 59 | |
Elad Alon | cde8ab2 | 2019-03-20 11:56:20 +0100 | [diff] [blame] | 60 | Vp8TemporalLayersFactory factory; |
Elad Alon | 45befc5 | 2019-07-02 11:20:09 +0200 | [diff] [blame] | 61 | frame_buffer_controller_ = |
| 62 | factory.Create(*config, settings, &fec_controller_override_); |
Ilya Nikolaevskiy | b0588e6 | 2018-08-27 14:12:27 +0200 | [diff] [blame] | 63 | |
| 64 | return WEBRTC_VIDEO_CODEC_OK; |
| 65 | } |
| 66 | |
| 67 | int32_t FakeVP8Encoder::Release() { |
| 68 | auto result = FakeEncoder::Release(); |
| 69 | sequence_checker_.Detach(); |
| 70 | return result; |
| 71 | } |
| 72 | |
Erik Språng | 59021ba | 2018-10-03 11:05:16 +0200 | [diff] [blame] | 73 | void FakeVP8Encoder::PopulateCodecSpecific(CodecSpecificInfo* codec_specific, |
| 74 | size_t size_bytes, |
Niels Möller | 87e2d78 | 2019-03-07 10:18:23 +0100 | [diff] [blame] | 75 | VideoFrameType frame_type, |
Erik Språng | 59021ba | 2018-10-03 11:05:16 +0200 | [diff] [blame] | 76 | int stream_idx, |
| 77 | uint32_t timestamp) { |
Sebastian Jansson | b55015e | 2019-04-09 13:44:04 +0200 | [diff] [blame] | 78 | RTC_DCHECK_RUN_ON(&sequence_checker_); |
Ilya Nikolaevskiy | b0588e6 | 2018-08-27 14:12:27 +0200 | [diff] [blame] | 79 | codec_specific->codecType = kVideoCodecVP8; |
philipel | 9df3353 | 2019-03-04 16:37:50 +0100 | [diff] [blame] | 80 | codec_specific->codecSpecific.VP8.keyIdx = kNoKeyIdx; |
| 81 | codec_specific->codecSpecific.VP8.nonReference = false; |
Elad Alon | 6796ec2 | 2019-04-15 10:07:50 +0200 | [diff] [blame] | 82 | if (size_bytes > 0) { |
| 83 | frame_buffer_controller_->OnEncodeDone( |
| 84 | stream_idx, timestamp, size_bytes, |
| 85 | frame_type == VideoFrameType::kVideoFrameKey, -1, codec_specific); |
| 86 | } else { |
| 87 | frame_buffer_controller_->OnFrameDropped(stream_idx, timestamp); |
| 88 | } |
Ilya Nikolaevskiy | b0588e6 | 2018-08-27 14:12:27 +0200 | [diff] [blame] | 89 | } |
| 90 | |
Niels Möller | d738071 | 2019-03-06 10:09:47 +0100 | [diff] [blame] | 91 | std::unique_ptr<RTPFragmentationHeader> FakeVP8Encoder::EncodeHook( |
| 92 | EncodedImage* encoded_image, |
| 93 | CodecSpecificInfo* codec_specific) { |
Sebastian Jansson | b55015e | 2019-04-09 13:44:04 +0200 | [diff] [blame] | 94 | RTC_DCHECK_RUN_ON(&sequence_checker_); |
Niels Möller | d738071 | 2019-03-06 10:09:47 +0100 | [diff] [blame] | 95 | uint8_t stream_idx = encoded_image->SpatialIndex().value_or(0); |
Elad Alon | 979c442 | 2019-04-17 12:53:08 +0200 | [diff] [blame] | 96 | frame_buffer_controller_->NextFrameConfig(stream_idx, |
| 97 | encoded_image->Timestamp()); |
Niels Möller | d738071 | 2019-03-06 10:09:47 +0100 | [diff] [blame] | 98 | PopulateCodecSpecific(codec_specific, encoded_image->size(), |
| 99 | encoded_image->_frameType, stream_idx, |
| 100 | encoded_image->Timestamp()); |
Ilya Nikolaevskiy | b0588e6 | 2018-08-27 14:12:27 +0200 | [diff] [blame] | 101 | |
Per Kjellander | 841c912 | 2018-10-04 18:40:28 +0200 | [diff] [blame] | 102 | // Write width and height to the payload the same way as the real encoder |
| 103 | // does. |
Niels Möller | d738071 | 2019-03-06 10:09:47 +0100 | [diff] [blame] | 104 | WriteFakeVp8(encoded_image->data(), encoded_image->_encodedWidth, |
| 105 | encoded_image->_encodedHeight, |
Niels Möller | 8f7ce22 | 2019-03-21 15:43:58 +0100 | [diff] [blame] | 106 | encoded_image->_frameType == VideoFrameType::kVideoFrameKey); |
Niels Möller | d738071 | 2019-03-06 10:09:47 +0100 | [diff] [blame] | 107 | return nullptr; |
Ilya Nikolaevskiy | b0588e6 | 2018-08-27 14:12:27 +0200 | [diff] [blame] | 108 | } |
| 109 | |
Erik Språng | 86336a5 | 2018-11-15 15:38:05 +0100 | [diff] [blame] | 110 | VideoEncoder::EncoderInfo FakeVP8Encoder::GetEncoderInfo() const { |
| 111 | EncoderInfo info; |
| 112 | info.implementation_name = "FakeVp8Encoder"; |
| 113 | return info; |
| 114 | } |
| 115 | |
Ilya Nikolaevskiy | b0588e6 | 2018-08-27 14:12:27 +0200 | [diff] [blame] | 116 | } // namespace test |
| 117 | } // namespace webrtc |