1 /*
2  * Copyright (C) 2017 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 #define LOG_TAG "AAudioMixer"
18 //#define LOG_NDEBUG 0
19 #include <utils/Log.h>
20 
21 #define ATRACE_TAG ATRACE_TAG_AUDIO
22 
23 #include <cstring>
24 #include <utils/Trace.h>
25 
26 #include "AAudioMixer.h"
27 
28 #ifndef AAUDIO_MIXER_ATRACE_ENABLED
29 #define AAUDIO_MIXER_ATRACE_ENABLED    1
30 #endif
31 
32 using android::WrappingBuffer;
33 using android::FifoBuffer;
34 using android::fifo_frames_t;
35 
allocate(int32_t samplesPerFrame,int32_t framesPerBurst)36 void AAudioMixer::allocate(int32_t samplesPerFrame, int32_t framesPerBurst) {
37     mSamplesPerFrame = samplesPerFrame;
38     mFramesPerBurst = framesPerBurst;
39     int32_t samplesPerBuffer = samplesPerFrame * framesPerBurst;
40     mOutputBuffer = std::make_unique<float[]>(samplesPerBuffer);
41     mBufferSizeInBytes = samplesPerBuffer * sizeof(float);
42 }
43 
clear()44 void AAudioMixer::clear() {
45     memset(mOutputBuffer.get(), 0, mBufferSizeInBytes);
46 }
47 
mix(int streamIndex,const std::shared_ptr<FifoBuffer> & fifo,bool allowUnderflow)48 int32_t AAudioMixer::mix(
49         int streamIndex, const std::shared_ptr<FifoBuffer>& fifo, bool allowUnderflow) {
50     WrappingBuffer wrappingBuffer;
51     float *destination = mOutputBuffer.get();
52 
53 #if AAUDIO_MIXER_ATRACE_ENABLED
54     ATRACE_BEGIN("aaMix");
55 #endif /* AAUDIO_MIXER_ATRACE_ENABLED */
56 
57     // Gather the data from the client. May be in two parts.
58     fifo_frames_t fullFrames = fifo->getFullDataAvailable(&wrappingBuffer);
59 #if AAUDIO_MIXER_ATRACE_ENABLED
60     if (ATRACE_ENABLED()) {
61         char rdyText[] = "aaMixRdy#";
62         char letter = 'A' + (streamIndex % 26);
63         rdyText[sizeof(rdyText) - 2] = letter;
64         ATRACE_INT(rdyText, fullFrames);
65     }
66 #else /* MIXER_ATRACE_ENABLED */
67     (void) trackIndex;
68 #endif /* AAUDIO_MIXER_ATRACE_ENABLED */
69 
70     // If allowUnderflow then always advance by one burst even if we do not have the data.
71     // Otherwise the stream timing will drift whenever there is an underflow.
72     // This actual underflow can then be detected by the client for XRun counting.
73     //
74     // Generally, allowUnderflow will be false when stopping a stream and we want to
75     // use up whatever data is in the queue.
76     fifo_frames_t framesDesired = mFramesPerBurst;
77     if (!allowUnderflow && fullFrames < framesDesired) {
78         framesDesired = fullFrames; // just use what is available then stop
79     }
80 
81     // Mix data in one or two parts.
82     int partIndex = 0;
83     int32_t framesLeft = framesDesired;
84     while (framesLeft > 0 && partIndex < WrappingBuffer::SIZE) {
85         fifo_frames_t framesToMixFromPart = framesLeft;
86         fifo_frames_t framesAvailableFromPart = wrappingBuffer.numFrames[partIndex];
87         if (framesAvailableFromPart > 0) {
88             if (framesToMixFromPart > framesAvailableFromPart) {
89                 framesToMixFromPart = framesAvailableFromPart;
90             }
91             mixPart(destination, (float *)wrappingBuffer.data[partIndex],
92                     framesToMixFromPart);
93 
94             destination += framesToMixFromPart * mSamplesPerFrame;
95             framesLeft -= framesToMixFromPart;
96         }
97         partIndex++;
98     }
99     fifo->advanceReadIndex(framesDesired);
100 
101 #if AAUDIO_MIXER_ATRACE_ENABLED
102     ATRACE_END();
103 #endif /* AAUDIO_MIXER_ATRACE_ENABLED */
104 
105     return (framesDesired - framesLeft); // framesRead
106 }
107 
mixPart(float * destination,float * source,int32_t numFrames)108 void AAudioMixer::mixPart(float *destination, float *source, int32_t numFrames) {
109     int32_t numSamples = numFrames * mSamplesPerFrame;
110     // TODO maybe optimize using SIMD
111     for (int sampleIndex = 0; sampleIndex < numSamples; sampleIndex++) {
112         *destination++ += *source++;
113     }
114 }
115 
getOutputBuffer()116 float *AAudioMixer::getOutputBuffer() {
117     return mOutputBuffer.get();
118 }
119