1 2 #include <private/matimpl.h> 3 4 #undef __FUNCT__ 5 #define __FUNCT__ "MatConvert_Basic" 6 /* 7 MatConvert_Basic - Converts from any input format to another format. For 8 parallel formats, the new matrix distribution is determined by PETSc. 9 10 Does not do preallocation so in general will be slow 11 */ 12 PetscErrorCode MatConvert_Basic(Mat mat, const MatType newtype,MatReuse reuse,Mat *newmat) 13 { 14 Mat M; 15 const PetscScalar *vwork; 16 PetscErrorCode ierr; 17 PetscInt i,nz,m,n,rstart,rend,lm,ln; 18 const PetscInt *cwork; 19 PetscBool isseqsbaij,ismpisbaij; 20 21 PetscFunctionBegin; 22 ierr = MatGetSize(mat,&m,&n);CHKERRQ(ierr); 23 ierr = MatGetLocalSize(mat,&lm,&ln);CHKERRQ(ierr); 24 25 if (ln == n) ln = PETSC_DECIDE; /* try to preserve column ownership */ 26 27 ierr = MatCreate(((PetscObject)mat)->comm,&M);CHKERRQ(ierr); 28 ierr = MatSetSizes(M,lm,ln,m,n);CHKERRQ(ierr); 29 ierr = MatSetType(M,newtype);CHKERRQ(ierr); 30 ierr = PetscTypeCompare((PetscObject)M,MATSEQSBAIJ,&isseqsbaij);CHKERRQ(ierr); 31 ierr = PetscTypeCompare((PetscObject)M,MATMPISBAIJ,&ismpisbaij);CHKERRQ(ierr); 32 if (isseqsbaij || ismpisbaij) {ierr = MatSetOption(M,MAT_IGNORE_LOWER_TRIANGULAR,PETSC_TRUE);CHKERRQ(ierr);} 33 34 ierr = MatGetOwnershipRange(mat,&rstart,&rend);CHKERRQ(ierr); 35 for (i=rstart; i<rend; i++) { 36 ierr = MatGetRow(mat,i,&nz,&cwork,&vwork);CHKERRQ(ierr); 37 ierr = MatSetValues(M,1,&i,nz,cwork,vwork,INSERT_VALUES);CHKERRQ(ierr); 38 ierr = MatRestoreRow(mat,i,&nz,&cwork,&vwork);CHKERRQ(ierr); 39 } 40 ierr = MatAssemblyBegin(M,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 41 ierr = MatAssemblyEnd(M,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 42 if (mat->hermitian){ 43 ierr = MatSetOption(M,MAT_HERMITIAN,PETSC_TRUE);CHKERRQ(ierr); 44 } 45 46 if (reuse == MAT_REUSE_MATRIX) { 47 ierr = MatHeaderReplace(mat,M);CHKERRQ(ierr); 48 } else { 49 *newmat = M; 50 } 51 PetscFunctionReturn(0); 52 } 53