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

   1 /*****************************************************************************
   2  * common.c: misc common functions
   3  *****************************************************************************
   4  * Copyright (C) 2003-2016 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  * This program is also available under a commercial proprietary license.
  24  * For more information, contact us at licensing@x264.com.
  25  *****************************************************************************/
  26
  27 #include "common.h"
  28
  29 #include <ctype.h>
  30
  31 #if HAVE_MALLOC_H
  32 #include <malloc.h>
  33 #endif
  34 #if HAVE_THP
  35 #include <sys/mman.h>
  36 #endif
  37
  38 const int x264_bit_depth = BIT_DEPTH;
  39
  40 const int x264_chroma_format = X264_CHROMA_FORMAT;
  41
  42 static void x264_log_default( void *, int, const char *, va_list );
  43
  44 /****************************************************************************
  45  * x264_param_default:
  46  ****************************************************************************/
  47 void x264_param_default( x264_param_t *param )
  48 {
  49     /* */
  50     memset( param, 0, sizeof( x264_param_t ) );
  51
  52     /* CPU autodetect */
  53     param->cpu = x264_cpu_detect();
  54     param->i_threads = X264_THREADS_AUTO;
  55     param->i_lookahead_threads = X264_THREADS_AUTO;
  56     param->b_deterministic = 1;
  57     param->i_sync_lookahead = X264_SYNC_LOOKAHEAD_AUTO;
  58
  59     /* Video properties */
  60     param->i_csp           = X264_CHROMA_FORMAT ? X264_CHROMA_FORMAT : X264_CSP_I420;
  61     param->i_width         = 0;
  62     param->i_height        = 0;
  63     param->vui.i_sar_width = 0;
  64     param->vui.i_sar_height= 0;
  65     param->vui.i_overscan  = 0;  /* undef */
  66     param->vui.i_vidformat = 5;  /* undef */
  67     param->vui.b_fullrange = -1; /* default depends on input */
  68     param->vui.i_colorprim = 2;  /* undef */
  69     param->vui.i_transfer  = 2;  /* undef */
  70     param->vui.i_colmatrix = -1; /* default depends on input */
  71     param->vui.i_chroma_loc= 0;  /* left center */
  72     param->i_fps_num       = 25;
  73     param->i_fps_den       = 1;
  74     param->i_level_idc     = -1;
  75     param->i_slice_max_size = 0;
  76     param->i_slice_max_mbs = 0;
  77     param->i_slice_count = 0;
  78
  79     /* Encoder parameters */
  80     param->i_frame_reference = 3;
  81     param->i_keyint_max = 250;
  82     param->i_keyint_min = X264_KEYINT_MIN_AUTO;
  83     param->i_bframe = 3;
  84     param->i_scenecut_threshold = 40;
  85     param->i_bframe_adaptive = X264_B_ADAPT_FAST;
  86     param->i_bframe_bias = 0;
  87     param->i_bframe_pyramid = X264_B_PYRAMID_NORMAL;
  88     param->b_interlaced = 0;
  89     param->b_constrained_intra = 0;
  90
  91     param->b_deblocking_filter = 1;
  92     param->i_deblocking_filter_alphac0 = 0;
  93     param->i_deblocking_filter_beta = 0;
  94
  95     param->b_cabac = 1;
  96     param->i_cabac_init_idc = 0;
  97
  98     param->rc.i_rc_method = X264_RC_CRF;
  99     param->rc.i_bitrate = 0;
 100     param->rc.f_rate_tolerance = 1.0;
 101     param->rc.i_vbv_max_bitrate = 0;
 102     param->rc.i_vbv_buffer_size = 0;
 103     param->rc.f_vbv_buffer_init = 0.9;
 104     param->rc.i_qp_constant = 23 + QP_BD_OFFSET;
 105     param->rc.f_rf_constant = 23;
 106     param->rc.i_qp_min = 0;
 107     param->rc.i_qp_max = QP_MAX;
 108     param->rc.i_qp_step = 4;
 109     param->rc.f_ip_factor = 1.4;
 110     param->rc.f_pb_factor = 1.3;
 111     param->rc.i_aq_mode = X264_AQ_VARIANCE;
 112     param->rc.f_aq_strength = 1.0;
 113     param->rc.i_lookahead = 40;
 114
 115     param->rc.b_stat_write = 0;
 116     param->rc.psz_stat_out = "x264_2pass.log";
 117     param->rc.b_stat_read = 0;
 118     param->rc.psz_stat_in = "x264_2pass.log";
 119     param->rc.f_qcompress = 0.6;
 120     param->rc.f_qblur = 0.5;
 121     param->rc.f_complexity_blur = 20;
 122     param->rc.i_zones = 0;
 123     param->rc.b_mb_tree = 1;
 124
 125     // speedcontrol
 126     param->sc.f_speed = 0;
 127     param->sc.i_buffer_size = 12;
 128     param->sc.f_buffer_init = 0.75;
 129
 130     /* Log */
 131     param->pf_log = x264_log_default;
 132     param->p_log_private = NULL;
 133     param->i_log_level = X264_LOG_INFO;
 134
 135     /* */
 136     param->analyse.intra = X264_ANALYSE_I4x4 | X264_ANALYSE_I8x8;
 137     param->analyse.inter = X264_ANALYSE_I4x4 | X264_ANALYSE_I8x8
 138                          | X264_ANALYSE_PSUB16x16 | X264_ANALYSE_BSUB16x16;
 139     param->analyse.i_direct_mv_pred = X264_DIRECT_PRED_SPATIAL;
 140     param->analyse.i_me_method = X264_ME_HEX;
 141     param->analyse.f_psy_rd = 1.0;
 142     param->analyse.b_psy = 1;
 143     param->analyse.f_psy_trellis = 0;
 144     param->analyse.i_me_range = 16;
 145     param->analyse.i_subpel_refine = 7;
 146     param->analyse.b_mixed_references = 1;
 147     param->analyse.b_chroma_me = 1;
 148     param->analyse.i_mv_range_thread = -1;
 149     param->analyse.i_mv_range = -1; // set from level_idc
 150     param->analyse.i_chroma_qp_offset = 0;
 151     param->analyse.b_fast_pskip = 1;
 152     param->analyse.b_weighted_bipred = 1;
 153     param->analyse.i_weighted_pred = X264_WEIGHTP_SMART;
 154     param->analyse.b_dct_decimate = 1;
 155     param->analyse.b_transform_8x8 = 1;
 156     param->analyse.i_trellis = 1;
 157     param->analyse.i_luma_deadzone[0] = 21;
 158     param->analyse.i_luma_deadzone[1] = 11;
 159     param->analyse.b_psnr = 0;
 160     param->analyse.b_ssim = 0;
 161
 162     param->i_cqm_preset = X264_CQM_FLAT;
 163     memset( param->cqm_4iy, 16, sizeof( param->cqm_4iy ) );
 164     memset( param->cqm_4py, 16, sizeof( param->cqm_4py ) );
 165     memset( param->cqm_4ic, 16, sizeof( param->cqm_4ic ) );
 166     memset( param->cqm_4pc, 16, sizeof( param->cqm_4pc ) );
 167     memset( param->cqm_8iy, 16, sizeof( param->cqm_8iy ) );
 168     memset( param->cqm_8py, 16, sizeof( param->cqm_8py ) );
 169     memset( param->cqm_8ic, 16, sizeof( param->cqm_8ic ) );
 170     memset( param->cqm_8pc, 16, sizeof( param->cqm_8pc ) );
 171
 172     param->b_repeat_headers = 1;
 173     param->b_annexb = 1;
 174     param->b_aud = 0;
 175     param->b_vfr_input = 1;
 176     param->i_nal_hrd = X264_NAL_HRD_NONE;
 177     param->b_tff = 1;
 178     param->b_pic_struct = 0;
 179     param->b_fake_interlaced = 0;
 180     param->i_frame_packing = -1;
 181     param->b_opencl = 0;
 182     param->i_opencl_device = 0;
 183     param->opencl_device_id = NULL;
 184     param->psz_clbin_file = NULL;
 185 }
 186
 187 static int x264_param_apply_preset( x264_param_t *param, const char *preset )
 188 {
 189     char *end;
 190     int i = strtol( preset, &end, 10 );
 191     if( *end == 0 && i >= 0 && i < sizeof(x264_preset_names)/sizeof(*x264_preset_names)-1 )
 192         preset = x264_preset_names[i];
 193
 194     if( !strcasecmp( preset, "ultrafast" ) )
 195     {
 196         param->i_frame_reference = 1;
 197         param->i_scenecut_threshold = 0;
 198         param->b_deblocking_filter = 0;
 199         param->b_cabac = 0;
 200         param->i_bframe = 0;
 201         param->analyse.intra = 0;
 202         param->analyse.inter = 0;
 203         param->analyse.b_transform_8x8 = 0;
 204         param->analyse.i_me_method = X264_ME_DIA;
 205         param->analyse.i_subpel_refine = 0;
 206         param->rc.i_aq_mode = 0;
 207         param->analyse.b_mixed_references = 0;
 208         param->analyse.i_trellis = 0;
 209         param->i_bframe_adaptive = X264_B_ADAPT_NONE;
 210         param->rc.b_mb_tree = 0;
 211         param->analyse.i_weighted_pred = X264_WEIGHTP_NONE;
 212         param->analyse.b_weighted_bipred = 0;
 213         param->rc.i_lookahead = 0;
 214     }
 215     else if( !strcasecmp( preset, "superfast" ) )
 216     {
 217         param->analyse.inter = X264_ANALYSE_I8x8|X264_ANALYSE_I4x4;
 218         param->analyse.i_me_method = X264_ME_DIA;
 219         param->analyse.i_subpel_refine = 1;
 220         param->i_frame_reference = 1;
 221         param->analyse.b_mixed_references = 0;
 222         param->analyse.i_trellis = 0;
 223         param->rc.b_mb_tree = 0;
 224         param->analyse.i_weighted_pred = X264_WEIGHTP_SIMPLE;
 225         param->rc.i_lookahead = 0;
 226     }
 227     else if( !strcasecmp( preset, "veryfast" ) )
 228     {
 229         param->analyse.i_me_method = X264_ME_HEX;
 230         param->analyse.i_subpel_refine = 2;
 231         param->i_frame_reference = 1;
 232         param->analyse.b_mixed_references = 0;
 233         param->analyse.i_trellis = 0;
 234         param->analyse.i_weighted_pred = X264_WEIGHTP_SIMPLE;
 235         param->rc.i_lookahead = 10;
 236     }
 237     else if( !strcasecmp( preset, "faster" ) )
 238     {
 239         param->analyse.b_mixed_references = 0;
 240         param->i_frame_reference = 2;
 241         param->analyse.i_subpel_refine = 4;
 242         param->analyse.i_weighted_pred = X264_WEIGHTP_SIMPLE;
 243         param->rc.i_lookahead = 20;
 244     }
 245     else if( !strcasecmp( preset, "fast" ) )
 246     {
 247         param->i_frame_reference = 2;
 248         param->analyse.i_subpel_refine = 6;
 249         param->analyse.i_weighted_pred = X264_WEIGHTP_SIMPLE;
 250         param->rc.i_lookahead = 30;
 251     }
 252     else if( !strcasecmp( preset, "medium" ) )
 253     {
 254         /* Default is medium */
 255     }
 256     else if( !strcasecmp( preset, "slow" ) )
 257     {
 258         param->analyse.i_me_method = X264_ME_UMH;
 259         param->analyse.i_subpel_refine = 8;
 260         param->i_frame_reference = 5;
 261         param->i_bframe_adaptive = X264_B_ADAPT_TRELLIS;
 262         param->analyse.i_direct_mv_pred = X264_DIRECT_PRED_AUTO;
 263         param->rc.i_lookahead = 50;
 264     }
 265     else if( !strcasecmp( preset, "slower" ) )
 266     {
 267         param->analyse.i_me_method = X264_ME_UMH;
 268         param->analyse.i_subpel_refine = 9;
 269         param->i_frame_reference = 8;
 270         param->i_bframe_adaptive = X264_B_ADAPT_TRELLIS;
 271         param->analyse.i_direct_mv_pred = X264_DIRECT_PRED_AUTO;
 272         param->analyse.inter |= X264_ANALYSE_PSUB8x8;
 273         param->analyse.i_trellis = 2;
 274         param->rc.i_lookahead = 60;
 275     }
 276     else if( !strcasecmp( preset, "veryslow" ) )
 277     {
 278         param->analyse.i_me_method = X264_ME_UMH;
 279         param->analyse.i_subpel_refine = 10;
 280         param->analyse.i_me_range = 24;
 281         param->i_frame_reference = 16;
 282         param->i_bframe_adaptive = X264_B_ADAPT_TRELLIS;
 283         param->analyse.i_direct_mv_pred = X264_DIRECT_PRED_AUTO;
 284         param->analyse.inter |= X264_ANALYSE_PSUB8x8;
 285         param->analyse.i_trellis = 2;
 286         param->i_bframe = 8;
 287         param->rc.i_lookahead = 60;
 288     }
 289     else if( !strcasecmp( preset, "placebo" ) )
 290     {
 291         param->analyse.i_me_method = X264_ME_TESA;
 292         param->analyse.i_subpel_refine = 11;
 293         param->analyse.i_me_range = 24;
 294         param->i_frame_reference = 16;
 295         param->i_bframe_adaptive = X264_B_ADAPT_TRELLIS;
 296         param->analyse.i_direct_mv_pred = X264_DIRECT_PRED_AUTO;
 297         param->analyse.inter |= X264_ANALYSE_PSUB8x8;
 298         param->analyse.b_fast_pskip = 0;
 299         param->analyse.i_trellis = 2;
 300         param->i_bframe = 16;
 301         param->rc.i_lookahead = 60;
 302     }
 303     else
 304     {
 305         x264_log( NULL, X264_LOG_ERROR, "invalid preset '%s'\n", preset );
 306         return -1;
 307     }
 308     return 0;
 309 }
 310
 311 static int x264_param_apply_tune( x264_param_t *param, const char *tune )
 312 {
 313     char *tmp = x264_malloc( strlen( tune ) + 1 );
 314     if( !tmp )
 315         return -1;
 316     tmp = strcpy( tmp, tune );
 317     char *s = strtok( tmp, ",./-+" );
 318     int psy_tuning_used = 0;
 319     while( s )
 320     {
 321         if( !strncasecmp( s, "film", 4 ) )
 322         {
 323             if( psy_tuning_used++ ) goto psy_failure;
 324             param->i_deblocking_filter_alphac0 = -1;
 325             param->i_deblocking_filter_beta = -1;
 326             param->analyse.f_psy_trellis = 0.15;
 327         }
 328         else if( !strncasecmp( s, "animation", 9 ) )
 329         {
 330             if( psy_tuning_used++ ) goto psy_failure;
 331             param->i_frame_reference = param->i_frame_reference > 1 ? param->i_frame_reference*2 : 1;
 332             param->i_deblocking_filter_alphac0 = 1;
 333             param->i_deblocking_filter_beta = 1;
 334             param->analyse.f_psy_rd = 0.4;
 335             param->rc.f_aq_strength = 0.6;
 336             param->i_bframe += 2;
 337         }
 338         else if( !strncasecmp( s, "grain", 5 ) )
 339         {
 340             if( psy_tuning_used++ ) goto psy_failure;
 341             param->i_deblocking_filter_alphac0 = -2;
 342             param->i_deblocking_filter_beta = -2;
 343             param->analyse.f_psy_trellis = 0.25;
 344             param->analyse.b_dct_decimate = 0;
 345             param->rc.f_pb_factor = 1.1;
 346             param->rc.f_ip_factor = 1.1;
 347             param->rc.f_aq_strength = 0.5;
 348             param->analyse.i_luma_deadzone[0] = 6;
 349             param->analyse.i_luma_deadzone[1] = 6;
 350             param->rc.f_qcompress = 0.8;
 351         }
 352         else if( !strncasecmp( s, "stillimage", 10 ) )
 353         {
 354             if( psy_tuning_used++ ) goto psy_failure;
 355             param->i_deblocking_filter_alphac0 = -3;
 356             param->i_deblocking_filter_beta = -3;
 357             param->analyse.f_psy_rd = 2.0;
 358             param->analyse.f_psy_trellis = 0.7;
 359             param->rc.f_aq_strength = 1.2;
 360         }
 361         else if( !strncasecmp( s, "psnr", 4 ) )
 362         {
 363             if( psy_tuning_used++ ) goto psy_failure;
 364             param->rc.i_aq_mode = X264_AQ_NONE;
 365             param->analyse.b_psy = 0;
 366         }
 367         else if( !strncasecmp( s, "ssim", 4 ) )
 368         {
 369             if( psy_tuning_used++ ) goto psy_failure;
 370             param->rc.i_aq_mode = X264_AQ_AUTOVARIANCE;
 371             param->analyse.b_psy = 0;
 372         }
 373         else if( !strncasecmp( s, "fastdecode", 10 ) )
 374         {
 375             param->b_deblocking_filter = 0;
 376             param->b_cabac = 0;
 377             param->analyse.b_weighted_bipred = 0;
 378             param->analyse.i_weighted_pred = X264_WEIGHTP_NONE;
 379         }
 380         else if( !strncasecmp( s, "zerolatency", 11 ) )
 381         {
 382             param->rc.i_lookahead = 0;
 383             param->i_sync_lookahead = 0;
 384             param->i_bframe = 0;
 385             param->b_sliced_threads = 1;
 386             param->b_vfr_input = 0;
 387             param->rc.b_mb_tree = 0;
 388         }
 389         else if( !strncasecmp( s, "touhou", 6 ) )
 390         {
 391             if( psy_tuning_used++ ) goto psy_failure;
 392             param->i_frame_reference = param->i_frame_reference > 1 ? param->i_frame_reference*2 : 1;
 393             param->i_deblocking_filter_alphac0 = -1;
 394             param->i_deblocking_filter_beta = -1;
 395             param->analyse.f_psy_trellis = 0.2;
 396             param->rc.f_aq_strength = 1.3;
 397             if( param->analyse.inter & X264_ANALYSE_PSUB16x16 )
 398                 param->analyse.inter |= X264_ANALYSE_PSUB8x8;
 399         }
 400         else
 401         {
 402             x264_log( NULL, X264_LOG_ERROR, "invalid tune '%s'\n", s );
 403             x264_free( tmp );
 404             return -1;
 405         }
 406         if( 0 )
 407         {
 408     psy_failure:
 409             x264_log( NULL, X264_LOG_WARNING, "only 1 psy tuning can be used: ignoring tune %s\n", s );
 410         }
 411         s = strtok( NULL, ",./-+" );
 412     }
 413     x264_free( tmp );
 414     return 0;
 415 }
 416
 417 int x264_param_default_preset( x264_param_t *param, const char *preset, const char *tune )
 418 {
 419     x264_param_default( param );
 420
 421     if( preset && x264_param_apply_preset( param, preset ) < 0 )
 422         return -1;
 423     if( tune && x264_param_apply_tune( param, tune ) < 0 )
 424         return -1;
 425     return 0;
 426 }
 427
 428 void x264_param_apply_fastfirstpass( x264_param_t *param )
 429 {
 430     /* Set faster options in case of turbo firstpass. */
 431     if( param->rc.b_stat_write && !param->rc.b_stat_read )
 432     {
 433         param->i_frame_reference = 1;
 434         param->analyse.b_transform_8x8 = 0;
 435         param->analyse.inter = 0;
 436         param->analyse.i_me_method = X264_ME_DIA;
 437         param->analyse.i_subpel_refine = X264_MIN( 2, param->analyse.i_subpel_refine );
 438         param->analyse.i_trellis = 0;
 439         param->analyse.b_fast_pskip = 1;
 440     }
 441 }
 442
 443 static int profile_string_to_int( const char *str )
 444 {
 445     if( !strcasecmp( str, "baseline" ) )
 446         return PROFILE_BASELINE;
 447     if( !strcasecmp( str, "main" ) )
 448         return PROFILE_MAIN;
 449     if( !strcasecmp( str, "high" ) )
 450         return PROFILE_HIGH;
 451     if( !strcasecmp( str, "high10" ) )
 452         return PROFILE_HIGH10;
 453     if( !strcasecmp( str, "high422" ) )
 454         return PROFILE_HIGH422;
 455     if( !strcasecmp( str, "high444" ) )
 456         return PROFILE_HIGH444_PREDICTIVE;
 457     return -1;
 458 }
 459
 460 int x264_param_apply_profile( x264_param_t *param, const char *profile )
 461 {
 462     if( !profile )
 463         return 0;
 464
 465     int p = profile_string_to_int( profile );
 466     if( p < 0 )
 467     {
 468         x264_log( NULL, X264_LOG_ERROR, "invalid profile: %s\n", profile );
 469         return -1;
 470     }
 471     if( p < PROFILE_HIGH444_PREDICTIVE && ((param->rc.i_rc_method == X264_RC_CQP && param->rc.i_qp_constant <= 0) ||
 472         (param->rc.i_rc_method == X264_RC_CRF && (int)(param->rc.f_rf_constant + QP_BD_OFFSET) <= 0)) )
 473     {
 474         x264_log( NULL, X264_LOG_ERROR, "%s profile doesn't support lossless\n", profile );
 475         return -1;
 476     }
 477     if( p < PROFILE_HIGH444_PREDICTIVE && (param->i_csp & X264_CSP_MASK) >= X264_CSP_I444 )
 478     {
 479         x264_log( NULL, X264_LOG_ERROR, "%s profile doesn't support 4:4:4\n", profile );
 480         return -1;
 481     }
 482     if( p < PROFILE_HIGH422 && (param->i_csp & X264_CSP_MASK) >= X264_CSP_I422 )
 483     {
 484         x264_log( NULL, X264_LOG_ERROR, "%s profile doesn't support 4:2:2\n", profile );
 485         return -1;
 486     }
 487     if( p < PROFILE_HIGH10 && BIT_DEPTH > 8 )
 488     {
 489         x264_log( NULL, X264_LOG_ERROR, "%s profile doesn't support a bit depth of %d\n", profile, BIT_DEPTH );
 490         return -1;
 491     }
 492
 493     if( p == PROFILE_BASELINE )
 494     {
 495         param->analyse.b_transform_8x8 = 0;
 496         param->b_cabac = 0;
 497         param->i_cqm_preset = X264_CQM_FLAT;
 498         param->psz_cqm_file = NULL;
 499         param->i_bframe = 0;
 500         param->analyse.i_weighted_pred = X264_WEIGHTP_NONE;
 501         if( param->b_interlaced )
 502         {
 503             x264_log( NULL, X264_LOG_ERROR, "baseline profile doesn't support interlacing\n" );
 504             return -1;
 505         }
 506         if( param->b_fake_interlaced )
 507         {
 508             x264_log( NULL, X264_LOG_ERROR, "baseline profile doesn't support fake interlacing\n" );
 509             return -1;
 510         }
 511     }
 512     else if( p == PROFILE_MAIN )
 513     {
 514         param->analyse.b_transform_8x8 = 0;
 515         param->i_cqm_preset = X264_CQM_FLAT;
 516         param->psz_cqm_file = NULL;
 517     }
 518     return 0;
 519 }
 520
 521 static int parse_enum( const char *arg, const char * const *names, int *dst )
 522 {
 523     for( int i = 0; names[i]; i++ )
 524         if( !strcasecmp( arg, names[i] ) )
 525         {
 526             *dst = i;
 527             return 0;
 528         }
 529     return -1;
 530 }
 531
 532 static int parse_cqm( const char *str, uint8_t *cqm, int length )
 533 {
 534     int i = 0;
 535     do {
 536         int coef;
 537         if( !sscanf( str, "%d", &coef ) || coef < 1 || coef > 255 )
 538             return -1;
 539         cqm[i++] = coef;
 540     } while( i < length && (str = strchr( str, ',' )) && str++ );
 541     return (i == length) ? 0 : -1;
 542 }
 543
 544 static int x264_atobool( const char *str, int *b_error )
 545 {
 546     if( !strcmp(str, "1") ||
 547         !strcasecmp(str, "true") ||
 548         !strcasecmp(str, "yes") )
 549         return 1;
 550     if( !strcmp(str, "0") ||
 551         !strcasecmp(str, "false") ||
 552         !strcasecmp(str, "no") )
 553         return 0;
 554     *b_error = 1;
 555     return 0;
 556 }
 557
 558 static int x264_atoi( const char *str, int *b_error )
 559 {
 560     char *end;
 561     int v = strtol( str, &end, 0 );
 562     if( end == str || *end != '\0' )
 563         *b_error = 1;
 564     return v;
 565 }
 566
 567 static double x264_atof( const char *str, int *b_error )
 568 {
 569     char *end;
 570     double v = strtod( str, &end );
 571     if( end == str || *end != '\0' )
 572         *b_error = 1;
 573     return v;
 574 }
 575
 576 #define atobool(str) ( name_was_bool = 1, x264_atobool( str, &b_error ) )
 577 #undef atoi
 578 #undef atof
 579 #define atoi(str) x264_atoi( str, &b_error )
 580 #define atof(str) x264_atof( str, &b_error )
 581
 582 int x264_param_parse( x264_param_t *p, const char *name, const char *value )
 583 {
 584     char *name_buf = NULL;
 585     int b_error = 0;
 586     int errortype = X264_PARAM_BAD_VALUE;
 587     int name_was_bool;
 588     int value_was_null = !value;
 589
 590     if( !name )
 591         return X264_PARAM_BAD_NAME;
 592     if( !value )
 593         value = "true";
 594
 595     if( value[0] == '=' )
 596         value++;
 597
 598     if( strchr( name, '_' ) ) // s/_/-/g
 599     {
 600         char *c;
 601         name_buf = strdup(name);
 602         if( !name_buf )
 603             return X264_PARAM_BAD_NAME;
 604         while( (c = strchr( name_buf, '_' )) )
 605             *c = '-';
 606         name = name_buf;
 607     }
 608
 609     if( !strncmp( name, "no", 2 ) )
 610     {
 611         name += 2;
 612         if( name[0] == '-' )
 613             name++;
 614         value = atobool(value) ? "false" : "true";
 615     }
 616     name_was_bool = 0;
 617
 618 #define OPT(STR) else if( !strcmp( name, STR ) )
 619 #define OPT2(STR0, STR1) else if( !strcmp( name, STR0 ) || !strcmp( name, STR1 ) )
 620     if(0);
 621     OPT("asm")
 622     {
 623         p->cpu = isdigit(value[0]) ? atoi(value) :
 624                  !strcasecmp(value, "auto") || atobool(value) ? x264_cpu_detect() : 0;
 625         if( b_error )
 626         {
 627             char *buf = strdup( value );
 628             if( buf )
 629             {
 630                 char *tok, UNUSED *saveptr=NULL, *init;
 631                 b_error = 0;
 632                 p->cpu = 0;
 633                 for( init=buf; (tok=strtok_r(init, ",", &saveptr)); init=NULL )
 634                 {
 635                     int i = 0;
 636                     while( x264_cpu_names[i].flags && strcasecmp(tok, x264_cpu_names[i].name) )
 637                         i++;
 638                     p->cpu |= x264_cpu_names[i].flags;
 639                     if( !x264_cpu_names[i].flags )
 640                         b_error = 1;
 641                 }
 642                 free( buf );
 643                 if( (p->cpu&X264_CPU_SSSE3) && !(p->cpu&X264_CPU_SSE2_IS_SLOW) )
 644                     p->cpu |= X264_CPU_SSE2_IS_FAST;
 645             }
 646         }
 647     }
 648     OPT("threads")
 649     {
 650         if( !strcasecmp(value, "auto") )
 651             p->i_threads = X264_THREADS_AUTO;
 652         else
 653             p->i_threads = atoi(value);
 654     }
 655     OPT("lookahead-threads")
 656     {
 657         if( !strcasecmp(value, "auto") )
 658             p->i_lookahead_threads = X264_THREADS_AUTO;
 659         else
 660             p->i_lookahead_threads = atoi(value);
 661     }
 662     OPT("sliced-threads")
 663         p->b_sliced_threads = atobool(value);
 664     OPT("sync-lookahead")
 665     {
 666         if( !strcasecmp(value, "auto") )
 667             p->i_sync_lookahead = X264_SYNC_LOOKAHEAD_AUTO;
 668         else
 669             p->i_sync_lookahead = atoi(value);
 670     }
 671     OPT2("deterministic", "n-deterministic")
 672         p->b_deterministic = atobool(value);
 673     OPT("cpu-independent")
 674         p->b_cpu_independent = atobool(value);
 675     OPT2("level", "level-idc")
 676     {
 677         if( !strcmp(value, "1b") )
 678             p->i_level_idc = 9;
 679         else if( atof(value) < 6 )
 680             p->i_level_idc = (int)(10*atof(value)+.5);
 681         else
 682             p->i_level_idc = atoi(value);
 683     }
 684     OPT("bluray-compat")
 685         p->b_bluray_compat = atobool(value);
 686     OPT("avcintra-class")
 687         p->i_avcintra_class = atoi(value);
 688     OPT("sar")
 689     {
 690         b_error = ( 2 != sscanf( value, "%d:%d", &p->vui.i_sar_width, &p->vui.i_sar_height ) &&
 691                     2 != sscanf( value, "%d/%d", &p->vui.i_sar_width, &p->vui.i_sar_height ) );
 692     }
 693     OPT("overscan")
 694         b_error |= parse_enum( value, x264_overscan_names, &p->vui.i_overscan );
 695     OPT("videoformat")
 696         b_error |= parse_enum( value, x264_vidformat_names, &p->vui.i_vidformat );
 697     OPT("fullrange")
 698         b_error |= parse_enum( value, x264_fullrange_names, &p->vui.b_fullrange );
 699     OPT("colorprim")
 700         b_error |= parse_enum( value, x264_colorprim_names, &p->vui.i_colorprim );
 701     OPT("transfer")
 702         b_error |= parse_enum( value, x264_transfer_names, &p->vui.i_transfer );
 703     OPT("colormatrix")
 704         b_error |= parse_enum( value, x264_colmatrix_names, &p->vui.i_colmatrix );
 705     OPT("chromaloc")
 706     {
 707         p->vui.i_chroma_loc = atoi(value);
 708         b_error = ( p->vui.i_chroma_loc < 0 || p->vui.i_chroma_loc > 5 );
 709     }
 710     OPT("fps")
 711     {
 712         if( sscanf( value, "%u/%u", &p->i_fps_num, &p->i_fps_den ) != 2 )
 713         {
 714             double fps = atof(value);
 715             if( fps > 0.0 && fps <= INT_MAX/1000.0 )
 716             {
 717                 p->i_fps_num = (int)(fps * 1000.0 + .5);
 718                 p->i_fps_den = 1000;
 719             }
 720             else
 721             {
 722                 p->i_fps_num = atoi(value);
 723                 p->i_fps_den = 1;
 724             }
 725         }
 726     }
 727     OPT2("ref", "frameref")
 728         p->i_frame_reference = atoi(value);
 729     OPT("dpb-size")
 730         p->i_dpb_size = atoi(value);
 731     OPT("keyint")
 732     {
 733         if( strstr( value, "infinite" ) )
 734             p->i_keyint_max = X264_KEYINT_MAX_INFINITE;
 735         else
 736             p->i_keyint_max = atoi(value);
 737     }
 738     OPT2("min-keyint", "keyint-min")
 739     {
 740         p->i_keyint_min = atoi(value);
 741         if( p->i_keyint_max < p->i_keyint_min )
 742             p->i_keyint_max = p->i_keyint_min;
 743     }
 744     OPT("scenecut")
 745     {
 746         p->i_scenecut_threshold = atobool(value);
 747         if( b_error || p->i_scenecut_threshold )
 748         {
 749             b_error = 0;
 750             p->i_scenecut_threshold = atoi(value);
 751         }
 752     }
 753     OPT("intra-refresh")
 754         p->b_intra_refresh = atobool(value);
 755     OPT("bframes")
 756         p->i_bframe = atoi(value);
 757     OPT("b-adapt")
 758     {
 759         p->i_bframe_adaptive = atobool(value);
 760         if( b_error )
 761         {
 762             b_error = 0;
 763             p->i_bframe_adaptive = atoi(value);
 764         }
 765     }
 766     OPT("b-bias")
 767         p->i_bframe_bias = atoi(value);
 768     OPT("b-pyramid")
 769     {
 770         b_error |= parse_enum( value, x264_b_pyramid_names, &p->i_bframe_pyramid );
 771         if( b_error )
 772         {
 773             b_error = 0;
 774             p->i_bframe_pyramid = atoi(value);
 775         }
 776     }
 777     OPT("open-gop")
 778         p->b_open_gop = atobool(value);
 779     OPT("nf")
 780         p->b_deblocking_filter = !atobool(value);
 781     OPT2("filter", "deblock")
 782     {
 783         if( 2 == sscanf( value, "%d:%d", &p->i_deblocking_filter_alphac0, &p->i_deblocking_filter_beta ) ||
 784             2 == sscanf( value, "%d,%d", &p->i_deblocking_filter_alphac0, &p->i_deblocking_filter_beta ) )
 785         {
 786             p->b_deblocking_filter = 1;
 787         }
 788         else if( sscanf( value, "%d", &p->i_deblocking_filter_alphac0 ) )
 789         {
 790             p->b_deblocking_filter = 1;
 791             p->i_deblocking_filter_beta = p->i_deblocking_filter_alphac0;
 792         }
 793         else
 794             p->b_deblocking_filter = atobool(value);
 795     }
 796     OPT("slice-max-size")
 797         p->i_slice_max_size = atoi(value);
 798     OPT("slice-max-mbs")
 799         p->i_slice_max_mbs = atoi(value);
 800     OPT("slice-min-mbs")
 801         p->i_slice_min_mbs = atoi(value);
 802     OPT("slices")
 803         p->i_slice_count = atoi(value);
 804     OPT("slices-max")
 805         p->i_slice_count_max = atoi(value);
 806     OPT("cabac")
 807         p->b_cabac = atobool(value);
 808     OPT("cabac-idc")
 809         p->i_cabac_init_idc = atoi(value);
 810     OPT("interlaced")
 811         p->b_interlaced = atobool(value);
 812     OPT("tff")
 813         p->b_interlaced = p->b_tff = atobool(value);
 814     OPT("bff")
 815     {
 816         p->b_interlaced = atobool(value);
 817         p->b_tff = !p->b_interlaced;
 818     }
 819     OPT("constrained-intra")
 820         p->b_constrained_intra = atobool(value);
 821     OPT("cqm")
 822     {
 823         if( strstr( value, "flat" ) )
 824             p->i_cqm_preset = X264_CQM_FLAT;
 825         else if( strstr( value, "jvt" ) )
 826             p->i_cqm_preset = X264_CQM_JVT;
 827         else
 828             p->psz_cqm_file = strdup(value);
 829     }
 830     OPT("cqmfile")
 831         p->psz_cqm_file = strdup(value);
 832     OPT("cqm4")
 833     {
 834         p->i_cqm_preset = X264_CQM_CUSTOM;
 835         b_error |= parse_cqm( value, p->cqm_4iy, 16 );
 836         b_error |= parse_cqm( value, p->cqm_4py, 16 );
 837         b_error |= parse_cqm( value, p->cqm_4ic, 16 );
 838         b_error |= parse_cqm( value, p->cqm_4pc, 16 );
 839     }
 840     OPT("cqm8")
 841     {
 842         p->i_cqm_preset = X264_CQM_CUSTOM;
 843         b_error |= parse_cqm( value, p->cqm_8iy, 64 );
 844         b_error |= parse_cqm( value, p->cqm_8py, 64 );
 845         b_error |= parse_cqm( value, p->cqm_8ic, 64 );
 846         b_error |= parse_cqm( value, p->cqm_8pc, 64 );
 847     }
 848     OPT("cqm4i")
 849     {
 850         p->i_cqm_preset = X264_CQM_CUSTOM;
 851         b_error |= parse_cqm( value, p->cqm_4iy, 16 );
 852         b_error |= parse_cqm( value, p->cqm_4ic, 16 );
 853     }
 854     OPT("cqm4p")
 855     {
 856         p->i_cqm_preset = X264_CQM_CUSTOM;
 857         b_error |= parse_cqm( value, p->cqm_4py, 16 );
 858         b_error |= parse_cqm( value, p->cqm_4pc, 16 );
 859     }
 860     OPT("cqm4iy")
 861     {
 862         p->i_cqm_preset = X264_CQM_CUSTOM;
 863         b_error |= parse_cqm( value, p->cqm_4iy, 16 );
 864     }
 865     OPT("cqm4ic")
 866     {
 867         p->i_cqm_preset = X264_CQM_CUSTOM;
 868         b_error |= parse_cqm( value, p->cqm_4ic, 16 );
 869     }
 870     OPT("cqm4py")
 871     {
 872         p->i_cqm_preset = X264_CQM_CUSTOM;
 873         b_error |= parse_cqm( value, p->cqm_4py, 16 );
 874     }
 875     OPT("cqm4pc")
 876     {
 877         p->i_cqm_preset = X264_CQM_CUSTOM;
 878         b_error |= parse_cqm( value, p->cqm_4pc, 16 );
 879     }
 880     OPT("cqm8i")
 881     {
 882         p->i_cqm_preset = X264_CQM_CUSTOM;
 883         b_error |= parse_cqm( value, p->cqm_8iy, 64 );
 884         b_error |= parse_cqm( value, p->cqm_8ic, 64 );
 885     }
 886     OPT("cqm8p")
 887     {
 888         p->i_cqm_preset = X264_CQM_CUSTOM;
 889         b_error |= parse_cqm( value, p->cqm_8py, 64 );
 890         b_error |= parse_cqm( value, p->cqm_8pc, 64 );
 891     }
 892     OPT("log")
 893         p->i_log_level = atoi(value);
 894     OPT("dump-yuv")
 895         p->psz_dump_yuv = strdup(value);
 896     OPT2("analyse", "partitions")
 897     {
 898         p->analyse.inter = 0;
 899         if( strstr( value, "none" ) )  p->analyse.inter =  0;
 900         if( strstr( value, "all" ) )   p->analyse.inter = ~0;
 901
 902         if( strstr( value, "i4x4" ) )  p->analyse.inter |= X264_ANALYSE_I4x4;
 903         if( strstr( value, "i8x8" ) )  p->analyse.inter |= X264_ANALYSE_I8x8;
 904         if( strstr( value, "p8x8" ) )  p->analyse.inter |= X264_ANALYSE_PSUB16x16;
 905         if( strstr( value, "p4x4" ) )  p->analyse.inter |= X264_ANALYSE_PSUB8x8;
 906         if( strstr( value, "b8x8" ) )  p->analyse.inter |= X264_ANALYSE_BSUB16x16;
 907     }
 908     OPT("8x8dct")
 909         p->analyse.b_transform_8x8 = atobool(value);
 910     OPT2("weightb", "weight-b")
 911         p->analyse.b_weighted_bipred = atobool(value);
 912     OPT("weightp")
 913         p->analyse.i_weighted_pred = atoi(value);
 914     OPT2("direct", "direct-pred")
 915         b_error |= parse_enum( value, x264_direct_pred_names, &p->analyse.i_direct_mv_pred );
 916     OPT("chroma-qp-offset")
 917         p->analyse.i_chroma_qp_offset = atoi(value);
 918     OPT("me")
 919         b_error |= parse_enum( value, x264_motion_est_names, &p->analyse.i_me_method );
 920     OPT2("merange", "me-range")
 921         p->analyse.i_me_range = atoi(value);
 922     OPT2("mvrange", "mv-range")
 923         p->analyse.i_mv_range = atoi(value);
 924     OPT2("mvrange-thread", "mv-range-thread")
 925         p->analyse.i_mv_range_thread = atoi(value);
 926     OPT2("subme", "subq")
 927         p->analyse.i_subpel_refine = atoi(value);
 928     OPT("psy-rd")
 929     {
 930         if( 2 == sscanf( value, "%f:%f", &p->analyse.f_psy_rd, &p->analyse.f_psy_trellis ) ||
 931             2 == sscanf( value, "%f,%f", &p->analyse.f_psy_rd, &p->analyse.f_psy_trellis ) ||
 932             2 == sscanf( value, "%f|%f", &p->analyse.f_psy_rd, &p->analyse.f_psy_trellis ))
 933         { }
 934         else if( sscanf( value, "%f", &p->analyse.f_psy_rd ) )
 935         {
 936             p->analyse.f_psy_trellis = 0;
 937         }
 938         else
 939         {
 940             p->analyse.f_psy_rd = 0;
 941             p->analyse.f_psy_trellis = 0;
 942         }
 943     }
 944     OPT("psy")
 945         p->analyse.b_psy = atobool(value);
 946     OPT("chroma-me")
 947         p->analyse.b_chroma_me = atobool(value);
 948     OPT("mixed-refs")
 949         p->analyse.b_mixed_references = atobool(value);
 950     OPT("trellis")
 951         p->analyse.i_trellis = atoi(value);
 952     OPT("fast-pskip")
 953         p->analyse.b_fast_pskip = atobool(value);
 954     OPT("dct-decimate")
 955         p->analyse.b_dct_decimate = atobool(value);
 956     OPT("deadzone-inter")
 957         p->analyse.i_luma_deadzone[0] = atoi(value);
 958     OPT("deadzone-intra")
 959         p->analyse.i_luma_deadzone[1] = atoi(value);
 960     OPT("nr")
 961         p->analyse.i_noise_reduction = atoi(value);
 962     OPT("bitrate")
 963     {
 964         p->rc.i_bitrate = atoi(value);
 965         p->rc.i_rc_method = X264_RC_ABR;
 966     }
 967     OPT2("qp", "qp_constant")
 968     {
 969         p->rc.i_qp_constant = atoi(value);
 970         p->rc.i_rc_method = X264_RC_CQP;
 971     }
 972     OPT("crf")
 973     {
 974         p->rc.f_rf_constant = atof(value);
 975         p->rc.i_rc_method = X264_RC_CRF;
 976     }
 977     OPT("crf-max")
 978         p->rc.f_rf_constant_max = atof(value);
 979     OPT("rc-lookahead")
 980         p->rc.i_lookahead = atoi(value);
 981     OPT2("qpmin", "qp-min")
 982         p->rc.i_qp_min = atoi(value);
 983     OPT2("qpmax", "qp-max")
 984         p->rc.i_qp_max = atoi(value);
 985     OPT2("qpstep", "qp-step")
 986         p->rc.i_qp_step = atoi(value);
 987     OPT("ratetol")
 988         p->rc.f_rate_tolerance = !strncmp("inf", value, 3) ? 1e9 : atof(value);
 989     OPT("vbv-maxrate")
 990         p->rc.i_vbv_max_bitrate = atoi(value);
 991     OPT("vbv-bufsize")
 992         p->rc.i_vbv_buffer_size = atoi(value);
 993     OPT("vbv-init")
 994         p->rc.f_vbv_buffer_init = atof(value);
 995     OPT2("ipratio", "ip-factor")
 996         p->rc.f_ip_factor = atof(value);
 997     OPT2("pbratio", "pb-factor")
 998         p->rc.f_pb_factor = atof(value);
 999     OPT("aq-mode")
1000         p->rc.i_aq_mode = atoi(value);
1001     OPT("aq-strength")
1002         p->rc.f_aq_strength = atof(value);
1003     OPT("pass")
1004     {
1005         int pass = x264_clip3( atoi(value), 0, 3 );
1006         p->rc.b_stat_write = pass & 1;
1007         p->rc.b_stat_read = pass & 2;
1008     }
1009     OPT("stats")
1010     {
1011         p->rc.psz_stat_in = strdup(value);
1012         p->rc.psz_stat_out = strdup(value);
1013     }
1014     OPT("qcomp")
1015         p->rc.f_qcompress = atof(value);
1016     OPT("mbtree")
1017         p->rc.b_mb_tree = atobool(value);
1018     OPT("qblur")
1019         p->rc.f_qblur = atof(value);
1020     OPT2("cplxblur", "cplx-blur")
1021         p->rc.f_complexity_blur = atof(value);
1022     OPT("zones")
1023         p->rc.psz_zones = strdup(value);
1024     OPT("speed")
1025         p->sc.f_speed = atof(value);
1026     OPT("speed-bufsize")
1027         p->sc.i_buffer_size = atoi(value);
1028     OPT("speed-init")
1029         p->sc.f_buffer_init = atof(value);
1030     OPT("speed-alt-timer")
1031         p->sc.b_alt_timer = atobool(value);
1032     OPT("crop-rect")
1033         b_error |= sscanf( value, "%u,%u,%u,%u", &p->crop_rect.i_left, &p->crop_rect.i_top,
1034                                                  &p->crop_rect.i_right, &p->crop_rect.i_bottom ) != 4;
1035     OPT("psnr")
1036         p->analyse.b_psnr = atobool(value);
1037     OPT("ssim")
1038         p->analyse.b_ssim = atobool(value);
1039     OPT("aud")
1040         p->b_aud = atobool(value);
1041     OPT("sps-id")
1042         p->i_sps_id = atoi(value);
1043     OPT("global-header")
1044         p->b_repeat_headers = !atobool(value);
1045     OPT("repeat-headers")
1046         p->b_repeat_headers = atobool(value);
1047     OPT("annexb")
1048         p->b_annexb = atobool(value);
1049     OPT("force-cfr")
1050         p->b_vfr_input = !atobool(value);
1051     OPT("nal-hrd")
1052         b_error |= parse_enum( value, x264_nal_hrd_names, &p->i_nal_hrd );
1053     OPT("filler")
1054         p->rc.b_filler = atobool(value);
1055     OPT("pic-struct")
1056         p->b_pic_struct = atobool(value);
1057     OPT("fake-interlaced")
1058         p->b_fake_interlaced = atobool(value);
1059     OPT("frame-packing")
1060         p->i_frame_packing = atoi(value);
1061     OPT("stitchable")
1062         p->b_stitchable = atobool(value);
1063     OPT("opencl")
1064         p->b_opencl = atobool( value );
1065     OPT("opencl-clbin")
1066         p->psz_clbin_file = strdup( value );
1067     OPT("opencl-device")
1068         p->i_opencl_device = atoi( value );
1069     else
1070     {
1071         b_error = 1;
1072         errortype = X264_PARAM_BAD_NAME;
1073     }
1074 #undef OPT
1075 #undef OPT2
1076 #undef atobool
1077 #undef atoi
1078 #undef atof
1079
1080     if( name_buf )
1081         free( name_buf );
1082
1083     b_error |= value_was_null && !name_was_bool;
1084     return b_error ? errortype : 0;
1085 }
1086
1087 /****************************************************************************
1088  * x264_log:
1089  ****************************************************************************/
1090 void x264_log( x264_t *h, int i_level, const char *psz_fmt, ... )
1091 {
1092     if( !h || i_level <= h->param.i_log_level )
1093     {
1094         va_list arg;
1095         va_start( arg, psz_fmt );
1096         if( !h )
1097             x264_log_default( NULL, i_level, psz_fmt, arg );
1098         else
1099             h->param.pf_log( h->param.p_log_private, i_level, psz_fmt, arg );
1100         va_end( arg );
1101     }
1102 }
1103
1104 static void x264_log_default( void *p_unused, int i_level, const char *psz_fmt, va_list arg )
1105 {
1106     char *psz_prefix;
1107     switch( i_level )
1108     {
1109         case X264_LOG_ERROR:
1110             psz_prefix = "error";
1111             break;
1112         case X264_LOG_WARNING:
1113             psz_prefix = "warning";
1114             break;
1115         case X264_LOG_INFO:
1116             psz_prefix = "info";
1117             break;
1118         case X264_LOG_DEBUG:
1119             psz_prefix = "debug";
1120             break;
1121         default:
1122             psz_prefix = "unknown";
1123             break;
1124     }
1125     fprintf( stderr, "x264 [%s]: ", psz_prefix );
1126     x264_vfprintf( stderr, psz_fmt, arg );
1127 }
1128
1129 /****************************************************************************
1130  * x264_picture_init:
1131  ****************************************************************************/
1132 void x264_picture_init( x264_picture_t *pic )
1133 {
1134     memset( pic, 0, sizeof( x264_picture_t ) );
1135     pic->i_type = X264_TYPE_AUTO;
1136     pic->i_qpplus1 = X264_QP_AUTO;
1137     pic->i_pic_struct = PIC_STRUCT_AUTO;
1138 }
1139
1140 /****************************************************************************
1141  * x264_picture_alloc:
1142  ****************************************************************************/
1143 int x264_picture_alloc( x264_picture_t *pic, int i_csp, int i_width, int i_height )
1144 {
1145     typedef struct
1146     {
1147         int planes;
1148         int width_fix8[3];
1149         int height_fix8[3];
1150     } x264_csp_tab_t;
1151
1152     static const x264_csp_tab_t x264_csp_tab[] =
1153     {
1154         [X264_CSP_I420] = { 3, { 256*1, 256/2, 256/2 }, { 256*1, 256/2, 256/2 } },
1155         [X264_CSP_YV12] = { 3, { 256*1, 256/2, 256/2 }, { 256*1, 256/2, 256/2 } },
1156         [X264_CSP_NV12] = { 2, { 256*1, 256*1 },        { 256*1, 256/2 },       },
1157         [X264_CSP_NV21] = { 2, { 256*1, 256*1 },        { 256*1, 256/2 },       },
1158         [X264_CSP_I422] = { 3, { 256*1, 256/2, 256/2 }, { 256*1, 256*1, 256*1 } },
1159         [X264_CSP_YV16] = { 3, { 256*1, 256/2, 256/2 }, { 256*1, 256*1, 256*1 } },
1160         [X264_CSP_NV16] = { 2, { 256*1, 256*1 },        { 256*1, 256*1 },       },
1161         [X264_CSP_I444] = { 3, { 256*1, 256*1, 256*1 }, { 256*1, 256*1, 256*1 } },
1162         [X264_CSP_YV24] = { 3, { 256*1, 256*1, 256*1 }, { 256*1, 256*1, 256*1 } },
1163         [X264_CSP_BGR]  = { 1, { 256*3 },               { 256*1 },              },
1164         [X264_CSP_BGRA] = { 1, { 256*4 },               { 256*1 },              },
1165         [X264_CSP_RGB]  = { 1, { 256*3 },               { 256*1 },              },
1166     };
1167
1168     int csp = i_csp & X264_CSP_MASK;
1169     if( csp <= X264_CSP_NONE || csp >= X264_CSP_MAX || csp == X264_CSP_V210 )
1170         return -1;
1171     x264_picture_init( pic );
1172     pic->img.i_csp = i_csp;
1173     pic->img.i_plane = x264_csp_tab[csp].planes;
1174     int depth_factor = i_csp & X264_CSP_HIGH_DEPTH ? 2 : 1;
1175     int plane_offset[3] = {0};
1176     int frame_size = 0;
1177     for( int i = 0; i < pic->img.i_plane; i++ )
1178     {
1179         int stride = (((int64_t)i_width * x264_csp_tab[csp].width_fix8[i]) >> 8) * depth_factor;
1180         int plane_size = (((int64_t)i_height * x264_csp_tab[csp].height_fix8[i]) >> 8) * stride;
1181         pic->img.i_stride[i] = stride;
1182         plane_offset[i] = frame_size;
1183         frame_size += plane_size;
1184     }
1185     pic->img.plane[0] = x264_malloc( frame_size );
1186     if( !pic->img.plane[0] )
1187         return -1;
1188     for( int i = 1; i < pic->img.i_plane; i++ )
1189         pic->img.plane[i] = pic->img.plane[0] + plane_offset[i];
1190     return 0;
1191 }
1192
1193 /****************************************************************************
1194  * x264_picture_clean:
1195  ****************************************************************************/
1196 void x264_picture_clean( x264_picture_t *pic )
1197 {
1198     x264_free( pic->img.plane[0] );
1199
1200     /* just to be safe */
1201     memset( pic, 0, sizeof( x264_picture_t ) );
1202 }
1203
1204 /****************************************************************************
1205  * x264_malloc:
1206  ****************************************************************************/
1207 void *x264_malloc( int i_size )
1208 {
1209     uint8_t *align_buf = NULL;
1210 #if HAVE_MALLOC_H
1211 #if HAVE_THP
1212 #define HUGE_PAGE_SIZE 2*1024*1024
1213 #define HUGE_PAGE_THRESHOLD HUGE_PAGE_SIZE*7/8 /* FIXME: Is this optimal? */
1214     /* Attempt to allocate huge pages to reduce TLB misses. */
1215     if( i_size >= HUGE_PAGE_THRESHOLD )
1216     {
1217         align_buf = memalign( HUGE_PAGE_SIZE, i_size );
1218         if( align_buf )
1219         {
1220             /* Round up to the next huge page boundary if we are close enough. */
1221             size_t madv_size = (i_size + HUGE_PAGE_SIZE - HUGE_PAGE_THRESHOLD) & ~(HUGE_PAGE_SIZE-1);
1222             madvise( align_buf, madv_size, MADV_HUGEPAGE );
1223         }
1224     }
1225     else
1226 #undef HUGE_PAGE_SIZE
1227 #undef HUGE_PAGE_THRESHOLD
1228 #endif
1229         align_buf = memalign( NATIVE_ALIGN, i_size );
1230 #else
1231     uint8_t *buf = malloc( i_size + (NATIVE_ALIGN-1) + sizeof(void **) );
1232     if( buf )
1233     {
1234         align_buf = buf + (NATIVE_ALIGN-1) + sizeof(void **);
1235         align_buf -= (intptr_t) align_buf & (NATIVE_ALIGN-1);
1236         *( (void **) ( align_buf - sizeof(void **) ) ) = buf;
1237     }
1238 #endif
1239     if( !align_buf )
1240         x264_log( NULL, X264_LOG_ERROR, "malloc of size %d failed\n", i_size );
1241     return align_buf;
1242 }
1243
1244 /****************************************************************************
1245  * x264_free:
1246  ****************************************************************************/
1247 void x264_free( void *p )
1248 {
1249     if( p )
1250     {
1251 #if HAVE_MALLOC_H
1252         free( p );
1253 #else
1254         free( *( ( ( void **) p ) - 1 ) );
1255 #endif
1256     }
1257 }
1258
1259 /****************************************************************************
1260  * x264_reduce_fraction:
1261  ****************************************************************************/
1262 #define REDUCE_FRACTION( name, type )\
1263 void name( type *n, type *d )\
1264 {                   \
1265     type a = *n;    \
1266     type b = *d;    \
1267     type c;         \
1268     if( !a || !b )  \
1269         return;     \
1270     c = a % b;      \
1271     while( c )      \
1272     {               \
1273         a = b;      \
1274         b = c;      \
1275         c = a % b;  \
1276     }               \
1277     *n /= b;        \
1278     *d /= b;        \
1279 }
1280
1281 REDUCE_FRACTION( x264_reduce_fraction  , uint32_t )
1282 REDUCE_FRACTION( x264_reduce_fraction64, uint64_t )
1283
1284 /****************************************************************************
1285  * x264_slurp_file:
1286  ****************************************************************************/
1287 char *x264_slurp_file( const char *filename )
1288 {
1289     int b_error = 0;
1290     int64_t i_size;
1291     char *buf;
1292     FILE *fh = x264_fopen( filename, "rb" );
1293     if( !fh )
1294         return NULL;
1295
1296     b_error |= fseek( fh, 0, SEEK_END ) < 0;
1297     b_error |= ( i_size = ftell( fh ) ) <= 0;
1298     if( WORD_SIZE == 4 )
1299         b_error |= i_size > INT32_MAX;
1300     b_error |= fseek( fh, 0, SEEK_SET ) < 0;
1301     if( b_error )
1302         goto error;
1303
1304     buf = x264_malloc( i_size+2 );
1305     if( !buf )
1306         goto error;
1307
1308     b_error |= fread( buf, 1, i_size, fh ) != i_size;
1309     fclose( fh );
1310     if( b_error )
1311     {
1312         x264_free( buf );
1313         return NULL;
1314     }
1315
1316     if( buf[i_size-1] != '\n' )
1317         buf[i_size++] = '\n';
1318     buf[i_size] = '\0';
1319
1320     return buf;
1321 error:
1322     fclose( fh );
1323     return NULL;
1324 }
1325
1326 /****************************************************************************
1327  * x264_param2string:
1328  ****************************************************************************/
1329 char *x264_param2string( x264_param_t *p, int b_res )
1330 {
1331     int len = 1000;
1332     char *buf, *s;
1333     if( p->rc.psz_zones )
1334         len += strlen(p->rc.psz_zones);
1335     buf = s = x264_malloc( len );
1336     if( !buf )
1337         return NULL;
1338
1339     if( b_res )
1340     {
1341         s += sprintf( s, "%dx%d ", p->i_width, p->i_height );
1342         s += sprintf( s, "fps=%u/%u ", p->i_fps_num, p->i_fps_den );
1343         s += sprintf( s, "timebase=%u/%u ", p->i_timebase_num, p->i_timebase_den );
1344         s += sprintf( s, "bitdepth=%d ", BIT_DEPTH );
1345     }
1346
1347     if( p->b_opencl )
1348         s += sprintf( s, "opencl=%d ", p->b_opencl );
1349
1350     // FIXME speedcontrol stuff
1351
1352     s += sprintf( s, "cabac=%d", p->b_cabac );
1353     s += sprintf( s, " ref=%d", p->i_frame_reference );
1354     s += sprintf( s, " deblock=%d:%d:%d", p->b_deblocking_filter,
1355                   p->i_deblocking_filter_alphac0, p->i_deblocking_filter_beta );
1356     s += sprintf( s, " analyse=%#x:%#x", p->analyse.intra, p->analyse.inter );
1357     s += sprintf( s, " me=%s", x264_motion_est_names[ p->analyse.i_me_method ] );
1358     s += sprintf( s, " subme=%d", p->analyse.i_subpel_refine );
1359     s += sprintf( s, " psy=%d", p->analyse.b_psy );
1360     if( p->analyse.b_psy )
1361         s += sprintf( s, " psy_rd=%.2f:%.2f", p->analyse.f_psy_rd, p->analyse.f_psy_trellis );
1362     s += sprintf( s, " mixed_ref=%d", p->analyse.b_mixed_references );
1363     s += sprintf( s, " me_range=%d", p->analyse.i_me_range );
1364     s += sprintf( s, " chroma_me=%d", p->analyse.b_chroma_me );
1365     s += sprintf( s, " trellis=%d", p->analyse.i_trellis );
1366     s += sprintf( s, " 8x8dct=%d", p->analyse.b_transform_8x8 );
1367     s += sprintf( s, " cqm=%d", p->i_cqm_preset );
1368     s += sprintf( s, " deadzone=%d,%d", p->analyse.i_luma_deadzone[0], p->analyse.i_luma_deadzone[1] );
1369     s += sprintf( s, " fast_pskip=%d", p->analyse.b_fast_pskip );
1370     s += sprintf( s, " chroma_qp_offset=%d", p->analyse.i_chroma_qp_offset );
1371     s += sprintf( s, " threads=%d", p->i_threads );
1372     s += sprintf( s, " lookahead_threads=%d", p->i_lookahead_threads );
1373     s += sprintf( s, " sliced_threads=%d", p->b_sliced_threads );
1374     if( p->i_slice_count )
1375         s += sprintf( s, " slices=%d", p->i_slice_count );
1376     if( p->i_slice_count_max )
1377         s += sprintf( s, " slices_max=%d", p->i_slice_count_max );
1378     if( p->i_slice_max_size )
1379         s += sprintf( s, " slice_max_size=%d", p->i_slice_max_size );
1380     if( p->i_slice_max_mbs )
1381         s += sprintf( s, " slice_max_mbs=%d", p->i_slice_max_mbs );
1382     if( p->i_slice_min_mbs )
1383         s += sprintf( s, " slice_min_mbs=%d", p->i_slice_min_mbs );
1384     s += sprintf( s, " nr=%d", p->analyse.i_noise_reduction );
1385     s += sprintf( s, " decimate=%d", p->analyse.b_dct_decimate );
1386     s += sprintf( s, " interlaced=%s", p->b_interlaced ? p->b_tff ? "tff" : "bff" : p->b_fake_interlaced ? "fake" : "0" );
1387     s += sprintf( s, " bluray_compat=%d", p->b_bluray_compat );
1388     if( p->b_stitchable )
1389         s += sprintf( s, " stitchable=%d", p->b_stitchable );
1390
1391     s += sprintf( s, " constrained_intra=%d", p->b_constrained_intra );
1392
1393     s += sprintf( s, " bframes=%d", p->i_bframe );
1394     if( p->i_bframe )
1395     {
1396         s += sprintf( s, " b_pyramid=%d b_adapt=%d b_bias=%d direct=%d weightb=%d open_gop=%d",
1397                       p->i_bframe_pyramid, p->i_bframe_adaptive, p->i_bframe_bias,
1398                       p->analyse.i_direct_mv_pred, p->analyse.b_weighted_bipred, p->b_open_gop );
1399     }
1400     s += sprintf( s, " weightp=%d", p->analyse.i_weighted_pred > 0 ? p->analyse.i_weighted_pred : 0 );
1401
1402     if( p->i_keyint_max == X264_KEYINT_MAX_INFINITE )
1403         s += sprintf( s, " keyint=infinite" );
1404     else
1405         s += sprintf( s, " keyint=%d", p->i_keyint_max );
1406     s += sprintf( s, " keyint_min=%d scenecut=%d intra_refresh=%d",
1407                   p->i_keyint_min, p->i_scenecut_threshold, p->b_intra_refresh );
1408
1409     if( p->rc.b_mb_tree || p->rc.i_vbv_buffer_size )
1410         s += sprintf( s, " rc_lookahead=%d", p->rc.i_lookahead );
1411
1412     s += sprintf( s, " rc=%s mbtree=%d", p->rc.i_rc_method == X264_RC_ABR ?
1413                                ( p->rc.b_stat_read ? "2pass" : p->rc.i_vbv_max_bitrate == p->rc.i_bitrate ? "cbr" : "abr" )
1414                                : p->rc.i_rc_method == X264_RC_CRF ? "crf" : "cqp", p->rc.b_mb_tree );
1415     if( p->rc.i_rc_method == X264_RC_ABR || p->rc.i_rc_method == X264_RC_CRF )
1416     {
1417         if( p->rc.i_rc_method == X264_RC_CRF )
1418             s += sprintf( s, " crf=%.1f", p->rc.f_rf_constant );
1419         else
1420             s += sprintf( s, " bitrate=%d ratetol=%.1f",
1421                           p->rc.i_bitrate, p->rc.f_rate_tolerance );
1422         s += sprintf( s, " qcomp=%.2f qpmin=%d qpmax=%d qpstep=%d",
1423                       p->rc.f_qcompress, p->rc.i_qp_min, p->rc.i_qp_max, p->rc.i_qp_step );
1424         if( p->rc.b_stat_read )
1425             s += sprintf( s, " cplxblur=%.1f qblur=%.1f",
1426                           p->rc.f_complexity_blur, p->rc.f_qblur );
1427         if( p->rc.i_vbv_buffer_size )
1428         {
1429             s += sprintf( s, " vbv_maxrate=%d vbv_bufsize=%d",
1430                           p->rc.i_vbv_max_bitrate, p->rc.i_vbv_buffer_size );
1431             if( p->rc.i_rc_method == X264_RC_CRF )
1432                 s += sprintf( s, " crf_max=%.1f", p->rc.f_rf_constant_max );
1433         }
1434     }
1435     else if( p->rc.i_rc_method == X264_RC_CQP )
1436         s += sprintf( s, " qp=%d", p->rc.i_qp_constant );
1437
1438     if( p->rc.i_vbv_buffer_size )
1439         s += sprintf( s, " nal_hrd=%s filler=%d", x264_nal_hrd_names[p->i_nal_hrd], p->rc.b_filler );
1440     if( p->crop_rect.i_left | p->crop_rect.i_top | p->crop_rect.i_right | p->crop_rect.i_bottom )
1441         s += sprintf( s, " crop_rect=%u,%u,%u,%u", p->crop_rect.i_left, p->crop_rect.i_top,
1442                                                    p->crop_rect.i_right, p->crop_rect.i_bottom );
1443     if( p->i_frame_packing >= 0 )
1444         s += sprintf( s, " frame-packing=%d", p->i_frame_packing );
1445
1446     if( !(p->rc.i_rc_method == X264_RC_CQP && p->rc.i_qp_constant == 0) )
1447     {
1448         s += sprintf( s, " ip_ratio=%.2f", p->rc.f_ip_factor );
1449         if( p->i_bframe && !p->rc.b_mb_tree )
1450             s += sprintf( s, " pb_ratio=%.2f", p->rc.f_pb_factor );
1451         s += sprintf( s, " aq=%d", p->rc.i_aq_mode );
1452         if( p->rc.i_aq_mode )
1453             s += sprintf( s, ":%.2f", p->rc.f_aq_strength );
1454         if( p->rc.psz_zones )
1455             s += sprintf( s, " zones=%s", p->rc.psz_zones );
1456         else if( p->rc.i_zones )
1457             s += sprintf( s, " zones" );
1458     }
1459
1460     return buf;
1461 }
1462