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 * Initializes a scan of the by calling hb_scan_init
299 * @param h Handle to hb_handle_t
300 * @param path location of VIDEO_TS folder.
301 * @param title_index Desired title to scan. 0 for all titles.
302 * @param preview_count Number of preview images to generate.
303 * @param store_previews Whether or not to write previews to disk.
305 void hb_scan( hb_handle_t * h, const char * path, int title_index,
306 int preview_count, int store_previews )
310 /* Clean up from previous scan */
311 while( ( title = hb_list_item( h->list_title, 0 ) ) )
313 hb_list_rem( h->list_title, title );
314 hb_title_close( &title );
317 hb_log( "hb_scan: path=%s, title_index=%d", path, title_index );
318 h->scan_thread = hb_scan_init( h, path, title_index, h->list_title,
319 preview_count, store_previews );
323 * Returns the list of titles found.
324 * @param h Handle to hb_handle_t
325 * @return Handle to hb_list_t of the title list.
327 hb_list_t * hb_get_titles( hb_handle_t * h )
329 return h->list_title;
333 * Create preview image of desired title a index of picture.
334 * @param h Handle to hb_handle_t.
335 * @param title Handle to hb_title_t of desired title.
336 * @param picture Index in title.
337 * @param buffer Handle to buufer were inage will be drawn.
339 void hb_get_preview( hb_handle_t * h, hb_title_t * title, int picture,
342 hb_job_t * job = title->job;
345 uint8_t * buf1, * buf2, * buf3, * buf4, * pen;
346 uint32_t * p32, swsflags;
347 AVPicture pic_in, pic_preview, pic_deint, pic_crop, pic_scale;
348 struct SwsContext * context;
350 int rgb_width = ((job->width + 7) >> 3) << 3;
353 swsflags = SWS_LANCZOS | SWS_ACCURATE_RND;
355 buf1 = av_malloc( avpicture_get_size( PIX_FMT_YUV420P, title->width, title->height ) );
356 buf2 = av_malloc( avpicture_get_size( PIX_FMT_YUV420P, title->width, title->height ) );
357 buf3 = av_malloc( avpicture_get_size( PIX_FMT_YUV420P, job->width, job->height ) );
358 buf4 = av_malloc( avpicture_get_size( PIX_FMT_RGBA32, rgb_width, job->height ) );
359 avpicture_fill( &pic_in, buf1, PIX_FMT_YUV420P,
360 title->width, title->height );
361 avpicture_fill( &pic_deint, buf2, PIX_FMT_YUV420P,
362 title->width, title->height );
363 avpicture_fill( &pic_scale, buf3, PIX_FMT_YUV420P,
364 job->width, job->height );
365 avpicture_fill( &pic_preview, buf4, PIX_FMT_RGBA32,
366 rgb_width, job->height );
368 // Allocate the AVPicture frames and fill in
370 memset( filename, 0, 1024 );
372 hb_get_tempory_filename( h, filename, "%x%d",
373 (intptr_t) title, picture );
375 file = fopen( filename, "r" );
378 hb_log( "hb_get_preview: fopen failed" );
382 fread( buf1, avpicture_get_size( PIX_FMT_YUV420P, title->width, title->height), 1, file );
385 if( job->deinterlace )
387 // Deinterlace and crop
388 avpicture_deinterlace( &pic_deint, &pic_in, PIX_FMT_YUV420P, title->width, title->height );
389 av_picture_crop( &pic_crop, &pic_deint, PIX_FMT_YUV420P, job->crop[0], job->crop[2] );
394 av_picture_crop( &pic_crop, &pic_in, PIX_FMT_YUV420P, job->crop[0], job->crop[2] );
397 // Get scaling context
398 context = sws_getContext(title->width - (job->crop[2] + job->crop[3]),
399 title->height - (job->crop[0] + job->crop[1]),
401 job->width, job->height, PIX_FMT_YUV420P,
402 swsflags, NULL, NULL, NULL);
406 pic_crop.data, pic_crop.linesize,
407 0, title->height - (job->crop[0] + job->crop[1]),
408 pic_scale.data, pic_scale.linesize);
411 sws_freeContext( context );
413 // Get preview context
414 context = sws_getContext(rgb_width, job->height, PIX_FMT_YUV420P,
415 rgb_width, job->height, PIX_FMT_RGBA32,
416 swsflags, NULL, NULL, NULL);
420 pic_scale.data, pic_scale.linesize,
422 pic_preview.data, pic_preview.linesize);
425 sws_freeContext( context );
427 /* Gray background */
428 p32 = (uint32_t *) buffer;
429 for( i = 0; i < ( title->width + 2 ) * ( title->height + 2 ); i++ )
434 /* Draw the picture, centered, and draw the cropping zone */
435 preview_size = pic_preview.linesize[0];
436 pen = buffer + ( title->height - job->height ) *
437 ( title->width + 2 ) * 2 + ( title->width - job->width ) * 2;
438 memset( pen, 0xFF, 4 * ( job->width + 2 ) );
439 pen += 4 * ( title->width + 2 );
440 for( i = 0; i < job->height; i++ )
443 nextLine = pen + 4 * ( title->width + 2 );
444 memset( pen, 0xFF, 4 );
446 memcpy( pen, buf4 + preview_size * i, 4 * job->width );
447 pen += 4 * job->width;
448 memset( pen, 0xFF, 4 );
451 memset( pen, 0xFF, 4 * ( job->width + 2 ) );
454 avpicture_free( &pic_preview );
455 avpicture_free( &pic_scale );
456 avpicture_free( &pic_deint );
457 avpicture_free( &pic_in );
461 * Analyzes a frame to detect interlacing artifacts
462 * and returns true if interlacing (combing) is found.
464 * Code taken from Thomas Oestreich's 32detect filter
465 * in the Transcode project, with minor formatting changes.
467 * @param buf An hb_buffer structure holding valid frame data
468 * @param width The frame's width in pixels
469 * @param height The frame's height in pixels
470 * @param color_equal Sensitivity for detecting similar colors
471 * @param color_diff Sensitivity for detecting different colors
472 * @param threshold Sensitivity for flagging planes as combed
473 * @param prog_equal Sensitivity for detecting similar colors on progressive frames
474 * @param prog_diff Sensitivity for detecting different colors on progressive frames
475 * @param prog_threshold Sensitivity for flagging progressive frames as combed
477 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 )
479 int j, k, n, off, cc_1, cc_2, cc[3];
480 // int flag[3] ; // debugging flag
481 uint16_t s1, s2, s3, s4;
486 if ( buf->flags & 16 )
488 /* Frame is progressive, be more discerning. */
489 color_diff = prog_diff;
490 color_equal = prog_equal;
491 threshold = prog_threshold;
494 /* One pas for Y, one pass for Cb, one pass for Cr */
495 for( k = 0; k < 3; k++ )
499 /* Y has already been checked, now offset by Y's dimensions
500 and divide all the other values by 2, since Cr and Cb
501 are half-size compared to Y. */
502 offset = width * height;
508 /* Y and Cb are done, so the offset needs to be bumped
509 so it's width*height + (width / 2) * (height / 2) */
513 for( j = 0; j < width; ++j )
517 for( n = 0; n < ( height - 4 ); n = n + 2 )
519 /* Look at groups of 4 sequential horizontal lines */
520 s1 = ( ( buf->data + offset )[ off + j ] & 0xff );
521 s2 = ( ( buf->data + offset )[ off + j + width ] & 0xff );
522 s3 = ( ( buf->data + offset )[ off + j + 2 * width ] & 0xff );
523 s4 = ( ( buf->data + offset )[ off + j + 3 * width ] & 0xff );
525 /* Note if the 1st and 2nd lines are more different in
526 color than the 1st and 3rd lines are similar in color.*/
527 if ( ( abs( s1 - s3 ) < color_equal ) &&
528 ( abs( s1 - s2 ) > color_diff ) )
531 /* Note if the 2nd and 3rd lines are more different in
532 color than the 2nd and 4th lines are similar in color.*/
533 if ( ( abs( s2 - s4 ) < color_equal ) &&
534 ( abs( s2 - s3 ) > color_diff) )
537 /* Now move down 2 horizontal lines before starting over.*/
543 /* The final cc score for a plane is the percentage of combed pixels it contains.
544 Because sensitivity goes down to hundreths of a percent, multiply by 1000
545 so it will be easy to compare against the threhold value which is an integer. */
546 cc[k] = (int)( ( cc_1 + cc_2 ) * 1000.0 / ( width * height ) );
550 /* HandBrake is all yuv420, so weight the average percentage of all 3 planes accordingly.*/
551 int average_cc = ( 2 * cc[0] + ( cc[1] / 2 ) + ( cc[2] / 2 ) ) / 3;
553 /* Now see if that average percentage of combed pixels surpasses the threshold percentage given by the user.*/
554 if( average_cc > threshold )
557 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" );
563 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" );
566 /* Reaching this point means no combing detected. */
572 * Calculates job width and height for anamorphic content,
574 * @param job Handle to hb_job_t
575 * @param output_width Pointer to returned storage width
576 * @param output_height Pointer to returned storage height
577 * @param output_par_width Pointer to returned pixel width
578 @ param output_par_height Pointer to returned pixel height
580 void hb_set_anamorphic_size( hb_job_t * job,
581 int *output_width, int *output_height,
582 int *output_par_width, int *output_par_height )
584 /* Set up some variables to make the math easier to follow. */
585 hb_title_t * title = job->title;
586 int cropped_width = title->width - job->crop[2] - job->crop[3] ;
587 int cropped_height = title->height - job->crop[0] - job->crop[1] ;
588 double storage_aspect = (double)cropped_width / (double)cropped_height;
589 int mod = job->anamorphic.modulus ? job->anamorphic.modulus : 16;
590 double aspect = title->aspect;
592 int pixel_aspect_width = job->anamorphic.par_width;
593 int pixel_aspect_height = job->anamorphic.par_height;
595 /* If a source was really NTSC or PAL and the user specified ITU PAR
596 values, replace the standard PAR values with the ITU broadcast ones. */
597 if( title->width == 720 && job->anamorphic.itu_par )
599 // convert aspect to a scaled integer so we can test for 16:9 & 4:3
600 // aspect ratios ignoring insignificant differences in the LSBs of
601 // the floating point representation.
602 int iaspect = aspect * 9.;
604 /* Handle ITU PARs */
605 if (title->height == 480)
610 /* It's widescreen */
611 pixel_aspect_width = 40;
612 pixel_aspect_height = 33;
614 else if (iaspect == 12)
617 pixel_aspect_width = 10;
618 pixel_aspect_height = 11;
621 else if (title->height == 576)
626 /* It's widescreen */
627 pixel_aspect_width = 16;
628 pixel_aspect_height = 11;
630 else if (iaspect == 12)
633 pixel_aspect_width = 12;
634 pixel_aspect_height = 11;
639 /* Figure out what width the source would display at. */
640 int source_display_width = cropped_width * (double)pixel_aspect_width /
641 (double)pixel_aspect_height ;
644 3 different ways of deciding output dimensions:
645 - 1: Strict anamorphic, preserve source dimensions
646 - 2: Loose anamorphic, round to mod16 and preserve storage aspect ratio
647 - 3: Power user anamorphic, specify everything
650 switch( job->anamorphic.mode )
653 /* Strict anamorphic */
654 *output_width = cropped_width;
655 *output_height = cropped_height;
656 *output_par_width = title->pixel_aspect_width;
657 *output_par_height = title->pixel_aspect_height;
661 /* "Loose" anamorphic.
662 - Uses mod16-compliant dimensions,
663 - Allows users to set the width
666 // height: Gets set later, ignore user job->height value
668 /* Gotta handle bounding dimensions.
669 If the width is too big, just reset it with no rescaling.
670 Instead of using the aspect-scaled job height,
671 we need to see if the job width divided by the storage aspect
672 is bigger than the max. If so, set it to the max (this is sloppy).
673 If not, set job height to job width divided by storage aspect.
676 if ( job->maxWidth && (job->maxWidth < job->width) )
677 width = job->maxWidth;
679 /* Time to get picture width that divide cleanly.*/
680 width = MULTIPLE_MOD( width, mod);
682 /* Verify these new dimensions don't violate max height and width settings */
683 if ( job->maxWidth && (job->maxWidth < job->width) )
684 width = job->maxWidth;
686 height = ((double)width / storage_aspect) + 0.5;
688 if ( job->maxHeight && (job->maxHeight < height) )
689 height = job->maxHeight;
691 /* Time to get picture height that divide cleanly.*/
692 height = MULTIPLE_MOD( height, mod);
694 /* Verify these new dimensions don't violate max height and width settings */
695 if ( job->maxHeight && (job->maxHeight < height) )
696 height = job->maxHeight;
698 /* The film AR is the source's display width / cropped source height.
699 The output display width is the output height * film AR.
700 The output PAR is the output display width / output storage width. */
701 pixel_aspect_width = height * source_display_width / cropped_height;
702 pixel_aspect_height = width;
704 /* Pass the results back to the caller */
705 *output_width = width;
706 *output_height = height;
710 /* Anamorphic 3: Power User Jamboree
711 - Set everything based on specified values */
713 /* Use specified storage dimensions */
715 height = job->height;
717 /* Bind to max dimensions */
718 if( job->maxWidth && width > job->maxWidth )
719 width = job->maxWidth;
720 if( job->maxHeight && height > job->maxHeight )
721 height = job->maxHeight;
723 /* Time to get picture dimensions that divide cleanly.*/
724 width = MULTIPLE_MOD( width, mod);
725 height = MULTIPLE_MOD( height, mod);
727 /* Verify we're still within max dimensions */
728 if( job->maxWidth && width > job->maxWidth )
729 width = job->maxWidth - (mod/2);
730 if( job->maxHeight && height > job->maxHeight )
731 height = job->maxHeight - (mod/2);
733 /* Re-ensure we have picture dimensions that divide cleanly. */
734 width = MULTIPLE_MOD( width, mod );
735 height = MULTIPLE_MOD( height, mod );
737 /* That finishes the storage dimensions. On to display. */
738 if( job->anamorphic.dar_width && job->anamorphic.dar_height )
740 /* We need to adjust the PAR to produce this aspect. */
741 pixel_aspect_width = height * job->anamorphic.dar_width / job->anamorphic.dar_height;
742 pixel_aspect_height = width;
746 /* We first need the display ar.
747 That's the source display width divided by the source height after cropping.
748 Then we multiple the output height by that to get the pixel aspect width,
749 and the pixel aspect height is the storage width.*/
750 pixel_aspect_width = height * source_display_width / cropped_height;
751 pixel_aspect_height = width;
755 *output_width = width;
756 *output_height = height;
760 /* While x264 is smart enough to reduce fractions on its own, libavcodec
761 needs some help with the math, so lose superfluous factors. */
762 hb_reduce( output_par_width, output_par_height,
763 pixel_aspect_width, pixel_aspect_height );
767 * Calculates job width, height, and cropping parameters.
768 * @param job Handle to hb_job_t.
769 * @param aspect Desired aspect ratio. Value of -1 uses title aspect.
770 * @param pixels Maximum desired pixel count.
772 void hb_set_size( hb_job_t * job, double aspect, int pixels )
774 hb_title_t * title = job->title;
776 int croppedWidth = title->width - title->crop[2] - title->crop[3];
777 int croppedHeight = title->height - title->crop[0] - title->crop[1];
778 double croppedAspect = title->aspect * title->height * croppedWidth /
779 croppedHeight / title->width;
785 /* Keep the best possible aspect ratio */
786 aspect = croppedAspect;
789 /* Crop if necessary to obtain the desired ratio */
790 memcpy( job->crop, title->crop, 4 * sizeof( int ) );
791 if( aspect < croppedAspect )
793 /* Need to crop on the left and right */
794 addCrop = croppedWidth - aspect * croppedHeight * title->width /
795 title->aspect / title->height;
798 addCrop = ( addCrop + 1 ) / 2;
799 job->crop[2] += addCrop;
800 job->crop[3] += addCrop;
802 else if( addCrop & 2 )
805 job->crop[2] += addCrop - 1;
806 job->crop[3] += addCrop + 1;
811 job->crop[2] += addCrop;
812 job->crop[3] += addCrop;
815 else if( aspect > croppedAspect )
817 /* Need to crop on the top and bottom */
818 addCrop = croppedHeight - croppedWidth * title->aspect *
819 title->height / aspect / title->width;
822 addCrop = ( addCrop + 1 ) / 2;
823 job->crop[0] += addCrop;
824 job->crop[1] += addCrop;
826 else if( addCrop & 2 )
829 job->crop[0] += addCrop - 1;
830 job->crop[1] += addCrop + 1;
835 job->crop[0] += addCrop;
836 job->crop[1] += addCrop;
840 /* Compute a resolution from the number of pixels and aspect */
844 h = MULTIPLE_16( (int)( (double)w / aspect ) );
852 job->height = MULTIPLE_16( (int)( (double)job->width / aspect ) );
856 * Returns the number of jobs in the queue.
857 * @param h Handle to hb_handle_t.
858 * @return Number of jobs.
860 int hb_count( hb_handle_t * h )
862 return hb_list_count( h->jobs );
866 * Returns handle to job at index i within the job list.
867 * @param h Handle to hb_handle_t.
868 * @param i Index of job.
869 * @returns Handle to hb_job_t of desired job.
871 hb_job_t * hb_job( hb_handle_t * h, int i )
873 return hb_list_item( h->jobs, i );
876 hb_job_t * hb_current_job( hb_handle_t * h )
878 return( h->current_job );
882 * Adds a job to the job list.
883 * @param h Handle to hb_handle_t.
884 * @param job Handle to hb_job_t.
886 void hb_add( hb_handle_t * h, hb_job_t * job )
889 hb_title_t * title, * title_copy;
890 hb_chapter_t * chapter, * chapter_copy;
892 hb_subtitle_t * subtitle, * subtitle_copy;
898 title_copy = malloc( sizeof( hb_title_t ) );
899 memcpy( title_copy, title, sizeof( hb_title_t ) );
901 title_copy->list_chapter = hb_list_init();
902 for( i = 0; i < hb_list_count( title->list_chapter ); i++ )
904 chapter = hb_list_item( title->list_chapter, i );
905 chapter_copy = malloc( sizeof( hb_chapter_t ) );
906 memcpy( chapter_copy, chapter, sizeof( hb_chapter_t ) );
907 hb_list_add( title_copy->list_chapter, chapter_copy );
913 if( title->metadata )
915 title_copy->metadata = malloc( sizeof( hb_metadata_t ) );
917 if( title_copy->metadata )
919 memcpy( title_copy->metadata, title->metadata, sizeof( hb_metadata_t ) );
922 * Need to copy the artwork seperatly (TODO).
924 if( title->metadata->coverart )
926 title_copy->metadata->coverart = malloc( title->metadata->coverart_size );
927 if( title_copy->metadata->coverart )
929 memcpy( title_copy->metadata->coverart, title->metadata->coverart,
930 title->metadata->coverart_size );
932 title_copy->metadata->coverart_size = 0;
938 /* Copy the audio track(s) we want */
939 title_copy->list_audio = hb_list_init();
941 for( i = 0; i < hb_list_count(job->list_audio); i++ )
943 if( ( audio = hb_list_item( job->list_audio, i ) ) )
945 hb_list_add( title_copy->list_audio, hb_audio_copy(audio) );
949 title_copy->list_subtitle = hb_list_init();
952 * The following code is confusing, there are three ways in which
953 * we select subtitles and it depends on whether this is single or
956 * subtitle_scan may be enabled, in which case the first pass
957 * scans all subtitles of that language. The second pass does not
958 * select any because they are set at the end of the first pass.
960 * native_language may have a preferred language, in which case we
961 * may be switching the language we want for the subtitles in the
962 * first pass of a single pass, or the second pass of a two pass.
964 * We may have manually selected a subtitle, in which case that is
965 * selected in the first pass of a single pass, or the second of a
968 memset( audio_lang, 0, sizeof( audio_lang ) );
970 if ( job->indepth_scan || job->native_language ) {
973 * Find the first audio language that is being encoded
975 for( i = 0; i < hb_list_count(job->list_audio); i++ )
977 if( ( audio = hb_list_item( job->list_audio, i ) ) )
979 strncpy(audio_lang, audio->config.lang.iso639_2, sizeof(audio_lang));
985 * In all cases switch the language if we need to to our native
988 if( job->native_language )
990 if( strncasecmp( job->native_language, audio_lang,
991 sizeof( audio_lang ) ) != 0 )
996 hb_log( "Enabled subtitles in native language '%s', audio is in '%s'",
997 job->native_language, audio_lang);
1000 * The main audio track is not in our native language, so switch
1001 * the subtitles to use our native language instead.
1003 strncpy( audio_lang, job->native_language, sizeof( audio_lang ) );
1006 * native language is irrelevent, free it.
1008 free( job->native_language );
1009 job->native_language = NULL;
1015 * If doing a subtitle scan then add all the matching subtitles for this
1018 if ( job->indepth_scan )
1020 for( i=0; i < hb_list_count( title->list_subtitle ); i++ )
1022 subtitle = hb_list_item( title->list_subtitle, i );
1023 if( strcmp( subtitle->iso639_2, audio_lang ) == 0 )
1026 * Matched subtitle language with audio language, so
1027 * add this to our list to scan.
1029 * We will update the subtitle list on the second pass
1030 * later after the first pass has completed.
1032 subtitle_copy = malloc( sizeof( hb_subtitle_t ) );
1033 memcpy( subtitle_copy, subtitle, sizeof( hb_subtitle_t ) );
1034 hb_list_add( title_copy->list_subtitle, subtitle_copy );
1035 if ( job->native_language ) {
1037 * With native language just select the
1038 * first match in our langiage, not all of
1039 * them. Subsequent ones are likely to be commentary
1047 * Not doing a subtitle scan in this pass, but maybe we are in the
1050 if( job->select_subtitle )
1053 * Don't add subtitles here, we'll add them via select_subtitle
1054 * at the end of the subtitle_scan.
1058 * Definitely not doing a subtitle scan.
1060 if( job->pass != 1 && job->native_language )
1063 * We are not doing a subtitle scan but do want the
1064 * native langauge subtitle selected, so select it
1065 * for pass 0 or pass 2 of a two pass.
1067 for( i=0; i < hb_list_count( title->list_subtitle ); i++ )
1069 subtitle = hb_list_item( title->list_subtitle, i );
1070 if( strcmp( subtitle->iso639_2, audio_lang ) == 0 )
1073 * Matched subtitle language with audio language, so
1074 * add this to our list to scan.
1076 subtitle_copy = malloc( sizeof( hb_subtitle_t ) );
1077 memcpy( subtitle_copy, subtitle, sizeof( hb_subtitle_t ) );
1078 hb_list_add( title_copy->list_subtitle, subtitle_copy );
1084 * Manually selected subtitle, in which case only
1085 * bother adding them for pass 0 or pass 2 of a two
1088 if( job->pass != 1 )
1090 if( ( subtitle = hb_list_item( title->list_subtitle, job->subtitle ) ) )
1092 subtitle_copy = malloc( sizeof( hb_subtitle_t ) );
1093 memcpy( subtitle_copy, subtitle, sizeof( hb_subtitle_t ) );
1094 hb_list_add( title_copy->list_subtitle, subtitle_copy );
1102 job_copy = calloc( sizeof( hb_job_t ), 1 );
1103 memcpy( job_copy, job, sizeof( hb_job_t ) );
1104 title_copy->job = job_copy;
1105 job_copy->title = title_copy;
1106 job_copy->list_audio = title_copy->list_audio;
1107 job_copy->file = strdup( job->file );
1109 job_copy->pause = h->pause_lock;
1111 /* Copy the job filter list */
1115 int filter_count = hb_list_count( job->filters );
1116 job_copy->filters = hb_list_init();
1117 for( i = 0; i < filter_count; i++ )
1120 * Copy the filters, since the MacGui reuses the global filter objects
1121 * meaning that queued up jobs overwrite the previous filter settings.
1122 * In reality, settings is probably the only field that needs duplicating
1123 * since it's the only value that is ever changed. But name is duplicated
1124 * as well for completeness. Not copying private_data since it gets
1125 * created for each job in renderInit.
1127 hb_filter_object_t * filter = hb_list_item( job->filters, i );
1128 hb_filter_object_t * filter_copy = malloc( sizeof( hb_filter_object_t ) );
1129 memcpy( filter_copy, filter, sizeof( hb_filter_object_t ) );
1131 filter_copy->name = strdup( filter->name );
1132 if( filter->settings )
1133 filter_copy->settings = strdup( filter->settings );
1134 hb_list_add( job_copy->filters, filter_copy );
1138 /* Add the job to the list */
1139 hb_list_add( h->jobs, job_copy );
1140 h->job_count = hb_count(h);
1141 h->job_count_permanent++;
1145 * Removes a job from the job list.
1146 * @param h Handle to hb_handle_t.
1147 * @param job Handle to hb_job_t.
1149 void hb_rem( hb_handle_t * h, hb_job_t * job )
1151 hb_list_rem( h->jobs, job );
1153 h->job_count = hb_count(h);
1154 if (h->job_count_permanent)
1155 h->job_count_permanent--;
1157 /* XXX free everything XXX */
1161 * Starts the conversion process.
1162 * Sets state to HB_STATE_WORKING.
1163 * calls hb_work_init, to launch work thread. Stores handle to work thread.
1164 * @param h Handle to hb_handle_t.
1166 void hb_start( hb_handle_t * h )
1169 h->job_count = hb_list_count( h->jobs );
1170 h->job_count_permanent = h->job_count;
1172 hb_lock( h->state_lock );
1173 h->state.state = HB_STATE_WORKING;
1174 #define p h->state.param.working
1177 p.job_count = h->job_count;
1185 hb_unlock( h->state_lock );
1190 h->work_thread = hb_work_init( h->jobs, h->cpu_count,
1191 &h->work_die, &h->work_error, &h->current_job );
1195 * Pauses the conversion process.
1196 * @param h Handle to hb_handle_t.
1198 void hb_pause( hb_handle_t * h )
1202 hb_lock( h->pause_lock );
1205 hb_lock( h->state_lock );
1206 h->state.state = HB_STATE_PAUSED;
1207 hb_unlock( h->state_lock );
1212 * Resumes the conversion process.
1213 * @param h Handle to hb_handle_t.
1215 void hb_resume( hb_handle_t * h )
1219 hb_unlock( h->pause_lock );
1225 * Stops the conversion process.
1226 * @param h Handle to hb_handle_t.
1228 void hb_stop( hb_handle_t * h )
1232 h->job_count = hb_count(h);
1233 h->job_count_permanent = 0;
1239 * Returns the state of the conversion process.
1240 * @param h Handle to hb_handle_t.
1241 * @param s Handle to hb_state_t which to copy the state data.
1243 void hb_get_state( hb_handle_t * h, hb_state_t * s )
1245 hb_lock( h->state_lock );
1247 memcpy( s, &h->state, sizeof( hb_state_t ) );
1248 if ( h->state.state == HB_STATE_SCANDONE || h->state.state == HB_STATE_WORKDONE )
1249 h->state.state = HB_STATE_IDLE;
1251 hb_unlock( h->state_lock );
1254 void hb_get_state2( hb_handle_t * h, hb_state_t * s )
1256 hb_lock( h->state_lock );
1258 memcpy( s, &h->state, sizeof( hb_state_t ) );
1260 hb_unlock( h->state_lock );
1264 * Called in MacGui in UpdateUI to check
1265 * for a new scan being completed to set a new source
1267 int hb_get_scancount( hb_handle_t * h)
1269 return h->scanCount;
1273 * Closes access to libhb by freeing the hb_handle_t handle ontained in hb_init_real.
1274 * @param _h Pointer to handle to hb_handle_t.
1276 void hb_close( hb_handle_t ** _h )
1278 hb_handle_t * h = *_h;
1282 hb_thread_close( &h->main_thread );
1284 while( ( title = hb_list_item( h->list_title, 0 ) ) )
1286 hb_list_rem( h->list_title, title );
1287 if( title->job && title->job->filters )
1289 hb_list_close( &title->job->filters );
1292 hb_title_close( &title );
1294 hb_list_close( &h->list_title );
1296 hb_list_close( &h->jobs );
1297 hb_lock_close( &h->state_lock );
1298 hb_lock_close( &h->pause_lock );
1305 * Monitors the state of the update, scan, and work threads.
1306 * Sets scan done state when scan thread exits.
1307 * Sets work done state when work thread exits.
1308 * @param _h Handle to hb_handle_t
1310 static void thread_func( void * _h )
1312 hb_handle_t * h = (hb_handle_t *) _h;
1315 struct dirent * entry;
1319 /* Create folder for temporary files */
1320 memset( dirname, 0, 1024 );
1321 hb_get_tempory_directory( h, dirname );
1323 hb_mkdir( dirname );
1327 /* In case the check_update thread hangs, it'll die sooner or
1328 later. Then, we join it here */
1329 if( h->update_thread &&
1330 hb_thread_has_exited( h->update_thread ) )
1332 hb_thread_close( &h->update_thread );
1335 /* Check if the scan thread is done */
1336 if( h->scan_thread &&
1337 hb_thread_has_exited( h->scan_thread ) )
1339 hb_thread_close( &h->scan_thread );
1341 hb_log( "libhb: scan thread found %d valid title(s)",
1342 hb_list_count( h->list_title ) );
1343 hb_lock( h->state_lock );
1344 h->state.state = HB_STATE_SCANDONE; //originally state.state
1345 hb_unlock( h->state_lock );
1346 /*we increment this sessions scan count by one for the MacGui
1347 to trigger a new source being set */
1351 /* Check if the work thread is done */
1352 if( h->work_thread &&
1353 hb_thread_has_exited( h->work_thread ) )
1355 hb_thread_close( &h->work_thread );
1357 hb_log( "libhb: work result = %d",
1359 hb_lock( h->state_lock );
1360 h->state.state = HB_STATE_WORKDONE;
1361 h->state.param.workdone.error = h->work_error;
1363 h->job_count = hb_count(h);
1364 if (h->job_count < 1)
1365 h->job_count_permanent = 0;
1366 hb_unlock( h->state_lock );
1372 if( h->work_thread )
1375 hb_thread_close( &h->work_thread );
1378 /* Remove temp folder */
1379 dir = opendir( dirname );
1382 while( ( entry = readdir( dir ) ) )
1384 char filename[1024];
1385 if( entry->d_name[0] == '.' )
1389 memset( filename, 0, 1024 );
1390 snprintf( filename, 1023, "%s/%s", dirname, entry->d_name );
1400 * @param h Handle to hb_handle_t
1402 int hb_get_pid( hb_handle_t * h )
1408 * Sets the current state.
1409 * @param h Handle to hb_handle_t
1410 * @param s Handle to new hb_state_t
1412 void hb_set_state( hb_handle_t * h, hb_state_t * s )
1414 hb_lock( h->pause_lock );
1415 hb_lock( h->state_lock );
1416 memcpy( &h->state, s, sizeof( hb_state_t ) );
1417 if( h->state.state == HB_STATE_WORKING )
1420 if (h->job_count < 1)
1421 h->job_count_permanent = 1;
1423 h->state.param.working.job_cur =
1424 h->job_count_permanent - hb_list_count( h->jobs );
1425 h->state.param.working.job_count = h->job_count_permanent;
1427 // Set which job is being worked on
1429 h->state.param.working.sequence_id = h->current_job->sequence_id;
1431 h->state.param.working.sequence_id = 0;
1433 hb_unlock( h->state_lock );
1434 hb_unlock( h->pause_lock );