|  | /* | 
|  | * Copyright (c) 2016, Alliance for Open Media. All rights reserved | 
|  | * | 
|  | * This source code is subject to the terms of the BSD 2 Clause License and | 
|  | * the Alliance for Open Media Patent License 1.0. If the BSD 2 Clause License | 
|  | * was not distributed with this source code in the LICENSE file, you can | 
|  | * obtain it at www.aomedia.org/license/software. If the Alliance for Open | 
|  | * Media Patent License 1.0 was not distributed with this source code in the | 
|  | * PATENTS file, you can obtain it at www.aomedia.org/license/patent. | 
|  | */ | 
|  |  | 
|  | #include <assert.h> | 
|  |  | 
|  | #include "config/aom_config.h" | 
|  | #include "config/aom_scale_rtcd.h" | 
|  |  | 
|  | #include "aom/aom_integer.h" | 
|  | #include "aom_mem/aom_mem.h" | 
|  | #include "aom_ports/mem.h" | 
|  | #include "aom_scale/yv12config.h" | 
|  |  | 
|  | static void extend_plane(uint8_t *const src, int src_stride, int width, | 
|  | int height, int extend_top, int extend_left, | 
|  | int extend_bottom, int extend_right) { | 
|  | int i; | 
|  | const int linesize = extend_left + extend_right + width; | 
|  |  | 
|  | /* copy the left and right most columns out */ | 
|  | uint8_t *src_ptr1 = src; | 
|  | uint8_t *src_ptr2 = src + width - 1; | 
|  | uint8_t *dst_ptr1 = src - extend_left; | 
|  | uint8_t *dst_ptr2 = src + width; | 
|  |  | 
|  | for (i = 0; i < height; ++i) { | 
|  | memset(dst_ptr1, src_ptr1[0], extend_left); | 
|  | memset(dst_ptr2, src_ptr2[0], extend_right); | 
|  | src_ptr1 += src_stride; | 
|  | src_ptr2 += src_stride; | 
|  | dst_ptr1 += src_stride; | 
|  | dst_ptr2 += src_stride; | 
|  | } | 
|  |  | 
|  | /* Now copy the top and bottom lines into each line of the respective | 
|  | * borders | 
|  | */ | 
|  | src_ptr1 = src - extend_left; | 
|  | src_ptr2 = src + src_stride * (height - 1) - extend_left; | 
|  | dst_ptr1 = src + src_stride * -extend_top - extend_left; | 
|  | dst_ptr2 = src + src_stride * height - extend_left; | 
|  |  | 
|  | for (i = 0; i < extend_top; ++i) { | 
|  | memcpy(dst_ptr1, src_ptr1, linesize); | 
|  | dst_ptr1 += src_stride; | 
|  | } | 
|  |  | 
|  | for (i = 0; i < extend_bottom; ++i) { | 
|  | memcpy(dst_ptr2, src_ptr2, linesize); | 
|  | dst_ptr2 += src_stride; | 
|  | } | 
|  | } | 
|  |  | 
|  | static void extend_plane_high(uint8_t *const src8, int src_stride, int width, | 
|  | int height, int extend_top, int extend_left, | 
|  | int extend_bottom, int extend_right) { | 
|  | int i; | 
|  | const int linesize = extend_left + extend_right + width; | 
|  | uint16_t *src = CONVERT_TO_SHORTPTR(src8); | 
|  |  | 
|  | /* copy the left and right most columns out */ | 
|  | uint16_t *src_ptr1 = src; | 
|  | uint16_t *src_ptr2 = src + width - 1; | 
|  | uint16_t *dst_ptr1 = src - extend_left; | 
|  | uint16_t *dst_ptr2 = src + width; | 
|  |  | 
|  | for (i = 0; i < height; ++i) { | 
|  | aom_memset16(dst_ptr1, src_ptr1[0], extend_left); | 
|  | aom_memset16(dst_ptr2, src_ptr2[0], extend_right); | 
|  | src_ptr1 += src_stride; | 
|  | src_ptr2 += src_stride; | 
|  | dst_ptr1 += src_stride; | 
|  | dst_ptr2 += src_stride; | 
|  | } | 
|  |  | 
|  | /* Now copy the top and bottom lines into each line of the respective | 
|  | * borders | 
|  | */ | 
|  | src_ptr1 = src - extend_left; | 
|  | src_ptr2 = src + src_stride * (height - 1) - extend_left; | 
|  | dst_ptr1 = src + src_stride * -extend_top - extend_left; | 
|  | dst_ptr2 = src + src_stride * height - extend_left; | 
|  |  | 
|  | for (i = 0; i < extend_top; ++i) { | 
|  | memcpy(dst_ptr1, src_ptr1, linesize * sizeof(uint16_t)); | 
|  | dst_ptr1 += src_stride; | 
|  | } | 
|  |  | 
|  | for (i = 0; i < extend_bottom; ++i) { | 
|  | memcpy(dst_ptr2, src_ptr2, linesize * sizeof(uint16_t)); | 
|  | dst_ptr2 += src_stride; | 
|  | } | 
|  | } | 
|  |  | 
|  | void aom_yv12_extend_frame_borders_c(YV12_BUFFER_CONFIG *ybf, | 
|  | const int num_planes) { | 
|  | assert(ybf->border % 2 == 0); | 
|  | assert(ybf->y_height - ybf->y_crop_height < 16); | 
|  | assert(ybf->y_width - ybf->y_crop_width < 16); | 
|  | assert(ybf->y_height - ybf->y_crop_height >= 0); | 
|  | assert(ybf->y_width - ybf->y_crop_width >= 0); | 
|  |  | 
|  | if (ybf->flags & YV12_FLAG_HIGHBITDEPTH) { | 
|  | for (int plane = 0; plane < num_planes; ++plane) { | 
|  | const int is_uv = plane > 0; | 
|  | const int plane_border = ybf->border >> is_uv; | 
|  | extend_plane_high( | 
|  | ybf->buffers[plane], ybf->strides[is_uv], ybf->crop_widths[is_uv], | 
|  | ybf->crop_heights[is_uv], plane_border, plane_border, | 
|  | plane_border + ybf->heights[is_uv] - ybf->crop_heights[is_uv], | 
|  | plane_border + ybf->widths[is_uv] - ybf->crop_widths[is_uv]); | 
|  | } | 
|  | return; | 
|  | } | 
|  | for (int plane = 0; plane < num_planes; ++plane) { | 
|  | const int is_uv = plane > 0; | 
|  | const int plane_border = ybf->border >> is_uv; | 
|  | extend_plane(ybf->buffers[plane], ybf->strides[is_uv], | 
|  | ybf->crop_widths[is_uv], ybf->crop_heights[is_uv], | 
|  | plane_border, plane_border, | 
|  | plane_border + ybf->heights[is_uv] - ybf->crop_heights[is_uv], | 
|  | plane_border + ybf->widths[is_uv] - ybf->crop_widths[is_uv]); | 
|  | } | 
|  | } | 
|  |  | 
|  | static void extend_frame(YV12_BUFFER_CONFIG *const ybf, int ext_size, | 
|  | const int num_planes) { | 
|  | const int ss_x = ybf->uv_width < ybf->y_width; | 
|  | const int ss_y = ybf->uv_height < ybf->y_height; | 
|  |  | 
|  | assert(ybf->y_height - ybf->y_crop_height < 16); | 
|  | assert(ybf->y_width - ybf->y_crop_width < 16); | 
|  | assert(ybf->y_height - ybf->y_crop_height >= 0); | 
|  | assert(ybf->y_width - ybf->y_crop_width >= 0); | 
|  |  | 
|  | if (ybf->flags & YV12_FLAG_HIGHBITDEPTH) { | 
|  | for (int plane = 0; plane < num_planes; ++plane) { | 
|  | const int is_uv = plane > 0; | 
|  | const int top = ext_size >> (is_uv ? ss_y : 0); | 
|  | const int left = ext_size >> (is_uv ? ss_x : 0); | 
|  | const int bottom = top + ybf->heights[is_uv] - ybf->crop_heights[is_uv]; | 
|  | const int right = left + ybf->widths[is_uv] - ybf->crop_widths[is_uv]; | 
|  | extend_plane_high(ybf->buffers[plane], ybf->strides[is_uv], | 
|  | ybf->crop_widths[is_uv], ybf->crop_heights[is_uv], top, | 
|  | left, bottom, right); | 
|  | } | 
|  | return; | 
|  | } | 
|  | for (int plane = 0; plane < num_planes; ++plane) { | 
|  | const int is_uv = plane > 0; | 
|  | const int top = ext_size >> (is_uv ? ss_y : 0); | 
|  | const int left = ext_size >> (is_uv ? ss_x : 0); | 
|  | const int bottom = top + ybf->heights[is_uv] - ybf->crop_heights[is_uv]; | 
|  | const int right = left + ybf->widths[is_uv] - ybf->crop_widths[is_uv]; | 
|  | extend_plane(ybf->buffers[plane], ybf->strides[is_uv], | 
|  | ybf->crop_widths[is_uv], ybf->crop_heights[is_uv], top, left, | 
|  | bottom, right); | 
|  | } | 
|  | } | 
|  |  | 
|  | void aom_extend_frame_borders_c(YV12_BUFFER_CONFIG *ybf, const int num_planes) { | 
|  | extend_frame(ybf, ybf->border, num_planes); | 
|  | } | 
|  |  | 
|  | void aom_extend_frame_inner_borders_c(YV12_BUFFER_CONFIG *ybf, | 
|  | const int num_planes) { | 
|  | const int inner_bw = (ybf->border > AOMINNERBORDERINPIXELS) | 
|  | ? AOMINNERBORDERINPIXELS | 
|  | : ybf->border; | 
|  | extend_frame(ybf, inner_bw, num_planes); | 
|  | } | 
|  |  | 
|  | void aom_extend_frame_borders_y_c(YV12_BUFFER_CONFIG *ybf) { | 
|  | int ext_size = ybf->border; | 
|  | assert(ybf->y_height - ybf->y_crop_height < 16); | 
|  | assert(ybf->y_width - ybf->y_crop_width < 16); | 
|  | assert(ybf->y_height - ybf->y_crop_height >= 0); | 
|  | assert(ybf->y_width - ybf->y_crop_width >= 0); | 
|  |  | 
|  | if (ybf->flags & YV12_FLAG_HIGHBITDEPTH) { | 
|  | extend_plane_high(ybf->y_buffer, ybf->y_stride, ybf->y_crop_width, | 
|  | ybf->y_crop_height, ext_size, ext_size, | 
|  | ext_size + ybf->y_height - ybf->y_crop_height, | 
|  | ext_size + ybf->y_width - ybf->y_crop_width); | 
|  | return; | 
|  | } | 
|  | extend_plane(ybf->y_buffer, ybf->y_stride, ybf->y_crop_width, | 
|  | ybf->y_crop_height, ext_size, ext_size, | 
|  | ext_size + ybf->y_height - ybf->y_crop_height, | 
|  | ext_size + ybf->y_width - ybf->y_crop_width); | 
|  | } | 
|  |  | 
|  | static void memcpy_short_addr(uint8_t *dst8, const uint8_t *src8, int num) { | 
|  | uint16_t *dst = CONVERT_TO_SHORTPTR(dst8); | 
|  | uint16_t *src = CONVERT_TO_SHORTPTR(src8); | 
|  | memcpy(dst, src, num * sizeof(uint16_t)); | 
|  | } | 
|  |  | 
|  | // Copies the source image into the destination image and updates the | 
|  | // destination's UMV borders. | 
|  | // Note: The frames are assumed to be identical in size. | 
|  | void aom_yv12_copy_frame_c(const YV12_BUFFER_CONFIG *src_bc, | 
|  | YV12_BUFFER_CONFIG *dst_bc, const int num_planes) { | 
|  | #if 0 | 
|  | /* These assertions are valid in the codec, but the libaom-tester uses | 
|  | * this code slightly differently. | 
|  | */ | 
|  | assert(src_bc->y_width == dst_bc->y_width); | 
|  | assert(src_bc->y_height == dst_bc->y_height); | 
|  | #endif | 
|  |  | 
|  | assert((src_bc->flags & YV12_FLAG_HIGHBITDEPTH) == | 
|  | (dst_bc->flags & YV12_FLAG_HIGHBITDEPTH)); | 
|  |  | 
|  | if (src_bc->flags & YV12_FLAG_HIGHBITDEPTH) { | 
|  | for (int plane = 0; plane < num_planes; ++plane) { | 
|  | const uint8_t *plane_src = src_bc->buffers[plane]; | 
|  | uint8_t *plane_dst = dst_bc->buffers[plane]; | 
|  | const int is_uv = plane > 0; | 
|  |  | 
|  | for (int row = 0; row < src_bc->heights[is_uv]; ++row) { | 
|  | memcpy_short_addr(plane_dst, plane_src, src_bc->widths[is_uv]); | 
|  | plane_src += src_bc->strides[is_uv]; | 
|  | plane_dst += dst_bc->strides[is_uv]; | 
|  | } | 
|  | } | 
|  | aom_yv12_extend_frame_borders_c(dst_bc, num_planes); | 
|  | return; | 
|  | } | 
|  | for (int plane = 0; plane < num_planes; ++plane) { | 
|  | const uint8_t *plane_src = src_bc->buffers[plane]; | 
|  | uint8_t *plane_dst = dst_bc->buffers[plane]; | 
|  | const int is_uv = plane > 0; | 
|  |  | 
|  | for (int row = 0; row < src_bc->heights[is_uv]; ++row) { | 
|  | memcpy(plane_dst, plane_src, src_bc->widths[is_uv]); | 
|  | plane_src += src_bc->strides[is_uv]; | 
|  | plane_dst += dst_bc->strides[is_uv]; | 
|  | } | 
|  | } | 
|  | aom_yv12_extend_frame_borders_c(dst_bc, num_planes); | 
|  | } | 
|  |  | 
|  | void aom_yv12_copy_y_c(const YV12_BUFFER_CONFIG *src_ybc, | 
|  | YV12_BUFFER_CONFIG *dst_ybc) { | 
|  | int row; | 
|  | const uint8_t *src = src_ybc->y_buffer; | 
|  | uint8_t *dst = dst_ybc->y_buffer; | 
|  |  | 
|  | if (src_ybc->flags & YV12_FLAG_HIGHBITDEPTH) { | 
|  | const uint16_t *src16 = CONVERT_TO_SHORTPTR(src); | 
|  | uint16_t *dst16 = CONVERT_TO_SHORTPTR(dst); | 
|  | for (row = 0; row < src_ybc->y_height; ++row) { | 
|  | memcpy(dst16, src16, src_ybc->y_width * sizeof(uint16_t)); | 
|  | src16 += src_ybc->y_stride; | 
|  | dst16 += dst_ybc->y_stride; | 
|  | } | 
|  | return; | 
|  | } | 
|  |  | 
|  | for (row = 0; row < src_ybc->y_height; ++row) { | 
|  | memcpy(dst, src, src_ybc->y_width); | 
|  | src += src_ybc->y_stride; | 
|  | dst += dst_ybc->y_stride; | 
|  | } | 
|  | } | 
|  |  | 
|  | void aom_yv12_copy_u_c(const YV12_BUFFER_CONFIG *src_bc, | 
|  | YV12_BUFFER_CONFIG *dst_bc) { | 
|  | int row; | 
|  | const uint8_t *src = src_bc->u_buffer; | 
|  | uint8_t *dst = dst_bc->u_buffer; | 
|  |  | 
|  | if (src_bc->flags & YV12_FLAG_HIGHBITDEPTH) { | 
|  | const uint16_t *src16 = CONVERT_TO_SHORTPTR(src); | 
|  | uint16_t *dst16 = CONVERT_TO_SHORTPTR(dst); | 
|  | for (row = 0; row < src_bc->uv_height; ++row) { | 
|  | memcpy(dst16, src16, src_bc->uv_width * sizeof(uint16_t)); | 
|  | src16 += src_bc->uv_stride; | 
|  | dst16 += dst_bc->uv_stride; | 
|  | } | 
|  | return; | 
|  | } | 
|  |  | 
|  | for (row = 0; row < src_bc->uv_height; ++row) { | 
|  | memcpy(dst, src, src_bc->uv_width); | 
|  | src += src_bc->uv_stride; | 
|  | dst += dst_bc->uv_stride; | 
|  | } | 
|  | } | 
|  |  | 
|  | void aom_yv12_copy_v_c(const YV12_BUFFER_CONFIG *src_bc, | 
|  | YV12_BUFFER_CONFIG *dst_bc) { | 
|  | int row; | 
|  | const uint8_t *src = src_bc->v_buffer; | 
|  | uint8_t *dst = dst_bc->v_buffer; | 
|  |  | 
|  | if (src_bc->flags & YV12_FLAG_HIGHBITDEPTH) { | 
|  | const uint16_t *src16 = CONVERT_TO_SHORTPTR(src); | 
|  | uint16_t *dst16 = CONVERT_TO_SHORTPTR(dst); | 
|  | for (row = 0; row < src_bc->uv_height; ++row) { | 
|  | memcpy(dst16, src16, src_bc->uv_width * sizeof(uint16_t)); | 
|  | src16 += src_bc->uv_stride; | 
|  | dst16 += dst_bc->uv_stride; | 
|  | } | 
|  | return; | 
|  | } | 
|  |  | 
|  | for (row = 0; row < src_bc->uv_height; ++row) { | 
|  | memcpy(dst, src, src_bc->uv_width); | 
|  | src += src_bc->uv_stride; | 
|  | dst += dst_bc->uv_stride; | 
|  | } | 
|  | } | 
|  |  | 
|  | void aom_yv12_partial_copy_y_c(const YV12_BUFFER_CONFIG *src_ybc, | 
|  | YV12_BUFFER_CONFIG *dst_ybc, int hstart, | 
|  | int hend, int vstart, int vend) { | 
|  | int row; | 
|  | const uint8_t *src = src_ybc->y_buffer; | 
|  | uint8_t *dst = dst_ybc->y_buffer; | 
|  |  | 
|  | if (src_ybc->flags & YV12_FLAG_HIGHBITDEPTH) { | 
|  | const uint16_t *src16 = | 
|  | CONVERT_TO_SHORTPTR(src + vstart * src_ybc->y_stride + hstart); | 
|  | uint16_t *dst16 = | 
|  | CONVERT_TO_SHORTPTR(dst + vstart * dst_ybc->y_stride + hstart); | 
|  | for (row = vstart; row < vend; ++row) { | 
|  | memcpy(dst16, src16, (hend - hstart) * sizeof(uint16_t)); | 
|  | src16 += src_ybc->y_stride; | 
|  | dst16 += dst_ybc->y_stride; | 
|  | } | 
|  | return; | 
|  | } | 
|  | src = (src + vstart * src_ybc->y_stride + hstart); | 
|  | dst = (dst + vstart * dst_ybc->y_stride + hstart); | 
|  |  | 
|  | for (row = vstart; row < vend; ++row) { | 
|  | memcpy(dst, src, (hend - hstart)); | 
|  | src += src_ybc->y_stride; | 
|  | dst += dst_ybc->y_stride; | 
|  | } | 
|  | } | 
|  |  | 
|  | void aom_yv12_partial_copy_u_c(const YV12_BUFFER_CONFIG *src_bc, | 
|  | YV12_BUFFER_CONFIG *dst_bc, int hstart, int hend, | 
|  | int vstart, int vend) { | 
|  | int row; | 
|  | const uint8_t *src = src_bc->u_buffer; | 
|  | uint8_t *dst = dst_bc->u_buffer; | 
|  |  | 
|  | if (src_bc->flags & YV12_FLAG_HIGHBITDEPTH) { | 
|  | const uint16_t *src16 = | 
|  | CONVERT_TO_SHORTPTR(src + vstart * src_bc->uv_stride + hstart); | 
|  | uint16_t *dst16 = | 
|  | CONVERT_TO_SHORTPTR(dst + vstart * dst_bc->uv_stride + hstart); | 
|  | for (row = vstart; row < vend; ++row) { | 
|  | memcpy(dst16, src16, (hend - hstart) * sizeof(uint16_t)); | 
|  | src16 += src_bc->uv_stride; | 
|  | dst16 += dst_bc->uv_stride; | 
|  | } | 
|  | return; | 
|  | } | 
|  |  | 
|  | src = (src + vstart * src_bc->uv_stride + hstart); | 
|  | dst = (dst + vstart * dst_bc->uv_stride + hstart); | 
|  |  | 
|  | for (row = vstart; row < vend; ++row) { | 
|  | memcpy(dst, src, (hend - hstart)); | 
|  | src += src_bc->uv_stride; | 
|  | dst += dst_bc->uv_stride; | 
|  | } | 
|  | } | 
|  |  | 
|  | void aom_yv12_partial_copy_v_c(const YV12_BUFFER_CONFIG *src_bc, | 
|  | YV12_BUFFER_CONFIG *dst_bc, int hstart, int hend, | 
|  | int vstart, int vend) { | 
|  | int row; | 
|  | const uint8_t *src = src_bc->v_buffer; | 
|  | uint8_t *dst = dst_bc->v_buffer; | 
|  |  | 
|  | if (src_bc->flags & YV12_FLAG_HIGHBITDEPTH) { | 
|  | const uint16_t *src16 = | 
|  | CONVERT_TO_SHORTPTR(src + vstart * src_bc->uv_stride + hstart); | 
|  | uint16_t *dst16 = | 
|  | CONVERT_TO_SHORTPTR(dst + vstart * dst_bc->uv_stride + hstart); | 
|  | for (row = vstart; row < vend; ++row) { | 
|  | memcpy(dst16, src16, (hend - hstart) * sizeof(uint16_t)); | 
|  | src16 += src_bc->uv_stride; | 
|  | dst16 += dst_bc->uv_stride; | 
|  | } | 
|  | return; | 
|  | } | 
|  |  | 
|  | src = (src + vstart * src_bc->uv_stride + hstart); | 
|  | dst = (dst + vstart * dst_bc->uv_stride + hstart); | 
|  |  | 
|  | for (row = vstart; row < vend; ++row) { | 
|  | memcpy(dst, src, (hend - hstart)); | 
|  | src += src_bc->uv_stride; | 
|  | dst += dst_bc->uv_stride; | 
|  | } | 
|  | } |