test/test.c

   1 /* $Id: test.c,v 1.82 2005/11/19 08:25:54 titer Exp $
   2
   3    This file is part of the HandBrake source code.
   4    Homepage: <http://handbrake.m0k.org/>.
   5    It may be used under the terms of the GNU General Public License. */
   6
   7 #include <signal.h>
   8 #include <getopt.h>
   9 #include <sys/time.h>
  10 #include <time.h>
  11 #include <unistd.h>
  12
  13 #include "hb.h"
  14 #include "parsecsv.h"
  15
  16 /* Options */
  17 static int    debug       = HB_DEBUG_NONE;
  18 static int    update      = 0;
  19 static char * input       = NULL;
  20 static char * output      = NULL;
  21 static char * format      = NULL;
  22 static int    titleindex  = 1;
  23 static int    longest_title = 0;
  24 static int    subtitle_scan = 0;
  25 static int    subtitle_force = 0;
  26 static char * native_language = NULL;
  27 static int    twoPass     = 0;
  28 static int    deinterlace           = 0;
  29 static char * deinterlace_opt       = 0;
  30 static int    deblock               = 0;
  31 static char * deblock_opt           = 0;
  32 static int    denoise               = 0;
  33 static char * denoise_opt           = 0;
  34 static int    detelecine            = 0;
  35 static char * detelecine_opt        = 0;
  36 static int    grayscale   = 0;
  37 static int    vcodec      = HB_VCODEC_FFMPEG;
  38 static int    h264_13     = 0;
  39 static int    h264_30     = 0;
  40 static char * audios      = NULL;
  41 static int    audio_mixdown = HB_AMIXDOWN_DOLBYPLII;
  42 static int    sub         = 0;
  43 static int    width       = 0;
  44 static int    height      = 0;
  45 static int    crop[4]     = { -1,-1,-1,-1 };
  46 static int    cpu         = 0;
  47 static int    vrate       = 0;
  48 static int    arate       = 0;
  49 static float  vquality    = -1.0;
  50 static int    vbitrate    = 0;
  51 static int    size        = 0;
  52 static int    abitrate    = 0;
  53 static int    mux         = 0;
  54 static int    acodec      = 0;
  55 static int    pixelratio  = 0;
  56 static int    loosePixelratio = 0;
  57 static int    modulus       = 0;
  58 static int    chapter_start = 0;
  59 static int    chapter_end   = 0;
  60 static int    chapter_markers = 0;
  61 static char * marker_file   = NULL;
  62 static int        crf                   = 0;
  63 static char       *x264opts             = NULL;
  64 static char       *x264opts2    = NULL;
  65 static int        maxHeight             = 0;
  66 static int        maxWidth              = 0;
  67 static int    turbo_opts_enabled = 0;
  68 static char * turbo_opts = "ref=1:subme=1:me=dia:analyse=none:trellis=0:no-fast-pskip=0:8x8dct=0";
  69 static int    largeFileSize = 0;
  70
  71 /* Exit cleanly on Ctrl-C */
  72 static volatile int die = 0;
  73 static void SigHandler( int );
  74
  75 /* Utils */
  76 static void ShowCommands();
  77 static void ShowHelp();
  78 static int  ParseOptions( int argc, char ** argv );
  79 static int  CheckOptions( int argc, char ** argv );
  80 static int  HandleEvents( hb_handle_t * h );
  81
  82 /****************************************************************************
  83  * hb_error_handler
  84  *
  85  * When using the CLI just display using hb_log as we always did in the past
  86  * make sure that we prefix with a nice ERROR message to catch peoples eyes.
  87  ****************************************************************************/
  88 static void hb_cli_error_handler ( const char *errmsg )
  89 {
  90     fprintf( stderr, "ERROR: %s", errmsg );
  91 }
  92
  93 int main( int argc, char ** argv )
  94 {
  95     hb_handle_t * h;
  96     int           build;
  97     char        * version;
  98
  99     /* Parse command line */
 100     if( ParseOptions( argc, argv ) ||
 101         CheckOptions( argc, argv ) )
 102     {
 103         return 1;
 104     }
 105
 106     /* Register our error handler */
 107     hb_register_error_handler(&hb_cli_error_handler);
 108
 109     /* Init libhb */
 110     h = hb_init( debug, update );
 111
 112     /* Show version */
 113     fprintf( stderr, "HandBrake %s (%d) - http://handbrake.m0k.org/\n",
 114              hb_get_version( h ), hb_get_build( h ) );
 115
 116     /* Check for update */
 117     if( update )
 118     {
 119         if( ( build = hb_check_update( h, &version ) ) > -1 )
 120         {
 121             fprintf( stderr, "You are using an old version of "
 122                      "HandBrake.\nLatest is %s (build %d).\n", version,
 123                      build );
 124         }
 125         else
 126         {
 127             fprintf( stderr, "Your version of HandBrake is up to "
 128                      "date.\n" );
 129         }
 130         hb_close( &h );
 131         return 0;
 132     }
 133
 134     /* Geeky */
 135     fprintf( stderr, "%d CPU%s detected\n", hb_get_cpu_count(),
 136              hb_get_cpu_count( h ) > 1 ? "s" : "" );
 137     if( cpu )
 138     {
 139         fprintf( stderr, "Forcing %d CPU%s\n", cpu,
 140                  cpu > 1 ? "s" : "" );
 141         hb_set_cpu_count( h, cpu );
 142     }
 143
 144     /* Exit ASAP on Ctrl-C */
 145     signal( SIGINT, SigHandler );
 146
 147     /* Feed libhb with a DVD to scan */
 148     fprintf( stderr, "Opening %s...\n", input );
 149
 150     if (longest_title) {
 151         /*
 152          * We need to scan for all the titles in order to find the longest
 153          */
 154         titleindex = 0;
 155     }
 156     hb_scan( h, input, titleindex );
 157
 158     /* Wait... */
 159     while( !die )
 160     {
 161 #if !defined(SYS_BEOS)
 162         fd_set         fds;
 163         struct timeval tv;
 164         int            ret;
 165         char           buf[257];
 166
 167         tv.tv_sec  = 0;
 168         tv.tv_usec = 100000;
 169
 170         FD_ZERO( &fds );
 171         FD_SET( STDIN_FILENO, &fds );
 172         ret = select( STDIN_FILENO + 1, &fds, NULL, NULL, &tv );
 173
 174         if( ret > 0 )
 175         {
 176             int size = 0;
 177
 178             while( size < 256 &&
 179                    read( STDIN_FILENO, &buf[size], 1 ) > 0 )
 180             {
 181                 if( buf[size] == '\n' )
 182                 {
 183                     break;
 184                 }
 185                 size++;
 186             }
 187
 188             if( size >= 256 || buf[size] == '\n' )
 189             {
 190                 switch( buf[0] )
 191                 {
 192                     case 'q':
 193                         fprintf( stdout, "\nEncoding Quit by user command\n" );
 194                         die = 1;
 195                         break;
 196                     case 'p':
 197                         fprintf( stdout, "\nEncoding Paused by user command, 'r' to resume\n" );
 198                         hb_pause( h );
 199                         break;
 200                     case 'r':
 201                         hb_resume( h );
 202                         break;
 203                     case 'h':
 204                         ShowCommands();
 205                         break;
 206                 }
 207             }
 208         }
 209         hb_snooze( 200 );
 210 #else
 211         hb_snooze( 200 );
 212 #endif
 213
 214         HandleEvents( h );
 215     }
 216
 217     /* Clean up */
 218     hb_close( &h );
 219     if( input )  free( input );
 220     if( output ) free( output );
 221     if( format ) free( format );
 222     if( audios ) free( audios );
 223     if (native_language ) free (native_language );
 224         if( x264opts ) free (x264opts );
 225         if( x264opts2 ) free (x264opts2 );
 226
 227     fprintf( stderr, "HandBrake has exited.\n" );
 228
 229     return 0;
 230 }
 231
 232 static void ShowCommands()
 233 {
 234     fprintf( stdout, "\nCommands:\n" );
 235     fprintf( stdout, " [h]elp    Show this message\n" );
 236     fprintf( stdout, " [q]uit    Exit HandBrakeCLI\n" );
 237     fprintf( stdout, " [p]ause   Pause encoding\n" );
 238     fprintf( stdout, " [r]esume  Resume encoding\n" );
 239 }
 240
 241 static void PrintTitleInfo( hb_title_t * title )
 242 {
 243     hb_chapter_t  * chapter;
 244     hb_audio_t    * audio;
 245     hb_subtitle_t * subtitle;
 246     int i;
 247
 248     fprintf( stderr, "+ title %d:\n", title->index );
 249     fprintf( stderr, "  + vts %d, ttn %d, cells %d->%d (%d blocks)\n",
 250              title->vts, title->ttn, title->cell_start, title->cell_end,
 251              title->block_count );
 252     fprintf( stderr, "  + duration: %02d:%02d:%02d\n",
 253              title->hours, title->minutes, title->seconds );
 254     fprintf( stderr, "  + size: %dx%d, aspect: %.2f, %.3f fps\n",
 255              title->width, title->height,
 256              (float) title->aspect / HB_ASPECT_BASE,
 257              (float) title->rate / title->rate_base );
 258     fprintf( stderr, "  + autocrop: %d/%d/%d/%d\n", title->crop[0],
 259              title->crop[1], title->crop[2], title->crop[3] );
 260     fprintf( stderr, "  + chapters:\n" );
 261     for( i = 0; i < hb_list_count( title->list_chapter ); i++ )
 262     {
 263         chapter = hb_list_item( title->list_chapter, i );
 264         fprintf( stderr, "    + %d: cells %d->%d, %d blocks, duration "
 265                  "%02d:%02d:%02d\n", chapter->index,
 266                  chapter->cell_start, chapter->cell_end,
 267                  chapter->block_count, chapter->hours, chapter->minutes,
 268                  chapter->seconds );
 269     }
 270     fprintf( stderr, "  + audio tracks:\n" );
 271     for( i = 0; i < hb_list_count( title->list_audio ); i++ )
 272     {
 273         audio = hb_list_item( title->list_audio, i );
 274         if( ( audio->codec & HB_ACODEC_AC3 ) || ( audio->codec & HB_ACODEC_DCA) )
 275         {
 276             fprintf( stderr, "    + %d, %s, %dHz, %dbps\n", i + 1,
 277                      audio->lang, audio->rate, audio->bitrate );
 278         }
 279         else
 280         {
 281             fprintf( stderr, "    + %d, %s\n", i + 1, audio->lang );
 282         }
 283     }
 284     fprintf( stderr, "  + subtitle tracks:\n" );
 285     for( i = 0; i < hb_list_count( title->list_subtitle ); i++ )
 286     {
 287         subtitle = hb_list_item( title->list_subtitle, i );
 288         fprintf( stderr, "    + %d, %s (iso639-2: %s)\n", i + 1, subtitle->lang,
 289             subtitle->iso639_2);
 290     }
 291 }
 292
 293 static int HandleEvents( hb_handle_t * h )
 294 {
 295     hb_state_t s;
 296     hb_get_state( h, &s );
 297     switch( s.state )
 298     {
 299         case HB_STATE_IDLE:
 300             /* Nothing to do */
 301             break;
 302
 303 #define p s.param.scanning
 304         case HB_STATE_SCANNING:
 305             /* Show what title is currently being scanned */
 306             fprintf( stderr, "Scanning title %d", p.title_cur );
 307             if( !titleindex )
 308                 fprintf( stderr, " of %d", p.title_count );
 309             fprintf( stderr, "...\n" );
 310             break;
 311 #undef p
 312
 313         case HB_STATE_SCANDONE:
 314         {
 315             hb_list_t  * list;
 316             hb_title_t * title;
 317             hb_job_t   * job;
 318
 319             list = hb_get_titles( h );
 320
 321             if( !hb_list_count( list ) )
 322             {
 323                 /* No valid title, stop right there */
 324                 fprintf( stderr, "No title found.\n" );
 325                 die = 1;
 326                 break;
 327             }
 328             if( longest_title )
 329             {
 330                 int i;
 331                 int longest_title_idx=0;
 332                 int longest_title_pos=-1;
 333                 int longest_title_time=0;
 334                 int title_time;
 335
 336                 fprintf( stderr, "Searching for longest title...\n" );
 337
 338                 for( i = 0; i < hb_list_count( list ); i++ )
 339                 {
 340                     title = hb_list_item( list, i );
 341                     title_time = (title->hours*60*60 ) + (title->minutes *60) + (title->seconds);
 342                     fprintf( stderr, " + Title (%d) index %d has length %dsec\n",
 343                              i, title->index, title_time );
 344                     if( longest_title_time < title_time )
 345                     {
 346                         longest_title_time = title_time;
 347                         longest_title_pos = i;
 348                         longest_title_idx = title->index;
 349                     }
 350                 }
 351                 if( longest_title_pos == -1 )
 352                 {
 353                     fprintf( stderr, "No longest title found.\n" );
 354                     die = 1;
 355                     break;
 356                 }
 357                 titleindex = longest_title_idx;
 358                 fprintf( stderr, "Found longest title, setting title to %d\n",
 359                          longest_title_idx);
 360
 361                 title = hb_list_item( list, longest_title_pos);
 362             } else {
 363                 title = hb_list_item( list, 0 );
 364             }
 365
 366             if( !titleindex )
 367             {
 368                 /* Scan-only mode, print infos and exit */
 369                 int i;
 370                 for( i = 0; i < hb_list_count( list ); i++ )
 371                 {
 372                     title = hb_list_item( list, i );
 373                     PrintTitleInfo( title );
 374                 }
 375                 die = 1;
 376                 break;
 377             }
 378
 379             /* Set job settings */
 380             job   = title->job;
 381
 382             PrintTitleInfo( title );
 383
 384             if( chapter_start && chapter_end )
 385             {
 386                 job->chapter_start = MAX( job->chapter_start,
 387                                           chapter_start );
 388                 job->chapter_end   = MIN( job->chapter_end,
 389                                           chapter_end );
 390                 job->chapter_end   = MAX( job->chapter_start,
 391                                           job->chapter_end );
 392             }
 393
 394                         if ( chapter_markers )
 395                         {
 396                                 job->chapter_markers = chapter_markers;
 397
 398                 if( marker_file != NULL )
 399                 {
 400                     hb_csv_file_t * file = hb_open_csv_file( marker_file );
 401                     hb_csv_cell_t * cell;
 402                     int row = 0;
 403                     int chapter = 0;
 404
 405                     fprintf( stderr, "Reading chapter markers from file %s\n", marker_file );
 406
 407                     if( file == NULL )
 408                     {
 409                          fprintf( stderr, "Cannot open chapter marker file, using defaults\n" );
 410                     }
 411                     else
 412                     {
 413                         /* Parse the cells */
 414                         while( NULL != ( cell = hb_read_next_cell( file ) ) )
 415                         {
 416                             /* We have a chapter number */
 417                             if( cell->cell_col == 0 )
 418                             {
 419                                 row = cell->cell_row;
 420                                 chapter = atoi( cell->cell_text );
 421                             }
 422
 423                             /* We have a chapter name */
 424                             if( cell->cell_col == 1 && row == cell->cell_row )
 425                             {
 426                                 /* If we have a valid chapter, copy the string an terminate it */
 427                                 if( chapter >= job->chapter_start && chapter <= job->chapter_end )
 428                                 {
 429                                     hb_chapter_t * chapter_s;
 430
 431                                     chapter_s = hb_list_item( job->title->list_chapter, chapter - 1);
 432                                     strncpy(chapter_s->title, cell->cell_text, 1023);
 433                                     chapter_s->title[1023] = '\0';
 434                                 }
 435                             }
 436
 437
 438                             hb_dispose_cell( cell );
 439                         }
 440
 441                         hb_close_csv_file( file );
 442                     }
 443                 }
 444                         }
 445
 446             if( crop[0] >= 0 && crop[1] >= 0 &&
 447                 crop[2] >= 0 && crop[3] >= 0 )
 448             {
 449                 memcpy( job->crop, crop, 4 * sizeof( int ) );
 450             }
 451
 452             job->deinterlace = deinterlace;
 453             job->grayscale   = grayscale;
 454             if (loosePixelratio)
 455             {
 456                 job->pixel_ratio = 2;
 457                 if (modulus)
 458                 {
 459                     job->modulus = modulus;
 460                 }
 461             }
 462             else
 463             {
 464                 job->pixel_ratio = pixelratio;
 465             }
 466             /* Add selected filters */
 467             job->filters = hb_list_init();
 468             if( detelecine )
 469             {
 470                 hb_filter_detelecine.settings = detelecine_opt;
 471                 hb_list_add( job->filters, &hb_filter_detelecine );
 472             }
 473             if( deinterlace )
 474             {
 475                 hb_filter_deinterlace.settings = deinterlace_opt;
 476                 hb_list_add( job->filters, &hb_filter_deinterlace );
 477             }
 478             if( deblock )
 479             {
 480                 hb_filter_deblock.settings = deblock_opt;
 481                 hb_list_add( job->filters, &hb_filter_deblock );
 482             }
 483             if( denoise )
 484             {
 485                 hb_log("Added denoise setting of '%s'", hb_filter_denoise.settings);
 486                 hb_filter_denoise.settings = denoise_opt;
 487                 hb_list_add( job->filters, &hb_filter_denoise );
 488             }
 489
 490             if( width && height )
 491             {
 492                 job->width  = width;
 493                 job->height = height;
 494             }
 495             else if( width )
 496             {
 497                 job->width = width;
 498                 hb_fix_aspect( job, HB_KEEP_WIDTH );
 499             }
 500             else if( height && !loosePixelratio)
 501             {
 502                 job->height = height;
 503                 hb_fix_aspect( job, HB_KEEP_HEIGHT );
 504             }
 505             else if( !width && !height && !pixelratio && !loosePixelratio )
 506             {
 507                 hb_fix_aspect( job, HB_KEEP_WIDTH );
 508             }
 509             else if (!width && loosePixelratio)
 510             {
 511                 /* Default to full width when one isn't specified for loose anamorphic */
 512                 job->width = title->width - job->crop[2] - job->crop[3];
 513                 /* The height will be thrown away in hb.c but calculate it anyway */
 514                 hb_fix_aspect( job, HB_KEEP_WIDTH );
 515             }
 516
 517             if( vquality >= 0.0 && vquality <= 1.0 )
 518             {
 519                 job->vquality = vquality;
 520                 job->vbitrate = 0;
 521             }
 522             else if( vbitrate )
 523             {
 524                 job->vquality = -1.0;
 525                 job->vbitrate = vbitrate;
 526             }
 527             if( vcodec )
 528             {
 529                 job->vcodec = vcodec;
 530             }
 531             if( h264_13 )
 532             {
 533                 job->h264_level = 13;
 534             }
 535                 if( h264_30 )
 536                 {
 537                     job->h264_level = 30;
 538             }
 539             if( vrate )
 540             {
 541                 job->vrate = 27000000;
 542                 job->vrate_base = vrate;
 543             }
 544             if( arate )
 545             {
 546                 job->arate = arate;
 547             }
 548
 549             if( audios )
 550             {
 551                 if( strcasecmp( audios, "none" ) )
 552                 {
 553                     int    audio_count = 0;
 554                     char * tmp         = audios;
 555                     while( *tmp )
 556                     {
 557                         if( *tmp < '0' || *tmp > '9' )
 558                         {
 559                             /* Skip non numeric char */
 560                             tmp++;
 561                             continue;
 562                         }
 563                                                 job->audio_mixdowns[audio_count] = audio_mixdown;
 564                         job->audios[audio_count++] =
 565                             strtol( tmp, &tmp, 0 ) - 1;
 566                     }
 567                     job->audios[audio_count] = -1;
 568                 }
 569                 else
 570                 {
 571                     job->audios[0] = -1;
 572                 }
 573             }
 574                         else
 575                         {
 576                             /* default to the first audio track if none has been specified */
 577                             job->audios[0] = 0;
 578                             job->audio_mixdowns[0] = audio_mixdown;
 579                         }
 580             if( abitrate )
 581             {
 582                 job->abitrate = abitrate;
 583             }
 584             if( acodec )
 585             {
 586                 job->acodec = acodec;
 587             }
 588
 589             if( size )
 590             {
 591                 job->vbitrate = hb_calc_bitrate( job, size );
 592                 fprintf( stderr, "Calculated bitrate: %d kbps\n",
 593                          job->vbitrate );
 594             }
 595
 596             if( sub )
 597             {
 598                 job->subtitle = sub - 1;
 599             }
 600
 601             if( native_language )
 602             {
 603                 job->native_language = strdup( native_language );
 604             }
 605
 606             if( job->mux )
 607             {
 608                 job->mux = mux;
 609             }
 610
 611             if ( largeFileSize )
 612             {
 613                 job->largeFileSize = 1;
 614             }
 615
 616             job->file = strdup( output );
 617
 618             if( crf )
 619             {
 620                 job->crf = 1;
 621             }
 622
 623             if( x264opts != NULL && *x264opts != '\0' )
 624             {
 625                 fprintf( stderr, "Applying the following x264 options: %s\n",
 626                          x264opts);
 627                 job->x264opts = x264opts;
 628             }
 629             else /*avoids a bus error crash when options aren't specified*/
 630             {
 631                 job->x264opts =  NULL;
 632             }
 633             if (maxWidth)
 634                 job->maxWidth = maxWidth;
 635             if (maxHeight)
 636                 job->maxHeight = maxHeight;
 637
 638             if( subtitle_force )
 639             {
 640                 job->subtitle_force = subtitle_force;
 641             }
 642
 643             if( subtitle_scan )
 644             {
 645                 char *x264opts_tmp;
 646
 647                 /*
 648                  * When subtitle scan is enabled do a fast pre-scan job
 649                  * which will determine which subtitles to enable, if any.
 650                  */
 651                 job->pass = -1;
 652
 653                 x264opts_tmp = job->x264opts;
 654
 655                 job->x264opts = NULL;
 656
 657                 job->indepth_scan = subtitle_scan;
 658                 fprintf( stderr, "Subtitle Scan Enabled - enabling "
 659                          "subtitles if found for foreign language segments\n");
 660                 job->select_subtitle = malloc(sizeof(hb_subtitle_t*));
 661                 *(job->select_subtitle) = NULL;
 662
 663                 /*
 664                  * Add the pre-scan job
 665                  */
 666                 hb_add( h, job );
 667
 668                 job->x264opts = x264opts_tmp;
 669             }
 670
 671             if( twoPass )
 672             {
 673                 /*
 674                  * If subtitle_scan is enabled then only turn it on
 675                  * for the first pass and then off again for the
 676                  * second.
 677                  */
 678                 hb_subtitle_t **subtitle_tmp = job->select_subtitle;
 679
 680                 job->select_subtitle = NULL;
 681
 682                 job->pass = 1;
 683
 684                 job->indepth_scan = 0;
 685
 686                 /*
 687                  * If turbo options have been selected then append them
 688                  * to the x264opts now (size includes one ':' and the '\0')
 689                  */
 690                 if( turbo_opts_enabled )
 691                 {
 692                     int size = (x264opts ? strlen(x264opts) : 0) + strlen(turbo_opts) + 2;
 693                     char *tmp_x264opts;
 694
 695                     tmp_x264opts = malloc(size * sizeof(char));
 696                     if( x264opts )
 697                     {
 698                         snprintf( tmp_x264opts, size, "%s:%s",
 699                                   x264opts, turbo_opts );
 700                         free( x264opts );
 701                     } else {
 702                         /*
 703                          * No x264opts to modify, but apply the turbo options
 704                          * anyway as they may be modifying defaults
 705                          */
 706                         snprintf( tmp_x264opts, size, "%s",
 707                                   turbo_opts );
 708                     }
 709                     x264opts = tmp_x264opts;
 710
 711                     fprintf( stderr, "Modified x264 options for pass 1 to append turbo options: %s\n",
 712                              x264opts );
 713
 714                     job->x264opts = x264opts;
 715                 }
 716                 hb_add( h, job );
 717
 718                 job->select_subtitle = subtitle_tmp;
 719
 720                 job->pass = 2;
 721                 /*
 722                  * On the second pass we turn off subtitle scan so that we
 723                  * can actually encode using any subtitles that were auto
 724                  * selected in the first pass (using the whacky select-subtitle
 725                  * attribute of the job).
 726                  */
 727                 job->indepth_scan = 0;
 728
 729                 job->x264opts = x264opts2;
 730
 731                 hb_add( h, job );
 732             }
 733             else
 734             {
 735                 /*
 736                  * Turn on subtitle scan if requested, note that this option
 737                  * precludes encoding of any actual subtitles.
 738                  */
 739
 740                 job->indepth_scan = 0;
 741                 job->pass = 0;
 742                 hb_add( h, job );
 743             }
 744             hb_start( h );
 745             break;
 746         }
 747
 748 #define p s.param.working
 749         case HB_STATE_WORKING:
 750             fprintf( stdout, "\rEncoding: task %d of %d, %.2f %%",
 751                      p.job_cur, p.job_count, 100.0 * p.progress );
 752             if( p.seconds > -1 )
 753             {
 754                 fprintf( stdout, " (%.2f fps, avg %.2f fps, ETA "
 755                          "%02dh%02dm%02ds)", p.rate_cur, p.rate_avg,
 756                          p.hours, p.minutes, p.seconds );
 757             }
 758             fflush(stdout);
 759             break;
 760 #undef p
 761
 762 #define p s.param.muxing
 763         case HB_STATE_MUXING:
 764         {
 765             fprintf( stdout, "\rMuxing: %.2f %%", 100.0 * p.progress );
 766             fflush(stdout);
 767             break;
 768         }
 769 #undef p
 770
 771 #define p s.param.workdone
 772         case HB_STATE_WORKDONE:
 773             /* Print error if any, then exit */
 774             switch( p.error )
 775             {
 776                 case HB_ERROR_NONE:
 777                     fprintf( stderr, "\nRip done!\n" );
 778                     break;
 779                 case HB_ERROR_CANCELED:
 780                     fprintf( stderr, "\nRip canceled.\n" );
 781                     break;
 782                 default:
 783                     fprintf( stderr, "\nRip failed (error %x).\n",
 784                              p.error );
 785             }
 786             die = 1;
 787             break;
 788 #undef p
 789     }
 790     return 0;
 791 }
 792
 793 /****************************************************************************
 794  * SigHandler:
 795  ****************************************************************************/
 796 static volatile int64_t i_die_date = 0;
 797 void SigHandler( int i_signal )
 798 {
 799     if( die == 0 )
 800     {
 801         die = 1;
 802         i_die_date = hb_get_date();
 803         fprintf( stderr, "Signal %d received, terminating - do it "
 804                  "again in case it gets stuck\n", i_signal );
 805     }
 806     else if( i_die_date + 500 < hb_get_date() )
 807     {
 808         fprintf( stderr, "Dying badly, files might remain in your /tmp\n" );
 809         exit( 1 );
 810     }
 811 }
 812
 813 /****************************************************************************
 814  * ShowHelp:
 815  ****************************************************************************/
 816 static void ShowHelp()
 817 {
 818     int i;
 819
 820     fprintf( stderr,
 821     "Syntax: HandBrakeCLI [options] -i <device> -o <file>\n"
 822     "\n"
 823         "### General Handbrake Options------------------------------------------------\n\n"
 824     "    -h, --help              Print help\n"
 825     "    -u, --update            Check for updates and exit\n"
 826     "    -v, --verbose           Be verbose\n"
 827     "    -C, --cpu               Set CPU count (default: autodetected)\n"
 828     "\n"
 829
 830         "### Source Options-----------------------------------------------------------\n\n"
 831         "    -i, --input <string>    Set input device\n"
 832         "    -t, --title <number>    Select a title to encode (0 to scan only,\n"
 833     "                            default: 1)\n"
 834     "    -L, --longest           Select the longest title\n"
 835     "    -c, --chapters <string> Select chapters (e.g. \"1-3\" for chapters\n"
 836     "                            1 to 3, or \"3\" for chapter 3 only,\n"
 837     "                            default: all chapters)\n"
 838         "\n"
 839
 840         "### Destination Options------------------------------------------------------\n\n"
 841     "    -o, --output <string>   Set output file name\n"
 842         "    -f, --format <string>   Set output format (avi/mp4/ogm/mkv, default:\n"
 843     "                            autodetected from file name)\n"
 844     "    -4, --large-file        Use 64-bit mp4 files that can hold more than\n"
 845     "                            4 GB. Note: Breaks iPod, @TV, PS3 compatibility.\n"""
 846     "\n"
 847
 848         "### Picture Settings---------------------------------------------------------\n\n"
 849     "    -w, --width <number>    Set picture width\n"
 850     "    -l, --height <number>   Set picture height\n"
 851     "        --crop <T:B:L:R>    Set cropping values (default: autocrop)\n"
 852         "    -Y, --maxHeight <#>     Set maximum height\n"
 853         "    -X, --maxWidth <#>      Set maximum width\n"
 854         "    -s, --subtitle <number> Select subtitle (default: none)\n"
 855     "    -U, --subtitle-scan     Scan for subtitles in an extra first pass, and choose\n"
 856     "                            the one that's only used 10 percent of the time\n"
 857     "                            or less. This should locate subtitles for short\n"
 858     "                            foreign language segments. Best used in conjunction\n"
 859     "                            with --subtitle-forced.\n"
 860     "    -F, --subtitle-forced   Only display subtitles from the selected stream if\n"
 861     "                            the subtitle has the forced flag set. May be used in\n"
 862     "                            conjunction with --subtitle-scan to auto-select\n"
 863     "                            a stream if it contains forced subtitles.\n"
 864     "    -N, --native-language   Select subtitles with this language if it does not\n"
 865     "          <string>          match the Audio language. Provide the language's\n"
 866     "                            iso639-2 code (fre, eng, spa, dut, et cetera)\n"
 867         "    -m, --markers           Add chapter markers (mp4 output format only)\n"
 868         "\n"
 869
 870         "### Video Options------------------------------------------------------------\n\n"
 871         "    -e, --encoder <string>  Set video library encoder (ffmpeg,xvid,\n"
 872     "                            x264,x264b13,x264b30 default: ffmpeg)\n"
 873         "    -q, --quality <float>   Set video quality (0.0..1.0)\n"
 874         "    -Q, --crf               Use with -q for CRF instead of CQP\n"
 875     "    -S, --size <MB>         Set target size\n"
 876         "    -b, --vb <kb/s>         Set video bitrate (default: 1000)\n"
 877         "    -r, --rate              Set video framerate (" );
 878     for( i = 0; i < hb_video_rates_count; i++ )
 879     {
 880         fprintf( stderr, hb_video_rates[i].string );
 881         if( i != hb_video_rates_count - 1 )
 882             fprintf( stderr, "/" );
 883     }
 884     fprintf( stderr, ")\n"
 885         "\n"
 886         "    -2, --two-pass          Use two-pass mode\n"
 887      "    -d, --deinterlace       Deinterlace video with yadif/mcdeint filter\n"
 888      "          <YM:FD:MM:QP>     (default 0:-1:-1:1)\n"
 889      "           or\n"
 890      "          <fast/slow/slower/slowest>\n"
 891      "    -7, --deblock           Deblock video with pp7 filter\n"
 892      "          <QP:M>            (default 0:2)\n"
 893      "    -8, --denoise           Denoise video with hqdn3d filter\n"
 894      "          <SL:SC:TL:TC>     (default 4:3:6:4.5)\n"
 895      "           or\n"
 896      "          <weak/medium/strong>\n"
 897      "    -9, --detelecine        Detelecine video with pullup filter\n"
 898      "          <L:R:T:B:SB:MP>   (default 1:1:4:4:0:0)\n"
 899     "    -g, --grayscale         Grayscale encoding\n"
 900     "    -p, --pixelratio        Store pixel aspect ratio in video stream\n"
 901     "    -P, --loosePixelratio   Store pixel aspect ratio with specified width\n"
 902     "           <modulus>        Takes as optional argument what number you want\n"
 903     "                            the dimensions to divide cleanly by (default 16)\n"
 904
 905
 906         "\n"
 907
 908
 909         "### Audio Options-----------------------------------------------------------\n\n"
 910         "    -E, --aencoder <string> Set audio encoder (faac/lame/vorbis/ac3, ac3\n"
 911     "                            meaning passthrough, default: guessed)\n"
 912         "    -B, --ab <kb/s>         Set audio bitrate (default: 128)\n"
 913         "    -a, --audio <string>    Select audio channel(s), separated by commas\n"
 914         "                            (\"none\" for no audio, \"1,2,3\" for multiple\n"
 915         "                             tracks, default: first one, max: eight)\n"
 916     "    -6, --mixdown <string>  Format for surround sound downmixing\n"
 917     "                            (mono/stereo/dpl1/dpl2/6ch, default: dpl2)\n"
 918     "    -R, --arate             Set audio samplerate (" );
 919     for( i = 0; i < hb_audio_rates_count; i++ )
 920     {
 921         fprintf( stderr, hb_audio_rates[i].string );
 922         if( i != hb_audio_rates_count - 1 )
 923             fprintf( stderr, "/" );
 924     }
 925     fprintf( stderr, " kHz)\n"
 926
 927
 928
 929         "\n"
 930
 931
 932     "### Advanced H264 Options----------------------------------------------------\n\n"
 933     "    -x, --x264opts <string> Specify advanced x264 options in the\n"
 934     "                            same style as mencoder:\n"
 935     "                            option1=value1:option2=value2\n"
 936     "    -T, --turbo             When using 2-pass use the turbo options\n"
 937     "                            on the first pass to improve speed\n"
 938     "                            (only works with x264, affects PSNR by about 0.05dB,\n"
 939     "                            and increases first pass speed two to four times)\n");
 940 }
 941
 942 /****************************************************************************
 943  * ParseOptions:
 944  ****************************************************************************/
 945 static int ParseOptions( int argc, char ** argv )
 946 {
 947     for( ;; )
 948     {
 949         static struct option long_options[] =
 950           {
 951             { "help",        no_argument,       NULL,    'h' },
 952             { "update",      no_argument,       NULL,    'u' },
 953             { "verbose",     no_argument,       NULL,    'v' },
 954             { "cpu",         required_argument, NULL,    'C' },
 955
 956             { "format",      required_argument, NULL,    'f' },
 957             { "input",       required_argument, NULL,    'i' },
 958             { "output",      required_argument, NULL,    'o' },
 959             { "large-file",  no_argument,       NULL,    '4' },
 960
 961             { "title",       required_argument, NULL,    't' },
 962             { "longest",     no_argument,       NULL,    'L' },
 963             { "chapters",    required_argument, NULL,    'c' },
 964             { "markers",     optional_argument, NULL,    'm' },
 965             { "audio",       required_argument, NULL,    'a' },
 966             { "mixdown",     required_argument, NULL,    '6' },
 967             { "subtitle",    required_argument, NULL,    's' },
 968             { "subtitle-scan", no_argument,     NULL,    'U' },
 969             { "subtitle-forced", no_argument,   NULL,    'F' },
 970             { "native-language", required_argument, NULL,'N' },
 971
 972             { "encoder",     required_argument, NULL,    'e' },
 973             { "aencoder",    required_argument, NULL,    'E' },
 974             { "two-pass",    no_argument,       NULL,    '2' },
 975             { "deinterlace", optional_argument, NULL,    'd' },
 976             { "deblock",     optional_argument, NULL,    '7' },
 977             { "denoise",     optional_argument, NULL,    '8' },
 978             { "detelecine",  optional_argument, NULL,    '9' },
 979             { "grayscale",   no_argument,       NULL,    'g' },
 980             { "pixelratio",  no_argument,       NULL,    'p' },
 981             { "loosePixelratio", optional_argument,   NULL,    'P' },
 982             { "width",       required_argument, NULL,    'w' },
 983             { "height",      required_argument, NULL,    'l' },
 984             { "crop",        required_argument, NULL,    'n' },
 985
 986             { "vb",          required_argument, NULL,    'b' },
 987             { "quality",     required_argument, NULL,    'q' },
 988             { "size",        required_argument, NULL,    'S' },
 989             { "ab",          required_argument, NULL,    'B' },
 990             { "rate",        required_argument, NULL,    'r' },
 991             { "arate",       required_argument, NULL,    'R' },
 992             { "crf",         no_argument,       NULL,    'Q' },
 993             { "x264opts",    required_argument, NULL,    'x' },
 994             { "turbo",       no_argument,       NULL,    'T' },
 995
 996             { "maxHeight",   required_argument, NULL,    'Y' },
 997             { "maxWidth",    required_argument, NULL,    'X' },
 998
 999             { 0, 0, 0, 0 }
1000           };
1001
1002         int option_index = 0;
1003         int c;
1004
1005         c = getopt_long( argc, argv,
1006                          "hvuC:f:4i:o:t:Lc:ma:6:s:UFN:e:E:2d789gpP::w:l:n:b:q:S:B:r:R:Qx:TY:X:",
1007                          long_options, &option_index );
1008         if( c < 0 )
1009         {
1010             break;
1011         }
1012
1013         switch( c )
1014         {
1015             case 'h':
1016                 ShowHelp();
1017                 exit( 0 );
1018             case 'u':
1019                 update = 1;
1020                 break;
1021             case 'v':
1022                 debug = HB_DEBUG_ALL;
1023                 break;
1024             case 'C':
1025                 cpu = atoi( optarg );
1026                 break;
1027
1028             case 'f':
1029                 format = strdup( optarg );
1030                 break;
1031             case 'i':
1032                 input = strdup( optarg );
1033                 break;
1034             case 'o':
1035                 output = strdup( optarg );
1036                 break;
1037             case '4':
1038                 largeFileSize = 1;
1039                 break;
1040             case 't':
1041                 titleindex = atoi( optarg );
1042                 break;
1043             case 'L':
1044                 longest_title = 1;
1045                 break;
1046             case 'c':
1047             {
1048                 int start, end;
1049                 if( sscanf( optarg, "%d-%d", &start, &end ) == 2 )
1050                 {
1051                     chapter_start = start;
1052                     chapter_end   = end;
1053                 }
1054                 else if( sscanf( optarg, "%d", &start ) == 1 )
1055                 {
1056                     chapter_start = start;
1057                     chapter_end   = chapter_start;
1058                 }
1059                 else
1060                 {
1061                     fprintf( stderr, "chapters: invalid syntax (%s)\n",
1062                              optarg );
1063                     return -1;
1064                 }
1065                 break;
1066             }
1067             case 'm':
1068                 if( optarg != NULL )
1069                 {
1070                     marker_file = strdup( optarg );
1071                 }
1072                 chapter_markers = 1;
1073                 break;
1074             case 'a':
1075                 audios = strdup( optarg );
1076                 break;
1077             case '6':
1078                 if( !strcasecmp( optarg, "mono" ) )
1079                 {
1080                     audio_mixdown = HB_AMIXDOWN_MONO;
1081                 }
1082                 else if( !strcasecmp( optarg, "stereo" ) )
1083                 {
1084                     audio_mixdown = HB_AMIXDOWN_STEREO;
1085                 }
1086                 else if( !strcasecmp( optarg, "dpl1" ) )
1087                 {
1088                     audio_mixdown = HB_AMIXDOWN_DOLBY;
1089                 }
1090                 else if( !strcasecmp( optarg, "dpl2" ) )
1091                 {
1092                     audio_mixdown = HB_AMIXDOWN_DOLBYPLII;
1093                 }
1094                 else if( !strcasecmp( optarg, "6ch" ) )
1095                 {
1096                     audio_mixdown = HB_AMIXDOWN_6CH;
1097                 }
1098                 break;
1099             case 's':
1100                 sub = atoi( optarg );
1101                 break;
1102             case 'U':
1103                 subtitle_scan = 1;
1104                 break;
1105             case 'F':
1106                 subtitle_force = 1;
1107                 break;
1108             case 'N':
1109                 native_language = strdup( optarg );
1110                 break;
1111             case '2':
1112                 twoPass = 1;
1113                 break;
1114             case 'd':
1115                 if( optarg != NULL )
1116                 {
1117                     if (!( strcmp( optarg, "fast" ) ))
1118                     {
1119                         deinterlace_opt = "-1";
1120                     }
1121                     else if (!( strcmp( optarg, "slow" ) ))
1122                     {
1123                         deinterlace_opt = "0";
1124                     }
1125                     else if (!( strcmp( optarg, "slower" ) ))
1126                     {
1127                         deinterlace_opt = "2:-1:1";
1128                     }
1129                     else if (!( strcmp( optarg, "slowest" ) ))
1130                     {
1131                         deinterlace_opt = "1:-1:1";
1132                     }
1133                     else
1134                     {
1135                         deinterlace_opt = strdup( optarg );
1136                     }
1137                 }
1138                 deinterlace = 1;
1139                 break;
1140             case '7':
1141                 if( optarg != NULL )
1142                 {
1143                     deblock_opt = strdup( optarg );
1144                 }
1145                 deblock = 1;
1146                 break;
1147             case '8':
1148                 if( optarg != NULL )
1149                 {
1150                     if (!( strcmp( optarg, "weak" ) ))
1151                     {
1152                         denoise_opt = "2:1:2:3";
1153                     }
1154                     else if (!( strcmp( optarg, "medium" ) ))
1155                     {
1156                         denoise_opt = "3:2:2:3";
1157                     }
1158                     else if (!( strcmp( optarg, "strong" ) ))
1159                     {
1160                         denoise_opt = "7:7:5:5";
1161                     }
1162                     else
1163                     {
1164                         denoise_opt = strdup( optarg );
1165                     }
1166                 }
1167                 denoise = 1;
1168                 break;
1169             case '9':
1170                 if( optarg != NULL )
1171                 {
1172                     detelecine_opt = strdup( optarg );
1173                 }
1174                 detelecine = 1;
1175                 break;
1176             case 'g':
1177                 grayscale = 1;
1178                 break;
1179             case 'p':
1180                 pixelratio = 1;
1181                 break;
1182             case 'P':
1183                 loosePixelratio = 1;
1184                 if( optarg != NULL )
1185                 {
1186                     modulus = atoi( optarg );
1187                 }
1188                 break;
1189             case 'e':
1190                 if( !strcasecmp( optarg, "ffmpeg" ) )
1191                 {
1192                     vcodec = HB_VCODEC_FFMPEG;
1193                 }
1194                 else if( !strcasecmp( optarg, "xvid" ) )
1195                 {
1196                     vcodec = HB_VCODEC_XVID;
1197                 }
1198                 else if( !strcasecmp( optarg, "x264" ) )
1199                 {
1200                     vcodec = HB_VCODEC_X264;
1201                 }
1202                 else if( !strcasecmp( optarg, "x264b13" ) )
1203                 {
1204                     vcodec = HB_VCODEC_X264;
1205                     h264_13 = 1;
1206                 }
1207                 else if( !strcasecmp( optarg, "x264b30" ) )
1208                 {
1209                     vcodec = HB_VCODEC_X264;
1210                     h264_30 = 1;
1211                 }
1212                 else
1213                 {
1214                     fprintf( stderr, "invalid codec (%s)\n", optarg );
1215                     return -1;
1216                 }
1217                 break;
1218             case 'E':
1219                 if( !strcasecmp( optarg, "ac3" ) )
1220                 {
1221                     acodec = HB_ACODEC_AC3;
1222                 }
1223                 else if( !strcasecmp( optarg, "lame" ) )
1224                 {
1225                     acodec = HB_ACODEC_LAME;
1226                 }
1227                 else if( !strcasecmp( optarg, "faac" ) )
1228                 {
1229                     acodec = HB_ACODEC_FAAC;
1230                 }
1231                 else if( !strcasecmp( optarg, "vorbis") )
1232                 {
1233                     acodec = HB_ACODEC_VORBIS;
1234                 }
1235                 break;
1236             case 'w':
1237                 width = atoi( optarg );
1238                 break;
1239             case 'l':
1240                 height = atoi( optarg );
1241                 break;
1242             case 'n':
1243             {
1244                 int    i;
1245                 char * tmp = optarg;
1246                 for( i = 0; i < 4; i++ )
1247                 {
1248                     if( !*tmp )
1249                         break;
1250                     crop[i] = strtol( tmp, &tmp, 0 );
1251                     tmp++;
1252                 }
1253                 break;
1254             }
1255             case 'r':
1256             {
1257                 int i;
1258                 vrate = 0;
1259                 for( i = 0; i < hb_video_rates_count; i++ )
1260                 {
1261                     if( !strcmp( optarg, hb_video_rates[i].string ) )
1262                     {
1263                         vrate = hb_video_rates[i].rate;
1264                         break;
1265                     }
1266                 }
1267                 if( !vrate )
1268                 {
1269                     fprintf( stderr, "invalid framerate %s\n", optarg );
1270                 }
1271                 break;
1272             }
1273             case 'R':
1274             {
1275                 int i;
1276                 arate = 0;
1277                 for( i = 0; i < hb_audio_rates_count; i++ )
1278                 {
1279                     if( !strcmp( optarg, hb_audio_rates[i].string ) )
1280                     {
1281                         arate = hb_audio_rates[i].rate;
1282                         break;
1283                     }
1284                 }
1285                 if( !arate )
1286                 {
1287                     fprintf( stderr, "invalid framerate %s\n", optarg );
1288                 }
1289                 break;
1290             }
1291             case 'b':
1292                 vbitrate = atoi( optarg );
1293                 break;
1294             case 'q':
1295                 vquality = atof( optarg );
1296                 break;
1297             case 'S':
1298                 size = atoi( optarg );
1299                 break;
1300             case 'B':
1301                 abitrate = atoi( optarg );
1302                 break;
1303             case 'Q':
1304                 crf = 1;
1305                 break;
1306             case 'x':
1307                 x264opts = strdup( optarg );
1308                 x264opts2 = strdup( optarg );
1309                 break;
1310             case 'T':
1311                 turbo_opts_enabled = 1;
1312                 break;
1313             case 'Y':
1314                 maxHeight = atoi( optarg );
1315                 break;
1316             case 'X':
1317                 maxWidth = atoi (optarg );
1318                 break;
1319
1320             default:
1321                 fprintf( stderr, "unknown option (%s)\n", argv[optind] );
1322                 return -1;
1323         }
1324     }
1325
1326     return 0;
1327 }
1328
1329 static int CheckOptions( int argc, char ** argv )
1330 {
1331     if( update )
1332     {
1333         return 0;
1334     }
1335
1336     if( input == NULL || *input == '\0' )
1337     {
1338         fprintf( stderr, "Missing input device. Run %s --help for "
1339                  "syntax.\n", argv[0] );
1340         return 1;
1341     }
1342
1343     /* Parse format */
1344     if( titleindex > 0 )
1345     {
1346         if( output == NULL || *output == '\0' )
1347         {
1348             fprintf( stderr, "Missing output file name. Run %s --help "
1349                      "for syntax.\n", argv[0] );
1350             return 1;
1351         }
1352
1353         if( !format )
1354         {
1355             char * p = strrchr( output, '.' );
1356
1357             /* autodetect */
1358             if( p && !strcasecmp( p, ".avi" ) )
1359             {
1360                 mux = HB_MUX_AVI;
1361             }
1362             else if( p && ( !strcasecmp( p, ".mp4" )  ||
1363                             !strcasecmp( p, ".m4v" ) ) )
1364             {
1365                 if ( h264_30 == 1 )
1366                     mux = HB_MUX_IPOD;
1367                 else
1368                     mux = HB_MUX_MP4;
1369             }
1370             else if( p && ( !strcasecmp( p, ".ogm" ) ||
1371                             !strcasecmp( p, ".ogg" ) ) )
1372             {
1373                 mux = HB_MUX_OGM;
1374             }
1375             else if( p && !strcasecmp(p, ".mkv" ) )
1376             {
1377                 mux = HB_MUX_MKV;
1378             }
1379             else
1380             {
1381                 fprintf( stderr, "Output format couldn't be guessed "
1382                          "from file name, using default.\n" );
1383                 return 0;
1384             }
1385         }
1386         else if( !strcasecmp( format, "avi" ) )
1387         {
1388             mux = HB_MUX_AVI;
1389         }
1390         else if( !strcasecmp( format, "mp4" ) )
1391         {
1392             if ( h264_30 == 1)
1393                 mux = HB_MUX_IPOD;
1394             else
1395                 mux = HB_MUX_MP4;
1396         }
1397         else if( !strcasecmp( format, "ogm" ) ||
1398                  !strcasecmp( format, "ogg" ) )
1399         {
1400             mux = HB_MUX_OGM;
1401         }
1402         else if( !strcasecmp( format, "mkv" ) )
1403         {
1404             mux = HB_MUX_MKV;
1405         }
1406         else
1407         {
1408             fprintf( stderr, "Invalid output format (%s). Possible "
1409                      "choices are avi, mp4, m4v, ogm, ogg and mkv\n.", format );
1410             return 1;
1411         }
1412
1413         if( !acodec )
1414         {
1415             if( mux == HB_MUX_MP4 || mux == HB_MUX_IPOD )
1416             {
1417                 acodec = HB_ACODEC_FAAC;
1418             }
1419             else if( mux == HB_MUX_AVI )
1420             {
1421                 acodec = HB_ACODEC_LAME;
1422             }
1423             else if( mux == HB_MUX_OGM )
1424             {
1425                 acodec = HB_ACODEC_VORBIS;
1426             }
1427             else if( mux == HB_MUX_MKV )
1428             {
1429                 acodec = HB_ACODEC_AC3;
1430             }
1431         }
1432
1433     }
1434
1435     return 0;
1436 }