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 )
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 )
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;
355 swsflags |= SWS_ACCURATE_RND;
356 #endif /* __x86_64__ */
358 buf1 = av_malloc( avpicture_get_size( PIX_FMT_YUV420P, title->width, title->height ) );
359 buf2 = av_malloc( avpicture_get_size( PIX_FMT_YUV420P, title->width, title->height ) );
360 buf3 = av_malloc( avpicture_get_size( PIX_FMT_YUV420P, job->width, job->height ) );
361 buf4 = av_malloc( avpicture_get_size( PIX_FMT_RGBA32, rgb_width, job->height ) );
362 avpicture_fill( &pic_in, buf1, PIX_FMT_YUV420P,
363 title->width, title->height );
364 avpicture_fill( &pic_deint, buf2, PIX_FMT_YUV420P,
365 title->width, title->height );
366 avpicture_fill( &pic_scale, buf3, PIX_FMT_YUV420P,
367 job->width, job->height );
368 avpicture_fill( &pic_preview, buf4, PIX_FMT_RGBA32,
369 rgb_width, job->height );
371 // Allocate the AVPicture frames and fill in
373 memset( filename, 0, 1024 );
375 hb_get_tempory_filename( h, filename, "%x%d",
376 (intptr_t) title, picture );
378 file = fopen( filename, "r" );
381 hb_log( "hb_get_preview: fopen failed" );
385 fread( buf1, avpicture_get_size( PIX_FMT_YUV420P, title->width, title->height), 1, file );
388 if( job->deinterlace )
390 // Deinterlace and crop
391 avpicture_deinterlace( &pic_deint, &pic_in, PIX_FMT_YUV420P, title->width, title->height );
392 av_picture_crop( &pic_crop, &pic_deint, PIX_FMT_YUV420P, job->crop[0], job->crop[2] );
397 av_picture_crop( &pic_crop, &pic_in, PIX_FMT_YUV420P, job->crop[0], job->crop[2] );
400 // Get scaling context
401 context = sws_getContext(title->width - (job->crop[2] + job->crop[3]),
402 title->height - (job->crop[0] + job->crop[1]),
404 job->width, job->height, PIX_FMT_YUV420P,
405 swsflags, NULL, NULL, NULL);
409 pic_crop.data, pic_crop.linesize,
410 0, title->height - (job->crop[0] + job->crop[1]),
411 pic_scale.data, pic_scale.linesize);
414 sws_freeContext( context );
416 // Get preview context
417 context = sws_getContext(rgb_width, job->height, PIX_FMT_YUV420P,
418 rgb_width, job->height, PIX_FMT_RGBA32,
419 swsflags, NULL, NULL, NULL);
423 pic_scale.data, pic_scale.linesize,
425 pic_preview.data, pic_preview.linesize);
428 sws_freeContext( context );
430 /* Gray background */
431 p32 = (uint32_t *) buffer;
432 for( i = 0; i < ( title->width + 2 ) * ( title->height + 2 ); i++ )
437 /* Draw the picture, centered, and draw the cropping zone */
438 preview_size = pic_preview.linesize[0];
439 pen = buffer + ( title->height - job->height ) *
440 ( title->width + 2 ) * 2 + ( title->width - job->width ) * 2;
441 memset( pen, 0xFF, 4 * ( job->width + 2 ) );
442 pen += 4 * ( title->width + 2 );
443 for( i = 0; i < job->height; i++ )
446 nextLine = pen + 4 * ( title->width + 2 );
447 memset( pen, 0xFF, 4 );
449 memcpy( pen, buf4 + preview_size * i, 4 * job->width );
450 pen += 4 * job->width;
451 memset( pen, 0xFF, 4 );
454 memset( pen, 0xFF, 4 * ( job->width + 2 ) );
457 avpicture_free( &pic_preview );
458 avpicture_free( &pic_scale );
459 avpicture_free( &pic_deint );
460 avpicture_free( &pic_in );
464 * Analyzes a frame to detect interlacing artifacts
465 * and returns true if interlacing (combing) is found.
467 * Code taken from Thomas Oestreich's 32detect filter
468 * in the Transcode project, with minor formatting changes.
470 * @param buf An hb_buffer structure holding valid frame data
471 * @param width The frame's width in pixels
472 * @param height The frame's height in pixels
473 * @param color_equal Sensitivity for detecting similar colors
474 * @param color_diff Sensitivity for detecting different colors
475 * @param threshold Sensitivity for flagging planes as combed
476 * @param prog_equal Sensitivity for detecting similar colors on progressive frames
477 * @param prog_diff Sensitivity for detecting different colors on progressive frames
478 * @param prog_threshold Sensitivity for flagging progressive frames as combed
480 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 )
482 int j, k, n, off, cc_1, cc_2, cc[3], flag[3] ;
483 uint16_t s1, s2, s3, s4;
488 if ( buf->flags & 16 )
490 /* Frame is progressive, be more discerning. */
491 color_diff = prog_diff;
492 color_equal = prog_equal;
493 threshold = prog_threshold;
496 /* One pas for Y, one pass for Cb, one pass for Cr */
497 for( k = 0; k < 3; k++ )
501 /* Y has already been checked, now offset by Y's dimensions
502 and divide all the other values by 2, since Cr and Cb
503 are half-size compared to Y. */
504 offset = width * height;
510 /* Y and Cb are done, so the offset needs to be bumped
511 so it's width*height + (width / 2) * (height / 2) */
515 for( j = 0; j < width; ++j )
519 for( n = 0; n < ( height - 4 ); n = n + 2 )
521 /* Look at groups of 4 sequential horizontal lines */
522 s1 = ( ( buf->data + offset )[ off + j ] & 0xff );
523 s2 = ( ( buf->data + offset )[ off + j + width ] & 0xff );
524 s3 = ( ( buf->data + offset )[ off + j + 2 * width ] & 0xff );
525 s4 = ( ( buf->data + offset )[ off + j + 3 * width ] & 0xff );
527 /* Note if the 1st and 2nd lines are more different in
528 color than the 1st and 3rd lines are similar in color.*/
529 if ( ( abs( s1 - s3 ) < color_equal ) &&
530 ( abs( s1 - s2 ) > color_diff ) )
533 /* Note if the 2nd and 3rd lines are more different in
534 color than the 2nd and 4th lines are similar in color.*/
535 if ( ( abs( s2 - s4 ) < color_equal ) &&
536 ( abs( s2 - s3 ) > color_diff) )
539 /* Now move down 2 horizontal lines before starting over.*/
545 /* The final cc score for a plane is the percentage of combed pixels it contains.
546 Because sensitivity goes down to hundreths of a percent, multiply by 1000
547 so it will be easy to compare against the threhold value which is an integer. */
548 cc[k] = (int)( ( cc_1 + cc_2 ) * 1000.0 / ( width * height ) );
552 /* HandBrake is all yuv420, so weight the average percentage of all 3 planes accordingly.*/
553 int average_cc = ( 2 * cc[0] + ( cc[1] / 2 ) + ( cc[2] / 2 ) ) / 3;
555 /* Now see if that average percentage of combed pixels surpasses the threshold percentage given by the user.*/
556 if( average_cc > threshold )
559 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" );
565 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" );
568 /* Reaching this point means no combing detected. */
574 * Calculates job width and height for anamorphic content,
576 * @param job Handle to hb_job_t
577 * @param output_width Pointer to returned storage width
578 * @param output_height Pointer to returned storage height
579 * @param output_par_width Pointer to returned pixel width
580 @ param output_par_height Pointer to returned pixel height
582 void hb_set_anamorphic_size( hb_job_t * job,
583 int *output_width, int *output_height,
584 int *output_par_width, int *output_par_height )
586 /* Set up some variables to make the math easier to follow. */
587 hb_title_t * title = job->title;
588 int cropped_width = title->width - job->crop[2] - job->crop[3] ;
589 int cropped_height = title->height - job->crop[0] - job->crop[1] ;
590 double storage_aspect = (double)cropped_width / (double)cropped_height;
591 int mod = job->anamorphic.modulus ? job->anamorphic.modulus : 16;
592 double aspect = title->aspect;
594 int pixel_aspect_width = job->anamorphic.par_width;
595 int pixel_aspect_height = job->anamorphic.par_height;
597 /* If a source was really NTSC or PAL and the user specified ITU PAR
598 values, replace the standard PAR values with the ITU broadcast ones. */
599 if( title->width == 720 && job->anamorphic.itu_par )
601 // convert aspect to a scaled integer so we can test for 16:9 & 4:3
602 // aspect ratios ignoring insignificant differences in the LSBs of
603 // the floating point representation.
604 int iaspect = aspect * 9.;
606 /* Handle ITU PARs */
607 if (title->height == 480)
612 /* It's widescreen */
613 pixel_aspect_width = 40;
614 pixel_aspect_height = 33;
616 else if (iaspect == 12)
619 pixel_aspect_width = 10;
620 pixel_aspect_height = 11;
623 else if (title->height == 576)
628 /* It's widescreen */
629 pixel_aspect_width = 16;
630 pixel_aspect_height = 11;
632 else if (iaspect == 12)
635 pixel_aspect_width = 12;
636 pixel_aspect_height = 11;
641 /* Figure out what width the source would display at. */
642 int source_display_width = cropped_width * (double)pixel_aspect_width /
643 (double)pixel_aspect_height ;
646 3 different ways of deciding output dimensions:
647 - 1: Strict anamorphic, preserve source dimensions
648 - 2: Loose anamorphic, round to mod16 and preserve storage aspect ratio
649 - 3: Power user anamorphic, specify everything
652 switch( job->anamorphic.mode )
655 /* Strict anamorphic */
656 *output_width = cropped_width;
657 *output_height = cropped_height;
658 *output_par_width = title->pixel_aspect_width;
659 *output_par_height = title->pixel_aspect_height;
663 /* "Loose" anamorphic.
664 - Uses mod16-compliant dimensions,
665 - Allows users to set the width
668 height; // Gets set later, ignore user job->height value
670 /* Gotta handle bounding dimensions.
671 If the width is too big, just reset it with no rescaling.
672 Instead of using the aspect-scaled job height,
673 we need to see if the job width divided by the storage aspect
674 is bigger than the max. If so, set it to the max (this is sloppy).
675 If not, set job height to job width divided by storage aspect.
678 if ( job->maxWidth && (job->maxWidth < job->width) )
679 width = job->maxWidth;
680 height = ((double)width / storage_aspect) + 0.5;
682 if ( job->maxHeight && (job->maxHeight < height) )
683 height = job->maxHeight;
685 /* Time to get picture dimensions that divide cleanly.*/
686 width = MULTIPLE_MOD( width, mod);
687 height = MULTIPLE_MOD( height, mod);
689 /* Verify these new dimensions don't violate max height and width settings */
690 if ( job->maxWidth && (job->maxWidth < job->width) )
691 width = job->maxWidth;
692 if ( job->maxHeight && (job->maxHeight < height) )
693 height = job->maxHeight;
695 /* The film AR is the source's display width / cropped source height.
696 The output display width is the output height * film AR.
697 The output PAR is the output display width / output storage width. */
698 pixel_aspect_width = height * source_display_width / cropped_height;
699 pixel_aspect_height = width;
701 /* Pass the results back to the caller */
702 *output_width = width;
703 *output_height = height;
707 /* Anamorphic 3: Power User Jamboree
708 - Set everything based on specified values */
710 /* Use specified storage dimensions */
712 height = job->height;
714 /* Bind to max dimensions */
715 if( job->maxWidth && width > job->maxWidth )
716 width = job->maxWidth;
717 if( job->maxHeight && height > job->maxHeight )
718 height = job->maxHeight;
720 /* Time to get picture dimensions that divide cleanly.*/
721 width = MULTIPLE_MOD( width, mod);
722 height = MULTIPLE_MOD( height, mod);
724 /* Verify we're still within max dimensions */
725 if( job->maxWidth && width > job->maxWidth )
726 width = job->maxWidth - (mod/2);
727 if( job->maxHeight && height > job->maxHeight )
728 height = job->maxHeight - (mod/2);
730 /* Re-ensure we have picture dimensions that divide cleanly. */
731 width = MULTIPLE_MOD( width, mod );
732 height = MULTIPLE_MOD( height, mod );
734 /* That finishes the storage dimensions. On to display. */
735 if( job->anamorphic.dar_width && job->anamorphic.dar_height )
737 /* We need to adjust the PAR to produce this aspect. */
738 pixel_aspect_width = height * job->anamorphic.dar_width / job->anamorphic.dar_height;
739 pixel_aspect_height = width;
743 /* We first need the display ar.
744 That's the source display width divided by the source height after cropping.
745 Then we multiple the output height by that to get the pixel aspect width,
746 and the pixel aspect height is the storage width.*/
747 pixel_aspect_width = height * source_display_width / cropped_height;
748 pixel_aspect_height = width;
752 *output_width = width;
753 *output_height = height;
757 /* While x264 is smart enough to reduce fractions on its own, libavcodec
758 needs some help with the math, so lose superfluous factors. */
759 hb_reduce( output_par_width, output_par_height,
760 pixel_aspect_width, pixel_aspect_height );
764 * Calculates job width, height, and cropping parameters.
765 * @param job Handle to hb_job_t.
766 * @param aspect Desired aspect ratio. Value of -1 uses title aspect.
767 * @param pixels Maximum desired pixel count.
769 void hb_set_size( hb_job_t * job, double aspect, int pixels )
771 hb_title_t * title = job->title;
773 int croppedWidth = title->width - title->crop[2] - title->crop[3];
774 int croppedHeight = title->height - title->crop[0] - title->crop[1];
775 double croppedAspect = title->aspect * title->height * croppedWidth /
776 croppedHeight / title->width;
782 /* Keep the best possible aspect ratio */
783 aspect = croppedAspect;
786 /* Crop if necessary to obtain the desired ratio */
787 memcpy( job->crop, title->crop, 4 * sizeof( int ) );
788 if( aspect < croppedAspect )
790 /* Need to crop on the left and right */
791 addCrop = croppedWidth - aspect * croppedHeight * title->width /
792 title->aspect / title->height;
795 addCrop = ( addCrop + 1 ) / 2;
796 job->crop[2] += addCrop;
797 job->crop[3] += addCrop;
799 else if( addCrop & 2 )
802 job->crop[2] += addCrop - 1;
803 job->crop[3] += addCrop + 1;
808 job->crop[2] += addCrop;
809 job->crop[3] += addCrop;
812 else if( aspect > croppedAspect )
814 /* Need to crop on the top and bottom */
815 addCrop = croppedHeight - croppedWidth * title->aspect *
816 title->height / aspect / title->width;
819 addCrop = ( addCrop + 1 ) / 2;
820 job->crop[0] += addCrop;
821 job->crop[1] += addCrop;
823 else if( addCrop & 2 )
826 job->crop[0] += addCrop - 1;
827 job->crop[1] += addCrop + 1;
832 job->crop[0] += addCrop;
833 job->crop[1] += addCrop;
837 /* Compute a resolution from the number of pixels and aspect */
841 h = MULTIPLE_16( (int)( (double)w / aspect ) );
849 job->height = MULTIPLE_16( (int)( (double)job->width / aspect ) );
853 * Returns the number of jobs in the queue.
854 * @param h Handle to hb_handle_t.
855 * @return Number of jobs.
857 int hb_count( hb_handle_t * h )
859 return hb_list_count( h->jobs );
863 * Returns handle to job at index i within the job list.
864 * @param h Handle to hb_handle_t.
865 * @param i Index of job.
866 * @returns Handle to hb_job_t of desired job.
868 hb_job_t * hb_job( hb_handle_t * h, int i )
870 return hb_list_item( h->jobs, i );
873 hb_job_t * hb_current_job( hb_handle_t * h )
875 return( h->current_job );
879 * Adds a job to the job list.
880 * @param h Handle to hb_handle_t.
881 * @param job Handle to hb_job_t.
883 void hb_add( hb_handle_t * h, hb_job_t * job )
886 hb_title_t * title, * title_copy;
887 hb_chapter_t * chapter, * chapter_copy;
889 hb_subtitle_t * subtitle, * subtitle_copy;
895 title_copy = malloc( sizeof( hb_title_t ) );
896 memcpy( title_copy, title, sizeof( hb_title_t ) );
898 title_copy->list_chapter = hb_list_init();
899 for( i = 0; i < hb_list_count( title->list_chapter ); i++ )
901 chapter = hb_list_item( title->list_chapter, i );
902 chapter_copy = malloc( sizeof( hb_chapter_t ) );
903 memcpy( chapter_copy, chapter, sizeof( hb_chapter_t ) );
904 hb_list_add( title_copy->list_chapter, chapter_copy );
910 if( title->metadata )
912 title_copy->metadata = malloc( sizeof( hb_metadata_t ) );
914 if( title_copy->metadata )
916 memcpy( title_copy->metadata, title->metadata, sizeof( hb_metadata_t ) );
919 * Need to copy the artwork seperatly (TODO).
921 if( title->metadata->coverart )
923 title_copy->metadata->coverart = malloc( title->metadata->coverart_size );
924 if( title_copy->metadata->coverart )
926 memcpy( title_copy->metadata->coverart, title->metadata->coverart,
927 title->metadata->coverart_size );
929 title_copy->metadata->coverart_size = 0;
935 /* Copy the audio track(s) we want */
936 title_copy->list_audio = hb_list_init();
938 for( i = 0; i < hb_list_count(job->list_audio); i++ )
940 if( ( audio = hb_list_item( job->list_audio, i ) ) )
942 hb_list_add( title_copy->list_audio, hb_audio_copy(audio) );
946 title_copy->list_subtitle = hb_list_init();
949 * The following code is confusing, there are three ways in which
950 * we select subtitles and it depends on whether this is single or
953 * subtitle_scan may be enabled, in which case the first pass
954 * scans all subtitles of that language. The second pass does not
955 * select any because they are set at the end of the first pass.
957 * native_language may have a preferred language, in which case we
958 * may be switching the language we want for the subtitles in the
959 * first pass of a single pass, or the second pass of a two pass.
961 * We may have manually selected a subtitle, in which case that is
962 * selected in the first pass of a single pass, or the second of a
965 memset( audio_lang, 0, sizeof( audio_lang ) );
967 if ( job->indepth_scan || job->native_language ) {
970 * Find the first audio language that is being encoded
972 for( i = 0; i < hb_list_count(job->list_audio); i++ )
974 if( ( audio = hb_list_item( job->list_audio, i ) ) )
976 strncpy(audio_lang, audio->config.lang.iso639_2, sizeof(audio_lang));
982 * In all cases switch the language if we need to to our native
985 if( job->native_language )
987 if( strncasecmp( job->native_language, audio_lang,
988 sizeof( audio_lang ) ) != 0 )
993 hb_log( "Enabled subtitles in native language '%s', audio is in '%s'",
994 job->native_language, audio_lang);
997 * The main audio track is not in our native language, so switch
998 * the subtitles to use our native language instead.
1000 strncpy( audio_lang, job->native_language, sizeof( audio_lang ) );
1003 * native language is irrelevent, free it.
1005 free( job->native_language );
1006 job->native_language = NULL;
1012 * If doing a subtitle scan then add all the matching subtitles for this
1015 if ( job->indepth_scan )
1017 for( i=0; i < hb_list_count( title->list_subtitle ); i++ )
1019 subtitle = hb_list_item( title->list_subtitle, i );
1020 if( strcmp( subtitle->iso639_2, audio_lang ) == 0 )
1023 * Matched subtitle language with audio language, so
1024 * add this to our list to scan.
1026 * We will update the subtitle list on the second pass
1027 * later after the first pass has completed.
1029 subtitle_copy = malloc( sizeof( hb_subtitle_t ) );
1030 memcpy( subtitle_copy, subtitle, sizeof( hb_subtitle_t ) );
1031 hb_list_add( title_copy->list_subtitle, subtitle_copy );
1032 if ( job->native_language ) {
1034 * With native language just select the
1035 * first match in our langiage, not all of
1036 * them. Subsequent ones are likely to be commentary
1044 * Not doing a subtitle scan in this pass, but maybe we are in the
1047 if( job->select_subtitle )
1050 * Don't add subtitles here, we'll add them via select_subtitle
1051 * at the end of the subtitle_scan.
1055 * Definitely not doing a subtitle scan.
1057 if( job->pass != 1 && job->native_language )
1060 * We are not doing a subtitle scan but do want the
1061 * native langauge subtitle selected, so select it
1062 * for pass 0 or pass 2 of a two pass.
1064 for( i=0; i < hb_list_count( title->list_subtitle ); i++ )
1066 subtitle = hb_list_item( title->list_subtitle, i );
1067 if( strcmp( subtitle->iso639_2, audio_lang ) == 0 )
1070 * Matched subtitle language with audio language, so
1071 * add this to our list to scan.
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 );
1081 * Manually selected subtitle, in which case only
1082 * bother adding them for pass 0 or pass 2 of a two
1085 if( job->pass != 1 )
1087 if( ( subtitle = hb_list_item( title->list_subtitle, job->subtitle ) ) )
1089 subtitle_copy = malloc( sizeof( hb_subtitle_t ) );
1090 memcpy( subtitle_copy, subtitle, sizeof( hb_subtitle_t ) );
1091 hb_list_add( title_copy->list_subtitle, subtitle_copy );
1099 job_copy = calloc( sizeof( hb_job_t ), 1 );
1100 memcpy( job_copy, job, sizeof( hb_job_t ) );
1101 title_copy->job = job_copy;
1102 job_copy->title = title_copy;
1103 job_copy->list_audio = title_copy->list_audio;
1104 job_copy->file = strdup( job->file );
1106 job_copy->pause = h->pause_lock;
1108 /* Copy the job filter list */
1112 int filter_count = hb_list_count( job->filters );
1113 job_copy->filters = hb_list_init();
1114 for( i = 0; i < filter_count; i++ )
1117 * Copy the filters, since the MacGui reuses the global filter objects
1118 * meaning that queued up jobs overwrite the previous filter settings.
1119 * In reality, settings is probably the only field that needs duplicating
1120 * since it's the only value that is ever changed. But name is duplicated
1121 * as well for completeness. Not copying private_data since it gets
1122 * created for each job in renderInit.
1124 hb_filter_object_t * filter = hb_list_item( job->filters, i );
1125 hb_filter_object_t * filter_copy = malloc( sizeof( hb_filter_object_t ) );
1126 memcpy( filter_copy, filter, sizeof( hb_filter_object_t ) );
1128 filter_copy->name = strdup( filter->name );
1129 if( filter->settings )
1130 filter_copy->settings = strdup( filter->settings );
1131 hb_list_add( job_copy->filters, filter_copy );
1135 /* Add the job to the list */
1136 hb_list_add( h->jobs, job_copy );
1137 h->job_count = hb_count(h);
1138 h->job_count_permanent++;
1142 * Removes a job from the job list.
1143 * @param h Handle to hb_handle_t.
1144 * @param job Handle to hb_job_t.
1146 void hb_rem( hb_handle_t * h, hb_job_t * job )
1148 hb_list_rem( h->jobs, job );
1150 h->job_count = hb_count(h);
1151 if (h->job_count_permanent)
1152 h->job_count_permanent--;
1154 /* XXX free everything XXX */
1158 * Starts the conversion process.
1159 * Sets state to HB_STATE_WORKING.
1160 * calls hb_work_init, to launch work thread. Stores handle to work thread.
1161 * @param h Handle to hb_handle_t.
1163 void hb_start( hb_handle_t * h )
1166 h->job_count = hb_list_count( h->jobs );
1167 h->job_count_permanent = h->job_count;
1169 hb_lock( h->state_lock );
1170 h->state.state = HB_STATE_WORKING;
1171 #define p h->state.param.working
1174 p.job_count = h->job_count;
1182 hb_unlock( h->state_lock );
1187 h->work_thread = hb_work_init( h->jobs, h->cpu_count,
1188 &h->work_die, &h->work_error, &h->current_job );
1192 * Pauses the conversion process.
1193 * @param h Handle to hb_handle_t.
1195 void hb_pause( hb_handle_t * h )
1199 hb_lock( h->pause_lock );
1202 hb_lock( h->state_lock );
1203 h->state.state = HB_STATE_PAUSED;
1204 hb_unlock( h->state_lock );
1209 * Resumes the conversion process.
1210 * @param h Handle to hb_handle_t.
1212 void hb_resume( hb_handle_t * h )
1216 hb_unlock( h->pause_lock );
1222 * Stops the conversion process.
1223 * @param h Handle to hb_handle_t.
1225 void hb_stop( hb_handle_t * h )
1229 h->job_count = hb_count(h);
1230 h->job_count_permanent = 0;
1236 * Returns the state of the conversion process.
1237 * @param h Handle to hb_handle_t.
1238 * @param s Handle to hb_state_t which to copy the state data.
1240 void hb_get_state( hb_handle_t * h, hb_state_t * s )
1242 hb_lock( h->state_lock );
1244 memcpy( s, &h->state, sizeof( hb_state_t ) );
1245 if ( h->state.state == HB_STATE_SCANDONE || h->state.state == HB_STATE_WORKDONE )
1246 h->state.state = HB_STATE_IDLE;
1248 hb_unlock( h->state_lock );
1251 void hb_get_state2( hb_handle_t * h, hb_state_t * s )
1253 hb_lock( h->state_lock );
1255 memcpy( s, &h->state, sizeof( hb_state_t ) );
1257 hb_unlock( h->state_lock );
1261 * Called in MacGui in UpdateUI to check
1262 * for a new scan being completed to set a new source
1264 int hb_get_scancount( hb_handle_t * h)
1266 return h->scanCount;
1270 * Closes access to libhb by freeing the hb_handle_t handle ontained in hb_init_real.
1271 * @param _h Pointer to handle to hb_handle_t.
1273 void hb_close( hb_handle_t ** _h )
1275 hb_handle_t * h = *_h;
1279 hb_thread_close( &h->main_thread );
1281 while( ( title = hb_list_item( h->list_title, 0 ) ) )
1283 hb_list_rem( h->list_title, title );
1284 if( title->job && title->job->filters )
1286 hb_list_close( &title->job->filters );
1289 hb_title_close( &title );
1291 hb_list_close( &h->list_title );
1293 hb_list_close( &h->jobs );
1294 hb_lock_close( &h->state_lock );
1295 hb_lock_close( &h->pause_lock );
1302 * Monitors the state of the update, scan, and work threads.
1303 * Sets scan done state when scan thread exits.
1304 * Sets work done state when work thread exits.
1305 * @param _h Handle to hb_handle_t
1307 static void thread_func( void * _h )
1309 hb_handle_t * h = (hb_handle_t *) _h;
1312 struct dirent * entry;
1316 /* Create folder for temporary files */
1317 memset( dirname, 0, 1024 );
1318 hb_get_tempory_directory( h, dirname );
1320 hb_mkdir( dirname );
1324 /* In case the check_update thread hangs, it'll die sooner or
1325 later. Then, we join it here */
1326 if( h->update_thread &&
1327 hb_thread_has_exited( h->update_thread ) )
1329 hb_thread_close( &h->update_thread );
1332 /* Check if the scan thread is done */
1333 if( h->scan_thread &&
1334 hb_thread_has_exited( h->scan_thread ) )
1336 hb_thread_close( &h->scan_thread );
1338 hb_log( "libhb: scan thread found %d valid title(s)",
1339 hb_list_count( h->list_title ) );
1340 hb_lock( h->state_lock );
1341 h->state.state = HB_STATE_SCANDONE; //originally state.state
1342 hb_unlock( h->state_lock );
1343 /*we increment this sessions scan count by one for the MacGui
1344 to trigger a new source being set */
1348 /* Check if the work thread is done */
1349 if( h->work_thread &&
1350 hb_thread_has_exited( h->work_thread ) )
1352 hb_thread_close( &h->work_thread );
1354 hb_log( "libhb: work result = %d",
1356 hb_lock( h->state_lock );
1357 h->state.state = HB_STATE_WORKDONE;
1358 h->state.param.workdone.error = h->work_error;
1360 h->job_count = hb_count(h);
1361 if (h->job_count < 1)
1362 h->job_count_permanent = 0;
1363 hb_unlock( h->state_lock );
1369 if( h->work_thread )
1372 hb_thread_close( &h->work_thread );
1375 /* Remove temp folder */
1376 dir = opendir( dirname );
1379 while( ( entry = readdir( dir ) ) )
1381 char filename[1024];
1382 if( entry->d_name[0] == '.' )
1386 memset( filename, 0, 1024 );
1387 snprintf( filename, 1023, "%s/%s", dirname, entry->d_name );
1397 * @param h Handle to hb_handle_t
1399 int hb_get_pid( hb_handle_t * h )
1405 * Sets the current state.
1406 * @param h Handle to hb_handle_t
1407 * @param s Handle to new hb_state_t
1409 void hb_set_state( hb_handle_t * h, hb_state_t * s )
1411 hb_lock( h->pause_lock );
1412 hb_lock( h->state_lock );
1413 memcpy( &h->state, s, sizeof( hb_state_t ) );
1414 if( h->state.state == HB_STATE_WORKING )
1417 if (h->job_count < 1)
1418 h->job_count_permanent = 1;
1420 h->state.param.working.job_cur =
1421 h->job_count_permanent - hb_list_count( h->jobs );
1422 h->state.param.working.job_count = h->job_count_permanent;
1424 // Set which job is being worked on
1426 h->state.param.working.sequence_id = h->current_job->sequence_id;
1428 h->state.param.working.sequence_id = 0;
1430 hb_unlock( h->state_lock );
1431 hb_unlock( h->pause_lock );