104546884bf7f816e52e1a6db03d6bba49a12edc5minyue@webrtc.org/* 204546884bf7f816e52e1a6db03d6bba49a12edc5minyue@webrtc.org * Copyright (c) 2014 The WebRTC project authors. All Rights Reserved. 304546884bf7f816e52e1a6db03d6bba49a12edc5minyue@webrtc.org * 404546884bf7f816e52e1a6db03d6bba49a12edc5minyue@webrtc.org * Use of this source code is governed by a BSD-style license 504546884bf7f816e52e1a6db03d6bba49a12edc5minyue@webrtc.org * that can be found in the LICENSE file in the root of the source 604546884bf7f816e52e1a6db03d6bba49a12edc5minyue@webrtc.org * tree. An additional intellectual property rights grant can be found 704546884bf7f816e52e1a6db03d6bba49a12edc5minyue@webrtc.org * in the file PATENTS. All contributing project authors may 804546884bf7f816e52e1a6db03d6bba49a12edc5minyue@webrtc.org * be found in the AUTHORS file in the root of the source tree. 904546884bf7f816e52e1a6db03d6bba49a12edc5minyue@webrtc.org */ 1004546884bf7f816e52e1a6db03d6bba49a12edc5minyue@webrtc.org 113c652b67468d182bd36aee4c31557621be50cc92kjellander@webrtc.org#include "webrtc/modules/audio_coding/codecs/opus/opus_interface.h" 1204546884bf7f816e52e1a6db03d6bba49a12edc5minyue@webrtc.org#include "webrtc/modules/audio_coding/codecs/tools/audio_codec_speed_test.h" 1304546884bf7f816e52e1a6db03d6bba49a12edc5minyue@webrtc.org 14a28c697d93cdf857b5b5be01b37c721db45843f7solenberg@webrtc.orgusing ::std::string; 1504546884bf7f816e52e1a6db03d6bba49a12edc5minyue@webrtc.org 1604546884bf7f816e52e1a6db03d6bba49a12edc5minyue@webrtc.orgnamespace webrtc { 1704546884bf7f816e52e1a6db03d6bba49a12edc5minyue@webrtc.org 1804546884bf7f816e52e1a6db03d6bba49a12edc5minyue@webrtc.orgstatic const int kOpusBlockDurationMs = 20; 19f563e85ab0bac7d2f5e70f70af7790595726832bminyue@webrtc.orgstatic const int kOpusSamplingKhz = 48; 2004546884bf7f816e52e1a6db03d6bba49a12edc5minyue@webrtc.org 2104546884bf7f816e52e1a6db03d6bba49a12edc5minyue@webrtc.orgclass OpusSpeedTest : public AudioCodecSpeedTest { 2204546884bf7f816e52e1a6db03d6bba49a12edc5minyue@webrtc.org protected: 2304546884bf7f816e52e1a6db03d6bba49a12edc5minyue@webrtc.org OpusSpeedTest(); 2414665ff7d4024d07e58622f498b23fd980001871kjellander@webrtc.org void SetUp() override; 2514665ff7d4024d07e58622f498b23fd980001871kjellander@webrtc.org void TearDown() override; 2604546884bf7f816e52e1a6db03d6bba49a12edc5minyue@webrtc.org virtual float EncodeABlock(int16_t* in_data, uint8_t* bit_stream, 27dce40cf804019a9898b6ab8d8262466b697c56e0Peter Kasting size_t max_bytes, size_t* encoded_bytes); 28dce40cf804019a9898b6ab8d8262466b697c56e0Peter Kasting virtual float DecodeABlock(const uint8_t* bit_stream, size_t encoded_bytes, 2904546884bf7f816e52e1a6db03d6bba49a12edc5minyue@webrtc.org int16_t* out_data); 3004546884bf7f816e52e1a6db03d6bba49a12edc5minyue@webrtc.org WebRtcOpusEncInst* opus_encoder_; 3104546884bf7f816e52e1a6db03d6bba49a12edc5minyue@webrtc.org WebRtcOpusDecInst* opus_decoder_; 3204546884bf7f816e52e1a6db03d6bba49a12edc5minyue@webrtc.org}; 3304546884bf7f816e52e1a6db03d6bba49a12edc5minyue@webrtc.org 3404546884bf7f816e52e1a6db03d6bba49a12edc5minyue@webrtc.orgOpusSpeedTest::OpusSpeedTest() 3504546884bf7f816e52e1a6db03d6bba49a12edc5minyue@webrtc.org : AudioCodecSpeedTest(kOpusBlockDurationMs, 36f563e85ab0bac7d2f5e70f70af7790595726832bminyue@webrtc.org kOpusSamplingKhz, 37f563e85ab0bac7d2f5e70f70af7790595726832bminyue@webrtc.org kOpusSamplingKhz), 3804546884bf7f816e52e1a6db03d6bba49a12edc5minyue@webrtc.org opus_encoder_(NULL), 3904546884bf7f816e52e1a6db03d6bba49a12edc5minyue@webrtc.org opus_decoder_(NULL) { 4004546884bf7f816e52e1a6db03d6bba49a12edc5minyue@webrtc.org} 4104546884bf7f816e52e1a6db03d6bba49a12edc5minyue@webrtc.org 4204546884bf7f816e52e1a6db03d6bba49a12edc5minyue@webrtc.orgvoid OpusSpeedTest::SetUp() { 4304546884bf7f816e52e1a6db03d6bba49a12edc5minyue@webrtc.org AudioCodecSpeedTest::SetUp(); 447dba7860c79652593f0a643fc81fe35f8707e0dbminyue@webrtc.org // If channels_ == 1, use Opus VOIP mode, otherwise, audio mode. 457dba7860c79652593f0a643fc81fe35f8707e0dbminyue@webrtc.org int app = channels_ == 1 ? 0 : 1; 4604546884bf7f816e52e1a6db03d6bba49a12edc5minyue@webrtc.org /* Create encoder memory. */ 477dba7860c79652593f0a643fc81fe35f8707e0dbminyue@webrtc.org EXPECT_EQ(0, WebRtcOpus_EncoderCreate(&opus_encoder_, channels_, app)); 4804546884bf7f816e52e1a6db03d6bba49a12edc5minyue@webrtc.org EXPECT_EQ(0, WebRtcOpus_DecoderCreate(&opus_decoder_, channels_)); 4904546884bf7f816e52e1a6db03d6bba49a12edc5minyue@webrtc.org /* Set bitrate. */ 5004546884bf7f816e52e1a6db03d6bba49a12edc5minyue@webrtc.org EXPECT_EQ(0, WebRtcOpus_SetBitRate(opus_encoder_, bit_rate_)); 5104546884bf7f816e52e1a6db03d6bba49a12edc5minyue@webrtc.org} 5204546884bf7f816e52e1a6db03d6bba49a12edc5minyue@webrtc.org 5304546884bf7f816e52e1a6db03d6bba49a12edc5minyue@webrtc.orgvoid OpusSpeedTest::TearDown() { 5404546884bf7f816e52e1a6db03d6bba49a12edc5minyue@webrtc.org AudioCodecSpeedTest::TearDown(); 5504546884bf7f816e52e1a6db03d6bba49a12edc5minyue@webrtc.org /* Free memory. */ 5604546884bf7f816e52e1a6db03d6bba49a12edc5minyue@webrtc.org EXPECT_EQ(0, WebRtcOpus_EncoderFree(opus_encoder_)); 5704546884bf7f816e52e1a6db03d6bba49a12edc5minyue@webrtc.org EXPECT_EQ(0, WebRtcOpus_DecoderFree(opus_decoder_)); 5804546884bf7f816e52e1a6db03d6bba49a12edc5minyue@webrtc.org} 5904546884bf7f816e52e1a6db03d6bba49a12edc5minyue@webrtc.org 6004546884bf7f816e52e1a6db03d6bba49a12edc5minyue@webrtc.orgfloat OpusSpeedTest::EncodeABlock(int16_t* in_data, uint8_t* bit_stream, 61dce40cf804019a9898b6ab8d8262466b697c56e0Peter Kasting size_t max_bytes, size_t* encoded_bytes) { 6204546884bf7f816e52e1a6db03d6bba49a12edc5minyue@webrtc.org clock_t clocks = clock(); 6304546884bf7f816e52e1a6db03d6bba49a12edc5minyue@webrtc.org int value = WebRtcOpus_Encode(opus_encoder_, in_data, 6404546884bf7f816e52e1a6db03d6bba49a12edc5minyue@webrtc.org input_length_sample_, max_bytes, 6504546884bf7f816e52e1a6db03d6bba49a12edc5minyue@webrtc.org bit_stream); 6604546884bf7f816e52e1a6db03d6bba49a12edc5minyue@webrtc.org clocks = clock() - clocks; 6704546884bf7f816e52e1a6db03d6bba49a12edc5minyue@webrtc.org EXPECT_GT(value, 0); 68dce40cf804019a9898b6ab8d8262466b697c56e0Peter Kasting *encoded_bytes = static_cast<size_t>(value); 6904546884bf7f816e52e1a6db03d6bba49a12edc5minyue@webrtc.org return 1000.0 * clocks / CLOCKS_PER_SEC; 7004546884bf7f816e52e1a6db03d6bba49a12edc5minyue@webrtc.org} 7104546884bf7f816e52e1a6db03d6bba49a12edc5minyue@webrtc.org 7204546884bf7f816e52e1a6db03d6bba49a12edc5minyue@webrtc.orgfloat OpusSpeedTest::DecodeABlock(const uint8_t* bit_stream, 73dce40cf804019a9898b6ab8d8262466b697c56e0Peter Kasting size_t encoded_bytes, int16_t* out_data) { 7404546884bf7f816e52e1a6db03d6bba49a12edc5minyue@webrtc.org int value; 7504546884bf7f816e52e1a6db03d6bba49a12edc5minyue@webrtc.org int16_t audio_type; 7604546884bf7f816e52e1a6db03d6bba49a12edc5minyue@webrtc.org clock_t clocks = clock(); 7733ccdfa1f555e00170e2b98cd0f575eed3e46236minyue@webrtc.org value = WebRtcOpus_Decode(opus_decoder_, bit_stream, encoded_bytes, out_data, 7833ccdfa1f555e00170e2b98cd0f575eed3e46236minyue@webrtc.org &audio_type); 7904546884bf7f816e52e1a6db03d6bba49a12edc5minyue@webrtc.org clocks = clock() - clocks; 8025702cb1628941427fa55e528f53483f239ae011pkasting EXPECT_EQ(output_length_sample_, static_cast<size_t>(value)); 8104546884bf7f816e52e1a6db03d6bba49a12edc5minyue@webrtc.org return 1000.0 * clocks / CLOCKS_PER_SEC; 8204546884bf7f816e52e1a6db03d6bba49a12edc5minyue@webrtc.org} 8304546884bf7f816e52e1a6db03d6bba49a12edc5minyue@webrtc.org 8404546884bf7f816e52e1a6db03d6bba49a12edc5minyue@webrtc.org#define ADD_TEST(complexity) \ 8504546884bf7f816e52e1a6db03d6bba49a12edc5minyue@webrtc.orgTEST_P(OpusSpeedTest, OpusSetComplexityTest##complexity) { \ 8604546884bf7f816e52e1a6db03d6bba49a12edc5minyue@webrtc.org /* Test audio length in second. */ \ 8704546884bf7f816e52e1a6db03d6bba49a12edc5minyue@webrtc.org size_t kDurationSec = 400; \ 8804546884bf7f816e52e1a6db03d6bba49a12edc5minyue@webrtc.org /* Set complexity. */ \ 8904546884bf7f816e52e1a6db03d6bba49a12edc5minyue@webrtc.org printf("Setting complexity to %d ...\n", complexity); \ 9004546884bf7f816e52e1a6db03d6bba49a12edc5minyue@webrtc.org EXPECT_EQ(0, WebRtcOpus_SetComplexity(opus_encoder_, complexity)); \ 9104546884bf7f816e52e1a6db03d6bba49a12edc5minyue@webrtc.org EncodeDecode(kDurationSec); \ 9204546884bf7f816e52e1a6db03d6bba49a12edc5minyue@webrtc.org} 9304546884bf7f816e52e1a6db03d6bba49a12edc5minyue@webrtc.org 9404546884bf7f816e52e1a6db03d6bba49a12edc5minyue@webrtc.orgADD_TEST(10); 9504546884bf7f816e52e1a6db03d6bba49a12edc5minyue@webrtc.orgADD_TEST(9); 9604546884bf7f816e52e1a6db03d6bba49a12edc5minyue@webrtc.orgADD_TEST(8); 9704546884bf7f816e52e1a6db03d6bba49a12edc5minyue@webrtc.orgADD_TEST(7); 9804546884bf7f816e52e1a6db03d6bba49a12edc5minyue@webrtc.orgADD_TEST(6); 9904546884bf7f816e52e1a6db03d6bba49a12edc5minyue@webrtc.orgADD_TEST(5); 10004546884bf7f816e52e1a6db03d6bba49a12edc5minyue@webrtc.orgADD_TEST(4); 10104546884bf7f816e52e1a6db03d6bba49a12edc5minyue@webrtc.orgADD_TEST(3); 10204546884bf7f816e52e1a6db03d6bba49a12edc5minyue@webrtc.orgADD_TEST(2); 10304546884bf7f816e52e1a6db03d6bba49a12edc5minyue@webrtc.orgADD_TEST(1); 10404546884bf7f816e52e1a6db03d6bba49a12edc5minyue@webrtc.orgADD_TEST(0); 10504546884bf7f816e52e1a6db03d6bba49a12edc5minyue@webrtc.org 10604546884bf7f816e52e1a6db03d6bba49a12edc5minyue@webrtc.org// List all test cases: (channel, bit rat, filename, extension). 10704546884bf7f816e52e1a6db03d6bba49a12edc5minyue@webrtc.orgconst coding_param param_set[] = 10834a865a0385b6d5f79d177f16fa393cbbedcb844kjellander@webrtc.org {::std::tr1::make_tuple(1, 64000, 10934a865a0385b6d5f79d177f16fa393cbbedcb844kjellander@webrtc.org string("audio_coding/speech_mono_32_48kHz"), 11034a865a0385b6d5f79d177f16fa393cbbedcb844kjellander@webrtc.org string("pcm"), true), 11134a865a0385b6d5f79d177f16fa393cbbedcb844kjellander@webrtc.org ::std::tr1::make_tuple(1, 32000, 11234a865a0385b6d5f79d177f16fa393cbbedcb844kjellander@webrtc.org string("audio_coding/speech_mono_32_48kHz"), 11334a865a0385b6d5f79d177f16fa393cbbedcb844kjellander@webrtc.org string("pcm"), true), 11434a865a0385b6d5f79d177f16fa393cbbedcb844kjellander@webrtc.org ::std::tr1::make_tuple(2, 64000, 11534a865a0385b6d5f79d177f16fa393cbbedcb844kjellander@webrtc.org string("audio_coding/music_stereo_48kHz"), 11634a865a0385b6d5f79d177f16fa393cbbedcb844kjellander@webrtc.org string("pcm"), true)}; 11704546884bf7f816e52e1a6db03d6bba49a12edc5minyue@webrtc.org 11804546884bf7f816e52e1a6db03d6bba49a12edc5minyue@webrtc.orgINSTANTIATE_TEST_CASE_P(AllTest, OpusSpeedTest, 11934a865a0385b6d5f79d177f16fa393cbbedcb844kjellander@webrtc.org ::testing::ValuesIn(param_set)); 12004546884bf7f816e52e1a6db03d6bba49a12edc5minyue@webrtc.org 12104546884bf7f816e52e1a6db03d6bba49a12edc5minyue@webrtc.org} // namespace webrtc 122