xref: /petsc/src/mat/utils/gcreate.c (revision dae03382a1045a5f522b2689cebebda3f56b9289)
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