1 /* $Id: hb.c,v 1.43 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. */
9 #include "ffmpeg/avcodec.h"
13 /* The "Check for update" thread */
16 hb_thread_t * update_thread;
18 /* This thread's only purpose is to check other threads'
21 hb_thread_t * main_thread;
24 /* DVD/file scan thread */
25 hb_list_t * list_title;
26 hb_thread_t * scan_thread;
28 /* The thread which processes the jobs. Others threads are launched
29 from this one (see work.c) */
32 volatile int work_die;
34 hb_thread_t * work_thread;
38 hb_lock_t * state_lock;
42 hb_lock_t * pause_lock;
45 hb_work_object_t * hb_objects = NULL;
47 static void thread_func( void * );
49 void hb_register( hb_work_object_t * w )
55 hb_handle_t * hb_init_real( int verbose, int update_check )
57 hb_handle_t * h = calloc( sizeof( hb_handle_t ), 1 );
60 /* See hb_log() in common.c */
61 if( verbose > HB_DEBUG_NONE )
63 putenv( "HB_DEBUG=1" );
66 /* Check for an update on the website if asked to */
71 hb_log( "hb_init: checking for updates" );
73 h->update_thread = hb_update_init( &h->build, h->version );
77 if( hb_thread_has_exited( h->update_thread ) )
79 /* Immediate success or failure */
80 hb_thread_close( &h->update_thread );
83 if( hb_get_date() > date + 1000 )
85 /* Still nothing after one second. Connection problem,
87 hb_log( "hb_init: connection problem, not waiting for "
95 /* CPU count detection */
96 hb_log( "hb_init: checking cpu count" );
97 h->cpu_count = hb_get_cpu_count();
99 h->list_title = hb_list_init();
100 h->jobs = hb_list_init();
102 h->state_lock = hb_lock_init();
103 h->state.state = HB_STATE_IDLE;
105 h->pause_lock = hb_lock_init();
109 register_avcodec( &mpeg4_encoder );
110 register_avcodec( &mp2_decoder );
111 register_avcodec( &ac3_encoder );
113 /* Start library thread */
114 hb_log( "hb_init: starting libhb thread" );
116 h->main_thread = hb_thread_init( "libhb", thread_func, h,
117 HB_NORMAL_PRIORITY );
122 char * hb_get_version( hb_handle_t * h )
124 return "0.7.1a2";//HB_VERSION;
127 int hb_get_build( hb_handle_t * h )
132 int hb_check_update( hb_handle_t * h, char ** version )
134 *version = ( h->build < 0 ) ? NULL : h->version;
138 void hb_set_cpu_count( hb_handle_t * h, int cpu_count )
140 cpu_count = MAX( 1, cpu_count );
141 cpu_count = MIN( cpu_count, 8 );
142 h->cpu_count = cpu_count;
145 void hb_scan( hb_handle_t * h, const char * path, int title_index )
149 /* Clean up from previous scan */
150 while( ( title = hb_list_item( h->list_title, 0 ) ) )
152 hb_list_rem( h->list_title, title );
153 hb_title_close( &title );
156 hb_log( "hb_scan: path=%s, title_index=%d", path, title_index );
157 h->scan_thread = hb_scan_init( h, path, title_index, h->list_title );
160 hb_list_t * hb_get_titles( hb_handle_t * h )
162 return h->list_title;
165 void hb_get_preview( hb_handle_t * h, hb_title_t * title, int picture,
168 hb_job_t * job = title->job;
171 uint8_t * buf1, * buf2, * buf3, * buf4, * pen;
173 AVPicture pic1, pic2, pic3, pic4;
174 ImgReSampleContext * context;
177 buf1 = malloc( title->width * title->height * 3 / 2 );
178 buf2 = malloc( title->width * title->height * 3 / 2 );
179 buf3 = malloc( title->width * title->height * 3 / 2 );
180 buf4 = malloc( title->width * title->height * 4 );
181 avpicture_fill( &pic1, buf1, PIX_FMT_YUV420P,
182 title->width, title->height );
183 avpicture_fill( &pic2, buf2, PIX_FMT_YUV420P,
184 title->width, title->height );
185 avpicture_fill( &pic3, buf3, PIX_FMT_YUV420P,
186 job->width, job->height );
187 avpicture_fill( &pic4, buf4, PIX_FMT_RGBA32,
188 job->width, job->height );
190 memset( filename, 0, 1024 );
192 hb_get_tempory_filename( h, filename, "%x%d",
193 (int) title, picture );
195 file = fopen( filename, "r" );
198 hb_log( "hb_get_preview: fopen failed" );
202 fread( buf1, title->width * title->height * 3 / 2, 1, file );
205 context = img_resample_full_init(
206 job->width, job->height, title->width, title->height,
207 job->crop[0], job->crop[1], job->crop[2], job->crop[3],
210 if( job->deinterlace )
212 avpicture_deinterlace( &pic2, &pic1, PIX_FMT_YUV420P,
213 title->width, title->height );
214 img_resample( context, &pic3, &pic2 );
218 img_resample( context, &pic3, &pic1 );
220 img_convert( &pic4, PIX_FMT_RGBA32, &pic3, PIX_FMT_YUV420P,
221 job->width, job->height );
223 /* Gray background */
224 p32 = (uint32_t *) buffer;
225 for( i = 0; i < ( title->width + 2 ) * ( title->height + 2 ); i++ )
230 /* Draw the picture, centered, and draw the cropping zone */
231 pen = buffer + ( title->height - job->height ) *
232 ( title->width + 2 ) * 2 + ( title->width - job->width ) * 2;
233 memset( pen, 0xFF, 4 * ( job->width + 2 ) );
234 pen += 4 * ( title->width + 2 );
235 for( i = 0; i < job->height; i++ )
238 nextLine = pen + 4 * ( title->width + 2 );
239 memset( pen, 0xFF, 4 );
241 memcpy( pen, buf4 + 4 * job->width * i, 4 * job->width );
242 pen += 4 * job->width;
243 memset( pen, 0xFF, 4 );
246 memset( pen, 0xFF, 4 * ( job->width + 2 ) );
254 void hb_set_size( hb_job_t * job, int aspect, int pixels )
256 hb_title_t * title = job->title;
258 int croppedWidth = title->width - title->crop[2] - title->crop[3];
259 int croppedHeight = title->height - title->crop[0] - title->crop[1];
260 int croppedAspect = title->aspect * title->height * croppedWidth /
261 croppedHeight / title->width;
267 /* Keep the best possible aspect ratio */
268 aspect = croppedAspect;
271 /* Crop if necessary to obtain the desired ratio */
272 memcpy( job->crop, title->crop, 4 * sizeof( int ) );
273 if( aspect < croppedAspect )
275 /* Need to crop on the left and right */
276 addCrop = croppedWidth - aspect * croppedHeight * title->width /
277 title->aspect / title->height;
280 addCrop = ( addCrop + 1 ) / 2;
281 job->crop[2] += addCrop;
282 job->crop[3] += addCrop;
284 else if( addCrop & 2 )
287 job->crop[2] += addCrop - 1;
288 job->crop[3] += addCrop + 1;
293 job->crop[2] += addCrop;
294 job->crop[3] += addCrop;
297 else if( aspect > croppedAspect )
299 /* Need to crop on the top and bottom */
300 addCrop = croppedHeight - croppedWidth * title->aspect *
301 title->height / aspect / title->width;
304 addCrop = ( addCrop + 1 ) / 2;
305 job->crop[0] += addCrop;
306 job->crop[1] += addCrop;
308 else if( addCrop & 2 )
311 job->crop[0] += addCrop - 1;
312 job->crop[1] += addCrop + 1;
317 job->crop[0] += addCrop;
318 job->crop[1] += addCrop;
322 /* Compute a resolution from the number of pixels and aspect */
326 h = MULTIPLE_16( w * HB_ASPECT_BASE / aspect );
334 job->height = MULTIPLE_16( 16 * i * HB_ASPECT_BASE / aspect );
337 int hb_count( hb_handle_t * h )
339 return hb_list_count( h->jobs );
342 hb_job_t * hb_job( hb_handle_t * h, int i )
344 return hb_list_item( h->jobs, i );
347 /* hb_add: memcpy() party. That's ugly, for if someone has a better
349 void hb_add( hb_handle_t * h, hb_job_t * job )
352 hb_title_t * title, * title_copy;
353 hb_chapter_t * chapter, * chapter_copy;
354 hb_audio_t * audio, * audio_copy;
355 hb_subtitle_t * subtitle, * subtitle_copy;
360 title_copy = malloc( sizeof( hb_title_t ) );
361 memcpy( title_copy, title, sizeof( hb_title_t ) );
363 title_copy->list_chapter = hb_list_init();
364 for( i = 0; i < hb_list_count( title->list_chapter ); i++ )
366 chapter = hb_list_item( title->list_chapter, i );
367 chapter_copy = malloc( sizeof( hb_chapter_t ) );
368 memcpy( chapter_copy, chapter, sizeof( hb_chapter_t ) );
369 hb_list_add( title_copy->list_chapter, chapter_copy );
372 /* Copy the audio track(s) we want */
373 title_copy->list_audio = hb_list_init();
375 /* Do nothing about audio during first pass */
378 for( i = 0; i < 8; i++ )
380 if( job->audios[i] < 0 )
384 if( ( audio = hb_list_item( title->list_audio, job->audios[i] ) ) )
386 audio_copy = malloc( sizeof( hb_audio_t ) );
387 memcpy( audio_copy, audio, sizeof( hb_audio_t ) );
388 hb_list_add( title_copy->list_audio, audio_copy );
393 /* Copy the subtitle we want (or not) */
394 title_copy->list_subtitle = hb_list_init();
395 if( ( subtitle = hb_list_item( title->list_subtitle, job->subtitle ) ) )
397 subtitle_copy = malloc( sizeof( hb_subtitle_t ) );
398 memcpy( subtitle_copy, subtitle, sizeof( hb_subtitle_t ) );
399 hb_list_add( title_copy->list_subtitle, subtitle_copy );
403 job_copy = calloc( sizeof( hb_job_t ), 1 );
404 memcpy( job_copy, job, sizeof( hb_job_t ) );
405 job_copy->title = title_copy;
406 job_copy->file = strdup( job->file );
408 job_copy->pause = h->pause_lock;
410 /* Add the job to the list */
411 hb_list_add( h->jobs, job_copy );
414 void hb_rem( hb_handle_t * h, hb_job_t * job )
416 hb_list_rem( h->jobs, job );
418 /* XXX free everything XXX */
421 void hb_start( hb_handle_t * h )
424 h->job_count = hb_list_count( h->jobs );
426 hb_lock( h->state_lock );
427 h->state.state = HB_STATE_WORKING;
428 #define p h->state.param.working
431 p.job_count = h->job_count;
438 hb_unlock( h->state_lock );
443 h->work_thread = hb_work_init( h->jobs, h->cpu_count,
444 &h->work_die, &h->work_error );
447 void hb_pause( hb_handle_t * h )
451 hb_lock( h->pause_lock );
454 hb_lock( h->state_lock );
455 h->state.state = HB_STATE_PAUSED;
456 hb_unlock( h->state_lock );
460 void hb_resume( hb_handle_t * h )
464 hb_unlock( h->pause_lock );
469 void hb_stop( hb_handle_t * h )
476 void hb_get_state( hb_handle_t * h, hb_state_t * s )
478 hb_lock( h->state_lock );
480 memcpy( s, &h->state, sizeof( hb_state_t ) );
481 h->state.state = HB_STATE_IDLE;
483 hb_unlock( h->state_lock );
486 void hb_close( hb_handle_t ** _h )
488 hb_handle_t * h = *_h;
492 hb_thread_close( &h->main_thread );
494 while( ( title = hb_list_item( h->list_title, 0 ) ) )
496 hb_list_rem( h->list_title, title );
497 hb_title_close( &title );
499 hb_list_close( &h->list_title );
501 hb_list_close( &h->jobs );
502 hb_lock_close( &h->state_lock );
503 hb_lock_close( &h->pause_lock );
508 static void thread_func( void * _h )
510 hb_handle_t * h = (hb_handle_t *) _h;
513 struct dirent * entry;
517 /* Create folder for temporary files */
518 memset( dirname, 0, 1024 );
519 hb_get_tempory_directory( h, dirname );
525 /* In case the check_update thread hangs, it'll die sooner or
526 later. Then, we join it here */
527 if( h->update_thread &&
528 hb_thread_has_exited( h->update_thread ) )
530 hb_thread_close( &h->update_thread );
533 /* Check if the scan thread is done */
534 if( h->scan_thread &&
535 hb_thread_has_exited( h->scan_thread ) )
537 hb_thread_close( &h->scan_thread );
539 hb_log( "libhb: scan thread found %d valid title(s)",
540 hb_list_count( h->list_title ) );
541 hb_lock( h->state_lock );
542 h->state.state = HB_STATE_SCANDONE;
543 hb_unlock( h->state_lock );
546 /* Check if the work thread is done */
547 if( h->work_thread &&
548 hb_thread_has_exited( h->work_thread ) )
550 hb_thread_close( &h->work_thread );
552 hb_log( "libhb: work result = %d",
554 hb_lock( h->state_lock );
555 h->state.state = HB_STATE_WORKDONE;
556 h->state.param.workdone.error = h->work_error;
557 hb_unlock( h->state_lock );
566 hb_thread_close( &h->work_thread );
569 /* Remove temp folder */
570 dir = opendir( dirname );
571 while( ( entry = readdir( dir ) ) )
574 if( entry->d_name[0] == '.' )
578 memset( filename, 0, 1024 );
579 snprintf( filename, 1023, "%s/%s", dirname, entry->d_name );
586 int hb_get_pid( hb_handle_t * h )
591 void hb_set_state( hb_handle_t * h, hb_state_t * s )
593 hb_lock( h->pause_lock );
594 hb_lock( h->state_lock );
595 memcpy( &h->state, s, sizeof( hb_state_t ) );
596 if( h->state.state == HB_STATE_WORKING )
599 h->state.param.working.job_cur =
600 h->job_count - hb_list_count( h->jobs );
601 h->state.param.working.job_count = h->job_count;
603 hb_unlock( h->state_lock );
604 hb_unlock( h->pause_lock );