x264.c 36.7 KB
Newer Older
Laurent Aimar's avatar
Laurent Aimar committed
1
/*****************************************************************************
2
 * x264: h264 encoder testing program.
Laurent Aimar's avatar
Laurent Aimar committed
3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30
 *****************************************************************************
 * Copyright (C) 2003 Laurent Aimar
 * $Id: x264.c,v 1.1 2004/06/03 19:24:12 fenrir Exp $
 *
 * Authors: Laurent Aimar <fenrir@via.ecp.fr>
 *
 * This program is free software; you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation; either version 2 of the License, or
 * (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program; if not, write to the Free Software
 * Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA  02111, USA.
 *****************************************************************************/

#include <stdlib.h>
#include <math.h>

#include <signal.h>
#define _GNU_SOURCE
#include <getopt.h>

31
#include "common/common.h"
32
#include "common/cpu.h"
33 34
#include "x264.h"
#include "muxers.h"
Laurent Aimar's avatar
Laurent Aimar committed
35

Loren Merritt's avatar
Loren Merritt committed
36 37 38 39
#ifndef _MSC_VER
#include "config.h"
#endif

40 41
uint8_t *mux_buffer = NULL;
int mux_buffer_size = 0;
Laurent Aimar's avatar
Laurent Aimar committed
42 43

/* Ctrl-C handler */
Loren Merritt's avatar
Loren Merritt committed
44 45
static int     b_ctrl_c = 0;
static int     b_exit_on_ctrl_c = 0;
Laurent Aimar's avatar
Laurent Aimar committed
46 47
static void    SigIntHandler( int a )
{
Loren Merritt's avatar
Loren Merritt committed
48 49 50
    if( b_exit_on_ctrl_c )
        exit(0);
    b_ctrl_c = 1;
Laurent Aimar's avatar
Laurent Aimar committed
51 52
}

Loren Merritt's avatar
Loren Merritt committed
53 54 55 56 57
typedef struct {
    int b_progress;
    int i_seek;
    hnd_t hin;
    hnd_t hout;
58
    FILE *qpfile;
Loren Merritt's avatar
Loren Merritt committed
59 60
} cli_opt_t;

61
/* input file operation function pointers */
62 63 64 65
int (*p_open_infile)( char *psz_filename, hnd_t *p_handle, x264_param_t *p_param );
int (*p_get_frame_total)( hnd_t handle );
int (*p_read_frame)( x264_picture_t *p_pic, hnd_t handle, int i_frame );
int (*p_close_infile)( hnd_t handle );
66

Loren Merritt's avatar
Loren Merritt committed
67 68 69 70 71 72 73
/* output file operation function pointers */
static int (*p_open_outfile)( char *psz_filename, hnd_t *p_handle );
static int (*p_set_outfile_param)( hnd_t handle, x264_param_t *p_param );
static int (*p_write_nalu)( hnd_t handle, uint8_t *p_nal, int i_size );
static int (*p_set_eop)( hnd_t handle, x264_picture_t *p_picture );
static int (*p_close_outfile)( hnd_t handle );

74
static void Help( x264_param_t *defaults, int b_longhelp );
Loren Merritt's avatar
Loren Merritt committed
75 76
static int  Parse( int argc, char **argv, x264_param_t *param, cli_opt_t *opt );
static int  Encode( x264_param_t *param, cli_opt_t *opt );
Laurent Aimar's avatar
Laurent Aimar committed
77

78

Laurent Aimar's avatar
Laurent Aimar committed
79 80 81 82 83 84
/****************************************************************************
 * main:
 ****************************************************************************/
int main( int argc, char **argv )
{
    x264_param_t param;
Loren Merritt's avatar
Loren Merritt committed
85
    cli_opt_t opt;
86 87 88 89 90 91
    int ret;

#ifdef PTW32_STATIC_LIB
    pthread_win32_process_attach_np();
    pthread_win32_thread_attach_np();
#endif
Laurent Aimar's avatar
Laurent Aimar committed
92 93

#ifdef _MSC_VER
Loren Merritt's avatar
Loren Merritt committed
94
    _setmode(_fileno(stdin), _O_BINARY);
Laurent Aimar's avatar
Laurent Aimar committed
95 96 97 98 99 100
    _setmode(_fileno(stdout), _O_BINARY);
#endif

    x264_param_default( &param );

    /* Parse command line */
Loren Merritt's avatar
Loren Merritt committed
101
    if( Parse( argc, argv, &param, &opt ) < 0 )
Laurent Aimar's avatar
Laurent Aimar committed
102 103 104 105 106
        return -1;

    /* Control-C handler */
    signal( SIGINT, SigIntHandler );

107 108 109 110 111 112 113 114
    ret = Encode( &param, &opt );

#ifdef PTW32_STATIC_LIB
    pthread_win32_thread_detach_np();
    pthread_win32_process_detach_np();
#endif

    return ret;
Laurent Aimar's avatar
Laurent Aimar committed
115 116
}

Loren Merritt's avatar
Loren Merritt committed
117
static char const *strtable_lookup( const char * const table[], int index )
118 119 120 121 122
{
    int i = 0; while( table[i] ) i++;
    return ( ( index >= 0 && index < i ) ? table[ index ] : "???" );
}

Laurent Aimar's avatar
Laurent Aimar committed
123 124 125
/*****************************************************************************
 * Help:
 *****************************************************************************/
126
static void Help( x264_param_t *defaults, int b_longhelp )
Laurent Aimar's avatar
Laurent Aimar committed
127
{
128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146
#define H0 printf
#define H1 if(b_longhelp) printf
    H0( "x264 core:%d%s\n"
        "Syntax: x264 [options] -o outfile infile [widthxheight]\n"
        "\n"
        "Infile can be raw YUV 4:2:0 (in which case resolution is required),\n"
        "  or YUV4MPEG 4:2:0 (*.y4m),\n"
        "  or AVI or Avisynth if compiled with AVIS support (%s).\n"
        "Outfile type is selected by filename:\n"
        " .264 -> Raw bytestream\n"
        " .mkv -> Matroska\n"
        " .mp4 -> MP4 if compiled with GPAC support (%s)\n"
        "\n"
        "Options:\n"
        "\n"
        "  -h, --help                  List the more commonly used options\n"
        "      --longhelp              List all options\n"
        "\n",
        X264_BUILD, X264_VERSION,
Loren Merritt's avatar
Loren Merritt committed
147
#ifdef AVIS_INPUT
148
        "yes",
Loren Merritt's avatar
Loren Merritt committed
149
#else
150
        "no",
Loren Merritt's avatar
Loren Merritt committed
151 152
#endif
#ifdef MP4_OUTPUT
153
        "yes"
Loren Merritt's avatar
Loren Merritt committed
154
#else
155
        "no"
Loren Merritt's avatar
Loren Merritt committed
156
#endif
157 158 159 160 161 162
      );
    H0( "Frame-type options:\n" );
    H0( "\n" );
    H0( "  -I, --keyint <integer>      Maximum GOP size [%d]\n", defaults->i_keyint_max );
    H1( "  -i, --min-keyint <integer>  Minimum GOP size [%d]\n", defaults->i_keyint_min );
    H1( "      --scenecut <integer>    How aggressively to insert extra I-frames [%d]\n", defaults->i_scenecut_threshold );
Loren Merritt's avatar
Loren Merritt committed
163 164
    H1( "      --pre-scenecut          Faster, less precise scenecut detection.\n"
        "                                  Required and implied by multi-threading.\n" );
165 166 167 168 169 170
    H0( "  -b, --bframes <integer>     Number of B-frames between I and P [%d]\n", defaults->i_bframe );
    H1( "      --no-b-adapt            Disable adaptive B-frame decision\n" );
    H1( "      --b-bias <integer>      Influences how often B-frames are used [%d]\n", defaults->i_bframe_bias );
    H0( "      --b-pyramid             Keep some B-frames as references\n" );
    H0( "      --no-cabac              Disable CABAC\n" );
    H0( "  -r, --ref <integer>         Number of reference frames [%d]\n", defaults->i_frame_reference );
171 172
    H1( "      --no-deblock            Disable loop filter\n" );
    H0( "  -f, --deblock <alpha:beta>  Loop filter AlphaC0 and Beta parameters [%d:%d]\n",
173
                                       defaults->i_deblocking_filter_alphac0, defaults->i_deblocking_filter_beta );
174
    H0( "      --interlaced            Enable pure-interlaced mode\n" );
175 176 177 178 179
    H0( "\n" );
    H0( "Ratecontrol:\n" );
    H0( "\n" );
    H0( "  -q, --qp <integer>          Set QP (0=lossless) [%d]\n", defaults->rc.i_qp_constant );
    H0( "  -B, --bitrate <integer>     Set bitrate (kbit/s)\n" );
Loren Merritt's avatar
Loren Merritt committed
180
    H0( "      --crf <float>           Quality-based VBR (nominal QP)\n" );
181 182 183 184 185 186 187 188 189 190
    H1( "      --vbv-maxrate <integer> Max local bitrate (kbit/s) [%d]\n", defaults->rc.i_vbv_max_bitrate );
    H0( "      --vbv-bufsize <integer> Enable CBR and set size of the VBV buffer (kbit) [%d]\n", defaults->rc.i_vbv_buffer_size );
    H1( "      --vbv-init <float>      Initial VBV buffer occupancy [%.1f]\n", defaults->rc.f_vbv_buffer_init );
    H1( "      --qpmin <integer>       Set min QP [%d]\n", defaults->rc.i_qp_min );
    H1( "      --qpmax <integer>       Set max QP [%d]\n", defaults->rc.i_qp_max );
    H1( "      --qpstep <integer>      Set max QP step [%d]\n", defaults->rc.i_qp_step );
    H0( "      --ratetol <float>       Allowed variance of average bitrate [%.1f]\n", defaults->rc.f_rate_tolerance );
    H0( "      --ipratio <float>       QP factor between I and P [%.2f]\n", defaults->rc.f_ip_factor );
    H0( "      --pbratio <float>       QP factor between P and B [%.2f]\n", defaults->rc.f_pb_factor );
    H1( "      --chroma-qp-offset <integer>  QP difference between chroma and luma [%d]\n", defaults->analyse.i_chroma_qp_offset );
191 192 193 194 195 196 197 198
    H0( "      --aq-mode <integer>     How AQ distributes bits [%d]\n"
        "                                  - 0: Disabled\n"
        "                                  - 1: Avoid moving bits between frames\n"
        "                                  - 2: Move bits between frames\n", defaults->rc.i_aq_mode );
    H0( "      --aq-strength <float>   Reduces blocking and blurring in flat and\n"
        "                              textured areas. [%.1f]\n"
        "                                  - 0.5: weak AQ\n"
        "                                  - 1.5: strong AQ\n", defaults->rc.f_aq_strength );
199 200 201 202 203 204 205 206 207 208 209 210 211 212 213 214 215 216 217 218
    H0( "\n" );
    H0( "  -p, --pass <1|2|3>          Enable multipass ratecontrol\n"
        "                                  - 1: First pass, creates stats file\n"
        "                                  - 2: Last pass, does not overwrite stats file\n"
        "                                  - 3: Nth pass, overwrites stats file\n" );
    H0( "      --stats <string>        Filename for 2 pass stats [\"%s\"]\n", defaults->rc.psz_stat_out );
    H1( "      --rceq <string>         Ratecontrol equation [\"%s\"]\n", defaults->rc.psz_rc_eq );
    H0( "      --qcomp <float>         QP curve compression: 0.0 => CBR, 1.0 => CQP [%.2f]\n", defaults->rc.f_qcompress );
    H1( "      --cplxblur <float>      Reduce fluctuations in QP (before curve compression) [%.1f]\n", defaults->rc.f_complexity_blur );
    H1( "      --qblur <float>         Reduce fluctuations in QP (after curve compression) [%.1f]\n", defaults->rc.f_qblur );
    H0( "      --zones <zone0>/<zone1>/...  Tweak the bitrate of some regions of the video\n" );
    H1( "                              Each zone is of the form\n"
        "                                  <start frame>,<end frame>,<option>\n"
        "                                  where <option> is either\n"
        "                                      q=<integer> (force QP)\n"
        "                                  or  b=<float> (bitrate multiplier)\n" );
    H1( "      --qpfile <string>       Force frametypes and QPs\n" );
    H0( "\n" );
    H0( "Analysis:\n" );
    H0( "\n" );
219
    H0( "  -A, --partitions <string>   Partitions to consider [\"p8x8,b8x8,i8x8,i4x4\"]\n"
220 221 222 223 224 225
        "                                  - p8x8, p4x4, b8x8, i8x8, i4x4\n"
        "                                  - none, all\n"
        "                                  (p4x4 requires p8x8. i8x8 requires --8x8dct.)\n" );
    H0( "      --direct <string>       Direct MV prediction mode [\"%s\"]\n"
        "                                  - none, spatial, temporal, auto\n",
                                       strtable_lookup( x264_direct_pred_names, defaults->analyse.i_direct_mv_pred ) );
226 227 228 229 230
    H1( "      --direct-8x8 <-1|0|1>   Direct prediction size [%d]\n"
        "                                  -  0: 4x4\n"
        "                                  -  1: 8x8\n"
        "                                  - -1: smallest possible according to level\n",
                                       defaults->analyse.i_direct_8x8_inference );
231 232 233 234 235 236
    H0( "  -w, --weightb               Weighted prediction for B-frames\n" );
    H0( "      --me <string>           Integer pixel motion estimation method [\"%s\"]\n",
                                       strtable_lookup( x264_motion_est_names, defaults->analyse.i_me_method ) );
    H1( "                                  - dia: diamond search, radius 1 (fast)\n"
        "                                  - hex: hexagonal search, radius 2\n"
        "                                  - umh: uneven multi-hexagon search\n"
237 238
        "                                  - esa: exhaustive search\n"
        "                                  - tesa: hadamard exhaustive search (slow)\n" );
239 240
    else H0( "                                  - dia, hex, umh\n" );
    H0( "      --merange <integer>     Maximum motion vector search range [%d]\n", defaults->analyse.i_me_range );
Loren Merritt's avatar
Loren Merritt committed
241 242
    H1( "      --mvrange <integer>     Maximum motion vector length [-1 (auto)]\n" );
    H1( "      --mvrange-thread <int>  Minimum buffer between threads [-1 (auto)]\n" );
243 244 245 246 247 248 249 250 251 252 253 254 255 256 257
    H0( "  -m, --subme <integer>       Subpixel motion estimation and partition\n"
        "                                  decision quality: 1=fast, 7=best. [%d]\n", defaults->analyse.i_subpel_refine );
    H0( "      --b-rdo                 RD based mode decision for B-frames. Requires subme 6.\n" );
    H0( "      --mixed-refs            Decide references on a per partition basis\n" );
    H1( "      --no-chroma-me          Ignore chroma in motion estimation\n" );
    H1( "      --bime                  Jointly optimize both MVs in B-frames\n" );
    H0( "  -8, --8x8dct                Adaptive spatial transform size\n" );
    H0( "  -t, --trellis <integer>     Trellis RD quantization. Requires CABAC. [%d]\n"
        "                                  - 0: disabled\n"
        "                                  - 1: enabled only on the final encode of a MB\n"
        "                                  - 2: enabled on all mode decisions\n", defaults->analyse.i_trellis );
    H0( "      --no-fast-pskip         Disables early SKIP detection on P-frames\n" );
    H0( "      --no-dct-decimate       Disables coefficient thresholding on P-frames\n" );
    H0( "      --nr <integer>          Noise reduction [%d]\n", defaults->analyse.i_noise_reduction );
    H1( "\n" );
258 259 260
    H1( "      --deadzone-inter <int>  Set the size of the inter luma quantization deadzone [%d]\n", defaults->analyse.i_luma_deadzone[0] );
    H1( "      --deadzone-intra <int>  Set the size of the intra luma quantization deadzone [%d]\n", defaults->analyse.i_luma_deadzone[1] );
    H1( "                                  Deadzones should be in the range 0 - 32.\n" );
261 262 263 264 265 266 267 268 269 270 271 272 273 274 275 276 277 278 279 280 281 282 283 284 285 286 287 288 289 290 291 292 293 294 295 296 297 298 299 300 301 302 303 304 305 306 307 308 309 310 311 312 313 314
    H1( "      --cqm <string>          Preset quant matrices [\"flat\"]\n"
        "                                  - jvt, flat\n" );
    H0( "      --cqmfile <string>      Read custom quant matrices from a JM-compatible file\n" );
    H1( "                                  Overrides any other --cqm* options.\n" );
    H1( "      --cqm4 <list>           Set all 4x4 quant matrices\n"
        "                                  Takes a comma-separated list of 16 integers.\n" );
    H1( "      --cqm8 <list>           Set all 8x8 quant matrices\n"
        "                                  Takes a comma-separated list of 64 integers.\n" );
    H1( "      --cqm4i, --cqm4p, --cqm8i, --cqm8p\n"
        "                              Set both luma and chroma quant matrices\n" );
    H1( "      --cqm4iy, --cqm4ic, --cqm4py, --cqm4pc\n"
        "                              Set individual quant matrices\n" );
    H1( "\n" );
    H1( "Video Usability Info (Annex E):\n" );
    H1( "The VUI settings are not used by the encoder but are merely suggestions to\n" );
    H1( "the playback equipment. See doc/vui.txt for details. Use at your own risk.\n" );
    H1( "\n" );
    H1( "      --overscan <string>     Specify crop overscan setting [\"%s\"]\n"
        "                                  - undef, show, crop\n",
                                       strtable_lookup( x264_overscan_names, defaults->vui.i_overscan ) );
    H1( "      --videoformat <string>  Specify video format [\"%s\"]\n"
        "                                  - component, pal, ntsc, secam, mac, undef\n",
                                       strtable_lookup( x264_vidformat_names, defaults->vui.i_vidformat ) );
    H1( "      --fullrange <string>    Specify full range samples setting [\"%s\"]\n"
        "                                  - off, on\n",
                                       strtable_lookup( x264_fullrange_names, defaults->vui.b_fullrange ) );
    H1( "      --colorprim <string>    Specify color primaries [\"%s\"]\n"
        "                                  - undef, bt709, bt470m, bt470bg\n"
        "                                    smpte170m, smpte240m, film\n",
                                       strtable_lookup( x264_colorprim_names, defaults->vui.i_colorprim ) );
    H1( "      --transfer <string>     Specify transfer characteristics [\"%s\"]\n"
        "                                  - undef, bt709, bt470m, bt470bg, linear,\n"
        "                                    log100, log316, smpte170m, smpte240m\n",
                                       strtable_lookup( x264_transfer_names, defaults->vui.i_transfer ) );
    H1( "      --colormatrix <string>  Specify color matrix setting [\"%s\"]\n"
        "                                  - undef, bt709, fcc, bt470bg\n"
        "                                    smpte170m, smpte240m, GBR, YCgCo\n",
                                       strtable_lookup( x264_colmatrix_names, defaults->vui.i_colmatrix ) );
    H1( "      --chromaloc <integer>   Specify chroma sample location (0 to 5) [%d]\n",
                                       defaults->vui.i_chroma_loc );
    H0( "\n" );
    H0( "Input/Output:\n" );
    H0( "\n" );
    H0( "  -o, --output                Specify output file\n" );
    H0( "      --sar width:height      Specify Sample Aspect Ratio\n" );
    H0( "      --fps <float|rational>  Specify framerate\n" );
    H0( "      --seek <integer>        First frame to encode\n" );
    H0( "      --frames <integer>      Maximum number of frames to encode\n" );
    H0( "      --level <string>        Specify level (as defined by Annex A)\n" );
    H0( "\n" );
    H0( "  -v, --verbose               Print stats for each frame\n" );
    H0( "      --progress              Show a progress indicator while encoding\n" );
    H0( "      --quiet                 Quiet Mode\n" );
    H0( "      --no-psnr               Disable PSNR computation\n" );
315
    H0( "      --no-ssim               Disable SSIM computation\n" );
Loren Merritt's avatar
Loren Merritt committed
316
    H0( "      --threads <integer>     Parallel encoding\n" );
317
    H0( "      --thread-input          Run Avisynth in its own thread\n" );
Loren Merritt's avatar
Loren Merritt committed
318
    H1( "      --non-deterministic     Slightly improve quality of SMP, at the cost of repeatability\n" );
319
    H1( "      --asm <integer>         Override CPU detection\n" );
320 321
    H1( "      --no-asm                Disable all CPU optimizations\n" );
    H1( "      --visualize             Show MB types overlayed on the encoded video\n" );
322
    H1( "      --dump-yuv <string>     Save reconstructed frames\n" );
323 324 325
    H1( "      --sps-id <integer>      Set SPS and PPS id numbers [%d]\n", defaults->i_sps_id );
    H1( "      --aud                   Use access unit delimiters\n" );
    H0( "\n" );
Laurent Aimar's avatar
Laurent Aimar committed
326 327 328 329 330 331
}

/*****************************************************************************
 * Parse:
 *****************************************************************************/
static int  Parse( int argc, char **argv,
Loren Merritt's avatar
Loren Merritt committed
332
                   x264_param_t *param, cli_opt_t *opt )
Laurent Aimar's avatar
Laurent Aimar committed
333 334
{
    char *psz_filename = NULL;
335
    x264_param_t defaults = *param;
336
    char *psz;
337
    int b_avis = 0;
Loren Merritt's avatar
Loren Merritt committed
338
    int b_y4m = 0;
339
    int b_thread_input = 0;
Laurent Aimar's avatar
Laurent Aimar committed
340

Loren Merritt's avatar
Loren Merritt committed
341
    memset( opt, 0, sizeof(cli_opt_t) );
Laurent Aimar's avatar
Laurent Aimar committed
342

343
    /* Default input file driver */
Loren Merritt's avatar
Loren Merritt committed
344
    p_open_infile = open_file_yuv;
345 346
    p_get_frame_total = get_frame_total_yuv;
    p_read_frame = read_frame_yuv;
Loren Merritt's avatar
Loren Merritt committed
347 348 349 350 351 352 353 354
    p_close_infile = close_file_yuv;

    /* Default output file driver */
    p_open_outfile = open_file_bsf;
    p_set_outfile_param = set_param_bsf;
    p_write_nalu = write_nalu_bsf;
    p_set_eop = set_eop_bsf;
    p_close_outfile = close_file_bsf;
355

Laurent Aimar's avatar
Laurent Aimar committed
356 357 358
    /* Parse command line options */
    for( ;; )
    {
Loren Merritt's avatar
Loren Merritt committed
359
        int b_error = 0;
360 361 362 363 364 365 366 367 368
        int long_options_index = -1;

#define OPT_FRAMES 256
#define OPT_SEEK 257
#define OPT_QPFILE 258
#define OPT_THREAD_INPUT 259
#define OPT_QUIET 260
#define OPT_PROGRESS 261
#define OPT_VISUALIZE 262
369
#define OPT_LONGHELP 263
370

Laurent Aimar's avatar
Laurent Aimar committed
371 372 373
        static struct option long_options[] =
        {
            { "help",    no_argument,       NULL, 'h' },
374 375
            { "longhelp",no_argument,       NULL, OPT_LONGHELP },
            { "version", no_argument,       NULL, 'V' },
Laurent Aimar's avatar
Laurent Aimar committed
376
            { "bitrate", required_argument, NULL, 'B' },
Loren Merritt's avatar
Loren Merritt committed
377
            { "bframes", required_argument, NULL, 'b' },
378 379 380
            { "no-b-adapt", no_argument,    NULL, 0 },
            { "b-bias",  required_argument, NULL, 0 },
            { "b-pyramid", no_argument,     NULL, 0 },
381 382
            { "min-keyint",required_argument,NULL,'i' },
            { "keyint",  required_argument, NULL, 'I' },
383
            { "scenecut",required_argument, NULL, 0 },
Loren Merritt's avatar
Loren Merritt committed
384
            { "pre-scenecut", no_argument,  NULL, 0 },
385
            { "nf",      no_argument,       NULL, 0 },
Loren Merritt's avatar
Loren Merritt committed
386
            { "no-deblock", no_argument,    NULL, 0 },
387 388
            { "filter",  required_argument, NULL, 0 },
            { "deblock", required_argument, NULL, 'f' },
389
            { "interlaced", no_argument,    NULL, 0 },
390
            { "no-cabac",no_argument,       NULL, 0 },
Laurent Aimar's avatar
Laurent Aimar committed
391
            { "qp",      required_argument, NULL, 'q' },
392 393 394 395
            { "qpmin",   required_argument, NULL, 0 },
            { "qpmax",   required_argument, NULL, 0 },
            { "qpstep",  required_argument, NULL, 0 },
            { "crf",     required_argument, NULL, 0 },
Laurent Aimar's avatar
Laurent Aimar committed
396
            { "ref",     required_argument, NULL, 'r' },
397
            { "asm",     required_argument, NULL, 0 },
398 399 400
            { "no-asm",  no_argument,       NULL, 0 },
            { "sar",     required_argument, NULL, 0 },
            { "fps",     required_argument, NULL, 0 },
401
            { "frames",  required_argument, NULL, OPT_FRAMES },
Loren Merritt's avatar
Loren Merritt committed
402
            { "seek",    required_argument, NULL, OPT_SEEK },
Laurent Aimar's avatar
Laurent Aimar committed
403
            { "output",  required_argument, NULL, 'o' },
404 405
            { "analyse", required_argument, NULL, 0 },
            { "partitions", required_argument, NULL, 'A' },
406
            { "direct",  required_argument, NULL, 0 },
407
            { "direct-8x8", required_argument, NULL, 0 },
408
            { "weightb", no_argument,       NULL, 'w' },
409 410
            { "me",      required_argument, NULL, 0 },
            { "merange", required_argument, NULL, 0 },
Loren Merritt's avatar
Loren Merritt committed
411 412
            { "mvrange", required_argument, NULL, 0 },
            { "mvrange-thread", required_argument, NULL, 0 },
413
            { "subme",   required_argument, NULL, 'm' },
414 415 416 417
            { "b-rdo",   no_argument,       NULL, 0 },
            { "mixed-refs", no_argument,    NULL, 0 },
            { "no-chroma-me", no_argument,  NULL, 0 },
            { "bime",    no_argument,       NULL, 0 },
418
            { "8x8dct",  no_argument,       NULL, '8' },
Loren Merritt's avatar
Loren Merritt committed
419
            { "trellis", required_argument, NULL, 't' },
420 421
            { "no-fast-pskip", no_argument, NULL, 0 },
            { "no-dct-decimate", no_argument, NULL, 0 },
422 423
            { "aq-strength", required_argument, NULL, 0 },
            { "aq-mode", required_argument, NULL, 0 },
424 425
            { "deadzone-inter", required_argument, NULL, '0' },
            { "deadzone-intra", required_argument, NULL, '0' },
426 427 428 429 430 431 432 433
            { "level",   required_argument, NULL, 0 },
            { "ratetol", required_argument, NULL, 0 },
            { "vbv-maxrate", required_argument, NULL, 0 },
            { "vbv-bufsize", required_argument, NULL, 0 },
            { "vbv-init", required_argument,NULL,  0 },
            { "ipratio", required_argument, NULL, 0 },
            { "pbratio", required_argument, NULL, 0 },
            { "chroma-qp-offset", required_argument, NULL, 0 },
434
            { "pass",    required_argument, NULL, 'p' },
435 436 437 438 439 440
            { "stats",   required_argument, NULL, 0 },
            { "rceq",    required_argument, NULL, 0 },
            { "qcomp",   required_argument, NULL, 0 },
            { "qblur",   required_argument, NULL, 0 },
            { "cplxblur",required_argument, NULL, 0 },
            { "zones",   required_argument, NULL, 0 },
441
            { "qpfile",  required_argument, NULL, OPT_QPFILE },
442
            { "threads", required_argument, NULL, 0 },
443
            { "thread-input", no_argument,  NULL, OPT_THREAD_INPUT },
Loren Merritt's avatar
Loren Merritt committed
444
            { "non-deterministic", no_argument, NULL, 0 },
445
            { "no-psnr", no_argument,       NULL, 0 },
446
            { "no-ssim", no_argument,       NULL, 0 },
447
            { "quiet",   no_argument,       NULL, OPT_QUIET },
448
            { "verbose", no_argument,       NULL, 'v' },
Loren Merritt's avatar
Loren Merritt committed
449
            { "progress",no_argument,       NULL, OPT_PROGRESS },
Loren Merritt's avatar
Loren Merritt committed
450
            { "visualize",no_argument,      NULL, OPT_VISUALIZE },
451
            { "dump-yuv",required_argument, NULL, 0 },
452 453 454 455 456 457 458 459 460 461 462 463 464 465 466 467 468 469 470 471 472 473
            { "sps-id",  required_argument, NULL, 0 },
            { "aud",     no_argument,       NULL, 0 },
            { "nr",      required_argument, NULL, 0 },
            { "cqm",     required_argument, NULL, 0 },
            { "cqmfile", required_argument, NULL, 0 },
            { "cqm4",    required_argument, NULL, 0 },
            { "cqm4i",   required_argument, NULL, 0 },
            { "cqm4iy",  required_argument, NULL, 0 },
            { "cqm4ic",  required_argument, NULL, 0 },
            { "cqm4p",   required_argument, NULL, 0 },
            { "cqm4py",  required_argument, NULL, 0 },
            { "cqm4pc",  required_argument, NULL, 0 },
            { "cqm8",    required_argument, NULL, 0 },
            { "cqm8i",   required_argument, NULL, 0 },
            { "cqm8p",   required_argument, NULL, 0 },
            { "overscan", required_argument, NULL, 0 },
            { "videoformat", required_argument, NULL, 0 },
            { "fullrange", required_argument, NULL, 0 },
            { "colorprim", required_argument, NULL, 0 },
            { "transfer", required_argument, NULL, 0 },
            { "colormatrix", required_argument, NULL, 0 },
            { "chromaloc", required_argument, NULL, 0 },
Laurent Aimar's avatar
Laurent Aimar committed
474 475 476
            {0, 0, 0, 0}
        };

477
        int c = getopt_long( argc, argv, "8A:B:b:f:hI:i:m:o:p:q:r:t:Vvw",
478
                             long_options, &long_options_index);
Laurent Aimar's avatar
Laurent Aimar committed
479 480 481 482 483 484 485 486 487

        if( c == -1 )
        {
            break;
        }

        switch( c )
        {
            case 'h':
488 489 490 491 492 493 494 495 496 497 498 499
                Help( &defaults, 0 );
                exit(0);
            case OPT_LONGHELP:
                Help( &defaults, 1 );
                exit(0);
            case 'V':
#ifdef X264_POINTVER
                printf( "x264 "X264_POINTVER"\n" );
#else
                printf( "x264 0.%d.X\n", X264_BUILD );
#endif
                exit(0);
500
            case OPT_FRAMES:
501
                param->i_frame_total = atoi( optarg );
Loren Merritt's avatar
Loren Merritt committed
502 503 504
                break;
            case OPT_SEEK:
                opt->i_seek = atoi( optarg );
505
                break;
Loren Merritt's avatar
Loren Merritt committed
506
            case 'o':
507
                if( !strncasecmp(optarg + strlen(optarg) - 4, ".mp4", 4) )
Loren Merritt's avatar
Loren Merritt committed
508
                {
509 510 511 512 513 514 515
#ifdef MP4_OUTPUT
                    p_open_outfile = open_file_mp4;
                    p_write_nalu = write_nalu_mp4;
                    p_set_outfile_param = set_param_mp4;
                    p_set_eop = set_eop_mp4;
                    p_close_outfile = close_file_mp4;
#else
516
                    fprintf( stderr, "x264 [error]: not compiled with MP4 output support\n" );
517
                    return -1;
Loren Merritt's avatar
Loren Merritt committed
518
#endif
519
                }
Loren Merritt's avatar
Loren Merritt committed
520 521 522
                else if( !strncasecmp(optarg + strlen(optarg) - 4, ".mkv", 4) )
                {
                    p_open_outfile = open_file_mkv;
523 524 525 526
                    p_write_nalu = write_nalu_mkv;
                    p_set_outfile_param = set_param_mkv;
                    p_set_eop = set_eop_mkv;
                    p_close_outfile = close_file_mkv;
Loren Merritt's avatar
Loren Merritt committed
527
                }
528
                if( !strcmp(optarg, "-") )
Loren Merritt's avatar
Loren Merritt committed
529 530
                    opt->hout = stdout;
                else if( p_open_outfile( optarg, &opt->hout ) )
Laurent Aimar's avatar
Laurent Aimar committed
531
                {
532
                    fprintf( stderr, "x264 [error]: can't open output file `%s'\n", optarg );
Laurent Aimar's avatar
Laurent Aimar committed
533 534 535
                    return -1;
                }
                break;
536 537 538 539
            case OPT_QPFILE:
                opt->qpfile = fopen( optarg, "r" );
                if( !opt->qpfile )
                {
540
                    fprintf( stderr, "x264 [error]: can't open `%s'\n", optarg );
541 542 543 544 545
                    return -1;
                }
                param->i_scenecut_threshold = -1;
                param->b_bframe_adaptive = 0;
                break;
546 547 548
            case OPT_THREAD_INPUT:
                b_thread_input = 1;
                break;
549 550 551
            case OPT_QUIET:
                param->i_log_level = X264_LOG_NONE;
                break;
552 553 554
            case 'v':
                param->i_log_level = X264_LOG_DEBUG;
                break;
Loren Merritt's avatar
Loren Merritt committed
555
            case OPT_PROGRESS:
Loren Merritt's avatar
Loren Merritt committed
556
                opt->b_progress = 1;
Loren Merritt's avatar
Loren Merritt committed
557
                break;
Loren Merritt's avatar
Loren Merritt committed
558 559 560
            case OPT_VISUALIZE:
#ifdef VISUALIZE
                param->b_visualize = 1;
Loren Merritt's avatar
Loren Merritt committed
561
                b_exit_on_ctrl_c = 1;
Loren Merritt's avatar
Loren Merritt committed
562
#else
563
                fprintf( stderr, "x264 [warning]: not compiled with visualization support\n" );
Loren Merritt's avatar
Loren Merritt committed
564
#endif
Loren Merritt's avatar
Loren Merritt committed
565
                break;
566 567 568 569
            default:
            {
                int i;
                if( long_options_index < 0 )
Loren Merritt's avatar
Loren Merritt committed
570
                {
571 572 573 574 575 576 577 578 579 580 581
                    for( i = 0; long_options[i].name; i++ )
                        if( long_options[i].val == c )
                        {
                            long_options_index = i;
                            break;
                        }
                    if( long_options_index < 0 )
                    {
                        /* getopt_long already printed an error message */
                        return -1;
                    }
Loren Merritt's avatar
Loren Merritt committed
582
                }
583

584
                b_error |= x264_param_parse( param, long_options[long_options_index].name, optarg );
585
            }
Laurent Aimar's avatar
Laurent Aimar committed
586
        }
Loren Merritt's avatar
Loren Merritt committed
587 588 589

        if( b_error )
        {
590 591
            const char *name = long_options_index > 0 ? long_options[long_options_index].name : argv[optind-2];
            fprintf( stderr, "x264 [error]: invalid argument: %s = %s\n", name, optarg );
Loren Merritt's avatar
Loren Merritt committed
592 593
            return -1;
        }
Laurent Aimar's avatar
Laurent Aimar committed
594 595 596
    }

    /* Get the file name */
Loren Merritt's avatar
Loren Merritt committed
597
    if( optind > argc - 1 || !opt->hout )
Laurent Aimar's avatar
Laurent Aimar committed
598
    {
599 600
        fprintf( stderr, "x264 [error]: No %s file. Run x264 --help for a list of options.\n",
                 optind > argc - 1 ? "input" : "output" );
Laurent Aimar's avatar
Laurent Aimar committed
601 602 603 604
        return -1;
    }
    psz_filename = argv[optind++];

605 606 607 608 609 610 611 612 613 614
    /* check demuxer type */
    psz = psz_filename + strlen(psz_filename) - 1;
    while( psz > psz_filename && *psz != '.' )
        psz--;
    if( !strncasecmp( psz, ".avi", 4 ) || !strncasecmp( psz, ".avs", 4 ) )
        b_avis = 1;
    if( !strncasecmp( psz, ".y4m", 4 ) )
        b_y4m = 1;

    if( !(b_avis || b_y4m) ) // raw yuv
Laurent Aimar's avatar
Laurent Aimar committed
615 616 617 618
    {
        if( optind > argc - 1 )
        {
            /* try to parse the file name */
619
            for( psz = psz_filename; *psz; psz++ )
Laurent Aimar's avatar
Laurent Aimar committed
620
            {
621 622
                if( *psz >= '0' && *psz <= '9'
                    && sscanf( psz, "%ux%u", &param->i_width, &param->i_height ) == 2 )
Laurent Aimar's avatar
Laurent Aimar committed
623
                {
624 625
                    if( param->i_log_level >= X264_LOG_INFO )
                        fprintf( stderr, "x264 [info]: file name gives %dx%d\n", param->i_width, param->i_height );
Laurent Aimar's avatar
Laurent Aimar committed
626 627 628 629 630 631
                    break;
                }
            }
        }
        else
        {
632
            sscanf( argv[optind++], "%ux%u", &param->i_width, &param->i_height );
Laurent Aimar's avatar
Laurent Aimar committed
633
        }
634
    }
635
        
636 637
    if( !(b_avis || b_y4m) && ( !param->i_width || !param->i_height ) )
    {
638
        fprintf( stderr, "x264 [error]: Rawyuv input requires a resolution.\n" );
639
        return -1;
Laurent Aimar's avatar
Laurent Aimar committed
640 641 642 643
    }

    /* open the input */
    {
644 645
        if( b_avis )
        {
646
#ifdef AVIS_INPUT
Loren Merritt's avatar
Loren Merritt committed
647
            p_open_infile = open_file_avis;
648 649
            p_get_frame_total = get_frame_total_avis;
            p_read_frame = read_frame_avis;
Loren Merritt's avatar
Loren Merritt committed
650
            p_close_infile = close_file_avis;
651
#else
652
            fprintf( stderr, "x264 [error]: not compiled with AVIS input support\n" );
653
            return -1;
654
#endif
655
        }
Loren Merritt's avatar
Loren Merritt committed
656 657 658 659 660 661 662 663
        if ( b_y4m )
        {
            p_open_infile = open_file_y4m;
            p_get_frame_total = get_frame_total_y4m;
            p_read_frame = read_frame_y4m;
            p_close_infile = close_file_y4m;
        }

Loren Merritt's avatar
Loren Merritt committed
664
        if( p_open_infile( psz_filename, &opt->hin, param ) )
665
        {
666
            fprintf( stderr, "x264 [error]: could not open input file '%s'\n", psz_filename );
667 668
            return -1;
        }
Laurent Aimar's avatar
Laurent Aimar committed
669 670
    }

671
#ifdef HAVE_PTHREAD
672 673
    if( b_thread_input || param->i_threads > 1
        || (param->i_threads == 0 && x264_cpu_num_processors() > 1) )
674 675 676
    {
        if( open_file_thread( NULL, &opt->hin, param ) )
        {
677
            fprintf( stderr, "x264 [warning]: threaded input failed\n" );
678 679 680 681 682 683 684 685 686 687 688
        }
        else
        {
            p_open_infile = open_file_thread;
            p_get_frame_total = get_frame_total_thread;
            p_read_frame = read_frame_thread;
            p_close_infile = close_file_thread;
        }
    }
#endif

Laurent Aimar's avatar
Laurent Aimar committed
689 690 691
    return 0;
}

692 693 694 695 696 697 698 699 700 701 702 703 704 705 706 707 708 709
static void parse_qpfile( cli_opt_t *opt, x264_picture_t *pic, int i_frame )
{
    int num = -1, qp;
    char type;
    while( num < i_frame )
    {
        int ret = fscanf( opt->qpfile, "%d %c %d\n", &num, &type, &qp );
        if( num < i_frame )
            continue;
        pic->i_qpplus1 = qp+1;
        if     ( type == 'I' ) pic->i_type = X264_TYPE_IDR;
        else if( type == 'i' ) pic->i_type = X264_TYPE_I;
        else if( type == 'P' ) pic->i_type = X264_TYPE_P;
        else if( type == 'B' ) pic->i_type = X264_TYPE_BREF;
        else if( type == 'b' ) pic->i_type = X264_TYPE_B;
        else ret = 0;
        if( ret != 3 || qp < 0 || qp > 51 || num > i_frame )
        {
710
            fprintf( stderr, "x264 [error]: can't parse qpfile for frame %d\n", i_frame );
711 712 713 714 715 716 717 718 719
            fclose( opt->qpfile );
            opt->qpfile = NULL;
            pic->i_type = X264_TYPE_AUTO;
            pic->i_qpplus1 = 0;
            break;
        }
    }
}

Laurent Aimar's avatar
Laurent Aimar committed
720
/*****************************************************************************
Loren Merritt's avatar
Loren Merritt committed
721
 * Encode:
Laurent Aimar's avatar
Laurent Aimar committed
722 723
 *****************************************************************************/

Loren Merritt's avatar
Loren Merritt committed
724
static int  Encode_frame( x264_t *h, hnd_t hout, x264_picture_t *pic )
725 726 727 728 729 730 731 732
{
    x264_picture_t pic_out;
    x264_nal_t *nal;
    int i_nal, i;
    int i_file = 0;

    if( x264_encoder_encode( h, &nal, &i_nal, pic, &pic_out ) < 0 )
    {
733
        fprintf( stderr, "x264 [error]: x264_encoder_encode failed\n" );
734 735 736 737 738 739
    }

    for( i = 0; i < i_nal; i++ )
    {
        int i_size;

740
        if( mux_buffer_size < nal[i].i_payload * 3/2 + 4 )
741
        {
742 743 744
            mux_buffer_size = nal[i].i_payload * 2 + 4;
            x264_free( mux_buffer );
            mux_buffer = x264_malloc( mux_buffer_size );
745
        }
746 747 748 749

        i_size = mux_buffer_size;
        x264_nal_encode( mux_buffer, &i_size, 1, &nal[i] );
        i_file += p_write_nalu( hout, mux_buffer, i_size );
750
    }
Loren Merritt's avatar
Loren Merritt committed
751
    if (i_nal)
752
        p_set_eop( hout, &pic_out );
Loren Merritt's avatar
Loren Merritt committed
753

754 755 756
    return i_file;
}

Loren Merritt's avatar
Loren Merritt committed
757
static int  Encode( x264_param_t *param, cli_opt_t *opt )
Laurent Aimar's avatar
Laurent Aimar committed
758 759 760 761 762 763 764
{
    x264_t *h;
    x264_picture_t pic;

    int     i_frame, i_frame_total;
    int64_t i_start, i_end;
    int64_t i_file;
765
    int     i_frame_size;
Loren Merritt's avatar
Loren Merritt committed
766
    int     i_progress;
Laurent Aimar's avatar
Laurent Aimar committed
767

768
    i_frame_total = p_get_frame_total( opt->hin );
769 770 771 772 773
    i_frame_total -= opt->i_seek;
    if( ( i_frame_total == 0 || param->i_frame_total < i_frame_total )
        && param->i_frame_total > 0 )
        i_frame_total = param->i_frame_total;
    param->i_frame_total = i_frame_total;
Laurent Aimar's avatar
Laurent Aimar committed
774 775 776

    if( ( h = x264_encoder_open( param ) ) == NULL )
    {
777
        fprintf( stderr, "x264 [error]: x264_encoder_open failed\n" );
Loren Merritt's avatar
Loren Merritt committed
778
        p_close_infile( opt->hin );
Loren Merritt's avatar
Loren Merritt committed
779 780 781
        return -1;
    }

Loren Merritt's avatar
Loren Merritt committed
782
    if( p_set_outfile_param( opt->hout, param ) )
Loren Merritt's avatar
Loren Merritt committed
783
    {
784
        fprintf( stderr, "x264 [error]: can't set outfile param\n" );
Loren Merritt's avatar
Loren Merritt committed
785 786
        p_close_infile( opt->hin );
        p_close_outfile( opt->hout );
Laurent Aimar's avatar
Laurent Aimar committed
787 788 789 790 791 792 793
        return -1;
    }

    /* Create a new pic */
    x264_picture_alloc( &pic, X264_CSP_I420, param->i_width, param->i_height );

    i_start = x264_mdate();
794

795
    /* Encode frames */
Loren Merritt's avatar
Loren Merritt committed
796 797
    for( i_frame = 0, i_file = 0, i_progress = 0;
         b_ctrl_c == 0 && (i_frame < i_frame_total || i_frame_total == 0); )
Laurent Aimar's avatar
Laurent Aimar committed
798
    {
799
        if( p_read_frame( &pic, opt->hin, i_frame + opt->i_seek ) )
Laurent Aimar's avatar
Laurent Aimar committed
800 801
            break;

Steve Lhomme's avatar
Steve Lhomme committed
802
        pic.i_pts = (int64_t)i_frame * param->i_fps_den;
Loren Merritt's avatar
Loren Merritt committed
803

804 805 806 807 808 809 810 811 812
        if( opt->qpfile )
            parse_qpfile( opt, &pic, i_frame + opt->i_seek );
        else
        {
            /* Do not force any parameters */
            pic.i_type = X264_TYPE_AUTO;
            pic.i_qpplus1 = 0;
        }

Loren Merritt's avatar
Loren Merritt committed
813
        i_file += Encode_frame( h, opt->hout, &pic );
Loren Merritt's avatar
Loren Merritt committed
814 815 816 817

        i_frame++;

        /* update status line (up to 1000 times per input file) */
Loren Merritt's avatar
Loren Merritt committed
818
        if( opt->b_progress && param->i_log_level < X264_LOG_DEBUG && 
819 820
            ( i_frame_total ? i_frame * 1000 / i_frame_total > i_progress
                            : i_frame % 10 == 0 ) )
Loren Merritt's avatar
Loren Merritt committed
821 822 823
        {
            int64_t i_elapsed = x264_mdate() - i_start;
            double fps = i_elapsed > 0 ? i_frame * 1000000. / i_elapsed : 0;
824 825
            if( i_frame_total )
            {
826
                int eta = i_elapsed * (i_frame_total - i_frame) / ((int64_t)i_frame * 1000000);
827
                i_progress = i_frame * 1000 / i_frame_total;
828 829 830
                fprintf( stderr, "encoded frames: %d/%d (%.1f%%), %.2f fps, eta %d:%02d:%02d  \r",
                         i_frame, i_frame_total, (float)i_progress / 10, fps,
                         eta/3600, (eta/60)%60, eta%60 );
831 832 833
            }
            else
                fprintf( stderr, "encoded frames: %d, %.2f fps   \r", i_frame, fps );
834
            fflush( stderr ); // needed in windows
Loren Merritt's avatar
Loren Merritt committed
835
        }
Laurent Aimar's avatar
Laurent Aimar committed
836
    }
837 838 839
    /* Flush delayed B-frames */
    do {
        i_file +=
Loren Merritt's avatar
Loren Merritt committed
840
        i_frame_size = Encode_frame( h, opt->hout, NULL );
841 842
    } while( i_frame_size );

Laurent Aimar's avatar
Laurent Aimar committed
843 844 845 846 847
    i_end = x264_mdate();
    x264_picture_clean( &pic );
    x264_encoder_close( h );
    fprintf( stderr, "\n" );

Loren Merritt's avatar
Loren Merritt committed
848 849 850 851 852
    if( b_ctrl_c )
        fprintf( stderr, "aborted at input frame %d\n", opt->i_seek + i_frame );

    p_close_infile( opt->hin );
    p_close_outfile( opt->hout );
Laurent Aimar's avatar
Laurent Aimar committed
853 854 855 856 857 858

    if( i_frame > 0 )
    {
        double fps = (double)i_frame * (double)1000000 /
                     (double)( i_end - i_start );

Loren Merritt's avatar
Loren Merritt committed
859
        fprintf( stderr, "encoded %d frames, %.2f fps, %.2f kb/s\n", i_frame, fps,
860 861
                 (double) i_file * 8 * param->i_fps_num /
                 ( (double) param->i_fps_den * i_frame * 1000 ) );
Laurent Aimar's avatar
Laurent Aimar committed
862 863 864 865
    }

    return 0;
}