#include "PicData.hpp"
|
|
#include <stdexcept>
|
|
extern "C" {
|
#include <libavcodec/avcodec.h>
|
}
|
|
#include "../log/log.hpp"
|
using namespace logif;
|
|
namespace ffwrapper{
|
PicData::PicData(const int width, const int height, const int pix_fmt)
|
:frame_(nullptr)
|
,data_(nullptr)
|
,data_size_(0)
|
,pix_fmt_(pix_fmt)
|
,width_(width)
|
,height_(height)
|
{
|
bool flag = init_frame();
|
flag = init_AVPicture() && flag;
|
if(!flag){
|
throw std::runtime_error("init PicData error");
|
}
|
}
|
|
|
PicData::~PicData(){
|
free_frame();
|
}
|
|
bool PicData::init_frame(){
|
frame_ = av_frame_alloc();
|
|
return !!frame_;
|
}
|
|
void PicData::free_frame(){
|
avpicture_free((AVPicture *)frame_);
|
frame_ = NULL;
|
data_ = NULL;
|
|
}
|
|
bool PicData::init_AVPicture(){
|
data_size_ = avpicture_get_size((AVPixelFormat)pix_fmt_,
|
width_, height_);
|
if(data_size_ <= 0){
|
logIt("avpicture_get_size error");
|
return false;
|
}
|
data_ = (uint8_t*)av_malloc(data_size_);
|
const int ret = avpicture_fill((AVPicture *)frame_, data_,
|
(AVPixelFormat)pix_fmt_, width_, height_);
|
|
if(ret < 0){
|
logIt("avpicture_fill error");
|
return false;
|
|
}
|
|
return true;
|
}
|
}
|
