]> git.sesse.net Git - vlc/blob - modules/visualization/goom.c
Include vlc_plugin.h as needed
[vlc] / modules / visualization / goom.c
1 /*****************************************************************************
2  * goom.c: based on libgoom (see http://ios.free.fr/?page=projet&quoi=1)
3  *****************************************************************************
4  * Copyright (C) 2003 the VideoLAN team
5  * $Id$
6  *
7  * Authors: Laurent Aimar <fenrir@via.ecp.fr>
8  *          Gildas Bazin <gbazin@videolan.org>
9  *
10  * This program is free software; you can redistribute it and/or modify
11  * it under the terms of the GNU General Public License as published by
12  * the Free Software Foundation; either version 2 of the License, or
13  * (at your option) any later version.
14  *
15  * This program is distributed in the hope that it will be useful,
16  * but WITHOUT ANY WARRANTY; without even the implied warranty of
17  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
18  * GNU General Public License for more details.
19  *
20  * You should have received a copy of the GNU General Public License
21  * along with this program; if not, write to the Free Software
22  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston MA 02110-1301, USA.
23  *****************************************************************************/
24
25 /*****************************************************************************
26  * Preamble
27  *****************************************************************************/
28 #include <errno.h>
29
30 #ifdef HAVE_CONFIG_H
31 # include "config.h"
32 #endif
33
34 #include <vlc/vlc.h>
35 #include <vlc_plugin.h>
36 #include <vlc_aout.h>
37 #include <vlc_vout.h>
38 #include <vlc_block.h>
39 #include <vlc_input.h>
40
41 #ifdef USE_GOOM_TREE
42 #   ifdef OLD_GOOM
43 #       include "goom_core.h"
44 #       define PluginInfo void
45 #       define goom_update(a,b,c,d,e,f) goom_update(b,c,d,e,f)
46 #       define goom_close(a) goom_close()
47 #       define goom_init(a,b) NULL; goom_init(a,b,0); goom_set_font(0,0,0)
48 #   else
49 #       include "goom.h"
50 #   endif
51 #else
52 #   include <goom/goom.h>
53 #endif
54
55 /*****************************************************************************
56  * Module descriptor
57  *****************************************************************************/
58 static int  Open         ( vlc_object_t * );
59 static void Close        ( vlc_object_t * );
60
61 #define WIDTH_TEXT N_("Goom display width")
62 #define HEIGHT_TEXT N_("Goom display height")
63 #define RES_LONGTEXT N_("This allows you to set the resolution of the " \
64   "Goom display (bigger resolution will be prettier but more CPU intensive).")
65
66 #define SPEED_TEXT N_("Goom animation speed")
67 #define SPEED_LONGTEXT N_("This allows you to set the animation speed " \
68   "(between 1 and 10, defaults to 6).")
69
70 #define MAX_SPEED 10
71
72 vlc_module_begin();
73     set_shortname( _("Goom"));
74     set_description( _("Goom effect") );
75     set_category( CAT_AUDIO );
76     set_subcategory( SUBCAT_AUDIO_VISUAL );
77     set_capability( "visualization", 0 );
78     add_integer( "goom-width", 320, NULL,
79                  WIDTH_TEXT, RES_LONGTEXT, false );
80     add_integer( "goom-height", 240, NULL,
81                  HEIGHT_TEXT, RES_LONGTEXT, false );
82     add_integer( "goom-speed", 6, NULL,
83                  SPEED_TEXT, SPEED_LONGTEXT, false );
84     set_callbacks( Open, Close );
85     add_shortcut( "goom" );
86 vlc_module_end();
87
88 /*****************************************************************************
89  * Local prototypes
90  *****************************************************************************/
91 #define MAX_BLOCKS 100
92 #define GOOM_DELAY 400000
93
94 typedef struct
95 {
96     VLC_COMMON_MEMBERS
97     vout_thread_t *p_vout;
98
99     char          *psz_title;
100
101     vlc_mutex_t   lock;
102     vlc_cond_t    wait;
103
104     /* Audio properties */
105     int i_channels;
106
107     /* Audio samples queue */
108     block_t       *pp_blocks[MAX_BLOCKS];
109     int           i_blocks;
110
111     audio_date_t  date;
112
113 } goom_thread_t;
114
115 typedef struct aout_filter_sys_t
116 {
117     goom_thread_t *p_thread;
118
119 } aout_filter_sys_t;
120
121 static void DoWork   ( aout_instance_t *, aout_filter_t *, aout_buffer_t *,
122                        aout_buffer_t * );
123
124 static void Thread   ( vlc_object_t * );
125
126 static char *TitleGet( vlc_object_t * );
127
128 /*****************************************************************************
129  * Open: open a scope effect plugin
130  *****************************************************************************/
131 static int Open( vlc_object_t *p_this )
132 {
133     aout_filter_t     *p_filter = (aout_filter_t *)p_this;
134     aout_filter_sys_t *p_sys;
135     goom_thread_t     *p_thread;
136     vlc_value_t       width, height;
137     video_format_t    fmt;
138
139
140     if ( p_filter->input.i_format != VLC_FOURCC('f','l','3','2' )
141          || p_filter->output.i_format != VLC_FOURCC('f','l','3','2') )
142     {
143         msg_Warn( p_filter, "bad input or output format" );
144         return VLC_EGENERIC;
145     }
146     if ( !AOUT_FMTS_SIMILAR( &p_filter->input, &p_filter->output ) )
147     {
148         msg_Warn( p_filter, "input and output formats are not similar" );
149         return VLC_EGENERIC;
150     }
151
152     p_filter->pf_do_work = DoWork;
153     p_filter->b_in_place = 1;
154
155     /* Allocate structure */
156     p_sys = p_filter->p_sys = malloc( sizeof( aout_filter_sys_t ) );
157
158     /* Create goom thread */
159     p_sys->p_thread = p_thread =
160         vlc_object_create( p_filter, sizeof( goom_thread_t ) );
161     vlc_object_attach( p_thread, p_this );
162
163     var_Create( p_thread, "goom-width", VLC_VAR_INTEGER|VLC_VAR_DOINHERIT );
164     var_Get( p_thread, "goom-width", &width );
165     var_Create( p_thread, "goom-height", VLC_VAR_INTEGER|VLC_VAR_DOINHERIT );
166     var_Get( p_thread, "goom-height", &height );
167
168     memset( &fmt, 0, sizeof(video_format_t) );
169
170     fmt.i_width = fmt.i_visible_width = width.i_int;
171     fmt.i_height = fmt.i_visible_height = height.i_int;
172     fmt.i_chroma = VLC_FOURCC('R','V','3','2');
173     fmt.i_aspect = VOUT_ASPECT_FACTOR * width.i_int/height.i_int;
174     fmt.i_sar_num = fmt.i_sar_den = 1;
175
176     p_thread->p_vout = vout_Request( p_filter, NULL, &fmt );
177     if( p_thread->p_vout == NULL )
178     {
179         msg_Err( p_filter, "no suitable vout module" );
180         vlc_object_detach( p_thread );
181         vlc_object_release( p_thread );
182         free( p_sys );
183         return VLC_EGENERIC;
184     }
185     vlc_mutex_init( &p_thread->lock );
186     vlc_cond_init( p_filter, &p_thread->wait );
187
188     p_thread->i_blocks = 0;
189     aout_DateInit( &p_thread->date, p_filter->output.i_rate );
190     aout_DateSet( &p_thread->date, 0 );
191     p_thread->i_channels = aout_FormatNbChannels( &p_filter->input );
192
193     p_thread->psz_title = TitleGet( VLC_OBJECT( p_filter ) );
194
195     if( vlc_thread_create( p_thread, "Goom Update Thread", Thread,
196                            VLC_THREAD_PRIORITY_LOW, false ) )
197     {
198         msg_Err( p_filter, "cannot lauch goom thread" );
199         vout_Destroy( p_thread->p_vout );
200         vlc_mutex_destroy( &p_thread->lock );
201         vlc_cond_destroy( &p_thread->wait );
202         free( p_thread->psz_title );
203         vlc_object_detach( p_thread );
204         vlc_object_release( p_thread );
205         free( p_sys );
206         return VLC_EGENERIC;
207     }
208
209     return VLC_SUCCESS;
210 }
211
212 /*****************************************************************************
213  * DoWork: process samples buffer
214  *****************************************************************************
215  * This function queues the audio buffer to be processed by the goom thread
216  *****************************************************************************/
217 static void DoWork( aout_instance_t * p_aout, aout_filter_t * p_filter,
218                     aout_buffer_t * p_in_buf, aout_buffer_t * p_out_buf )
219 {
220     VLC_UNUSED( p_aout );
221
222     aout_filter_sys_t *p_sys = p_filter->p_sys;
223     block_t *p_block;
224
225     p_out_buf->i_nb_samples = p_in_buf->i_nb_samples;
226     p_out_buf->i_nb_bytes = p_in_buf->i_nb_bytes;
227
228     /* Queue sample */
229     vlc_mutex_lock( &p_sys->p_thread->lock );
230     if( p_sys->p_thread->i_blocks == MAX_BLOCKS )
231     {
232         vlc_mutex_unlock( &p_sys->p_thread->lock );
233         return;
234     }
235
236     p_block = block_New( p_sys->p_thread, p_in_buf->i_nb_bytes );
237     if( !p_block ) return;
238     memcpy( p_block->p_buffer, p_in_buf->p_buffer, p_in_buf->i_nb_bytes );
239     p_block->i_pts = p_in_buf->start_date;
240
241     p_sys->p_thread->pp_blocks[p_sys->p_thread->i_blocks++] = p_block;
242
243     vlc_cond_signal( &p_sys->p_thread->wait );
244     vlc_mutex_unlock( &p_sys->p_thread->lock );
245 }
246
247 /*****************************************************************************
248  * float to s16 conversion
249  *****************************************************************************/
250 static inline int16_t FloatToInt16( float f )
251 {
252     if( f >= 1.0 )
253         return 32767;
254     else if( f < -1.0 )
255         return -32768;
256     else
257         return (int16_t)( f * 32768.0 );
258 }
259
260 /*****************************************************************************
261  * Fill buffer
262  *****************************************************************************/
263 static int FillBuffer( int16_t *p_data, int *pi_data,
264                        audio_date_t *pi_date, audio_date_t *pi_date_end,
265                        goom_thread_t *p_this )
266 {
267     int i_samples = 0;
268     block_t *p_block;
269
270     while( *pi_data < 512 )
271     {
272         if( !p_this->i_blocks ) return VLC_EGENERIC;
273
274         p_block = p_this->pp_blocks[0];
275         i_samples = __MIN( 512 - *pi_data, p_block->i_buffer /
276                            sizeof(float) / p_this->i_channels );
277
278         /* Date management */
279         if( p_block->i_pts > 0 &&
280             p_block->i_pts != aout_DateGet( pi_date_end ) )
281         {
282            aout_DateSet( pi_date_end, p_block->i_pts );
283         }
284         p_block->i_pts = 0;
285
286         aout_DateIncrement( pi_date_end, i_samples );
287
288         while( i_samples > 0 )
289         {
290             float *p_float = (float *)p_block->p_buffer;
291
292             p_data[*pi_data] = FloatToInt16( p_float[0] );
293             if( p_this->i_channels > 1 )
294                 p_data[512 + *pi_data] = FloatToInt16( p_float[1] );
295
296             (*pi_data)++;
297             p_block->p_buffer += (sizeof(float) * p_this->i_channels);
298             p_block->i_buffer -= (sizeof(float) * p_this->i_channels);
299             i_samples--;
300         }
301
302         if( !p_block->i_buffer )
303         {
304             block_Release( p_block );
305             p_this->i_blocks--;
306             if( p_this->i_blocks )
307                 memmove( p_this->pp_blocks, p_this->pp_blocks + 1,
308                          p_this->i_blocks * sizeof(block_t *) );
309         }
310     }
311
312     *pi_date = *pi_date_end;
313     *pi_data = 0;
314     return VLC_SUCCESS;
315 }
316
317 /*****************************************************************************
318  * Thread:
319  *****************************************************************************/
320 static void Thread( vlc_object_t *p_this )
321 {
322     goom_thread_t *p_thread = (goom_thread_t*)p_this;
323     vlc_value_t width, height, speed;
324     audio_date_t i_pts;
325     int16_t p_data[2][512];
326     int i_data = 0, i_count = 0;
327     PluginInfo *p_plugin_info;
328
329     var_Get( p_this, "goom-width", &width );
330     var_Get( p_this, "goom-height", &height );
331
332     var_Create( p_thread, "goom-speed", VLC_VAR_INTEGER|VLC_VAR_DOINHERIT );
333     var_Get( p_thread, "goom-speed", &speed );
334     speed.i_int = MAX_SPEED - speed.i_int;
335     if( speed.i_int < 0 ) speed.i_int = 0;
336
337     p_plugin_info = goom_init( width.i_int, height.i_int );
338
339     while( !p_thread->b_die )
340     {
341         uint32_t  *plane;
342         picture_t *p_pic;
343
344         /* goom_update is damn slow, so just copy data and release the lock */
345         vlc_mutex_lock( &p_thread->lock );
346         if( FillBuffer( (int16_t *)p_data, &i_data, &i_pts,
347                         &p_thread->date, p_thread ) != VLC_SUCCESS )
348             vlc_cond_wait( &p_thread->wait, &p_thread->lock );
349         vlc_mutex_unlock( &p_thread->lock );
350
351         /* Speed selection */
352         if( speed.i_int && (++i_count % (speed.i_int+1)) ) continue;
353
354         /* Frame dropping if necessary */
355         if( aout_DateGet( &i_pts ) + GOOM_DELAY <= mdate() ) continue;
356
357         plane = goom_update( p_plugin_info, p_data, 0, 0.0,
358                              p_thread->psz_title, NULL );
359
360         free( p_thread->psz_title );
361         p_thread->psz_title = NULL;
362
363         while( !( p_pic = vout_CreatePicture( p_thread->p_vout, 0, 0, 0 ) ) &&
364                !p_thread->b_die )
365         {
366             msleep( VOUT_OUTMEM_SLEEP );
367         }
368
369         if( p_pic == NULL ) break;
370
371         memcpy( p_pic->p[0].p_pixels, plane, width.i_int * height.i_int * 4 );
372         vout_DatePicture( p_thread->p_vout, p_pic,
373                           aout_DateGet( &i_pts ) + GOOM_DELAY );
374         vout_DisplayPicture( p_thread->p_vout, p_pic );
375     }
376
377     goom_close( p_plugin_info );
378 }
379
380 /*****************************************************************************
381  * Close: close the plugin
382  *****************************************************************************/
383 static void Close( vlc_object_t *p_this )
384 {
385     aout_filter_t     *p_filter = (aout_filter_t *)p_this;
386     aout_filter_sys_t *p_sys = p_filter->p_sys;
387
388     /* Stop Goom Thread */
389     vlc_object_kill( p_sys->p_thread );
390
391     vlc_mutex_lock( &p_sys->p_thread->lock );
392     vlc_cond_signal( &p_sys->p_thread->wait );
393     vlc_mutex_unlock( &p_sys->p_thread->lock );
394
395     vlc_thread_join( p_sys->p_thread );
396
397     /* Free data */
398     vout_Request( p_filter, p_sys->p_thread->p_vout, 0 );
399     vlc_mutex_destroy( &p_sys->p_thread->lock );
400     vlc_cond_destroy( &p_sys->p_thread->wait );
401     vlc_object_detach( p_sys->p_thread );
402
403     while( p_sys->p_thread->i_blocks-- )
404     {
405         block_Release( p_sys->p_thread->pp_blocks[p_sys->p_thread->i_blocks] );
406     }
407
408     vlc_object_release( p_sys->p_thread );
409
410     free( p_sys );
411 }
412
413 static char *TitleGet( vlc_object_t *p_this )
414 {
415     char *psz_title = NULL;
416     input_thread_t *p_input =
417         vlc_object_find( p_this, VLC_OBJECT_INPUT, FIND_ANYWHERE );
418
419     if( p_input )
420     {
421         psz_title = input_item_GetTitle( input_GetItem( p_input ) );
422         if( EMPTY_STR( psz_title ) )
423         {
424             free( psz_title );
425             char *psz_orig = input_item_GetURI( input_GetItem( p_input ) );
426             char *psz = strrchr( psz_orig, '/' );
427
428             if( psz )
429             {
430                 psz++;
431             }
432             else
433             {
434                 psz = psz_orig;
435             }
436             if( psz && *psz )
437             {
438                 psz_title = strdup( psz );
439             }
440             free( psz_orig );
441         }
442         vlc_object_release( p_input );
443     }
444
445     return psz_title;
446 }