1be1d678aSKris Buschelman #define PETSCMAT_DLL 2be1d678aSKris Buschelman 3*7c4f633dSBarry Smith #include "../src/mat/utils/freespace.h" 470f19b1fSKris Buschelman 570f19b1fSKris Buschelman #undef __FUNCT__ 6a1a86e44SBarry Smith #define __FUNCT__ "PetscFreeSpaceGet" 7a1a86e44SBarry Smith PetscErrorCode PetscFreeSpaceGet(PetscInt n,PetscFreeSpaceList *list) 82e111b49SBarry Smith { 9a1a86e44SBarry Smith PetscFreeSpaceList a; 10dfbe8321SBarry Smith PetscErrorCode ierr; 1170f19b1fSKris Buschelman 1270f19b1fSKris Buschelman PetscFunctionBegin; 13a1a86e44SBarry Smith ierr = PetscMalloc(sizeof(struct _Space),&a);CHKERRQ(ierr); 142e111b49SBarry Smith ierr = PetscMalloc(n*sizeof(PetscInt),&(a->array_head));CHKERRQ(ierr); 1570f19b1fSKris Buschelman a->array = a->array_head; 162e111b49SBarry Smith a->local_remaining = n; 1770f19b1fSKris Buschelman a->local_used = 0; 1870f19b1fSKris Buschelman a->total_array_size = 0; 1970f19b1fSKris Buschelman a->more_space = NULL; 2070f19b1fSKris Buschelman 2170f19b1fSKris Buschelman if (*list) { 2270f19b1fSKris Buschelman (*list)->more_space = a; 2370f19b1fSKris Buschelman a->total_array_size = (*list)->total_array_size; 2470f19b1fSKris Buschelman } 2570f19b1fSKris Buschelman 262e111b49SBarry Smith a->total_array_size += n; 2770f19b1fSKris Buschelman *list = a; 2870f19b1fSKris Buschelman PetscFunctionReturn(0); 2970f19b1fSKris Buschelman } 3070f19b1fSKris Buschelman 3170f19b1fSKris Buschelman #undef __FUNCT__ 32a1a86e44SBarry Smith #define __FUNCT__ "PetscFreeSpaceContiguous" 33a1a86e44SBarry Smith PetscErrorCode PetscFreeSpaceContiguous(PetscFreeSpaceList *head,PetscInt *space) 3438baddfdSBarry Smith { 35a1a86e44SBarry Smith PetscFreeSpaceList a; 36dfbe8321SBarry Smith PetscErrorCode ierr; 3770f19b1fSKris Buschelman 3870f19b1fSKris Buschelman PetscFunctionBegin; 3970f19b1fSKris Buschelman while ((*head)!=NULL) { 4070f19b1fSKris Buschelman a = (*head)->more_space; 412e111b49SBarry Smith ierr = PetscMemcpy(space,(*head)->array_head,((*head)->local_used)*sizeof(PetscInt));CHKERRQ(ierr); 4270f19b1fSKris Buschelman space += (*head)->local_used; 4370f19b1fSKris Buschelman ierr = PetscFree((*head)->array_head);CHKERRQ(ierr); 4470f19b1fSKris Buschelman ierr = PetscFree(*head);CHKERRQ(ierr); 4570f19b1fSKris Buschelman *head = a; 4670f19b1fSKris Buschelman } 4770f19b1fSKris Buschelman PetscFunctionReturn(0); 4870f19b1fSKris Buschelman } 497a48dd6fSHong Zhang 507a48dd6fSHong Zhang #undef __FUNCT__ 51a1a86e44SBarry Smith #define __FUNCT__ "PetscFreeSpaceDestroy" 52a1a86e44SBarry Smith PetscErrorCode PetscFreeSpaceDestroy(PetscFreeSpaceList head) 537a48dd6fSHong Zhang { 54a1a86e44SBarry Smith PetscFreeSpaceList a; 557a48dd6fSHong Zhang PetscErrorCode ierr; 567a48dd6fSHong Zhang 577a48dd6fSHong Zhang PetscFunctionBegin; 587a48dd6fSHong Zhang while ((head)!=NULL) { 597a48dd6fSHong Zhang a = (head)->more_space; 607a48dd6fSHong Zhang ierr = PetscFree((head)->array_head);CHKERRQ(ierr); 617a48dd6fSHong Zhang ierr = PetscFree(head);CHKERRQ(ierr); 627a48dd6fSHong Zhang head = a; 637a48dd6fSHong Zhang } 647a48dd6fSHong Zhang PetscFunctionReturn(0); 657a48dd6fSHong Zhang } 66