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 }
  81 #if 0
  82 enum AVPixelFormat ff_vpx_imgfmt_to_pixfmt(vpx_img_fmt_t img)
  83 {
  84     switch (img) {
  85     case VPX_IMG_FMT_RGB24:     return AV_PIX_FMT_RGB24;
  86     case VPX_IMG_FMT_RGB565:    return AV_PIX_FMT_RGB565BE;
  87     case VPX_IMG_FMT_RGB555:    return AV_PIX_FMT_RGB555BE;
  88     case VPX_IMG_FMT_UYVY:      return AV_PIX_FMT_UYVY422;
  89     case VPX_IMG_FMT_YUY2:      return AV_PIX_FMT_YUYV422;
  90     case VPX_IMG_FMT_YVYU:      return AV_PIX_FMT_YVYU422;
  91     case VPX_IMG_FMT_BGR24:     return AV_PIX_FMT_BGR24;
  92     case VPX_IMG_FMT_ARGB:      return AV_PIX_FMT_ARGB;
  93     case VPX_IMG_FMT_ARGB_LE:   return AV_PIX_FMT_BGRA;
  94     case VPX_IMG_FMT_RGB565_LE: return AV_PIX_FMT_RGB565LE;
  95     case VPX_IMG_FMT_RGB555_LE: return AV_PIX_FMT_RGB555LE;
  96     case VPX_IMG_FMT_I420:      return AV_PIX_FMT_YUV420P;
  97     case VPX_IMG_FMT_I422:      return AV_PIX_FMT_YUV422P;
  98     case VPX_IMG_FMT_I444:      return AV_PIX_FMT_YUV444P;
  99     case VPX_IMG_FMT_444A:      return AV_PIX_FMT_YUVA444P;
 100 #if VPX_IMAGE_ABI_VERSION >= 3
 101     case VPX_IMG_FMT_I440:      return AV_PIX_FMT_YUV440P;
 102     case VPX_IMG_FMT_I42016:    return AV_PIX_FMT_YUV420P16BE;
 103     case VPX_IMG_FMT_I42216:    return AV_PIX_FMT_YUV422P16BE;
 104     case VPX_IMG_FMT_I44416:    return AV_PIX_FMT_YUV444P16BE;
 105 #endif
 106     default:                    return AV_PIX_FMT_NONE;
 107     }
 108 }
 109
 110 vpx_img_fmt_t ff_vpx_pixfmt_to_imgfmt(enum AVPixelFormat pix)
 111 {
 112     switch (pix) {
 113     case AV_PIX_FMT_RGB24:        return VPX_IMG_FMT_RGB24;
 114     case AV_PIX_FMT_RGB565BE:     return VPX_IMG_FMT_RGB565;
 115     case AV_PIX_FMT_RGB555BE:     return VPX_IMG_FMT_RGB555;
 116     case AV_PIX_FMT_UYVY422:      return VPX_IMG_FMT_UYVY;
 117     case AV_PIX_FMT_YUYV422:      return VPX_IMG_FMT_YUY2;
 118     case AV_PIX_FMT_YVYU422:      return VPX_IMG_FMT_YVYU;
 119     case AV_PIX_FMT_BGR24:        return VPX_IMG_FMT_BGR24;
 120     case AV_PIX_FMT_ARGB:         return VPX_IMG_FMT_ARGB;
 121     case AV_PIX_FMT_BGRA:         return VPX_IMG_FMT_ARGB_LE;
 122     case AV_PIX_FMT_RGB565LE:     return VPX_IMG_FMT_RGB565_LE;
 123     case AV_PIX_FMT_RGB555LE:     return VPX_IMG_FMT_RGB555_LE;
 124     case AV_PIX_FMT_YUV420P:      return VPX_IMG_FMT_I420;
 125     case AV_PIX_FMT_YUV422P:      return VPX_IMG_FMT_I422;
 126     case AV_PIX_FMT_YUV444P:      return VPX_IMG_FMT_I444;
 127     case AV_PIX_FMT_YUVA444P:     return VPX_IMG_FMT_444A;
 128 #if VPX_IMAGE_ABI_VERSION >= 3
 129     case AV_PIX_FMT_YUV440P:      return VPX_IMG_FMT_I440;
 130     case AV_PIX_FMT_YUV420P16BE:  return VPX_IMG_FMT_I42016;
 131     case AV_PIX_FMT_YUV422P16BE:  return VPX_IMG_FMT_I42216;
 132     case AV_PIX_FMT_YUV444P16BE:  return VPX_IMG_FMT_I44416;
 133 #endif
 134     default:                      return VPX_IMG_FMT_NONE;
 135     }
 136 }
 137 #endif