common/y4menc.c

   1 /*
   2  * Copyright (c) 2016, Alliance for Open Media. All rights reserved
   3  *
   4  * This source code is subject to the terms of the BSD 2 Clause License and
   5  * the Alliance for Open Media Patent License 1.0. If the BSD 2 Clause License
   6  * was not distributed with this source code in the LICENSE file, you can
   7  * obtain it at www.aomedia.org/license/software. If the Alliance for Open
   8  * Media Patent License 1.0 was not distributed with this source code in the
   9  * PATENTS file, you can obtain it at www.aomedia.org/license/patent.
  10  */
  11
  12 #include <assert.h>
  13
  14 #include "common/rawenc.h"
  15 #include "common/y4menc.h"
  16
  17 // Returns the Y4M name associated with the monochrome colorspace.
  18 static const char *monochrome_colorspace(unsigned int bit_depth) {
  19   switch (bit_depth) {
  20     case 8: return "Cmono";
  21     case 9: return "Cmono9";
  22     case 10: return "Cmono10";
  23     case 12: return "Cmono12";
  24     case 16: return "Cmono16";
  25     default: assert(0); return NULL;
  26   }
  27 }
  28
  29 // Return the Y4M name of the 8-bit colorspace, given the chroma position and
  30 // image format.
  31 const char *colorspace8(aom_chroma_sample_position_t csp, aom_img_fmt_t fmt) {
  32   switch (fmt) {
  33     case AOM_IMG_FMT_I444: return "C444";
  34     case AOM_IMG_FMT_I422: return "C422";
  35     default:
  36       if (csp == AOM_CSP_VERTICAL) {
  37         return "C420mpeg2 XYSCSS=420MPEG2";
  38       } else if (csp == AOM_CSP_COLOCATED) {
  39         // Note that Y4M does not have a dedicated header for colocated chroma,
  40         // and that FFMPEG interprets C420 as C420jpeg.
  41         return "C420";
  42       } else {
  43         return "C420jpeg";
  44       }
  45   }
  46 }
  47
  48 // Return the Y4M name of the colorspace, given the bit depth and image format.
  49 static const char *colorspace(unsigned int bit_depth,
  50                               aom_chroma_sample_position_t csp,
  51                               aom_img_fmt_t fmt) {
  52   switch (bit_depth) {
  53     case 8: return colorspace8(csp, fmt);
  54     case 9:
  55       return fmt == AOM_IMG_FMT_I44416
  56                  ? "C444p9 XYSCSS=444P9"
  57                  : fmt == AOM_IMG_FMT_I42216 ? "C422p9 XYSCSS=422P9"
  58                                              : "C420p9 XYSCSS=420P9";
  59     case 10:
  60       return fmt == AOM_IMG_FMT_I44416
  61                  ? "C444p10 XYSCSS=444P10"
  62                  : fmt == AOM_IMG_FMT_I42216 ? "C422p10 XYSCSS=422P10"
  63                                              : "C420p10 XYSCSS=420P10";
  64     case 12:
  65       return fmt == AOM_IMG_FMT_I44416
  66                  ? "C444p12 XYSCSS=444P12"
  67                  : fmt == AOM_IMG_FMT_I42216 ? "C422p12 XYSCSS=422P12"
  68                                              : "C420p12 XYSCSS=420P12";
  69     case 14:
  70       return fmt == AOM_IMG_FMT_I44416
  71                  ? "C444p14 XYSCSS=444P14"
  72                  : fmt == AOM_IMG_FMT_I42216 ? "C422p14 XYSCSS=422P14"
  73                                              : "C420p14 XYSCSS=420P14";
  74     case 16:
  75       return fmt == AOM_IMG_FMT_I44416
  76                  ? "C444p16 XYSCSS=444P16"
  77                  : fmt == AOM_IMG_FMT_I42216 ? "C422p16 XYSCSS=422P16"
  78                                              : "C420p16 XYSCSS=420P16";
  79     default: assert(0); return NULL;
  80   }
  81 }
  82
  83 int y4m_write_file_header(char *buf, size_t len, int width, int height,
  84                           const struct AvxRational *framerate, int monochrome,
  85                           aom_chroma_sample_position_t csp, aom_img_fmt_t fmt,
  86                           unsigned int bit_depth, aom_color_range_t range) {
  87   const char *color = monochrome ? monochrome_colorspace(bit_depth)
  88                                  : colorspace(bit_depth, csp, fmt);
  89   const char *color_range = "";  // Default assumption is studio range.
  90   if (range == AOM_CR_FULL_RANGE) {
  91     color_range = " XCOLORRANGE=FULL";
  92   }
  93   return snprintf(buf, len, "YUV4MPEG2 W%d H%d F%d:%d Ip %s%s\n", width, height,
  94                   framerate->numerator, framerate->denominator, color,
  95                   color_range);
  96 }
  97
  98 int y4m_write_frame_header(char *buf, size_t len) {
  99   return snprintf(buf, len, "FRAME\n");
 100 }
 101
 102 void y4m_write_image_file(const aom_image_t *img, const int *planes,
 103                           FILE *file) {
 104   int num_planes = img->monochrome ? 1 : 3;
 105   raw_write_image_file(img, planes, num_planes, file);
 106 }
 107
 108 void y4m_update_image_md5(const aom_image_t *img, const int *planes,
 109                           MD5Context *md5) {
 110   int num_planes = img->monochrome ? 1 : 3;
 111   raw_update_image_md5(img, planes, num_planes, md5);
 112 }