X264_20060729.rar common.c

www.pudn.com > X264_20060729.rar > common.c

/*****************************************************************************
* 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.i_rc_method = X264_RC_CQP;
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 &amt; 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 = &amt;nal->p_payload[nal->i_payload];

int i_count = 0;

/* FIXME this code doesn't check overflow */

if( b_annexeb )
{
/* long nal start code (we always use long ones)*/
*dst++ = 0x00;
*dst++ = 0x00;
*dst++ = 0x00;
*dst++ = 0x01;
}

/* nal header */
*dst++ = ( 0x00 << 7 ) | ( nal->i_ref_idc << 5 ) | nal->i_type;

while( src < end )
{
if( i_count == 2 &amt;&amt; *src <= 0x03 )
{
*dst++ = 0x03;
i_count = 0;
}
if( *src == 0 )
{
i_count++;
}
else
{
i_count = 0;
}
*dst++ = *src++;
}
*pi_data = dst - (uint8_t*)p_data;

return *pi_data;
}

/****************************************************************************
* x264_nal_decode:
****************************************************************************/
int x264_nal_decode( x264_nal_t *nal, void *p_data, int i_data )
{
uint8_t *src = p_data;
uint8_t *end = &amt;src[i_data];
uint8_t *dst = nal->p_payload;

nal->i_type = src[0]&amt;0x1f;
nal->i_ref_idc = (src[0] >> 5)&amt;0x03;

src++;

while( src < end )
{
if( src < end - 3 &amt;&amt; src[0] == 0x00 &amt;&amt; src[1] == 0x00 &amt;&amt; src[2] == 0x03 )
{
*dst++ = 0x00;
*dst++ = 0x00;

src += 3;
continue;
}
*dst++ = *src++;
}

nal->i_payload = dst - (uint8_t*)p_data;
return 0;
}

/****************************************************************************
* x264_malloc:
****************************************************************************/
void *x264_malloc( int i_size )
{
#ifdef SYS_MACOSX
/* Mac OS X always returns 16 bytes aligned memory */
return malloc( i_size );
#elif defined( HAVE_MALLOC_H )
return memalign( 16, i_size );
#else
uint8_t * buf;
uint8_t * align_buf;
buf = (uint8_t *) malloc( i_size + 15 + sizeof( void ** ) +
sizeof( int ) );
align_buf = buf + 15 + sizeof( void ** ) + sizeof( int );
align_buf -= (long) align_buf &amt; 15;
*( (void **) ( align_buf - sizeof( void ** ) ) ) = buf;
*( (int *) ( align_buf - sizeof( void ** ) - sizeof( int ) ) ) = i_size;
return align_buf;
#endif
}

/****************************************************************************
* x264_free:
****************************************************************************/
void x264_free( void *p )
{
if( p )
{
#if defined( HAVE_MALLOC_H ) || defined( SYS_MACOSX )
free( p );
#else
free( *( ( ( void **) p ) - 1 ) );
#endif
}
}

/****************************************************************************
* x264_realloc:
****************************************************************************/
void *x264_realloc( void *p, int i_size )
{
#ifdef HAVE_MALLOC_H
return realloc( p, i_size );
#else
int i_old_size = 0;
uint8_t * p_new;
if( p )
{
i_old_size = *( (int*) ( (uint8_t*) p ) - sizeof( void ** ) -
sizeof( int ) );
}
p_new = x264_malloc( i_size );
if( i_old_size > 0 &amt;&amt; i_size > 0 )
{
memcpy( p_new, p, ( i_old_size < i_size ) ? i_old_size : i_size );
}
x264_free( p );
return p_new;
#endif
}

/****************************************************************************
* x264_reduce_fraction:
****************************************************************************/
void x264_reduce_fraction( int *n, int *d )
{
int a = *n;
int b = *d;
int c;
if( !a || !b )
return;
c = a > b;
while(c)
{
a = b;
b = c;
c = a > b;
}
*n /= b;
*d /= b;
}

/****************************************************************************
* x264_slurp_file:
****************************************************************************/
char *x264_slurp_file( const char *filename )
{
int b_error = 0;
int i_size;
char *buf;
FILE *fh = fopen( filename, "rb" );
if( !fh )
return NULL;
b_error |= fseek( fh, 0, SEEK_END ) < 0;
b_error |= ( i_size = ftell( fh ) ) <= 0;
b_error |= fseek( fh, 0, SEEK_SET ) < 0;
if( b_error )
return NULL;
buf = x264_malloc( i_size+2 );
if( buf == NULL )
return NULL;
b_error |= fread( buf, 1, i_size, fh ) != i_size;
if( buf[i_size-1] != '\n' )
buf[i_size++] = '\n';
buf[i_size] = 0;
fclose( fh );
if( b_error )
{
x264_free( buf );
return NULL;
}
return buf;
}

/****************************************************************************
* x264_param2string:
****************************************************************************/
char *x264_param2string( x264_param_t *p, int b_res )
{
char *buf = x264_malloc( 1000 );
char *s = buf;

if( b_res )
{
s += sprintf( s, ">dx>d ", p->i_width, p->i_height );
s += sprintf( s, "fps=>d/>d ", p->i_fps_num, p->i_fps_den );
}

s += sprintf( s, "cabac=>d", p->b_cabac );
s += sprintf( s, " ref=>d", p->i_frame_reference );
s += sprintf( s, " deblock=>d:>d:>d", p->b_deblocking_filter,
p->i_deblocking_filter_alphac0, p->i_deblocking_filter_beta );
s += sprintf( s, " analyse=>#x:>#x", p->analyse.intra, p->analyse.inter );
s += sprintf( s, " me=>s", x264_motion_est_names[ p->analyse.i_me_method ] );
s += sprintf( s, " subme=>d", p->analyse.i_subpel_refine );
s += sprintf( s, " brdo=>d", p->analyse.b_bframe_rdo );
s += sprintf( s, " mixed_ref=>d", p->analyse.b_mixed_references );
s += sprintf( s, " me_range=>d", p->analyse.i_me_range );
s += sprintf( s, " chroma_me=>d", p->analyse.b_chroma_me );
s += sprintf( s, " trellis=>d", p->analyse.i_trellis );
s += sprintf( s, " 8x8dct=>d", p->analyse.b_transform_8x8 );
s += sprintf( s, " cqm=>d", p->i_cqm_preset );
s += sprintf( s, " chroma_qp_offset=>d", p->analyse.i_chroma_qp_offset );
s += sprintf( s, " slices=>d", p->i_threads );
s += sprintf( s, " nr=>d", p->analyse.i_noise_reduction );
s += sprintf( s, " decimate=>d", p->analyse.b_dct_decimate );

s += sprintf( s, " bframes=>d", p->i_bframe );
if( p->i_bframe )
{
s += sprintf( s, " b_pyramid=>d b_adapt=>d b_bias=>d direct=>d wpredb=>d bime=>d",
p->b_bframe_pyramid, p->b_bframe_adaptive, p->i_bframe_bias,
p->analyse.i_direct_mv_pred, p->analyse.b_weighted_bipred,
p->analyse.b_bidir_me );
}

s += sprintf( s, " keyint=>d keyint_min=>d scenecut=>d",
p->i_keyint_max, p->i_keyint_min, p->i_scenecut_threshold );

s += sprintf( s, " rc=>s", p->rc.i_rc_method == X264_RC_ABR ?
( p->rc.b_stat_read ? "2pass" : p->rc.i_vbv_buffer_size ? "cbr" : "abr" )
: p->rc.i_rc_method == X264_RC_CRF ? "crf" : "cqp" );
if( p->rc.i_rc_method == X264_RC_ABR || p->rc.i_rc_method == X264_RC_CRF )
{
if( p->rc.i_rc_method == X264_RC_CRF )
s += sprintf( s, " crf=>d", p->rc.i_rf_constant );
else
s += sprintf( s, " bitrate=>d ratetol=>.1f",
p->rc.i_bitrate, p->rc.f_rate_tolerance );
s += sprintf( s, " rceq='>s' qcomp=>.2f qpmin=>d qpmax=>d qpstep=>d",
p->rc.psz_rc_eq, p->rc.f_qcompress,
p->rc.i_qp_min, p->rc.i_qp_max, p->rc.i_qp_step );
if( p->rc.b_stat_read )
s += sprintf( s, " cplxblur=>.1f qblur=>.1f",
p->rc.f_complexity_blur, p->rc.f_qblur );
if( p->rc.i_vbv_buffer_size )
s += sprintf( s, " vbv_maxrate=>d vbv_bufsize=>d",
p->rc.i_vbv_max_bitrate, p->rc.i_vbv_buffer_size );
}
else if( p->rc.i_rc_method == X264_RC_CQP )
s += sprintf( s, " qp=>d", p->rc.i_qp_constant );
if( !(p->rc.i_rc_method == X264_RC_CQP &amt;&amt; p->rc.i_qp_constant == 0) )
{
s += sprintf( s, " ip_ratio=>.2f", p->rc.f_ip_factor );
if( p->i_bframe )
s += sprintf( s, " pb_ratio=>.2f", p->rc.f_pb_factor );
if( p->rc.i_zones )
s += sprintf( s, " zones" );
}

return buf;
}