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

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