1 /*$Id: aijsbaij.c,v 1.9 2001/08/07 03:02:55 balay Exp $*/ 2 3 #include "src/mat/impls/aij/seq/aij.h" 4 5 EXTERN_C_BEGIN 6 #undef __FUNCT__ 7 #define __FUNCT__ "MatConvert_SeqAIJ_SeqSBAIJ" 8 int MatConvert_SeqAIJ_SeqSBAIJ(Mat A,MatType newtype,Mat *B) 9 { 10 Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data; 11 int ierr,*ai=a->i,*aj,m=A->M,n=A->N,i; 12 int *rowlengths; 13 PetscScalar *av; 14 15 PetscFunctionBegin; 16 if (n != m) SETERRQ(PETSC_ERR_SUP,"Matrix must be a square matrix"); 17 if (!a->diag){ 18 ierr = MatMarkDiagonal_SeqAIJ(A);CHKERRQ(ierr); 19 } 20 21 ierr = PetscMalloc(m*sizeof(int),&rowlengths);CHKERRQ(ierr); 22 for (i=0; i<m; i++) { 23 rowlengths[i] = ai[i+1] - a->diag[i]; 24 } 25 ierr = MatCreateSeqSBAIJ(PETSC_COMM_SELF,1,m,n,0,rowlengths,B);CHKERRQ(ierr); 26 27 ierr = MatSetOption(*B,MAT_ROW_ORIENTED);CHKERRQ(ierr); 28 ierr = MatSetOption(*B,MAT_ROWS_SORTED);CHKERRQ(ierr); 29 ierr = MatSetOption(*B,MAT_COLUMNS_SORTED);CHKERRQ(ierr); 30 31 for (i=0; i<m; i++) { 32 aj = a->j + a->diag[i]; 33 av = a->a + a->diag[i]; 34 ierr = MatSetValues(*B,1,&i,rowlengths[i],aj,av,INSERT_VALUES);CHKERRQ(ierr); 35 } 36 37 ierr = PetscFree(rowlengths);CHKERRQ(ierr); 38 ierr = MatAssemblyBegin(*B,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 39 ierr = MatAssemblyEnd(*B,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 40 PetscFunctionReturn(0); 41 } 42 EXTERN_C_END 43 44