63 #define CACHE_SIZE (1<<(3*NBITS))
103 #define OFFSET(x) offsetof(PaletteUseContext, x)
104 #define FLAGS AV_OPT_FLAG_FILTERING_PARAM|AV_OPT_FLAG_VIDEO_PARAM
137 if (!in || !inpal || !out) {
151 return av_clip_uint8((px >> 16 & 0xff) + ((er * scale) / (1<<shift))) << 16
152 | av_clip_uint8((px >> 8 & 0xff) + ((eg * scale) / (1<<shift))) << 8
153 | av_clip_uint8((px & 0xff) + ((eb * scale) / (1<<shift)));
159 const int dr = c1[0] - c2[0];
160 const int dg = c1[1] - c2[1];
161 const int db = c1[2] - c2[2];
162 return dr*dr + dg*dg + db*db;
167 int i, pal_id = -1, min_dist = INT_MAX;
170 const uint32_t
c = palette[i];
172 if ((c & 0xff000000) == 0xff000000) {
174 palette[i]>>16 & 0xff,
175 palette[i]>> 8 & 0xff,
178 const int d =
diff(palrgb, rgb);
200 const int s = kd->
split;
201 int dx, nearer_kd_id, further_kd_id;
203 const int current_to_target =
diff(target, current);
205 if (current_to_target < nearest->dist_sqd) {
207 nearest->
dist_sqd = current_to_target;
211 dx = target[
s] - current[
s];
216 if (nearer_kd_id != -1)
219 if (further_kd_id != -1 && dx*dx < nearest->dist_sqd)
238 int pos = 0, best_node_id = -1, best_dist = INT_MAX, cur_color_id = 0;
244 const struct color_node *kd = &root[cur_color_id];
246 const int current_to_target =
diff(target, current);
250 if (current_to_target < best_dist) {
251 best_node_id = cur_color_id;
252 if (!current_to_target)
254 best_dist = current_to_target;
260 const int dx = target[
split] - current[
split];
261 int nearer_kd_id, further_kd_id;
267 if (nearer_kd_id != -1) {
268 if (further_kd_id != -1) {
279 cur_color_id = nearer_kd_id;
281 }
else if (dx*dx < best_dist) {
286 cur_color_id = further_kd_id;
297 }
while (node->
dx2 >= best_dist);
308 #define COLORMAP_NEAREST(search, palette, root, target) \
309 search == COLOR_SEARCH_NNS_ITERATIVE ? colormap_nearest_iterative(root, target) : \
310 search == COLOR_SEARCH_NNS_RECURSIVE ? colormap_nearest_recursive(root, target) : \
311 colormap_nearest_bruteforce(palette, target)
336 if (e->
color == color)
352 int *er,
int *eg,
int *eb,
358 const int dstx =
color_get(cache, c, r, g, b, map, palette, search_method);
359 const uint32_t dstc = palette[dstx];
360 *er = r - (dstc >> 16 & 0xff);
361 *eg = g - (dstc >> 8 & 0xff);
362 *eb = b - (dstc & 0xff);
367 int x_start,
int y_start,
int w,
int h,
375 const int src_linesize = in ->
linesize[0] >> 2;
376 const int dst_linesize = out->
linesize[0];
377 uint32_t *
src = ((uint32_t *)in ->
data[0]) + y_start*src_linesize;
378 uint8_t *dst = out->
data[0] + y_start*dst_linesize;
383 for (y = y_start; y <
h; y++) {
384 for (x = x_start; x < w; x++) {
389 const uint8_t r8 = src[x] >> 16 & 0xff;
390 const uint8_t g8 = src[x] >> 8 & 0xff;
391 const uint8_t b8 = src[x] & 0xff;
392 const uint8_t r = av_clip_uint8(r8 + d);
393 const uint8_t g = av_clip_uint8(g8 + d);
394 const uint8_t b = av_clip_uint8(b8 + d);
395 const uint32_t
c = r<<16 | g<<8 |
b;
396 const int color =
color_get(cache, c, r, g, b, map, palette, search_method);
403 const int right = x < w - 1, down = y < h - 1;
410 if (right) src[ x + 1] =
dither_color(src[ x + 1], er, eg, eb, 3, 3);
411 if ( down) src[src_linesize + x ] =
dither_color(src[src_linesize + x ], er, eg, eb, 3, 3);
412 if (right && down) src[src_linesize + x + 1] =
dither_color(src[src_linesize + x + 1], er, eg, eb, 2, 3);
415 const int right = x < w - 1, down = y < h - 1, left = x > x_start;
422 if (right) src[ x + 1] =
dither_color(src[ x + 1], er, eg, eb, 7, 4);
423 if (left && down) src[src_linesize + x - 1] =
dither_color(src[src_linesize + x - 1], er, eg, eb, 3, 4);
424 if ( down) src[src_linesize + x ] =
dither_color(src[src_linesize + x ], er, eg, eb, 5, 4);
425 if (right && down) src[src_linesize + x + 1] =
dither_color(src[src_linesize + x + 1], er, eg, eb, 1, 4);
428 const int right = x < w - 1, down = y < h - 1, left = x > x_start;
429 const int right2 = x < w - 2, left2 = x > x_start + 1;
436 if (right) src[ x + 1] =
dither_color(src[ x + 1], er, eg, eb, 4, 4);
437 if (right2) src[ x + 2] =
dither_color(src[ x + 2], er, eg, eb, 3, 4);
440 if (left2) src[ src_linesize + x - 2] =
dither_color(src[ src_linesize + x - 2], er, eg, eb, 1, 4);
441 if (left) src[ src_linesize + x - 1] =
dither_color(src[ src_linesize + x - 1], er, eg, eb, 2, 4);
442 src[ src_linesize + x ] =
dither_color(src[ src_linesize + x ], er, eg, eb, 3, 4);
443 if (right) src[ src_linesize + x + 1] =
dither_color(src[ src_linesize + x + 1], er, eg, eb, 2, 4);
444 if (right2) src[ src_linesize + x + 2] =
dither_color(src[ src_linesize + x + 2], er, eg, eb, 1, 4);
448 const int right = x < w - 1, down = y < h - 1, left = x > x_start;
455 if (right) src[ x + 1] =
dither_color(src[ x + 1], er, eg, eb, 2, 2);
456 if (left && down) src[src_linesize + x - 1] =
dither_color(src[src_linesize + x - 1], er, eg, eb, 1, 2);
457 if ( down) src[src_linesize + x ] =
dither_color(src[src_linesize + x ], er, eg, eb, 1, 2);
460 const uint8_t r = src[x] >> 16 & 0xff;
461 const uint8_t g = src[x] >> 8 & 0xff;
463 const int color =
color_get(cache, src[x] & 0xffffff, r, g, b, map, palette, search_method);
479 int parent_id,
int node_id,
482 const struct color_node *node = &map[node_id];
483 const uint32_t fontcolor = node->
val[0] > 0x50 &&
484 node->
val[1] > 0x50 &&
485 node->
val[2] > 0x50 ? 0 : 0xffffff;
487 "label=\"%c%02X%c%02X%c%02X%c\" "
488 "fillcolor=\"#%02x%02x%02x\" "
489 "fontcolor=\"#%06X\"]\n",
495 node->
val[0], node->
val[1], node->
val[2],
498 av_bprintf(buf,
"%*cnode%d -> node%d\n", depth*INDENT,
' ',
520 av_bprintf(&buf,
" node [style=filled fontsize=10 shape=box]\n");
524 fwrite(buf.str, 1, buf.len, f);
535 for (r = 0; r < 256; r++) {
536 for (g = 0; g < 256; g++) {
537 for (b = 0; b < 256; b++) {
542 const uint32_t
c1 = palette[r1];
543 const uint32_t
c2 = palette[r2];
544 const uint8_t palrgb1[] = { c1>>16 & 0xff, c1>> 8 & 0xff, c1 & 0xff };
545 const uint8_t palrgb2[] = { c2>>16 & 0xff, c2>> 8 & 0xff, c2 & 0xff };
546 const int d1 =
diff(palrgb1, rgb);
547 const int d2 =
diff(palrgb2, rgb);
550 "/!\\ %02X%02X%02X: %d ! %d (%06X ! %06X) / dist: %d ! %d\n",
551 r, g, b, r1, r2, c1 & 0xffffff, c2 & 0xffffff, d1, d2);
571 typedef int (*
cmp_func)(
const void *,
const void *);
573 #define DECLARE_CMP_FUNC(name, pos) \
574 static int cmp_##name(const void *pa, const void *pb) \
576 const struct color *a = pa; \
577 const struct color *b = pb; \
578 return (a->value >> (8 * (2 - (pos))) & 0xff) \
579 - (b->value >> (8 * (2 - (pos))) & 0xff); \
593 unsigned nb_color = 0;
595 struct color tmp_pal[256];
598 ranges.
min[0] = ranges.
min[1] = ranges.
min[2] = 0xff;
599 ranges.
max[0] = ranges.
max[1] = ranges.
max[2] = 0x00;
602 const uint32_t
c = palette[i];
608 r < box->
min[0] || g < box->
min[1] || b < box->
min[2] ||
609 r > box->
max[0] || g > box->
max[1] || b > box->
max[2])
612 if (r < ranges.
min[0]) ranges.
min[0] =
r;
613 if (g < ranges.
min[1]) ranges.
min[1] =
g;
614 if (b < ranges.
min[2]) ranges.
min[2] =
b;
616 if (r > ranges.
max[0]) ranges.
max[0] =
r;
617 if (g > ranges.
max[1]) ranges.
max[1] =
g;
618 if (b > ranges.
max[2]) ranges.
max[2] =
b;
620 tmp_pal[nb_color].
value =
c;
621 tmp_pal[nb_color].
pal_id = i;
630 wr = ranges.
max[0] - ranges.
min[0];
631 wg = ranges.
max[1] - ranges.
min[1];
632 wb = ranges.
max[2] - ranges.
min[2];
633 if (wr >= wg && wr >= wb) longest = 0;
634 if (wg >= wr && wg >= wb) longest = 1;
635 if (wb >= wr && wb >= wg) longest = 2;
637 *component = longest;
642 return tmp_pal[nb_color >> 1].
pal_id;
652 int component, cur_id;
653 int node_left_id = -1, node_right_id = -1;
656 const int pal_id =
get_next_color(color_used, palette, &component, box);
662 cur_id = (*nb_used)++;
665 node->
split = component;
667 node->
val[0] = c>>16 & 0xff;
668 node->
val[1] = c>> 8 & 0xff;
669 node->
val[2] = c & 0xff;
671 color_used[pal_id] = 1;
675 box1.
max[component] = node->
val[component];
676 box2.
min[component] = node->
val[component] + 1;
678 node_left_id =
colormap_insert(map, color_used, nb_used, palette, &box1);
680 if (box2.
min[component] <= box2.
max[component])
681 node_right_id =
colormap_insert(map, color_used, nb_used, palette, &box2);
691 const int c1 = *(
const uint32_t *)a & 0xffffff;
692 const int c2 = *(
const uint32_t *)b & 0xffffff;
700 uint32_t last_color = 0;
707 if (i != 0 && c == last_color) {
712 if ((c & 0xff000000) != 0xff000000) {
718 box.
min[0] = box.
min[1] = box.
min[2] = 0x00;
719 box.
max[0] = box.
max[1] = box.
max[2] = 0xff;
733 const AVFrame *in2,
int frame_count)
737 uint32_t *src1 = (uint32_t *)in1->
data[0];
739 const int src1_linesize = in1->
linesize[0] >> 2;
740 const int src2_linesize = in2->
linesize[0];
742 unsigned mean_err = 0;
745 for (x = 0; x < in1->
width; x++) {
746 const uint32_t
c1 = src1[x];
747 const uint32_t
c2 = palette[src2[x]];
748 const uint8_t rgb1[] = {c1 >> 16 & 0xff, c1 >> 8 & 0xff, c1 & 0xff};
749 const uint8_t rgb2[] = {c2 >> 16 & 0xff, c2 >> 8 & 0xff, c2 & 0xff};
750 mean_err +=
diff(rgb1, rgb2);
752 src1 += src1_linesize;
753 src2 += src2_linesize;
765 int *xp,
int *yp,
int *wp,
int *hp)
767 int x_start = 0, y_start = 0;
773 int x_end = cur_src->
width - 1,
774 y_end = cur_src->
height - 1;
775 const uint32_t *prv_srcp = (
const uint32_t *)prv_src->
data[0];
776 const uint32_t *cur_srcp = (
const uint32_t *)cur_src->
data[0];
780 const int prv_src_linesize = prv_src->
linesize[0] >> 2;
781 const int cur_src_linesize = cur_src->
linesize[0] >> 2;
782 const int prv_dst_linesize = prv_dst->
linesize[0];
783 const int cur_dst_linesize = cur_dst->
linesize[0];
786 while (y_start < y_end && !memcmp(prv_srcp + y_start*prv_src_linesize,
787 cur_srcp + y_start*cur_src_linesize,
788 cur_src->
width * 4)) {
789 memcpy(cur_dstp + y_start*cur_dst_linesize,
790 prv_dstp + y_start*prv_dst_linesize,
794 while (y_end > y_start && !memcmp(prv_srcp + y_end*prv_src_linesize,
795 cur_srcp + y_end*cur_src_linesize,
796 cur_src->
width * 4)) {
797 memcpy(cur_dstp + y_end*cur_dst_linesize,
798 prv_dstp + y_end*prv_dst_linesize,
803 height = y_end + 1 - y_start;
806 while (x_start < x_end) {
808 for (y = y_start; y <= y_end; y++) {
809 if (prv_srcp[y*prv_src_linesize + x_start] != cur_srcp[y*cur_src_linesize + x_start]) {
818 while (x_end > x_start) {
820 for (y = y_start; y <= y_end; y++) {
821 if (prv_srcp[y*prv_src_linesize + x_end] != cur_srcp[y*cur_src_linesize + x_end]) {
830 width = x_end + 1 - x_start;
833 for (y = y_start; y <= y_end; y++)
834 memcpy(cur_dstp + y*cur_dst_linesize,
835 prv_dstp + y*prv_dst_linesize, x_start);
837 if (x_end != cur_src->
width - 1) {
838 const int copy_len = cur_src->
width - 1 - x_end;
839 for (y = y_start; y <= y_end; y++)
840 memcpy(cur_dstp + y*cur_dst_linesize + x_end + 1,
841 prv_dstp + y*prv_dst_linesize + x_end + 1,
878 av_dlog(ctx,
"%dx%d rect: (%d;%d) -> (%d,%d) [area:%dx%d]\n",
881 if (s->
set_frame(s, out, in, x, y, w, h) < 0) {
913 "Palette input must contain exactly %d pixels. "
914 "Specified input has %dx%d=%d pixels\n",
916 inlink->
w * inlink->
h);
925 const uint32_t *p = (
const uint32_t *)palette_frame->
data[0];
926 const int p_linesize = palette_frame->
linesize[0] >> 2;
929 for (y = 0; y < palette_frame->
height; y++) {
930 for (x = 0; x < palette_frame->
width; x++)
957 #define DEFINE_SET_FRAME(color_search, name, value) \
958 static int set_frame_##name(PaletteUseContext *s, AVFrame *out, AVFrame *in, \
959 int x_start, int y_start, int w, int h) \
961 return set_frame(s, out, in, x_start, y_start, w, h, value, color_search); \
964 #define DEFINE_SET_FRAME_COLOR_SEARCH(color_search, color_search_macro) \
965 DEFINE_SET_FRAME(color_search_macro, color_search##_##none, DITHERING_NONE) \
966 DEFINE_SET_FRAME(color_search_macro, color_search##_##bayer, DITHERING_BAYER) \
967 DEFINE_SET_FRAME(color_search_macro, color_search##_##heckbert, DITHERING_HECKBERT) \
968 DEFINE_SET_FRAME(color_search_macro, color_search##_##floyd_steinberg, DITHERING_FLOYD_STEINBERG) \
969 DEFINE_SET_FRAME(color_search_macro, color_search##_##sierra2, DITHERING_SIERRA2) \
970 DEFINE_SET_FRAME(color_search_macro, color_search##_##sierra2_4a, DITHERING_SIERRA2_4A) \
976 #define DITHERING_ENTRIES(color_search) { \
977 set_frame_##color_search##_none, \
978 set_frame_##color_search##_bayer, \
979 set_frame_##color_search##_heckbert, \
980 set_frame_##color_search##_floyd_steinberg, \
981 set_frame_##color_search##_sierra2, \
982 set_frame_##color_search##_sierra2_4a, \
993 const int q = p ^ (p >> 3);
994 return (p & 4) >> 2 | (q & 4) >> 1 \
995 | (p & 2) << 1 | (q & 2) << 2 \
996 | (p & 1) << 4 | (q & 1) << 5;
1041 .needs_writable = 1,
1062 .
name =
"paletteuse",
1068 .
inputs = paletteuse_inputs,
1069 .
outputs = paletteuse_outputs,
1070 .priv_class = &paletteuse_class,
int(* cmp_func)(const void *, const void *)
AVFILTER_DEFINE_CLASS(paletteuse)
void av_bprintf(AVBPrint *buf, const char *fmt,...)
static int shift(int a, int b)
static void load_palette(PaletteUseContext *s, const AVFrame *palette_frame)
static AVFrame * load_apply_palette(AVFilterContext *ctx, AVFrame *main, const AVFrame *second)
This structure describes decoded (raw) audio or video data.
ptrdiff_t const GLvoid * data
static const AVFilterPad outputs[]
Main libavfilter public API header.
static AVFrame * apply_palette(AVFilterLink *inlink, AVFrame *in)
int h
agreed upon image height
static int query_formats(AVFilterContext *ctx)
FILE * av_fopen_utf8(const char *path, const char *mode)
Open a file using a UTF-8 filename.
static const AVFilterPad paletteuse_outputs[]
static int debug_accuracy(const struct color_node *node, const uint32_t *palette, const enum color_search_method search_method)
void * av_dynarray2_add(void **tab_ptr, int *nb_ptr, size_t elem_size, const uint8_t *elem_data)
Add an element of size elem_size to a dynamic array.
#define FF_ARRAY_ELEMS(a)
static av_always_inline int dither_color(uint32_t px, int er, int eg, int eb, int scale, int shift)
AVFrame * ff_get_video_buffer(AVFilterLink *link, int w, int h)
Request a picture buffer with a specific set of permissions.
static av_cold int init(AVFilterContext *ctx)
int av_bprint_finalize(AVBPrint *buf, char **ret_str)
Finalize a print buffer.
const char * name
Pad name.
AVFilterLink ** inputs
array of pointers to input links
int(* set_frame_func)(struct PaletteUseContext *s, AVFrame *out, AVFrame *in, int x_start, int y_start, int width, int height)
AVFilter ff_vf_paletteuse
8 bit with AV_PIX_FMT_RGB32 palette
static const uint32_t color[16+AV_CLASS_CATEGORY_NB]
static av_cold int end(AVCodecContext *avctx)
static const cmp_func cmp_funcs[]
static int config_input_palette(AVFilterLink *inlink)
uint32_t palette[AVPALETTE_COUNT]
static const AVOption paletteuse_options[]
A filter pad used for either input or output.
#define DEFINE_SET_FRAME_COLOR_SEARCH(color_search, color_search_macro)
static int disp_tree(const struct color_node *node, const char *fname)
A link between two filters.
int width
width and height of the video frame
#define AV_LOG_ERROR
Something went wrong and cannot losslessly be recovered.
#define AV_BPRINT_SIZE_UNLIMITED
static void colormap_nearest_node(const struct color_node *map, const int node_pos, const uint8_t *target, struct nearest_color *nearest)
void av_frame_free(AVFrame **frame)
Free the frame and any dynamically allocated objects in it, e.g.
#define NULL_IF_CONFIG_SMALL(x)
Return NULL if CONFIG_SMALL is true, otherwise the argument without modification. ...
void av_bprint_init(AVBPrint *buf, unsigned size_init, unsigned size_max)
static const uint8_t dither[8][8]
void * priv
private data for use by the filter
static const set_frame_func set_frame_lut[NB_COLOR_SEARCHES][NB_DITHERING]
AVRational time_base
Define the time base used by the PTS of the frames/samples which will pass through this link...
AVFilterFormats * in_formats
Lists of formats and channel layouts supported by the input and output filters respectively.
static int request_frame(AVFilterLink *outlink)
static int dither_value(int p)
static char * split(char *message, char delim)
int w
agreed upon image width
static av_always_inline uint8_t colormap_nearest_iterative(const struct color_node *root, const uint8_t *target)
static void disp_node(AVBPrint *buf, const struct color_node *map, int parent_id, int node_id, int depth)
static void set_processing_window(enum diff_mode diff_mode, const AVFrame *prv_src, const AVFrame *cur_src, const AVFrame *prv_dst, AVFrame *cur_dst, int *xp, int *yp, int *wp, int *hp)
#define av_err2str(errnum)
Convenience macro, the return value should be used only directly in function arguments but never stan...
static av_always_inline int diff(const uint8_t *c1, const uint8_t *c2)
AVFilterContext * src
source filter
AVFrame *(* process)(AVFilterContext *ctx, AVFrame *main, const AVFrame *second)
AVFrame * av_frame_clone(const AVFrame *src)
Create a new frame that references the same data as src.
static av_always_inline int get_dst_color_err(struct cache_node *cache, uint32_t c, const struct color_node *map, const uint32_t *palette, int *er, int *eg, int *eb, const enum color_search_method search_method)
static int config_output(AVFilterLink *outlink)
#define av_dlog(pctx,...)
av_dlog macros
static struct AVHashContext * hash
#define AV_LOG_INFO
Standard information.
static int cmp_pal_entry(const void *a, const void *b)
int linesize[AV_NUM_DATA_POINTERS]
For video, size in bytes of each picture line.
static int filter_frame(AVFilterLink *inlink, AVFrame *in)
uint8_t pi<< 24) CONV_FUNC_GROUP(AV_SAMPLE_FMT_FLT, float, AV_SAMPLE_FMT_U8, uint8_t,(*(constuint8_t *) pi-0x80)*(1.0f/(1<< 7))) CONV_FUNC_GROUP(AV_SAMPLE_FMT_DBL, double, AV_SAMPLE_FMT_U8, uint8_t,(*(constuint8_t *) pi-0x80)*(1.0/(1<< 7))) CONV_FUNC_GROUP(AV_SAMPLE_FMT_U8, uint8_t, AV_SAMPLE_FMT_S16, int16_t,(*(constint16_t *) pi >>8)+0x80) CONV_FUNC_GROUP(AV_SAMPLE_FMT_FLT, float, AV_SAMPLE_FMT_S16, int16_t,*(constint16_t *) pi *(1.0f/(1<< 15))) CONV_FUNC_GROUP(AV_SAMPLE_FMT_DBL, double, AV_SAMPLE_FMT_S16, int16_t,*(constint16_t *) pi *(1.0/(1<< 15))) CONV_FUNC_GROUP(AV_SAMPLE_FMT_U8, uint8_t, AV_SAMPLE_FMT_S32, int32_t,(*(constint32_t *) pi >>24)+0x80) CONV_FUNC_GROUP(AV_SAMPLE_FMT_FLT, float, AV_SAMPLE_FMT_S32, int32_t,*(constint32_t *) pi *(1.0f/(1U<< 31))) CONV_FUNC_GROUP(AV_SAMPLE_FMT_DBL, double, AV_SAMPLE_FMT_S32, int32_t,*(constint32_t *) pi *(1.0/(1U<< 31))) CONV_FUNC_GROUP(AV_SAMPLE_FMT_U8, uint8_t, AV_SAMPLE_FMT_FLT, float, av_clip_uint8(lrintf(*(constfloat *) pi *(1<< 7))+0x80)) CONV_FUNC_GROUP(AV_SAMPLE_FMT_S16, int16_t, AV_SAMPLE_FMT_FLT, float, av_clip_int16(lrintf(*(constfloat *) pi *(1<< 15)))) CONV_FUNC_GROUP(AV_SAMPLE_FMT_S32, int32_t, AV_SAMPLE_FMT_FLT, float, av_clipl_int32(llrintf(*(constfloat *) pi *(1U<< 31)))) CONV_FUNC_GROUP(AV_SAMPLE_FMT_U8, uint8_t, AV_SAMPLE_FMT_DBL, double, av_clip_uint8(lrint(*(constdouble *) pi *(1<< 7))+0x80)) CONV_FUNC_GROUP(AV_SAMPLE_FMT_S16, int16_t, AV_SAMPLE_FMT_DBL, double, av_clip_int16(lrint(*(constdouble *) pi *(1<< 15)))) CONV_FUNC_GROUP(AV_SAMPLE_FMT_S32, int32_t, AV_SAMPLE_FMT_DBL, double, av_clipl_int32(llrint(*(constdouble *) pi *(1U<< 31))))#defineSET_CONV_FUNC_GROUP(ofmt, ifmt) staticvoidset_generic_function(AudioConvert *ac){}voidff_audio_convert_free(AudioConvert **ac){if(!*ac) return;ff_dither_free(&(*ac) ->dc);av_freep(ac);}AudioConvert *ff_audio_convert_alloc(AVAudioResampleContext *avr, enumAVSampleFormatout_fmt, enumAVSampleFormatin_fmt, intchannels, intsample_rate, intapply_map){AudioConvert *ac;intin_planar, out_planar;ac=av_mallocz(sizeof(*ac));if(!ac) returnNULL;ac->avr=avr;ac->out_fmt=out_fmt;ac->in_fmt=in_fmt;ac->channels=channels;ac->apply_map=apply_map;if(avr->dither_method!=AV_RESAMPLE_DITHER_NONE &&av_get_packed_sample_fmt(out_fmt)==AV_SAMPLE_FMT_S16 &&av_get_bytes_per_sample(in_fmt)>2){ac->dc=ff_dither_alloc(avr, out_fmt, in_fmt, channels, sample_rate, apply_map);if(!ac->dc){av_free(ac);returnNULL;}returnac;}in_planar=ff_sample_fmt_is_planar(in_fmt, channels);out_planar=ff_sample_fmt_is_planar(out_fmt, channels);if(in_planar==out_planar){ac->func_type=CONV_FUNC_TYPE_FLAT;ac->planes=in_planar?ac->channels:1;}elseif(in_planar) ac->func_type=CONV_FUNC_TYPE_INTERLEAVE;elseac->func_type=CONV_FUNC_TYPE_DEINTERLEAVE;set_generic_function(ac);if(ARCH_AARCH64) ff_audio_convert_init_aarch64(ac);if(ARCH_ARM) ff_audio_convert_init_arm(ac);if(ARCH_X86) ff_audio_convert_init_x86(ac);returnac;}intff_audio_convert(AudioConvert *ac, AudioData *out, AudioData *in){intuse_generic=1;intlen=in->nb_samples;intp;if(ac->dc){av_log(ac->avr, AV_LOG_TRACE,"%dsamples-audio_convert:%sto%s(dithered)\n", len, av_get_sample_fmt_name(ac->in_fmt), av_get_sample_fmt_name(ac->out_fmt));returnff_convert_dither(ac-> in
int repeatlast
repeat last second frame
BYTE int const BYTE int int int height
Describe the class of an AVClass context structure.
#define DITHERING_ENTRIES(color_search)
static const AVFilterPad inputs[]
static int get_next_color(const uint8_t *color_used, const uint32_t *palette, int *component, const struct color_rect *box)
const char * name
Filter name.
static void load_colormap(PaletteUseContext *s)
static int colormap_insert(struct color_node *map, uint8_t *color_used, int *nb_used, const uint32_t *palette, const struct color_rect *box)
#define DECLARE_CMP_FUNC(name, pos)
AVFilterLink ** outputs
array of pointers to output links
#define COLORMAP_NEAREST(search, palette, root, target)
int av_frame_make_writable(AVFrame *frame)
Ensure that the frame data is writable, avoiding data copy if possible.
uint8_t * data[AV_NUM_DATA_POINTERS]
pointer to the picture/channel planes.
FFDualInputContext dinput
struct cache_node cache[CACHE_SIZE]
struct cached_color * entries
static av_always_inline uint8_t colormap_nearest_recursive(const struct color_node *node, const uint8_t *rgb)
static av_always_inline int set_frame(PaletteUseContext *s, AVFrame *out, AVFrame *in, int x_start, int y_start, int w, int h, enum dithering_mode dither, const enum color_search_method search_method)
static av_cold void uninit(AVFilterContext *ctx)
static av_always_inline uint8_t colormap_nearest_bruteforce(const uint32_t *palette, const uint8_t *rgb)
AVFilterContext * dst
dest filter
static av_always_inline int color_get(struct cache_node *cache, uint32_t color, uint8_t r, uint8_t g, uint8_t b, const struct color_node *map, const uint32_t *palette, const enum color_search_method search_method)
Check if the requested color is in the cache already.
uint8_t pi<< 24) CONV_FUNC_GROUP(AV_SAMPLE_FMT_FLT, float, AV_SAMPLE_FMT_U8, uint8_t,(*(constuint8_t *) pi-0x80)*(1.0f/(1<< 7))) CONV_FUNC_GROUP(AV_SAMPLE_FMT_DBL, double, AV_SAMPLE_FMT_U8, uint8_t,(*(constuint8_t *) pi-0x80)*(1.0/(1<< 7))) CONV_FUNC_GROUP(AV_SAMPLE_FMT_U8, uint8_t, AV_SAMPLE_FMT_S16, int16_t,(*(constint16_t *) pi >>8)+0x80) CONV_FUNC_GROUP(AV_SAMPLE_FMT_FLT, float, AV_SAMPLE_FMT_S16, int16_t,*(constint16_t *) pi *(1.0f/(1<< 15))) CONV_FUNC_GROUP(AV_SAMPLE_FMT_DBL, double, AV_SAMPLE_FMT_S16, int16_t,*(constint16_t *) pi *(1.0/(1<< 15))) CONV_FUNC_GROUP(AV_SAMPLE_FMT_U8, uint8_t, AV_SAMPLE_FMT_S32, int32_t,(*(constint32_t *) pi >>24)+0x80) CONV_FUNC_GROUP(AV_SAMPLE_FMT_FLT, float, AV_SAMPLE_FMT_S32, int32_t,*(constint32_t *) pi *(1.0f/(1U<< 31))) CONV_FUNC_GROUP(AV_SAMPLE_FMT_DBL, double, AV_SAMPLE_FMT_S32, int32_t,*(constint32_t *) pi *(1.0/(1U<< 31))) CONV_FUNC_GROUP(AV_SAMPLE_FMT_U8, uint8_t, AV_SAMPLE_FMT_FLT, float, av_clip_uint8(lrintf(*(constfloat *) pi *(1<< 7))+0x80)) CONV_FUNC_GROUP(AV_SAMPLE_FMT_S16, int16_t, AV_SAMPLE_FMT_FLT, float, av_clip_int16(lrintf(*(constfloat *) pi *(1<< 15)))) CONV_FUNC_GROUP(AV_SAMPLE_FMT_S32, int32_t, AV_SAMPLE_FMT_FLT, float, av_clipl_int32(llrintf(*(constfloat *) pi *(1U<< 31)))) CONV_FUNC_GROUP(AV_SAMPLE_FMT_U8, uint8_t, AV_SAMPLE_FMT_DBL, double, av_clip_uint8(lrint(*(constdouble *) pi *(1<< 7))+0x80)) CONV_FUNC_GROUP(AV_SAMPLE_FMT_S16, int16_t, AV_SAMPLE_FMT_DBL, double, av_clip_int16(lrint(*(constdouble *) pi *(1<< 15)))) CONV_FUNC_GROUP(AV_SAMPLE_FMT_S32, int32_t, AV_SAMPLE_FMT_DBL, double, av_clipl_int32(llrint(*(constdouble *) pi *(1U<< 31))))#defineSET_CONV_FUNC_GROUP(ofmt, ifmt) staticvoidset_generic_function(AudioConvert *ac){}voidff_audio_convert_free(AudioConvert **ac){if(!*ac) return;ff_dither_free(&(*ac) ->dc);av_freep(ac);}AudioConvert *ff_audio_convert_alloc(AVAudioResampleContext *avr, enumAVSampleFormatout_fmt, enumAVSampleFormatin_fmt, intchannels, intsample_rate, intapply_map){AudioConvert *ac;intin_planar, out_planar;ac=av_mallocz(sizeof(*ac));if(!ac) returnNULL;ac->avr=avr;ac->out_fmt=out_fmt;ac->in_fmt=in_fmt;ac->channels=channels;ac->apply_map=apply_map;if(avr->dither_method!=AV_RESAMPLE_DITHER_NONE &&av_get_packed_sample_fmt(out_fmt)==AV_SAMPLE_FMT_S16 &&av_get_bytes_per_sample(in_fmt)>2){ac->dc=ff_dither_alloc(avr, out_fmt, in_fmt, channels, sample_rate, apply_map);if(!ac->dc){av_free(ac);returnNULL;}returnac;}in_planar=ff_sample_fmt_is_planar(in_fmt, channels);out_planar=ff_sample_fmt_is_planar(out_fmt, channels);if(in_planar==out_planar){ac->func_type=CONV_FUNC_TYPE_FLAT;ac->planes=in_planar?ac->channels:1;}elseif(in_planar) ac->func_type=CONV_FUNC_TYPE_INTERLEAVE;elseac->func_type=CONV_FUNC_TYPE_DEINTERLEAVE;set_generic_function(ac);if(ARCH_AARCH64) ff_audio_convert_init_aarch64(ac);if(ARCH_ARM) ff_audio_convert_init_arm(ac);if(ARCH_X86) ff_audio_convert_init_x86(ac);returnac;}intff_audio_convert(AudioConvert *ac, AudioData *out, AudioData *in){intuse_generic=1;intlen=in->nb_samples;intp;if(ac->dc){av_log(ac->avr, AV_LOG_TRACE,"%dsamples-audio_convert:%sto%s(dithered)\n", len, av_get_sample_fmt_name(ac->in_fmt), av_get_sample_fmt_name(ac->out_fmt));returnff_convert_dither(ac-> out
static const AVFilterPad paletteuse_inputs[]
int main(int argc, char **argv)
struct color_node map[AVPALETTE_COUNT]
AVPixelFormat
Pixel format.
#define AV_QSORT(p, num, type, cmp)
Quicksort This sort is fast, and fully inplace but not stable and it is possible to construct input t...
int64_t frame_count
Number of past frames sent through the link.
int av_frame_copy_props(AVFrame *dst, const AVFrame *src)
Copy only "metadata" fields from src to dst.
static void debug_mean_error(PaletteUseContext *s, const AVFrame *in1, const AVFrame *in2, int frame_count)
AVFilterFormats * out_formats