diff options
Diffstat (limited to 'third_party/aom/test/firstpass_test.cc')
-rw-r--r-- | third_party/aom/test/firstpass_test.cc | 166 |
1 files changed, 166 insertions, 0 deletions
diff --git a/third_party/aom/test/firstpass_test.cc b/third_party/aom/test/firstpass_test.cc new file mode 100644 index 0000000000..1f4f3b7853 --- /dev/null +++ b/third_party/aom/test/firstpass_test.cc @@ -0,0 +1,166 @@ +/* + * Copyright (c) 2021, Alliance for Open Media. All rights reserved + * + * This source code is subject to the terms of the BSD 2 Clause License and + * the Alliance for Open Media Patent License 1.0. If the BSD 2 Clause License + * was not distributed with this source code in the LICENSE file, you can + * obtain it at www.aomedia.org/license/software. 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 www.aomedia.org/license/patent. + */ + +#include <stddef.h> + +#include "av1/common/common.h" +#include "av1/encoder/firstpass.h" +#include "third_party/googletest/src/googletest/include/gtest/gtest.h" + +namespace { + +TEST(FirstpassTest, FirstpassInfoInitWithExtBuf) { + FIRSTPASS_INFO firstpass_info; + FIRSTPASS_STATS ext_stats_buf[10]; + const int ref_stats_size = 10; + for (int i = 0; i < ref_stats_size; ++i) { + av1_zero(ext_stats_buf[i]); + ext_stats_buf[i].frame = i; + } + aom_codec_err_t ret = + av1_firstpass_info_init(&firstpass_info, ext_stats_buf, 10); + EXPECT_EQ(firstpass_info.stats_count, ref_stats_size); + EXPECT_EQ(firstpass_info.future_stats_count + firstpass_info.past_stats_count, + firstpass_info.stats_count); + EXPECT_EQ(firstpass_info.cur_index, 0); + EXPECT_EQ(ret, AOM_CODEC_OK); +} + +TEST(FirstpassTest, FirstpassInfoInitWithStaticBuf) { + FIRSTPASS_INFO firstpass_info; + aom_codec_err_t ret = av1_firstpass_info_init(&firstpass_info, nullptr, 0); + EXPECT_EQ(firstpass_info.stats_count, 0); + EXPECT_EQ(firstpass_info.cur_index, 0); + EXPECT_EQ(ret, AOM_CODEC_OK); +} + +TEST(FirstpassTest, FirstpassInfoPushPop) { + FIRSTPASS_INFO firstpass_info; + av1_firstpass_info_init(&firstpass_info, nullptr, 0); + EXPECT_EQ(firstpass_info.stats_buf_size, FIRSTPASS_INFO_STATIC_BUF_SIZE); + for (int i = 0; i < FIRSTPASS_INFO_STATIC_BUF_SIZE; ++i) { + FIRSTPASS_STATS stats; + av1_zero(stats); + stats.frame = i; + aom_codec_err_t ret = av1_firstpass_info_push(&firstpass_info, &stats); + EXPECT_EQ(ret, AOM_CODEC_OK); + } + EXPECT_EQ(firstpass_info.stats_count, FIRSTPASS_INFO_STATIC_BUF_SIZE); + const int pop_count = FIRSTPASS_INFO_STATIC_BUF_SIZE / 2; + for (int i = 0; i < pop_count; ++i) { + const FIRSTPASS_STATS *stats = av1_firstpass_info_peek(&firstpass_info, 0); + aom_codec_err_t ret = + av1_firstpass_info_move_cur_index_and_pop(&firstpass_info); + EXPECT_NE(stats, nullptr); + EXPECT_EQ(stats->frame, i); + EXPECT_EQ(ret, AOM_CODEC_OK); + } + EXPECT_EQ(firstpass_info.stats_count, + FIRSTPASS_INFO_STATIC_BUF_SIZE - pop_count); + + const int push_count = FIRSTPASS_INFO_STATIC_BUF_SIZE / 2; + for (int i = 0; i < push_count; ++i) { + FIRSTPASS_STATS stats; + av1_zero(stats); + aom_codec_err_t ret = av1_firstpass_info_push(&firstpass_info, &stats); + EXPECT_EQ(ret, AOM_CODEC_OK); + } + EXPECT_EQ(firstpass_info.stats_count, FIRSTPASS_INFO_STATIC_BUF_SIZE); + + EXPECT_EQ(firstpass_info.stats_count, firstpass_info.stats_buf_size); + { + // Push the stats when the queue is full. + FIRSTPASS_STATS stats; + av1_zero(stats); + aom_codec_err_t ret = av1_firstpass_info_push(&firstpass_info, &stats); + EXPECT_EQ(ret, AOM_CODEC_ERROR); + } +} + +TEST(FirstpassTest, FirstpassInfoTotalStats) { + FIRSTPASS_INFO firstpass_info; + av1_firstpass_info_init(&firstpass_info, nullptr, 0); + EXPECT_EQ(firstpass_info.total_stats.frame, 0); + for (int i = 0; i < 10; ++i) { + FIRSTPASS_STATS stats; + av1_zero(stats); + stats.count = 1; + av1_firstpass_info_push(&firstpass_info, &stats); + } + EXPECT_EQ(firstpass_info.total_stats.count, 10); +} + +TEST(FirstpassTest, FirstpassInfoMoveCurr) { + FIRSTPASS_INFO firstpass_info; + av1_firstpass_info_init(&firstpass_info, nullptr, 0); + int frame_cnt = 0; + EXPECT_EQ(firstpass_info.stats_buf_size, FIRSTPASS_INFO_STATIC_BUF_SIZE); + for (int i = 0; i < FIRSTPASS_INFO_STATIC_BUF_SIZE; ++i) { + FIRSTPASS_STATS stats; + av1_zero(stats); + stats.frame = frame_cnt; + ++frame_cnt; + aom_codec_err_t ret = av1_firstpass_info_push(&firstpass_info, &stats); + EXPECT_EQ(ret, AOM_CODEC_OK); + } + EXPECT_EQ(firstpass_info.cur_index, firstpass_info.start_index); + { + aom_codec_err_t ret = av1_firstpass_info_pop(&firstpass_info); + // We cannot pop when cur_index == start_index + EXPECT_EQ(ret, AOM_CODEC_ERROR); + } + int ref_frame_cnt = 0; + const int move_count = FIRSTPASS_INFO_STATIC_BUF_SIZE * 2 / 3; + for (int i = 0; i < move_count; ++i) { + const FIRSTPASS_STATS *this_stats = + av1_firstpass_info_peek(&firstpass_info, 0); + EXPECT_EQ(this_stats->frame, ref_frame_cnt); + ++ref_frame_cnt; + av1_firstpass_info_move_cur_index(&firstpass_info); + } + EXPECT_EQ(firstpass_info.future_stats_count, + FIRSTPASS_INFO_STATIC_BUF_SIZE - move_count); + EXPECT_EQ(firstpass_info.past_stats_count, move_count); + EXPECT_EQ(firstpass_info.stats_count, FIRSTPASS_INFO_STATIC_BUF_SIZE); + + const int test_count = FIRSTPASS_INFO_STATIC_BUF_SIZE / 2; + for (int i = 0; i < test_count; ++i) { + aom_codec_err_t ret = av1_firstpass_info_pop(&firstpass_info); + EXPECT_EQ(ret, AOM_CODEC_OK); + } + + // Pop #test_count stats + for (int i = 0; i < test_count; ++i) { + FIRSTPASS_STATS stats; + av1_zero(stats); + stats.frame = frame_cnt; + ++frame_cnt; + aom_codec_err_t ret = av1_firstpass_info_push(&firstpass_info, &stats); + EXPECT_EQ(ret, AOM_CODEC_OK); + } + + // peek and move #test_count stats + for (int i = 0; i < test_count; ++i) { + const FIRSTPASS_STATS *this_stats = + av1_firstpass_info_peek(&firstpass_info, 0); + EXPECT_EQ(this_stats->frame, ref_frame_cnt); + ++ref_frame_cnt; + av1_firstpass_info_move_cur_index(&firstpass_info); + } + + // pop #test_count stats + for (int i = 0; i < test_count; ++i) { + aom_codec_err_t ret = av1_firstpass_info_pop(&firstpass_info); + EXPECT_EQ(ret, AOM_CODEC_OK); + } +} + +} // namespace |