record.h

Go to the documentation of this file.

#ifndef RECORD_H
#define RECORD_H
 
#include <filesystem>
#include <cstdint>
#include <vector>
 
#include <glm/glm.hpp>
 
class Record
{
public:
 
    Record(std::filesystem::path file, glm::ivec2 resolution, uint8_t fps)
    :   file(file),
        resolution(resolution),
        fps(fps)
        {}
 
    virtual ~Record(){}
 
    virtual void open() = 0;
 
    void queueFrame(std::vector<uint8_t> pixels)
    {
        if (fileOpen) { inBuffer.push_back(pixels); }
    }
 
    uint8_t queueSize() { return inBuffer.size(); }
 
    uint64_t framesLeft() { return inBuffer.size() + outBuffer.size(); }
 
    void writeFrames()
    {
        if (writing || inBuffer.size() == 0) { return; }
        writing = true;
        outBuffer = inBuffer;
        inBuffer.clear();
        std::thread io = std::thread
        (
            &Record::writeBuffer,
            this
        );
        io.detach();
    }
 
    virtual void close() = 0;
 
    bool isWriting() const { return writing; }
 
    bool isOpen() const { return fileOpen; }
 
    bool finalise()
    {
        if (!isWriting())
        {
            if (queueSize() > 0)
            {
                writeFrames();
            }
            else
            {
                close();
                return true;
            }
        }
        return false;
    }
 
protected:
 
    std::filesystem::path file;
    glm::ivec2 resolution;
    uint8_t fps;
    bool fileOpen = false;
    bool writing = false;
 
    std::vector<std::vector<uint8_t>> inBuffer, outBuffer;
 
    virtual void write(const uint8_t * frame) = 0;
 
    void writeBuffer()
    {
        for (const auto & frame : outBuffer)
        {
            write(frame.data());
        }
        outBuffer.clear();
        writing = false;
    }
};
 
#endif /* RECORD_H */