libhb/hb.c

   1 #include "hb.h"
   2 #include "hbffmpeg.h"
   3
   4 struct hb_handle_s
   5 {
   6     /* The "Check for update" thread */
   7     int            build;
   8     char           version[16];
   9     hb_thread_t  * update_thread;
  10
  11     /* This thread's only purpose is to check other threads'
  12        states */
  13     volatile int   die;
  14     hb_thread_t  * main_thread;
  15     int            pid;
  16
  17     /* DVD/file scan thread */
  18     hb_list_t    * list_title;
  19     hb_thread_t  * scan_thread;
  20
  21     /* The thread which processes the jobs. Others threads are launched
  22        from this one (see work.c) */
  23     hb_list_t    * jobs;
  24     hb_job_t     * current_job;
  25     int            job_count;
  26     int            job_count_permanent;
  27     volatile int   work_die;
  28     int            work_error;
  29     hb_thread_t  * work_thread;
  30
  31     int            cpu_count;
  32
  33     hb_lock_t    * state_lock;
  34     hb_state_t     state;
  35
  36     int            paused;
  37     hb_lock_t    * pause_lock;
  38     /* For MacGui active queue
  39        increments each time the scan thread completes*/
  40     int            scanCount;
  41
  42     /* Stash of persistent data between jobs, for stuff
  43        like correcting frame count and framerate estimates
  44        on multi-pass encodes where frames get dropped.     */
  45     hb_interjob_t * interjob;
  46
  47 };
  48
  49 hb_lock_t *hb_avcodec_lock;
  50 hb_work_object_t * hb_objects = NULL;
  51
  52 static void thread_func( void * );
  53
  54 void hb_avcodec_init()
  55 {
  56     hb_avcodec_lock  = hb_lock_init();
  57     av_register_all();
  58 }
  59
  60 int hb_avcodec_open(AVCodecContext *avctx, AVCodec *codec)
  61 {
  62     int ret;
  63     hb_lock( hb_avcodec_lock );
  64     ret = avcodec_open(avctx, codec);
  65     hb_unlock( hb_avcodec_lock );
  66     return ret;
  67 }
  68
  69 int hb_avcodec_close(AVCodecContext *avctx)
  70 {
  71     int ret;
  72     hb_lock( hb_avcodec_lock );
  73     ret = avcodec_close(avctx);
  74     hb_unlock( hb_avcodec_lock );
  75     return ret;
  76 }
  77
  78 /**
  79  * Registers work objects, by adding the work object to a liked list.
  80  * @param w Handle to hb_work_object_t to register.
  81  */
  82 void hb_register( hb_work_object_t * w )
  83 {
  84     w->next    = hb_objects;
  85     hb_objects = w;
  86 }
  87
  88 /**
  89  * libhb initialization routine.
  90  * @param verbose HB_DEBUG_NONE or HB_DEBUG_ALL.
  91  * @param update_check signals libhb to check for updated version from HandBrake website.
  92  * @return Handle to hb_handle_t for use on all subsequent calls to libhb.
  93  */
  94 hb_handle_t * hb_init( int verbose, int update_check )
  95 {
  96     hb_handle_t * h = calloc( sizeof( hb_handle_t ), 1 );
  97     uint64_t      date;
  98
  99     /* See hb_deep_log() and hb_log() in common.c */
 100     global_verbosity_level = verbose;
 101     if( verbose )
 102         putenv( "HB_DEBUG=1" );
 103
 104     /* Check for an update on the website if asked to */
 105     h->build = -1;
 106
 107     if( update_check )
 108     {
 109         hb_log( "hb_init: checking for updates" );
 110         date             = hb_get_date();
 111         h->update_thread = hb_update_init( &h->build, h->version );
 112
 113         for( ;; )
 114         {
 115             if( hb_thread_has_exited( h->update_thread ) )
 116             {
 117                 /* Immediate success or failure */
 118                 hb_thread_close( &h->update_thread );
 119                 break;
 120             }
 121             if( hb_get_date() > date + 1000 )
 122             {
 123                 /* Still nothing after one second. Connection problem,
 124                    let the thread die */
 125                 hb_log( "hb_init: connection problem, not waiting for "
 126                         "update_thread" );
 127                 break;
 128             }
 129             hb_snooze( 500 );
 130         }
 131     }
 132
 133     /*
 134      * Initialise buffer pool
 135      */
 136     hb_buffer_pool_init();
 137
 138     /* CPU count detection */
 139     hb_log( "hb_init: checking cpu count" );
 140     h->cpu_count = hb_get_cpu_count();
 141
 142     h->list_title = hb_list_init();
 143     h->jobs       = hb_list_init();
 144
 145     h->state_lock  = hb_lock_init();
 146     h->state.state = HB_STATE_IDLE;
 147
 148     h->pause_lock = hb_lock_init();
 149
 150     h->interjob = calloc( sizeof( hb_interjob_t ), 1 );
 151
 152     /* libavcodec */
 153     hb_avcodec_init();
 154
 155     /* Start library thread */
 156     hb_log( "hb_init: starting libhb thread" );
 157     h->die         = 0;
 158     h->main_thread = hb_thread_init( "libhb", thread_func, h,
 159                                      HB_NORMAL_PRIORITY );
 160     hb_register( &hb_sync );
 161         hb_register( &hb_decmpeg2 );
 162         hb_register( &hb_decvobsub );
 163     hb_register( &hb_encvobsub );
 164     hb_register( &hb_deccc608 );
 165         hb_register( &hb_render );
 166         hb_register( &hb_encavcodec );
 167         hb_register( &hb_encx264 );
 168     hb_register( &hb_enctheora );
 169         hb_register( &hb_deca52 );
 170         hb_register( &hb_decdca );
 171         hb_register( &hb_decavcodec );
 172         hb_register( &hb_decavcodecv );
 173         hb_register( &hb_decavcodecvi );
 174         hb_register( &hb_decavcodecai );
 175         hb_register( &hb_declpcm );
 176         hb_register( &hb_encfaac );
 177         hb_register( &hb_enclame );
 178         hb_register( &hb_encvorbis );
 179 #ifdef __APPLE__
 180         hb_register( &hb_encca_aac );
 181 #endif
 182
 183     return h;
 184 }
 185
 186 /**
 187  * libhb initialization routine.
 188  * This version is to use when calling the dylib, the macro hb_init isn't available from a dylib call!
 189  * @param verbose HB_DEBUG_NONE or HB_DEBUG_ALL.
 190  * @param update_check signals libhb to check for updated version from HandBrake website.
 191  * @return Handle to hb_handle_t for use on all subsequent calls to libhb.
 192  */
 193 hb_handle_t * hb_init_dl( int verbose, int update_check )
 194 {
 195     hb_handle_t * h = calloc( sizeof( hb_handle_t ), 1 );
 196     uint64_t      date;
 197
 198     /* See hb_log() in common.c */
 199     if( verbose > HB_DEBUG_NONE )
 200     {
 201         putenv( "HB_DEBUG=1" );
 202     }
 203
 204     /* Check for an update on the website if asked to */
 205     h->build = -1;
 206
 207     if( update_check )
 208     {
 209         hb_log( "hb_init: checking for updates" );
 210         date             = hb_get_date();
 211         h->update_thread = hb_update_init( &h->build, h->version );
 212
 213         for( ;; )
 214         {
 215             if( hb_thread_has_exited( h->update_thread ) )
 216             {
 217                 /* Immediate success or failure */
 218                 hb_thread_close( &h->update_thread );
 219                 break;
 220             }
 221             if( hb_get_date() > date + 1000 )
 222             {
 223                 /* Still nothing after one second. Connection problem,
 224                    let the thread die */
 225                 hb_log( "hb_init: connection problem, not waiting for "
 226                         "update_thread" );
 227                 break;
 228             }
 229             hb_snooze( 500 );
 230         }
 231     }
 232
 233     /* CPU count detection */
 234     hb_log( "hb_init: checking cpu count" );
 235     h->cpu_count = hb_get_cpu_count();
 236
 237     h->list_title = hb_list_init();
 238     h->jobs       = hb_list_init();
 239     h->current_job = NULL;
 240
 241     h->state_lock  = hb_lock_init();
 242     h->state.state = HB_STATE_IDLE;
 243
 244     h->pause_lock = hb_lock_init();
 245
 246     /* libavcodec */
 247     avcodec_init();
 248     avcodec_register_all();
 249
 250     /* Start library thread */
 251     hb_log( "hb_init: starting libhb thread" );
 252     h->die         = 0;
 253     h->main_thread = hb_thread_init( "libhb", thread_func, h,
 254                                      HB_NORMAL_PRIORITY );
 255
 256     hb_register( &hb_sync );
 257         hb_register( &hb_decmpeg2 );
 258         hb_register( &hb_decvobsub );
 259     hb_register( &hb_encvobsub );
 260     hb_register( &hb_deccc608 );
 261         hb_register( &hb_render );
 262         hb_register( &hb_encavcodec );
 263         hb_register( &hb_encx264 );
 264     hb_register( &hb_enctheora );
 265         hb_register( &hb_deca52 );
 266         hb_register( &hb_decdca );
 267         hb_register( &hb_decavcodec );
 268         hb_register( &hb_decavcodecv );
 269         hb_register( &hb_decavcodecvi );
 270         hb_register( &hb_decavcodecai );
 271         hb_register( &hb_declpcm );
 272         hb_register( &hb_encfaac );
 273         hb_register( &hb_enclame );
 274         hb_register( &hb_encvorbis );
 275 #ifdef __APPLE__
 276         hb_register( &hb_encca_aac );
 277 #endif
 278
 279         return h;
 280 }
 281
 282
 283 /**
 284  * Returns current version of libhb.
 285  * @param h Handle to hb_handle_t.
 286  * @return character array of version number.
 287  */
 288 char * hb_get_version( hb_handle_t * h )
 289 {
 290     return HB_PROJECT_VERSION;
 291 }
 292
 293 /**
 294  * Returns current build of libhb.
 295  * @param h Handle to hb_handle_t.
 296  * @return character array of build number.
 297  */
 298 int hb_get_build( hb_handle_t * h )
 299 {
 300     return HB_PROJECT_BUILD;
 301 }
 302
 303 /**
 304  * Checks for needed update.
 305  * @param h Handle to hb_handle_t.
 306  * @param version Pointer to handle where version will be copied.
 307  * @return update indicator.
 308  */
 309 int hb_check_update( hb_handle_t * h, char ** version )
 310 {
 311     *version = ( h->build < 0 ) ? NULL : h->version;
 312     return h->build;
 313 }
 314
 315 /**
 316  * Sets the cpu count to the desired value.
 317  * @param h Handle to hb_handle_t
 318  * @param cpu_count Number of CPUs to use.
 319  */
 320 void hb_set_cpu_count( hb_handle_t * h, int cpu_count )
 321 {
 322     cpu_count    = MAX( 1, cpu_count );
 323     cpu_count    = MIN( cpu_count, 8 );
 324     h->cpu_count = cpu_count;
 325 }
 326
 327 /**
 328  * Deletes current previews associated with titles
 329  * @param h Handle to hb_handle_t
 330  */
 331 void hb_remove_previews( hb_handle_t * h )
 332 {
 333     char            filename[1024];
 334     char            dirname[1024];
 335     hb_title_t    * title;
 336     int             i, count, len;
 337     DIR           * dir;
 338     struct dirent * entry;
 339
 340     memset( dirname, 0, 1024 );
 341     hb_get_tempory_directory( h, dirname );
 342     dir = opendir( dirname );
 343     if (dir == NULL) return;
 344
 345     count = hb_list_count( h->list_title );
 346     while( ( entry = readdir( dir ) ) )
 347     {
 348         if( entry->d_name[0] == '.' )
 349         {
 350             continue;
 351         }
 352         for( i = 0; i < count; i++ )
 353         {
 354             title = hb_list_item( h->list_title, i );
 355             len = snprintf( filename, 1024, "%" PRIxPTR, (intptr_t) title );
 356             if (strncmp(entry->d_name, filename, len) == 0)
 357             {
 358                 snprintf( filename, 1024, "%s/%s", dirname, entry->d_name );
 359                 unlink( filename );
 360                 break;
 361             }
 362         }
 363     }
 364     closedir( dir );
 365 }
 366
 367 /**
 368  * Initializes a scan of the by calling hb_scan_init
 369  * @param h Handle to hb_handle_t
 370  * @param path location of VIDEO_TS folder.
 371  * @param title_index Desired title to scan.  0 for all titles.
 372  * @param preview_count Number of preview images to generate.
 373  * @param store_previews Whether or not to write previews to disk.
 374  */
 375 void hb_scan( hb_handle_t * h, const char * path, int title_index,
 376               int preview_count, int store_previews )
 377 {
 378     hb_title_t * title;
 379
 380     /* Clean up from previous scan */
 381     hb_remove_previews( h );
 382     while( ( title = hb_list_item( h->list_title, 0 ) ) )
 383     {
 384         hb_list_rem( h->list_title, title );
 385         hb_title_close( &title );
 386     }
 387
 388     hb_log( "hb_scan: path=%s, title_index=%d", path, title_index );
 389     h->scan_thread = hb_scan_init( h, path, title_index, h->list_title,
 390                                    preview_count, store_previews );
 391 }
 392
 393 /**
 394  * Returns the list of titles found.
 395  * @param h Handle to hb_handle_t
 396  * @return Handle to hb_list_t of the title list.
 397  */
 398 hb_list_t * hb_get_titles( hb_handle_t * h )
 399 {
 400     return h->list_title;
 401 }
 402
 403 /**
 404  * Create preview image of desired title a index of picture.
 405  * @param h Handle to hb_handle_t.
 406  * @param title Handle to hb_title_t of desired title.
 407  * @param picture Index in title.
 408  * @param buffer Handle to buufer were inage will be drawn.
 409  */
 410 void hb_get_preview( hb_handle_t * h, hb_title_t * title, int picture,
 411                      uint8_t * buffer )
 412 {
 413     hb_job_t           * job = title->job;
 414     char                 filename[1024];
 415     FILE               * file;
 416     uint8_t            * buf1, * buf2, * buf3, * buf4, * pen;
 417     uint32_t             swsflags;
 418     AVPicture            pic_in, pic_preview, pic_deint, pic_crop, pic_scale;
 419     struct SwsContext  * context;
 420     int                  i;
 421     int                  rgb_width = ((job->width + 7) >> 3) << 3;
 422     int                  preview_size;
 423
 424     swsflags = SWS_LANCZOS | SWS_ACCURATE_RND;
 425
 426     buf1 = av_malloc( avpicture_get_size( PIX_FMT_YUV420P, title->width, title->height ) );
 427     buf2 = av_malloc( avpicture_get_size( PIX_FMT_YUV420P, title->width, title->height ) );
 428     buf3 = av_malloc( avpicture_get_size( PIX_FMT_YUV420P, rgb_width, job->height ) );
 429     buf4 = av_malloc( avpicture_get_size( PIX_FMT_RGB32, rgb_width, job->height ) );
 430     avpicture_fill( &pic_in, buf1, PIX_FMT_YUV420P,
 431                     title->width, title->height );
 432     avpicture_fill( &pic_deint, buf2, PIX_FMT_YUV420P,
 433                     title->width, title->height );
 434     avpicture_fill( &pic_scale, buf3, PIX_FMT_YUV420P,
 435                     rgb_width, job->height );
 436     avpicture_fill( &pic_preview, buf4, PIX_FMT_RGB32,
 437                     rgb_width, job->height );
 438
 439     // Allocate the AVPicture frames and fill in
 440
 441     memset( filename, 0, 1024 );
 442
 443     hb_get_tempory_filename( h, filename, "%" PRIxPTR "%d",
 444                              (intptr_t) title, picture );
 445
 446     file = fopen( filename, "r" );
 447     if( !file )
 448     {
 449         hb_log( "hb_get_preview: fopen failed" );
 450         return;
 451     }
 452
 453     fread( buf1, avpicture_get_size( PIX_FMT_YUV420P, title->width, title->height), 1, file );
 454     fclose( file );
 455
 456     if( job->deinterlace )
 457     {
 458         // Deinterlace and crop
 459         avpicture_deinterlace( &pic_deint, &pic_in, PIX_FMT_YUV420P, title->width, title->height );
 460         av_picture_crop( &pic_crop, &pic_deint, PIX_FMT_YUV420P, job->crop[0], job->crop[2] );
 461     }
 462     else
 463     {
 464         // Crop
 465         av_picture_crop( &pic_crop, &pic_in, PIX_FMT_YUV420P, job->crop[0], job->crop[2] );
 466     }
 467
 468     // Get scaling context
 469     context = sws_getContext(title->width  - (job->crop[2] + job->crop[3]),
 470                              title->height - (job->crop[0] + job->crop[1]),
 471                              PIX_FMT_YUV420P,
 472                              rgb_width, job->height, PIX_FMT_YUV420P,
 473                              swsflags, NULL, NULL, NULL);
 474
 475     // Scale
 476     sws_scale(context,
 477               pic_crop.data, pic_crop.linesize,
 478               0, title->height - (job->crop[0] + job->crop[1]),
 479               pic_scale.data, pic_scale.linesize);
 480
 481     // Free context
 482     sws_freeContext( context );
 483
 484     // Get preview context
 485     context = sws_getContext(rgb_width, job->height, PIX_FMT_YUV420P,
 486                               rgb_width, job->height, PIX_FMT_RGB32,
 487                               swsflags, NULL, NULL, NULL);
 488
 489     // Create preview
 490     sws_scale(context,
 491               pic_scale.data, pic_scale.linesize,
 492               0, job->height,
 493               pic_preview.data, pic_preview.linesize);
 494
 495     // Free context
 496     sws_freeContext( context );
 497
 498     preview_size = pic_preview.linesize[0];
 499     pen = buffer;
 500     for( i = 0; i < job->height; i++ )
 501     {
 502         memcpy( pen, buf4 + preview_size * i, 4 * job->width );
 503         pen += 4 * job->width;
 504     }
 505
 506     // Clean up
 507     avpicture_free( &pic_preview );
 508     avpicture_free( &pic_scale );
 509     avpicture_free( &pic_deint );
 510     avpicture_free( &pic_in );
 511 }
 512
 513  /**
 514  * Analyzes a frame to detect interlacing artifacts
 515  * and returns true if interlacing (combing) is found.
 516  *
 517  * Code taken from Thomas Oestreich's 32detect filter
 518  * in the Transcode project, with minor formatting changes.
 519  *
 520  * @param buf         An hb_buffer structure holding valid frame data
 521  * @param width       The frame's width in pixels
 522  * @param height      The frame's height in pixels
 523  * @param color_equal Sensitivity for detecting similar colors
 524  * @param color_diff  Sensitivity for detecting different colors
 525  * @param threshold   Sensitivity for flagging planes as combed
 526  * @param prog_equal  Sensitivity for detecting similar colors on progressive frames
 527  * @param prog_diff   Sensitivity for detecting different colors on progressive frames
 528  * @param prog_threshold Sensitivity for flagging progressive frames as combed
 529  */
 530 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 )
 531 {
 532     int j, k, n, off, cc_1, cc_2, cc[3];
 533         // int flag[3] ; // debugging flag
 534     uint16_t s1, s2, s3, s4;
 535     cc_1 = 0; cc_2 = 0;
 536
 537     int offset = 0;
 538
 539     if ( buf->flags & 16 )
 540     {
 541         /* Frame is progressive, be more discerning. */
 542         color_diff = prog_diff;
 543         color_equal = prog_equal;
 544         threshold = prog_threshold;
 545     }
 546
 547     /* One pas for Y, one pass for Cb, one pass for Cr */
 548     for( k = 0; k < 3; k++ )
 549     {
 550         if( k == 1 )
 551         {
 552             /* Y has already been checked, now offset by Y's dimensions
 553                and divide all the other values by 2, since Cr and Cb
 554                are half-size compared to Y.                               */
 555             offset = width * height;
 556             width >>= 1;
 557             height >>= 1;
 558         }
 559         else if ( k == 2 )
 560         {
 561             /* Y and Cb are done, so the offset needs to be bumped
 562                so it's width*height + (width / 2) * (height / 2)  */
 563             offset *= 5/4;
 564         }
 565
 566         for( j = 0; j < width; ++j )
 567         {
 568             off = 0;
 569
 570             for( n = 0; n < ( height - 4 ); n = n + 2 )
 571             {
 572                 /* Look at groups of 4 sequential horizontal lines */
 573                 s1 = ( ( buf->data + offset )[ off + j             ] & 0xff );
 574                 s2 = ( ( buf->data + offset )[ off + j + width     ] & 0xff );
 575                 s3 = ( ( buf->data + offset )[ off + j + 2 * width ] & 0xff );
 576                 s4 = ( ( buf->data + offset )[ off + j + 3 * width ] & 0xff );
 577
 578                 /* Note if the 1st and 2nd lines are more different in
 579                    color than the 1st and 3rd lines are similar in color.*/
 580                 if ( ( abs( s1 - s3 ) < color_equal ) &&
 581                      ( abs( s1 - s2 ) > color_diff ) )
 582                         ++cc_1;
 583
 584                 /* Note if the 2nd and 3rd lines are more different in
 585                    color than the 2nd and 4th lines are similar in color.*/
 586                 if ( ( abs( s2 - s4 ) < color_equal ) &&
 587                      ( abs( s2 - s3 ) > color_diff) )
 588                         ++cc_2;
 589
 590                 /* Now move down 2 horizontal lines before starting over.*/
 591                 off += 2 * width;
 592             }
 593         }
 594
 595         // compare results
 596         /*  The final cc score for a plane is the percentage of combed pixels it contains.
 597             Because sensitivity goes down to hundreths of a percent, multiply by 1000
 598             so it will be easy to compare against the threhold value which is an integer. */
 599         cc[k] = (int)( ( cc_1 + cc_2 ) * 1000.0 / ( width * height ) );
 600     }
 601
 602
 603     /* HandBrake is all yuv420, so weight the average percentage of all 3 planes accordingly.*/
 604     int average_cc = ( 2 * cc[0] + ( cc[1] / 2 ) + ( cc[2] / 2 ) ) / 3;
 605
 606     /* Now see if that average percentage of combed pixels surpasses the threshold percentage given by the user.*/
 607     if( average_cc > threshold )
 608     {
 609 #if 0
 610             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" );
 611 #endif
 612         return 1;
 613     }
 614
 615 #if 0
 616     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" );
 617 #endif
 618
 619     /* Reaching this point means no combing detected. */
 620     return 0;
 621
 622 }
 623
 624 /**
 625  * Calculates job width and height for anamorphic content,
 626  *
 627  * @param job Handle to hb_job_t
 628  * @param output_width Pointer to returned storage width
 629  * @param output_height Pointer to returned storage height
 630  * @param output_par_width Pointer to returned pixel width
 631  @ param output_par_height Pointer to returned pixel height
 632  */
 633 void hb_set_anamorphic_size( hb_job_t * job,
 634         int *output_width, int *output_height,
 635         int *output_par_width, int *output_par_height )
 636 {
 637     /* Set up some variables to make the math easier to follow. */
 638     hb_title_t * title = job->title;
 639     int cropped_width = title->width - job->crop[2] - job->crop[3] ;
 640     int cropped_height = title->height - job->crop[0] - job->crop[1] ;
 641     double storage_aspect = (double)cropped_width / (double)cropped_height;
 642     int mod = job->anamorphic.modulus ? job->anamorphic.modulus : 16;
 643     double aspect = title->aspect;
 644
 645     int pixel_aspect_width  = job->anamorphic.par_width;
 646     int pixel_aspect_height = job->anamorphic.par_height;
 647
 648     /* If a source was really NTSC or PAL and the user specified ITU PAR
 649        values, replace the standard PAR values with the ITU broadcast ones. */
 650     if( title->width == 720 && job->anamorphic.itu_par )
 651     {
 652         // convert aspect to a scaled integer so we can test for 16:9 & 4:3
 653         // aspect ratios ignoring insignificant differences in the LSBs of
 654         // the floating point representation.
 655         int iaspect = aspect * 9.;
 656
 657         /* Handle ITU PARs */
 658         if (title->height == 480)
 659         {
 660             /* It's NTSC */
 661             if (iaspect == 16)
 662             {
 663                 /* It's widescreen */
 664                 pixel_aspect_width = 40;
 665                 pixel_aspect_height = 33;
 666             }
 667             else if (iaspect == 12)
 668             {
 669                 /* It's 4:3 */
 670                 pixel_aspect_width = 10;
 671                 pixel_aspect_height = 11;
 672             }
 673         }
 674         else if (title->height == 576)
 675         {
 676             /* It's PAL */
 677             if(iaspect == 16)
 678             {
 679                 /* It's widescreen */
 680                 pixel_aspect_width = 16;
 681                 pixel_aspect_height = 11;
 682             }
 683             else if (iaspect == 12)
 684             {
 685                 /* It's 4:3 */
 686                 pixel_aspect_width = 12;
 687                 pixel_aspect_height = 11;
 688             }
 689         }
 690     }
 691
 692     /* Figure out what width the source would display at. */
 693     int source_display_width = cropped_width * (double)pixel_aspect_width /
 694                                (double)pixel_aspect_height ;
 695
 696     /*
 697        3 different ways of deciding output dimensions:
 698         - 1: Strict anamorphic, preserve source dimensions
 699         - 2: Loose anamorphic, round to mod16 and preserve storage aspect ratio
 700         - 3: Power user anamorphic, specify everything
 701     */
 702     int width, height;
 703     switch( job->anamorphic.mode )
 704     {
 705         case 1:
 706             /* Strict anamorphic */
 707             *output_width = cropped_width;
 708             *output_height = cropped_height;
 709             *output_par_width = title->pixel_aspect_width;
 710             *output_par_height = title->pixel_aspect_height;
 711         break;
 712
 713         case 2:
 714             /* "Loose" anamorphic.
 715                 - Uses mod16-compliant dimensions,
 716                 - Allows users to set the width
 717             */
 718             width = job->width;
 719             // height: Gets set later, ignore user job->height value
 720
 721             /* Gotta handle bounding dimensions.
 722                If the width is too big, just reset it with no rescaling.
 723                Instead of using the aspect-scaled job height,
 724                we need to see if the job width divided by the storage aspect
 725                is bigger than the max. If so, set it to the max (this is sloppy).
 726                If not, set job height to job width divided by storage aspect.
 727             */
 728
 729             if ( job->maxWidth && (job->maxWidth < job->width) )
 730                 width = job->maxWidth;
 731
 732             /* Time to get picture width that divide cleanly.*/
 733             width  = MULTIPLE_MOD( width, mod);
 734
 735             /* Verify these new dimensions don't violate max height and width settings */
 736             if ( job->maxWidth && (job->maxWidth < job->width) )
 737                 width = job->maxWidth;
 738
 739             height = ((double)width / storage_aspect) + 0.5;
 740
 741             if ( job->maxHeight && (job->maxHeight < height) )
 742                 height = job->maxHeight;
 743
 744             /* Time to get picture height that divide cleanly.*/
 745             height = MULTIPLE_MOD( height, mod);
 746
 747             /* Verify these new dimensions don't violate max height and width settings */
 748             if ( job->maxHeight && (job->maxHeight < height) )
 749                 height = job->maxHeight;
 750
 751             /* The film AR is the source's display width / cropped source height.
 752                The output display width is the output height * film AR.
 753                The output PAR is the output display width / output storage width. */
 754             pixel_aspect_width = height * source_display_width / cropped_height;
 755             pixel_aspect_height = width;
 756
 757             /* Pass the results back to the caller */
 758             *output_width = width;
 759             *output_height = height;
 760         break;
 761
 762         case 3:
 763             /* Anamorphic 3: Power User Jamboree
 764                - Set everything based on specified values */
 765
 766             /* Use specified storage dimensions */
 767             width = job->width;
 768             height = job->height;
 769
 770             /* Bind to max dimensions */
 771             if( job->maxWidth && width > job->maxWidth )
 772                 width = job->maxWidth;
 773             if( job->maxHeight && height > job->maxHeight )
 774                 height = job->maxHeight;
 775
 776             /* Time to get picture dimensions that divide cleanly.*/
 777             width  = MULTIPLE_MOD( width, mod);
 778             height = MULTIPLE_MOD( height, mod);
 779
 780             /* Verify we're still within max dimensions */
 781             if( job->maxWidth && width > job->maxWidth )
 782                 width = job->maxWidth - (mod/2);
 783             if( job->maxHeight && height > job->maxHeight )
 784                 height = job->maxHeight - (mod/2);
 785
 786             /* Re-ensure we have picture dimensions that divide cleanly. */
 787             width  = MULTIPLE_MOD( width, mod );
 788             height = MULTIPLE_MOD( height, mod );
 789
 790             /* That finishes the storage dimensions. On to display. */
 791             if( job->anamorphic.dar_width && job->anamorphic.dar_height )
 792             {
 793                 /* We need to adjust the PAR to produce this aspect. */
 794                 pixel_aspect_width = height * job->anamorphic.dar_width / job->anamorphic.dar_height;
 795                 pixel_aspect_height = width;
 796             }
 797             else
 798             {
 799                 /* If we're doing ana 3 and not specifying a DAR, care needs to be taken.
 800                    This indicates a PAR is potentially being set by the interface. But
 801                    this is an output PAR, to correct a source, and it should not be assumed
 802                    that it properly creates a display aspect ratio when applied to the source,
 803                    which could easily be stored in a different resolution. */
 804
 805                 int output_display_width = width * (double)pixel_aspect_width /
 806                                            (double)pixel_aspect_height ;
 807             }
 808
 809             /* Back to caller */
 810             *output_width = width;
 811             *output_height = height;
 812         break;
 813     }
 814
 815     /* While x264 is smart enough to reduce fractions on its own, libavcodec
 816        needs some help with the math, so lose superfluous factors.            */
 817     hb_reduce( output_par_width, output_par_height,
 818                pixel_aspect_width, pixel_aspect_height );
 819 }
 820
 821 /**
 822  * Calculates job width, height, and cropping parameters.
 823  * @param job Handle to hb_job_t.
 824  * @param aspect Desired aspect ratio. Value of -1 uses title aspect.
 825  * @param pixels Maximum desired pixel count.
 826  */
 827 void hb_set_size( hb_job_t * job, double aspect, int pixels )
 828 {
 829     hb_title_t * title = job->title;
 830
 831     int croppedWidth  = title->width - title->crop[2] - title->crop[3];
 832     int croppedHeight = title->height - title->crop[0] - title->crop[1];
 833     double croppedAspect = title->aspect * title->height * croppedWidth /
 834                            croppedHeight / title->width;
 835     int addCrop;
 836     int i, w, h;
 837
 838     if( aspect <= 0 )
 839     {
 840         /* Keep the best possible aspect ratio */
 841         aspect = croppedAspect;
 842     }
 843
 844     /* Crop if necessary to obtain the desired ratio */
 845     memcpy( job->crop, title->crop, 4 * sizeof( int ) );
 846     if( aspect < croppedAspect )
 847     {
 848         /* Need to crop on the left and right */
 849         addCrop = croppedWidth - aspect * croppedHeight * title->width /
 850                     title->aspect / title->height;
 851         if( addCrop & 3 )
 852         {
 853             addCrop = ( addCrop + 1 ) / 2;
 854             job->crop[2] += addCrop;
 855             job->crop[3] += addCrop;
 856         }
 857         else if( addCrop & 2 )
 858         {
 859             addCrop /= 2;
 860             job->crop[2] += addCrop - 1;
 861             job->crop[3] += addCrop + 1;
 862         }
 863         else
 864         {
 865             addCrop /= 2;
 866             job->crop[2] += addCrop;
 867             job->crop[3] += addCrop;
 868         }
 869     }
 870     else if( aspect > croppedAspect )
 871     {
 872         /* Need to crop on the top and bottom */
 873         addCrop = croppedHeight - croppedWidth * title->aspect *
 874             title->height / aspect / title->width;
 875         if( addCrop & 3 )
 876         {
 877             addCrop = ( addCrop + 1 ) / 2;
 878             job->crop[0] += addCrop;
 879             job->crop[1] += addCrop;
 880         }
 881         else if( addCrop & 2 )
 882         {
 883             addCrop /= 2;
 884             job->crop[0] += addCrop - 1;
 885             job->crop[1] += addCrop + 1;
 886         }
 887         else
 888         {
 889             addCrop /= 2;
 890             job->crop[0] += addCrop;
 891             job->crop[1] += addCrop;
 892         }
 893     }
 894
 895     /* Compute a resolution from the number of pixels and aspect */
 896     for( i = 0;; i++ )
 897     {
 898         w = 16 * i;
 899         h = MULTIPLE_16( (int)( (double)w / aspect ) );
 900         if( w * h > pixels )
 901         {
 902             break;
 903         }
 904     }
 905     i--;
 906     job->width  = 16 * i;
 907     job->height = MULTIPLE_16( (int)( (double)job->width / aspect ) );
 908 }
 909
 910 /**
 911  * Returns the number of jobs in the queue.
 912  * @param h Handle to hb_handle_t.
 913  * @return Number of jobs.
 914  */
 915 int hb_count( hb_handle_t * h )
 916 {
 917     return hb_list_count( h->jobs );
 918 }
 919
 920 /**
 921  * Returns handle to job at index i within the job list.
 922  * @param h Handle to hb_handle_t.
 923  * @param i Index of job.
 924  * @returns Handle to hb_job_t of desired job.
 925  */
 926 hb_job_t * hb_job( hb_handle_t * h, int i )
 927 {
 928     return hb_list_item( h->jobs, i );
 929 }
 930
 931 hb_job_t * hb_current_job( hb_handle_t * h )
 932 {
 933     return( h->current_job );
 934 }
 935
 936 /**
 937  * Adds a job to the job list.
 938  * @param h Handle to hb_handle_t.
 939  * @param job Handle to hb_job_t.
 940  */
 941 void hb_add( hb_handle_t * h, hb_job_t * job )
 942 {
 943     hb_job_t      * job_copy;
 944     hb_title_t    * title,    * title_copy;
 945     hb_chapter_t  * chapter,  * chapter_copy;
 946     hb_audio_t    * audio;
 947     hb_subtitle_t * subtitle, * subtitle_copy;
 948     int             i;
 949     char            audio_lang[4];
 950
 951     /* Copy the title */
 952     title      = job->title;
 953     title_copy = malloc( sizeof( hb_title_t ) );
 954     memcpy( title_copy, title, sizeof( hb_title_t ) );
 955
 956     title_copy->list_chapter = hb_list_init();
 957     for( i = 0; i < hb_list_count( title->list_chapter ); i++ )
 958     {
 959         chapter      = hb_list_item( title->list_chapter, i );
 960         chapter_copy = malloc( sizeof( hb_chapter_t ) );
 961         memcpy( chapter_copy, chapter, sizeof( hb_chapter_t ) );
 962         hb_list_add( title_copy->list_chapter, chapter_copy );
 963     }
 964
 965     /*
 966      * Copy the metadata
 967      */
 968     if( title->metadata )
 969     {
 970         title_copy->metadata = malloc( sizeof( hb_metadata_t ) );
 971
 972         if( title_copy->metadata )
 973         {
 974             memcpy( title_copy->metadata, title->metadata, sizeof( hb_metadata_t ) );
 975
 976             /*
 977              * Need to copy the artwork seperatly (TODO).
 978              */
 979             if( title->metadata->coverart )
 980             {
 981                 title_copy->metadata->coverart = malloc( title->metadata->coverart_size );
 982                 if( title_copy->metadata->coverart )
 983                 {
 984                     memcpy( title_copy->metadata->coverart, title->metadata->coverart,
 985                             title->metadata->coverart_size );
 986                 } else {
 987                     title_copy->metadata->coverart_size = 0;
 988                 }
 989             }
 990         }
 991     }
 992
 993     /* Copy the audio track(s) we want */
 994     title_copy->list_audio = hb_list_init();
 995
 996     for( i = 0; i < hb_list_count(job->list_audio); i++ )
 997     {
 998         if( ( audio = hb_list_item( job->list_audio, i ) ) )
 999         {
1000             hb_list_add( title_copy->list_audio, hb_audio_copy(audio) );
1001         }
1002     }
1003
1004     title_copy->list_subtitle = hb_list_init();
1005
1006     /*
1007      * The following code is confusing, there are three ways in which
1008      * we select subtitles and it depends on whether this is single or
1009      * two pass mode.
1010      *
1011      * subtitle_scan may be enabled, in which case the first pass
1012      * scans all subtitles of that language. The second pass does not
1013      * select any because they are set at the end of the first pass.
1014      *
1015      * native_language may have a preferred language, in which case we
1016      * may be switching the language we want for the subtitles in the
1017      * first pass of a single pass, or the second pass of a two pass.
1018      *
1019      * We may have manually selected a subtitle, in which case that is
1020      * selected in the first pass of a single pass, or the second of a
1021      * two pass.
1022      */
1023     memset( audio_lang, 0, sizeof( audio_lang ) );
1024
1025     if ( job->indepth_scan || job->native_language ) {
1026
1027         /*
1028          * Find the first audio language that is being encoded
1029          */
1030         for( i = 0; i < hb_list_count(job->list_audio); i++ )
1031         {
1032             if( ( audio = hb_list_item( job->list_audio, i ) ) )
1033             {
1034                 strncpy(audio_lang, audio->config.lang.iso639_2, sizeof(audio_lang));
1035                 break;
1036             }
1037         }
1038
1039         /*
1040          * In all cases switch the language if we need to to our native
1041          * language.
1042          */
1043         if( job->native_language )
1044         {
1045             if( strncasecmp( job->native_language, audio_lang,
1046                              sizeof( audio_lang ) ) != 0 )
1047             {
1048
1049                 if( job->pass != 2 )
1050                 {
1051                     hb_log( "Enabled subtitles in native language '%s', audio is in '%s'",
1052                             job->native_language, audio_lang);
1053                 }
1054                 /*
1055                  * The main audio track is not in our native language, so switch
1056                  * the subtitles to use our native language instead.
1057                  */
1058                 strncpy( audio_lang, job->native_language, sizeof( audio_lang ) );
1059             } else {
1060                 /*
1061                  * native language is irrelevent, free it.
1062                  */
1063                 free( job->native_language );
1064                 job->native_language = NULL;
1065             }
1066         }
1067     }
1068
1069     /*
1070      * If doing a subtitle scan then add all the matching subtitles for this
1071      * language.
1072      */
1073     if ( job->indepth_scan )
1074     {
1075         for( i=0; i < hb_list_count( title->list_subtitle ); i++ )
1076         {
1077             subtitle = hb_list_item( title->list_subtitle, i );
1078             if( strcmp( subtitle->iso639_2, audio_lang ) == 0 &&
1079                 subtitle->source == VOBSUB )
1080             {
1081                 /*
1082                  * Matched subtitle language with audio language, so
1083                  * add this to our list to scan.
1084                  *
1085                  * We will update the subtitle list on the second pass
1086                  * later after the first pass has completed.
1087                  */
1088                 subtitle_copy = malloc( sizeof( hb_subtitle_t ) );
1089                 memcpy( subtitle_copy, subtitle, sizeof( hb_subtitle_t ) );
1090                 hb_list_add( title_copy->list_subtitle, subtitle_copy );
1091                 if ( job->native_language ) {
1092                     /*
1093                      * With native language just select the
1094                      * first match in our langiage, not all of
1095                      * them. Subsequent ones are likely to be commentary
1096                      */
1097                     break;
1098                 }
1099             }
1100         }
1101     } else {
1102         /*
1103          * Not doing a subtitle scan in this pass, but maybe we are in the
1104          * first pass?
1105          */
1106         if( job->pass != 1 && job->native_language )
1107         {
1108             /*
1109              * We are not doing a subtitle scan but do want the
1110              * native langauge subtitle selected, so select it
1111              * for pass 0 or pass 2 of a two pass.
1112              */
1113             for( i=0; i < hb_list_count( title->list_subtitle ); i++ )
1114             {
1115                 subtitle = hb_list_item( title->list_subtitle, i );
1116                 if( strcmp( subtitle->iso639_2, audio_lang ) == 0 )
1117                 {
1118                     /*
1119                      * Matched subtitle language with audio language, so
1120                      * add this to our list to scan.
1121                      */
1122                     subtitle_copy = malloc( sizeof( hb_subtitle_t ) );
1123                     memcpy( subtitle_copy, subtitle, sizeof( hb_subtitle_t ) );
1124                     hb_list_add( title_copy->list_subtitle, subtitle_copy );
1125                     break;
1126                 }
1127             }
1128         } else {
1129             /*
1130              * Manually selected subtitles, in which case only
1131              * bother adding them for pass 0 or pass 2 of a two
1132              * pass.
1133              */
1134             if( job->pass != 1 )
1135             {
1136                 /*
1137                  * Copy all of them from the input job, to the title_copy/job_copy.
1138                  */
1139                 for(  i = 0; i < hb_list_count(job->list_subtitle); i++ ) {
1140                     if( ( subtitle = hb_list_item( job->list_subtitle, i ) ) )
1141                     {
1142                         subtitle_copy = malloc( sizeof( hb_subtitle_t ) );
1143                         memcpy( subtitle_copy, subtitle, sizeof( hb_subtitle_t ) );
1144                         hb_list_add( title_copy->list_subtitle, subtitle_copy );
1145                     }
1146                 }
1147             }
1148         }
1149     }
1150
1151     /* Copy the job */
1152     job_copy        = calloc( sizeof( hb_job_t ), 1 );
1153     memcpy( job_copy, job, sizeof( hb_job_t ) );
1154     title_copy->job = job_copy;
1155     job_copy->title = title_copy;
1156     job_copy->list_audio = title_copy->list_audio;
1157     job_copy->list_subtitle = title_copy->list_subtitle;   // sharing list between title and job
1158     job_copy->file  = strdup( job->file );
1159     job_copy->h     = h;
1160     job_copy->pause = h->pause_lock;
1161
1162     /* Copy the job filter list */
1163     if( job->filters )
1164     {
1165         int i;
1166         int filter_count = hb_list_count( job->filters );
1167         job_copy->filters = hb_list_init();
1168         for( i = 0; i < filter_count; i++ )
1169         {
1170             /*
1171              * Copy the filters, since the MacGui reuses the global filter objects
1172              * meaning that queued up jobs overwrite the previous filter settings.
1173              * In reality, settings is probably the only field that needs duplicating
1174              * since it's the only value that is ever changed. But name is duplicated
1175              * as well for completeness. Not copying private_data since it gets
1176              * created for each job in renderInit.
1177              */
1178             hb_filter_object_t * filter = hb_list_item( job->filters, i );
1179             hb_filter_object_t * filter_copy = malloc( sizeof( hb_filter_object_t ) );
1180             memcpy( filter_copy, filter, sizeof( hb_filter_object_t ) );
1181             if( filter->name )
1182                 filter_copy->name = strdup( filter->name );
1183             if( filter->settings )
1184                 filter_copy->settings = strdup( filter->settings );
1185             hb_list_add( job_copy->filters, filter_copy );
1186         }
1187     }
1188
1189     /* Add the job to the list */
1190     hb_list_add( h->jobs, job_copy );
1191     h->job_count = hb_count(h);
1192     h->job_count_permanent++;
1193 }
1194
1195 /**
1196  * Removes a job from the job list.
1197  * @param h Handle to hb_handle_t.
1198  * @param job Handle to hb_job_t.
1199  */
1200 void hb_rem( hb_handle_t * h, hb_job_t * job )
1201 {
1202     hb_list_rem( h->jobs, job );
1203
1204     h->job_count = hb_count(h);
1205     if (h->job_count_permanent)
1206         h->job_count_permanent--;
1207
1208     /* XXX free everything XXX */
1209 }
1210
1211 /**
1212  * Starts the conversion process.
1213  * Sets state to HB_STATE_WORKING.
1214  * calls hb_work_init, to launch work thread. Stores handle to work thread.
1215  * @param h Handle to hb_handle_t.
1216  */
1217 void hb_start( hb_handle_t * h )
1218 {
1219     /* XXX Hack */
1220     h->job_count = hb_list_count( h->jobs );
1221     h->job_count_permanent = h->job_count;
1222
1223     hb_lock( h->state_lock );
1224     h->state.state = HB_STATE_WORKING;
1225 #define p h->state.param.working
1226     p.progress  = 0.0;
1227     p.job_cur   = 1;
1228     p.job_count = h->job_count;
1229     p.rate_cur  = 0.0;
1230     p.rate_avg  = 0.0;
1231     p.hours     = -1;
1232     p.minutes   = -1;
1233     p.seconds   = -1;
1234     p.sequence_id = 0;
1235 #undef p
1236     hb_unlock( h->state_lock );
1237
1238     h->paused = 0;
1239
1240     h->work_die    = 0;
1241     h->work_thread = hb_work_init( h->jobs, h->cpu_count,
1242                                    &h->work_die, &h->work_error, &h->current_job );
1243 }
1244
1245 /**
1246  * Pauses the conversion process.
1247  * @param h Handle to hb_handle_t.
1248  */
1249 void hb_pause( hb_handle_t * h )
1250 {
1251     if( !h->paused )
1252     {
1253         hb_lock( h->pause_lock );
1254         h->paused = 1;
1255
1256         hb_lock( h->state_lock );
1257         h->state.state = HB_STATE_PAUSED;
1258         hb_unlock( h->state_lock );
1259     }
1260 }
1261
1262 /**
1263  * Resumes the conversion process.
1264  * @param h Handle to hb_handle_t.
1265  */
1266 void hb_resume( hb_handle_t * h )
1267 {
1268     if( h->paused )
1269     {
1270         hb_unlock( h->pause_lock );
1271         h->paused = 0;
1272     }
1273 }
1274
1275 /**
1276  * Stops the conversion process.
1277  * @param h Handle to hb_handle_t.
1278  */
1279 void hb_stop( hb_handle_t * h )
1280 {
1281     h->work_die = 1;
1282
1283     h->job_count = hb_count(h);
1284     h->job_count_permanent = 0;
1285
1286     hb_resume( h );
1287 }
1288
1289 /**
1290  * Returns the state of the conversion process.
1291  * @param h Handle to hb_handle_t.
1292  * @param s Handle to hb_state_t which to copy the state data.
1293  */
1294 void hb_get_state( hb_handle_t * h, hb_state_t * s )
1295 {
1296     hb_lock( h->state_lock );
1297
1298     memcpy( s, &h->state, sizeof( hb_state_t ) );
1299     if ( h->state.state == HB_STATE_SCANDONE || h->state.state == HB_STATE_WORKDONE )
1300         h->state.state = HB_STATE_IDLE;
1301
1302     hb_unlock( h->state_lock );
1303 }
1304
1305 void hb_get_state2( hb_handle_t * h, hb_state_t * s )
1306 {
1307     hb_lock( h->state_lock );
1308
1309     memcpy( s, &h->state, sizeof( hb_state_t ) );
1310
1311     hb_unlock( h->state_lock );
1312 }
1313
1314 /**
1315  * Called in MacGui in UpdateUI to check
1316  *  for a new scan being completed to set a new source
1317  */
1318 int hb_get_scancount( hb_handle_t * h)
1319  {
1320      return h->scanCount;
1321  }
1322
1323 /**
1324  * Closes access to libhb by freeing the hb_handle_t handle ontained in hb_init.
1325  * @param _h Pointer to handle to hb_handle_t.
1326  */
1327 void hb_close( hb_handle_t ** _h )
1328 {
1329     hb_handle_t * h = *_h;
1330     hb_title_t * title;
1331
1332     h->die = 1;
1333     hb_thread_close( &h->main_thread );
1334
1335     while( ( title = hb_list_item( h->list_title, 0 ) ) )
1336     {
1337         hb_list_rem( h->list_title, title );
1338         if( title->job && title->job->filters )
1339         {
1340             hb_list_close( &title->job->filters );
1341         }
1342         free( title->job );
1343         hb_title_close( &title );
1344     }
1345     hb_list_close( &h->list_title );
1346
1347     hb_list_close( &h->jobs );
1348     hb_lock_close( &h->state_lock );
1349     hb_lock_close( &h->pause_lock );
1350     free( h );
1351     *_h = NULL;
1352
1353 }
1354
1355 /**
1356  * Monitors the state of the update, scan, and work threads.
1357  * Sets scan done state when scan thread exits.
1358  * Sets work done state when work thread exits.
1359  * @param _h Handle to hb_handle_t
1360  */
1361 static void thread_func( void * _h )
1362 {
1363     hb_handle_t * h = (hb_handle_t *) _h;
1364     char dirname[1024];
1365     DIR * dir;
1366     struct dirent * entry;
1367
1368     h->pid = getpid();
1369
1370     /* Create folder for temporary files */
1371     memset( dirname, 0, 1024 );
1372     hb_get_tempory_directory( h, dirname );
1373
1374     hb_mkdir( dirname );
1375
1376     while( !h->die )
1377     {
1378         /* In case the check_update thread hangs, it'll die sooner or
1379            later. Then, we join it here */
1380         if( h->update_thread &&
1381             hb_thread_has_exited( h->update_thread ) )
1382         {
1383             hb_thread_close( &h->update_thread );
1384         }
1385
1386         /* Check if the scan thread is done */
1387         if( h->scan_thread &&
1388             hb_thread_has_exited( h->scan_thread ) )
1389         {
1390             hb_thread_close( &h->scan_thread );
1391
1392             hb_log( "libhb: scan thread found %d valid title(s)",
1393                     hb_list_count( h->list_title ) );
1394             hb_lock( h->state_lock );
1395             h->state.state = HB_STATE_SCANDONE; //originally state.state
1396                         hb_unlock( h->state_lock );
1397                         /*we increment this sessions scan count by one for the MacGui
1398                         to trigger a new source being set */
1399             h->scanCount++;
1400         }
1401
1402         /* Check if the work thread is done */
1403         if( h->work_thread &&
1404             hb_thread_has_exited( h->work_thread ) )
1405         {
1406             hb_thread_close( &h->work_thread );
1407
1408             hb_log( "libhb: work result = %d",
1409                     h->work_error );
1410             hb_lock( h->state_lock );
1411             h->state.state                = HB_STATE_WORKDONE;
1412             h->state.param.workdone.error = h->work_error;
1413
1414             h->job_count = hb_count(h);
1415             if (h->job_count < 1)
1416                 h->job_count_permanent = 0;
1417             hb_unlock( h->state_lock );
1418         }
1419
1420         hb_snooze( 50 );
1421     }
1422
1423     if( h->work_thread )
1424     {
1425         hb_stop( h );
1426         hb_thread_close( &h->work_thread );
1427     }
1428
1429     /* Remove temp folder */
1430     dir = opendir( dirname );
1431     if (dir)
1432     {
1433         while( ( entry = readdir( dir ) ) )
1434         {
1435             char filename[1024];
1436             if( entry->d_name[0] == '.' )
1437             {
1438                 continue;
1439             }
1440             memset( filename, 0, 1024 );
1441             snprintf( filename, 1023, "%s/%s", dirname, entry->d_name );
1442             unlink( filename );
1443         }
1444         closedir( dir );
1445         rmdir( dirname );
1446     }
1447 }
1448
1449 /**
1450  * Returns the PID.
1451  * @param h Handle to hb_handle_t
1452  */
1453 int hb_get_pid( hb_handle_t * h )
1454 {
1455     return h->pid;
1456 }
1457
1458 /**
1459  * Sets the current state.
1460  * @param h Handle to hb_handle_t
1461  * @param s Handle to new hb_state_t
1462  */
1463 void hb_set_state( hb_handle_t * h, hb_state_t * s )
1464 {
1465     hb_lock( h->pause_lock );
1466     hb_lock( h->state_lock );
1467     memcpy( &h->state, s, sizeof( hb_state_t ) );
1468     if( h->state.state == HB_STATE_WORKING )
1469     {
1470         /* XXX Hack */
1471         if (h->job_count < 1)
1472             h->job_count_permanent = 1;
1473
1474         h->state.param.working.job_cur =
1475             h->job_count_permanent - hb_list_count( h->jobs );
1476         h->state.param.working.job_count = h->job_count_permanent;
1477
1478         // Set which job is being worked on
1479         if (h->current_job)
1480             h->state.param.working.sequence_id = h->current_job->sequence_id;
1481         else
1482             h->state.param.working.sequence_id = 0;
1483     }
1484     hb_unlock( h->state_lock );
1485     hb_unlock( h->pause_lock );
1486 }
1487
1488 /* Passes a pointer to persistent data */
1489 hb_interjob_t * hb_interjob_get( hb_handle_t * h )
1490 {
1491     return h->interjob;
1492 }