1/* 2 * Copyright (c) 2013 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#ifndef VP9_VP9_IFACE_COMMON_H_ 11#define VP9_VP9_IFACE_COMMON_H_ 12 13#include "vpx_ports/mem.h" 14 15static void yuvconfig2image(vpx_image_t *img, const YV12_BUFFER_CONFIG *yv12, 16 void *user_priv) { 17 /** vpx_img_wrap() doesn't allow specifying independent strides for 18 * the Y, U, and V planes, nor other alignment adjustments that 19 * might be representable by a YV12_BUFFER_CONFIG, so we just 20 * initialize all the fields.*/ 21 int bps; 22 if (!yv12->subsampling_y) { 23 if (!yv12->subsampling_x) { 24 img->fmt = VPX_IMG_FMT_I444; 25 bps = 24; 26 } else { 27 img->fmt = VPX_IMG_FMT_I422; 28 bps = 16; 29 } 30 } else { 31 if (!yv12->subsampling_x) { 32 img->fmt = VPX_IMG_FMT_I440; 33 bps = 16; 34 } else { 35 img->fmt = VPX_IMG_FMT_I420; 36 bps = 12; 37 } 38 } 39 img->cs = yv12->color_space; 40 img->range = yv12->color_range; 41 img->bit_depth = 8; 42 img->w = yv12->y_stride; 43 img->h = ALIGN_POWER_OF_TWO(yv12->y_height + 2 * VP9_ENC_BORDER_IN_PIXELS, 3); 44 img->d_w = yv12->y_crop_width; 45 img->d_h = yv12->y_crop_height; 46 img->r_w = yv12->render_width; 47 img->r_h = yv12->render_height; 48 img->x_chroma_shift = yv12->subsampling_x; 49 img->y_chroma_shift = yv12->subsampling_y; 50 img->planes[VPX_PLANE_Y] = yv12->y_buffer; 51 img->planes[VPX_PLANE_U] = yv12->u_buffer; 52 img->planes[VPX_PLANE_V] = yv12->v_buffer; 53 img->planes[VPX_PLANE_ALPHA] = NULL; 54 img->stride[VPX_PLANE_Y] = yv12->y_stride; 55 img->stride[VPX_PLANE_U] = yv12->uv_stride; 56 img->stride[VPX_PLANE_V] = yv12->uv_stride; 57 img->stride[VPX_PLANE_ALPHA] = yv12->y_stride; 58#if CONFIG_VP9_HIGHBITDEPTH 59 if (yv12->flags & YV12_FLAG_HIGHBITDEPTH) { 60 // vpx_image_t uses byte strides and a pointer to the first byte 61 // of the image. 62 img->fmt = (vpx_img_fmt_t)(img->fmt | VPX_IMG_FMT_HIGHBITDEPTH); 63 img->bit_depth = yv12->bit_depth; 64 img->planes[VPX_PLANE_Y] = (uint8_t*)CONVERT_TO_SHORTPTR(yv12->y_buffer); 65 img->planes[VPX_PLANE_U] = (uint8_t*)CONVERT_TO_SHORTPTR(yv12->u_buffer); 66 img->planes[VPX_PLANE_V] = (uint8_t*)CONVERT_TO_SHORTPTR(yv12->v_buffer); 67 img->planes[VPX_PLANE_ALPHA] = NULL; 68 img->stride[VPX_PLANE_Y] = 2 * yv12->y_stride; 69 img->stride[VPX_PLANE_U] = 2 * yv12->uv_stride; 70 img->stride[VPX_PLANE_V] = 2 * yv12->uv_stride; 71 img->stride[VPX_PLANE_ALPHA] = 2 * yv12->y_stride; 72 } 73#endif // CONFIG_VP9_HIGHBITDEPTH 74 img->bps = bps; 75 img->user_priv = user_priv; 76 img->img_data = yv12->buffer_alloc; 77 img->img_data_owner = 0; 78 img->self_allocd = 0; 79} 80 81static vpx_codec_err_t image2yuvconfig(const vpx_image_t *img, 82 YV12_BUFFER_CONFIG *yv12) { 83 yv12->y_buffer = img->planes[VPX_PLANE_Y]; 84 yv12->u_buffer = img->planes[VPX_PLANE_U]; 85 yv12->v_buffer = img->planes[VPX_PLANE_V]; 86 87 yv12->y_crop_width = img->d_w; 88 yv12->y_crop_height = img->d_h; 89 yv12->render_width = img->r_w; 90 yv12->render_height = img->r_h; 91 yv12->y_width = img->d_w; 92 yv12->y_height = img->d_h; 93 94 yv12->uv_width = img->x_chroma_shift == 1 ? (1 + yv12->y_width) / 2 95 : yv12->y_width; 96 yv12->uv_height = img->y_chroma_shift == 1 ? (1 + yv12->y_height) / 2 97 : yv12->y_height; 98 yv12->uv_crop_width = yv12->uv_width; 99 yv12->uv_crop_height = yv12->uv_height; 100 101 yv12->y_stride = img->stride[VPX_PLANE_Y]; 102 yv12->uv_stride = img->stride[VPX_PLANE_U]; 103 yv12->color_space = img->cs; 104 yv12->color_range = img->range; 105 106#if CONFIG_VP9_HIGHBITDEPTH 107 if (img->fmt & VPX_IMG_FMT_HIGHBITDEPTH) { 108 // In vpx_image_t 109 // planes point to uint8 address of start of data 110 // stride counts uint8s to reach next row 111 // In YV12_BUFFER_CONFIG 112 // y_buffer, u_buffer, v_buffer point to uint16 address of data 113 // stride and border counts in uint16s 114 // This means that all the address calculations in the main body of code 115 // should work correctly. 116 // However, before we do any pixel operations we need to cast the address 117 // to a uint16 ponter and double its value. 118 yv12->y_buffer = CONVERT_TO_BYTEPTR(yv12->y_buffer); 119 yv12->u_buffer = CONVERT_TO_BYTEPTR(yv12->u_buffer); 120 yv12->v_buffer = CONVERT_TO_BYTEPTR(yv12->v_buffer); 121 yv12->y_stride >>= 1; 122 yv12->uv_stride >>= 1; 123 yv12->flags = YV12_FLAG_HIGHBITDEPTH; 124 } else { 125 yv12->flags = 0; 126 } 127 yv12->border = (yv12->y_stride - img->w) / 2; 128#else 129 yv12->border = (img->stride[VPX_PLANE_Y] - img->w) / 2; 130#endif // CONFIG_VP9_HIGHBITDEPTH 131 yv12->subsampling_x = img->x_chroma_shift; 132 yv12->subsampling_y = img->y_chroma_shift; 133 return VPX_CODEC_OK; 134} 135 136#endif // VP9_VP9_IFACE_COMMON_H_ 137