/src/aom/av1/encoder/extend.c
Line | Count | Source (jump to first uncovered line) |
1 | | /* |
2 | | * Copyright (c) 2016, Alliance for Open Media. All rights reserved |
3 | | * |
4 | | * This source code is subject to the terms of the BSD 2 Clause License and |
5 | | * the Alliance for Open Media Patent License 1.0. If the BSD 2 Clause License |
6 | | * was not distributed with this source code in the LICENSE file, you can |
7 | | * obtain it at www.aomedia.org/license/software. If the Alliance for Open |
8 | | * Media Patent License 1.0 was not distributed with this source code in the |
9 | | * PATENTS file, you can obtain it at www.aomedia.org/license/patent. |
10 | | */ |
11 | | |
12 | | #include <assert.h> |
13 | | |
14 | | #include "aom_dsp/aom_dsp_common.h" |
15 | | #include "aom_mem/aom_mem.h" |
16 | | #include "aom_ports/mem.h" |
17 | | |
18 | | #include "av1/common/common.h" |
19 | | #include "av1/encoder/extend.h" |
20 | | |
21 | | static void copy_and_extend_plane(const uint8_t *src, int src_pitch, |
22 | | uint8_t *dst, int dst_pitch, int w, int h, |
23 | | int extend_top, int extend_left, |
24 | 3.78k | int extend_bottom, int extend_right) { |
25 | 3.78k | int i, linesize; |
26 | | |
27 | | // copy the left and right most columns out |
28 | 3.78k | const uint8_t *src_ptr1 = src; |
29 | 3.78k | const uint8_t *src_ptr2 = src + w - 1; |
30 | 3.78k | uint8_t *dst_ptr1 = dst - extend_left; |
31 | 3.78k | uint8_t *dst_ptr2 = dst + w; |
32 | | |
33 | 364k | for (i = 0; i < h; i++) { |
34 | 360k | memset(dst_ptr1, src_ptr1[0], extend_left); |
35 | 360k | memcpy(dst_ptr1 + extend_left, src_ptr1, w); |
36 | 360k | memset(dst_ptr2, src_ptr2[0], extend_right); |
37 | 360k | src_ptr1 += src_pitch; |
38 | 360k | src_ptr2 += src_pitch; |
39 | 360k | dst_ptr1 += dst_pitch; |
40 | 360k | dst_ptr2 += dst_pitch; |
41 | 360k | } |
42 | | |
43 | | // Now copy the top and bottom lines into each line of the respective |
44 | | // borders |
45 | 3.78k | src_ptr1 = dst - extend_left; |
46 | 3.78k | src_ptr2 = dst + dst_pitch * (h - 1) - extend_left; |
47 | 3.78k | dst_ptr1 = dst + dst_pitch * (-extend_top) - extend_left; |
48 | 3.78k | dst_ptr2 = dst + dst_pitch * (h)-extend_left; |
49 | 3.78k | linesize = extend_left + extend_right + w; |
50 | 3.78k | assert(linesize <= dst_pitch); |
51 | | |
52 | 165k | for (i = 0; i < extend_top; i++) { |
53 | 161k | memcpy(dst_ptr1, src_ptr1, linesize); |
54 | 161k | dst_ptr1 += dst_pitch; |
55 | 161k | } |
56 | | |
57 | 165k | for (i = 0; i < extend_bottom; i++) { |
58 | 162k | memcpy(dst_ptr2, src_ptr2, linesize); |
59 | 162k | dst_ptr2 += dst_pitch; |
60 | 162k | } |
61 | 3.78k | } |
62 | | |
63 | | static void highbd_copy_and_extend_plane(const uint8_t *src8, int src_pitch, |
64 | | uint8_t *dst8, int dst_pitch, int w, |
65 | | int h, int extend_top, int extend_left, |
66 | 0 | int extend_bottom, int extend_right) { |
67 | 0 | int i, linesize; |
68 | 0 | uint16_t *src = CONVERT_TO_SHORTPTR(src8); |
69 | 0 | uint16_t *dst = CONVERT_TO_SHORTPTR(dst8); |
70 | | |
71 | | // copy the left and right most columns out |
72 | 0 | const uint16_t *src_ptr1 = src; |
73 | 0 | const uint16_t *src_ptr2 = src + w - 1; |
74 | 0 | uint16_t *dst_ptr1 = dst - extend_left; |
75 | 0 | uint16_t *dst_ptr2 = dst + w; |
76 | |
|
77 | 0 | for (i = 0; i < h; i++) { |
78 | 0 | aom_memset16(dst_ptr1, src_ptr1[0], extend_left); |
79 | 0 | memcpy(dst_ptr1 + extend_left, src_ptr1, w * sizeof(src_ptr1[0])); |
80 | 0 | aom_memset16(dst_ptr2, src_ptr2[0], extend_right); |
81 | 0 | src_ptr1 += src_pitch; |
82 | 0 | src_ptr2 += src_pitch; |
83 | 0 | dst_ptr1 += dst_pitch; |
84 | 0 | dst_ptr2 += dst_pitch; |
85 | 0 | } |
86 | | |
87 | | // Now copy the top and bottom lines into each line of the respective |
88 | | // borders |
89 | 0 | src_ptr1 = dst - extend_left; |
90 | 0 | src_ptr2 = dst + dst_pitch * (h - 1) - extend_left; |
91 | 0 | dst_ptr1 = dst + dst_pitch * (-extend_top) - extend_left; |
92 | 0 | dst_ptr2 = dst + dst_pitch * (h)-extend_left; |
93 | 0 | linesize = extend_left + extend_right + w; |
94 | 0 | assert(linesize <= dst_pitch); |
95 | |
|
96 | 0 | for (i = 0; i < extend_top; i++) { |
97 | 0 | memcpy(dst_ptr1, src_ptr1, linesize * sizeof(src_ptr1[0])); |
98 | 0 | dst_ptr1 += dst_pitch; |
99 | 0 | } |
100 | |
|
101 | 0 | for (i = 0; i < extend_bottom; i++) { |
102 | 0 | memcpy(dst_ptr2, src_ptr2, linesize * sizeof(src_ptr2[0])); |
103 | 0 | dst_ptr2 += dst_pitch; |
104 | 0 | } |
105 | 0 | } |
106 | | |
107 | | void av1_copy_and_extend_frame(const YV12_BUFFER_CONFIG *src, |
108 | 1.26k | YV12_BUFFER_CONFIG *dst) { |
109 | | // Extend src frame in buffer |
110 | 1.26k | const int et_y = dst->border; |
111 | 1.26k | const int el_y = dst->border; |
112 | 1.26k | const int er_y = |
113 | 1.26k | AOMMAX(src->y_width + dst->border, ALIGN_POWER_OF_TWO(src->y_width, 6)) - |
114 | 1.26k | src->y_crop_width; |
115 | 1.26k | const int eb_y = AOMMAX(src->y_height + dst->border, |
116 | 1.26k | ALIGN_POWER_OF_TWO(src->y_height, 6)) - |
117 | 1.26k | src->y_crop_height; |
118 | 1.26k | const int uv_width_subsampling = src->subsampling_x; |
119 | 1.26k | const int uv_height_subsampling = src->subsampling_y; |
120 | 1.26k | const int et_uv = et_y >> uv_height_subsampling; |
121 | 1.26k | const int el_uv = el_y >> uv_width_subsampling; |
122 | 1.26k | const int eb_uv = eb_y >> uv_height_subsampling; |
123 | 1.26k | const int er_uv = er_y >> uv_width_subsampling; |
124 | | |
125 | 1.26k | if (src->flags & YV12_FLAG_HIGHBITDEPTH) { |
126 | 0 | highbd_copy_and_extend_plane(src->y_buffer, src->y_stride, dst->y_buffer, |
127 | 0 | dst->y_stride, src->y_crop_width, |
128 | 0 | src->y_crop_height, et_y, el_y, eb_y, er_y); |
129 | 0 | if (!src->monochrome) { |
130 | 0 | highbd_copy_and_extend_plane( |
131 | 0 | src->u_buffer, src->uv_stride, dst->u_buffer, dst->uv_stride, |
132 | 0 | src->uv_crop_width, src->uv_crop_height, et_uv, el_uv, eb_uv, er_uv); |
133 | 0 | highbd_copy_and_extend_plane( |
134 | 0 | src->v_buffer, src->uv_stride, dst->v_buffer, dst->uv_stride, |
135 | 0 | src->uv_crop_width, src->uv_crop_height, et_uv, el_uv, eb_uv, er_uv); |
136 | 0 | } |
137 | 0 | return; |
138 | 0 | } |
139 | | |
140 | 1.26k | copy_and_extend_plane(src->y_buffer, src->y_stride, dst->y_buffer, |
141 | 1.26k | dst->y_stride, src->y_crop_width, src->y_crop_height, |
142 | 1.26k | et_y, el_y, eb_y, er_y); |
143 | 1.26k | if (!src->monochrome) { |
144 | 1.26k | copy_and_extend_plane(src->u_buffer, src->uv_stride, dst->u_buffer, |
145 | 1.26k | dst->uv_stride, src->uv_crop_width, |
146 | 1.26k | src->uv_crop_height, et_uv, el_uv, eb_uv, er_uv); |
147 | 1.26k | copy_and_extend_plane(src->v_buffer, src->uv_stride, dst->v_buffer, |
148 | 1.26k | dst->uv_stride, src->uv_crop_width, |
149 | 1.26k | src->uv_crop_height, et_uv, el_uv, eb_uv, er_uv); |
150 | 1.26k | } |
151 | 1.26k | } |