]> git.sesse.net Git - vlc/blob - modules/access/cdda.c
Simplify the CDDA module. Split more between parse and play
[vlc] / modules / access / cdda.c
1 /*****************************************************************************
2  * cdda.c : CD digital audio input module for vlc
3  *****************************************************************************
4  * Copyright (C) 2000, 2003 the VideoLAN team
5  * $Id$
6  *
7  * Authors: Laurent Aimar <fenrir@via.ecp.fr>
8  *          Gildas Bazin <gbazin@netcourrier.com>
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  * Todo:
27  *   - Improve CDDB support (non-blocking, cache, ...)
28  *   - Fix tracknumber in MRL
29  */
30
31 /*****************************************************************************
32  * Preamble
33  *****************************************************************************/
34 #include <stdlib.h>
35
36 #include <vlc/vlc.h>
37 #include <vlc/input.h>
38
39 #include "codecs.h"
40 #include "vcd/cdrom.h"
41
42 #include <vlc_playlist.h>
43
44 #ifdef HAVE_LIBCDDB
45 #include <cddb/cddb.h>
46 #endif
47
48 #ifdef HAVE_ERRNO_H
49 #include <errno.h>
50 #endif
51
52 /*****************************************************************************
53  * Module descriptior
54  *****************************************************************************/
55 static int  Open ( vlc_object_t * );
56 static void Close( vlc_object_t * );
57
58 #define CACHING_TEXT N_("Caching value in ms")
59 #define CACHING_LONGTEXT N_( \
60     "Default caching value for Audio CDs. This " \
61     "value should be set in milliseconds." )
62
63 vlc_module_begin();
64     set_shortname( _("Audio CD"));
65     set_description( _("Audio CD input") );
66     set_capability( "access2", 10 );
67     set_category( CAT_INPUT );
68     set_subcategory( SUBCAT_INPUT_ACCESS );
69     set_callbacks( Open, Close );
70
71     add_usage_hint( N_("[cdda:][device][@[track]]") );
72     add_integer( "cdda-caching", DEFAULT_PTS_DELAY / 1000, NULL, CACHING_TEXT,
73                  CACHING_LONGTEXT, VLC_TRUE );
74
75     add_integer( "cdda-track", -1 , NULL, NULL, NULL, VLC_TRUE );
76         change_internal();
77     add_integer( "cdda-first-sector", -1, NULL, NULL, NULL, VLC_TRUE );
78         change_internal();
79     add_integer( "cdda-last-sector", -1, NULL, NULL, NULL, VLC_TRUE );
80         change_internal();
81
82     add_string( "cddb-server", "freedb.freedb.org", NULL,
83                 N_( "CDDB Server" ), N_( "Address of the CDDB server to use." ),
84                 VLC_TRUE );
85     add_integer( "cddb-port", 8880, NULL,
86                 N_( "CDDB port" ), N_( "CDDB Server port to use." ),
87                 VLC_TRUE );
88     add_shortcut( "cdda" );
89     add_shortcut( "cddasimple" );
90 vlc_module_end();
91
92
93 /* how many blocks VCDRead will read in each loop */
94 #define CDDA_BLOCKS_ONCE 20
95 #define CDDA_DATA_ONCE   (CDDA_BLOCKS_ONCE * CDDA_DATA_SIZE)
96
97 /*****************************************************************************
98  * Access: local prototypes
99  *****************************************************************************/
100 struct access_sys_t
101 {
102     vcddev_t    *vcddev;                            /* vcd device descriptor */
103
104     /* Current position */
105     int         i_sector;                                  /* Current Sector */
106     int *       p_sectors;                                  /* Track sectors */
107
108     /* Wave header for the output data */
109     WAVEHEADER  waveheader;
110     vlc_bool_t  b_header;
111
112     int         i_track;
113     int         i_first_sector;
114     int         i_last_sector;
115
116 #ifdef HAVE_LIBCDDB
117     cddb_disc_t *p_disc;
118 #endif
119 };
120
121 static block_t *Block( access_t * );
122 static int      Seek( access_t *, int64_t );
123 static int      Control( access_t *, int, va_list );
124
125 static int GetTracks( access_t *p_access, playlist_t *p_playlist,
126                       playlist_item_t *p_parent );
127
128 #ifdef HAVE_LIBCDDB
129 static void GetCDDBInfo( access_t *p_access, int i_titles, int *p_sectors );
130 #endif
131
132 /*****************************************************************************
133  * Open: open cdda
134  *****************************************************************************/
135 static int Open( vlc_object_t *p_this )
136 {
137     access_t     *p_access = (access_t*)p_this;
138     access_sys_t *p_sys;
139     vcddev_t *vcddev;
140     char *psz_name;
141     int i_mrl_tracknum = -1;
142
143     input_thread_t *p_input;
144     playlist_item_t *p_item = NULL;
145     playlist_t *p_playlist  = NULL;
146     int i_ret;
147
148     if( !p_access->psz_path || !*p_access->psz_path )
149     {
150         /* Only when selected */
151         if( !p_this->b_force ) return VLC_EGENERIC;
152
153         psz_name = var_CreateGetString( p_this, "cd-audio" );
154         if( !psz_name || !*psz_name )
155         {
156             if( psz_name ) free( psz_name );
157             return VLC_EGENERIC;
158         }
159     }
160     else psz_name = strdup( p_access->psz_path );
161
162 #ifdef WIN32
163     if( psz_name[0] && psz_name[1] == ':' &&
164         psz_name[2] == '\\' && psz_name[3] == '\0' ) psz_name[2] = '\0';
165 #endif
166
167     /* Open CDDA */
168     if( (vcddev = ioctl_Open( VLC_OBJECT(p_access), psz_name )) == NULL )
169     {
170         msg_Warn( p_access, "could not open %s", psz_name );
171         free( psz_name );
172         return VLC_EGENERIC;
173     }
174     free( psz_name );
175
176     /* Set up p_access */
177     p_access->pf_read = NULL;
178     p_access->pf_block = Block;
179     p_access->pf_control = Control;
180     p_access->pf_seek = Seek;
181     p_access->info.i_update = 0;
182     p_access->info.i_size = 0;
183     p_access->info.i_pos = 0;
184     p_access->info.b_eof = VLC_FALSE;
185     p_access->info.i_title = 0;
186     p_access->info.i_seekpoint = 0;
187     p_access->p_sys = p_sys = malloc( sizeof( access_sys_t ) );
188     memset( p_sys, 0, sizeof( access_sys_t ) );
189     p_sys->vcddev = vcddev;
190
191     /* We only do separate items if the whole disc is requested -
192      *  Dirty hack we access some private data ! */
193     p_input = (input_thread_t *)( p_access->p_parent );
194
195    /* Do we play a single track ? */
196    p_sys->i_track = var_CreateGetInteger( p_access, "cdda-track" );
197
198    if( p_sys->i_track < 0 && i_mrl_tracknum <= 0 )
199    {
200         p_playlist = (playlist_t *) vlc_object_find( p_access,
201                                        VLC_OBJECT_PLAYLIST, FIND_ANYWHERE );
202         if( !p_playlist ) return VLC_EGENERIC;
203
204         if( p_playlist->status.p_item->p_input ==
205              ((input_thread_t *)p_access->p_parent)->input.p_item )
206             p_item = p_playlist->status.p_item;
207         else
208         {
209             input_item_t *p_current = ( (input_thread_t*)p_access->p_parent)->
210                                          input.p_item;
211             p_item = playlist_LockItemGetByInput( p_playlist, p_current );
212
213             if( !p_item )
214             {
215                 msg_Dbg( p_playlist, "unable to find item in playlist");
216                 return -1;
217             }
218         }
219
220         i_ret = GetTracks( p_access, p_playlist, p_item );
221
222         if( p_playlist ) vlc_object_release( p_playlist );
223         if( i_ret < 0 ) goto error;
224     }
225     else
226     {
227         /* Build a WAV header for the output data */
228         memset( &p_sys->waveheader, 0, sizeof(WAVEHEADER) );
229         SetWLE( &p_sys->waveheader.Format, 1 ); /*WAVE_FORMAT_PCM*/
230         SetWLE( &p_sys->waveheader.BitsPerSample, 16);
231         p_sys->waveheader.MainChunkID = VLC_FOURCC('R', 'I', 'F', 'F');
232         p_sys->waveheader.Length = 0;               /* we just don't know */
233         p_sys->waveheader.ChunkTypeID = VLC_FOURCC('W', 'A', 'V', 'E');
234         p_sys->waveheader.SubChunkID = VLC_FOURCC('f', 'm', 't', ' ');
235         SetDWLE( &p_sys->waveheader.SubChunkLength, 16);
236         SetWLE( &p_sys->waveheader.Modus, 2);
237         SetDWLE( &p_sys->waveheader.SampleFreq, 44100);
238         SetWLE( &p_sys->waveheader.BytesPerSample,
239                     2 /*Modus*/ * 16 /*BitsPerSample*/ / 8 );
240         SetDWLE( &p_sys->waveheader.BytesPerSec,
241                     2*16/8 /*BytesPerSample*/ * 44100 /*SampleFreq*/ );
242         p_sys->waveheader.DataChunkID = VLC_FOURCC('d', 'a', 't', 'a');
243         p_sys->waveheader.DataLength = 0;           /* we just don't know */
244
245         p_sys->i_first_sector = var_CreateGetInteger( p_access,
246                                                       "cdda-first-sector" );
247         p_sys->i_last_sector  = var_CreateGetInteger( p_access,
248                                                       "cdda-last-sector" );
249         /* Tracknumber in MRL */
250         if( p_sys->i_first_sector < 0 || p_sys->i_last_sector < 0 )
251         {
252             int i_titles;
253             if( i_mrl_tracknum <= 0 )
254             {
255                 msg_Err( p_access, "wrong sector information" );
256                 goto error;
257             }
258             i_titles = ioctl_GetTracksMap( VLC_OBJECT(p_access),
259                                             p_sys->vcddev, &p_sys->p_sectors );
260         }
261
262
263         p_sys->i_sector = p_sys->i_first_sector;
264         p_access->info.i_size = (p_sys->i_last_sector - p_sys->i_first_sector)
265                                      * (int64_t)CDDA_DATA_SIZE;
266     }
267
268     /* PTS delay */
269     var_Create( p_access, "cdda-caching", VLC_VAR_INTEGER|VLC_VAR_DOINHERIT );
270
271     return VLC_SUCCESS;
272
273 error:
274     ioctl_Close( VLC_OBJECT(p_access), p_sys->vcddev );
275     free( p_sys );
276     if( p_playlist ) vlc_object_release( p_playlist );
277     return VLC_EGENERIC;
278 }
279
280 /*****************************************************************************
281  * Close: closes cdda
282  *****************************************************************************/
283 static void Close( vlc_object_t *p_this )
284 {
285     access_t     *p_access = (access_t *)p_this;
286     access_sys_t *p_sys = p_access->p_sys;
287     ioctl_Close( p_this, p_sys->vcddev );
288     free( p_sys );
289 }
290
291 /*****************************************************************************
292  * Block: read data (CDDA_DATA_ONCE)
293  *****************************************************************************/
294 static block_t *Block( access_t *p_access )
295 {
296     access_sys_t *p_sys = p_access->p_sys;
297     int i_blocks = CDDA_BLOCKS_ONCE;
298     block_t *p_block;
299
300     if( p_sys->i_track < 0 ) p_access->info.b_eof = VLC_TRUE;
301
302     /* Check end of file */
303     if( p_access->info.b_eof ) return NULL;
304
305     if( !p_sys->b_header )
306     {
307         /* Return only the header */
308         p_block = block_New( p_access, sizeof( WAVEHEADER ) );
309         memcpy( p_block->p_buffer, &p_sys->waveheader, sizeof(WAVEHEADER) );
310         p_sys->b_header = VLC_TRUE;
311         return p_block;
312     }
313
314     if( p_sys->i_sector >= p_sys->i_last_sector )
315     {
316         p_access->info.b_eof = VLC_TRUE;
317         return NULL;
318     }
319
320     /* Don't read too far */
321     if( p_sys->i_sector + i_blocks >= p_sys->i_last_sector )
322         i_blocks = p_sys->i_last_sector - p_sys->i_sector;
323
324     /* Do the actual reading */
325     if( !( p_block = block_New( p_access, i_blocks * CDDA_DATA_SIZE ) ) )
326     {
327         msg_Err( p_access, "cannot get a new block of size: %i",
328                  i_blocks * CDDA_DATA_SIZE );
329         return NULL;
330     }
331
332     if( ioctl_ReadSectors( VLC_OBJECT(p_access), p_sys->vcddev,
333             p_sys->i_sector, p_block->p_buffer, i_blocks, CDDA_TYPE ) < 0 )
334     {
335         msg_Err( p_access, "cannot read sector %i", p_sys->i_sector );
336         block_Release( p_block );
337
338         /* Try to skip one sector (in case of bad sectors) */
339         p_sys->i_sector++;
340         p_access->info.i_pos += CDDA_DATA_SIZE;
341         return NULL;
342     }
343
344     /* Update a few values */
345     p_sys->i_sector += i_blocks;
346     p_access->info.i_pos += p_block->i_buffer;
347
348     return p_block;
349 }
350
351 /****************************************************************************
352  * Seek
353  ****************************************************************************/
354 static int Seek( access_t *p_access, int64_t i_pos )
355 {
356     access_sys_t *p_sys = p_access->p_sys;
357
358     /* Next sector to read */
359     p_sys->i_sector = p_sys->i_first_sector + i_pos / CDDA_DATA_SIZE;
360     p_access->info.i_pos = i_pos;
361
362     return VLC_SUCCESS;
363 }
364
365 /*****************************************************************************
366  * Control:
367  *****************************************************************************/
368 static int Control( access_t *p_access, int i_query, va_list args )
369 {
370     vlc_bool_t   *pb_bool;
371     int          *pi_int;
372     int64_t      *pi_64;
373
374     switch( i_query )
375     {
376         case ACCESS_CAN_SEEK:
377         case ACCESS_CAN_FASTSEEK:
378         case ACCESS_CAN_PAUSE:
379         case ACCESS_CAN_CONTROL_PACE:
380             pb_bool = (vlc_bool_t*)va_arg( args, vlc_bool_t* );
381             *pb_bool = VLC_TRUE;
382             break;
383
384         case ACCESS_GET_MTU:
385             pi_int = (int*)va_arg( args, int * );
386             *pi_int = CDDA_DATA_ONCE;
387             break;
388
389         case ACCESS_GET_PTS_DELAY:
390             pi_64 = (int64_t*)va_arg( args, int64_t * );
391             *pi_64 = var_GetInteger( p_access, "cdda-caching" ) * 1000;
392             break;
393
394         case ACCESS_SET_PAUSE_STATE:
395         case ACCESS_GET_TITLE_INFO:
396         case ACCESS_SET_TITLE:
397         case ACCESS_GET_META:
398         case ACCESS_SET_SEEKPOINT:
399         case ACCESS_SET_PRIVATE_ID_STATE:
400             return VLC_EGENERIC;
401
402         default:
403             msg_Warn( p_access, "unimplemented query in control" );
404             return VLC_EGENERIC;
405
406     }
407     return VLC_SUCCESS;
408 }
409
410 static int GetTracks( access_t *p_access,
411                       playlist_t *p_playlist, playlist_item_t *p_parent )
412 {
413     access_sys_t *p_sys = p_access->p_sys;
414     int i, i_titles;
415     input_item_t *p_input_item;
416     playlist_item_t *p_item_in_category;
417     char *psz_name;
418     i_titles = ioctl_GetTracksMap( VLC_OBJECT(p_access),
419                                    p_sys->vcddev, &p_sys->p_sectors );
420     if( i_titles < 0 )
421     {
422         msg_Err( p_access, "unable to count tracks" );
423         return VLC_EGENERIC;;
424     }
425     else if( i_titles <= 0 )
426     {
427         msg_Err( p_access, "no audio tracks found" );
428         return VLC_EGENERIC;
429     }
430
431     p_item_in_category = playlist_LockItemToNode( p_playlist, p_parent );
432     psz_name = strdup( "Audio CD" );
433     vlc_mutex_lock( &p_playlist->object_lock );
434     playlist_ItemSetName( p_parent, psz_name );
435     vlc_mutex_unlock( &p_playlist->object_lock );
436     var_SetInteger( p_playlist, "item-change", p_parent->p_input->i_id );
437     free( psz_name );
438
439 #ifdef HAVE_LIBCDDB
440     GetCDDBInfo( p_access, i_titles, p_sys->p_sectors );
441     if( p_sys->p_disc )
442     {
443         if( cddb_disc_get_title( p_sys->p_disc ) )
444         {
445             asprintf( &psz_name, "%s", cddb_disc_get_title( p_sys->p_disc ) );
446             vlc_mutex_lock( &p_playlist->object_lock );
447             playlist_ItemSetName( p_parent, psz_name );
448             vlc_mutex_unlock( &p_playlist->object_lock );
449             var_SetInteger( p_playlist, "item-change",
450                             p_parent->p_input->i_id );
451             free( psz_name );
452         }
453     }
454 #endif
455
456     /* Build title table */
457     for( i = 0; i < i_titles; i++ )
458     {
459         msg_Dbg( p_access, "track[%d] start=%d", i, p_sys->p_sectors[i] );
460         char *psz_uri, *psz_opt, *psz_first, *psz_last;
461         int i_path_len = p_access->psz_path ? strlen( p_access->psz_path ) : 0;
462
463         psz_name = (char*)malloc( strlen( _("Audio CD - Track ") ) + 5 );
464         psz_opt = (char*)malloc( strlen( "cdda-track=" ) + 3 );
465         psz_first = (char*)malloc( strlen( "cdda-first-sector=" ) + 7 );
466         psz_last = (char*)malloc( strlen( "cdda-last-sector=" ) + 7 );
467         psz_uri = (char*)malloc( i_path_len + 13 );
468
469         snprintf( psz_uri, i_path_len + 13, "cdda://%s",
470                            p_access->psz_path ? p_access->psz_path : "" );
471         sprintf( psz_opt, "cdda-track=%i", i+1 );
472         sprintf( psz_first, "cdda-first-sector=%i",p_sys->p_sectors[i] );
473
474 //        if( i != i_titles -1 )
475             sprintf( psz_last, "cdda-last-sector=%i", p_sys->p_sectors[i+1] );
476 //         else
477 //            sprintf( psz_last, "cdda-last-sector=%i", 1242 /* FIXME */);
478
479         /* Define a "default name" */
480         sprintf( psz_name, _("Audio CD - Track %i"), (i+1) );
481
482         /* Create playlist items */
483         p_input_item = input_ItemNewWithType( VLC_OBJECT( p_playlist ),
484                                               psz_uri, psz_name, 0, NULL, -1,
485                                               ITEM_TYPE_DISC );
486         vlc_input_item_AddOption( p_input_item, psz_first );
487         vlc_input_item_AddOption( p_input_item, psz_last );
488         vlc_input_item_AddOption( p_input_item, psz_opt );
489
490 #ifdef HAVE_LIBCDDB
491         /* If we have CDDB info, change the name */
492         if( p_sys->p_disc )
493         {
494             char *psz_result;
495             cddb_track_t *t = cddb_disc_get_track( p_sys->p_disc, i );
496             if( t!= NULL )
497             {
498                 if( cddb_track_get_title( t )  != NULL )
499                 {
500                     vlc_input_item_AddInfo( p_input_item, _(VLC_META_INFO_CAT),
501                                             _(VLC_META_TITLE),
502                                             cddb_track_get_title( t ) );
503                     if( p_input_item->psz_name )
504                         free( p_input_item->psz_name );
505                     asprintf( &p_input_item->psz_name, "%s",
506                               cddb_track_get_title( t ) );
507                 }
508                 psz_result = cddb_track_get_artist( t );
509                 if( psz_result )
510                 {
511                     vlc_input_item_AddInfo( p_input_item, _(VLC_META_INFO_CAT),
512                                             _(VLC_META_ARTIST), psz_result );
513                 }
514             }
515         }
516 #endif
517         playlist_AddWhereverNeeded( p_playlist, p_input_item, p_parent,
518                                p_item_in_category, VLC_FALSE, PLAYLIST_APPEND );
519         free( psz_uri ); free( psz_opt ); free( psz_name );
520         free( psz_first ); free( psz_last );
521     }
522     return VLC_SUCCESS;
523 }
524
525 #ifdef HAVE_LIBCDDB
526 static void GetCDDBInfo( access_t *p_access, int i_titles, int *p_sectors )
527 {
528     int i, i_matches;
529     int64_t  i_length = 0, i_size = 0;
530     cddb_conn_t  *p_cddb = cddb_new();
531
532     if( !p_cddb )
533     {
534         msg_Warn( p_access, "unable to use CDDB" );
535         goto cddb_destroy;
536     }
537
538     cddb_set_email_address( p_cddb, "vlc@videolan.org" );
539     cddb_set_server_name( p_cddb, config_GetPsz( p_access, "cddb-server" ) );
540     cddb_set_server_port( p_cddb, config_GetInt( p_access, "cddb-port" ) );
541
542     /// \todo
543     cddb_cache_disable( p_cddb );
544
545 //    cddb_cache_set_dir( p_cddb,
546 //                     config_GetPsz( p_access,
547 //                                    MODULE_STRING "-cddb-cachedir") );
548
549     cddb_set_timeout( p_cddb, 10 );
550
551     /// \todo
552     cddb_http_disable( p_cddb);
553
554     p_access->p_sys->p_disc = cddb_disc_new();
555
556     if(! p_access->p_sys->p_disc )
557     {
558         msg_Err( p_access, "unable to create CDDB disc structure." );
559         goto cddb_end;
560     }
561
562     for(i = 0; i < i_titles ; i++ )
563     {
564         cddb_track_t *t = cddb_track_new();
565         cddb_track_set_frame_offset(t, p_sectors[i] );
566         cddb_disc_add_track( p_access->p_sys->p_disc, t );
567         i_size = ( p_sectors[i+1] - p_sectors[i] ) *
568                    (int64_t)CDDA_DATA_SIZE;
569         i_length += I64C(1000000) * i_size / 44100 / 4  ;
570     }
571
572     cddb_disc_set_length( p_access->p_sys->p_disc, (int)(i_length/1000000) );
573
574     if (!cddb_disc_calc_discid(p_access->p_sys->p_disc ))
575     {
576         msg_Err( p_access, "CDDB disc ID calculation failed" );
577         goto cddb_destroy;
578     }
579
580     i_matches = cddb_query( p_cddb, p_access->p_sys->p_disc);
581
582     if (i_matches > 0)
583     {
584         if (i_matches > 1)
585              msg_Warn( p_access, "found %d matches in CDDB. Using first one.",
586                                  i_matches);
587         cddb_read( p_cddb, p_access->p_sys->p_disc );
588     }
589     else
590         msg_Warn( p_access, "CDDB error: %s", cddb_error_str(errno));
591
592 cddb_destroy:
593     cddb_destroy( p_cddb);
594
595 cddb_end: ;
596 }
597 #endif /*HAVE_LIBCDDB*/