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_decvobsub );
233 hb_register( &hb_encvobsub );
234 hb_register( &hb_deccc608 );
235 hb_register( &hb_render );
236 hb_register( &hb_encavcodec );
237 hb_register( &hb_encxvid );
238 hb_register( &hb_encx264 );
239 hb_register( &hb_enctheora );
240 hb_register( &hb_deca52 );
241 hb_register( &hb_decdca );
242 hb_register( &hb_decavcodec );
243 hb_register( &hb_decavcodecv );
244 hb_register( &hb_decavcodecvi );
245 hb_register( &hb_decavcodecai );
246 hb_register( &hb_declpcm );
247 hb_register( &hb_encfaac );
248 hb_register( &hb_enclame );
249 hb_register( &hb_encvorbis );
256 * Returns current version of libhb.
257 * @param h Handle to hb_handle_t.
258 * @return character array of version number.
260 char * hb_get_version( hb_handle_t * h )
262 return HB_PROJECT_VERSION;
266 * Returns current build of libhb.
267 * @param h Handle to hb_handle_t.
268 * @return character array of build number.
270 int hb_get_build( hb_handle_t * h )
272 return HB_PROJECT_BUILD;
276 * Checks for needed update.
277 * @param h Handle to hb_handle_t.
278 * @param version Pointer to handle where version will be copied.
279 * @return update indicator.
281 int hb_check_update( hb_handle_t * h, char ** version )
283 *version = ( h->build < 0 ) ? NULL : h->version;
288 * Sets the cpu count to the desired value.
289 * @param h Handle to hb_handle_t
290 * @param cpu_count Number of CPUs to use.
292 void hb_set_cpu_count( hb_handle_t * h, int cpu_count )
294 cpu_count = MAX( 1, cpu_count );
295 cpu_count = MIN( cpu_count, 8 );
296 h->cpu_count = cpu_count;
300 * Deletes current previews associated with titles
301 * @param h Handle to hb_handle_t
303 void hb_remove_previews( hb_handle_t * h )
310 struct dirent * entry;
312 memset( dirname, 0, 1024 );
313 hb_get_tempory_directory( h, dirname );
314 dir = opendir( dirname );
315 if (dir == NULL) return;
317 count = hb_list_count( h->list_title );
318 while( ( entry = readdir( dir ) ) )
320 if( entry->d_name[0] == '.' )
324 for( i = 0; i < count; i++ )
326 title = hb_list_item( h->list_title, i );
327 len = snprintf( filename, 1024, "%" PRIxPTR, (intptr_t) title );
328 if (strncmp(entry->d_name, filename, len) == 0)
330 snprintf( filename, 1024, "%s/%s", dirname, entry->d_name );
340 * Initializes a scan of the by calling hb_scan_init
341 * @param h Handle to hb_handle_t
342 * @param path location of VIDEO_TS folder.
343 * @param title_index Desired title to scan. 0 for all titles.
344 * @param preview_count Number of preview images to generate.
345 * @param store_previews Whether or not to write previews to disk.
347 void hb_scan( hb_handle_t * h, const char * path, int title_index,
348 int preview_count, int store_previews )
352 /* Clean up from previous scan */
353 hb_remove_previews( h );
354 while( ( title = hb_list_item( h->list_title, 0 ) ) )
356 hb_list_rem( h->list_title, title );
357 hb_title_close( &title );
360 hb_log( "hb_scan: path=%s, title_index=%d", path, title_index );
361 h->scan_thread = hb_scan_init( h, path, title_index, h->list_title,
362 preview_count, store_previews );
366 * Returns the list of titles found.
367 * @param h Handle to hb_handle_t
368 * @return Handle to hb_list_t of the title list.
370 hb_list_t * hb_get_titles( hb_handle_t * h )
372 return h->list_title;
376 * Create preview image of desired title a index of picture.
377 * @param h Handle to hb_handle_t.
378 * @param title Handle to hb_title_t of desired title.
379 * @param picture Index in title.
380 * @param buffer Handle to buufer were inage will be drawn.
382 void hb_get_preview( hb_handle_t * h, hb_title_t * title, int picture,
385 hb_job_t * job = title->job;
388 uint8_t * buf1, * buf2, * buf3, * buf4, * pen;
390 AVPicture pic_in, pic_preview, pic_deint, pic_crop, pic_scale;
391 struct SwsContext * context;
393 int rgb_width = ((job->width + 7) >> 3) << 3;
396 swsflags = SWS_LANCZOS | SWS_ACCURATE_RND;
398 buf1 = av_malloc( avpicture_get_size( PIX_FMT_YUV420P, title->width, title->height ) );
399 buf2 = av_malloc( avpicture_get_size( PIX_FMT_YUV420P, title->width, title->height ) );
400 buf3 = av_malloc( avpicture_get_size( PIX_FMT_YUV420P, rgb_width, job->height ) );
401 buf4 = av_malloc( avpicture_get_size( PIX_FMT_RGBA32, rgb_width, job->height ) );
402 avpicture_fill( &pic_in, buf1, PIX_FMT_YUV420P,
403 title->width, title->height );
404 avpicture_fill( &pic_deint, buf2, PIX_FMT_YUV420P,
405 title->width, title->height );
406 avpicture_fill( &pic_scale, buf3, PIX_FMT_YUV420P,
407 rgb_width, job->height );
408 avpicture_fill( &pic_preview, buf4, PIX_FMT_RGBA32,
409 rgb_width, job->height );
411 // Allocate the AVPicture frames and fill in
413 memset( filename, 0, 1024 );
415 hb_get_tempory_filename( h, filename, "%" PRIxPTR "%d",
416 (intptr_t) title, picture );
418 file = fopen( filename, "r" );
421 hb_log( "hb_get_preview: fopen failed" );
425 fread( buf1, avpicture_get_size( PIX_FMT_YUV420P, title->width, title->height), 1, file );
428 if( job->deinterlace )
430 // Deinterlace and crop
431 avpicture_deinterlace( &pic_deint, &pic_in, PIX_FMT_YUV420P, title->width, title->height );
432 av_picture_crop( &pic_crop, &pic_deint, PIX_FMT_YUV420P, job->crop[0], job->crop[2] );
437 av_picture_crop( &pic_crop, &pic_in, PIX_FMT_YUV420P, job->crop[0], job->crop[2] );
440 // Get scaling context
441 context = sws_getContext(title->width - (job->crop[2] + job->crop[3]),
442 title->height - (job->crop[0] + job->crop[1]),
444 rgb_width, job->height, PIX_FMT_YUV420P,
445 swsflags, NULL, NULL, NULL);
449 pic_crop.data, pic_crop.linesize,
450 0, title->height - (job->crop[0] + job->crop[1]),
451 pic_scale.data, pic_scale.linesize);
454 sws_freeContext( context );
456 // Get preview context
457 context = sws_getContext(rgb_width, job->height, PIX_FMT_YUV420P,
458 rgb_width, job->height, PIX_FMT_RGBA32,
459 swsflags, NULL, NULL, NULL);
463 pic_scale.data, pic_scale.linesize,
465 pic_preview.data, pic_preview.linesize);
468 sws_freeContext( context );
470 preview_size = pic_preview.linesize[0];
472 for( i = 0; i < job->height; i++ )
474 memcpy( pen, buf4 + preview_size * i, 4 * job->width );
475 pen += 4 * job->width;
479 avpicture_free( &pic_preview );
480 avpicture_free( &pic_scale );
481 avpicture_free( &pic_deint );
482 avpicture_free( &pic_in );
486 * Analyzes a frame to detect interlacing artifacts
487 * and returns true if interlacing (combing) is found.
489 * Code taken from Thomas Oestreich's 32detect filter
490 * in the Transcode project, with minor formatting changes.
492 * @param buf An hb_buffer structure holding valid frame data
493 * @param width The frame's width in pixels
494 * @param height The frame's height in pixels
495 * @param color_equal Sensitivity for detecting similar colors
496 * @param color_diff Sensitivity for detecting different colors
497 * @param threshold Sensitivity for flagging planes as combed
498 * @param prog_equal Sensitivity for detecting similar colors on progressive frames
499 * @param prog_diff Sensitivity for detecting different colors on progressive frames
500 * @param prog_threshold Sensitivity for flagging progressive frames as combed
502 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 )
504 int j, k, n, off, cc_1, cc_2, cc[3];
505 // int flag[3] ; // debugging flag
506 uint16_t s1, s2, s3, s4;
511 if ( buf->flags & 16 )
513 /* Frame is progressive, be more discerning. */
514 color_diff = prog_diff;
515 color_equal = prog_equal;
516 threshold = prog_threshold;
519 /* One pas for Y, one pass for Cb, one pass for Cr */
520 for( k = 0; k < 3; k++ )
524 /* Y has already been checked, now offset by Y's dimensions
525 and divide all the other values by 2, since Cr and Cb
526 are half-size compared to Y. */
527 offset = width * height;
533 /* Y and Cb are done, so the offset needs to be bumped
534 so it's width*height + (width / 2) * (height / 2) */
538 for( j = 0; j < width; ++j )
542 for( n = 0; n < ( height - 4 ); n = n + 2 )
544 /* Look at groups of 4 sequential horizontal lines */
545 s1 = ( ( buf->data + offset )[ off + j ] & 0xff );
546 s2 = ( ( buf->data + offset )[ off + j + width ] & 0xff );
547 s3 = ( ( buf->data + offset )[ off + j + 2 * width ] & 0xff );
548 s4 = ( ( buf->data + offset )[ off + j + 3 * width ] & 0xff );
550 /* Note if the 1st and 2nd lines are more different in
551 color than the 1st and 3rd lines are similar in color.*/
552 if ( ( abs( s1 - s3 ) < color_equal ) &&
553 ( abs( s1 - s2 ) > color_diff ) )
556 /* Note if the 2nd and 3rd lines are more different in
557 color than the 2nd and 4th lines are similar in color.*/
558 if ( ( abs( s2 - s4 ) < color_equal ) &&
559 ( abs( s2 - s3 ) > color_diff) )
562 /* Now move down 2 horizontal lines before starting over.*/
568 /* The final cc score for a plane is the percentage of combed pixels it contains.
569 Because sensitivity goes down to hundreths of a percent, multiply by 1000
570 so it will be easy to compare against the threhold value which is an integer. */
571 cc[k] = (int)( ( cc_1 + cc_2 ) * 1000.0 / ( width * height ) );
575 /* HandBrake is all yuv420, so weight the average percentage of all 3 planes accordingly.*/
576 int average_cc = ( 2 * cc[0] + ( cc[1] / 2 ) + ( cc[2] / 2 ) ) / 3;
578 /* Now see if that average percentage of combed pixels surpasses the threshold percentage given by the user.*/
579 if( average_cc > threshold )
582 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" );
588 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" );
591 /* Reaching this point means no combing detected. */
597 * Calculates job width and height for anamorphic content,
599 * @param job Handle to hb_job_t
600 * @param output_width Pointer to returned storage width
601 * @param output_height Pointer to returned storage height
602 * @param output_par_width Pointer to returned pixel width
603 @ param output_par_height Pointer to returned pixel height
605 void hb_set_anamorphic_size( hb_job_t * job,
606 int *output_width, int *output_height,
607 int *output_par_width, int *output_par_height )
609 /* Set up some variables to make the math easier to follow. */
610 hb_title_t * title = job->title;
611 int cropped_width = title->width - job->crop[2] - job->crop[3] ;
612 int cropped_height = title->height - job->crop[0] - job->crop[1] ;
613 double storage_aspect = (double)cropped_width / (double)cropped_height;
614 int mod = job->anamorphic.modulus ? job->anamorphic.modulus : 16;
615 double aspect = title->aspect;
617 int pixel_aspect_width = job->anamorphic.par_width;
618 int pixel_aspect_height = job->anamorphic.par_height;
620 /* If a source was really NTSC or PAL and the user specified ITU PAR
621 values, replace the standard PAR values with the ITU broadcast ones. */
622 if( title->width == 720 && job->anamorphic.itu_par )
624 // convert aspect to a scaled integer so we can test for 16:9 & 4:3
625 // aspect ratios ignoring insignificant differences in the LSBs of
626 // the floating point representation.
627 int iaspect = aspect * 9.;
629 /* Handle ITU PARs */
630 if (title->height == 480)
635 /* It's widescreen */
636 pixel_aspect_width = 40;
637 pixel_aspect_height = 33;
639 else if (iaspect == 12)
642 pixel_aspect_width = 10;
643 pixel_aspect_height = 11;
646 else if (title->height == 576)
651 /* It's widescreen */
652 pixel_aspect_width = 16;
653 pixel_aspect_height = 11;
655 else if (iaspect == 12)
658 pixel_aspect_width = 12;
659 pixel_aspect_height = 11;
664 /* Figure out what width the source would display at. */
665 int source_display_width = cropped_width * (double)pixel_aspect_width /
666 (double)pixel_aspect_height ;
669 3 different ways of deciding output dimensions:
670 - 1: Strict anamorphic, preserve source dimensions
671 - 2: Loose anamorphic, round to mod16 and preserve storage aspect ratio
672 - 3: Power user anamorphic, specify everything
675 switch( job->anamorphic.mode )
678 /* Strict anamorphic */
679 *output_width = cropped_width;
680 *output_height = cropped_height;
681 *output_par_width = title->pixel_aspect_width;
682 *output_par_height = title->pixel_aspect_height;
686 /* "Loose" anamorphic.
687 - Uses mod16-compliant dimensions,
688 - Allows users to set the width
691 // height: Gets set later, ignore user job->height value
693 /* Gotta handle bounding dimensions.
694 If the width is too big, just reset it with no rescaling.
695 Instead of using the aspect-scaled job height,
696 we need to see if the job width divided by the storage aspect
697 is bigger than the max. If so, set it to the max (this is sloppy).
698 If not, set job height to job width divided by storage aspect.
701 if ( job->maxWidth && (job->maxWidth < job->width) )
702 width = job->maxWidth;
704 /* Time to get picture width that divide cleanly.*/
705 width = MULTIPLE_MOD( width, mod);
707 /* Verify these new dimensions don't violate max height and width settings */
708 if ( job->maxWidth && (job->maxWidth < job->width) )
709 width = job->maxWidth;
711 height = ((double)width / storage_aspect) + 0.5;
713 if ( job->maxHeight && (job->maxHeight < height) )
714 height = job->maxHeight;
716 /* Time to get picture height that divide cleanly.*/
717 height = MULTIPLE_MOD( height, mod);
719 /* Verify these new dimensions don't violate max height and width settings */
720 if ( job->maxHeight && (job->maxHeight < height) )
721 height = job->maxHeight;
723 /* The film AR is the source's display width / cropped source height.
724 The output display width is the output height * film AR.
725 The output PAR is the output display width / output storage width. */
726 pixel_aspect_width = height * source_display_width / cropped_height;
727 pixel_aspect_height = width;
729 /* Pass the results back to the caller */
730 *output_width = width;
731 *output_height = height;
735 /* Anamorphic 3: Power User Jamboree
736 - Set everything based on specified values */
738 /* Use specified storage dimensions */
740 height = job->height;
742 /* Bind to max dimensions */
743 if( job->maxWidth && width > job->maxWidth )
744 width = job->maxWidth;
745 if( job->maxHeight && height > job->maxHeight )
746 height = job->maxHeight;
748 /* Time to get picture dimensions that divide cleanly.*/
749 width = MULTIPLE_MOD( width, mod);
750 height = MULTIPLE_MOD( height, mod);
752 /* Verify we're still within max dimensions */
753 if( job->maxWidth && width > job->maxWidth )
754 width = job->maxWidth - (mod/2);
755 if( job->maxHeight && height > job->maxHeight )
756 height = job->maxHeight - (mod/2);
758 /* Re-ensure we have picture dimensions that divide cleanly. */
759 width = MULTIPLE_MOD( width, mod );
760 height = MULTIPLE_MOD( height, mod );
762 /* That finishes the storage dimensions. On to display. */
763 if( job->anamorphic.dar_width && job->anamorphic.dar_height )
765 /* We need to adjust the PAR to produce this aspect. */
766 pixel_aspect_width = height * job->anamorphic.dar_width / job->anamorphic.dar_height;
767 pixel_aspect_height = width;
771 /* We first need the display ar.
772 That's the source display width divided by the source height after cropping.
773 Then we multiple the output height by that to get the pixel aspect width,
774 and the pixel aspect height is the storage width.*/
775 pixel_aspect_width = height * source_display_width / cropped_height;
776 pixel_aspect_height = width;
780 *output_width = width;
781 *output_height = height;
785 /* While x264 is smart enough to reduce fractions on its own, libavcodec
786 needs some help with the math, so lose superfluous factors. */
787 hb_reduce( output_par_width, output_par_height,
788 pixel_aspect_width, pixel_aspect_height );
792 * Calculates job width, height, and cropping parameters.
793 * @param job Handle to hb_job_t.
794 * @param aspect Desired aspect ratio. Value of -1 uses title aspect.
795 * @param pixels Maximum desired pixel count.
797 void hb_set_size( hb_job_t * job, double aspect, int pixels )
799 hb_title_t * title = job->title;
801 int croppedWidth = title->width - title->crop[2] - title->crop[3];
802 int croppedHeight = title->height - title->crop[0] - title->crop[1];
803 double croppedAspect = title->aspect * title->height * croppedWidth /
804 croppedHeight / title->width;
810 /* Keep the best possible aspect ratio */
811 aspect = croppedAspect;
814 /* Crop if necessary to obtain the desired ratio */
815 memcpy( job->crop, title->crop, 4 * sizeof( int ) );
816 if( aspect < croppedAspect )
818 /* Need to crop on the left and right */
819 addCrop = croppedWidth - aspect * croppedHeight * title->width /
820 title->aspect / title->height;
823 addCrop = ( addCrop + 1 ) / 2;
824 job->crop[2] += addCrop;
825 job->crop[3] += addCrop;
827 else if( addCrop & 2 )
830 job->crop[2] += addCrop - 1;
831 job->crop[3] += addCrop + 1;
836 job->crop[2] += addCrop;
837 job->crop[3] += addCrop;
840 else if( aspect > croppedAspect )
842 /* Need to crop on the top and bottom */
843 addCrop = croppedHeight - croppedWidth * title->aspect *
844 title->height / aspect / title->width;
847 addCrop = ( addCrop + 1 ) / 2;
848 job->crop[0] += addCrop;
849 job->crop[1] += addCrop;
851 else if( addCrop & 2 )
854 job->crop[0] += addCrop - 1;
855 job->crop[1] += addCrop + 1;
860 job->crop[0] += addCrop;
861 job->crop[1] += addCrop;
865 /* Compute a resolution from the number of pixels and aspect */
869 h = MULTIPLE_16( (int)( (double)w / aspect ) );
877 job->height = MULTIPLE_16( (int)( (double)job->width / aspect ) );
881 * Returns the number of jobs in the queue.
882 * @param h Handle to hb_handle_t.
883 * @return Number of jobs.
885 int hb_count( hb_handle_t * h )
887 return hb_list_count( h->jobs );
891 * Returns handle to job at index i within the job list.
892 * @param h Handle to hb_handle_t.
893 * @param i Index of job.
894 * @returns Handle to hb_job_t of desired job.
896 hb_job_t * hb_job( hb_handle_t * h, int i )
898 return hb_list_item( h->jobs, i );
901 hb_job_t * hb_current_job( hb_handle_t * h )
903 return( h->current_job );
907 * Adds a job to the job list.
908 * @param h Handle to hb_handle_t.
909 * @param job Handle to hb_job_t.
911 void hb_add( hb_handle_t * h, hb_job_t * job )
914 hb_title_t * title, * title_copy;
915 hb_chapter_t * chapter, * chapter_copy;
917 hb_subtitle_t * subtitle, * subtitle_copy;
923 title_copy = malloc( sizeof( hb_title_t ) );
924 memcpy( title_copy, title, sizeof( hb_title_t ) );
926 title_copy->list_chapter = hb_list_init();
927 for( i = 0; i < hb_list_count( title->list_chapter ); i++ )
929 chapter = hb_list_item( title->list_chapter, i );
930 chapter_copy = malloc( sizeof( hb_chapter_t ) );
931 memcpy( chapter_copy, chapter, sizeof( hb_chapter_t ) );
932 hb_list_add( title_copy->list_chapter, chapter_copy );
938 if( title->metadata )
940 title_copy->metadata = malloc( sizeof( hb_metadata_t ) );
942 if( title_copy->metadata )
944 memcpy( title_copy->metadata, title->metadata, sizeof( hb_metadata_t ) );
947 * Need to copy the artwork seperatly (TODO).
949 if( title->metadata->coverart )
951 title_copy->metadata->coverart = malloc( title->metadata->coverart_size );
952 if( title_copy->metadata->coverart )
954 memcpy( title_copy->metadata->coverart, title->metadata->coverart,
955 title->metadata->coverart_size );
957 title_copy->metadata->coverart_size = 0;
963 /* Copy the audio track(s) we want */
964 title_copy->list_audio = hb_list_init();
966 for( i = 0; i < hb_list_count(job->list_audio); i++ )
968 if( ( audio = hb_list_item( job->list_audio, i ) ) )
970 hb_list_add( title_copy->list_audio, hb_audio_copy(audio) );
974 title_copy->list_subtitle = hb_list_init();
977 * The following code is confusing, there are three ways in which
978 * we select subtitles and it depends on whether this is single or
981 * subtitle_scan may be enabled, in which case the first pass
982 * scans all subtitles of that language. The second pass does not
983 * select any because they are set at the end of the first pass.
985 * native_language may have a preferred language, in which case we
986 * may be switching the language we want for the subtitles in the
987 * first pass of a single pass, or the second pass of a two pass.
989 * We may have manually selected a subtitle, in which case that is
990 * selected in the first pass of a single pass, or the second of a
993 memset( audio_lang, 0, sizeof( audio_lang ) );
995 if ( job->indepth_scan || job->native_language ) {
998 * Find the first audio language that is being encoded
1000 for( i = 0; i < hb_list_count(job->list_audio); i++ )
1002 if( ( audio = hb_list_item( job->list_audio, i ) ) )
1004 strncpy(audio_lang, audio->config.lang.iso639_2, sizeof(audio_lang));
1010 * In all cases switch the language if we need to to our native
1013 if( job->native_language )
1015 if( strncasecmp( job->native_language, audio_lang,
1016 sizeof( audio_lang ) ) != 0 )
1019 if( job->pass != 2 )
1021 hb_log( "Enabled subtitles in native language '%s', audio is in '%s'",
1022 job->native_language, audio_lang);
1025 * The main audio track is not in our native language, so switch
1026 * the subtitles to use our native language instead.
1028 strncpy( audio_lang, job->native_language, sizeof( audio_lang ) );
1031 * native language is irrelevent, free it.
1033 free( job->native_language );
1034 job->native_language = NULL;
1040 * If doing a subtitle scan then add all the matching subtitles for this
1043 if ( job->indepth_scan )
1045 for( i=0; i < hb_list_count( title->list_subtitle ); i++ )
1047 subtitle = hb_list_item( title->list_subtitle, i );
1048 if( strcmp( subtitle->iso639_2, audio_lang ) == 0 )
1051 * Matched subtitle language with audio language, so
1052 * add this to our list to scan.
1054 * We will update the subtitle list on the second pass
1055 * later after the first pass has completed.
1057 subtitle_copy = malloc( sizeof( hb_subtitle_t ) );
1058 memcpy( subtitle_copy, subtitle, sizeof( hb_subtitle_t ) );
1059 hb_list_add( title_copy->list_subtitle, subtitle_copy );
1060 if ( job->native_language ) {
1062 * With native language just select the
1063 * first match in our langiage, not all of
1064 * them. Subsequent ones are likely to be commentary
1072 * Not doing a subtitle scan in this pass, but maybe we are in the
1075 if( job->select_subtitle )
1078 * Don't add subtitles here, we'll add them via select_subtitle
1079 * at the end of the subtitle_scan.
1083 * Definitely not doing a subtitle scan.
1085 if( job->pass != 1 && job->native_language )
1088 * We are not doing a subtitle scan but do want the
1089 * native langauge subtitle selected, so select it
1090 * for pass 0 or pass 2 of a two pass.
1092 for( i=0; i < hb_list_count( title->list_subtitle ); i++ )
1094 subtitle = hb_list_item( title->list_subtitle, i );
1095 if( strcmp( subtitle->iso639_2, audio_lang ) == 0 )
1098 * Matched subtitle language with audio language, so
1099 * add this to our list to scan.
1101 subtitle_copy = malloc( sizeof( hb_subtitle_t ) );
1102 memcpy( subtitle_copy, subtitle, sizeof( hb_subtitle_t ) );
1103 hb_list_add( title_copy->list_subtitle, subtitle_copy );
1109 * Manually selected subtitles, in which case only
1110 * bother adding them for pass 0 or pass 2 of a two
1113 if( job->pass != 1 )
1116 * Copy all of them from the input job, to the title_copy/job_copy.
1118 for( i = 0; i < hb_list_count(job->list_subtitle); i++ ) {
1119 if( ( subtitle = hb_list_item( job->list_subtitle, i ) ) )
1121 subtitle_copy = malloc( sizeof( hb_subtitle_t ) );
1122 memcpy( subtitle_copy, subtitle, sizeof( hb_subtitle_t ) );
1123 hb_list_add( title_copy->list_subtitle, subtitle_copy );
1132 job_copy = calloc( sizeof( hb_job_t ), 1 );
1133 memcpy( job_copy, job, sizeof( hb_job_t ) );
1134 title_copy->job = job_copy;
1135 job_copy->title = title_copy;
1136 job_copy->list_audio = title_copy->list_audio;
1137 job_copy->list_subtitle = title_copy->list_subtitle; // sharing list between title and job
1138 job_copy->file = strdup( job->file );
1140 job_copy->pause = h->pause_lock;
1142 /* Copy the job filter list */
1146 int filter_count = hb_list_count( job->filters );
1147 job_copy->filters = hb_list_init();
1148 for( i = 0; i < filter_count; i++ )
1151 * Copy the filters, since the MacGui reuses the global filter objects
1152 * meaning that queued up jobs overwrite the previous filter settings.
1153 * In reality, settings is probably the only field that needs duplicating
1154 * since it's the only value that is ever changed. But name is duplicated
1155 * as well for completeness. Not copying private_data since it gets
1156 * created for each job in renderInit.
1158 hb_filter_object_t * filter = hb_list_item( job->filters, i );
1159 hb_filter_object_t * filter_copy = malloc( sizeof( hb_filter_object_t ) );
1160 memcpy( filter_copy, filter, sizeof( hb_filter_object_t ) );
1162 filter_copy->name = strdup( filter->name );
1163 if( filter->settings )
1164 filter_copy->settings = strdup( filter->settings );
1165 hb_list_add( job_copy->filters, filter_copy );
1169 /* Add the job to the list */
1170 hb_list_add( h->jobs, job_copy );
1171 h->job_count = hb_count(h);
1172 h->job_count_permanent++;
1176 * Removes a job from the job list.
1177 * @param h Handle to hb_handle_t.
1178 * @param job Handle to hb_job_t.
1180 void hb_rem( hb_handle_t * h, hb_job_t * job )
1182 hb_list_rem( h->jobs, job );
1184 h->job_count = hb_count(h);
1185 if (h->job_count_permanent)
1186 h->job_count_permanent--;
1188 /* XXX free everything XXX */
1192 * Starts the conversion process.
1193 * Sets state to HB_STATE_WORKING.
1194 * calls hb_work_init, to launch work thread. Stores handle to work thread.
1195 * @param h Handle to hb_handle_t.
1197 void hb_start( hb_handle_t * h )
1200 h->job_count = hb_list_count( h->jobs );
1201 h->job_count_permanent = h->job_count;
1203 hb_lock( h->state_lock );
1204 h->state.state = HB_STATE_WORKING;
1205 #define p h->state.param.working
1208 p.job_count = h->job_count;
1216 hb_unlock( h->state_lock );
1221 h->work_thread = hb_work_init( h->jobs, h->cpu_count,
1222 &h->work_die, &h->work_error, &h->current_job );
1226 * Pauses the conversion process.
1227 * @param h Handle to hb_handle_t.
1229 void hb_pause( hb_handle_t * h )
1233 hb_lock( h->pause_lock );
1236 hb_lock( h->state_lock );
1237 h->state.state = HB_STATE_PAUSED;
1238 hb_unlock( h->state_lock );
1243 * Resumes the conversion process.
1244 * @param h Handle to hb_handle_t.
1246 void hb_resume( hb_handle_t * h )
1250 hb_unlock( h->pause_lock );
1256 * Stops the conversion process.
1257 * @param h Handle to hb_handle_t.
1259 void hb_stop( hb_handle_t * h )
1263 h->job_count = hb_count(h);
1264 h->job_count_permanent = 0;
1270 * Returns the state of the conversion process.
1271 * @param h Handle to hb_handle_t.
1272 * @param s Handle to hb_state_t which to copy the state data.
1274 void hb_get_state( hb_handle_t * h, hb_state_t * s )
1276 hb_lock( h->state_lock );
1278 memcpy( s, &h->state, sizeof( hb_state_t ) );
1279 if ( h->state.state == HB_STATE_SCANDONE || h->state.state == HB_STATE_WORKDONE )
1280 h->state.state = HB_STATE_IDLE;
1282 hb_unlock( h->state_lock );
1285 void hb_get_state2( hb_handle_t * h, hb_state_t * s )
1287 hb_lock( h->state_lock );
1289 memcpy( s, &h->state, sizeof( hb_state_t ) );
1291 hb_unlock( h->state_lock );
1295 * Called in MacGui in UpdateUI to check
1296 * for a new scan being completed to set a new source
1298 int hb_get_scancount( hb_handle_t * h)
1300 return h->scanCount;
1304 * Closes access to libhb by freeing the hb_handle_t handle ontained in hb_init_real.
1305 * @param _h Pointer to handle to hb_handle_t.
1307 void hb_close( hb_handle_t ** _h )
1309 hb_handle_t * h = *_h;
1313 hb_thread_close( &h->main_thread );
1315 while( ( title = hb_list_item( h->list_title, 0 ) ) )
1317 hb_list_rem( h->list_title, title );
1318 if( title->job && title->job->filters )
1320 hb_list_close( &title->job->filters );
1323 hb_title_close( &title );
1325 hb_list_close( &h->list_title );
1327 hb_list_close( &h->jobs );
1328 hb_lock_close( &h->state_lock );
1329 hb_lock_close( &h->pause_lock );
1336 * Monitors the state of the update, scan, and work threads.
1337 * Sets scan done state when scan thread exits.
1338 * Sets work done state when work thread exits.
1339 * @param _h Handle to hb_handle_t
1341 static void thread_func( void * _h )
1343 hb_handle_t * h = (hb_handle_t *) _h;
1346 struct dirent * entry;
1350 /* Create folder for temporary files */
1351 memset( dirname, 0, 1024 );
1352 hb_get_tempory_directory( h, dirname );
1354 hb_mkdir( dirname );
1358 /* In case the check_update thread hangs, it'll die sooner or
1359 later. Then, we join it here */
1360 if( h->update_thread &&
1361 hb_thread_has_exited( h->update_thread ) )
1363 hb_thread_close( &h->update_thread );
1366 /* Check if the scan thread is done */
1367 if( h->scan_thread &&
1368 hb_thread_has_exited( h->scan_thread ) )
1370 hb_thread_close( &h->scan_thread );
1372 hb_log( "libhb: scan thread found %d valid title(s)",
1373 hb_list_count( h->list_title ) );
1374 hb_lock( h->state_lock );
1375 h->state.state = HB_STATE_SCANDONE; //originally state.state
1376 hb_unlock( h->state_lock );
1377 /*we increment this sessions scan count by one for the MacGui
1378 to trigger a new source being set */
1382 /* Check if the work thread is done */
1383 if( h->work_thread &&
1384 hb_thread_has_exited( h->work_thread ) )
1386 hb_thread_close( &h->work_thread );
1388 hb_log( "libhb: work result = %d",
1390 hb_lock( h->state_lock );
1391 h->state.state = HB_STATE_WORKDONE;
1392 h->state.param.workdone.error = h->work_error;
1394 h->job_count = hb_count(h);
1395 if (h->job_count < 1)
1396 h->job_count_permanent = 0;
1397 hb_unlock( h->state_lock );
1403 if( h->work_thread )
1406 hb_thread_close( &h->work_thread );
1409 /* Remove temp folder */
1410 dir = opendir( dirname );
1413 while( ( entry = readdir( dir ) ) )
1415 char filename[1024];
1416 if( entry->d_name[0] == '.' )
1420 memset( filename, 0, 1024 );
1421 snprintf( filename, 1023, "%s/%s", dirname, entry->d_name );
1431 * @param h Handle to hb_handle_t
1433 int hb_get_pid( hb_handle_t * h )
1439 * Sets the current state.
1440 * @param h Handle to hb_handle_t
1441 * @param s Handle to new hb_state_t
1443 void hb_set_state( hb_handle_t * h, hb_state_t * s )
1445 hb_lock( h->pause_lock );
1446 hb_lock( h->state_lock );
1447 memcpy( &h->state, s, sizeof( hb_state_t ) );
1448 if( h->state.state == HB_STATE_WORKING )
1451 if (h->job_count < 1)
1452 h->job_count_permanent = 1;
1454 h->state.param.working.job_cur =
1455 h->job_count_permanent - hb_list_count( h->jobs );
1456 h->state.param.working.job_count = h->job_count_permanent;
1458 // Set which job is being worked on
1460 h->state.param.working.sequence_id = h->current_job->sequence_id;
1462 h->state.param.working.sequence_id = 0;
1464 hb_unlock( h->state_lock );
1465 hb_unlock( h->pause_lock );