1 /******************************************************************************
2 *
3 * Copyright 2017 The Android Open Source Project
4 *
5 * Licensed under the Apache License, Version 2.0 (the "License");
6 * you may not use this file except in compliance with the License.
7 * You may obtain a copy of the License at:
8 *
9 * http://www.apache.org/licenses/LICENSE-2.0
10 *
11 * Unless required by applicable law or agreed to in writing, software
12 * distributed under the License is distributed on an "AS IS" BASIS,
13 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
14 * See the License for the specific language governing permissions and
15 * limitations under the License.
16 *
17 ******************************************************************************/
18
19 #include "audio_hearing_aid_hw/include/audio_hearing_aid_hw.h"
20
21 #include <gtest/gtest.h>
22
23 namespace {
codec_sample_rate2value(btav_a2dp_codec_sample_rate_t codec_sample_rate)24 static uint32_t codec_sample_rate2value(btav_a2dp_codec_sample_rate_t codec_sample_rate) {
25 switch (codec_sample_rate) {
26 case BTAV_A2DP_CODEC_SAMPLE_RATE_44100:
27 return 44100;
28 case BTAV_A2DP_CODEC_SAMPLE_RATE_48000:
29 return 48000;
30 case BTAV_A2DP_CODEC_SAMPLE_RATE_88200:
31 return 88200;
32 case BTAV_A2DP_CODEC_SAMPLE_RATE_96000:
33 return 96000;
34 case BTAV_A2DP_CODEC_SAMPLE_RATE_176400:
35 return 176400;
36 case BTAV_A2DP_CODEC_SAMPLE_RATE_192000:
37 return 192000;
38 case BTAV_A2DP_CODEC_SAMPLE_RATE_16000:
39 return 16000;
40 case BTAV_A2DP_CODEC_SAMPLE_RATE_24000:
41 return 24000;
42 case BTAV_A2DP_CODEC_SAMPLE_RATE_NONE:
43 break;
44 }
45 return 0;
46 }
47
codec_bits_per_sample2value(btav_a2dp_codec_bits_per_sample_t codec_bits_per_sample)48 static uint32_t codec_bits_per_sample2value(
49 btav_a2dp_codec_bits_per_sample_t codec_bits_per_sample) {
50 switch (codec_bits_per_sample) {
51 case BTAV_A2DP_CODEC_BITS_PER_SAMPLE_16:
52 return 16;
53 case BTAV_A2DP_CODEC_BITS_PER_SAMPLE_24:
54 return 24;
55 case BTAV_A2DP_CODEC_BITS_PER_SAMPLE_32:
56 return 32;
57 case BTAV_A2DP_CODEC_BITS_PER_SAMPLE_NONE:
58 break;
59 }
60 return 0;
61 }
62
codec_channel_mode2value(btav_a2dp_codec_channel_mode_t codec_channel_mode)63 static uint32_t codec_channel_mode2value(btav_a2dp_codec_channel_mode_t codec_channel_mode) {
64 switch (codec_channel_mode) {
65 case BTAV_A2DP_CODEC_CHANNEL_MODE_MONO:
66 return 1;
67 case BTAV_A2DP_CODEC_CHANNEL_MODE_STEREO:
68 return 2;
69 case BTAV_A2DP_CODEC_CHANNEL_MODE_NONE:
70 break;
71 }
72 return 0;
73 }
74
75 } // namespace
76
77 class AudioA2dpHwTest : public ::testing::Test {
78 protected:
AudioA2dpHwTest()79 AudioA2dpHwTest() {}
80
81 private:
82 };
83
TEST_F(AudioA2dpHwTest,test_compute_buffer_size)84 TEST_F(AudioA2dpHwTest, test_compute_buffer_size) {
85 const btav_a2dp_codec_sample_rate_t codec_sample_rate_array[] = {
86 BTAV_A2DP_CODEC_SAMPLE_RATE_NONE, BTAV_A2DP_CODEC_SAMPLE_RATE_44100,
87 BTAV_A2DP_CODEC_SAMPLE_RATE_48000, BTAV_A2DP_CODEC_SAMPLE_RATE_88200,
88 BTAV_A2DP_CODEC_SAMPLE_RATE_96000, BTAV_A2DP_CODEC_SAMPLE_RATE_176400,
89 BTAV_A2DP_CODEC_SAMPLE_RATE_192000};
90
91 const btav_a2dp_codec_bits_per_sample_t codec_bits_per_sample_array[] = {
92 BTAV_A2DP_CODEC_BITS_PER_SAMPLE_NONE, BTAV_A2DP_CODEC_BITS_PER_SAMPLE_16,
93 BTAV_A2DP_CODEC_BITS_PER_SAMPLE_24, BTAV_A2DP_CODEC_BITS_PER_SAMPLE_32};
94
95 const btav_a2dp_codec_channel_mode_t codec_channel_mode_array[] = {
96 BTAV_A2DP_CODEC_CHANNEL_MODE_NONE, BTAV_A2DP_CODEC_CHANNEL_MODE_MONO,
97 BTAV_A2DP_CODEC_CHANNEL_MODE_STEREO};
98
99 for (const auto codec_sample_rate : codec_sample_rate_array) {
100 for (const auto codec_bits_per_sample : codec_bits_per_sample_array) {
101 for (const auto codec_channel_mode : codec_channel_mode_array) {
102 size_t buffer_size = audio_ha_hw_stream_compute_buffer_size(
103 codec_sample_rate, codec_bits_per_sample, codec_channel_mode);
104
105 // Check for invalid input
106 if ((codec_sample_rate == BTAV_A2DP_CODEC_SAMPLE_RATE_NONE) ||
107 (codec_bits_per_sample == BTAV_A2DP_CODEC_BITS_PER_SAMPLE_NONE) ||
108 (codec_channel_mode == BTAV_A2DP_CODEC_CHANNEL_MODE_NONE)) {
109 EXPECT_EQ(buffer_size, static_cast<size_t>(AUDIO_STREAM_OUTPUT_BUFFER_SZ));
110 continue;
111 }
112
113 uint32_t sample_rate = codec_sample_rate2value(codec_sample_rate);
114 EXPECT_NE(0u, sample_rate);
115
116 uint32_t bits_per_sample = codec_bits_per_sample2value(codec_bits_per_sample);
117 EXPECT_NE(0u, bits_per_sample);
118
119 uint32_t number_of_channels = codec_channel_mode2value(codec_channel_mode);
120 EXPECT_NE(0u, number_of_channels);
121
122 const uint64_t time_period_ms = 20; // TODO: Must be a parameter
123 size_t expected_buffer_size = (time_period_ms * AUDIO_STREAM_OUTPUT_BUFFER_PERIODS *
124 sample_rate * number_of_channels * (bits_per_sample / 8)) /
125 1000;
126
127 // Compute the divisor and adjust the buffer size
128 const size_t divisor =
129 (AUDIO_STREAM_OUTPUT_BUFFER_PERIODS * 16 * number_of_channels * bits_per_sample) /
130 8;
131 const size_t remainder = expected_buffer_size % divisor;
132 if (remainder != 0) {
133 expected_buffer_size += divisor - remainder;
134 }
135
136 EXPECT_EQ(buffer_size, expected_buffer_size);
137 }
138 }
139 }
140 }
141