libhb/hb.c

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