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