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 VPX_VP9_VP9_IFACE_COMMON_H_
11 #define VPX_VP9_VP9_IFACE_COMMON_H_
12
13 #include "vpx_ports/mem.h"
14
yuvconfig2image(vpx_image_t * img,const YV12_BUFFER_CONFIG * yv12,void * user_priv)15 static 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
image2yuvconfig(const vpx_image_t * img,YV12_BUFFER_CONFIG * yv12)81 static 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 =
95 img->x_chroma_shift == 1 ? (1 + yv12->y_width) / 2 : yv12->y_width;
96 yv12->uv_height =
97 img->y_chroma_shift == 1 ? (1 + yv12->y_height) / 2 : 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
ref_frame_to_vp9_reframe(vpx_ref_frame_type_t frame)136 static VP9_REFFRAME ref_frame_to_vp9_reframe(vpx_ref_frame_type_t frame) {
137 switch (frame) {
138 case VP8_LAST_FRAME: return VP9_LAST_FLAG;
139 case VP8_GOLD_FRAME: return VP9_GOLD_FLAG;
140 case VP8_ALTR_FRAME: return VP9_ALT_FLAG;
141 }
142 assert(0 && "Invalid Reference Frame");
143 return VP9_LAST_FLAG;
144 }
145 #endif // VPX_VP9_VP9_IFACE_COMMON_H_
146