libavcodec/libwebpenc.c

   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 "codec_internal.h"
  28 #include "encode.h"
  29 #include "libwebpenc_common.h"
  30
  31 typedef LibWebPContextCommon LibWebPContext;
  32
  33 static av_cold int libwebp_encode_init(AVCodecContext *avctx)
  34 {
  35     return ff_libwebp_encode_init_common(avctx);
  36 }
  37
  38 static int libwebp_encode_frame(AVCodecContext *avctx, AVPacket *pkt,
  39                                 const AVFrame *frame, int *got_packet)
  40 {
  41     LibWebPContext *s  = avctx->priv_data;
  42     WebPPicture *pic = NULL;
  43     AVFrame *alt_frame = NULL;
  44     WebPMemoryWriter mw = { 0 };
  45
  46     int ret = ff_libwebp_get_frame(avctx, s, frame, &alt_frame, &pic);
  47     if (ret < 0)
  48         goto end;
  49
  50     WebPMemoryWriterInit(&mw);
  51     pic->custom_ptr = &mw;
  52     pic->writer     = WebPMemoryWrite;
  53
  54     ret = WebPEncode(&s->config, pic);
  55     if (!ret) {
  56         av_log(avctx, AV_LOG_ERROR, "WebPEncode() failed with error: %d\n",
  57                pic->error_code);
  58         ret = ff_libwebp_error_to_averror(pic->error_code);
  59         goto end;
  60     }
  61
  62     ret = ff_get_encode_buffer(avctx, pkt, mw.size, 0);
  63     if (ret < 0)
  64         goto end;
  65     memcpy(pkt->data, mw.mem, mw.size);
  66
  67     *got_packet = 1;
  68
  69 end:
  70 #if (WEBP_ENCODER_ABI_VERSION > 0x0203)
  71     WebPMemoryWriterClear(&mw);
  72 #else
  73     free(mw.mem); /* must use free() according to libwebp documentation */
  74 #endif
  75     WebPPictureFree(pic);
  76     av_freep(&pic);
  77     av_frame_free(&alt_frame);
  78
  79     return ret;
  80 }
  81
  82 static int libwebp_encode_close(AVCodecContext *avctx)
  83 {
  84     LibWebPContextCommon *s  = avctx->priv_data;
  85     av_frame_free(&s->ref);
  86
  87     return 0;
  88 }
  89
  90 const FFCodec ff_libwebp_encoder = {
  91     .p.name         = "libwebp",
  92     .p.long_name    = NULL_IF_CONFIG_SMALL("libwebp WebP image"),
  93     .p.type         = AVMEDIA_TYPE_VIDEO,
  94     .p.id           = AV_CODEC_ID_WEBP,
  95     .p.capabilities = AV_CODEC_CAP_DR1,
  96     .p.pix_fmts     = ff_libwebpenc_pix_fmts,
  97     .p.priv_class   = &ff_libwebpenc_class,
  98     .p.wrapper_name = "libwebp",
  99     .priv_data_size = sizeof(LibWebPContext),
 100     .defaults       = ff_libwebp_defaults,
 101     .init           = libwebp_encode_init,
 102     FF_CODEC_ENCODE_CB(libwebp_encode_frame),
 103     .close          = libwebp_encode_close,
 104 };