common.c

在dsp上实现h.264编解码

/*****************************************************************************
 * common.c: h264 library
 *****************************************************************************
 * Copyright (C) 2003 Laurent Aimar
 * $Id: SHU264.h,v 1.1 2006/02/03 19:24:12 fenrir Exp $
 *
 * Authors: jsslq <[e-mail]>
 *
 * 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 <stdlib.h>
#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 SHU264_log_default( void *, int, const char *, va_list );

/****************************************************************************
 * SHU264_param_default:
 ****************************************************************************/
void    SHU264_param_default( SHU264_param_t *param )
{
    /* */
    memset( param, 0, sizeof( SHU264_param_t ) );

    /* CPU autodetect */
    param->cpu = SHU264_cpu_detect();
    param->i_threads =1;
    param->i_csp           = 1;

    /* Video properties */
    param->i_width         = 0;
    param->i_height        = 0;
  
    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_bframe = 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 = 1000;
    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 =(float) 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 = (float) 1.4;
    param->rc.f_pb_factor =(float) 1.3;

    param->rc.psz_rc_eq = "blurCplx^(1-qComp)";
    param->rc.f_qcompress =(float) 0.6;
    param->rc.f_qblur = 0.5;
    param->rc.f_complexity_blur = 20;
    param->rc.i_zones = 0;

    /* Log */
    param->pf_log = SHU264_log_default;
    param->p_log_private = NULL;
    param->i_log_level = SHU264_LOG_INFO;

    /* */
    param->analyse.intra = SHU264_ANALYSE_I4x4 | SHU264_ANALYSE_I8x8;
    param->analyse.inter = SHU264_ANALYSE_I4x4 | SHU264_ANALYSE_I8x8
                         | SHU264_ANALYSE_PSUB16x16 | SHU264_ANALYSE_BSUB16x16;
    param->analyse.i_direct_mv_pred = SHU264_DIRECT_PRED_TEMPORAL;
    param->analyse.i_me_method = SHU264_ME_HEX;
    param->analyse.i_me_range = 16;
    param->analyse.i_subpel_refine = 5;

    param->analyse.i_mv_range = -1; // set from level_idc
 
    param->analyse.b_fast_pskip = 1;
    param->analyse.b_psnr = 1;
    param->b_repeat_headers = 1;
}

/****************************************************************************
 * SHU264_log:
 ****************************************************************************/
void SHU264_log( SHU264_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 );
    }
}



/****************************************************************************
 * SHU264_picture_alloc:
 ****************************************************************************/
static void SHU264_log_default( void *p_unused, int i_level, const char *psz_fmt, va_list arg )
{
    char *psz_prefix;
    switch( i_level )
    {
        case SHU264_LOG_ERROR:
            psz_prefix = "error";
            break;
        case SHU264_LOG_WARNING:
            psz_prefix = "warning";
            break;
        case SHU264_LOG_INFO:
            psz_prefix = "info";
            break;
        case SHU264_LOG_DEBUG:
            psz_prefix = "debug";
            break;
        default:
            psz_prefix = "unknown";
            break;
    }
    fprintf( stderr, "SHU264 [%s]: ", psz_prefix );
    vfprintf( stderr, psz_fmt, arg );
}

void SHU264_picture_alloc( SHU264_picture_t *pic, int i_width, int i_height )
{
    pic->i_type = SHU264_TYPE_AUTO;
    pic->i_qpplus1 = 0;
    pic->img.i_csp = 1;
    pic->img.i_plane = 3;
    pic->img.plane[0] = SHU264_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;
}

/****************************************************************************
 * SHU264_picture_clean:
 ****************************************************************************/
void SHU264_picture_clean( SHU264_picture_t *pic )
{
    SHU264_free( pic->img.plane[0] );

    /* just to be safe */
    memset( pic, 0, sizeof( SHU264_picture_t ) );
}

/****************************************************************************
 * SHU264_nal_encode:
 ****************************************************************************/
int SHU264_nal_encode( void *p_data, int *pi_data, int b_annexeb, SHU264_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;

    /* 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 && *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;
}

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

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

    src++;

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

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

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



/****************************************************************************
 * SHU264_malloc:
 ****************************************************************************/
void *SHU264_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 & 15;
    *( (void **) ( align_buf - sizeof( void ** ) ) ) = buf;
    *( (int *) ( align_buf - sizeof( void ** ) - sizeof( int ) ) ) = i_size;
    return align_buf;
#endif
}

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

/****************************************************************************
 * SHU264_realloc:
 ****************************************************************************/
void *SHU264_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 = SHU264_malloc( i_size );
    if( i_old_size > 0 && i_size > 0 )
    {
        memcpy( p_new, p, ( i_old_size < i_size ) ? i_old_size : i_size );
    }
    SHU264_free( p );
    return p_new;
#endif
}