SignalSaver.h

Go to the documentation of this file.

namespace krotos
{
    //==============================================================================
    class SignalSaver : public AsyncUpdater
    {
      public:
        SignalSaver(const String& path, int numChannels, int64 numSamples, double sampleRate)
            : m_numChannels(numChannels), m_length(numSamples), m_counter(0), m_file(path)
        {
            // The file path must end in .wav
            jassert(path.contains(".wav"));
 
            WavAudioFormat format;
            m_writer.reset(format.createWriterFor(new FileOutputStream(m_file), sampleRate,
                                                  static_cast<unsigned int>(numChannels), 16, {}, 0));
        }
 
        void process(AudioBuffer<float> buffer)
        {
            jassert(buffer.getNumChannels() >= m_numChannels);
            process(buffer.getArrayOfReadPointers(), buffer.getNumSamples());
        }
 
        void process(const float* const* buffer, int blockSize)
        {
            if (m_counter == -1)
                return;
 
            int numSamplesToWrite = (m_counter + int64(blockSize) < m_length) ? blockSize : int(m_length - m_counter);
            m_writer->writeFromFloatArrays(buffer, m_numChannels, numSamplesToWrite);
            m_counter += numSamplesToWrite;
 
            if (m_counter >= m_length)
            {
                triggerAsyncUpdate();
                m_counter = -1;
            }
        }
 
        void handleAsyncUpdate() override { m_writer.reset(nullptr); }
 
      private:
        const int m_numChannels;
        const int64 m_length;
        int64 m_counter;
 
        const File m_file;
        std::unique_ptr<AudioFormatWriter> m_writer;
 
        std::vector<float> m_data;
 
        JUCE_DECLARE_NON_COPYABLE_WITH_LEAK_DETECTOR(SignalSaver)
    };
} // namespace krotos