mirror of
https://github.com/mollyim/webrtc.git
synced 2025-05-15 14:50:39 +01:00
73588223c1
Bug: webrtc:15860 Change-Id: I76517ee3603847ff064f16fddc9423a3568818a7 Reviewed-on: https://webrtc-review.googlesource.com/c/src/+/345741 Commit-Queue: Danil Chapovalov <danilchap@webrtc.org> Reviewed-by: Philip Eliasson <philipel@webrtc.org> Cr-Commit-Position: refs/heads/main@{#42004}
79 lines
2.7 KiB
C++
79 lines
2.7 KiB
C++
/*
|
|
* Copyright (c) 2015 The WebRTC 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 "modules/video_coding/codecs/h264/h264_encoder_impl.h"
|
|
|
|
#include "api/environment/environment_factory.h"
|
|
#include "api/video_codecs/video_encoder.h"
|
|
#include "modules/video_coding/include/video_error_codes.h"
|
|
#include "test/gtest.h"
|
|
|
|
namespace webrtc {
|
|
|
|
namespace {
|
|
|
|
const int kMaxPayloadSize = 1024;
|
|
const int kNumCores = 1;
|
|
|
|
const VideoEncoder::Capabilities kCapabilities(false);
|
|
const VideoEncoder::Settings kSettings(kCapabilities,
|
|
kNumCores,
|
|
kMaxPayloadSize);
|
|
|
|
void SetDefaultSettings(VideoCodec* codec_settings) {
|
|
codec_settings->codecType = kVideoCodecH264;
|
|
codec_settings->maxFramerate = 60;
|
|
codec_settings->width = 640;
|
|
codec_settings->height = 480;
|
|
// If frame dropping is false, we get a warning that bitrate can't
|
|
// be controlled for RC_QUALITY_MODE; RC_BITRATE_MODE and RC_TIMESTAMP_MODE
|
|
codec_settings->SetFrameDropEnabled(true);
|
|
codec_settings->startBitrate = 2000;
|
|
codec_settings->maxBitrate = 4000;
|
|
}
|
|
|
|
TEST(H264EncoderImplTest, CanInitializeWithDefaultParameters) {
|
|
H264EncoderImpl encoder(CreateEnvironment(), {});
|
|
VideoCodec codec_settings;
|
|
SetDefaultSettings(&codec_settings);
|
|
EXPECT_EQ(WEBRTC_VIDEO_CODEC_OK,
|
|
encoder.InitEncode(&codec_settings, kSettings));
|
|
EXPECT_EQ(H264PacketizationMode::NonInterleaved,
|
|
encoder.PacketizationModeForTesting());
|
|
}
|
|
|
|
TEST(H264EncoderImplTest, CanInitializeWithNonInterleavedModeExplicitly) {
|
|
H264EncoderImpl encoder(
|
|
CreateEnvironment(),
|
|
{.packetization_mode = H264PacketizationMode::NonInterleaved});
|
|
VideoCodec codec_settings;
|
|
SetDefaultSettings(&codec_settings);
|
|
EXPECT_EQ(WEBRTC_VIDEO_CODEC_OK,
|
|
encoder.InitEncode(&codec_settings, kSettings));
|
|
EXPECT_EQ(H264PacketizationMode::NonInterleaved,
|
|
encoder.PacketizationModeForTesting());
|
|
}
|
|
|
|
TEST(H264EncoderImplTest, CanInitializeWithSingleNalUnitModeExplicitly) {
|
|
H264EncoderImpl encoder(
|
|
CreateEnvironment(),
|
|
{.packetization_mode = H264PacketizationMode::SingleNalUnit});
|
|
VideoCodec codec_settings;
|
|
SetDefaultSettings(&codec_settings);
|
|
EXPECT_EQ(WEBRTC_VIDEO_CODEC_OK,
|
|
encoder.InitEncode(&codec_settings, kSettings));
|
|
EXPECT_EQ(H264PacketizationMode::SingleNalUnit,
|
|
encoder.PacketizationModeForTesting());
|
|
}
|
|
|
|
} // anonymous namespace
|
|
|
|
} // namespace webrtc
|