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