/* motion.c, 運(yùn)動預(yù)測程序模塊                               */

#include <stdio.h>#include "global.h"/* 私有函數(shù)*/static void frame_ME(unsigned char *oldorg, unsigned char *neworg,  unsigned char *oldref, unsigned char *newref, unsigned char *cur,  int i, int j, int sxf, int syf, int sxb, int syb, struct mbinfo *mbi);static void field_ME(unsigned char *oldorg, unsigned char *neworg,  unsigned char *oldref, unsigned char *newref, unsigned char *cur,  unsigned char *curref, int i, int j, int sxf, int syf, int sxb, int syb,  struct mbinfo *mbi, int secondfield, int ipflag);static void frame_estimate(unsigned char *org,  unsigned char *ref, unsigned char *mb,  int i, int j,  int sx, int sy, int *iminp, int *jminp, int *imintp, int *jmintp,  int *iminbp, int *jminbp, int *dframep, int *dfieldp,  int *tselp, int *bselp, int imins[2][2], int jmins[2][2]);static void field_estimate(unsigned char *toporg,  unsigned char *topref, unsigned char *botorg, unsigned char *botref,  unsigned char *mb, int i, int j, int sx, int sy, int ipflag,  int *iminp, int *jminp, int *imin8up, int *jmin8up, int *imin8lp,  int *jmin8lp, int *dfieldp, int *d8p, int *selp, int *sel8up, int *sel8lp,  int *iminsp, int *jminsp, int *dsp);static void dpframe_estimate(unsigned char *ref,  unsigned char *mb, int i, int j, int iminf[2][2], int jminf[2][2],  int *iminp, int *jminp, int *imindmvp, int *jmindmvp,  int *dmcp, int *vmcp);static void dpfield_estimate(unsigned char *topref,  unsigned char *botref, unsigned char *mb,  int i, int j, int imins, int jmins, int *imindmvp, int *jmindmvp,  int *dmcp, int *vmcp);static int fullsearch(unsigned char *org, unsigned char *ref,  unsigned char *blk,  int lx, int i0, int j0, int sx, int sy, int h, int xmax, int ymax,  int *iminp, int *jminp);static int dist1(unsigned char *blk1, unsigned char *blk2,  int lx, int hx, int hy, int h, int distlim);static int dist2(unsigned char *blk1, unsigned char *blk2,  int lx, int hx, int hy, int h);static int bdist1(unsigned char *pf, unsigned char *pb,  unsigned char *p2, int lx, int hxf, int hyf, int hxb, int hyb, int h);static int bdist2(unsigned char *pf, unsigned char *pb,  unsigned char *p2, int lx, int hxf, int hyf, int hxb, int hyb, int h);
static int variance(unsigned char *p, int lx);
static int calcDist16(unsigned char *p1In, unsigned char *p2In);

/* 私有變量 */

static unsigned char tmpResult[16];

// 下面的常數(shù)將用在MMX部分
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ùn)動估計 * *oldorg：前向預(yù)測的原始參考圖（供P圖像和B圖像使用）
 *neworg：后向預(yù)測的原始參考圖（僅供B圖像使用）
 *oldref：前向預(yù)測的重建圖像（P圖像和B圖像）
 *newref：后向預(yù)測的重建圖像（B圖像）
 *cur：當(dāng)前原始圖像幀（由預(yù)測圖像產(chǎn)生的）
 *curref：當(dāng)前的重建圖像幀（用來根據(jù)第一個場預(yù)測第二個場）
 *sxf，syf：前向搜索窗口
 *sxb，syb：后向搜索窗口
 *mbi：指向宏塊信息結(jié)構(gòu)體的指針
 *輸出結(jié)果mbi->
 *mb_type: 0, MB_INTRA, MB_FORWARD, MB_BACKWARD, MB_FORWARD|MB_BACKWARD
 *MV[][][]: 運(yùn)動向量 (場格式)
 *mv_field_sel: 頂部/底部場
 *motion_type: MC_FIELD, MC_16X8 */void motion_estimation(oldorg,neworg,oldref,newref,cur,curref,  sxf,syf,sxb,syb,mbi,secondfield,ipflag)unsigned char *oldorg,*neworg,*oldref,*newref,*cur,*curref;int sxf,syf,sxb,syb;struct mbinfo *mbi;int secondfield,ipflag;{  int i, j;  for (j=0; j<height2; j+=16)  {    for (i=0; i<width; i+=16)    {      if (pict_struct==FRAME_PICTURE)        frame_ME(oldorg,neworg,oldref,newref,cur,i,j,sxf,syf,sxb,syb,mbi);      else        field_ME(oldorg,neworg,oldref,newref,cur,curref,i,j,sxf,syf,sxb,syb,          mbi,secondfield,ipflag);      mbi++;    }    if (!quiet)    {      putc('.',stderr);      fflush(stderr);    }  }  if (!quiet)    putc('\r',stderr);}static void frame_ME(oldorg,neworg,oldref,newref,cur,i,j,sxf,syf,sxb,syb,mbi)unsigned char *oldorg,*neworg,*oldref,*newref,*cur;int i,j,sxf,syf,sxb,syb;struct mbinfo *mbi;{  int imin,jmin,iminf,jminf,iminr,jminr;  int imint,jmint,iminb,jminb;  int imintf,jmintf,iminbf,jminbf;  int imintr,jmintr,iminbr,jminbr;  int var,v0;  int dmc,dmcf,dmcr,dmci,vmc,vmcf,vmcr,vmci;  int dmcfield,dmcfieldf,dmcfieldr,dmcfieldi;  int tsel,bsel,tself,bself,tselr,bselr;  unsigned char *mb;  int imins[2][2],jmins[2][2];  int imindp,jmindp,imindmv,jmindmv,dmc_dp,vmc_dp;  mb = cur + i + width*j;  var = variance(mb,width);  if (pict_type==I_TYPE)    mbi->mb_type = MB_INTRA;  else if (pict_type==P_TYPE)  {    if (frame_pred_dct)    {      dmc = fullsearch(oldorg,oldref,mb,                       width,i,j,sxf,syf,16,width,height,&imin,&jmin);      vmc = dist2(oldref+(imin>>1)+width*(jmin>>1),mb,                  width,imin&1,jmin&1,16);      mbi->motion_type = MC_FRAME;    }    else    {      frame_estimate(oldorg,oldref,mb,i,j,sxf,syf,        &imin,&jmin,&imint,&jmint,&iminb,&jminb,        &dmc,&dmcfield,&tsel,&bsel,imins,jmins);      if (M==1)        dpframe_estimate(oldref,mb,i,j>>1,imins,jmins,          &imindp,&jmindp,&imindmv,&jmindmv,&dmc_dp,&vmc_dp);      if (M==1 && dmc_dp<dmc && dmc_dp<dmcfield)      {        mbi->motion_type = MC_DMV;        dmc = dmc_dp;        vmc = vmc_dp;      }      else if (dmc<=dmcfield)      {        mbi->motion_type = MC_FRAME;        vmc = dist2(oldref+(imin>>1)+width*(jmin>>1),mb,                    width,imin&1,jmin&1,16);      }      else      {        mbi->motion_type = MC_FIELD;        dmc = dmcfield;        vmc = dist2(oldref+(tsel?width:0)+(imint>>1)+(width<<1)*(jmint>>1),                    mb,width<<1,imint&1,jmint&1,8);        vmc+= dist2(oldref+(bsel?width:0)+(iminb>>1)+(width<<1)*(jminb>>1),                    mb+width,width<<1,iminb&1,jminb&1,8);      }    }     if (vmc>var && vmc>=9*256)      mbi->mb_type = MB_INTRA;    else    {            v0 = dist2(oldref+i+width*j,mb,width,0,0,16);      if (4*v0>5*vmc && v0>=9*256)      {        /* use MC */        var = vmc;        mbi->mb_type = MB_FORWARD;        if (mbi->motion_type==MC_FRAME)        {          mbi->MV[0][0][0] = imin - (i<<1);          mbi->MV[0][0][1] = jmin - (j<<1);        }        else if (mbi->motion_type==MC_DMV)        {                    /* same parity vector */          mbi->MV[0][0][0] = imindp - (i<<1);          mbi->MV[0][0][1] = (jmindp<<1) - (j<<1);          /* opposite parity vector */          mbi->dmvector[0] = imindmv;          mbi->dmvector[1] = jmindmv;        }        else        {                   mbi->MV[0][0][0] = imint - (i<<1);          mbi->MV[0][0][1] = (jmint<<1) - (j<<1);          mbi->MV[1][0][0] = iminb - (i<<1);          mbi->MV[1][0][1] = (jminb<<1) - (j<<1);          mbi->mv_field_sel[0][0] = tsel;          mbi->mv_field_sel[1][0] = bsel;        }      }      else      {        /* No-MC */        var = v0;        mbi->mb_type = 0;        mbi->motion_type = MC_FRAME;        mbi->MV[0][0][0] = 0;        mbi->MV[0][0][1] = 0;      }    }  }  else /* B幀圖像*/  {    if (frame_pred_dct)    {      /* 前向 */      dmcf = fullsearch(oldorg,oldref,mb,                        width,i,j,sxf,syf,16,width,height,&iminf,&jminf);      vmcf = dist2(oldref+(iminf>>1)+width*(jminf>>1),mb,                   width,iminf&1,jminf&1,16);      /* 后向 */      dmcr = fullsearch(neworg,newref,mb,                        width,i,j,sxb,syb,16,width,height,&iminr,&jminr);      vmcr = dist2(newref+(iminr>>1)+width*(jminr>>1),mb,                   width,iminr&1,jminr&1,16);      /* 雙向幀 */      vmci = bdist2(oldref+(iminf>>1)+width*(jminf>>1),                    newref+(iminr>>1)+width*(jminr>>1),                    mb,width,iminf&1,jminf&1,iminr&1,jminr&1,16);            if (vmcf<=vmcr && vmcf<=vmci)      {        vmc = vmcf;        mbi->mb_type = MB_FORWARD;      }      else if (vmcr<=vmci)      {        vmc = vmcr;        mbi->mb_type = MB_BACKWARD;      }      else      {        vmc = vmci;        mbi->mb_type = MB_FORWARD|MB_BACKWARD;      }      mbi->motion_type = MC_FRAME;    }    else    {      /* 前向預(yù)測 */      frame_estimate(oldorg,oldref,mb,i,j,sxf,syf,        &iminf,&jminf,&imintf,&jmintf,&iminbf,&jminbf,        &dmcf,&dmcfieldf,&tself,&bself,imins,jmins);      /* 后向預(yù)測 */      frame_estimate(neworg,newref,mb,i,j,sxb,syb,        &iminr,&jminr,&imintr,&jmintr,&iminbr,&jminbr,        &dmcr,&dmcfieldr,&tselr,&bselr,imins,jmins);      /* 計算交織距離 */      /* 幀 */      dmci = bdist1(oldref+(iminf>>1)+width*(jminf>>1),                    newref+(iminr>>1)+width*(jminr>>1),                    mb,width,iminf&1,jminf&1,iminr&1,jminr&1,16);      /* 上半部分場 */      dmcfieldi = bdist1(                    oldref+(imintf>>1)+(tself?width:0)+(width<<1)*(jmintf>>1),                    newref+(imintr>>1)+(tselr?width:0)+(width<<1)*(jmintr>>1),                    mb,width<<1,imintf&1,jmintf&1,imintr&1,jmintr&1,8);      /* 下半部分場 */      dmcfieldi+= bdist1(                    oldref+(iminbf>>1)+(bself?width:0)+(width<<1)*(jminbf>>1),                    newref+(iminbr>>1)+(bselr?width:0)+(width<<1)*(jminbr>>1),                    mb+width,width<<1,iminbf&1,jminbf&1,iminbr&1,jminbr&1,8);            if (dmci<dmcfieldi && dmci<dmcf && dmci<dmcfieldf          && dmci<dmcr && dmci<dmcfieldr)      {        /* 交織幀 */        mbi->mb_type = MB_FORWARD|MB_BACKWARD;        mbi->motion_type = MC_FRAME;        vmc = bdist2(oldref+(iminf>>1)+width*(jminf>>1),                     newref+(iminr>>1)+width*(jminr>>1),                     mb,width,iminf&1,jminf&1,iminr&1,jminr&1,16);      }      else if (dmcfieldi<dmcf && dmcfieldi<dmcfieldf               && dmcfieldi<dmcr && dmcfieldi<dmcfieldr)      {        /* 交織場*/        mbi->mb_type = MB_FORWARD|MB_BACKWARD;        mbi->motion_type = MC_FIELD;        vmc = bdist2(oldref+(imintf>>1)+(tself?width:0)+(width<<1)*(jmintf>>1),                     newref+(imintr>>1)+(tselr?width:0)+(width<<1)*(jmintr>>1),                     mb,width<<1,imintf&1,jmintf&1,imintr&1,jmintr&1,8);        vmc+= bdist2(oldref+(iminbf>>1)+(bself?width:0)+(width<<1)*(jminbf>>1),                     newref+(iminbr>>1)+(bselr?width:0)+(width<<1)*(jminbr>>1),                     mb+width,width<<1,iminbf&1,jminbf&1,iminbr&1,jminbr&1,8);      }      else if (dmcf<dmcfieldf && dmcf<dmcr && dmcf<dmcfieldr)      {        /* 前向幀 */        mbi->mb_type = MB_FORWARD;        mbi->motion_type = MC_FRAME;        vmc = dist2(oldref+(iminf>>1)+width*(jminf>>1),mb,                    width,iminf&1,jminf&1,16);      }      else if (dmcfieldf<dmcr && dmcfieldf<dmcfieldr)      {        /* 前向場 */        mbi->mb_type = MB_FORWARD;        mbi->motion_type = MC_FIELD;        vmc = dist2(oldref+(tself?width:0)+(imintf>>1)+(width<<1)*(jmintf>>1),                    mb,width<<1,imintf&1,jmintf&1,8);        vmc+= dist2(oldref+(bself?width:0)+(iminbf>>1)+(width<<1)*(jminbf>>1),                    mb+width,width<<1,iminbf&1,jminbf&1,8);      }      else if (dmcr<dmcfieldr)      {        /* 后向幀 */        mbi->mb_type = MB_BACKWARD;        mbi->motion_type = MC_FRAME;        vmc = dist2(newref+(iminr>>1)+width*(jminr>>1),mb,                    width,iminr&1,jminr&1,16);      }      else      {        /* 后向場 */        mbi->mb_type = MB_BACKWARD;        mbi->motion_type = MC_FIELD;        vmc = dist2(newref+(tselr?width:0)+(imintr>>1)+(width<<1)*(jmintr>>1),                    mb,width<<1,imintr&1,jmintr&1,8);        vmc+= dist2(newref+(bselr?width:0)+(iminbr>>1)+(width<<1)*(jminbr>>1),                    mb+width,width<<1,iminbr&1,jminbr&1,8);      }    }    /* 選擇幀內(nèi)編碼或者非幀內(nèi)編碼     *     * 選擇是基于幀內(nèi)塊方差和預(yù)測誤差的方差     *          */    if (vmc>var && vmc>=9*256)      mbi->mb_type = MB_INTRA;    else    {      var = vmc;      if (mbi->motion_type==MC_FRAME)      {        /* 前向*/        mbi->MV[0][0][0] = iminf - (i<<1);        mbi->MV[0][0][1] = jminf - (j<<1);        /* 后向*/        mbi->MV[0][1][0] = iminr - (i<<1);        mbi->MV[0][1][1] = jminr - (j<<1);      }      else      {                /* 前向 */        mbi->MV[0][0][0] = imintf - (i<<1);        mbi->MV[0][0][1] = (jmintf<<1) - (j<<1);        mbi->MV[1][0][0] = iminbf - (i<<1);        mbi->MV[1][0][1] = (jminbf<<1) - (j<<1);        mbi->mv_field_sel[0][0] = tself;        mbi->mv_field_sel[1][0] = bself;        /* 后向*/        mbi->MV[0][1][0] = imintr - (i<<1);        mbi->MV[0][1][1] = (jmintr<<1) - (j<<1);        mbi->MV[1][1][0] = iminbr - (i<<1);        mbi->MV[1][1][1] = (jminbr<<1) - (j<<1);        mbi->mv_field_sel[0][1] = tselr;        mbi->mv_field_sel[1][1] = bselr;      }    }  }  mbi->var = var;}/* * 場圖像的運(yùn)動估計 * *oldorg：前向預(yù)測的原始參考圖（供P圖像和B圖像使用）
 *neworg：后向預(yù)測的原始參考圖（僅供B圖像使用）
 *oldref：前向預(yù)測的重建圖像（P圖像和B圖像）
 *newref：后向預(yù)測的重建圖像（B圖像）
 *cur：當(dāng)前原始圖像幀（由預(yù)測圖像產(chǎn)生的）