lammps/tools/phonon/memory.h

#ifndef LMP_MEMORY_H
#define LMP_MEMORY_H

#define __STDC_LIMIT_MACROS
#define __STDC_FORMAT_MACROS

#include <cstdlib>
#include <stdint.h>
#include <inttypes.h>

typedef int64_t bigint;
#define BIGINT_FORMAT "%" PRId64
#define ATOBIGINT atoll

class Memory {
public:
  Memory(){};

  void *smalloc(bigint n, const char *);
  void *srealloc(void *, bigint n, const char *);
  void sfree(void *);
  void fail(const char *);

/* ----------------------------------------------------------------------
   create a 1d array
   ------------------------------------------------------------------------- */

  template <typename TYPE>
  TYPE *create(TYPE *&array, int n, const char *name)
    {
      bigint nbytes = sizeof(TYPE) * n;
      array = (TYPE *) smalloc(nbytes,name);
      return array;
    };

  template <typename TYPE>
  TYPE **create(TYPE **&, int, const char *name) {fail(name);}

/* ----------------------------------------------------------------------
   grow or shrink 1d array
   ------------------------------------------------------------------------- */

  template <typename TYPE>
  TYPE *grow(TYPE *&array, int n, const char *name)
    {
      if (array == NULL) return create(array,n,name);

      bigint nbytes = sizeof(TYPE) * n;
      array = (TYPE *) srealloc(array,nbytes,name);
      return array;
    };

  template <typename TYPE>
  TYPE **grow(TYPE **&, int, const char *name) {fail(name);}

/* ----------------------------------------------------------------------
   destroy a 1d array
   ------------------------------------------------------------------------- */

  template <typename TYPE>
  void destroy(TYPE *array)
    {
      sfree(array);
    };

/* ----------------------------------------------------------------------
   create a 1d array with index from nlo to nhi inclusive
   cannot grow it
   ------------------------------------------------------------------------- */

  template <typename TYPE>
  TYPE *create1d_offset(TYPE *&array, int nlo, int nhi, const char *name)
    {
      bigint nbytes = sizeof(TYPE) * (nhi-nlo+1);
      array = (TYPE *) smalloc(nbytes,name);
      array -= nlo;
      return array;
    }

  template <typename TYPE>
  TYPE **create1d_offset(TYPE **&, int, int, const char *name)
    {fail(name);}

/* ----------------------------------------------------------------------
   destroy a 1d array with index offset
   ------------------------------------------------------------------------- */

  template <typename TYPE>
  void destroy1d_offset(TYPE *array, int offset)
    {
      if (array) sfree(&array[offset]);
    }

/* ----------------------------------------------------------------------
   create a 2d array
   ------------------------------------------------------------------------- */

  template <typename TYPE>
  TYPE **create(TYPE **&array, int n1, int n2, const char *name)
    {
      bigint nbytes = sizeof(TYPE) * n1*n2;
      TYPE *data = (TYPE *) smalloc(nbytes,name);
      nbytes = sizeof(TYPE *) * n1;
      array = (TYPE **) smalloc(nbytes,name);

      int n = 0;
      for (int i = 0; i < n1; i++) {
        array[i] = &data[n];
        n += n2;
      }
      return array;
    }

  template <typename TYPE>
  TYPE ***create(TYPE ***&, int, int, const char *name)
    {fail(name);}

/* ----------------------------------------------------------------------
   grow or shrink 1st dim of a 2d array
   last dim must stay the same
   ------------------------------------------------------------------------- */

  template <typename TYPE>
  TYPE **grow(TYPE **&array, int n1, int n2, const char *name)
    {
      if (array == NULL) return create(array,n1,n2,name);

      bigint nbytes = sizeof(TYPE) * n1*n2;
      TYPE *data = (TYPE *) srealloc(array[0],nbytes,name);
      nbytes = sizeof(TYPE *) * n1;
      array = (TYPE **) srealloc(array,nbytes,name);

      int n = 0;
      for (int i = 0; i < n1; i++) {
        array[i] = &data[n];
        n += n2;
      }
      return array;
    }

  template <typename TYPE>
  TYPE ***grow(TYPE ***&, int, int, const char *name)
    {fail(name);}

/* ----------------------------------------------------------------------
   destroy a 2d array
   ------------------------------------------------------------------------- */

  template <typename TYPE>
  void destroy(TYPE **array)
    {
      if (array == NULL) return;
      sfree(array[0]);
      sfree(array);
    }

/* ----------------------------------------------------------------------
   create a 3d array
   ------------------------------------------------------------------------- */

  template <typename TYPE>
  TYPE ***create(TYPE ***&array, int n1, int n2, int n3, const char *name)
    {
      bigint nbytes = sizeof(TYPE) * n1*n2*n3;
      TYPE *data = (TYPE *) smalloc(nbytes,name);
      nbytes = sizeof(TYPE *) * n1*n2;
      TYPE **plane = (TYPE **) smalloc(nbytes,name);
      nbytes = sizeof(TYPE **) * n1;
      array = (TYPE ***) smalloc(nbytes,name);

      int i,j;
      int n = 0;
      for (i = 0; i < n1; i++) {
        array[i] = &plane[i*n2];
        for (j = 0; j < n2; j++) {
          plane[i*n2+j] = &data[n];
          n += n3;
        }
      }
      return array;
    }

  template <typename TYPE>
  TYPE ****create(TYPE ****&, int, int, int, const char *name)
    {fail(name);}

/* ----------------------------------------------------------------------
   grow or shrink 1st dim of a 3d array
   last 2 dims must stay the same
   ------------------------------------------------------------------------- */

  template <typename TYPE>
  TYPE ***grow(TYPE ***&array, int n1, int n2, int n3, const char *name)
    {
      if (array == NULL) return create(array,n1,n2,n3,name);

      bigint nbytes = sizeof(TYPE) * n1*n2*n3;
      TYPE *data = (TYPE *) srealloc(array[0][0],nbytes,name);
      nbytes = sizeof(TYPE *) * n1*n2;
      TYPE **plane = (TYPE **) srealloc(array[0],nbytes,name);
      nbytes = sizeof(TYPE **) * n1;
      array = (TYPE ***) srealloc(array,nbytes,name);

      int i,j;
      int n = 0;
      for (i = 0; i < n1; i++) {
        array[i] = &plane[i*n2];
        for (j = 0; j < n2; j++) {
          plane[i*n2+j] = &data[n];
          n += n3;
        }
      }
      return array;
    }

  template <typename TYPE>
  TYPE ****grow(TYPE ****&, int, int, int, const char *name)
    {fail(name);}

/* ----------------------------------------------------------------------
   destroy a 3d array
   ------------------------------------------------------------------------- */

  template <typename TYPE>
  void destroy(TYPE ***array)
    {
      if (array == NULL) return;
      sfree(array[0][0]);
      sfree(array[0]);
      sfree(array);
    }

/* ----------------------------------------------------------------------
   memory usage of arrays, including pointers
   ------------------------------------------------------------------------- */

  template <typename TYPE>
  bigint usage(TYPE *, int n)
    {
      bigint bytes = sizeof(TYPE) * n;
      return bytes;
    }

  template <typename TYPE>
  bigint usage(TYPE **, int n1, int n2)
    {
      bigint bytes = sizeof(TYPE) * n1*n2;
      bytes += sizeof(TYPE *) * n1;
      return bytes;
    }
};
#endif