FFmpeg
 All Data Structures Namespaces Files Functions Variables Typedefs Enumerations Enumerator Macros Groups Pages
mpegvideodsp.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 "config.h"
20 #include "libavutil/attributes.h"
21 #include "libavutil/cpu.h"
22 #include "libavutil/x86/cpu.h"
24 #include "libavcodec/videodsp.h"
25 
26 #if HAVE_INLINE_ASM
27 
28 static void gmc_mmx(uint8_t *dst, uint8_t *src,
29  int stride, int h, int ox, int oy,
30  int dxx, int dxy, int dyx, int dyy,
31  int shift, int r, int width, int height)
32 {
33  const int w = 8;
34  const int ix = ox >> (16 + shift);
35  const int iy = oy >> (16 + shift);
36  const int oxs = ox >> 4;
37  const int oys = oy >> 4;
38  const int dxxs = dxx >> 4;
39  const int dxys = dxy >> 4;
40  const int dyxs = dyx >> 4;
41  const int dyys = dyy >> 4;
42  const uint16_t r4[4] = { r, r, r, r };
43  const uint16_t dxy4[4] = { dxys, dxys, dxys, dxys };
44  const uint16_t dyy4[4] = { dyys, dyys, dyys, dyys };
45  const uint64_t shift2 = 2 * shift;
46 #define MAX_STRIDE 4096U
47 #define MAX_H 8U
48  uint8_t edge_buf[(MAX_H + 1) * MAX_STRIDE];
49  int x, y;
50 
51  const int dxw = (dxx - (1 << (16 + shift))) * (w - 1);
52  const int dyh = (dyy - (1 << (16 + shift))) * (h - 1);
53  const int dxh = dxy * (h - 1);
54  const int dyw = dyx * (w - 1);
55  int need_emu = (unsigned) ix >= width - w ||
56  (unsigned) iy >= height - h;
57 
58  if ( // non-constant fullpel offset (3% of blocks)
59  ((ox ^ (ox + dxw)) | (ox ^ (ox + dxh)) | (ox ^ (ox + dxw + dxh)) |
60  (oy ^ (oy + dyw)) | (oy ^ (oy + dyh)) | (oy ^ (oy + dyw + dyh))) >> (16 + shift) ||
61  // uses more than 16 bits of subpel mv (only at huge resolution)
62  (dxx | dxy | dyx | dyy) & 15 ||
63  (need_emu && (h > MAX_H || stride > MAX_STRIDE))) {
64  // FIXME could still use mmx for some of the rows
65  ff_gmc_c(dst, src, stride, h, ox, oy, dxx, dxy, dyx, dyy,
66  shift, r, width, height);
67  return;
68  }
69 
70  src += ix + iy * stride;
71  if (need_emu) {
72  ff_emulated_edge_mc_8(edge_buf, src, stride, stride, w + 1, h + 1, ix, iy, width, height);
73  src = edge_buf;
74  }
75 
76  __asm__ volatile (
77  "movd %0, %%mm6 \n\t"
78  "pxor %%mm7, %%mm7 \n\t"
79  "punpcklwd %%mm6, %%mm6 \n\t"
80  "punpcklwd %%mm6, %%mm6 \n\t"
81  :: "r" (1 << shift));
82 
83  for (x = 0; x < w; x += 4) {
84  uint16_t dx4[4] = { oxs - dxys + dxxs * (x + 0),
85  oxs - dxys + dxxs * (x + 1),
86  oxs - dxys + dxxs * (x + 2),
87  oxs - dxys + dxxs * (x + 3) };
88  uint16_t dy4[4] = { oys - dyys + dyxs * (x + 0),
89  oys - dyys + dyxs * (x + 1),
90  oys - dyys + dyxs * (x + 2),
91  oys - dyys + dyxs * (x + 3) };
92 
93  for (y = 0; y < h; y++) {
94  __asm__ volatile (
95  "movq %0, %%mm4 \n\t"
96  "movq %1, %%mm5 \n\t"
97  "paddw %2, %%mm4 \n\t"
98  "paddw %3, %%mm5 \n\t"
99  "movq %%mm4, %0 \n\t"
100  "movq %%mm5, %1 \n\t"
101  "psrlw $12, %%mm4 \n\t"
102  "psrlw $12, %%mm5 \n\t"
103  : "+m" (*dx4), "+m" (*dy4)
104  : "m" (*dxy4), "m" (*dyy4));
105 
106  __asm__ volatile (
107  "movq %%mm6, %%mm2 \n\t"
108  "movq %%mm6, %%mm1 \n\t"
109  "psubw %%mm4, %%mm2 \n\t"
110  "psubw %%mm5, %%mm1 \n\t"
111  "movq %%mm2, %%mm0 \n\t"
112  "movq %%mm4, %%mm3 \n\t"
113  "pmullw %%mm1, %%mm0 \n\t" // (s - dx) * (s - dy)
114  "pmullw %%mm5, %%mm3 \n\t" // dx * dy
115  "pmullw %%mm5, %%mm2 \n\t" // (s - dx) * dy
116  "pmullw %%mm4, %%mm1 \n\t" // dx * (s - dy)
117 
118  "movd %4, %%mm5 \n\t"
119  "movd %3, %%mm4 \n\t"
120  "punpcklbw %%mm7, %%mm5 \n\t"
121  "punpcklbw %%mm7, %%mm4 \n\t"
122  "pmullw %%mm5, %%mm3 \n\t" // src[1, 1] * dx * dy
123  "pmullw %%mm4, %%mm2 \n\t" // src[0, 1] * (s - dx) * dy
124 
125  "movd %2, %%mm5 \n\t"
126  "movd %1, %%mm4 \n\t"
127  "punpcklbw %%mm7, %%mm5 \n\t"
128  "punpcklbw %%mm7, %%mm4 \n\t"
129  "pmullw %%mm5, %%mm1 \n\t" // src[1, 0] * dx * (s - dy)
130  "pmullw %%mm4, %%mm0 \n\t" // src[0, 0] * (s - dx) * (s - dy)
131  "paddw %5, %%mm1 \n\t"
132  "paddw %%mm3, %%mm2 \n\t"
133  "paddw %%mm1, %%mm0 \n\t"
134  "paddw %%mm2, %%mm0 \n\t"
135 
136  "psrlw %6, %%mm0 \n\t"
137  "packuswb %%mm0, %%mm0 \n\t"
138  "movd %%mm0, %0 \n\t"
139 
140  : "=m" (dst[x + y * stride])
141  : "m" (src[0]), "m" (src[1]),
142  "m" (src[stride]), "m" (src[stride + 1]),
143  "m" (*r4), "m" (shift2));
144  src += stride;
145  }
146  src += 4 - h * stride;
147  }
148 }
149 
150 #endif /* HAVE_INLINE_ASM */
151 
153 {
154 #if HAVE_INLINE_ASM
155  int cpu_flags = av_get_cpu_flags();
156 
157  if (INLINE_MMX(cpu_flags))
158  c->gmc = gmc_mmx;
159 #endif /* HAVE_INLINE_ASM */
160 }
static int shift(int a, int b)
Definition: sonic.c:82
Macro definitions for various function/variable attributes.
uint8_t
#define av_cold
Definition: attributes.h:82
#define height
void(* gmc)(uint8_t *dst, uint8_t *src, int stride, int h, int ox, int oy, int dxx, int dxy, int dyx, int dyy, int shift, int r, int width, int height)
global motion compensation.
Definition: mpegvideodsp.h:37
#define INLINE_MMX(flags)
Definition: cpu.h:81
const char * r
Definition: vf_curves.c:107
av_cold void ff_mpegvideodsp_init_x86(MpegVideoDSPContext *c)
Definition: mpegvideodsp.c:152
void ff_gmc_c(uint8_t *dst, uint8_t *src, int stride, int h, int ox, int oy, int dxx, int dxy, int dyx, int dyy, int shift, int r, int width, int height)
Definition: mpegvideodsp.c:47
#define width
#define src
Definition: vp9dsp.c:530
int av_get_cpu_flags(void)
Return the flags which specify extensions supported by the CPU.
Definition: cpu.c:76
GLint GLenum GLboolean GLsizei stride
Definition: opengl_enc.c:105
static const int shift2[6]
Definition: dxa.c:51
static double c[64]
Core video DSP helper functions.
#define stride