On 5/3/2018 3:08 PM, Paul B Mahol wrote:
> +#define DEFINE_INTERP_FUNC_PLANAR(name, nbits, depth)
> \
> +static int interp_##nbits##_##name##_p##depth(AVFilterContext *ctx, void
> *arg, int jobnr, int nb_jobs) \
> +{
> \
> + int x, y;
> \
> + const LUT3DContext *lut3d = ctx->priv;
> \
> + const ThreadData *td = arg;
> \
> + const AVFrame *in = td->in;
> \
> + const AVFrame *out = td->out;
> \
> + const int direct = out == in;
> \
> + const int slice_start = (in->height * jobnr ) / nb_jobs;
> \
> + const int slice_end = (in->height * (jobnr+1)) / nb_jobs;
> \
> + uint8_t *grow = out->data[0] + slice_start * out->linesize[0];
> \
> + uint8_t *brow = out->data[1] + slice_start * out->linesize[1];
> \
> + uint8_t *rrow = out->data[2] + slice_start * out->linesize[2];
> \
> + uint8_t *arow = out->data[2] + slice_start * out->linesize[2];
> \
> + const uint8_t *srcgrow = in->data[0] + slice_start * in->linesize[0];
> \
> + const uint8_t *srcbrow = in->data[1] + slice_start * in->linesize[1];
> \
> + const uint8_t *srcrrow = in->data[2] + slice_start * in->linesize[2];
> \
> + const uint8_t *srcarow = in->data[3] + slice_start * in->linesize[3];
> \
> + const float scale = (1. / ((1<<depth) - 1)) * (lut3d->lutsize - 1);
> \
> +
> \
> + for (y = slice_start; y < slice_end; y++) {
> \
> + uint##nbits##_t *dstg = (uint##nbits##_t *)grow;
> \
> + uint##nbits##_t *dstb = (uint##nbits##_t *)brow;
> \
> + uint##nbits##_t *dstr = (uint##nbits##_t *)rrow;
> \
> + uint##nbits##_t *dsta = (uint##nbits##_t *)arow;
> \
> + const uint##nbits##_t *srcg = (const uint##nbits##_t *)srcgrow;
> \
> + const uint##nbits##_t *srcb = (const uint##nbits##_t *)srcbrow;
> \
> + const uint##nbits##_t *srcr = (const uint##nbits##_t *)srcrrow;
> \
> + const uint##nbits##_t *srca = (const uint##nbits##_t *)srcarow;
> \
> + for (x = 0; x < in->width; x++) {
> \
> + const struct rgbvec scaled_rgb = {srcr[x] * scale,
> \
> + srcg[x] * scale,
> \
> + srcb[x] * scale};
> \
> + struct rgbvec vec = interp_##name(lut3d, &scaled_rgb);
> \
> + dstr[x] = av_clip(vec.r * (float)((1<<depth) - 1), 0, (1<<depth)
> - 1); \
> + dstg[x] = av_clip(vec.g * (float)((1<<depth) - 1), 0, (1<<depth)
> - 1); \
> + dstb[x] = av_clip(vec.b * (float)((1<<depth) - 1), 0, (1<<depth)
> - 1); \
av_clip_uintp2(vec.r * (float)((1<<depth) - 1), depth);
av_clip_uintp2(vec.g * (float)((1<<depth) - 1), depth);
av_clip_uintp2(vec.b * (float)((1<<depth) - 1), depth);
> + if (!direct && in->linesize[3])
> \
> + dsta[x] = srca[x];
> \
> + }
> \
> + grow += out->linesize[0];
> \
> + brow += out->linesize[1];
> \
> + rrow += out->linesize[2];
> \
> + arow += out->linesize[3];
> \
> + srcgrow += in->linesize[0];
> \
> + srcbrow += in->linesize[1];
> \
> + srcrrow += in->linesize[2];
> \
> + srcarow += in->linesize[3];
> \
> + }
> \
> + return 0;
> \
> +}
_______________________________________________
ffmpeg-devel mailing list
[email protected]
http://ffmpeg.org/mailman/listinfo/ffmpeg-devel
