vf_mp: remove boxblur wrapper
[ffmpeg.git] / libavfilter / libmpcodecs / vf_geq.c
1 /*
2  * Copyright (C) 2006 Michael Niedermayer <michaelni@gmx.at>
3  *
4  * This file is part of MPlayer.
5  *
6  * MPlayer is free software; you can redistribute it and/or modify
7  * it under the terms of the GNU General Public License as published by
8  * the Free Software Foundation; either version 2 of the License, or
9  * (at your option) any later version.
10  *
11  * MPlayer is distributed in the hope that it will be useful,
12  * but WITHOUT ANY WARRANTY; without even the implied warranty of
13  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
14  * GNU General Public License for more details.
15  *
16  * You should have received a copy of the GNU General Public License along
17  * with MPlayer; if not, write to the Free Software Foundation, Inc.,
18  * 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
19  */
20
21 #include <stdio.h>
22 #include <stdlib.h>
23 #include <string.h>
24 #include <math.h>
25 #include <inttypes.h>
26
27 #include "config.h"
28
29 #include "mp_msg.h"
30 #include "cpudetect.h"
31
32 #include "img_format.h"
33 #include "mp_image.h"
34 #include "vf.h"
35
36 #include "libavcodec/avcodec.h"
37 #include "libavutil/eval.h"
38
39 struct vf_priv_s {
40     AVExpr * e[3];
41     int framenum;
42     mp_image_t *mpi;
43 };
44
45 static int config(struct vf_instance *vf,
46         int width, int height, int d_width, int d_height,
47         unsigned int flags, unsigned int outfmt){
48     return vf_next_config(vf,width,height,d_width,d_height,flags,outfmt);
49 }
50
51 static inline double getpix(struct vf_instance *vf, double x, double y, int plane){
52     int xi, yi;
53     mp_image_t *mpi= vf->priv->mpi;
54     int stride= mpi->stride[plane];
55     uint8_t *src=  mpi->planes[plane];
56     xi=x= FFMIN(FFMAX(x, 0), (mpi->w >> (plane ? mpi->chroma_x_shift : 0))-1);
57     yi=y= FFMIN(FFMAX(y, 0), (mpi->h >> (plane ? mpi->chroma_y_shift : 0))-1);
58
59     x-=xi;
60     y-=yi;
61
62     return
63      (1-y)*((1-x)*src[xi +  yi    * stride] + x*src[xi + 1 +  yi    * stride])
64     +   y *((1-x)*src[xi + (yi+1) * stride] + x*src[xi + 1 + (yi+1) * stride]);
65 }
66
67 //FIXME cubic interpolate
68 //FIXME keep the last few frames
69 static double lum(void *vf, double x, double y){
70     return getpix(vf, x, y, 0);
71 }
72
73 static double cb(void *vf, double x, double y){
74     return getpix(vf, x, y, 1);
75 }
76
77 static double cr(void *vf, double x, double y){
78     return getpix(vf, x, y, 2);
79 }
80
81 static int put_image(struct vf_instance *vf, mp_image_t *mpi, double pts){
82     mp_image_t *dmpi;
83     int x,y, plane;
84
85     if(!(mpi->flags&MP_IMGFLAG_DIRECT)){
86         // no DR, so get a new image! hope we'll get DR buffer:
87         vf->dmpi=vf_get_image(vf->next,mpi->imgfmt, MP_IMGTYPE_TEMP,
88                               MP_IMGFLAG_ACCEPT_STRIDE|MP_IMGFLAG_PREFER_ALIGNED_STRIDE,
89                               mpi->w,mpi->h);
90     }
91
92     dmpi= vf->dmpi;
93     vf->priv->mpi= mpi;
94
95     vf_clone_mpi_attributes(dmpi, mpi);
96
97     for(plane=0; plane<3; plane++){
98         int w= mpi->w >> (plane ? mpi->chroma_x_shift : 0);
99         int h= mpi->h >> (plane ? mpi->chroma_y_shift : 0);
100         uint8_t *dst  = dmpi->planes[plane];
101         int dst_stride= dmpi->stride[plane];
102         double const_values[]={
103             M_PI,
104             M_E,
105             0,
106             0,
107             w,
108             h,
109             vf->priv->framenum,
110             w/(double)mpi->w,
111             h/(double)mpi->h,
112             0
113         };
114         if (!vf->priv->e[plane]) continue;
115         for(y=0; y<h; y++){
116             const_values[3]=y;
117             for(x=0; x<w; x++){
118                 const_values[2]=x;
119                 dst[x + y * dst_stride] = av_expr_eval(vf->priv->e[plane],
120                                                        const_values, vf);
121             }
122         }
123     }
124
125     vf->priv->framenum++;
126
127     return vf_next_put_image(vf,dmpi, pts);
128 }
129
130 static void uninit(struct vf_instance *vf){
131     av_free(vf->priv);
132     vf->priv=NULL;
133 }
134
135 //===========================================================================//
136 static int vf_open(vf_instance_t *vf, char *args){
137     char eq[3][2000] = { { 0 }, { 0 }, { 0 } };
138     int plane, res;
139
140     vf->config=config;
141     vf->put_image=put_image;
142 //    vf->get_image=get_image;
143     vf->uninit=uninit;
144     vf->priv=av_malloc(sizeof(struct vf_priv_s));
145     memset(vf->priv, 0, sizeof(struct vf_priv_s));
146
147     if (args) sscanf(args, "%1999[^:]:%1999[^:]:%1999[^:]", eq[0], eq[1], eq[2]);
148
149     if (!eq[1][0]) strncpy(eq[1], eq[0], sizeof(eq[0])-1);
150     if (!eq[2][0]) strncpy(eq[2], eq[1], sizeof(eq[0])-1);
151
152     for(plane=0; plane<3; plane++){
153         static const char *const_names[]={
154             "PI",
155             "E",
156             "X",
157             "Y",
158             "W",
159             "H",
160             "N",
161             "SW",
162             "SH",
163             NULL
164         };
165         static const char *func2_names[]={
166             "lum",
167             "cb",
168             "cr",
169             "p",
170             NULL
171         };
172         double (*func2[])(void *, double, double)={
173             lum,
174             cb,
175             cr,
176             plane==0 ? lum : (plane==1 ? cb : cr),
177             NULL
178         };
179         res = av_expr_parse(&vf->priv->e[plane], eq[plane], const_names, NULL, NULL, func2_names, func2, 0, NULL);
180
181         if (res < 0) {
182             mp_msg(MSGT_VFILTER, MSGL_ERR, "geq: error loading equation `%s'\n", eq[plane]);
183             return 0;
184         }
185     }
186
187     return 1;
188 }
189
190 const vf_info_t vf_info_geq = {
191     "generic equation filter",
192     "geq",
193     "Michael Niedermayer",
194     "",
195     vf_open,
196     NULL
197 };