1*05767d91SRobert Wu /*
2*05767d91SRobert Wu * Copyright 2018 The Android Open Source Project
3*05767d91SRobert Wu *
4*05767d91SRobert Wu * Licensed under the Apache License, Version 2.0 (the "License");
5*05767d91SRobert Wu * you may not use this file except in compliance with the License.
6*05767d91SRobert Wu * You may obtain a copy of the License at
7*05767d91SRobert Wu *
8*05767d91SRobert Wu * http://www.apache.org/licenses/LICENSE-2.0
9*05767d91SRobert Wu *
10*05767d91SRobert Wu * Unless required by applicable law or agreed to in writing, software
11*05767d91SRobert Wu * distributed under the License is distributed on an "AS IS" BASIS,
12*05767d91SRobert Wu * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13*05767d91SRobert Wu * See the License for the specific language governing permissions and
14*05767d91SRobert Wu * limitations under the License.
15*05767d91SRobert Wu */
16*05767d91SRobert Wu
17*05767d91SRobert Wu #include <algorithm>
18*05767d91SRobert Wu #include <unistd.h>
19*05767d91SRobert Wu
20*05767d91SRobert Wu #include "SinkI16.h"
21*05767d91SRobert Wu
22*05767d91SRobert Wu #if FLOWGRAPH_ANDROID_INTERNAL
23*05767d91SRobert Wu #include <audio_utils/primitives.h>
24*05767d91SRobert Wu #endif
25*05767d91SRobert Wu
26*05767d91SRobert Wu using namespace FLOWGRAPH_OUTER_NAMESPACE::flowgraph;
27*05767d91SRobert Wu
SinkI16(int32_t channelCount)28*05767d91SRobert Wu SinkI16::SinkI16(int32_t channelCount)
29*05767d91SRobert Wu : FlowGraphSink(channelCount) {}
30*05767d91SRobert Wu
read(void * data,int32_t numFrames)31*05767d91SRobert Wu int32_t SinkI16::read(void *data, int32_t numFrames) {
32*05767d91SRobert Wu int16_t *shortData = (int16_t *) data;
33*05767d91SRobert Wu const int32_t channelCount = input.getSamplesPerFrame();
34*05767d91SRobert Wu
35*05767d91SRobert Wu int32_t framesLeft = numFrames;
36*05767d91SRobert Wu while (framesLeft > 0) {
37*05767d91SRobert Wu // Run the graph and pull data through the input port.
38*05767d91SRobert Wu int32_t framesRead = pullData(framesLeft);
39*05767d91SRobert Wu if (framesRead <= 0) {
40*05767d91SRobert Wu break;
41*05767d91SRobert Wu }
42*05767d91SRobert Wu const float *signal = input.getBuffer();
43*05767d91SRobert Wu int32_t numSamples = framesRead * channelCount;
44*05767d91SRobert Wu #if FLOWGRAPH_ANDROID_INTERNAL
45*05767d91SRobert Wu memcpy_to_i16_from_float(shortData, signal, numSamples);
46*05767d91SRobert Wu shortData += numSamples;
47*05767d91SRobert Wu signal += numSamples;
48*05767d91SRobert Wu #else
49*05767d91SRobert Wu for (int i = 0; i < numSamples; i++) {
50*05767d91SRobert Wu int32_t n = (int32_t) (*signal++ * 32768.0f);
51*05767d91SRobert Wu *shortData++ = std::min(INT16_MAX, std::max(INT16_MIN, n)); // clip
52*05767d91SRobert Wu }
53*05767d91SRobert Wu #endif
54*05767d91SRobert Wu framesLeft -= framesRead;
55*05767d91SRobert Wu }
56*05767d91SRobert Wu return numFrames - framesLeft;
57*05767d91SRobert Wu }
58