/* recon.c, motion compensation routines                                    *//* Copyright (C) 1994, MPEG Software Simulation Group. All Rights Reserved. *//* * Disclaimer of Warranty * * These software programs are available to the user without any license fee or * royalty on an "as is" basis.  The MPEG Software Simulation Group disclaims * any and all warranties, whether express, implied, or statuary, including any * implied warranties or merchantability or of fitness for a particular * purpose.  In no event shall the copyright-holder be liable for any * incidental, punitive, or consequential damages of any kind whatsoever * arising from the use of these programs. * * This disclaimer of warranty extends to the user of these programs and user's * customers, employees, agents, transferees, successors, and assigns. * * The MPEG Software Simulation Group does not represent or warrant that the * programs furnished hereunder are free of infringement of any third-party * patents. * * Commercial implementations of MPEG-1 and MPEG-2 video, including shareware, * are subject to royalty fees to patent holders.  Many of these patents are * general enough such that they are unavoidable regardless of implementation * design. * */#include <stdio.h>#include "config.h"#include "global.h"/* private prototypes */static void recon _ANSI_ARGS_((unsigned char *src[], int sfield,  unsigned char *dst[], int dfield,  int lx, int lx2, int w, int h, int x, int y, int dx, int dy,  int addflag));static void recon_comp _ANSI_ARGS_((unsigned char *src, unsigned char *dst,  int lx, int lx2, int w, int h, int x, int y, int dx, int dy, int addflag));void reconstruct(bx,by,mb_type,motion_type,PMV,mv_field_sel,dmvector,stwtype)int bx, by;int mb_type;int motion_type;int PMV[2][2][2], mv_field_sel[2][2], dmvector[2];int stwtype;{  int currentfield;  unsigned char **predframe;  int DMV[2][2];  int stwtop, stwbot;  stwtop = stwtype%3; /* 0:temporal, 1:(spat+temp)/2, 2:spatial */  stwbot = stwtype/3;  if ((mb_type & MB_FORWARD) || (pict_type==P_TYPE))  {    if (pict_struct==FRAME_PICTURE)    {      if ((motion_type==MC_FRAME) || !(mb_type & MB_FORWARD))      {        /* frame-based prediction */        if (stwtop<2)          recon(oldrefframe,0,newframe,0,            coded_picture_width,coded_picture_width<<1,16,8,bx,by,            PMV[0][0][0],PMV[0][0][1],stwtop);        if (stwbot<2)          recon(oldrefframe,1,newframe,1,            coded_picture_width,coded_picture_width<<1,16,8,bx,by,            PMV[0][0][0],PMV[0][0][1],stwbot);      }      else if (motion_type==MC_FIELD) /* field-based prediction */      {        /* top field prediction */        if (stwtop<2)          recon(oldrefframe,mv_field_sel[0][0],newframe,0,            coded_picture_width<<1,coded_picture_width<<1,16,8,bx,by>>1,            PMV[0][0][0],PMV[0][0][1]>>1,stwtop);        /* bottom field prediction */        if (stwbot<2)          recon(oldrefframe,mv_field_sel[1][0],newframe,1,            coded_picture_width<<1,coded_picture_width<<1,16,8,bx,by>>1,            PMV[1][0][0],PMV[1][0][1]>>1,stwbot);      }      else if (motion_type==MC_DMV) /* dual prime prediction */      {        /* calculate derived motion vectors */        calc_DMV(DMV,dmvector,PMV[0][0][0],PMV[0][0][1]>>1);        if (stwtop<2)        {          /* predict top field from top field */          recon(oldrefframe,0,newframe,0,            coded_picture_width<<1,coded_picture_width<<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 */          recon(oldrefframe,1,newframe,0,            coded_picture_width<<1,coded_picture_width<<1,16,8,bx,by>>1,            DMV[0][0],DMV[0][1],1);        }        if (stwbot<2)        {          /* predict bottom field from bottom field */          recon(oldrefframe,1,newframe,1,            coded_picture_width<<1,coded_picture_width<<1,16,8,bx,by>>1,            PMV[0][0][0],PMV[0][0][1]>>1,0);          /* predict and add to bottom field from top field */          recon(oldrefframe,0,newframe,1,            coded_picture_width<<1,coded_picture_width<<1,16,8,bx,by>>1,            DMV[1][0],DMV[1][1],1);        }      }      else        /* invalid motion_type */        printf("invalid motion_type\n");    }    else /* TOP_FIELD or BOTTOM_FIELD */    {      /* field picture */      currentfield = (pict_struct==BOTTOM_FIELD);      /* determine which frame to use for prediction */      if ((pict_type==P_TYPE) && secondfield         && (currentfield!=mv_field_sel[0][0]))        predframe = refframe; /* same frame */      else        predframe = oldrefframe; /* previous frame */      if ((motion_type==MC_FIELD) || !(mb_type & MB_FORWARD))      {        /* field-based prediction */        if (stwtop<2)          recon(predframe,mv_field_sel[0][0],newframe,0,            coded_picture_width<<1,coded_picture_width<<1,16,16,bx,by,            PMV[0][0][0],PMV[0][0][1],stwtop);      }      else if (motion_type==MC_16X8)      {        if (stwtop<2)        {          recon(predframe,mv_field_sel[0][0],newframe,0,            coded_picture_width<<1,coded_picture_width<<1,16,8,bx,by,            PMV[0][0][0],PMV[0][0][1],stwtop);          /* determine which frame to use for lower half prediction */          if ((pict_type==P_TYPE) && secondfield             && (currentfield!=mv_field_sel[1][0]))            predframe = refframe; /* same frame */          else            predframe = oldrefframe; /* previous frame */          recon(predframe,mv_field_sel[1][0],newframe,0,            coded_picture_width<<1,coded_picture_width<<1,16,8,bx,by+8,            PMV[1][0][0],PMV[1][0][1],stwtop);        }      }      else if (motion_type==MC_DMV) /* dual prime prediction */      {        if (secondfield)          predframe = refframe; /* same frame */        else          predframe = oldrefframe; /* previous frame */        /* calculate derived motion vectors */        calc_DMV(DMV,dmvector,PMV[0][0][0],PMV[0][0][1]);        /* predict from field of same parity */        recon(oldrefframe,currentfield,newframe,0,          coded_picture_width<<1,coded_picture_width<<1,16,16,bx,by,          PMV[0][0][0],PMV[0][0][1],0);        /* predict from field of opposite parity */        recon(predframe,!currentfield,newframe,0,          coded_picture_width<<1,coded_picture_width<<1,16,16,bx,by,          DMV[0][0],DMV[0][1],1);      }      else        /* invalid motion_type */        printf("invalid motion_type\n");    }    stwtop = stwbot = 1;  }  if (mb_type & MB_BACKWARD)  {    if (pict_struct==FRAME_PICTURE)    {      if (motion_type==MC_FRAME)      {        /* frame-based prediction */        if (stwtop<2)          recon(refframe,0,newframe,0,            coded_picture_width,coded_picture_width<<1,16,8,bx,by,            PMV[0][1][0],PMV[0][1][1],stwtop);        if (stwbot<2)          recon(refframe,1,newframe,1,            coded_picture_width,coded_picture_width<<1,16,8,bx,by,            PMV[0][1][0],PMV[0][1][1],stwbot);      }      else /* field-based prediction */      {        /* top field prediction */        if (stwtop<2)          recon(refframe,mv_field_sel[0][1],newframe,0,            coded_picture_width<<1,coded_picture_width<<1,16,8,bx,by>>1,            PMV[0][1][0],PMV[0][1][1]>>1,stwtop);        /* bottom field prediction */        if (stwbot<2)          recon(refframe,mv_field_sel[1][1],newframe,1,            coded_picture_width<<1,coded_picture_width<<1,16,8,bx,by>>1,            PMV[1][1][0],PMV[1][1][1]>>1,stwbot);      }    }    else /* TOP_FIELD or BOTTOM_FIELD */    {      /* field picture */      if (motion_type==MC_FIELD)      {        /* field-based prediction */        recon(refframe,mv_field_sel[0][1],newframe,0,          coded_picture_width<<1,coded_picture_width<<1,16,16,bx,by,          PMV[0][1][0],PMV[0][1][1],stwtop);      }      else if (motion_type==MC_16X8)      {        recon(refframe,mv_field_sel[0][1],newframe,0,          coded_picture_width<<1,coded_picture_width<<1,16,8,bx,by,          PMV[0][1][0],PMV[0][1][1],stwtop);        recon(refframe,mv_field_sel[1][1],newframe,0,          coded_picture_width<<1,coded_picture_width<<1,16,8,bx,by+8,          PMV[1][1][0],PMV[1][1][1],stwtop);      }      else        /* invalid motion_type */        printf("invalid motion_type\n");    }  }}static void recon(src,sfield,dst,dfield,lx,lx2,w,h,x,y,dx,dy,addflag)unsigned char *src[]; /* prediction source buffer */int sfield;           /* prediction source field number (0 or 1) */unsigned char *dst[]; /* prediction destination buffer */int dfield;           /* prediction destination field number (0 or 1)*/int lx,lx2;           /* horizontal offsets */int w,h;              /* prediction block/sub-block width, height */int x,y;              /* pixel co-ordinates of top-left sample in current MB */int dx,dy;            /* horizontal, vertical motion vector */int addflag;          /* add prediction error to prediction ? */{  /* Y */  recon_comp(src[0]+(sfield?lx2>>1:0),dst[0]+(dfield?lx2>>1:0),    lx,lx2,w,h,x,y,dx,dy,addflag);  if (chroma_format!=CHROMA444)  {    lx>>=1; lx2>>=1; w>>=1; x>>=1; dx/=2;  }  if (chroma_format==CHROMA420)  {    h>>=1; y>>=1; dy/=2;  }  /* Cb */  recon_comp(src[1]+(sfield?lx2>>1:0),dst[1]+(dfield?lx2>>1:0),    lx,lx2,w,h,x,y,dx,dy,addflag);  /* Cr */  recon_comp(src[2]+(sfield?lx2>>1:0),dst[2]+(dfield?lx2>>1:0),    lx,lx2,w,h,x,y,dx,dy,addflag);}static void recon_comp(src,dst,lx,lx2,w,h,x,y,dx,dy,addflag)unsigned char *src;unsigned char *dst;int lx,lx2;int w,h;int x,y;int dx,dy;int addflag;{  int xint, xh, yint, yh;  int i, j, v;  unsigned char *s, *d;  /* half pel scaling */  xint = dx>>1;  xh = dx & 1;  yint = dy>>1;  yh = dy & 1;  /* origins */  s = src + lx*(y+yint) + x + xint;  d = dst + lx*y + x;  if (!xh && !yh)    if (addflag)      for (j=0; j<h; j++)      {        for (i=0; i<w; i++)        {          v = d[i]+s[i];          d[i] = (v+(v>=0?1:0))>>1;        }        s+= lx2;        d+= lx2;      }    else      for (j=0; j<h; j++)      {        for (i=0; i<w; i++)          d[i] = s[i];        s+= lx2;        d+= lx2;      }  else if (!xh && yh)    if (addflag)      for (j=0; j<h; j++)      {        for (i=0; i<w; i++)        {          v = d[i] + ((unsigned int)(s[i]+s[i+lx]+1)>>1);          d[i]=(v+(v>=0?1:0))>>1;        }        s+= lx2;        d+= lx2;      }    else      for (j=0; j<h; j++)      {        for (i=0; i<w; i++)          d[i] = (unsigned int)(s[i]+s[i+lx]+1)>>1;        s+= lx2;        d+= lx2;      }  else if (xh && !yh)    if (addflag)      for (j=0; j<h; j++)      {        for (i=0; i<w; i++)        {          v = d[i] + ((unsigned int)(s[i]+s[i+1]+1)>>1);          d[i] = (v+(v>=0?1:0))>>1;        }        s+= lx2;        d+= lx2;      }    else      for (j=0; j<h; j++)      {        for (i=0; i<w; i++)          d[i] = (unsigned int)(s[i]+s[i+1]+1)>>1;        s+= lx2;        d+= lx2;      }  else /* if (xh && yh) */    if (addflag)      for (j=0; j<h; j++)      {        for (i=0; i<w; i++)        {          v = d[i] + ((unsigned int)(s[i]+s[i+1]+s[i+lx]+s[i+lx+1]+2)>>2);          d[i] = (v+(v>=0?1:0))>>1;        }        s+= lx2;        d+= lx2;      }    else      for (j=0; j<h; j++)      {        for (i=0; i<w; i++)          d[i] = (unsigned int)(s[i]+s[i+1]+s[i+lx]+s[i+lx+1]+2)>>2;        s+= lx2;        d+= lx2;      }}