Actual source code: matstashspace.c
petsc-3.13.6 2020-09-29
2: #include <petsc/private/matimpl.h>
4: /* Get new PetscMatStashSpace into the existing space */
5: PetscErrorCode PetscMatStashSpaceGet(PetscInt bs2,PetscInt n,PetscMatStashSpace *space)
6: {
7: PetscMatStashSpace a;
8: PetscErrorCode ierr;
11: if (!n) return(0);
13: PetscMalloc(sizeof(struct _MatStashSpace),&a);
14: PetscMalloc3(n*bs2,&(a->space_head),n,&a->idx,n,&a->idy);
16: a->val = a->space_head;
17: a->local_remaining = n;
18: a->local_used = 0;
19: a->total_space_size = 0;
20: a->next = NULL;
22: if (*space) {
23: (*space)->next = a;
24: a->total_space_size = (*space)->total_space_size;
25: }
26: a->total_space_size += n;
27: *space = a;
28: return(0);
29: }
31: /* Copy the values in space into arrays val, idx and idy. Then destroy space */
32: PetscErrorCode PetscMatStashSpaceContiguous(PetscInt bs2,PetscMatStashSpace *space,PetscScalar *val,PetscInt *idx,PetscInt *idy)
33: {
34: PetscMatStashSpace a;
35: PetscErrorCode ierr;
38: while ((*space)) {
39: a = (*space)->next;
40: PetscArraycpy(val,(*space)->val,(*space)->local_used*bs2);
41: val += bs2*(*space)->local_used;
42: PetscArraycpy(idx,(*space)->idx,(*space)->local_used);
43: idx += (*space)->local_used;
44: PetscArraycpy(idy,(*space)->idy,(*space)->local_used);
45: idy += (*space)->local_used;
47: PetscFree3((*space)->space_head,(*space)->idx,(*space)->idy);
48: PetscFree(*space);
49: *space = a;
50: }
51: return(0);
52: }
54: PetscErrorCode PetscMatStashSpaceDestroy(PetscMatStashSpace *space)
55: {
56: PetscMatStashSpace a;
57: PetscErrorCode ierr;
60: while (*space) {
61: a = (*space)->next;
62: PetscFree3((*space)->space_head,(*space)->idx,(*space)->idy);
63: PetscFree((*space));
64: *space = a;
65: }
66: *space = NULL;
67: return(0);
68: }