开通博客赚积分
发布资源赚积分
分类

源码开发语言/平台
当前位置：首页 > 源码/资料 > 多媒体 > 流媒体/Mpeg4/MP4 > 查看源码
common.c
资源名称：X264CODEC.rar　[点击查看]
上传用户：lctgjx
上传日期：2022-06-04
资源大小：8887k
文件大小：30k
源码类别：

流媒体/Mpeg4/MP4

开发平台：

Visual C++

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