x264.c 36 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
 *****************************************************************************
 * 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.
 *****************************************************************************/

Loren Merritt's avatar
Loren Merritt committed
24 25 26
#define _LARGEFILE_SOURCE
#define _FILE_OFFSET_BITS 64

Laurent Aimar's avatar
Laurent Aimar committed
27 28 29 30 31 32 33 34 35 36 37 38 39 40
#include <stdlib.h>
#include <stdio.h>
#include <string.h>
#include <math.h>

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

#ifdef _MSC_VER
#include <io.h>     /* _setmode() */
#include <fcntl.h>  /* _O_BINARY */
#endif

Loren Merritt's avatar
Loren Merritt committed
41 42 43 44
#ifndef _MSC_VER
#include "config.h"
#endif

45
#include "common/common.h"
46
#include "x264.h"
47
#include "muxers.h"
48

49 50
uint8_t *mux_buffer = NULL;
int mux_buffer_size = 0;
Laurent Aimar's avatar
Laurent Aimar committed
51 52

/* Ctrl-C handler */
Loren Merritt's avatar
Loren Merritt committed
53 54
static int     b_ctrl_c = 0;
static int     b_exit_on_ctrl_c = 0;
Laurent Aimar's avatar
Laurent Aimar committed
55 56
static void    SigIntHandler( int a )
{
Loren Merritt's avatar
Loren Merritt committed
57 58 59
    if( b_exit_on_ctrl_c )
        exit(0);
    b_ctrl_c = 1;
Laurent Aimar's avatar
Laurent Aimar committed
60 61
}

Loren Merritt's avatar
Loren Merritt committed
62 63 64 65 66
typedef struct {
    int b_progress;
    int i_seek;
    hnd_t hin;
    hnd_t hout;
67
    FILE *qpfile;
Loren Merritt's avatar
Loren Merritt committed
68 69
} cli_opt_t;

70
/* input file operation function pointers */
71 72 73 74
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 );
75

Loren Merritt's avatar
Loren Merritt committed
76 77 78 79 80 81 82
/* 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 );

83
static void Help( x264_param_t *defaults, int b_longhelp );
Loren Merritt's avatar
Loren Merritt committed
84 85
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
86

87

Laurent Aimar's avatar
Laurent Aimar committed
88 89 90 91 92 93
/****************************************************************************
 * main:
 ****************************************************************************/
int main( int argc, char **argv )
{
    x264_param_t param;
Loren Merritt's avatar
Loren Merritt committed
94
    cli_opt_t opt;
95 96 97 98 99 100
    int ret;

#ifdef PTW32_STATIC_LIB
    pthread_win32_process_attach_np();
    pthread_win32_thread_attach_np();
#endif
Laurent Aimar's avatar
Laurent Aimar committed
101 102

#ifdef _MSC_VER
103
    _setmode(_fileno(stdin), _O_BINARY);
Laurent Aimar's avatar
Laurent Aimar committed
104 105 106 107 108 109
    _setmode(_fileno(stdout), _O_BINARY);
#endif

    x264_param_default( &param );

    /* Parse command line */
Loren Merritt's avatar
Loren Merritt committed
110
    if( Parse( argc, argv, &param, &opt ) < 0 )
Laurent Aimar's avatar
Laurent Aimar committed
111 112 113 114 115
        return -1;

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

116 117 118 119 120 121 122 123
    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
124 125
}

Loren Merritt's avatar
Loren Merritt committed
126
static char const *strtable_lookup( const char * const table[], int index )
127 128 129 130 131
{
    int i = 0; while( table[i] ) i++;
    return ( ( index >= 0 && index < i ) ? table[ index ] : "???" );
}

Laurent Aimar's avatar
Laurent Aimar committed
132 133 134
/*****************************************************************************
 * Help:
 *****************************************************************************/
135
static void Help( x264_param_t *defaults, int b_longhelp )
Laurent Aimar's avatar
Laurent Aimar committed
136
{
137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155
#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
156
#ifdef AVIS_INPUT
157
        "yes",
Loren Merritt's avatar
Loren Merritt committed
158
#else
159
        "no",
Loren Merritt's avatar
Loren Merritt committed
160 161
#endif
#ifdef MP4_OUTPUT
162
        "yes"
Loren Merritt's avatar
Loren Merritt committed
163
#else
164
        "no"
Loren Merritt's avatar
Loren Merritt committed
165
#endif
166 167 168 169 170 171
      );
    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
172 173
    H1( "      --pre-scenecut          Faster, less precise scenecut detection.\n"
        "                                  Required and implied by multi-threading.\n" );
174 175 176 177 178 179
    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 );
180 181
    H1( "      --no-deblock            Disable loop filter\n" );
    H0( "  -f, --deblock <alpha:beta>  Loop filter AlphaC0 and Beta parameters [%d:%d]\n",
182
                                       defaults->i_deblocking_filter_alphac0, defaults->i_deblocking_filter_beta );
183
    H0( "      --interlaced            Enable pure-interlaced mode\n" );
184 185 186 187 188
    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
189
    H0( "      --crf <float>           Quality-based VBR (nominal QP)\n" );
190 191 192 193 194 195 196 197 198 199 200 201 202 203 204 205 206 207 208 209 210 211 212 213 214 215 216 217 218 219
    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 );
    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" );
220
    H0( "  -A, --partitions <string>   Partitions to consider [\"p8x8,b8x8,i8x8,i4x4\"]\n"
221 222 223 224 225 226
        "                                  - 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 ) );
227 228 229 230 231
    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 );
232 233 234 235 236 237 238 239 240
    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"
        "                                  - esa: exhaustive search (slow)\n" );
    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 320 321 322 323
    H1( "      --no-asm                Disable all CPU optimizations\n" );
    H1( "      --visualize             Show MB types overlayed on the encoded video\n" );
    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
324 325 326 327 328 329
}

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

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

341
    /* Default input file driver */
Loren Merritt's avatar
Loren Merritt committed
342
    p_open_infile = open_file_yuv;
343 344
    p_get_frame_total = get_frame_total_yuv;
    p_read_frame = read_frame_yuv;
Loren Merritt's avatar
Loren Merritt committed
345 346 347 348 349 350 351 352
    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;
353

Laurent Aimar's avatar
Laurent Aimar committed
354 355 356
    /* Parse command line options */
    for( ;; )
    {
Loren Merritt's avatar
Loren Merritt committed
357
        int b_error = 0;
358 359 360 361 362 363 364 365 366
        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
367
#define OPT_LONGHELP 263
368

Laurent Aimar's avatar
Laurent Aimar committed
369 370 371
        static struct option long_options[] =
        {
            { "help",    no_argument,       NULL, 'h' },
372 373
            { "longhelp",no_argument,       NULL, OPT_LONGHELP },
            { "version", no_argument,       NULL, 'V' },
Laurent Aimar's avatar
Laurent Aimar committed
374
            { "bitrate", required_argument, NULL, 'B' },
Loren Merritt's avatar
Loren Merritt committed
375
            { "bframes", required_argument, NULL, 'b' },
376 377 378
            { "no-b-adapt", no_argument,    NULL, 0 },
            { "b-bias",  required_argument, NULL, 0 },
            { "b-pyramid", no_argument,     NULL, 0 },
379 380
            { "min-keyint",required_argument,NULL,'i' },
            { "keyint",  required_argument, NULL, 'I' },
381
            { "scenecut",required_argument, NULL, 0 },
Loren Merritt's avatar
Loren Merritt committed
382
            { "pre-scenecut", no_argument,  NULL, 0 },
383
            { "nf",      no_argument,       NULL, 0 },
384
            { "no-deblock", no_argument,    NULL, 0 },
385 386
            { "filter",  required_argument, NULL, 0 },
            { "deblock", required_argument, NULL, 'f' },
387
            { "interlaced", no_argument,    NULL, 0 },
388
            { "no-cabac",no_argument,       NULL, 0 },
Laurent Aimar's avatar
Laurent Aimar committed
389
            { "qp",      required_argument, NULL, 'q' },
390 391 392 393
            { "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
394
            { "ref",     required_argument, NULL, 'r' },
395 396 397
            { "no-asm",  no_argument,       NULL, 0 },
            { "sar",     required_argument, NULL, 0 },
            { "fps",     required_argument, NULL, 0 },
398
            { "frames",  required_argument, NULL, OPT_FRAMES },
Loren Merritt's avatar
Loren Merritt committed
399
            { "seek",    required_argument, NULL, OPT_SEEK },
Laurent Aimar's avatar
Laurent Aimar committed
400
            { "output",  required_argument, NULL, 'o' },
401 402
            { "analyse", required_argument, NULL, 0 },
            { "partitions", required_argument, NULL, 'A' },
403
            { "direct",  required_argument, NULL, 0 },
404
            { "direct-8x8", required_argument, NULL, 0 },
405
            { "weightb", no_argument,       NULL, 'w' },
406 407
            { "me",      required_argument, NULL, 0 },
            { "merange", required_argument, NULL, 0 },
Loren Merritt's avatar
Loren Merritt committed
408 409
            { "mvrange", required_argument, NULL, 0 },
            { "mvrange-thread", required_argument, NULL, 0 },
410
            { "subme",   required_argument, NULL, 'm' },
411 412 413 414
            { "b-rdo",   no_argument,       NULL, 0 },
            { "mixed-refs", no_argument,    NULL, 0 },
            { "no-chroma-me", no_argument,  NULL, 0 },
            { "bime",    no_argument,       NULL, 0 },
415
            { "8x8dct",  no_argument,       NULL, '8' },
Loren Merritt's avatar
Loren Merritt committed
416
            { "trellis", required_argument, NULL, 't' },
417 418
            { "no-fast-pskip", no_argument, NULL, 0 },
            { "no-dct-decimate", no_argument, NULL, 0 },
419 420
            { "deadzone-inter", required_argument, NULL, '0' },
            { "deadzone-intra", required_argument, NULL, '0' },
421 422 423 424 425 426 427 428
            { "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 },
429
            { "pass",    required_argument, NULL, 'p' },
430 431 432 433 434 435
            { "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 },
436
            { "qpfile",  required_argument, NULL, OPT_QPFILE },
437
            { "threads", required_argument, NULL, 0 },
438
            { "thread-input", no_argument,  NULL, OPT_THREAD_INPUT },
Loren Merritt's avatar
Loren Merritt committed
439
            { "non-deterministic", no_argument, NULL, 0 },
440
            { "no-psnr", no_argument,       NULL, 0 },
441
            { "no-ssim", no_argument,       NULL, 0 },
442
            { "quiet",   no_argument,       NULL, OPT_QUIET },
443
            { "verbose", no_argument,       NULL, 'v' },
Loren Merritt's avatar
Loren Merritt committed
444
            { "progress",no_argument,       NULL, OPT_PROGRESS },
Loren Merritt's avatar
Loren Merritt committed
445
            { "visualize",no_argument,      NULL, OPT_VISUALIZE },
446 447 448 449 450 451 452 453 454 455 456 457 458 459 460 461 462 463 464 465 466 467
            { "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
468 469 470
            {0, 0, 0, 0}
        };

471
        int c = getopt_long( argc, argv, "8A:B:b:f:hI:i:m:o:p:q:r:t:Vvw",
472
                             long_options, &long_options_index);
Laurent Aimar's avatar
Laurent Aimar committed
473 474 475 476 477 478 479 480 481

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

        switch( c )
        {
            case 'h':
482 483 484 485 486 487 488 489 490 491 492 493
                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);
494
            case OPT_FRAMES:
495
                param->i_frame_total = atoi( optarg );
Loren Merritt's avatar
Loren Merritt committed
496 497 498
                break;
            case OPT_SEEK:
                opt->i_seek = atoi( optarg );
499
                break;
Loren Merritt's avatar
Loren Merritt committed
500
            case 'o':
501
                if( !strncasecmp(optarg + strlen(optarg) - 4, ".mp4", 4) )
Loren Merritt's avatar
Loren Merritt committed
502
                {
503 504 505 506 507 508 509
#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
510
                    fprintf( stderr, "x264 [error]: not compiled with MP4 output support\n" );
511
                    return -1;
Loren Merritt's avatar
Loren Merritt committed
512
#endif
513
                }
Loren Merritt's avatar
Loren Merritt committed
514 515 516
                else if( !strncasecmp(optarg + strlen(optarg) - 4, ".mkv", 4) )
                {
                    p_open_outfile = open_file_mkv;
517 518 519 520
                    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
521
                }
522
                if( !strcmp(optarg, "-") )
Loren Merritt's avatar
Loren Merritt committed
523 524
                    opt->hout = stdout;
                else if( p_open_outfile( optarg, &opt->hout ) )
Laurent Aimar's avatar
Laurent Aimar committed
525
                {
526
                    fprintf( stderr, "x264 [error]: can't open output file `%s'\n", optarg );
Laurent Aimar's avatar
Laurent Aimar committed
527 528 529
                    return -1;
                }
                break;
530 531 532 533
            case OPT_QPFILE:
                opt->qpfile = fopen( optarg, "r" );
                if( !opt->qpfile )
                {
534
                    fprintf( stderr, "x264 [error]: can't open `%s'\n", optarg );
535 536 537 538 539
                    return -1;
                }
                param->i_scenecut_threshold = -1;
                param->b_bframe_adaptive = 0;
                break;
540 541 542
            case OPT_THREAD_INPUT:
                b_thread_input = 1;
                break;
543 544 545
            case OPT_QUIET:
                param->i_log_level = X264_LOG_NONE;
                break;
546 547 548
            case 'v':
                param->i_log_level = X264_LOG_DEBUG;
                break;
Loren Merritt's avatar
Loren Merritt committed
549
            case OPT_PROGRESS:
Loren Merritt's avatar
Loren Merritt committed
550
                opt->b_progress = 1;
Loren Merritt's avatar
Loren Merritt committed
551
                break;
Loren Merritt's avatar
Loren Merritt committed
552 553 554
            case OPT_VISUALIZE:
#ifdef VISUALIZE
                param->b_visualize = 1;
Loren Merritt's avatar
Loren Merritt committed
555
                b_exit_on_ctrl_c = 1;
Loren Merritt's avatar
Loren Merritt committed
556
#else
557
                fprintf( stderr, "x264 [warning]: not compiled with visualization support\n" );
Loren Merritt's avatar
Loren Merritt committed
558
#endif
Loren Merritt's avatar
Loren Merritt committed
559
                break;
560 561 562 563
            default:
            {
                int i;
                if( long_options_index < 0 )
Loren Merritt's avatar
Loren Merritt committed
564
                {
565 566 567 568 569 570 571 572 573 574 575
                    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
576
                }
577

578
                b_error |= x264_param_parse( param, long_options[long_options_index].name, optarg );
579
            }
Laurent Aimar's avatar
Laurent Aimar committed
580
        }
Loren Merritt's avatar
Loren Merritt committed
581 582 583

        if( b_error )
        {
584 585
            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
586 587
            return -1;
        }
Laurent Aimar's avatar
Laurent Aimar committed
588 589 590
    }

    /* Get the file name */
Loren Merritt's avatar
Loren Merritt committed
591
    if( optind > argc - 1 || !opt->hout )
Laurent Aimar's avatar
Laurent Aimar committed
592
    {
593 594
        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
595 596 597 598
        return -1;
    }
    psz_filename = argv[optind++];

599 600 601 602 603 604 605 606 607 608
    /* 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
609 610 611 612
    {
        if( optind > argc - 1 )
        {
            /* try to parse the file name */
613
            for( psz = psz_filename; *psz; psz++ )
Laurent Aimar's avatar
Laurent Aimar committed
614
            {
615 616
                if( *psz >= '0' && *psz <= '9'
                    && sscanf( psz, "%ux%u", &param->i_width, &param->i_height ) == 2 )
Laurent Aimar's avatar
Laurent Aimar committed
617
                {
618 619
                    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
620 621 622 623 624 625
                    break;
                }
            }
        }
        else
        {
626
            sscanf( argv[optind++], "%ux%u", &param->i_width, &param->i_height );
Laurent Aimar's avatar
Laurent Aimar committed
627
        }
628
    }
629
        
630 631
    if( !(b_avis || b_y4m) && ( !param->i_width || !param->i_height ) )
    {
632
        fprintf( stderr, "x264 [error]: Rawyuv input requires a resolution.\n" );
633
        return -1;
Laurent Aimar's avatar
Laurent Aimar committed
634 635 636 637
    }

    /* open the input */
    {
638 639
        if( b_avis )
        {
640
#ifdef AVIS_INPUT
Loren Merritt's avatar
Loren Merritt committed
641
            p_open_infile = open_file_avis;
642 643
            p_get_frame_total = get_frame_total_avis;
            p_read_frame = read_frame_avis;
Loren Merritt's avatar
Loren Merritt committed
644
            p_close_infile = close_file_avis;
645
#else
646
            fprintf( stderr, "x264 [error]: not compiled with AVIS input support\n" );
647
            return -1;
648
#endif
649
        }
Loren Merritt's avatar
Loren Merritt committed
650 651 652 653 654 655 656 657
        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
658
        if( p_open_infile( psz_filename, &opt->hin, param ) )
659
        {
660
            fprintf( stderr, "x264 [error]: could not open input file '%s'\n", psz_filename );
661 662
            return -1;
        }
Laurent Aimar's avatar
Laurent Aimar committed
663 664
    }

665
#ifdef HAVE_PTHREAD
666
    if( b_thread_input || param->i_threads > 1 )
667 668 669
    {
        if( open_file_thread( NULL, &opt->hin, param ) )
        {
670
            fprintf( stderr, "x264 [warning]: threaded input failed\n" );
671 672 673 674 675 676 677 678 679 680 681
        }
        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
682 683 684
    return 0;
}

685 686 687 688 689 690 691 692 693 694 695 696 697 698 699 700 701 702
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 )
        {
703
            fprintf( stderr, "x264 [error]: can't parse qpfile for frame %d\n", i_frame );
704 705 706 707 708 709 710 711 712
            fclose( opt->qpfile );
            opt->qpfile = NULL;
            pic->i_type = X264_TYPE_AUTO;
            pic->i_qpplus1 = 0;
            break;
        }
    }
}

Laurent Aimar's avatar
Laurent Aimar committed
713 714 715 716
/*****************************************************************************
 * Decode:
 *****************************************************************************/

Loren Merritt's avatar
Loren Merritt committed
717
static int  Encode_frame( x264_t *h, hnd_t hout, x264_picture_t *pic )
718 719 720 721 722 723 724 725
{
    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 )
    {
726
        fprintf( stderr, "x264 [error]: x264_encoder_encode failed\n" );
727 728 729 730 731 732
    }

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

733
        if( mux_buffer_size < nal[i].i_payload * 3/2 + 4 )
734
        {
735 736 737
            mux_buffer_size = nal[i].i_payload * 2 + 4;
            x264_free( mux_buffer );
            mux_buffer = x264_malloc( mux_buffer_size );
738
        }
739 740 741 742

        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 );
743
    }
Loren Merritt's avatar
Loren Merritt committed
744
    if (i_nal)
745
        p_set_eop( hout, &pic_out );
Loren Merritt's avatar
Loren Merritt committed
746

747 748 749
    return i_file;
}

Laurent Aimar's avatar
Laurent Aimar committed
750 751 752
/*****************************************************************************
 * Encode:
 *****************************************************************************/
Loren Merritt's avatar
Loren Merritt committed
753
static int  Encode( x264_param_t *param, cli_opt_t *opt )
Laurent Aimar's avatar
Laurent Aimar committed
754 755 756 757 758 759 760
{
    x264_t *h;
    x264_picture_t pic;

    int     i_frame, i_frame_total;
    int64_t i_start, i_end;
    int64_t i_file;
761
    int     i_frame_size;
Loren Merritt's avatar
Loren Merritt committed
762
    int     i_progress;
Laurent Aimar's avatar
Laurent Aimar committed
763

764
    i_frame_total = p_get_frame_total( opt->hin );
765 766 767 768 769
    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
770 771 772

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

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

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

    i_start = x264_mdate();
791

792
    /* Encode frames */
Loren Merritt's avatar
Loren Merritt committed
793 794
    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
795
    {
796
        if( p_read_frame( &pic, opt->hin, i_frame + opt->i_seek ) )
Laurent Aimar's avatar
Laurent Aimar committed
797 798
            break;

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

801 802 803 804 805 806 807 808 809
        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
810
        i_file += Encode_frame( h, opt->hout, &pic );
Loren Merritt's avatar
Loren Merritt committed
811 812 813 814

        i_frame++;

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

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

Loren Merritt's avatar
Loren Merritt committed
845 846 847 848 849
    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
850 851 852 853 854 855

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

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

    return 0;
}