Index: webrtc/modules/video_coding/codecs/test/videoprocessor_integrationtest.cc |
diff --git a/webrtc/modules/video_coding/codecs/test/videoprocessor_integrationtest.cc b/webrtc/modules/video_coding/codecs/test/videoprocessor_integrationtest.cc |
index e4b20e2c96f6a7642ab1f3cfb40377233d0bab75..88f5b0e3f8d1c234da80cc97c02acd453257631c 100644 |
--- a/webrtc/modules/video_coding/codecs/test/videoprocessor_integrationtest.cc |
+++ b/webrtc/modules/video_coding/codecs/test/videoprocessor_integrationtest.cc |
@@ -8,578 +8,10 @@ |
* be found in the AUTHORS file in the root of the source tree. |
*/ |
-#include <math.h> |
- |
-#include <memory> |
- |
-#include "webrtc/modules/video_coding/codecs/h264/include/h264.h" |
-#include "webrtc/modules/video_coding/codecs/test/packet_manipulator.h" |
-#include "webrtc/modules/video_coding/codecs/test/videoprocessor.h" |
-#include "webrtc/modules/video_coding/codecs/vp8/include/vp8.h" |
-#include "webrtc/modules/video_coding/codecs/vp8/include/vp8_common_types.h" |
-#include "webrtc/modules/video_coding/codecs/vp8/temporal_layers.h" |
-#include "webrtc/modules/video_coding/codecs/vp9/include/vp9.h" |
-#include "webrtc/modules/video_coding/include/video_codec_interface.h" |
-#include "webrtc/modules/video_coding/include/video_coding.h" |
-#include "webrtc/test/gtest.h" |
-#include "webrtc/test/testsupport/fileutils.h" |
-#include "webrtc/test/testsupport/frame_reader.h" |
-#include "webrtc/test/testsupport/frame_writer.h" |
-#include "webrtc/test/testsupport/metrics/video_metrics.h" |
-#include "webrtc/test/testsupport/packet_reader.h" |
-#include "webrtc/typedefs.h" |
+#include "webrtc/modules/video_coding/codecs/test/videoprocessor_integrationtest.h" |
namespace webrtc { |
-namespace { |
-// Maximum number of rate updates (i.e., calls to encoder to change bitrate |
-// and/or frame rate) for the current tests. |
-const int kMaxNumRateUpdates = 3; |
- |
-const int kPercTargetvsActualMismatch = 20; |
-const int kBaseKeyFrameInterval = 3000; |
- |
-// Codec and network settings. |
-struct CodecConfigPars { |
- VideoCodecType codec_type; |
- float packet_loss; |
- int num_temporal_layers; |
- int key_frame_interval; |
- bool error_concealment_on; |
- bool denoising_on; |
- bool frame_dropper_on; |
- bool spatial_resize_on; |
-}; |
- |
-// Quality metrics. |
-struct QualityMetrics { |
- double minimum_avg_psnr; |
- double minimum_min_psnr; |
- double minimum_avg_ssim; |
- double minimum_min_ssim; |
-}; |
- |
-// The sequence of bitrate and frame rate changes for the encoder, the frame |
-// number where the changes are made, and the total number of frames for the |
-// test. |
-struct RateProfile { |
- int target_bit_rate[kMaxNumRateUpdates]; |
- int input_frame_rate[kMaxNumRateUpdates]; |
- int frame_index_rate_update[kMaxNumRateUpdates + 1]; |
- int num_frames; |
-}; |
- |
-// Metrics for the rate control. The rate mismatch metrics are defined as |
-// percentages.|max_time_hit_target| is defined as number of frames, after a |
-// rate update is made to the encoder, for the encoder to reach within |
-// |kPercTargetvsActualMismatch| of new target rate. The metrics are defined for |
-// each rate update sequence. |
-struct RateControlMetrics { |
- int max_num_dropped_frames; |
- int max_key_frame_size_mismatch; |
- int max_delta_frame_size_mismatch; |
- int max_encoding_rate_mismatch; |
- int max_time_hit_target; |
- int num_spatial_resizes; |
- int num_key_frames; |
-}; |
- |
-// Sequence used is foreman (CIF): may be better to use VGA for resize test. |
-const int kCIFWidth = 352; |
-const int kCIFHeight = 288; |
-#if !defined(WEBRTC_IOS) |
-const int kNbrFramesShort = 100; // Some tests are run for shorter sequence. |
-#endif |
-const int kNbrFramesLong = 299; |
- |
-// Parameters from VP8 wrapper, which control target size of key frames. |
-const float kInitialBufferSize = 0.5f; |
-const float kOptimalBufferSize = 0.6f; |
-const float kScaleKeyFrameSize = 0.5f; |
- |
-void SetRateProfilePars(RateProfile* rate_profile, |
- int update_index, |
- int bit_rate, |
- int frame_rate, |
- int frame_index_rate_update) { |
- rate_profile->target_bit_rate[update_index] = bit_rate; |
- rate_profile->input_frame_rate[update_index] = frame_rate; |
- rate_profile->frame_index_rate_update[update_index] = frame_index_rate_update; |
-} |
- |
-void SetCodecParameters(CodecConfigPars* process_settings, |
- VideoCodecType codec_type, |
- float packet_loss, |
- int key_frame_interval, |
- int num_temporal_layers, |
- bool error_concealment_on, |
- bool denoising_on, |
- bool frame_dropper_on, |
- bool spatial_resize_on) { |
- process_settings->codec_type = codec_type; |
- process_settings->packet_loss = packet_loss; |
- process_settings->key_frame_interval = key_frame_interval; |
- process_settings->num_temporal_layers = num_temporal_layers, |
- process_settings->error_concealment_on = error_concealment_on; |
- process_settings->denoising_on = denoising_on; |
- process_settings->frame_dropper_on = frame_dropper_on; |
- process_settings->spatial_resize_on = spatial_resize_on; |
-} |
- |
-void SetQualityMetrics(QualityMetrics* quality_metrics, |
- double minimum_avg_psnr, |
- double minimum_min_psnr, |
- double minimum_avg_ssim, |
- double minimum_min_ssim) { |
- quality_metrics->minimum_avg_psnr = minimum_avg_psnr; |
- quality_metrics->minimum_min_psnr = minimum_min_psnr; |
- quality_metrics->minimum_avg_ssim = minimum_avg_ssim; |
- quality_metrics->minimum_min_ssim = minimum_min_ssim; |
-} |
- |
-void SetRateControlMetrics(RateControlMetrics* rc_metrics, |
- int update_index, |
- int max_num_dropped_frames, |
- int max_key_frame_size_mismatch, |
- int max_delta_frame_size_mismatch, |
- int max_encoding_rate_mismatch, |
- int max_time_hit_target, |
- int num_spatial_resizes, |
- int num_key_frames) { |
- rc_metrics[update_index].max_num_dropped_frames = max_num_dropped_frames; |
- rc_metrics[update_index].max_key_frame_size_mismatch = |
- max_key_frame_size_mismatch; |
- rc_metrics[update_index].max_delta_frame_size_mismatch = |
- max_delta_frame_size_mismatch; |
- rc_metrics[update_index].max_encoding_rate_mismatch = |
- max_encoding_rate_mismatch; |
- rc_metrics[update_index].max_time_hit_target = max_time_hit_target; |
- rc_metrics[update_index].num_spatial_resizes = num_spatial_resizes; |
- rc_metrics[update_index].num_key_frames = num_key_frames; |
-} |
-} // namespace |
- |
-// Integration test for video processor. Encodes+decodes a clip and |
-// writes it to the output directory. After completion, quality metrics |
-// (PSNR and SSIM) and rate control metrics are computed to verify that the |
-// quality and encoder response is acceptable. The rate control tests allow us |
-// to verify the behavior for changing bitrate, changing frame rate, frame |
-// dropping/spatial resize, and temporal layers. The limits for the rate |
-// control metrics are set to be fairly conservative, so failure should only |
-// happen when some significant regression or breakdown occurs. |
-class VideoProcessorIntegrationTest : public testing::Test { |
- protected: |
- std::unique_ptr<VideoEncoder> encoder_; |
- std::unique_ptr<VideoDecoder> decoder_; |
- std::unique_ptr<test::FrameReader> frame_reader_; |
- std::unique_ptr<test::FrameWriter> frame_writer_; |
- test::PacketReader packet_reader_; |
- std::unique_ptr<test::PacketManipulator> packet_manipulator_; |
- test::Stats stats_; |
- test::TestConfig config_; |
- VideoCodec codec_settings_; |
- std::unique_ptr<test::VideoProcessor> processor_; |
- TemporalLayersFactory tl_factory_; |
- |
- // Quantities defined/updated for every encoder rate update. |
- // Some quantities defined per temporal layer (at most 3 layers in this test). |
- int num_frames_per_update_[3]; |
- float sum_frame_size_mismatch_[3]; |
- float sum_encoded_frame_size_[3]; |
- float encoding_bitrate_[3]; |
- float per_frame_bandwidth_[3]; |
- float bit_rate_layer_[3]; |
- float frame_rate_layer_[3]; |
- int num_frames_total_; |
- float sum_encoded_frame_size_total_; |
- float encoding_bitrate_total_; |
- float perc_encoding_rate_mismatch_; |
- int num_frames_to_hit_target_; |
- bool encoding_rate_within_target_; |
- int bit_rate_; |
- int frame_rate_; |
- int layer_; |
- float target_size_key_frame_initial_; |
- float target_size_key_frame_; |
- float sum_key_frame_size_mismatch_; |
- int num_key_frames_; |
- float start_bitrate_; |
- |
- // Codec and network settings. |
- VideoCodecType codec_type_; |
- float packet_loss_; |
- int num_temporal_layers_; |
- int key_frame_interval_; |
- bool error_concealment_on_; |
- bool denoising_on_; |
- bool frame_dropper_on_; |
- bool spatial_resize_on_; |
- |
- VideoProcessorIntegrationTest() {} |
- virtual ~VideoProcessorIntegrationTest() {} |
- |
- void SetUpCodecConfig() { |
- if (codec_type_ == kVideoCodecH264) { |
- encoder_.reset(H264Encoder::Create(cricket::VideoCodec("H264"))); |
- decoder_.reset(H264Decoder::Create()); |
- VideoCodingModule::Codec(kVideoCodecH264, &codec_settings_); |
- } else if (codec_type_ == kVideoCodecVP8) { |
- encoder_.reset(VP8Encoder::Create()); |
- decoder_.reset(VP8Decoder::Create()); |
- VideoCodingModule::Codec(kVideoCodecVP8, &codec_settings_); |
- } else if (codec_type_ == kVideoCodecVP9) { |
- encoder_.reset(VP9Encoder::Create()); |
- decoder_.reset(VP9Decoder::Create()); |
- VideoCodingModule::Codec(kVideoCodecVP9, &codec_settings_); |
- } |
- |
- // CIF is currently used for all tests below. |
- // Setup the TestConfig struct for processing of a clip in CIF resolution. |
- config_.input_filename = webrtc::test::ResourcePath("foreman_cif", "yuv"); |
- |
- // Generate an output filename in a safe way. |
- config_.output_filename = webrtc::test::TempFilename( |
- webrtc::test::OutputPath(), "videoprocessor_integrationtest"); |
- config_.frame_length_in_bytes = |
- CalcBufferSize(kI420, kCIFWidth, kCIFHeight); |
- config_.verbose = false; |
- // Only allow encoder/decoder to use single core, for predictability. |
- config_.use_single_core = true; |
- // Key frame interval and packet loss are set for each test. |
- config_.keyframe_interval = key_frame_interval_; |
- config_.networking_config.packet_loss_probability = packet_loss_; |
- |
- // Configure codec settings. |
- config_.codec_settings = &codec_settings_; |
- config_.codec_settings->startBitrate = start_bitrate_; |
- config_.codec_settings->width = kCIFWidth; |
- config_.codec_settings->height = kCIFHeight; |
- |
- // These features may be set depending on the test. |
- switch (config_.codec_settings->codecType) { |
- case kVideoCodecH264: |
- config_.codec_settings->H264()->frameDroppingOn = frame_dropper_on_; |
- config_.codec_settings->H264()->keyFrameInterval = |
- kBaseKeyFrameInterval; |
- break; |
- case kVideoCodecVP8: |
- config_.codec_settings->VP8()->errorConcealmentOn = |
- error_concealment_on_; |
- config_.codec_settings->VP8()->denoisingOn = denoising_on_; |
- config_.codec_settings->VP8()->numberOfTemporalLayers = |
- num_temporal_layers_; |
- config_.codec_settings->VP8()->frameDroppingOn = frame_dropper_on_; |
- config_.codec_settings->VP8()->automaticResizeOn = spatial_resize_on_; |
- config_.codec_settings->VP8()->keyFrameInterval = kBaseKeyFrameInterval; |
- break; |
- case kVideoCodecVP9: |
- config_.codec_settings->VP9()->denoisingOn = denoising_on_; |
- config_.codec_settings->VP9()->numberOfTemporalLayers = |
- num_temporal_layers_; |
- config_.codec_settings->VP9()->frameDroppingOn = frame_dropper_on_; |
- config_.codec_settings->VP9()->automaticResizeOn = spatial_resize_on_; |
- config_.codec_settings->VP9()->keyFrameInterval = kBaseKeyFrameInterval; |
- break; |
- default: |
- assert(false); |
- break; |
- } |
- frame_reader_.reset(new test::FrameReaderImpl( |
- config_.input_filename, config_.codec_settings->width, |
- config_.codec_settings->height)); |
- frame_writer_.reset(new test::FrameWriterImpl( |
- config_.output_filename, config_.frame_length_in_bytes)); |
- ASSERT_TRUE(frame_reader_->Init()); |
- ASSERT_TRUE(frame_writer_->Init()); |
- |
- packet_manipulator_.reset(new test::PacketManipulatorImpl( |
- &packet_reader_, config_.networking_config, config_.verbose)); |
- processor_.reset(new test::VideoProcessorImpl( |
- encoder_.get(), decoder_.get(), frame_reader_.get(), |
- frame_writer_.get(), packet_manipulator_.get(), config_, &stats_)); |
- ASSERT_TRUE(processor_->Init()); |
- } |
- |
- // Reset quantities after each encoder update, update the target |
- // per-frame bandwidth. |
- void ResetRateControlMetrics(int num_frames) { |
- for (int i = 0; i < num_temporal_layers_; i++) { |
- num_frames_per_update_[i] = 0; |
- sum_frame_size_mismatch_[i] = 0.0f; |
- sum_encoded_frame_size_[i] = 0.0f; |
- encoding_bitrate_[i] = 0.0f; |
- // Update layer per-frame-bandwidth. |
- per_frame_bandwidth_[i] = static_cast<float>(bit_rate_layer_[i]) / |
- static_cast<float>(frame_rate_layer_[i]); |
- } |
- // Set maximum size of key frames, following setting in the VP8 wrapper. |
- float max_key_size = kScaleKeyFrameSize * kOptimalBufferSize * frame_rate_; |
- // We don't know exact target size of the key frames (except for first one), |
- // but the minimum in libvpx is ~|3 * per_frame_bandwidth| and maximum is |
- // set by |max_key_size_ * per_frame_bandwidth|. Take middle point/average |
- // as reference for mismatch. Note key frames always correspond to base |
- // layer frame in this test. |
- target_size_key_frame_ = 0.5 * (3 + max_key_size) * per_frame_bandwidth_[0]; |
- num_frames_total_ = 0; |
- sum_encoded_frame_size_total_ = 0.0f; |
- encoding_bitrate_total_ = 0.0f; |
- perc_encoding_rate_mismatch_ = 0.0f; |
- num_frames_to_hit_target_ = num_frames; |
- encoding_rate_within_target_ = false; |
- sum_key_frame_size_mismatch_ = 0.0; |
- num_key_frames_ = 0; |
- } |
- |
- // For every encoded frame, update the rate control metrics. |
- void UpdateRateControlMetrics(int frame_num, FrameType frame_type) { |
- float encoded_size_kbits = processor_->EncodedFrameSize() * 8.0f / 1000.0f; |
- // Update layer data. |
- // Update rate mismatch relative to per-frame bandwidth for delta frames. |
- if (frame_type == kVideoFrameDelta) { |
- // TODO(marpan): Should we count dropped (zero size) frames in mismatch? |
- sum_frame_size_mismatch_[layer_] += |
- fabs(encoded_size_kbits - per_frame_bandwidth_[layer_]) / |
- per_frame_bandwidth_[layer_]; |
- } else { |
- float target_size = (frame_num == 1) ? target_size_key_frame_initial_ |
- : target_size_key_frame_; |
- sum_key_frame_size_mismatch_ += |
- fabs(encoded_size_kbits - target_size) / target_size; |
- num_key_frames_ += 1; |
- } |
- sum_encoded_frame_size_[layer_] += encoded_size_kbits; |
- // Encoding bitrate per layer: from the start of the update/run to the |
- // current frame. |
- encoding_bitrate_[layer_] = sum_encoded_frame_size_[layer_] * |
- frame_rate_layer_[layer_] / |
- num_frames_per_update_[layer_]; |
- // Total encoding rate: from the start of the update/run to current frame. |
- sum_encoded_frame_size_total_ += encoded_size_kbits; |
- encoding_bitrate_total_ = |
- sum_encoded_frame_size_total_ * frame_rate_ / num_frames_total_; |
- perc_encoding_rate_mismatch_ = |
- 100 * fabs(encoding_bitrate_total_ - bit_rate_) / bit_rate_; |
- if (perc_encoding_rate_mismatch_ < kPercTargetvsActualMismatch && |
- !encoding_rate_within_target_) { |
- num_frames_to_hit_target_ = num_frames_total_; |
- encoding_rate_within_target_ = true; |
- } |
- } |
- |
- // Verify expected behavior of rate control and print out data. |
- void VerifyRateControl(int update_index, |
- int max_key_frame_size_mismatch, |
- int max_delta_frame_size_mismatch, |
- int max_encoding_rate_mismatch, |
- int max_time_hit_target, |
- int max_num_dropped_frames, |
- int num_spatial_resizes, |
- int num_key_frames) { |
- int num_dropped_frames = processor_->NumberDroppedFrames(); |
- int num_resize_actions = processor_->NumberSpatialResizes(); |
- printf( |
- "For update #: %d,\n " |
- " Target Bitrate: %d,\n" |
- " Encoding bitrate: %f,\n" |
- " Frame rate: %d \n", |
- update_index, bit_rate_, encoding_bitrate_total_, frame_rate_); |
- printf( |
- " Number of frames to approach target rate: %d, \n" |
- " Number of dropped frames: %d, \n" |
- " Number of spatial resizes: %d, \n", |
- num_frames_to_hit_target_, num_dropped_frames, num_resize_actions); |
- EXPECT_LE(perc_encoding_rate_mismatch_, max_encoding_rate_mismatch); |
- if (num_key_frames_ > 0) { |
- int perc_key_frame_size_mismatch = |
- 100 * sum_key_frame_size_mismatch_ / num_key_frames_; |
- printf( |
- " Number of Key frames: %d \n" |
- " Key frame rate mismatch: %d \n", |
- num_key_frames_, perc_key_frame_size_mismatch); |
- EXPECT_LE(perc_key_frame_size_mismatch, max_key_frame_size_mismatch); |
- } |
- printf("\n"); |
- printf("Rates statistics for Layer data \n"); |
- for (int i = 0; i < num_temporal_layers_; i++) { |
- printf("Temporal layer #%d \n", i); |
- int perc_frame_size_mismatch = |
- 100 * sum_frame_size_mismatch_[i] / num_frames_per_update_[i]; |
- int perc_encoding_rate_mismatch = |
- 100 * fabs(encoding_bitrate_[i] - bit_rate_layer_[i]) / |
- bit_rate_layer_[i]; |
- printf( |
- " Target Layer Bit rate: %f \n" |
- " Layer frame rate: %f, \n" |
- " Layer per frame bandwidth: %f, \n" |
- " Layer Encoding bit rate: %f, \n" |
- " Layer Percent frame size mismatch: %d, \n" |
- " Layer Percent encoding rate mismatch: %d, \n" |
- " Number of frame processed per layer: %d \n", |
- bit_rate_layer_[i], frame_rate_layer_[i], per_frame_bandwidth_[i], |
- encoding_bitrate_[i], perc_frame_size_mismatch, |
- perc_encoding_rate_mismatch, num_frames_per_update_[i]); |
- EXPECT_LE(perc_frame_size_mismatch, max_delta_frame_size_mismatch); |
- EXPECT_LE(perc_encoding_rate_mismatch, max_encoding_rate_mismatch); |
- } |
- printf("\n"); |
- EXPECT_LE(num_frames_to_hit_target_, max_time_hit_target); |
- EXPECT_LE(num_dropped_frames, max_num_dropped_frames); |
- EXPECT_EQ(num_resize_actions, num_spatial_resizes); |
- EXPECT_EQ(num_key_frames_, num_key_frames); |
- } |
- |
- // Layer index corresponding to frame number, for up to 3 layers. |
- void LayerIndexForFrame(int frame_number) { |
- if (num_temporal_layers_ == 1) { |
- layer_ = 0; |
- } else if (num_temporal_layers_ == 2) { |
- // layer 0: 0 2 4 ... |
- // layer 1: 1 3 |
- if (frame_number % 2 == 0) { |
- layer_ = 0; |
- } else { |
- layer_ = 1; |
- } |
- } else if (num_temporal_layers_ == 3) { |
- // layer 0: 0 4 8 ... |
- // layer 1: 2 6 |
- // layer 2: 1 3 5 7 |
- if (frame_number % 4 == 0) { |
- layer_ = 0; |
- } else if ((frame_number + 2) % 4 == 0) { |
- layer_ = 1; |
- } else if ((frame_number + 1) % 2 == 0) { |
- layer_ = 2; |
- } |
- } else { |
- assert(false); // Only up to 3 layers. |
- } |
- } |
- |
- // Set the bitrate and frame rate per layer, for up to 3 layers. |
- void SetLayerRates() { |
- assert(num_temporal_layers_ <= 3); |
- for (int i = 0; i < num_temporal_layers_; i++) { |
- float bit_rate_ratio = |
- kVp8LayerRateAlloction[num_temporal_layers_ - 1][i]; |
- if (i > 0) { |
- float bit_rate_delta_ratio = |
- kVp8LayerRateAlloction[num_temporal_layers_ - 1][i] - |
- kVp8LayerRateAlloction[num_temporal_layers_ - 1][i - 1]; |
- bit_rate_layer_[i] = bit_rate_ * bit_rate_delta_ratio; |
- } else { |
- bit_rate_layer_[i] = bit_rate_ * bit_rate_ratio; |
- } |
- frame_rate_layer_[i] = |
- frame_rate_ / static_cast<float>(1 << (num_temporal_layers_ - 1)); |
- } |
- if (num_temporal_layers_ == 3) { |
- frame_rate_layer_[2] = frame_rate_ / 2.0f; |
- } |
- } |
- |
- // Processes all frames in the clip and verifies the result. |
- void ProcessFramesAndVerify(QualityMetrics quality_metrics, |
- RateProfile rate_profile, |
- CodecConfigPars process, |
- RateControlMetrics* rc_metrics) { |
- // Codec/config settings. |
- codec_type_ = process.codec_type; |
- start_bitrate_ = rate_profile.target_bit_rate[0]; |
- packet_loss_ = process.packet_loss; |
- key_frame_interval_ = process.key_frame_interval; |
- num_temporal_layers_ = process.num_temporal_layers; |
- error_concealment_on_ = process.error_concealment_on; |
- denoising_on_ = process.denoising_on; |
- frame_dropper_on_ = process.frame_dropper_on; |
- spatial_resize_on_ = process.spatial_resize_on; |
- SetUpCodecConfig(); |
- // Update the layers and the codec with the initial rates. |
- bit_rate_ = rate_profile.target_bit_rate[0]; |
- frame_rate_ = rate_profile.input_frame_rate[0]; |
- SetLayerRates(); |
- // Set the initial target size for key frame. |
- target_size_key_frame_initial_ = |
- 0.5 * kInitialBufferSize * bit_rate_layer_[0]; |
- processor_->SetRates(bit_rate_, frame_rate_); |
- // Process each frame, up to |num_frames|. |
- int num_frames = rate_profile.num_frames; |
- int update_index = 0; |
- ResetRateControlMetrics( |
- rate_profile.frame_index_rate_update[update_index + 1]); |
- int frame_number = 0; |
- FrameType frame_type = kVideoFrameDelta; |
- while (processor_->ProcessFrame(frame_number) && |
- frame_number < num_frames) { |
- // Get the layer index for the frame |frame_number|. |
- LayerIndexForFrame(frame_number); |
- // Get the frame_type. |
- frame_type = processor_->EncodedFrameType(); |
- // Counter for whole sequence run. |
- ++frame_number; |
- // Counters for each rate update. |
- ++num_frames_per_update_[layer_]; |
- ++num_frames_total_; |
- UpdateRateControlMetrics(frame_number, frame_type); |
- // If we hit another/next update, verify stats for current state and |
- // update layers and codec with new rates. |
- if (frame_number == |
- rate_profile.frame_index_rate_update[update_index + 1]) { |
- VerifyRateControl( |
- update_index, rc_metrics[update_index].max_key_frame_size_mismatch, |
- rc_metrics[update_index].max_delta_frame_size_mismatch, |
- rc_metrics[update_index].max_encoding_rate_mismatch, |
- rc_metrics[update_index].max_time_hit_target, |
- rc_metrics[update_index].max_num_dropped_frames, |
- rc_metrics[update_index].num_spatial_resizes, |
- rc_metrics[update_index].num_key_frames); |
- // Update layer rates and the codec with new rates. |
- ++update_index; |
- bit_rate_ = rate_profile.target_bit_rate[update_index]; |
- frame_rate_ = rate_profile.input_frame_rate[update_index]; |
- SetLayerRates(); |
- ResetRateControlMetrics( |
- rate_profile.frame_index_rate_update[update_index + 1]); |
- processor_->SetRates(bit_rate_, frame_rate_); |
- } |
- } |
- VerifyRateControl(update_index, |
- rc_metrics[update_index].max_key_frame_size_mismatch, |
- rc_metrics[update_index].max_delta_frame_size_mismatch, |
- rc_metrics[update_index].max_encoding_rate_mismatch, |
- rc_metrics[update_index].max_time_hit_target, |
- rc_metrics[update_index].max_num_dropped_frames, |
- rc_metrics[update_index].num_spatial_resizes, |
- rc_metrics[update_index].num_key_frames); |
- EXPECT_EQ(num_frames, frame_number); |
- EXPECT_EQ(num_frames + 1, static_cast<int>(stats_.stats_.size())); |
- |
- // Release encoder and decoder to make sure they have finished processing: |
- EXPECT_EQ(WEBRTC_VIDEO_CODEC_OK, encoder_->Release()); |
- EXPECT_EQ(WEBRTC_VIDEO_CODEC_OK, decoder_->Release()); |
- // Close the files before we start using them for SSIM/PSNR calculations. |
- frame_reader_->Close(); |
- frame_writer_->Close(); |
- |
- // TODO(marpan): should compute these quality metrics per SetRates update. |
- test::QualityMetricsResult psnr_result, ssim_result; |
- EXPECT_EQ(0, test::I420MetricsFromFiles(config_.input_filename.c_str(), |
- config_.output_filename.c_str(), |
- config_.codec_settings->width, |
- config_.codec_settings->height, |
- &psnr_result, &ssim_result)); |
- printf("PSNR avg: %f, min: %f\nSSIM avg: %f, min: %f\n", |
- psnr_result.average, psnr_result.min, ssim_result.average, |
- ssim_result.min); |
- stats_.PrintSummary(); |
- EXPECT_GT(psnr_result.average, quality_metrics.minimum_avg_psnr); |
- EXPECT_GT(psnr_result.min, quality_metrics.minimum_min_psnr); |
- EXPECT_GT(ssim_result.average, quality_metrics.minimum_avg_ssim); |
- EXPECT_GT(ssim_result.min, quality_metrics.minimum_min_ssim); |
- if (remove(config_.output_filename.c_str()) < 0) { |
- fprintf(stderr, "Failed to remove temporary file!\n"); |
- } |
- } |
-}; |
+namespace test { |
#if defined(WEBRTC_VIDEOPROCESSOR_H264_TESTS) |
@@ -961,4 +393,5 @@ TEST_F(VideoProcessorIntegrationTest, MAYBE_ProcessNoLossTemporalLayersVP8) { |
ProcessFramesAndVerify(quality_metrics, rate_profile, process_settings, |
rc_metrics); |
} |
+} // namespace test |
} // namespace webrtc |