1 /* $Id: scan.c,v 1.51 2005/04/27 21:05:24 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 hb_dvd_seek( data->dvd, (float) ( i + 1 ) / 11.0 );
244 hb_log( "scan: preview %d", i + 1 );
246 mpeg2 = hb_libmpeg2_init();
248 for( j = 0; j < 10240; j++ )
250 if( !hb_dvd_read( data->dvd, buf_ps ) )
254 hb_demux_ps( buf_ps, list_es );
256 while( ( buf_es = hb_list_item( list_es, 0 ) ) )
258 hb_list_rem( list_es, buf_es );
259 if( buf_es->id == 0xE0 && !hb_list_count( list_raw ) )
261 hb_libmpeg2_decode( mpeg2, buf_es, list_raw );
265 LookForAC3( title, buf_es );
267 hb_buffer_close( &buf_es );
269 if( hb_list_count( list_raw ) &&
270 ( i || AllAC3OK( title ) ) )
272 /* We got a picture */
277 if( hb_list_count( list_raw ) &&
278 ( i || AllAC3OK( title ) ) )
284 if( !hb_list_count( list_raw ) )
286 hb_log( "scan: could not get a decoded picture" );
292 /* Get size and rate infos */
293 title->rate = 27000000;
294 hb_libmpeg2_info( mpeg2, &title->width, &title->height,
296 title->crop[0] = title->crop[1] = title->height / 2;
297 title->crop[2] = title->crop[3] = title->width / 2;
300 hb_libmpeg2_close( &mpeg2 );
302 while( ( buf_es = hb_list_item( list_es, 0 ) ) )
304 hb_list_rem( list_es, buf_es );
305 hb_buffer_close( &buf_es );
308 buf_raw = hb_list_item( list_raw, 0 );
310 hb_get_tempory_filename( data->h, filename, "%x%d",
313 file_preview = fopen( filename, "w" );
316 fwrite( buf_raw->data, title->width * title->height * 3 / 2,
318 fclose( file_preview );
322 hb_log( "scan: fopen failed (%s)", filename );
325 #define Y buf_raw->data
328 /* Detect black borders */
330 for( j = 0; j < title->width; j++ )
332 for( k = 0; k < title->crop[0]; k++ )
333 if( Y[ k * title->width + j ] > DARK )
338 for( k = 0; k < title->crop[1]; k++ )
339 if( Y[ ( title->height - k - 1 ) *
340 title->width + j ] > DARK )
346 for( j = 0; j < title->height; j++ )
348 for( k = 0; k < title->crop[2]; k++ )
349 if( Y[ j * title->width + k ] > DARK )
354 for( k = 0; k < title->crop[3]; k++ )
355 if( Y[ j * title->width +
356 title->width - k - 1 ] > DARK )
363 while( ( buf_raw = hb_list_item( list_raw, 0 ) ) )
365 hb_list_rem( list_raw, buf_raw );
366 hb_buffer_close( &buf_raw );
370 title->crop[0] = EVEN( title->crop[0] );
371 title->crop[1] = EVEN( title->crop[1] );
372 title->crop[2] = EVEN( title->crop[2] );
373 title->crop[3] = EVEN( title->crop[3] );
375 hb_log( "scan: %dx%d, %.3f fps, autocrop = %d/%d/%d/%d",
376 title->width, title->height, (float) title->rate /
377 (float) title->rate_base, title->crop[0], title->crop[1],
378 title->crop[2], title->crop[3] );
387 hb_buffer_close( &buf_ps );
388 while( ( buf_es = hb_list_item( list_es, 0 ) ) )
390 hb_list_rem( list_es, buf_es );
391 hb_buffer_close( &buf_es );
393 hb_list_close( &list_es );
394 while( ( buf_raw = hb_list_item( list_raw, 0 ) ) )
396 hb_list_rem( list_raw, buf_raw );
397 hb_buffer_close( &buf_raw );
399 hb_list_close( &list_raw );
403 static void LookForAC3( hb_title_t * title, hb_buffer_t * b )
410 /* Figure out if this is a AC3 buffer for a known track */
411 hb_audio_t * audio = NULL;
412 for( i = 0; i < hb_list_count( title->list_audio ); i++ )
414 audio = hb_list_item( title->list_audio, i );
415 if( audio->codec == HB_ACODEC_AC3 &&
432 /* Already done for this track */
436 for( i = 0; i < b->size - 7; i++ )
438 if( a52_syncinfo( &b->data[i], &flags, &rate, &bitrate ) )
440 hb_log( "scan: rate=%dHz, bitrate=%d", rate, bitrate );
442 audio->bitrate = bitrate;
443 switch( flags & A52_CHANNEL_MASK )
468 sprintf( audio->lang + strlen( audio->lang ),
469 " (%d ch)", audio->channels );
475 static int AllAC3OK( hb_title_t * title )
480 for( i = 0; i < hb_list_count( title->list_audio ); i++ )
482 audio = hb_list_item( title->list_audio, i );
483 if( audio->codec == HB_ACODEC_AC3 &&