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 if( title->aspect == 16 )
182 hb_reduce( &job->pixel_aspect_width, &job->pixel_aspect_height,
183 16 * title->height, 9 * title->width );
187 hb_reduce( &job->pixel_aspect_width, &job->pixel_aspect_height,
188 4 * title->height, 3 * title->width );
191 job->width = title->width - job->crop[2] - job->crop[3];
192 // job->height = title->height - job->crop[0] - job->crop[1];
193 hb_fix_aspect( job, HB_KEEP_WIDTH );
194 if( job->height > title->height - job->crop[0] - job->crop[1] )
196 job->height = title->height - job->crop[0] - job->crop[1];
197 hb_fix_aspect( job, HB_KEEP_HEIGHT );
200 hb_log( "scan: title (%d) job->width:%d, job->height:%d",
201 i,job->width, job->height );
205 job->vcodec = HB_VCODEC_FFMPEG;
206 job->vquality = -1.0;
207 job->vbitrate = 1000;
209 job->vrate = title->rate;
210 job->vrate_base = title->rate_base;
215 job->acodec = HB_ACODEC_FAAC;
221 job->mux = HB_MUX_MP4;
226 hb_dvd_close( &data->dvd );
230 /***********************************************************************
232 ***********************************************************************
233 * Decode 10 pictures for the given title.
234 * It assumes that data->reader and data->vts have successfully been
235 * DVDOpen()ed and ifoOpen()ed.
236 **********************************************************************/
237 static int DecodePreviews( hb_scan_t * data, hb_title_t * title )
240 hb_buffer_t * buf_ps, * buf_es, * buf_raw;
241 hb_list_t * list_es, * list_raw;
242 hb_libmpeg2_t * mpeg2;
244 buf_ps = hb_buffer_init( 2048 );
245 list_es = hb_list_init();
246 list_raw = hb_list_init();
248 hb_log( "scan: decoding previews for title %d", title->index );
250 hb_dvd_start( data->dvd, title->index, 1 );
252 for( i = 0; i < 10; i++ )
258 if( !hb_dvd_seek( data->dvd, (float) ( i + 1 ) / 11.0 ) )
263 hb_log( "scan: preview %d", i + 1 );
265 mpeg2 = hb_libmpeg2_init();
267 for( j = 0; j < 10240 ; j++ )
269 if( !hb_dvd_read( data->dvd, buf_ps ) )
273 hb_demux_ps( buf_ps, list_es );
275 while( ( buf_es = hb_list_item( list_es, 0 ) ) )
277 hb_list_rem( list_es, buf_es );
278 if( buf_es->id == 0xE0 && !hb_list_count( list_raw ) )
280 hb_libmpeg2_decode( mpeg2, buf_es, list_raw );
284 LookForAC3( title, buf_es );
286 hb_buffer_close( &buf_es );
288 if( hb_list_count( list_raw ) &&
289 ( i || AllAC3OK( title ) ) )
291 /* We got a picture */
296 if( hb_list_count( list_raw ) &&
297 ( i || AllAC3OK( title ) ) )
303 if( !hb_list_count( list_raw ) )
305 hb_log( "scan: could not get a decoded picture" );
311 /* Get size and rate infos */
312 title->rate = 27000000;
313 hb_libmpeg2_info( mpeg2, &title->width, &title->height,
315 title->crop[0] = title->crop[1] = title->height / 2;
316 title->crop[2] = title->crop[3] = title->width / 2;
319 hb_libmpeg2_close( &mpeg2 );
321 while( ( buf_es = hb_list_item( list_es, 0 ) ) )
323 hb_list_rem( list_es, buf_es );
324 hb_buffer_close( &buf_es );
327 buf_raw = hb_list_item( list_raw, 0 );
329 hb_get_tempory_filename( data->h, filename, "%x%d",
332 file_preview = fopen( filename, "w" );
335 fwrite( buf_raw->data, title->width * title->height * 3 / 2,
337 fclose( file_preview );
341 hb_log( "scan: fopen failed (%s)", filename );
344 #define Y buf_raw->data
347 /* Detect black borders */
349 for( j = 0; j < title->width; j++ )
351 for( k = 0; k < title->crop[0]; k++ )
352 if( Y[ k * title->width + j ] > DARK )
357 for( k = 0; k < title->crop[1]; k++ )
358 if( Y[ ( title->height - k - 1 ) *
359 title->width + j ] > DARK )
365 for( j = 0; j < title->height; j++ )
367 for( k = 0; k < title->crop[2]; k++ )
368 if( Y[ j * title->width + k ] > DARK )
373 for( k = 0; k < title->crop[3]; k++ )
374 if( Y[ j * title->width +
375 title->width - k - 1 ] > DARK )
382 while( ( buf_raw = hb_list_item( list_raw, 0 ) ) )
384 hb_list_rem( list_raw, buf_raw );
385 hb_buffer_close( &buf_raw );
389 title->crop[0] = EVEN( title->crop[0] );
390 title->crop[1] = EVEN( title->crop[1] );
391 title->crop[2] = EVEN( title->crop[2] );
392 title->crop[3] = EVEN( title->crop[3] );
394 hb_log( "scan: %dx%d, %.3f fps, autocrop = %d/%d/%d/%d",
395 title->width, title->height, (float) title->rate /
396 (float) title->rate_base, title->crop[0], title->crop[1],
397 title->crop[2], title->crop[3] );
406 hb_buffer_close( &buf_ps );
407 while( ( buf_es = hb_list_item( list_es, 0 ) ) )
409 hb_list_rem( list_es, buf_es );
410 hb_buffer_close( &buf_es );
412 hb_list_close( &list_es );
413 while( ( buf_raw = hb_list_item( list_raw, 0 ) ) )
415 hb_list_rem( list_raw, buf_raw );
416 hb_buffer_close( &buf_raw );
418 hb_list_close( &list_raw );
419 hb_dvd_stop( data->dvd );
423 static void LookForAC3( hb_title_t * title, hb_buffer_t * b )
430 /* Figure out if this is a AC3 buffer for a known track */
431 hb_audio_t * audio = NULL;
432 for( i = 0; i < hb_list_count( title->list_audio ); i++ )
434 audio = hb_list_item( title->list_audio, i );
435 if( audio->codec == HB_ACODEC_AC3 &&
452 /* Already done for this track */
456 for( i = 0; i < b->size - 7; i++ )
458 if( a52_syncinfo( &b->data[i], &flags, &rate, &bitrate ) )
460 hb_log( "scan: rate=%dHz, bitrate=%d", rate, bitrate );
462 audio->bitrate = bitrate;
463 switch( flags & A52_CHANNEL_MASK )
488 sprintf( audio->lang + strlen( audio->lang ),
489 " (%d ch)", audio->channels );
495 static int AllAC3OK( hb_title_t * title )
500 for( i = 0; i < hb_list_count( title->list_audio ); i++ )
502 audio = hb_list_item( title->list_audio, i );
503 if( audio->codec == HB_ACODEC_AC3 &&