1 /*
2 * WebP encoding support via libwebp
3 * Copyright (c) 2013 Justin Ruggles <justin.ruggles@gmail.com>
4 *
5 * This file is part of FFmpeg.
6 *
7 * FFmpeg is free software; you can redistribute it and/or
8 * modify it under the terms of the GNU Lesser General Public
9 * License as published by the Free Software Foundation; either
10 * version 2.1 of the License, or (at your option) any later version.
11 *
12 * FFmpeg is distributed in the hope that it will be useful,
13 * but WITHOUT ANY WARRANTY; without even the implied warranty of
14 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
15 * Lesser General Public License for more details.
16 *
17 * You should have received a copy of the GNU Lesser General Public
18 * License along with FFmpeg; if not, write to the Free Software
19 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
20 */
21
22 /**
23 * @file
24 * WebP encoder using libwebp (WebPEncode API)
25 */
26
27 #include "libwebpenc_common.h"
28
29 typedef LibWebPContextCommon LibWebPContext;
30
libwebp_encode_init(AVCodecContext * avctx)31 static av_cold int libwebp_encode_init(AVCodecContext *avctx)
32 {
33 return ff_libwebp_encode_init_common(avctx);
34 }
35
libwebp_encode_frame(AVCodecContext * avctx,AVPacket * pkt,const AVFrame * frame,int * got_packet)36 static int libwebp_encode_frame(AVCodecContext *avctx, AVPacket *pkt,
37 const AVFrame *frame, int *got_packet)
38 {
39 LibWebPContext *s = avctx->priv_data;
40 WebPPicture *pic = NULL;
41 AVFrame *alt_frame = NULL;
42 WebPMemoryWriter mw = { 0 };
43
44 int ret = ff_libwebp_get_frame(avctx, s, frame, &alt_frame, &pic);
45 if (ret < 0)
46 goto end;
47
48 WebPMemoryWriterInit(&mw);
49 pic->custom_ptr = &mw;
50 pic->writer = WebPMemoryWrite;
51
52 ret = WebPEncode(&s->config, pic);
53 if (!ret) {
54 av_log(avctx, AV_LOG_ERROR, "WebPEncode() failed with error: %d\n",
55 pic->error_code);
56 ret = ff_libwebp_error_to_averror(pic->error_code);
57 goto end;
58 }
59
60 ret = ff_alloc_packet2(avctx, pkt, mw.size, mw.size);
61 if (ret < 0)
62 goto end;
63 memcpy(pkt->data, mw.mem, mw.size);
64
65 pkt->flags |= AV_PKT_FLAG_KEY;
66 *got_packet = 1;
67
68 end:
69 #if (WEBP_ENCODER_ABI_VERSION > 0x0203)
70 WebPMemoryWriterClear(&mw);
71 #else
72 free(mw.mem); /* must use free() according to libwebp documentation */
73 #endif
74 WebPPictureFree(pic);
75 av_freep(&pic);
76 av_frame_free(&alt_frame);
77
78 return ret;
79 }
80
libwebp_encode_close(AVCodecContext * avctx)81 static int libwebp_encode_close(AVCodecContext *avctx)
82 {
83 LibWebPContextCommon *s = avctx->priv_data;
84 av_frame_free(&s->ref);
85
86 return 0;
87 }
88
89 static const AVClass class = {
90 .class_name = "libwebp",
91 .item_name = av_default_item_name,
92 .option = options,
93 .version = LIBAVUTIL_VERSION_INT,
94 };
95
96 AVCodec ff_libwebp_encoder = {
97 .name = "libwebp",
98 .long_name = NULL_IF_CONFIG_SMALL("libwebp WebP image"),
99 .type = AVMEDIA_TYPE_VIDEO,
100 .id = AV_CODEC_ID_WEBP,
101 .priv_data_size = sizeof(LibWebPContext),
102 .init = libwebp_encode_init,
103 .encode2 = libwebp_encode_frame,
104 .close = libwebp_encode_close,
105 .pix_fmts = (const enum AVPixelFormat[]) {
106 AV_PIX_FMT_RGB32,
107 AV_PIX_FMT_YUV420P, AV_PIX_FMT_YUVA420P,
108 AV_PIX_FMT_NONE
109 },
110 .priv_class = &class,
111 .defaults = libwebp_defaults,
112 .wrapper_name = "libwebp",
113 };
114