1 #ifndef lint 2 static char vcid[] = "$Id: convert.c,v 1.1 1995/03/23 15:16:46 curfman Exp curfman $"; 3 #endif 4 5 /* Matrix conversion routines. For now, this supports only AIJ */ 6 7 #include "mpiaij.h" 8 9 /* 10 MatConvert_AIJ - Converts from MATAIJ format to another sequential format. 11 */ 12 int MatConvert_AIJ(Mat mat, MATTYPE newtype, Mat *newmat) 13 { 14 Mat_AIJ *aij = (Mat_AIJ *) mat->data; 15 Scalar *vwork; 16 int i, ierr, nz, m = aij->m, n = aij->n, *cwork; 17 18 if (mat->type != MATAIJ) SETERR(1,"Input matrix must be MATAIJ."); 19 switch (newtype) { 20 case MATROW: 21 ierr = MatCreateSequentialRow(m,n,0,aij->ilen,newmat); 22 CHKERR(ierr); break; 23 case MATDENSE: 24 ierr = MatCreateSequentialDense(m,n,newmat); 25 CHKERR(ierr); break; 26 default: 27 SETERR(1,"Only MATROW and MATDENSE are currently suported."); 28 } 29 for (i=0; i<m; i++) { 30 ierr = MatGetRow(mat,i,&nz,&cwork,&vwork); CHKERR(ierr); 31 ierr = MatSetValues(*newmat,1,&i,nz,cwork,vwork,InsertValues); CHKERR(ierr); 32 ierr = MatRestoreRow(mat,i,&nz,&cwork,&vwork); CHKERR(ierr); 33 } 34 ierr = MatBeginAssembly(*newmat); CHKERR(ierr); 35 ierr = MatEndAssembly(*newmat); CHKERR(ierr); 36 return 0; 37 } 38 /* ------------------------------------------------------------------ */ 39 /* 40 MatConvert_MPIAIJ - Converts from MATMPIAIJ format to another 41 parallel format. 42 */ 43 int MatConvert_MPIAIJ(Mat mat, MATTYPE newtype, Mat *newmat) 44 { 45 Mat_MPIAIJ *aij = (Mat_MPIAIJ *) mat->data; 46 Mat A = aij->A, B = aij->B, OldM; 47 Mat_AIJ *Ad = (Mat_AIJ *)(A->data), *Bd = (Mat_AIJ *)(B->data); 48 Scalar *vwork; 49 int ierr, m = aij->m, n = aij->n, M = aij->M, N = aij->N, *cwork; 50 int nz,i,j,ig,submat, rstart = aij->rstart, cstart = aij->cstart; 51 52 if (mat->type != MATMPIAIJ) SETERR(1,"Input matrix must be MATMPIAIJ."); 53 switch (newtype) { 54 case MATMPIROW: 55 for (i=0; i<m; i++) 56 printf("row=%d, A-nz=%d, B-nz=%d\n",i,Ad->ilen[i],Bd->ilen[i]); 57 ierr = MatCreateMPIRow(mat->comm,m,n,M,N,0,Ad->ilen,0,Bd->ilen,newmat); 58 CHKERR(ierr); break; 59 default: 60 SETERR(1,"Only MATMPIROW is currently suported."); 61 } 62 for (submat = 0; submat<2; submat++) { 63 if (submat == 0) OldM = A; 64 else OldM = B; 65 for (i=0; i<m; i++) { 66 ig = i + rstart; 67 ierr = MatGetRow(OldM,i,&nz,&cwork,&vwork); CHKERR(ierr); 68 if (submat == 0) { 69 for (j=0; j<nz; j++) cwork[j] += cstart; 70 } else { 71 /* This part is incorrect! */ 72 for (j=0; j<nz; j++) cwork[j] += cstart; 73 } 74 ierr = MatSetValues(*newmat,1,&ig,nz,cwork,vwork, 75 InsertValues); CHKERR(ierr); 76 ierr = MatRestoreRow(OldM,ig,&nz,&cwork,&vwork); CHKERR(ierr); 77 } 78 } 79 ierr = MatBeginAssembly(*newmat); CHKERR(ierr); 80 ierr = MatEndAssembly(*newmat); CHKERR(ierr); 81 return 0; 82 } 83