Home | History | Annotate | Download | only in vp9
      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 static void yuvconfig2image(vpx_image_t *img, const YV12_BUFFER_CONFIG  *yv12,
     14                             void *user_priv) {
     15   /** vpx_img_wrap() doesn't allow specifying independent strides for
     16     * the Y, U, and V planes, nor other alignment adjustments that
     17     * might be representable by a YV12_BUFFER_CONFIG, so we just
     18     * initialize all the fields.*/
     19   int bps = 12;
     20   if (yv12->uv_height == yv12->y_height) {
     21     if (yv12->uv_width == yv12->y_width) {
     22       img->fmt = VPX_IMG_FMT_I444;
     23       bps = 24;
     24     } else {
     25       img->fmt = VPX_IMG_FMT_I422;
     26       bps = 16;
     27     }
     28   } else {
     29     img->fmt = VPX_IMG_FMT_I420;
     30   }
     31   img->w = yv12->y_stride;
     32   img->h = ALIGN_POWER_OF_TWO(yv12->y_height + 2 * VP9_ENC_BORDER_IN_PIXELS, 3);
     33   img->d_w = yv12->y_crop_width;
     34   img->d_h = yv12->y_crop_height;
     35   img->x_chroma_shift = yv12->uv_width < yv12->y_width;
     36   img->y_chroma_shift = yv12->uv_height < yv12->y_height;
     37   img->planes[VPX_PLANE_Y] = yv12->y_buffer;
     38   img->planes[VPX_PLANE_U] = yv12->u_buffer;
     39   img->planes[VPX_PLANE_V] = yv12->v_buffer;
     40   img->planes[VPX_PLANE_ALPHA] = yv12->alpha_buffer;
     41   img->stride[VPX_PLANE_Y] = yv12->y_stride;
     42   img->stride[VPX_PLANE_U] = yv12->uv_stride;
     43   img->stride[VPX_PLANE_V] = yv12->uv_stride;
     44   img->stride[VPX_PLANE_ALPHA] = yv12->alpha_stride;
     45   img->bps = bps;
     46   img->user_priv = user_priv;
     47   img->img_data = yv12->buffer_alloc;
     48   img->img_data_owner = 0;
     49   img->self_allocd = 0;
     50   img->fb_index = yv12->fb_index;
     51 }
     52 
     53 static vpx_codec_err_t image2yuvconfig(const vpx_image_t *img,
     54                                        YV12_BUFFER_CONFIG *yv12) {
     55   yv12->y_buffer = img->planes[VPX_PLANE_Y];
     56   yv12->u_buffer = img->planes[VPX_PLANE_U];
     57   yv12->v_buffer = img->planes[VPX_PLANE_V];
     58   yv12->alpha_buffer = img->planes[VPX_PLANE_ALPHA];
     59 
     60   yv12->y_crop_width  = img->d_w;
     61   yv12->y_crop_height = img->d_h;
     62   yv12->y_width  = img->d_w;
     63   yv12->y_height = img->d_h;
     64 
     65   yv12->uv_width = img->x_chroma_shift == 1 ? (1 + yv12->y_width) / 2
     66                                             : yv12->y_width;
     67   yv12->uv_height = img->y_chroma_shift == 1 ? (1 + yv12->y_height) / 2
     68                                              : yv12->y_height;
     69 
     70   yv12->alpha_width = yv12->alpha_buffer ? img->d_w : 0;
     71   yv12->alpha_height = yv12->alpha_buffer ? img->d_h : 0;
     72 
     73   yv12->y_stride = img->stride[VPX_PLANE_Y];
     74   yv12->uv_stride = img->stride[VPX_PLANE_U];
     75   yv12->alpha_stride = yv12->alpha_buffer ? img->stride[VPX_PLANE_ALPHA] : 0;
     76 
     77   yv12->border  = (img->stride[VPX_PLANE_Y] - img->w) / 2;
     78 #if CONFIG_ALPHA
     79   // For development purposes, force alpha to hold the same data as Y for now.
     80   yv12->alpha_buffer = yv12->y_buffer;
     81   yv12->alpha_width = yv12->y_width;
     82   yv12->alpha_height = yv12->y_height;
     83   yv12->alpha_stride = yv12->y_stride;
     84 #endif
     85   return VPX_CODEC_OK;
     86 }
     87 
     88 #endif  // VP9_VP9_IFACE_COMMON_H_
     89