www.gusucode.com > 一个可以在局域网进行视频聊天的源代码 > 一个可以在局域网进行视频聊天的源代码/VoIP/encoder/pred.cpp
////////////////////////////////////////////////////////////////////////////
//
//
// Project : VideoNet version 1.1.
// Description : Peer to Peer Video Conferencing over the LAN.
// Author : Nagareshwar Y Talekar ( nsry2002@yahoo.co.in)
// Date : 15-6-2004.
//
// I have converted origional fast h.263 encoder library from C to C++
// so that it can be integrated into any windows application easily.
// I have removed some of unnecessary codes/files from the
// fast h263 library.Also moved definitions and declarations
// in their proper .h and .cpp files.
//
// File description :
// Name : pred.cpp
//
//
/////////////////////////////////////////////////////////////////////////////
/*************************************************
* libr263: fast H.263 encoder library
*
* Copyright (C) 1996, Roalt Aalmoes, Twente University
* SPA multimedia group
*
* Based on Telenor TMN 1.6 encoder (Copyright (C) 1995, Telenor R&D)
* created by Karl Lillevold
*
* Author encoder: Roalt Aalmoes, <aalmoes@huygens.nl>
*
* Date: 31-07-96
**************************************************/
#include "pred.h"
#include "mot_est.h"
MB_Structure *Predict_P(unsigned int *curr_image, unsigned int *prev_image,
unsigned int *prev_ipol, int x, int y,
MotionVector *MV_ptr)
{
int m,n;
unsigned int curr[16][16];
int pred[16][16];
MotionVector *fr0;
int dx, dy;
int xmb, ymb;
MB_Structure *pred_error;
pred_error = (MB_Structure *)malloc(sizeof(MB_Structure));
xmb = x/MB_SIZE;
ymb = y/MB_SIZE;
fr0 = MV_ptr + ymb*Global::mbc + xmb;
/* Find MB in current image */
#ifdef PRINT_DEBUG
fprintf(debugf,"FindMB(%d,%d,%p,%p)\n",x,y,curr_image->Y,curr);
#endif
FindMB(x, y, curr_image,curr);
/* Find prediction based on half pel MV */
FindPred(x, y, fr0, prev_ipol, &pred[0][0]);
/* Do the actual prediction */
if (fr0->Mode == MODE_INTER) {
for (n = 0; n < MB_SIZE; n++)
for (m = 0; m < MB_SIZE; m++)
pred_error->lum[n][m] = (int)(curr[n][m] - pred[n][m]);
dx = 2*fr0->x + fr0->x_half;
dy = 2*fr0->y + fr0->y_half;
dx = ( dx % 4 == 0 ? dx >> 1 : (dx>>1)|1 );
dy = ( dy % 4 == 0 ? dy >> 1 : (dy>>1)|1 );
#ifdef PRINTDEBUG
fprintf(debugf,"DoPredChrom_P(%d,%d,%d,%d,%p,%p,%p)\n",x, y, dx, dy, curr_image, prev_image, pred_error);
#endif
DoPredChrom_P(x, y, dx, dy, curr_image, prev_image, pred_error);
}
else
fprintf(stderr,"Illegal Mode in Predict_P (pred.c)\n");
return pred_error;
}
void DoPredChrom_P(int x_curr, int y_curr, int dx, int dy,
unsigned int *curr, unsigned int *prev,
MB_Structure *pred_error)
{
int m,n;
int x, y, ofx, ofy, pel, lx;
int xint, yint;
int xh, yh;
lx = Global::pels/2;
x = x_curr>>1;
y = y_curr>>1;
xint = dx>>1;
xh = dx & 1;
yint = dy>>1;
yh = dy & 1;
if (!xh && !yh) {
for (n = 0; n < 8; n++) {
for (m = 0; m < 8; m++) {
ofx = x + xint + m;
ofy = y + yint + n;
/* prev+Global::vskip does not work directly... */
pel=*(prev+Global::vskip + ofx + (ofy )*lx);
pred_error->Cr[n][m] =
(int)(*(curr+Global::vskip + x+m + (y+n)*Global::cpels) - pel);
pel=*(prev+Global::uskip + ofx + (ofy )*lx);
pred_error->Cb[n][m] =
(int)(*(curr + Global::uskip + x+m + (y+n)*Global::cpels) - pel);
}
}
}
else if (!xh && yh) {
for (n = 0; n < 8; n++) {
for (m = 0; m < 8; m++) {
ofx = x + xint + m;
ofy = y + yint + n;
pel=(*(prev+Global::vskip + ofx + (ofy )*lx)+
*(prev+Global::vskip + ofx + (ofy+yh)*lx) + 1)>>1;
pred_error->Cr[n][m] =
(int)(*(curr+Global::vskip + x+m + (y+n)*Global::cpels) - pel);
pel=(*(prev+Global::uskip + ofx + (ofy )*lx)+
*(prev+Global::uskip + ofx + (ofy+yh)*lx) + 1)>>1;
pred_error->Cb[n][m] =
(int)(*( curr+Global::uskip + x+m + (y+n)*Global::cpels) - pel);
}
}
}
else if (xh && !yh) {
for (n = 0; n < 8; n++) {
for (m = 0; m < 8; m++) {
ofx = x + xint + m;
ofy = y + yint + n;
pel=(*(prev+Global::vskip+ofx + (ofy )*lx)+
*(prev+Global::vskip+ofx+xh + (ofy )*lx) + 1)>>1;
pred_error->Cr[n][m] =
(int)(*(curr+Global::vskip + x+m + (y+n)*Global::cpels) - pel);
pel=(*(prev+Global::uskip+ofx + (ofy )*lx)+
*(prev+Global::uskip+ofx+xh + (ofy )*lx) + 1)>>1;
pred_error->Cb[n][m] =
(int)( *( curr+Global::uskip + x+m + (y+n)*Global::cpels) - pel);
}
}
}
else { /* xh && yh */
for (n = 0; n < 8; n++) {
for (m = 0; m < 8; m++) {
ofx = x + xint + m;
ofy = y + yint + n;
pel=(*(prev+Global::vskip+ofx + (ofy )*lx)+
*(prev+Global::vskip+ofx+xh + (ofy )*lx)+
*(prev+Global::vskip+ofx + (ofy+yh)*lx)+
*(prev+Global::vskip+ofx+xh + (ofy+yh)*lx)+
2)>>2;
pred_error->Cr[n][m] =
(int)(*(curr+Global::vskip + x+m + (y+n)*Global::cpels) - pel);
pel=(*(prev+Global::uskip+ofx + (ofy )*lx)+
*(prev+Global::uskip+ofx+xh + (ofy )*lx)+
*(prev+Global::uskip+ofx + (ofy+yh)*lx)+
*(prev+Global::uskip+ofx+xh + (ofy+yh)*lx)+
2)>>2;
pred_error->Cb[n][m] =
(int)(*(curr+Global::uskip + x+m + (y+n)*Global::cpels) - pel);
}
}
}
return;
}
void FindPred(int x, int y, MotionVector *fr, unsigned int *prev,
int *pred)
{
int n;
int new_x, new_y;
int *prev_ptr;
new_x = 2*(x + fr->x) + fr->x_half;
new_y = 2*(y + fr->y) + fr->y_half;
prev_ptr = (int *) prev + 2*new_y*Global::pels + new_x;
/* Fill pred. data */
for (n = 0; n < 16; n++) {
/* Interpolated data -> too bad it can't be done in longs... */
*(pred++) = *(prev_ptr);
prev_ptr += 2;
*(pred++) = *(prev_ptr);
prev_ptr += 2;
*(pred++) = *(prev_ptr);
prev_ptr += 2;
*(pred++) = *(prev_ptr);
prev_ptr += 2;
*(pred++) = *(prev_ptr);
prev_ptr += 2;
*(pred++) = *(prev_ptr);
prev_ptr += 2;
*(pred++) = *(prev_ptr);
prev_ptr += 2;
*(pred++) = *(prev_ptr);
prev_ptr += 2;
*(pred++) = *(prev_ptr);
prev_ptr += 2;
*(pred++) = *(prev_ptr);
prev_ptr += 2;
*(pred++) = *(prev_ptr);
prev_ptr += 2;
*(pred++) = *(prev_ptr);
prev_ptr += 2;
*(pred++) = *(prev_ptr);
prev_ptr += 2;
*(pred++) = *(prev_ptr);
prev_ptr += 2;
*(pred++) = *(prev_ptr);
prev_ptr += 2;
*(pred++) = *(prev_ptr);
prev_ptr += Global::pels*4 - 30;
}
return;
}
MB_Structure *MB_Recon_P(unsigned int *prev_image, unsigned int *prev_ipol,
MB_Structure *diff, int x_curr, int y_curr,
MotionVector *MV_ptr)
{
MB_Structure *recon_data = (MB_Structure *)malloc(sizeof(MB_Structure));
MotionVector *fr0;
int dx, dy;
fr0 = MV_ptr + (y_curr/MB_SIZE)*Global::mbc + (x_curr/MB_SIZE);
if (fr0->Mode == MODE_INTER) {
/* Inter 16x16 */
ReconLumBlock_P(x_curr,y_curr,fr0,prev_ipol,&diff->lum[0][0]);
dx = 2*fr0->x + fr0->x_half;
dy = 2*fr0->y + fr0->y_half;
dx = ( dx % 4 == 0 ? dx >> 1 : (dx>>1)|1 );
dy = ( dy % 4 == 0 ? dy >> 1 : (dy>>1)|1 );
ReconChromBlock_P(x_curr, y_curr, dx, dy, prev_image, diff);
}
memcpy(recon_data, diff, sizeof(MB_Structure));
return recon_data;
}
void ReconLumBlock_P(int x, int y, MotionVector *fr,
unsigned int *prev, int *data)
{
int m, n;
int x1, y1;
int *data_ptr = data;
unsigned int *prev_ptr;
x1 = 2*(x + fr->x) + fr->x_half;
y1 = 2*(y + fr->y) + fr->y_half;
prev_ptr = prev + x1 + y1*2*Global::pels;
for (n = 0; n < 16; n++) {
for (m = 0; m < 16; m++) {
*data_ptr++ += (int)(*prev_ptr);
prev_ptr += 2;
}
prev_ptr += 2*2*Global::pels - 32;
}
return;
}
void ReconChromBlock_P(int x_curr, int y_curr, int dx, int dy,
unsigned int *prev, MB_Structure *data)
{
int m,n;
int x, y, ofx, ofy, pel,lx;
int xint, yint;
int xh, yh;
lx = Global::pels/2;
x = x_curr>>1;
y = y_curr>>1;
xint = dx>>1;
xh = dx & 1;
yint = dy>>1;
yh = dy & 1;
if (!xh && !yh) {
for (n = 0; n < 8; n++) {
for (m = 0; m < 8; m++) {
ofx = x + xint + m;
ofy = y + yint + n;
pel=*(prev+Global::vskip+ ofx + (ofy )*lx);
data->Cr[n][m] += pel;
pel=*(prev+Global::uskip + ofx + (ofy )*lx);
data->Cb[n][m] += pel;
}
}
}
else if (!xh && yh) {
for (n = 0; n < 8; n++) {
for (m = 0; m < 8; m++) {
ofx = x + xint + m;
ofy = y + yint + n;
pel=(*(prev+Global::vskip+ ofx + (ofy )*lx)+
*(prev+Global::vskip+ ofx + (ofy+yh)*lx) + 1)>>1;
data->Cr[n][m] += pel;
pel=(*(prev+Global::uskip+ofx + (ofy )*lx)+
*(prev+Global::uskip+ofx + (ofy+yh)*lx) + 1)>>1;
data->Cb[n][m] += pel;
}
}
}
else if (xh && !yh) {
for (n = 0; n < 8; n++) {
for (m = 0; m < 8; m++) {
ofx = x + xint + m;
ofy = y + yint + n;
pel=(*(prev+Global::vskip+ofx + (ofy )*lx)+
*(prev+Global::vskip+ofx+xh + (ofy )*lx) + 1)>>1;
data->Cr[n][m] += pel;
pel=(*(prev+Global::uskip+ofx + (ofy )*lx)+
*(prev+Global::uskip+ofx+xh + (ofy )*lx) + 1)>>1;
data->Cb[n][m] += pel;
}
}
}
else { /* xh && yh */
for (n = 0; n < 8; n++) {
for (m = 0; m < 8; m++) {
ofx = x + xint + m;
ofy = y + yint + n;
pel=(*(prev+Global::vskip+ofx + (ofy )*lx)+
*(prev+Global::vskip+ofx+xh + (ofy )*lx)+
*(prev+Global::vskip+ofx + (ofy+yh)*lx)+
*(prev+Global::vskip+ofx+xh + (ofy+yh)*lx)+
2)>>2;
data->Cr[n][m] += pel;
pel=(*(prev+Global::uskip+ofx + (ofy )*lx)+
*(prev+Global::uskip+ofx+xh + (ofy )*lx)+
*(prev+Global::uskip+ofx + (ofy+yh)*lx)+
*(prev+Global::uskip+ofx+xh + (ofy+yh)*lx)+
2)>>2;
data->Cb[n][m] += pel;
}
}
}
return;
}
void FindChromBlock_P(int x_curr, int y_curr, int dx, int dy,
unsigned int *prev, MB_Structure *data)
{
int m,n;
int x, y, ofx, ofy, pel,lx;
int xint, yint;
int xh, yh;
lx = Global::pels/2;
x = x_curr>>1;
y = y_curr>>1;
xint = dx>>1;
xh = dx & 1;
yint = dy>>1;
yh = dy & 1;
if (!xh && !yh) {
for (n = 0; n < 8; n++) {
for (m = 0; m < 8; m++) {
ofx = x + xint + m;
ofy = y + yint + n;
pel=*(prev+Global::vskip+ofx + (ofy )*lx);
data->Cr[n][m] = pel;
pel=*(prev+Global::uskip+ofx + (ofy )*lx);
data->Cb[n][m] = pel;
}
}
}
else if (!xh && yh) {
for (n = 0; n < 8; n++) {
for (m = 0; m < 8; m++) {
ofx = x + xint + m;
ofy = y + yint + n;
pel=(*(prev+Global::vskip+ofx + (ofy )*lx)+
*(prev+Global::vskip+ofx + (ofy+yh)*lx) + 1)>>1;
data->Cr[n][m] = pel;
pel=(*(prev+Global::uskip+ofx + (ofy )*lx)+
*(prev+Global::uskip+ofx + (ofy+yh)*lx) + 1)>>1;
data->Cb[n][m] = pel;
}
}
}
else if (xh && !yh) {
for (n = 0; n < 8; n++) {
for (m = 0; m < 8; m++) {
ofx = x + xint + m;
ofy = y + yint + n;
pel=(*(prev+Global::vskip+ofx + (ofy )*lx)+
*(prev+Global::vskip+ofx+xh + (ofy )*lx) + 1)>>1;
data->Cr[n][m] = pel;
pel=(*(prev+Global::uskip+ofx + (ofy )*lx)+
*(prev+Global::uskip+ofx+xh + (ofy )*lx) + 1)>>1;
data->Cb[n][m] = pel;
}
}
}
else { /* xh && yh */
for (n = 0; n < 8; n++) {
for (m = 0; m < 8; m++) {
ofx = x + xint + m;
ofy = y + yint + n;
pel=(*(prev+Global::vskip+ofx + (ofy )*lx)+
*(prev+Global::vskip+ofx+xh + (ofy )*lx)+
*(prev+Global::vskip+ofx + (ofy+yh)*lx)+
*(prev+Global::vskip+ofx+xh + (ofy+yh)*lx)+
2)>>2;
data->Cr[n][m] = pel;
pel=(*(prev+Global::uskip+ofx + (ofy )*lx)+
*(prev+Global::uskip+ofx+xh + (ofy )*lx)+
*(prev+Global::uskip+ofx + (ofy+yh)*lx)+
*(prev+Global::uskip+ofx+xh + (ofy+yh)*lx)+
2)>>2;
data->Cb[n][m] = pel;
}
}
}
return;
}
int ChooseMode(unsigned int *curr, int x_pos, int y_pos, int min_SAD)
{
int i,j;
int MB_mean = 0, A = 0;
int y_off;
for (j = 0; j < MB_SIZE; j++) {
y_off = (y_pos + j) * Global::pels;
for (i = 0; i < MB_SIZE; i++) {
MB_mean += *(curr + x_pos + i + y_off);
}
}
MB_mean /= (MB_SIZE*MB_SIZE);
for (j = 0; j < MB_SIZE; j++) {
y_off = (y_pos + j) * Global::pels;
for (i = 0; i < MB_SIZE; i++) {
A += abs( *(curr + x_pos + i + y_off) - MB_mean );
}
}
if (A < (min_SAD - 500))
return MODE_INTRA;
else
return MODE_INTER;
}