#pragma once
#include"DataType.h"
extern "C"
{
#include <libavformat/avformat.h>
    #include<libavcodec/avcodec.h>
}

class EncodeOutput
{
public:

    virtual ~EncodeOutput()=default;

    virtual bool Write(const void *,const uint)=0;
};//class EncodeOutput

class VideoEncoder
{
protected:

    AVFormatContext* fmt_ctx;
    AVCodecContext* codec_ctx;

    uint width;
    uint height;
    AVRational frame_rate;
    uint bit_rate;

    char filename[_MAX_PATH];

public:

    VideoEncoder(const char *fn,const uint br)
    {
        strcpy_s(filename,_MAX_PATH,fn);

        width=height=0;
        bit_rate=br;
    }

    AVFormatContext* GetFrmCtx()
    {
        return fmt_ctx;
    }

    AVCodecContext* GetCodecCtx() {
        return codec_ctx
            ;
    };

    virtual ~VideoEncoder()=default;

    virtual void Set(const uint w,const uint h,const AVRational &fr)
    {
        width=w;
        height=h;
        frame_rate=fr;
    }

    virtual bool Init()=0;

    virtual bool WriteFrame(const uint8 *rgba_data)=0;

    virtual bool Finish()=0;
};//class VideoEncoder

VideoEncoder *CreateVideoEncoder(const char *filename,const uint bit_rate,const bool);