AuxMemory.c

Go to the documentation of this file.

 
/*---------------------------------*/
/*-- Declare External Functions  --*/
/*---------------------------------*/
 
#include "fasp.h"
#include "fasp_functs.h"
 
#if DLMALLOC
#include "dlmalloc.h"
#elif NEDMALLOC
#include "nedmalloc.h"
#ifdef __cplusplus
extern "C" {
#endif
void* nedcalloc(size_t no, size_t size);
void* nedrealloc(void* mem, size_t size);
void  nedfree(void* mem);
#ifdef __cplusplus
}
#endif
#endif
 
#if DEBUG_MODE > 1
extern unsigned long total_alloc_mem;
extern unsigned long total_alloc_count;
#endif
 
/*---------------------------------*/
/*--      Global Variables       --*/
/*---------------------------------*/
 
const int Million = 1048576; 
/*---------------------------------*/
/*--      Public Functions       --*/
/*---------------------------------*/
 
void* fasp_mem_calloc(const unsigned int size, const unsigned int type)
{
    const LONGLONG tsize = size * type;
    void*          mem   = NULL;
 
#if DEBUG_MODE > 1
    printf("### DEBUG: Trying to allocate %.3lfMB RAM!\n", (REAL)tsize / Million);
#endif
 
    if (tsize > 0) {
 
#if DLMALLOC
        mem = dlcalloc(size, type);
#elif NEDMALLOC
        mem = nedcalloc(size, type);
#else
        mem = calloc(size, type);
#endif
 
#if DEBUG_MODE > 1
        total_alloc_mem += tsize;
        total_alloc_count++;
#endif
    }
 
    if (mem == NULL) {
        printf("### WARNING: Trying to allocate %lldB RAM...\n", tsize);
        printf("### WARNING: Cannot allocate %.4fMB RAM!\n", (REAL)tsize / Million);
    }
 
    return mem;
}
 
void* fasp_mem_realloc(void* oldmem, const LONGLONG tsize)
{
    void* mem = NULL;
 
#if DEBUG_MODE > 1
    printf("### DEBUG: Trying to allocate %.3lfMB RAM!\n", (REAL)tsize / Million);
#endif
 
    if (tsize > 0) {
 
#if DLMALLOC
        mem = dlrealloc(oldmem, tsize);
#elif NEDMALLOC
        mem = nedrealloc(oldmem, tsize);
#else
        mem = realloc(oldmem, tsize);
#endif
    }
 
    if (mem == NULL) {
        printf("### WARNING: Trying to allocate %lldB RAM!\n", tsize);
        printf("### WARNING: Cannot allocate %.3lfMB RAM!\n", (REAL)tsize / Million);
    }
 
    return mem;
}
 
void fasp_mem_free(void* mem)
{
    if (mem) {
#if DLMALLOC
        dlfree(mem);
#elif NEDMALLOC
        nedfree(mem);
#else
        free(mem);
#endif
 
        mem = NULL;
 
#if DEBUG_MODE > 1
        total_alloc_count--;
#endif
    } else {
#if DEBUG_MODE > 1
        printf("### WARNING: Trying to free an empty pointer!\n");
#endif
    }
}
 
void fasp_mem_usage(void)
{
#if DEBUG_MODE > 1
    printf("### DEBUG: Number of alloc = %ld, allocated memory = %.3fMB.\n",
           total_alloc_count, (REAL)total_alloc_mem / Million);
#endif
}
 
SHORT fasp_mem_iludata_check(const ILU_data* iludata)
{
    const INT memneed = 2 * iludata->row; // estimated memory usage
 
    if (iludata->nwork >= memneed) {
        return FASP_SUCCESS;
    } else {
        printf("### ERROR: ILU needs %d RAM, only %d available!\n", memneed,
               iludata->nwork);
        return ERROR_ALLOC_MEM;
    }
}
 
/*---------------------------------*/
/*--        End of File          --*/
/*---------------------------------*/