59 lines
1.1 KiB
C++
59 lines
1.1 KiB
C++
#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;
|
|
|
|
Size2u frame_size;
|
|
AVRational frame_rate;
|
|
uint bit_rate;
|
|
|
|
char filename[_MAX_PATH];
|
|
|
|
public:
|
|
|
|
VideoEncoder(const char *fn,const uint br)
|
|
{
|
|
strcpy_s(filename,_MAX_PATH,fn);
|
|
|
|
bit_rate=br;
|
|
}
|
|
|
|
AVFormatContext *GetFrmCtx(){return fmt_ctx;}
|
|
AVCodecContext *GetCodecCtx(){return codec_ctx;}
|
|
|
|
virtual ~VideoEncoder()=default;
|
|
|
|
virtual void Set(const AVRational &fr,const Size2u &size)
|
|
{
|
|
frame_size=size;
|
|
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);
|