1 /*
2 * Copyright 2018 The Android Open Source Project
3 *
4 * Licensed under the Apache License, Version 2.0 (the "License");
5 * you may not use this file except in compliance with the License.
6 * You may obtain a copy of the License at
7 *
8 * http://www.apache.org/licenses/LICENSE-2.0
9 *
10 * Unless required by applicable law or agreed to in writing, software
11 * distributed under the License is distributed on an "AS IS" BASIS,
12 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13 * See the License for the specific language governing permissions and
14 * limitations under the License.
15 */
16
17 /*
18 * Test FlowGraph
19 */
20
21 #include "stdio.h"
22
23 #include <gtest/gtest.h>
24 #include <oboe/Oboe.h>
25
26 #include "flowgraph/ClipToRange.h"
27 #include "flowgraph/Limiter.h"
28 #include "flowgraph/MonoToMultiConverter.h"
29 #include "flowgraph/SourceFloat.h"
30 #include "flowgraph/RampLinear.h"
31 #include "flowgraph/SinkFloat.h"
32 #include "flowgraph/SinkI16.h"
33 #include "flowgraph/SinkI24.h"
34 #include "flowgraph/SinkI32.h"
35 #include "flowgraph/SourceI16.h"
36 #include "flowgraph/SourceI24.h"
37
38 using namespace oboe::flowgraph;
39
40 constexpr int kBytesPerI24Packed = 3;
41
TEST(test_flowgraph,module_sinki16)42 TEST(test_flowgraph, module_sinki16) {
43 static const float input[] = {1.0f, 0.5f, -0.25f, -1.0f, 0.0f, 53.9f, -87.2f};
44 static const int16_t expected[] = {32767, 16384, -8192, -32768, 0, 32767, -32768};
45 int16_t output[20];
46 SourceFloat sourceFloat{1};
47 SinkI16 sinkI16{1};
48
49 int numInputFrames = sizeof(input) / sizeof(input[0]);
50 sourceFloat.setData(input, numInputFrames);
51 sourceFloat.output.connect(&sinkI16.input);
52
53 int numOutputFrames = sizeof(output) / sizeof(int16_t);
54 int32_t numRead = sinkI16.read(output, numOutputFrames);
55 ASSERT_EQ(numInputFrames, numRead);
56 for (int i = 0; i < numRead; i++) {
57 EXPECT_EQ(expected[i], output[i]);
58 }
59 }
60
TEST(test_flowgraph,module_mono_to_stereo)61 TEST(test_flowgraph, module_mono_to_stereo) {
62 static const float input[] = {1.0f, 2.0f, 3.0f};
63 float output[100] = {};
64 SourceFloat sourceFloat{1};
65 MonoToMultiConverter monoToStereo{2};
66 SinkFloat sinkFloat{2};
67
68 sourceFloat.setData(input, 3);
69
70 sourceFloat.output.connect(&monoToStereo.input);
71 monoToStereo.output.connect(&sinkFloat.input);
72
73 int32_t numRead = sinkFloat.read(output, 8);
74 ASSERT_EQ(3, numRead);
75 EXPECT_EQ(input[0], output[0]);
76 EXPECT_EQ(input[0], output[1]);
77 EXPECT_EQ(input[1], output[2]);
78 EXPECT_EQ(input[1], output[3]);
79 }
80
TEST(test_flowgraph,module_ramp_linear)81 TEST(test_flowgraph, module_ramp_linear) {
82 constexpr int singleNumOutput = 1;
83 constexpr int rampSize = 5;
84 constexpr int numOutput = 100;
85 constexpr float value = 1.0f;
86 constexpr float initialTarget = 10.0f;
87 constexpr float finalTarget = 100.0f;
88 constexpr float tolerance = 0.0001f; // arbitrary
89 float output[numOutput] = {};
90 RampLinear rampLinear{1};
91 SinkFloat sinkFloat{1};
92
93 rampLinear.input.setValue(value);
94 rampLinear.setLengthInFrames(rampSize);
95 rampLinear.output.connect(&sinkFloat.input);
96
97 // Check that the values go to the initial target instantly.
98 rampLinear.setTarget(initialTarget);
99 int32_t singleNumRead = sinkFloat.read(output, singleNumOutput);
100 ASSERT_EQ(singleNumRead, singleNumOutput);
101 EXPECT_NEAR(value * initialTarget, output[0], tolerance);
102
103 // Now set target and check that the linear ramp works as expected.
104 rampLinear.setTarget(finalTarget);
105 int32_t numRead = sinkFloat.read(output, numOutput);
106 const float incrementSize = (finalTarget - initialTarget) / rampSize;
107 ASSERT_EQ(numOutput, numRead);
108
109 int i = 0;
110 for (; i < rampSize; i++) {
111 float expected = value * (initialTarget + i * incrementSize);
112 EXPECT_NEAR(expected, output[i], tolerance);
113 }
114 for (; i < numOutput; i++) {
115 float expected = value * finalTarget;
116 EXPECT_NEAR(expected, output[i], tolerance);
117 }
118 }
119
120 // It is easiest to represent packed 24-bit data as a byte array.
121 // This test will read from input, convert to float, then write
122 // back to output as bytes.
TEST(test_flowgraph,module_packed_24)123 TEST(test_flowgraph, module_packed_24) {
124 static const uint8_t input[] = {0x01, 0x23, 0x45,
125 0x67, 0x89, 0xAB,
126 0xCD, 0xEF, 0x5A};
127 uint8_t output[99] = {};
128 SourceI24 sourceI24{1};
129 SinkI24 sinkI24{1};
130
131 int numInputFrames = sizeof(input) / kBytesPerI24Packed;
132 sourceI24.setData(input, numInputFrames);
133 sourceI24.output.connect(&sinkI24.input);
134
135 int32_t numRead = sinkI24.read(output, sizeof(output) / kBytesPerI24Packed);
136 ASSERT_EQ(numInputFrames, numRead);
137 for (size_t i = 0; i < sizeof(input); i++) {
138 EXPECT_EQ(input[i], output[i]);
139 }
140 }
141
TEST(test_flowgraph,module_clip_to_range)142 TEST(test_flowgraph, module_clip_to_range) {
143 constexpr float myMin = -2.0f;
144 constexpr float myMax = 1.5f;
145
146 static const float input[] = {-9.7, 0.5f, -0.25, 1.0f, 12.3};
147 static const float expected[] = {myMin, 0.5f, -0.25, 1.0f, myMax};
148 float output[100];
149 SourceFloat sourceFloat{1};
150 ClipToRange clipper{1};
151 SinkFloat sinkFloat{1};
152
153 int numInputFrames = sizeof(input) / sizeof(input[0]);
154 sourceFloat.setData(input, numInputFrames);
155
156 clipper.setMinimum(myMin);
157 clipper.setMaximum(myMax);
158
159 sourceFloat.output.connect(&clipper.input);
160 clipper.output.connect(&sinkFloat.input);
161
162 int numOutputFrames = sizeof(output) / sizeof(output[0]);
163 int32_t numRead = sinkFloat.read(output, numOutputFrames);
164 ASSERT_EQ(numInputFrames, numRead);
165 constexpr float tolerance = 0.000001f; // arbitrary
166 for (int i = 0; i < numRead; i++) {
167 EXPECT_NEAR(expected[i], output[i], tolerance);
168 }
169 }
170
TEST(test_flowgraph,module_sinki32)171 TEST(test_flowgraph, module_sinki32) {
172 static constexpr int kNumSamples = 8;
173 static const float input[] = {
174 1.0f, 0.5f, -0.25f, -1.0f,
175 0.0f, 53.9f, -87.2f, -1.02f};
176 static const int32_t expected[] = {
177 INT32_MAX, 1 << 30, INT32_MIN / 4, INT32_MIN,
178 0, INT32_MAX, INT32_MIN, INT32_MIN};
179 int32_t output[kNumSamples + 10]; // larger than input
180
181 SourceFloat sourceFloat{1};
182 SinkI32 sinkI32{1};
183
184 sourceFloat.setData(input, kNumSamples);
185 sourceFloat.output.connect(&sinkI32.input);
186
187 int numOutputFrames = sizeof(output) / sizeof(int32_t);
188 int32_t numRead = sinkI32.read(output, numOutputFrames);
189 ASSERT_EQ(kNumSamples, numRead);
190 for (int i = 0; i < numRead; i++) {
191 EXPECT_EQ(expected[i], output[i]) << ", i = " << i;
192 }
193 }
194
TEST(test_flowgraph,module_limiter)195 TEST(test_flowgraph, module_limiter) {
196 constexpr int kNumSamples = 101;
197 constexpr float kLastSample = 3.0f;
198 constexpr float kFirstSample = -kLastSample;
199 constexpr float kDeltaBetweenSamples = (kLastSample - kFirstSample) / (kNumSamples - 1);
200 constexpr float kTolerance = 0.00001f;
201
202 float input[kNumSamples];
203 float output[kNumSamples];
204 SourceFloat sourceFloat{1};
205 Limiter limiter{1};
206 SinkFloat sinkFloat{1};
207
208 for (int i = 0; i < kNumSamples; i++) {
209 input[i] = kFirstSample + i * kDeltaBetweenSamples;
210 }
211
212 const int numInputFrames = std::size(input);
213 sourceFloat.setData(input, numInputFrames);
214
215 sourceFloat.output.connect(&limiter.input);
216 limiter.output.connect(&sinkFloat.input);
217
218 const int numOutputFrames = std::size(output);
219 int32_t numRead = sinkFloat.read(output, numOutputFrames);
220 ASSERT_EQ(numInputFrames, numRead);
221
222 for (int i = 0; i < numRead; i++) {
223 // limiter must be symmetric wrt 0.
224 EXPECT_NEAR(output[i], -output[kNumSamples - i - 1], kTolerance);
225 if (i > 0) {
226 EXPECT_GE(output[i], output[i - 1]); // limiter must be monotonic
227 }
228 if (input[i] == 0.f) {
229 EXPECT_EQ(0.f, output[i]);
230 } else if (input[i] > 0.0f) {
231 EXPECT_GE(output[i], 0.0f);
232 EXPECT_LE(output[i], M_SQRT2); // limiter actually limits
233 EXPECT_LE(output[i], input[i]); // a limiter, gain <= 1
234 } else {
235 EXPECT_LE(output[i], 0.0f);
236 EXPECT_GE(output[i], -M_SQRT2); // limiter actually limits
237 EXPECT_GE(output[i], input[i]); // a limiter, gain <= 1
238 }
239 if (-1.f <= input[i] && input[i] <= 1.f) {
240 EXPECT_EQ(input[i], output[i]);
241 }
242 }
243 }
244
TEST(test_flowgraph,module_limiter_nan)245 TEST(test_flowgraph, module_limiter_nan) {
246 constexpr int kArbitraryOutputSize = 100;
247 constexpr float kFloatNan = NAN;
248 static const float input[] = {kFloatNan, 0.5f, kFloatNan, kFloatNan, -10.0f, kFloatNan};
249 static const float expected[] = {0.0f, 0.5f, 0.5f, 0.5f, -M_SQRT2, -M_SQRT2};
250 constexpr float tolerance = 0.00001f;
251 float output[kArbitraryOutputSize];
252 SourceFloat sourceFloat{1};
253 Limiter limiter{1};
254 SinkFloat sinkFloat{1};
255
256 const int numInputFrames = std::size(input);
257 sourceFloat.setData(input, numInputFrames);
258
259 sourceFloat.output.connect(&limiter.input);
260 limiter.output.connect(&sinkFloat.input);
261
262 const int numOutputFrames = std::size(output);
263 int32_t numRead = sinkFloat.read(output, numOutputFrames);
264 ASSERT_EQ(numInputFrames, numRead);
265
266 for (int i = 0; i < numRead; i++) {
267 EXPECT_NEAR(expected[i], output[i], tolerance);
268 }
269 }
270