/***************************************************************************** * common.c: h264 library ***************************************************************************** * Copyright (C) 2003 Laurent Aimar * $Id: common.c,v 1.1 2004/06/03 19:27:06 fenrir Exp $ * * Authors: Laurent Aimar <fenrir@via.ecp.fr> * * This program is free software; you can redistribute it and/or modify * it under the terms of the GNU General Public License as published by * the Free Software Foundation; either version 2 of the License, or * (at your option) any later version. * * This program is distributed in the hope that it will be useful, * but WITHOUT ANY WARRANTY; without even the implied warranty of * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the * GNU General Public License for more details. * * You should have received a copy of the GNU General Public License * along with this program; if not, write to the Free Software * Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA  02111, USA. *****************************************************************************/#include <stdio.h>#include <string.h>#include <stdarg.h>#ifdef HAVE_MALLOC_H#include <malloc.h>#endif#include "common.h"#include "cpu.h"static void x264_log_default( void *, int, const char *, va_list );/**************************************************************************** * x264_param_default: ****************************************************************************/void    x264_param_default( x264_param_t *param ){    /* */    memset( param, 0, sizeof( x264_param_t ) );    /* CPU autodetect */    param->cpu = x264_cpu_detect();    param->i_threads = 1;    /* Video properties */    param->i_csp           = X264_CSP_I420;    param->i_width         = 0;    param->i_height        = 0;    param->vui.i_sar_width = 0;    param->vui.i_sar_height= 0;    param->vui.i_overscan  = 0;  /* undef */    param->vui.i_vidformat = 5;  /* undef */    param->vui.b_fullrange = 0;  /* off */    param->vui.i_colorprim = 2;  /* undef */    param->vui.i_transfer  = 2;  /* undef */    param->vui.i_colmatrix = 2;  /* undef */    param->vui.i_chroma_loc= 0;  /* left center */    param->i_fps_num       = 25;    param->i_fps_den       = 1;    param->i_level_idc     = 51; /* as close to "unrestricted" as we can get */    /* Encoder parameters */    param->i_frame_reference = 1;    param->i_keyint_max = 250;    param->i_keyint_min = 25;    param->i_bframe = 0;    param->i_scenecut_threshold = 40;    param->b_bframe_adaptive = 1;    param->i_bframe_bias = 0;    param->b_bframe_pyramid = 0;    param->b_deblocking_filter = 1;    param->i_deblocking_filter_alphac0 = 0;    param->i_deblocking_filter_beta = 0;    param->b_cabac = 1;    param->i_cabac_init_idc = 0;    param->rc.b_cbr = 0;    param->rc.i_bitrate = 0;    param->rc.f_rate_tolerance = 1.0;    param->rc.i_vbv_max_bitrate = 0;    param->rc.i_vbv_buffer_size = 0;    param->rc.f_vbv_buffer_init = 0.9;    param->rc.i_qp_constant = 26;    param->rc.i_rf_constant = 0;    param->rc.i_qp_min = 10;    param->rc.i_qp_max = 51;    param->rc.i_qp_step = 4;    param->rc.f_ip_factor = 1.4;    param->rc.f_pb_factor = 1.3;    param->rc.b_stat_write = 0;    param->rc.psz_stat_out = "x264_2pass.log";    param->rc.b_stat_read = 0;    param->rc.psz_stat_in = "x264_2pass.log";    param->rc.psz_rc_eq = "blurCplx^(1-qComp)";    param->rc.f_qcompress = 0.6;    param->rc.f_qblur = 0.5;    param->rc.f_complexity_blur = 20;    param->rc.i_zones = 0;    /* Log */    param->pf_log = x264_log_default;    param->p_log_private = NULL;    param->i_log_level = X264_LOG_INFO;    /* */    param->analyse.intra = X264_ANALYSE_I4x4 | X264_ANALYSE_I8x8;    param->analyse.inter = X264_ANALYSE_I4x4 | X264_ANALYSE_I8x8                         | X264_ANALYSE_PSUB16x16 | X264_ANALYSE_BSUB16x16;    param->analyse.i_direct_mv_pred = X264_DIRECT_PRED_SPATIAL;    param->analyse.i_me_method = X264_ME_HEX;    param->analyse.i_me_range = 16;    param->analyse.i_subpel_refine = 5;    param->analyse.b_chroma_me = 1;    param->analyse.i_mv_range = -1; // set from level_idc    param->analyse.i_chroma_qp_offset = 0;    param->analyse.b_fast_pskip = 1;    param->analyse.b_dct_decimate = 1;    param->analyse.b_psnr = 1;    param->i_cqm_preset = X264_CQM_FLAT;    memset( param->cqm_4iy, 16, 16 );    memset( param->cqm_4ic, 16, 16 );    memset( param->cqm_4py, 16, 16 );    memset( param->cqm_4pc, 16, 16 );    memset( param->cqm_8iy, 16, 64 );    memset( param->cqm_8py, 16, 64 );    param->b_repeat_headers = 1;    param->b_aud = 0;}/**************************************************************************** * x264_log: ****************************************************************************/void x264_log( x264_t *h, int i_level, const char *psz_fmt, ... ){    if( i_level <= h->param.i_log_level )    {        va_list arg;        va_start( arg, psz_fmt );        h->param.pf_log( h->param.p_log_private, i_level, psz_fmt, arg );        va_end( arg );    }}static void x264_log_default( void *p_unused, int i_level, const char *psz_fmt, va_list arg ){    char *psz_prefix;    switch( i_level )    {        case X264_LOG_ERROR:            psz_prefix = "error";            break;        case X264_LOG_WARNING:            psz_prefix = "warning";            break;        case X264_LOG_INFO:            psz_prefix = "info";            break;        case X264_LOG_DEBUG:            psz_prefix = "debug";            break;        default:            psz_prefix = "unknown";            break;    }    fprintf( stderr, "x264 [%s]: ", psz_prefix );    vfprintf( stderr, psz_fmt, arg );}/**************************************************************************** * x264_picture_alloc: ****************************************************************************/void x264_picture_alloc( x264_picture_t *pic, int i_csp, int i_width, int i_height ){    pic->i_type = X264_TYPE_AUTO;    pic->i_qpplus1 = 0;    pic->img.i_csp = i_csp;    switch( i_csp & X264_CSP_MASK )    {        case X264_CSP_I420:        case X264_CSP_YV12:            pic->img.i_plane = 3;            pic->img.plane[0] = x264_malloc( 3 * i_width * i_height / 2 );            pic->img.plane[1] = pic->img.plane[0] + i_width * i_height;            pic->img.plane[2] = pic->img.plane[1] + i_width * i_height / 4;            pic->img.i_stride[0] = i_width;            pic->img.i_stride[1] = i_width / 2;            pic->img.i_stride[2] = i_width / 2;            break;        case X264_CSP_I422:            pic->img.i_plane = 3;            pic->img.plane[0] = x264_malloc( 2 * i_width * i_height );            pic->img.plane[1] = pic->img.plane[0] + i_width * i_height;            pic->img.plane[2] = pic->img.plane[1] + i_width * i_height / 2;            pic->img.i_stride[0] = i_width;            pic->img.i_stride[1] = i_width / 2;            pic->img.i_stride[2] = i_width / 2;            break;        case X264_CSP_I444:            pic->img.i_plane = 3;            pic->img.plane[0] = x264_malloc( 3 * i_width * i_height );            pic->img.plane[1] = pic->img.plane[0] + i_width * i_height;            pic->img.plane[2] = pic->img.plane[1] + i_width * i_height;            pic->img.i_stride[0] = i_width;            pic->img.i_stride[1] = i_width;            pic->img.i_stride[2] = i_width;            break;        case X264_CSP_YUYV:            pic->img.i_plane = 1;            pic->img.plane[0] = x264_malloc( 2 * i_width * i_height );            pic->img.i_stride[0] = 2 * i_width;            break;        case X264_CSP_RGB:        case X264_CSP_BGR:            pic->img.i_plane = 1;            pic->img.plane[0] = x264_malloc( 3 * i_width * i_height );            pic->img.i_stride[0] = 3 * i_width;            break;        case X264_CSP_BGRA:            pic->img.i_plane = 1;            pic->img.plane[0] = x264_malloc( 4 * i_width * i_height );            pic->img.i_stride[0] = 4 * i_width;            break;        default:            fprintf( stderr, "invalid CSP\n" );            pic->img.i_plane = 0;            break;    }}/**************************************************************************** * x264_picture_clean: ****************************************************************************/void x264_picture_clean( x264_picture_t *pic ){    x264_free( pic->img.plane[0] );    /* just to be safe */    memset( pic, 0, sizeof( x264_picture_t ) );}/**************************************************************************** * x264_nal_encode: ****************************************************************************/int x264_nal_encode( void *p_data, int *pi_data, int b_annexeb, x264_nal_t *nal ){    uint8_t *dst = p_data;    uint8_t *src = nal->p_payload;    uint8_t *end = &nal->p_payload[nal->i_payload];    int i_count = 0;