1 #ifndef lint 2 static char vcid[] = "$Id: convert.c,v 1.5 1995/04/05 21:52:03 curfman Exp bsmith $"; 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(mat->comm,m,n,0,aij->ilen,newmat); 22 CHKERR(ierr); break; 23 case MATDENSE: 24 ierr = MatCreateSequentialDense(mat->comm,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); 32 CHKERR(ierr); 33 ierr = MatRestoreRow(mat,i,&nz,&cwork,&vwork); CHKERR(ierr); 34 } 35 ierr = MatBeginAssembly(*newmat,FINAL_ASSEMBLY); CHKERR(ierr); 36 ierr = MatEndAssembly(*newmat,FINAL_ASSEMBLY); CHKERR(ierr); 37 return 0; 38 } 39 /* ------------------------------------------------------------------ */ 40 /* 41 MatConvert_MPIAIJ - Converts from MATMPIAIJ format to another 42 parallel format. 43 */ 44 int MatConvert_MPIAIJ(Mat mat, MATTYPE newtype, Mat *newmat) 45 { 46 Mat_MPIAIJ *aij = (Mat_MPIAIJ *) mat->data; 47 Mat_AIJ *Ad = (Mat_AIJ *)(aij->A->data), *Bd = (Mat_AIJ *)(aij->B->data); 48 int ierr, nz, i, ig,rstart = aij->rstart, m = aij->m, *cwork; 49 Scalar *vwork; 50 51 if (mat->type != MATMPIAIJ) SETERR(1,"Input matrix must be MATMPIAIJ."); 52 switch (newtype) { 53 case MATMPIROW: 54 for (i=0; i<m; i++) 55 {ierr = MatCreateMPIRow(mat->comm,m,aij->n,aij->M,aij->N,0,Ad->ilen, 56 0,Bd->ilen,newmat); CHKERR(ierr); } 57 break; 58 default: 59 SETERR(1,"Only MATMPIROW is currently suported."); 60 } 61 /* Each processor converts its local rows */ 62 for (i=0; i<m; i++) { 63 ig = i + rstart; 64 ierr = MatGetRow(mat,ig,&nz,&cwork,&vwork); CHKERR(ierr); 65 ierr = MatSetValues(*newmat,1,&ig,nz,cwork,vwork, 66 InsertValues); CHKERR(ierr); 67 ierr = MatRestoreRow(mat,ig,&nz,&cwork,&vwork); CHKERR(ierr); 68 } 69 ierr = MatBeginAssembly(*newmat,FINAL_ASSEMBLY); CHKERR(ierr); 70 ierr = MatEndAssembly(*newmat,FINAL_ASSEMBLY); CHKERR(ierr); 71 return 0; 72 } 73