MpegFrameCapture.hh

Go to the documentation of this file.

#ifdef FRAME_CAPTURE_MPEG
 
#ifndef _MPEG_FRAME_CAPTURE_HH_
#define _MPEG_FRAME_CAPTURE_HH_
 
#include "FrameCapture.hh"
#include "MpegFrameCaptureOptions.hh"
#include <string>
 
extern "C" {
#include <libavcodec/avcodec.h>
#include <libavutil/imgutils.h>
#include <libavutil/opt.h>
#include <libavutil/mem.h>
#include <libswscale/swscale.h>
}
 
namespace OpenGLMbs{
 
class MpegFrameCapture: public MbsFrameCapture
{
 
    public:
        MpegFrameCapture(MbsViewPointRenderer* vp_renderer, std::string filename = "test", int fps=25, OpenGLMbs::MpegFrameCaptureOptions const &opt=OpenGLMbs::MpegFrameCaptureOptions());
        ~MpegFrameCapture();
 
        void ProcessScreenShot();
        void ffmpeg_encoder_start(int codec_id, int fps);
        void ffmpeg_encoder_glread_rgb();
        void ffmpeg_encoder_set_frame_yuv_from_rgb();
        void ffmpeg_encoder_encode_frame();
        void ffmpeg_encoder_finish();
 
    private:
        std::string filename;
 
        AVCodecContext *c;
        AVFrame *frame;
        AVPacket pkt;
        FILE *file;
        struct SwsContext *sws_context;
        uint8_t *rgb;
        const size_t format_nchannels;
        int nvals;
 
        // parameters coming from options
        int bit_rate;
 
};
 
} //namespace OpenGLMbs
 
#endif //_MPEG_FRAME_CAPTURE_HH_
 
#endif