69 lines
1.2 KiB
C
69 lines
1.2 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;
|
|||
|
|
|
|||
|
|
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);
|