audio/include/utilite/UAudioRecorder.h

Go to the documentation of this file.

#ifndef UAUDIORECORDER_H
#define UAUDIORECORDER_H

#include "utilite/UtiLiteExp.h" // DLL export/import defines

#include <utilite/USemaphore.h>
#include <utilite/UMutex.h>
#include <utilite/UThreadNode.h>

#include <vector>

class UTILITE_EXP UAudioRecorder : public UThreadNode {
public:
    virtual ~UAudioRecorder();

    virtual bool init()
    {
        _samplesMutex.lock();
                _samples.clear();
                _samplesMutex.unlock();
                return true;
    }
    virtual void close() {}

    void stop() {join(true);} // for convenience

    bool getNextFrame(std::vector<char> & frame,
                      bool removeOldFrames = false);

    bool getNextFrame(std::vector<char> & frame,
                      int &frameId);

    bool getMultiFrame(std::vector<char> & frame,
                      int frameIdBeg,
                      int frameIdEnd);
    
    bool getFrame(std::vector<char> & frame, int frameId);

    void removeFrames(int frameIdBeg,
                      int frameIdEnd);

    int getNumFrames();
    int getNextFrameToGet();

    int samples();

    int fs() const {return _fs;}
    int frameLength() const {return _frameLength;}
    int bytesPerSample() const {return _bytesPerSample;}
    int channels() const {return _channels;}

protected:
    UAudioRecorder(int fs = 44100,
                 int frameLength = 1024,
                 int bytesPerSample = 2,
                 int channels = 1);

    virtual void mainLoopEnd();
    void pushBackSamples(void * data, int dataLengthInBytes);

    void setFs(int fs) {_fs = fs;}
    void setFrameLength(int frameLength) {_frameLength = frameLength;}
        void setBytesPerSample(int bytesPerSample) {_bytesPerSample = bytesPerSample;}
        void setChannels(int channels) {_channels = channels;}

private:
    UMutex _samplesMutex;
    std::vector<char> _samples;

    int _fs;
    int _frameLength;
    int _bytesPerSample;
    int _channels;

    int _nextFrameToGet;
    USemaphore _getTrameSemaphore;

};

#endif