/*
* Copyright (c) 2016, Alliance for Open Media. All rights reserved.
*
* 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.
*/
#include <assert.h>
#include "common/rawenc.h"
#include "common/y4menc.h"
// Returns the Y4M name associated with the monochrome colorspace.
static const char *monochrome_colorspace(
unsigned int bit_depth) {
switch (bit_depth) {
case 8:
return "Cmono";
case 9:
return "Cmono9";
case 10:
return "Cmono10";
case 12:
return "Cmono12";
case 16:
return "Cmono16";
default: assert(0);
return NULL;
}
}
// Return the Y4M name of the 8-bit colorspace, given the chroma position and
// image format.
static const char *colorspace8(aom_chroma_sample_position_t csp,
aom_img_fmt_t fmt) {
switch (fmt) {
case AOM_IMG_FMT_I444:
return "C444";
case AOM_IMG_FMT_I422:
return "C422";
default:
if (csp == AOM_CSP_VERTICAL) {
return "C420mpeg2 XYSCSS=420MPEG2";
}
else if (csp == AOM_CSP_COLOCATED) {
// Note that Y4M does not have a dedicated header for colocated chroma,
// and that FFMPEG interprets C420 as C420jpeg.
return "C420";
}
else {
return "C420jpeg";
}
}
}
// Return the Y4M name of the colorspace, given the bit depth and image format.
static const char *colorspace(
unsigned int bit_depth,
aom_chroma_sample_position_t csp,
aom_img_fmt_t fmt) {
switch (bit_depth) {
case 8:
return colorspace8(csp, fmt);
case 9:
return fmt == AOM_IMG_FMT_I44416 ?
"C444p9 XYSCSS=444P9"
: fmt == AOM_IMG_FMT_I42216 ?
"C422p9 XYSCSS=422P9"
:
"C420p9 XYSCSS=420P9";
case 10:
return fmt == AOM_IMG_FMT_I44416 ?
"C444p10 XYSCSS=444P10"
: fmt == AOM_IMG_FMT_I42216 ?
"C422p10 XYSCSS=422P10"
:
"C420p10 XYSCSS=420P10";
case 12:
return fmt == AOM_IMG_FMT_I44416 ?
"C444p12 XYSCSS=444P12"
: fmt == AOM_IMG_FMT_I42216 ?
"C422p12 XYSCSS=422P12"
:
"C420p12 XYSCSS=420P12";
case 14:
return fmt == AOM_IMG_FMT_I44416 ?
"C444p14 XYSCSS=444P14"
: fmt == AOM_IMG_FMT_I42216 ?
"C422p14 XYSCSS=422P14"
:
"C420p14 XYSCSS=420P14";
case 16:
return fmt == AOM_IMG_FMT_I44416 ?
"C444p16 XYSCSS=444P16"
: fmt == AOM_IMG_FMT_I42216 ?
"C422p16 XYSCSS=422P16"
:
"C420p16 XYSCSS=420P16";
default: assert(0);
return NULL;
}
}
int 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) {
const char *color = monochrome ? monochrome_colorspace(bit_depth)
: colorspace(bit_depth, csp, fmt);
const char *color_range =
"";
// Default assumption is studio range.
if (range == AOM_CR_FULL_RANGE) {
color_range =
" XCOLORRANGE=FULL";
}
return snprintf(buf, len,
"YUV4MPEG2 W%d H%d F%d:%d Ip %s%s\n", width, height,
framerate->numerator, framerate->denominator, color,
color_range);
}
int y4m_write_frame_header(
char *buf, size_t len) {
return snprintf(buf, len,
"FRAME\n");
}
void y4m_write_image_file(
const aom_image_t *img,
const int *planes,
FILE *file) {
int num_planes = img->monochrome ? 1 : 3;
raw_write_image_file(img, planes, num_planes, file);
}
void y4m_update_image_md5(
const aom_image_t *img,
const int *planes,
MD5Context *md5) {
int num_planes = img->monochrome ? 1 : 3;
raw_update_image_md5(img, planes, num_planes, md5);
}
