git.sesse.net Git - ffmpeg/blob - libavcodec/libvpx.c

   1 /*
   2  * Copyright (c) 2013 Guillaume Martres <smarter@ubuntu.com>
   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 #include <vpx/vpx_codec.h>
  22 #include "libvpx.h"
  23 #include "config.h"
  24
  25 #if CONFIG_LIBVPX_VP9_ENCODER
  26 #include <vpx/vpx_encoder.h>
  27 #include <vpx/vp8cx.h>
  28 #endif
  29
  30 static const enum AVPixelFormat vp9_pix_fmts_def[] = {
  31     AV_PIX_FMT_YUV420P,
  32     AV_PIX_FMT_YUVA420P,
  33     AV_PIX_FMT_NONE
  34 };
  35
  36 #if CONFIG_LIBVPX_VP9_ENCODER
  37 static const enum AVPixelFormat vp9_pix_fmts_highcol[] = {
  38     AV_PIX_FMT_YUV420P,
  39     AV_PIX_FMT_YUVA420P,
  40     AV_PIX_FMT_YUV422P,
  41     AV_PIX_FMT_YUV440P,
  42     AV_PIX_FMT_YUV444P,
  43     AV_PIX_FMT_GBRP,
  44     AV_PIX_FMT_NONE
  45 };
  46
  47 static const enum AVPixelFormat vp9_pix_fmts_highbd[] = {
  48     AV_PIX_FMT_YUV420P,
  49     AV_PIX_FMT_YUVA420P,
  50     AV_PIX_FMT_YUV422P,
  51     AV_PIX_FMT_YUV440P,
  52     AV_PIX_FMT_YUV444P,
  53     AV_PIX_FMT_YUV420P10,
  54     AV_PIX_FMT_YUV422P10,
  55     AV_PIX_FMT_YUV440P10,
  56     AV_PIX_FMT_YUV444P10,
  57     AV_PIX_FMT_YUV420P12,
  58     AV_PIX_FMT_YUV422P12,
  59     AV_PIX_FMT_YUV440P12,
  60     AV_PIX_FMT_YUV444P12,
  61     AV_PIX_FMT_GBRP,
  62     AV_PIX_FMT_GBRP10,
  63     AV_PIX_FMT_GBRP12,
  64     AV_PIX_FMT_NONE
  65 };
  66 #endif
  67
  68 av_cold void ff_vp9_init_static(AVCodec *codec)
  69 {
  70     codec->pix_fmts = vp9_pix_fmts_def;
  71 #if CONFIG_LIBVPX_VP9_ENCODER
  72     {
  73         vpx_codec_caps_t codec_caps = vpx_codec_get_caps(vpx_codec_vp9_cx());
  74         if (codec_caps & VPX_CODEC_CAP_HIGHBITDEPTH)
  75             codec->pix_fmts = vp9_pix_fmts_highbd;
  76         else
  77             codec->pix_fmts = vp9_pix_fmts_highcol;
  78     }
  79 #endif
  80 }