Skip to content

F
ffmpeg.wasm-core
Commit d5232d47 authored by Yayoi's avatar Yayoi Committed by Michael Niedermayer
Browse files
Options

avfilter/colormatrix:add slice threading 

Signed-off-by: 's avatarMichael Niedermayer <michaelni@gmx.at>
parent 26d81b57
Hide whitespace changes
Inline Side-by-side
Showing with 94 additions and 52 deletions
libavfilter/vf_colormatrix.c
View file @ d5232d47
...@@ -73,6 +73,17 @@ typedef struct { ...@@ -73,6 +73,17 @@ typedef struct {
int hsub, vsub; int hsub, vsub;
} ColorMatrixContext; } ColorMatrixContext;
typedef struct ThreadData {
AVFrame *dst;
const AVFrame *src;
int c2;
int c3;
int c4;
int c5;
int c6;
int c7;
} ThreadData;
#define OFFSET(x) offsetof(ColorMatrixContext, x) #define OFFSET(x) offsetof(ColorMatrixContext, x)
#define FLAGS AV_OPT_FLAG_VIDEO_PARAM|AV_OPT_FLAG_FILTERING_PARAM #define FLAGS AV_OPT_FLAG_VIDEO_PARAM|AV_OPT_FLAG_FILTERING_PARAM
...@@ -179,24 +190,28 @@ static av_cold int init(AVFilterContext *ctx) ...@@ -179,24 +190,28 @@ static av_cold int init(AVFilterContext *ctx)
return 0; return 0;
} }
static void process_frame_uyvy422(ColorMatrixContext *color, static int process_slice_uyvy422(AVFilterContext *ctx, void *arg, int jobnr, int nb_jobs)
AVFrame *dst, AVFrame *src)
{ {
const unsigned char *srcp = src->data[0]; const ThreadData *td = arg;
const int src_pitch = src->linesize[0]; const AVFrame *src = td->src;
AVFrame *dst = td->dst;
const int height = src->height; const int height = src->height;
const int width = src->width*2; const int width = src->width*2;
unsigned char *dstp = dst->data[0]; const int src_pitch = src->linesize[0];
const int dst_pitch = dst->linesize[0]; const int dst_pitch = dst->linesize[0];
const int c2 = color->yuv_convert[color->mode][0][1]; const int slice_start = (height * jobnr ) / nb_jobs;
const int c3 = color->yuv_convert[color->mode][0][2]; const int slice_end = (height * (jobnr+1)) / nb_jobs;
const int c4 = color->yuv_convert[color->mode][1][1]; const unsigned char *srcp = src->data[0] + slice_start * src_pitch;
const int c5 = color->yuv_convert[color->mode][1][2]; unsigned char *dstp = dst->data[0] + slice_start * dst_pitch;
const int c6 = color->yuv_convert[color->mode][2][1]; const int c2 = td->c2;
const int c7 = color->yuv_convert[color->mode][2][2]; 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; int x, y;
for (y = 0; y < height; y++) { for (y = slice_start; y < slice_end; y++) {
for (x = 0; x < width; x += 4) { for (x = 0; x < width; x += 4) {
const int u = srcp[x + 0] - 128; const int u = srcp[x + 0] - 128;
const int v = srcp[x + 2] - 128; const int v = srcp[x + 2] - 128;
...@@ -209,32 +224,38 @@ static void process_frame_uyvy422(ColorMatrixContext *color, ...@@ -209,32 +224,38 @@ static void process_frame_uyvy422(ColorMatrixContext *color,
srcp += src_pitch; srcp += src_pitch;
dstp += dst_pitch; dstp += dst_pitch;
} }
return 0;
} }
static void process_frame_yuv422p(ColorMatrixContext *color, static int process_slice_yuv422p(AVFilterContext *ctx, void *arg, int jobnr, int nb_jobs)
AVFrame *dst, AVFrame *src)
{ {
const unsigned char *srcpU = src->data[1]; const ThreadData *td = arg;
const unsigned char *srcpV = src->data[2]; const AVFrame *src = td->src;
const unsigned char *srcpY = src->data[0]; AVFrame *dst = td->dst;
const int src_pitchY = src->linesize[0];
const int src_pitchUV = src->linesize[1];
const int height = src->height; const int height = src->height;
const int width = src->width; const int width = src->width;
unsigned char *dstpU = dst->data[1]; const int slice_start = (height * jobnr ) / nb_jobs;
unsigned char *dstpV = dst->data[2]; const int slice_end = (height * (jobnr+1)) / nb_jobs;
unsigned char *dstpY = dst->data[0]; 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_pitchY = dst->linesize[0];
const int dst_pitchUV = dst->linesize[1]; const int dst_pitchUV = dst->linesize[1];
const int c2 = color->yuv_convert[color->mode][0][1]; unsigned char *dstpU = dst->data[1] + slice_start * dst_pitchUV;
const int c3 = color->yuv_convert[color->mode][0][2]; unsigned char *dstpV = dst->data[2] + slice_start * dst_pitchUV;
const int c4 = color->yuv_convert[color->mode][1][1]; unsigned char *dstpY = dst->data[0] + slice_start * dst_pitchY;
const int c5 = color->yuv_convert[color->mode][1][2]; const int c2 = td->c2;
const int c6 = color->yuv_convert[color->mode][2][1]; const int c3 = td->c3;
const int c7 = color->yuv_convert[color->mode][2][2]; const int c4 = td->c4;
const int c5 = td->c5;
const int c6 = td->c6;
const int c7 = td->c7;
int x, y; int x, y;
for (y = 0; y < height; y++) { for (y = slice_start; y < slice_end; y++) {
for (x = 0; x < width; x += 2) { for (x = 0; x < width; x += 2) {
const int u = srcpU[x >> 1] - 128; const int u = srcpU[x >> 1] - 128;
const int v = srcpV[x >> 1] - 128; const int v = srcpV[x >> 1] - 128;
...@@ -251,34 +272,40 @@ static void process_frame_yuv422p(ColorMatrixContext *color, ...@@ -251,34 +272,40 @@ static void process_frame_yuv422p(ColorMatrixContext *color,
dstpU += dst_pitchUV; dstpU += dst_pitchUV;
dstpV += dst_pitchUV; dstpV += dst_pitchUV;
} }
return 0;
} }
static void process_frame_yuv420p(ColorMatrixContext *color, static int process_slice_yuv420p(AVFilterContext *ctx, void *arg, int jobnr, int nb_jobs)
AVFrame *dst, AVFrame *src)
{ {
const unsigned char *srcpU = src->data[1]; const ThreadData *td = arg;
const unsigned char *srcpV = src->data[2]; const AVFrame *src = td->src;
const unsigned char *srcpY = src->data[0]; AVFrame *dst = td->dst;
const unsigned char *srcpN = src->data[0] + src->linesize[0];
const int src_pitchY = src->linesize[0];
const int src_pitchUV = src->linesize[1];
const int height = src->height; const int height = src->height;
const int width = src->width; const int width = src->width;
unsigned char *dstpU = dst->data[1]; const int slice_start = (height * jobnr ) / nb_jobs;
unsigned char *dstpV = dst->data[2]; const int slice_end = (height * (jobnr+1)) / nb_jobs;
unsigned char *dstpY = dst->data[0]; const int src_pitchY = src->linesize[0];
unsigned char *dstpN = dst->data[0] + dst->linesize[0]; const int src_pitchUV = src->linesize[1];
const int dst_pitchY = dst->linesize[0]; const int dst_pitchY = dst->linesize[0];
const int dst_pitchUV = dst->linesize[1]; const int dst_pitchUV = dst->linesize[1];
const int c2 = color->yuv_convert[color->mode][0][1]; const unsigned char *srcpY = src->data[0] + src_pitchY * slice_start;
const int c3 = color->yuv_convert[color->mode][0][2]; const unsigned char *srcpU = src->data[1] + src_pitchUV * (slice_start >> 1);
const int c4 = color->yuv_convert[color->mode][1][1]; const unsigned char *srcpV = src->data[2] + src_pitchUV * (slice_start >> 1);
const int c5 = color->yuv_convert[color->mode][1][2]; const unsigned char *srcpN = src->data[0] + src_pitchY * (slice_start + 1);
const int c6 = color->yuv_convert[color->mode][2][1]; unsigned char *dstpU = dst->data[1] + dst_pitchUV * (slice_start >> 1);
const int c7 = color->yuv_convert[color->mode][2][2]; unsigned char *dstpV = dst->data[2] + dst_pitchUV * (slice_start >> 1);
unsigned char *dstpY = dst->data[0] + dst_pitchY * slice_start;
unsigned char *dstpN = dst->data[0] + dst_pitchY * (slice_start + 1);
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; int x, y;
for (y = 0; y < height; y += 2) { for (y = slice_start; y < slice_end; y += 2) {
for (x = 0; x < width; x += 2) { for (x = 0; x < width; x += 2) {
const int u = srcpU[x >> 1] - 128; const int u = srcpU[x >> 1] - 128;
const int v = srcpV[x >> 1] - 128; const int v = srcpV[x >> 1] - 128;
...@@ -299,6 +326,8 @@ static void process_frame_yuv420p(ColorMatrixContext *color, ...@@ -299,6 +326,8 @@ static void process_frame_yuv420p(ColorMatrixContext *color,
dstpU += dst_pitchUV; dstpU += dst_pitchUV;
dstpV += dst_pitchUV; dstpV += dst_pitchUV;
} }
return 0;
} }
static int config_input(AVFilterLink *inlink) static int config_input(AVFilterLink *inlink)
...@@ -336,6 +365,7 @@ static int filter_frame(AVFilterLink *link, AVFrame *in) ...@@ -336,6 +365,7 @@ static int filter_frame(AVFilterLink *link, AVFrame *in)
ColorMatrixContext *color = ctx->priv; ColorMatrixContext *color = ctx->priv;
AVFilterLink *outlink = ctx->outputs[0]; AVFilterLink *outlink = ctx->outputs[0];
AVFrame *out; AVFrame *out;
ThreadData td = {0};
out = ff_get_video_buffer(outlink, outlink->w, outlink->h); out = ff_get_video_buffer(outlink, outlink->w, outlink->h);
if (!out) { if (!out) {
...@@ -372,12 +402,24 @@ static int filter_frame(AVFilterLink *link, AVFrame *in) ...@@ -372,12 +402,24 @@ static int filter_frame(AVFilterLink *link, AVFrame *in)
calc_coefficients(ctx); calc_coefficients(ctx);
td.src = in;
td.dst = out;
td.c2 = color->yuv_convert[color->mode][0][1];
td.c3 = color->yuv_convert[color->mode][0][2];
td.c4 = color->yuv_convert[color->mode][1][1];
td.c5 = color->yuv_convert[color->mode][1][2];
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_YUV422P)
process_frame_yuv422p(color, out, in); ctx->internal->execute(ctx, process_slice_yuv422p, &td, NULL,
FFMIN(in->height, ctx->graph->nb_threads));
else if (in->format == AV_PIX_FMT_YUV420P) else if (in->format == AV_PIX_FMT_YUV420P)
process_frame_yuv420p(color, out, in); ctx->internal->execute(ctx, process_slice_yuv420p, &td, NULL,
FFMIN(in->height, ctx->graph->nb_threads));
else else
process_frame_uyvy422(color, out, in); ctx->internal->execute(ctx, process_slice_uyvy422, &td, NULL,
FFMIN(in->height, ctx->graph->nb_threads));
av_frame_free(&in); av_frame_free(&in);
return ff_filter_frame(outlink, out); return ff_filter_frame(outlink, out);
...@@ -410,5 +452,5 @@ AVFilter ff_vf_colormatrix = { ...@@ -410,5 +452,5 @@ AVFilter ff_vf_colormatrix = {
.inputs = colormatrix_inputs, .inputs = colormatrix_inputs,
.outputs = colormatrix_outputs, .outputs = colormatrix_outputs,
.priv_class = &colormatrix_class, .priv_class = &colormatrix_class,
.flags = AVFILTER_FLAG_SUPPORT_TIMELINE_GENERIC, .flags = AVFILTER_FLAG_SUPPORT_TIMELINE_GENERIC | AVFILTER_FLAG_SLICE_THREADS,
}; };
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment