ScopeDataCollector.h

Go to the documentation of this file.

/*
  ==============================================================================
 
    ScopeDataCollector.h
    Created: 24 Feb 2023 6:49:28pm
    Author:  wadda
 
  ==============================================================================
*/
 
#pragma once
#include "AudioBufferQueue.h"
#include <cstdlib>
 
template <typename SampleType>
class ScopeDataCollector
{
public:
    //==============================================================================
    ScopeDataCollector(AudioBufferQueue<SampleType> &queueToUse)
        : audioBufferQueue(queueToUse)
    {
        // resize buffer
        buffer.resize(queueToUse.getBufferSize());
        bufferSize = buffer.size();
 
        // reset num collected
        numCollected = 0;
    }
 
    //==============================================================================
    void process(const SampleType *data, size_t numSamples)
    {
        // declare variables
        size_t index = 0;
        int numBuffersToPush = numSamples / bufferSize;
        int numSamplesExceeding = numSamples % bufferSize;
        int numEmptyBuffers = 0;
        int additionalSamples = 0;
 
        // if no samples still collected
        if (state == State::waitingForTrigger)
        {
            std::fill(buffer.begin(), buffer.end(), 0);
            // If no sample over treshold push silence.
            if (std::all_of(data, data + numSamples, [](SampleType i)
                            { return i < triggerLevel; }))
            {
                for (size_t i = 0; i < numBuffersToPush; i++)
                {
                    audioBufferQueue.push(buffer.data(), bufferSize);
                }
                numCollected = numSamplesExceeding;
            }
            // Else setup collecting stage
            else
            {
                auto result = data;
                while (*result < triggerLevel)
                {
                    result++;
                }
                prevSample = *result;
                state = State::collecting;
                int firstSample = result - data;
                numEmptyBuffers = firstSample / bufferSize;
                additionalSamples = firstSample % bufferSize;
                for (size_t i = 0; i < numEmptyBuffers; i++)
                {
                    audioBufferQueue.push(buffer.data(), bufferSize);
                }
                numCollected = additionalSamples;
                index = firstSample;
            }
        }
 
        // if already collecting
        if (state == State::collecting)
        {
            while (index++ < numSamples)
            {
                buffer[numCollected++] = *data++;
 
                if (numCollected == buffer.size())
                {
                    audioBufferQueue.push(buffer.data(), buffer.size());
                    numCollected = 0;
                }
            }
        }
    }
 
private:
    //==============================================================================
    AudioBufferQueue<SampleType> &audioBufferQueue; 
    std::vector<SampleType> buffer;                 
    int bufferSize;                                 
    size_t numCollected;                            
    SampleType prevSample = SampleType(100);        
    static constexpr auto triggerLevel = SampleType(0); 
    enum class State
    {
        waitingForTrigger,
        collecting
    } state{State::waitingForTrigger};
};