2 * Copyright (C) 2002 Jindrich Makovicka <makovick@gmail.com>
4 * This file is part of MPlayer.
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.
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.
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.
21 /* A very simple tv station logo remover */
32 #include "cpudetect.h"
33 #include "img_format.h"
36 #include "libvo/fastmemcpy.h"
41 //===========================================================================//
43 static struct vf_priv_s
{
45 int xoff
, yoff
, lw
, lh
, band
, show
;
47 struct timed_rectangle
{
48 int ts
, x
, y
, w
, h
, b
;
52 } const vf_priv_dflt
= {
58 #define MIN(a,b) (((a) < (b)) ? (a) : (b))
59 #define MAX(a,b) (((a) > (b)) ? (a) : (b))
62 * Adjust the coordinates to suit the band width
63 * Also print a notice in verbose mode
65 static void fix_band(struct vf_priv_s
*p
)
76 mp_msg(MSGT_VFILTER
, MSGL_V
, "delogo: %d x %d, %d x %d, band = %d\n",
77 p
->xoff
, p
->yoff
, p
->lw
, p
->lh
, p
->band
);
80 static void update_sub(struct vf_priv_s
*p
, double pts
)
82 int ipts
= pts
* 1000;
83 int tr
= p
->cur_timed_rect
;
84 while (tr
< p
->n_timed_rect
- 1 && ipts
>= p
->timed_rect
[tr
+ 1].ts
)
86 while (tr
>= 0 && ipts
< p
->timed_rect
[tr
].ts
)
88 if (tr
== p
->cur_timed_rect
)
90 p
->cur_timed_rect
= tr
;
92 p
->xoff
= p
->timed_rect
[tr
].x
;
93 p
->yoff
= p
->timed_rect
[tr
].y
;
94 p
->lw
= p
->timed_rect
[tr
].w
;
95 p
->lh
= p
->timed_rect
[tr
].h
;
96 p
->band
= p
->timed_rect
[tr
].b
;
98 p
->xoff
= p
->yoff
= p
->lw
= p
->lh
= p
->band
= 0;
103 static void delogo(uint8_t *dst
, uint8_t *src
, int dstStride
, int srcStride
, int width
, int height
,
104 int logo_x
, int logo_y
, int logo_w
, int logo_h
, int band
, int show
, int direct
) {
107 uint8_t *xdst
, *xsrc
;
109 uint8_t *topleft
, *botleft
, *topright
;
110 int xclipl
, xclipr
, yclipt
, yclipb
;
111 int logo_x1
, logo_x2
, logo_y1
, logo_y2
;
113 xclipl
= MAX(-logo_x
, 0);
114 xclipr
= MAX(logo_x
+logo_w
-width
, 0);
115 yclipt
= MAX(-logo_y
, 0);
116 yclipb
= MAX(logo_y
+logo_h
-height
, 0);
118 logo_x1
= logo_x
+ xclipl
;
119 logo_x2
= logo_x
+ logo_w
- xclipr
;
120 logo_y1
= logo_y
+ yclipt
;
121 logo_y2
= logo_y
+ logo_h
- yclipb
;
123 topleft
= src
+logo_y1
*srcStride
+logo_x1
;
124 topright
= src
+logo_y1
*srcStride
+logo_x2
-1;
125 botleft
= src
+(logo_y2
-1)*srcStride
+logo_x1
;
127 if (!direct
) memcpy_pic(dst
, src
, width
, height
, dstStride
, srcStride
);
129 dst
+= (logo_y1
+1)*dstStride
;
130 src
+= (logo_y1
+1)*srcStride
;
132 for(y
= logo_y1
+1; y
< logo_y2
-1; y
++)
134 for (x
= logo_x1
+1, xdst
= dst
+logo_x1
+1, xsrc
= src
+logo_x1
+1; x
< logo_x2
-1; x
++, xdst
++, xsrc
++) {
135 interp
= ((topleft
[srcStride
*(y
-logo_y
-yclipt
)]
136 + topleft
[srcStride
*(y
-logo_y
-1-yclipt
)]
137 + topleft
[srcStride
*(y
-logo_y
+1-yclipt
)])*(logo_w
-(x
-logo_x
))/logo_w
138 + (topright
[srcStride
*(y
-logo_y
-yclipt
)]
139 + topright
[srcStride
*(y
-logo_y
-1-yclipt
)]
140 + topright
[srcStride
*(y
-logo_y
+1-yclipt
)])*(x
-logo_x
)/logo_w
141 + (topleft
[x
-logo_x
-xclipl
]
142 + topleft
[x
-logo_x
-1-xclipl
]
143 + topleft
[x
-logo_x
+1-xclipl
])*(logo_h
-(y
-logo_y
))/logo_h
144 + (botleft
[x
-logo_x
-xclipl
]
145 + botleft
[x
-logo_x
-1-xclipl
]
146 + botleft
[x
-logo_x
+1-xclipl
])*(y
-logo_y
)/logo_h
148 /* interp = (topleft[srcStride*(y-logo_y)]*(logo_w-(x-logo_x))/logo_w
149 + topright[srcStride*(y-logo_y)]*(x-logo_x)/logo_w
150 + topleft[x-logo_x]*(logo_h-(y-logo_y))/logo_h
151 + botleft[x-logo_x]*(y-logo_y)/logo_h
153 if (y
>= logo_y
+band
&& y
< logo_y
+logo_h
-band
&& x
>= logo_x
+band
&& x
< logo_x
+logo_w
-band
) {
157 if (x
< logo_x
+band
) dist
= MAX(dist
, logo_x
-x
+band
);
158 else if (x
>= logo_x
+logo_w
-band
) dist
= MAX(dist
, x
-(logo_x
+logo_w
-1-band
));
159 if (y
< logo_y
+band
) dist
= MAX(dist
, logo_y
-y
+band
);
160 else if (y
>= logo_y
+logo_h
-band
) dist
= MAX(dist
, y
-(logo_y
+logo_h
-1-band
));
161 *xdst
= (*xsrc
*dist
+ interp
*(band
-dist
))/band
;
162 if (show
&& (dist
== band
-1)) *xdst
= 0;
171 static int config(struct vf_instance
*vf
,
172 int width
, int height
, int d_width
, int d_height
,
173 unsigned int flags
, unsigned int outfmt
){
175 return vf_next_config(vf
,width
,height
,d_width
,d_height
,flags
,outfmt
);
179 static void get_image(struct vf_instance
*vf
, mp_image_t
*mpi
){
180 if(mpi
->flags
&MP_IMGFLAG_PRESERVE
) return; // don't change
181 if(mpi
->imgfmt
!=vf
->priv
->outfmt
) return; // colorspace differ
182 // ok, we can do pp in-place (or pp disabled):
183 vf
->dmpi
=vf_get_image(vf
->next
,mpi
->imgfmt
,
184 mpi
->type
, mpi
->flags
, mpi
->w
, mpi
->h
);
185 mpi
->planes
[0]=vf
->dmpi
->planes
[0];
186 mpi
->stride
[0]=vf
->dmpi
->stride
[0];
187 mpi
->width
=vf
->dmpi
->width
;
188 if(mpi
->flags
&MP_IMGFLAG_PLANAR
){
189 mpi
->planes
[1]=vf
->dmpi
->planes
[1];
190 mpi
->planes
[2]=vf
->dmpi
->planes
[2];
191 mpi
->stride
[1]=vf
->dmpi
->stride
[1];
192 mpi
->stride
[2]=vf
->dmpi
->stride
[2];
194 mpi
->flags
|=MP_IMGFLAG_DIRECT
;
197 static int put_image(struct vf_instance
*vf
, mp_image_t
*mpi
, double pts
){
200 if(!(mpi
->flags
&MP_IMGFLAG_DIRECT
)){
201 // no DR, so get a new image! hope we'll get DR buffer:
202 vf
->dmpi
=vf_get_image(vf
->next
,vf
->priv
->outfmt
,
203 MP_IMGTYPE_TEMP
, MP_IMGFLAG_ACCEPT_STRIDE
,
208 if (vf
->priv
->timed_rect
)
209 update_sub(vf
->priv
, pts
);
210 delogo(dmpi
->planes
[0], mpi
->planes
[0], dmpi
->stride
[0], mpi
->stride
[0], mpi
->w
, mpi
->h
,
211 vf
->priv
->xoff
, vf
->priv
->yoff
, vf
->priv
->lw
, vf
->priv
->lh
, vf
->priv
->band
, vf
->priv
->show
,
212 mpi
->flags
&MP_IMGFLAG_DIRECT
);
213 delogo(dmpi
->planes
[1], mpi
->planes
[1], dmpi
->stride
[1], mpi
->stride
[1], mpi
->w
/2, mpi
->h
/2,
214 vf
->priv
->xoff
/2, vf
->priv
->yoff
/2, vf
->priv
->lw
/2, vf
->priv
->lh
/2, vf
->priv
->band
/2, vf
->priv
->show
,
215 mpi
->flags
&MP_IMGFLAG_DIRECT
);
216 delogo(dmpi
->planes
[2], mpi
->planes
[2], dmpi
->stride
[2], mpi
->stride
[2], mpi
->w
/2, mpi
->h
/2,
217 vf
->priv
->xoff
/2, vf
->priv
->yoff
/2, vf
->priv
->lw
/2, vf
->priv
->lh
/2, vf
->priv
->band
/2, vf
->priv
->show
,
218 mpi
->flags
&MP_IMGFLAG_DIRECT
);
220 vf_clone_mpi_attributes(dmpi
, mpi
);
222 return vf_next_put_image(vf
,dmpi
, pts
);
225 static void uninit(struct vf_instance
*vf
){
226 if(!vf
->priv
) return;
232 //===========================================================================//
234 static int query_format(struct vf_instance
*vf
, unsigned int fmt
){
240 return vf_next_query_format(vf
,vf
->priv
->outfmt
);
245 static const unsigned int fmt_list
[]={
252 static int load_timed_rectangles(struct vf_priv_s
*delogo
)
257 double ts
, last_ts
= 0;
258 struct timed_rectangle
*rect
= NULL
, *nr
;
259 int n_rect
= 0, alloc_rect
= 0;
261 f
= fopen(delogo
->file
, "r");
263 mp_msg(MSGT_VFILTER
, MSGL_ERR
, "delogo: unable to load %s: %s\n",
264 delogo
->file
, strerror(errno
));
267 while (fgets(line
, sizeof(line
), f
)) {
269 if (*line
== '#' || *line
== '\n')
271 if (n_rect
== alloc_rect
) {
272 if (alloc_rect
> INT_MAX
/ 2 / (int)sizeof(*rect
)) {
273 mp_msg(MSGT_VFILTER
, MSGL_WARN
,
274 "delogo: too many rectangles\n");
277 alloc_rect
= alloc_rect
? 2 * alloc_rect
: 256;
278 nr
= realloc(rect
, alloc_rect
* sizeof(*rect
));
280 mp_msg(MSGT_VFILTER
, MSGL_WARN
, "delogo: out of memory\n");
286 memset(nr
, 0, sizeof(*nr
));
287 p
= sscanf(line
, "%lf %d:%d:%d:%d:%d",
288 &ts
, &nr
->x
, &nr
->y
, &nr
->w
, &nr
->h
, &nr
->b
);
289 if ((p
== 2 && !nr
->x
) || p
== 5 || p
== 6) {
291 mp_msg(MSGT_VFILTER
, MSGL_WARN
, "delogo: %s:%d: wrong time\n",
292 delogo
->file
, lineno
);
293 nr
->ts
= 1000 * ts
+ 0.5;
296 mp_msg(MSGT_VFILTER
, MSGL_WARN
, "delogo: %s:%d: syntax error\n",
297 delogo
->file
, lineno
);
302 mp_msg(MSGT_VFILTER
, MSGL_ERR
, "delogo: %s: no rectangles found\n",
307 nr
= realloc(rect
, n_rect
* sizeof(*rect
));
310 delogo
->timed_rect
= rect
;
311 delogo
->n_timed_rect
= n_rect
;
320 static int vf_open(vf_instance_t
*vf
, char *args
){
322 vf
->put_image
=put_image
;
323 vf
->get_image
=get_image
;
324 vf
->query_format
=query_format
;
327 if (vf
->priv
->file
) {
328 if (load_timed_rectangles(vf
->priv
))
330 mp_msg(MSGT_VFILTER
, MSGL_V
, "delogo: %d from %s\n",
331 vf
->priv
->n_timed_rect
, vf
->priv
->file
);
332 vf
->priv
->cur_timed_rect
= -1;
337 vf
->priv
->outfmt
=vf_match_csp(&vf
->next
,fmt_list
,IMGFMT_YV12
);
338 if(!vf
->priv
->outfmt
)
341 return 0; // no csp match :(
347 #define ST_OFF(f) M_ST_OFF(struct vf_priv_s,f)
348 static const m_option_t vf_opts_fields
[] = {
349 { "x", ST_OFF(xoff
), CONF_TYPE_INT
, 0, 0, 0, NULL
},
350 { "y", ST_OFF(yoff
), CONF_TYPE_INT
, 0, 0, 0, NULL
},
351 { "w", ST_OFF(lw
), CONF_TYPE_INT
, 0, 0, 0, NULL
},
352 { "h", ST_OFF(lh
), CONF_TYPE_INT
, 0, 0, 0, NULL
},
353 { "t", ST_OFF(band
), CONF_TYPE_INT
, 0, 0, 0, NULL
},
354 { "band", ST_OFF(band
), CONF_TYPE_INT
, 0, 0, 0, NULL
}, // alias
355 { "file", ST_OFF(file
), CONF_TYPE_STRING
, 0, 0, 0, NULL
},
356 { NULL
, NULL
, 0, 0, 0, 0, NULL
}
359 static const m_struct_t vf_opts
= {
361 sizeof(struct vf_priv_s
),
366 const vf_info_t vf_info_delogo
= {
367 "simple logo remover",
369 "Jindrich Makovicka, Alex Beregszaszi",
375 //===========================================================================//