tracktion_AudioBufferStack.h

Go to the documentation of this file.

 /*
    ,--.                     ,--.     ,--.  ,--.
  ,-'  '-.,--.--.,--,--.,---.|  |,-.,-'  '-.`--' ,---. ,--,--,      Copyright 2024
  '-.  .-'|  .--' ,-.  | .--'|     /'-.  .-',--.| .-. ||      \   Tracktion Software
    |  |  |  |  \ '-'  \ `--.|  \  \  |  |  |  |' '-' '|  ||  |       Corporation
    `---' `--'   `--`--'`---'`--'`--' `---' `--' `---' `--''--'    www.tracktion.com
 
    Tracktion Engine uses a GPL/commercial licence - see LICENCE.md for details.
*/
 
#include <stack>
 
namespace tracktion { inline namespace graph
{
 
//==============================================================================
class AudioBufferStack
{
public:
    AudioBufferStack() = default;
 
    choc::buffer::ChannelArrayBuffer<float> allocate (choc::buffer::Size);
 
    void release (choc::buffer::ChannelArrayBuffer<float>&&);
 
    //==============================================================================
    void reset();
 
    void reserve (size_t numBuffers, choc::buffer::Size);
 
    //==============================================================================
    size_t getNumBuffers();
 
    size_t getAllocatedSize();
 
private:
    std::stack<choc::buffer::ChannelArrayBuffer<float>> stack;
};
 
 
//==============================================================================
//        _        _           _  _
//     __| |  ___ | |_   __ _ (_)| | ___
//    / _` | / _ \| __| / _` || || |/ __|
//   | (_| ||  __/| |_ | (_| || || |\__ \ _  _  _
//    \__,_| \___| \__| \__,_||_||_||___/(_)(_)(_)
//
//   Code beyond this point is implementation detail...
//
//==============================================================================
inline choc::buffer::ChannelArrayBuffer<float> AudioBufferStack::allocate (choc::buffer::Size size)
{
    choc::buffer::ChannelArrayBuffer<float> buffer;
 
    if (stack.size() > 0)
    {
        buffer = std::move (stack.top());
        stack.pop();
 
        if (auto bufferSize = buffer.getSize();
            bufferSize.numChannels < size.numChannels
            || bufferSize.numFrames < size.numFrames)
        {
            buffer.resize (size);
        }
    }
    else
    {
        buffer.resize (size);
    }
 
    return buffer;
}
 
inline void AudioBufferStack::release (choc::buffer::ChannelArrayBuffer<float>&& buffer)
{
    stack.push (std::move (buffer));
}
 
//==============================================================================
inline void AudioBufferStack::reserve (size_t numBuffers, choc::buffer::Size size)
{
    std::vector<choc::buffer::ChannelArrayBuffer<float>> buffers;
 
    // Remove all the buffers
    for (size_t i = 0; i < std::min (numBuffers, stack.size()); ++i)
    {
        buffers.emplace_back (std::move (stack.top()));
        stack.pop();
    }
 
    // Ensure their size is as big as required
    for (auto& b : buffers)
    {
        if (auto bufferSize = b.getSize();
            bufferSize.numChannels < size.numChannels
            || bufferSize.numFrames < size.numFrames)
        {
            b.resize (size);
        }
    }
 
    // Reset the fifo storage to hold the new number of buffers
    const int numToAdd = static_cast<int> (numBuffers) - static_cast<int> (buffers.size());
    assert (numToAdd >= 0);
 
    // Push the temp buffers back
    for (auto& b : buffers)
        stack.push (std::move (b));
 
    // Push any additional buffers
    for (int i = 0; i < numToAdd; ++i)
        stack.push (choc::buffer::ChannelArrayBuffer<float> (size));
}
 
inline size_t AudioBufferStack::getNumBuffers()
{
    return stack.size();
}
 
inline size_t AudioBufferStack::getAllocatedSize()
{
    if (stack.size() == 0)
        return 0;
 
    auto& b = stack.top();
    return b.getView().data.getBytesNeeded (b.getSize());
}
 
}} // namespace tracktion