6 /* The "Check for update" thread */
9 hb_thread_t * update_thread;
11 /* This thread's only purpose is to check other threads'
14 hb_thread_t * main_thread;
17 /* DVD/file scan thread */
18 hb_list_t * list_title;
19 hb_thread_t * scan_thread;
21 /* The thread which processes the jobs. Others threads are launched
22 from this one (see work.c) */
24 hb_job_t * current_job;
26 int job_count_permanent;
27 volatile int work_die;
29 hb_thread_t * work_thread;
33 hb_lock_t * state_lock;
37 hb_lock_t * pause_lock;
38 /* For MacGui active queue
39 increments each time the scan thread completes*/
44 hb_lock_t *hb_avcodec_lock;
45 hb_work_object_t * hb_objects = NULL;
47 static void thread_func( void * );
49 void hb_avcodec_init()
51 hb_avcodec_lock = hb_lock_init();
55 int hb_avcodec_open(AVCodecContext *avctx, AVCodec *codec)
58 hb_lock( hb_avcodec_lock );
59 ret = avcodec_open(avctx, codec);
60 hb_unlock( hb_avcodec_lock );
64 int hb_avcodec_close(AVCodecContext *avctx)
67 hb_lock( hb_avcodec_lock );
68 ret = avcodec_close(avctx);
69 hb_unlock( hb_avcodec_lock );
74 * Registers work objects, by adding the work object to a liked list.
75 * @param w Handle to hb_work_object_t to register.
77 void hb_register( hb_work_object_t * w )
84 * libhb initialization routine.
85 * @param verbose HB_DEBUG_NONE or HB_DEBUG_ALL.
86 * @param update_check signals libhb to check for updated version from HandBrake website.
87 * @return Handle to hb_handle_t for use on all subsequent calls to libhb.
89 hb_handle_t * hb_init_real( int verbose, int update_check )
91 hb_handle_t * h = calloc( sizeof( hb_handle_t ), 1 );
94 /* See hb_deep_log() and hb_log() in common.c */
95 global_verbosity_level = verbose;
97 putenv( "HB_DEBUG=1" );
99 /* Check for an update on the website if asked to */
104 hb_log( "hb_init: checking for updates" );
105 date = hb_get_date();
106 h->update_thread = hb_update_init( &h->build, h->version );
110 if( hb_thread_has_exited( h->update_thread ) )
112 /* Immediate success or failure */
113 hb_thread_close( &h->update_thread );
116 if( hb_get_date() > date + 1000 )
118 /* Still nothing after one second. Connection problem,
119 let the thread die */
120 hb_log( "hb_init: connection problem, not waiting for "
129 * Initialise buffer pool
131 hb_buffer_pool_init();
133 /* CPU count detection */
134 hb_log( "hb_init: checking cpu count" );
135 h->cpu_count = hb_get_cpu_count();
137 h->list_title = hb_list_init();
138 h->jobs = hb_list_init();
140 h->state_lock = hb_lock_init();
141 h->state.state = HB_STATE_IDLE;
143 h->pause_lock = hb_lock_init();
148 /* Start library thread */
149 hb_log( "hb_init: starting libhb thread" );
151 h->main_thread = hb_thread_init( "libhb", thread_func, h,
152 HB_NORMAL_PRIORITY );
156 /* Set the scan count to start at 0 */
161 * libhb initialization routine.
162 * This version is to use when calling the dylib, the macro hb_init isn't available from a dylib call!
163 * @param verbose HB_DEBUG_NONE or HB_DEBUG_ALL.
164 * @param update_check signals libhb to check for updated version from HandBrake website.
165 * @return Handle to hb_handle_t for use on all subsequent calls to libhb.
167 hb_handle_t * hb_init_dl( int verbose, int update_check )
169 hb_handle_t * h = calloc( sizeof( hb_handle_t ), 1 );
172 /* See hb_log() in common.c */
173 if( verbose > HB_DEBUG_NONE )
175 putenv( "HB_DEBUG=1" );
178 /* Check for an update on the website if asked to */
183 hb_log( "hb_init: checking for updates" );
184 date = hb_get_date();
185 h->update_thread = hb_update_init( &h->build, h->version );
189 if( hb_thread_has_exited( h->update_thread ) )
191 /* Immediate success or failure */
192 hb_thread_close( &h->update_thread );
195 if( hb_get_date() > date + 1000 )
197 /* Still nothing after one second. Connection problem,
198 let the thread die */
199 hb_log( "hb_init: connection problem, not waiting for "
207 /* CPU count detection */
208 hb_log( "hb_init: checking cpu count" );
209 h->cpu_count = hb_get_cpu_count();
211 h->list_title = hb_list_init();
212 h->jobs = hb_list_init();
213 h->current_job = NULL;
215 h->state_lock = hb_lock_init();
216 h->state.state = HB_STATE_IDLE;
218 h->pause_lock = hb_lock_init();
222 avcodec_register_all();
224 /* Start library thread */
225 hb_log( "hb_init: starting libhb thread" );
227 h->main_thread = hb_thread_init( "libhb", thread_func, h,
228 HB_NORMAL_PRIORITY );
230 hb_register( &hb_sync );
231 hb_register( &hb_decmpeg2 );
232 hb_register( &hb_decsub );
233 hb_register( &hb_render );
234 hb_register( &hb_encavcodec );
235 hb_register( &hb_encxvid );
236 hb_register( &hb_encx264 );
237 hb_register( &hb_enctheora );
238 hb_register( &hb_deca52 );
239 hb_register( &hb_decdca );
240 hb_register( &hb_decavcodec );
241 hb_register( &hb_decavcodecv );
242 hb_register( &hb_decavcodecvi );
243 hb_register( &hb_decavcodecai );
244 hb_register( &hb_declpcm );
245 hb_register( &hb_encfaac );
246 hb_register( &hb_enclame );
247 hb_register( &hb_encvorbis );
254 * Returns current version of libhb.
255 * @param h Handle to hb_handle_t.
256 * @return character array of version number.
258 char * hb_get_version( hb_handle_t * h )
260 return HB_PROJECT_VERSION;
264 * Returns current build of libhb.
265 * @param h Handle to hb_handle_t.
266 * @return character array of build number.
268 int hb_get_build( hb_handle_t * h )
270 return HB_PROJECT_BUILD;
274 * Checks for needed update.
275 * @param h Handle to hb_handle_t.
276 * @param version Pointer to handle where version will be copied.
277 * @return update indicator.
279 int hb_check_update( hb_handle_t * h, char ** version )
281 *version = ( h->build < 0 ) ? NULL : h->version;
286 * Sets the cpu count to the desired value.
287 * @param h Handle to hb_handle_t
288 * @param cpu_count Number of CPUs to use.
290 void hb_set_cpu_count( hb_handle_t * h, int cpu_count )
292 cpu_count = MAX( 1, cpu_count );
293 cpu_count = MIN( cpu_count, 8 );
294 h->cpu_count = cpu_count;
298 * Deletes current previews associated with titles
299 * @param h Handle to hb_handle_t
301 void hb_remove_previews( hb_handle_t * h )
308 struct dirent * entry;
310 memset( dirname, 0, 1024 );
311 hb_get_tempory_directory( h, dirname );
312 dir = opendir( dirname );
313 if (dir == NULL) return;
315 count = hb_list_count( h->list_title );
316 while( ( entry = readdir( dir ) ) )
318 if( entry->d_name[0] == '.' )
322 for( i = 0; i < count; i++ )
324 title = hb_list_item( h->list_title, i );
325 len = snprintf( filename, 1024, "%" PRIxPTR, (intptr_t) title );
326 if (strncmp(entry->d_name, filename, len) == 0)
328 snprintf( filename, 1024, "%s/%s", dirname, entry->d_name );
338 * Initializes a scan of the by calling hb_scan_init
339 * @param h Handle to hb_handle_t
340 * @param path location of VIDEO_TS folder.
341 * @param title_index Desired title to scan. 0 for all titles.
342 * @param preview_count Number of preview images to generate.
343 * @param store_previews Whether or not to write previews to disk.
345 void hb_scan( hb_handle_t * h, const char * path, int title_index,
346 int preview_count, int store_previews )
350 /* Clean up from previous scan */
351 hb_remove_previews( h );
352 while( ( title = hb_list_item( h->list_title, 0 ) ) )
354 hb_list_rem( h->list_title, title );
355 hb_title_close( &title );
358 hb_log( "hb_scan: path=%s, title_index=%d", path, title_index );
359 h->scan_thread = hb_scan_init( h, path, title_index, h->list_title,
360 preview_count, store_previews );
364 * Returns the list of titles found.
365 * @param h Handle to hb_handle_t
366 * @return Handle to hb_list_t of the title list.
368 hb_list_t * hb_get_titles( hb_handle_t * h )
370 return h->list_title;
374 * Create preview image of desired title a index of picture.
375 * @param h Handle to hb_handle_t.
376 * @param title Handle to hb_title_t of desired title.
377 * @param picture Index in title.
378 * @param buffer Handle to buufer were inage will be drawn.
380 void hb_get_preview( hb_handle_t * h, hb_title_t * title, int picture,
383 hb_job_t * job = title->job;
386 uint8_t * buf1, * buf2, * buf3, * buf4, * pen;
387 uint32_t * p32, swsflags;
388 AVPicture pic_in, pic_preview, pic_deint, pic_crop, pic_scale;
389 struct SwsContext * context;
391 int rgb_width = ((job->width + 7) >> 3) << 3;
394 swsflags = SWS_LANCZOS | SWS_ACCURATE_RND;
396 buf1 = av_malloc( avpicture_get_size( PIX_FMT_YUV420P, title->width, title->height ) );
397 buf2 = av_malloc( avpicture_get_size( PIX_FMT_YUV420P, title->width, title->height ) );
398 buf3 = av_malloc( avpicture_get_size( PIX_FMT_YUV420P, job->width, job->height ) );
399 buf4 = av_malloc( avpicture_get_size( PIX_FMT_RGBA32, rgb_width, job->height ) );
400 avpicture_fill( &pic_in, buf1, PIX_FMT_YUV420P,
401 title->width, title->height );
402 avpicture_fill( &pic_deint, buf2, PIX_FMT_YUV420P,
403 title->width, title->height );
404 avpicture_fill( &pic_scale, buf3, PIX_FMT_YUV420P,
405 job->width, job->height );
406 avpicture_fill( &pic_preview, buf4, PIX_FMT_RGBA32,
407 rgb_width, job->height );
409 // Allocate the AVPicture frames and fill in
411 memset( filename, 0, 1024 );
413 hb_get_tempory_filename( h, filename, "%" PRIxPTR "%d",
414 (intptr_t) title, picture );
416 file = fopen( filename, "r" );
419 hb_log( "hb_get_preview: fopen failed" );
423 fread( buf1, avpicture_get_size( PIX_FMT_YUV420P, title->width, title->height), 1, file );
426 if( job->deinterlace )
428 // Deinterlace and crop
429 avpicture_deinterlace( &pic_deint, &pic_in, PIX_FMT_YUV420P, title->width, title->height );
430 av_picture_crop( &pic_crop, &pic_deint, PIX_FMT_YUV420P, job->crop[0], job->crop[2] );
435 av_picture_crop( &pic_crop, &pic_in, PIX_FMT_YUV420P, job->crop[0], job->crop[2] );
438 // Get scaling context
439 context = sws_getContext(title->width - (job->crop[2] + job->crop[3]),
440 title->height - (job->crop[0] + job->crop[1]),
442 job->width, job->height, PIX_FMT_YUV420P,
443 swsflags, NULL, NULL, NULL);
447 pic_crop.data, pic_crop.linesize,
448 0, title->height - (job->crop[0] + job->crop[1]),
449 pic_scale.data, pic_scale.linesize);
452 sws_freeContext( context );
454 // Get preview context
455 context = sws_getContext(rgb_width, job->height, PIX_FMT_YUV420P,
456 rgb_width, job->height, PIX_FMT_RGBA32,
457 swsflags, NULL, NULL, NULL);
461 pic_scale.data, pic_scale.linesize,
463 pic_preview.data, pic_preview.linesize);
466 sws_freeContext( context );
468 /* Gray background */
469 p32 = (uint32_t *) buffer;
470 for( i = 0; i < ( title->width + 2 ) * ( title->height + 2 ); i++ )
475 /* Draw the picture, centered, and draw the cropping zone */
476 preview_size = pic_preview.linesize[0];
477 pen = buffer + ( title->height - job->height ) *
478 ( title->width + 2 ) * 2 + ( title->width - job->width ) * 2;
479 memset( pen, 0xFF, 4 * ( job->width + 2 ) );
480 pen += 4 * ( title->width + 2 );
481 for( i = 0; i < job->height; i++ )
484 nextLine = pen + 4 * ( title->width + 2 );
485 memset( pen, 0xFF, 4 );
487 memcpy( pen, buf4 + preview_size * i, 4 * job->width );
488 pen += 4 * job->width;
489 memset( pen, 0xFF, 4 );
492 memset( pen, 0xFF, 4 * ( job->width + 2 ) );
495 avpicture_free( &pic_preview );
496 avpicture_free( &pic_scale );
497 avpicture_free( &pic_deint );
498 avpicture_free( &pic_in );
502 * Analyzes a frame to detect interlacing artifacts
503 * and returns true if interlacing (combing) is found.
505 * Code taken from Thomas Oestreich's 32detect filter
506 * in the Transcode project, with minor formatting changes.
508 * @param buf An hb_buffer structure holding valid frame data
509 * @param width The frame's width in pixels
510 * @param height The frame's height in pixels
511 * @param color_equal Sensitivity for detecting similar colors
512 * @param color_diff Sensitivity for detecting different colors
513 * @param threshold Sensitivity for flagging planes as combed
514 * @param prog_equal Sensitivity for detecting similar colors on progressive frames
515 * @param prog_diff Sensitivity for detecting different colors on progressive frames
516 * @param prog_threshold Sensitivity for flagging progressive frames as combed
518 int hb_detect_comb( hb_buffer_t * buf, int width, int height, int color_equal, int color_diff, int threshold, int prog_equal, int prog_diff, int prog_threshold )
520 int j, k, n, off, cc_1, cc_2, cc[3];
521 // int flag[3] ; // debugging flag
522 uint16_t s1, s2, s3, s4;
527 if ( buf->flags & 16 )
529 /* Frame is progressive, be more discerning. */
530 color_diff = prog_diff;
531 color_equal = prog_equal;
532 threshold = prog_threshold;
535 /* One pas for Y, one pass for Cb, one pass for Cr */
536 for( k = 0; k < 3; k++ )
540 /* Y has already been checked, now offset by Y's dimensions
541 and divide all the other values by 2, since Cr and Cb
542 are half-size compared to Y. */
543 offset = width * height;
549 /* Y and Cb are done, so the offset needs to be bumped
550 so it's width*height + (width / 2) * (height / 2) */
554 for( j = 0; j < width; ++j )
558 for( n = 0; n < ( height - 4 ); n = n + 2 )
560 /* Look at groups of 4 sequential horizontal lines */
561 s1 = ( ( buf->data + offset )[ off + j ] & 0xff );
562 s2 = ( ( buf->data + offset )[ off + j + width ] & 0xff );
563 s3 = ( ( buf->data + offset )[ off + j + 2 * width ] & 0xff );
564 s4 = ( ( buf->data + offset )[ off + j + 3 * width ] & 0xff );
566 /* Note if the 1st and 2nd lines are more different in
567 color than the 1st and 3rd lines are similar in color.*/
568 if ( ( abs( s1 - s3 ) < color_equal ) &&
569 ( abs( s1 - s2 ) > color_diff ) )
572 /* Note if the 2nd and 3rd lines are more different in
573 color than the 2nd and 4th lines are similar in color.*/
574 if ( ( abs( s2 - s4 ) < color_equal ) &&
575 ( abs( s2 - s3 ) > color_diff) )
578 /* Now move down 2 horizontal lines before starting over.*/
584 /* The final cc score for a plane is the percentage of combed pixels it contains.
585 Because sensitivity goes down to hundreths of a percent, multiply by 1000
586 so it will be easy to compare against the threhold value which is an integer. */
587 cc[k] = (int)( ( cc_1 + cc_2 ) * 1000.0 / ( width * height ) );
591 /* HandBrake is all yuv420, so weight the average percentage of all 3 planes accordingly.*/
592 int average_cc = ( 2 * cc[0] + ( cc[1] / 2 ) + ( cc[2] / 2 ) ) / 3;
594 /* Now see if that average percentage of combed pixels surpasses the threshold percentage given by the user.*/
595 if( average_cc > threshold )
598 hb_log("Average %i combed (Threshold %i) %i/%i/%i | PTS: %lld (%fs) %s", average_cc, threshold, cc[0], cc[1], cc[2], buf->start, (float)buf->start / 90000, (buf->flags & 16) ? "Film" : "Video" );
604 hb_log("SKIPPED Average %i combed (Threshold %i) %i/%i/%i | PTS: %lld (%fs) %s", average_cc, threshold, cc[0], cc[1], cc[2], buf->start, (float)buf->start / 90000, (buf->flags & 16) ? "Film" : "Video" );
607 /* Reaching this point means no combing detected. */
613 * Calculates job width and height for anamorphic content,
615 * @param job Handle to hb_job_t
616 * @param output_width Pointer to returned storage width
617 * @param output_height Pointer to returned storage height
618 * @param output_par_width Pointer to returned pixel width
619 @ param output_par_height Pointer to returned pixel height
621 void hb_set_anamorphic_size( hb_job_t * job,
622 int *output_width, int *output_height,
623 int *output_par_width, int *output_par_height )
625 /* Set up some variables to make the math easier to follow. */
626 hb_title_t * title = job->title;
627 int cropped_width = title->width - job->crop[2] - job->crop[3] ;
628 int cropped_height = title->height - job->crop[0] - job->crop[1] ;
629 double storage_aspect = (double)cropped_width / (double)cropped_height;
630 int mod = job->anamorphic.modulus ? job->anamorphic.modulus : 16;
631 double aspect = title->aspect;
633 int pixel_aspect_width = job->anamorphic.par_width;
634 int pixel_aspect_height = job->anamorphic.par_height;
636 /* If a source was really NTSC or PAL and the user specified ITU PAR
637 values, replace the standard PAR values with the ITU broadcast ones. */
638 if( title->width == 720 && job->anamorphic.itu_par )
640 // convert aspect to a scaled integer so we can test for 16:9 & 4:3
641 // aspect ratios ignoring insignificant differences in the LSBs of
642 // the floating point representation.
643 int iaspect = aspect * 9.;
645 /* Handle ITU PARs */
646 if (title->height == 480)
651 /* It's widescreen */
652 pixel_aspect_width = 40;
653 pixel_aspect_height = 33;
655 else if (iaspect == 12)
658 pixel_aspect_width = 10;
659 pixel_aspect_height = 11;
662 else if (title->height == 576)
667 /* It's widescreen */
668 pixel_aspect_width = 16;
669 pixel_aspect_height = 11;
671 else if (iaspect == 12)
674 pixel_aspect_width = 12;
675 pixel_aspect_height = 11;
680 /* Figure out what width the source would display at. */
681 int source_display_width = cropped_width * (double)pixel_aspect_width /
682 (double)pixel_aspect_height ;
685 3 different ways of deciding output dimensions:
686 - 1: Strict anamorphic, preserve source dimensions
687 - 2: Loose anamorphic, round to mod16 and preserve storage aspect ratio
688 - 3: Power user anamorphic, specify everything
691 switch( job->anamorphic.mode )
694 /* Strict anamorphic */
695 *output_width = cropped_width;
696 *output_height = cropped_height;
697 *output_par_width = title->pixel_aspect_width;
698 *output_par_height = title->pixel_aspect_height;
702 /* "Loose" anamorphic.
703 - Uses mod16-compliant dimensions,
704 - Allows users to set the width
707 // height: Gets set later, ignore user job->height value
709 /* Gotta handle bounding dimensions.
710 If the width is too big, just reset it with no rescaling.
711 Instead of using the aspect-scaled job height,
712 we need to see if the job width divided by the storage aspect
713 is bigger than the max. If so, set it to the max (this is sloppy).
714 If not, set job height to job width divided by storage aspect.
717 if ( job->maxWidth && (job->maxWidth < job->width) )
718 width = job->maxWidth;
720 /* Time to get picture width that divide cleanly.*/
721 width = MULTIPLE_MOD( width, mod);
723 /* Verify these new dimensions don't violate max height and width settings */
724 if ( job->maxWidth && (job->maxWidth < job->width) )
725 width = job->maxWidth;
727 height = ((double)width / storage_aspect) + 0.5;
729 if ( job->maxHeight && (job->maxHeight < height) )
730 height = job->maxHeight;
732 /* Time to get picture height that divide cleanly.*/
733 height = MULTIPLE_MOD( height, mod);
735 /* Verify these new dimensions don't violate max height and width settings */
736 if ( job->maxHeight && (job->maxHeight < height) )
737 height = job->maxHeight;
739 /* The film AR is the source's display width / cropped source height.
740 The output display width is the output height * film AR.
741 The output PAR is the output display width / output storage width. */
742 pixel_aspect_width = height * source_display_width / cropped_height;
743 pixel_aspect_height = width;
745 /* Pass the results back to the caller */
746 *output_width = width;
747 *output_height = height;
751 /* Anamorphic 3: Power User Jamboree
752 - Set everything based on specified values */
754 /* Use specified storage dimensions */
756 height = job->height;
758 /* Bind to max dimensions */
759 if( job->maxWidth && width > job->maxWidth )
760 width = job->maxWidth;
761 if( job->maxHeight && height > job->maxHeight )
762 height = job->maxHeight;
764 /* Time to get picture dimensions that divide cleanly.*/
765 width = MULTIPLE_MOD( width, mod);
766 height = MULTIPLE_MOD( height, mod);
768 /* Verify we're still within max dimensions */
769 if( job->maxWidth && width > job->maxWidth )
770 width = job->maxWidth - (mod/2);
771 if( job->maxHeight && height > job->maxHeight )
772 height = job->maxHeight - (mod/2);
774 /* Re-ensure we have picture dimensions that divide cleanly. */
775 width = MULTIPLE_MOD( width, mod );
776 height = MULTIPLE_MOD( height, mod );
778 /* That finishes the storage dimensions. On to display. */
779 if( job->anamorphic.dar_width && job->anamorphic.dar_height )
781 /* We need to adjust the PAR to produce this aspect. */
782 pixel_aspect_width = height * job->anamorphic.dar_width / job->anamorphic.dar_height;
783 pixel_aspect_height = width;
787 /* We first need the display ar.
788 That's the source display width divided by the source height after cropping.
789 Then we multiple the output height by that to get the pixel aspect width,
790 and the pixel aspect height is the storage width.*/
791 pixel_aspect_width = height * source_display_width / cropped_height;
792 pixel_aspect_height = width;
796 *output_width = width;
797 *output_height = height;
801 /* While x264 is smart enough to reduce fractions on its own, libavcodec
802 needs some help with the math, so lose superfluous factors. */
803 hb_reduce( output_par_width, output_par_height,
804 pixel_aspect_width, pixel_aspect_height );
808 * Calculates job width, height, and cropping parameters.
809 * @param job Handle to hb_job_t.
810 * @param aspect Desired aspect ratio. Value of -1 uses title aspect.
811 * @param pixels Maximum desired pixel count.
813 void hb_set_size( hb_job_t * job, double aspect, int pixels )
815 hb_title_t * title = job->title;
817 int croppedWidth = title->width - title->crop[2] - title->crop[3];
818 int croppedHeight = title->height - title->crop[0] - title->crop[1];
819 double croppedAspect = title->aspect * title->height * croppedWidth /
820 croppedHeight / title->width;
826 /* Keep the best possible aspect ratio */
827 aspect = croppedAspect;
830 /* Crop if necessary to obtain the desired ratio */
831 memcpy( job->crop, title->crop, 4 * sizeof( int ) );
832 if( aspect < croppedAspect )
834 /* Need to crop on the left and right */
835 addCrop = croppedWidth - aspect * croppedHeight * title->width /
836 title->aspect / title->height;
839 addCrop = ( addCrop + 1 ) / 2;
840 job->crop[2] += addCrop;
841 job->crop[3] += addCrop;
843 else if( addCrop & 2 )
846 job->crop[2] += addCrop - 1;
847 job->crop[3] += addCrop + 1;
852 job->crop[2] += addCrop;
853 job->crop[3] += addCrop;
856 else if( aspect > croppedAspect )
858 /* Need to crop on the top and bottom */
859 addCrop = croppedHeight - croppedWidth * title->aspect *
860 title->height / aspect / title->width;
863 addCrop = ( addCrop + 1 ) / 2;
864 job->crop[0] += addCrop;
865 job->crop[1] += addCrop;
867 else if( addCrop & 2 )
870 job->crop[0] += addCrop - 1;
871 job->crop[1] += addCrop + 1;
876 job->crop[0] += addCrop;
877 job->crop[1] += addCrop;
881 /* Compute a resolution from the number of pixels and aspect */
885 h = MULTIPLE_16( (int)( (double)w / aspect ) );
893 job->height = MULTIPLE_16( (int)( (double)job->width / aspect ) );
897 * Returns the number of jobs in the queue.
898 * @param h Handle to hb_handle_t.
899 * @return Number of jobs.
901 int hb_count( hb_handle_t * h )
903 return hb_list_count( h->jobs );
907 * Returns handle to job at index i within the job list.
908 * @param h Handle to hb_handle_t.
909 * @param i Index of job.
910 * @returns Handle to hb_job_t of desired job.
912 hb_job_t * hb_job( hb_handle_t * h, int i )
914 return hb_list_item( h->jobs, i );
917 hb_job_t * hb_current_job( hb_handle_t * h )
919 return( h->current_job );
923 * Adds a job to the job list.
924 * @param h Handle to hb_handle_t.
925 * @param job Handle to hb_job_t.
927 void hb_add( hb_handle_t * h, hb_job_t * job )
930 hb_title_t * title, * title_copy;
931 hb_chapter_t * chapter, * chapter_copy;
933 hb_subtitle_t * subtitle, * subtitle_copy;
939 title_copy = malloc( sizeof( hb_title_t ) );
940 memcpy( title_copy, title, sizeof( hb_title_t ) );
942 title_copy->list_chapter = hb_list_init();
943 for( i = 0; i < hb_list_count( title->list_chapter ); i++ )
945 chapter = hb_list_item( title->list_chapter, i );
946 chapter_copy = malloc( sizeof( hb_chapter_t ) );
947 memcpy( chapter_copy, chapter, sizeof( hb_chapter_t ) );
948 hb_list_add( title_copy->list_chapter, chapter_copy );
954 if( title->metadata )
956 title_copy->metadata = malloc( sizeof( hb_metadata_t ) );
958 if( title_copy->metadata )
960 memcpy( title_copy->metadata, title->metadata, sizeof( hb_metadata_t ) );
963 * Need to copy the artwork seperatly (TODO).
965 if( title->metadata->coverart )
967 title_copy->metadata->coverart = malloc( title->metadata->coverart_size );
968 if( title_copy->metadata->coverart )
970 memcpy( title_copy->metadata->coverart, title->metadata->coverart,
971 title->metadata->coverart_size );
973 title_copy->metadata->coverart_size = 0;
979 /* Copy the audio track(s) we want */
980 title_copy->list_audio = hb_list_init();
982 for( i = 0; i < hb_list_count(job->list_audio); i++ )
984 if( ( audio = hb_list_item( job->list_audio, i ) ) )
986 hb_list_add( title_copy->list_audio, hb_audio_copy(audio) );
990 title_copy->list_subtitle = hb_list_init();
993 * The following code is confusing, there are three ways in which
994 * we select subtitles and it depends on whether this is single or
997 * subtitle_scan may be enabled, in which case the first pass
998 * scans all subtitles of that language. The second pass does not
999 * select any because they are set at the end of the first pass.
1001 * native_language may have a preferred language, in which case we
1002 * may be switching the language we want for the subtitles in the
1003 * first pass of a single pass, or the second pass of a two pass.
1005 * We may have manually selected a subtitle, in which case that is
1006 * selected in the first pass of a single pass, or the second of a
1009 memset( audio_lang, 0, sizeof( audio_lang ) );
1011 if ( job->indepth_scan || job->native_language ) {
1014 * Find the first audio language that is being encoded
1016 for( i = 0; i < hb_list_count(job->list_audio); i++ )
1018 if( ( audio = hb_list_item( job->list_audio, i ) ) )
1020 strncpy(audio_lang, audio->config.lang.iso639_2, sizeof(audio_lang));
1026 * In all cases switch the language if we need to to our native
1029 if( job->native_language )
1031 if( strncasecmp( job->native_language, audio_lang,
1032 sizeof( audio_lang ) ) != 0 )
1035 if( job->pass != 2 )
1037 hb_log( "Enabled subtitles in native language '%s', audio is in '%s'",
1038 job->native_language, audio_lang);
1041 * The main audio track is not in our native language, so switch
1042 * the subtitles to use our native language instead.
1044 strncpy( audio_lang, job->native_language, sizeof( audio_lang ) );
1047 * native language is irrelevent, free it.
1049 free( job->native_language );
1050 job->native_language = NULL;
1056 * If doing a subtitle scan then add all the matching subtitles for this
1059 if ( job->indepth_scan )
1061 for( i=0; i < hb_list_count( title->list_subtitle ); i++ )
1063 subtitle = hb_list_item( title->list_subtitle, i );
1064 if( strcmp( subtitle->iso639_2, audio_lang ) == 0 )
1067 * Matched subtitle language with audio language, so
1068 * add this to our list to scan.
1070 * We will update the subtitle list on the second pass
1071 * later after the first pass has completed.
1073 subtitle_copy = malloc( sizeof( hb_subtitle_t ) );
1074 memcpy( subtitle_copy, subtitle, sizeof( hb_subtitle_t ) );
1075 hb_list_add( title_copy->list_subtitle, subtitle_copy );
1076 if ( job->native_language ) {
1078 * With native language just select the
1079 * first match in our langiage, not all of
1080 * them. Subsequent ones are likely to be commentary
1088 * Not doing a subtitle scan in this pass, but maybe we are in the
1091 if( job->select_subtitle )
1094 * Don't add subtitles here, we'll add them via select_subtitle
1095 * at the end of the subtitle_scan.
1099 * Definitely not doing a subtitle scan.
1101 if( job->pass != 1 && job->native_language )
1104 * We are not doing a subtitle scan but do want the
1105 * native langauge subtitle selected, so select it
1106 * for pass 0 or pass 2 of a two pass.
1108 for( i=0; i < hb_list_count( title->list_subtitle ); i++ )
1110 subtitle = hb_list_item( title->list_subtitle, i );
1111 if( strcmp( subtitle->iso639_2, audio_lang ) == 0 )
1114 * Matched subtitle language with audio language, so
1115 * add this to our list to scan.
1117 subtitle_copy = malloc( sizeof( hb_subtitle_t ) );
1118 memcpy( subtitle_copy, subtitle, sizeof( hb_subtitle_t ) );
1119 hb_list_add( title_copy->list_subtitle, subtitle_copy );
1125 * Manually selected subtitle, in which case only
1126 * bother adding them for pass 0 or pass 2 of a two
1129 if( job->pass != 1 )
1131 if( ( subtitle = hb_list_item( title->list_subtitle, job->subtitle ) ) )
1133 subtitle_copy = malloc( sizeof( hb_subtitle_t ) );
1134 memcpy( subtitle_copy, subtitle, sizeof( hb_subtitle_t ) );
1135 hb_list_add( title_copy->list_subtitle, subtitle_copy );
1143 job_copy = calloc( sizeof( hb_job_t ), 1 );
1144 memcpy( job_copy, job, sizeof( hb_job_t ) );
1145 title_copy->job = job_copy;
1146 job_copy->title = title_copy;
1147 job_copy->list_audio = title_copy->list_audio;
1148 job_copy->file = strdup( job->file );
1150 job_copy->pause = h->pause_lock;
1152 /* Copy the job filter list */
1156 int filter_count = hb_list_count( job->filters );
1157 job_copy->filters = hb_list_init();
1158 for( i = 0; i < filter_count; i++ )
1161 * Copy the filters, since the MacGui reuses the global filter objects
1162 * meaning that queued up jobs overwrite the previous filter settings.
1163 * In reality, settings is probably the only field that needs duplicating
1164 * since it's the only value that is ever changed. But name is duplicated
1165 * as well for completeness. Not copying private_data since it gets
1166 * created for each job in renderInit.
1168 hb_filter_object_t * filter = hb_list_item( job->filters, i );
1169 hb_filter_object_t * filter_copy = malloc( sizeof( hb_filter_object_t ) );
1170 memcpy( filter_copy, filter, sizeof( hb_filter_object_t ) );
1172 filter_copy->name = strdup( filter->name );
1173 if( filter->settings )
1174 filter_copy->settings = strdup( filter->settings );
1175 hb_list_add( job_copy->filters, filter_copy );
1179 /* Add the job to the list */
1180 hb_list_add( h->jobs, job_copy );
1181 h->job_count = hb_count(h);
1182 h->job_count_permanent++;
1186 * Removes a job from the job list.
1187 * @param h Handle to hb_handle_t.
1188 * @param job Handle to hb_job_t.
1190 void hb_rem( hb_handle_t * h, hb_job_t * job )
1192 hb_list_rem( h->jobs, job );
1194 h->job_count = hb_count(h);
1195 if (h->job_count_permanent)
1196 h->job_count_permanent--;
1198 /* XXX free everything XXX */
1202 * Starts the conversion process.
1203 * Sets state to HB_STATE_WORKING.
1204 * calls hb_work_init, to launch work thread. Stores handle to work thread.
1205 * @param h Handle to hb_handle_t.
1207 void hb_start( hb_handle_t * h )
1210 h->job_count = hb_list_count( h->jobs );
1211 h->job_count_permanent = h->job_count;
1213 hb_lock( h->state_lock );
1214 h->state.state = HB_STATE_WORKING;
1215 #define p h->state.param.working
1218 p.job_count = h->job_count;
1226 hb_unlock( h->state_lock );
1231 h->work_thread = hb_work_init( h->jobs, h->cpu_count,
1232 &h->work_die, &h->work_error, &h->current_job );
1236 * Pauses the conversion process.
1237 * @param h Handle to hb_handle_t.
1239 void hb_pause( hb_handle_t * h )
1243 hb_lock( h->pause_lock );
1246 hb_lock( h->state_lock );
1247 h->state.state = HB_STATE_PAUSED;
1248 hb_unlock( h->state_lock );
1253 * Resumes the conversion process.
1254 * @param h Handle to hb_handle_t.
1256 void hb_resume( hb_handle_t * h )
1260 hb_unlock( h->pause_lock );
1266 * Stops the conversion process.
1267 * @param h Handle to hb_handle_t.
1269 void hb_stop( hb_handle_t * h )
1273 h->job_count = hb_count(h);
1274 h->job_count_permanent = 0;
1280 * Returns the state of the conversion process.
1281 * @param h Handle to hb_handle_t.
1282 * @param s Handle to hb_state_t which to copy the state data.
1284 void hb_get_state( hb_handle_t * h, hb_state_t * s )
1286 hb_lock( h->state_lock );
1288 memcpy( s, &h->state, sizeof( hb_state_t ) );
1289 if ( h->state.state == HB_STATE_SCANDONE || h->state.state == HB_STATE_WORKDONE )
1290 h->state.state = HB_STATE_IDLE;
1292 hb_unlock( h->state_lock );
1295 void hb_get_state2( hb_handle_t * h, hb_state_t * s )
1297 hb_lock( h->state_lock );
1299 memcpy( s, &h->state, sizeof( hb_state_t ) );
1301 hb_unlock( h->state_lock );
1305 * Called in MacGui in UpdateUI to check
1306 * for a new scan being completed to set a new source
1308 int hb_get_scancount( hb_handle_t * h)
1310 return h->scanCount;
1314 * Closes access to libhb by freeing the hb_handle_t handle ontained in hb_init_real.
1315 * @param _h Pointer to handle to hb_handle_t.
1317 void hb_close( hb_handle_t ** _h )
1319 hb_handle_t * h = *_h;
1323 hb_thread_close( &h->main_thread );
1325 while( ( title = hb_list_item( h->list_title, 0 ) ) )
1327 hb_list_rem( h->list_title, title );
1328 if( title->job && title->job->filters )
1330 hb_list_close( &title->job->filters );
1333 hb_title_close( &title );
1335 hb_list_close( &h->list_title );
1337 hb_list_close( &h->jobs );
1338 hb_lock_close( &h->state_lock );
1339 hb_lock_close( &h->pause_lock );
1346 * Monitors the state of the update, scan, and work threads.
1347 * Sets scan done state when scan thread exits.
1348 * Sets work done state when work thread exits.
1349 * @param _h Handle to hb_handle_t
1351 static void thread_func( void * _h )
1353 hb_handle_t * h = (hb_handle_t *) _h;
1356 struct dirent * entry;
1360 /* Create folder for temporary files */
1361 memset( dirname, 0, 1024 );
1362 hb_get_tempory_directory( h, dirname );
1364 hb_mkdir( dirname );
1368 /* In case the check_update thread hangs, it'll die sooner or
1369 later. Then, we join it here */
1370 if( h->update_thread &&
1371 hb_thread_has_exited( h->update_thread ) )
1373 hb_thread_close( &h->update_thread );
1376 /* Check if the scan thread is done */
1377 if( h->scan_thread &&
1378 hb_thread_has_exited( h->scan_thread ) )
1380 hb_thread_close( &h->scan_thread );
1382 hb_log( "libhb: scan thread found %d valid title(s)",
1383 hb_list_count( h->list_title ) );
1384 hb_lock( h->state_lock );
1385 h->state.state = HB_STATE_SCANDONE; //originally state.state
1386 hb_unlock( h->state_lock );
1387 /*we increment this sessions scan count by one for the MacGui
1388 to trigger a new source being set */
1392 /* Check if the work thread is done */
1393 if( h->work_thread &&
1394 hb_thread_has_exited( h->work_thread ) )
1396 hb_thread_close( &h->work_thread );
1398 hb_log( "libhb: work result = %d",
1400 hb_lock( h->state_lock );
1401 h->state.state = HB_STATE_WORKDONE;
1402 h->state.param.workdone.error = h->work_error;
1404 h->job_count = hb_count(h);
1405 if (h->job_count < 1)
1406 h->job_count_permanent = 0;
1407 hb_unlock( h->state_lock );
1413 if( h->work_thread )
1416 hb_thread_close( &h->work_thread );
1419 /* Remove temp folder */
1420 dir = opendir( dirname );
1423 while( ( entry = readdir( dir ) ) )
1425 char filename[1024];
1426 if( entry->d_name[0] == '.' )
1430 memset( filename, 0, 1024 );
1431 snprintf( filename, 1023, "%s/%s", dirname, entry->d_name );
1441 * @param h Handle to hb_handle_t
1443 int hb_get_pid( hb_handle_t * h )
1449 * Sets the current state.
1450 * @param h Handle to hb_handle_t
1451 * @param s Handle to new hb_state_t
1453 void hb_set_state( hb_handle_t * h, hb_state_t * s )
1455 hb_lock( h->pause_lock );
1456 hb_lock( h->state_lock );
1457 memcpy( &h->state, s, sizeof( hb_state_t ) );
1458 if( h->state.state == HB_STATE_WORKING )
1461 if (h->job_count < 1)
1462 h->job_count_permanent = 1;
1464 h->state.param.working.job_cur =
1465 h->job_count_permanent - hb_list_count( h->jobs );
1466 h->state.param.working.job_count = h->job_count_permanent;
1468 // Set which job is being worked on
1470 h->state.param.working.sequence_id = h->current_job->sequence_id;
1472 h->state.param.working.sequence_id = 0;
1474 hb_unlock( h->state_lock );
1475 hb_unlock( h->pause_lock );