generic-library/ffmpeg
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

avfilter/vf_colormatrix: add yuv444p support

Browse Source 
Signed-off-by: Paul B Mahol <onemda@gmail.com>
This commit is contained in:
Paul B Mahol 2015-07-06 14:07:52 +00:00
parent 6a1204a1a4
commit 7efe81a813
1 changed files with 52 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

53
libavfilter/vf_colormatrix.c
View File

@ -230,6 +230,53 @@ static int process_slice_uyvy422(AVFilterContext *ctx, void *arg, int jobnr, int
return 0;
}
static int process_slice_yuv444p(AVFilterContext *ctx, void *arg, int jobnr, int nb_jobs)
{
const ThreadData *td = arg;
const AVFrame *src = td->src;
AVFrame *dst = td->dst;
const int height = src->height;
const int width = src->width;
const int slice_start = (height * jobnr ) / nb_jobs;
const int slice_end = (height * (jobnr+1)) / nb_jobs;
const int src_pitchY = src->linesize[0];
const int src_pitchUV = src->linesize[1];
const unsigned char *srcpU = src->data[1] + slice_start * src_pitchUV;
const unsigned char *srcpV = src->data[2] + slice_start * src_pitchUV;
const unsigned char *srcpY = src->data[0] + slice_start * src_pitchY;
const int dst_pitchY = dst->linesize[0];
const int dst_pitchUV = dst->linesize[1];
unsigned char *dstpU = dst->data[1] + slice_start * dst_pitchUV;
unsigned char *dstpV = dst->data[2] + slice_start * dst_pitchUV;
unsigned char *dstpY = dst->data[0] + slice_start * dst_pitchY;
const int c2 = td->c2;
const int c3 = td->c3;
const int c4 = td->c4;
const int c5 = td->c5;
const int c6 = td->c6;
const int c7 = td->c7;
int x, y;
for (y = slice_start; y < slice_end; y++) {
for (x = 0; x < width; x++) {
const int u = srcpU[x] - 128;
const int v = srcpV[x] - 128;
const int uvval = c2 * u + c3 * v + 1081344;
dstpY[x] = CB((65536 * (srcpY[x] - 16) + uvval) >> 16);
dstpU[x] = CB((c4 * u + c5 * v + 8421376) >> 16);
dstpV[x] = CB((c6 * u + c7 * v + 8421376) >> 16);
}
srcpY += src_pitchY;
dstpY += dst_pitchY;
srcpU += src_pitchUV;
srcpV += src_pitchUV;
dstpU += dst_pitchUV;
dstpV += dst_pitchUV;
}
return 0;
}
static int process_slice_yuv422p(AVFilterContext *ctx, void *arg, int jobnr, int nb_jobs)
{
const ThreadData *td = arg;
@ -350,6 +397,7 @@ static int config_input(AVFilterLink *inlink)
static int query_formats(AVFilterContext *ctx)
{
static const enum AVPixelFormat pix_fmts[] = {
AV_PIX_FMT_YUV444P,
AV_PIX_FMT_YUV422P,
AV_PIX_FMT_YUV420P,
AV_PIX_FMT_UYVY422,
@ -411,7 +459,10 @@ static int filter_frame(AVFilterLink *link, AVFrame *in)
td.c6 = color->yuv_convert[color->mode][2][1];
td.c7 = color->yuv_convert[color->mode][2][2];
if (in->format == AV_PIX_FMT_YUV422P)
if (in->format == AV_PIX_FMT_YUV444P)
ctx->internal->execute(ctx, process_slice_yuv444p, &td, NULL,
FFMIN(in->height, ctx->graph->nb_threads));
else if (in->format == AV_PIX_FMT_YUV422P)
ctx->internal->execute(ctx, process_slice_yuv422p, &td, NULL,
FFMIN(in->height, ctx->graph->nb_threads));
else if (in->format == AV_PIX_FMT_YUV420P)