Almond/VideoStream.h

#pragma once
#define FFMPEG_ENABLED
#ifdef FFMPEG_ENABLED


#ifndef VIDEO_STREAM_H_
#define VIDEO_STREAM_H_

#include <string>
#include "Bitmap.h"

extern "C" {
#   include <libavformat/avformat.h>
#   include <libavformat/avio.h>
#   include <libavcodec/avcodec.h>
#   include <libavformat/avformat.h>
#   include <libavutil/imgutils.h>
#   include <libavutil/opt.h>
#   include <libswscale/swscale.h>
}

class VideoStream
{
    const AVCodec* codec;
    AVCodecContext* codecContext;
    AVFormatContext* formatContext;
    AVCodecParameters* params;
    //FILE* file;
    AVFrame* picture;
    AVPacket* pkt;
    AVStream* stream;
    SwsContext* swsContext;
    static const uint8_t endcode[];

    int width;
    int height;

    int64_t frameIndex = 0;
public:
    VideoStream(int width, int height, const std::string& filename, int bitrate, int fps, const char* preset);
    ~VideoStream(void);

    void encode(AVFrame* frame);

    void addFrame(const Bitmap<RGBColor>& frame);
};

#endif // VIDEO_STREAM_H_

#endif // FFMPEG_ENABLED