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.
monochrome_colorspace(unsigned int bit_depth)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.
colorspace8(aom_chroma_sample_position_t csp,aom_img_fmt_t fmt)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.
colorspace(unsigned int bit_depth,aom_chroma_sample_position_t csp,aom_img_fmt_t fmt)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 ? "C444p9 XYSCSS=444P9"
56 : fmt == AOM_IMG_FMT_I42216 ? "C422p9 XYSCSS=422P9"
57 : "C420p9 XYSCSS=420P9";
58 case 10:
59 return fmt == AOM_IMG_FMT_I44416 ? "C444p10 XYSCSS=444P10"
60 : fmt == AOM_IMG_FMT_I42216 ? "C422p10 XYSCSS=422P10"
61 : "C420p10 XYSCSS=420P10";
62 case 12:
63 return fmt == AOM_IMG_FMT_I44416 ? "C444p12 XYSCSS=444P12"
64 : fmt == AOM_IMG_FMT_I42216 ? "C422p12 XYSCSS=422P12"
65 : "C420p12 XYSCSS=420P12";
66 case 14:
67 return fmt == AOM_IMG_FMT_I44416 ? "C444p14 XYSCSS=444P14"
68 : fmt == AOM_IMG_FMT_I42216 ? "C422p14 XYSCSS=422P14"
69 : "C420p14 XYSCSS=420P14";
70 case 16:
71 return fmt == AOM_IMG_FMT_I44416 ? "C444p16 XYSCSS=444P16"
72 : fmt == AOM_IMG_FMT_I42216 ? "C422p16 XYSCSS=422P16"
73 : "C420p16 XYSCSS=420P16";
74 default: assert(0); return NULL;
75 }
76 }
77
y4m_write_file_header(char * buf,size_t len,int width,int height,const struct AvxRational * framerate,int monochrome,aom_chroma_sample_position_t csp,aom_img_fmt_t fmt,unsigned int bit_depth,aom_color_range_t range)78 int y4m_write_file_header(char *buf, size_t len, int width, int height,
79 const struct AvxRational *framerate, int monochrome,
80 aom_chroma_sample_position_t csp, aom_img_fmt_t fmt,
81 unsigned int bit_depth, aom_color_range_t range) {
82 const char *color = monochrome ? monochrome_colorspace(bit_depth)
83 : colorspace(bit_depth, csp, fmt);
84 const char *color_range = ""; // Default assumption is studio range.
85 if (range == AOM_CR_FULL_RANGE) {
86 color_range = " XCOLORRANGE=FULL";
87 }
88 return snprintf(buf, len, "YUV4MPEG2 W%d H%d F%d:%d Ip %s%s\n", width, height,
89 framerate->numerator, framerate->denominator, color,
90 color_range);
91 }
92
y4m_write_frame_header(char * buf,size_t len)93 int y4m_write_frame_header(char *buf, size_t len) {
94 return snprintf(buf, len, "FRAME\n");
95 }
96
y4m_write_image_file(const aom_image_t * img,const int * planes,FILE * file)97 void y4m_write_image_file(const aom_image_t *img, const int *planes,
98 FILE *file) {
99 int num_planes = img->monochrome ? 1 : 3;
100 raw_write_image_file(img, planes, num_planes, file);
101 }
102
y4m_update_image_md5(const aom_image_t * img,const int * planes,MD5Context * md5)103 void y4m_update_image_md5(const aom_image_t *img, const int *planes,
104 MD5Context *md5) {
105 int num_planes = img->monochrome ? 1 : 3;
106 raw_update_image_md5(img, planes, num_planes, md5);
107 }
108