memalloc.c

包含了从MPEG4的视频解码到H.264的视频编码部分的源代码

/*
***********************************************************************
* COPYRIGHT AND WARRANTY INFORMATION
*
* Copyright 2001, International Telecommunications Union, Geneva
*
* DISCLAIMER OF WARRANTY 
*
* These software programs are available to the user without any
* license fee or royalty on an "as is" basis. The ITU disclaims
* any and all warranties, whether express, implied, or
* statutory, including any implied warranties of merchantability
* or of fitness for a particular purpose.  In no event shall the
* contributor or the ITU 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 ITU does not represent or warrant that the programs furnished
* hereunder are free of infringement of any third-party patents.
* Commercial implementations of ITU-T Recommendations, including
* shareware, may be subject to royalty fees to patent holders.
* Information regarding the ITU-T patent policy is available from
* the ITU Web site at http://www.itu.int.
*
* THIS IS NOT A GRANT OF PATENT RIGHTS - SEE THE ITU-T PATENT POLICY.
************************************************************************
*/

/*!
 ************************************************************************
 * \file  memalloc.c
 *
 * \brief
 *    Memory allocation and free helper funtions
 *
 * \author
 *    Main contributors (see contributors.h for copyright, address and affiliation details)
 ************************************************************************
 */

#include <stdlib.h>
#include "memalloc.h"

/*!
 ************************************************************************
 * \brief
 *    Allocate 2D memory array -> unsigned char array2D[rows][columns]
 *
 * \par Output:
 *    memory size in bytes
 ************************************************************************/
// Change 9-Aug-2001 P. List: dont allocate independant row arrays anymore
// but one complete array and move row-pointers to array. Now you can step
// to the next line with an offset of img->width
int get_mem2D(byte ***array2D, int rows, int columns)
{
  int i;

  if((*array2D      = (byte**)calloc(rows,        sizeof(byte*))) == NULL)
    no_mem_exit("get_mem2D: array2D");
  if(((*array2D)[0] = (byte* )calloc(columns*rows,sizeof(byte ))) == NULL)
    no_mem_exit("get_mem2D: array2D");

  for(i=1;i<rows;i++)
    (*array2D)[i] = (*array2D)[i-1] + columns ;

  return rows*columns;
}
//add by zdd
int get_mem2Dwithextend(byte ***array2D, int rows, int columns,int extw,int exth)
{
  int i;

  if((*array2D      = (byte**)calloc(rows+exth*2,        sizeof(byte*))) == NULL)
    no_mem_exit("get_mem2D: array2D");  
  if(((*array2D)[0] = (byte* )calloc((columns+extw*2)*(rows+exth*2),sizeof(byte ))) == NULL)
    no_mem_exit("get_mem2D: array2D");
  (*array2D)[0]+=extw;
  for(i=1;i<rows+exth*2;i++)
    (*array2D)[i] = (*array2D)[i-1] + (columns+extw*2) ;
  (*array2D)+=exth;
  return (columns+extw*2)*(rows+exth*2);
}
void free_mem2Dwithextend(byte **array2D,int columns,int extw,int exth)
{
  array2D-=exth;
  if (array2D>0)
  {
	  array2D[0]-=extw;
    if (array2D[0]>0)
      free (array2D[0]);
 
    free (array2D);
  } 
}

int get_mem3Dwithextend(byte ****array3D, int frames, int rows, int columns,int extw,int exth)
{
  int  j;

  if(((*array3D) = (byte***)calloc(frames,sizeof(byte**))) == NULL)
    no_mem_exit("get_mem3D: array3D");

  for(j=0;j<frames;j++)
    get_mem2Dwithextend( (*array3D)+j, rows, columns,extw, exth ) ;

  return frames*(rows+exth*2)*(columns+extw*2);
}

void free_mem3Dwithextend(byte ***array3D, int frames,int columns,int extw,int exth)
{
  int i;

  if (array3D)
  {
    for (i=0;i<frames;i++)
    { 
      free_mem2Dwithextend(array3D[i],columns,extw, exth);
    }
   free (array3D);
  } 
}
/*!
 ************************************************************************
 * \brief
 *    Allocate 2D memory array -> int array2D[rows][columns]
 *
 * \par Output:
 *    memory size in bytes
 ************************************************************************
 */
// same change as in get_mem2Dint
int get_mem2Dint(int ***array2D, int rows, int columns)
{
  int i;

  if((*array2D      = (int**)calloc(rows,        sizeof(int*))) == NULL)
    no_mem_exit("get_mem2Dint: array2D");
  if(((*array2D)[0] = (int* )calloc(rows*columns,sizeof(int ))) == NULL)
    no_mem_exit("get_mem2Dint: array2D");

  for(i=1 ; i<rows ; i++)
    (*array2D)[i] =  (*array2D)[i-1] + columns  ;

  return rows*columns*sizeof(int);
}

/*!
 ************************************************************************
 * \brief
 *    Allocate 3D memory array -> unsigned char array3D[frames][rows][columns]
 *
 * \par Output:
 *    memory size in bytes
 ************************************************************************
 */
// same change as in get_mem2Dint
int get_mem3D(byte ****array3D, int frames, int rows, int columns)
{
  int  j;
  byte  *buff;

  if(((*array3D) = (byte***)calloc(frames,sizeof(byte**))) == NULL)
    no_mem_exit("get_mem3Dint: array3D");
  if((buff=calloc(frames*rows*columns,1))==NULL)
	  no_mem_exit("get_mem3Dint: array3D");
  for(j=0;j<frames;j++)
  {    
	int i;
	if(((*array3D)[j]= (byte**)calloc(rows,sizeof(byte*))) == NULL)
		no_mem_exit("get_mem2Dint: array2D");
	(*array3D)[j][0]=buff+j*rows*columns;
	for(i=1 ; i<rows ; i++)
		(*array3D)[j][i] =  (*array3D)[j][i-1] + columns  ;
  }

  return frames*rows*columns;
}

/*!
 ************************************************************************
 * \brief
 *    Allocate 3D memory array -> int array3D[frames][rows][columns]
 *
 * \par Output:
 *    memory size in bytes
 ************************************************************************
 */
// same change as in get_mem2Dint
int get_mem3Dint(int ****array3D, int frames, int rows, int columns)
{
  int  j,*buff;

  if(((*array3D) = (int***)calloc(frames,sizeof(int**))) == NULL)
    no_mem_exit("get_mem3Dint: array3D");
  if((buff=calloc(frames*rows*columns,sizeof(int)))==NULL)
	  no_mem_exit("get_mem3Dint: array3D");
  for(j=0;j<frames;j++)
  {    
	int i;
	if(((*array3D)[j] = (int**)calloc(rows,sizeof(int*))) == NULL)
		no_mem_exit("get_mem2Dint: array2D");
	(*array3D)[j][0]=buff+j*rows*columns;
	for(i=1 ; i<rows ; i++)
		(*array3D)[j][i] =  (*array3D)[j][i-1] + columns  ;
  }

  return frames*rows*columns*sizeof(int);
}

/*!
 ************************************************************************
 * \brief
 *    free 2D memory array
 *    which was alocated with get_mem2D()
 ************************************************************************
 */
void free_mem2D(byte **array2D)
{
  if (array2D)
  {
    if (array2D[0])
      free (array2D[0]);

    free (array2D);
  } 
}

/*!
 ************************************************************************
 * \brief
 *    free 2D memory array
 *    which was alocated with get_mem2Dint()
 ************************************************************************
 */
void free_mem2Dint(int **array2D)
{
  if (array2D)
  {
    if (array2D[0]) 
      free (array2D[0]);
   
    free (array2D);

  } 
}

/*!
 ************************************************************************
 * \brief
 *    free 3D memory array
 *    which was alocated with get_mem3D()
 ************************************************************************
 */
void free_mem3D(byte ***array3D, int frames)
{
  int i;

  if (array3D)
  {
	free(array3D[0][0]);
    for (i=0;i<frames;i++)
    {       
      free(array3D[i]);
    }
   free (array3D);
  }
}

/*!
 ************************************************************************
 * \brief
 *    free 3D memory array 
 *    which was alocated with get_mem3Dint()
 ************************************************************************
 */
void free_mem3Dint(int ***array3D, int frames)
{
  int i;

  if (array3D)
  {
	  free(array3D[0][0]);
    for (i=0;i<frames;i++)
    { 
      free(array3D[i]);
    }
   free (array3D);
  } 
}

/*!
 ************************************************************************
 * \brief
 *    Exit program if memory allocation failed (using error())
 * \param where
 *    string indicating which memory allocation failed
 ************************************************************************
 */
void no_mem_exit(char *where)
{
  
}

/*!
 ************************************************************************
 * \brief
 *    free 4D memory array 
 *    which was alocated with get_mem4Dint()
 ************************************************************************
 */
void free_mem4Dint(int ****array4D, int idx, int frames )
{
  int  j;

  if (array4D)
  {
    for(j=0;j<idx;j++)
      free_mem3Dint( array4D[j], frames) ;
    free (array4D);
  } else
  {
    error ("",100);
  }
}
/*!
 ************************************************************************
 * \brief
 *    Allocate 4D memory array -> int array3D[frames][rows][columns][component]
 *
 * \par Output:
 *    memory size in bytes
 ************************************************************************
 */
// same change as in get_mem2Dint
int get_mem4Dint(int *****array4D, int idx, int frames, int rows, int columns )
{
  int  j;

  if(((*array4D) = (int****)calloc(idx,sizeof(int**))) == NULL)
    no_mem_exit("get_mem4Dint: array4D");

  for(j=0;j<idx;j++)
    get_mem3Dint( (*array4D)+j, frames, rows, columns ) ;

  return idx*frames*rows*columns*sizeof(int);
}