AudioFile.h

#pragma once
#include <string>
#include <vector>

 
namespace AudioFileUtilities {
        int getSamples(const std::string& file, float *buf, unsigned int channel, unsigned int startFrame, unsigned int endFrame);
        int getNumChannels(const std::string& file);
        int getNumFrames(const std::string& file);
        int write(const std::string& filename, float *buf, unsigned int channels, unsigned int frames, unsigned int samplerate);
        int write(const std::string& filename, const std::vector<std::vector<float> >& dataIn, unsigned int sampleRate);
        std::vector<std::vector<float> > load(const std::string& filename, int maxCount = -1, unsigned int start = 0);
        std::vector<float> loadMono(const std::string& file);
};
 
#include <libraries/sndfile/sndfile.h>
#include <thread>
#include <array>
 
class AudioFile
{
protected:
        typedef enum {
                kRead,
                kWrite,
        } Mode;
        enum { kNumBufs = 2};
protected:
        int setup(const std::string& path, size_t bufferSize, Mode mode, size_t channels = 0, unsigned int sampleRate = 0);
public:
        size_t getLength() const { return sfinfo.frames; };
        size_t getChannels() const { return sfinfo.channels; };
        int getSampleRate() const { return sfinfo.samplerate; };
        virtual ~AudioFile();
private:
        void cleanup();
protected:
        volatile size_t ioBuffer;
        size_t ioBufferOld;
protected:
        void scheduleIo();
        std::vector<float>& getRtBuffer();
        std::array<std::vector<float>,kNumBufs> internalBuffers;
        void threadLoop();
        virtual void io(std::vector<float>& buffer) = 0;
        std::thread diskIo;
        size_t size;
        volatile bool stop;
        bool ramOnly;
        size_t rtIdx;
        SNDFILE* sndfile = NULL;
        SF_INFO sfinfo = { 0 };
};
 
class AudioFileReader : public AudioFile
{
public:
        int setup(const std::string& path, size_t bufferSize);void getSamples(std::vector<float>& buffer);
        void getSamples(float* dst, size_t samplesCount);
        int setLoop(bool loop);
        int setLoop(size_t start, size_t end);
        size_t getIdx();
private:
        void io(std::vector<float>& buffer) override;
        bool loop;
        size_t loopStart;
        size_t loopStop;
        size_t idx;
};
 
class AudioFileWriter : public AudioFile
{
public:
        int setup(const std::string& path, size_t bufferSize, size_t channels, unsigned int sampleRate);
        void setSamples(std::vector<float>& buffer);
        void setSamples(float const * src, size_t samplesCount);
private:
        void io(std::vector<float>& buffer) override;
};