| /* |
| * 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 "webrtc/modules/video_coding/codecs/h264/h264_encoder_impl.h" |
| |
| #include "webrtc/test/gtest.h" |
| |
| namespace webrtc { |
| |
| namespace { |
| |
| const int kMaxPayloadSize = 1024; |
| const int kNumCores = 1; |
| |
| 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->H264()->frameDroppingOn = true; |
| codec_settings->targetBitrate = 2000; |
| codec_settings->maxBitrate = 4000; |
| } |
| |
| TEST(H264EncoderImplTest, CanInitializeWithDefaultParameters) { |
| H264EncoderImpl encoder(cricket::VideoCodec("H264")); |
| VideoCodec codec_settings; |
| SetDefaultSettings(&codec_settings); |
| EXPECT_EQ(WEBRTC_VIDEO_CODEC_OK, |
| encoder.InitEncode(&codec_settings, kNumCores, kMaxPayloadSize)); |
| EXPECT_EQ(H264PacketizationMode::NonInterleaved, |
| encoder.PacketizationModeForTesting()); |
| } |
| |
| TEST(H264EncoderImplTest, CanInitializeWithNonInterleavedModeExplicitly) { |
| cricket::VideoCodec codec("H264"); |
| codec.SetParam(cricket::kH264FmtpPacketizationMode, "1"); |
| H264EncoderImpl encoder(codec); |
| VideoCodec codec_settings; |
| SetDefaultSettings(&codec_settings); |
| EXPECT_EQ(WEBRTC_VIDEO_CODEC_OK, |
| encoder.InitEncode(&codec_settings, kNumCores, kMaxPayloadSize)); |
| EXPECT_EQ(H264PacketizationMode::NonInterleaved, |
| encoder.PacketizationModeForTesting()); |
| } |
| |
| TEST(H264EncoderImplTest, CanInitializeWithSingleNalUnitModeExplicitly) { |
| cricket::VideoCodec codec("H264"); |
| codec.SetParam(cricket::kH264FmtpPacketizationMode, "0"); |
| H264EncoderImpl encoder(codec); |
| VideoCodec codec_settings; |
| SetDefaultSettings(&codec_settings); |
| EXPECT_EQ(WEBRTC_VIDEO_CODEC_OK, |
| encoder.InitEncode(&codec_settings, kNumCores, kMaxPayloadSize)); |
| EXPECT_EQ(H264PacketizationMode::SingleNalUnit, |
| encoder.PacketizationModeForTesting()); |
| } |
| |
| TEST(H264EncoderImplTest, CanInitializeWithRemovedParameter) { |
| cricket::VideoCodec codec("H264"); |
| codec.RemoveParam(cricket::kH264FmtpPacketizationMode); |
| H264EncoderImpl encoder(codec); |
| VideoCodec codec_settings; |
| SetDefaultSettings(&codec_settings); |
| EXPECT_EQ(WEBRTC_VIDEO_CODEC_OK, |
| encoder.InitEncode(&codec_settings, kNumCores, kMaxPayloadSize)); |
| EXPECT_EQ(H264PacketizationMode::SingleNalUnit, |
| encoder.PacketizationModeForTesting()); |
| } |
| |
| } // anonymous namespace |
| |
| } // namespace webrtc |