#include "PL_AVFrameYUV420.h"
|
|
extern "C"
|
{
|
#include <libavcodec/avcodec.h>
|
#include <libavutil/frame.h>
|
#include <libavformat/avformat.h>
|
}
|
|
struct AVFrameYUV420_Internal
|
{
|
uint8_t buffer[1920*1080*3];
|
size_t buffSize;
|
size_t buffSizeMax;
|
|
AVFrameYUV420_Internal() :
|
buffSize(0), buffSizeMax(sizeof(buffer))
|
{
|
}
|
|
~AVFrameYUV420_Internal()
|
{
|
}
|
|
void reset()
|
{
|
buffSize = 0;
|
}
|
};
|
|
PipeLineElem* create_PL_AVFrameYUV420()
|
{
|
return new PL_AVFrameYUV420;
|
}
|
|
PL_AVFrameYUV420::PL_AVFrameYUV420() : internal(new AVFrameYUV420_Internal)
|
{
|
}
|
|
PL_AVFrameYUV420::~PL_AVFrameYUV420()
|
{
|
delete (AVFrameYUV420_Internal*)internal;
|
internal= nullptr;
|
}
|
|
bool PL_AVFrameYUV420::init(void* args)
|
{
|
AVFrameYUV420_Internal* in = (AVFrameYUV420_Internal*)internal;
|
in->reset();
|
|
return true;
|
}
|
|
void PL_AVFrameYUV420::finit()
|
{
|
AVFrameYUV420_Internal* in = (AVFrameYUV420_Internal*)internal;
|
|
}
|
|
bool PL_AVFrameYUV420::pay(const PipeMaterial& pm)
|
{
|
AVFrameYUV420_Internal* in = (AVFrameYUV420_Internal*)internal;
|
|
AVFrame* pAVFrame = (AVFrame*)pm.buffer;
|
if (pAVFrame == nullptr)
|
return false;
|
|
int picSize = pAVFrame->height * pAVFrame->width;
|
in->buffSize = picSize * 1.5;
|
|
int height = pAVFrame->height;
|
int width = pAVFrame->width;
|
|
uint8_t* pBuff = in->buffer;
|
|
memcpy(pBuff, pAVFrame->data[0], height * width);
|
pBuff += height * width;
|
|
memcpy(pBuff, pAVFrame->data[1], height * width / 4);
|
pBuff += height * width / 4;
|
|
memcpy(pBuff, pAVFrame->data[2], height * width / 4);
|
pBuff += height * width / 4;
|
|
in->buffSize = pBuff - in->buffer;
|
|
// write yuv420
|
//int a=0;
|
//for (int i = 0; i < height; i++)
|
//{
|
// memcpy(in->buffer + a, pAVFrame->data[0] + i * pAVFrame->linesize[0], width);
|
// a += width;
|
//}
|
//for (int i=0; i<height/2; i++)
|
//{
|
// memcpy(in->buffer + a, pAVFrame->data[1] + i * pAVFrame->linesize[1], width / 2);
|
// a += width / 2;//#todo 4
|
//}
|
//for (int i=0; i<height/2; i++)
|
//{
|
// memcpy(in->buffer + a, pAVFrame->data[2] + i * pAVFrame->linesize[2], width / 2);
|
// a += width / 2;
|
//}
|
|
//in->buffer readly
|
|
//static size_t f=0;
|
//char fname[50];
|
//sprintf(fname, "%u.yuv420", ++f);
|
//FILE * pFile = fopen (fname,"wb");
|
//fwrite (in->buffer , sizeof(char), in->buffSize, pFile);
|
//fclose(pFile);
|
|
return true;
|
}
|
|
bool PL_AVFrameYUV420::gain(PipeMaterial& pm)
|
{
|
AVFrameYUV420_Internal* in = (AVFrameYUV420_Internal*)internal;
|
|
pm.buffer = in->buffer;
|
pm.buffSize = in->buffSize;
|
pm.former = this;
|
return true;
|
}
|
