PetscErrorCode PetscStackCreate(void)
{
PetscStack *petscstack_in;
PetscInt i;
if (PetscStackActive()) return 0;
petscstack_in = (PetscStack*)malloc(sizeof(PetscStack));
petscstack_in->currentsize = 0;
petscstack_in->hotdepth = 0;
for (i=0; i<PETSCSTACKSIZE; i++) {
petscstack_in->function[i] = 0;
petscstack_in->file[i] = 0;
}
PetscThreadLocalSetValue((PetscThreadKey*)&petscstack,petscstack_in);
#if defined(PETSC_HAVE_SAWS)
{
PetscBool flg = PETSC_FALSE;
PetscOptionsHasName(NULL,"-stack_view",&flg);
if (flg) PetscStackViewSAWs();
}
#endif
return 0;
}
PetscErrorCode PetscStackDestroy(void)
{
if (PetscStackActive()) {
PetscStack *petscstack_in;
petscstack_in = (PetscStack*)PetscThreadLocalGetValue(petscstack);
free(petscstack_in);
PetscThreadLocalSetValue((PetscThreadKey*)&petscstack,NULL);
}
return 0;
}
PetscErrorCode PetscStackCreate(void)
{
PetscStack *petscstack_in;
if (PetscStackActive()) return 0;
petscstack_in = (PetscStack*)malloc(sizeof(PetscStack));
petscstack_in->currentsize = 0;
PetscThreadLocalSetValue((PetscThreadKey*)&petscstack,petscstack_in);
#if defined(PETSC_HAVE_AMS)
{
PetscBool flg = PETSC_FALSE;
PetscOptionsHasName(NULL,"-stack_view",&flg);
if (flg) PetscStackViewAMS();
}
#endif
return 0;
}
/*
PetscTrMallocDefault - Malloc with tracing.
Input Parameters:
+ a - number of bytes to allocate
. lineno - line number where used. Use __LINE__ for this
. function - function calling routine. Use __FUNCT__ for this
- filename - file name where used. Use __FILE__ for this
Returns:
double aligned pointer to requested storage, or null if not
available.
*/
PetscErrorCode PetscTrMallocDefault(size_t a,int lineno,const char function[],const char filename[],void **result)
{
TRSPACE *head;
char *inew;
size_t nsize;
PetscErrorCode ierr;
PetscFunctionBegin;
if (TRdebugLevel) {
ierr = PetscMallocValidate(lineno,function,filename); if (ierr) PetscFunctionReturn(ierr);
}
nsize = (a + (PETSC_MEMALIGN-1)) & ~(PETSC_MEMALIGN-1);
ierr = PetscMallocAlign(nsize+sizeof(TrSPACE)+sizeof(PetscClassId),lineno,function,filename,(void**)&inew);CHKERRQ(ierr);
head = (TRSPACE*)inew;
inew += sizeof(TrSPACE);
if (TRhead) TRhead->prev = head;
head->next = TRhead;
TRhead = head;
head->prev = 0;
head->size = nsize;
head->id = TRid;
head->lineno = lineno;
head->filename = filename;
head->functionname = function;
head->classid = CLASSID_VALUE;
*(PetscClassId*)(inew + nsize) = CLASSID_VALUE;
TRallocated += nsize;
if (TRallocated > TRMaxMem) TRMaxMem = TRallocated;
TRfrags++;
#if defined(PETSC_USE_DEBUG)
if (PetscStackActive()) {
ierr = PetscStackCopy((PetscStack*)PetscThreadLocalGetValue(petscstack),&head->stack);CHKERRQ(ierr);
/* fix the line number to where the malloc() was called, not the PetscFunctionBegin; */
head->stack.line[head->stack.currentsize-2] = lineno;
}
#endif
/*
Allow logging of all mallocs made
*/
if (PetscLogMalloc > -1 && PetscLogMalloc < PetscLogMallocMax && a >= PetscLogMallocThreshold) {
if (!PetscLogMalloc) {
PetscLogMallocLength = (size_t*)malloc(PetscLogMallocMax*sizeof(size_t));
if (!PetscLogMallocLength) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_MEM," ");
PetscLogMallocFile = (const char**)malloc(PetscLogMallocMax*sizeof(char*));
if (!PetscLogMallocFile) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_MEM," ");
PetscLogMallocFunction = (const char**)malloc(PetscLogMallocMax*sizeof(char*));
if (!PetscLogMallocFunction) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_MEM," ");
}
PetscLogMallocLength[PetscLogMalloc] = nsize;
PetscLogMallocFile[PetscLogMalloc] = filename;
PetscLogMallocFunction[PetscLogMalloc++] = function;
}
*result = (void*)inew;
PetscFunctionReturn(0);
}
/*
PetscTrReallocDefault - Realloc with tracing.
Input Parameters:
+ len - number of bytes to allocate
. lineno - line number where used. Use __LINE__ for this
. filename - file name where used. Use __FILE__ for this
- result - double aligned pointer to initial storage.
Output Parameter:
. result - double aligned pointer to requested storage, or null if not available.
Level: developer
.seealso: PetscTrMallocDefault(), PetscTrFreeDefault()
*/
PetscErrorCode PetscTrReallocDefault(size_t len, int lineno, const char function[], const char filename[], void **result)
{
char *a = (char *) *result;
TRSPACE *head;
char *ahead, *inew;
PetscClassId *nend;
size_t nsize;
PetscErrorCode ierr;
PetscFunctionBegin;
/* Realloc to zero = free */
if (!len) {
ierr = PetscTrFreeDefault(*result,lineno,function,filename);CHKERRQ(ierr);
*result = NULL;
PetscFunctionReturn(0);
}
if (TRdebugLevel) {ierr = PetscMallocValidate(lineno,function,filename); if (ierr) PetscFunctionReturn(ierr);}
ahead = a;
a = a - sizeof(TrSPACE);
head = (TRSPACE *) a;
inew = a;
if (head->classid != CLASSID_VALUE) {
(*PetscErrorPrintf)("PetscTrReallocDefault() called from %s() line %d in %s\n",function,lineno,filename);
(*PetscErrorPrintf)("Block at address %p is corrupted; cannot free;\nmay be block not allocated with PetscMalloc()\n",a);
SETERRQ(PETSC_COMM_SELF,PETSC_ERR_MEMC,"Bad location or corrupted memory");
}
nend = (PetscClassId *)(ahead + head->size);
if (*nend != CLASSID_VALUE) {
if (*nend == ALREADY_FREED) {
(*PetscErrorPrintf)("PetscTrReallocDefault() called from %s() line %d in %s\n",function,lineno,filename);
(*PetscErrorPrintf)("Block [id=%d(%.0f)] at address %p was already freed\n",head->id,(PetscLogDouble)head->size,a + sizeof(TrSPACE));
if (head->lineno > 0 && head->lineno < 50000 /* sanity check */) {
(*PetscErrorPrintf)("Block freed in %s() line %d in %s\n",head->functionname,head->lineno,head->filename);
} else {
(*PetscErrorPrintf)("Block allocated in %s() line %d in %s\n",head->functionname,-head->lineno,head->filename);
}
SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONG,"Memory already freed");
} else {
/* Damaged tail */
(*PetscErrorPrintf)("PetscTrReallocDefault() called from %s() line %d in %s\n",function,lineno,filename);
(*PetscErrorPrintf)("Block [id=%d(%.0f)] at address %p is corrupted (probably write past end of array)\n",head->id,(PetscLogDouble)head->size,a);
(*PetscErrorPrintf)("Block allocated in %s() line %d in %s\n",head->functionname,head->lineno,head->filename);
SETERRQ(PETSC_COMM_SELF,PETSC_ERR_MEMC,"Corrupted memory");
}
}
TRallocated -= head->size;
TRfrags--;
if (head->prev) head->prev->next = head->next;
else TRhead = head->next;
if (head->next) head->next->prev = head->prev;
nsize = (len + (PETSC_MEMALIGN-1)) & ~(PETSC_MEMALIGN-1);
ierr = PetscReallocAlign(nsize+sizeof(TrSPACE)+sizeof(PetscClassId),lineno,function,filename,(void**)&inew);CHKERRQ(ierr);
head = (TRSPACE*)inew;
inew += sizeof(TrSPACE);
if (TRhead) TRhead->prev = head;
head->next = TRhead;
TRhead = head;
head->prev = NULL;
head->size = nsize;
head->id = TRid;
head->lineno = lineno;
head->filename = filename;
head->functionname = function;
head->classid = CLASSID_VALUE;
*(PetscClassId*)(inew + nsize) = CLASSID_VALUE;
TRallocated += nsize;
if (TRallocated > TRMaxMem) TRMaxMem = TRallocated;
TRfrags++;
#if defined(PETSC_USE_DEBUG)
if (PetscStackActive()) {
ierr = PetscStackCopy(petscstack,&head->stack);CHKERRQ(ierr);
/* fix the line number to where the malloc() was called, not the PetscFunctionBegin; */
head->stack.line[head->stack.currentsize-2] = lineno;
} else {
head->stack.currentsize = 0;
}
#endif
/*
Allow logging of all mallocs made
*/
if (PetscLogMalloc > -1 && PetscLogMalloc < PetscLogMallocMax && len >= PetscLogMallocThreshold) {
if (!PetscLogMalloc) {
PetscLogMallocLength = (size_t*)malloc(PetscLogMallocMax*sizeof(size_t));
if (!PetscLogMallocLength) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_MEM," ");
PetscLogMallocFile = (const char**)malloc(PetscLogMallocMax*sizeof(char*));
if (!PetscLogMallocFile) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_MEM," ");
PetscLogMallocFunction = (const char**)malloc(PetscLogMallocMax*sizeof(char*));
if (!PetscLogMallocFunction) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_MEM," ");
}
PetscLogMallocLength[PetscLogMalloc] = nsize;
PetscLogMallocFile[PetscLogMalloc] = filename;
PetscLogMallocFunction[PetscLogMalloc++] = function;
}
*result = (void*)inew;
PetscFunctionReturn(0);
}
