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_NONE
  33 };
  34
  35 #if CONFIG_LIBVPX_VP9_ENCODER
  36 static const enum AVPixelFormat vp9_pix_fmts_highcol[] = {
  37     AV_PIX_FMT_YUV420P,
  38     AV_PIX_FMT_YUV422P,
  39     AV_PIX_FMT_YUV444P,
  40     AV_PIX_FMT_NONE
  41 };
  42
  43 static const enum AVPixelFormat vp9_pix_fmts_highbd[] = {
  44     AV_PIX_FMT_YUV420P,
  45     AV_PIX_FMT_YUV422P,
  46     AV_PIX_FMT_YUV444P,
  47     AV_PIX_FMT_YUV420P10LE,
  48     AV_PIX_FMT_YUV422P10LE,
  49     AV_PIX_FMT_YUV444P10LE,
  50     AV_PIX_FMT_YUV420P12LE,
  51     AV_PIX_FMT_YUV422P12LE,
  52     AV_PIX_FMT_YUV444P12LE,
  53     AV_PIX_FMT_NONE
  54 };
  55 #endif
  56
  57 av_cold void ff_vp9_init_static(AVCodec *codec)
  58 {
  59     if (    vpx_codec_version_major() < 1
  60         || (vpx_codec_version_major() == 1 && vpx_codec_version_minor() < 3))
  61         codec->capabilities |= CODEC_CAP_EXPERIMENTAL;
  62     codec->pix_fmts = vp9_pix_fmts_def;
  63 #if CONFIG_LIBVPX_VP9_ENCODER
  64     if (    vpx_codec_version_major() > 1
  65         || (vpx_codec_version_major() == 1 && vpx_codec_version_minor() >= 4)) {
  66 #ifdef VPX_CODEC_CAP_HIGHBITDEPTH
  67         vpx_codec_caps_t codec_caps = vpx_codec_get_caps(vpx_codec_vp9_cx());
  68         if (codec_caps & VPX_CODEC_CAP_HIGHBITDEPTH)
  69             codec->pix_fmts = vp9_pix_fmts_highbd;
  70         else
  71 #endif
  72             codec->pix_fmts = vp9_pix_fmts_highcol;
  73     }
  74 #endif
  75 }