/*
 *  Copyright (C) 2004-2021 Edward F. Valeev
 *
 *  This file is part of Libint.
 *
 *  Libint is free software: you can redistribute it and/or modify
 *  it under the terms of the GNU Lesser General Public License as published by
 *  the Free Software Foundation, either version 3 of the License, or
 *  (at your option) any later version.
 *
 *  Libint is distributed in the hope that it will be useful,
 *  but WITHOUT ANY WARRANTY; without even the implied warranty of
 *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 *  GNU Lesser General Public License for more details.
 *
 *  You should have received a copy of the GNU Lesser General Public License
 *  along with Libint.  If not, see <http://www.gnu.org/licenses/>.
 *
 */

#ifndef _libint2_src_lib_libint_libint2memory_h_
#define _libint2_src_lib_libint_libint2memory_h_

#include <cstdlib>
#include <libint2/util/generated/libint2_params.h>

#ifdef _MSC_VER
#define posix_memalign(p, a, s) (((*(p)) = _aligned_malloc((s), (a))), *(p) ?0 :errno)
#endif

namespace libint2 {

  /// Aligned version of malloc().

  /** Allocates a memory block aligned to LIBINT2_ALIGN_SIZE*sizeof(LIBINT2_REALTYPE).
      If LIBINT2_ALIGN_SIZE, no alignment is assumed. Use free() to deallocate. */
  inline void* malloc(size_t nbytes) {
    void* result;
#if (LIBINT2_ALIGN_SIZE == 0)
    result = ::malloc(nbytes);
#elif defined(HAVE_POSIX_MEMALIGN)
    posix_memalign(&result, LIBINT2_ALIGN_SIZE*sizeof(LIBINT2_REALTYPE), nbytes);
#else
#   error "LIBINT2_ALIGN_SIZE!=0 but posix_memalign is not available"
#endif
    return result;
  }

  /// type-specific version of libint2::malloc()
  template <typename T>
  inline T* malloc(size_t n) {
    return reinterpret_cast<T*>(malloc(n * sizeof(T)));
  }

}

#endif /* _libint2_src_lib_libint_libint2memory_h_ */
