/* * DSP utils * Copyright (c) 2000, 2001 Fabrice Bellard * Copyright (c) 2002-2004 Michael Niedermayer * * gmc & q-pel & 32/64 bit based MC by Michael Niedermayer * * This file is part of Libav. * * Libav is free software; you can redistribute it and/or * modify it under the terms of the GNU Lesser General Public * License as published by the Free Software Foundation; either * version 2.1 of the License, or (at your option) any later version. * * Libav is distributed in the hope that it will be useful, * but WITHOUT ANY WARRANTY; without even the implied warranty of * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU * Lesser General Public License for more details. * * You should have received a copy of the GNU Lesser General Public * License along with Libav; if not, write to the Free Software * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA */ /** * @file * DSP utils */ #include "bit_depth_template.c" #if BIT_DEPTH == 8 /* draw the edges of width 'w' of an image of size width, height */ // FIXME: Check that this is OK for MPEG-4 interlaced. static void FUNCC(draw_edges)(uint8_t *_buf, int _wrap, int width, int height, int w, int h, int sides) { pixel *buf = (pixel *) _buf; int wrap = _wrap / sizeof(pixel); pixel *ptr = buf, *last_line; int i; /* left and right */ for (i = 0; i < height; i++) { memset(ptr - w, ptr[0], w); memset(ptr + width, ptr[width - 1], w); ptr += wrap; } /* top and bottom + corners */ buf -= w; last_line = buf + (height - 1) * wrap; if (sides & EDGE_TOP) for (i = 0; i < h; i++) // top memcpy(buf - (i + 1) * wrap, buf, (width + w + w) * sizeof(pixel)); if (sides & EDGE_BOTTOM) for (i = 0; i < h; i++) // bottom memcpy(last_line + (i + 1) * wrap, last_line, (width + w + w) * sizeof(pixel)); } #endif static void FUNCC(get_pixels)(int16_t *restrict block, const uint8_t *_pixels, int line_size) { const pixel *pixels = (const pixel *) _pixels; int i; /* read the pixels */ for (i = 0; i < 8; i++) { block[0] = pixels[0]; block[1] = pixels[1]; block[2] = pixels[2]; block[3] = pixels[3]; block[4] = pixels[4]; block[5] = pixels[5]; block[6] = pixels[6]; block[7] = pixels[7]; pixels += line_size / sizeof(pixel); block += 8; } } #if BIT_DEPTH == 8 static void FUNCC(clear_block)(int16_t *block) { memset(block, 0, sizeof(int16_t) * 64); } static void FUNCC(clear_blocks)(int16_t *blocks) { memset(blocks, 0, sizeof(int16_t) * 6 * 64); } #endif #if BIT_DEPTH == 8 #include "hpel_template.c" #endif #define PIXOP2(OPNAME, OP) \ static inline void FUNC(OPNAME ## _no_rnd_pixels8_l2)(uint8_t *dst, \ const uint8_t *src1, \ const uint8_t *src2, \ int dst_stride, \ int src_stride1, \ int src_stride2, \ int h) \ { \ int i; \ \ for (i = 0; i < h; i++) { \ pixel4 a, b; \ a = AV_RN4P(&src1[i * src_stride1]); \ b = AV_RN4P(&src2[i * src_stride2]); \ OP(*((pixel4 *) &dst[i * dst_stride]), \ no_rnd_avg_pixel4(a, b)); \ a = AV_RN4P(&src1[i * src_stride1 + 4 * sizeof(pixel)]); \ b = AV_RN4P(&src2[i * src_stride2 + 4 * sizeof(pixel)]); \ OP(*((pixel4 *) &dst[i * dst_stride + 4 * sizeof(pixel)]), \ no_rnd_avg_pixel4(a, b)); \ } \ } \ \ static inline void FUNC(OPNAME ## _no_rnd_pixels16_l2)(uint8_t *dst, \ const uint8_t *src1, \ const uint8_t *src2, \ int dst_stride, \ int src_stride1, \ int src_stride2, \ int h) \ { \ FUNC(OPNAME ## _no_rnd_pixels8_l2)(dst, src1, src2, dst_stride, \ src_stride1, src_stride2, h); \ FUNC(OPNAME ## _no_rnd_pixels8_l2)(dst + 8 * sizeof(pixel), \ src1 + 8 * sizeof(pixel), \ src2 + 8 * sizeof(pixel), \ dst_stride, src_stride1, \ src_stride2, h); \ } \ \ static inline void FUNC(OPNAME ## _pixels8_l4)(uint8_t *dst, \ const uint8_t *src1, \ const uint8_t *src2, \ const uint8_t *src3, \ const uint8_t *src4, \ int dst_stride, \ int src_stride1, \ int src_stride2, \ int src_stride3, \ int src_stride4, \ int h) \ { \ /* FIXME HIGH BIT DEPTH */ \ int i; \ \ for (i = 0; i < h; i++) { \ uint32_t a, b, c, d, l0, l1, h0, h1; \ a = AV_RN32(&src1[i * src_stride1]); \ b = AV_RN32(&src2[i * src_stride2]); \ c = AV_RN32(&src3[i * src_stride3]); \ d = AV_RN32(&src4[i * src_stride4]); \ l0 = (a & 0x03030303UL) + \ (b & 0x03030303UL) + \ 0x02020202UL; \ h0 = ((a & 0xFCFCFCFCUL) >> 2) + \ ((b & 0xFCFCFCFCUL) >> 2); \ l1 = (c & 0x03030303UL) + \ (d & 0x03030303UL); \ h1 = ((c & 0xFCFCFCFCUL) >> 2) + \ ((d & 0xFCFCFCFCUL) >> 2); \ OP(*((uint32_t *) &dst[i * dst_stride]), \ h0 + h1 + (((l0 + l1) >> 2) & 0x0F0F0F0FUL)); \ a = AV_RN32(&src1[i * src_stride1 + 4]); \ b = AV_RN32(&src2[i * src_stride2 + 4]); \ c = AV_RN32(&src3[i * src_stride3 + 4]); \ d = AV_RN32(&src4[i * src_stride4 + 4]); \ l0 = (a & 0x03030303UL) + \ (b & 0x03030303UL) + \ 0x02020202UL; \ h0 = ((a & 0xFCFCFCFCUL) >> 2) + \ ((b & 0xFCFCFCFCUL) >> 2); \ l1 = (c & 0x03030303UL) + \ (d & 0x03030303UL); \ h1 = ((c & 0xFCFCFCFCUL) >> 2) + \ ((d & 0xFCFCFCFCUL) >> 2); \ OP(*((uint32_t *) &dst[i * dst_stride + 4]), \ h0 + h1 + (((l0 + l1) >> 2) & 0x0F0F0F0FUL)); \ } \ } \ \ static inline void FUNC(OPNAME ## _no_rnd_pixels8_l4)(uint8_t *dst, \ const uint8_t *src1, \ const uint8_t *src2, \ const uint8_t *src3, \ const uint8_t *src4, \ int dst_stride, \ int src_stride1, \ int src_stride2, \ int src_stride3, \ int src_stride4, \ int h) \ { \ /* FIXME HIGH BIT DEPTH */ \ int i; \ \ for (i = 0; i < h; i++) { \ uint32_t a, b, c, d, l0, l1, h0, h1; \ a = AV_RN32(&src1[i * src_stride1]); \ b = AV_RN32(&src2[i * src_stride2]); \ c = AV_RN32(&src3[i * src_stride3]); \ d = AV_RN32(&src4[i * src_stride4]); \ l0 = (a & 0x03030303UL) + \ (b & 0x03030303UL) + \ 0x01010101UL; \ h0 = ((a & 0xFCFCFCFCUL) >> 2) + \ ((b & 0xFCFCFCFCUL) >> 2); \ l1 = (c & 0x03030303UL) + \ (d & 0x03030303UL); \ h1 = ((c & 0xFCFCFCFCUL) >> 2) + \ ((d & 0xFCFCFCFCUL) >> 2); \ OP(*((uint32_t *) &dst[i * dst_stride]), \ h0 + h1 + (((l0 + l1) >> 2) & 0x0F0F0F0FUL)); \ a = AV_RN32(&src1[i * src_stride1 + 4]); \ b = AV_RN32(&src2[i * src_stride2 + 4]); \ c = AV_RN32(&src3[i * src_stride3 + 4]); \ d = AV_RN32(&src4[i * src_stride4 + 4]); \ l0 = (a & 0x03030303UL) + \ (b & 0x03030303UL) + \ 0x01010101UL; \ h0 = ((a & 0xFCFCFCFCUL) >> 2) + \ ((b & 0xFCFCFCFCUL) >> 2); \ l1 = (c & 0x03030303UL) + \ (d & 0x03030303UL); \ h1 = ((c & 0xFCFCFCFCUL) >> 2) + \ ((d & 0xFCFCFCFCUL) >> 2); \ OP(*((uint32_t *) &dst[i * dst_stride + 4]), \ h0 + h1 + (((l0 + l1) >> 2) & 0x0F0F0F0FUL)); \ } \ } \ static inline void FUNC(OPNAME ## _pixels16_l4)(uint8_t *dst, \ const uint8_t *src1, \ const uint8_t *src2, \ const uint8_t *src3, \ const uint8_t *src4, \ int dst_stride, \ int src_stride1, \ int src_stride2, \ int src_stride3, \ int src_stride4, \ int h) \ { \ FUNC(OPNAME ## _pixels8_l4)(dst, src1, src2, src3, src4, dst_stride, \ src_stride1, src_stride2, src_stride3, \ src_stride4, h); \ FUNC(OPNAME ## _pixels8_l4)(dst + 8 * sizeof(pixel), \ src1 + 8 * sizeof(pixel), \ src2 + 8 * sizeof(pixel), \ src3 + 8 * sizeof(pixel), \ src4 + 8 * sizeof(pixel), \ dst_stride, src_stride1, src_stride2, \ src_stride3, src_stride4, h); \ } \ static inline void FUNC(OPNAME ## _no_rnd_pixels16_l4)(uint8_t *dst, \ const uint8_t *src1, \ const uint8_t *src2, \ const uint8_t *src3, \ const uint8_t *src4, \ int dst_stride, \ int src_stride1, \ int src_stride2, \ int src_stride3, \ int src_stride4, \ int h) \ { \ FUNC(OPNAME ## _no_rnd_pixels8_l4)(dst, src1, src2, src3, src4, \ dst_stride, src_stride1, src_stride2, \ src_stride3, src_stride4, h); \ FUNC(OPNAME ## _no_rnd_pixels8_l4)(dst + 8 * sizeof(pixel), \ src1 + 8 * sizeof(pixel), \ src2 + 8 * sizeof(pixel), \ src3 + 8 * sizeof(pixel), \ src4 + 8 * sizeof(pixel), \ dst_stride, src_stride1, src_stride2, \ src_stride3, src_stride4, h); \ } \ \ static inline void FUNCC(OPNAME ## _pixels8_xy2)(uint8_t *block, \ const uint8_t *pixels, \ ptrdiff_t line_size, \ int h) \ { \ /* FIXME HIGH BIT DEPTH */ \ int j; \ \ for (j = 0; j < 2; j++) { \ int i; \ const uint32_t a = AV_RN32(pixels); \ const uint32_t b = AV_RN32(pixels + 1); \ uint32_t l0 = (a & 0x03030303UL) + \ (b & 0x03030303UL) + \ 0x02020202UL; \ uint32_t h0 = ((a & 0xFCFCFCFCUL) >> 2) + \ ((b & 0xFCFCFCFCUL) >> 2); \ uint32_t l1, h1; \ \ pixels += line_size; \ for (i = 0; i < h; i += 2) { \ uint32_t a = AV_RN32(pixels); \ uint32_t b = AV_RN32(pixels + 1); \ l1 = (a & 0x03030303UL) + \ (b & 0x03030303UL); \ h1 = ((a & 0xFCFCFCFCUL) >> 2) + \ ((b & 0xFCFCFCFCUL) >> 2); \ OP(*((uint32_t *) block), \ h0 + h1 + (((l0 + l1) >> 2) & 0x0F0F0F0FUL)); \ pixels += line_size; \ block += line_size; \ a = AV_RN32(pixels); \ b = AV_RN32(pixels + 1); \ l0 = (a & 0x03030303UL) + \ (b & 0x03030303UL) + \ 0x02020202UL; \ h0 = ((a & 0xFCFCFCFCUL) >> 2) + \ ((b & 0xFCFCFCFCUL) >> 2); \ OP(*((uint32_t *) block), \ h0 + h1 + (((l0 + l1) >> 2) & 0x0F0F0F0FUL)); \ pixels += line_size; \ block += line_size; \ } \ pixels += 4 - line_size * (h + 1); \ block += 4 - line_size * h; \ } \ } \ \ CALL_2X_PIXELS(FUNCC(OPNAME ## _pixels16_xy2), \ FUNCC(OPNAME ## _pixels8_xy2), \ 8 * sizeof(pixel)) \ #define op_avg(a, b) a = rnd_avg_pixel4(a, b) #define op_put(a, b) a = b #if BIT_DEPTH == 8 #define put_no_rnd_pixels8_8_c put_pixels8_8_c PIXOP2(avg, op_avg) PIXOP2(put, op_put) #endif #undef op_avg #undef op_put