1 /* $Id: scan.c,v 1.52 2005/11/25 15:05:25 titer Exp $
3 This file is part of the HandBrake source code.
4 Homepage: <http://handbrake.m0k.org/>.
5 It may be used under the terms of the GNU General Public License. */
8 #include "a52dec/a52.h"
16 hb_list_t * list_title;
22 static void ScanFunc( void * );
23 static int DecodePreviews( hb_scan_t *, hb_title_t * title );
24 static void LookForAC3( hb_title_t * title, hb_buffer_t * b );
25 static int AllAC3OK( hb_title_t * title );
27 hb_thread_t * hb_scan_init( hb_handle_t * handle, const char * path,
28 int title_index, hb_list_t * list_title )
30 hb_scan_t * data = calloc( sizeof( hb_scan_t ), 1 );
33 data->path = strdup( path );
34 data->title_index = title_index;
35 data->list_title = list_title;
37 return hb_thread_init( "scan", ScanFunc, data, HB_NORMAL_PRIORITY );
40 static void ScanFunc( void * _data )
42 hb_scan_t * data = (hb_scan_t *) _data;
46 /* Try to open the path as a DVD. If it fails, try as a file */
47 hb_log( "scan: trying to open with libdvdread" );
48 if( ( data->dvd = hb_dvd_init( data->path ) ) )
50 hb_log( "scan: DVD has %d title(s)",
51 hb_dvd_title_count( data->dvd ) );
52 if( data->title_index )
54 /* Scan this title only */
55 hb_list_add( data->list_title, hb_dvd_title_scan( data->dvd,
56 data->title_index ) );
61 for( i = 0; i < hb_dvd_title_count( data->dvd ); i++ )
63 hb_list_add( data->list_title,
64 hb_dvd_title_scan( data->dvd, i + 1 ) );
70 /* Open as a VOB file */
72 hb_log( "scan: trying to open as VOB file" );
73 file = fopen( data->path, "rb" );
81 hb_log( "scan: fopen failed" );
86 for( i = 0; i < hb_list_count( data->list_title ); )
91 hb_title_t * title_tmp = NULL;
93 title = hb_list_item( data->list_title, i );
95 /* I've seen a DVD with strictly identical titles. Check this
96 here and ignore it if redundant */
97 for( j = 0; j < i; j++ )
99 title_tmp = hb_list_item( data->list_title, j );
100 if( title->vts == title_tmp->vts &&
101 title->block_start == title_tmp->block_start &&
102 title->block_end == title_tmp->block_end &&
103 title->block_count == title_tmp->block_count )
114 hb_log( "scan: title %d is duplicate with title %d",
115 title->index, title_tmp->index );
116 hb_list_rem( data->list_title, title );
121 #define p state.param.scanning
123 state.state = HB_STATE_SCANNING;
124 p.title_cur = title->index;
125 p.title_count = hb_dvd_title_count( data->dvd );
126 hb_set_state( data->h, &state );
129 /* Decode previews */
130 if( !DecodePreviews( data, title ) )
132 /* TODO: free things */
133 hb_list_rem( data->list_title, title );
137 /* Make sure we found AC3 rates and bitrates */
138 for( j = 0; j < hb_list_count( title->list_audio ); )
140 audio = hb_list_item( title->list_audio, j );
141 if( audio->codec == HB_ACODEC_AC3 &&
144 hb_list_rem( title->list_audio, audio );
151 /* Do we still have audio */
152 if( !hb_list_count( title->list_audio ) )
154 hb_list_rem( data->list_title, title );
162 /* Init jobs templates */
163 for( i = 0; i < hb_list_count( data->list_title ); i++ )
167 title = hb_list_item( data->list_title, i );
168 job = calloc( sizeof( hb_job_t ), 1 );
173 /* Set defaults settings */
174 job->chapter_start = 1;
175 job->chapter_end = hb_list_count( title->list_chapter );
177 /* Autocrop by default. Gnark gnark */
178 memcpy( job->crop, title->crop, 4 * sizeof( int ) );
180 job->width = title->width - job->crop[2] - job->crop[3];
181 hb_fix_aspect( job, HB_KEEP_WIDTH );
182 if( job->height > title->height - job->crop[0] - job->crop[1] )
184 job->height = title->height - job->crop[0] - job->crop[1];
185 hb_fix_aspect( job, HB_KEEP_HEIGHT );
189 job->vcodec = HB_VCODEC_FFMPEG;
190 job->vquality = -1.0;
191 job->vbitrate = 1000;
193 job->vrate = title->rate;
194 job->vrate_base = title->rate_base;
199 job->acodec = HB_ACODEC_FAAC;
205 job->mux = HB_MUX_MP4;
210 hb_dvd_close( &data->dvd );
214 /***********************************************************************
216 ***********************************************************************
217 * Decode 10 pictures for the given title.
218 * It assumes that data->reader and data->vts have successfully been
219 * DVDOpen()ed and ifoOpen()ed.
220 **********************************************************************/
221 static int DecodePreviews( hb_scan_t * data, hb_title_t * title )
224 hb_buffer_t * buf_ps, * buf_es, * buf_raw;
225 hb_list_t * list_es, * list_raw;
226 hb_libmpeg2_t * mpeg2;
228 buf_ps = hb_buffer_init( 2048 );
229 list_es = hb_list_init();
230 list_raw = hb_list_init();
232 hb_log( "scan: decoding previews for title %d", title->index );
234 hb_dvd_start( data->dvd, title->index, 1 );
236 for( i = 0; i < 10; i++ )
242 if( !hb_dvd_seek( data->dvd, (float) ( i + 1 ) / 11.0 ) )
247 hb_log( "scan: preview %d", i + 1 );
249 mpeg2 = hb_libmpeg2_init();
251 for( j = 0; j < 10240 ; j++ )
253 if( !hb_dvd_read( data->dvd, buf_ps ) )
257 hb_demux_ps( buf_ps, list_es );
259 while( ( buf_es = hb_list_item( list_es, 0 ) ) )
261 hb_list_rem( list_es, buf_es );
262 if( buf_es->id == 0xE0 && !hb_list_count( list_raw ) )
264 hb_libmpeg2_decode( mpeg2, buf_es, list_raw );
268 LookForAC3( title, buf_es );
270 hb_buffer_close( &buf_es );
272 if( hb_list_count( list_raw ) &&
273 ( i || AllAC3OK( title ) ) )
275 /* We got a picture */
280 if( hb_list_count( list_raw ) &&
281 ( i || AllAC3OK( title ) ) )
287 if( !hb_list_count( list_raw ) )
289 hb_log( "scan: could not get a decoded picture" );
295 /* Get size and rate infos */
296 title->rate = 27000000;
297 hb_libmpeg2_info( mpeg2, &title->width, &title->height,
299 title->crop[0] = title->crop[1] = title->height / 2;
300 title->crop[2] = title->crop[3] = title->width / 2;
303 hb_libmpeg2_close( &mpeg2 );
305 while( ( buf_es = hb_list_item( list_es, 0 ) ) )
307 hb_list_rem( list_es, buf_es );
308 hb_buffer_close( &buf_es );
311 buf_raw = hb_list_item( list_raw, 0 );
313 hb_get_tempory_filename( data->h, filename, "%x%d",
316 file_preview = fopen( filename, "w" );
319 fwrite( buf_raw->data, title->width * title->height * 3 / 2,
321 fclose( file_preview );
325 hb_log( "scan: fopen failed (%s)", filename );
328 #define Y buf_raw->data
331 /* Detect black borders */
333 for( j = 0; j < title->width; j++ )
335 for( k = 0; k < title->crop[0]; k++ )
336 if( Y[ k * title->width + j ] > DARK )
341 for( k = 0; k < title->crop[1]; k++ )
342 if( Y[ ( title->height - k - 1 ) *
343 title->width + j ] > DARK )
349 for( j = 0; j < title->height; j++ )
351 for( k = 0; k < title->crop[2]; k++ )
352 if( Y[ j * title->width + k ] > DARK )
357 for( k = 0; k < title->crop[3]; k++ )
358 if( Y[ j * title->width +
359 title->width - k - 1 ] > DARK )
366 while( ( buf_raw = hb_list_item( list_raw, 0 ) ) )
368 hb_list_rem( list_raw, buf_raw );
369 hb_buffer_close( &buf_raw );
373 title->crop[0] = EVEN( title->crop[0] );
374 title->crop[1] = EVEN( title->crop[1] );
375 title->crop[2] = EVEN( title->crop[2] );
376 title->crop[3] = EVEN( title->crop[3] );
378 hb_log( "scan: %dx%d, %.3f fps, autocrop = %d/%d/%d/%d",
379 title->width, title->height, (float) title->rate /
380 (float) title->rate_base, title->crop[0], title->crop[1],
381 title->crop[2], title->crop[3] );
390 hb_buffer_close( &buf_ps );
391 while( ( buf_es = hb_list_item( list_es, 0 ) ) )
393 hb_list_rem( list_es, buf_es );
394 hb_buffer_close( &buf_es );
396 hb_list_close( &list_es );
397 while( ( buf_raw = hb_list_item( list_raw, 0 ) ) )
399 hb_list_rem( list_raw, buf_raw );
400 hb_buffer_close( &buf_raw );
402 hb_list_close( &list_raw );
403 hb_dvd_stop( data->dvd );
407 static void LookForAC3( hb_title_t * title, hb_buffer_t * b )
414 /* Figure out if this is a AC3 buffer for a known track */
415 hb_audio_t * audio = NULL;
416 for( i = 0; i < hb_list_count( title->list_audio ); i++ )
418 audio = hb_list_item( title->list_audio, i );
419 if( audio->codec == HB_ACODEC_AC3 &&
436 /* Already done for this track */
440 for( i = 0; i < b->size - 7; i++ )
442 if( a52_syncinfo( &b->data[i], &flags, &rate, &bitrate ) )
444 hb_log( "scan: rate=%dHz, bitrate=%d", rate, bitrate );
446 audio->bitrate = bitrate;
447 switch( flags & A52_CHANNEL_MASK )
472 sprintf( audio->lang + strlen( audio->lang ),
473 " (%d ch)", audio->channels );
479 static int AllAC3OK( hb_title_t * title )
484 for( i = 0; i < hb_list_count( title->list_audio ); i++ )
486 audio = hb_list_item( title->list_audio, i );
487 if( audio->codec == HB_ACODEC_AC3 &&