image.c

avs-s最新代码,包括编码器和解码器源码

		for (y=0; y<input->img_height/2; y++)
			for (x=input->img_width/2; x<img->width/2; x++)
			{
				imgUV_org_frm [0][y][x] = imgUV_org_frm [0][y][x-1];
				imgUV_org_frm [1][y][x] = imgUV_org_frm [1][y][x-1];
			}

			//padding bottom border
			for (y=input->img_width/2; y<img->width/2; y++)
				for (x=0; x<img->height/2; x++)
				{
					imgUV_org_frm [0][y][x] = imgUV_org_frm [0][y-1][x];
					imgUV_org_frm [1][y][x] = imgUV_org_frm [1][y-1][x];
				}
				if(input->InterlaceCodingOption != FRAME_CODING)
				{
					for (y=0; y<img->height; y += 2)
						for (x=0; x<img->width; x++)
						{
							imgY_org_top [y/2][x] = imgY_org_buffer[y*img->width    +x]; // !! Lum component for top field 
							imgY_org_bot [y/2][x] = imgY_org_buffer[(y+1)*img->width+x]; // !! Lum component for bot field
						}

						for (y=0; y<img->height/2; y += 2)
							for (x=0; x<img->width/2; x++)
							{
								imgUV_org_top[0][y/2][x] = u_buffer[y*img->width/2+x];    // !! Cr and Cb component for top field
								imgUV_org_top[1][y/2][x] = v_buffer[y*img->width/2+x];
								imgUV_org_bot[0][y/2][x] = u_buffer[(y+1)*img->width/2+x]; // !! Cr and Cb component for bot field
								imgUV_org_bot[1][y/2][x] = v_buffer[(y+1)*img->width/2+x];
							}
				}
}
  
/*
*************************************************************************
* Function: Calculates the absolute frame number in the source file out
        of various variables in img-> and input->
* Input:
* Output:
* Return: frame number in the file to be read
* Attention: \side effects
    global variable frame_no updated -- dunno, for what this one is necessary
*************************************************************************
*/


static int CalculateFrameNumber()
{
  if (img->type == B_IMG)
    frame_no = (IMG_NUMBER - 1) * (input->jumpd + 1) + img->b_interval * img->b_frame_to_code;
  else
  {
    frame_no = IMG_NUMBER * (input->jumpd + 1);
  }

  return frame_no;
}
  
/*
*************************************************************************
* Function:Reads one new frame from file
* Input: FrameNoInFile: Frame number in the source file
    HeaderSize: Number of bytes in the source file to be skipped
    xs: horizontal size of frame in pixels, must be divisible by 16
    ys: vertical size of frame in pixels, must be divisible by 16 or
        32 in case of MB-adaptive frame/field coding
    sf: Sourceframe structure to which the frame is written
* Output:
* Return: 
* Attention:
*************************************************************************
*/

static void ReadOneFrame (int FrameNoInFile, int HeaderSize, int xs, int ys)

{
	//const unsigned int  bytes_y = input->img_width *input->stuff_height; //modify by wuzhongmou 0610
	const unsigned int  bytes_y = input->img_width *input->img_height; //add by wuzhongmou 0610
	const unsigned int bytes_uv = bytes_y/4;
	const int framesize_in_bytes = bytes_y + 2*bytes_uv;
	//int stuff_height_cr = (input->img_height-input->stuff_height)/2;
	int off_y = input->img_width*input->img_height;
	int off_uv = off_y/4;

	assert (FrameNumberInFile == FrameNoInFile);

	//4:0:0 WANGJP START
	if (input->chroma_format || input->input_chroma_format)
	{
		if (fseek (p_in, framesize_in_bytes*FrameNoInFile + HeaderSize, SEEK_SET) != 0)
			error ("ReadOneFrame: cannot fseek to (Header size) in p_in", -1);
	}
	else
	{
		if (fseek (p_in, bytes_y*FrameNoInFile + HeaderSize, SEEK_SET) != 0)
			error ("ReadOneFrame: cannot fseek to (Header size) in p_in", -1);
	}
	//WANGJP END

	if (fread (imgY_org_buffer, 1, bytes_y, p_in) != (unsigned )bytes_y)
	{
		printf ("ReadOneFrame: cannot read %d bytes from input file, unexpected EOF?, exiting", bytes_y);
		exit (-1);
	}

	//4:0:0 WANGJP START
	if (input->chroma_format || input->input_chroma_format)
	{
		if (fread (imgY_org_buffer + off_y, 1, bytes_uv, p_in) != (unsigned )bytes_uv)
		{
			printf ("ReadOneFrame: cannot read %d bytes from input file, unexpected EOF?, exiting", bytes_y);
			exit (-1);
		}


		if (fread (imgY_org_buffer + input->img_height*input->img_width + input->img_height*input->img_width/4, 1, bytes_uv, p_in) != (unsigned )bytes_uv)
		{
			printf ("ReadOneFrame: cannot read %d bytes from input file, unexpected EOF?, exiting", bytes_y);
			exit (-1);
		}
	}
	else
	{
		memset(imgY_org_buffer+off_y, 128, 2*bytes_uv);
	}
	//WANGJP END
}
  
/*
*************************************************************************
* Function:point to frame coding variables 
* Input:
* Output:
* Return: 
* Attention:
*************************************************************************
*/

void put_buffer_frame()
{
	int i,j;

  imgY_org  = imgY_org_frm;
  imgUV_org = imgUV_org_frm;  
  tmp_mv    = tmp_mv_frm;

	//initialize ref index 1/4 pixel
#ifdef COREFRAME
   if(cur_core_frame_flag ==1)
   {
	   //1/4象素参考
	   mref[0] = core_frame_mref_frm[core_frame_ref_num];
	   mref[1] = mref_frm[0];
       //chroma

	for(j=0;j<2;j++)
	{
		mcef[0][j]   = core_reference_frame[core_frame_ref_num][j+1];  
	}
	for(j=0;j<2;j++)
	{
		mcef[1][j]   = ref_frm[0][j+1];  
	}
		//integer pixel for luma
	
  	Refbuf11[0] = &core_reference_frame[core_frame_ref_num][0][0][0];
	Refbuf11[1] = &ref_frm[0][0][0][0];	

   }
   else
   {
		for(i=0;i<2;i++)
		{
			mref[i] = mref_frm[i];
		}
			
		//integer pixel for chroma
		for(i=0;i<2;i++)
		for(j=0;j<2;j++)
		{
			mcef[i][j]   = ref_frm[i][j+1];  
		}

		//integer pixel for luma
	  for(i=0;i<2;i++)
  		Refbuf11[i] = &ref_frm[i][0][0][0];	   
   }
#else

	for(i=0;i<2;i++)
	{
		mref[i] = mref_frm[i];
	}
		
	//integer pixel for chroma
	for(i=0;i<2;i++)
	for(j=0;j<2;j++)
	{
		mcef[i][j]   = ref_frm[i][j+1];  
	}

	//integer pixel for luma
  for(i=0;i<2;i++)
  	Refbuf11[i] = &ref_frm[i][0][0][0];
#endif		

	//current reconstructed image 
	
	imgY = imgY_frm = current_frame[0];
	imgUV = imgUV_frm =  &current_frame[1];
 	
  refFrArr    = refFrArr_frm;
  fw_refFrArr = fw_refFrArr_frm;
  bw_refFrArr = bw_refFrArr_frm;
}

/*
*************************************************************************
* Function:point to top field coding variables 
* Input:
* Output:
* Return: 
* Attention:
*************************************************************************
*/
void put_buffer_top()
{
	int i,j;
	
  img->fld_type = 0;
	
	
  imgY_org  = imgY_org_top;
  imgUV_org = imgUV_org_top;
#ifdef COREFRAME
    //核心帧
	if(cur_core_frame_flag) 
	{
	   //1/4象素参考luma
	   mref[0] = core_frame_mref_fld[core_frame_ref_num*2+1];
	   mref[1] = core_frame_mref_fld[core_frame_ref_num*2];
	   mref[2] = mref_fld[0];
	   mref[3] = mref_fld[1];
       //	//intial ref index for field coding 
	   ref[0] = core_reference_field[core_frame_ref_num*2+1];
	   ref[1] = core_reference_field[core_frame_ref_num*2];
	   ref[2] = ref_fld[0];
	   ref[3] = ref_fld[1];

		//integer luma pixel for interlace
		for(i=0;i<4;i++)
		{
			Refbuf11[i] = &ref[i][0][0][0];
		}

		//integer chroma pixel for interlace 
		

		for(j=0;j<2;j++)
		{
			mcef[0][j]   = core_reference_field[core_frame_ref_num*2+1][j+1];  
		}
		for(j=0;j<2;j++)
		{
			mcef[1][j]   = core_reference_field[core_frame_ref_num*2][j+1];  
		}
		for (j=0;j<2;j++)//ref_index = 0
			for (i=0;i<2;i++)
			{
				mcef[j+2][i] = ref_fld[j][i+1];
			}		
	}
	else
	{
		//intial ref index for field coding 
	  for(i=0;i<4;i++)              // 3 2 1 0
			ref[i] = ref_fld[i]; 
		
		//initialize ref index 1/4 pixel
		for(i=0;i<4;i++)
		{
			mref[i] = mref_fld[i];
		}
		
		//integer chroma pixel for interlace 
		for (j=0;j<4;j++)//ref_index = 0
			for (i=0;i<2;i++)
			{
				mcef[j][i] = ref_fld[j][i+1];
			}
			
			//integer luma pixel for interlace
			for(i=0;i<4;i++)
			{
				Refbuf11[i] = &ref[i][0][0][0];
			}
	}
#else
	 
	//intial ref index for field coding 
  for(i=0;i<4;i++)              // 3 2 1 0
		ref[i] = ref_fld[i]; 
	
	//initialize ref index 1/4 pixel
	for(i=0;i<4;i++)
	{
		mref[i] = mref_fld[i];
	}
	
	//integer chroma pixel for interlace 
	for (j=0;j<4;j++)//ref_index = 0
		for (i=0;i<2;i++)
		{
			mcef[j][i] = ref_fld[j][i+1];
		}
		
		//integer luma pixel for interlace
		for(i=0;i<4;i++)
		{
			Refbuf11[i] = &ref[i][0][0][0];
		}
#endif
		
		imgY  = imgY_top   = current_field[0];
		imgUV = imgUV_top  = &current_field[1];
		
		tmp_mv = tmp_mv_top;
		refFrArr = refFrArr_top;
		fw_refFrArr = fw_refFrArr_top;
		bw_refFrArr = bw_refFrArr_top;
}
  
/*
*************************************************************************
* Function:point to bottom field coding variables 
* Input:
* Output:
* Return: 
* Attention:
*************************************************************************
*/

void put_buffer_bot()
{
	int i,j;

  img->fld_type = 1; 
  
  imgY_org = imgY_org_bot;
  imgUV_org = imgUV_org_bot;
  
  tmp_mv = tmp_mv_bot;
  refFrArr = refFrArr_bot;
  fw_refFrArr = fw_refFrArr_bot;
  bw_refFrArr = bw_refFrArr_bot;
#ifdef COREFRAME  
  if(cur_core_frame_flag)
  {
	   //1/4象素参考luma
	   mref[0] = mref_fld[0];
	   mref[1] = core_frame_mref_fld[core_frame_ref_num*2+1];
	   mref[2] = core_frame_mref_fld[core_frame_ref_num*2];
	   mref[3] = mref_fld[1];
       //	//intial ref index for field coding
	   ref[0] = ref_fld[0];	   
	   ref[1] = core_reference_field[core_frame_ref_num*2+1];
	   ref[2] = core_reference_field[core_frame_ref_num*2];
	   ref[3] = ref_fld[1];
		//integer luma pixel for interlace
		for(i=0;i<4;i++)
		{
			Refbuf11[i] = &ref[i][0][0][0];
		}

		//integer chroma pixel for interlace 
		

		for(j=0;j<2;j++)
		{
			mcef[0][j]   = ref_fld[0][j+1];  
		}
		for(j=0;j<2;j++)
		{
			mcef[1][j]   = core_reference_field[core_frame_ref_num*2+1][j+1];  
		}
		for(j=0;j<2;j++)
		{
			mcef[2][j]   = core_reference_field[core_frame_ref_num*2][j+1];  
		}
		for(j=0;j<2;j++)
		{
			mcef[3][j]   = ref_fld[1][j+1];  
		}		  
  }
  else
  {
	   //intial ref index for field coding 
	  for(i=0;i<4;i++)
  		ref[i] = ref_fld[i]; 

		//initialize ref index 1/4 pixel
		for(i=0;i<4;i++)
		{
			mref[i] = mref_fld[i];
		}

		//integer chroma pixel for interlace 
		for (j=0;j<4;j++)//ref_index = 0
		for (i=0;i<2;i++)
		{
			mcef[j][i] = ref_fld[j][i+1];
		}
		
		//integer luma pixel for interlace
		for(i=0;i<4;i++)
		{
			Refbuf11[i] = &ref[i][0][0][0];
		}
  }
#else
   //intial ref index for field coding 
  for(i=0;i<4;i++)
  	ref[i] = ref_fld[i]; 

	//initialize ref index 1/4 pixel
	for(i=0;i<4;i++)
	{
		mref[i] = mref_fld[i];
	}

	//integer chroma pixel for interlace 
	for (j=0;j<4;j++)//ref_index = 0
	for (i=0;i<2;i++)
	{