git.sesse.net Git - ffmpeg/blob - libavformat/vpcc.c

   1 /*
   2  * Copyright (c) 2016 Google Inc.
   3  * Copyright (c) 2016 KongQun Yang (kqyang@google.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 #include "libavutil/pixdesc.h"
  23 #include "libavutil/pixfmt.h"
  24 #include "vpcc.h"
  25
  26 enum VPX_CHROMA_SUBSAMPLING
  27 {
  28     VPX_SUBSAMPLING_420_VERTICAL = 0,
  29     VPX_SUBSAMPLING_420_COLLOCATED_WITH_LUMA = 1,
  30     VPX_SUBSAMPLING_422 = 2,
  31     VPX_SUBSAMPLING_444 = 3,
  32 };
  33
  34 static int get_vpx_chroma_subsampling(AVFormatContext *s,
  35                                       enum AVPixelFormat pixel_format,
  36                                       enum AVChromaLocation chroma_location)
  37 {
  38     int chroma_w, chroma_h;
  39     if (av_pix_fmt_get_chroma_sub_sample(pixel_format, &chroma_w, &chroma_h) == 0) {
  40         if (chroma_w == 1 && chroma_h == 1) {
  41             return (chroma_location == AVCHROMA_LOC_LEFT)
  42                        ? VPX_SUBSAMPLING_420_VERTICAL
  43                        : VPX_SUBSAMPLING_420_COLLOCATED_WITH_LUMA;
  44         } else if (chroma_w == 1 && chroma_h == 0) {
  45             return VPX_SUBSAMPLING_422;
  46         } else if (chroma_w == 0 && chroma_h == 0) {
  47             return VPX_SUBSAMPLING_444;
  48         }
  49     }
  50     av_log(s, AV_LOG_ERROR, "Unsupported pixel format (%d)\n", pixel_format);
  51     return -1;
  52 }
  53
  54 static int get_bit_depth(AVFormatContext *s, enum AVPixelFormat pixel_format)
  55 {
  56     const AVPixFmtDescriptor *desc = av_pix_fmt_desc_get(pixel_format);
  57     if (desc == NULL) {
  58         av_log(s, AV_LOG_ERROR, "Unsupported pixel format (%d)\n",
  59                pixel_format);
  60         return -1;
  61     }
  62     return desc->comp[0].depth;
  63 }
  64
  65 static int get_vpx_video_full_range_flag(enum AVColorRange color_range)
  66 {
  67     return color_range == AVCOL_RANGE_JPEG;
  68 }
  69
  70 int ff_isom_write_vpcc(AVFormatContext *s, AVIOContext *pb,
  71                        AVCodecParameters *par)
  72 {
  73     int profile = par->profile;
  74     int level = par->level == FF_LEVEL_UNKNOWN ? 0 : par->level;
  75     int bit_depth = get_bit_depth(s, par->format);
  76     int vpx_chroma_subsampling =
  77         get_vpx_chroma_subsampling(s, par->format, par->chroma_location);
  78     int vpx_video_full_range_flag =
  79         get_vpx_video_full_range_flag(par->color_range);
  80
  81     if (bit_depth < 0 || vpx_chroma_subsampling < 0)
  82         return AVERROR_INVALIDDATA;
  83
  84     if (profile == FF_PROFILE_UNKNOWN) {
  85         if (vpx_chroma_subsampling == VPX_SUBSAMPLING_420_VERTICAL ||
  86             vpx_chroma_subsampling == VPX_SUBSAMPLING_420_COLLOCATED_WITH_LUMA) {
  87             profile = (bit_depth == 8) ? FF_PROFILE_VP9_0 : FF_PROFILE_VP9_2;
  88         } else {
  89             profile = (bit_depth == 8) ? FF_PROFILE_VP9_1 : FF_PROFILE_VP9_3;
  90         }
  91     }
  92
  93     avio_w8(pb, profile);
  94     avio_w8(pb, level);
  95     avio_w8(pb, (bit_depth << 4) | (vpx_chroma_subsampling << 1) | vpx_video_full_range_flag);
  96     avio_w8(pb, par->color_primaries);
  97     avio_w8(pb, par->color_trc);
  98     avio_w8(pb, par->color_space);
  99
 100     // vp9 does not have codec initialization data.
 101     avio_wb16(pb, 0);
 102     return 0;
 103 }