fdtd-alloc.h

来自「利用VC++进行FDTD三维远近场变换」· C头文件 代码 · 共 36 行

#ifndef _FDTD_ALLOC_H
#define _FDTD_ALLOC_H

#include <stdio.h>
#include <stdlib.h>

/* memory allocation macros */
#define ALLOC_1D(name,nx,type)                                      \
    name = (type *)calloc(nx,sizeof(type));                         \
    if (!name) {                                                    \
      perror("ALLOC_1D");                                           \
      fprintf(stderr,                                               \
              "Allocation failed for " #name ". Terminating...\n"); \
      exit(-1);                                                     \
    }

#define ALLOC_2D(name,nx,ny,type)                                   \
    name = (type *)calloc((nx)*(ny),sizeof(type));                  \
    if (!name) {                                                    \
      perror("ALLOC_2D");                                           \
      fprintf(stderr,                                               \
              "Allocation failed for " #name ".  Terminating...\n");\
      exit(-1);                                                     \
    }

#define ALLOC_3D(name,nx,ny,nz,type)                                \
    name = (type *)calloc((nx)*(ny)*(nz),sizeof(type));             \
    if (!name) {                                                    \
      perror("ALLOC_2D");                                           \
      fprintf(stderr,                                               \
              "Allocation failed for " #name ".  Terminating...\n");\
      exit(-1);                                                     \
    }

#endif