diff --git a/libavfilter/colorchannelmixer_template.c b/libavfilter/colorchannelmixer_template.c new file mode 100644 index 0000000000..0f15f8fd07 --- /dev/null +++ b/libavfilter/colorchannelmixer_template.c @@ -0,0 +1,193 @@ +/* + * Copyright (c) 2013 Paul B Mahol + * + * This file is part of FFmpeg. + * + * FFmpeg is free software; you can redistribute it and/or + * modify it under the terms of the GNU Lesser General Public + * License as published by the Free Software Foundation; either + * version 2.1 of the License, or (at your option) any later version. + * + * FFmpeg is distributed in the hope that it will be useful, + * but WITHOUT ANY WARRANTY; without even the implied warranty of + * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU + * Lesser General Public License for more details. + * + * You should have received a copy of the GNU Lesser General Public + * License along with FFmpeg; if not, write to the Free Software + * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA + */ + +#include + +#undef pixel +#if DEPTH == 8 +#define pixel uint8_t +#elif DEPTH == 16 +#define pixel uint16_t +#endif + +#undef fn +#undef fn2 +#undef fn3 +#define fn3(a,b) a##_##b +#define fn2(a,b) fn3(a,b) +#define fn(a) fn2(a, DEPTH) + +static av_always_inline int fn(filter_slice_rgba_planar)(AVFilterContext *ctx, void *arg, int jobnr, int nb_jobs, + int have_alpha, int depth, int pc) +{ + ColorChannelMixerContext *s = ctx->priv; + ThreadData *td = arg; + AVFrame *in = td->in; + AVFrame *out = td->out; + const float pa = s->preserve_amount; + const float max = (1 << depth) - 1; + const int slice_start = (out->height * jobnr) / nb_jobs; + const int slice_end = (out->height * (jobnr+1)) / nb_jobs; + const pixel *srcg = (const pixel *)(in->data[0] + slice_start * in->linesize[0]); + const pixel *srcb = (const pixel *)(in->data[1] + slice_start * in->linesize[1]); + const pixel *srcr = (const pixel *)(in->data[2] + slice_start * in->linesize[2]); + const pixel *srca = (const pixel *)(in->data[3] + slice_start * in->linesize[3]); + pixel *dstg = (pixel *)(out->data[0] + slice_start * out->linesize[0]); + pixel *dstb = (pixel *)(out->data[1] + slice_start * out->linesize[1]); + pixel *dstr = (pixel *)(out->data[2] + slice_start * out->linesize[2]); + pixel *dsta = (pixel *)(out->data[3] + slice_start * out->linesize[3]); + + for (int i = slice_start; i < slice_end; i++) { + for (int j = 0; j < out->width; j++) { + const pixel rin = srcr[j]; + const pixel gin = srcg[j]; + const pixel bin = srcb[j]; + const pixel ain = have_alpha ? srca[j] : 0; + int rout, gout, bout; + + rout = s->lut[R][R][rin] + + s->lut[R][G][gin] + + s->lut[R][B][bin] + + (have_alpha == 1 ? s->lut[R][A][ain] : 0); + gout = s->lut[G][R][rin] + + s->lut[G][G][gin] + + s->lut[G][B][bin] + + (have_alpha == 1 ? s->lut[G][A][ain] : 0); + bout = s->lut[B][R][rin] + + s->lut[B][G][gin] + + s->lut[B][B][bin] + + (have_alpha == 1 ? s->lut[B][A][ain] : 0); + + if (pc) { + float frout = av_clipf(rout, 0.f, max); + float fgout = av_clipf(gout, 0.f, max); + float fbout = av_clipf(bout, 0.f, max); + float lin, lout; + + preserve_color(s->preserve_color, rin, gin, bin, + rout, gout, bout, max, &lin, &lout); + preservel(&frout, &fgout, &fbout, lin, lout, max); + + rout = lrintf(lerpf(rout, frout, pa)); + gout = lrintf(lerpf(gout, fgout, pa)); + bout = lrintf(lerpf(bout, fbout, pa)); + } + + dstr[j] = av_clip_uintp2(rout, depth); + dstg[j] = av_clip_uintp2(gout, depth); + dstb[j] = av_clip_uintp2(bout, depth); + + if (have_alpha == 1) { + dsta[j] = av_clip_uintp2(s->lut[A][R][rin] + + s->lut[A][G][gin] + + s->lut[A][B][bin] + + s->lut[A][A][ain], depth); + } + } + + srcg += in->linesize[0] / sizeof(pixel); + srcb += in->linesize[1] / sizeof(pixel); + srcr += in->linesize[2] / sizeof(pixel); + srca += in->linesize[3] / sizeof(pixel); + dstg += out->linesize[0] / sizeof(pixel); + dstb += out->linesize[1] / sizeof(pixel); + dstr += out->linesize[2] / sizeof(pixel); + dsta += out->linesize[3] / sizeof(pixel); + } + + return 0; +} + +static av_always_inline int fn(filter_slice_rgba_packed)(AVFilterContext *ctx, void *arg, int jobnr, int nb_jobs, + int have_alpha, int step, int pc, int depth) +{ + ColorChannelMixerContext *s = ctx->priv; + ThreadData *td = arg; + AVFrame *in = td->in; + AVFrame *out = td->out; + const float pa = s->preserve_amount; + const float max = (1 << depth) - 1; + const int slice_start = (out->height * jobnr) / nb_jobs; + const int slice_end = (out->height * (jobnr+1)) / nb_jobs; + const uint8_t roffset = s->rgba_map[R]; + const uint8_t goffset = s->rgba_map[G]; + const uint8_t boffset = s->rgba_map[B]; + const uint8_t aoffset = s->rgba_map[A]; + const uint8_t *srcrow = in->data[0] + slice_start * in->linesize[0]; + uint8_t *dstrow = out->data[0] + slice_start * out->linesize[0]; + int i, j; + + for (i = slice_start; i < slice_end; i++) { + const pixel *src = (const pixel *)srcrow; + pixel *dst = (pixel *)dstrow; + + for (j = 0; j < out->width * step; j += step) { + const pixel rin = src[j + roffset]; + const pixel gin = src[j + goffset]; + const pixel bin = src[j + boffset]; + const pixel ain = src[j + aoffset]; + int rout, gout, bout; + + rout = s->lut[R][R][rin] + + s->lut[R][G][gin] + + s->lut[R][B][bin] + + (have_alpha == 1 ? s->lut[R][A][ain] : 0); + gout = s->lut[G][R][rin] + + s->lut[G][G][gin] + + s->lut[G][B][bin] + + (have_alpha == 1 ? s->lut[G][A][ain] : 0); + bout = s->lut[B][R][rin] + + s->lut[B][G][gin] + + s->lut[B][B][bin] + + (have_alpha == 1 ? s->lut[B][A][ain] : 0); + + if (pc) { + float frout = av_clipf(rout, 0.f, max); + float fgout = av_clipf(gout, 0.f, max); + float fbout = av_clipf(bout, 0.f, max); + float lin, lout; + + preserve_color(s->preserve_color, rin, gin, bin, + rout, gout, bout, max, &lin, &lout); + preservel(&frout, &fgout, &fbout, lin, lout, max); + + rout = lrintf(lerpf(rout, frout, pa)); + gout = lrintf(lerpf(gout, fgout, pa)); + bout = lrintf(lerpf(bout, fbout, pa)); + } + + dst[j + roffset] = av_clip_uintp2(rout, depth); + dst[j + goffset] = av_clip_uintp2(gout, depth); + dst[j + boffset] = av_clip_uintp2(bout, depth); + + if (have_alpha == 1) { + dst[j + aoffset] = av_clip_uintp2(s->lut[A][R][rin] + + s->lut[A][G][gin] + + s->lut[A][B][bin] + + s->lut[A][A][ain], depth); + } + } + + srcrow += in->linesize[0]; + dstrow += out->linesize[0]; + } + + return 0; +} diff --git a/libavfilter/vf_colorchannelmixer.c b/libavfilter/vf_colorchannelmixer.c index dfbc082b3f..b32e0a9d0e 100644 --- a/libavfilter/vf_colorchannelmixer.c +++ b/libavfilter/vf_colorchannelmixer.c @@ -56,6 +56,27 @@ typedef struct ColorChannelMixerContext { int (*filter_slice[2])(AVFilterContext *ctx, void *arg, int jobnr, int nb_jobs); } ColorChannelMixerContext; +static float lerpf(float v0, float v1, float f) +{ + return v0 + (v1 - v0) * f; +} + +static void preservel(float *r, float *g, float *b, float lin, float lout, float max) +{ + if (lout <= 0.f) + lout = 1.f / (max * 2.f); + *r *= lin / lout; + *g *= lin / lout; + *b *= lin / lout; +} + +#define DEPTH 8 +#include "colorchannelmixer_template.c" + +#undef DEPTH +#define DEPTH 16 +#include "colorchannelmixer_template.c" + #define OFFSET(x) offsetof(ColorChannelMixerContext, x) #define FLAGS AV_OPT_FLAG_FILTERING_PARAM|AV_OPT_FLAG_VIDEO_PARAM|AV_OPT_FLAG_RUNTIME_PARAM @@ -107,484 +128,154 @@ static const enum AVPixelFormat pix_fmts[] = { AV_PIX_FMT_NONE }; -static float lerpf(float v0, float v1, float f) -{ - return v0 + (v1 - v0) * f; -} - -static void preservel(float *r, float *g, float *b, float lin, float lout, float max) -{ - if (lout <= 0.f) - lout = 1.f / (max * 2.f); - *r *= lin / lout; - *g *= lin / lout; - *b *= lin / lout; -} - -static av_always_inline int filter_slice_rgba_planar(AVFilterContext *ctx, void *arg, int jobnr, int nb_jobs, - int have_alpha, int pc) -{ - ColorChannelMixerContext *s = ctx->priv; - ThreadData *td = arg; - AVFrame *in = td->in; - AVFrame *out = td->out; - const float pa = s->preserve_amount; - const int slice_start = (out->height * jobnr) / nb_jobs; - const int slice_end = (out->height * (jobnr+1)) / nb_jobs; - const uint8_t *srcg = in->data[0] + slice_start * in->linesize[0]; - const uint8_t *srcb = in->data[1] + slice_start * in->linesize[1]; - const uint8_t *srcr = in->data[2] + slice_start * in->linesize[2]; - const uint8_t *srca = in->data[3] + slice_start * in->linesize[3]; - uint8_t *dstg = out->data[0] + slice_start * out->linesize[0]; - uint8_t *dstb = out->data[1] + slice_start * out->linesize[1]; - uint8_t *dstr = out->data[2] + slice_start * out->linesize[2]; - uint8_t *dsta = out->data[3] + slice_start * out->linesize[3]; - int i, j; - - for (i = slice_start; i < slice_end; i++) { - for (j = 0; j < out->width; j++) { - const uint8_t rin = srcr[j]; - const uint8_t gin = srcg[j]; - const uint8_t bin = srcb[j]; - const uint8_t ain = have_alpha ? srca[j] : 0; - int rout, gout, bout; - - rout = s->lut[R][R][rin] + - s->lut[R][G][gin] + - s->lut[R][B][bin] + - (have_alpha == 1 ? s->lut[R][A][ain] : 0); - gout = s->lut[G][R][rin] + - s->lut[G][G][gin] + - s->lut[G][B][bin] + - (have_alpha == 1 ? s->lut[G][A][ain] : 0); - bout = s->lut[B][R][rin] + - s->lut[B][G][gin] + - s->lut[B][B][bin] + - (have_alpha == 1 ? s->lut[B][A][ain] : 0); - - if (pc) { - float frout = av_clipf(rout, 0.f, 255.f); - float fgout = av_clipf(gout, 0.f, 255.f); - float fbout = av_clipf(bout, 0.f, 255.f); - float lin, lout; - - preserve_color(s->preserve_color, rin, gin, bin, - rout, gout, bout, 255.f, &lin, &lout); - preservel(&frout, &fgout, &fbout, lin, lout, 255.f); - - rout = lrintf(lerpf(rout, frout, pa)); - gout = lrintf(lerpf(gout, fgout, pa)); - bout = lrintf(lerpf(bout, fbout, pa)); - } - - dstr[j] = av_clip_uint8(rout); - dstg[j] = av_clip_uint8(gout); - dstb[j] = av_clip_uint8(bout); - - if (have_alpha == 1) { - dsta[j] = av_clip_uint8(s->lut[A][R][rin] + - s->lut[A][G][gin] + - s->lut[A][B][bin] + - s->lut[A][A][ain]); - } - } - - srcg += in->linesize[0]; - srcb += in->linesize[1]; - srcr += in->linesize[2]; - srca += in->linesize[3]; - dstg += out->linesize[0]; - dstb += out->linesize[1]; - dstr += out->linesize[2]; - dsta += out->linesize[3]; - } - - return 0; -} - -static av_always_inline int filter_slice_rgba16_planar(AVFilterContext *ctx, void *arg, int jobnr, int nb_jobs, - int have_alpha, int depth, int pc) -{ - ColorChannelMixerContext *s = ctx->priv; - ThreadData *td = arg; - AVFrame *in = td->in; - AVFrame *out = td->out; - const float pa = s->preserve_amount; - const float max = (1 << depth) - 1; - const int slice_start = (out->height * jobnr) / nb_jobs; - const int slice_end = (out->height * (jobnr+1)) / nb_jobs; - const uint16_t *srcg = (const uint16_t *)(in->data[0] + slice_start * in->linesize[0]); - const uint16_t *srcb = (const uint16_t *)(in->data[1] + slice_start * in->linesize[1]); - const uint16_t *srcr = (const uint16_t *)(in->data[2] + slice_start * in->linesize[2]); - const uint16_t *srca = (const uint16_t *)(in->data[3] + slice_start * in->linesize[3]); - uint16_t *dstg = (uint16_t *)(out->data[0] + slice_start * out->linesize[0]); - uint16_t *dstb = (uint16_t *)(out->data[1] + slice_start * out->linesize[1]); - uint16_t *dstr = (uint16_t *)(out->data[2] + slice_start * out->linesize[2]); - uint16_t *dsta = (uint16_t *)(out->data[3] + slice_start * out->linesize[3]); - int i, j; - - for (i = slice_start; i < slice_end; i++) { - for (j = 0; j < out->width; j++) { - const uint16_t rin = srcr[j]; - const uint16_t gin = srcg[j]; - const uint16_t bin = srcb[j]; - const uint16_t ain = have_alpha ? srca[j] : 0; - int rout, gout, bout; - - rout = s->lut[R][R][rin] + - s->lut[R][G][gin] + - s->lut[R][B][bin] + - (have_alpha == 1 ? s->lut[R][A][ain] : 0); - gout = s->lut[G][R][rin] + - s->lut[G][G][gin] + - s->lut[G][B][bin] + - (have_alpha == 1 ? s->lut[G][A][ain] : 0); - bout = s->lut[B][R][rin] + - s->lut[B][G][gin] + - s->lut[B][B][bin] + - (have_alpha == 1 ? s->lut[B][A][ain] : 0); - - if (pc) { - float frout = av_clipf(rout, 0.f, max); - float fgout = av_clipf(gout, 0.f, max); - float fbout = av_clipf(bout, 0.f, max); - float lin, lout; - - preserve_color(s->preserve_color, rin, gin, bin, - rout, gout, bout, max, &lin, &lout); - preservel(&frout, &fgout, &fbout, lin, lout, max); - - rout = lrintf(lerpf(rout, frout, pa)); - gout = lrintf(lerpf(gout, fgout, pa)); - bout = lrintf(lerpf(bout, fbout, pa)); - } - - dstr[j] = av_clip_uintp2(rout, depth); - dstg[j] = av_clip_uintp2(gout, depth); - dstb[j] = av_clip_uintp2(bout, depth); - - if (have_alpha == 1) { - dsta[j] = av_clip_uintp2(s->lut[A][R][rin] + - s->lut[A][G][gin] + - s->lut[A][B][bin] + - s->lut[A][A][ain], depth); - } - } - - srcg += in->linesize[0] / 2; - srcb += in->linesize[1] / 2; - srcr += in->linesize[2] / 2; - srca += in->linesize[3] / 2; - dstg += out->linesize[0] / 2; - dstb += out->linesize[1] / 2; - dstr += out->linesize[2] / 2; - dsta += out->linesize[3] / 2; - } - - return 0; -} - static int filter_slice_gbrp(AVFilterContext *ctx, void *arg, int jobnr, int nb_jobs) { - return filter_slice_rgba_planar(ctx, arg, jobnr, nb_jobs, 0, 0); + return filter_slice_rgba_planar_8(ctx, arg, jobnr, nb_jobs, 0, 8, 0); } static int filter_slice_gbrap(AVFilterContext *ctx, void *arg, int jobnr, int nb_jobs) { - return filter_slice_rgba_planar(ctx, arg, jobnr, nb_jobs, 1, 0); + return filter_slice_rgba_planar_8(ctx, arg, jobnr, nb_jobs, 1, 8, 0); } static int filter_slice_gbrp_pl(AVFilterContext *ctx, void *arg, int jobnr, int nb_jobs) { - return filter_slice_rgba_planar(ctx, arg, jobnr, nb_jobs, 0, 1); + return filter_slice_rgba_planar_8(ctx, arg, jobnr, nb_jobs, 0, 8, 1); } static int filter_slice_gbrap_pl(AVFilterContext *ctx, void *arg, int jobnr, int nb_jobs) { - return filter_slice_rgba_planar(ctx, arg, jobnr, nb_jobs, 1, 1); + return filter_slice_rgba_planar_8(ctx, arg, jobnr, nb_jobs, 1, 8, 1); } static int filter_slice_gbrp9(AVFilterContext *ctx, void *arg, int jobnr, int nb_jobs) { - return filter_slice_rgba16_planar(ctx, arg, jobnr, nb_jobs, 0, 9, 0); + return filter_slice_rgba_planar_16(ctx, arg, jobnr, nb_jobs, 0, 9, 0); } static int filter_slice_gbrp10(AVFilterContext *ctx, void *arg, int jobnr, int nb_jobs) { - return filter_slice_rgba16_planar(ctx, arg, jobnr, nb_jobs, 0, 10, 0); + return filter_slice_rgba_planar_16(ctx, arg, jobnr, nb_jobs, 0, 10, 0); } static int filter_slice_gbrap10(AVFilterContext *ctx, void *arg, int jobnr, int nb_jobs) { - return filter_slice_rgba16_planar(ctx, arg, jobnr, nb_jobs, 1, 10, 0); + return filter_slice_rgba_planar_16(ctx, arg, jobnr, nb_jobs, 1, 10, 0); } static int filter_slice_gbrp12(AVFilterContext *ctx, void *arg, int jobnr, int nb_jobs) { - return filter_slice_rgba16_planar(ctx, arg, jobnr, nb_jobs, 0, 12, 0); + return filter_slice_rgba_planar_16(ctx, arg, jobnr, nb_jobs, 0, 12, 0); } static int filter_slice_gbrap12(AVFilterContext *ctx, void *arg, int jobnr, int nb_jobs) { - return filter_slice_rgba16_planar(ctx, arg, jobnr, nb_jobs, 1, 12, 0); + return filter_slice_rgba_planar_16(ctx, arg, jobnr, nb_jobs, 1, 12, 0); } static int filter_slice_gbrp14(AVFilterContext *ctx, void *arg, int jobnr, int nb_jobs) { - return filter_slice_rgba16_planar(ctx, arg, jobnr, nb_jobs, 0, 14, 0); + return filter_slice_rgba_planar_16(ctx, arg, jobnr, nb_jobs, 0, 14, 0); } static int filter_slice_gbrp16(AVFilterContext *ctx, void *arg, int jobnr, int nb_jobs) { - return filter_slice_rgba16_planar(ctx, arg, jobnr, nb_jobs, 0, 16, 0); + return filter_slice_rgba_planar_16(ctx, arg, jobnr, nb_jobs, 0, 16, 0); } static int filter_slice_gbrap16(AVFilterContext *ctx, void *arg, int jobnr, int nb_jobs) { - return filter_slice_rgba16_planar(ctx, arg, jobnr, nb_jobs, 1, 16, 0); + return filter_slice_rgba_planar_16(ctx, arg, jobnr, nb_jobs, 1, 16, 0); } static int filter_slice_gbrp9_pl(AVFilterContext *ctx, void *arg, int jobnr, int nb_jobs) { - return filter_slice_rgba16_planar(ctx, arg, jobnr, nb_jobs, 0, 9, 1); + return filter_slice_rgba_planar_16(ctx, arg, jobnr, nb_jobs, 0, 9, 1); } static int filter_slice_gbrp10_pl(AVFilterContext *ctx, void *arg, int jobnr, int nb_jobs) { - return filter_slice_rgba16_planar(ctx, arg, jobnr, nb_jobs, 0, 10, 1); + return filter_slice_rgba_planar_16(ctx, arg, jobnr, nb_jobs, 0, 10, 1); } static int filter_slice_gbrap10_pl(AVFilterContext *ctx, void *arg, int jobnr, int nb_jobs) { - return filter_slice_rgba16_planar(ctx, arg, jobnr, nb_jobs, 1, 10, 1); + return filter_slice_rgba_planar_16(ctx, arg, jobnr, nb_jobs, 1, 10, 1); } static int filter_slice_gbrp12_pl(AVFilterContext *ctx, void *arg, int jobnr, int nb_jobs) { - return filter_slice_rgba16_planar(ctx, arg, jobnr, nb_jobs, 0, 12, 1); + return filter_slice_rgba_planar_16(ctx, arg, jobnr, nb_jobs, 0, 12, 1); } static int filter_slice_gbrap12_pl(AVFilterContext *ctx, void *arg, int jobnr, int nb_jobs) { - return filter_slice_rgba16_planar(ctx, arg, jobnr, nb_jobs, 1, 12, 1); + return filter_slice_rgba_planar_16(ctx, arg, jobnr, nb_jobs, 1, 12, 1); } static int filter_slice_gbrp14_pl(AVFilterContext *ctx, void *arg, int jobnr, int nb_jobs) { - return filter_slice_rgba16_planar(ctx, arg, jobnr, nb_jobs, 0, 14, 1); + return filter_slice_rgba_planar_16(ctx, arg, jobnr, nb_jobs, 0, 14, 1); } static int filter_slice_gbrp16_pl(AVFilterContext *ctx, void *arg, int jobnr, int nb_jobs) { - return filter_slice_rgba16_planar(ctx, arg, jobnr, nb_jobs, 0, 16, 1); + return filter_slice_rgba_planar_16(ctx, arg, jobnr, nb_jobs, 0, 16, 1); } static int filter_slice_gbrap16_pl(AVFilterContext *ctx, void *arg, int jobnr, int nb_jobs) { - return filter_slice_rgba16_planar(ctx, arg, jobnr, nb_jobs, 1, 16, 1); -} - -static av_always_inline int filter_slice_rgba_packed(AVFilterContext *ctx, void *arg, int jobnr, int nb_jobs, - int have_alpha, int step, int pc) -{ - ColorChannelMixerContext *s = ctx->priv; - ThreadData *td = arg; - AVFrame *in = td->in; - AVFrame *out = td->out; - const float pa = s->preserve_amount; - const int slice_start = (out->height * jobnr) / nb_jobs; - const int slice_end = (out->height * (jobnr+1)) / nb_jobs; - const uint8_t roffset = s->rgba_map[R]; - const uint8_t goffset = s->rgba_map[G]; - const uint8_t boffset = s->rgba_map[B]; - const uint8_t aoffset = s->rgba_map[A]; - const uint8_t *srcrow = in->data[0] + slice_start * in->linesize[0]; - uint8_t *dstrow = out->data[0] + slice_start * out->linesize[0]; - int i, j; - - for (i = slice_start; i < slice_end; i++) { - const uint8_t *src = srcrow; - uint8_t *dst = dstrow; - - for (j = 0; j < out->width * step; j += step) { - const uint8_t rin = src[j + roffset]; - const uint8_t gin = src[j + goffset]; - const uint8_t bin = src[j + boffset]; - const uint8_t ain = src[j + aoffset]; - int rout, gout, bout; - - rout = s->lut[R][R][rin] + - s->lut[R][G][gin] + - s->lut[R][B][bin] + - (have_alpha == 1 ? s->lut[R][A][ain] : 0); - gout = s->lut[G][R][rin] + - s->lut[G][G][gin] + - s->lut[G][B][bin] + - (have_alpha == 1 ? s->lut[G][A][ain] : 0); - bout = s->lut[B][R][rin] + - s->lut[B][G][gin] + - s->lut[B][B][bin] + - (have_alpha == 1 ? s->lut[B][A][ain] : 0); - - if (pc) { - float frout = av_clipf(rout, 0.f, 255.f); - float fgout = av_clipf(gout, 0.f, 255.f); - float fbout = av_clipf(bout, 0.f, 255.f); - float lin, lout; - - preserve_color(s->preserve_color, rin, gin, bin, - rout, gout, bout, 255.f, &lin, &lout); - preservel(&frout, &fgout, &fbout, lin, lout, 255.f); - - rout = lrintf(lerpf(rout, frout, pa)); - gout = lrintf(lerpf(gout, fgout, pa)); - bout = lrintf(lerpf(bout, fbout, pa)); - } - - dst[j + roffset] = av_clip_uint8(rout); - dst[j + goffset] = av_clip_uint8(gout); - dst[j + boffset] = av_clip_uint8(bout); - - if (have_alpha == 1) { - dst[j + aoffset] = av_clip_uint8(s->lut[A][R][rin] + - s->lut[A][G][gin] + - s->lut[A][B][bin] + - s->lut[A][A][ain]); - } else if (have_alpha == -1 && in != out) - dst[j + aoffset] = 0; - } - - srcrow += in->linesize[0]; - dstrow += out->linesize[0]; - } - - return 0; -} - -static av_always_inline int filter_slice_rgba16_packed(AVFilterContext *ctx, void *arg, int jobnr, int nb_jobs, - int have_alpha, int step, int pc) -{ - ColorChannelMixerContext *s = ctx->priv; - ThreadData *td = arg; - AVFrame *in = td->in; - AVFrame *out = td->out; - const float pa = s->preserve_amount; - const int slice_start = (out->height * jobnr) / nb_jobs; - const int slice_end = (out->height * (jobnr+1)) / nb_jobs; - const uint8_t roffset = s->rgba_map[R]; - const uint8_t goffset = s->rgba_map[G]; - const uint8_t boffset = s->rgba_map[B]; - const uint8_t aoffset = s->rgba_map[A]; - const uint8_t *srcrow = in->data[0] + slice_start * in->linesize[0]; - uint8_t *dstrow = out->data[0] + slice_start * out->linesize[0]; - int i, j; - - for (i = slice_start; i < slice_end; i++) { - const uint16_t *src = (const uint16_t *)srcrow; - uint16_t *dst = (uint16_t *)dstrow; - - for (j = 0; j < out->width * step; j += step) { - const uint16_t rin = src[j + roffset]; - const uint16_t gin = src[j + goffset]; - const uint16_t bin = src[j + boffset]; - const uint16_t ain = src[j + aoffset]; - int rout, gout, bout; - - rout = s->lut[R][R][rin] + - s->lut[R][G][gin] + - s->lut[R][B][bin] + - (have_alpha == 1 ? s->lut[R][A][ain] : 0); - gout = s->lut[G][R][rin] + - s->lut[G][G][gin] + - s->lut[G][B][bin] + - (have_alpha == 1 ? s->lut[G][A][ain] : 0); - bout = s->lut[B][R][rin] + - s->lut[B][G][gin] + - s->lut[B][B][bin] + - (have_alpha == 1 ? s->lut[B][A][ain] : 0); - - if (pc) { - float frout = av_clipf(rout, 0.f, 65535.f); - float fgout = av_clipf(gout, 0.f, 65535.f); - float fbout = av_clipf(bout, 0.f, 65535.f); - float lin, lout; - - preserve_color(s->preserve_color, rin, gin, bin, - rout, gout, bout, 65535.f, &lin, &lout); - preservel(&frout, &fgout, &fbout, lin, lout, 65535.f); - - rout = lrintf(lerpf(rout, frout, pa)); - gout = lrintf(lerpf(gout, fgout, pa)); - bout = lrintf(lerpf(bout, fbout, pa)); - } - - dst[j + roffset] = av_clip_uint16(rout); - dst[j + goffset] = av_clip_uint16(gout); - dst[j + boffset] = av_clip_uint16(bout); - - if (have_alpha == 1) { - dst[j + aoffset] = av_clip_uint16(s->lut[A][R][rin] + - s->lut[A][G][gin] + - s->lut[A][B][bin] + - s->lut[A][A][ain]); - } - } - - srcrow += in->linesize[0]; - dstrow += out->linesize[0]; - } - - return 0; + return filter_slice_rgba_planar_16(ctx, arg, jobnr, nb_jobs, 1, 16, 1); } static int filter_slice_rgba64(AVFilterContext *ctx, void *arg, int jobnr, int nb_jobs) { - return filter_slice_rgba16_packed(ctx, arg, jobnr, nb_jobs, 1, 4, 0); + return filter_slice_rgba_packed_16(ctx, arg, jobnr, nb_jobs, 1, 4, 0, 16); } static int filter_slice_rgb48(AVFilterContext *ctx, void *arg, int jobnr, int nb_jobs) { - return filter_slice_rgba16_packed(ctx, arg, jobnr, nb_jobs, 0, 3, 0); + return filter_slice_rgba_packed_16(ctx, arg, jobnr, nb_jobs, 0, 3, 0, 16); } static int filter_slice_rgba64_pl(AVFilterContext *ctx, void *arg, int jobnr, int nb_jobs) { - return filter_slice_rgba16_packed(ctx, arg, jobnr, nb_jobs, 1, 4, 1); + return filter_slice_rgba_packed_16(ctx, arg, jobnr, nb_jobs, 1, 4, 1, 16); } static int filter_slice_rgb48_pl(AVFilterContext *ctx, void *arg, int jobnr, int nb_jobs) { - return filter_slice_rgba16_packed(ctx, arg, jobnr, nb_jobs, 0, 3, 1); + return filter_slice_rgba_packed_16(ctx, arg, jobnr, nb_jobs, 0, 3, 1, 16); } static int filter_slice_rgba(AVFilterContext *ctx, void *arg, int jobnr, int nb_jobs) { - return filter_slice_rgba_packed(ctx, arg, jobnr, nb_jobs, 1, 4, 0); + return filter_slice_rgba_packed_8(ctx, arg, jobnr, nb_jobs, 1, 4, 0, 8); } static int filter_slice_rgb24(AVFilterContext *ctx, void *arg, int jobnr, int nb_jobs) { - return filter_slice_rgba_packed(ctx, arg, jobnr, nb_jobs, 0, 3, 0); + return filter_slice_rgba_packed_8(ctx, arg, jobnr, nb_jobs, 0, 3, 0, 8); } static int filter_slice_rgb0(AVFilterContext *ctx, void *arg, int jobnr, int nb_jobs) { - return filter_slice_rgba_packed(ctx, arg, jobnr, nb_jobs, -1, 4, 0); + return filter_slice_rgba_packed_8(ctx, arg, jobnr, nb_jobs, -1, 4, 0, 8); } static int filter_slice_rgba_pl(AVFilterContext *ctx, void *arg, int jobnr, int nb_jobs) { - return filter_slice_rgba_packed(ctx, arg, jobnr, nb_jobs, 1, 4, 1); + return filter_slice_rgba_packed_8(ctx, arg, jobnr, nb_jobs, 1, 4, 1, 8); } static int filter_slice_rgb24_pl(AVFilterContext *ctx, void *arg, int jobnr, int nb_jobs) { - return filter_slice_rgba_packed(ctx, arg, jobnr, nb_jobs, 0, 3, 1); + return filter_slice_rgba_packed_8(ctx, arg, jobnr, nb_jobs, 0, 3, 1, 8); } static int filter_slice_rgb0_pl(AVFilterContext *ctx, void *arg, int jobnr, int nb_jobs) { - return filter_slice_rgba_packed(ctx, arg, jobnr, nb_jobs, -1, 4, 1); + return filter_slice_rgba_packed_8(ctx, arg, jobnr, nb_jobs, -1, 4, 1, 8); } static int config_output(AVFilterLink *outlink)