libhb/hb.c

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