FFmpeg
 All Data Structures Namespaces Files Functions Variables Typedefs Enumerations Enumerator Macros Groups Pages
mpegvideoencdsp.c
Go to the documentation of this file.
1 /*
2  * This file is part of FFmpeg.
3  *
4  * FFmpeg is free software; you can redistribute it and/or
5  * modify it under the terms of the GNU Lesser General Public
6  * License as published by the Free Software Foundation; either
7  * version 2.1 of the License, or (at your option) any later version.
8  *
9  * FFmpeg is distributed in the hope that it will be useful,
10  * but WITHOUT ANY WARRANTY; without even the implied warranty of
11  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
12  * Lesser General Public License for more details.
13  *
14  * You should have received a copy of the GNU Lesser General Public
15  * License along with FFmpeg; if not, write to the Free Software
16  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
17  */
18 
19 #include <assert.h>
20 #include <stdint.h>
21 #include <string.h>
22 
23 #include "config.h"
24 #include "libavutil/avassert.h"
25 #include "libavutil/attributes.h"
26 #include "libavutil/imgutils.h"
27 #include "avcodec.h"
28 #include "imgconvert.h"
29 #include "me_cmp.h"
30 #include "mpegvideoencdsp.h"
31 
32 static int try_8x8basis_c(int16_t rem[64], int16_t weight[64],
33  int16_t basis[64], int scale)
34 {
35  int i;
36  unsigned int sum = 0;
37 
38  for (i = 0; i < 8 * 8; i++) {
39  int b = rem[i] + ((basis[i] * scale +
40  (1 << (BASIS_SHIFT - RECON_SHIFT - 1))) >>
42  int w = weight[i];
43  b >>= RECON_SHIFT;
44  av_assert2(-512 < b && b < 512);
45 
46  sum += (w * b) * (w * b) >> 4;
47  }
48  return sum >> 2;
49 }
50 
51 static void add_8x8basis_c(int16_t rem[64], int16_t basis[64], int scale)
52 {
53  int i;
54 
55  for (i = 0; i < 8 * 8; i++)
56  rem[i] += (basis[i] * scale +
57  (1 << (BASIS_SHIFT - RECON_SHIFT - 1))) >>
59 }
60 
61 static int pix_sum_c(uint8_t *pix, int line_size)
62 {
63  int s = 0, i, j;
64 
65  for (i = 0; i < 16; i++) {
66  for (j = 0; j < 16; j += 8) {
67  s += pix[0];
68  s += pix[1];
69  s += pix[2];
70  s += pix[3];
71  s += pix[4];
72  s += pix[5];
73  s += pix[6];
74  s += pix[7];
75  pix += 8;
76  }
77  pix += line_size - 16;
78  }
79  return s;
80 }
81 
82 static int pix_norm1_c(uint8_t *pix, int line_size)
83 {
84  int s = 0, i, j;
85  uint32_t *sq = ff_square_tab + 256;
86 
87  for (i = 0; i < 16; i++) {
88  for (j = 0; j < 16; j += 8) {
89 #if 0
90  s += sq[pix[0]];
91  s += sq[pix[1]];
92  s += sq[pix[2]];
93  s += sq[pix[3]];
94  s += sq[pix[4]];
95  s += sq[pix[5]];
96  s += sq[pix[6]];
97  s += sq[pix[7]];
98 #else
99 #if HAVE_FAST_64BIT
100  register uint64_t x = *(uint64_t *) pix;
101  s += sq[x & 0xff];
102  s += sq[(x >> 8) & 0xff];
103  s += sq[(x >> 16) & 0xff];
104  s += sq[(x >> 24) & 0xff];
105  s += sq[(x >> 32) & 0xff];
106  s += sq[(x >> 40) & 0xff];
107  s += sq[(x >> 48) & 0xff];
108  s += sq[(x >> 56) & 0xff];
109 #else
110  register uint32_t x = *(uint32_t *) pix;
111  s += sq[x & 0xff];
112  s += sq[(x >> 8) & 0xff];
113  s += sq[(x >> 16) & 0xff];
114  s += sq[(x >> 24) & 0xff];
115  x = *(uint32_t *) (pix + 4);
116  s += sq[x & 0xff];
117  s += sq[(x >> 8) & 0xff];
118  s += sq[(x >> 16) & 0xff];
119  s += sq[(x >> 24) & 0xff];
120 #endif
121 #endif
122  pix += 8;
123  }
124  pix += line_size - 16;
125  }
126  return s;
127 }
128 
129 /* draw the edges of width 'w' of an image of size width, height */
130 // FIXME: Check that this is OK for MPEG-4 interlaced.
131 static void draw_edges_8_c(uint8_t *buf, int wrap, int width, int height,
132  int w, int h, int sides)
133 {
134  uint8_t *ptr = buf, *last_line;
135  int i;
136 
137  /* left and right */
138  for (i = 0; i < height; i++) {
139  memset(ptr - w, ptr[0], w);
140  memset(ptr + width, ptr[width - 1], w);
141  ptr += wrap;
142  }
143 
144  /* top and bottom + corners */
145  buf -= w;
146  last_line = buf + (height - 1) * wrap;
147  if (sides & EDGE_TOP)
148  for (i = 0; i < h; i++)
149  // top
150  memcpy(buf - (i + 1) * wrap, buf, width + w + w);
151  if (sides & EDGE_BOTTOM)
152  for (i = 0; i < h; i++)
153  // bottom
154  memcpy(last_line + (i + 1) * wrap, last_line, width + w + w);
155 }
156 
158  AVCodecContext *avctx)
159 {
162 
163  c->shrink[0] = av_image_copy_plane;
164  c->shrink[1] = ff_shrink22;
165  c->shrink[2] = ff_shrink44;
166  c->shrink[3] = ff_shrink88;
167 
168  c->pix_sum = pix_sum_c;
169  c->pix_norm1 = pix_norm1_c;
170 
172 
173  if (ARCH_ARM)
174  ff_mpegvideoencdsp_init_arm(c, avctx);
175  if (ARCH_PPC)
176  ff_mpegvideoencdsp_init_ppc(c, avctx);
177  if (ARCH_X86)
178  ff_mpegvideoencdsp_init_x86(c, avctx);
179 }
int(* try_8x8basis)(int16_t rem[64], int16_t weight[64], int16_t basis[64], int scale)
const char * s
Definition: avisynth_c.h:631
#define RECON_SHIFT
static int pix_norm1_c(uint8_t *pix, int line_size)
misc image utilities
void(* shrink[4])(uint8_t *dst, int dst_wrap, const uint8_t *src, int src_wrap, int width, int height)
void(* add_8x8basis)(int16_t rem[64], int16_t basis[64], int scale)
const char * b
Definition: vf_curves.c:109
av_cold void ff_mpegvideoencdsp_init_ppc(MpegvideoEncDSPContext *c, AVCodecContext *avctx)
#define EDGE_TOP
uint32_t ff_square_tab[512]
Definition: me_cmp.c:32
Macro definitions for various function/variable attributes.
av_cold void ff_mpegvideoencdsp_init(MpegvideoEncDSPContext *c, AVCodecContext *avctx)
uint8_t
#define av_cold
Definition: attributes.h:74
#define av_assert2(cond)
assert() equivalent, that does lie in speed critical code.
Definition: avassert.h:63
void ff_shrink44(uint8_t *dst, int dst_wrap, const uint8_t *src, int src_wrap, int width, int height)
Definition: imgconvert.c:125
static void draw_edges_8_c(uint8_t *buf, int wrap, int width, int height, int w, int h, int sides)
#define wrap(func)
Definition: neontest.h:62
simple assert() macros that are a bit more flexible than ISO C assert().
Libavcodec external API header.
void ff_shrink88(uint8_t *dst, int dst_wrap, const uint8_t *src, int src_wrap, int width, int height)
Definition: imgconvert.c:156
void ff_shrink22(uint8_t *dst, int dst_wrap, const uint8_t *src, int src_wrap, int width, int height)
Definition: imgconvert.c:92
int(* pix_sum)(uint8_t *pix, int line_size)
static void add_8x8basis_c(int16_t rem[64], int16_t basis[64], int scale)
av_cold void ff_mpegvideoencdsp_init_arm(MpegvideoEncDSPContext *c, AVCodecContext *avctx)
main external API structure.
Definition: avcodec.h:1241
void * buf
Definition: avisynth_c.h:553
BYTE int const BYTE int int int height
Definition: avisynth_c.h:676
int(* pix_norm1)(uint8_t *pix, int line_size)
static int pix_sum_c(uint8_t *pix, int line_size)
static int weight(int i, int blen, int offset)
Definition: diracdec.c:1298
#define BASIS_SHIFT
void ff_mpegvideoencdsp_init_x86(MpegvideoEncDSPContext *c, AVCodecContext *avctx)
static double c[64]
static int16_t basis[64][64]
void(* draw_edges)(uint8_t *buf, int wrap, int width, int height, int w, int h, int sides)
#define EDGE_BOTTOM
static int try_8x8basis_c(int16_t rem[64], int16_t weight[64], int16_t basis[64], int scale)
void av_image_copy_plane(uint8_t *dst, int dst_linesize, const uint8_t *src, int src_linesize, int bytewidth, int height)
Copy image plane from src to dst.
Definition: imgutils.c:273
static int width