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                 /* We first need the display ar.
 800                    That's the source display width divided by the source height after cropping.
 801                    Then we multiple the output height by that to get the pixel aspect width,
 802                    and the pixel aspect height is the storage width.*/
 803                 pixel_aspect_width = height * source_display_width / cropped_height;
 804                 pixel_aspect_height = width;
 805             }
 806
 807             /* Back to caller */
 808             *output_width = width;
 809             *output_height = height;
 810         break;
 811     }
 812
 813     /* While x264 is smart enough to reduce fractions on its own, libavcodec
 814        needs some help with the math, so lose superfluous factors.            */
 815     hb_reduce( output_par_width, output_par_height,
 816                pixel_aspect_width, pixel_aspect_height );
 817 }
 818
 819 /**
 820  * Calculates job width, height, and cropping parameters.
 821  * @param job Handle to hb_job_t.
 822  * @param aspect Desired aspect ratio. Value of -1 uses title aspect.
 823  * @param pixels Maximum desired pixel count.
 824  */
 825 void hb_set_size( hb_job_t * job, double aspect, int pixels )
 826 {
 827     hb_title_t * title = job->title;
 828
 829     int croppedWidth  = title->width - title->crop[2] - title->crop[3];
 830     int croppedHeight = title->height - title->crop[0] - title->crop[1];
 831     double croppedAspect = title->aspect * title->height * croppedWidth /
 832                            croppedHeight / title->width;
 833     int addCrop;
 834     int i, w, h;
 835
 836     if( aspect <= 0 )
 837     {
 838         /* Keep the best possible aspect ratio */
 839         aspect = croppedAspect;
 840     }
 841
 842     /* Crop if necessary to obtain the desired ratio */
 843     memcpy( job->crop, title->crop, 4 * sizeof( int ) );
 844     if( aspect < croppedAspect )
 845     {
 846         /* Need to crop on the left and right */
 847         addCrop = croppedWidth - aspect * croppedHeight * title->width /
 848                     title->aspect / title->height;
 849         if( addCrop & 3 )
 850         {
 851             addCrop = ( addCrop + 1 ) / 2;
 852             job->crop[2] += addCrop;
 853             job->crop[3] += addCrop;
 854         }
 855         else if( addCrop & 2 )
 856         {
 857             addCrop /= 2;
 858             job->crop[2] += addCrop - 1;
 859             job->crop[3] += addCrop + 1;
 860         }
 861         else
 862         {
 863             addCrop /= 2;
 864             job->crop[2] += addCrop;
 865             job->crop[3] += addCrop;
 866         }
 867     }
 868     else if( aspect > croppedAspect )
 869     {
 870         /* Need to crop on the top and bottom */
 871         addCrop = croppedHeight - croppedWidth * title->aspect *
 872             title->height / aspect / title->width;
 873         if( addCrop & 3 )
 874         {
 875             addCrop = ( addCrop + 1 ) / 2;
 876             job->crop[0] += addCrop;
 877             job->crop[1] += addCrop;
 878         }
 879         else if( addCrop & 2 )
 880         {
 881             addCrop /= 2;
 882             job->crop[0] += addCrop - 1;
 883             job->crop[1] += addCrop + 1;
 884         }
 885         else
 886         {
 887             addCrop /= 2;
 888             job->crop[0] += addCrop;
 889             job->crop[1] += addCrop;
 890         }
 891     }
 892
 893     /* Compute a resolution from the number of pixels and aspect */
 894     for( i = 0;; i++ )
 895     {
 896         w = 16 * i;
 897         h = MULTIPLE_16( (int)( (double)w / aspect ) );
 898         if( w * h > pixels )
 899         {
 900             break;
 901         }
 902     }
 903     i--;
 904     job->width  = 16 * i;
 905     job->height = MULTIPLE_16( (int)( (double)job->width / aspect ) );
 906 }
 907
 908 /**
 909  * Returns the number of jobs in the queue.
 910  * @param h Handle to hb_handle_t.
 911  * @return Number of jobs.
 912  */
 913 int hb_count( hb_handle_t * h )
 914 {
 915     return hb_list_count( h->jobs );
 916 }
 917
 918 /**
 919  * Returns handle to job at index i within the job list.
 920  * @param h Handle to hb_handle_t.
 921  * @param i Index of job.
 922  * @returns Handle to hb_job_t of desired job.
 923  */
 924 hb_job_t * hb_job( hb_handle_t * h, int i )
 925 {
 926     return hb_list_item( h->jobs, i );
 927 }
 928
 929 hb_job_t * hb_current_job( hb_handle_t * h )
 930 {
 931     return( h->current_job );
 932 }
 933
 934 /**
 935  * Adds a job to the job list.
 936  * @param h Handle to hb_handle_t.
 937  * @param job Handle to hb_job_t.
 938  */
 939 void hb_add( hb_handle_t * h, hb_job_t * job )
 940 {
 941     hb_job_t      * job_copy;
 942     hb_title_t    * title,    * title_copy;
 943     hb_chapter_t  * chapter,  * chapter_copy;
 944     hb_audio_t    * audio;
 945     hb_subtitle_t * subtitle, * subtitle_copy;
 946     int             i;
 947     char            audio_lang[4];
 948
 949     /* Copy the title */
 950     title      = job->title;
 951     title_copy = malloc( sizeof( hb_title_t ) );
 952     memcpy( title_copy, title, sizeof( hb_title_t ) );
 953
 954     title_copy->list_chapter = hb_list_init();
 955     for( i = 0; i < hb_list_count( title->list_chapter ); i++ )
 956     {
 957         chapter      = hb_list_item( title->list_chapter, i );
 958         chapter_copy = malloc( sizeof( hb_chapter_t ) );
 959         memcpy( chapter_copy, chapter, sizeof( hb_chapter_t ) );
 960         hb_list_add( title_copy->list_chapter, chapter_copy );
 961     }
 962
 963     /*
 964      * Copy the metadata
 965      */
 966     if( title->metadata )
 967     {
 968         title_copy->metadata = malloc( sizeof( hb_metadata_t ) );
 969
 970         if( title_copy->metadata )
 971         {
 972             memcpy( title_copy->metadata, title->metadata, sizeof( hb_metadata_t ) );
 973
 974             /*
 975              * Need to copy the artwork seperatly (TODO).
 976              */
 977             if( title->metadata->coverart )
 978             {
 979                 title_copy->metadata->coverart = malloc( title->metadata->coverart_size );
 980                 if( title_copy->metadata->coverart )
 981                 {
 982                     memcpy( title_copy->metadata->coverart, title->metadata->coverart,
 983                             title->metadata->coverart_size );
 984                 } else {
 985                     title_copy->metadata->coverart_size = 0;
 986                 }
 987             }
 988         }
 989     }
 990
 991     /* Copy the audio track(s) we want */
 992     title_copy->list_audio = hb_list_init();
 993
 994     for( i = 0; i < hb_list_count(job->list_audio); i++ )
 995     {
 996         if( ( audio = hb_list_item( job->list_audio, i ) ) )
 997         {
 998             hb_list_add( title_copy->list_audio, hb_audio_copy(audio) );
 999         }
1000     }
1001
1002     title_copy->list_subtitle = hb_list_init();
1003
1004     /*
1005      * The following code is confusing, there are three ways in which
1006      * we select subtitles and it depends on whether this is single or
1007      * two pass mode.
1008      *
1009      * subtitle_scan may be enabled, in which case the first pass
1010      * scans all subtitles of that language. The second pass does not
1011      * select any because they are set at the end of the first pass.
1012      *
1013      * native_language may have a preferred language, in which case we
1014      * may be switching the language we want for the subtitles in the
1015      * first pass of a single pass, or the second pass of a two pass.
1016      *
1017      * We may have manually selected a subtitle, in which case that is
1018      * selected in the first pass of a single pass, or the second of a
1019      * two pass.
1020      */
1021     memset( audio_lang, 0, sizeof( audio_lang ) );
1022
1023     if ( job->indepth_scan || job->native_language ) {
1024
1025         /*
1026          * Find the first audio language that is being encoded
1027          */
1028         for( i = 0; i < hb_list_count(job->list_audio); i++ )
1029         {
1030             if( ( audio = hb_list_item( job->list_audio, i ) ) )
1031             {
1032                 strncpy(audio_lang, audio->config.lang.iso639_2, sizeof(audio_lang));
1033                 break;
1034             }
1035         }
1036
1037         /*
1038          * In all cases switch the language if we need to to our native
1039          * language.
1040          */
1041         if( job->native_language )
1042         {
1043             if( strncasecmp( job->native_language, audio_lang,
1044                              sizeof( audio_lang ) ) != 0 )
1045             {
1046
1047                 if( job->pass != 2 )
1048                 {
1049                     hb_log( "Enabled subtitles in native language '%s', audio is in '%s'",
1050                             job->native_language, audio_lang);
1051                 }
1052                 /*
1053                  * The main audio track is not in our native language, so switch
1054                  * the subtitles to use our native language instead.
1055                  */
1056                 strncpy( audio_lang, job->native_language, sizeof( audio_lang ) );
1057             } else {
1058                 /*
1059                  * native language is irrelevent, free it.
1060                  */
1061                 free( job->native_language );
1062                 job->native_language = NULL;
1063             }
1064         }
1065     }
1066
1067     /*
1068      * If doing a subtitle scan then add all the matching subtitles for this
1069      * language.
1070      */
1071     if ( job->indepth_scan )
1072     {
1073         for( i=0; i < hb_list_count( title->list_subtitle ); i++ )
1074         {
1075             subtitle = hb_list_item( title->list_subtitle, i );
1076             if( strcmp( subtitle->iso639_2, audio_lang ) == 0 &&
1077                 subtitle->source == VOBSUB )
1078             {
1079                 /*
1080                  * Matched subtitle language with audio language, so
1081                  * add this to our list to scan.
1082                  *
1083                  * We will update the subtitle list on the second pass
1084                  * later after the first pass has completed.
1085                  */
1086                 subtitle_copy = malloc( sizeof( hb_subtitle_t ) );
1087                 memcpy( subtitle_copy, subtitle, sizeof( hb_subtitle_t ) );
1088                 hb_list_add( title_copy->list_subtitle, subtitle_copy );
1089                 if ( job->native_language ) {
1090                     /*
1091                      * With native language just select the
1092                      * first match in our langiage, not all of
1093                      * them. Subsequent ones are likely to be commentary
1094                      */
1095                     break;
1096                 }
1097             }
1098         }
1099     } else {
1100         /*
1101          * Not doing a subtitle scan in this pass, but maybe we are in the
1102          * first pass?
1103          */
1104         if( job->pass != 1 && job->native_language )
1105         {
1106             /*
1107              * We are not doing a subtitle scan but do want the
1108              * native langauge subtitle selected, so select it
1109              * for pass 0 or pass 2 of a two pass.
1110              */
1111             for( i=0; i < hb_list_count( title->list_subtitle ); i++ )
1112             {
1113                 subtitle = hb_list_item( title->list_subtitle, i );
1114                 if( strcmp( subtitle->iso639_2, audio_lang ) == 0 )
1115                 {
1116                     /*
1117                      * Matched subtitle language with audio language, so
1118                      * add this to our list to scan.
1119                      */
1120                     subtitle_copy = malloc( sizeof( hb_subtitle_t ) );
1121                     memcpy( subtitle_copy, subtitle, sizeof( hb_subtitle_t ) );
1122                     hb_list_add( title_copy->list_subtitle, subtitle_copy );
1123                     break;
1124                 }
1125             }
1126         } else {
1127             /*
1128              * Manually selected subtitles, in which case only
1129              * bother adding them for pass 0 or pass 2 of a two
1130              * pass.
1131              */
1132             if( job->pass != 1 )
1133             {
1134                 /*
1135                  * Copy all of them from the input job, to the title_copy/job_copy.
1136                  */
1137                 for(  i = 0; i < hb_list_count(job->list_subtitle); i++ ) {
1138                     if( ( subtitle = hb_list_item( job->list_subtitle, i ) ) )
1139                     {
1140                         subtitle_copy = malloc( sizeof( hb_subtitle_t ) );
1141                         memcpy( subtitle_copy, subtitle, sizeof( hb_subtitle_t ) );
1142                         hb_list_add( title_copy->list_subtitle, subtitle_copy );
1143                     }
1144                 }
1145             }
1146         }
1147     }
1148
1149     /* Copy the job */
1150     job_copy        = calloc( sizeof( hb_job_t ), 1 );
1151     memcpy( job_copy, job, sizeof( hb_job_t ) );
1152     title_copy->job = job_copy;
1153     job_copy->title = title_copy;
1154     job_copy->list_audio = title_copy->list_audio;
1155     job_copy->list_subtitle = title_copy->list_subtitle;   // sharing list between title and job
1156     job_copy->file  = strdup( job->file );
1157     job_copy->h     = h;
1158     job_copy->pause = h->pause_lock;
1159
1160     /* Copy the job filter list */
1161     if( job->filters )
1162     {
1163         int i;
1164         int filter_count = hb_list_count( job->filters );
1165         job_copy->filters = hb_list_init();
1166         for( i = 0; i < filter_count; i++ )
1167         {
1168             /*
1169              * Copy the filters, since the MacGui reuses the global filter objects
1170              * meaning that queued up jobs overwrite the previous filter settings.
1171              * In reality, settings is probably the only field that needs duplicating
1172              * since it's the only value that is ever changed. But name is duplicated
1173              * as well for completeness. Not copying private_data since it gets
1174              * created for each job in renderInit.
1175              */
1176             hb_filter_object_t * filter = hb_list_item( job->filters, i );
1177             hb_filter_object_t * filter_copy = malloc( sizeof( hb_filter_object_t ) );
1178             memcpy( filter_copy, filter, sizeof( hb_filter_object_t ) );
1179             if( filter->name )
1180                 filter_copy->name = strdup( filter->name );
1181             if( filter->settings )
1182                 filter_copy->settings = strdup( filter->settings );
1183             hb_list_add( job_copy->filters, filter_copy );
1184         }
1185     }
1186
1187     /* Add the job to the list */
1188     hb_list_add( h->jobs, job_copy );
1189     h->job_count = hb_count(h);
1190     h->job_count_permanent++;
1191 }
1192
1193 /**
1194  * Removes a job from the job list.
1195  * @param h Handle to hb_handle_t.
1196  * @param job Handle to hb_job_t.
1197  */
1198 void hb_rem( hb_handle_t * h, hb_job_t * job )
1199 {
1200     hb_list_rem( h->jobs, job );
1201
1202     h->job_count = hb_count(h);
1203     if (h->job_count_permanent)
1204         h->job_count_permanent--;
1205
1206     /* XXX free everything XXX */
1207 }
1208
1209 /**
1210  * Starts the conversion process.
1211  * Sets state to HB_STATE_WORKING.
1212  * calls hb_work_init, to launch work thread. Stores handle to work thread.
1213  * @param h Handle to hb_handle_t.
1214  */
1215 void hb_start( hb_handle_t * h )
1216 {
1217     /* XXX Hack */
1218     h->job_count = hb_list_count( h->jobs );
1219     h->job_count_permanent = h->job_count;
1220
1221     hb_lock( h->state_lock );
1222     h->state.state = HB_STATE_WORKING;
1223 #define p h->state.param.working
1224     p.progress  = 0.0;
1225     p.job_cur   = 1;
1226     p.job_count = h->job_count;
1227     p.rate_cur  = 0.0;
1228     p.rate_avg  = 0.0;
1229     p.hours     = -1;
1230     p.minutes   = -1;
1231     p.seconds   = -1;
1232     p.sequence_id = 0;
1233 #undef p
1234     hb_unlock( h->state_lock );
1235
1236     h->paused = 0;
1237
1238     h->work_die    = 0;
1239     h->work_thread = hb_work_init( h->jobs, h->cpu_count,
1240                                    &h->work_die, &h->work_error, &h->current_job );
1241 }
1242
1243 /**
1244  * Pauses the conversion process.
1245  * @param h Handle to hb_handle_t.
1246  */
1247 void hb_pause( hb_handle_t * h )
1248 {
1249     if( !h->paused )
1250     {
1251         hb_lock( h->pause_lock );
1252         h->paused = 1;
1253
1254         hb_lock( h->state_lock );
1255         h->state.state = HB_STATE_PAUSED;
1256         hb_unlock( h->state_lock );
1257     }
1258 }
1259
1260 /**
1261  * Resumes the conversion process.
1262  * @param h Handle to hb_handle_t.
1263  */
1264 void hb_resume( hb_handle_t * h )
1265 {
1266     if( h->paused )
1267     {
1268         hb_unlock( h->pause_lock );
1269         h->paused = 0;
1270     }
1271 }
1272
1273 /**
1274  * Stops the conversion process.
1275  * @param h Handle to hb_handle_t.
1276  */
1277 void hb_stop( hb_handle_t * h )
1278 {
1279     h->work_die = 1;
1280
1281     h->job_count = hb_count(h);
1282     h->job_count_permanent = 0;
1283
1284     hb_resume( h );
1285 }
1286
1287 /**
1288  * Returns the state of the conversion process.
1289  * @param h Handle to hb_handle_t.
1290  * @param s Handle to hb_state_t which to copy the state data.
1291  */
1292 void hb_get_state( hb_handle_t * h, hb_state_t * s )
1293 {
1294     hb_lock( h->state_lock );
1295
1296     memcpy( s, &h->state, sizeof( hb_state_t ) );
1297     if ( h->state.state == HB_STATE_SCANDONE || h->state.state == HB_STATE_WORKDONE )
1298         h->state.state = HB_STATE_IDLE;
1299
1300     hb_unlock( h->state_lock );
1301 }
1302
1303 void hb_get_state2( hb_handle_t * h, hb_state_t * s )
1304 {
1305     hb_lock( h->state_lock );
1306
1307     memcpy( s, &h->state, sizeof( hb_state_t ) );
1308
1309     hb_unlock( h->state_lock );
1310 }
1311
1312 /**
1313  * Called in MacGui in UpdateUI to check
1314  *  for a new scan being completed to set a new source
1315  */
1316 int hb_get_scancount( hb_handle_t * h)
1317  {
1318      return h->scanCount;
1319  }
1320
1321 /**
1322  * Closes access to libhb by freeing the hb_handle_t handle ontained in hb_init.
1323  * @param _h Pointer to handle to hb_handle_t.
1324  */
1325 void hb_close( hb_handle_t ** _h )
1326 {
1327     hb_handle_t * h = *_h;
1328     hb_title_t * title;
1329
1330     h->die = 1;
1331     hb_thread_close( &h->main_thread );
1332
1333     while( ( title = hb_list_item( h->list_title, 0 ) ) )
1334     {
1335         hb_list_rem( h->list_title, title );
1336         if( title->job && title->job->filters )
1337         {
1338             hb_list_close( &title->job->filters );
1339         }
1340         free( title->job );
1341         hb_title_close( &title );
1342     }
1343     hb_list_close( &h->list_title );
1344
1345     hb_list_close( &h->jobs );
1346     hb_lock_close( &h->state_lock );
1347     hb_lock_close( &h->pause_lock );
1348     free( h );
1349     *_h = NULL;
1350
1351 }
1352
1353 /**
1354  * Monitors the state of the update, scan, and work threads.
1355  * Sets scan done state when scan thread exits.
1356  * Sets work done state when work thread exits.
1357  * @param _h Handle to hb_handle_t
1358  */
1359 static void thread_func( void * _h )
1360 {
1361     hb_handle_t * h = (hb_handle_t *) _h;
1362     char dirname[1024];
1363     DIR * dir;
1364     struct dirent * entry;
1365
1366     h->pid = getpid();
1367
1368     /* Create folder for temporary files */
1369     memset( dirname, 0, 1024 );
1370     hb_get_tempory_directory( h, dirname );
1371
1372     hb_mkdir( dirname );
1373
1374     while( !h->die )
1375     {
1376         /* In case the check_update thread hangs, it'll die sooner or
1377            later. Then, we join it here */
1378         if( h->update_thread &&
1379             hb_thread_has_exited( h->update_thread ) )
1380         {
1381             hb_thread_close( &h->update_thread );
1382         }
1383
1384         /* Check if the scan thread is done */
1385         if( h->scan_thread &&
1386             hb_thread_has_exited( h->scan_thread ) )
1387         {
1388             hb_thread_close( &h->scan_thread );
1389
1390             hb_log( "libhb: scan thread found %d valid title(s)",
1391                     hb_list_count( h->list_title ) );
1392             hb_lock( h->state_lock );
1393             h->state.state = HB_STATE_SCANDONE; //originally state.state
1394                         hb_unlock( h->state_lock );
1395                         /*we increment this sessions scan count by one for the MacGui
1396                         to trigger a new source being set */
1397             h->scanCount++;
1398         }
1399
1400         /* Check if the work thread is done */
1401         if( h->work_thread &&
1402             hb_thread_has_exited( h->work_thread ) )
1403         {
1404             hb_thread_close( &h->work_thread );
1405
1406             hb_log( "libhb: work result = %d",
1407                     h->work_error );
1408             hb_lock( h->state_lock );
1409             h->state.state                = HB_STATE_WORKDONE;
1410             h->state.param.workdone.error = h->work_error;
1411
1412             h->job_count = hb_count(h);
1413             if (h->job_count < 1)
1414                 h->job_count_permanent = 0;
1415             hb_unlock( h->state_lock );
1416         }
1417
1418         hb_snooze( 50 );
1419     }
1420
1421     if( h->work_thread )
1422     {
1423         hb_stop( h );
1424         hb_thread_close( &h->work_thread );
1425     }
1426
1427     /* Remove temp folder */
1428     dir = opendir( dirname );
1429     if (dir)
1430     {
1431         while( ( entry = readdir( dir ) ) )
1432         {
1433             char filename[1024];
1434             if( entry->d_name[0] == '.' )
1435             {
1436                 continue;
1437             }
1438             memset( filename, 0, 1024 );
1439             snprintf( filename, 1023, "%s/%s", dirname, entry->d_name );
1440             unlink( filename );
1441         }
1442         closedir( dir );
1443         rmdir( dirname );
1444     }
1445 }
1446
1447 /**
1448  * Returns the PID.
1449  * @param h Handle to hb_handle_t
1450  */
1451 int hb_get_pid( hb_handle_t * h )
1452 {
1453     return h->pid;
1454 }
1455
1456 /**
1457  * Sets the current state.
1458  * @param h Handle to hb_handle_t
1459  * @param s Handle to new hb_state_t
1460  */
1461 void hb_set_state( hb_handle_t * h, hb_state_t * s )
1462 {
1463     hb_lock( h->pause_lock );
1464     hb_lock( h->state_lock );
1465     memcpy( &h->state, s, sizeof( hb_state_t ) );
1466     if( h->state.state == HB_STATE_WORKING )
1467     {
1468         /* XXX Hack */
1469         if (h->job_count < 1)
1470             h->job_count_permanent = 1;
1471
1472         h->state.param.working.job_cur =
1473             h->job_count_permanent - hb_list_count( h->jobs );
1474         h->state.param.working.job_count = h->job_count_permanent;
1475
1476         // Set which job is being worked on
1477         if (h->current_job)
1478             h->state.param.working.sequence_id = h->current_job->sequence_id;
1479         else
1480             h->state.param.working.sequence_id = 0;
1481     }
1482     hb_unlock( h->state_lock );
1483     hb_unlock( h->pause_lock );
1484 }
1485
1486 /* Passes a pointer to persistent data */
1487 hb_interjob_t * hb_interjob_get( hb_handle_t * h )
1488 {
1489     return h->interjob;
1490 }