/***************************************************************************** * predict.c: h264 encoder ***************************************************************************** * Copyright (C) 2003 Laurent Aimar * $Id: predict.c,v 1.1 2003/11/09 23:25:04 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. *****************************************************************************//* XXX predict4x4 are inspired from ffmpeg h264 decoder */#include <stdlib.h>#include <string.h>#include <stdint.h>#include "predict.h"#define X264_CLIP3(a,b,c) ( (a)<(b)?(b):( (a)>(c)?(c):(a)) )/**************************************************************************** * 16x16 prediction for intra block DC, H, V, P ****************************************************************************/static void predict_16x16_dc( uint8_t *src, int i_stride ){    int dc = 0;    int i, j;    /* calculate DC value */    for( i = 0; i < 16; i++ )    {        dc += src[-1 + i * i_stride];        dc += src[i - i_stride];    }    dc = ( dc + 16 ) >> 5;    for( i = 0; i < 16; i++ )    {        for( j = 0; j < 16; j++ )        {            src[j] = dc;        }        src += i_stride;    }}static void predict_16x16_dc_left( uint8_t *src, int i_stride ){    int dc = 0;    int i,j;    for( i = 0; i < 16; i++ )    {        dc += src[-1 + i * i_stride];    }    dc = ( dc + 8 ) >> 4;    for( i = 0; i < 16; i++ )    {        for( j = 0; j < 16; j++ )        {            src[j] = dc;        }        src += i_stride;    }}static void predict_16x16_dc_top( uint8_t *src, int i_stride ){    int dc = 0;    int i,j;    for( i = 0; i < 16; i++ )    {        dc += src[i - i_stride];    }    dc = ( dc + 8 ) >> 4;    for( i = 0; i < 16; i++ )    {        for( j = 0; j < 16; j++ )        {            src[j] = dc;        }        src += i_stride;    }}static void predict_16x16_dc_128( uint8_t *src, int i_stride ){    int i,j;    for( i = 0; i < 16; i++ )    {        for( j = 0; j < 16; j++ )        {            src[j] = 128;        }        src += i_stride;    }}static void predict_16x16_h( uint8_t *src, int i_stride ){    int i,j;    for( i = 0; i < 16; i++ )    {        uint8_t v;        v = src[-1];        for( j = 0; j < 16; j++ )        {            src[j] = v;        }        src += i_stride;    }}static void predict_16x16_v( uint8_t *src, int i_stride ){    int i,j;    for( i = 0; i < 16; i++ )    {        for( j = 0; j < 16; j++ )        {            src[i * i_stride +j] = src[j - i_stride];        }    }}static void predict_16x16_p( uint8_t *src, int i_stride ){    int x, y, i;    int a, b, c;    int H = 0;    int V = 0;    int i00;    /* calcule H and V */    for( i = 0; i <= 7; i++ )    {        H += ( i + 1 ) * ( src[ 8 + i - i_stride ] - src[6 -i -i_stride] );        V += ( i + 1 ) * ( src[-1 + (8+i)*i_stride] - src[-1 + (6-i)*i_stride] );    }    a = 16 * ( src[-1 + 15*i_stride] + src[15 - i_stride] );    b = ( 5 * H + 32 ) >> 6;    c = ( 5 * V + 32 ) >> 6;    i00 = a - b * 7 - c * 7 + 16;    for( y = 0; y < 16; y++ )    {        for( x = 0; x < 16; x++ )        {            int pix;            pix = (i00+b*x)>>5;            src[x] = X264_CLIP3( pix, 0, 255 );        }        src += i_stride;        i00 += c;    }}/**************************************************************************** * 8x8 prediction for intra chroma block DC, H, V, P ****************************************************************************/static void predict_8x8_dc_128( uint8_t *src, int i_stride ){    int x,y;    for( y = 0; y < 8; y++ )    {        for( x = 0; x < 8; x++ )        {            src[x] = 128;        }        src += i_stride;    }}static void predict_8x8_dc_left( uint8_t *src, int i_stride ){    int x,y;    int dc0 = 0, dc1 = 0;    for( y = 0; y < 4; y++ )    {        dc0 += src[y * i_stride     - 1];        dc1 += src[(y+4) * i_stride - 1];    }    dc0 = ( dc0 + 2 ) >> 2;    dc1 = ( dc1 + 2 ) >> 2;    for( y = 0; y < 4; y++ )    {        for( x = 0; x < 8; x++ )        {            src[           x] = dc0;            src[4*i_stride+x] = dc1;        }        src += i_stride;    }}static void predict_8x8_dc_top( uint8_t *src, int i_stride ){    int x,y;    int dc0 = 0, dc1 = 0;    for( x = 0; x < 4; x++ )    {        dc0 += src[x     - i_stride];        dc1 += src[x + 4 - i_stride];    }    dc0 = ( dc0 + 2 ) >> 2;    dc1 = ( dc1 + 2 ) >> 2;    for( y = 0; y < 8; y++ )    {        for( x = 0; x < 4; x++ )        {            src[x    ] = dc0;            src[x + 4] = dc1;        }        src += i_stride;    }}static void predict_8x8_dc( uint8_t *src, int i_stride ){    int x,y;    int s0 = 0, s1 = 0, s2 = 0, s3 = 0;    int dc0, dc1, dc2, dc3;    int i;    /* First do :          s0 s1       s2       s3    */    for( i = 0; i < 4; i++ )    {        s0 += src[i - i_stride];        s1 += src[i + 4 - i_stride];        s2 += src[-1 + i * i_stride];        s3 += src[-1 + (i+4)*i_stride];    }    /* now calculate       dc0 dc1       dc2 dc3     */    dc0 = ( s0 + s2 + 4 ) >> 3;    dc1 = ( s1 + 2 ) >> 2;    dc2 = ( s3 + 2 ) >> 2;    dc3 = ( s1 + s3 + 4 ) >> 3;    for( y = 0; y < 4; y++ )    {        for( x = 0; x < 4; x++ )        {            src[             x    ] = dc0;            src[             x + 4] = dc1;            src[4*i_stride + x    ] = dc2;            src[4*i_stride + x + 4] = dc3;        }        src += i_stride;    }}static void predict_8x8_h( uint8_t *src, int i_stride ){    int i,j;    for( i = 0; i < 8; i++ )    {        uint8_t v;        v = src[-1];        for( j = 0; j < 8; j++ )        {            src[j] = v;        }        src += i_stride;    }}static void predict_8x8_v( uint8_t *src, int i_stride ){    int i,j;    for( i = 0; i < 8; i++ )    {        for( j = 0; j < 8; j++ )        {            src[i * i_stride +j] = src[j - i_stride];        }    }}static void predict_8x8_p( uint8_t *src, int i_stride ){    int i;    int x,y;    int a, b, c;    int H = 0;    int V = 0;    int i00;    for( i = 0; i < 4; i++ )    {        H += ( i + 1 ) * ( src[4+i - i_stride] - src[2 - i -i_stride] );        V += ( i + 1 ) * ( src[-1 +(i+4)*i_stride] - src[-1+(2-i)*i_stride] );    }    a = 16 * ( src[-1+7*i_stride] + src[7 - i_stride] );    b = ( 17 * H + 16 ) >> 5;    c = ( 17 * V + 16 ) >> 5;    i00 = a -3*b -3*c + 16;    for( y = 0; y < 8; y++ )