148b35521SBarry Smith 2cb512458SBarry Smith #ifndef lint 3*dae03382SLois Curfman McInnes static char vcid[] = "$Id: gcreate.c,v 1.30 1995/07/06 17:20:03 bsmith Exp curfman $"; 4cb512458SBarry Smith #endif 57807a1faSBarry Smith 67807a1faSBarry Smith #include "sys.h" 748b35521SBarry Smith #include "mat.h" /*I "mat.h" I*/ 87807a1faSBarry Smith 948b35521SBarry Smith /*@ 106469c4f9SBarry Smith MatCreate - Creates a matrix, where the type is determined 1102a82ca1SLois Curfman McInnes from the options database. Generates a parallel MPI matrix if the 1202a82ca1SLois Curfman McInnes communicator has more than one processor. 137807a1faSBarry Smith 147807a1faSBarry Smith Input Parameters: 15e0b365e2SLois Curfman McInnes . m - number of global rows 16e0b365e2SLois Curfman McInnes . n - number of global columns 176b5873e3SBarry Smith . comm - MPI communicator 187807a1faSBarry Smith 197807a1faSBarry Smith Output Parameter: 20df998da4SLois Curfman McInnes . V - location to stash resulting matrix 21e0b365e2SLois Curfman McInnes 22e0b365e2SLois Curfman McInnes Options Database Keywords: 23a8bb4b34SLois Curfman McInnes $ -mat_aij : AIJ type, uses MatCreateSequentialAIJ() 24a8bb4b34SLois Curfman McInnes $ and MatCreateMPIAIJ() 2527b4b7ebSLois Curfman McInnes $ -mat_dense : dense type, uses MatCreateSequentialDense() 2627b4b7ebSLois Curfman McInnes $ -mat_row : row type, uses MatCreateSequentialRow() 27e0b365e2SLois Curfman McInnes $ and MatCreateMPIRow() 28a8bb4b34SLois Curfman McInnes $ -mat_rowbs : rowbs type (for parallel symmetric matrices only), 29e0b365e2SLois Curfman McInnes $ uses MatCreateMPIRowbs() 3027b4b7ebSLois Curfman McInnes $ -mat_bdiag : block diagonal type, uses 31e0b365e2SLois Curfman McInnes $ MatCreateSequentialBDiag() and 32e0b365e2SLois Curfman McInnes $ MatCreateMPIBDiag() 33e0b365e2SLois Curfman McInnes $ 34a8bb4b34SLois Curfman McInnes $ -mpi_objects : uses MPI matrix (parallel format), even for one processor 35e0b365e2SLois Curfman McInnes 36e0b365e2SLois Curfman McInnes Notes: 37e0b365e2SLois Curfman McInnes The default matrix type is AIJ, using MatCreateSequentialAIJ() and 38e0b365e2SLois Curfman McInnes MatCreateMPIAIJ(). 39e0b365e2SLois Curfman McInnes 40e0b365e2SLois Curfman McInnes .keywords: matrix, create, initial 41e0b365e2SLois Curfman McInnes 42e0b365e2SLois Curfman McInnes .seealso: MatCreateSequentialAIJ((), MatCreateMPIAIJ(), 43e0b365e2SLois Curfman McInnes MatCreateSequentialRow(), MatCreateMPIRow(), 440c28153cSLois Curfman McInnes MatCreateSequentialBDiag(),MatCreateMPIBDiag(), 450c28153cSLois Curfman McInnes MatCreateSequentialDense(), MatCreateMPIRowbs(), MatConvert() 467807a1faSBarry Smith @*/ 476469c4f9SBarry Smith int MatCreate(MPI_Comm comm,int m,int n,Mat *V) 487807a1faSBarry Smith { 49d6dfbf8fSBarry Smith int numtid; 506b5873e3SBarry Smith MPI_Comm_size(comm,&numtid); 517f813858SBarry Smith if (OptionsHasName(0,"-help")) { 527f813858SBarry Smith MPIU_printf(comm,"MatCreate() options: -mat_dense, -mat_row -mat_rowbs\n"); 537f813858SBarry Smith MPIU_printf(comm," -mat_bdiag, -mpi_objects\n"); 547f813858SBarry Smith } 55df60cc22SBarry Smith if (OptionsHasName(0,"-mat_dense")) { 566b5873e3SBarry Smith return MatCreateSequentialDense(comm,m,n,V); 577807a1faSBarry Smith } 58e452cd2fSLois Curfman McInnes if (OptionsHasName(0,"-mat_bdiag")) { 59e452cd2fSLois Curfman McInnes int nb = 1, ndiag = 0, ndiag2, *d, ierr; 60e452cd2fSLois Curfman McInnes OptionsGetInt(0,"-mat_bdiag_bsize",&nb); 61e452cd2fSLois Curfman McInnes OptionsGetInt(0,"-mat_bdiag_ndiag",&ndiag); 6278b31e54SBarry Smith if (!ndiag) SETERRQ(1,"Must set diagonals before creating matrix."); 6378b31e54SBarry Smith d = (int *)PETSCMALLOC( ndiag * sizeof(int) ); CHKPTRQ(d); 64e452cd2fSLois Curfman McInnes ndiag2 = ndiag; 65e452cd2fSLois Curfman McInnes OptionsGetIntArray(0,"-mat_bdiag_dvals",d,&ndiag2); 66e452cd2fSLois Curfman McInnes if (ndiag2 != ndiag) { 6778b31e54SBarry Smith SETERRQ(1,"Incompatible number of diagonals and diagonal values."); 68e452cd2fSLois Curfman McInnes } 69e452cd2fSLois Curfman McInnes if (numtid > 1 || OptionsHasName(0,"-mpi_objects")) 70e452cd2fSLois Curfman McInnes ierr = MatCreateMPIBDiag(comm,PETSC_DECIDE,m,n,ndiag,nb,d,0,V); 71e452cd2fSLois Curfman McInnes else 72e452cd2fSLois Curfman McInnes ierr = MatCreateSequentialBDiag(comm,m,n,ndiag,nb,d,0,V); 7378b31e54SBarry Smith CHKERRQ(ierr); 7478b31e54SBarry Smith if (d) PETSCFREE(d); 75e452cd2fSLois Curfman McInnes return ierr; 76e452cd2fSLois Curfman McInnes } 779b779142SLois Curfman McInnes #if defined(HAVE_BLOCKSOLVE) && !defined(__cplusplus) 789b779142SLois Curfman McInnes if (OptionsHasName(0,"-mat_rowbs")) { 799b779142SLois Curfman McInnes return MatCreateMPIRowbs(comm,PETSC_DECIDE,m,5,0,0,V); 809b779142SLois Curfman McInnes } 819b779142SLois Curfman McInnes #endif 82df60cc22SBarry Smith if (numtid > 1 || OptionsHasName(0,"-mpi_objects")) { 83df60cc22SBarry Smith if (OptionsHasName(0,"-mat_row")) { 846b5873e3SBarry Smith return MatCreateMPIRow(comm,PETSC_DECIDE,PETSC_DECIDE,m,n,5,0,0,0,V); 85df998da4SLois Curfman McInnes } 860b5753b4SLois Curfman McInnes return MatCreateMPIAIJ(comm,PETSC_DECIDE,PETSC_DECIDE,m,n,5,0,0,0,V); 870b5753b4SLois Curfman McInnes } 88df60cc22SBarry Smith if (OptionsHasName(0,"-mat_row")) { 893dfeaf31SLois Curfman McInnes return MatCreateSequentialRow(comm,m,n,10,0,V); 903dfeaf31SLois Curfman McInnes } 916b5873e3SBarry Smith return MatCreateSequentialAIJ(comm,m,n,10,0,V); 927807a1faSBarry Smith } 937807a1faSBarry Smith 94*dae03382SLois Curfman McInnes #include "matimpl.h" 95*dae03382SLois Curfman McInnes /*@C 96*dae03382SLois Curfman McInnes MatGetName - Gets the matrix type name (as a string) from the matrix. 97*dae03382SLois Curfman McInnes 98*dae03382SLois Curfman McInnes Input Parameter: 99*dae03382SLois Curfman McInnes . mat - the matrix 100*dae03382SLois Curfman McInnes 101*dae03382SLois Curfman McInnes Output Parameter: 102*dae03382SLois Curfman McInnes . name - name of matrix type 103*dae03382SLois Curfman McInnes 104*dae03382SLois Curfman McInnes .keywords: matrix, get, name 105*dae03382SLois Curfman McInnes 106*dae03382SLois Curfman McInnes .seealso: MatGetType() 107*dae03382SLois Curfman McInnes @*/ 108*dae03382SLois Curfman McInnes int MatGetName(Mat mat,char **name) 109*dae03382SLois Curfman McInnes { 110*dae03382SLois Curfman McInnes /* Note: Be sure that this list corresponds to the enum in mat.h */ 111*dae03382SLois Curfman McInnes int itype = (int)mat->type; 112*dae03382SLois Curfman McInnes char *matname[9]; 113*dae03382SLois Curfman McInnes matname[0] = "MATDENSE"; 114*dae03382SLois Curfman McInnes matname[1] = "MATAIJ"; 115*dae03382SLois Curfman McInnes matname[2] = "MATMPIAIJ"; 116*dae03382SLois Curfman McInnes matname[3] = "MATSHELL"; 117*dae03382SLois Curfman McInnes matname[4] = "MATROW"; 118*dae03382SLois Curfman McInnes matname[5] = "MATMPIROW"; 119*dae03382SLois Curfman McInnes matname[6] = "MATMPIROW_BS"; 120*dae03382SLois Curfman McInnes matname[7] = "MATMPIBDIAG"; 121*dae03382SLois Curfman McInnes matname[8] = "MATBDIAG"; 122*dae03382SLois Curfman McInnes if (itype < 0 || itype > 8) *name = "unknown matrix type"; 123*dae03382SLois Curfman McInnes else *name = matname[itype]; 124*dae03382SLois Curfman McInnes return 0; 125*dae03382SLois Curfman McInnes } 126*dae03382SLois Curfman McInnes 127