| Line | Branch | Exec | Source |
|---|---|---|---|
| 1 | /* | ||
| 2 | * Copyright (c) 2000 Fabrice Bellard | ||
| 3 | * Copyright (c) 2002 Francois Revol | ||
| 4 | * Copyright (c) 2006 Baptiste Coudurier | ||
| 5 | * Copyright (c) 2018 Bjorn Roche | ||
| 6 | * Copyright (c) 2018 Paul B Mahol | ||
| 7 | * | ||
| 8 | * first version by Francois Revol <revol@free.fr> | ||
| 9 | * | ||
| 10 | * This file is part of FFmpeg. | ||
| 11 | * | ||
| 12 | * FFmpeg is free software; you can redistribute it and/or | ||
| 13 | * modify it under the terms of the GNU Lesser General Public | ||
| 14 | * License as published by the Free Software Foundation; either | ||
| 15 | * version 2.1 of the License, or (at your option) any later version. | ||
| 16 | * | ||
| 17 | * FFmpeg is distributed in the hope that it will be useful, | ||
| 18 | * but WITHOUT ANY WARRANTY; without even the implied warranty of | ||
| 19 | * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU | ||
| 20 | * Lesser General Public License for more details. | ||
| 21 | * | ||
| 22 | * You should have received a copy of the GNU Lesser General Public | ||
| 23 | * License along with FFmpeg; if not, write to the Free Software | ||
| 24 | * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA | ||
| 25 | */ | ||
| 26 | |||
| 27 | /** | ||
| 28 | * @file | ||
| 29 | * GIF encoder | ||
| 30 | * @see http://www.w3.org/Graphics/GIF/spec-gif89a.txt | ||
| 31 | */ | ||
| 32 | |||
| 33 | #include "libavutil/imgutils_internal.h" | ||
| 34 | #include "libavutil/mem.h" | ||
| 35 | #include "libavutil/opt.h" | ||
| 36 | #include "avcodec.h" | ||
| 37 | #include "bytestream.h" | ||
| 38 | #include "codec_internal.h" | ||
| 39 | #include "encode.h" | ||
| 40 | #include "lzw.h" | ||
| 41 | #include "gif.h" | ||
| 42 | |||
| 43 | #define DEFAULT_TRANSPARENCY_INDEX 0x1f | ||
| 44 | |||
| 45 | typedef struct GIFContext { | ||
| 46 | const AVClass *class; | ||
| 47 | LZWState *lzw; | ||
| 48 | uint8_t *buf; | ||
| 49 | uint8_t *shrunk_buf; | ||
| 50 | int buf_size; | ||
| 51 | AVFrame *last_frame; | ||
| 52 | int flags; | ||
| 53 | int image; | ||
| 54 | int use_global_palette; | ||
| 55 | uint32_t palette[AVPALETTE_COUNT]; ///< local reference palette for !pal8 | ||
| 56 | int palette_loaded; | ||
| 57 | int transparent_index; | ||
| 58 | uint8_t *tmpl; ///< temporary line buffer | ||
| 59 | } GIFContext; | ||
| 60 | |||
| 61 | enum { | ||
| 62 | GF_OFFSETTING = 1<<0, | ||
| 63 | GF_TRANSDIFF = 1<<1, | ||
| 64 | }; | ||
| 65 | |||
| 66 | 1 | static void shrink_palette(const uint32_t *src, uint8_t *map, | |
| 67 | uint32_t *dst, size_t *palette_count) | ||
| 68 | { | ||
| 69 | 1 | size_t colors_seen = 0; | |
| 70 | |||
| 71 |
2/2✓ Branch 0 taken 256 times.
✓ Branch 1 taken 1 times.
|
257 | for (size_t i = 0; i < AVPALETTE_COUNT; i++) { |
| 72 | 256 | int seen = 0; | |
| 73 |
2/2✓ Branch 0 taken 32640 times.
✓ Branch 1 taken 256 times.
|
32896 | for (size_t c = 0; c < colors_seen; c++) { |
| 74 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 32640 times.
|
32640 | if (src[i] == dst[c]) { |
| 75 | ✗ | seen = 1; | |
| 76 | ✗ | break; | |
| 77 | } | ||
| 78 | } | ||
| 79 |
1/2✓ Branch 0 taken 256 times.
✗ Branch 1 not taken.
|
256 | if (!seen) { |
| 80 | 256 | dst[colors_seen] = src[i]; | |
| 81 | 256 | map[i] = colors_seen; | |
| 82 | 256 | colors_seen++; | |
| 83 | } | ||
| 84 | } | ||
| 85 | |||
| 86 | 1 | *palette_count = colors_seen; | |
| 87 | 1 | } | |
| 88 | |||
| 89 | 1 | static void remap_frame_to_palette(const uint8_t *src, int src_linesize, | |
| 90 | uint8_t *dst, int dst_linesize, | ||
| 91 | int w, int h, uint8_t *map) | ||
| 92 | { | ||
| 93 |
2/2✓ Branch 0 taken 217 times.
✓ Branch 1 taken 1 times.
|
218 | for (int i = 0; i < h; i++) |
| 94 |
2/2✓ Branch 0 taken 47089 times.
✓ Branch 1 taken 217 times.
|
47306 | for (int j = 0; j < w; j++) |
| 95 | 47089 | dst[i * dst_linesize + j] = map[src[i * src_linesize + j]]; | |
| 96 | 1 | } | |
| 97 | |||
| 98 | 1063 | static int is_image_translucent(AVCodecContext *avctx, | |
| 99 | const uint8_t *buf, const int linesize) | ||
| 100 | { | ||
| 101 | 1063 | GIFContext *s = avctx->priv_data; | |
| 102 | 1063 | int trans = s->transparent_index; | |
| 103 | |||
| 104 |
1/2✓ Branch 0 taken 1063 times.
✗ Branch 1 not taken.
|
1063 | if (trans < 0) |
| 105 | 1063 | return 0; | |
| 106 | |||
| 107 | ✗ | for (int y = 0; y < avctx->height; y++) { | |
| 108 | ✗ | for (int x = 0; x < avctx->width; x++) { | |
| 109 | ✗ | if (buf[x] == trans) { | |
| 110 | ✗ | return 1; | |
| 111 | } | ||
| 112 | } | ||
| 113 | ✗ | buf += linesize; | |
| 114 | } | ||
| 115 | |||
| 116 | ✗ | return 0; | |
| 117 | } | ||
| 118 | |||
| 119 | 39 | static int get_palette_transparency_index(const uint32_t *palette) | |
| 120 | { | ||
| 121 | 39 | int transparent_color_index = -1; | |
| 122 | 39 | unsigned i, smallest_alpha = 0xff; | |
| 123 | |||
| 124 |
2/2✓ Branch 0 taken 30 times.
✓ Branch 1 taken 9 times.
|
39 | if (!palette) |
| 125 | 30 | return -1; | |
| 126 | |||
| 127 |
2/2✓ Branch 0 taken 2304 times.
✓ Branch 1 taken 9 times.
|
2313 | for (i = 0; i < AVPALETTE_COUNT; i++) { |
| 128 | 2304 | const uint32_t v = palette[i]; | |
| 129 |
2/2✓ Branch 0 taken 2 times.
✓ Branch 1 taken 2302 times.
|
2304 | if (v >> 24 < smallest_alpha) { |
| 130 | 2 | smallest_alpha = v >> 24; | |
| 131 | 2 | transparent_color_index = i; | |
| 132 | } | ||
| 133 | } | ||
| 134 |
2/2✓ Branch 0 taken 2 times.
✓ Branch 1 taken 7 times.
|
9 | return smallest_alpha < 128 ? transparent_color_index : -1; |
| 135 | } | ||
| 136 | |||
| 137 | 1056 | static int pick_palette_entry(const uint8_t *buf, int linesize, int w, int h) | |
| 138 | { | ||
| 139 | 1056 | int histogram[AVPALETTE_COUNT] = {0}; | |
| 140 | int x, y, i; | ||
| 141 | |||
| 142 |
2/2✓ Branch 0 taken 32487 times.
✓ Branch 1 taken 1056 times.
|
33543 | for (y = 0; y < h; y++) { |
| 143 |
2/2✓ Branch 0 taken 3157775 times.
✓ Branch 1 taken 32487 times.
|
3190262 | for (x = 0; x < w; x++) |
| 144 | 3157775 | histogram[buf[x]]++; | |
| 145 | 32487 | buf += linesize; | |
| 146 | } | ||
| 147 |
2/2✓ Branch 0 taken 7288 times.
✓ Branch 1 taken 24 times.
|
7312 | for (i = 0; i < FF_ARRAY_ELEMS(histogram); i++) |
| 148 |
2/2✓ Branch 0 taken 1032 times.
✓ Branch 1 taken 6256 times.
|
7288 | if (!histogram[i]) |
| 149 | 1032 | return i; | |
| 150 | 24 | return -1; | |
| 151 | } | ||
| 152 | |||
| 153 | ✗ | static void gif_crop_translucent(AVCodecContext *avctx, | |
| 154 | const uint8_t *buf, const int linesize, | ||
| 155 | int *width, int *height, | ||
| 156 | int *x_start, int *y_start) | ||
| 157 | { | ||
| 158 | ✗ | GIFContext *s = avctx->priv_data; | |
| 159 | ✗ | int trans = s->transparent_index; | |
| 160 | |||
| 161 | /* Crop image */ | ||
| 162 | ✗ | if ((s->flags & GF_OFFSETTING) && trans >= 0) { | |
| 163 | ✗ | const int w = avctx->width; | |
| 164 | ✗ | const int h = avctx->height; | |
| 165 | ✗ | int x_end = w - 1, | |
| 166 | ✗ | y_end = h - 1; | |
| 167 | |||
| 168 | // crop top | ||
| 169 | ✗ | while (*y_start < y_end) { | |
| 170 | ✗ | int is_trans = 1; | |
| 171 | ✗ | for (int i = 0; i < w; i++) { | |
| 172 | ✗ | if (buf[linesize * *y_start + i] != trans) { | |
| 173 | ✗ | is_trans = 0; | |
| 174 | ✗ | break; | |
| 175 | } | ||
| 176 | } | ||
| 177 | |||
| 178 | ✗ | if (!is_trans) | |
| 179 | ✗ | break; | |
| 180 | ✗ | (*y_start)++; | |
| 181 | } | ||
| 182 | |||
| 183 | // crop bottom | ||
| 184 | ✗ | while (y_end > *y_start) { | |
| 185 | ✗ | int is_trans = 1; | |
| 186 | ✗ | for (int i = 0; i < w; i++) { | |
| 187 | ✗ | if (buf[linesize * y_end + i] != trans) { | |
| 188 | ✗ | is_trans = 0; | |
| 189 | ✗ | break; | |
| 190 | } | ||
| 191 | } | ||
| 192 | ✗ | if (!is_trans) | |
| 193 | ✗ | break; | |
| 194 | ✗ | y_end--; | |
| 195 | } | ||
| 196 | |||
| 197 | // crop left | ||
| 198 | ✗ | while (*x_start < x_end) { | |
| 199 | ✗ | int is_trans = 1; | |
| 200 | ✗ | for (int i = *y_start; i < y_end; i++) { | |
| 201 | ✗ | if (buf[linesize * i + *x_start] != trans) { | |
| 202 | ✗ | is_trans = 0; | |
| 203 | ✗ | break; | |
| 204 | } | ||
| 205 | } | ||
| 206 | ✗ | if (!is_trans) | |
| 207 | ✗ | break; | |
| 208 | ✗ | (*x_start)++; | |
| 209 | } | ||
| 210 | |||
| 211 | // crop right | ||
| 212 | ✗ | while (x_end > *x_start) { | |
| 213 | ✗ | int is_trans = 1; | |
| 214 | ✗ | for (int i = *y_start; i < y_end; i++) { | |
| 215 | ✗ | if (buf[linesize * i + x_end] != trans) { | |
| 216 | ✗ | is_trans = 0; | |
| 217 | ✗ | break; | |
| 218 | } | ||
| 219 | } | ||
| 220 | ✗ | if (!is_trans) | |
| 221 | ✗ | break; | |
| 222 | ✗ | x_end--; | |
| 223 | } | ||
| 224 | |||
| 225 | ✗ | *height = y_end + 1 - *y_start; | |
| 226 | ✗ | *width = x_end + 1 - *x_start; | |
| 227 | ✗ | av_log(avctx, AV_LOG_DEBUG,"%dx%d image at pos (%d;%d) [area:%dx%d]\n", | |
| 228 | *width, *height, *x_start, *y_start, avctx->width, avctx->height); | ||
| 229 | } | ||
| 230 | ✗ | } | |
| 231 | |||
| 232 | 1063 | static void gif_crop_opaque(AVCodecContext *avctx, | |
| 233 | const uint32_t *palette, | ||
| 234 | const uint8_t *buf, const int linesize, | ||
| 235 | int *width, int *height, int *x_start, int *y_start) | ||
| 236 | { | ||
| 237 | 1063 | GIFContext *s = avctx->priv_data; | |
| 238 | |||
| 239 | /* Crop image */ | ||
| 240 |
4/6✓ Branch 0 taken 1063 times.
✗ Branch 1 not taken.
✓ Branch 2 taken 1056 times.
✓ Branch 3 taken 7 times.
✓ Branch 4 taken 1056 times.
✗ Branch 5 not taken.
|
1063 | if ((s->flags & GF_OFFSETTING) && s->last_frame && !palette) { |
| 241 | 1056 | const uint8_t *ref = s->last_frame->data[0]; | |
| 242 | 1056 | const int ref_linesize = s->last_frame->linesize[0]; | |
| 243 | 1056 | int x_end = avctx->width - 1, | |
| 244 | 1056 | y_end = avctx->height - 1; | |
| 245 | |||
| 246 | /* skip common lines */ | ||
| 247 |
1/2✓ Branch 0 taken 100704 times.
✗ Branch 1 not taken.
|
100704 | while (*y_start < y_end) { |
| 248 |
2/2✓ Branch 0 taken 1056 times.
✓ Branch 1 taken 99648 times.
|
100704 | if (memcmp(ref + *y_start*ref_linesize, buf + *y_start*linesize, *width)) |
| 249 | 1056 | break; | |
| 250 | 99648 | (*y_start)++; | |
| 251 | } | ||
| 252 |
1/2✓ Branch 0 taken 99777 times.
✗ Branch 1 not taken.
|
99777 | while (y_end > *y_start) { |
| 253 |
2/2✓ Branch 0 taken 1056 times.
✓ Branch 1 taken 98721 times.
|
99777 | if (memcmp(ref + y_end*ref_linesize, buf + y_end*linesize, *width)) |
| 254 | 1056 | break; | |
| 255 | 98721 | y_end--; | |
| 256 | } | ||
| 257 | 1056 | *height = y_end + 1 - *y_start; | |
| 258 | |||
| 259 | /* skip common columns */ | ||
| 260 |
1/2✓ Branch 0 taken 96923 times.
✗ Branch 1 not taken.
|
96923 | while (*x_start < x_end) { |
| 261 | 96923 | int same_column = 1; | |
| 262 |
2/2✓ Branch 0 taken 2402634 times.
✓ Branch 1 taken 95867 times.
|
2498501 | for (int y = *y_start; y <= y_end; y++) { |
| 263 |
2/2✓ Branch 0 taken 1056 times.
✓ Branch 1 taken 2401578 times.
|
2402634 | if (ref[y*ref_linesize + *x_start] != buf[y*linesize + *x_start]) { |
| 264 | 1056 | same_column = 0; | |
| 265 | 1056 | break; | |
| 266 | } | ||
| 267 | } | ||
| 268 |
2/2✓ Branch 0 taken 1056 times.
✓ Branch 1 taken 95867 times.
|
96923 | if (!same_column) |
| 269 | 1056 | break; | |
| 270 | 95867 | (*x_start)++; | |
| 271 | } | ||
| 272 |
1/2✓ Branch 0 taken 104460 times.
✗ Branch 1 not taken.
|
104460 | while (x_end > *x_start) { |
| 273 | 104460 | int same_column = 1; | |
| 274 |
2/2✓ Branch 0 taken 2442709 times.
✓ Branch 1 taken 103404 times.
|
2546113 | for (int y = *y_start; y <= y_end; y++) { |
| 275 |
2/2✓ Branch 0 taken 1056 times.
✓ Branch 1 taken 2441653 times.
|
2442709 | if (ref[y*ref_linesize + x_end] != buf[y*linesize + x_end]) { |
| 276 | 1056 | same_column = 0; | |
| 277 | 1056 | break; | |
| 278 | } | ||
| 279 | } | ||
| 280 |
2/2✓ Branch 0 taken 1056 times.
✓ Branch 1 taken 103404 times.
|
104460 | if (!same_column) |
| 281 | 1056 | break; | |
| 282 | 103404 | x_end--; | |
| 283 | } | ||
| 284 | 1056 | *width = x_end + 1 - *x_start; | |
| 285 | |||
| 286 | 1056 | av_log(avctx, AV_LOG_DEBUG,"%dx%d image at pos (%d;%d) [area:%dx%d]\n", | |
| 287 | *width, *height, *x_start, *y_start, avctx->width, avctx->height); | ||
| 288 | } | ||
| 289 | 1063 | } | |
| 290 | |||
| 291 | 1063 | static int gif_image_write_image(AVCodecContext *avctx, | |
| 292 | uint8_t **bytestream, uint8_t *end, | ||
| 293 | const uint32_t *palette, | ||
| 294 | const uint8_t *buf, const int linesize, | ||
| 295 | AVPacket *pkt) | ||
| 296 | { | ||
| 297 | 1063 | GIFContext *s = avctx->priv_data; | |
| 298 | 1063 | int disposal, len = 0, height = avctx->height, width = avctx->width, x, y; | |
| 299 | 1063 | int x_start = 0, y_start = 0, trans = s->transparent_index; | |
| 300 |
4/6✓ Branch 0 taken 1063 times.
✗ Branch 1 not taken.
✓ Branch 2 taken 1056 times.
✓ Branch 3 taken 7 times.
✓ Branch 4 taken 1056 times.
✗ Branch 5 not taken.
|
1063 | int bcid = -1, honor_transparency = (s->flags & GF_TRANSDIFF) && s->last_frame && !palette; |
| 301 | const uint8_t *ptr; | ||
| 302 | uint32_t shrunk_palette[AVPALETTE_COUNT]; | ||
| 303 | 1063 | uint8_t map[AVPALETTE_COUNT] = { 0 }; | |
| 304 | 1063 | size_t shrunk_palette_count = 0; | |
| 305 | |||
| 306 | /* | ||
| 307 | * We memset to 0xff instead of 0x00 so that the transparency detection | ||
| 308 | * doesn't pick anything after the palette entries as the transparency | ||
| 309 | * index, and because GIF89a requires us to always write a power-of-2 | ||
| 310 | * number of palette entries. | ||
| 311 | */ | ||
| 312 | 1063 | memset(shrunk_palette, 0xff, AVPALETTE_SIZE); | |
| 313 | |||
| 314 |
2/4✓ Branch 0 taken 1063 times.
✗ Branch 1 not taken.
✗ Branch 3 not taken.
✓ Branch 4 taken 1063 times.
|
1063 | if (!s->image && is_image_translucent(avctx, buf, linesize)) { |
| 315 | ✗ | gif_crop_translucent(avctx, buf, linesize, &width, &height, &x_start, &y_start); | |
| 316 | ✗ | honor_transparency = 0; | |
| 317 | ✗ | disposal = GCE_DISPOSAL_BACKGROUND; | |
| 318 | } else { | ||
| 319 | 1063 | gif_crop_opaque(avctx, palette, buf, linesize, &width, &height, &x_start, &y_start); | |
| 320 | 1063 | disposal = GCE_DISPOSAL_INPLACE; | |
| 321 | } | ||
| 322 | |||
| 323 |
3/4✓ Branch 0 taken 1063 times.
✗ Branch 1 not taken.
✓ Branch 2 taken 7 times.
✓ Branch 3 taken 1056 times.
|
1063 | if (s->image || !avctx->frame_num) { /* GIF header */ |
| 324 |
2/2✓ Branch 0 taken 6 times.
✓ Branch 1 taken 1 times.
|
7 | const uint32_t *global_palette = palette ? palette : s->palette; |
| 325 | 7 | const AVRational sar = avctx->sample_aspect_ratio; | |
| 326 | 7 | int64_t aspect = 0; | |
| 327 | |||
| 328 |
1/4✗ Branch 0 not taken.
✓ Branch 1 taken 7 times.
✗ Branch 2 not taken.
✗ Branch 3 not taken.
|
7 | if (sar.num > 0 && sar.den > 0) { |
| 329 | ✗ | aspect = sar.num * 64LL / sar.den - 15; | |
| 330 | ✗ | if (aspect < 0 || aspect > 255) | |
| 331 | ✗ | aspect = 0; | |
| 332 | } | ||
| 333 | |||
| 334 | 7 | bytestream_put_buffer(bytestream, gif89a_sig, sizeof(gif89a_sig)); | |
| 335 | 7 | bytestream_put_le16(bytestream, avctx->width); | |
| 336 | 7 | bytestream_put_le16(bytestream, avctx->height); | |
| 337 | |||
| 338 | 7 | bcid = get_palette_transparency_index(global_palette); | |
| 339 | |||
| 340 |
1/2✓ Branch 0 taken 7 times.
✗ Branch 1 not taken.
|
7 | bytestream_put_byte(bytestream, ((uint8_t) s->use_global_palette << 7) | 0x70 | (s->use_global_palette ? 7 : 0)); /* flags: global clut, 256 entries */ |
| 341 |
2/2✓ Branch 0 taken 2 times.
✓ Branch 1 taken 5 times.
|
7 | bytestream_put_byte(bytestream, bcid < 0 ? DEFAULT_TRANSPARENCY_INDEX : bcid); /* background color index */ |
| 342 | 7 | bytestream_put_byte(bytestream, aspect); | |
| 343 |
1/2✓ Branch 0 taken 7 times.
✗ Branch 1 not taken.
|
7 | if (s->use_global_palette) { |
| 344 |
2/2✓ Branch 0 taken 1792 times.
✓ Branch 1 taken 7 times.
|
1799 | for (int i = 0; i < 256; i++) { |
| 345 | 1792 | const uint32_t v = global_palette[i] & 0xffffff; | |
| 346 | 1792 | bytestream_put_be24(bytestream, v); | |
| 347 | } | ||
| 348 | } | ||
| 349 | } | ||
| 350 | |||
| 351 |
3/4✓ Branch 0 taken 1056 times.
✓ Branch 1 taken 7 times.
✓ Branch 2 taken 1056 times.
✗ Branch 3 not taken.
|
1063 | if (honor_transparency && trans < 0) { |
| 352 | 1056 | trans = pick_palette_entry(buf + y_start*linesize + x_start, | |
| 353 | linesize, width, height); | ||
| 354 |
2/2✓ Branch 0 taken 24 times.
✓ Branch 1 taken 1032 times.
|
1056 | if (trans < 0) // TODO, patch welcome |
| 355 | 24 | av_log(avctx, AV_LOG_DEBUG, "No available color, can not use transparency\n"); | |
| 356 | } | ||
| 357 | |||
| 358 |
2/2✓ Branch 0 taken 31 times.
✓ Branch 1 taken 1032 times.
|
1063 | if (trans < 0) |
| 359 | 31 | honor_transparency = 0; | |
| 360 | |||
| 361 |
3/4✓ Branch 0 taken 1062 times.
✓ Branch 1 taken 1 times.
✗ Branch 2 not taken.
✓ Branch 3 taken 1062 times.
|
1063 | if (palette || !s->use_global_palette) { |
| 362 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 1 times.
|
1 | const uint32_t *pal = palette ? palette : s->palette; |
| 363 | 1 | shrink_palette(pal, map, shrunk_palette, &shrunk_palette_count); | |
| 364 | } | ||
| 365 | |||
| 366 |
3/4✓ Branch 0 taken 31 times.
✓ Branch 1 taken 1032 times.
✓ Branch 2 taken 31 times.
✗ Branch 3 not taken.
|
1063 | bcid = honor_transparency || disposal == GCE_DISPOSAL_BACKGROUND ? trans : get_palette_transparency_index(palette); |
| 367 | |||
| 368 | /* graphic control extension */ | ||
| 369 | 1063 | bytestream_put_byte(bytestream, GIF_EXTENSION_INTRODUCER); | |
| 370 | 1063 | bytestream_put_byte(bytestream, GIF_GCE_EXT_LABEL); | |
| 371 | 1063 | bytestream_put_byte(bytestream, 0x04); /* block size */ | |
| 372 | 1063 | bytestream_put_byte(bytestream, disposal<<2 | (bcid >= 0)); | |
| 373 | 1063 | bytestream_put_le16(bytestream, 5); // default delay | |
| 374 |
3/4✓ Branch 0 taken 1032 times.
✓ Branch 1 taken 31 times.
✗ Branch 2 not taken.
✓ Branch 3 taken 1032 times.
|
1063 | bytestream_put_byte(bytestream, bcid < 0 ? DEFAULT_TRANSPARENCY_INDEX : (shrunk_palette_count ? map[bcid] : bcid)); |
| 375 | 1063 | bytestream_put_byte(bytestream, 0x00); | |
| 376 | |||
| 377 | /* image block */ | ||
| 378 | 1063 | bytestream_put_byte(bytestream, GIF_IMAGE_SEPARATOR); | |
| 379 | 1063 | bytestream_put_le16(bytestream, x_start); | |
| 380 | 1063 | bytestream_put_le16(bytestream, y_start); | |
| 381 | 1063 | bytestream_put_le16(bytestream, width); | |
| 382 | 1063 | bytestream_put_le16(bytestream, height); | |
| 383 | |||
| 384 |
3/4✓ Branch 0 taken 1062 times.
✓ Branch 1 taken 1 times.
✗ Branch 2 not taken.
✓ Branch 3 taken 1062 times.
|
1064 | if (palette || !s->use_global_palette) { |
| 385 | 1 | unsigned pow2_count = av_log2(shrunk_palette_count - 1); | |
| 386 | unsigned i; | ||
| 387 | |||
| 388 | 1 | bytestream_put_byte(bytestream, 1<<7 | pow2_count); /* flags */ | |
| 389 |
2/2✓ Branch 0 taken 256 times.
✓ Branch 1 taken 1 times.
|
257 | for (i = 0; i < 1 << (pow2_count + 1); i++) { |
| 390 | 256 | const uint32_t v = shrunk_palette[i]; | |
| 391 | 256 | bytestream_put_be24(bytestream, v); | |
| 392 | } | ||
| 393 | } else { | ||
| 394 | 1062 | bytestream_put_byte(bytestream, 0x00); /* flags */ | |
| 395 | } | ||
| 396 | |||
| 397 | 1063 | bytestream_put_byte(bytestream, 0x08); | |
| 398 | |||
| 399 | 1063 | ff_lzw_encode_init(s->lzw, s->buf, s->buf_size, | |
| 400 | 12, FF_LZW_GIF, 1); | ||
| 401 | |||
| 402 |
2/2✓ Branch 0 taken 1 times.
✓ Branch 1 taken 1062 times.
|
1063 | if (shrunk_palette_count) { |
| 403 |
1/2✓ Branch 0 taken 1 times.
✗ Branch 1 not taken.
|
1 | if (!s->shrunk_buf) { |
| 404 | 1 | s->shrunk_buf = av_malloc(avctx->height * linesize); | |
| 405 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 1 times.
|
1 | if (!s->shrunk_buf) { |
| 406 | ✗ | av_log(avctx, AV_LOG_ERROR, "Could not allocated remapped frame buffer.\n"); | |
| 407 | ✗ | return AVERROR(ENOMEM); | |
| 408 | } | ||
| 409 | } | ||
| 410 | 1 | remap_frame_to_palette(buf, linesize, s->shrunk_buf, linesize, avctx->width, avctx->height, map); | |
| 411 | 1 | ptr = s->shrunk_buf + y_start*linesize + x_start; | |
| 412 | } else { | ||
| 413 | 1062 | ptr = buf + y_start*linesize + x_start; | |
| 414 | } | ||
| 415 |
2/2✓ Branch 0 taken 1032 times.
✓ Branch 1 taken 31 times.
|
1063 | if (honor_transparency) { |
| 416 | 1032 | const int ref_linesize = s->last_frame->linesize[0]; | |
| 417 | 1032 | const uint8_t *ref = s->last_frame->data[0] + y_start*ref_linesize + x_start; | |
| 418 | |||
| 419 |
2/2✓ Branch 0 taken 25575 times.
✓ Branch 1 taken 1032 times.
|
26607 | for (y = 0; y < height; y++) { |
| 420 | 25575 | memcpy(s->tmpl, ptr, width); | |
| 421 |
2/2✓ Branch 0 taken 724751 times.
✓ Branch 1 taken 25575 times.
|
750326 | for (x = 0; x < width; x++) |
| 422 |
2/2✓ Branch 0 taken 367271 times.
✓ Branch 1 taken 357480 times.
|
724751 | if (ref[x] == ptr[x]) |
| 423 | 367271 | s->tmpl[x] = trans; | |
| 424 | 25575 | len += ff_lzw_encode(s->lzw, s->tmpl, width); | |
| 425 | 25575 | ptr += linesize; | |
| 426 | 25575 | ref += ref_linesize; | |
| 427 | } | ||
| 428 | } else { | ||
| 429 |
2/2✓ Branch 0 taken 8502 times.
✓ Branch 1 taken 31 times.
|
8533 | for (y = 0; y < height; y++) { |
| 430 | 8502 | len += ff_lzw_encode(s->lzw, ptr, width); | |
| 431 | 8502 | ptr += linesize; | |
| 432 | } | ||
| 433 | } | ||
| 434 | 1063 | len += ff_lzw_encode_flush(s->lzw); | |
| 435 | |||
| 436 | 1063 | ptr = s->buf; | |
| 437 |
2/2✓ Branch 0 taken 9251 times.
✓ Branch 1 taken 1063 times.
|
10314 | while (len > 0) { |
| 438 | 9251 | int size = FFMIN(255, len); | |
| 439 | 9251 | bytestream_put_byte(bytestream, size); | |
| 440 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 9251 times.
|
9251 | if (end - *bytestream < size) |
| 441 | ✗ | return -1; | |
| 442 | 9251 | bytestream_put_buffer(bytestream, ptr, size); | |
| 443 | 9251 | ptr += size; | |
| 444 | 9251 | len -= size; | |
| 445 | } | ||
| 446 | 1063 | bytestream_put_byte(bytestream, 0x00); /* end of image block */ | |
| 447 | 1063 | return 0; | |
| 448 | } | ||
| 449 | |||
| 450 | 7 | static av_cold int gif_encode_init(AVCodecContext *avctx) | |
| 451 | { | ||
| 452 | 7 | GIFContext *s = avctx->priv_data; | |
| 453 | |||
| 454 |
2/4✓ Branch 0 taken 7 times.
✗ Branch 1 not taken.
✗ Branch 2 not taken.
✓ Branch 3 taken 7 times.
|
7 | if (avctx->width > 65535 || avctx->height > 65535) { |
| 455 | ✗ | av_log(avctx, AV_LOG_ERROR, "GIF does not support resolutions above 65535x65535\n"); | |
| 456 | ✗ | return AVERROR(EINVAL); | |
| 457 | } | ||
| 458 | |||
| 459 | 7 | s->transparent_index = -1; | |
| 460 | |||
| 461 | 7 | s->lzw = av_mallocz(ff_lzw_encode_state_size); | |
| 462 | 7 | s->buf_size = avctx->width*avctx->height*2 + 1000; | |
| 463 | 7 | s->buf = av_malloc(s->buf_size); | |
| 464 | 7 | s->tmpl = av_malloc(avctx->width); | |
| 465 |
3/6✓ Branch 0 taken 7 times.
✗ Branch 1 not taken.
✓ Branch 2 taken 7 times.
✗ Branch 3 not taken.
✗ Branch 4 not taken.
✓ Branch 5 taken 7 times.
|
7 | if (!s->tmpl || !s->buf || !s->lzw) |
| 466 | ✗ | return AVERROR(ENOMEM); | |
| 467 | |||
| 468 |
2/2✓ Branch 1 taken 1 times.
✓ Branch 2 taken 6 times.
|
7 | if (avpriv_set_systematic_pal2(s->palette, avctx->pix_fmt) < 0) |
| 469 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 1 times.
|
1 | av_assert0(avctx->pix_fmt == AV_PIX_FMT_PAL8); |
| 470 | |||
| 471 | 7 | return 0; | |
| 472 | } | ||
| 473 | |||
| 474 | 1063 | static int gif_encode_frame(AVCodecContext *avctx, AVPacket *pkt, | |
| 475 | const AVFrame *pict, int *got_packet) | ||
| 476 | { | ||
| 477 | 1063 | GIFContext *s = avctx->priv_data; | |
| 478 | uint8_t *outbuf_ptr, *end; | ||
| 479 | 1063 | const uint32_t *palette = NULL; | |
| 480 | int ret; | ||
| 481 | |||
| 482 |
1/2✗ Branch 1 not taken.
✓ Branch 2 taken 1063 times.
|
1063 | if ((ret = ff_alloc_packet(avctx, pkt, avctx->width*avctx->height*7/5 + FF_INPUT_BUFFER_MIN_SIZE)) < 0) |
| 483 | ✗ | return ret; | |
| 484 | 1063 | outbuf_ptr = pkt->data; | |
| 485 | 1063 | end = pkt->data + pkt->size; | |
| 486 | |||
| 487 |
2/2✓ Branch 0 taken 173 times.
✓ Branch 1 taken 890 times.
|
1063 | if (avctx->pix_fmt == AV_PIX_FMT_PAL8) { |
| 488 | 173 | palette = (uint32_t*)pict->data[1]; | |
| 489 | |||
| 490 |
2/2✓ Branch 0 taken 1 times.
✓ Branch 1 taken 172 times.
|
173 | if (!s->palette_loaded) { |
| 491 | 1 | memcpy(s->palette, palette, AVPALETTE_SIZE); | |
| 492 | 1 | s->transparent_index = get_palette_transparency_index(palette); | |
| 493 | 1 | s->palette_loaded = 1; | |
| 494 |
1/2✓ Branch 0 taken 172 times.
✗ Branch 1 not taken.
|
172 | } else if (!memcmp(s->palette, palette, AVPALETTE_SIZE)) { |
| 495 | 172 | palette = NULL; | |
| 496 | } | ||
| 497 | } | ||
| 498 | |||
| 499 | 1063 | gif_image_write_image(avctx, &outbuf_ptr, end, palette, | |
| 500 | 1063 | pict->data[0], pict->linesize[0], pkt); | |
| 501 |
3/4✓ Branch 0 taken 7 times.
✓ Branch 1 taken 1056 times.
✓ Branch 2 taken 7 times.
✗ Branch 3 not taken.
|
1063 | if (!s->last_frame && !s->image) { |
| 502 | 7 | s->last_frame = av_frame_alloc(); | |
| 503 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 7 times.
|
7 | if (!s->last_frame) |
| 504 | ✗ | return AVERROR(ENOMEM); | |
| 505 | } | ||
| 506 | |||
| 507 |
1/2✓ Branch 0 taken 1063 times.
✗ Branch 1 not taken.
|
1063 | if (!s->image) { |
| 508 | 1063 | ret = av_frame_replace(s->last_frame, pict); | |
| 509 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 1063 times.
|
1063 | if (ret < 0) |
| 510 | ✗ | return ret; | |
| 511 | } | ||
| 512 | |||
| 513 | 1063 | pkt->size = outbuf_ptr - pkt->data; | |
| 514 |
3/4✓ Branch 0 taken 1063 times.
✗ Branch 1 not taken.
✓ Branch 2 taken 7 times.
✓ Branch 3 taken 1056 times.
|
1063 | if (s->image || !avctx->frame_num) |
| 515 | 7 | pkt->flags |= AV_PKT_FLAG_KEY; | |
| 516 | 1063 | *got_packet = 1; | |
| 517 | |||
| 518 | 1063 | return 0; | |
| 519 | } | ||
| 520 | |||
| 521 | 7 | static int gif_encode_close(AVCodecContext *avctx) | |
| 522 | { | ||
| 523 | 7 | GIFContext *s = avctx->priv_data; | |
| 524 | |||
| 525 | 7 | av_freep(&s->lzw); | |
| 526 | 7 | av_freep(&s->buf); | |
| 527 | 7 | av_freep(&s->shrunk_buf); | |
| 528 | 7 | s->buf_size = 0; | |
| 529 | 7 | av_frame_free(&s->last_frame); | |
| 530 | 7 | av_freep(&s->tmpl); | |
| 531 | 7 | return 0; | |
| 532 | } | ||
| 533 | |||
| 534 | #define OFFSET(x) offsetof(GIFContext, x) | ||
| 535 | #define FLAGS AV_OPT_FLAG_VIDEO_PARAM | AV_OPT_FLAG_ENCODING_PARAM | ||
| 536 | static const AVOption gif_options[] = { | ||
| 537 | { "gifflags", "set GIF flags", OFFSET(flags), AV_OPT_TYPE_FLAGS, {.i64 = GF_OFFSETTING|GF_TRANSDIFF}, 0, INT_MAX, FLAGS, .unit = "flags" }, | ||
| 538 | { "offsetting", "enable picture offsetting", 0, AV_OPT_TYPE_CONST, {.i64=GF_OFFSETTING}, INT_MIN, INT_MAX, FLAGS, .unit = "flags" }, | ||
| 539 | { "transdiff", "enable transparency detection between frames", 0, AV_OPT_TYPE_CONST, {.i64=GF_TRANSDIFF}, INT_MIN, INT_MAX, FLAGS, .unit = "flags" }, | ||
| 540 | { "gifimage", "enable encoding only images per frame", OFFSET(image), AV_OPT_TYPE_BOOL, {.i64=0}, 0, 1, FLAGS }, | ||
| 541 | { "global_palette", "write a palette to the global gif header where feasible", OFFSET(use_global_palette), AV_OPT_TYPE_BOOL, {.i64=1}, 0, 1, FLAGS }, | ||
| 542 | { NULL } | ||
| 543 | }; | ||
| 544 | |||
| 545 | static const AVClass gif_class = { | ||
| 546 | .class_name = "GIF encoder", | ||
| 547 | .item_name = av_default_item_name, | ||
| 548 | .option = gif_options, | ||
| 549 | .version = LIBAVUTIL_VERSION_INT, | ||
| 550 | }; | ||
| 551 | |||
| 552 | const FFCodec ff_gif_encoder = { | ||
| 553 | .p.name = "gif", | ||
| 554 | CODEC_LONG_NAME("GIF (Graphics Interchange Format)"), | ||
| 555 | .p.type = AVMEDIA_TYPE_VIDEO, | ||
| 556 | .p.id = AV_CODEC_ID_GIF, | ||
| 557 | .p.capabilities = AV_CODEC_CAP_DR1 | AV_CODEC_CAP_ENCODER_REORDERED_OPAQUE, | ||
| 558 | .priv_data_size = sizeof(GIFContext), | ||
| 559 | .init = gif_encode_init, | ||
| 560 | FF_CODEC_ENCODE_CB(gif_encode_frame), | ||
| 561 | .close = gif_encode_close, | ||
| 562 | .p.pix_fmts = (const enum AVPixelFormat[]){ | ||
| 563 | AV_PIX_FMT_RGB8, AV_PIX_FMT_BGR8, AV_PIX_FMT_RGB4_BYTE, AV_PIX_FMT_BGR4_BYTE, | ||
| 564 | AV_PIX_FMT_GRAY8, AV_PIX_FMT_PAL8, AV_PIX_FMT_NONE | ||
| 565 | }, | ||
| 566 | .p.priv_class = &gif_class, | ||
| 567 | .caps_internal = FF_CODEC_CAP_INIT_CLEANUP, | ||
| 568 | }; | ||
| 569 |