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
14int 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_444A
21                  ? "C444alpha\n"
22                  : fmt == VPX_IMG_FMT_I444
23                        ? "C444\n"
24                        : fmt == VPX_IMG_FMT_I422 ? "C422\n" : "C420jpeg\n";
25      break;
26    case 9:
27      color = fmt == VPX_IMG_FMT_I44416
28                  ? "C444p9 XYSCSS=444P9\n"
29                  : fmt == VPX_IMG_FMT_I42216 ? "C422p9 XYSCSS=422P9\n"
30                                              : "C420p9 XYSCSS=420P9\n";
31      break;
32    case 10:
33      color = fmt == VPX_IMG_FMT_I44416
34                  ? "C444p10 XYSCSS=444P10\n"
35                  : fmt == VPX_IMG_FMT_I42216 ? "C422p10 XYSCSS=422P10\n"
36                                              : "C420p10 XYSCSS=420P10\n";
37      break;
38    case 12:
39      color = fmt == VPX_IMG_FMT_I44416
40                  ? "C444p12 XYSCSS=444P12\n"
41                  : fmt == VPX_IMG_FMT_I42216 ? "C422p12 XYSCSS=422P12\n"
42                                              : "C420p12 XYSCSS=420P12\n";
43      break;
44    case 14:
45      color = fmt == VPX_IMG_FMT_I44416
46                  ? "C444p14 XYSCSS=444P14\n"
47                  : fmt == VPX_IMG_FMT_I42216 ? "C422p14 XYSCSS=422P14\n"
48                                              : "C420p14 XYSCSS=420P14\n";
49      break;
50    case 16:
51      color = fmt == VPX_IMG_FMT_I44416
52                  ? "C444p16 XYSCSS=444P16\n"
53                  : fmt == VPX_IMG_FMT_I42216 ? "C422p16 XYSCSS=422P16\n"
54                                              : "C420p16 XYSCSS=420P16\n";
55      break;
56    default: color = NULL; assert(0);
57  }
58  return snprintf(buf, len, "YUV4MPEG2 W%u H%u F%u:%u I%c %s", width, height,
59                  framerate->numerator, framerate->denominator, 'p', color);
60}
61
62int y4m_write_frame_header(char *buf, size_t len) {
63  return snprintf(buf, len, "FRAME\n");
64}
65