| 1 | /* | 
| 2 | Copyright 2018 Google Inc. All Rights Reserved. | 
| 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 | #ifndef RESONANCE_AUDIO_UTILS_BUFFER_UNPARTITIONER_H_ | 
| 18 | #define RESONANCE_AUDIO_UTILS_BUFFER_UNPARTITIONER_H_ | 
| 19 |  | 
| 20 | #include <functional> | 
| 21 | #include <memory> | 
| 22 |  | 
| 23 | #include "base/audio_buffer.h" | 
| 24 | #include "base/logging.h" | 
| 25 |  | 
| 26 | namespace vraudio { | 
| 27 |  | 
| 28 | // Unpackages |AudioBuffer| into output buffers of arbitrary sizes. | 
| 29 | class BufferUnpartitioner { | 
| 30 |  public: | 
| 31 |   // Callback to obtain input |AudioBuffer|s. Returns a nullptr if no buffers | 
| 32 |   // are available. | 
| 33 |   typedef std::function<const AudioBuffer*()> GetBufferCallback; | 
| 34 |  | 
| 35 |   // Constructor. | 
| 36 |   // | 
| 37 |   // @param num_channels Number of audio channels in input and output buffers. | 
| 38 |   // @param frames_per_buffer Number of frames per input buffer. | 
| 39 |   // @param buffer_callback Callback to receive output |AudioBuffer|s. | 
| 40 |   BufferUnpartitioner(size_t num_channels, size_t frames_per_buffer, | 
| 41 |                       GetBufferCallback buffer_callback); | 
| 42 |  | 
| 43 |   // Returns the number of input buffers required for a given number of output | 
| 44 |   // frames and based on the current fill state of the internal |temp_buffer_|. | 
| 45 |   // | 
| 46 |   // @param num_output_frames Number of output frames. | 
| 47 |   // @return Number of required input |AudioBuffer|s. | 
| 48 |   size_t GetNumBuffersRequestedForNumInputFrames( | 
| 49 |       size_t num_output_frames) const; | 
| 50 |  | 
| 51 |   // Returns the number of buffered frames in internal |temp_buffer_|. | 
| 52 |   size_t GetNumBufferedFrames() const; | 
| 53 |  | 
| 54 |   // Requests an interleaved int16 output buffer. This method triggers | 
| 55 |   // |GetBufferCallback|. | 
| 56 |   // | 
| 57 |   // @param output_buffer Interleaved output buffer to write to. | 
| 58 |   // @param num_channels Number of channels in output buffer. | 
| 59 |   // @param num_frames Number of frames in output buffer. | 
| 60 |   // @return Number of frames actually written. | 
| 61 |   size_t GetBuffer(int16* output_buffer, size_t num_channels, | 
| 62 |                    size_t num_frames); | 
| 63 |  | 
| 64 |   // Requests an interleaved float output buffer. This method triggers | 
| 65 |   // |GetBufferCallback|. | 
| 66 |   // | 
| 67 |   // @param output_buffer Interleaved output buffer to write to. | 
| 68 |   // @param num_channels Number of channels in output buffer. | 
| 69 |   // @param num_frames Number of frames in output buffer. | 
| 70 |   // @return Number of frames actually written. | 
| 71 |   size_t GetBuffer(float* output_buffer, size_t num_channels, | 
| 72 |                    size_t num_frames); | 
| 73 |  | 
| 74 |   // Requests a planar int16 output buffer. This method triggers | 
| 75 |   // |GetBufferCallback|. | 
| 76 |   // | 
| 77 |   // @param output_buffer Planar output buffer to write to. | 
| 78 |   // @param num_channels Number of channels in output buffer. | 
| 79 |   // @param num_frames Number of frames in output buffer. | 
| 80 |   // @return Number of frames actually written. | 
| 81 |   size_t GetBuffer(int16** output_buffer, size_t num_channels, | 
| 82 |                    size_t num_frames); | 
| 83 |  | 
| 84 |   // Requests a planar float output buffer. This method triggers | 
| 85 |   // |GetBufferCallback|. | 
| 86 |   // | 
| 87 |   // @param output_buffer Planar output buffer to write to. | 
| 88 |   // @param num_channels Number of channels in output buffer. | 
| 89 |   // @param num_frames Number of frames in output buffer. | 
| 90 |   // @return Number of frames actually written. | 
| 91 |   size_t GetBuffer(float** output_buffer, size_t num_channels, | 
| 92 |                    size_t num_frames); | 
| 93 |  | 
| 94 |   // Clears internal temporary buffer that holds remaining audio frames. | 
| 95 |   void Clear(); | 
| 96 |  | 
| 97 |  private: | 
| 98 |   // Returns the number of frames that are buffered in |current_input_buffer_|. | 
| 99 |   // | 
| 100 |   // @return Number of frames in |current_input_buffer_|. If | 
| 101 |   //     |current_input_buffer_| is undefined, it returns 0. | 
| 102 |   size_t GetNumFramesAvailableInBuffer() const; | 
| 103 |  | 
| 104 |   // Writes output buffer into target buffer. Supported buffer types are planar | 
| 105 |   // and interleaved floating point abd interleaved int16 output. This method | 
| 106 |   // triggers |GetBufferCallback|. | 
| 107 |   // | 
| 108 |   // @tparam BufferType Output buffer type. | 
| 109 |   // @param buffer Output buffer to write to. | 
| 110 |   // @param num_channels Number of channels in output buffer. | 
| 111 |   // @param num_frames Number of frames in output buffer. | 
| 112 |   // @return Number of frames actually written. | 
| 113 |   template <typename BufferType> | 
| 114 |   size_t GetBufferTemplated(BufferType buffer, size_t num_channels, | 
| 115 |                             size_t num_frames); | 
| 116 |  | 
| 117 |   // Number of channels in output buffers. | 
| 118 |   const size_t num_channels_; | 
| 119 |  | 
| 120 |   // Number of frames per buffer in output buffers. | 
| 121 |   const size_t frames_per_buffer_; | 
| 122 |  | 
| 123 |   // Callback to request input |AudioBuffer|s. | 
| 124 |   const GetBufferCallback buffer_callback_; | 
| 125 |  | 
| 126 |   // Temporary buffer containing remaining audio frames. | 
| 127 |   const AudioBuffer* current_input_buffer_;  // Not owned. | 
| 128 |  | 
| 129 |   // Current read position in |current_input_buffer_|. | 
| 130 |   size_t current_buffer_read_offset_frames_; | 
| 131 | }; | 
| 132 |  | 
| 133 | }  // namespace vraudio | 
| 134 |  | 
| 135 | #endif  // RESONANCE_AUDIO_UTILS_BUFFER_UNPARTITIONER_H_ | 
| 136 |  |