| /* |
| * Copyright (C) 2017 The Android Open Source Project |
| * |
| * Licensed under the Apache License, Version 2.0 (the "License"); |
| * you may not use this file except in compliance with the License. |
| * You may obtain a copy of the License at |
| * |
| * http://www.apache.org/licenses/LICENSE-2.0 |
| * |
| * Unless required by applicable law or agreed to in writing, software |
| * distributed under the License is distributed on an "AS IS" BASIS, |
| * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
| * See the License for the specific language governing permissions and |
| * limitations under the License. |
| */ |
| |
| #include <math.h> |
| #include <vector> |
| |
| #include <gtest/gtest.h> |
| |
| #include <audio_utils/channels.h> |
| |
| // TODO: Make a common include file for helper functions. |
| |
| template<typename T> |
| void checkMonotone(const T *ary, size_t size) |
| { |
| for (size_t i = 1; i < size; ++i) { |
| EXPECT_LT(ary[i-1], ary[i]); |
| } |
| } |
| |
| template<typename T> |
| void checkUnsignedMonotoneOrZero(const T *ary, size_t size) |
| { |
| if (size == 0) return; |
| |
| T least = ary[0]; |
| for (size_t i = 1; i < size; ++i) { |
| if (ary[i]) { |
| EXPECT_LT(least, ary[i]); |
| least = ary[i]; |
| } |
| } |
| } |
| |
| template<typename T> |
| void expectEq(const T &c1, const T &c2) { |
| EXPECT_EQ(c1.size(), c2.size()); |
| EXPECT_EQ(0, memcmp(c1.data(), c2.data(), sizeof(c1[0]) * std::min(c1.size(), c2.size()))); |
| } |
| |
| TEST(audio_utils_channels, geometry_constexpr) { |
| using namespace android::audio_utils::channels; |
| // fails to compile if not const. |
| constexpr size_t RIGHT_IDX = 1; // bit position of AUDIO_CHANNEL_OUT_FRONT_RIGHT; |
| static constexpr AUDIO_GEOMETRY_SIDE checkConstexprSide = sideFromChannelIdx(RIGHT_IDX); |
| static constexpr AUDIO_GEOMETRY_HEIGHT checkConstexprHeight = heightFromChannelIdx(RIGHT_IDX); |
| static constexpr AUDIO_GEOMETRY_DEPTH checkConstexprDepth = depthFromChannelIdx(RIGHT_IDX); |
| (void) checkConstexprSide; |
| (void) checkConstexprHeight; |
| (void) checkConstexprDepth; |
| |
| static constexpr ssize_t leftIdx = pairIdxFromChannelIdx(RIGHT_IDX); |
| ASSERT_EQ(0, leftIdx); |
| } |
| |
| TEST(audio_utils_channels, geometry_range) { |
| using namespace android::audio_utils::channels; |
| for (size_t i = 0; i < FCC_26 + 2 /* sic */; ++i) { |
| const AUDIO_GEOMETRY_SIDE side = sideFromChannelIdx(i); |
| const AUDIO_GEOMETRY_HEIGHT height = heightFromChannelIdx(i); |
| const AUDIO_GEOMETRY_DEPTH depth = depthFromChannelIdx(i); |
| ASSERT_TRUE(side == AUDIO_GEOMETRY_SIDE_LEFT |
| || side == AUDIO_GEOMETRY_SIDE_RIGHT |
| || side == AUDIO_GEOMETRY_SIDE_CENTER); |
| ASSERT_TRUE(height == AUDIO_GEOMETRY_HEIGHT_BOTTOM |
| || height == AUDIO_GEOMETRY_HEIGHT_MIDDLE |
| || height == AUDIO_GEOMETRY_HEIGHT_TOP); |
| ASSERT_TRUE(depth == AUDIO_GEOMETRY_DEPTH_FRONT |
| || depth == AUDIO_GEOMETRY_DEPTH_MIDDLE |
| || depth == AUDIO_GEOMETRY_DEPTH_BACK); |
| } |
| } |
| |
| TEST(audio_utils_channels, array_lr_pair_matching) { |
| using namespace android::audio_utils::channels; |
| for (size_t i = 0; i < FCC_26; ++i) { |
| const AUDIO_GEOMETRY_SIDE side = sideFromChannelIdx(i); |
| const ssize_t pairIdx = pairIdxFromChannelIdx(i); |
| switch (side) { |
| case AUDIO_GEOMETRY_SIDE_LEFT: |
| case AUDIO_GEOMETRY_SIDE_RIGHT: { |
| ASSERT_GE(pairIdx, 0); |
| ASSERT_LT(pairIdx, FCC_26); |
| const AUDIO_GEOMETRY_SIDE pairSide = side == AUDIO_GEOMETRY_SIDE_LEFT |
| ? AUDIO_GEOMETRY_SIDE_RIGHT : AUDIO_GEOMETRY_SIDE_LEFT; |
| ASSERT_EQ(pairSide, sideFromChannelIdx(pairIdx)); |
| } break; |
| case AUDIO_GEOMETRY_SIDE_CENTER: |
| ASSERT_EQ(-1, pairIdx); |
| break; |
| } |
| } |
| } |
| |
| TEST(audio_utils_channels, adjust_channels) { |
| constexpr size_t size = 65536; |
| std::vector<uint16_t> u16ref(size); |
| std::vector<uint16_t> u16expand(size * 2); |
| std::vector<uint16_t> u16ary(size); |
| |
| // reference buffer is monotonic. |
| for (size_t i = 0; i < u16ref.size(); ++i) { |
| u16ref[i] = i; |
| } |
| |
| // expand channels from stereo to quad. |
| adjust_channels( |
| u16ref.data() /*in_buff*/, |
| 2 /*in_channels*/, |
| u16expand.data() /*out_buff*/, |
| 4 /*out_channels*/, |
| sizeof(u16ref[0]) /*sample_size_in_bytes*/, |
| sizeof(u16ref[0]) * u16ref.size() /*num_in_bytes*/); |
| |
| // expanded buffer must increase (or be zero). |
| checkUnsignedMonotoneOrZero(u16expand.data(), u16expand.size()); |
| |
| // contract channels back to stereo. |
| adjust_channels( |
| u16expand.data() /*in_buff*/, |
| 4 /*in_channels*/, |
| u16ary.data() /*out_buff*/, |
| 2 /*out_channels*/, |
| sizeof(u16expand[0]) /*sample_size_in_bytes*/, |
| sizeof(u16expand[0]) * u16expand.size() /*num_in_bytes*/); |
| |
| // contracted array must be identical to original. |
| expectEq(u16ary, u16ref); |
| } |
| |
| TEST(audio_utils_channels, adjust_selected_channels) { |
| constexpr size_t size = 65536; |
| std::vector<uint16_t> u16ref(size); |
| std::vector<uint16_t> u16contract(size / 2); |
| std::vector<uint16_t> u16ary(size); |
| |
| // reference buffer is monotonic. |
| for (size_t i = 0; i < u16ref.size(); ++i) { |
| u16ref[i] = i; |
| } |
| |
| // contract from quad to stereo. |
| adjust_selected_channels( |
| u16ref.data() /*in_buff*/, |
| 4 /*in_channels*/, |
| u16contract.data() /*out_buff*/, |
| 2 /*out_channels*/, |
| sizeof(u16ref[0]) /*sample_size_in_bytes*/, |
| sizeof(u16ref[0]) * u16ref.size() /*num_in_bytes*/); |
| |
| // contracted buffer must increase. |
| checkMonotone(u16contract.data(), u16contract.size()); |
| |
| // initialize channels 3 and 4 of final comparison array. |
| for (size_t i = 0; i < u16ary.size() / 4; ++i) { |
| u16ary[i * 4 + 2] = u16ref[i * 4 + 2]; |
| u16ary[i * 4 + 3] = u16ref[i * 4 + 3]; |
| } |
| |
| // expand stereo into channels 1 and 2 of quad comparison array. |
| adjust_selected_channels( |
| u16contract.data() /*in_buff*/, |
| 2 /*in_channels*/, |
| u16ary.data() /*out_buff*/, |
| 4 /*out_channels*/, |
| sizeof(u16contract[0]) /*sample_size_in_bytes*/, |
| sizeof(u16contract[0]) * u16contract.size() /*num_in_bytes*/); |
| |
| // comparison array must be identical to original. |
| expectEq(u16ary, u16ref); |
| } |
| |
| TEST(audio_utils_channels, adjust_channels_non_destructive) { |
| constexpr size_t size = 65536; /* arbitrary large multiple of 8 */ |
| std::vector<uint16_t> u16ref(size); |
| std::vector<uint16_t> u16contracted(size); |
| std::vector<uint16_t> u16expanded(size); |
| std::vector<uint16_t> u16inout(size); |
| |
| // Reference buffer increases monotonically. |
| // For second test, in/out buffer begins identical to ref. |
| for (size_t i = 0; i < u16ref.size(); ++i) { |
| u16ref[i] = i; |
| u16inout[i] = i; |
| } |
| |
| // *** First test: different in/out buffers *** |
| |
| // Contract from quad to stereo. |
| adjust_channels_non_destructive( |
| u16ref.data() /*in_buff*/, |
| 4 /*in_channels*/, |
| u16contracted.data() /*out_buff*/, |
| 2 /*out_channels*/, |
| sizeof(u16ref[0]) /*sample_size_in_bytes*/, |
| sizeof(u16ref[0]) * u16ref.size() /*num_in_bytes*/); |
| |
| // Each half of contracted buffer should increase monotonically. |
| checkMonotone(u16contracted.data(), u16contracted.size() / 2); |
| checkMonotone(&u16contracted[u16contracted.size() / 2], u16contracted.size() / 2); |
| |
| // Expand stereo to quad |
| adjust_channels_non_destructive( |
| u16contracted.data() /*in_buff*/, |
| 2 /*in_channels*/, |
| u16expanded.data() /*out_buff*/, |
| 4 /*out_channels*/, |
| sizeof(u16contracted[0]) /*sample_size_in_bytes*/, |
| sizeof(u16contracted[0]) * (u16contracted.size() / 2) /*num_in_bytes*/); |
| |
| // Comparison array must be identical to reference. |
| expectEq(u16expanded, u16ref); |
| |
| // *** Second test: in_buff == out_buff *** |
| |
| // Contract from eight channels to stereo. |
| adjust_channels_non_destructive( |
| u16inout.data() /*in_buff*/, |
| 8 /*in_channels*/, |
| u16inout.data() /*out_buff*/, |
| 2 /*out_channels*/, |
| sizeof(u16inout[0]) /*sample_size_in_bytes*/, |
| sizeof(u16inout[0]) * u16inout.size() /*num_in_bytes*/); |
| |
| // Each section [1/4][3/4] of contracted buffer should increase monotonically. |
| checkMonotone(u16inout.data(), u16inout.size() / 4); |
| checkMonotone(&u16inout[u16inout.size() / 4], (u16inout.size() * 3) / 4); |
| |
| // Expand stereo to eight channels. |
| adjust_channels_non_destructive( |
| u16inout.data() /*in_buff*/, |
| 2 /*in_channels*/, |
| u16inout.data() /*out_buff*/, |
| 8 /*out_channels*/, |
| sizeof(u16inout[0]) /*sample_size_in_bytes*/, |
| sizeof(u16inout[0]) * (u16inout.size() / 4) /*num_in_bytes*/); |
| |
| // Comparison array must be identical to reference. |
| expectEq(u16inout, u16ref); |
| } |