?  search:
summary | log | graphiclog1 | graphiclog2 | commit | commitdiff | tree | refs | edit | fork
blame | history | raw | HEAD
lib: media_player: fix libvlc_MediaPlayerMediaChanged event
[vlc.git] / modules / video_filter / gradient.c
blob22675305f5ea159f9fe5ddb511c04bd54270148c
1 /*****************************************************************************
2 * gradient.c : Gradient and edge detection video effects plugin for vlc
3 *****************************************************************************
4 * Copyright (C) 2000-2008 VLC authors and VideoLAN
5 *
6 * Authors: Samuel Hocevar <sam@zoy.org>
7 * Antoine Cellerier <dionoea -at- videolan -dot- org>
8 *
9 * This program is free software; you can redistribute it and/or modify it
10 * under the terms of the GNU Lesser General Public License as published by
11 * the Free Software Foundation; either version 2.1 of the License, or
12 * (at your option) any later version.
13 *
14 * This program is distributed in the hope that it will be useful,
15 * but WITHOUT ANY WARRANTY; without even the implied warranty of
16 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
17 * GNU Lesser General Public License for more details.
18 *
19 * You should have received a copy of the GNU Lesser General Public License
20 * along with this program; if not, write to the Free Software Foundation,
21 * Inc., 51 Franklin Street, Fifth Floor, Boston MA 02110-1301, USA.
22 *****************************************************************************/
23
24 /*****************************************************************************
25 * Preamble
26 *****************************************************************************/
27
28 #ifdef HAVE_CONFIG_H
29 # include "config.h"
30 #endif
31
32 #include <math.h> /* sin(), cos() */
33
34 #include <vlc_common.h>
35 #include <vlc_plugin.h>
36 #include <vlc_sout.h>
37 #include <vlc_filter.h>
38 #include <vlc_picture.h>
39 #include "filter_picture.h"
40
41 enum { GRADIENT, EDGE, HOUGH };
42
43 /*****************************************************************************
44 * Local prototypes
45 *****************************************************************************/
46 static int Create ( vlc_object_t * );
47 static void Destroy ( vlc_object_t * );
48
49 static picture_t *Filter( filter_t *, picture_t * );
50 static int GradientCallback( vlc_object_t *, char const *,
51 vlc_value_t, vlc_value_t,
52 void * );
53
54 static void FilterGradient( filter_t *, picture_t *, picture_t * );
55 static void FilterEdge ( filter_t *, picture_t *, picture_t * );
56 static void FilterHough ( filter_t *, picture_t *, picture_t * );
57
58 /*****************************************************************************
59 * Module descriptor
60 *****************************************************************************/
61 #define MODE_TEXT N_("Distort mode")
62 #define MODE_LONGTEXT N_("Distort mode, one of \"gradient\", \"edge\" and \"hough\".")
63
64 #define GRADIENT_TEXT N_("Gradient image type")
65 #define GRADIENT_LONGTEXT N_("Gradient image type (0 or 1). 0 will " \
66 "turn the image to white while 1 will keep colors." )
67
68 #define CARTOON_TEXT N_("Apply cartoon effect")
69 #define CARTOON_LONGTEXT N_("Apply cartoon effect. It is only used by " \
70 "\"gradient\" and \"edge\".")
71
72 #define GRADIENT_HELP N_("Apply color gradient or edge detection effects")
73
74 static const char *const mode_list[] = { "gradient", "edge", "hough" };
75 static const char *const mode_list_text[] = { N_("Gradient"), N_("Edge"), N_("Hough") };
76
77 #define FILTER_PREFIX "gradient-"
78
79 vlc_module_begin ()
80 set_description( N_("Gradient video filter") )
81 set_shortname( N_( "Gradient" ))
82 set_help(GRADIENT_HELP)
83 set_capability( "video filter", 0 )
84 set_category( CAT_VIDEO )
85 set_subcategory( SUBCAT_VIDEO_VFILTER )
86
87 add_string( FILTER_PREFIX "mode", "gradient",
88 MODE_TEXT, MODE_LONGTEXT, false )
89 change_string_list( mode_list, mode_list_text )
90
91 add_integer_with_range( FILTER_PREFIX "type", 0, 0, 1,
92 GRADIENT_TEXT, GRADIENT_LONGTEXT, false )
93 add_bool( FILTER_PREFIX "cartoon", true,
94 CARTOON_TEXT, CARTOON_LONGTEXT, false )
95
96 add_shortcut( "gradient" )
97 set_callbacks( Create, Destroy )
98 vlc_module_end ()
99
100 static const char *const ppsz_filter_options[] = {
101 "mode", "type", "cartoon", NULL
102 };
103
104 /*****************************************************************************
105 * filter_sys_t: Distort video output method descriptor
106 *****************************************************************************
107 * This structure is part of the video output thread descriptor.
108 * It describes the Distort specific properties of an output thread.
109 *****************************************************************************/
110 typedef struct
111 {
112 vlc_mutex_t lock;
113 int i_mode;
114
115 /* For the gradient mode */
116 int i_gradient_type;
117 bool b_cartoon;
118
119 uint32_t *p_buf32;
120 uint32_t *p_buf32_bis;
121 uint8_t *p_buf8;
122
123 /* For hough mode */
124 int *p_pre_hough;
125 } filter_sys_t;
126
127 /*****************************************************************************
128 * Create: allocates Distort video thread output method
129 *****************************************************************************
130 * This function allocates and initializes a Distort vout method.
131 *****************************************************************************/
132 static int Create( vlc_object_t *p_this )
133 {
134 filter_t *p_filter = (filter_t *)p_this;
135 char *psz_method;
136
137 switch( p_filter->fmt_in.video.i_chroma )
138 {
139 CASE_PLANAR_YUV
140 break;
141
142 default:
143 msg_Err( p_filter, "Unsupported input chroma (%4.4s)",
144 (char*)&(p_filter->fmt_in.video.i_chroma) );
145 return VLC_EGENERIC;
146 }
147
148 /* Allocate structure */
149 filter_sys_t *p_sys = malloc( sizeof( filter_sys_t ) );
150 if( p_sys == NULL )
151 return VLC_ENOMEM;
152 p_filter->p_sys = p_sys;
153
154 p_filter->pf_video_filter = Filter;
155
156 p_sys->p_pre_hough = NULL;
157
158 config_ChainParse( p_filter, FILTER_PREFIX, ppsz_filter_options,
159 p_filter->p_cfg );
160
161 if( !(psz_method =
162 var_CreateGetNonEmptyStringCommand( p_filter, FILTER_PREFIX "mode" )) )
163 {
164 msg_Err( p_filter, "configuration variable "
165 FILTER_PREFIX "mode empty" );
166 p_sys->i_mode = GRADIENT;
167 }
168 else
169 {
170 if( !strcmp( psz_method, "gradient" ) )
171 {
172 p_sys->i_mode = GRADIENT;
173 }
174 else if( !strcmp( psz_method, "edge" ) )
175 {
176 p_sys->i_mode = EDGE;
177 }
178 else if( !strcmp( psz_method, "hough" ) )
179 {
180 p_sys->i_mode = HOUGH;
181 }
182 else
183 {
184 msg_Err( p_filter, "no valid gradient mode provided (%s)", psz_method );
185 p_sys->i_mode = GRADIENT;
186 }
187 }
188 free( psz_method );
189
190 p_sys->i_gradient_type =
191 var_CreateGetIntegerCommand( p_filter, FILTER_PREFIX "type" );
192 p_sys->b_cartoon =
193 var_CreateGetBoolCommand( p_filter, FILTER_PREFIX "cartoon" );
194
195 vlc_mutex_init( &p_sys->lock );
196 var_AddCallback( p_filter, FILTER_PREFIX "mode",
197 GradientCallback, p_sys );
198 var_AddCallback( p_filter, FILTER_PREFIX "type",
199 GradientCallback, p_sys );
200 var_AddCallback( p_filter, FILTER_PREFIX "cartoon",
201 GradientCallback, p_sys );
202
203 p_sys->p_buf32 = NULL;
204 p_sys->p_buf32_bis = NULL;
205 p_sys->p_buf8 = NULL;
206
207 return VLC_SUCCESS;
208 }
209
210 /*****************************************************************************
211 * Destroy: destroy Distort video thread output method
212 *****************************************************************************
213 * Terminate an output method created by DistortCreateOutputMethod
214 *****************************************************************************/
215 static void Destroy( vlc_object_t *p_this )
216 {
217 filter_t *p_filter = (filter_t *)p_this;
218 filter_sys_t *p_sys = p_filter->p_sys;
219
220 var_DelCallback( p_filter, FILTER_PREFIX "mode",
221 GradientCallback, p_sys );
222 var_DelCallback( p_filter, FILTER_PREFIX "type",
223 GradientCallback, p_sys );
224 var_DelCallback( p_filter, FILTER_PREFIX "cartoon",
225 GradientCallback, p_sys );
226
227 free( p_sys->p_buf32 );
228 free( p_sys->p_buf32_bis );
229 free( p_sys->p_buf8 );
230 free( p_sys->p_pre_hough );
231
232 free( p_sys );
233 }
234
235 /*****************************************************************************
236 * Render: displays previously rendered output
237 *****************************************************************************
238 * This function send the currently rendered image to Distort image, waits
239 * until it is displayed and switch the two rendering buffers, preparing next
240 * frame.
241 *****************************************************************************/
242 static picture_t *Filter( filter_t *p_filter, picture_t *p_pic )
243 {
244 picture_t *p_outpic;
245
246 if( !p_pic ) return NULL;
247
248 p_outpic = filter_NewPicture( p_filter );
249 if( !p_outpic )
250 {
251 picture_Release( p_pic );
252 return NULL;
253 }
254
255 filter_sys_t *p_sys = p_filter->p_sys;
256
257 vlc_mutex_lock( &p_sys->lock );
258 switch( p_sys->i_mode )
259 {
260 case EDGE:
261 FilterEdge( p_filter, p_pic, p_outpic );
262 break;
263
264 case GRADIENT:
265 FilterGradient( p_filter, p_pic, p_outpic );
266 break;
267
268 case HOUGH:
269 FilterHough( p_filter, p_pic, p_outpic );
270 break;
271
272 default:
273 break;
274 }
275 vlc_mutex_unlock( &p_sys->lock );
276
277 return CopyInfoAndRelease( p_outpic, p_pic );
278 }
279
280 /*****************************************************************************
281 * Gaussian Convolution
282 *****************************************************************************
283 * Gaussian convolution ( sigma == 1.4 )
284 *
285 * | 2 4 5 4 2 | | 2 4 4 4 2 |
286 * | 4 9 12 9 4 | | 4 8 12 8 4 |
287 * | 5 12 15 12 5 | ~ | 4 12 16 12 4 |
288 * | 4 9 12 9 4 | | 4 8 12 8 4 |
289 * | 2 4 5 4 2 | | 2 4 4 4 2 |
290 *****************************************************************************/
291 static void GaussianConvolution( picture_t *p_inpic, uint32_t *p_smooth )
292 {
293 const uint8_t *p_inpix = p_inpic->p[Y_PLANE].p_pixels;
294 const int i_src_pitch = p_inpic->p[Y_PLANE].i_pitch;
295 const int i_src_visible = p_inpic->p[Y_PLANE].i_visible_pitch;
296 const int i_num_lines = p_inpic->p[Y_PLANE].i_visible_lines;
297
298 for( int y = 2; y < i_num_lines - 2; y++ )
299 {
300 for( int x = 2; x < i_src_visible - 2; x++ )
301 {
302 p_smooth[y*i_src_visible+x] = (uint32_t)(
303 /* 2 rows up */
304 ( p_inpix[(y-2)*i_src_pitch+x-2] )
305 + ((p_inpix[(y-2)*i_src_pitch+x-1]
306 + p_inpix[(y-2)*i_src_pitch+x]
307 + p_inpix[(y-2)*i_src_pitch+x+1])<<1 )
308 + ( p_inpix[(y-2)*i_src_pitch+x+2] )
309 /* 1 row up */
310 + ((p_inpix[(y-1)*i_src_pitch+x-2]
311 + ( p_inpix[(y-1)*i_src_pitch+x-1]<<1 )
312 + ( p_inpix[(y-1)*i_src_pitch+x]*3 )
313 + ( p_inpix[(y-1)*i_src_pitch+x+1]<<1 )
314 + p_inpix[(y-1)*i_src_pitch+x+2]
315 /* */
316 + p_inpix[y*i_src_pitch+x-2]
317 + ( p_inpix[y*i_src_pitch+x-1]*3 )
318 + ( p_inpix[y*i_src_pitch+x]<<2 )
319 + ( p_inpix[y*i_src_pitch+x+1]*3 )
320 + p_inpix[y*i_src_pitch+x+2]
321 /* 1 row down */
322 + p_inpix[(y+1)*i_src_pitch+x-2]
323 + ( p_inpix[(y+1)*i_src_pitch+x-1]<<1 )
324 + ( p_inpix[(y+1)*i_src_pitch+x]*3 )
325 + ( p_inpix[(y+1)*i_src_pitch+x+1]<<1 )
326 + p_inpix[(y+1)*i_src_pitch+x+2] )<<1 )
327 /* 2 rows down */
328 + ( p_inpix[(y+2)*i_src_pitch+x-2] )
329 + ((p_inpix[(y+2)*i_src_pitch+x-1]
330 + p_inpix[(y+2)*i_src_pitch+x]
331 + p_inpix[(y+2)*i_src_pitch+x+1])<<1 )
332 + ( p_inpix[(y+2)*i_src_pitch+x+2] )
333 ) >> 6 /* 115 */;
334 }
335 }
336 }
337
338 /*****************************************************************************
339 * FilterGradient: Sobel
340 *****************************************************************************/
341 static void FilterGradient( filter_t *p_filter, picture_t *p_inpic,
342 picture_t *p_outpic )
343 {
344 filter_sys_t *p_sys = p_filter->p_sys;
345
346 const int i_src_pitch = p_inpic->p[Y_PLANE].i_pitch;
347 const int i_src_visible = p_inpic->p[Y_PLANE].i_visible_pitch;
348 const int i_dst_pitch = p_outpic->p[Y_PLANE].i_pitch;
349 const int i_num_lines = p_inpic->p[Y_PLANE].i_visible_lines;
350
351 const uint8_t *p_inpix = p_inpic->p[Y_PLANE].p_pixels;
352 uint8_t *p_outpix = p_outpic->p[Y_PLANE].p_pixels;
353
354 uint32_t *p_smooth;
355 if( !p_sys->p_buf32 )
356 p_sys->p_buf32 =
357 vlc_alloc( i_num_lines * i_src_visible, sizeof(uint32_t));
358 p_smooth = p_sys->p_buf32;
359
360 if( !p_smooth ) return;
361
362 if( p_sys->b_cartoon )
363 {
364 plane_CopyPixels( &p_outpic->p[U_PLANE], &p_inpic->p[U_PLANE] );
365 plane_CopyPixels( &p_outpic->p[V_PLANE], &p_inpic->p[V_PLANE] );
366 }
367 else
368 {
369 memset( p_outpic->p[U_PLANE].p_pixels, 0x80,
370 p_outpic->p[U_PLANE].i_lines * p_outpic->p[U_PLANE].i_pitch );
371 memset( p_outpic->p[V_PLANE].p_pixels, 0x80,
372 p_outpic->p[V_PLANE].i_lines * p_outpic->p[V_PLANE].i_pitch );
373 }
374
375 GaussianConvolution( p_inpic, p_smooth );
376
377 /* Sobel gradient
378
379 | -1 0 1 | | 1 2 1 |
380 | -2 0 2 | and | 0 0 0 |
381 | -1 0 1 | | -1 -2 -1 | */
382
383 #define FOR \
384 for( int y = 1; y < i_num_lines - 1; y++ ) \
385 { \
386 for( int x = 1; x < i_src_visible - 1; x++ ) \
387 { \
388 const uint32_t a = \
389 abs(((int)p_smooth[(y - 1) * i_src_visible + x - 1] \
390 - (int)p_smooth[(y + 1) * i_src_visible + x - 1]) \
391 + (((int)p_smooth[(y - 1) * i_src_visible + x] \
392 - (int)p_smooth[(y + 1) * i_src_visible + x]) * 2) \
393 + ((int)p_smooth[(y - 1) * i_src_visible + x + 1] \
394 - (int)p_smooth[(y + 1) * i_src_visible + x + 1])) \
395 + abs(((int)p_smooth[(y - 1) * i_src_visible + x - 1] \
396 - (int)p_smooth[(y - 1) * i_src_visible + x + 1]) \
397 + (((int)p_smooth[y * i_src_visible + x - 1] \
398 - (int)p_smooth[y * i_src_visible + x + 1]) * 2) \
399 + ((int)p_smooth[(y + 1) * i_src_visible + x - 1] \
400 - (int)p_smooth[(y + 1) * i_src_visible + x + 1]));
401
402 if( p_sys->i_gradient_type )
403 {
404 if( p_sys->b_cartoon )
405 {
406 FOR
407 if( a > 60 )
408 {
409 p_outpix[y*i_dst_pitch+x] = 0x00;
410 }
411 else
412 {
413 if( p_smooth[y*i_src_visible+x] > 0xa0 )
414 p_outpix[y*i_dst_pitch+x] =
415 0xff - ((0xff - p_inpix[y*i_src_pitch+x] )>>2);
416 else if( p_smooth[y*i_src_visible+x] > 0x70 )
417 p_outpix[y*i_dst_pitch+x] =
418 0xa0 - ((0xa0 - p_inpix[y*i_src_pitch+x] )>>2);
419 else if( p_smooth[y*i_src_visible+x] > 0x28 )
420 p_outpix[y*i_dst_pitch+x] =
421 0x70 - ((0x70 - p_inpix[y*i_src_pitch+x] )>>2);
422 else
423 p_outpix[y*i_dst_pitch+x] =
424 0x28 - ((0x28 - p_inpix[y*i_src_pitch+x] )>>2);
425 }
426 }}
427 }
428 else
429 {
430 FOR
431 p_outpix[y*i_dst_pitch+x] = clip_uint8_vlc( a );
432 }}
433 }
434 }
435 else
436 {
437 FOR
438 if( a>>8 )
439 p_outpix[y*i_dst_pitch+x] = 0;
440 else
441 p_outpix[y*i_dst_pitch+x] = 0xff-(uint8_t)a;
442 }}
443 }
444 #undef FOR
445 }
446
447 /*****************************************************************************
448 * FilterEdge: Canny edge detection algorithm
449 *****************************************************************************
450 * http://fourier.eng.hmc.edu/e161/lectures/canny/node1.html
451 * (well ... my implementation isn't really the canny algorithm ... but some
452 * ideas are the same)
453 *****************************************************************************/
454 /* angle : | */
455 #define THETA_Y 0
456 /* angle : - */
457 #define THETA_X 1
458 /* angle : / */
459 #define THETA_P 2
460 /* angle : \ */
461 #define THETA_M 3
462 static void FilterEdge( filter_t *p_filter, picture_t *p_inpic,
463 picture_t *p_outpic )
464 {
465 filter_sys_t *p_sys = p_filter->p_sys;
466
467 const int i_src_pitch = p_inpic->p[Y_PLANE].i_pitch;
468 const int i_src_visible = p_inpic->p[Y_PLANE].i_visible_pitch;
469 const int i_dst_pitch = p_outpic->p[Y_PLANE].i_pitch;
470 const int i_num_lines = p_inpic->p[Y_PLANE].i_visible_lines;
471
472 const uint8_t *p_inpix = p_inpic->p[Y_PLANE].p_pixels;
473 uint8_t *p_outpix = p_outpic->p[Y_PLANE].p_pixels;
474
475 uint32_t *p_smooth;
476 uint32_t *p_grad;
477 uint8_t *p_theta;
478
479 if( !p_sys->p_buf32 )
480 p_sys->p_buf32 =
481 vlc_alloc( i_num_lines * i_src_visible, sizeof(uint32_t));
482 p_smooth = p_sys->p_buf32;
483
484 if( !p_sys->p_buf32_bis )
485 p_sys->p_buf32_bis =
486 vlc_alloc( i_num_lines * i_src_visible, sizeof(uint32_t));
487 p_grad = p_sys->p_buf32_bis;
488
489 if( !p_sys->p_buf8 )
490 p_sys->p_buf8 =
491 vlc_alloc( i_num_lines * i_src_visible, sizeof(uint8_t));
492 p_theta = p_sys->p_buf8;
493
494 if( !p_smooth || !p_grad || !p_theta ) return;
495
496 if( p_sys->b_cartoon )
497 {
498 plane_CopyPixels( &p_outpic->p[U_PLANE], &p_inpic->p[U_PLANE] );
499 plane_CopyPixels( &p_outpic->p[V_PLANE], &p_inpic->p[V_PLANE] );
500 }
501 else
502 {
503 memset( p_outpic->p[Y_PLANE].p_pixels, 0xff,
504 p_outpic->p[Y_PLANE].i_lines * p_outpic->p[Y_PLANE].i_pitch );
505 memset( p_outpic->p[U_PLANE].p_pixels, 0x80,
506 p_outpic->p[U_PLANE].i_lines * p_outpic->p[U_PLANE].i_pitch );
507 memset( p_outpic->p[V_PLANE].p_pixels, 0x80,
508 p_outpic->p[V_PLANE].i_lines * p_outpic->p[V_PLANE].i_pitch );
509 }
510
511 GaussianConvolution( p_inpic, p_smooth );
512
513 /* Sobel gradient
514
515 | -1 0 1 | | 1 2 1 |
516 | -2 0 2 | and | 0 0 0 |
517 | -1 0 1 | | -1 -2 -1 | */
518
519 for( int y = 1; y < i_num_lines - 1; y++ )
520 {
521 for( int x = 1; x < i_src_visible - 1; x++ )
522 {
523
524 const int gradx =
525 ( p_smooth[(y-1)*i_src_visible+x-1]
526 - p_smooth[(y+1)*i_src_visible+x-1] )
527 + ( ( p_smooth[(y-1)*i_src_visible+x]
528 - p_smooth[(y+1)*i_src_visible+x] ) <<1 )
529 + ( p_smooth[(y-1)*i_src_visible+x+1]
530 - p_smooth[(y+1)*i_src_visible+x+1] );
531 const int grady =
532 ( p_smooth[(y-1)*i_src_visible+x-1]
533 - p_smooth[(y-1)*i_src_visible+x+1] )
534 + ( ( p_smooth[y*i_src_visible+x-1]
535 - p_smooth[y*i_src_visible+x+1] ) <<1 )
536 + ( p_smooth[(y+1)*i_src_visible+x-1]
537 - p_smooth[(y+1)*i_src_visible+x+1] );
538
539 p_grad[y*i_src_visible+x] = (uint32_t)(abs( gradx ) + abs( grady ));
540
541 /* tan( 22.5 ) = 0,414213562 .. * 128 = 53
542 * tan( 26,565051177 ) = 0.5
543 * tan( 45 + 22.5 ) = 2,414213562 .. * 128 = 309
544 * tan( 63,434948823 ) 2 */
545 if( (grady<<1) > gradx )
546 p_theta[y*i_src_visible+x] = THETA_P;
547 else if( (grady<<1) < -gradx )
548 p_theta[y*i_src_visible+x] = THETA_M;
549 else if( !gradx || abs(grady) > abs(gradx)<<1 )
550 p_theta[y*i_src_visible+x] = THETA_Y;
551 else
552 p_theta[y*i_src_visible+x] = THETA_X;
553 }
554 }
555
556 /* edge computing */
557 for( int y = 1; y < i_num_lines - 1; y++ )
558 {
559 for( int x = 1; x < i_src_visible - 1; x++ )
560 {
561 if( p_grad[y*i_src_visible+x] > 40 )
562 {
563 switch( p_theta[y*i_src_visible+x] )
564 {
565 case THETA_Y:
566 if( p_grad[y*i_src_visible+x] > p_grad[(y-1)*i_src_visible+x]
567 && p_grad[y*i_src_visible+x] > p_grad[(y+1)*i_src_visible+x] )
568 {
569 p_outpix[y*i_dst_pitch+x] = 0;
570 break;
571 } else goto colorize;
572 case THETA_P:
573 if( p_grad[y*i_src_visible+x] > p_grad[(y-1)*i_src_visible+x-1]
574 && p_grad[y*i_src_visible+x] > p_grad[(y+1)*i_src_visible+x+1] )
575 {
576 p_outpix[y*i_dst_pitch+x] = 0;
577 break;
578 } else goto colorize;
579 case THETA_M:
580 if( p_grad[y*i_src_visible+x] > p_grad[(y-1)*i_src_visible+x+1]
581 && p_grad[y*i_src_visible+x] > p_grad[(y+1)*i_src_visible+x-1] )
582 {
583 p_outpix[y*i_dst_pitch+x] = 0;
584 break;
585 } else goto colorize;
586 case THETA_X:
587 if( p_grad[y*i_src_visible+x] > p_grad[y*i_src_visible+x-1]
588 && p_grad[y*i_src_visible+x] > p_grad[y*i_src_visible+x+1] )
589 {
590 p_outpix[y*i_dst_pitch+x] = 0;
591 break;
592 } else goto colorize;
593 }
594 }
595 else
596 {
597 colorize:
598 if( p_sys->b_cartoon )
599 {
600 if( p_smooth[y*i_src_visible+x] > 0xa0 )
601 p_outpix[y*i_dst_pitch+x] = (uint8_t)
602 0xff - ((0xff - p_inpix[y*i_src_pitch+x] )>>2);
603 else if( p_smooth[y*i_src_visible+x] > 0x70 )
604 p_outpix[y*i_dst_pitch+x] =(uint8_t)
605 0xa0 - ((0xa0 - p_inpix[y*i_src_pitch+x] )>>2);
606 else if( p_smooth[y*i_src_visible+x] > 0x28 )
607 p_outpix[y*i_dst_pitch+x] =(uint8_t)
608 0x70 - ((0x70 - p_inpix[y*i_src_pitch+x] )>>2);
609 else
610 p_outpix[y*i_dst_pitch+x] =(uint8_t)
611 0x28 - ((0x28 - p_inpix[y*i_src_pitch+x] )>>2);
612 }
613 }
614 }
615 }
616 }
617
618 /*****************************************************************************
619 * FilterHough
620 *****************************************************************************/
621 #define p_pre_hough p_sys->p_pre_hough
622 static void FilterHough( filter_t *p_filter, picture_t *p_inpic,
623 picture_t *p_outpic )
624 {
625 filter_sys_t *p_sys = p_filter->p_sys;
626
627 int i_src_visible = p_inpic->p[Y_PLANE].i_visible_pitch;
628 int i_dst_pitch = p_outpic->p[Y_PLANE].i_pitch;
629 int i_num_lines = p_inpic->p[Y_PLANE].i_visible_lines;
630
631 uint8_t *p_outpix = p_outpic->p[Y_PLANE].p_pixels;
632
633 int i_diag = sqrt( i_num_lines * i_num_lines +
634 i_src_visible * i_src_visible);
635 int i_max, i_phi_max, i_rho, i_rho_max;
636 int i_nb_steps = 90;
637 double d_step = M_PI / i_nb_steps;
638 double d_sin;
639 double d_cos;
640 uint32_t *p_smooth;
641
642 int *p_hough = vlc_alloc( i_diag * i_nb_steps, sizeof(int) );
643 if( ! p_hough ) return;
644
645 p_smooth = vlc_alloc( i_num_lines * i_src_visible, sizeof(uint32_t));
646 if( !p_smooth )
647 {
648 free( p_hough );
649 return;
650 }
651
652 if( ! p_pre_hough )
653 {
654 msg_Dbg(p_filter, "Starting precalculation");
655 p_pre_hough = vlc_alloc( i_num_lines*i_src_visible*i_nb_steps,sizeof(int));
656 if( ! p_pre_hough )
657 {
658 free( p_smooth );
659 free( p_hough );
660 return;
661 }
662 for( int i = 0; i < i_nb_steps; i++)
663 {
664 d_sin = sin(d_step * i);
665 d_cos = cos(d_step * i);
666 for( int y = 0; y < i_num_lines; y++ )
667 for( int x = 0; x < i_src_visible; x++ )
668 {
669 p_pre_hough[(i*i_num_lines+y)*i_src_visible + x] =
670 ceil(x*d_sin + y*d_cos);
671 }
672 }
673 msg_Dbg(p_filter, "Precalculation done");
674 }
675
676 memset( p_hough, 0, i_diag * i_nb_steps * sizeof(int) );
677
678 plane_CopyPixels( &p_outpic->p[Y_PLANE], &p_inpic->p[Y_PLANE] );
679 plane_CopyPixels( &p_outpic->p[U_PLANE], &p_inpic->p[U_PLANE] );
680 plane_CopyPixels( &p_outpic->p[V_PLANE], &p_inpic->p[V_PLANE] );
681
682 GaussianConvolution( p_inpic, p_smooth );
683
684 /* Sobel gradient
685
686 | -1 0 1 | | 1 2 1 |
687 | -2 0 2 | and | 0 0 0 |
688 | -1 0 1 | | -1 -2 -1 | */
689
690 i_max = 0;
691 i_rho_max = 0;
692 i_phi_max = 0;
693 for( int y = 4; y < i_num_lines - 4; y++ )
694 {
695 for( int x = 4; x < i_src_visible - 4; x++ )
696 {
697 uint32_t a =
698 abs((((int)p_smooth[(y - 1) * i_src_visible + x]
699 - (int)p_smooth[(y + 1) * i_src_visible + x]) * 2)
700 + ((int)p_smooth[(y - 1) * i_src_visible + x - 1]
701 - (int)p_smooth[(y + 1) * i_src_visible + x - 1])
702 + ((int)p_smooth[(y - 1) * i_src_visible + x + 1]
703 - (int)p_smooth[(y + 1) * i_src_visible + x + 1]))
704 + abs((((int)p_smooth[y * i_src_visible + x - 1]
705 - (int)p_smooth[y * i_src_visible + x + 1]) * 2)
706 + ((int)p_smooth[(y - 1) * i_src_visible + x - 1]
707 - (int)p_smooth[(y - 1) * i_src_visible + x + 1])
708 + ((int)p_smooth[(y + 1) * i_src_visible + x - 1]
709 - (int)p_smooth[(y + 1) * i_src_visible + x + 1]));
710 if( a>>8 )
711 {
712 for( int i = 0; i < i_nb_steps; i++ )
713 {
714 i_rho = p_pre_hough[(i*i_num_lines+y)*i_src_visible + x];
715 if( p_hough[i_rho + i_diag/2 + i * i_diag]++ > i_max )
716 {
717 i_max = p_hough[i_rho + i_diag/2 + i * i_diag];
718 i_rho_max = i_rho;
719 i_phi_max = i;
720 }
721 }
722 }
723 }
724 }
725
726 d_sin = sin(i_phi_max*d_step);
727 d_cos = cos(i_phi_max*d_step);
728 if( d_cos != 0 )
729 {
730 for( int x = 0; x < i_src_visible; x++ )
731 {
732 int y = (i_rho_max - x * d_sin) / d_cos;
733 if( y >= 0 && y < i_num_lines )
734 p_outpix[y*i_dst_pitch+x] = 255;
735 }
736 }
737
738 free( p_hough );
739 free( p_smooth );
740 }
741 #undef p_pre_hough
742
743
744 static int GradientCallback( vlc_object_t *p_this, char const *psz_var,
745 vlc_value_t oldval, vlc_value_t newval,
746 void *p_data )
747 {
748 VLC_UNUSED(oldval);
749 filter_sys_t *p_sys = (filter_sys_t *)p_data;
750
751 vlc_mutex_lock( &p_sys->lock );
752 if( !strcmp( psz_var, FILTER_PREFIX "mode" ) )
753 {
754 if( !strcmp( newval.psz_string, "gradient" ) )
755 {
756 p_sys->i_mode = GRADIENT;
757 }
758 else if( !strcmp( newval.psz_string, "edge" ) )
759 {
760 p_sys->i_mode = EDGE;
761 }
762 else if( !strcmp( newval.psz_string, "hough" ) )
763 {
764 p_sys->i_mode = HOUGH;
765 }
766 else
767 {
768 msg_Err( p_this, "no valid gradient mode provided (%s)", newval.psz_string );
769 p_sys->i_mode = GRADIENT;
770 }
771 }
772 else if( !strcmp( psz_var, FILTER_PREFIX "type" ) )
773 {
774 p_sys->i_gradient_type = newval.i_int;
775 }
776 else if( !strcmp( psz_var, FILTER_PREFIX "cartoon" ) )
777 {
778 p_sys->b_cartoon = newval.b_bool;
779 }
780 vlc_mutex_unlock( &p_sys->lock );
781
782 return VLC_SUCCESS;
783 }
VLC Development tree mirror