jompegRecord.h

Go to the documentation of this file.

#ifndef JOMPEGRECORD_H
#define JOMPEGRECORD_H
 
#include <stdio.h>
#include <math.h>
#include <iostream>
 
#include <jo_mpeg.h>
 
#include <record.h>
 
class JompegRecord : public Record
{
public:
 
    JompegRecord
    (
        std::filesystem::path file,
        glm::ivec2 resolution,
        uint8_t fps
    )
    :
      Record(file, resolution, fps)
    {
        bool fpsOk = false;
        for (const auto & option : {24, 25, 30, 50, 60})
        {
            if (fps == option)
            {
                fpsOk = true;
                break;
            }
        }
 
        if (!fpsOk)
        {
            uint8_t closest = 0;
            for (const auto & option : {24, 25, 30, 50, 60})
            {
                if (std::abs(int(option)-int(fps)) < std::abs(int(closest)-int(fps)))
                {
                    closest = option;
                }
            }
            std::cout << "jo_mpeg recording only supports 24, 25, 30, 50, and 60 fps.\nSelected "+std::to_string(closest)+" fps.\n";
        }
    }
 
    ~JompegRecord(){};
 
    void open()
    {
        if (fileOpen) { return; }
        fp = fopen(file.string().c_str(), "wb");
        fileOpen = true;
    }
 
    void close()
    {
        if (!fileOpen) { return; }
        fileOpen = false;
        fclose(fp);
    }
 
private:
 
    FILE * fp = nullptr;
 
    void write(const uint8_t * frame)
    {
        jo_write_mpeg(fp, frame, resolution.x, resolution.y, fps);
    }
};
 
#endif /* JOMPEGRECORD_H */