/* predict.c, 運動補(bǔ)償預(yù)測*/
#include <stdio.h>#include "global.h"/* 函數(shù)聲明*/static void predict_mb(  unsigned char *oldref[], unsigned char *newref[], unsigned char *cur[],  int lx, int bx, int by, int pict_type, int pict_struct, int mb_type,  int motion_type, int secondfield,  int PMV[2][2][2], int mv_field_sel[2][2], int dmvector[2]);static void pred(unsigned char *src[], int sfield,  unsigned char *dst[], int dfield,  int lx, int w, int h, int x, int y, int dx, int dy, int addflag);static void pred_comp(unsigned char *src, unsigned char *dst,  int lx, int w, int h, int x, int y, int dx, int dy, int addflag);static void calc_DMV(int DMV[][2], int *dmvector, int mvx,  int mvy);static void clearblock(unsigned char *cur[], int i0, int j0);
static short PACKED_0[4] = { 0, 0, 0, 0 };
static short PACKED_1[4] = { 1, 1, 1, 1 };
static short PACKED_2[4] = { 2, 2, 2, 2 };
/* 為一個完整的圖像形成預(yù)測 * * reff: 前向預(yù)測參考幀 * refb: 后向預(yù)測參考幀 * cur:  目標(biāo)幀 * secondfield: 預(yù)測幀的第二場 * mbi:  宏塊信息 *  */void predict(reff,refb,cur,secondfield,mbi)unsigned char *reff[],*refb[],*cur[3];int secondfield;struct mbinfo *mbi;{  int i, j, k;  k = 0;  for (j=0; j<height2; j+=16)    for (i=0; i<width; i+=16)    {      predict_mb(reff,refb,cur,width,i,j,pict_type,pict_struct,                 mbi[k].mb_type,mbi[k].motion_type,secondfield,                 mbi[k].MV,mbi[k].mv_field_sel,mbi[k].dmvector);      k++;    }}/* 為一個宏塊形成預(yù)測
 * * oldref: 前向預(yù)測的參考幀 * newref: 后向預(yù)測的參考幀 * cur:    目標(biāo)幀 * lx:     幀的寬度 * bx,by:  要預(yù)測的宏塊的圖像 (場或幀) 坐標(biāo)
 * pict_type: I, P 或 B * pict_struct: FRAME_PICTURE, TOP_FIELD, BOTTOM_FIELD * mb_type:     MB_FORWARD, MB_BACKWARD, MB_INTRA * motion_type: MC_FRAME, MC_FIELD, MC_16X8, MC_DMV * secondfield: 預(yù)測一幀的第二場 * PMV[2][2][2]: 運動向量 * mv_field_sel[2][2]: 運動向量場選擇 * dmvector: 差分運動向量 *  */static void predict_mb(oldref,newref,cur,lx,bx,by,pict_type,pict_struct,  mb_type,motion_type,secondfield,PMV,mv_field_sel,dmvector)unsigned char *oldref[],*newref[],*cur[];int lx;int bx,by;int pict_type;int pict_struct;int mb_type;int motion_type;int secondfield;int PMV[2][2][2], mv_field_sel[2][2], dmvector[2];{  int addflag, currentfield;  unsigned char **predframe;  int DMV[2][2];  if (mb_type&MB_INTRA)  {    clearblock(cur,bx,by);    return;  }  addflag = 0; /* first prediction is stored, second is added and averaged */  if ((mb_type & MB_FORWARD) || (pict_type==P_TYPE))  {    /* forward prediction, including zero MV in P pictures */    if (pict_struct==FRAME_PICTURE)    {      /* frame picture */      if ((motion_type==MC_FRAME) || !(mb_type & MB_FORWARD))      {        /* frame-based prediction in frame picture */        pred(oldref,0,cur,0,          lx,16,16,bx,by,PMV[0][0][0],PMV[0][0][1],0);      }      else if (motion_type==MC_FIELD)      {               /* top field prediction */        pred(oldref,mv_field_sel[0][0],cur,0,          lx<<1,16,8,bx,by>>1,PMV[0][0][0],PMV[0][0][1]>>1,0);        /* bottom field prediction */        pred(oldref,mv_field_sel[1][0],cur,1,          lx<<1,16,8,bx,by>>1,PMV[1][0][0],PMV[1][0][1]>>1,0);      }      else if (motion_type==MC_DMV)      {        /* calculate derived motion vectors */        calc_DMV(DMV,dmvector,PMV[0][0][0],PMV[0][0][1]>>1);        /* predict top field from top field */        pred(oldref,0,cur,0,          lx<<1,16,8,bx,by>>1,PMV[0][0][0],PMV[0][0][1]>>1,0);        /* predict bottom field from bottom field */        pred(oldref,1,cur,1,          lx<<1,16,8,bx,by>>1,PMV[0][0][0],PMV[0][0][1]>>1,0);        /* predict and add to top field from bottom field */        pred(oldref,1,cur,0,          lx<<1,16,8,bx,by>>1,DMV[0][0],DMV[0][1],1);        /* predict and add to bottom field from top field */        pred(oldref,0,cur,1,          lx<<1,16,8,bx,by>>1,DMV[1][0],DMV[1][1],1);      }      else      {        /* invalid motion_type in frame picture */        if (!quiet)          fprintf(stderr,"invalid motion_type\n");      }    }    else /* TOP_FIELD or BOTTOM_FIELD */    {      /* field picture */      currentfield = (pict_struct==BOTTOM_FIELD);      if ((pict_type==P_TYPE) && secondfield          && (currentfield!=mv_field_sel[0][0]))        predframe = newref; /* same frame */      else        predframe = oldref; /* previous frame */      if ((motion_type==MC_FIELD) || !(mb_type & MB_FORWARD))      {        /* field-based prediction in field picture */        pred(predframe,mv_field_sel[0][0],cur,currentfield,          lx<<1,16,16,bx,by,PMV[0][0][0],PMV[0][0][1],0);      }      else if (motion_type==MC_16X8)      {        /* 16 x 8 motion compensation in field picture */        /* upper half */        pred(predframe,mv_field_sel[0][0],cur,currentfield,          lx<<1,16,8,bx,by,PMV[0][0][0],PMV[0][0][1],0);        if ((pict_type==P_TYPE) && secondfield            && (currentfield!=mv_field_sel[1][0]))          predframe = newref; /* same frame */        else          predframe = oldref; /* previous frame */        /* lower half */        pred(predframe,mv_field_sel[1][0],cur,currentfield,          lx<<1,16,8,bx,by+8,PMV[1][0][0],PMV[1][0][1],0);      }      else if (motion_type==MC_DMV)      {                /* determine which frame to use for prediction */        if (secondfield)          predframe = newref; /* same frame */        else          predframe = oldref; /* previous frame */        /* calculate derived motion vectors */        calc_DMV(DMV,dmvector,PMV[0][0][0],PMV[0][0][1]);        /* predict from field of same parity */        pred(oldref,currentfield,cur,currentfield,          lx<<1,16,16,bx,by,PMV[0][0][0],PMV[0][0][1],0);        /* predict from field of opposite parity */        pred(predframe,!currentfield,cur,currentfield,          lx<<1,16,16,bx,by,DMV[0][0],DMV[0][1],1);      }      else      {        if (!quiet)          fprintf(stderr,"invalid motion_type\n");      }    }    addflag = 1;   }  if (mb_type & MB_BACKWARD)  {    /* backward prediction */    if (pict_struct==FRAME_PICTURE)    {      /* frame picture */      if (motion_type==MC_FRAME)      {        /* frame-based prediction in frame picture */        pred(newref,0,cur,0,          lx,16,16,bx,by,PMV[0][1][0],PMV[0][1][1],addflag);      }      else      {                /* top field prediction */        pred(newref,mv_field_sel[0][1],cur,0,          lx<<1,16,8,bx,by>>1,PMV[0][1][0],PMV[0][1][1]>>1,addflag);        /* bottom field prediction */        pred(newref,mv_field_sel[1][1],cur,1,          lx<<1,16,8,bx,by>>1,PMV[1][1][0],PMV[1][1][1]>>1,addflag);      }    }    else /* TOP_FIELD or BOTTOM_FIELD */    {      /* field picture */      currentfield = (pict_struct==BOTTOM_FIELD);      if (motion_type==MC_FIELD)      {        /* field-based prediction in field picture */        pred(newref,mv_field_sel[0][1],cur,currentfield,          lx<<1,16,16,bx,by,PMV[0][1][0],PMV[0][1][1],addflag);      }      else if (motion_type==MC_16X8)      {        /* 16 x 8 motion compensation in field picture */        /* upper half */        pred(newref,mv_field_sel[0][1],cur,currentfield,          lx<<1,16,8,bx,by,PMV[0][1][0],PMV[0][1][1],addflag);        /* lower half */        pred(newref,mv_field_sel[1][1],cur,currentfield,          lx<<1,16,8,bx,by+8,PMV[1][1][0],PMV[1][1][1],addflag);      }      else      {        /* invalid motion_type in field picture */        if (!quiet)          fprintf(stderr,"invalid motion_type\n");      }    }  }}/* 預(yù)測矩形塊 * * src:     源幀(Y,U,V) * sfield:  源場選擇(0: 幀或上半場, 1: 下半場) * dst:     目標(biāo)幀(Y,U,V) * dfield:  目標(biāo)場選擇 * * 下面的數(shù)值是亮度圖像的參數(shù) * lx:      相鄰像素的垂直距離 * w,h:     塊的寬度和高度 * x,y:     目標(biāo)塊的坐標(biāo) * dx,dy:   半像素運動向量 * addflag: 存儲或加預(yù)測 */static void pred(src,sfield,dst,dfield,lx,w,h,x,y,dx,dy,addflag)unsigned char *src[];int sfield;unsigned char *dst[];int dfield;int lx;int w, h;int x, y;int dx, dy;int addflag;{  int cc;  for (cc=0; cc<3; cc++)  {    if (cc==1)    {      /* scale for color components */      if (chroma_format==CHROMA420)      {        /* vertical */        h >>= 1; y >>= 1; dy /= 2;      }      if (chroma_format!=CHROMA444)      {        /* horizontal */        w >>= 1; x >>= 1; dx /= 2;        lx >>= 1;      }    }    pred_comp(src[cc]+(sfield?lx>>1:0),dst[cc]+(dfield?lx>>1:0),      lx,w,h,x,y,dx,dy,addflag);  }}/* 低等級預(yù)測
 *
 * src:     源幀(Y,U,V)
 * dst:     目標(biāo)幀(Y,U,V)
 *
 * lx:      相鄰像素的垂直距離
 * w,h:     塊的寬度和高度
 * x,y:     目標(biāo)塊的坐標(biāo)
 * dx,dy:   半像素運動向量
 * addflag: 存儲或加預(yù)測