Actual source code: mpiaijsbaij.c

  1: #define PETSCMAT_DLL

 3:  #include mpisbaij.h
 4:  #include ../src/mat/impls/aij/mpi/mpiaij.h
 5:  #include private/matimpl.h
 6:  #include petscmat.h

 11: PetscErrorCode  MatConvert_MPIAIJ_MPISBAIJ(Mat A, MatType newtype,MatReuse reuse,Mat *newmat)
 12: {
 13:   PetscErrorCode     ierr;
 14:   Mat                M;
 15:   Mat_MPIAIJ         *mpimat = (Mat_MPIAIJ*)A->data;
 16:   Mat_SeqAIJ         *Aa = (Mat_SeqAIJ*)mpimat->A->data,*Ba = (Mat_SeqAIJ*)mpimat->B->data;
 17:   PetscInt           *d_nnz,*o_nnz;
 18:   PetscInt           i,j,nz;
 19:   PetscInt           m,n,lm,ln;
 20:   PetscInt           rstart,rend;
 21:   const PetscScalar  *vwork;
 22:   const PetscInt     *cwork;

 25:   MatGetSize(A,&m,&n);
 26:   MatGetLocalSize(A,&lm,&ln);
 27:   PetscMalloc2(lm*sizeof(PetscInt),PetscInt,&d_nnz,lm*sizeof(PetscInt),PetscInt,&o_nnz);

 29:   for(i=0;i<lm;i++){
 30:     d_nnz[i] = Aa->i[i+1] - Aa->diag[i];
 31:     o_nnz[i] = Ba->i[i+1] - Ba->i[i];
 32:   }

 34:   MatCreate(((PetscObject)A)->comm,&M);
 35:   MatSetSizes(M,lm,ln,m,n);
 36:   MatSetType(M,newtype);
 37:   MatMPISBAIJSetPreallocation(M,1,0,d_nnz,0,o_nnz);

 39:   PetscFree2(d_nnz,o_nnz);

 41:   MatGetOwnershipRange(A,&rstart,&rend);
 42:   for(i=rstart;i<rend;i++){
 43:     MatGetRow(A,i,&nz,&cwork,&vwork);
 44:     j = 0;
 45:     while (cwork[j] < i){ j++; nz--;}
 46:     MatSetValues(M,1,&i,nz,cwork+j,vwork+j,INSERT_VALUES);
 47:     MatRestoreRow(A,i,&nz,&cwork,&vwork);
 48:   }
 49:   MatAssemblyBegin(M,MAT_FINAL_ASSEMBLY);
 50:   MatAssemblyEnd(M,MAT_FINAL_ASSEMBLY);

 52:   if (A->hermitian){
 53:     MatSetOption(M,MAT_HERMITIAN,PETSC_TRUE);
 54:   }

 56:   if (reuse == MAT_REUSE_MATRIX) {
 57:     MatHeaderReplace(A,M);
 58:   } else {
 59:     *newmat = M;
 60:   }
 61:   return(0);
 62: }