| /* |
| * Copyright (c) 2017 The WebRTC project authors. All Rights Reserved. |
| * |
| * Use of this source code is governed by a BSD-style license |
| * that can be found in the LICENSE file in the root of the source |
| * tree. An additional intellectual property rights grant can be found |
| * in the file PATENTS. All contributing project authors may |
| * be found in the AUTHORS file in the root of the source tree. |
| */ |
| |
| #include "modules/video_coding/codecs/test/test_config.h" |
| |
| #include <sstream> |
| |
| #include "modules/video_coding/include/video_codec_interface.h" |
| #include "rtc_base/checks.h" |
| #include "system_wrappers/include/cpu_info.h" |
| #include "test/video_codec_settings.h" |
| |
| namespace webrtc { |
| namespace test { |
| |
| namespace { |
| const int kBaseKeyFrameInterval = 3000; |
| |
| std::string CodecSpecificToString(const webrtc::VideoCodec& codec) { |
| std::stringstream ss; |
| switch (codec.codecType) { |
| case kVideoCodecVP8: |
| ss << "\n Complexity : " << codec.VP8().complexity; |
| ss << "\n Resilience : " << codec.VP8().resilience; |
| ss << "\n # temporal layers : " |
| << static_cast<int>(codec.VP8().numberOfTemporalLayers); |
| ss << "\n Denoising : " << codec.VP8().denoisingOn; |
| ss << "\n Error concealment : " << codec.VP8().errorConcealmentOn; |
| ss << "\n Automatic resize : " << codec.VP8().automaticResizeOn; |
| ss << "\n Frame dropping : " << codec.VP8().frameDroppingOn; |
| ss << "\n Key frame interval: " << codec.VP8().keyFrameInterval; |
| break; |
| case kVideoCodecVP9: |
| ss << "\n Complexity : " << codec.VP9().complexity; |
| ss << "\n Resilience : " << codec.VP9().resilienceOn; |
| ss << "\n # temporal layers : " |
| << static_cast<int>(codec.VP9().numberOfTemporalLayers); |
| ss << "\n Denoising : " << codec.VP9().denoisingOn; |
| ss << "\n Frame dropping : " << codec.VP9().frameDroppingOn; |
| ss << "\n Key frame interval: " << codec.VP9().keyFrameInterval; |
| ss << "\n Adaptive QP mode : " << codec.VP9().adaptiveQpMode; |
| ss << "\n Automatic resize : " << codec.VP9().automaticResizeOn; |
| ss << "\n # spatial layers : " |
| << static_cast<int>(codec.VP9().numberOfSpatialLayers); |
| ss << "\n Flexible mode : " << codec.VP9().flexibleMode; |
| break; |
| case kVideoCodecH264: |
| ss << "\n Frame dropping : " << codec.H264().frameDroppingOn; |
| ss << "\n Key frame interval: " << codec.H264().keyFrameInterval; |
| ss << "\n Profile : " << codec.H264().profile; |
| break; |
| default: |
| break; |
| } |
| ss << "\n"; |
| return ss.str(); |
| } |
| } // namespace |
| |
| void TestConfig::SetCodecSettings(VideoCodecType codec_type, |
| int num_temporal_layers, |
| bool error_concealment_on, |
| bool denoising_on, |
| bool frame_dropper_on, |
| bool spatial_resize_on, |
| bool resilience_on, |
| int width, |
| int height) { |
| webrtc::test::CodecSettings(codec_type, &codec_settings); |
| |
| // TODO(brandtr): Move the setting of |width| and |height| to the tests, and |
| // DCHECK that they are set before initializing the codec instead. |
| codec_settings.width = width; |
| codec_settings.height = height; |
| |
| switch (codec_settings.codecType) { |
| case kVideoCodecVP8: |
| codec_settings.VP8()->resilience = |
| resilience_on ? kResilientStream : kResilienceOff; |
| codec_settings.VP8()->numberOfTemporalLayers = num_temporal_layers; |
| codec_settings.VP8()->denoisingOn = denoising_on; |
| codec_settings.VP8()->errorConcealmentOn = error_concealment_on; |
| codec_settings.VP8()->automaticResizeOn = spatial_resize_on; |
| codec_settings.VP8()->frameDroppingOn = frame_dropper_on; |
| codec_settings.VP8()->keyFrameInterval = kBaseKeyFrameInterval; |
| break; |
| case kVideoCodecVP9: |
| codec_settings.VP9()->resilienceOn = resilience_on; |
| codec_settings.VP9()->numberOfTemporalLayers = num_temporal_layers; |
| codec_settings.VP9()->denoisingOn = denoising_on; |
| codec_settings.VP9()->frameDroppingOn = frame_dropper_on; |
| codec_settings.VP9()->keyFrameInterval = kBaseKeyFrameInterval; |
| codec_settings.VP9()->automaticResizeOn = spatial_resize_on; |
| break; |
| case kVideoCodecH264: |
| codec_settings.H264()->frameDroppingOn = frame_dropper_on; |
| codec_settings.H264()->keyFrameInterval = kBaseKeyFrameInterval; |
| break; |
| default: |
| RTC_NOTREACHED(); |
| break; |
| } |
| } |
| |
| int TestConfig::NumberOfCores() const { |
| return use_single_core ? 1 : CpuInfo::DetectNumberOfCores(); |
| } |
| |
| int TestConfig::NumberOfTemporalLayers() const { |
| if (codec_settings.codecType == kVideoCodecVP8) { |
| return codec_settings.VP8().numberOfTemporalLayers; |
| } else if (codec_settings.codecType == kVideoCodecVP9) { |
| return codec_settings.VP9().numberOfTemporalLayers; |
| } else { |
| return 1; |
| } |
| } |
| |
| int TestConfig::TemporalLayerForFrame(int frame_idx) const { |
| int tl = -1; |
| switch (NumberOfTemporalLayers()) { |
| case 1: |
| tl = 0; |
| break; |
| case 2: |
| // temporal layer 1: 1 3 |
| // temporal layer 0: 0 2 4 ... |
| tl = (frame_idx % 2 == 0) ? 0 : 1; |
| break; |
| case 3: |
| // temporal layer 2: 1 3 5 7 |
| // temporal layer 1: 2 6 |
| // temporal layer 0: 0 4 8 ... |
| if (frame_idx % 4 == 0) { |
| tl = 0; |
| } else if ((frame_idx + 2) % 4 == 0) { |
| tl = 1; |
| } else if ((frame_idx + 1) % 2 == 0) { |
| tl = 2; |
| } |
| break; |
| default: |
| RTC_NOTREACHED(); |
| break; |
| } |
| return tl; |
| } |
| |
| std::vector<FrameType> TestConfig::FrameTypeForFrame(int frame_idx) const { |
| if (keyframe_interval > 0 && (frame_idx % keyframe_interval == 0)) { |
| return {kVideoFrameKey}; |
| } |
| return {kVideoFrameDelta}; |
| } |
| |
| std::string TestConfig::ToString() const { |
| std::string codec_type = CodecTypeToPayloadString(codec_settings.codecType); |
| std::stringstream ss; |
| ss << "\n Filename : " << filename; |
| ss << "\n # CPU cores used : " << NumberOfCores(); |
| ss << "\n General:"; |
| ss << "\n Codec type : " << codec_type; |
| ss << "\n Start bitrate : " << codec_settings.startBitrate << " kbps"; |
| ss << "\n Max bitrate : " << codec_settings.maxBitrate << " kbps"; |
| ss << "\n Min bitrate : " << codec_settings.minBitrate << " kbps"; |
| ss << "\n Width : " << codec_settings.width; |
| ss << "\n Height : " << codec_settings.height; |
| ss << "\n Max frame rate : " << codec_settings.maxFramerate; |
| ss << "\n QPmax : " << codec_settings.qpMax; |
| ss << "\n " << codec_type << " specific: "; |
| ss << CodecSpecificToString(codec_settings); |
| return ss.str(); |
| } |
| |
| } // namespace test |
| } // namespace webrtc |