git.sesse.net Git - x264/blob - common/common.c

   1 /*****************************************************************************
   2  * common.c: h264 library
   3  *****************************************************************************
   4  * Copyright (C) 2003-2008 x264 project
   5  *
   6  * Authors: Loren Merritt <lorenm@u.washington.edu>
   7  *          Laurent Aimar <fenrir@via.ecp.fr>
   8  *
   9  * This program is free software; you can redistribute it and/or modify
  10  * it under the terms of the GNU General Public License as published by
  11  * the Free Software Foundation; either version 2 of the License, or
  12  * (at your option) any later version.
  13  *
  14  * This program is distributed in the hope that it will be useful,
  15  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  16  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  17  * GNU General Public License for more details.
  18  *
  19  * You should have received a copy of the GNU General Public License
  20  * along with this program; if not, write to the Free Software
  21  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02111, USA.
  22  *****************************************************************************/
  23
  24 #include "common.h"
  25 #include "cpu.h"
  26
  27 #include <stdarg.h>
  28 #include <ctype.h>
  29
  30 #ifdef HAVE_MALLOC_H
  31 #include <malloc.h>
  32 #endif
  33
  34 static void x264_log_default( void *, int, const char *, va_list );
  35
  36 /****************************************************************************
  37  * x264_param_default:
  38  ****************************************************************************/
  39 void    x264_param_default( x264_param_t *param )
  40 {
  41     /* */
  42     memset( param, 0, sizeof( x264_param_t ) );
  43
  44     /* CPU autodetect */
  45     param->cpu = x264_cpu_detect();
  46     param->i_threads = X264_THREADS_AUTO;
  47     param->b_deterministic = 1;
  48     param->i_sync_lookahead = X264_SYNC_LOOKAHEAD_AUTO;
  49
  50     /* Video properties */
  51     param->i_csp           = X264_CSP_I420;
  52     param->i_width         = 0;
  53     param->i_height        = 0;
  54     param->vui.i_sar_width = 0;
  55     param->vui.i_sar_height= 0;
  56     param->vui.i_overscan  = 0;  /* undef */
  57     param->vui.i_vidformat = 5;  /* undef */
  58     param->vui.b_fullrange = 0;  /* off */
  59     param->vui.i_colorprim = 2;  /* undef */
  60     param->vui.i_transfer  = 2;  /* undef */
  61     param->vui.i_colmatrix = 2;  /* undef */
  62     param->vui.i_chroma_loc= 0;  /* left center */
  63     param->i_fps_num       = 25;
  64     param->i_fps_den       = 1;
  65     param->i_level_idc     = -1;
  66     param->i_slice_max_size = 0;
  67     param->i_slice_max_mbs = 0;
  68     param->i_slice_count = 0;
  69
  70     /* Encoder parameters */
  71     param->i_frame_reference = 3;
  72     param->i_keyint_max = 250;
  73     param->i_keyint_min = 25;
  74     param->i_bframe = 3;
  75     param->i_scenecut_threshold = 40;
  76     param->i_bframe_adaptive = X264_B_ADAPT_FAST;
  77     param->i_bframe_bias = 0;
  78     param->i_bframe_pyramid = 0;
  79     param->b_interlaced = 0;
  80     param->b_constrained_intra = 0;
  81
  82     param->b_deblocking_filter = 1;
  83     param->i_deblocking_filter_alphac0 = 0;
  84     param->i_deblocking_filter_beta = 0;
  85
  86     param->b_cabac = 1;
  87     param->i_cabac_init_idc = 0;
  88
  89     param->rc.i_rc_method = X264_RC_CRF;
  90     param->rc.i_bitrate = 0;
  91     param->rc.f_rate_tolerance = 1.0;
  92     param->rc.i_vbv_max_bitrate = 0;
  93     param->rc.i_vbv_buffer_size = 0;
  94     param->rc.f_vbv_buffer_init = 0.9;
  95     param->rc.i_qp_constant = 23;
  96     param->rc.f_rf_constant = 23;
  97     param->rc.i_qp_min = 10;
  98     param->rc.i_qp_max = 51;
  99     param->rc.i_qp_step = 4;
 100     param->rc.f_ip_factor = 1.4;
 101     param->rc.f_pb_factor = 1.3;
 102     param->rc.i_aq_mode = X264_AQ_VARIANCE;
 103     param->rc.f_aq_strength = 1.0;
 104     param->rc.i_lookahead = 40;
 105
 106     param->rc.b_stat_write = 0;
 107     param->rc.psz_stat_out = "x264_2pass.log";
 108     param->rc.b_stat_read = 0;
 109     param->rc.psz_stat_in = "x264_2pass.log";
 110     param->rc.f_qcompress = 0.6;
 111     param->rc.f_qblur = 0.5;
 112     param->rc.f_complexity_blur = 20;
 113     param->rc.i_zones = 0;
 114     param->rc.b_mb_tree = 1;
 115
 116     /* Log */
 117     param->pf_log = x264_log_default;
 118     param->p_log_private = NULL;
 119     param->i_log_level = X264_LOG_INFO;
 120
 121     /* */
 122     param->analyse.intra = X264_ANALYSE_I4x4 | X264_ANALYSE_I8x8;
 123     param->analyse.inter = X264_ANALYSE_I4x4 | X264_ANALYSE_I8x8
 124                          | X264_ANALYSE_PSUB16x16 | X264_ANALYSE_BSUB16x16;
 125     param->analyse.i_direct_mv_pred = X264_DIRECT_PRED_SPATIAL;
 126     param->analyse.i_me_method = X264_ME_HEX;
 127     param->analyse.f_psy_rd = 1.0;
 128     param->analyse.b_psy = 1;
 129     param->analyse.f_psy_trellis = 0;
 130     param->analyse.i_me_range = 16;
 131     param->analyse.i_subpel_refine = 7;
 132     param->analyse.b_mixed_references = 1;
 133     param->analyse.b_chroma_me = 1;
 134     param->analyse.i_mv_range_thread = -1;
 135     param->analyse.i_mv_range = -1; // set from level_idc
 136     param->analyse.i_chroma_qp_offset = 0;
 137     param->analyse.b_fast_pskip = 1;
 138     param->analyse.b_weighted_bipred = 1;
 139     param->analyse.i_weighted_pred = X264_WEIGHTP_SMART;
 140     param->analyse.b_dct_decimate = 1;
 141     param->analyse.b_transform_8x8 = 1;
 142     param->analyse.i_trellis = 1;
 143     param->analyse.i_luma_deadzone[0] = 21;
 144     param->analyse.i_luma_deadzone[1] = 11;
 145     param->analyse.b_psnr = 0;
 146     param->analyse.b_ssim = 0;
 147
 148     param->i_cqm_preset = X264_CQM_FLAT;
 149     memset( param->cqm_4iy, 16, 16 );
 150     memset( param->cqm_4ic, 16, 16 );
 151     memset( param->cqm_4py, 16, 16 );
 152     memset( param->cqm_4pc, 16, 16 );
 153     memset( param->cqm_8iy, 16, 64 );
 154     memset( param->cqm_8py, 16, 64 );
 155
 156     param->b_repeat_headers = 1;
 157     param->b_annexb = 1;
 158     param->b_aud = 0;
 159     param->b_vfr_input = 1;
 160     param->b_dts_compress = 0;
 161 }
 162
 163 static int parse_enum( const char *arg, const char * const *names, int *dst )
 164 {
 165     int i;
 166     for( i = 0; names[i]; i++ )
 167         if( !strcmp( arg, names[i] ) )
 168         {
 169             *dst = i;
 170             return 0;
 171         }
 172     return -1;
 173 }
 174
 175 static int parse_cqm( const char *str, uint8_t *cqm, int length )
 176 {
 177     int i = 0;
 178     do {
 179         int coef;
 180         if( !sscanf( str, "%d", &coef ) || coef < 1 || coef > 255 )
 181             return -1;
 182         cqm[i++] = coef;
 183     } while( i < length && (str = strchr( str, ',' )) && str++ );
 184     return (i == length) ? 0 : -1;
 185 }
 186
 187 static int x264_atobool( const char *str, int *b_error )
 188 {
 189     if( !strcmp(str, "1") ||
 190         !strcmp(str, "true") ||
 191         !strcmp(str, "yes") )
 192         return 1;
 193     if( !strcmp(str, "0") ||
 194         !strcmp(str, "false") ||
 195         !strcmp(str, "no") )
 196         return 0;
 197     *b_error = 1;
 198     return 0;
 199 }
 200
 201 static int x264_atoi( const char *str, int *b_error )
 202 {
 203     char *end;
 204     int v = strtol( str, &end, 0 );
 205     if( end == str || *end != '\0' )
 206         *b_error = 1;
 207     return v;
 208 }
 209
 210 static double x264_atof( const char *str, int *b_error )
 211 {
 212     char *end;
 213     double v = strtod( str, &end );
 214     if( end == str || *end != '\0' )
 215         *b_error = 1;
 216     return v;
 217 }
 218
 219 #define atobool(str) ( name_was_bool = 1, x264_atobool( str, &b_error ) )
 220 #define atoi(str) x264_atoi( str, &b_error )
 221 #define atof(str) x264_atof( str, &b_error )
 222
 223 int x264_param_parse( x264_param_t *p, const char *name, const char *value )
 224 {
 225     char *name_buf = NULL;
 226     int b_error = 0;
 227     int name_was_bool;
 228     int value_was_null = !value;
 229     int i;
 230
 231     if( !name )
 232         return X264_PARAM_BAD_NAME;
 233     if( !value )
 234         value = "true";
 235
 236     if( value[0] == '=' )
 237         value++;
 238
 239     if( strchr( name, '_' ) ) // s/_/-/g
 240     {
 241         char *p;
 242         name_buf = strdup(name);
 243         while( (p = strchr( name_buf, '_' )) )
 244             *p = '-';
 245         name = name_buf;
 246     }
 247
 248     if( (!strncmp( name, "no-", 3 ) && (i = 3)) ||
 249         (!strncmp( name, "no", 2 ) && (i = 2)) )
 250     {
 251         name += i;
 252         value = atobool(value) ? "false" : "true";
 253     }
 254     name_was_bool = 0;
 255
 256 #define OPT(STR) else if( !strcmp( name, STR ) )
 257 #define OPT2(STR0, STR1) else if( !strcmp( name, STR0 ) || !strcmp( name, STR1 ) )
 258     if(0);
 259     OPT("asm")
 260     {
 261         p->cpu = isdigit(value[0]) ? atoi(value) :
 262                  !strcmp(value, "auto") || atobool(value) ? x264_cpu_detect() : 0;
 263         if( b_error )
 264         {
 265             char *buf = strdup(value);
 266             char *tok, UNUSED *saveptr=NULL, *init;
 267             b_error = 0;
 268             p->cpu = 0;
 269             for( init=buf; (tok=strtok_r(init, ",", &saveptr)); init=NULL )
 270             {
 271                 for( i=0; x264_cpu_names[i].flags && strcasecmp(tok, x264_cpu_names[i].name); i++ );
 272                 p->cpu |= x264_cpu_names[i].flags;
 273                 if( !x264_cpu_names[i].flags )
 274                     b_error = 1;
 275             }
 276             free( buf );
 277         }
 278     }
 279     OPT("threads")
 280     {
 281         if( !strcmp(value, "auto") )
 282             p->i_threads = X264_THREADS_AUTO;
 283         else
 284             p->i_threads = atoi(value);
 285     }
 286     OPT("sliced-threads")
 287         p->b_sliced_threads = atobool(value);
 288     OPT("sync-lookahead")
 289     {
 290         if( !strcmp(value, "auto") )
 291             p->i_sync_lookahead = X264_SYNC_LOOKAHEAD_AUTO;
 292         else
 293             p->i_sync_lookahead = atoi(value);
 294     }
 295     OPT2("deterministic", "n-deterministic")
 296         p->b_deterministic = atobool(value);
 297     OPT2("level", "level-idc")
 298     {
 299         if( atof(value) < 6 )
 300             p->i_level_idc = (int)(10*atof(value)+.5);
 301         else
 302             p->i_level_idc = atoi(value);
 303     }
 304     OPT("sar")
 305     {
 306         b_error = ( 2 != sscanf( value, "%d:%d", &p->vui.i_sar_width, &p->vui.i_sar_height ) &&
 307                     2 != sscanf( value, "%d/%d", &p->vui.i_sar_width, &p->vui.i_sar_height ) );
 308     }
 309     OPT("overscan")
 310         b_error |= parse_enum( value, x264_overscan_names, &p->vui.i_overscan );
 311     OPT("videoformat")
 312         b_error |= parse_enum( value, x264_vidformat_names, &p->vui.i_vidformat );
 313     OPT("fullrange")
 314         b_error |= parse_enum( value, x264_fullrange_names, &p->vui.b_fullrange );
 315     OPT("colorprim")
 316         b_error |= parse_enum( value, x264_colorprim_names, &p->vui.i_colorprim );
 317     OPT("transfer")
 318         b_error |= parse_enum( value, x264_transfer_names, &p->vui.i_transfer );
 319     OPT("colormatrix")
 320         b_error |= parse_enum( value, x264_colmatrix_names, &p->vui.i_colmatrix );
 321     OPT("chromaloc")
 322     {
 323         p->vui.i_chroma_loc = atoi(value);
 324         b_error = ( p->vui.i_chroma_loc < 0 || p->vui.i_chroma_loc > 5 );
 325     }
 326     OPT("fps")
 327     {
 328         if( sscanf( value, "%d/%d", &p->i_fps_num, &p->i_fps_den ) == 2 )
 329             ;
 330         else
 331         {
 332             float fps = atof(value);
 333             p->i_fps_num = (int)(fps * 1000 + .5);
 334             p->i_fps_den = 1000;
 335         }
 336     }
 337     OPT2("ref", "frameref")
 338         p->i_frame_reference = atoi(value);
 339     OPT("keyint")
 340     {
 341         p->i_keyint_max = atoi(value);
 342         if( p->i_keyint_min > p->i_keyint_max )
 343             p->i_keyint_min = p->i_keyint_max;
 344     }
 345     OPT2("min-keyint", "keyint-min")
 346     {
 347         p->i_keyint_min = atoi(value);
 348         if( p->i_keyint_max < p->i_keyint_min )
 349             p->i_keyint_max = p->i_keyint_min;
 350     }
 351     OPT("scenecut")
 352     {
 353         p->i_scenecut_threshold = atobool(value);
 354         if( b_error || p->i_scenecut_threshold )
 355         {
 356             b_error = 0;
 357             p->i_scenecut_threshold = atoi(value);
 358         }
 359     }
 360     OPT("intra-refresh")
 361         p->b_intra_refresh = atobool(value);
 362     OPT("bframes")
 363         p->i_bframe = atoi(value);
 364     OPT("b-adapt")
 365     {
 366         p->i_bframe_adaptive = atobool(value);
 367         if( b_error )
 368         {
 369             b_error = 0;
 370             p->i_bframe_adaptive = atoi(value);
 371         }
 372     }
 373     OPT("b-bias")
 374         p->i_bframe_bias = atoi(value);
 375     OPT("b-pyramid")
 376         b_error |= parse_enum( value, x264_b_pyramid_names, &p->i_bframe_pyramid );
 377     OPT("nf")
 378         p->b_deblocking_filter = !atobool(value);
 379     OPT2("filter", "deblock")
 380     {
 381         if( 2 == sscanf( value, "%d:%d", &p->i_deblocking_filter_alphac0, &p->i_deblocking_filter_beta ) ||
 382             2 == sscanf( value, "%d,%d", &p->i_deblocking_filter_alphac0, &p->i_deblocking_filter_beta ) )
 383         {
 384             p->b_deblocking_filter = 1;
 385         }
 386         else if( sscanf( value, "%d", &p->i_deblocking_filter_alphac0 ) )
 387         {
 388             p->b_deblocking_filter = 1;
 389             p->i_deblocking_filter_beta = p->i_deblocking_filter_alphac0;
 390         }
 391         else
 392             p->b_deblocking_filter = atobool(value);
 393     }
 394     OPT("slice-max-size")
 395         p->i_slice_max_size = atoi(value);
 396     OPT("slice-max-mbs")
 397         p->i_slice_max_mbs = atoi(value);
 398     OPT("slices")
 399         p->i_slice_count = atoi(value);
 400     OPT("cabac")
 401         p->b_cabac = atobool(value);
 402     OPT("cabac-idc")
 403         p->i_cabac_init_idc = atoi(value);
 404     OPT("interlaced")
 405         p->b_interlaced = atobool(value);
 406     OPT("constrained-intra")
 407         p->b_constrained_intra = atobool(value);
 408     OPT("cqm")
 409     {
 410         if( strstr( value, "flat" ) )
 411             p->i_cqm_preset = X264_CQM_FLAT;
 412         else if( strstr( value, "jvt" ) )
 413             p->i_cqm_preset = X264_CQM_JVT;
 414         else
 415             p->psz_cqm_file = strdup(value);
 416     }
 417     OPT("cqmfile")
 418         p->psz_cqm_file = strdup(value);
 419     OPT("cqm4")
 420     {
 421         p->i_cqm_preset = X264_CQM_CUSTOM;
 422         b_error |= parse_cqm( value, p->cqm_4iy, 16 );
 423         b_error |= parse_cqm( value, p->cqm_4ic, 16 );
 424         b_error |= parse_cqm( value, p->cqm_4py, 16 );
 425         b_error |= parse_cqm( value, p->cqm_4pc, 16 );
 426     }
 427     OPT("cqm8")
 428     {
 429         p->i_cqm_preset = X264_CQM_CUSTOM;
 430         b_error |= parse_cqm( value, p->cqm_8iy, 64 );
 431         b_error |= parse_cqm( value, p->cqm_8py, 64 );
 432     }
 433     OPT("cqm4i")
 434     {
 435         p->i_cqm_preset = X264_CQM_CUSTOM;
 436         b_error |= parse_cqm( value, p->cqm_4iy, 16 );
 437         b_error |= parse_cqm( value, p->cqm_4ic, 16 );
 438     }
 439     OPT("cqm4p")
 440     {
 441         p->i_cqm_preset = X264_CQM_CUSTOM;
 442         b_error |= parse_cqm( value, p->cqm_4py, 16 );
 443         b_error |= parse_cqm( value, p->cqm_4pc, 16 );
 444     }
 445     OPT("cqm4iy")
 446     {
 447         p->i_cqm_preset = X264_CQM_CUSTOM;
 448         b_error |= parse_cqm( value, p->cqm_4iy, 16 );
 449     }
 450     OPT("cqm4ic")
 451     {
 452         p->i_cqm_preset = X264_CQM_CUSTOM;
 453         b_error |= parse_cqm( value, p->cqm_4ic, 16 );
 454     }
 455     OPT("cqm4py")
 456     {
 457         p->i_cqm_preset = X264_CQM_CUSTOM;
 458         b_error |= parse_cqm( value, p->cqm_4py, 16 );
 459     }
 460     OPT("cqm4pc")
 461     {
 462         p->i_cqm_preset = X264_CQM_CUSTOM;
 463         b_error |= parse_cqm( value, p->cqm_4pc, 16 );
 464     }
 465     OPT("cqm8i")
 466     {
 467         p->i_cqm_preset = X264_CQM_CUSTOM;
 468         b_error |= parse_cqm( value, p->cqm_8iy, 64 );
 469     }
 470     OPT("cqm8p")
 471     {
 472         p->i_cqm_preset = X264_CQM_CUSTOM;
 473         b_error |= parse_cqm( value, p->cqm_8py, 64 );
 474     }
 475     OPT("log")
 476         p->i_log_level = atoi(value);
 477 #ifdef HAVE_VISUALIZE
 478     OPT("visualize")
 479         p->b_visualize = atobool(value);
 480 #endif
 481     OPT("dump-yuv")
 482         p->psz_dump_yuv = strdup(value);
 483     OPT2("analyse", "partitions")
 484     {
 485         p->analyse.inter = 0;
 486         if( strstr( value, "none" ) )  p->analyse.inter =  0;
 487         if( strstr( value, "all" ) )   p->analyse.inter = ~0;
 488
 489         if( strstr( value, "i4x4" ) )  p->analyse.inter |= X264_ANALYSE_I4x4;
 490         if( strstr( value, "i8x8" ) )  p->analyse.inter |= X264_ANALYSE_I8x8;
 491         if( strstr( value, "p8x8" ) )  p->analyse.inter |= X264_ANALYSE_PSUB16x16;
 492         if( strstr( value, "p4x4" ) )  p->analyse.inter |= X264_ANALYSE_PSUB8x8;
 493         if( strstr( value, "b8x8" ) )  p->analyse.inter |= X264_ANALYSE_BSUB16x16;
 494     }
 495     OPT("8x8dct")
 496         p->analyse.b_transform_8x8 = atobool(value);
 497     OPT2("weightb", "weight-b")
 498         p->analyse.b_weighted_bipred = atobool(value);
 499     OPT("weightp")
 500         p->analyse.i_weighted_pred = atoi(value);
 501     OPT2("direct", "direct-pred")
 502         b_error |= parse_enum( value, x264_direct_pred_names, &p->analyse.i_direct_mv_pred );
 503     OPT("chroma-qp-offset")
 504         p->analyse.i_chroma_qp_offset = atoi(value);
 505     OPT("me")
 506         b_error |= parse_enum( value, x264_motion_est_names, &p->analyse.i_me_method );
 507     OPT2("merange", "me-range")
 508         p->analyse.i_me_range = atoi(value);
 509     OPT2("mvrange", "mv-range")
 510         p->analyse.i_mv_range = atoi(value);
 511     OPT2("mvrange-thread", "mv-range-thread")
 512         p->analyse.i_mv_range_thread = atoi(value);
 513     OPT2("subme", "subq")
 514         p->analyse.i_subpel_refine = atoi(value);
 515     OPT("psy-rd")
 516     {
 517         if( 2 == sscanf( value, "%f:%f", &p->analyse.f_psy_rd, &p->analyse.f_psy_trellis ) ||
 518             2 == sscanf( value, "%f,%f", &p->analyse.f_psy_rd, &p->analyse.f_psy_trellis ) ||
 519             2 == sscanf( value, "%f|%f", &p->analyse.f_psy_rd, &p->analyse.f_psy_trellis ))
 520         { }
 521         else if( sscanf( value, "%f", &p->analyse.f_psy_rd ) )
 522         {
 523             p->analyse.f_psy_trellis = 0;
 524         }
 525         else
 526         {
 527             p->analyse.f_psy_rd = 0;
 528             p->analyse.f_psy_trellis = 0;
 529         }
 530     }
 531     OPT("psy")
 532         p->analyse.b_psy = atobool(value);
 533     OPT("chroma-me")
 534         p->analyse.b_chroma_me = atobool(value);
 535     OPT("mixed-refs")
 536         p->analyse.b_mixed_references = atobool(value);
 537     OPT("trellis")
 538         p->analyse.i_trellis = atoi(value);
 539     OPT("fast-pskip")
 540         p->analyse.b_fast_pskip = atobool(value);
 541     OPT("dct-decimate")
 542         p->analyse.b_dct_decimate = atobool(value);
 543     OPT("deadzone-inter")
 544         p->analyse.i_luma_deadzone[0] = atoi(value);
 545     OPT("deadzone-intra")
 546         p->analyse.i_luma_deadzone[1] = atoi(value);
 547     OPT("nr")
 548         p->analyse.i_noise_reduction = atoi(value);
 549     OPT("bitrate")
 550     {
 551         p->rc.i_bitrate = atoi(value);
 552         p->rc.i_rc_method = X264_RC_ABR;
 553     }
 554     OPT2("qp", "qp_constant")
 555     {
 556         p->rc.i_qp_constant = atoi(value);
 557         p->rc.i_rc_method = X264_RC_CQP;
 558     }
 559     OPT("crf")
 560     {
 561         p->rc.f_rf_constant = atof(value);
 562         p->rc.i_rc_method = X264_RC_CRF;
 563     }
 564     OPT("rc-lookahead")
 565         p->rc.i_lookahead = atoi(value);
 566     OPT2("qpmin", "qp-min")
 567         p->rc.i_qp_min = atoi(value);
 568     OPT2("qpmax", "qp-max")
 569         p->rc.i_qp_max = atoi(value);
 570     OPT2("qpstep", "qp-step")
 571         p->rc.i_qp_step = atoi(value);
 572     OPT("ratetol")
 573         p->rc.f_rate_tolerance = !strncmp("inf", value, 3) ? 1e9 : atof(value);
 574     OPT("vbv-maxrate")
 575         p->rc.i_vbv_max_bitrate = atoi(value);
 576     OPT("vbv-bufsize")
 577         p->rc.i_vbv_buffer_size = atoi(value);
 578     OPT("vbv-init")
 579         p->rc.f_vbv_buffer_init = atof(value);
 580     OPT2("ipratio", "ip-factor")
 581         p->rc.f_ip_factor = atof(value);
 582     OPT2("pbratio", "pb-factor")
 583         p->rc.f_pb_factor = atof(value);
 584     OPT("aq-mode")
 585         p->rc.i_aq_mode = atoi(value);
 586     OPT("aq-strength")
 587         p->rc.f_aq_strength = atof(value);
 588     OPT("pass")
 589     {
 590         int i = x264_clip3( atoi(value), 0, 3 );
 591         p->rc.b_stat_write = i & 1;
 592         p->rc.b_stat_read = i & 2;
 593     }
 594     OPT("stats")
 595     {
 596         p->rc.psz_stat_in = strdup(value);
 597         p->rc.psz_stat_out = strdup(value);
 598     }
 599     OPT("qcomp")
 600         p->rc.f_qcompress = atof(value);
 601     OPT("mbtree")
 602         p->rc.b_mb_tree = atobool(value);
 603     OPT("qblur")
 604         p->rc.f_qblur = atof(value);
 605     OPT2("cplxblur", "cplx-blur")
 606         p->rc.f_complexity_blur = atof(value);
 607     OPT("zones")
 608         p->rc.psz_zones = strdup(value);
 609     OPT("psnr")
 610         p->analyse.b_psnr = atobool(value);
 611     OPT("ssim")
 612         p->analyse.b_ssim = atobool(value);
 613     OPT("aud")
 614         p->b_aud = atobool(value);
 615     OPT("sps-id")
 616         p->i_sps_id = atoi(value);
 617     OPT("global-header")
 618         p->b_repeat_headers = !atobool(value);
 619     OPT("repeat-headers")
 620         p->b_repeat_headers = atobool(value);
 621     OPT("annexb")
 622         p->b_annexb = atobool(value);
 623     OPT("force-cfr")
 624         p->b_vfr_input = !atobool(value);
 625     else
 626         return X264_PARAM_BAD_NAME;
 627 #undef OPT
 628 #undef OPT2
 629 #undef atobool
 630 #undef atoi
 631 #undef atof
 632
 633     if( name_buf )
 634         free( name_buf );
 635
 636     b_error |= value_was_null && !name_was_bool;
 637     return b_error ? X264_PARAM_BAD_VALUE : 0;
 638 }
 639
 640 /****************************************************************************
 641  * x264_log:
 642  ****************************************************************************/
 643 void x264_log( x264_t *h, int i_level, const char *psz_fmt, ... )
 644 {
 645     if( !h || i_level <= h->param.i_log_level )
 646     {
 647         va_list arg;
 648         va_start( arg, psz_fmt );
 649         if( !h )
 650             x264_log_default( NULL, i_level, psz_fmt, arg );
 651         else
 652             h->param.pf_log( h->param.p_log_private, i_level, psz_fmt, arg );
 653         va_end( arg );
 654     }
 655 }
 656
 657 static void x264_log_default( void *p_unused, int i_level, const char *psz_fmt, va_list arg )
 658 {
 659     char *psz_prefix;
 660     switch( i_level )
 661     {
 662         case X264_LOG_ERROR:
 663             psz_prefix = "error";
 664             break;
 665         case X264_LOG_WARNING:
 666             psz_prefix = "warning";
 667             break;
 668         case X264_LOG_INFO:
 669             psz_prefix = "info";
 670             break;
 671         case X264_LOG_DEBUG:
 672             psz_prefix = "debug";
 673             break;
 674         default:
 675             psz_prefix = "unknown";
 676             break;
 677     }
 678     fprintf( stderr, "x264 [%s]: ", psz_prefix );
 679     vfprintf( stderr, psz_fmt, arg );
 680 }
 681
 682 /****************************************************************************
 683  * x264_picture_alloc:
 684  ****************************************************************************/
 685 int x264_picture_alloc( x264_picture_t *pic, int i_csp, int i_width, int i_height )
 686 {
 687     pic->i_type = X264_TYPE_AUTO;
 688     pic->i_qpplus1 = 0;
 689     pic->img.i_csp = i_csp;
 690     pic->img.i_plane = 3;
 691     pic->img.plane[0] = x264_malloc( 3 * i_width * i_height / 2 );
 692     if( !pic->img.plane[0] )
 693         return -1;
 694     pic->img.plane[1] = pic->img.plane[0] + i_width * i_height;
 695     pic->img.plane[2] = pic->img.plane[1] + i_width * i_height / 4;
 696     pic->img.i_stride[0] = i_width;
 697     pic->img.i_stride[1] = i_width / 2;
 698     pic->img.i_stride[2] = i_width / 2;
 699     pic->param = NULL;
 700     return 0;
 701 }
 702
 703 /****************************************************************************
 704  * x264_picture_clean:
 705  ****************************************************************************/
 706 void x264_picture_clean( x264_picture_t *pic )
 707 {
 708     x264_free( pic->img.plane[0] );
 709
 710     /* just to be safe */
 711     memset( pic, 0, sizeof( x264_picture_t ) );
 712 }
 713
 714 /****************************************************************************
 715  * x264_nal_encode:
 716  ****************************************************************************/
 717 int x264_nal_encode( uint8_t *dst, int b_annexb, x264_nal_t *nal )
 718 {
 719     uint8_t *src = nal->p_payload;
 720     uint8_t *end = nal->p_payload + nal->i_payload;
 721     uint8_t *orig_dst = dst;
 722     int i_count = 0, size;
 723
 724     /* long nal start code (we always use long ones) */
 725     if( b_annexb )
 726     {
 727         *dst++ = 0x00;
 728         *dst++ = 0x00;
 729         *dst++ = 0x00;
 730         *dst++ = 0x01;
 731     }
 732     else /* save room for size later */
 733         dst += 4;
 734
 735     /* nal header */
 736     *dst++ = ( 0x00 << 7 ) | ( nal->i_ref_idc << 5 ) | nal->i_type;
 737
 738     while( src < end )
 739     {
 740         if( i_count == 2 && *src <= 0x03 )
 741         {
 742             *dst++ = 0x03;
 743             i_count = 0;
 744         }
 745         if( *src == 0 )
 746             i_count++;
 747         else
 748             i_count = 0;
 749         *dst++ = *src++;
 750     }
 751     size = (dst - orig_dst) - 4;
 752
 753     /* Write the size header for mp4/etc */
 754     if( !b_annexb )
 755     {
 756         /* Size doesn't include the size of the header we're writing now. */
 757         orig_dst[0] = size>>24;
 758         orig_dst[1] = size>>16;
 759         orig_dst[2] = size>> 8;
 760         orig_dst[3] = size>> 0;
 761     }
 762
 763     return size+4;
 764 }
 765
 766
 767
 768 /****************************************************************************
 769  * x264_malloc:
 770  ****************************************************************************/
 771 void *x264_malloc( int i_size )
 772 {
 773     uint8_t *align_buf = NULL;
 774 #ifdef SYS_MACOSX
 775     /* Mac OS X always returns 16 bytes aligned memory */
 776     align_buf = malloc( i_size );
 777 #elif defined( HAVE_MALLOC_H )
 778     align_buf = memalign( 16, i_size );
 779 #else
 780     uint8_t *buf = malloc( i_size + 15 + sizeof(void **) + sizeof(int) );
 781     if( buf )
 782     {
 783         align_buf = buf + 15 + sizeof(void **) + sizeof(int);
 784         align_buf -= (intptr_t) align_buf & 15;
 785         *( (void **) ( align_buf - sizeof(void **) ) ) = buf;
 786         *( (int *) ( align_buf - sizeof(void **) - sizeof(int) ) ) = i_size;
 787     }
 788 #endif
 789     if( !align_buf )
 790         x264_log( NULL, X264_LOG_ERROR, "malloc of size %d failed\n", i_size );
 791     return align_buf;
 792 }
 793
 794 /****************************************************************************
 795  * x264_free:
 796  ****************************************************************************/
 797 void x264_free( void *p )
 798 {
 799     if( p )
 800     {
 801 #if defined( HAVE_MALLOC_H ) || defined( SYS_MACOSX )
 802         free( p );
 803 #else
 804         free( *( ( ( void **) p ) - 1 ) );
 805 #endif
 806     }
 807 }
 808
 809 /****************************************************************************
 810  * x264_reduce_fraction:
 811  ****************************************************************************/
 812 void x264_reduce_fraction( int *n, int *d )
 813 {
 814     int a = *n;
 815     int b = *d;
 816     int c;
 817     if( !a || !b )
 818         return;
 819     c = a % b;
 820     while(c)
 821     {
 822         a = b;
 823         b = c;
 824         c = a % b;
 825     }
 826     *n /= b;
 827     *d /= b;
 828 }
 829
 830 /****************************************************************************
 831  * x264_slurp_file:
 832  ****************************************************************************/
 833 char *x264_slurp_file( const char *filename )
 834 {
 835     int b_error = 0;
 836     int i_size;
 837     char *buf;
 838     FILE *fh = fopen( filename, "rb" );
 839     if( !fh )
 840         return NULL;
 841     b_error |= fseek( fh, 0, SEEK_END ) < 0;
 842     b_error |= ( i_size = ftell( fh ) ) <= 0;
 843     b_error |= fseek( fh, 0, SEEK_SET ) < 0;
 844     if( b_error )
 845         return NULL;
 846     buf = x264_malloc( i_size+2 );
 847     if( buf == NULL )
 848         return NULL;
 849     b_error |= fread( buf, 1, i_size, fh ) != i_size;
 850     if( buf[i_size-1] != '\n' )
 851         buf[i_size++] = '\n';
 852     buf[i_size] = 0;
 853     fclose( fh );
 854     if( b_error )
 855     {
 856         x264_free( buf );
 857         return NULL;
 858     }
 859     return buf;
 860 }
 861
 862 /****************************************************************************
 863  * x264_param2string:
 864  ****************************************************************************/
 865 char *x264_param2string( x264_param_t *p, int b_res )
 866 {
 867     int len = 1000;
 868     char *buf, *s;
 869     if( p->rc.psz_zones )
 870         len += strlen(p->rc.psz_zones);
 871     buf = s = x264_malloc( len );
 872     if( !buf )
 873         return NULL;
 874
 875     if( b_res )
 876     {
 877         s += sprintf( s, "%dx%d ", p->i_width, p->i_height );
 878         s += sprintf( s, "fps=%d/%d ", p->i_fps_num, p->i_fps_den );
 879     }
 880
 881     s += sprintf( s, "cabac=%d", p->b_cabac );
 882     s += sprintf( s, " ref=%d", p->i_frame_reference );
 883     s += sprintf( s, " deblock=%d:%d:%d", p->b_deblocking_filter,
 884                   p->i_deblocking_filter_alphac0, p->i_deblocking_filter_beta );
 885     s += sprintf( s, " analyse=%#x:%#x", p->analyse.intra, p->analyse.inter );
 886     s += sprintf( s, " me=%s", x264_motion_est_names[ p->analyse.i_me_method ] );
 887     s += sprintf( s, " subme=%d", p->analyse.i_subpel_refine );
 888     s += sprintf( s, " psy=%d", p->analyse.b_psy );
 889     if( p->analyse.b_psy )
 890         s += sprintf( s, " psy_rd=%.2f:%.2f", p->analyse.f_psy_rd, p->analyse.f_psy_trellis );
 891     s += sprintf( s, " mixed_ref=%d", p->analyse.b_mixed_references );
 892     s += sprintf( s, " me_range=%d", p->analyse.i_me_range );
 893     s += sprintf( s, " chroma_me=%d", p->analyse.b_chroma_me );
 894     s += sprintf( s, " trellis=%d", p->analyse.i_trellis );
 895     s += sprintf( s, " 8x8dct=%d", p->analyse.b_transform_8x8 );
 896     s += sprintf( s, " cqm=%d", p->i_cqm_preset );
 897     s += sprintf( s, " deadzone=%d,%d", p->analyse.i_luma_deadzone[0], p->analyse.i_luma_deadzone[1] );
 898     s += sprintf( s, " fast_pskip=%d", p->analyse.b_fast_pskip );
 899     s += sprintf( s, " chroma_qp_offset=%d", p->analyse.i_chroma_qp_offset );
 900     s += sprintf( s, " threads=%d", p->i_threads );
 901     s += sprintf( s, " sliced_threads=%d", p->b_sliced_threads );
 902     if( p->i_slice_count )
 903         s += sprintf( s, " slices=%d", p->i_slice_count );
 904     if( p->i_slice_max_size )
 905         s += sprintf( s, " slice_max_size=%d", p->i_slice_max_size );
 906     if( p->i_slice_max_mbs )
 907         s += sprintf( s, " slice_max_mbs=%d", p->i_slice_max_mbs );
 908     s += sprintf( s, " nr=%d", p->analyse.i_noise_reduction );
 909     s += sprintf( s, " decimate=%d", p->analyse.b_dct_decimate );
 910     s += sprintf( s, " mbaff=%d", p->b_interlaced );
 911     s += sprintf( s, " constrained_intra=%d", p->b_constrained_intra );
 912
 913     s += sprintf( s, " bframes=%d", p->i_bframe );
 914     if( p->i_bframe )
 915     {
 916         s += sprintf( s, " b_pyramid=%d b_adapt=%d b_bias=%d direct=%d wpredb=%d",
 917                       p->i_bframe_pyramid, p->i_bframe_adaptive, p->i_bframe_bias,
 918                       p->analyse.i_direct_mv_pred, p->analyse.b_weighted_bipred );
 919     }
 920     s += sprintf( s, " wpredp=%d", p->analyse.i_weighted_pred > 0 ? p->analyse.i_weighted_pred : 0 );
 921
 922     s += sprintf( s, " keyint=%d keyint_min=%d scenecut=%d intra_refresh=%d",
 923                   p->i_keyint_max, p->i_keyint_min, p->i_scenecut_threshold, p->b_intra_refresh );
 924
 925     if( p->rc.b_mb_tree || p->rc.i_vbv_buffer_size )
 926         s += sprintf( s, " rc_lookahead=%d", p->rc.i_lookahead );
 927
 928     s += sprintf( s, " rc=%s mbtree=%d", p->rc.i_rc_method == X264_RC_ABR ?
 929                                ( p->rc.b_stat_read ? "2pass" : p->rc.i_vbv_buffer_size ? "cbr" : "abr" )
 930                                : p->rc.i_rc_method == X264_RC_CRF ? "crf" : "cqp", p->rc.b_mb_tree );
 931     if( p->rc.i_rc_method == X264_RC_ABR || p->rc.i_rc_method == X264_RC_CRF )
 932     {
 933         if( p->rc.i_rc_method == X264_RC_CRF )
 934             s += sprintf( s, " crf=%.1f", p->rc.f_rf_constant );
 935         else
 936             s += sprintf( s, " bitrate=%d ratetol=%.1f",
 937                           p->rc.i_bitrate, p->rc.f_rate_tolerance );
 938         s += sprintf( s, " qcomp=%.2f qpmin=%d qpmax=%d qpstep=%d",
 939                       p->rc.f_qcompress, p->rc.i_qp_min, p->rc.i_qp_max, p->rc.i_qp_step );
 940         if( p->rc.b_stat_read )
 941             s += sprintf( s, " cplxblur=%.1f qblur=%.1f",
 942                           p->rc.f_complexity_blur, p->rc.f_qblur );
 943         if( p->rc.i_vbv_buffer_size )
 944             s += sprintf( s, " vbv_maxrate=%d vbv_bufsize=%d",
 945                           p->rc.i_vbv_max_bitrate, p->rc.i_vbv_buffer_size );
 946     }
 947     else if( p->rc.i_rc_method == X264_RC_CQP )
 948         s += sprintf( s, " qp=%d", p->rc.i_qp_constant );
 949     if( !(p->rc.i_rc_method == X264_RC_CQP && p->rc.i_qp_constant == 0) )
 950     {
 951         s += sprintf( s, " ip_ratio=%.2f", p->rc.f_ip_factor );
 952         if( p->i_bframe && !p->rc.b_mb_tree )
 953             s += sprintf( s, " pb_ratio=%.2f", p->rc.f_pb_factor );
 954         s += sprintf( s, " aq=%d", p->rc.i_aq_mode );
 955         if( p->rc.i_aq_mode )
 956             s += sprintf( s, ":%.2f", p->rc.f_aq_strength );
 957         if( p->rc.psz_zones )
 958             s += sprintf( s, " zones=%s", p->rc.psz_zones );
 959         else if( p->rc.i_zones )
 960             s += sprintf( s, " zones" );
 961     }
 962
 963     return buf;
 964 }
 965