[FFmpeg-devel] [PATCH] avfilter/vf_lut3d: add planar rgb support

[James Almer]

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;                                                                                       \
> +}