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