/***************************************************************************** * frame.c: h264 encoder library ***************************************************************************** * Copyright (C) 2003 Laurent Aimar * $Id: frame.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 <stdlib.h>#include <stdio.h>#include <string.h>#include "common.h"#include "macroblock.h"x264_frame_t *x264_frame_new( x264_t *h ){    x264_frame_t   *frame = x264_malloc( sizeof( x264_frame_t ) );    int i;    int i_mb_count = h->mb.i_mb_count;    int i_stride;    int i_lines;    /* allocate frame data (+64 for extra data for me) */    i_stride = ( ( h->param.i_width  + 15 )&0xfffff0 )+ 64;    i_lines  = ( ( h->param.i_height + 15 )&0xfffff0 );    frame->i_plane = 3;    for( i = 0; i < 3; i++ )    {        int i_divh = 1;        int i_divw = 1;        if( i > 0 )        {            if( h->param.i_csp == X264_CSP_I420 )                i_divh = i_divw = 2;            else if( h->param.i_csp == X264_CSP_I422 )                i_divw = 2;        }        frame->i_stride[i] = i_stride / i_divw;        frame->i_lines[i] = i_lines / i_divh;        frame->buffer[i] = x264_malloc( frame->i_stride[i] *                                        ( frame->i_lines[i] + 64 / i_divh ) );        frame->plane[i] = ((uint8_t*)frame->buffer[i]) +                          frame->i_stride[i] * 32 / i_divh + 32 / i_divw;    }    frame->i_stride[3] = 0;    frame->i_lines[3] = 0;    frame->buffer[3] = NULL;    frame->plane[3] = NULL;    frame->filtered[0] = frame->plane[0];    for( i = 0; i < 3; i++ )    {        frame->buffer[4+i] = x264_malloc( frame->i_stride[0] *                                        ( frame->i_lines[0] + 64 ) );        frame->filtered[i+1] = ((uint8_t*)frame->buffer[4+i]) +                                frame->i_stride[0] * 32 + 32;    }    frame->i_stride_lowres = frame->i_stride[0]/2 + 32;    frame->i_lines_lowres = frame->i_lines[0]/2;    for( i = 0; i < 4; i++ )    {        frame->buffer[7+i] = x264_malloc( frame->i_stride_lowres *                                        ( frame->i_lines[0]/2 + 64 ) );        frame->lowres[i] = ((uint8_t*)frame->buffer[7+i]) +                            frame->i_stride_lowres * 32 + 32;    }    frame->i_poc = -1;    frame->i_type = X264_TYPE_AUTO;    frame->i_qpplus1 = 0;    frame->i_pts = -1;    frame->i_frame = -1;    frame->i_frame_num = -1;

	frame->is_reference = 0;    frame->mb_type= x264_malloc( i_mb_count * sizeof( int8_t) );    frame->mv[0]  = x264_malloc( 2*16 * i_mb_count * sizeof( int16_t ) );    frame->ref[0] = x264_malloc( 4 * i_mb_count * sizeof( int8_t ) );    //if( h->param.i_bframe )
	if( h->sh.sps->i_profile_idc == PROFILE_MAIN )    {        frame->mv[1]  = x264_malloc( 2*16 * i_mb_count * sizeof( int16_t ) );        frame->ref[1] = x264_malloc( 4 * i_mb_count * sizeof( int8_t ) );    }    else    {        frame->mv[1]  = NULL;        frame->ref[1] = NULL;    }    return frame;}void x264_frame_delete( x264_frame_t *frame ){    int i;    for( i = 0; i < frame->i_plane; i++ )    {        x264_free( frame->buffer[i] );    }    for( i = 4; i < 11; i++ ) /* filtered planes */    {        x264_free( frame->buffer[i] );    }    x264_free( frame->mb_type );    x264_free( frame->mv[0] );    x264_free( frame->mv[1] );    x264_free( frame->ref[0] );    x264_free( frame->ref[1] );    x264_free( frame );}#if 0void x264_frame_copy_picture( x264_t *h, x264_frame_t *dst, x264_picture_t *src ){    dst->i_type     = src->i_type;    dst->i_qpplus1  = src->i_qpplus1;    dst->i_pts      = src->i_pts;    switch( src->img.i_csp & X264_CSP_MASK )    {        case X264_CSP_I420:            h->csp.i420( dst, &src->img, h->param.i_width, h->param.i_height );            break;        case X264_CSP_YV12:            h->csp.yv12( dst, &src->img, h->param.i_width, h->param.i_height );            break;        case X264_CSP_I422:            h->csp.i422( dst, &src->img, h->param.i_width, h->param.i_height );            break;        case X264_CSP_I444:            h->csp.i444( dst, &src->img, h->param.i_width, h->param.i_height );            break;        case X264_CSP_YUYV:            h->csp.yuyv( dst, &src->img, h->param.i_width, h->param.i_height );            break;        case X264_CSP_RGB:            h->csp.rgb( dst, &src->img, h->param.i_width, h->param.i_height );            break;        case X264_CSP_BGR:            h->csp.bgr( dst, &src->img, h->param.i_width, h->param.i_height );            break;        case X264_CSP_BGRA:            h->csp.bgra( dst, &src->img, h->param.i_width, h->param.i_height );            break;        default:            x264_log( h, X264_LOG_ERROR, "Arg invalid CSP\n" );            break;    }}#endifvoid plane_expand_border( uint8_t *pix, int i_stride, int i_height, int i_pad ){#define PPIXEL(x, y) ( pix + (x) + (y)*i_stride )    const int i_width = i_stride - 2*i_pad;    int y;    for( y = 0; y < i_height; y++ )    {        /* left band */        memset( PPIXEL(-i_pad, y), PPIXEL(0, y)[0], i_pad );        /* right band */        memset( PPIXEL(i_width, y), PPIXEL(i_width-1, y)[0], i_pad );    }    /* upper band */    for( y = 0; y < i_pad; y++ )        memcpy( PPIXEL(-i_pad, -y-1), PPIXEL(-i_pad, 0), i_stride );    /* lower band */    for( y = 0; y < i_pad; y++ )        memcpy( PPIXEL(-i_pad, i_height+y), PPIXEL(-i_pad, i_height-1), i_stride );#undef PPIXEL}void x264_frame_expand_border( x264_frame_t *frame ){    int i;    for( i = 0; i < frame->i_plane; i++ )    {        int i_pad = i ? 16 : 32;        plane_expand_border( frame->plane[i], frame->i_stride[i], frame->i_lines[i], i_pad );    }}void x264_frame_expand_border_filtered( x264_frame_t *frame ){    /* during filtering, 8 extra pixels were filtered on each edge.        we want to expand border from the last filtered pixel */    int i;    for( i = 1; i < 4; i++ )        plane_expand_border( frame->filtered[i] - 8*frame->i_stride[0] - 8, frame->i_stride[0], frame->i_lines[0]+2*8, 24 );}void x264_frame_expand_lowres( x264_frame_t *frame ){    int i;    for( i = 0; i < 4; i++ )        plane_expand_border( frame->lowres[i], frame->i_stride_lowres, frame->i_lines_lowres, 32 );}/* FIXME theses tables are duplicated with the ones in macroblock.c */static const uint8_t block_idx_xy[4][4] ={    { 0, 2, 8,  10},    { 1, 3, 9,  11},    { 4, 6, 12, 14},    { 5, 7, 13, 15}};static const int i_chroma_qp_table[52] ={     0,  1,  2,  3,  4,  5,  6,  7,  8,  9,    10, 11, 12, 13, 14, 15, 16, 17, 18, 19,    20, 21, 22, 23, 24, 25, 26, 27, 28, 29,    29, 30, 31, 32, 32, 33, 34, 34, 35, 35,    36, 36, 37, 37, 37, 38, 38, 38, 39, 39,    39, 39};/* Deblocking filter (p153) */static const int i_alpha_table[52] ={     0,  0,  0,  0,  0,  0,  0,  0,  0,  0,     0,  0,  0,  0,  0,  0,  4,  4,  5,  6,     7,  8,  9, 10, 12, 13, 15, 17, 20, 22,    25, 28, 32, 36, 40, 45, 50, 56, 63, 71,    80, 90,101,113,127,144,162,182,203,226,    255, 255};static const int i_beta_table[52] ={     0,  0,  0,  0,  0,  0,  0,  0,  0,  0,     0,  0,  0,  0,  0,  0,  2,  2,  2,  3,     3,  3,  3,  4,  4,  4,  6,  6,  7,  7,     8,  8,  9,  9, 10, 10, 11, 11, 12, 12,    13, 13, 14, 14, 15, 15, 16, 16, 17, 17,    18, 18};static const int i_tc0_table[52][3] ={    { 0, 0, 0 }, { 0, 0, 0 }, { 0, 0, 0 }, { 0, 0, 0 }, { 0, 0, 0 }, { 0, 0, 0 },    { 0, 0, 0 }, { 0, 0, 0 }, { 0, 0, 0 }, { 0, 0, 0 }, { 0, 0, 0 }, { 0, 0, 0 },    { 0, 0, 0 }, { 0, 0, 0 }, { 0, 0, 0 }, { 0, 0, 0 }, { 0, 0, 0 }, { 0, 0, 1 },    { 0, 0, 1 }, { 0, 0, 1 }, { 0, 0, 1 }, { 0, 1, 1 }, { 0, 1, 1 }, { 1, 1, 1 },    { 1, 1, 1 }, { 1, 1, 1 }, { 1, 1, 1 }, { 1, 1, 2 }, { 1, 1, 2 }, { 1, 1, 2 },    { 1, 1, 2 }, { 1, 2, 3 }, { 1, 2, 3 }, { 2, 2, 3 }, { 2, 2, 4 }, { 2, 3, 4 },    { 2, 3, 4 }, { 3, 3, 5 }, { 3, 4, 6 }, { 3, 4, 6 }, { 4, 5, 7 }, { 4, 5, 8 },    { 4, 6, 9 }, { 5, 7,10 }, { 6, 8,11 }, { 6, 8,13 }, { 7,10,14 }, { 8,11,16 },    { 9,12,18 }, {10,13,20 }, {11,15,23 }, {13,17,25 }};/* From FF */static inline int clip_uint8( int a ){    if (a&(~255))        return (-a)>>31;    else        return a;}