mirror of https://github.com/mpv-player/mpv
499 lines
13 KiB
C
499 lines
13 KiB
C
#include <stdio.h>
|
|
#include <stdlib.h>
|
|
#include <string.h>
|
|
|
|
#include "../config.h"
|
|
#include "../mp_msg.h"
|
|
#include "../cpudetect.h"
|
|
|
|
#include "img_format.h"
|
|
#include "mp_image.h"
|
|
#include "vf.h"
|
|
|
|
#include "../libvo/fastmemcpy.h"
|
|
|
|
struct vf_priv_s {
|
|
int mode;
|
|
};
|
|
|
|
static inline void *my_memcpy_pic(void * dst, void * src, int bytesPerLine, int height, int dstStride, int srcStride)
|
|
{
|
|
int i;
|
|
void *retval=dst;
|
|
|
|
for(i=0; i<height; i++)
|
|
{
|
|
memcpy(dst, src, bytesPerLine);
|
|
src+= srcStride;
|
|
dst+= dstStride;
|
|
}
|
|
|
|
return retval;
|
|
}
|
|
|
|
static void deint(unsigned char *dest, int ds, unsigned char *src, int ss, int w, int h, int field)
|
|
{
|
|
int x, y;
|
|
src += ss;
|
|
dest += ds;
|
|
if (field) {
|
|
src += ss;
|
|
dest += ds;
|
|
h -= 2;
|
|
}
|
|
for (y=h/2; y; y--) {
|
|
for (x=0; x<w; x++) {
|
|
if (((src[x-ss] < src[x]) && (src[x+ss] < src[x])) ||
|
|
((src[x-ss] > src[x]) && (src[x+ss] > src[x]))) {
|
|
//dest[x] = (src[x+ss] + src[x-ss])>>1;
|
|
dest[x] = ((src[x+ss]<<1) + (src[x-ss]<<1)
|
|
+ src[x+ss+1] + src[x-ss+1]
|
|
+ src[x+ss-1] + src[x-ss-1])>>3;
|
|
}
|
|
else dest[x] = src[x];
|
|
}
|
|
dest += ds<<1;
|
|
src += ss<<1;
|
|
}
|
|
}
|
|
|
|
#ifdef HAVE_3DNOW
|
|
static void qpel_li_3DNOW(unsigned char *d, unsigned char *s, int w, int h, int ds, int ss, int up)
|
|
{
|
|
int i, j, ssd=ss;
|
|
int crap1, crap2;
|
|
if (up) {
|
|
ssd = -ss;
|
|
memcpy(d, s, w);
|
|
d += ds;
|
|
s += ss;
|
|
}
|
|
for (i=h-1; i; i--) {
|
|
asm volatile(
|
|
"1: \n\t"
|
|
"movq (%%esi), %%mm0 \n\t"
|
|
"movq (%%esi,%%eax), %%mm1 \n\t"
|
|
"pavgusb %%mm0, %%mm1 \n\t"
|
|
"addl $8, %%esi \n\t"
|
|
"pavgusb %%mm0, %%mm1 \n\t"
|
|
"movq %%mm1, (%%edi) \n\t"
|
|
"addl $8, %%edi \n\t"
|
|
"decl %%ecx \n\t"
|
|
"jnz 1b \n\t"
|
|
: "=S"(crap1), "=D"(crap2)
|
|
: "c"(w>>3), "S"(s), "D"(d), "a"(ssd)
|
|
);
|
|
for (j=w-(w&7); j<w; j++)
|
|
d[j] = (s[j+ssd] + 3*s[j])>>2;
|
|
d += ds;
|
|
s += ss;
|
|
}
|
|
if (!up) memcpy(d, s, w);
|
|
asm volatile("emms \n\t" : : : "memory");
|
|
}
|
|
#endif
|
|
|
|
#ifdef HAVE_MMX2
|
|
static void qpel_li_MMX2(unsigned char *d, unsigned char *s, int w, int h, int ds, int ss, int up)
|
|
{
|
|
int i, j, ssd=ss;
|
|
int crap1, crap2;
|
|
if (up) {
|
|
ssd = -ss;
|
|
memcpy(d, s, w);
|
|
d += ds;
|
|
s += ss;
|
|
}
|
|
for (i=h-1; i; i--) {
|
|
asm volatile(
|
|
"pxor %%mm7, %%mm7 \n\t"
|
|
"2: \n\t"
|
|
"movq (%%esi), %%mm0 \n\t"
|
|
"movq (%%esi,%%eax), %%mm1 \n\t"
|
|
"pavgb %%mm0, %%mm1 \n\t"
|
|
"addl $8, %%esi \n\t"
|
|
"pavgb %%mm0, %%mm1 \n\t"
|
|
"movq %%mm1, (%%edi) \n\t"
|
|
"addl $8, %%edi \n\t"
|
|
"decl %%ecx \n\t"
|
|
"jnz 2b \n\t"
|
|
: "=S"(crap1), "=D"(crap2)
|
|
: "c"(w>>3), "S"(s), "D"(d), "a"(ssd)
|
|
);
|
|
for (j=w-(w&7); j<w; j++)
|
|
d[j] = (s[j+ssd] + 3*s[j])>>2;
|
|
d += ds;
|
|
s += ss;
|
|
}
|
|
if (!up) memcpy(d, s, w);
|
|
asm volatile("emms \n\t" : : : "memory");
|
|
}
|
|
#endif
|
|
|
|
#ifdef HAVE_MMX
|
|
static void qpel_li_MMX(unsigned char *d, unsigned char *s, int w, int h, int ds, int ss, int up)
|
|
{
|
|
int i, j, ssd=ss;
|
|
int crap1, crap2;
|
|
if (up) {
|
|
ssd = -ss;
|
|
memcpy(d, s, w);
|
|
d += ds;
|
|
s += ss;
|
|
}
|
|
for (i=h-1; i; i--) {
|
|
asm volatile(
|
|
"pxor %%mm7, %%mm7 \n\t"
|
|
"3: \n\t"
|
|
"movq (%%esi), %%mm0 \n\t"
|
|
"movq (%%esi), %%mm1 \n\t"
|
|
"movq (%%esi,%%eax), %%mm2 \n\t"
|
|
"movq (%%esi,%%eax), %%mm3 \n\t"
|
|
"addl $8, %%esi \n\t"
|
|
"punpcklbw %%mm7, %%mm0 \n\t"
|
|
"punpckhbw %%mm7, %%mm1 \n\t"
|
|
"punpcklbw %%mm7, %%mm2 \n\t"
|
|
"punpckhbw %%mm7, %%mm3 \n\t"
|
|
"paddw %%mm0, %%mm2 \n\t"
|
|
"paddw %%mm1, %%mm3 \n\t"
|
|
"paddw %%mm0, %%mm2 \n\t"
|
|
"paddw %%mm1, %%mm3 \n\t"
|
|
"paddw %%mm0, %%mm2 \n\t"
|
|
"paddw %%mm1, %%mm3 \n\t"
|
|
"psrlw $2, %%mm2 \n\t"
|
|
"psrlw $2, %%mm3 \n\t"
|
|
"packsswb %%mm3, %%mm2 \n\t"
|
|
"movq %%mm2, (%%edi) \n\t"
|
|
"addl $8, %%edi \n\t"
|
|
"decl %%ecx \n\t"
|
|
"jnz 3b \n\t"
|
|
: "=S"(crap1), "=D"(crap2)
|
|
: "c"(w>>3), "S"(s), "D"(d), "a"(ssd)
|
|
);
|
|
for (j=w-(w&7); j<w; j++)
|
|
d[j] = (s[j+ssd] + 3*s[j])>>2;
|
|
d += ds;
|
|
s += ss;
|
|
}
|
|
if (!up) memcpy(d, s, w);
|
|
asm volatile("emms \n\t" : : : "memory");
|
|
}
|
|
|
|
static void qpel_4tap_MMX(unsigned char *d, unsigned char *s, int w, int h, int ds, int ss, int up)
|
|
{
|
|
int i, j, ssd=ss;
|
|
static const short filter[] = {
|
|
29, 29, 29, 29, 110, 110, 110, 110,
|
|
9, 9, 9, 9, 3, 3, 3, 3,
|
|
64, 64, 64, 64 };
|
|
int crap1, crap2;
|
|
if (up) {
|
|
ssd = -ss;
|
|
memcpy(d, s, w);
|
|
d += ds; s += ss;
|
|
}
|
|
for (j=0; j<w; j++)
|
|
d[j] = (s[j+ssd] + 3*s[j])>>2;
|
|
d += ds; s += ss;
|
|
for (i=h-3; i; i--) {
|
|
asm volatile(
|
|
"pxor %%mm0, %%mm0 \n\t"
|
|
"movq (%%edx), %%mm4 \n\t"
|
|
"movq 8(%%edx), %%mm5 \n\t"
|
|
"movq 16(%%edx), %%mm6 \n\t"
|
|
"movq 24(%%edx), %%mm7 \n\t"
|
|
"4: \n\t"
|
|
|
|
"movq (%%esi,%%eax), %%mm1 \n\t"
|
|
"movq (%%esi), %%mm2 \n\t"
|
|
"movq (%%esi,%%ebx), %%mm3 \n\t"
|
|
"punpcklbw %%mm0, %%mm1 \n\t"
|
|
"punpcklbw %%mm0, %%mm2 \n\t"
|
|
"pmullw %%mm4, %%mm1 \n\t"
|
|
"punpcklbw %%mm0, %%mm3 \n\t"
|
|
"pmullw %%mm5, %%mm2 \n\t"
|
|
"paddusw %%mm2, %%mm1 \n\t"
|
|
"pmullw %%mm6, %%mm3 \n\t"
|
|
"movq (%%esi,%%eax,2), %%mm2 \n\t"
|
|
"psubusw %%mm3, %%mm1 \n\t"
|
|
"punpcklbw %%mm0, %%mm2 \n\t"
|
|
"pmullw %%mm7, %%mm2 \n\t"
|
|
"psubusw %%mm2, %%mm1 \n\t"
|
|
"psrlw $7, %%mm1 \n\t"
|
|
|
|
"movq (%%esi,%%eax), %%mm2 \n\t"
|
|
"movq (%%esi), %%mm3 \n\t"
|
|
"punpckhbw %%mm0, %%mm2 \n\t"
|
|
"punpckhbw %%mm0, %%mm3 \n\t"
|
|
"pmullw %%mm4, %%mm2 \n\t"
|
|
"pmullw %%mm5, %%mm3 \n\t"
|
|
"paddusw %%mm3, %%mm2 \n\t"
|
|
"movq (%%esi,%%ebx), %%mm3 \n\t"
|
|
"punpckhbw %%mm0, %%mm3 \n\t"
|
|
"pmullw %%mm6, %%mm3 \n\t"
|
|
"psubusw %%mm3, %%mm2 \n\t"
|
|
"movq (%%esi,%%eax,2), %%mm3 \n\t"
|
|
"punpckhbw %%mm0, %%mm3 \n\t"
|
|
"addl $8, %%esi \n\t"
|
|
"pmullw %%mm7, %%mm3 \n\t"
|
|
"psubusw %%mm3, %%mm2 \n\t"
|
|
"psrlw $7, %%mm2 \n\t"
|
|
|
|
"packuswb %%mm2, %%mm1 \n\t"
|
|
"movq %%mm1, (%%edi) \n\t"
|
|
"addl $8, %%edi \n\t"
|
|
"decl %%ecx \n\t"
|
|
"jnz 4b \n\t"
|
|
: "=S"(crap1), "=D"(crap2)
|
|
: "c"(w>>3), "S"(s), "D"(d), "a"(ssd), "b"(-ssd), "d"(filter)
|
|
);
|
|
for (j=w-(w&7); j<w; j++)
|
|
d[j] = (-9*s[j-ssd] + 111*s[j] + 29*s[j+ssd] - 3*s[j+ssd+ssd])>>7;
|
|
d += ds;
|
|
s += ss;
|
|
}
|
|
for (j=0; j<w; j++)
|
|
d[j] = (s[j+ssd] + 3*s[j])>>2;
|
|
d += ds; s += ss;
|
|
if (!up) memcpy(d, s, w);
|
|
asm volatile("emms \n\t" : : : "memory");
|
|
}
|
|
#endif
|
|
|
|
static inline int clamp(int a)
|
|
{
|
|
// If a<512, this is equivalent to:
|
|
// return (a<0) ? 0 : ( (a>255) ? 255 : a);
|
|
return (~(a>>31)) & (a | ((a<<23)>>31));
|
|
}
|
|
|
|
static void qpel_li_C(unsigned char *d, unsigned char *s, int w, int h, int ds, int ss, int up)
|
|
{
|
|
int i, j, ssd=ss;
|
|
if (up) {
|
|
ssd = -ss;
|
|
memcpy(d, s, w);
|
|
d += ds;
|
|
s += ss;
|
|
}
|
|
for (i=h-1; i; i--) {
|
|
for (j=0; j<w; j++)
|
|
d[j] = (s[j+ssd] + 3*s[j])>>2;
|
|
d += ds;
|
|
s += ss;
|
|
}
|
|
if (!up) memcpy(d, s, w);
|
|
}
|
|
|
|
static void qpel_4tap_C(unsigned char *d, unsigned char *s, int w, int h, int ds, int ss, int up)
|
|
{
|
|
int i, j, ssd=ss;
|
|
if (up) {
|
|
ssd = -ss;
|
|
memcpy(d, s, w);
|
|
d += ds; s += ss;
|
|
}
|
|
for (j=0; j<w; j++)
|
|
d[j] = (s[j+ssd] + 3*s[j] + 2)>>2;
|
|
d += ds; s += ss;
|
|
for (i=h-3; i; i--) {
|
|
for (j=0; j<w; j++)
|
|
d[j] = clamp((-9*s[j-ssd] + 111*s[j] + 29*s[j+ssd] - 3*s[j+ssd+ssd] + 64)>>7);
|
|
d += ds; s += ss;
|
|
}
|
|
for (j=0; j<w; j++)
|
|
d[j] = (s[j+ssd] + 3*s[j] + 2)>>2;
|
|
d += ds; s += ss;
|
|
if (!up) memcpy(d, s, w);
|
|
}
|
|
|
|
static void (*qpel_li)(unsigned char *d, unsigned char *s, int w, int h, int ds, int ss, int up);
|
|
static void (*qpel_4tap)(unsigned char *d, unsigned char *s, int w, int h, int ds, int ss, int up);
|
|
|
|
static int put_image(struct vf_instance_s* vf, mp_image_t *mpi)
|
|
{
|
|
int ret;
|
|
mp_image_t *dmpi;
|
|
void (*qpel)(unsigned char *, unsigned char *, int, int, int, int, int);
|
|
|
|
switch (vf->priv->mode) {
|
|
case 2:
|
|
qpel = qpel_li;
|
|
break;
|
|
case 3:
|
|
// TODO: add 3tap filter
|
|
qpel = qpel_4tap;
|
|
break;
|
|
case 4:
|
|
qpel = qpel_4tap;
|
|
break;
|
|
}
|
|
|
|
switch (vf->priv->mode) {
|
|
case 0:
|
|
dmpi = vf_get_image(vf->next, mpi->imgfmt,
|
|
MP_IMGTYPE_TEMP, MP_IMGFLAG_ACCEPT_STRIDE,
|
|
mpi->width, mpi->height/2);
|
|
memcpy_pic(dmpi->planes[0], mpi->planes[0], mpi->w, mpi->h/2,
|
|
dmpi->stride[0], mpi->stride[0]*2);
|
|
if (mpi->flags & MP_IMGFLAG_PLANAR) {
|
|
memcpy_pic(dmpi->planes[1], mpi->planes[1],
|
|
mpi->chroma_width, mpi->chroma_height/2,
|
|
dmpi->stride[1], mpi->stride[1]*2);
|
|
memcpy_pic(dmpi->planes[2], mpi->planes[2],
|
|
mpi->chroma_width, mpi->chroma_height/2,
|
|
dmpi->stride[2], mpi->stride[2]*2);
|
|
}
|
|
ret = vf_next_put_image(vf, dmpi);
|
|
|
|
memcpy_pic(dmpi->planes[0], mpi->planes[0] + mpi->stride[0],
|
|
mpi->w, mpi->h/2, dmpi->stride[0], mpi->stride[0]*2);
|
|
if (mpi->flags & MP_IMGFLAG_PLANAR) {
|
|
memcpy_pic(dmpi->planes[1], mpi->planes[1] + mpi->stride[1],
|
|
mpi->chroma_width, mpi->chroma_height/2,
|
|
dmpi->stride[1], mpi->stride[1]*2);
|
|
memcpy_pic(dmpi->planes[2], mpi->planes[2] + mpi->stride[2],
|
|
mpi->chroma_width, mpi->chroma_height/2,
|
|
dmpi->stride[2], mpi->stride[2]*2);
|
|
}
|
|
return vf_next_put_image(vf, dmpi) || ret;
|
|
case 1:
|
|
dmpi = vf_get_image(vf->next, mpi->imgfmt,
|
|
MP_IMGTYPE_TEMP, MP_IMGFLAG_ACCEPT_STRIDE,
|
|
mpi->width, mpi->height);
|
|
my_memcpy_pic(dmpi->planes[0], mpi->planes[0], mpi->w, mpi->h/2,
|
|
dmpi->stride[0]*2, mpi->stride[0]*2);
|
|
deint(dmpi->planes[0], dmpi->stride[0], mpi->planes[0], mpi->stride[0], mpi->w, mpi->h, 0);
|
|
if (mpi->flags & MP_IMGFLAG_PLANAR) {
|
|
my_memcpy_pic(dmpi->planes[1], mpi->planes[1],
|
|
mpi->chroma_width, mpi->chroma_height/2,
|
|
dmpi->stride[1]*2, mpi->stride[1]*2);
|
|
my_memcpy_pic(dmpi->planes[2], mpi->planes[2],
|
|
mpi->chroma_width, mpi->chroma_height/2,
|
|
dmpi->stride[2]*2, mpi->stride[2]*2);
|
|
deint(dmpi->planes[1], dmpi->stride[1], mpi->planes[1], mpi->stride[1],
|
|
mpi->chroma_width, mpi->chroma_height, 0);
|
|
deint(dmpi->planes[2], dmpi->stride[2], mpi->planes[2], mpi->stride[2],
|
|
mpi->chroma_width, mpi->chroma_height, 0);
|
|
}
|
|
ret = vf_next_put_image(vf, dmpi);
|
|
|
|
my_memcpy_pic(dmpi->planes[0] + dmpi->stride[0], mpi->planes[0] + mpi->stride[0],
|
|
mpi->w, mpi->h/2, dmpi->stride[0]*2, mpi->stride[0]*2);
|
|
deint(dmpi->planes[0], dmpi->stride[0], mpi->planes[0], mpi->stride[0], mpi->w, mpi->h, 1);
|
|
if (mpi->flags & MP_IMGFLAG_PLANAR) {
|
|
my_memcpy_pic(dmpi->planes[1] + dmpi->stride[1], mpi->planes[1] + mpi->stride[1],
|
|
mpi->chroma_width, mpi->chroma_height/2,
|
|
dmpi->stride[1]*2, mpi->stride[1]*2);
|
|
my_memcpy_pic(dmpi->planes[2] + dmpi->stride[2], mpi->planes[2] + mpi->stride[2],
|
|
mpi->chroma_width, mpi->chroma_height/2,
|
|
dmpi->stride[2]*2, mpi->stride[2]*2);
|
|
deint(dmpi->planes[1], dmpi->stride[1], mpi->planes[1], mpi->stride[1],
|
|
mpi->chroma_width, mpi->chroma_height, 1);
|
|
deint(dmpi->planes[2], dmpi->stride[2], mpi->planes[2], mpi->stride[2],
|
|
mpi->chroma_width, mpi->chroma_height, 1);
|
|
}
|
|
return vf_next_put_image(vf, dmpi) || ret;
|
|
case 2:
|
|
case 3:
|
|
case 4:
|
|
dmpi = vf_get_image(vf->next, mpi->imgfmt,
|
|
MP_IMGTYPE_TEMP, MP_IMGFLAG_ACCEPT_STRIDE,
|
|
mpi->width, mpi->height/2);
|
|
qpel(dmpi->planes[0], mpi->planes[0], mpi->w, mpi->h/2,
|
|
dmpi->stride[0], mpi->stride[0]*2, 0);
|
|
if (mpi->flags & MP_IMGFLAG_PLANAR) {
|
|
qpel(dmpi->planes[1], mpi->planes[1],
|
|
mpi->chroma_width, mpi->chroma_height/2,
|
|
dmpi->stride[1], mpi->stride[1]*2, 0);
|
|
qpel(dmpi->planes[2], mpi->planes[2],
|
|
mpi->chroma_width, mpi->chroma_height/2,
|
|
dmpi->stride[2], mpi->stride[2]*2, 0);
|
|
}
|
|
ret = vf_next_put_image(vf, dmpi);
|
|
|
|
qpel(dmpi->planes[0], mpi->planes[0] + mpi->stride[0],
|
|
mpi->w, mpi->h/2, dmpi->stride[0], mpi->stride[0]*2, 1);
|
|
if (mpi->flags & MP_IMGFLAG_PLANAR) {
|
|
qpel(dmpi->planes[1], mpi->planes[1] + mpi->stride[1],
|
|
mpi->chroma_width, mpi->chroma_height/2,
|
|
dmpi->stride[1], mpi->stride[1]*2, 1);
|
|
qpel(dmpi->planes[2], mpi->planes[2] + mpi->stride[2],
|
|
mpi->chroma_width, mpi->chroma_height/2,
|
|
dmpi->stride[2], mpi->stride[2]*2, 1);
|
|
}
|
|
return vf_next_put_image(vf, dmpi) || ret;
|
|
}
|
|
return 0;
|
|
}
|
|
|
|
static int query_format(struct vf_instance_s* vf, unsigned int fmt)
|
|
{
|
|
/* FIXME - figure out which other formats work */
|
|
switch (fmt) {
|
|
case IMGFMT_YV12:
|
|
case IMGFMT_IYUV:
|
|
case IMGFMT_I420:
|
|
return vf_next_query_format(vf, fmt);
|
|
}
|
|
return 0;
|
|
}
|
|
|
|
static int config(struct vf_instance_s* vf,
|
|
int width, int height, int d_width, int d_height,
|
|
unsigned int flags, unsigned int outfmt)
|
|
{
|
|
switch (vf->priv->mode) {
|
|
case 0:
|
|
case 2:
|
|
case 3:
|
|
case 4:
|
|
return vf_next_config(vf,width,height/2,d_width,d_height,flags,outfmt);
|
|
case 1:
|
|
return vf_next_config(vf,width,height,d_width,d_height,flags,outfmt);
|
|
}
|
|
return 0;
|
|
}
|
|
|
|
static void uninit(struct vf_instance_s* vf)
|
|
{
|
|
free(vf->priv);
|
|
}
|
|
|
|
static int open(vf_instance_t *vf, char* args)
|
|
{
|
|
struct vf_priv_s *p;
|
|
vf->config = config;
|
|
vf->put_image = put_image;
|
|
vf->query_format = query_format;
|
|
vf->uninit = uninit;
|
|
vf->default_reqs = VFCAP_ACCEPT_STRIDE;
|
|
vf->priv = p = calloc(1, sizeof(struct vf_priv_s));
|
|
vf->priv->mode = 0;
|
|
if (args) sscanf(args, "%d", &vf->priv->mode);
|
|
qpel_li = qpel_li_C;
|
|
qpel_4tap = qpel_4tap_C;
|
|
#ifdef HAVE_MMX
|
|
if(gCpuCaps.hasMMX) qpel_li = qpel_li_MMX;
|
|
if(gCpuCaps.hasMMX) qpel_4tap = qpel_4tap_MMX;
|
|
#endif
|
|
#ifdef HAVE_MMX2
|
|
if(gCpuCaps.hasMMX2) qpel_li = qpel_li_MMX2;
|
|
#endif
|
|
#ifdef HAVE_3DNOW
|
|
if(gCpuCaps.has3DNow) qpel_li = qpel_li_3DNOW;
|
|
#endif
|
|
return 1;
|
|
}
|
|
|
|
vf_info_t vf_info_tfields = {
|
|
"temporal field separation",
|
|
"tfields",
|
|
"Rich Felker",
|
|
"",
|
|
open,
|
|
NULL
|
|
};
|
|
|
|
|