Date: 2022-03-04T17:38:18+00:00

Actual source code: matstashspace.c

      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: }