libavcodec/mpegvideoencdsp.c - chromium/third_party/ffmpeg - Git at Google

 /*
  * This file is part of FFmpeg.
  *
  * FFmpeg 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.
  *
  * FFmpeg 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 FFmpeg; if not, write to the Free Software
  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
  */

 #include <assert.h>
 #include <stdint.h>
 #include <string.h>

 #include "config.h"
 #include "libavutil/avassert.h"
 #include "libavutil/attributes.h"
 #include "libavutil/imgutils.h"
 #include "avcodec.h"
 #include "me_cmp.h"
 #include "mpegvideoencdsp.h"

 static int try_8x8basis_c(int16_t rem[64], int16_t weight[64],
                           int16_t basis[64], int scale)
 {
     int i;
     unsigned int sum = 0;

     for (i = 0; i < 8 * 8; i++) {
         int b = rem[i] + ((basis[i] * scale +
                            (1 << (BASIS_SHIFT - RECON_SHIFT - 1))) >>
                           (BASIS_SHIFT - RECON_SHIFT));
         int w = weight[i];
         b >>= RECON_SHIFT;
         av_assert2(-512 < b && b < 512);

         sum += (w * b) * (w * b) >> 4;
     }
     return sum >> 2;
 }

 static void add_8x8basis_c(int16_t rem[64], int16_t basis[64], int scale)
 {
     int i;

     for (i = 0; i < 8 * 8; i++)
         rem[i] += (basis[i] * scale +
                    (1 << (BASIS_SHIFT - RECON_SHIFT - 1))) >>
                   (BASIS_SHIFT - RECON_SHIFT);
 }

 static int pix_sum_c(uint8_t *pix, int line_size)
 {
     int s = 0, i, j;

     for (i = 0; i < 16; i++) {
         for (j = 0; j < 16; j += 8) {
             s   += pix[0];
             s   += pix[1];
             s   += pix[2];
             s   += pix[3];
             s   += pix[4];
             s   += pix[5];
             s   += pix[6];
             s   += pix[7];
             pix += 8;
         }
         pix += line_size - 16;
     }
     return s;
 }

 static int pix_norm1_c(uint8_t *pix, int line_size)
 {
     int s = 0, i, j;
     uint32_t *sq = ff_square_tab + 256;

     for (i = 0; i < 16; i++) {
         for (j = 0; j < 16; j += 8) {
 #if 0
             s += sq[pix[0]];
             s += sq[pix[1]];
             s += sq[pix[2]];
             s += sq[pix[3]];
             s += sq[pix[4]];
             s += sq[pix[5]];
             s += sq[pix[6]];
             s += sq[pix[7]];
 #else
 #if HAVE_FAST_64BIT
             register uint64_t x = *(uint64_t *) pix;
             s += sq[x         & 0xff];
             s += sq[(x >>  8) & 0xff];
             s += sq[(x >> 16) & 0xff];
             s += sq[(x >> 24) & 0xff];
             s += sq[(x >> 32) & 0xff];
             s += sq[(x >> 40) & 0xff];
             s += sq[(x >> 48) & 0xff];
             s += sq[(x >> 56) & 0xff];
 #else
             register uint32_t x = *(uint32_t *) pix;
             s += sq[x         & 0xff];
             s += sq[(x >>  8) & 0xff];
             s += sq[(x >> 16) & 0xff];
             s += sq[(x >> 24) & 0xff];
             x  = *(uint32_t *) (pix + 4);
             s += sq[x         & 0xff];
             s += sq[(x >>  8) & 0xff];
             s += sq[(x >> 16) & 0xff];
             s += sq[(x >> 24) & 0xff];
 #endif
 #endif
             pix += 8;
         }
         pix += line_size - 16;
     }
     return s;
 }

 /* 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 draw_edges_8_c(uint8_t *buf, int wrap, int width, int height,
                            int w, int h, int sides)
 {
     uint8_t *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);
     if (sides & EDGE_BOTTOM)
         for (i = 0; i < h; i++)
             // bottom
             memcpy(last_line + (i + 1) * wrap, last_line, width + w + w);
 }

 /* 2x2 -> 1x1 */
 static void shrink22(uint8_t *dst, int dst_wrap,
                      const uint8_t *src, int src_wrap,
                      int width, int height)
 {
     int w;
     const uint8_t *s1, *s2;
     uint8_t *d;

     for (; height > 0; height--) {
         s1 = src;
         s2 = s1 + src_wrap;
         d = dst;
         for (w = width; w >= 4; w -= 4) {
             d[0] = (s1[0] + s1[1] + s2[0] + s2[1] + 2) >> 2;
             d[1] = (s1[2] + s1[3] + s2[2] + s2[3] + 2) >> 2;
             d[2] = (s1[4] + s1[5] + s2[4] + s2[5] + 2) >> 2;
             d[3] = (s1[6] + s1[7] + s2[6] + s2[7] + 2) >> 2;
             s1 += 8;
             s2 += 8;
             d += 4;
         }
         for (; w > 0; w--) {
             d[0] = (s1[0] + s1[1] + s2[0] + s2[1] + 2) >> 2;
             s1 += 2;
             s2 += 2;
             d++;
         }
         src += 2 * src_wrap;
         dst += dst_wrap;
     }
 }

 /* 4x4 -> 1x1 */
 static void shrink44(uint8_t *dst, int dst_wrap,
                      const uint8_t *src, int src_wrap,
                      int width, int height)
 {
     int w;
     const uint8_t *s1, *s2, *s3, *s4;
     uint8_t *d;

     for (; height > 0; height--) {
         s1 = src;
         s2 = s1 + src_wrap;
         s3 = s2 + src_wrap;
         s4 = s3 + src_wrap;
         d = dst;
         for (w = width; w > 0; w--) {
             d[0] = (s1[0] + s1[1] + s1[2] + s1[3] +
                     s2[0] + s2[1] + s2[2] + s2[3] +
                     s3[0] + s3[1] + s3[2] + s3[3] +
                     s4[0] + s4[1] + s4[2] + s4[3] + 8) >> 4;
             s1 += 4;
             s2 += 4;
             s3 += 4;
             s4 += 4;
             d++;
         }
         src += 4 * src_wrap;
         dst += dst_wrap;
     }
 }

 /* 8x8 -> 1x1 */
 static void shrink88(uint8_t *dst, int dst_wrap,
                      const uint8_t *src, int src_wrap,
                      int width, int height)
 {
     int w, i;

     for (; height > 0; height--) {
         for(w = width;w > 0; w--) {
             int tmp = 0;
             for (i = 0; i < 8; i++) {
                 tmp += src[0] + src[1] + src[2] + src[3] +
                        src[4] + src[5] + src[6] + src[7];
                 src += src_wrap;
             }
             *(dst++) = (tmp + 32) >> 6;
             src += 8 - 8 * src_wrap;
         }
         src += 8 * src_wrap - 8 * width;
         dst += dst_wrap - width;
     }
 }

 av_cold void ff_mpegvideoencdsp_init(MpegvideoEncDSPContext *c,
                                      AVCodecContext *avctx)
 {
     c->try_8x8basis = try_8x8basis_c;
     c->add_8x8basis = add_8x8basis_c;

     c->shrink[0] = av_image_copy_plane;
     c->shrink[1] = shrink22;
     c->shrink[2] = shrink44;
     c->shrink[3] = shrink88;

     c->pix_sum   = pix_sum_c;
     c->pix_norm1 = pix_norm1_c;

     c->draw_edges = draw_edges_8_c;

     if (ARCH_ARM)
         ff_mpegvideoencdsp_init_arm(c, avctx);
     if (ARCH_PPC)
         ff_mpegvideoencdsp_init_ppc(c, avctx);
     if (ARCH_X86)
         ff_mpegvideoencdsp_init_x86(c, avctx);
     if (ARCH_MIPS)
         ff_mpegvideoencdsp_init_mips(c, avctx);
 }
	/*
	* This file is part of FFmpeg.
	*
	* FFmpeg 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.
	*
	* FFmpeg 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 FFmpeg; if not, write to the Free Software
	* Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
	*/

	#include <assert.h>
	#include <stdint.h>
	#include <string.h>

	#include "config.h"
	#include "libavutil/avassert.h"
	#include "libavutil/attributes.h"
	#include "libavutil/imgutils.h"
	#include "avcodec.h"
	#include "me_cmp.h"
	#include "mpegvideoencdsp.h"

	static int try_8x8basis_c(int16_t rem[64], int16_t weight[64],
	int16_t basis[64], int scale)
	{
	int i;
	unsigned int sum = 0;

	for (i = 0; i < 8 * 8; i++) {
	int b = rem[i] + ((basis[i] * scale +
	(1 << (BASIS_SHIFT - RECON_SHIFT - 1))) >>
	(BASIS_SHIFT - RECON_SHIFT));
	int w = weight[i];
	b >>= RECON_SHIFT;
	av_assert2(-512 < b && b < 512);

	sum += (w * b) * (w * b) >> 4;
	}
	return sum >> 2;
	}

	static void add_8x8basis_c(int16_t rem[64], int16_t basis[64], int scale)
	{
	int i;

	for (i = 0; i < 8 * 8; i++)
	rem[i] += (basis[i] * scale +
	(1 << (BASIS_SHIFT - RECON_SHIFT - 1))) >>
	(BASIS_SHIFT - RECON_SHIFT);
	}

	static int pix_sum_c(uint8_t *pix, int line_size)
	{
	int s = 0, i, j;

	for (i = 0; i < 16; i++) {
	for (j = 0; j < 16; j += 8) {
	s += pix[0];
	s += pix[1];
	s += pix[2];
	s += pix[3];
	s += pix[4];
	s += pix[5];
	s += pix[6];
	s += pix[7];
	pix += 8;
	}
	pix += line_size - 16;
	}
	return s;
	}

	static int pix_norm1_c(uint8_t *pix, int line_size)
	{
	int s = 0, i, j;
	uint32_t *sq = ff_square_tab + 256;

	for (i = 0; i < 16; i++) {
	for (j = 0; j < 16; j += 8) {
	#if 0
	s += sq[pix[0]];
	s += sq[pix[1]];
	s += sq[pix[2]];
	s += sq[pix[3]];
	s += sq[pix[4]];
	s += sq[pix[5]];
	s += sq[pix[6]];
	s += sq[pix[7]];
	#else
	#if HAVE_FAST_64BIT
	register uint64_t x = (uint64_t ) pix;
	s += sq[x & 0xff];
	s += sq[(x >> 8) & 0xff];
	s += sq[(x >> 16) & 0xff];
	s += sq[(x >> 24) & 0xff];
	s += sq[(x >> 32) & 0xff];
	s += sq[(x >> 40) & 0xff];
	s += sq[(x >> 48) & 0xff];
	s += sq[(x >> 56) & 0xff];
	#else
	register uint32_t x = (uint32_t ) pix;
	s += sq[x & 0xff];
	s += sq[(x >> 8) & 0xff];
	s += sq[(x >> 16) & 0xff];
	s += sq[(x >> 24) & 0xff];
	x = (uint32_t ) (pix + 4);
	s += sq[x & 0xff];
	s += sq[(x >> 8) & 0xff];
	s += sq[(x >> 16) & 0xff];
	s += sq[(x >> 24) & 0xff];
	#endif
	#endif
	pix += 8;
	}
	pix += line_size - 16;
	}
	return s;
	}

	/* 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 draw_edges_8_c(uint8_t *buf, int wrap, int width, int height,
	int w, int h, int sides)
	{
	uint8_t 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);
	if (sides & EDGE_BOTTOM)
	for (i = 0; i < h; i++)
	// bottom
	memcpy(last_line + (i + 1) * wrap, last_line, width + w + w);
	}

	/* 2x2 -> 1x1 */
	static void shrink22(uint8_t *dst, int dst_wrap,
	const uint8_t *src, int src_wrap,
	int width, int height)
	{
	int w;
	const uint8_t s1, s2;
	uint8_t *d;

	for (; height > 0; height--) {
	s1 = src;
	s2 = s1 + src_wrap;
	d = dst;
	for (w = width; w >= 4; w -= 4) {
	d[0] = (s1[0] + s1[1] + s2[0] + s2[1] + 2) >> 2;
	d[1] = (s1[2] + s1[3] + s2[2] + s2[3] + 2) >> 2;
	d[2] = (s1[4] + s1[5] + s2[4] + s2[5] + 2) >> 2;
	d[3] = (s1[6] + s1[7] + s2[6] + s2[7] + 2) >> 2;
	s1 += 8;
	s2 += 8;
	d += 4;
	}
	for (; w > 0; w--) {
	d[0] = (s1[0] + s1[1] + s2[0] + s2[1] + 2) >> 2;
	s1 += 2;
	s2 += 2;
	d++;
	}
	src += 2 * src_wrap;
	dst += dst_wrap;
	}
	}

	/* 4x4 -> 1x1 */
	static void shrink44(uint8_t *dst, int dst_wrap,
	const uint8_t *src, int src_wrap,
	int width, int height)
	{
	int w;
	const uint8_t s1, s2, s3, s4;
	uint8_t *d;

	for (; height > 0; height--) {
	s1 = src;
	s2 = s1 + src_wrap;
	s3 = s2 + src_wrap;
	s4 = s3 + src_wrap;
	d = dst;
	for (w = width; w > 0; w--) {
	d[0] = (s1[0] + s1[1] + s1[2] + s1[3] +
	s2[0] + s2[1] + s2[2] + s2[3] +
	s3[0] + s3[1] + s3[2] + s3[3] +
	s4[0] + s4[1] + s4[2] + s4[3] + 8) >> 4;
	s1 += 4;
	s2 += 4;
	s3 += 4;
	s4 += 4;
	d++;
	}
	src += 4 * src_wrap;
	dst += dst_wrap;
	}
	}

	/* 8x8 -> 1x1 */
	static void shrink88(uint8_t *dst, int dst_wrap,
	const uint8_t *src, int src_wrap,
	int width, int height)
	{
	int w, i;

	for (; height > 0; height--) {
	for(w = width;w > 0; w--) {
	int tmp = 0;
	for (i = 0; i < 8; i++) {
	tmp += src[0] + src[1] + src[2] + src[3] +
	src[4] + src[5] + src[6] + src[7];
	src += src_wrap;
	}
	*(dst++) = (tmp + 32) >> 6;
	src += 8 - 8 * src_wrap;
	}
	src += 8 * src_wrap - 8 * width;
	dst += dst_wrap - width;
	}
	}

	av_cold void ff_mpegvideoencdsp_init(MpegvideoEncDSPContext *c,
	AVCodecContext *avctx)
	{
	c->try_8x8basis = try_8x8basis_c;
	c->add_8x8basis = add_8x8basis_c;

	c->shrink[0] = av_image_copy_plane;
	c->shrink[1] = shrink22;
	c->shrink[2] = shrink44;
	c->shrink[3] = shrink88;

	c->pix_sum = pix_sum_c;
	c->pix_norm1 = pix_norm1_c;

	c->draw_edges = draw_edges_8_c;

	if (ARCH_ARM)
	ff_mpegvideoencdsp_init_arm(c, avctx);
	if (ARCH_PPC)
	ff_mpegvideoencdsp_init_ppc(c, avctx);
	if (ARCH_X86)
	ff_mpegvideoencdsp_init_x86(c, avctx);
	if (ARCH_MIPS)
	ff_mpegvideoencdsp_init_mips(c, avctx);
	}