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

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