Home | History | Annotate | Download | only in generic
      1 /*
      2  *  Copyright (c) 2010 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 
     11 #include <assert.h>
     12 #include "./vpx_config.h"
     13 #include "vpx/vpx_integer.h"
     14 #include "vpx_mem/vpx_mem.h"
     15 #include "vpx_scale/yv12config.h"
     16 
     17 static void extend_plane(uint8_t *const src, int src_stride,
     18                          int width, int height,
     19                          int extend_top, int extend_left,
     20                          int extend_bottom, int extend_right) {
     21   int i;
     22   const int linesize = extend_left + extend_right + width;
     23 
     24   /* copy the left and right most columns out */
     25   uint8_t *src_ptr1 = src;
     26   uint8_t *src_ptr2 = src + width - 1;
     27   uint8_t *dst_ptr1 = src - extend_left;
     28   uint8_t *dst_ptr2 = src + width;
     29 
     30   for (i = 0; i < height; ++i) {
     31     vpx_memset(dst_ptr1, src_ptr1[0], extend_left);
     32     vpx_memset(dst_ptr2, src_ptr2[0], extend_right);
     33     src_ptr1 += src_stride;
     34     src_ptr2 += src_stride;
     35     dst_ptr1 += src_stride;
     36     dst_ptr2 += src_stride;
     37   }
     38 
     39   /* Now copy the top and bottom lines into each line of the respective
     40    * borders
     41    */
     42   src_ptr1 = src - extend_left;
     43   src_ptr2 = src + src_stride * (height - 1) - extend_left;
     44   dst_ptr1 = src + src_stride * -extend_top - extend_left;
     45   dst_ptr2 = src + src_stride * height - extend_left;
     46 
     47   for (i = 0; i < extend_top; ++i) {
     48     vpx_memcpy(dst_ptr1, src_ptr1, linesize);
     49     dst_ptr1 += src_stride;
     50   }
     51 
     52   for (i = 0; i < extend_bottom; ++i) {
     53     vpx_memcpy(dst_ptr2, src_ptr2, linesize);
     54     dst_ptr2 += src_stride;
     55   }
     56 }
     57 
     58 void vp8_yv12_extend_frame_borders_c(YV12_BUFFER_CONFIG *ybf) {
     59   assert(ybf->y_height - ybf->y_crop_height < 16);
     60   assert(ybf->y_width - ybf->y_crop_width < 16);
     61   assert(ybf->y_height - ybf->y_crop_height >= 0);
     62   assert(ybf->y_width - ybf->y_crop_width >= 0);
     63 
     64   extend_plane(ybf->y_buffer, ybf->y_stride,
     65                ybf->y_crop_width, ybf->y_crop_height,
     66                ybf->border, ybf->border,
     67                ybf->border + ybf->y_height - ybf->y_crop_height,
     68                ybf->border + ybf->y_width - ybf->y_crop_width);
     69 
     70   extend_plane(ybf->u_buffer, ybf->uv_stride,
     71                (ybf->y_crop_width + 1) / 2, (ybf->y_crop_height + 1) / 2,
     72                ybf->border / 2, ybf->border / 2,
     73                (ybf->border + ybf->y_height - ybf->y_crop_height + 1) / 2,
     74                (ybf->border + ybf->y_width - ybf->y_crop_width + 1) / 2);
     75 
     76   extend_plane(ybf->v_buffer, ybf->uv_stride,
     77                (ybf->y_crop_width + 1) / 2, (ybf->y_crop_height + 1) / 2,
     78                ybf->border / 2, ybf->border / 2,
     79                (ybf->border + ybf->y_height - ybf->y_crop_height + 1) / 2,
     80                (ybf->border + ybf->y_width - ybf->y_crop_width + 1) / 2);
     81 }
     82 
     83 #if CONFIG_VP9
     84 static void extend_frame(YV12_BUFFER_CONFIG *const ybf,
     85                          int subsampling_x, int subsampling_y,
     86                          int ext_size) {
     87   const int c_w = (ybf->y_crop_width + subsampling_x) >> subsampling_x;
     88   const int c_h = (ybf->y_crop_height + subsampling_y) >> subsampling_y;
     89   const int c_et = ext_size >> subsampling_y;
     90   const int c_el = ext_size >> subsampling_x;
     91   const int c_eb = (ext_size + ybf->y_height - ybf->y_crop_height +
     92                     subsampling_y) >> subsampling_y;
     93   const int c_er = (ext_size + ybf->y_width - ybf->y_crop_width +
     94                     subsampling_x) >> subsampling_x;
     95 
     96   assert(ybf->y_height - ybf->y_crop_height < 16);
     97   assert(ybf->y_width - ybf->y_crop_width < 16);
     98   assert(ybf->y_height - ybf->y_crop_height >= 0);
     99   assert(ybf->y_width - ybf->y_crop_width >= 0);
    100 
    101   extend_plane(ybf->y_buffer, ybf->y_stride,
    102                ybf->y_crop_width, ybf->y_crop_height,
    103                ext_size, ext_size,
    104                ext_size + ybf->y_height - ybf->y_crop_height,
    105                ext_size + ybf->y_width - ybf->y_crop_width);
    106 
    107   extend_plane(ybf->u_buffer, ybf->uv_stride,
    108                c_w, c_h, c_et, c_el, c_eb, c_er);
    109 
    110   extend_plane(ybf->v_buffer, ybf->uv_stride,
    111                c_w, c_h, c_et, c_el, c_eb, c_er);
    112 }
    113 
    114 void vp9_extend_frame_borders_c(YV12_BUFFER_CONFIG *ybf,
    115                                 int subsampling_x, int subsampling_y) {
    116   extend_frame(ybf, subsampling_x, subsampling_y, ybf->border);
    117 }
    118 
    119 void vp9_extend_frame_inner_borders_c(YV12_BUFFER_CONFIG *ybf,
    120                                       int subsampling_x, int subsampling_y) {
    121   const int inner_bw = (ybf->border > VP9INNERBORDERINPIXELS) ?
    122                        VP9INNERBORDERINPIXELS : ybf->border;
    123   extend_frame(ybf, subsampling_x, subsampling_y, inner_bw);
    124 }
    125 #endif  // CONFIG_VP9
    126 
    127 // Copies the source image into the destination image and updates the
    128 // destination's UMV borders.
    129 // Note: The frames are assumed to be identical in size.
    130 void vp8_yv12_copy_frame_c(const YV12_BUFFER_CONFIG *src_ybc,
    131                            YV12_BUFFER_CONFIG *dst_ybc) {
    132   int row;
    133   const uint8_t *src = src_ybc->y_buffer;
    134   uint8_t *dst = dst_ybc->y_buffer;
    135 
    136 #if 0
    137   /* These assertions are valid in the codec, but the libvpx-tester uses
    138    * this code slightly differently.
    139    */
    140   assert(src_ybc->y_width == dst_ybc->y_width);
    141   assert(src_ybc->y_height == dst_ybc->y_height);
    142 #endif
    143 
    144   for (row = 0; row < src_ybc->y_height; ++row) {
    145     vpx_memcpy(dst, src, src_ybc->y_width);
    146     src += src_ybc->y_stride;
    147     dst += dst_ybc->y_stride;
    148   }
    149 
    150   src = src_ybc->u_buffer;
    151   dst = dst_ybc->u_buffer;
    152 
    153   for (row = 0; row < src_ybc->uv_height; ++row) {
    154     vpx_memcpy(dst, src, src_ybc->uv_width);
    155     src += src_ybc->uv_stride;
    156     dst += dst_ybc->uv_stride;
    157   }
    158 
    159   src = src_ybc->v_buffer;
    160   dst = dst_ybc->v_buffer;
    161 
    162   for (row = 0; row < src_ybc->uv_height; ++row) {
    163     vpx_memcpy(dst, src, src_ybc->uv_width);
    164     src += src_ybc->uv_stride;
    165     dst += dst_ybc->uv_stride;
    166   }
    167 
    168   vp8_yv12_extend_frame_borders_c(dst_ybc);
    169 }
    170 
    171 void vpx_yv12_copy_y_c(const YV12_BUFFER_CONFIG *src_ybc,
    172                        YV12_BUFFER_CONFIG *dst_ybc) {
    173   int row;
    174   const uint8_t *src = src_ybc->y_buffer;
    175   uint8_t *dst = dst_ybc->y_buffer;
    176 
    177   for (row = 0; row < src_ybc->y_height; ++row) {
    178     vpx_memcpy(dst, src, src_ybc->y_width);
    179     src += src_ybc->y_stride;
    180     dst += dst_ybc->y_stride;
    181   }
    182 }
    183