mirror of https://github.com/mpv-player/mpv
350 lines
8.9 KiB
C
350 lines
8.9 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"
|
|
|
|
#include "pullup.h"
|
|
|
|
#undef MAX
|
|
#define MAX(a,b) ((a)>(b)?(a):(b))
|
|
|
|
struct vf_priv_s {
|
|
struct pullup_context *ctx;
|
|
int init;
|
|
int fakecount;
|
|
char *qbuf;
|
|
};
|
|
|
|
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 inline void *il_memcpy_pic(void *dst, void *src0, void *src1, int w, int h, int ds, int ss)
|
|
{
|
|
int i;
|
|
void *retval=dst;
|
|
ss += ss;
|
|
|
|
for(i=h>>1; i; i--)
|
|
{
|
|
memcpy(dst, src0, w);
|
|
src0 += ss;
|
|
dst += ds;
|
|
memcpy(dst, src1, w);
|
|
src1 += ss;
|
|
dst += ds;
|
|
}
|
|
|
|
return retval;
|
|
}
|
|
|
|
static void init_pullup(struct vf_instance_s* vf, mp_image_t *mpi)
|
|
{
|
|
struct pullup_context *c = vf->priv->ctx;
|
|
|
|
if (mpi->flags & MP_IMGFLAG_PLANAR) {
|
|
c->format = PULLUP_FMT_Y;
|
|
c->nplanes = 4;
|
|
pullup_preinit_context(c);
|
|
c->bpp[0] = c->bpp[1] = c->bpp[2] = 8;
|
|
c->w[0] = mpi->w;
|
|
c->h[0] = mpi->h;
|
|
c->w[1] = c->w[2] = mpi->chroma_width;
|
|
c->h[1] = c->h[2] = mpi->chroma_height;
|
|
c->w[3] = ((mpi->w+15)/16) * ((mpi->h+15)/16);
|
|
c->h[3] = 2;
|
|
c->stride[0] = mpi->width;
|
|
c->stride[1] = c->stride[2] = mpi->chroma_width;
|
|
c->stride[3] = c->w[3];
|
|
c->background[1] = c->background[2] = 128;
|
|
}
|
|
|
|
if (gCpuCaps.hasMMX) c->cpu |= PULLUP_CPU_MMX;
|
|
if (gCpuCaps.hasMMX2) c->cpu |= PULLUP_CPU_MMX2;
|
|
if (gCpuCaps.has3DNow) c->cpu |= PULLUP_CPU_3DNOW;
|
|
if (gCpuCaps.has3DNowExt) c->cpu |= PULLUP_CPU_3DNOWEXT;
|
|
if (gCpuCaps.hasSSE) c->cpu |= PULLUP_CPU_SSE;
|
|
if (gCpuCaps.hasSSE2) c->cpu |= PULLUP_CPU_SSE2;
|
|
|
|
pullup_init_context(c);
|
|
|
|
vf->priv->init = 1;
|
|
vf->priv->qbuf = malloc(c->w[3]);
|
|
}
|
|
|
|
|
|
static void get_image(struct vf_instance_s* vf, mp_image_t *mpi)
|
|
{
|
|
struct pullup_context *c = vf->priv->ctx;
|
|
struct pullup_buffer *b;
|
|
|
|
if (mpi->type == MP_IMGTYPE_STATIC) return;
|
|
|
|
if (!vf->priv->init) init_pullup(vf, mpi);
|
|
|
|
b = pullup_get_buffer(c, 2);
|
|
if (!b) return; /* shouldn't happen... */
|
|
|
|
mpi->priv = b;
|
|
|
|
mpi->planes[0] = b->planes[0];
|
|
mpi->planes[1] = b->planes[1];
|
|
mpi->planes[2] = b->planes[2];
|
|
mpi->stride[0] = c->stride[0];
|
|
mpi->stride[1] = c->stride[1];
|
|
mpi->stride[2] = c->stride[2];
|
|
|
|
mpi->flags |= MP_IMGFLAG_DIRECT;
|
|
mpi->flags &= ~MP_IMGFLAG_DRAW_CALLBACK;
|
|
}
|
|
|
|
static int put_image(struct vf_instance_s* vf, mp_image_t *mpi)
|
|
{
|
|
struct pullup_context *c = vf->priv->ctx;
|
|
struct pullup_buffer *b;
|
|
struct pullup_frame *f;
|
|
mp_image_t *dmpi;
|
|
int ret;
|
|
int p;
|
|
int i;
|
|
|
|
if (!vf->priv->init) init_pullup(vf, mpi);
|
|
|
|
if (mpi->flags & MP_IMGFLAG_DIRECT) {
|
|
b = mpi->priv;
|
|
mpi->priv = 0;
|
|
} else {
|
|
b = pullup_get_buffer(c, 2);
|
|
if (!b) {
|
|
mp_msg(MSGT_VFILTER,MSGL_ERR,"Could not get buffer from pullup!\n");
|
|
f = pullup_get_frame(c);
|
|
pullup_release_frame(f);
|
|
return 0;
|
|
}
|
|
memcpy_pic(b->planes[0], mpi->planes[0], mpi->w, mpi->h,
|
|
c->stride[0], mpi->stride[0]);
|
|
if (mpi->flags & MP_IMGFLAG_PLANAR) {
|
|
memcpy_pic(b->planes[1], mpi->planes[1],
|
|
mpi->chroma_width, mpi->chroma_height,
|
|
c->stride[1], mpi->stride[1]);
|
|
memcpy_pic(b->planes[2], mpi->planes[2],
|
|
mpi->chroma_width, mpi->chroma_height,
|
|
c->stride[2], mpi->stride[2]);
|
|
}
|
|
}
|
|
if (mpi->qscale) {
|
|
memcpy(b->planes[3], mpi->qscale, c->w[3]);
|
|
memcpy(b->planes[3]+c->w[3], mpi->qscale, c->w[3]);
|
|
}
|
|
|
|
p = mpi->fields & MP_IMGFIELD_TOP_FIRST ? 0 :
|
|
(mpi->fields & MP_IMGFIELD_ORDERED ? 1 : 0);
|
|
pullup_submit_field(c, b, p);
|
|
pullup_submit_field(c, b, p^1);
|
|
if (mpi->fields & MP_IMGFIELD_REPEAT_FIRST)
|
|
pullup_submit_field(c, b, p);
|
|
|
|
pullup_release_buffer(b, 2);
|
|
|
|
f = pullup_get_frame(c);
|
|
|
|
/* Fake yes for first few frames (buffer depth) to keep from
|
|
* breaking A/V sync with G1's bad architecture... */
|
|
if (!f) return vf->priv->fakecount ? (--vf->priv->fakecount,1) : 0;
|
|
|
|
if (f->length < 2) {
|
|
pullup_release_frame(f);
|
|
f = pullup_get_frame(c);
|
|
if (!f) return 0;
|
|
if (f->length < 2) {
|
|
pullup_release_frame(f);
|
|
if (!(mpi->fields & MP_IMGFIELD_REPEAT_FIRST))
|
|
return 0;
|
|
f = pullup_get_frame(c);
|
|
if (!f) return 0;
|
|
if (f->length < 2) {
|
|
pullup_release_frame(f);
|
|
return 0;
|
|
}
|
|
}
|
|
}
|
|
|
|
#if 0
|
|
/* Average qscale tables from both frames. */
|
|
if (mpi->qscale) {
|
|
for (i=0; i<c->w[3]; i++) {
|
|
vf->priv->qbuf[i] = (f->ofields[0]->planes[3][i]
|
|
+ f->ofields[1]->planes[3][i+c->w[3]])>>1;
|
|
}
|
|
}
|
|
#else
|
|
/* Take worst of qscale tables from both frames. */
|
|
if (mpi->qscale) {
|
|
for (i=0; i<c->w[3]; i++) {
|
|
vf->priv->qbuf[i] = MAX(f->ofields[0]->planes[3][i], f->ofields[1]->planes[3][i+c->w[3]]);
|
|
}
|
|
}
|
|
#endif
|
|
|
|
/* If the frame isn't already exportable... */
|
|
while (!f->buffer) {
|
|
dmpi = vf_get_image(vf->next, mpi->imgfmt,
|
|
MP_IMGTYPE_TEMP, MP_IMGFLAG_ACCEPT_STRIDE,
|
|
mpi->width, mpi->height);
|
|
/* FIXME: Is it ok to discard dmpi if it's not direct? */
|
|
if (!(dmpi->flags & MP_IMGFLAG_DIRECT)) {
|
|
pullup_pack_frame(c, f);
|
|
break;
|
|
}
|
|
/* Direct render fields into output buffer */
|
|
#if 0
|
|
/* Write-order copy seems to have worse cache performance
|
|
* than read-order, but both should be checked on
|
|
* various cpus to see which is actually better...*/
|
|
il_memcpy_pic(dmpi->planes[0], f->ofields[0]->planes[0],
|
|
f->ofields[1]->planes[0] + c->stride[0],
|
|
mpi->w, mpi->h, dmpi->stride[0], c->stride[0]);
|
|
if (mpi->flags & MP_IMGFLAG_PLANAR) {
|
|
il_memcpy_pic(dmpi->planes[1], f->ofields[0]->planes[1],
|
|
f->ofields[1]->planes[1] + c->stride[1],
|
|
mpi->chroma_width, mpi->chroma_height,
|
|
dmpi->stride[1], c->stride[1]);
|
|
il_memcpy_pic(dmpi->planes[2], f->ofields[0]->planes[2],
|
|
f->ofields[1]->planes[2] + c->stride[2],
|
|
mpi->chroma_width, mpi->chroma_height,
|
|
dmpi->stride[2], c->stride[2]);
|
|
}
|
|
#else
|
|
my_memcpy_pic(dmpi->planes[0], f->ofields[0]->planes[0],
|
|
mpi->w, mpi->h/2, dmpi->stride[0]*2, c->stride[0]*2);
|
|
my_memcpy_pic(dmpi->planes[0] + dmpi->stride[0],
|
|
f->ofields[1]->planes[0] + c->stride[0],
|
|
mpi->w, mpi->h/2, dmpi->stride[0]*2, c->stride[0]*2);
|
|
if (mpi->flags & MP_IMGFLAG_PLANAR) {
|
|
my_memcpy_pic(dmpi->planes[1], f->ofields[0]->planes[1],
|
|
mpi->chroma_width, mpi->chroma_height/2,
|
|
dmpi->stride[1]*2, c->stride[1]*2);
|
|
my_memcpy_pic(dmpi->planes[1] + dmpi->stride[1],
|
|
f->ofields[1]->planes[1] + c->stride[1],
|
|
mpi->chroma_width, mpi->chroma_height/2,
|
|
dmpi->stride[1]*2, c->stride[1]*2);
|
|
my_memcpy_pic(dmpi->planes[2], f->ofields[0]->planes[2],
|
|
mpi->chroma_width, mpi->chroma_height/2,
|
|
dmpi->stride[2]*2, c->stride[2]*2);
|
|
my_memcpy_pic(dmpi->planes[2] + dmpi->stride[2],
|
|
f->ofields[1]->planes[2] + c->stride[2],
|
|
mpi->chroma_width, mpi->chroma_height/2,
|
|
dmpi->stride[2]*2, c->stride[2]*2);
|
|
}
|
|
#endif
|
|
pullup_release_frame(f);
|
|
if (mpi->qscale) {
|
|
dmpi->qscale = vf->priv->qbuf;
|
|
dmpi->qstride = mpi->qstride;
|
|
dmpi->qscale_type = mpi->qscale_type;
|
|
}
|
|
return vf_next_put_image(vf, dmpi);
|
|
}
|
|
dmpi = vf_get_image(vf->next, mpi->imgfmt,
|
|
MP_IMGTYPE_EXPORT, MP_IMGFLAG_ACCEPT_STRIDE,
|
|
mpi->width, mpi->height);
|
|
|
|
dmpi->planes[0] = f->buffer->planes[0];
|
|
dmpi->planes[1] = f->buffer->planes[1];
|
|
dmpi->planes[2] = f->buffer->planes[2];
|
|
|
|
dmpi->stride[0] = c->stride[0];
|
|
dmpi->stride[1] = c->stride[1];
|
|
dmpi->stride[2] = c->stride[2];
|
|
|
|
if (mpi->qscale) {
|
|
dmpi->qscale = vf->priv->qbuf;
|
|
dmpi->qstride = mpi->qstride;
|
|
dmpi->qscale_type = mpi->qscale_type;
|
|
}
|
|
ret = vf_next_put_image(vf, dmpi);
|
|
pullup_release_frame(f);
|
|
return ret;
|
|
}
|
|
|
|
static int query_format(struct vf_instance_s* vf, unsigned int fmt)
|
|
{
|
|
/* FIXME - support more formats */
|
|
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)
|
|
{
|
|
if (height&3) return 0;
|
|
return vf_next_config(vf, width, height, d_width, d_height, flags, outfmt);
|
|
}
|
|
|
|
static void uninit(struct vf_instance_s* vf)
|
|
{
|
|
pullup_free_context(vf->priv->ctx);
|
|
free(vf->priv);
|
|
}
|
|
|
|
static int open(vf_instance_t *vf, char* args)
|
|
{
|
|
struct vf_priv_s *p;
|
|
struct pullup_context *c;
|
|
//vf->get_image = get_image;
|
|
vf->put_image = put_image;
|
|
vf->config = config;
|
|
vf->query_format = query_format;
|
|
vf->uninit = uninit;
|
|
vf->default_reqs = VFCAP_ACCEPT_STRIDE;
|
|
vf->priv = p = calloc(1, sizeof(struct vf_priv_s));
|
|
p->ctx = c = pullup_alloc_context();
|
|
p->fakecount = 1;
|
|
c->verbose = verbose>0;
|
|
c->junk_left = c->junk_right = 1;
|
|
c->junk_top = c->junk_bottom = 4;
|
|
c->strict_breaks = 0;
|
|
c->metric_plane = 0;
|
|
if (args) {
|
|
sscanf(args, "%d:%d:%d:%d:%d:%d", &c->junk_left, &c->junk_right, &c->junk_top, &c->junk_bottom, &c->strict_breaks, &c->metric_plane);
|
|
}
|
|
return 1;
|
|
}
|
|
|
|
vf_info_t vf_info_pullup = {
|
|
"pullup (from field sequence to frames)",
|
|
"pullup",
|
|
"Rich Felker",
|
|
"",
|
|
open,
|
|
NULL
|
|
};
|
|
|
|
|