y4menc.c 2.61 KB
Newer Older
1
/*
2
 * Copyright (c) 2016, Alliance for Open Media. All rights reserved
3
 *
4 5 6 7 8 9
 * This source code is subject to the terms of the BSD 2 Clause License and
 * the Alliance for Open Media Patent License 1.0. If the BSD 2 Clause License
 * was not distributed with this source code in the LICENSE file, you can
 * obtain it at www.aomedia.org/license/software. If the Alliance for Open
 * Media Patent License 1.0 was not distributed with this source code in the
 * PATENTS file, you can obtain it at www.aomedia.org/license/patent.
10 11
 */

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

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

63 64
int y4m_write_frame_header(char *buf, size_t len) {
  return snprintf(buf, len, "FRAME\n");
65
}