y4menc.c (4353B)
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. 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. 31 static const char *colorspace8(aom_chroma_sample_position_t csp, 32 aom_img_fmt_t fmt) { 33 switch (fmt) { 34 case AOM_IMG_FMT_I444: return "C444"; 35 case AOM_IMG_FMT_I422: return "C422"; 36 default: 37 if (csp == AOM_CSP_VERTICAL) { 38 return "C420mpeg2 XYSCSS=420MPEG2"; 39 } else if (csp == AOM_CSP_COLOCATED) { 40 // Note that Y4M does not have a dedicated header for colocated chroma, 41 // and that FFMPEG interprets C420 as C420jpeg. 42 return "C420"; 43 } else { 44 return "C420jpeg"; 45 } 46 } 47 } 48 49 // Return the Y4M name of the colorspace, given the bit depth and image format. 50 static const char *colorspace(unsigned int bit_depth, 51 aom_chroma_sample_position_t csp, 52 aom_img_fmt_t fmt) { 53 switch (bit_depth) { 54 case 8: return colorspace8(csp, fmt); 55 case 9: 56 return fmt == AOM_IMG_FMT_I44416 ? "C444p9 XYSCSS=444P9" 57 : fmt == AOM_IMG_FMT_I42216 ? "C422p9 XYSCSS=422P9" 58 : "C420p9 XYSCSS=420P9"; 59 case 10: 60 return fmt == AOM_IMG_FMT_I44416 ? "C444p10 XYSCSS=444P10" 61 : fmt == AOM_IMG_FMT_I42216 ? "C422p10 XYSCSS=422P10" 62 : "C420p10 XYSCSS=420P10"; 63 case 12: 64 return fmt == AOM_IMG_FMT_I44416 ? "C444p12 XYSCSS=444P12" 65 : fmt == AOM_IMG_FMT_I42216 ? "C422p12 XYSCSS=422P12" 66 : "C420p12 XYSCSS=420P12"; 67 case 14: 68 return fmt == AOM_IMG_FMT_I44416 ? "C444p14 XYSCSS=444P14" 69 : fmt == AOM_IMG_FMT_I42216 ? "C422p14 XYSCSS=422P14" 70 : "C420p14 XYSCSS=420P14"; 71 case 16: 72 return fmt == AOM_IMG_FMT_I44416 ? "C444p16 XYSCSS=444P16" 73 : fmt == AOM_IMG_FMT_I42216 ? "C422p16 XYSCSS=422P16" 74 : "C420p16 XYSCSS=420P16"; 75 default: assert(0); return NULL; 76 } 77 } 78 79 int y4m_write_file_header(char *buf, size_t len, int width, int height, 80 const struct AvxRational *framerate, int monochrome, 81 aom_chroma_sample_position_t csp, aom_img_fmt_t fmt, 82 unsigned int bit_depth, aom_color_range_t range) { 83 const char *color = monochrome ? monochrome_colorspace(bit_depth) 84 : colorspace(bit_depth, csp, fmt); 85 const char *color_range = ""; // Default assumption is studio range. 86 if (range == AOM_CR_FULL_RANGE) { 87 color_range = " XCOLORRANGE=FULL"; 88 } 89 return snprintf(buf, len, "YUV4MPEG2 W%d H%d F%d:%d Ip %s%s\n", width, height, 90 framerate->numerator, framerate->denominator, color, 91 color_range); 92 } 93 94 int y4m_write_frame_header(char *buf, size_t len) { 95 return snprintf(buf, len, "FRAME\n"); 96 } 97 98 void y4m_write_image_file(const aom_image_t *img, const int *planes, 99 FILE *file) { 100 int num_planes = img->monochrome ? 1 : 3; 101 raw_write_image_file(img, planes, num_planes, file); 102 } 103 104 void y4m_update_image_md5(const aom_image_t *img, const int *planes, 105 MD5Context *md5) { 106 int num_planes = img->monochrome ? 1 : 3; 107 raw_update_image_md5(img, planes, num_planes, md5); 108 }