[deb_ffmpeg.git] / ffmpeg / libavfilter / vf_elbg.c

/*
 * Copyright (c) 2013 Stefano Sabatini
 *
 * 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
 */

/**
 * @file
 * video quantizer filter based on ELBG
 */

#include "libavcodec/elbg.h"
#include "libavutil/opt.h"
#include "libavutil/pixdesc.h"
#include "libavutil/random_seed.h"

#include "avfilter.h"
#include "drawutils.h"
#include "internal.h"
#include "video.h"

typedef struct ColorContext {
    const AVClass *class;
    AVLFG lfg;
    unsigned int lfg_seed;
    int max_steps_nb;
    int *codeword;
    int codeword_length;
    int *codeword_closest_codebook_idxs;
    int *codebook;
    int codebook_length;
    const AVPixFmtDescriptor *pix_desc;
    uint8_t rgba_map[4];
} ELBGContext;

#define OFFSET(x) offsetof(ELBGContext, x)
#define FLAGS AV_OPT_FLAG_VIDEO_PARAM|AV_OPT_FLAG_FILTERING_PARAM

static const AVOption elbg_options[] = {
    { "codebook_length", "set codebook length", OFFSET(codebook_length), AV_OPT_TYPE_INT, { .i64 = 256 }, 1, INT_MAX, FLAGS },
    { "l",               "set codebook length", OFFSET(codebook_length), AV_OPT_TYPE_INT, { .i64 = 256 }, 1, INT_MAX, FLAGS },
    { "nb_steps", "set max number of steps used to compute the mapping", OFFSET(max_steps_nb), AV_OPT_TYPE_INT, { .i64 = 1 }, 1, INT_MAX, FLAGS },
    { "n",        "set max number of steps used to compute the mapping", OFFSET(max_steps_nb), AV_OPT_TYPE_INT, { .i64 = 1 }, 1, INT_MAX, FLAGS },
    { "seed", "set the random seed", OFFSET(lfg_seed), AV_OPT_TYPE_INT, {.i64 = -1}, -1, UINT32_MAX, FLAGS },
    { "s",    "set the random seed", OFFSET(lfg_seed), AV_OPT_TYPE_INT, { .i64 = -1 }, -1, UINT32_MAX, FLAGS },
    { NULL }
};

AVFILTER_DEFINE_CLASS(elbg);

static av_cold int init(AVFilterContext *ctx)
{
    ELBGContext *elbg = ctx->priv;

    if (elbg->lfg_seed == -1)
        elbg->lfg_seed = av_get_random_seed();

    av_lfg_init(&elbg->lfg, elbg->lfg_seed);
    return 0;
}

static int query_formats(AVFilterContext *ctx)
{
    static const enum PixelFormat pix_fmts[] = {
        AV_PIX_FMT_ARGB, AV_PIX_FMT_RGBA, AV_PIX_FMT_ABGR, AV_PIX_FMT_BGRA,
        AV_PIX_FMT_RGB24, AV_PIX_FMT_BGR24,
        AV_PIX_FMT_NONE
    };

    ff_set_common_formats(ctx, ff_make_format_list(pix_fmts));

    return 0;
}

#define NB_COMPONENTS 3

static int config_input(AVFilterLink *inlink)
{
    AVFilterContext *ctx = inlink->dst;
    ELBGContext *elbg = ctx->priv;

    elbg->pix_desc = av_pix_fmt_desc_get(inlink->format);
    elbg->codeword_length = inlink->w * inlink->h;
    elbg->codeword = av_realloc_f(elbg->codeword, elbg->codeword_length,
                                  NB_COMPONENTS * sizeof(*elbg->codeword));
    if (!elbg->codeword)
        return AVERROR(ENOMEM);

    elbg->codeword_closest_codebook_idxs =
        av_realloc_f(elbg->codeword_closest_codebook_idxs, elbg->codeword_length,
                     sizeof(*elbg->codeword_closest_codebook_idxs));
    if (!elbg->codeword_closest_codebook_idxs)
        return AVERROR(ENOMEM);

    elbg->codebook = av_realloc_f(elbg->codebook, elbg->codebook_length,
                                  NB_COMPONENTS * sizeof(*elbg->codebook));
    if (!elbg->codebook)
        return AVERROR(ENOMEM);

    ff_fill_rgba_map(elbg->rgba_map, inlink->format);

    return 0;
}

#define R 0
#define G 1
#define B 2

static int filter_frame(AVFilterLink *inlink, AVFrame *frame)
{
    ELBGContext *elbg = inlink->dst->priv;
    int i, j, k;
    uint8_t *p, *p0;

    const uint8_t r_idx  = elbg->rgba_map[R];
    const uint8_t g_idx  = elbg->rgba_map[G];
    const uint8_t b_idx  = elbg->rgba_map[B];

    /* build the codeword */
    p0 = frame->data[0];
    k = 0;
    for (i = 0; i < inlink->h; i++) {
        p = p0;
        for (j = 0; j < inlink->w; j++) {
            elbg->codeword[k++] = p[r_idx];
            elbg->codeword[k++] = p[g_idx];
            elbg->codeword[k++] = p[b_idx];
            p += elbg->pix_desc->nb_components;
        }
        p0 += frame->linesize[0];
    }

    /* compute the codebook */
    avpriv_init_elbg(elbg->codeword, NB_COMPONENTS, elbg->codeword_length,
                     elbg->codebook, elbg->codebook_length, elbg->max_steps_nb,
                     elbg->codeword_closest_codebook_idxs, &elbg->lfg);
    avpriv_do_elbg(elbg->codeword, NB_COMPONENTS, elbg->codeword_length,
                   elbg->codebook, elbg->codebook_length, elbg->max_steps_nb,
                   elbg->codeword_closest_codebook_idxs, &elbg->lfg);

    /* fill the output with the codebook values */
    p0 = frame->data[0];

    k = 0;
    for (i = 0; i < inlink->h; i++) {
        p = p0;
        for (j = 0; j < inlink->w; j++) {
            int cb_idx = NB_COMPONENTS * elbg->codeword_closest_codebook_idxs[k++];
            p[r_idx] = elbg->codebook[cb_idx];
            p[g_idx] = elbg->codebook[cb_idx+1];
            p[b_idx] = elbg->codebook[cb_idx+2];
            p += elbg->pix_desc->nb_components;
        }
        p0 += frame->linesize[0];
    }

    return ff_filter_frame(inlink->dst->outputs[0], frame);
}

static av_cold void uninit(AVFilterContext *ctx)
{
    ELBGContext *elbg = ctx->priv;

    av_freep(&elbg->codebook);
    av_freep(&elbg->codeword);
    av_freep(&elbg->codeword_closest_codebook_idxs);
}

static const AVFilterPad elbg_inputs[] = {
    {
        .name           = "default",
        .type           = AVMEDIA_TYPE_VIDEO,
        .config_props   = config_input,
        .filter_frame   = filter_frame,
        .needs_writable = 1,
    },
    { NULL }
};

static const AVFilterPad elbg_outputs[] = {
    {
        .name = "default",
        .type = AVMEDIA_TYPE_VIDEO,
    },
    { NULL }
};

AVFilter ff_vf_elbg = {
    .name          = "elbg",
    .description   = NULL_IF_CONFIG_SMALL("Apply posterize effect, using the ELBG algorithm."),
    .priv_size     = sizeof(ELBGContext),
    .priv_class    = &elbg_class,
    .query_formats = query_formats,
    .init          = init,
    .uninit        = uninit,
    .inputs        = elbg_inputs,
    .outputs       = elbg_outputs,
};
Commit	Line	Data
2ba45a60 DM	1	/*
	2	* Copyright (c) 2013 Stefano Sabatini
	3	*
	4	* This file is part of FFmpeg.
	5	*
	6	* FFmpeg is free software; you can redistribute it and/or
	7	* modify it under the terms of the GNU Lesser General Public
	8	* License as published by the Free Software Foundation; either
	9	* version 2.1 of the License, or (at your option) any later version.
	10	*
	11	* FFmpeg is distributed in the hope that it will be useful,
	12	* but WITHOUT ANY WARRANTY; without even the implied warranty of
	13	* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
	14	* Lesser General Public License for more details.
	15	*
	16	* You should have received a copy of the GNU Lesser General Public
	17	* License along with FFmpeg; if not, write to the Free Software
	18	* Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
	19	*/
	20
	21	/**
	22	* @file
	23	* video quantizer filter based on ELBG
	24	*/
	25
	26	#include "libavcodec/elbg.h"
	27	#include "libavutil/opt.h"
	28	#include "libavutil/pixdesc.h"
	29	#include "libavutil/random_seed.h"
	30
	31	#include "avfilter.h"
	32	#include "drawutils.h"
	33	#include "internal.h"
	34	#include "video.h"
	35
	36	typedef struct ColorContext {
	37	const AVClass *class;
	38	AVLFG lfg;
	39	unsigned int lfg_seed;
	40	int max_steps_nb;
	41	int *codeword;
	42	int codeword_length;
	43	int *codeword_closest_codebook_idxs;
	44	int *codebook;
	45	int codebook_length;
	46	const AVPixFmtDescriptor *pix_desc;
	47	uint8_t rgba_map[4];
	48	} ELBGContext;
	49
	50	#define OFFSET(x) offsetof(ELBGContext, x)
	51	#define FLAGS AV_OPT_FLAG_VIDEO_PARAM\|AV_OPT_FLAG_FILTERING_PARAM
	52
	53	static const AVOption elbg_options[] = {
	54	{ "codebook_length", "set codebook length", OFFSET(codebook_length), AV_OPT_TYPE_INT, { .i64 = 256 }, 1, INT_MAX, FLAGS },
	55	{ "l", "set codebook length", OFFSET(codebook_length), AV_OPT_TYPE_INT, { .i64 = 256 }, 1, INT_MAX, FLAGS },
	56	{ "nb_steps", "set max number of steps used to compute the mapping", OFFSET(max_steps_nb), AV_OPT_TYPE_INT, { .i64 = 1 }, 1, INT_MAX, FLAGS },
	57	{ "n", "set max number of steps used to compute the mapping", OFFSET(max_steps_nb), AV_OPT_TYPE_INT, { .i64 = 1 }, 1, INT_MAX, FLAGS },
	58	{ "seed", "set the random seed", OFFSET(lfg_seed), AV_OPT_TYPE_INT, {.i64 = -1}, -1, UINT32_MAX, FLAGS },
	59	{ "s", "set the random seed", OFFSET(lfg_seed), AV_OPT_TYPE_INT, { .i64 = -1 }, -1, UINT32_MAX, FLAGS },
	60	{ NULL }
	61	};
	62
	63	AVFILTER_DEFINE_CLASS(elbg);
	64
65	static av_cold int init(AVFilterContext *ctx)
66	{
67	ELBGContext *elbg = ctx->priv;
68
69	if (elbg->lfg_seed == -1)
70	elbg->lfg_seed = av_get_random_seed();
71
72	av_lfg_init(&elbg->lfg, elbg->lfg_seed);
73	return 0;
74	}
75
76	static int query_formats(AVFilterContext *ctx)
77	{
78	static const enum PixelFormat pix_fmts[] = {
79	AV_PIX_FMT_ARGB, AV_PIX_FMT_RGBA, AV_PIX_FMT_ABGR, AV_PIX_FMT_BGRA,
80	AV_PIX_FMT_RGB24, AV_PIX_FMT_BGR24,
81	AV_PIX_FMT_NONE
82	};
83
84	ff_set_common_formats(ctx, ff_make_format_list(pix_fmts));
85
86	return 0;
87	}
88
89	#define NB_COMPONENTS 3
90
91	static int config_input(AVFilterLink *inlink)
92	{
93	AVFilterContext *ctx = inlink->dst;
94	ELBGContext *elbg = ctx->priv;
95
96	elbg->pix_desc = av_pix_fmt_desc_get(inlink->format);
97	elbg->codeword_length = inlink->w * inlink->h;
98	elbg->codeword = av_realloc_f(elbg->codeword, elbg->codeword_length,
99	NB_COMPONENTS * sizeof(*elbg->codeword));
100	if (!elbg->codeword)
101	return AVERROR(ENOMEM);
102
103	elbg->codeword_closest_codebook_idxs =
104	av_realloc_f(elbg->codeword_closest_codebook_idxs, elbg->codeword_length,
105	sizeof(*elbg->codeword_closest_codebook_idxs));
106	if (!elbg->codeword_closest_codebook_idxs)
107	return AVERROR(ENOMEM);
108
109	elbg->codebook = av_realloc_f(elbg->codebook, elbg->codebook_length,
110	NB_COMPONENTS * sizeof(*elbg->codebook));
111	if (!elbg->codebook)
112	return AVERROR(ENOMEM);
113
114	ff_fill_rgba_map(elbg->rgba_map, inlink->format);
115
116	return 0;
117	}
118
119	#define R 0
120	#define G 1
121	#define B 2
122
123	static int filter_frame(AVFilterLink inlink, AVFrame frame)
124	{
125	ELBGContext *elbg = inlink->dst->priv;
126	int i, j, k;
127	uint8_t p, p0;
128
129	const uint8_t r_idx = elbg->rgba_map[R];
130	const uint8_t g_idx = elbg->rgba_map[G];
131	const uint8_t b_idx = elbg->rgba_map[B];
132
133	/* build the codeword */
134	p0 = frame->data[0];
135	k = 0;
136	for (i = 0; i < inlink->h; i++) {
137	p = p0;
138	for (j = 0; j < inlink->w; j++) {
139	elbg->codeword[k++] = p[r_idx];
140	elbg->codeword[k++] = p[g_idx];
141	elbg->codeword[k++] = p[b_idx];
142	p += elbg->pix_desc->nb_components;
143	}
144	p0 += frame->linesize[0];
145	}
146
147	/* compute the codebook */
148	avpriv_init_elbg(elbg->codeword, NB_COMPONENTS, elbg->codeword_length,
149	elbg->codebook, elbg->codebook_length, elbg->max_steps_nb,
150	elbg->codeword_closest_codebook_idxs, &elbg->lfg);
151	avpriv_do_elbg(elbg->codeword, NB_COMPONENTS, elbg->codeword_length,
152	elbg->codebook, elbg->codebook_length, elbg->max_steps_nb,
153	elbg->codeword_closest_codebook_idxs, &elbg->lfg);
154
155	/* fill the output with the codebook values */
156	p0 = frame->data[0];
157
158	k = 0;
159	for (i = 0; i < inlink->h; i++) {
160	p = p0;
161	for (j = 0; j < inlink->w; j++) {
162	int cb_idx = NB_COMPONENTS * elbg->codeword_closest_codebook_idxs[k++];
163	p[r_idx] = elbg->codebook[cb_idx];
164	p[g_idx] = elbg->codebook[cb_idx+1];
165	p[b_idx] = elbg->codebook[cb_idx+2];
166	p += elbg->pix_desc->nb_components;
167	}
168	p0 += frame->linesize[0];
169	}
170
171	return ff_filter_frame(inlink->dst->outputs[0], frame);
172	}
173
174	static av_cold void uninit(AVFilterContext *ctx)
175	{
176	ELBGContext *elbg = ctx->priv;
177
178	av_freep(&elbg->codebook);
179	av_freep(&elbg->codeword);
180	av_freep(&elbg->codeword_closest_codebook_idxs);
181	}
182
183	static const AVFilterPad elbg_inputs[] = {
184	{
185	.name = "default",
186	.type = AVMEDIA_TYPE_VIDEO,
187	.config_props = config_input,
188	.filter_frame = filter_frame,
189	.needs_writable = 1,
190	},
191	{ NULL }
192	};
193
194	static const AVFilterPad elbg_outputs[] = {
195	{
196	.name = "default",
197	.type = AVMEDIA_TYPE_VIDEO,
198	},
199	{ NULL }
200	};
201
202	AVFilter ff_vf_elbg = {
203	.name = "elbg",
204	.description = NULL_IF_CONFIG_SMALL("Apply posterize effect, using the ELBG algorithm."),
205	.priv_size = sizeof(ELBGContext),
206	.priv_class = &elbg_class,
207	.query_formats = query_formats,
208	.init = init,
209	.uninit = uninit,
210	.inputs = elbg_inputs,
211	.outputs = elbg_outputs,
212	};