/vol/vipdata/irtk/common++/include/irtkAllocate.h

/*=========================================================================

  Library   : Image Registration Toolkit (IRTK)
  Module    : $Id: irtkAllocate.h 2 2008-12-23 12:40:14Z dr $
  Copyright : Imperial College, Department of Computing
              Visual Information Processing (VIP), 2008 onwards
  Date      : $Date: 2008-12-23 12:40:14 +0000 (Tue, 23 Dec 2008) $
  Version   : $Revision: 2 $
  Changes   : $Author: dr $

=========================================================================*/

#ifndef _IRTKALLOCATE_H

#define _IRTKALLOCATE_H

#include <irtkCommon.h>

template <class Type> inline Type **Allocate(Type **matrix, int x, int y)
{
  int i;

  if ((matrix = new Type *[y]) == NULL) {
    cerr << "Allocate: malloc failed for " << x << " x " << y << "\n";
    exit(1);
  }

  if ((matrix[0] = new Type[y*x]) == NULL) {
    cerr << "Allocate: malloc failed for " << x << " x " << y << "\n";
    exit(1);
  }

  for (i = 1; i < y; i++) {
    matrix[i] = matrix[i-1] + x;
  }

#ifdef DEBUG
  memory_allocated += x*y*sizeof(Type);
  cout << "Allocate: Memory allocated is " << memory_allocated << " bytes.\n";
#endif

  return matrix;
}

template <class Type> inline Type ***Allocate(Type ***matrix, int x, int y, int z)
{
  int i, j;

  if ((matrix = new Type **[z]) == NULL) {
    cerr << "Allocate: malloc failed for " << x << " x " << y << " x ";
    cerr << z << "\n";
    exit(1);
  }

  if ((matrix[0] = new Type *[z*y]) == NULL) {
    cerr << "Allocate: malloc failed for " << x << " x " << y << " x ";
    cerr << z << "\n";
    exit(1);
  }

  for (i = 1; i < z; i++) {
    matrix[i] = matrix[i-1] + y;
  }

  if ((matrix[0][0] = new Type[z*y*x]) == NULL) {
    cerr << "Allocate: malloc failed for " << x << " x " << y << " x ";
    cerr << z << "\n";
    exit(1);
  }

  for (i = 0; i < z; i++) {
    for (j = 0; j < y; j++) {
      matrix[i][j] = matrix[0][0] + (i*y+j)*x;
    }
  }

#ifdef DEBUG
  memory_allocated += x*y*z*sizeof(Type);
  cout << "Allocate: Memory allocated is " << memory_allocated << " bytes.\n";
#endif

  return matrix;
}

template <class Type> inline Type ****Allocate(Type ****matrix, int x, int y, int z, int t)
{
  int i, j, k;

  if ((matrix = new Type ***[t]) == NULL) {
    cerr << "Allocate: malloc failed for " << x << " x " << y << " x ";
    cerr << z << " x " << t << "\n";
    exit(1);
  }

  if ((matrix[0] = new Type **[t*z]) == NULL) {
    cerr << "Allocate: malloc failed for " << x << " x " << y << " x ";
    cerr << z << " x " << t << "\n";
    exit(1);
  }

  for (i = 1; i < t; i++) {
    matrix[i] = matrix[i-1] + z;
  }

  if ((matrix[0][0] = new Type*[t*z*y]) == NULL) {
    cerr << "Allocate: malloc failed for " << x << " x " << y << " x ";
    cerr << z << " x " << t << "\n";
    exit(1);
  }

  for (i = 0; i < t; i++) {
    for (j = 0; j < z; j++) {
      matrix[i][j] = matrix[0][0] + i*z*y + j*y;
    }
  }

  if ((matrix[0][0][0] = new Type[t*z*y*x]) == NULL) {
    cerr << "Allocate: malloc failed for " << x << " x " << y << " x ";
    cerr << z << " x " << t << "\n";
    exit(1);
  }

  for (i = 0; i < t; i++) {
    for (j = 0; j < z; j++) {
      for (k = 0; k < y; k++) {
        matrix[i][j][k] = matrix[0][0][0] + i*z*y*x + j*y*x + k*x;
      }
    }
  }

#ifdef DEBUG
  memory_allocated += x*y*z*t*sizeof(Type);
  cout << "Allocate: Memory allocated is " << memory_allocated << " bytes.\n";
#endif

  return matrix;
}

#endif

Generated on Mon Mar 23 16:55:52 2009 for IRTK by Doxygen 1.5.4