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