common.c 29 KB
Newer Older
Laurent Aimar's avatar
Laurent Aimar committed
1 2 3
/*****************************************************************************
 * common.c: h264 library
 *****************************************************************************
4
 * Copyright (C) 2003-2008 x264 project
Laurent Aimar's avatar
Laurent Aimar committed
5
 *
6 7
 * Authors: Loren Merritt <lorenm@u.washington.edu>
 *          Laurent Aimar <fenrir@via.ecp.fr>
Laurent Aimar's avatar
Laurent Aimar committed
8 9 10 11 12 13 14 15 16 17 18 19 20
 *
 * 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
21
 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02111, USA.
Laurent Aimar's avatar
Laurent Aimar committed
22 23
 *****************************************************************************/

24
#include <stdarg.h>
25
#include <ctype.h>
Laurent Aimar's avatar
Laurent Aimar committed
26 27 28 29 30 31 32 33

#ifdef HAVE_MALLOC_H
#include <malloc.h>
#endif

#include "common.h"
#include "cpu.h"

34 35
static void x264_log_default( void *, int, const char *, va_list );

Laurent Aimar's avatar
Laurent Aimar committed
36 37 38 39 40 41 42 43 44 45
/****************************************************************************
 * x264_param_default:
 ****************************************************************************/
void    x264_param_default( x264_param_t *param )
{
    /* */
    memset( param, 0, sizeof( x264_param_t ) );

    /* CPU autodetect */
    param->cpu = x264_cpu_detect();
46
    param->i_threads = X264_THREADS_AUTO;
Loren Merritt's avatar
Loren Merritt committed
47
    param->b_deterministic = 1;
Laurent Aimar's avatar
Laurent Aimar committed
48 49 50 51 52 53 54

    /* Video properties */
    param->i_csp           = X264_CSP_I420;
    param->i_width         = 0;
    param->i_height        = 0;
    param->vui.i_sar_width = 0;
    param->vui.i_sar_height= 0;
55 56 57 58 59 60 61
    param->vui.i_overscan  = 0;  /* undef */
    param->vui.i_vidformat = 5;  /* undef */
    param->vui.b_fullrange = 0;  /* off */
    param->vui.i_colorprim = 2;  /* undef */
    param->vui.i_transfer  = 2;  /* undef */
    param->vui.i_colmatrix = 2;  /* undef */
    param->vui.i_chroma_loc= 0;  /* left center */
62 63
    param->i_fps_num       = 25;
    param->i_fps_den       = 1;
64
    param->i_level_idc     = -1;
Fiona Glaser's avatar
Fiona Glaser committed
65 66 67
    param->i_slice_max_size = 0;
    param->i_slice_max_mbs = 0;
    param->i_slice_count = 0;
Laurent Aimar's avatar
Laurent Aimar committed
68 69

    /* Encoder parameters */
70
    param->i_frame_reference = 3;
71
    param->i_keyint_max = 250;
72
    param->i_keyint_min = 25;
73
    param->i_bframe = 3;
74
    param->i_scenecut_threshold = 40;
75
    param->i_bframe_adaptive = X264_B_ADAPT_FAST;
76
    param->i_bframe_bias = 0;
77
    param->b_bframe_pyramid = 0;
Laurent Aimar's avatar
Laurent Aimar committed
78 79 80 81 82

    param->b_deblocking_filter = 1;
    param->i_deblocking_filter_alphac0 = 0;
    param->i_deblocking_filter_beta = 0;

83
    param->b_cabac = 1;
84
    param->i_cabac_init_idc = 0;
Laurent Aimar's avatar
Laurent Aimar committed
85

86
    param->rc.i_rc_method = X264_RC_CRF;
87
    param->rc.i_bitrate = 0;
Loren Merritt's avatar
Loren Merritt committed
88 89 90 91
    param->rc.f_rate_tolerance = 1.0;
    param->rc.i_vbv_max_bitrate = 0;
    param->rc.i_vbv_buffer_size = 0;
    param->rc.f_vbv_buffer_init = 0.9;
92 93
    param->rc.i_qp_constant = 23;
    param->rc.f_rf_constant = 23;
Loren Merritt's avatar
Loren Merritt committed
94
    param->rc.i_qp_min = 10;
95 96
    param->rc.i_qp_max = 51;
    param->rc.i_qp_step = 4;
Loren Merritt's avatar
Loren Merritt committed
97
    param->rc.f_ip_factor = 1.4;
98
    param->rc.f_pb_factor = 1.3;
99
    param->rc.i_aq_mode = X264_AQ_VARIANCE;
100
    param->rc.f_aq_strength = 1.0;
Fiona Glaser's avatar
Fiona Glaser committed
101
    param->rc.i_lookahead = 40;
102 103 104 105 106 107 108

    param->rc.b_stat_write = 0;
    param->rc.psz_stat_out = "x264_2pass.log";
    param->rc.b_stat_read = 0;
    param->rc.psz_stat_in = "x264_2pass.log";
    param->rc.f_qcompress = 0.6;
    param->rc.f_qblur = 0.5;
Loren Merritt's avatar
Loren Merritt committed
109
    param->rc.f_complexity_blur = 20;
Loren Merritt's avatar
Loren Merritt committed
110
    param->rc.i_zones = 0;
Fiona Glaser's avatar
Fiona Glaser committed
111
    param->rc.b_mb_tree = 1;
Laurent Aimar's avatar
Laurent Aimar committed
112

113 114 115
    /* Log */
    param->pf_log = x264_log_default;
    param->p_log_private = NULL;
116
    param->i_log_level = X264_LOG_INFO;
117 118

    /* */
119
    param->analyse.intra = X264_ANALYSE_I4x4 | X264_ANALYSE_I8x8;
120 121
    param->analyse.inter = X264_ANALYSE_I4x4 | X264_ANALYSE_I8x8
                         | X264_ANALYSE_PSUB16x16 | X264_ANALYSE_BSUB16x16;
122
    param->analyse.i_direct_mv_pred = X264_DIRECT_PRED_SPATIAL;
123
    param->analyse.i_me_method = X264_ME_HEX;
124
    param->analyse.f_psy_rd = 1.0;
Fiona Glaser's avatar
Fiona Glaser committed
125
    param->analyse.b_psy = 1;
126
    param->analyse.f_psy_trellis = 0;
127
    param->analyse.i_me_range = 16;
128 129
    param->analyse.i_subpel_refine = 7;
    param->analyse.b_mixed_references = 1;
Loren Merritt's avatar
Loren Merritt committed
130
    param->analyse.b_chroma_me = 1;
Loren Merritt's avatar
Loren Merritt committed
131
    param->analyse.i_mv_range_thread = -1;
132
    param->analyse.i_mv_range = -1; // set from level_idc
133
    param->analyse.i_chroma_qp_offset = 0;
Loren Merritt's avatar
Loren Merritt committed
134
    param->analyse.b_fast_pskip = 1;
135
    param->analyse.b_weighted_bipred = 1;
136
    param->analyse.b_dct_decimate = 1;
137 138
    param->analyse.b_transform_8x8 = 1;
    param->analyse.i_trellis = 1;
139 140
    param->analyse.i_luma_deadzone[0] = 21;
    param->analyse.i_luma_deadzone[1] = 11;
141 142
    param->analyse.b_psnr = 0;
    param->analyse.b_ssim = 0;
143

Loren Merritt's avatar
Loren Merritt committed
144 145 146 147 148 149 150 151
    param->i_cqm_preset = X264_CQM_FLAT;
    memset( param->cqm_4iy, 16, 16 );
    memset( param->cqm_4ic, 16, 16 );
    memset( param->cqm_4py, 16, 16 );
    memset( param->cqm_4pc, 16, 16 );
    memset( param->cqm_8iy, 16, 64 );
    memset( param->cqm_8py, 16, 64 );

152
    param->b_repeat_headers = 1;
153
    param->b_aud = 0;
Laurent Aimar's avatar
Laurent Aimar committed
154 155
}

156 157 158 159 160 161 162 163 164 165 166 167 168 169 170 171 172 173 174 175 176 177 178 179
static int parse_enum( const char *arg, const char * const *names, int *dst )
{
    int i;
    for( i = 0; names[i]; i++ )
        if( !strcmp( arg, names[i] ) )
        {
            *dst = i;
            return 0;
        }
    return -1;
}

static int parse_cqm( const char *str, uint8_t *cqm, int length )
{
    int i = 0;
    do {
        int coef;
        if( !sscanf( str, "%d", &coef ) || coef < 1 || coef > 255 )
            return -1;
        cqm[i++] = coef;
    } while( i < length && (str = strchr( str, ',' )) && str++ );
    return (i == length) ? 0 : -1;
}

180
static int x264_atobool( const char *str, int *b_error )
181
{
Loren Merritt's avatar
Loren Merritt committed
182 183
    if( !strcmp(str, "1") ||
        !strcmp(str, "true") ||
184 185
        !strcmp(str, "yes") )
        return 1;
Loren Merritt's avatar
Loren Merritt committed
186 187
    if( !strcmp(str, "0") ||
        !strcmp(str, "false") ||
188 189
        !strcmp(str, "no") )
        return 0;
190 191 192 193 194 195 196 197 198 199 200
    *b_error = 1;
    return 0;
}

static int x264_atoi( const char *str, int *b_error )
{
    char *end;
    int v = strtol( str, &end, 0 );
    if( end == str || *end != '\0' )
        *b_error = 1;
    return v;
201 202
}

203 204 205 206 207 208 209 210 211 212 213 214
static double x264_atof( const char *str, int *b_error )
{
    char *end;
    double v = strtod( str, &end );
    if( end == str || *end != '\0' )
        *b_error = 1;
    return v;
}

#define atobool(str) ( name_was_bool = 1, x264_atobool( str, &b_error ) )
#define atoi(str) x264_atoi( str, &b_error )
#define atof(str) x264_atof( str, &b_error )
215 216 217

int x264_param_parse( x264_param_t *p, const char *name, const char *value )
{
218
    char *name_buf = NULL;
219
    int b_error = 0;
220 221
    int name_was_bool;
    int value_was_null = !value;
222 223 224 225 226
    int i;

    if( !name )
        return X264_PARAM_BAD_NAME;
    if( !value )
227
        value = "true";
228 229 230 231

    if( value[0] == '=' )
        value++;

232 233 234 235 236 237 238 239 240
    if( strchr( name, '_' ) ) // s/_/-/g
    {
        char *p;
        name_buf = strdup(name);
        while( (p = strchr( name_buf, '_' )) )
            *p = '-';
        name = name_buf;
    }

241 242 243 244 245 246
    if( (!strncmp( name, "no-", 3 ) && (i = 3)) ||
        (!strncmp( name, "no", 2 ) && (i = 2)) )
    {
        name += i;
        value = atobool(value) ? "false" : "true";
    }
247
    name_was_bool = 0;
248 249

#define OPT(STR) else if( !strcmp( name, STR ) )
250
#define OPT2(STR0, STR1) else if( !strcmp( name, STR0 ) || !strcmp( name, STR1 ) )
251 252
    if(0);
    OPT("asm")
253 254 255 256 257 258
    {
        p->cpu = isdigit(value[0]) ? atoi(value) :
                 !strcmp(value, "auto") || atobool(value) ? x264_cpu_detect() : 0;
        if( b_error )
        {
            char *buf = strdup(value);
259
            char *tok, UNUSED *saveptr, *init;
260 261 262 263 264 265 266 267 268 269 270 271
            b_error = 0;
            p->cpu = 0;
            for( init=buf; (tok=strtok_r(init, ",", &saveptr)); init=NULL )
            {
                for( i=0; x264_cpu_names[i].flags && strcasecmp(tok, x264_cpu_names[i].name); i++ );
                p->cpu |= x264_cpu_names[i].flags;
                if( !x264_cpu_names[i].flags )
                    b_error = 1;
            }
            free( buf );
        }
    }
272
    OPT("threads")
273 274
    {
        if( !strcmp(value, "auto") )
275
            p->i_threads = X264_THREADS_AUTO;
276 277 278
        else
            p->i_threads = atoi(value);
    }
Loren Merritt's avatar
Loren Merritt committed
279 280
    OPT2("deterministic", "n-deterministic")
        p->b_deterministic = atobool(value);
281
    OPT2("level", "level-idc")
282 283 284 285 286 287 288 289 290 291 292 293 294
    {
        if( atof(value) < 6 )
            p->i_level_idc = (int)(10*atof(value)+.5);
        else
            p->i_level_idc = atoi(value);
    }
    OPT("sar")
    {
        b_error = ( 2 != sscanf( value, "%d:%d", &p->vui.i_sar_width, &p->vui.i_sar_height ) &&
                    2 != sscanf( value, "%d/%d", &p->vui.i_sar_width, &p->vui.i_sar_height ) );
    }
    OPT("overscan")
        b_error |= parse_enum( value, x264_overscan_names, &p->vui.i_overscan );
295
    OPT("videoformat")
296 297 298
        b_error |= parse_enum( value, x264_vidformat_names, &p->vui.i_vidformat );
    OPT("fullrange")
        b_error |= parse_enum( value, x264_fullrange_names, &p->vui.b_fullrange );
299
    OPT("colorprim")
300 301 302
        b_error |= parse_enum( value, x264_colorprim_names, &p->vui.i_colorprim );
    OPT("transfer")
        b_error |= parse_enum( value, x264_transfer_names, &p->vui.i_transfer );
303
    OPT("colormatrix")
304 305 306 307 308 309 310 311 312 313
        b_error |= parse_enum( value, x264_colmatrix_names, &p->vui.i_colmatrix );
    OPT("chromaloc")
    {
        p->vui.i_chroma_loc = atoi(value);
        b_error = ( p->vui.i_chroma_loc < 0 || p->vui.i_chroma_loc > 5 );
    }
    OPT("fps")
    {
        if( sscanf( value, "%d/%d", &p->i_fps_num, &p->i_fps_den ) == 2 )
            ;
314
        else
315
        {
316
            float fps = atof(value);
317 318 319 320
            p->i_fps_num = (int)(fps * 1000 + .5);
            p->i_fps_den = 1000;
        }
    }
321
    OPT2("ref", "frameref")
322 323 324 325 326 327 328
        p->i_frame_reference = atoi(value);
    OPT("keyint")
    {
        p->i_keyint_max = atoi(value);
        if( p->i_keyint_min > p->i_keyint_max )
            p->i_keyint_min = p->i_keyint_max;
    }
329
    OPT2("min-keyint", "keyint-min")
330 331 332 333 334 335
    {
        p->i_keyint_min = atoi(value);
        if( p->i_keyint_max < p->i_keyint_min )
            p->i_keyint_max = p->i_keyint_min;
    }
    OPT("scenecut")
Fiona Glaser's avatar
Fiona Glaser committed
336 337 338 339 340 341 342 343
    {
        p->i_scenecut_threshold = atobool(value);
        if( b_error || p->i_scenecut_threshold )
        {
            b_error = 0;
            p->i_scenecut_threshold = atoi(value);
        }
    }
344 345 346
    OPT("bframes")
        p->i_bframe = atoi(value);
    OPT("b-adapt")
347 348 349 350 351 352 353 354
    {
        p->i_bframe_adaptive = atobool(value);
        if( b_error )
        {
            b_error = 0;
            p->i_bframe_adaptive = atoi(value);
        }
    }
355 356 357 358 359
    OPT("b-bias")
        p->i_bframe_bias = atoi(value);
    OPT("b-pyramid")
        p->b_bframe_pyramid = atobool(value);
    OPT("nf")
360
        p->b_deblocking_filter = !atobool(value);
361
    OPT2("filter", "deblock")
362
    {
363 364
        if( 2 == sscanf( value, "%d:%d", &p->i_deblocking_filter_alphac0, &p->i_deblocking_filter_beta ) ||
            2 == sscanf( value, "%d,%d", &p->i_deblocking_filter_alphac0, &p->i_deblocking_filter_beta ) )
365 366
        {
            p->b_deblocking_filter = 1;
367 368 369 370 371
        }
        else if( sscanf( value, "%d", &p->i_deblocking_filter_alphac0 ) )
        {
            p->b_deblocking_filter = 1;
            p->i_deblocking_filter_beta = p->i_deblocking_filter_alphac0;
372 373 374 375
        }
        else
            p->b_deblocking_filter = atobool(value);
    }
Fiona Glaser's avatar
Fiona Glaser committed
376 377 378 379 380 381
    OPT("slice-max-size")
        p->i_slice_max_size = atoi(value);
    OPT("slice-max-mbs")
        p->i_slice_max_mbs = atoi(value);
    OPT("slices")
        p->i_slice_count = atoi(value);
382 383 384 385
    OPT("cabac")
        p->b_cabac = atobool(value);
    OPT("cabac-idc")
        p->i_cabac_init_idc = atoi(value);
386 387
    OPT("interlaced")
        p->b_interlaced = atobool(value);
388 389 390 391 392 393 394
    OPT("cqm")
    {
        if( strstr( value, "flat" ) )
            p->i_cqm_preset = X264_CQM_FLAT;
        else if( strstr( value, "jvt" ) )
            p->i_cqm_preset = X264_CQM_JVT;
        else
395
            p->psz_cqm_file = strdup(value);
396 397 398 399 400 401 402 403 404 405 406 407 408 409 410 411 412 413 414 415 416 417 418 419 420 421 422 423 424 425 426 427 428 429 430 431 432 433 434 435 436 437 438 439 440 441 442 443 444 445 446 447 448 449 450 451 452 453 454 455 456
    }
    OPT("cqmfile")
        p->psz_cqm_file = strdup(value);
    OPT("cqm4")
    {
        p->i_cqm_preset = X264_CQM_CUSTOM;
        b_error |= parse_cqm( value, p->cqm_4iy, 16 );
        b_error |= parse_cqm( value, p->cqm_4ic, 16 );
        b_error |= parse_cqm( value, p->cqm_4py, 16 );
        b_error |= parse_cqm( value, p->cqm_4pc, 16 );
    }
    OPT("cqm8")
    {
        p->i_cqm_preset = X264_CQM_CUSTOM;
        b_error |= parse_cqm( value, p->cqm_8iy, 64 );
        b_error |= parse_cqm( value, p->cqm_8py, 64 );
    }
    OPT("cqm4i")
    {
        p->i_cqm_preset = X264_CQM_CUSTOM;
        b_error |= parse_cqm( value, p->cqm_4iy, 16 );
        b_error |= parse_cqm( value, p->cqm_4ic, 16 );
    }
    OPT("cqm4p")
    {
        p->i_cqm_preset = X264_CQM_CUSTOM;
        b_error |= parse_cqm( value, p->cqm_4py, 16 );
        b_error |= parse_cqm( value, p->cqm_4pc, 16 );
    }
    OPT("cqm4iy")
    {
        p->i_cqm_preset = X264_CQM_CUSTOM;
        b_error |= parse_cqm( value, p->cqm_4iy, 16 );
    }
    OPT("cqm4ic")
    {
        p->i_cqm_preset = X264_CQM_CUSTOM;
        b_error |= parse_cqm( value, p->cqm_4ic, 16 );
    }
    OPT("cqm4py")
    {
        p->i_cqm_preset = X264_CQM_CUSTOM;
        b_error |= parse_cqm( value, p->cqm_4py, 16 );
    }
    OPT("cqm4pc")
    {
        p->i_cqm_preset = X264_CQM_CUSTOM;
        b_error |= parse_cqm( value, p->cqm_4pc, 16 );
    }
    OPT("cqm8i")
    {
        p->i_cqm_preset = X264_CQM_CUSTOM;
        b_error |= parse_cqm( value, p->cqm_8iy, 64 );
    }
    OPT("cqm8p")
    {
        p->i_cqm_preset = X264_CQM_CUSTOM;
        b_error |= parse_cqm( value, p->cqm_8py, 64 );
    }
    OPT("log")
        p->i_log_level = atoi(value);
457 458 459 460
#ifdef VISUALIZE
    OPT("visualize")
        p->b_visualize = atobool(value);
#endif
461 462
    OPT("dump-yuv")
        p->psz_dump_yuv = strdup(value);
463
    OPT2("analyse", "partitions")
464 465 466 467 468 469 470 471 472 473 474 475 476
    {
        p->analyse.inter = 0;
        if( strstr( value, "none" ) )  p->analyse.inter =  0;
        if( strstr( value, "all" ) )   p->analyse.inter = ~0;

        if( strstr( value, "i4x4" ) )  p->analyse.inter |= X264_ANALYSE_I4x4;
        if( strstr( value, "i8x8" ) )  p->analyse.inter |= X264_ANALYSE_I8x8;
        if( strstr( value, "p8x8" ) )  p->analyse.inter |= X264_ANALYSE_PSUB16x16;
        if( strstr( value, "p4x4" ) )  p->analyse.inter |= X264_ANALYSE_PSUB8x8;
        if( strstr( value, "b8x8" ) )  p->analyse.inter |= X264_ANALYSE_BSUB16x16;
    }
    OPT("8x8dct")
        p->analyse.b_transform_8x8 = atobool(value);
477
    OPT2("weightb", "weight-b")
478
        p->analyse.b_weighted_bipred = atobool(value);
479
    OPT2("direct", "direct-pred")
480 481 482 483 484
        b_error |= parse_enum( value, x264_direct_pred_names, &p->analyse.i_direct_mv_pred );
    OPT("chroma-qp-offset")
        p->analyse.i_chroma_qp_offset = atoi(value);
    OPT("me")
        b_error |= parse_enum( value, x264_motion_est_names, &p->analyse.i_me_method );
485
    OPT2("merange", "me-range")
486
        p->analyse.i_me_range = atoi(value);
Loren Merritt's avatar
Loren Merritt committed
487
    OPT2("mvrange", "mv-range")
488
        p->analyse.i_mv_range = atoi(value);
Loren Merritt's avatar
Loren Merritt committed
489 490
    OPT2("mvrange-thread", "mv-range-thread")
        p->analyse.i_mv_range_thread = atoi(value);
491
    OPT2("subme", "subq")
492
        p->analyse.i_subpel_refine = atoi(value);
493 494 495 496 497 498 499 500 501 502 503 504 505 506 507
    OPT("psy-rd")
    {
        if( 2 == sscanf( value, "%f:%f", &p->analyse.f_psy_rd, &p->analyse.f_psy_trellis ) ||
            2 == sscanf( value, "%f,%f", &p->analyse.f_psy_rd, &p->analyse.f_psy_trellis ) )
        { }
        else if( sscanf( value, "%f", &p->analyse.f_psy_rd ) )
        {
            p->analyse.f_psy_trellis = 0;
        }
        else
        {
            p->analyse.f_psy_rd = 0;
            p->analyse.f_psy_trellis = 0;
        }
    }
Fiona Glaser's avatar
Fiona Glaser committed
508 509
    OPT("psy")
        p->analyse.b_psy = atobool(value);
510 511 512 513 514 515 516 517 518 519
    OPT("chroma-me")
        p->analyse.b_chroma_me = atobool(value);
    OPT("mixed-refs")
        p->analyse.b_mixed_references = atobool(value);
    OPT("trellis")
        p->analyse.i_trellis = atoi(value);
    OPT("fast-pskip")
        p->analyse.b_fast_pskip = atobool(value);
    OPT("dct-decimate")
        p->analyse.b_dct_decimate = atobool(value);
520 521 522 523
    OPT("deadzone-inter")
        p->analyse.i_luma_deadzone[0] = atoi(value);
    OPT("deadzone-intra")
        p->analyse.i_luma_deadzone[1] = atoi(value);
524 525 526 527 528 529 530
    OPT("nr")
        p->analyse.i_noise_reduction = atoi(value);
    OPT("bitrate")
    {
        p->rc.i_bitrate = atoi(value);
        p->rc.i_rc_method = X264_RC_ABR;
    }
531
    OPT2("qp", "qp_constant")
532 533 534 535 536 537
    {
        p->rc.i_qp_constant = atoi(value);
        p->rc.i_rc_method = X264_RC_CQP;
    }
    OPT("crf")
    {
538
        p->rc.f_rf_constant = atof(value);
539 540
        p->rc.i_rc_method = X264_RC_CRF;
    }
Fiona Glaser's avatar
Fiona Glaser committed
541 542
    OPT("rc-lookahead")
        p->rc.i_lookahead = atoi(value);
543
    OPT2("qpmin", "qp-min")
544
        p->rc.i_qp_min = atoi(value);
545
    OPT2("qpmax", "qp-max")
546
        p->rc.i_qp_max = atoi(value);
547
    OPT2("qpstep", "qp-step")
548 549 550 551 552 553 554 555 556
        p->rc.i_qp_step = atoi(value);
    OPT("ratetol")
        p->rc.f_rate_tolerance = !strncmp("inf", value, 3) ? 1e9 : atof(value);
    OPT("vbv-maxrate")
        p->rc.i_vbv_max_bitrate = atoi(value);
    OPT("vbv-bufsize")
        p->rc.i_vbv_buffer_size = atoi(value);
    OPT("vbv-init")
        p->rc.f_vbv_buffer_init = atof(value);
557
    OPT2("ipratio", "ip-factor")
558
        p->rc.f_ip_factor = atof(value);
559
    OPT2("pbratio", "pb-factor")
560
        p->rc.f_pb_factor = atof(value);
561 562 563 564
    OPT("aq-mode")
        p->rc.i_aq_mode = atoi(value);
    OPT("aq-strength")
        p->rc.f_aq_strength = atof(value);
565 566 567 568 569 570 571 572 573 574 575 576 577
    OPT("pass")
    {
        int i = x264_clip3( atoi(value), 0, 3 );
        p->rc.b_stat_write = i & 1;
        p->rc.b_stat_read = i & 2;
    }
    OPT("stats")
    {
        p->rc.psz_stat_in = strdup(value);
        p->rc.psz_stat_out = strdup(value);
    }
    OPT("qcomp")
        p->rc.f_qcompress = atof(value);
Fiona Glaser's avatar
Fiona Glaser committed
578 579
    OPT("mbtree")
        p->rc.b_mb_tree = atobool(value);
580 581
    OPT("qblur")
        p->rc.f_qblur = atof(value);
582
    OPT2("cplxblur", "cplx-blur")
583 584 585 586 587
        p->rc.f_complexity_blur = atof(value);
    OPT("zones")
        p->rc.psz_zones = strdup(value);
    OPT("psnr")
        p->analyse.b_psnr = atobool(value);
588 589
    OPT("ssim")
        p->analyse.b_ssim = atobool(value);
590 591 592 593
    OPT("aud")
        p->b_aud = atobool(value);
    OPT("sps-id")
        p->i_sps_id = atoi(value);
594 595
    OPT("global-header")
        p->b_repeat_headers = !atobool(value);
596 597 598 599 600
    OPT("repeat-headers")
        p->b_repeat_headers = atobool(value);
    else
        return X264_PARAM_BAD_NAME;
#undef OPT
601
#undef OPT2
602
#undef atobool
603 604 605 606 607
#undef atoi
#undef atof

    if( name_buf )
        free( name_buf );
608

609
    b_error |= value_was_null && !name_was_bool;
610 611 612
    return b_error ? X264_PARAM_BAD_VALUE : 0;
}

613 614 615 616 617
/****************************************************************************
 * x264_log:
 ****************************************************************************/
void x264_log( x264_t *h, int i_level, const char *psz_fmt, ... )
{
618
    if( !h || i_level <= h->param.i_log_level )
619 620 621
    {
        va_list arg;
        va_start( arg, psz_fmt );
622 623 624 625
        if( !h )
            x264_log_default( NULL, i_level, psz_fmt, arg );
        else
            h->param.pf_log( h->param.p_log_private, i_level, psz_fmt, arg );
626 627 628 629 630 631 632 633 634 635 636 637 638 639 640 641 642 643 644 645 646 647 648 649 650 651 652 653 654
        va_end( arg );
    }
}

static void x264_log_default( void *p_unused, int i_level, const char *psz_fmt, va_list arg )
{
    char *psz_prefix;
    switch( i_level )
    {
        case X264_LOG_ERROR:
            psz_prefix = "error";
            break;
        case X264_LOG_WARNING:
            psz_prefix = "warning";
            break;
        case X264_LOG_INFO:
            psz_prefix = "info";
            break;
        case X264_LOG_DEBUG:
            psz_prefix = "debug";
            break;
        default:
            psz_prefix = "unknown";
            break;
    }
    fprintf( stderr, "x264 [%s]: ", psz_prefix );
    vfprintf( stderr, psz_fmt, arg );
}

Laurent Aimar's avatar
Laurent Aimar committed
655 656 657
/****************************************************************************
 * x264_picture_alloc:
 ****************************************************************************/
658
int x264_picture_alloc( x264_picture_t *pic, int i_csp, int i_width, int i_height )
Laurent Aimar's avatar
Laurent Aimar committed
659 660 661 662
{
    pic->i_type = X264_TYPE_AUTO;
    pic->i_qpplus1 = 0;
    pic->img.i_csp = i_csp;
663 664
    pic->img.i_plane = 3;
    pic->img.plane[0] = x264_malloc( 3 * i_width * i_height / 2 );
665 666
    if( !pic->img.plane[0] )
        return -1;
667 668 669 670 671
    pic->img.plane[1] = pic->img.plane[0] + i_width * i_height;
    pic->img.plane[2] = pic->img.plane[1] + i_width * i_height / 4;
    pic->img.i_stride[0] = i_width;
    pic->img.i_stride[1] = i_width / 2;
    pic->img.i_stride[2] = i_width / 2;
672
    pic->param = NULL;
673
    return 0;
Laurent Aimar's avatar
Laurent Aimar committed
674 675 676 677 678 679 680 681 682 683 684 685 686 687 688 689 690 691 692 693 694
}

/****************************************************************************
 * x264_picture_clean:
 ****************************************************************************/
void x264_picture_clean( x264_picture_t *pic )
{
    x264_free( pic->img.plane[0] );

    /* just to be safe */
    memset( pic, 0, sizeof( x264_picture_t ) );
}

/****************************************************************************
 * x264_nal_encode:
 ****************************************************************************/
int x264_nal_encode( void *p_data, int *pi_data, int b_annexeb, x264_nal_t *nal )
{
    uint8_t *dst = p_data;
    uint8_t *src = nal->p_payload;
    uint8_t *end = &nal->p_payload[nal->i_payload];
Fiona Glaser's avatar
Fiona Glaser committed
695
    int i_count = 0;
Laurent Aimar's avatar
Laurent Aimar committed
696 697 698 699 700 701 702 703 704 705 706 707 708 709 710 711 712

    /* FIXME this code doesn't check overflow */

    if( b_annexeb )
    {
        /* long nal start code (we always use long ones)*/
        *dst++ = 0x00;
        *dst++ = 0x00;
        *dst++ = 0x00;
        *dst++ = 0x01;
    }

    /* nal header */
    *dst++ = ( 0x00 << 7 ) | ( nal->i_ref_idc << 5 ) | nal->i_type;

    while( src < end )
    {
Fiona Glaser's avatar
Fiona Glaser committed
713 714
        if( i_count == 2 && *src <= 0x03 )
        {
Laurent Aimar's avatar
Laurent Aimar committed
715
            *dst++ = 0x03;
Fiona Glaser's avatar
Fiona Glaser committed
716 717 718 719 720 721
            i_count = 0;
        }
        if( *src == 0 )
            i_count++;
        else
            i_count = 0;
Laurent Aimar's avatar
Laurent Aimar committed
722 723 724 725 726 727 728 729 730 731 732 733 734 735
        *dst++ = *src++;
    }
    *pi_data = dst - (uint8_t*)p_data;

    return *pi_data;
}



/****************************************************************************
 * x264_malloc:
 ****************************************************************************/
void *x264_malloc( int i_size )
{
736
    uint8_t *align_buf = NULL;
737 738
#ifdef SYS_MACOSX
    /* Mac OS X always returns 16 bytes aligned memory */
739
    align_buf = malloc( i_size );
740
#elif defined( HAVE_MALLOC_H )
741
    align_buf = memalign( 16, i_size );
Laurent Aimar's avatar
Laurent Aimar committed
742
#else
743 744 745 746 747 748 749 750
    uint8_t *buf = malloc( i_size + 15 + sizeof(void **) + sizeof(int) );
    if( buf )
    {
        align_buf = buf + 15 + sizeof(void **) + sizeof(int);
        align_buf -= (intptr_t) align_buf & 15;
        *( (void **) ( align_buf - sizeof(void **) ) ) = buf;
        *( (int *) ( align_buf - sizeof(void **) - sizeof(int) ) ) = i_size;
    }
Laurent Aimar's avatar
Laurent Aimar committed
751
#endif
752 753 754
    if( !align_buf )
        x264_log( NULL, X264_LOG_ERROR, "malloc of size %d failed\n", i_size );
    return align_buf;
Laurent Aimar's avatar
Laurent Aimar committed
755 756 757 758 759 760 761 762 763
}

/****************************************************************************
 * x264_free:
 ****************************************************************************/
void x264_free( void *p )
{
    if( p )
    {
764
#if defined( HAVE_MALLOC_H ) || defined( SYS_MACOSX )
Laurent Aimar's avatar
Laurent Aimar committed
765 766 767 768 769 770 771
        free( p );
#else
        free( *( ( ( void **) p ) - 1 ) );
#endif
    }
}

Loren Merritt's avatar
Loren Merritt committed
772 773 774 775 776 777 778 779 780 781 782 783 784
/****************************************************************************
 * x264_reduce_fraction:
 ****************************************************************************/
void x264_reduce_fraction( int *n, int *d )
{
    int a = *n;
    int b = *d;
    int c;
    if( !a || !b )
        return;
    c = a % b;
    while(c)
    {
Fiona Glaser's avatar
Fiona Glaser committed
785 786 787
        a = b;
        b = c;
        c = a % b;
Loren Merritt's avatar
Loren Merritt committed
788 789 790 791 792
    }
    *n /= b;
    *d /= b;
}

793 794 795 796 797 798 799 800 801 802 803 804 805 806 807 808 809 810 811 812 813 814 815 816 817 818 819 820 821 822 823
/****************************************************************************
 * x264_slurp_file:
 ****************************************************************************/
char *x264_slurp_file( const char *filename )
{
    int b_error = 0;
    int i_size;
    char *buf;
    FILE *fh = fopen( filename, "rb" );
    if( !fh )
        return NULL;
    b_error |= fseek( fh, 0, SEEK_END ) < 0;
    b_error |= ( i_size = ftell( fh ) ) <= 0;
    b_error |= fseek( fh, 0, SEEK_SET ) < 0;
    if( b_error )
        return NULL;
    buf = x264_malloc( i_size+2 );
    if( buf == NULL )
        return NULL;
    b_error |= fread( buf, 1, i_size, fh ) != i_size;
    if( buf[i_size-1] != '\n' )
        buf[i_size++] = '\n';
    buf[i_size] = 0;
    fclose( fh );
    if( b_error )
    {
        x264_free( buf );
        return NULL;
    }
    return buf;
}
824 825 826 827 828 829

/****************************************************************************
 * x264_param2string:
 ****************************************************************************/
char *x264_param2string( x264_param_t *p, int b_res )
{
830 831 832 833 834
    int len = 1000;
    char *buf, *s;
    if( p->rc.psz_zones )
        len += strlen(p->rc.psz_zones);
    buf = s = x264_malloc( len );
835 836
    if( !buf )
        return NULL;
837 838 839 840 841 842 843 844 845 846 847 848 849 850

    if( b_res )
    {
        s += sprintf( s, "%dx%d ", p->i_width, p->i_height );
        s += sprintf( s, "fps=%d/%d ", p->i_fps_num, p->i_fps_den );
    }

    s += sprintf( s, "cabac=%d", p->b_cabac );
    s += sprintf( s, " ref=%d", p->i_frame_reference );
    s += sprintf( s, " deblock=%d:%d:%d", p->b_deblocking_filter,
                  p->i_deblocking_filter_alphac0, p->i_deblocking_filter_beta );
    s += sprintf( s, " analyse=%#x:%#x", p->analyse.intra, p->analyse.inter );
    s += sprintf( s, " me=%s", x264_motion_est_names[ p->analyse.i_me_method ] );
    s += sprintf( s, " subme=%d", p->analyse.i_subpel_refine );
Fiona Glaser's avatar
Fiona Glaser committed
851 852 853
    s += sprintf( s, " psy=%d", p->analyse.b_psy );
    if( p->analyse.b_psy )
        s += sprintf( s, " psy_rd=%.1f:%.1f", p->analyse.f_psy_rd, p->analyse.f_psy_trellis );
854 855 856 857 858 859
    s += sprintf( s, " mixed_ref=%d", p->analyse.b_mixed_references );
    s += sprintf( s, " me_range=%d", p->analyse.i_me_range );
    s += sprintf( s, " chroma_me=%d", p->analyse.b_chroma_me );
    s += sprintf( s, " trellis=%d", p->analyse.i_trellis );
    s += sprintf( s, " 8x8dct=%d", p->analyse.b_transform_8x8 );
    s += sprintf( s, " cqm=%d", p->i_cqm_preset );
860
    s += sprintf( s, " deadzone=%d,%d", p->analyse.i_luma_deadzone[0], p->analyse.i_luma_deadzone[1] );
861
    s += sprintf( s, " chroma_qp_offset=%d", p->analyse.i_chroma_qp_offset );
Loren Merritt's avatar
Loren Merritt committed
862
    s += sprintf( s, " threads=%d", p->i_threads );
Fiona Glaser's avatar
Fiona Glaser committed
863 864 865 866 867 868
    if( p->i_slice_count )
        s += sprintf( s, " slices=%d", p->i_slice_count );
    if( p->i_slice_max_size )
        s += sprintf( s, " slice_max_size=%d", p->i_slice_max_size );
    if( p->i_slice_max_mbs )
        s += sprintf( s, " slice_max_mbs=%d", p->i_slice_max_mbs );
869
    s += sprintf( s, " nr=%d", p->analyse.i_noise_reduction );
870
    s += sprintf( s, " decimate=%d", p->analyse.b_dct_decimate );
871
    s += sprintf( s, " mbaff=%d", p->b_interlaced );
872 873 874 875

    s += sprintf( s, " bframes=%d", p->i_bframe );
    if( p->i_bframe )
    {
876
        s += sprintf( s, " b_pyramid=%d b_adapt=%d b_bias=%d direct=%d wpredb=%d",
877
                      p->b_bframe_pyramid, p->i_bframe_adaptive, p->i_bframe_bias,
878
                      p->analyse.i_direct_mv_pred, p->analyse.b_weighted_bipred );
879 880
    }

Fiona Glaser's avatar
Fiona Glaser committed
881 882
    s += sprintf( s, " keyint=%d keyint_min=%d scenecut=%d",
                  p->i_keyint_max, p->i_keyint_min, p->i_scenecut_threshold );
883

Fiona Glaser's avatar
Fiona Glaser committed
884
    if( p->rc.b_mb_tree || p->rc.i_vbv_buffer_size )
Fiona Glaser's avatar
Fiona Glaser committed
885 886 887
        s += sprintf( s, " rc_lookahead=%d", p->rc.i_lookahead );

    s += sprintf( s, " rc=%s mbtree=%d", p->rc.i_rc_method == X264_RC_ABR ?
888
                               ( p->rc.b_stat_read ? "2pass" : p->rc.i_vbv_buffer_size ? "cbr" : "abr" )
Fiona Glaser's avatar
Fiona Glaser committed
889
                               : p->rc.i_rc_method == X264_RC_CRF ? "crf" : "cqp", p->rc.b_mb_tree );
890
    if( p->rc.i_rc_method == X264_RC_ABR || p->rc.i_rc_method == X264_RC_CRF )
891
    {
892
        if( p->rc.i_rc_method == X264_RC_CRF )
893
            s += sprintf( s, " crf=%.1f", p->rc.f_rf_constant );
894 895 896
        else
            s += sprintf( s, " bitrate=%d ratetol=%.1f",
                          p->rc.i_bitrate, p->rc.f_rate_tolerance );
897 898
        s += sprintf( s, " qcomp=%.2f qpmin=%d qpmax=%d qpstep=%d",
                      p->rc.f_qcompress, p->rc.i_qp_min, p->rc.i_qp_max, p->rc.i_qp_step );
899 900 901
        if( p->rc.b_stat_read )
            s += sprintf( s, " cplxblur=%.1f qblur=%.1f",
                          p->rc.f_complexity_blur, p->rc.f_qblur );
902
        if( p->rc.i_vbv_buffer_size )
903 904 905
            s += sprintf( s, " vbv_maxrate=%d vbv_bufsize=%d",
                          p->rc.i_vbv_max_bitrate, p->rc.i_vbv_buffer_size );
    }
906
    else if( p->rc.i_rc_method == X264_RC_CQP )
907
        s += sprintf( s, " qp=%d", p->rc.i_qp_constant );
908
    if( !(p->rc.i_rc_method == X264_RC_CQP && p->rc.i_qp_constant == 0) )
909 910
    {
        s += sprintf( s, " ip_ratio=%.2f", p->rc.f_ip_factor );
Fiona Glaser's avatar
Fiona Glaser committed
911
        if( p->i_bframe && !p->rc.b_mb_tree )
912
            s += sprintf( s, " pb_ratio=%.2f", p->rc.f_pb_factor );
913 914 915
        s += sprintf( s, " aq=%d", p->rc.i_aq_mode );
        if( p->rc.i_aq_mode )
            s += sprintf( s, ":%.2f", p->rc.f_aq_strength );
916 917 918
        if( p->rc.psz_zones )
            s += sprintf( s, " zones=%s", p->rc.psz_zones );
        else if( p->rc.i_zones )
919 920 921 922 923 924
            s += sprintf( s, " zones" );
    }

    return buf;
}