1*b9b97703SBarry Smith /*$Id: aijbaij.c,v 1.1 2000/06/06 19:00:44 bsmith Exp bsmith $*/ 261e22dc2SBarry Smith 361e22dc2SBarry Smith #include "src/mat/impls/baij/seq/baij.h" 461e22dc2SBarry Smith 561e22dc2SBarry Smith #undef __FUNC__ 6*b9b97703SBarry Smith #define __FUNC__ /*<a name="MatConvert_SeqBAIJ_SeqAIJ"></a>*/"MatConvert_SeqBAI_SeqAIJ" 7*b9b97703SBarry Smith int MatConvert_SeqBAIJ_SeqAIJ(Mat A,MatType newtype,Mat *B) 861e22dc2SBarry Smith { 961e22dc2SBarry Smith Mat_SeqBAIJ *a = (Mat_SeqBAIJ*)A->data; 10*b9b97703SBarry Smith int ierr,bs = a->bs,*ai = a->i,*aj = a->j,n = A->M/bs,i,j,k; 11*b9b97703SBarry Smith int *rowlengths,*rows,*cols,maxlen = 0,ncols; 12*b9b97703SBarry Smith Scalar *aa = a->a; 1361e22dc2SBarry Smith 1461e22dc2SBarry Smith PetscFunctionBegin; 15*b9b97703SBarry Smith rowlengths = (int*)PetscMalloc(n*bs*sizeof(int));CHKPTRQ(rowlengths); 1661e22dc2SBarry Smith for (i=0; i<n; i++) { 17*b9b97703SBarry Smith maxlen = PetscMax(maxlen,(ai[i+1] - ai[i])); 1861e22dc2SBarry Smith for (j=0; j<bs; j++) { 19*b9b97703SBarry Smith rowlengths[i*bs+j] = bs*(ai[i+1] - ai[i]); 2061e22dc2SBarry Smith } 2161e22dc2SBarry Smith } 22*b9b97703SBarry Smith ierr = MatCreateSeqAIJ(PETSC_COMM_SELF,A->m,A->n,0,rowlengths,B);CHKERRQ(ierr); 23*b9b97703SBarry Smith ierr = MatSetOption(*B,MAT_COLUMN_ORIENTED);CHKERRQ(ierr); 24*b9b97703SBarry Smith ierr = MatSetOption(*B,MAT_ROWS_SORTED);CHKERRQ(ierr); 25*b9b97703SBarry Smith ierr = MatSetOption(*B,MAT_COLUMNS_SORTED);CHKERRQ(ierr); 2661e22dc2SBarry Smith ierr = PetscFree(rowlengths);CHKERRQ(ierr); 2761e22dc2SBarry Smith 28*b9b97703SBarry Smith rows = (int*)PetscMalloc(bs*sizeof(int));CHKPTRQ(rows); 29*b9b97703SBarry Smith cols = (int*)PetscMalloc(bs*maxlen*sizeof(int));CHKPTRQ(cols); 30*b9b97703SBarry Smith for (i=0; i<n; i++) { 31*b9b97703SBarry Smith for (j=0; j<bs; j++) { 32*b9b97703SBarry Smith rows[j] = i*bs+j; 33*b9b97703SBarry Smith } 34*b9b97703SBarry Smith ncols = ai[i+1] - ai[i]; 35*b9b97703SBarry Smith for (k=0; k<ncols; k++) { 36*b9b97703SBarry Smith for (j=0; j<bs; j++) { 37*b9b97703SBarry Smith cols[k*bs+j] = bs*(*aj) + j; 38*b9b97703SBarry Smith } 39*b9b97703SBarry Smith aj++; 40*b9b97703SBarry Smith } 41*b9b97703SBarry Smith ierr = MatSetValues(*B,bs,rows,bs*ncols,cols,aa,INSERT_VALUES); 42*b9b97703SBarry Smith aa += ncols*bs*bs; 43*b9b97703SBarry Smith } 44*b9b97703SBarry Smith ierr = PetscFree(cols);CHKERRQ(ierr); 45*b9b97703SBarry Smith ierr = PetscFree(rows);CHKERRQ(ierr); 46*b9b97703SBarry Smith ierr = MatAssemblyBegin(*B,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 47*b9b97703SBarry Smith ierr = MatAssemblyEnd(*B,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 4861e22dc2SBarry Smith PetscFunctionReturn(0); 4961e22dc2SBarry Smith } 5061e22dc2SBarry Smith 5161e22dc2SBarry Smith 52