andrew@webrtc.org | 19eefdc | 2011-09-14 17:02:44 +0000 | [diff] [blame] | 1 | /* |
| 2 | * Copyright (c) 2011 The WebRTC project authors. All Rights Reserved. |
| 3 | * |
| 4 | * Use of this source code is governed by a BSD-style license |
| 5 | * that can be found in the LICENSE file in the root of the source |
| 6 | * tree. An additional intellectual property rights grant can be found |
| 7 | * in the file PATENTS. All contributing project authors may |
| 8 | * be found in the AUTHORS file in the root of the source tree. |
| 9 | */ |
| 10 | |
pbos@webrtc.org | aa30bb7 | 2013-05-27 09:49:58 +0000 | [diff] [blame] | 11 | #include "testing/gtest/include/gtest/gtest.h" |
andrew@webrtc.org | 19eefdc | 2011-09-14 17:02:44 +0000 | [diff] [blame] | 12 | |
pbos@webrtc.org | aa30bb7 | 2013-05-27 09:49:58 +0000 | [diff] [blame] | 13 | #include "webrtc/common_audio/resampler/include/resampler.h" |
andrew@webrtc.org | 19eefdc | 2011-09-14 17:02:44 +0000 | [diff] [blame] | 14 | |
| 15 | // TODO(andrew): this is a work-in-progress. Many more tests are needed. |
| 16 | |
| 17 | namespace webrtc { |
| 18 | namespace { |
Andrew MacDonald | 2c9c83d | 2015-03-30 10:08:22 -0700 | [diff] [blame] | 19 | |
| 20 | const int kNumChannels[] = {1, 2}; |
| 21 | const size_t kNumChannelsSize = sizeof(kNumChannels) / sizeof(*kNumChannels); |
andrew@webrtc.org | 19eefdc | 2011-09-14 17:02:44 +0000 | [diff] [blame] | 22 | |
| 23 | // Rates we must support. |
| 24 | const int kMaxRate = 96000; |
| 25 | const int kRates[] = { |
| 26 | 8000, |
| 27 | 16000, |
| 28 | 32000, |
| 29 | 44000, |
| 30 | 48000, |
| 31 | kMaxRate |
| 32 | }; |
| 33 | const size_t kRatesSize = sizeof(kRates) / sizeof(*kRates); |
tina.legrand@webrtc.org | 5c43b1b | 2011-12-22 08:28:05 +0000 | [diff] [blame] | 34 | const int kMaxChannels = 2; |
| 35 | const size_t kDataSize = static_cast<size_t> (kMaxChannels * kMaxRate / 100); |
andrew@webrtc.org | 19eefdc | 2011-09-14 17:02:44 +0000 | [diff] [blame] | 36 | |
| 37 | // TODO(andrew): should we be supporting these combinations? |
| 38 | bool ValidRates(int in_rate, int out_rate) { |
| 39 | // Not the most compact notation, for clarity. |
| 40 | if ((in_rate == 44000 && (out_rate == 48000 || out_rate == 96000)) || |
bjornv@webrtc.org | 48b68c0 | 2011-11-23 13:50:41 +0000 | [diff] [blame] | 41 | (out_rate == 44000 && (in_rate == 48000 || in_rate == 96000))) { |
andrew@webrtc.org | 19eefdc | 2011-09-14 17:02:44 +0000 | [diff] [blame] | 42 | return false; |
| 43 | } |
| 44 | |
| 45 | return true; |
| 46 | } |
| 47 | |
| 48 | class ResamplerTest : public testing::Test { |
| 49 | protected: |
| 50 | ResamplerTest(); |
| 51 | virtual void SetUp(); |
| 52 | virtual void TearDown(); |
| 53 | |
| 54 | Resampler rs_; |
| 55 | int16_t data_in_[kDataSize]; |
| 56 | int16_t data_out_[kDataSize]; |
| 57 | }; |
| 58 | |
bjornv@webrtc.org | 48b68c0 | 2011-11-23 13:50:41 +0000 | [diff] [blame] | 59 | ResamplerTest::ResamplerTest() {} |
andrew@webrtc.org | 19eefdc | 2011-09-14 17:02:44 +0000 | [diff] [blame] | 60 | |
bjornv@webrtc.org | 0edb25d | 2011-12-13 09:06:54 +0000 | [diff] [blame] | 61 | void ResamplerTest::SetUp() { |
| 62 | // Initialize input data with anything. The tests are content independent. |
| 63 | memset(data_in_, 1, sizeof(data_in_)); |
| 64 | } |
andrew@webrtc.org | 19eefdc | 2011-09-14 17:02:44 +0000 | [diff] [blame] | 65 | |
bjornv@webrtc.org | 48b68c0 | 2011-11-23 13:50:41 +0000 | [diff] [blame] | 66 | void ResamplerTest::TearDown() {} |
andrew@webrtc.org | 19eefdc | 2011-09-14 17:02:44 +0000 | [diff] [blame] | 67 | |
| 68 | TEST_F(ResamplerTest, Reset) { |
| 69 | // The only failure mode for the constructor is if Reset() fails. For the |
| 70 | // time being then (until an Init function is added), we rely on Reset() |
| 71 | // to test the constructor. |
| 72 | |
| 73 | // Check that all required combinations are supported. |
| 74 | for (size_t i = 0; i < kRatesSize; ++i) { |
| 75 | for (size_t j = 0; j < kRatesSize; ++j) { |
Andrew MacDonald | 2c9c83d | 2015-03-30 10:08:22 -0700 | [diff] [blame] | 76 | for (size_t k = 0; k < kNumChannelsSize; ++k) { |
andrew@webrtc.org | 19eefdc | 2011-09-14 17:02:44 +0000 | [diff] [blame] | 77 | std::ostringstream ss; |
| 78 | ss << "Input rate: " << kRates[i] << ", output rate: " << kRates[j] |
Andrew MacDonald | 2c9c83d | 2015-03-30 10:08:22 -0700 | [diff] [blame] | 79 | << ", channels: " << kNumChannels[k]; |
andrew@webrtc.org | 19eefdc | 2011-09-14 17:02:44 +0000 | [diff] [blame] | 80 | SCOPED_TRACE(ss.str()); |
| 81 | if (ValidRates(kRates[i], kRates[j])) |
Andrew MacDonald | 2c9c83d | 2015-03-30 10:08:22 -0700 | [diff] [blame] | 82 | EXPECT_EQ(0, rs_.Reset(kRates[i], kRates[j], kNumChannels[k])); |
andrew@webrtc.org | 19eefdc | 2011-09-14 17:02:44 +0000 | [diff] [blame] | 83 | else |
Andrew MacDonald | 2c9c83d | 2015-03-30 10:08:22 -0700 | [diff] [blame] | 84 | EXPECT_EQ(-1, rs_.Reset(kRates[i], kRates[j], kNumChannels[k])); |
andrew@webrtc.org | 19eefdc | 2011-09-14 17:02:44 +0000 | [diff] [blame] | 85 | } |
| 86 | } |
| 87 | } |
| 88 | } |
| 89 | |
tina.legrand@webrtc.org | 5c43b1b | 2011-12-22 08:28:05 +0000 | [diff] [blame] | 90 | // TODO(tlegrand): Replace code inside the two tests below with a function |
| 91 | // with number of channels and ResamplerType as input. |
Andrew MacDonald | 2c9c83d | 2015-03-30 10:08:22 -0700 | [diff] [blame] | 92 | TEST_F(ResamplerTest, Mono) { |
| 93 | const int kChannels = 1; |
andrew@webrtc.org | 19eefdc | 2011-09-14 17:02:44 +0000 | [diff] [blame] | 94 | for (size_t i = 0; i < kRatesSize; ++i) { |
| 95 | for (size_t j = 0; j < kRatesSize; ++j) { |
| 96 | std::ostringstream ss; |
| 97 | ss << "Input rate: " << kRates[i] << ", output rate: " << kRates[j]; |
| 98 | SCOPED_TRACE(ss.str()); |
| 99 | |
| 100 | if (ValidRates(kRates[i], kRates[j])) { |
Peter Kasting | dce40cf | 2015-08-24 14:52:23 -0700 | [diff] [blame^] | 101 | size_t in_length = static_cast<size_t>(kRates[i] / 100); |
| 102 | size_t out_length = 0; |
Andrew MacDonald | 2c9c83d | 2015-03-30 10:08:22 -0700 | [diff] [blame] | 103 | EXPECT_EQ(0, rs_.Reset(kRates[i], kRates[j], kChannels)); |
andrew@webrtc.org | 19eefdc | 2011-09-14 17:02:44 +0000 | [diff] [blame] | 104 | EXPECT_EQ(0, rs_.Push(data_in_, in_length, data_out_, kDataSize, |
| 105 | out_length)); |
Peter Kasting | dce40cf | 2015-08-24 14:52:23 -0700 | [diff] [blame^] | 106 | EXPECT_EQ(static_cast<size_t>(kRates[j] / 100), out_length); |
andrew@webrtc.org | 19eefdc | 2011-09-14 17:02:44 +0000 | [diff] [blame] | 107 | } else { |
Andrew MacDonald | 2c9c83d | 2015-03-30 10:08:22 -0700 | [diff] [blame] | 108 | EXPECT_EQ(-1, rs_.Reset(kRates[i], kRates[j], kChannels)); |
andrew@webrtc.org | 19eefdc | 2011-09-14 17:02:44 +0000 | [diff] [blame] | 109 | } |
| 110 | } |
| 111 | } |
tina.legrand@webrtc.org | 5c43b1b | 2011-12-22 08:28:05 +0000 | [diff] [blame] | 112 | } |
andrew@webrtc.org | 19eefdc | 2011-09-14 17:02:44 +0000 | [diff] [blame] | 113 | |
Andrew MacDonald | 2c9c83d | 2015-03-30 10:08:22 -0700 | [diff] [blame] | 114 | TEST_F(ResamplerTest, Stereo) { |
tina.legrand@webrtc.org | 5c43b1b | 2011-12-22 08:28:05 +0000 | [diff] [blame] | 115 | const int kChannels = 2; |
| 116 | for (size_t i = 0; i < kRatesSize; ++i) { |
| 117 | for (size_t j = 0; j < kRatesSize; ++j) { |
| 118 | std::ostringstream ss; |
| 119 | ss << "Input rate: " << kRates[i] << ", output rate: " << kRates[j]; |
| 120 | SCOPED_TRACE(ss.str()); |
| 121 | |
| 122 | if (ValidRates(kRates[i], kRates[j])) { |
Peter Kasting | dce40cf | 2015-08-24 14:52:23 -0700 | [diff] [blame^] | 123 | size_t in_length = static_cast<size_t>(kChannels * kRates[i] / 100); |
| 124 | size_t out_length = 0; |
tina.legrand@webrtc.org | 5c43b1b | 2011-12-22 08:28:05 +0000 | [diff] [blame] | 125 | EXPECT_EQ(0, rs_.Reset(kRates[i], kRates[j], |
Andrew MacDonald | 2c9c83d | 2015-03-30 10:08:22 -0700 | [diff] [blame] | 126 | kChannels)); |
tina.legrand@webrtc.org | 5c43b1b | 2011-12-22 08:28:05 +0000 | [diff] [blame] | 127 | EXPECT_EQ(0, rs_.Push(data_in_, in_length, data_out_, kDataSize, |
| 128 | out_length)); |
Peter Kasting | dce40cf | 2015-08-24 14:52:23 -0700 | [diff] [blame^] | 129 | EXPECT_EQ(static_cast<size_t>(kChannels * kRates[j] / 100), out_length); |
tina.legrand@webrtc.org | 5c43b1b | 2011-12-22 08:28:05 +0000 | [diff] [blame] | 130 | } else { |
| 131 | EXPECT_EQ(-1, rs_.Reset(kRates[i], kRates[j], |
Andrew MacDonald | 2c9c83d | 2015-03-30 10:08:22 -0700 | [diff] [blame] | 132 | kChannels)); |
tina.legrand@webrtc.org | 5c43b1b | 2011-12-22 08:28:05 +0000 | [diff] [blame] | 133 | } |
| 134 | } |
| 135 | } |
andrew@webrtc.org | 19eefdc | 2011-09-14 17:02:44 +0000 | [diff] [blame] | 136 | } |
Andrew MacDonald | 2c9c83d | 2015-03-30 10:08:22 -0700 | [diff] [blame] | 137 | |
andrew@webrtc.org | 19eefdc | 2011-09-14 17:02:44 +0000 | [diff] [blame] | 138 | } // namespace |
| 139 | } // namespace webrtc |