mirror of https://github.com/FFmpeg/FFmpeg.git
Clément Bœsch
12 years ago
parent
8eecbaf5e4
commit
3a7f00810c
5 changed files with 1 additions and 201 deletions
@ -1,196 +0,0 @@ |
||||
/*
|
||||
* Copyright (C) 2006 Michael Niedermayer <michaelni@gmx.at> |
||||
* |
||||
* This file is part of MPlayer. |
||||
* |
||||
* MPlayer is free software; you can redistribute it and/or modify |
||||
* it under the terms of the GNU General Public License as published by |
||||
* the Free Software Foundation; either version 2 of the License, or |
||||
* (at your option) any later version. |
||||
* |
||||
* MPlayer 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 General Public License for more details. |
||||
* |
||||
* You should have received a copy of the GNU General Public License along |
||||
* with MPlayer; if not, write to the Free Software Foundation, Inc., |
||||
* 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA. |
||||
*/ |
||||
|
||||
#include <stdio.h> |
||||
#include <stdlib.h> |
||||
#include <string.h> |
||||
#include <math.h> |
||||
#include <inttypes.h> |
||||
|
||||
#include "config.h" |
||||
|
||||
#include "mp_msg.h" |
||||
|
||||
#include "img_format.h" |
||||
#include "mp_image.h" |
||||
#include "vf.h" |
||||
|
||||
#include "libavcodec/avcodec.h" |
||||
#include "libavutil/eval.h" |
||||
|
||||
struct vf_priv_s { |
||||
AVExpr * e[3]; |
||||
int framenum; |
||||
mp_image_t *mpi; |
||||
}; |
||||
|
||||
static int config(struct vf_instance *vf, |
||||
int width, int height, int d_width, int d_height, |
||||
unsigned int flags, unsigned int outfmt){ |
||||
return vf_next_config(vf,width,height,d_width,d_height,flags,outfmt); |
||||
} |
||||
|
||||
static inline double getpix(struct vf_instance *vf, double x, double y, int plane){ |
||||
int xi, yi; |
||||
mp_image_t *mpi= vf->priv->mpi; |
||||
int stride= mpi->stride[plane]; |
||||
uint8_t *src= mpi->planes[plane]; |
||||
xi=x= FFMIN(FFMAX(x, 0), (mpi->w >> (plane ? mpi->chroma_x_shift : 0))-1); |
||||
yi=y= FFMIN(FFMAX(y, 0), (mpi->h >> (plane ? mpi->chroma_y_shift : 0))-1); |
||||
|
||||
x-=xi; |
||||
y-=yi; |
||||
|
||||
return |
||||
(1-y)*((1-x)*src[xi + yi * stride] + x*src[xi + 1 + yi * stride]) |
||||
+ y *((1-x)*src[xi + (yi+1) * stride] + x*src[xi + 1 + (yi+1) * stride]); |
||||
} |
||||
|
||||
//FIXME cubic interpolate
|
||||
//FIXME keep the last few frames
|
||||
static double lum(void *vf, double x, double y){ |
||||
return getpix(vf, x, y, 0); |
||||
} |
||||
|
||||
static double cb(void *vf, double x, double y){ |
||||
return getpix(vf, x, y, 1); |
||||
} |
||||
|
||||
static double cr(void *vf, double x, double y){ |
||||
return getpix(vf, x, y, 2); |
||||
} |
||||
|
||||
static int put_image(struct vf_instance *vf, mp_image_t *mpi, double pts){ |
||||
mp_image_t *dmpi; |
||||
int x,y, plane; |
||||
|
||||
if(!(mpi->flags&MP_IMGFLAG_DIRECT)){ |
||||
// no DR, so get a new image! hope we'll get DR buffer:
|
||||
vf->dmpi=vf_get_image(vf->next,mpi->imgfmt, MP_IMGTYPE_TEMP, |
||||
MP_IMGFLAG_ACCEPT_STRIDE|MP_IMGFLAG_PREFER_ALIGNED_STRIDE, |
||||
mpi->w,mpi->h); |
||||
} |
||||
|
||||
dmpi= vf->dmpi; |
||||
vf->priv->mpi= mpi; |
||||
|
||||
vf_clone_mpi_attributes(dmpi, mpi); |
||||
|
||||
for(plane=0; plane<3; plane++){ |
||||
int w= mpi->w >> (plane ? mpi->chroma_x_shift : 0); |
||||
int h= mpi->h >> (plane ? mpi->chroma_y_shift : 0); |
||||
uint8_t *dst = dmpi->planes[plane]; |
||||
int dst_stride= dmpi->stride[plane]; |
||||
double const_values[]={ |
||||
M_PI, |
||||
M_E, |
||||
0, |
||||
0, |
||||
w, |
||||
h, |
||||
vf->priv->framenum, |
||||
w/(double)mpi->w, |
||||
h/(double)mpi->h, |
||||
0 |
||||
}; |
||||
if (!vf->priv->e[plane]) continue; |
||||
for(y=0; y<h; y++){ |
||||
const_values[3]=y; |
||||
for(x=0; x<w; x++){ |
||||
const_values[2]=x; |
||||
dst[x + y * dst_stride] = av_expr_eval(vf->priv->e[plane], |
||||
const_values, vf); |
||||
} |
||||
} |
||||
} |
||||
|
||||
vf->priv->framenum++; |
||||
|
||||
return vf_next_put_image(vf,dmpi, pts); |
||||
} |
||||
|
||||
static void uninit(struct vf_instance *vf){ |
||||
av_free(vf->priv); |
||||
vf->priv=NULL; |
||||
} |
||||
|
||||
//===========================================================================//
|
||||
static int vf_open(vf_instance_t *vf, char *args){ |
||||
char eq[3][2000] = { { 0 }, { 0 }, { 0 } }; |
||||
int plane, res; |
||||
|
||||
vf->config=config; |
||||
vf->put_image=put_image; |
||||
// vf->get_image=get_image;
|
||||
vf->uninit=uninit; |
||||
vf->priv=av_malloc(sizeof(struct vf_priv_s)); |
||||
memset(vf->priv, 0, sizeof(struct vf_priv_s)); |
||||
|
||||
if (args) sscanf(args, "%1999[^:]:%1999[^:]:%1999[^:]", eq[0], eq[1], eq[2]); |
||||
|
||||
if (!eq[1][0]) strncpy(eq[1], eq[0], sizeof(eq[0])-1); |
||||
if (!eq[2][0]) strncpy(eq[2], eq[1], sizeof(eq[0])-1); |
||||
|
||||
for(plane=0; plane<3; plane++){ |
||||
static const char *const_names[]={ |
||||
"PI", |
||||
"E", |
||||
"X", |
||||
"Y", |
||||
"W", |
||||
"H", |
||||
"N", |
||||
"SW", |
||||
"SH", |
||||
NULL |
||||
}; |
||||
static const char *func2_names[]={ |
||||
"lum", |
||||
"cb", |
||||
"cr", |
||||
"p", |
||||
NULL |
||||
}; |
||||
double (*func2[])(void *, double, double)={ |
||||
lum, |
||||
cb, |
||||
cr, |
||||
plane==0 ? lum : (plane==1 ? cb : cr), |
||||
NULL |
||||
}; |
||||
res = av_expr_parse(&vf->priv->e[plane], eq[plane], const_names, NULL, NULL, func2_names, func2, 0, NULL); |
||||
|
||||
if (res < 0) { |
||||
mp_msg(MSGT_VFILTER, MSGL_ERR, "geq: error loading equation `%s'\n", eq[plane]); |
||||
return 0; |
||||
} |
||||
} |
||||
|
||||
return 1; |
||||
} |
||||
|
||||
const vf_info_t vf_info_geq = { |
||||
"generic equation filter", |
||||
"geq", |
||||
"Michael Niedermayer", |
||||
"", |
||||
vf_open, |
||||
NULL |
||||
}; |
||||
Loading…
Reference in new issue