video_ostream.hpp

#pragma once


#include <string>
#include "core/image.hpp"

namespace cv{
    class VideoWriter;
    class Mat;
}

namespace rt{

class video_ostream{
public:
    video_ostream(int width, int height);
    ~video_ostream();

    int width() const;
    int height() const;

    bool begin(const std::string & filename, int framerate);
    void write(const rt::image & img);
    void end();

private:
    cv::VideoWriter *_writer;
    cv::Mat *_data;
    int _width;
    int _height;
};

}