mc.cpp

完整的RTP RTSP代码库

					ppxlcPred += (m_vopmd.RRVmode.iRRVOnOff == 1) ? (iSize) : (MB_SIZE);
//					ppxlcPred += MB_SIZE;
// ~RRV
				}
			}
			else {  //bXSubPxl && !bYSubPxl
				for (iy = 0; iy < iSize; iy++){
					for (ix = 0; ix < iSize; ix++)
						ppxlcPred [ix] = (ppxlcRef [ix] + ppxlcRef [ix + 1] + 1 - iRoundingControl) >> 1;
					ppxlcRef += m_iFrameWidthY;
// RRV modification
					ppxlcPred += (m_vopmd.RRVmode.iRRVOnOff == 1) ? (iSize) : (MB_SIZE);
//					ppxlcPred += MB_SIZE;
// ~RRV
				}
			}
		}
		else {
			const PixelC* ppxlcRefBot;
			if (!(xRef & 1)) {  //!bXSubPxl&& bYSubPxl
				for (iy = 0; iy < iSize; iy++) {
					ppxlcRefBot = ppxlcRef + m_iFrameWidthY;		//UPln -> pixels (xInt,yInt+1);
					for (ix = 0; ix < iSize; ix++) 
						ppxlcPred [ix] = (ppxlcRef [ix] + ppxlcRefBot [ix] + 1 - iRoundingControl) >> 1;
					ppxlcRef = ppxlcRefBot;
// RRV modification
					ppxlcPred += (m_vopmd.RRVmode.iRRVOnOff == 1) ? (iSize) : (MB_SIZE);
//					ppxlcPred += MB_SIZE;
// ~RRV
				}
			}
			else { // bXSubPxl && bYSubPxl
				for (iy = 0; iy < iSize; iy++) {
					ppxlcRefBot = ppxlcRef + m_iFrameWidthY;		//UPln -> pixels (xInt,yInt+1);
					for (ix = 0; ix < iSize; ix++){
						ppxlcPred [ix] = (
							ppxlcRef [ix + 1] + ppxlcRef [ix] +
							ppxlcRefBot [ix + 1] + ppxlcRefBot [ix] + 2 - iRoundingControl
						) >> 2;

					}
					ppxlcRef = ppxlcRefBot;
// RRV modification
					ppxlcPred += (m_vopmd.RRVmode.iRRVOnOff == 1) ? (iSize) : (MB_SIZE);
//					ppxlcPred += MB_SIZE;
// ~RRV
				}
			}
		}
	}
}

Void CVideoObject::motionCompQuarterSample (
	PixelC* ppxlcPred, // can be either Y or A
	const PixelC* ppxlcRefLeftTop, // point to left-top of the frame
	Int iSize, // either MB or BLOCK size
	CoordI xRef, CoordI yRef, // x + mvX in quarter pel unit
	Int iRoundingControl, // rounding control
	CRct *prctMVLimit // extended bounding box
    )
{
	CoordI ix, iy;
    U8 *ppxlcblk;
    
	limitMVRangeToExtendedBBQuarterPel(xRef,yRef,prctMVLimit,iSize);
    
    Int blkSizeX, blkSizeY;
    if (iSize == 0) { // INTERLACED
      blkSizeX = 16;
      blkSizeY = 8;
      
      ppxlcblk = (U8*) calloc(blkSizeX*blkSizeX,sizeof(U8));
      blkInterpolateY (ppxlcRefLeftTop,iSize,xRef,yRef,ppxlcblk,iRoundingControl);
      
      for (iy = 0; iy < blkSizeX; iy++) { 
        if (!(iy&1)) 
          for (ix = 0; ix < blkSizeX; ix++) { 
            ppxlcPred [ix] = *(ppxlcblk+ix+iy*blkSizeX);
          }
        ppxlcPred += blkSizeX;
      }
    } //~INTERLACED
    else { 
      blkSizeX = iSize;
      blkSizeY = iSize;
      
      ppxlcblk = (U8*) calloc(blkSizeX*blkSizeX,sizeof(U8));
      blkInterpolateY (ppxlcRefLeftTop,iSize,xRef,yRef,ppxlcblk,iRoundingControl);
      
      for (iy = 0; iy < blkSizeX; iy++) { 
        for (ix = 0; ix < blkSizeX; ix++) { 
          ppxlcPred [ix] = *(ppxlcblk+ix+iy*blkSizeX);
        }
        ppxlcPred += MB_SIZE;
      }
    }

    free(ppxlcblk);
}

Void CVideoObject::motionCompDirectMode(                // Interlaced direct mode
    CoordI x, CoordI y,
    CMBMode *pmbmd,
    const CMotionVector *pmvRef,
    CRct *prctMVLimitFwd, CRct *prctMVLimitBak,
	Int plane	// plane=1 for grey scale, plane=0 for texture, 02-17-99

                                        )
{
	Int* rgiMvRound = NULL;
	UInt uiDivisor = 0;
	Int xRefUVF,yRefUVF,xRefUVB,yRefUVB;

// begin of new changes 10/21/98
	Int iMBX,iMBY;
	const CMBMode *pmbmdRef; 

	if(m_volmd.fAUsage != RECTANGLE)
	{
		iMBX=(x-m_rctCurrVOPY.left)/MB_SIZE;
		iMBY=(y-m_rctCurrVOPY.top)/MB_SIZE;
		pmbmdRef= m_rgmbmdRef +
				(iMBX+iMBY*m_iNumMBXRef);
	}
	else
	{
		iMBX=x/MB_SIZE;
		iMBY=y/MB_SIZE;
		pmbmdRef= m_rgmbmdRef +
				iMBX+iMBY*m_iNumMBXRef;
	}
// end of new changes 10/21/98

  if (pmbmdRef->m_rgTranspStatus[0]==ALL) {
    static CMotionVector mvZero[5];
    pmvRef = mvZero;
  }

	if ((iMBX<m_iNumMBXRef && iMBX>=0 && iMBY<m_iNumMBYRef && iMBY>=0)&& // new change 10/21/98
		(pmbmdRef->m_bFieldMV&&pmbmdRef->m_rgTranspStatus[0]!=ALL)) {
      static I8 iTROffsetTop[] = {  0, 0,  1, 1, 0, 0, -1, -1 };
      static I8 iTROffsetBot[] = { -1, 0, -1, 0, 1, 0,  1,  0 };
      CoordI iXFwdTop, iXFwdBot, iXBakTop, iXBakBot;
      CoordI iYFwdTop, iYFwdBot, iYBakTop, iYBakBot;
      CoordI iXFTQSUV = 0, iXFBQSUV = 0, iXBTQSUV = 0, iXBBQSUV = 0;
      CoordI iYFTQSUV = 0, iYFBQSUV = 0, iYBTQSUV = 0, iYBBQSUV = 0;
      const CMotionVector *pmvRefTop, *pmvRefBot;
      Int iTopRefFldOffset = 0, iBotRefFldOffset = 0;
      Int iCode = (Int)(vopmd().bTopFieldFirst) << 2;
      
      pmbmd->m_bFieldMV = 1;  // set field direct mode for grey scale, Krit 02-17-99

      assert((pmbmdRef->m_dctMd != INTRA) && (pmbmdRef->m_dctMd != INTRAQ));
      if (pmbmdRef->m_bForwardTop) {
        iCode |= 2;
        iTopRefFldOffset = 1;
        pmvRefTop = pmvRef + 6;
      } else
        pmvRefTop = pmvRef + 5;
      if (pmbmdRef->m_bForwardBottom) {
        iCode |= 1;
        iBotRefFldOffset = 1;
        pmvRefBot = pmvRef + 8;
      } else
        pmvRefBot = pmvRef + 7;
      Int iTempRefDTop = 2*(m_tFutureRef - m_tPastRef) + iTROffsetTop[iCode];
      Int iTempRefDBot = 2*(m_tFutureRef - m_tPastRef) + iTROffsetBot[iCode];
      Int iTempRefBTop = 2*(m_t          - m_tPastRef) + iTROffsetTop[iCode];
      Int iTempRefBBot = 2*(m_t          - m_tPastRef) + iTROffsetBot[iCode];
      
      assert(iTempRefDTop > 0); assert(iTempRefDBot > 0); assert(iTempRefBTop > 0); assert(iTempRefBBot > 0);
      
      // Find MVs for the top field
      iXFwdTop = (pmvRefTop->m_vctTrueHalfPel.x * iTempRefBTop) / iTempRefDTop + pmbmd->m_vctDirectDeltaMV.x;
      iYFwdTop = (pmvRefTop->m_vctTrueHalfPel.y * iTempRefBTop) / iTempRefDTop + pmbmd->m_vctDirectDeltaMV.y;
      iXBakTop = pmbmd->m_vctDirectDeltaMV.x ? (iXFwdTop - pmvRefTop->m_vctTrueHalfPel.x) :
        ((pmvRefTop->m_vctTrueHalfPel.x * (iTempRefBTop - iTempRefDTop)) / iTempRefDTop);
      iYBakTop = pmbmd->m_vctDirectDeltaMV.y ? (iYFwdTop - pmvRefTop->m_vctTrueHalfPel.y) :
        ((pmvRefTop->m_vctTrueHalfPel.y * (iTempRefBTop - iTempRefDTop)) / iTempRefDTop);
      
      // Find MVs for the bottom field
      iXFwdBot = (pmvRefBot->m_vctTrueHalfPel.x * iTempRefBBot) / iTempRefDBot + pmbmd->m_vctDirectDeltaMV.x;
      iYFwdBot = (pmvRefBot->m_vctTrueHalfPel.y * iTempRefBBot) / iTempRefDBot + pmbmd->m_vctDirectDeltaMV.y;
      iXBakBot = pmbmd->m_vctDirectDeltaMV.x ? (iXFwdBot - pmvRefBot->m_vctTrueHalfPel.x) :
        ((pmvRefBot->m_vctTrueHalfPel.x * (iTempRefBBot - iTempRefDBot)) / iTempRefDBot);
      iYBakBot = pmbmd->m_vctDirectDeltaMV.y ? (iYFwdBot - pmvRefBot->m_vctTrueHalfPel.y) :
        ((pmvRefBot->m_vctTrueHalfPel.y * (iTempRefBBot - iTempRefDBot)) / iTempRefDBot);
    
      // Motion compensate the top field forward
      if (m_volmd.bQuarterSample) {
        iXFTQSUV = iXFwdTop/2; 
        iYFTQSUV = iYFwdTop/2;
        iXBTQSUV = iXBakTop/2; 
        iYBTQSUV = iYBakTop/2;

        iXFBQSUV = iXFwdBot/2;
        iYFBQSUV = iYFwdBot/2;
        iXBBQSUV = iXBakBot/2; 
        iYBBQSUV = iYBakBot/2;

        iYFwdTop = iYFwdTop & ~1;
        iYBakTop = iYBakTop & ~1;
        iYFwdBot = iYFwdBot & ~1;
        iYBakBot = iYBakBot & ~1;
    
        iXFwdTop += 4*x; iYFwdTop += 4*y;
//        limitMVRangeToExtendedBBQuarterPel(iXFwdTop, iYFwdTop, prctMVLimitFwd, MB_SIZE); // deleted by Y.Suzuki for the extended bounding box support
		if (plane==0) {  // texture MC
          motionCompQuarterSample(m_ppxlcPredMBY, 
                                  m_pvopcRefQ0->pixelsY() + iTopRefFldOffset * m_iFrameWidthY, 0, 
                                  iXFwdTop, iYFwdTop, m_vopmd.iRoundingControl, prctMVLimitFwd);
          
//          iXFTQSUV += 2*x; iYFTQSUV += 2*y; // deleted by Y.Suzuki for the extended bounding box support
//          limitMVRangeToExtendedBBHalfPel(iXFTQSUV, iYFTQSUV, prctMVLimitFwd, MB_SIZE); // deleted by Y.Suzuki for the extended bounding box support
//          iXFTQSUV -= 2*x; iYFTQSUV -= 2*y; // deleted by Y.Suzuki for the extended bounding box support
          motionCompFieldUV(m_ppxlcPredMBU, m_ppxlcPredMBV, m_pvopcRefQ0, x, y,
                            (iXFTQSUV & 3) ? ((iXFTQSUV >> 1) | 1) : (iXFTQSUV >> 1),
                            (iYFTQSUV & 6) ? ((iYFTQSUV >> 1) | 2) : (iYFTQSUV >> 1), 
                            iTopRefFldOffset,
                            prctMVLimitFwd); // added by Y.Suzuki for the extended bounding box support
		}
		else {  // plane=1, grey scale MC
          for(Int iAuxComp=0;iAuxComp<m_volmd.iAuxCompCount;iAuxComp++) { // MAC (SB) 26-Nov-99
            motionCompQuarterSample(m_ppxlcPredMBA[iAuxComp], 
              m_pvopcRefQ0->pixelsA(iAuxComp) + iTopRefFldOffset * m_iFrameWidthY, 0, 
              iXFwdTop, iYFwdTop, m_vopmd.iRoundingControl, prctMVLimitFwd);
          }
		}
      }
      else {
          iXFwdTop += 2*x; iYFwdTop += 2*y;
//        limitMVRangeToExtendedBBHalfPel(iXFwdTop, iYFwdTop, prctMVLimitFwd, MB_SIZE); // deleted by Y.Suzuki for the extended bounding box support
		if (plane==0) {  // texture MC
          motionCompYField(m_ppxlcPredMBY,
                           m_pvopcRefQ0->pixelsY() + iTopRefFldOffset * m_iFrameWidthY, iXFwdTop, iYFwdTop,
                           prctMVLimitFwd); // added by Y.Suzuki for the extended bounding box support
          iXFwdTop -= 2*x; iYFwdTop -= 2*y;
          motionCompFieldUV(m_ppxlcPredMBU, m_ppxlcPredMBV, m_pvopcRefQ0, x, y,
                            (iXFwdTop & 3) ? ((iXFwdTop >> 1) | 1) : (iXFwdTop >> 1),
                            (iYFwdTop & 6) ? ((iYFwdTop >> 1) | 2) : (iYFwdTop >> 1), iTopRefFldOffset,
                            prctMVLimitFwd); // added by Y.Suzuki for the extended bounding box support
		}
		else {  // plane=1, grey scale MC
          for(Int iAuxComp=0;iAuxComp<m_volmd.iAuxCompCount;iAuxComp++) { // MAC (SB) 26-Nov-99
            motionCompYField(m_ppxlcPredMBA[iAuxComp],
                           m_pvopcRefQ0->pixelsA(iAuxComp) + iTopRefFldOffset * m_iFrameWidthY, iXFwdTop, iYFwdTop,
						   prctMVLimitFwd); // added by Y.Suzuki for the extended bounding box support
          }

          iXFwdTop -= 2*x; iYFwdTop -= 2*y;
		}
      }
    

      // Motion compensate the top field backward
    if (m_volmd.bQuarterSample) {
      iXBakTop += 4*x; iYBakTop += 4*y;
//      limitMVRangeToExtendedBBQuarterPel(iXBakTop, iYBakTop, prctMVLimitFwd, MB_SIZE); // deleted by Y.Suzuki for the extended bounding box support
      if (plane==0) {  // texture MC
        motionCompQuarterSample(m_ppxlcPredMBBackY, 
                                m_pvopcRefQ1->pixelsY(), 0, 
                                iXBakTop, iYBakTop, m_vopmd.iRoundingControl, prctMVLimitBak);
        
//        iXBTQSUV += 2*x; iYBTQSUV += 2*y; // deleted by Y.Suzuki for the extended bounding box support
//        limitMVRangeToExtendedBBHalfPel(iXBTQSUV, iYBTQSUV, prctMVLimitBak, MB_SIZE); // deleted by Y.Suzuki for the extended bounding box support
//        iXBTQSUV -= 2*x; iYBTQSUV -= 2*y; // deleted by Y.Suzuki for the extended bounding box support
        motionCompFieldUV(m_ppxlcPredMBBackU, m_ppxlcPredMBBackV, m_pvopcRefQ1, x, y,
                          (iXBTQSUV & 3) ? ((iXBTQSUV >> 1) | 1) : (iXBTQSUV >> 1),
                          (iYBTQSUV & 6) ? ((iYBTQSUV >> 1) | 2) : (iYBTQSUV >> 1), 
                          0,
                          prctMVLimitBak); // added by Y.Suzuki for the extended bounding box support
      }
      else {  // plane=1, grey scale MC
        for(Int iAuxComp=0;iAuxComp<m_volmd.iAuxCompCount;iAuxComp++) { // MAC (SB) 26-Nov-99
          motionCompQuarterSample(m_ppxlcPredMBBackA[iAuxComp], 
                                  m_pvopcRefQ1->pixelsA(iAuxComp), 0, 
                                  iXBakTop, iYBakTop, m_vopmd.iRoundingControl, prctMVLimitBak);
        }
	  }
    }
    else {
      iXBakTop += 2*x; iYBakTop += 2*y;
//      limitMVRangeToExtendedBBHalfPel(iXBakTop, iYBakTop, prctMVLimitBak, MB_SIZE); // deleted by Y.Suzuki for the extended bounding box support
      if (plane==0) {  // texture MC
        motionCompYField(m_ppxlcPredMBBackY, m_pvopcRefQ1->pixelsY(), iXBakTop,  iYBakTop,
			prctMVLimitBak); // added by Y.Suzuki for the extended bounding box support
        iXBakTop -= 2*x; iYBakTop -= 2*y;
        motionCompFieldUV(m_ppxlcPredMBBackU, m_ppxlcPredMBBackV, m_pvopcRefQ1, x, y,
                          (iXBakTop & 3) ? ((iXBakTop >> 1) | 1) : (iXBakTop >> 1),
                          (iYBakTop & 6) ? ((iYBakTop >> 1) | 2) : (iYBakTop >> 1), 0,
			   prctMVLimitBak); // added by Y.Suzuki for the extended bounding box support
      }
      else {  // plane=1, grey scale MC
        for(Int iAuxComp=0;iAuxComp<m_volmd.iAuxCompCount;iAuxComp++) { // MAC (SB) 26-Nov-99
          motionCompYField(m_ppxlcPredMBBackA[iAuxComp], m_pvopcRefQ1->pixelsA(iAuxComp), iXBakTop,  iYBakTop,
			  prctMVLimitBak); // added by Y.Suzuki for the extended bounding box support
        }
        iXBakTop -= 2*x; iYBakTop -= 2*y;
      }
    }
    

    // Motion compensate the bottom field forward
    if (m_volmd.bQuarterSample) {
      iXFwdBot += 4*x; iYFwdBot += 4*y;
//      limitMVRangeToExtendedBBQuarterPel(iXFwdBot, iYFwdBot, prctMVLimitFwd, MB_SIZE); // deleted by Y.Suzuki for the extended bounding box support
      if (plane==0) {  // texture MC
        motionCompQuarterSample(m_ppxlcPredMBY + MB_SIZE, 
                                m_pvopcRefQ0->pixelsY() + iBotRefFldOffset * m_iFrameWidthY, 0, 
                                iXFwdBot, iYFwdBot, m_vopmd.iRoundingControl, prctMVLimitFwd);