| /* |
| * Copyright (c) 2021, Alliance for Open Media. All rights reserved |
| * |
| * This source code is subject to the terms of the BSD 3-Clause Clear License |
| * and the Alliance for Open Media Patent License 1.0. If the BSD 3-Clause Clear |
| * License was not distributed with this source code in the LICENSE file, you |
| * can obtain it at aomedia.org/license/software-license/bsd-3-c-c/. If the |
| * Alliance for Open Media Patent License 1.0 was not distributed with this |
| * source code in the PATENTS file, you can obtain it at |
| * aomedia.org/license/patent-license/. |
| */ |
| |
| #include <climits> |
| #include <tuple> |
| #include <vector> |
| #include "third_party/googletest/src/googletest/include/gtest/gtest.h" |
| #include "test/codec_factory.h" |
| #include "test/encode_test_driver.h" |
| #include "test/i420_video_source.h" |
| #include "test/util.h" |
| |
| namespace { |
| |
| const int kTestMode = 0; |
| const int kTileCols = 1; |
| const int kTileRows = 2; |
| |
| typedef std::tuple<libaom_test::TestMode, int, int> SuperframeTestParam; |
| |
| class SuperframeTest |
| : public ::libaom_test::CodecTestWithParam<SuperframeTestParam>, |
| public ::libaom_test::EncoderTest { |
| protected: |
| SuperframeTest() : EncoderTest(GET_PARAM(0)), last_sf_pts_(0) {} |
| virtual ~SuperframeTest() {} |
| |
| virtual void SetUp() { |
| InitializeConfig(); |
| const SuperframeTestParam input = GET_PARAM(1); |
| const libaom_test::TestMode mode = std::get<kTestMode>(input); |
| SetMode(mode); |
| sf_count_ = 0; |
| sf_count_max_ = INT_MAX; |
| n_tile_cols_ = std::get<kTileCols>(input); |
| n_tile_rows_ = std::get<kTileRows>(input); |
| } |
| |
| virtual void PreEncodeFrameHook(libaom_test::VideoSource *video, |
| libaom_test::Encoder *encoder) { |
| if (video->frame() == 0) { |
| encoder->Control(AOME_SET_ENABLEAUTOALTREF, 1); |
| encoder->Control(AOME_SET_CPUUSED, 2); |
| encoder->Control(AV1E_SET_TILE_COLUMNS, n_tile_cols_); |
| encoder->Control(AV1E_SET_TILE_ROWS, n_tile_rows_); |
| } |
| } |
| |
| virtual const aom_codec_cx_pkt_t *MutateEncoderOutputHook( |
| const aom_codec_cx_pkt_t *pkt) { |
| if (pkt->kind != AOM_CODEC_CX_FRAME_PKT |
| #if CONFIG_OUTPUT_FRAME_BASED_ON_ORDER_HINT |
| && pkt->kind != AOM_CODEC_CX_FRAME_NULL_PKT |
| #endif // CONFIG_OUTPUT_FRAME_BASED_ON_ORDER_HINT |
| ) { |
| return pkt; |
| } |
| |
| const uint8_t *buffer = reinterpret_cast<uint8_t *>(pkt->data.frame.buf); |
| const uint8_t marker = buffer[0]; |
| const int frames = (marker & 0x7) + 1; |
| const int mag = ((marker >> 3) & 3) + 1; |
| const unsigned int index_sz = 2 + mag * (frames - 1); |
| if ((marker & 0xe0) == 0xc0 && pkt->data.frame.sz >= index_sz && |
| buffer[index_sz - 1] == marker) { |
| // frame is a superframe. strip off the index. |
| modified_buf_.resize(pkt->data.frame.sz - index_sz); |
| memcpy(&modified_buf_[0], (uint8_t *)pkt->data.frame.buf + index_sz, |
| pkt->data.frame.sz - index_sz); |
| modified_pkt_ = *pkt; |
| modified_pkt_.data.frame.buf = &modified_buf_[0]; |
| modified_pkt_.data.frame.sz -= index_sz; |
| |
| sf_count_++; |
| last_sf_pts_ = pkt->data.frame.pts; |
| return &modified_pkt_; |
| } |
| |
| // Make sure we do a few frames after the last SF |
| abort_ |= |
| sf_count_ > sf_count_max_ && pkt->data.frame.pts - last_sf_pts_ >= 5; |
| return pkt; |
| } |
| |
| int sf_count_; |
| int sf_count_max_; |
| aom_codec_cx_pkt_t modified_pkt_; |
| std::vector<uint8_t> modified_buf_; |
| aom_codec_pts_t last_sf_pts_; |
| |
| private: |
| int n_tile_cols_; |
| int n_tile_rows_; |
| }; |
| GTEST_ALLOW_UNINSTANTIATED_PARAMETERIZED_TEST(SuperframeTest); |
| |
| TEST_P(SuperframeTest, TestSuperframeIndexIsOptional) { |
| sf_count_max_ = 0; // early exit on successful test. |
| cfg_.g_lag_in_frames = 25; |
| cfg_.large_scale_tile = 1; |
| ::libaom_test::I420VideoSource video("hantro_collage_w352h288.yuv", 352, 288, |
| 30, 1, 0, 40); |
| ASSERT_NO_FATAL_FAILURE(RunLoop(&video)); |
| // NOTE: The use of BWDREF_FRAME will enable the coding of more non-show |
| // frames besides ALTREF_FRAME. |
| EXPECT_GE(sf_count_, 1); |
| } |
| |
| } // namespace |