Mercurial > mplayer.hg
annotate libmpcodecs/vf_geq.c @ 24738:45189040161d
Reorder #includes to get rid of the FIXME
author | zuxy |
---|---|
date | Fri, 12 Oct 2007 03:17:43 +0000 |
parents | 90302c23df72 |
children | 14e4e1ca7988 |
rev | line source |
---|---|
20266 | 1 /* |
2 Copyright (C) 2006 Michael Niedermayer <michaelni@gmx.at> | |
3 | |
4 This program is free software; you can redistribute it and/or modify | |
5 it under the terms of the GNU General Public License as published by | |
6 the Free Software Foundation; either version 2 of the License, or | |
7 (at your option) any later version. | |
8 | |
9 This program 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 | |
12 GNU General Public License for more details. | |
13 | |
14 You should have received a copy of the GNU General Public License | |
15 along with this program; if not, write to the Free Software | |
16 Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA | |
17 */ | |
18 | |
19 #include <stdio.h> | |
20 #include <stdlib.h> | |
21 #include <string.h> | |
22 #include <math.h> | |
23 #include <inttypes.h> | |
24 | |
25 #include "config.h" | |
26 | |
27 #include "mp_msg.h" | |
28 #include "cpudetect.h" | |
29 | |
24738 | 30 #include "img_format.h" |
31 #include "mp_image.h" | |
32 #include "vf.h" | |
33 | |
20266 | 34 // Needed to bring in lrintf. |
35 #define HAVE_AV_CONFIG_H | |
36 | |
37 #include "libavcodec/avcodec.h" | |
38 #include "libavcodec/dsputil.h" | |
20455 | 39 #include "libavcodec/eval.h" |
20266 | 40 #include "libavutil/common.h" |
41 | |
42 struct vf_priv_s { | |
20456 | 43 AVEvalExpr * e[3]; |
44 int framenum; | |
45 mp_image_t *mpi; | |
20266 | 46 }; |
47 | |
48 static int config(struct vf_instance_s* vf, | |
49 int width, int height, int d_width, int d_height, | |
20456 | 50 unsigned int flags, unsigned int outfmt){ |
51 return vf_next_config(vf,width,height,d_width,d_height,flags,outfmt); | |
20266 | 52 } |
53 | |
54 static void get_image(struct vf_instance_s* vf, mp_image_t *mpi){ | |
55 if(mpi->flags&MP_IMGFLAG_PRESERVE) return; // don't change | |
56 // ok, we can do pp in-place (or pp disabled): | |
57 vf->dmpi=vf_get_image(vf->next,mpi->imgfmt, | |
58 mpi->type, mpi->flags, mpi->w, mpi->h); | |
59 mpi->planes[0]=vf->dmpi->planes[0]; | |
60 mpi->stride[0]=vf->dmpi->stride[0]; | |
61 mpi->width=vf->dmpi->width; | |
62 if(mpi->flags&MP_IMGFLAG_PLANAR){ | |
63 mpi->planes[1]=vf->dmpi->planes[1]; | |
64 mpi->planes[2]=vf->dmpi->planes[2]; | |
20456 | 65 mpi->stride[1]=vf->dmpi->stride[1]; |
66 mpi->stride[2]=vf->dmpi->stride[2]; | |
20266 | 67 } |
68 mpi->flags|=MP_IMGFLAG_DIRECT; | |
69 } | |
70 | |
23219
cdf6ab5cf76d
Fix "inline keyword not at beginning of declaration" warning.
diego
parents:
20466
diff
changeset
|
71 static inline double getpix(struct vf_instance_s* vf, double x, double y, int plane){ |
20270 | 72 int xi, yi; |
73 mp_image_t *mpi= vf->priv->mpi; | |
74 int stride= mpi->stride[plane]; | |
75 uint8_t *src= mpi->planes[plane]; | |
76 xi=x= FFMIN(FFMAX(x, 0), (mpi->w >> (plane ? mpi->chroma_x_shift : 0))-1); | |
77 yi=y= FFMIN(FFMAX(y, 0), (mpi->h >> (plane ? mpi->chroma_y_shift : 0))-1); | |
78 | |
79 x-=xi; | |
80 y-=yi; | |
81 | |
82 return | |
83 (1-y)*((1-x)*src[xi + yi * stride] + x*src[xi + 1 + yi * stride]) | |
84 + y *((1-x)*src[xi + (yi+1) * stride] + x*src[xi + 1 + (yi+1) * stride]); | |
85 } | |
86 | |
87 //FIXME cubic interpolate | |
20266 | 88 //FIXME keep the last few frames |
89 static double lum(struct vf_instance_s* vf, double x, double y){ | |
20270 | 90 return getpix(vf, x, y, 0); |
20266 | 91 } |
92 | |
93 static double cb(struct vf_instance_s* vf, double x, double y){ | |
20270 | 94 return getpix(vf, x, y, 1); |
20266 | 95 } |
96 | |
97 static double cr(struct vf_instance_s* vf, double x, double y){ | |
20270 | 98 return getpix(vf, x, y, 2); |
20266 | 99 } |
100 | |
101 static int put_image(struct vf_instance_s* vf, mp_image_t *mpi, double pts){ | |
20456 | 102 mp_image_t *dmpi; |
103 int x,y, plane; | |
20455 | 104 |
20456 | 105 if(!(mpi->flags&MP_IMGFLAG_DIRECT)){ |
106 // no DR, so get a new image! hope we'll get DR buffer: | |
107 vf->dmpi=vf_get_image(vf->next,mpi->imgfmt, MP_IMGTYPE_TEMP, | |
108 MP_IMGFLAG_ACCEPT_STRIDE|MP_IMGFLAG_PREFER_ALIGNED_STRIDE, | |
109 mpi->w,mpi->h); | |
110 } | |
20455 | 111 |
20456 | 112 dmpi= vf->dmpi; |
113 vf->priv->mpi= mpi; | |
20455 | 114 |
20456 | 115 vf_clone_mpi_attributes(dmpi, mpi); |
20455 | 116 |
20456 | 117 for(plane=0; plane<3; plane++){ |
118 int w= mpi->w >> (plane ? mpi->chroma_x_shift : 0); | |
119 int h= mpi->h >> (plane ? mpi->chroma_y_shift : 0); | |
120 uint8_t *dst = dmpi->planes[plane]; | |
121 int dst_stride= dmpi->stride[plane]; | |
122 double const_values[]={ | |
123 M_PI, | |
124 M_E, | |
125 0, | |
126 0, | |
127 w, | |
128 h, | |
129 vf->priv->framenum, | |
130 w/(double)mpi->w, | |
131 h/(double)mpi->h, | |
132 0 | |
133 }; | |
134 if (!vf->priv->e[plane]) continue; | |
135 for(y=0; y<h; y++){ | |
136 const_values[3]=y; | |
137 for(x=0; x<w; x++){ | |
138 const_values[2]=x; | |
139 dst[x+y* dst_stride]= ff_parse_eval(vf->priv->e[plane], const_values, vf); | |
20455 | 140 } |
141 } | |
20456 | 142 } |
20455 | 143 |
20456 | 144 vf->priv->framenum++; |
20455 | 145 |
20456 | 146 return vf_next_put_image(vf,dmpi, pts); |
20455 | 147 } |
148 | |
149 static void uninit(struct vf_instance_s* vf){ | |
20456 | 150 if(!vf->priv) return; |
20455 | 151 |
20456 | 152 av_free(vf->priv); |
153 vf->priv=NULL; | |
20455 | 154 } |
155 | |
156 //===========================================================================// | |
157 static int open(vf_instance_t *vf, char* args){ | |
158 char eq[3][2000] = { { 0 }, { 0 }, { 0 } }; | |
159 int plane; | |
160 | |
161 vf->config=config; | |
162 vf->put_image=put_image; | |
163 // vf->get_image=get_image; | |
164 vf->uninit=uninit; | |
165 vf->priv=av_malloc(sizeof(struct vf_priv_s)); | |
166 memset(vf->priv, 0, sizeof(struct vf_priv_s)); | |
167 | |
20466 | 168 if (args) sscanf(args, "%1999[^:]:%1999[^:]:%1999[^:]", eq[0], eq[1], eq[2]); |
20455 | 169 |
170 if (!eq[1][0]) strncpy(eq[1], eq[0], sizeof(eq[0])-1); | |
171 if (!eq[2][0]) strncpy(eq[2], eq[1], sizeof(eq[0])-1); | |
172 | |
173 for(plane=0; plane<3; plane++){ | |
20266 | 174 static const char *const_names[]={ |
175 "PI", | |
176 "E", | |
177 "X", | |
178 "Y", | |
179 "W", | |
180 "H", | |
181 "N", | |
182 "SW", | |
183 "SH", | |
184 NULL | |
185 }; | |
186 static const char *func2_names[]={ | |
187 "lum", | |
188 "cb", | |
189 "cr", | |
190 "p", | |
191 NULL | |
192 }; | |
20455 | 193 double (*func2[])(void *, double, double)={ |
194 lum, | |
195 cb, | |
196 cr, | |
197 plane==0 ? lum : (plane==1 ? cb : cr), | |
198 NULL | |
199 }; | |
200 char * a; | |
201 vf->priv->e[plane] = ff_parse(eq[plane], const_names, NULL, NULL, func2, func2_names, &a); | |
20266 | 202 |
20455 | 203 if (!vf->priv->e[plane]) { |
204 mp_msg(MSGT_VFILTER, MSGL_ERR, "geq: error loading equation `%s': %s\n", eq[plane], a); | |
20266 | 205 } |
20455 | 206 } |
20266 | 207 |
208 return 1; | |
209 } | |
210 | |
211 vf_info_t vf_info_geq = { | |
212 "generic equation filter", | |
213 "geq", | |
214 "Michael Niedermayer", | |
215 "", | |
216 open, | |
217 NULL | |
218 }; |