y4menc.c

   1 /*
   2  *  Copyright (c) 2014 The WebM project authors. All Rights Reserved.
   3  *
   4  *  Use of this source code is governed by a BSD-style license
   5  *  that can be found in the LICENSE file in the root of the source
   6  *  tree. An additional intellectual property rights grant can be found
   7  *  in the file PATENTS.  All contributing project authors may
   8  *  be found in the AUTHORS file in the root of the source tree.
   9  */
  10
  11 #include <assert.h>
  12 #include "./y4menc.h"
  13
  14 int y4m_write_file_header(char *buf, size_t len, int width, int height,
  15                           const struct VpxRational *framerate,
  16                           vpx_img_fmt_t fmt, unsigned int bit_depth) {
  17   const char *color;
  18   switch (bit_depth) {
  19     case 8:
  20       color = fmt == VPX_IMG_FMT_I444
  21                   ? "C444\n"
  22                   : fmt == VPX_IMG_FMT_I422 ? "C422\n" : "C420jpeg\n";
  23       break;
  24     case 9:
  25       color = fmt == VPX_IMG_FMT_I44416
  26                   ? "C444p9 XYSCSS=444P9\n"
  27                   : fmt == VPX_IMG_FMT_I42216 ? "C422p9 XYSCSS=422P9\n"
  28                                               : "C420p9 XYSCSS=420P9\n";
  29       break;
  30     case 10:
  31       color = fmt == VPX_IMG_FMT_I44416
  32                   ? "C444p10 XYSCSS=444P10\n"
  33                   : fmt == VPX_IMG_FMT_I42216 ? "C422p10 XYSCSS=422P10\n"
  34                                               : "C420p10 XYSCSS=420P10\n";
  35       break;
  36     case 12:
  37       color = fmt == VPX_IMG_FMT_I44416
  38                   ? "C444p12 XYSCSS=444P12\n"
  39                   : fmt == VPX_IMG_FMT_I42216 ? "C422p12 XYSCSS=422P12\n"
  40                                               : "C420p12 XYSCSS=420P12\n";
  41       break;
  42     case 14:
  43       color = fmt == VPX_IMG_FMT_I44416
  44                   ? "C444p14 XYSCSS=444P14\n"
  45                   : fmt == VPX_IMG_FMT_I42216 ? "C422p14 XYSCSS=422P14\n"
  46                                               : "C420p14 XYSCSS=420P14\n";
  47       break;
  48     case 16:
  49       color = fmt == VPX_IMG_FMT_I44416
  50                   ? "C444p16 XYSCSS=444P16\n"
  51                   : fmt == VPX_IMG_FMT_I42216 ? "C422p16 XYSCSS=422P16\n"
  52                                               : "C420p16 XYSCSS=420P16\n";
  53       break;
  54     default: color = NULL; assert(0);
  55   }
  56   return snprintf(buf, len, "YUV4MPEG2 W%u H%u F%u:%u I%c %s", width, height,
  57                   framerate->numerator, framerate->denominator, 'p', color);
  58 }
  59
  60 int y4m_write_frame_header(char *buf, size_t len) {
  61   return snprintf(buf, len, "FRAME\n");
  62 }