y4menc.c 2.2 KB
Newer Older
1 2 3 4 5 6 7 8 9 10
/*
 *  Copyright (c) 2014 The WebM project authors. All Rights Reserved.
 *
 *  Use of this source code is governed by a BSD-style license
 *  that can be found in the LICENSE file in the root of the source
 *  tree. An additional intellectual property rights grant can be found
 *  in the file PATENTS.  All contributing project authors may
 *  be found in the AUTHORS file in the root of the source tree.
 */

11
#include <assert.h>
12 13
#include "./y4menc.h"

14 15
int y4m_write_file_header(char *buf, size_t len, int width, int height,
                          const struct VpxRational *framerate,
16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50
                          vpx_img_fmt_t fmt, unsigned int bit_depth) {
  const char *color;
  switch (bit_depth) {
    case 8:
      color = fmt == VPX_IMG_FMT_444A ? "C444alpha\n" :
              fmt == VPX_IMG_FMT_I444 ? "C444\n" :
              fmt == VPX_IMG_FMT_I422 ? "C422\n" :
              "C420jpeg\n";
      break;
    case 9:
      color = fmt == VPX_IMG_FMT_I44416 ? "C444p9 XYSCSS=444P9\n" :
              fmt == VPX_IMG_FMT_I42216 ? "C422p9 XYSCSS=422P9\n" :
              "C420p9 XYSCSS=420P9\n";
      break;
    case 10:
      color = fmt == VPX_IMG_FMT_I44416 ? "C444p10 XYSCSS=444P10\n" :
              fmt == VPX_IMG_FMT_I42216 ? "C422p10 XYSCSS=422P10\n" :
              "C420p10 XYSCSS=420P10\n";
      break;
    case 12:
      color = fmt == VPX_IMG_FMT_I44416 ? "C444p12 XYSCSS=444P12\n" :
              fmt == VPX_IMG_FMT_I42216 ? "C422p12 XYSCSS=422P12\n" :
              "C420p12 XYSCSS=420P12\n";
      break;
    case 14:
      color = fmt == VPX_IMG_FMT_I44416 ? "C444p14 XYSCSS=444P14\n" :
              fmt == VPX_IMG_FMT_I42216 ? "C422p14 XYSCSS=422P14\n" :
              "C420p14 XYSCSS=420P14\n";
      break;
    case 16:
      color = fmt == VPX_IMG_FMT_I44416 ? "C444p16 XYSCSS=444P16\n" :
              fmt == VPX_IMG_FMT_I42216 ? "C422p16 XYSCSS=422P16\n" :
              "C420p16 XYSCSS=420P16\n";
      break;
    default:
51
      color = NULL;
52 53
      assert(0);
  }
54 55
  return snprintf(buf, len, "YUV4MPEG2 W%u H%u F%u:%u I%c %s", width, height,
                  framerate->numerator, framerate->denominator, 'p', color);
56 57
}

58 59
int y4m_write_frame_header(char *buf, size_t len) {
  return snprintf(buf, len, "FRAME\n");
60
}