force_key_frame_test.cc (3106B)
1 /* 2 * Copyright (c) 2022, Alliance for Open Media. All rights reserved. 3 * 4 * This source code is subject to the terms of the BSD 2 Clause License and 5 * the Alliance for Open Media Patent License 1.0. If the BSD 2 Clause License 6 * was not distributed with this source code in the LICENSE file, you can 7 * obtain it at www.aomedia.org/license/software. If the Alliance for Open 8 * Media Patent License 1.0 was not distributed with this source code in the 9 * PATENTS file, you can obtain it at www.aomedia.org/license/patent. 10 */ 11 12 // Tests for https://crbug.com/aomedia/3327. 13 // 14 // In good-quality mode, set cfg.g_lag_in_frames to 1 or 0 and encode two 15 // frames in one-pass mode. Pass AOM_EFLAG_FORCE_KF to the second 16 // aom_codec_encode() call. Both frames should be encoded as key frames. 17 18 #include <memory> 19 20 #include "aom/aomcx.h" 21 #include "aom/aom_encoder.h" 22 #include "gtest/gtest.h" 23 24 namespace { 25 26 void TestOnePassMode(unsigned int lag_in_frames) { 27 // A buffer of gray samples of size 128x128, YUV 4:2:0. 28 constexpr size_t kImageDataSize = 128 * 128 + 2 * 64 * 64; 29 std::unique_ptr<unsigned char[]> img_data(new unsigned char[kImageDataSize]); 30 ASSERT_NE(img_data, nullptr); 31 memset(img_data.get(), 128, kImageDataSize); 32 33 aom_codec_iface_t *iface = aom_codec_av1_cx(); 34 aom_codec_enc_cfg_t cfg; 35 ASSERT_EQ(AOM_CODEC_OK, 36 aom_codec_enc_config_default(iface, &cfg, AOM_USAGE_GOOD_QUALITY)); 37 cfg.g_w = 128; 38 cfg.g_h = 128; 39 cfg.g_pass = AOM_RC_ONE_PASS; 40 cfg.g_lag_in_frames = lag_in_frames; 41 aom_codec_ctx_t enc; 42 EXPECT_EQ(AOM_CODEC_OK, aom_codec_enc_init(&enc, iface, &cfg, 0)); 43 44 aom_image_t img; 45 EXPECT_EQ(&img, 46 aom_img_wrap(&img, AOM_IMG_FMT_I420, 128, 128, 1, img_data.get())); 47 48 aom_codec_iter_t iter; 49 const aom_codec_cx_pkt_t *pkt; 50 int frame_count = 0; 51 52 EXPECT_EQ(AOM_CODEC_OK, aom_codec_encode(&enc, &img, 0, 1, 0)); 53 54 iter = nullptr; 55 while ((pkt = aom_codec_get_cx_data(&enc, &iter)) != nullptr) { 56 ASSERT_EQ(pkt->kind, AOM_CODEC_CX_FRAME_PKT); 57 EXPECT_NE(pkt->data.frame.flags & AOM_FRAME_IS_KEY, 0u) 58 << "frame " << frame_count; 59 frame_count++; 60 } 61 62 EXPECT_EQ(AOM_CODEC_OK, 63 aom_codec_encode(&enc, &img, 1, 1, AOM_EFLAG_FORCE_KF)); 64 65 iter = nullptr; 66 while ((pkt = aom_codec_get_cx_data(&enc, &iter)) != nullptr) { 67 ASSERT_EQ(pkt->kind, AOM_CODEC_CX_FRAME_PKT); 68 EXPECT_NE(pkt->data.frame.flags & AOM_FRAME_IS_KEY, 0u) 69 << "frame " << frame_count; 70 frame_count++; 71 } 72 73 EXPECT_EQ(AOM_CODEC_OK, aom_codec_encode(&enc, nullptr, 0, 0, 0)); 74 75 iter = nullptr; 76 while ((pkt = aom_codec_get_cx_data(&enc, &iter)) != nullptr) { 77 ASSERT_EQ(pkt->kind, AOM_CODEC_CX_FRAME_PKT); 78 EXPECT_NE(pkt->data.frame.flags & AOM_FRAME_IS_KEY, 0u) 79 << "frame " << frame_count; 80 frame_count++; 81 } 82 83 EXPECT_EQ(frame_count, 2); 84 EXPECT_EQ(AOM_CODEC_OK, aom_codec_destroy(&enc)); 85 } 86 87 TEST(ForceKeyFrameTest, OnePassModeLag0) { TestOnePassMode(0); } 88 89 TEST(ForceKeyFrameTest, OnePassModeLag1) { TestOnePassMode(1); } 90 91 TEST(ForceKeyFrameTest, OnePassModeLag2) { TestOnePassMode(2); } 92 93 } // namespace