| /* |
| * Copyright (c) 2018 The WebM 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 "./vpx_config.h" |
| #include "third_party/googletest/src/include/gtest/gtest.h" |
| #include "test/codec_factory.h" |
| #include "test/encode_test_driver.h" |
| #include "test/i420_video_source.h" |
| #include "test/svc_test.h" |
| #include "test/util.h" |
| #include "test/y4m_video_source.h" |
| #include "vpx/vpx_codec.h" |
| #include "vpx_ports/bitops.h" |
| |
| namespace { |
| |
| // Params: Inter layer prediction modes. |
| class SyncFrameOnePassCbrSvc : public ::svc_test::OnePassCbrSvc, |
| public ::libvpx_test::CodecTestWithParam<int> { |
| public: |
| SyncFrameOnePassCbrSvc() |
| : OnePassCbrSvc(GET_PARAM(0)), current_video_frame_(0), |
| frame_to_start_decode_(0), frame_to_sync_(0), mismatch_nframes_(0), |
| num_nonref_frames_(0), inter_layer_pred_mode_(GET_PARAM(1)), |
| decode_to_layer_before_sync_(-1), decode_to_layer_after_sync_(-1) { |
| SetMode(::libvpx_test::kRealTime); |
| memset(&svc_layer_sync_, 0, sizeof(svc_layer_sync_)); |
| } |
| |
| protected: |
| virtual ~SyncFrameOnePassCbrSvc() {} |
| |
| virtual void SetUp() { |
| InitializeConfig(); |
| speed_setting_ = 7; |
| } |
| |
| void Set2SpatialLayerConfig() { |
| SetConfig(); |
| cfg_.ss_number_layers = 2; |
| cfg_.ts_number_layers = 3; |
| svc_params_.scaling_factor_num[0] = 144; |
| svc_params_.scaling_factor_den[0] = 288; |
| svc_params_.scaling_factor_num[1] = 288; |
| svc_params_.scaling_factor_den[1] = 288; |
| number_spatial_layers_ = cfg_.ss_number_layers; |
| number_temporal_layers_ = cfg_.ts_number_layers; |
| } |
| |
| void Set3SpatialLayerConfig() { |
| SetConfig(); |
| cfg_.ss_number_layers = 3; |
| cfg_.ts_number_layers = 3; |
| svc_params_.scaling_factor_num[0] = 72; |
| svc_params_.scaling_factor_den[0] = 288; |
| svc_params_.scaling_factor_num[1] = 144; |
| svc_params_.scaling_factor_den[1] = 288; |
| svc_params_.scaling_factor_num[2] = 288; |
| svc_params_.scaling_factor_den[2] = 288; |
| number_spatial_layers_ = cfg_.ss_number_layers; |
| number_temporal_layers_ = cfg_.ts_number_layers; |
| } |
| |
| virtual bool DoDecode() const { |
| return current_video_frame_ >= frame_to_start_decode_; |
| } |
| |
| virtual void PreEncodeFrameHook(::libvpx_test::VideoSource *video, |
| ::libvpx_test::Encoder *encoder) { |
| current_video_frame_ = video->frame(); |
| PreEncodeFrameHookSetup(video, encoder); |
| if (video->frame() == 0) |
| encoder->Control(VP9E_SET_SVC_INTER_LAYER_PRED, inter_layer_pred_mode_); |
| if (video->frame() == frame_to_sync_) { |
| encoder->Control(VP9E_SET_SVC_SPATIAL_LAYER_SYNC, &svc_layer_sync_); |
| } |
| } |
| |
| #if CONFIG_VP9_DECODER |
| virtual void PreDecodeFrameHook(::libvpx_test::VideoSource *video, |
| ::libvpx_test::Decoder *decoder) { |
| if (video->frame() < frame_to_sync_) { |
| if (decode_to_layer_before_sync_ >= 0) |
| decoder->Control(VP9_DECODE_SVC_SPATIAL_LAYER, |
| decode_to_layer_before_sync_); |
| } else { |
| if (decode_to_layer_after_sync_ >= 0) |
| decoder->Control(VP9_DECODE_SVC_SPATIAL_LAYER, |
| decode_to_layer_after_sync_); |
| } |
| } |
| #endif |
| |
| virtual void FramePktHook(const vpx_codec_cx_pkt_t *pkt) { |
| // Keep track of number of non-reference frames, needed for mismatch check. |
| // Non-reference frames are top spatial and temporal layer frames, |
| // for TL > 0. |
| if (temporal_layer_id_ == number_temporal_layers_ - 1 && |
| temporal_layer_id_ > 0 && |
| pkt->data.frame.spatial_layer_encoded[number_spatial_layers_ - 1] && |
| current_video_frame_ >= frame_to_sync_) |
| num_nonref_frames_++; |
| } |
| |
| virtual void MismatchHook(const vpx_image_t * /*img1*/, |
| const vpx_image_t * /*img2*/) { |
| if (current_video_frame_ >= frame_to_sync_) ++mismatch_nframes_; |
| } |
| |
| unsigned int GetMismatchFrames() const { return mismatch_nframes_; } |
| |
| unsigned int current_video_frame_; |
| unsigned int frame_to_start_decode_; |
| unsigned int frame_to_sync_; |
| unsigned int mismatch_nframes_; |
| unsigned int num_nonref_frames_; |
| int inter_layer_pred_mode_; |
| int decode_to_layer_before_sync_; |
| int decode_to_layer_after_sync_; |
| vpx_svc_spatial_layer_sync_t svc_layer_sync_; |
| |
| private: |
| void SetConfig() { |
| cfg_.rc_buf_initial_sz = 500; |
| cfg_.rc_buf_optimal_sz = 500; |
| cfg_.rc_buf_sz = 1000; |
| cfg_.rc_min_quantizer = 0; |
| cfg_.rc_max_quantizer = 63; |
| cfg_.rc_end_usage = VPX_CBR; |
| cfg_.g_lag_in_frames = 0; |
| cfg_.g_error_resilient = 1; |
| cfg_.g_threads = 1; |
| cfg_.rc_dropframe_thresh = 30; |
| cfg_.kf_max_dist = 9999; |
| cfg_.ts_rate_decimator[0] = 4; |
| cfg_.ts_rate_decimator[1] = 2; |
| cfg_.ts_rate_decimator[2] = 1; |
| cfg_.temporal_layering_mode = 3; |
| } |
| }; |
| |
| // Test for sync layer for 1 pass CBR SVC: 3 spatial layers and |
| // 3 temporal layers. Only start decoding on the sync layer. |
| // Full sync: insert key frame on base layer. |
| TEST_P(SyncFrameOnePassCbrSvc, OnePassCbrSvc3SL3TLSyncFrameFull) { |
| Set3SpatialLayerConfig(); |
| // Sync is on base layer so the frame to sync and the frame to start decoding |
| // is the same. |
| frame_to_start_decode_ = 20; |
| frame_to_sync_ = 20; |
| decode_to_layer_before_sync_ = -1; |
| decode_to_layer_after_sync_ = 2; |
| |
| // Set up svc layer sync structure. |
| svc_layer_sync_.base_layer_intra_only = 0; |
| svc_layer_sync_.spatial_layer_sync[0] = 1; |
| |
| ::libvpx_test::Y4mVideoSource video("niklas_1280_720_30.y4m", 0, 60); |
| |
| cfg_.rc_target_bitrate = 600; |
| AssignLayerBitrates(); |
| ASSERT_NO_FATAL_FAILURE(RunLoop(&video)); |
| #if CONFIG_VP9_DECODER |
| // The non-reference frames are expected to be mismatched frames as the |
| // encoder will avoid loopfilter on these frames. |
| EXPECT_EQ(num_nonref_frames_, GetMismatchFrames()); |
| #endif |
| } |
| |
| // Test for sync layer for 1 pass CBR SVC: 2 spatial layers and |
| // 3 temporal layers. Decoding QVGA before sync frame and decode up to |
| // VGA on and after sync. |
| TEST_P(SyncFrameOnePassCbrSvc, OnePassCbrSvc2SL3TLSyncFrameVGA) { |
| Set2SpatialLayerConfig(); |
| frame_to_start_decode_ = 0; |
| frame_to_sync_ = 100; |
| decode_to_layer_before_sync_ = 0; |
| decode_to_layer_after_sync_ = 1; |
| |
| // Set up svc layer sync structure. |
| svc_layer_sync_.base_layer_intra_only = 0; |
| svc_layer_sync_.spatial_layer_sync[0] = 0; |
| svc_layer_sync_.spatial_layer_sync[1] = 1; |
| |
| ::libvpx_test::I420VideoSource video("niklas_640_480_30.yuv", 640, 480, 30, 1, |
| 0, 400); |
| cfg_.rc_target_bitrate = 400; |
| AssignLayerBitrates(); |
| ASSERT_NO_FATAL_FAILURE(RunLoop(&video)); |
| #if CONFIG_VP9_DECODER |
| // The non-reference frames are expected to be mismatched frames as the |
| // encoder will avoid loopfilter on these frames. |
| EXPECT_EQ(num_nonref_frames_, GetMismatchFrames()); |
| #endif |
| } |
| |
| // Test for sync layer for 1 pass CBR SVC: 3 spatial layers and |
| // 3 temporal layers. Decoding QVGA and VGA before sync frame and decode up to |
| // HD on and after sync. |
| TEST_P(SyncFrameOnePassCbrSvc, OnePassCbrSvc3SL3TLSyncFrameHD) { |
| Set3SpatialLayerConfig(); |
| frame_to_start_decode_ = 0; |
| frame_to_sync_ = 20; |
| decode_to_layer_before_sync_ = 1; |
| decode_to_layer_after_sync_ = 2; |
| |
| // Set up svc layer sync structure. |
| svc_layer_sync_.base_layer_intra_only = 0; |
| svc_layer_sync_.spatial_layer_sync[0] = 0; |
| svc_layer_sync_.spatial_layer_sync[1] = 0; |
| svc_layer_sync_.spatial_layer_sync[2] = 1; |
| |
| ::libvpx_test::Y4mVideoSource video("niklas_1280_720_30.y4m", 0, 60); |
| cfg_.rc_target_bitrate = 600; |
| AssignLayerBitrates(); |
| ASSERT_NO_FATAL_FAILURE(RunLoop(&video)); |
| #if CONFIG_VP9_DECODER |
| // The non-reference frames are expected to be mismatched frames as the |
| // encoder will avoid loopfilter on these frames. |
| EXPECT_EQ(num_nonref_frames_, GetMismatchFrames()); |
| #endif |
| } |
| |
| // Test for sync layer for 1 pass CBR SVC: 3 spatial layers and |
| // 3 temporal layers. Decoding QVGA before sync frame and decode up to |
| // HD on and after sync. |
| TEST_P(SyncFrameOnePassCbrSvc, OnePassCbrSvc3SL3TLSyncFrameVGAHD) { |
| Set3SpatialLayerConfig(); |
| frame_to_start_decode_ = 0; |
| frame_to_sync_ = 20; |
| decode_to_layer_before_sync_ = 0; |
| decode_to_layer_after_sync_ = 2; |
| |
| // Set up svc layer sync structure. |
| svc_layer_sync_.base_layer_intra_only = 0; |
| svc_layer_sync_.spatial_layer_sync[0] = 0; |
| svc_layer_sync_.spatial_layer_sync[1] = 1; |
| svc_layer_sync_.spatial_layer_sync[2] = 1; |
| |
| ::libvpx_test::Y4mVideoSource video("niklas_1280_720_30.y4m", 0, 60); |
| cfg_.rc_target_bitrate = 600; |
| AssignLayerBitrates(); |
| ASSERT_NO_FATAL_FAILURE(RunLoop(&video)); |
| #if CONFIG_VP9_DECODER |
| // The non-reference frames are expected to be mismatched frames as the |
| // encoder will avoid loopfilter on these frames. |
| EXPECT_EQ(num_nonref_frames_, GetMismatchFrames()); |
| #endif |
| } |
| |
| VP9_INSTANTIATE_TEST_CASE(SyncFrameOnePassCbrSvc, ::testing::Range(0, 3)); |
| |
| } // namespace |