xref: /petsc/src/mat/interface/matrix.c (revision 146574ab6339e369828611fa942736f2bdad83fd)
1 
2 /*
3    This is where the abstract matrix operations are defined
4 */
5 
6 #include <petsc-private/matimpl.h>        /*I "petscmat.h" I*/
7 #include <petsc-private/vecimpl.h>
8 
9 /* Logging support */
10 PetscClassId MAT_CLASSID;
11 PetscClassId MAT_FDCOLORING_CLASSID;
12 PetscClassId MAT_TRANSPOSECOLORING_CLASSID;
13 
14 PetscLogEvent MAT_Mult, MAT_Mults, MAT_MultConstrained, MAT_MultAdd, MAT_MultTranspose;
15 PetscLogEvent MAT_MultTransposeConstrained, MAT_MultTransposeAdd, MAT_Solve, MAT_Solves, MAT_SolveAdd, MAT_SolveTranspose, MAT_MatSolve;
16 PetscLogEvent MAT_SolveTransposeAdd, MAT_SOR, MAT_ForwardSolve, MAT_BackwardSolve, MAT_LUFactor, MAT_LUFactorSymbolic;
17 PetscLogEvent MAT_LUFactorNumeric, MAT_CholeskyFactor, MAT_CholeskyFactorSymbolic, MAT_CholeskyFactorNumeric, MAT_ILUFactor;
18 PetscLogEvent MAT_ILUFactorSymbolic, MAT_ICCFactorSymbolic, MAT_Copy, MAT_Convert, MAT_Scale, MAT_AssemblyBegin;
19 PetscLogEvent MAT_AssemblyEnd, MAT_SetValues, MAT_GetValues, MAT_GetRow, MAT_GetRowIJ, MAT_GetSubMatrices, MAT_GetColoring, MAT_GetOrdering, MAT_GetRedundantMatrix, MAT_GetSeqNonzeroStructure;
20 PetscLogEvent MAT_IncreaseOverlap, MAT_Partitioning, MAT_Coarsen, MAT_ZeroEntries, MAT_Load, MAT_View, MAT_AXPY, MAT_FDColoringCreate;
21 PetscLogEvent MAT_FDColoringApply,MAT_Transpose,MAT_FDColoringFunction;
22 PetscLogEvent MAT_TransposeColoringCreate;
23 PetscLogEvent MAT_MatMult, MAT_MatMultSymbolic, MAT_MatMultNumeric;
24 PetscLogEvent MAT_PtAP, MAT_PtAPSymbolic, MAT_PtAPNumeric,MAT_RARt, MAT_RARtSymbolic, MAT_RARtNumeric;
25 PetscLogEvent MAT_MatTransposeMult, MAT_MatTransposeMultSymbolic, MAT_MatTransposeMultNumeric;
26 PetscLogEvent MAT_TransposeMatMult, MAT_TransposeMatMultSymbolic, MAT_TransposeMatMultNumeric;
27 PetscLogEvent MAT_MatMatMult, MAT_MatMatMultSymbolic, MAT_MatMatMultNumeric;
28 PetscLogEvent MAT_MultHermitianTranspose,MAT_MultHermitianTransposeAdd;
29 PetscLogEvent MAT_Getsymtranspose, MAT_Getsymtransreduced, MAT_Transpose_SeqAIJ, MAT_GetBrowsOfAcols;
30 PetscLogEvent MAT_GetBrowsOfAocols, MAT_Getlocalmat, MAT_Getlocalmatcondensed, MAT_Seqstompi, MAT_Seqstompinum, MAT_Seqstompisym;
31 PetscLogEvent MAT_Applypapt, MAT_Applypapt_numeric, MAT_Applypapt_symbolic, MAT_GetSequentialNonzeroStructure;
32 PetscLogEvent MAT_GetMultiProcBlock;
33 PetscLogEvent MAT_CUSPCopyToGPU, MAT_CUSPARSECopyToGPU, MAT_SetValuesBatch, MAT_SetValuesBatchI, MAT_SetValuesBatchII, MAT_SetValuesBatchIII, MAT_SetValuesBatchIV;
34 PetscLogEvent MAT_Merge;
35 
36 const char *const MatFactorTypes[] = {"NONE","LU","CHOLESKY","ILU","ICC","ILUDT","MatFactorType","MAT_FACTOR_",0};
37 
38 #undef __FUNCT__
39 #define __FUNCT__ "MatSetRandom"
40 /*@
41    MatSetRandom - Sets all components of a matrix to random numbers. For sparse matrices that have been preallocated it randomly selects appropriate locations
42 
43    Logically Collective on Vec
44 
45    Input Parameters:
46 +  x  - the vector
47 -  rctx - the random number context, formed by PetscRandomCreate(), or NULL and
48           it will create one internally.
49 
50    Output Parameter:
51 .  x  - the vector
52 
53    Example of Usage:
54 .vb
55      PetscRandomCreate(PETSC_COMM_WORLD,&rctx);
56      VecSetRandom(x,rctx);
57      PetscRandomDestroy(rctx);
58 .ve
59 
60    Level: intermediate
61 
62    Concepts: vector^setting to random
63    Concepts: random^vector
64 
65 .seealso: MatZeroEntries(), MatSetValues(), PetscRandomCreate(), PetscRandomDestroy()
66 @*/
67 PetscErrorCode  MatSetRandom(Mat x,PetscRandom rctx)
68 {
69   PetscErrorCode ierr;
70   PetscRandom    randObj = NULL;
71 
72   PetscFunctionBegin;
73   PetscValidHeaderSpecific(x,MAT_CLASSID,1);
74   if (rctx) PetscValidHeaderSpecific(rctx,PETSC_RANDOM_CLASSID,2);
75   PetscValidType(x,1);
76 
77   if (!rctx) {
78     MPI_Comm comm;
79     ierr = PetscObjectGetComm((PetscObject)x,&comm);CHKERRQ(ierr);
80     ierr = PetscRandomCreate(comm,&randObj);CHKERRQ(ierr);
81     ierr = PetscRandomSetFromOptions(randObj);CHKERRQ(ierr);
82     rctx = randObj;
83   }
84 
85   ierr = PetscLogEventBegin(VEC_SetRandom,x,rctx,0,0);CHKERRQ(ierr);
86   ierr = (*x->ops->setrandom)(x,rctx);CHKERRQ(ierr);
87   ierr = PetscLogEventEnd(VEC_SetRandom,x,rctx,0,0);CHKERRQ(ierr);
88 
89   x->assembled = PETSC_TRUE;
90   ierr         = PetscRandomDestroy(&randObj);CHKERRQ(ierr);
91   PetscFunctionReturn(0);
92 }
93 
94 
95 #undef __FUNCT__
96 #define __FUNCT__ "MatFindNonzeroRows"
97 /*@
98       MatFindNonzeroRows - Locate all rows that are not completely zero in the matrix
99 
100   Input Parameter:
101 .    A  - the matrix
102 
103   Output Parameter:
104 .    keptrows - the rows that are not completely zero
105 
106   Level: intermediate
107 
108  @*/
109 PetscErrorCode MatFindNonzeroRows(Mat mat,IS *keptrows)
110 {
111   PetscErrorCode ierr;
112 
113   PetscValidHeaderSpecific(mat,MAT_CLASSID,1);
114   PetscValidType(mat,1);
115   if (!mat->assembled) SETERRQ(PetscObjectComm((PetscObject)mat),PETSC_ERR_ARG_WRONGSTATE,"Not for unassembled matrix");
116   if (mat->factortype) SETERRQ(PetscObjectComm((PetscObject)mat),PETSC_ERR_ARG_WRONGSTATE,"Not for factored matrix");
117   if (!mat->ops->findnonzerorows) SETERRQ(PetscObjectComm((PetscObject)mat),PETSC_ERR_SUP,"Not coded for this matrix type");
118   ierr = (*mat->ops->findnonzerorows)(mat,keptrows);CHKERRQ(ierr);
119   PetscFunctionReturn(0);
120 }
121 
122 #undef __FUNCT__
123 #define __FUNCT__ "MatGetDiagonalBlock"
124 /*@
125    MatGetDiagonalBlock - Returns the part of the matrix associated with the on-process coupling
126 
127    Not Collective
128 
129    Input Parameters:
130 .   A - the matrix
131 
132    Output Parameters:
133 .   a - the diagonal part (which is a SEQUENTIAL matrix)
134 
135    Notes: see the manual page for MatCreateAIJ() for more information on the "diagonal part" of the matrix.
136 
137    Level: advanced
138 
139 @*/
140 PetscErrorCode  MatGetDiagonalBlock(Mat A,Mat *a)
141 {
142   PetscErrorCode ierr,(*f)(Mat,Mat*);
143   PetscMPIInt    size;
144 
145   PetscFunctionBegin;
146   PetscValidHeaderSpecific(A,MAT_CLASSID,1);
147   PetscValidType(A,1);
148   PetscValidPointer(a,3);
149   if (!A->assembled) SETERRQ(PetscObjectComm((PetscObject)A),PETSC_ERR_ARG_WRONGSTATE,"Not for unassembled matrix");
150   if (A->factortype) SETERRQ(PetscObjectComm((PetscObject)A),PETSC_ERR_ARG_WRONGSTATE,"Not for factored matrix");
151   ierr = MPI_Comm_size(PetscObjectComm((PetscObject)A),&size);CHKERRQ(ierr);
152   ierr = PetscObjectQueryFunction((PetscObject)A,"MatGetDiagonalBlock_C",&f);CHKERRQ(ierr);
153   if (f) {
154     ierr = (*f)(A,a);CHKERRQ(ierr);
155     PetscFunctionReturn(0);
156   } else if (size == 1) {
157     *a = A;
158   } else {
159     MatType mattype;
160     ierr = MatGetType(A,&mattype);CHKERRQ(ierr);
161     SETERRQ1(PetscObjectComm((PetscObject)A),PETSC_ERR_SUP,"Matrix type %s does not support getting diagonal block",mattype);
162   }
163   PetscFunctionReturn(0);
164 }
165 
166 #undef __FUNCT__
167 #define __FUNCT__ "MatGetTrace"
168 /*@
169    MatGetTrace - Gets the trace of a matrix. The sum of the diagonal entries.
170 
171    Collective on Mat
172 
173    Input Parameters:
174 .  mat - the matrix
175 
176    Output Parameter:
177 .   trace - the sum of the diagonal entries
178 
179    Level: advanced
180 
181 @*/
182 PetscErrorCode  MatGetTrace(Mat mat,PetscScalar *trace)
183 {
184   PetscErrorCode ierr;
185   Vec            diag;
186 
187   PetscFunctionBegin;
188   ierr = MatGetVecs(mat,&diag,NULL);CHKERRQ(ierr);
189   ierr = MatGetDiagonal(mat,diag);CHKERRQ(ierr);
190   ierr = VecSum(diag,trace);CHKERRQ(ierr);
191   ierr = VecDestroy(&diag);CHKERRQ(ierr);
192   PetscFunctionReturn(0);
193 }
194 
195 #undef __FUNCT__
196 #define __FUNCT__ "MatRealPart"
197 /*@
198    MatRealPart - Zeros out the imaginary part of the matrix
199 
200    Logically Collective on Mat
201 
202    Input Parameters:
203 .  mat - the matrix
204 
205    Level: advanced
206 
207 
208 .seealso: MatImaginaryPart()
209 @*/
210 PetscErrorCode  MatRealPart(Mat mat)
211 {
212   PetscErrorCode ierr;
213 
214   PetscFunctionBegin;
215   PetscValidHeaderSpecific(mat,MAT_CLASSID,1);
216   PetscValidType(mat,1);
217   if (!mat->assembled) SETERRQ(PetscObjectComm((PetscObject)mat),PETSC_ERR_ARG_WRONGSTATE,"Not for unassembled matrix");
218   if (mat->factortype) SETERRQ(PetscObjectComm((PetscObject)mat),PETSC_ERR_ARG_WRONGSTATE,"Not for factored matrix");
219   if (!mat->ops->realpart) SETERRQ1(PetscObjectComm((PetscObject)mat),PETSC_ERR_SUP,"Mat type %s",((PetscObject)mat)->type_name);
220   MatCheckPreallocated(mat,1);
221   ierr = (*mat->ops->realpart)(mat);CHKERRQ(ierr);
222 #if defined(PETSC_HAVE_CUSP)
223   if (mat->valid_GPU_matrix != PETSC_CUSP_UNALLOCATED) {
224     mat->valid_GPU_matrix = PETSC_CUSP_CPU;
225   }
226 #endif
227   PetscFunctionReturn(0);
228 }
229 
230 #undef __FUNCT__
231 #define __FUNCT__ "MatGetGhosts"
232 /*@C
233    MatGetGhosts - Get the global index of all ghost nodes defined by the sparse matrix
234 
235    Collective on Mat
236 
237    Input Parameter:
238 .  mat - the matrix
239 
240    Output Parameters:
241 +   nghosts - number of ghosts (note for BAIJ matrices there is one ghost for each block)
242 -   ghosts - the global indices of the ghost points
243 
244    Notes: the nghosts and ghosts are suitable to pass into VecCreateGhost()
245 
246    Level: advanced
247 
248 @*/
249 PetscErrorCode  MatGetGhosts(Mat mat,PetscInt *nghosts,const PetscInt *ghosts[])
250 {
251   PetscErrorCode ierr;
252 
253   PetscFunctionBegin;
254   PetscValidHeaderSpecific(mat,MAT_CLASSID,1);
255   PetscValidType(mat,1);
256   if (!mat->assembled) SETERRQ(PetscObjectComm((PetscObject)mat),PETSC_ERR_ARG_WRONGSTATE,"Not for unassembled matrix");
257   if (mat->factortype) SETERRQ(PetscObjectComm((PetscObject)mat),PETSC_ERR_ARG_WRONGSTATE,"Not for factored matrix");
258   if (!mat->ops->getghosts) {
259     if (nghosts) *nghosts = 0;
260     if (ghosts) *ghosts = 0;
261   } else {
262     ierr = (*mat->ops->getghosts)(mat,nghosts,ghosts);CHKERRQ(ierr);
263   }
264   PetscFunctionReturn(0);
265 }
266 
267 
268 #undef __FUNCT__
269 #define __FUNCT__ "MatImaginaryPart"
270 /*@
271    MatImaginaryPart - Moves the imaginary part of the matrix to the real part and zeros the imaginary part
272 
273    Logically Collective on Mat
274 
275    Input Parameters:
276 .  mat - the matrix
277 
278    Level: advanced
279 
280 
281 .seealso: MatRealPart()
282 @*/
283 PetscErrorCode  MatImaginaryPart(Mat mat)
284 {
285   PetscErrorCode ierr;
286 
287   PetscFunctionBegin;
288   PetscValidHeaderSpecific(mat,MAT_CLASSID,1);
289   PetscValidType(mat,1);
290   if (!mat->assembled) SETERRQ(PetscObjectComm((PetscObject)mat),PETSC_ERR_ARG_WRONGSTATE,"Not for unassembled matrix");
291   if (mat->factortype) SETERRQ(PetscObjectComm((PetscObject)mat),PETSC_ERR_ARG_WRONGSTATE,"Not for factored matrix");
292   if (!mat->ops->imaginarypart) SETERRQ1(PetscObjectComm((PetscObject)mat),PETSC_ERR_SUP,"Mat type %s",((PetscObject)mat)->type_name);
293   MatCheckPreallocated(mat,1);
294   ierr = (*mat->ops->imaginarypart)(mat);CHKERRQ(ierr);
295 #if defined(PETSC_HAVE_CUSP)
296   if (mat->valid_GPU_matrix != PETSC_CUSP_UNALLOCATED) {
297     mat->valid_GPU_matrix = PETSC_CUSP_CPU;
298   }
299 #endif
300   PetscFunctionReturn(0);
301 }
302 
303 #undef __FUNCT__
304 #define __FUNCT__ "MatMissingDiagonal"
305 /*@
306    MatMissingDiagonal - Determine if sparse matrix is missing a diagonal entry (or block entry for BAIJ matrices)
307 
308    Collective on Mat
309 
310    Input Parameter:
311 .  mat - the matrix
312 
313    Output Parameters:
314 +  missing - is any diagonal missing
315 -  dd - first diagonal entry that is missing (optional)
316 
317    Level: advanced
318 
319 
320 .seealso: MatRealPart()
321 @*/
322 PetscErrorCode  MatMissingDiagonal(Mat mat,PetscBool *missing,PetscInt *dd)
323 {
324   PetscErrorCode ierr;
325 
326   PetscFunctionBegin;
327   PetscValidHeaderSpecific(mat,MAT_CLASSID,1);
328   PetscValidType(mat,1);
329   if (!mat->assembled) SETERRQ(PetscObjectComm((PetscObject)mat),PETSC_ERR_ARG_WRONGSTATE,"Not for unassembled matrix");
330   if (mat->factortype) SETERRQ(PetscObjectComm((PetscObject)mat),PETSC_ERR_ARG_WRONGSTATE,"Not for factored matrix");
331   if (!mat->ops->missingdiagonal) SETERRQ1(PetscObjectComm((PetscObject)mat),PETSC_ERR_SUP,"Mat type %s",((PetscObject)mat)->type_name);
332   ierr = (*mat->ops->missingdiagonal)(mat,missing,dd);CHKERRQ(ierr);
333   PetscFunctionReturn(0);
334 }
335 
336 #undef __FUNCT__
337 #define __FUNCT__ "MatGetRow"
338 /*@C
339    MatGetRow - Gets a row of a matrix.  You MUST call MatRestoreRow()
340    for each row that you get to ensure that your application does
341    not bleed memory.
342 
343    Not Collective
344 
345    Input Parameters:
346 +  mat - the matrix
347 -  row - the row to get
348 
349    Output Parameters:
350 +  ncols -  if not NULL, the number of nonzeros in the row
351 .  cols - if not NULL, the column numbers
352 -  vals - if not NULL, the values
353 
354    Notes:
355    This routine is provided for people who need to have direct access
356    to the structure of a matrix.  We hope that we provide enough
357    high-level matrix routines that few users will need it.
358 
359    MatGetRow() always returns 0-based column indices, regardless of
360    whether the internal representation is 0-based (default) or 1-based.
361 
362    For better efficiency, set cols and/or vals to NULL if you do
363    not wish to extract these quantities.
364 
365    The user can only examine the values extracted with MatGetRow();
366    the values cannot be altered.  To change the matrix entries, one
367    must use MatSetValues().
368 
369    You can only have one call to MatGetRow() outstanding for a particular
370    matrix at a time, per processor. MatGetRow() can only obtain rows
371    associated with the given processor, it cannot get rows from the
372    other processors; for that we suggest using MatGetSubMatrices(), then
373    MatGetRow() on the submatrix. The row indix passed to MatGetRows()
374    is in the global number of rows.
375 
376    Fortran Notes:
377    The calling sequence from Fortran is
378 .vb
379    MatGetRow(matrix,row,ncols,cols,values,ierr)
380          Mat     matrix (input)
381          integer row    (input)
382          integer ncols  (output)
383          integer cols(maxcols) (output)
384          double precision (or double complex) values(maxcols) output
385 .ve
386    where maxcols >= maximum nonzeros in any row of the matrix.
387 
388 
389    Caution:
390    Do not try to change the contents of the output arrays (cols and vals).
391    In some cases, this may corrupt the matrix.
392 
393    Level: advanced
394 
395    Concepts: matrices^row access
396 
397 .seealso: MatRestoreRow(), MatSetValues(), MatGetValues(), MatGetSubMatrices(), MatGetDiagonal()
398 @*/
399 PetscErrorCode  MatGetRow(Mat mat,PetscInt row,PetscInt *ncols,const PetscInt *cols[],const PetscScalar *vals[])
400 {
401   PetscErrorCode ierr;
402   PetscInt       incols;
403 
404   PetscFunctionBegin;
405   PetscValidHeaderSpecific(mat,MAT_CLASSID,1);
406   PetscValidType(mat,1);
407   if (!mat->assembled) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONGSTATE,"Not for unassembled matrix");
408   if (mat->factortype) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONGSTATE,"Not for factored matrix");
409   if (!mat->ops->getrow) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_SUP,"Mat type %s",((PetscObject)mat)->type_name);
410   MatCheckPreallocated(mat,1);
411   ierr = PetscLogEventBegin(MAT_GetRow,mat,0,0,0);CHKERRQ(ierr);
412   ierr = (*mat->ops->getrow)(mat,row,&incols,(PetscInt**)cols,(PetscScalar**)vals);CHKERRQ(ierr);
413   if (ncols) *ncols = incols;
414   ierr = PetscLogEventEnd(MAT_GetRow,mat,0,0,0);CHKERRQ(ierr);
415   PetscFunctionReturn(0);
416 }
417 
418 #undef __FUNCT__
419 #define __FUNCT__ "MatConjugate"
420 /*@
421    MatConjugate - replaces the matrix values with their complex conjugates
422 
423    Logically Collective on Mat
424 
425    Input Parameters:
426 .  mat - the matrix
427 
428    Level: advanced
429 
430 .seealso:  VecConjugate()
431 @*/
432 PetscErrorCode  MatConjugate(Mat mat)
433 {
434 #if defined(PETSC_USE_COMPLEX)
435   PetscErrorCode ierr;
436 
437   PetscFunctionBegin;
438   PetscValidHeaderSpecific(mat,MAT_CLASSID,1);
439   if (!mat->assembled) SETERRQ(PetscObjectComm((PetscObject)mat),PETSC_ERR_ARG_WRONGSTATE,"Not for unassembled matrix");
440   if (!mat->ops->conjugate) SETERRQ(PetscObjectComm((PetscObject)mat),PETSC_ERR_SUP,"Not provided for this matrix format, send email to petsc-maint@mcs.anl.gov");
441   ierr = (*mat->ops->conjugate)(mat);CHKERRQ(ierr);
442 #if defined(PETSC_HAVE_CUSP)
443   if (mat->valid_GPU_matrix != PETSC_CUSP_UNALLOCATED) {
444     mat->valid_GPU_matrix = PETSC_CUSP_CPU;
445   }
446 #endif
447   PetscFunctionReturn(0);
448 #else
449   return 0;
450 #endif
451 }
452 
453 #undef __FUNCT__
454 #define __FUNCT__ "MatRestoreRow"
455 /*@C
456    MatRestoreRow - Frees any temporary space allocated by MatGetRow().
457 
458    Not Collective
459 
460    Input Parameters:
461 +  mat - the matrix
462 .  row - the row to get
463 .  ncols, cols - the number of nonzeros and their columns
464 -  vals - if nonzero the column values
465 
466    Notes:
467    This routine should be called after you have finished examining the entries.
468 
469    Fortran Notes:
470    The calling sequence from Fortran is
471 .vb
472    MatRestoreRow(matrix,row,ncols,cols,values,ierr)
473       Mat     matrix (input)
474       integer row    (input)
475       integer ncols  (output)
476       integer cols(maxcols) (output)
477       double precision (or double complex) values(maxcols) output
478 .ve
479    Where maxcols >= maximum nonzeros in any row of the matrix.
480 
481    In Fortran MatRestoreRow() MUST be called after MatGetRow()
482    before another call to MatGetRow() can be made.
483 
484    Level: advanced
485 
486 .seealso:  MatGetRow()
487 @*/
488 PetscErrorCode  MatRestoreRow(Mat mat,PetscInt row,PetscInt *ncols,const PetscInt *cols[],const PetscScalar *vals[])
489 {
490   PetscErrorCode ierr;
491 
492   PetscFunctionBegin;
493   PetscValidHeaderSpecific(mat,MAT_CLASSID,1);
494   if (ncols) PetscValidIntPointer(ncols,3);
495   if (!mat->assembled) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONGSTATE,"Not for unassembled matrix");
496   if (!mat->ops->restorerow) PetscFunctionReturn(0);
497   ierr = (*mat->ops->restorerow)(mat,row,ncols,(PetscInt **)cols,(PetscScalar **)vals);CHKERRQ(ierr);
498   if (ncols) *ncols = 0;
499   if (cols)  *cols = NULL;
500   if (vals)  *vals = NULL;
501   PetscFunctionReturn(0);
502 }
503 
504 #undef __FUNCT__
505 #define __FUNCT__ "MatGetRowUpperTriangular"
506 /*@
507    MatGetRowUpperTriangular - Sets a flag to enable calls to MatGetRow() for matrix in MATSBAIJ format.
508    You should call MatRestoreRowUpperTriangular() after calling MatGetRow/MatRestoreRow() to disable the flag.
509 
510    Not Collective
511 
512    Input Parameters:
513 +  mat - the matrix
514 
515    Notes:
516    The flag is to ensure that users are aware of MatGetRow() only provides the upper trianglular part of the row for the matrices in MATSBAIJ format.
517 
518    Level: advanced
519 
520    Concepts: matrices^row access
521 
522 .seealso: MatRestoreRowRowUpperTriangular()
523 @*/
524 PetscErrorCode  MatGetRowUpperTriangular(Mat mat)
525 {
526   PetscErrorCode ierr;
527 
528   PetscFunctionBegin;
529   PetscValidHeaderSpecific(mat,MAT_CLASSID,1);
530   PetscValidType(mat,1);
531   if (!mat->assembled) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONGSTATE,"Not for unassembled matrix");
532   if (mat->factortype) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONGSTATE,"Not for factored matrix");
533   if (!mat->ops->getrowuppertriangular) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_SUP,"Mat type %s",((PetscObject)mat)->type_name);
534   MatCheckPreallocated(mat,1);
535   ierr = (*mat->ops->getrowuppertriangular)(mat);CHKERRQ(ierr);
536   PetscFunctionReturn(0);
537 }
538 
539 #undef __FUNCT__
540 #define __FUNCT__ "MatRestoreRowUpperTriangular"
541 /*@
542    MatRestoreRowUpperTriangular - Disable calls to MatGetRow() for matrix in MATSBAIJ format.
543 
544    Not Collective
545 
546    Input Parameters:
547 +  mat - the matrix
548 
549    Notes:
550    This routine should be called after you have finished MatGetRow/MatRestoreRow().
551 
552 
553    Level: advanced
554 
555 .seealso:  MatGetRowUpperTriangular()
556 @*/
557 PetscErrorCode  MatRestoreRowUpperTriangular(Mat mat)
558 {
559   PetscErrorCode ierr;
560 
561   PetscFunctionBegin;
562   PetscValidHeaderSpecific(mat,MAT_CLASSID,1);
563   if (!mat->assembled) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONGSTATE,"Not for unassembled matrix");
564   if (!mat->ops->restorerowuppertriangular) PetscFunctionReturn(0);
565   ierr = (*mat->ops->restorerowuppertriangular)(mat);CHKERRQ(ierr);
566   PetscFunctionReturn(0);
567 }
568 
569 #undef __FUNCT__
570 #define __FUNCT__ "MatSetOptionsPrefix"
571 /*@C
572    MatSetOptionsPrefix - Sets the prefix used for searching for all
573    Mat options in the database.
574 
575    Logically Collective on Mat
576 
577    Input Parameter:
578 +  A - the Mat context
579 -  prefix - the prefix to prepend to all option names
580 
581    Notes:
582    A hyphen (-) must NOT be given at the beginning of the prefix name.
583    The first character of all runtime options is AUTOMATICALLY the hyphen.
584 
585    Level: advanced
586 
587 .keywords: Mat, set, options, prefix, database
588 
589 .seealso: MatSetFromOptions()
590 @*/
591 PetscErrorCode  MatSetOptionsPrefix(Mat A,const char prefix[])
592 {
593   PetscErrorCode ierr;
594 
595   PetscFunctionBegin;
596   PetscValidHeaderSpecific(A,MAT_CLASSID,1);
597   ierr = PetscObjectSetOptionsPrefix((PetscObject)A,prefix);CHKERRQ(ierr);
598   PetscFunctionReturn(0);
599 }
600 
601 #undef __FUNCT__
602 #define __FUNCT__ "MatAppendOptionsPrefix"
603 /*@C
604    MatAppendOptionsPrefix - Appends to the prefix used for searching for all
605    Mat options in the database.
606 
607    Logically Collective on Mat
608 
609    Input Parameters:
610 +  A - the Mat context
611 -  prefix - the prefix to prepend to all option names
612 
613    Notes:
614    A hyphen (-) must NOT be given at the beginning of the prefix name.
615    The first character of all runtime options is AUTOMATICALLY the hyphen.
616 
617    Level: advanced
618 
619 .keywords: Mat, append, options, prefix, database
620 
621 .seealso: MatGetOptionsPrefix()
622 @*/
623 PetscErrorCode  MatAppendOptionsPrefix(Mat A,const char prefix[])
624 {
625   PetscErrorCode ierr;
626 
627   PetscFunctionBegin;
628   PetscValidHeaderSpecific(A,MAT_CLASSID,1);
629   ierr = PetscObjectAppendOptionsPrefix((PetscObject)A,prefix);CHKERRQ(ierr);
630   PetscFunctionReturn(0);
631 }
632 
633 #undef __FUNCT__
634 #define __FUNCT__ "MatGetOptionsPrefix"
635 /*@C
636    MatGetOptionsPrefix - Sets the prefix used for searching for all
637    Mat options in the database.
638 
639    Not Collective
640 
641    Input Parameter:
642 .  A - the Mat context
643 
644    Output Parameter:
645 .  prefix - pointer to the prefix string used
646 
647    Notes: On the fortran side, the user should pass in a string 'prefix' of
648    sufficient length to hold the prefix.
649 
650    Level: advanced
651 
652 .keywords: Mat, get, options, prefix, database
653 
654 .seealso: MatAppendOptionsPrefix()
655 @*/
656 PetscErrorCode  MatGetOptionsPrefix(Mat A,const char *prefix[])
657 {
658   PetscErrorCode ierr;
659 
660   PetscFunctionBegin;
661   PetscValidHeaderSpecific(A,MAT_CLASSID,1);
662   ierr = PetscObjectGetOptionsPrefix((PetscObject)A,prefix);CHKERRQ(ierr);
663   PetscFunctionReturn(0);
664 }
665 
666 #undef __FUNCT__
667 #define __FUNCT__ "MatSetUp"
668 /*@
669    MatSetUp - Sets up the internal matrix data structures for the later use.
670 
671    Collective on Mat
672 
673    Input Parameters:
674 .  A - the Mat context
675 
676    Notes:
677    If the user has not set preallocation for this matrix then a default preallocation that is likely to be inefficient is used.
678 
679    If a suitable preallocation routine is used, this function does not need to be called.
680 
681    See the Performance chapter of the PETSc users manual for how to preallocate matrices
682 
683    Level: beginner
684 
685 .keywords: Mat, setup
686 
687 .seealso: MatCreate(), MatDestroy()
688 @*/
689 PetscErrorCode  MatSetUp(Mat A)
690 {
691   PetscMPIInt    size;
692   PetscErrorCode ierr;
693 
694   PetscFunctionBegin;
695   PetscValidHeaderSpecific(A,MAT_CLASSID,1);
696   if (!((PetscObject)A)->type_name) {
697     ierr = MPI_Comm_size(PetscObjectComm((PetscObject)A), &size);CHKERRQ(ierr);
698     if (size == 1) {
699       ierr = MatSetType(A, MATSEQAIJ);CHKERRQ(ierr);
700     } else {
701       ierr = MatSetType(A, MATMPIAIJ);CHKERRQ(ierr);
702     }
703   }
704   if (!A->preallocated && A->ops->setup) {
705     ierr = PetscInfo(A,"Warning not preallocating matrix storage\n");CHKERRQ(ierr);
706     ierr = (*A->ops->setup)(A);CHKERRQ(ierr);
707   }
708   A->preallocated = PETSC_TRUE;
709   PetscFunctionReturn(0);
710 }
711 
712 #if defined(PETSC_HAVE_AMS)
713 #include <petscviewerams.h>
714 #endif
715 #undef __FUNCT__
716 #define __FUNCT__ "MatView"
717 /*@C
718    MatView - Visualizes a matrix object.
719 
720    Collective on Mat
721 
722    Input Parameters:
723 +  mat - the matrix
724 -  viewer - visualization context
725 
726   Notes:
727   The available visualization contexts include
728 +    PETSC_VIEWER_STDOUT_SELF - standard output (default)
729 .    PETSC_VIEWER_STDOUT_WORLD - synchronized standard
730         output where only the first processor opens
731         the file.  All other processors send their
732         data to the first processor to print.
733 -     PETSC_VIEWER_DRAW_WORLD - graphical display of nonzero structure
734 
735    The user can open alternative visualization contexts with
736 +    PetscViewerASCIIOpen() - Outputs matrix to a specified file
737 .    PetscViewerBinaryOpen() - Outputs matrix in binary to a
738          specified file; corresponding input uses MatLoad()
739 .    PetscViewerDrawOpen() - Outputs nonzero matrix structure to
740          an X window display
741 -    PetscViewerSocketOpen() - Outputs matrix to Socket viewer.
742          Currently only the sequential dense and AIJ
743          matrix types support the Socket viewer.
744 
745    The user can call PetscViewerSetFormat() to specify the output
746    format of ASCII printed objects (when using PETSC_VIEWER_STDOUT_SELF,
747    PETSC_VIEWER_STDOUT_WORLD and PetscViewerASCIIOpen).  Available formats include
748 +    PETSC_VIEWER_DEFAULT - default, prints matrix contents
749 .    PETSC_VIEWER_ASCII_MATLAB - prints matrix contents in Matlab format
750 .    PETSC_VIEWER_ASCII_DENSE - prints entire matrix including zeros
751 .    PETSC_VIEWER_ASCII_COMMON - prints matrix contents, using a sparse
752          format common among all matrix types
753 .    PETSC_VIEWER_ASCII_IMPL - prints matrix contents, using an implementation-specific
754          format (which is in many cases the same as the default)
755 .    PETSC_VIEWER_ASCII_INFO - prints basic information about the matrix
756          size and structure (not the matrix entries)
757 .    PETSC_VIEWER_ASCII_INFO_DETAIL - prints more detailed information about
758          the matrix structure
759 
760    Options Database Keys:
761 +  -mat_view ::ascii_info - Prints info on matrix at conclusion of MatEndAssembly()
762 .  -mat_view ::ascii_info_detail - Prints more detailed info
763 .  -mat_view - Prints matrix in ASCII format
764 .  -mat_view ::ascii_matlab - Prints matrix in Matlab format
765 .  -mat_view draw - PetscDraws nonzero structure of matrix, using MatView() and PetscDrawOpenX().
766 .  -display <name> - Sets display name (default is host)
767 .  -draw_pause <sec> - Sets number of seconds to pause after display
768 .  -mat_view socket - Sends matrix to socket, can be accessed from Matlab (see the <a href="../../docs/manual.pdf">users manual</a> for details).
769 .  -viewer_socket_machine <machine>
770 .  -viewer_socket_port <port>
771 .  -mat_view binary - save matrix to file in binary format
772 -  -viewer_binary_filename <name>
773    Level: beginner
774 
775    Notes: see the manual page for MatLoad() for the exact format of the binary file when the binary
776       viewer is used.
777 
778       See bin/matlab/PetscBinaryRead.m for a Matlab code that can read in the binary file when the binary
779       viewer is used.
780 
781       One can use '-mat_view draw -draw_pause -1' to pause the graphical display of matrix nonzero structure.
782       And then use the following mouse functions:
783           left mouse: zoom in
784           middle mouse: zoom out
785           right mouse: continue with the simulation
786 
787    Concepts: matrices^viewing
788    Concepts: matrices^plotting
789    Concepts: matrices^printing
790 
791 .seealso: PetscViewerSetFormat(), PetscViewerASCIIOpen(), PetscViewerDrawOpen(),
792           PetscViewerSocketOpen(), PetscViewerBinaryOpen(), MatLoad()
793 @*/
794 PetscErrorCode  MatView(Mat mat,PetscViewer viewer)
795 {
796   PetscErrorCode    ierr;
797   PetscInt          rows,cols,bs;
798   PetscBool         iascii;
799   PetscViewerFormat format;
800 #if defined(PETSC_HAVE_AMS)
801   PetscBool      isams;
802 #endif
803 
804   PetscFunctionBegin;
805   PetscValidHeaderSpecific(mat,MAT_CLASSID,1);
806   PetscValidType(mat,1);
807   if (!viewer) {
808     ierr = PetscViewerASCIIGetStdout(PetscObjectComm((PetscObject)mat),&viewer);CHKERRQ(ierr);
809   }
810   PetscValidHeaderSpecific(viewer,PETSC_VIEWER_CLASSID,2);
811   PetscCheckSameComm(mat,1,viewer,2);
812   if (!mat->assembled) SETERRQ(PetscObjectComm((PetscObject)mat),PETSC_ERR_ORDER,"Must call MatAssemblyBegin/End() before viewing matrix");
813   MatCheckPreallocated(mat,1);
814 
815   ierr = PetscLogEventBegin(MAT_View,mat,viewer,0,0);CHKERRQ(ierr);
816   ierr = PetscObjectTypeCompare((PetscObject)viewer,PETSCVIEWERASCII,&iascii);CHKERRQ(ierr);
817 #if defined(PETSC_HAVE_AMS)
818   ierr = PetscObjectTypeCompare((PetscObject)viewer,PETSCVIEWERAMS,&isams);CHKERRQ(ierr);
819 #endif
820   if (iascii) {
821     ierr = PetscViewerGetFormat(viewer,&format);CHKERRQ(ierr);
822     if (format == PETSC_VIEWER_ASCII_INFO || format == PETSC_VIEWER_ASCII_INFO_DETAIL) {
823       ierr = PetscObjectPrintClassNamePrefixType((PetscObject)mat,viewer,"Matrix Object");CHKERRQ(ierr);
824       ierr = PetscViewerASCIIPushTab(viewer);CHKERRQ(ierr);
825       ierr = MatGetSize(mat,&rows,&cols);CHKERRQ(ierr);
826       ierr = MatGetBlockSize(mat,&bs);CHKERRQ(ierr);
827       if (bs != 1) {
828         ierr = PetscViewerASCIIPrintf(viewer,"rows=%D, cols=%D, bs=%D\n",rows,cols,bs);CHKERRQ(ierr);
829       } else {
830         ierr = PetscViewerASCIIPrintf(viewer,"rows=%D, cols=%D\n",rows,cols);CHKERRQ(ierr);
831       }
832       if (mat->factortype) {
833         const MatSolverPackage solver;
834         ierr = MatFactorGetSolverPackage(mat,&solver);CHKERRQ(ierr);
835         ierr = PetscViewerASCIIPrintf(viewer,"package used to perform factorization: %s\n",solver);CHKERRQ(ierr);
836       }
837       if (mat->ops->getinfo) {
838         MatInfo info;
839         ierr = MatGetInfo(mat,MAT_GLOBAL_SUM,&info);CHKERRQ(ierr);
840         ierr = PetscViewerASCIIPrintf(viewer,"total: nonzeros=%lld, allocated nonzeros=%lld\n",(Petsc64bitInt)info.nz_used,(Petsc64bitInt)info.nz_allocated);CHKERRQ(ierr);
841         ierr = PetscViewerASCIIPrintf(viewer,"total number of mallocs used during MatSetValues calls =%D\n",(PetscInt)info.mallocs);CHKERRQ(ierr);
842       }
843     }
844 #if defined(PETSC_HAVE_AMS)
845   } else if (isams) {
846     if (((PetscObject)mat)->amsmem == -1) {
847       ierr = PetscObjectViewAMS((PetscObject)mat,viewer);CHKERRQ(ierr);
848     }
849 #endif
850   }
851   if (mat->ops->view) {
852     ierr = PetscViewerASCIIPushTab(viewer);CHKERRQ(ierr);
853     ierr = (*mat->ops->view)(mat,viewer);CHKERRQ(ierr);
854     ierr = PetscViewerASCIIPopTab(viewer);CHKERRQ(ierr);
855   } else if (!iascii) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_SUP,"Viewer type %s not supported",((PetscObject)viewer)->type_name);
856   if (iascii) {
857     ierr = PetscViewerGetFormat(viewer,&format);CHKERRQ(ierr);
858     if (format == PETSC_VIEWER_ASCII_INFO || format == PETSC_VIEWER_ASCII_INFO_DETAIL) {
859       ierr = PetscViewerASCIIPopTab(viewer);CHKERRQ(ierr);
860     }
861   }
862   ierr = PetscLogEventEnd(MAT_View,mat,viewer,0,0);CHKERRQ(ierr);
863   PetscFunctionReturn(0);
864 }
865 
866 #if defined(PETSC_USE_DEBUG)
867 #include <../src/sys/totalview/tv_data_display.h>
868 PETSC_UNUSED static int TV_display_type(const struct _p_Mat *mat)
869 {
870   TV_add_row("Local rows", "int", &mat->rmap->n);
871   TV_add_row("Local columns", "int", &mat->cmap->n);
872   TV_add_row("Global rows", "int", &mat->rmap->N);
873   TV_add_row("Global columns", "int", &mat->cmap->N);
874   TV_add_row("Typename", TV_ascii_string_type, ((PetscObject)mat)->type_name);
875   return TV_format_OK;
876 }
877 #endif
878 
879 #undef __FUNCT__
880 #define __FUNCT__ "MatLoad"
881 /*@C
882    MatLoad - Loads a matrix that has been stored in binary format
883    with MatView().  The matrix format is determined from the options database.
884    Generates a parallel MPI matrix if the communicator has more than one
885    processor.  The default matrix type is AIJ.
886 
887    Collective on PetscViewer
888 
889    Input Parameters:
890 +  newmat - the newly loaded matrix, this needs to have been created with MatCreate()
891             or some related function before a call to MatLoad()
892 -  viewer - binary file viewer, created with PetscViewerBinaryOpen()
893 
894    Options Database Keys:
895    Used with block matrix formats (MATSEQBAIJ,  ...) to specify
896    block size
897 .    -matload_block_size <bs>
898 
899    Level: beginner
900 
901    Notes:
902    If the Mat type has not yet been given then MATAIJ is used, call MatSetFromOptions() on the
903    Mat before calling this routine if you wish to set it from the options database.
904 
905    MatLoad() automatically loads into the options database any options
906    given in the file filename.info where filename is the name of the file
907    that was passed to the PetscViewerBinaryOpen(). The options in the info
908    file will be ignored if you use the -viewer_binary_skip_info option.
909 
910    If the type or size of newmat is not set before a call to MatLoad, PETSc
911    sets the default matrix type AIJ and sets the local and global sizes.
912    If type and/or size is already set, then the same are used.
913 
914    In parallel, each processor can load a subset of rows (or the
915    entire matrix).  This routine is especially useful when a large
916    matrix is stored on disk and only part of it is desired on each
917    processor.  For example, a parallel solver may access only some of
918    the rows from each processor.  The algorithm used here reads
919    relatively small blocks of data rather than reading the entire
920    matrix and then subsetting it.
921 
922    Notes for advanced users:
923    Most users should not need to know the details of the binary storage
924    format, since MatLoad() and MatView() completely hide these details.
925    But for anyone who's interested, the standard binary matrix storage
926    format is
927 
928 $    int    MAT_FILE_CLASSID
929 $    int    number of rows
930 $    int    number of columns
931 $    int    total number of nonzeros
932 $    int    *number nonzeros in each row
933 $    int    *column indices of all nonzeros (starting index is zero)
934 $    PetscScalar *values of all nonzeros
935 
936    PETSc automatically does the byte swapping for
937 machines that store the bytes reversed, e.g.  DEC alpha, freebsd,
938 linux, Windows and the paragon; thus if you write your own binary
939 read/write routines you have to swap the bytes; see PetscBinaryRead()
940 and PetscBinaryWrite() to see how this may be done.
941 
942 .keywords: matrix, load, binary, input
943 
944 .seealso: PetscViewerBinaryOpen(), MatView(), VecLoad()
945 
946  @*/
947 PetscErrorCode  MatLoad(Mat newmat,PetscViewer viewer)
948 {
949   PetscErrorCode ierr;
950   PetscBool      isbinary,flg;
951 
952   PetscFunctionBegin;
953   PetscValidHeaderSpecific(newmat,MAT_CLASSID,1);
954   PetscValidHeaderSpecific(viewer,PETSC_VIEWER_CLASSID,2);
955   ierr = PetscObjectTypeCompare((PetscObject)viewer,PETSCVIEWERBINARY,&isbinary);CHKERRQ(ierr);
956   if (!isbinary) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONG,"Invalid viewer; open viewer with PetscViewerBinaryOpen()");
957 
958   if (!((PetscObject)newmat)->type_name) {
959     ierr = MatSetType(newmat,MATAIJ);CHKERRQ(ierr);
960   }
961 
962   if (!newmat->ops->load) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"MatLoad is not supported for type");
963   ierr = PetscLogEventBegin(MAT_Load,viewer,0,0,0);CHKERRQ(ierr);
964   ierr = (*newmat->ops->load)(newmat,viewer);CHKERRQ(ierr);
965   ierr = PetscLogEventEnd(MAT_Load,viewer,0,0,0);CHKERRQ(ierr);
966 
967   flg  = PETSC_FALSE;
968   ierr = PetscOptionsGetBool(((PetscObject)newmat)->prefix,"-matload_symmetric",&flg,NULL);CHKERRQ(ierr);
969   if (flg) {
970     ierr = MatSetOption(newmat,MAT_SYMMETRIC,PETSC_TRUE);CHKERRQ(ierr);
971     ierr = MatSetOption(newmat,MAT_SYMMETRY_ETERNAL,PETSC_TRUE);CHKERRQ(ierr);
972   }
973   flg  = PETSC_FALSE;
974   ierr = PetscOptionsGetBool(((PetscObject)newmat)->prefix,"-matload_spd",&flg,NULL);CHKERRQ(ierr);
975   if (flg) {
976     ierr = MatSetOption(newmat,MAT_SPD,PETSC_TRUE);CHKERRQ(ierr);
977   }
978   PetscFunctionReturn(0);
979 }
980 
981 #undef __FUNCT__
982 #define __FUNCT__ "MatDestroy"
983 /*@
984    MatDestroy - Frees space taken by a matrix.
985 
986    Collective on Mat
987 
988    Input Parameter:
989 .  A - the matrix
990 
991    Level: beginner
992 
993 @*/
994 PetscErrorCode  MatDestroy(Mat *A)
995 {
996   PetscErrorCode ierr;
997 
998   PetscFunctionBegin;
999   if (!*A) PetscFunctionReturn(0);
1000   PetscValidHeaderSpecific(*A,MAT_CLASSID,1);
1001   if (--((PetscObject)(*A))->refct > 0) {*A = NULL; PetscFunctionReturn(0);}
1002 
1003   ierr = PetscViewerDestroy(&(*A)->viewonassembly);CHKERRQ(ierr);
1004   /* if memory was published with AMS then destroy it */
1005   ierr = PetscObjectAMSViewOff((PetscObject)*A);CHKERRQ(ierr);
1006   if ((*A)->ops->destroy) {
1007     ierr = (*(*A)->ops->destroy)(*A);CHKERRQ(ierr);
1008   }
1009   ierr = MatNullSpaceDestroy(&(*A)->nullsp);CHKERRQ(ierr);
1010   ierr = MatNullSpaceDestroy(&(*A)->nearnullsp);CHKERRQ(ierr);
1011   ierr = PetscLayoutDestroy(&(*A)->rmap);CHKERRQ(ierr);
1012   ierr = PetscLayoutDestroy(&(*A)->cmap);CHKERRQ(ierr);
1013   ierr = PetscHeaderDestroy(A);CHKERRQ(ierr);
1014   PetscFunctionReturn(0);
1015 }
1016 
1017 #undef __FUNCT__
1018 #define __FUNCT__ "MatSetValues"
1019 /*@
1020    MatSetValues - Inserts or adds a block of values into a matrix.
1021    These values may be cached, so MatAssemblyBegin() and MatAssemblyEnd()
1022    MUST be called after all calls to MatSetValues() have been completed.
1023 
1024    Not Collective
1025 
1026    Input Parameters:
1027 +  mat - the matrix
1028 .  v - a logically two-dimensional array of values
1029 .  m, idxm - the number of rows and their global indices
1030 .  n, idxn - the number of columns and their global indices
1031 -  addv - either ADD_VALUES or INSERT_VALUES, where
1032    ADD_VALUES adds values to any existing entries, and
1033    INSERT_VALUES replaces existing entries with new values
1034 
1035    Notes:
1036    If you create the matrix yourself (that is not with a call to DMCreateMatrix()) then you MUST call MatXXXXSetPreallocation() or
1037       MatSetUp() before using this routine
1038 
1039    By default the values, v, are row-oriented. See MatSetOption() for other options.
1040 
1041    Calls to MatSetValues() with the INSERT_VALUES and ADD_VALUES
1042    options cannot be mixed without intervening calls to the assembly
1043    routines.
1044 
1045    MatSetValues() uses 0-based row and column numbers in Fortran
1046    as well as in C.
1047 
1048    Negative indices may be passed in idxm and idxn, these rows and columns are
1049    simply ignored. This allows easily inserting element stiffness matrices
1050    with homogeneous Dirchlet boundary conditions that you don't want represented
1051    in the matrix.
1052 
1053    Efficiency Alert:
1054    The routine MatSetValuesBlocked() may offer much better efficiency
1055    for users of block sparse formats (MATSEQBAIJ and MATMPIBAIJ).
1056 
1057    Level: beginner
1058 
1059    Concepts: matrices^putting entries in
1060 
1061 .seealso: MatSetOption(), MatAssemblyBegin(), MatAssemblyEnd(), MatSetValuesBlocked(), MatSetValuesLocal(),
1062           InsertMode, INSERT_VALUES, ADD_VALUES
1063 @*/
1064 PetscErrorCode  MatSetValues(Mat mat,PetscInt m,const PetscInt idxm[],PetscInt n,const PetscInt idxn[],const PetscScalar v[],InsertMode addv)
1065 {
1066   PetscErrorCode ierr;
1067 #if defined(PETSC_USE_DEBUG)
1068   PetscInt i,j;
1069 #endif
1070 
1071   PetscFunctionBegin;
1072   PetscValidHeaderSpecific(mat,MAT_CLASSID,1);
1073   PetscValidType(mat,1);
1074   if (!m || !n) PetscFunctionReturn(0); /* no values to insert */
1075   PetscValidIntPointer(idxm,3);
1076   PetscValidIntPointer(idxn,5);
1077   if (v) PetscValidScalarPointer(v,6);
1078   MatCheckPreallocated(mat,1);
1079   if (mat->insertmode == NOT_SET_VALUES) {
1080     mat->insertmode = addv;
1081   }
1082 #if defined(PETSC_USE_DEBUG)
1083   else if (mat->insertmode != addv) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONGSTATE,"Cannot mix add values and insert values");
1084   if (mat->factortype) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONGSTATE,"Not for factored matrix");
1085   if (!mat->ops->setvalues) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_SUP,"Mat type %s",((PetscObject)mat)->type_name);
1086 
1087   if (v) {
1088     for (i=0; i<m; i++) {
1089       for (j=0; j<n; j++) {
1090         if (PetscIsInfOrNanScalar(v[i*n+j]))
1091 #if defined(PETSC_USE_COMPLEX)
1092           SETERRQ4(PETSC_COMM_SELF,PETSC_ERR_FP,"Inserting %G+iG at matrix entry (%D,%D)",PetscRealPart(v[i*n+j]),PetscImaginaryPart(v[i*n+j]),idxm[i],idxn[j]);
1093 #else
1094           SETERRQ3(PETSC_COMM_SELF,PETSC_ERR_FP,"Inserting %G at matrix entry (%D,%D)",(PetscReal)v[i*n+j],idxm[i],idxn[j]);
1095 #endif
1096       }
1097     }
1098   }
1099 #endif
1100 
1101   if (mat->assembled) {
1102     mat->was_assembled = PETSC_TRUE;
1103     mat->assembled     = PETSC_FALSE;
1104   }
1105   ierr = PetscLogEventBegin(MAT_SetValues,mat,0,0,0);CHKERRQ(ierr);
1106   ierr = (*mat->ops->setvalues)(mat,m,idxm,n,idxn,v,addv);CHKERRQ(ierr);
1107   ierr = PetscLogEventEnd(MAT_SetValues,mat,0,0,0);CHKERRQ(ierr);
1108 #if defined(PETSC_HAVE_CUSP)
1109   if (mat->valid_GPU_matrix != PETSC_CUSP_UNALLOCATED) {
1110     mat->valid_GPU_matrix = PETSC_CUSP_CPU;
1111   }
1112 #endif
1113   PetscFunctionReturn(0);
1114 }
1115 
1116 
1117 #undef __FUNCT__
1118 #define __FUNCT__ "MatSetValuesRowLocal"
1119 /*@
1120    MatSetValuesRowLocal - Inserts a row (block row for BAIJ matrices) of nonzero
1121         values into a matrix
1122 
1123    Not Collective
1124 
1125    Input Parameters:
1126 +  mat - the matrix
1127 .  row - the (block) row to set
1128 -  v - a logically two-dimensional array of values
1129 
1130    Notes:
1131    By the values, v, are column-oriented (for the block version) and sorted
1132 
1133    All the nonzeros in the row must be provided
1134 
1135    The matrix must have previously had its column indices set
1136 
1137    The row must belong to this process
1138 
1139    Level: intermediate
1140 
1141    Concepts: matrices^putting entries in
1142 
1143 .seealso: MatSetOption(), MatAssemblyBegin(), MatAssemblyEnd(), MatSetValuesBlocked(), MatSetValuesLocal(),
1144           InsertMode, INSERT_VALUES, ADD_VALUES, MatSetValues(), MatSetValuesRow(), MatSetLocalToGlobalMapping()
1145 @*/
1146 PetscErrorCode  MatSetValuesRowLocal(Mat mat,PetscInt row,const PetscScalar v[])
1147 {
1148   PetscErrorCode ierr;
1149   PetscInt       globalrow;
1150 
1151   PetscFunctionBegin;
1152   PetscValidHeaderSpecific(mat,MAT_CLASSID,1);
1153   PetscValidType(mat,1);
1154   PetscValidScalarPointer(v,2);
1155   ierr = ISLocalToGlobalMappingApply(mat->rmap->mapping,1,&row,&globalrow);CHKERRQ(ierr);
1156   ierr = MatSetValuesRow(mat,globalrow,v);CHKERRQ(ierr);
1157 #if defined(PETSC_HAVE_CUSP)
1158   if (mat->valid_GPU_matrix != PETSC_CUSP_UNALLOCATED) {
1159     mat->valid_GPU_matrix = PETSC_CUSP_CPU;
1160   }
1161 #endif
1162   PetscFunctionReturn(0);
1163 }
1164 
1165 #undef __FUNCT__
1166 #define __FUNCT__ "MatSetValuesRow"
1167 /*@
1168    MatSetValuesRow - Inserts a row (block row for BAIJ matrices) of nonzero
1169         values into a matrix
1170 
1171    Not Collective
1172 
1173    Input Parameters:
1174 +  mat - the matrix
1175 .  row - the (block) row to set
1176 -  v - a logically two-dimensional array of values
1177 
1178    Notes:
1179    The values, v, are column-oriented for the block version.
1180 
1181    All the nonzeros in the row must be provided
1182 
1183    THE MATRIX MUSAT HAVE PREVIOUSLY HAD ITS COLUMN INDICES SET. IT IS RARE THAT THIS ROUTINE IS USED, usually MatSetValues() is used.
1184 
1185    The row must belong to this process
1186 
1187    Level: advanced
1188 
1189    Concepts: matrices^putting entries in
1190 
1191 .seealso: MatSetOption(), MatAssemblyBegin(), MatAssemblyEnd(), MatSetValuesBlocked(), MatSetValuesLocal(),
1192           InsertMode, INSERT_VALUES, ADD_VALUES, MatSetValues()
1193 @*/
1194 PetscErrorCode  MatSetValuesRow(Mat mat,PetscInt row,const PetscScalar v[])
1195 {
1196   PetscErrorCode ierr;
1197 
1198   PetscFunctionBegin;
1199   PetscValidHeaderSpecific(mat,MAT_CLASSID,1);
1200   PetscValidType(mat,1);
1201   MatCheckPreallocated(mat,1);
1202   PetscValidScalarPointer(v,2);
1203 #if defined(PETSC_USE_DEBUG)
1204   if (mat->insertmode == ADD_VALUES) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONGSTATE,"Cannot mix add and insert values");
1205   if (mat->factortype) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONGSTATE,"Not for factored matrix");
1206 #endif
1207   mat->insertmode = INSERT_VALUES;
1208 
1209   if (mat->assembled) {
1210     mat->was_assembled = PETSC_TRUE;
1211     mat->assembled     = PETSC_FALSE;
1212   }
1213   ierr = PetscLogEventBegin(MAT_SetValues,mat,0,0,0);CHKERRQ(ierr);
1214   if (!mat->ops->setvaluesrow) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_SUP,"Mat type %s",((PetscObject)mat)->type_name);
1215   ierr = (*mat->ops->setvaluesrow)(mat,row,v);CHKERRQ(ierr);
1216   ierr = PetscLogEventEnd(MAT_SetValues,mat,0,0,0);CHKERRQ(ierr);
1217 #if defined(PETSC_HAVE_CUSP)
1218   if (mat->valid_GPU_matrix != PETSC_CUSP_UNALLOCATED) {
1219     mat->valid_GPU_matrix = PETSC_CUSP_CPU;
1220   }
1221 #endif
1222   PetscFunctionReturn(0);
1223 }
1224 
1225 #undef __FUNCT__
1226 #define __FUNCT__ "MatSetValuesStencil"
1227 /*@
1228    MatSetValuesStencil - Inserts or adds a block of values into a matrix.
1229      Using structured grid indexing
1230 
1231    Not Collective
1232 
1233    Input Parameters:
1234 +  mat - the matrix
1235 .  m - number of rows being entered
1236 .  idxm - grid coordinates (and component number when dof > 1) for matrix rows being entered
1237 .  n - number of columns being entered
1238 .  idxn - grid coordinates (and component number when dof > 1) for matrix columns being entered
1239 .  v - a logically two-dimensional array of values
1240 -  addv - either ADD_VALUES or INSERT_VALUES, where
1241    ADD_VALUES adds values to any existing entries, and
1242    INSERT_VALUES replaces existing entries with new values
1243 
1244    Notes:
1245    By default the values, v, are row-oriented.  See MatSetOption() for other options.
1246 
1247    Calls to MatSetValuesStencil() with the INSERT_VALUES and ADD_VALUES
1248    options cannot be mixed without intervening calls to the assembly
1249    routines.
1250 
1251    The grid coordinates are across the entire grid, not just the local portion
1252 
1253    MatSetValuesStencil() uses 0-based row and column numbers in Fortran
1254    as well as in C.
1255 
1256    For setting/accessing vector values via array coordinates you can use the DMDAVecGetArray() routine
1257 
1258    In order to use this routine you must either obtain the matrix with DMCreateMatrix()
1259    or call MatSetLocalToGlobalMapping() and MatSetStencil() first.
1260 
1261    The columns and rows in the stencil passed in MUST be contained within the
1262    ghost region of the given process as set with DMDACreateXXX() or MatSetStencil(). For example,
1263    if you create a DMDA with an overlap of one grid level and on a particular process its first
1264    local nonghost x logical coordinate is 6 (so its first ghost x logical coordinate is 5) the
1265    first i index you can use in your column and row indices in MatSetStencil() is 5.
1266 
1267    In Fortran idxm and idxn should be declared as
1268 $     MatStencil idxm(4,m),idxn(4,n)
1269    and the values inserted using
1270 $    idxm(MatStencil_i,1) = i
1271 $    idxm(MatStencil_j,1) = j
1272 $    idxm(MatStencil_k,1) = k
1273 $    idxm(MatStencil_c,1) = c
1274    etc
1275 
1276    For periodic boundary conditions use negative indices for values to the left (below 0; that are to be
1277    obtained by wrapping values from right edge). For values to the right of the last entry using that index plus one
1278    etc to obtain values that obtained by wrapping the values from the left edge. This does not work for anything but the
1279    DMDA_BOUNDARY_PERIODIC boundary type.
1280 
1281    For indices that don't mean anything for your case (like the k index when working in 2d) or the c index when you have
1282    a single value per point) you can skip filling those indices.
1283 
1284    Inspired by the structured grid interface to the HYPRE package
1285    (http://www.llnl.gov/CASC/hypre)
1286 
1287    Efficiency Alert:
1288    The routine MatSetValuesBlockedStencil() may offer much better efficiency
1289    for users of block sparse formats (MATSEQBAIJ and MATMPIBAIJ).
1290 
1291    Level: beginner
1292 
1293    Concepts: matrices^putting entries in
1294 
1295 .seealso: MatSetOption(), MatAssemblyBegin(), MatAssemblyEnd(), MatSetValuesBlocked(), MatSetValuesLocal()
1296           MatSetValues(), MatSetValuesBlockedStencil(), MatSetStencil(), DMCreateMatrix(), DMDAVecGetArray(), MatStencil
1297 @*/
1298 PetscErrorCode  MatSetValuesStencil(Mat mat,PetscInt m,const MatStencil idxm[],PetscInt n,const MatStencil idxn[],const PetscScalar v[],InsertMode addv)
1299 {
1300   PetscErrorCode ierr;
1301   PetscInt       buf[8192],*bufm=0,*bufn=0,*jdxm,*jdxn;
1302   PetscInt       j,i,dim = mat->stencil.dim,*dims = mat->stencil.dims+1,tmp;
1303   PetscInt       *starts = mat->stencil.starts,*dxm = (PetscInt*)idxm,*dxn = (PetscInt*)idxn,sdim = dim - (1 - (PetscInt)mat->stencil.noc);
1304 
1305   PetscFunctionBegin;
1306   if (!m || !n) PetscFunctionReturn(0); /* no values to insert */
1307   PetscValidHeaderSpecific(mat,MAT_CLASSID,1);
1308   PetscValidType(mat,1);
1309   PetscValidIntPointer(idxm,3);
1310   PetscValidIntPointer(idxn,5);
1311   PetscValidScalarPointer(v,6);
1312 
1313   if ((m+n) <= (PetscInt)(sizeof(buf)/sizeof(PetscInt))) {
1314     jdxm = buf; jdxn = buf+m;
1315   } else {
1316     ierr = PetscMalloc2(m,PetscInt,&bufm,n,PetscInt,&bufn);CHKERRQ(ierr);
1317     jdxm = bufm; jdxn = bufn;
1318   }
1319   for (i=0; i<m; i++) {
1320     for (j=0; j<3-sdim; j++) dxm++;
1321     tmp = *dxm++ - starts[0];
1322     for (j=0; j<dim-1; j++) {
1323       if ((*dxm++ - starts[j+1]) < 0 || tmp < 0) tmp = -1;
1324       else                                       tmp = tmp*dims[j] + *(dxm-1) - starts[j+1];
1325     }
1326     if (mat->stencil.noc) dxm++;
1327     jdxm[i] = tmp;
1328   }
1329   for (i=0; i<n; i++) {
1330     for (j=0; j<3-sdim; j++) dxn++;
1331     tmp = *dxn++ - starts[0];
1332     for (j=0; j<dim-1; j++) {
1333       if ((*dxn++ - starts[j+1]) < 0 || tmp < 0) tmp = -1;
1334       else                                       tmp = tmp*dims[j] + *(dxn-1) - starts[j+1];
1335     }
1336     if (mat->stencil.noc) dxn++;
1337     jdxn[i] = tmp;
1338   }
1339   ierr = MatSetValuesLocal(mat,m,jdxm,n,jdxn,v,addv);CHKERRQ(ierr);
1340   ierr = PetscFree2(bufm,bufn);CHKERRQ(ierr);
1341   PetscFunctionReturn(0);
1342 }
1343 
1344 #undef __FUNCT__
1345 #define __FUNCT__ "MatSetValuesBlockedStencil"
1346 /*@C
1347    MatSetValuesBlockedStencil - Inserts or adds a block of values into a matrix.
1348      Using structured grid indexing
1349 
1350    Not Collective
1351 
1352    Input Parameters:
1353 +  mat - the matrix
1354 .  m - number of rows being entered
1355 .  idxm - grid coordinates for matrix rows being entered
1356 .  n - number of columns being entered
1357 .  idxn - grid coordinates for matrix columns being entered
1358 .  v - a logically two-dimensional array of values
1359 -  addv - either ADD_VALUES or INSERT_VALUES, where
1360    ADD_VALUES adds values to any existing entries, and
1361    INSERT_VALUES replaces existing entries with new values
1362 
1363    Notes:
1364    By default the values, v, are row-oriented and unsorted.
1365    See MatSetOption() for other options.
1366 
1367    Calls to MatSetValuesBlockedStencil() with the INSERT_VALUES and ADD_VALUES
1368    options cannot be mixed without intervening calls to the assembly
1369    routines.
1370 
1371    The grid coordinates are across the entire grid, not just the local portion
1372 
1373    MatSetValuesBlockedStencil() uses 0-based row and column numbers in Fortran
1374    as well as in C.
1375 
1376    For setting/accessing vector values via array coordinates you can use the DMDAVecGetArray() routine
1377 
1378    In order to use this routine you must either obtain the matrix with DMCreateMatrix()
1379    or call MatSetBlockSize(), MatSetLocalToGlobalMapping() and MatSetStencil() first.
1380 
1381    The columns and rows in the stencil passed in MUST be contained within the
1382    ghost region of the given process as set with DMDACreateXXX() or MatSetStencil(). For example,
1383    if you create a DMDA with an overlap of one grid level and on a particular process its first
1384    local nonghost x logical coordinate is 6 (so its first ghost x logical coordinate is 5) the
1385    first i index you can use in your column and row indices in MatSetStencil() is 5.
1386 
1387    In Fortran idxm and idxn should be declared as
1388 $     MatStencil idxm(4,m),idxn(4,n)
1389    and the values inserted using
1390 $    idxm(MatStencil_i,1) = i
1391 $    idxm(MatStencil_j,1) = j
1392 $    idxm(MatStencil_k,1) = k
1393    etc
1394 
1395    Negative indices may be passed in idxm and idxn, these rows and columns are
1396    simply ignored. This allows easily inserting element stiffness matrices
1397    with homogeneous Dirchlet boundary conditions that you don't want represented
1398    in the matrix.
1399 
1400    Inspired by the structured grid interface to the HYPRE package
1401    (http://www.llnl.gov/CASC/hypre)
1402 
1403    Level: beginner
1404 
1405    Concepts: matrices^putting entries in
1406 
1407 .seealso: MatSetOption(), MatAssemblyBegin(), MatAssemblyEnd(), MatSetValuesBlocked(), MatSetValuesLocal()
1408           MatSetValues(), MatSetValuesStencil(), MatSetStencil(), DMCreateMatrix(), DMDAVecGetArray(), MatStencil,
1409           MatSetBlockSize(), MatSetLocalToGlobalMapping()
1410 @*/
1411 PetscErrorCode  MatSetValuesBlockedStencil(Mat mat,PetscInt m,const MatStencil idxm[],PetscInt n,const MatStencil idxn[],const PetscScalar v[],InsertMode addv)
1412 {
1413   PetscErrorCode ierr;
1414   PetscInt       buf[8192],*bufm=0,*bufn=0,*jdxm,*jdxn;
1415   PetscInt       j,i,dim = mat->stencil.dim,*dims = mat->stencil.dims+1,tmp;
1416   PetscInt       *starts = mat->stencil.starts,*dxm = (PetscInt*)idxm,*dxn = (PetscInt*)idxn,sdim = dim - (1 - (PetscInt)mat->stencil.noc);
1417 
1418   PetscFunctionBegin;
1419   if (!m || !n) PetscFunctionReturn(0); /* no values to insert */
1420   PetscValidHeaderSpecific(mat,MAT_CLASSID,1);
1421   PetscValidType(mat,1);
1422   PetscValidIntPointer(idxm,3);
1423   PetscValidIntPointer(idxn,5);
1424   PetscValidScalarPointer(v,6);
1425 
1426   if ((m+n) <= (PetscInt)(sizeof(buf)/sizeof(PetscInt))) {
1427     jdxm = buf; jdxn = buf+m;
1428   } else {
1429     ierr = PetscMalloc2(m,PetscInt,&bufm,n,PetscInt,&bufn);CHKERRQ(ierr);
1430     jdxm = bufm; jdxn = bufn;
1431   }
1432   for (i=0; i<m; i++) {
1433     for (j=0; j<3-sdim; j++) dxm++;
1434     tmp = *dxm++ - starts[0];
1435     for (j=0; j<sdim-1; j++) {
1436       if ((*dxm++ - starts[j+1]) < 0 || tmp < 0) tmp = -1;
1437       else                                       tmp = tmp*dims[j] + *(dxm-1) - starts[j+1];
1438     }
1439     dxm++;
1440     jdxm[i] = tmp;
1441   }
1442   for (i=0; i<n; i++) {
1443     for (j=0; j<3-sdim; j++) dxn++;
1444     tmp = *dxn++ - starts[0];
1445     for (j=0; j<sdim-1; j++) {
1446       if ((*dxn++ - starts[j+1]) < 0 || tmp < 0) tmp = -1;
1447       else                                       tmp = tmp*dims[j] + *(dxn-1) - starts[j+1];
1448     }
1449     dxn++;
1450     jdxn[i] = tmp;
1451   }
1452   ierr = MatSetValuesBlockedLocal(mat,m,jdxm,n,jdxn,v,addv);CHKERRQ(ierr);
1453   ierr = PetscFree2(bufm,bufn);CHKERRQ(ierr);
1454 #if defined(PETSC_HAVE_CUSP)
1455   if (mat->valid_GPU_matrix != PETSC_CUSP_UNALLOCATED) {
1456     mat->valid_GPU_matrix = PETSC_CUSP_CPU;
1457   }
1458 #endif
1459   PetscFunctionReturn(0);
1460 }
1461 
1462 #undef __FUNCT__
1463 #define __FUNCT__ "MatSetStencil"
1464 /*@
1465    MatSetStencil - Sets the grid information for setting values into a matrix via
1466         MatSetValuesStencil()
1467 
1468    Not Collective
1469 
1470    Input Parameters:
1471 +  mat - the matrix
1472 .  dim - dimension of the grid 1, 2, or 3
1473 .  dims - number of grid points in x, y, and z direction, including ghost points on your processor
1474 .  starts - starting point of ghost nodes on your processor in x, y, and z direction
1475 -  dof - number of degrees of freedom per node
1476 
1477 
1478    Inspired by the structured grid interface to the HYPRE package
1479    (www.llnl.gov/CASC/hyper)
1480 
1481    For matrices generated with DMCreateMatrix() this routine is automatically called and so not needed by the
1482    user.
1483 
1484    Level: beginner
1485 
1486    Concepts: matrices^putting entries in
1487 
1488 .seealso: MatSetOption(), MatAssemblyBegin(), MatAssemblyEnd(), MatSetValuesBlocked(), MatSetValuesLocal()
1489           MatSetValues(), MatSetValuesBlockedStencil(), MatSetValuesStencil()
1490 @*/
1491 PetscErrorCode  MatSetStencil(Mat mat,PetscInt dim,const PetscInt dims[],const PetscInt starts[],PetscInt dof)
1492 {
1493   PetscInt i;
1494 
1495   PetscFunctionBegin;
1496   PetscValidHeaderSpecific(mat,MAT_CLASSID,1);
1497   PetscValidIntPointer(dims,3);
1498   PetscValidIntPointer(starts,4);
1499 
1500   mat->stencil.dim = dim + (dof > 1);
1501   for (i=0; i<dim; i++) {
1502     mat->stencil.dims[i]   = dims[dim-i-1];      /* copy the values in backwards */
1503     mat->stencil.starts[i] = starts[dim-i-1];
1504   }
1505   mat->stencil.dims[dim]   = dof;
1506   mat->stencil.starts[dim] = 0;
1507   mat->stencil.noc         = (PetscBool)(dof == 1);
1508   PetscFunctionReturn(0);
1509 }
1510 
1511 #undef __FUNCT__
1512 #define __FUNCT__ "MatSetValuesBlocked"
1513 /*@
1514    MatSetValuesBlocked - Inserts or adds a block of values into a matrix.
1515 
1516    Not Collective
1517 
1518    Input Parameters:
1519 +  mat - the matrix
1520 .  v - a logically two-dimensional array of values
1521 .  m, idxm - the number of block rows and their global block indices
1522 .  n, idxn - the number of block columns and their global block indices
1523 -  addv - either ADD_VALUES or INSERT_VALUES, where
1524    ADD_VALUES adds values to any existing entries, and
1525    INSERT_VALUES replaces existing entries with new values
1526 
1527    Notes:
1528    If you create the matrix yourself (that is not with a call to DMCreateMatrix()) then you MUST call
1529    MatXXXXSetPreallocation() or MatSetUp() before using this routine.
1530 
1531    The m and n count the NUMBER of blocks in the row direction and column direction,
1532    NOT the total number of rows/columns; for example, if the block size is 2 and
1533    you are passing in values for rows 2,3,4,5  then m would be 2 (not 4).
1534    The values in idxm would be 1 2; that is the first index for each block divided by
1535    the block size.
1536 
1537    Note that you must call MatSetBlockSize() when constructing this matrix (after
1538    preallocating it).
1539 
1540    By default the values, v, are row-oriented, so the layout of
1541    v is the same as for MatSetValues(). See MatSetOption() for other options.
1542 
1543    Calls to MatSetValuesBlocked() with the INSERT_VALUES and ADD_VALUES
1544    options cannot be mixed without intervening calls to the assembly
1545    routines.
1546 
1547    MatSetValuesBlocked() uses 0-based row and column numbers in Fortran
1548    as well as in C.
1549 
1550    Negative indices may be passed in idxm and idxn, these rows and columns are
1551    simply ignored. This allows easily inserting element stiffness matrices
1552    with homogeneous Dirchlet boundary conditions that you don't want represented
1553    in the matrix.
1554 
1555    Each time an entry is set within a sparse matrix via MatSetValues(),
1556    internal searching must be done to determine where to place the the
1557    data in the matrix storage space.  By instead inserting blocks of
1558    entries via MatSetValuesBlocked(), the overhead of matrix assembly is
1559    reduced.
1560 
1561    Example:
1562 $   Suppose m=n=2 and block size(bs) = 2 The array is
1563 $
1564 $   1  2  | 3  4
1565 $   5  6  | 7  8
1566 $   - - - | - - -
1567 $   9  10 | 11 12
1568 $   13 14 | 15 16
1569 $
1570 $   v[] should be passed in like
1571 $   v[] = [1,2,3,4,5,6,7,8,9,10,11,12,13,14,15,16]
1572 $
1573 $  If you are not using row oriented storage of v (that is you called MatSetOption(mat,MAT_ROW_ORIENTED,PETSC_FALSE)) then
1574 $   v[] = [1,5,9,13,2,6,10,14,3,7,11,15,4,8,12,16]
1575 
1576    Level: intermediate
1577 
1578    Concepts: matrices^putting entries in blocked
1579 
1580 .seealso: MatSetBlockSize(), MatSetOption(), MatAssemblyBegin(), MatAssemblyEnd(), MatSetValues(), MatSetValuesBlockedLocal()
1581 @*/
1582 PetscErrorCode  MatSetValuesBlocked(Mat mat,PetscInt m,const PetscInt idxm[],PetscInt n,const PetscInt idxn[],const PetscScalar v[],InsertMode addv)
1583 {
1584   PetscErrorCode ierr;
1585 
1586   PetscFunctionBegin;
1587   PetscValidHeaderSpecific(mat,MAT_CLASSID,1);
1588   PetscValidType(mat,1);
1589   if (!m || !n) PetscFunctionReturn(0); /* no values to insert */
1590   PetscValidIntPointer(idxm,3);
1591   PetscValidIntPointer(idxn,5);
1592   PetscValidScalarPointer(v,6);
1593   MatCheckPreallocated(mat,1);
1594   if (mat->insertmode == NOT_SET_VALUES) {
1595     mat->insertmode = addv;
1596   }
1597 #if defined(PETSC_USE_DEBUG)
1598   else if (mat->insertmode != addv) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONGSTATE,"Cannot mix add values and insert values");
1599   if (mat->factortype) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONGSTATE,"Not for factored matrix");
1600   if (!mat->ops->setvaluesblocked && !mat->ops->setvalues) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_SUP,"Mat type %s",((PetscObject)mat)->type_name);
1601 #endif
1602 
1603   if (mat->assembled) {
1604     mat->was_assembled = PETSC_TRUE;
1605     mat->assembled     = PETSC_FALSE;
1606   }
1607   ierr = PetscLogEventBegin(MAT_SetValues,mat,0,0,0);CHKERRQ(ierr);
1608   if (mat->ops->setvaluesblocked) {
1609     ierr = (*mat->ops->setvaluesblocked)(mat,m,idxm,n,idxn,v,addv);CHKERRQ(ierr);
1610   } else {
1611     PetscInt buf[8192],*bufr=0,*bufc=0,*iidxm,*iidxn;
1612     PetscInt i,j,bs=mat->rmap->bs;
1613     if ((m+n)*bs <= (PetscInt)(sizeof(buf)/sizeof(PetscInt))) {
1614       iidxm = buf; iidxn = buf + m*bs;
1615     } else {
1616       ierr  = PetscMalloc2(m*bs,PetscInt,&bufr,n*bs,PetscInt,&bufc);CHKERRQ(ierr);
1617       iidxm = bufr; iidxn = bufc;
1618     }
1619     for (i=0; i<m; i++) {
1620       for (j=0; j<bs; j++) {
1621         iidxm[i*bs+j] = bs*idxm[i] + j;
1622       }
1623     }
1624     for (i=0; i<n; i++) {
1625       for (j=0; j<bs; j++) {
1626         iidxn[i*bs+j] = bs*idxn[i] + j;
1627       }
1628     }
1629     ierr = MatSetValues(mat,m*bs,iidxm,n*bs,iidxn,v,addv);CHKERRQ(ierr);
1630     ierr = PetscFree2(bufr,bufc);CHKERRQ(ierr);
1631   }
1632   ierr = PetscLogEventEnd(MAT_SetValues,mat,0,0,0);CHKERRQ(ierr);
1633 #if defined(PETSC_HAVE_CUSP)
1634   if (mat->valid_GPU_matrix != PETSC_CUSP_UNALLOCATED) {
1635     mat->valid_GPU_matrix = PETSC_CUSP_CPU;
1636   }
1637 #endif
1638   PetscFunctionReturn(0);
1639 }
1640 
1641 #undef __FUNCT__
1642 #define __FUNCT__ "MatGetValues"
1643 /*@
1644    MatGetValues - Gets a block of values from a matrix.
1645 
1646    Not Collective; currently only returns a local block
1647 
1648    Input Parameters:
1649 +  mat - the matrix
1650 .  v - a logically two-dimensional array for storing the values
1651 .  m, idxm - the number of rows and their global indices
1652 -  n, idxn - the number of columns and their global indices
1653 
1654    Notes:
1655    The user must allocate space (m*n PetscScalars) for the values, v.
1656    The values, v, are then returned in a row-oriented format,
1657    analogous to that used by default in MatSetValues().
1658 
1659    MatGetValues() uses 0-based row and column numbers in
1660    Fortran as well as in C.
1661 
1662    MatGetValues() requires that the matrix has been assembled
1663    with MatAssemblyBegin()/MatAssemblyEnd().  Thus, calls to
1664    MatSetValues() and MatGetValues() CANNOT be made in succession
1665    without intermediate matrix assembly.
1666 
1667    Negative row or column indices will be ignored and those locations in v[] will be
1668    left unchanged.
1669 
1670    Level: advanced
1671 
1672    Concepts: matrices^accessing values
1673 
1674 .seealso: MatGetRow(), MatGetSubMatrices(), MatSetValues()
1675 @*/
1676 PetscErrorCode  MatGetValues(Mat mat,PetscInt m,const PetscInt idxm[],PetscInt n,const PetscInt idxn[],PetscScalar v[])
1677 {
1678   PetscErrorCode ierr;
1679 
1680   PetscFunctionBegin;
1681   PetscValidHeaderSpecific(mat,MAT_CLASSID,1);
1682   PetscValidType(mat,1);
1683   if (!m || !n) PetscFunctionReturn(0);
1684   PetscValidIntPointer(idxm,3);
1685   PetscValidIntPointer(idxn,5);
1686   PetscValidScalarPointer(v,6);
1687   if (!mat->assembled) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONGSTATE,"Not for unassembled matrix");
1688   if (mat->factortype) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONGSTATE,"Not for factored matrix");
1689   if (!mat->ops->getvalues) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_SUP,"Mat type %s",((PetscObject)mat)->type_name);
1690   MatCheckPreallocated(mat,1);
1691 
1692   ierr = PetscLogEventBegin(MAT_GetValues,mat,0,0,0);CHKERRQ(ierr);
1693   ierr = (*mat->ops->getvalues)(mat,m,idxm,n,idxn,v);CHKERRQ(ierr);
1694   ierr = PetscLogEventEnd(MAT_GetValues,mat,0,0,0);CHKERRQ(ierr);
1695   PetscFunctionReturn(0);
1696 }
1697 
1698 #undef __FUNCT__
1699 #define __FUNCT__ "MatSetValuesBatch"
1700 /*@
1701   MatSetValuesBatch - Adds (ADD_VALUES) many blocks of values into a matrix at once. The blocks must all be square and
1702   the same size. Currently, this can only be called once and creates the given matrix.
1703 
1704   Not Collective
1705 
1706   Input Parameters:
1707 + mat - the matrix
1708 . nb - the number of blocks
1709 . bs - the number of rows (and columns) in each block
1710 . rows - a concatenation of the rows for each block
1711 - v - a concatenation of logically two-dimensional arrays of values
1712 
1713   Notes:
1714   In the future, we will extend this routine to handle rectangular blocks, and to allow multiple calls for a given matrix.
1715 
1716   Level: advanced
1717 
1718   Concepts: matrices^putting entries in
1719 
1720 .seealso: MatSetOption(), MatAssemblyBegin(), MatAssemblyEnd(), MatSetValuesBlocked(), MatSetValuesLocal(),
1721           InsertMode, INSERT_VALUES, ADD_VALUES, MatSetValues()
1722 @*/
1723 PetscErrorCode MatSetValuesBatch(Mat mat, PetscInt nb, PetscInt bs, PetscInt rows[], const PetscScalar v[])
1724 {
1725   PetscErrorCode ierr;
1726 
1727   PetscFunctionBegin;
1728   PetscValidHeaderSpecific(mat,MAT_CLASSID,1);
1729   PetscValidType(mat,1);
1730   PetscValidScalarPointer(rows,4);
1731   PetscValidScalarPointer(v,5);
1732 #if defined(PETSC_USE_DEBUG)
1733   if (mat->factortype) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONGSTATE,"Not for factored matrix");
1734 #endif
1735 
1736   ierr = PetscLogEventBegin(MAT_SetValuesBatch,mat,0,0,0);CHKERRQ(ierr);
1737   if (mat->ops->setvaluesbatch) {
1738     ierr = (*mat->ops->setvaluesbatch)(mat,nb,bs,rows,v);CHKERRQ(ierr);
1739   } else {
1740     PetscInt b;
1741     for (b = 0; b < nb; ++b) {
1742       ierr = MatSetValues(mat, bs, &rows[b*bs], bs, &rows[b*bs], &v[b*bs*bs], ADD_VALUES);CHKERRQ(ierr);
1743     }
1744   }
1745   ierr = PetscLogEventEnd(MAT_SetValuesBatch,mat,0,0,0);CHKERRQ(ierr);
1746   PetscFunctionReturn(0);
1747 }
1748 
1749 #undef __FUNCT__
1750 #define __FUNCT__ "MatSetLocalToGlobalMapping"
1751 /*@
1752    MatSetLocalToGlobalMapping - Sets a local-to-global numbering for use by
1753    the routine MatSetValuesLocal() to allow users to insert matrix entries
1754    using a local (per-processor) numbering.
1755 
1756    Not Collective
1757 
1758    Input Parameters:
1759 +  x - the matrix
1760 .  rmapping - row mapping created with ISLocalToGlobalMappingCreate()
1761              or ISLocalToGlobalMappingCreateIS()
1762 - cmapping - column mapping
1763 
1764    Level: intermediate
1765 
1766    Concepts: matrices^local to global mapping
1767    Concepts: local to global mapping^for matrices
1768 
1769 .seealso:  MatAssemblyBegin(), MatAssemblyEnd(), MatSetValues(), MatSetValuesLocal()
1770 @*/
1771 PetscErrorCode  MatSetLocalToGlobalMapping(Mat x,ISLocalToGlobalMapping rmapping,ISLocalToGlobalMapping cmapping)
1772 {
1773   PetscErrorCode ierr;
1774 
1775   PetscFunctionBegin;
1776   PetscValidHeaderSpecific(x,MAT_CLASSID,1);
1777   PetscValidType(x,1);
1778   PetscValidHeaderSpecific(rmapping,IS_LTOGM_CLASSID,2);
1779   PetscValidHeaderSpecific(cmapping,IS_LTOGM_CLASSID,3);
1780 
1781   if (x->ops->setlocaltoglobalmapping) {
1782     ierr = (*x->ops->setlocaltoglobalmapping)(x,rmapping,cmapping);CHKERRQ(ierr);
1783   } else {
1784     ierr = PetscLayoutSetISLocalToGlobalMapping(x->rmap,rmapping);CHKERRQ(ierr);
1785     ierr = PetscLayoutSetISLocalToGlobalMapping(x->cmap,cmapping);CHKERRQ(ierr);
1786   }
1787   PetscFunctionReturn(0);
1788 }
1789 
1790 #undef __FUNCT__
1791 #define __FUNCT__ "MatSetLocalToGlobalMappingBlock"
1792 /*@
1793    MatSetLocalToGlobalMappingBlock - Sets a local-to-global numbering for use
1794    by the routine MatSetValuesBlockedLocal() to allow users to insert matrix
1795    entries using a local (per-processor) numbering.
1796 
1797    Not Collective
1798 
1799    Input Parameters:
1800 +  x - the matrix
1801 . rmapping - row mapping created with ISLocalToGlobalMappingCreate() or
1802              ISLocalToGlobalMappingCreateIS()
1803 - cmapping - column mapping
1804 
1805    Level: intermediate
1806 
1807    Concepts: matrices^local to global mapping blocked
1808    Concepts: local to global mapping^for matrices, blocked
1809 
1810 .seealso:  MatAssemblyBegin(), MatAssemblyEnd(), MatSetValues(), MatSetValuesBlockedLocal(),
1811            MatSetValuesBlocked(), MatSetValuesLocal()
1812 @*/
1813 PetscErrorCode  MatSetLocalToGlobalMappingBlock(Mat x,ISLocalToGlobalMapping rmapping,ISLocalToGlobalMapping cmapping)
1814 {
1815   PetscErrorCode ierr;
1816 
1817   PetscFunctionBegin;
1818   PetscValidHeaderSpecific(x,MAT_CLASSID,1);
1819   PetscValidType(x,1);
1820   PetscValidHeaderSpecific(rmapping,IS_LTOGM_CLASSID,2);
1821   PetscValidHeaderSpecific(cmapping,IS_LTOGM_CLASSID,3);
1822 
1823   ierr = PetscLayoutSetISLocalToGlobalMappingBlock(x->rmap,rmapping);CHKERRQ(ierr);
1824   ierr = PetscLayoutSetISLocalToGlobalMappingBlock(x->cmap,cmapping);CHKERRQ(ierr);
1825   PetscFunctionReturn(0);
1826 }
1827 
1828 #undef __FUNCT__
1829 #define __FUNCT__ "MatGetLocalToGlobalMapping"
1830 /*@
1831    MatGetLocalToGlobalMapping - Gets the local-to-global numbering set by MatSetLocalToGlobalMapping()
1832 
1833    Not Collective
1834 
1835    Input Parameters:
1836 .  A - the matrix
1837 
1838    Output Parameters:
1839 + rmapping - row mapping
1840 - cmapping - column mapping
1841 
1842    Level: advanced
1843 
1844    Concepts: matrices^local to global mapping
1845    Concepts: local to global mapping^for matrices
1846 
1847 .seealso:  MatSetValuesLocal(), MatGetLocalToGlobalMappingBlock()
1848 @*/
1849 PetscErrorCode  MatGetLocalToGlobalMapping(Mat A,ISLocalToGlobalMapping *rmapping,ISLocalToGlobalMapping *cmapping)
1850 {
1851   PetscFunctionBegin;
1852   PetscValidHeaderSpecific(A,MAT_CLASSID,1);
1853   PetscValidType(A,1);
1854   if (rmapping) PetscValidPointer(rmapping,2);
1855   if (cmapping) PetscValidPointer(cmapping,3);
1856   if (rmapping) *rmapping = A->rmap->mapping;
1857   if (cmapping) *cmapping = A->cmap->mapping;
1858   PetscFunctionReturn(0);
1859 }
1860 
1861 #undef __FUNCT__
1862 #define __FUNCT__ "MatGetLocalToGlobalMappingBlock"
1863 /*@
1864    MatGetLocalToGlobalMappingBlock - Gets the local-to-global numbering set by MatSetLocalToGlobalMappingBlock()
1865 
1866    Not Collective
1867 
1868    Input Parameters:
1869 .  A - the matrix
1870 
1871    Output Parameters:
1872 + rmapping - row mapping
1873 - cmapping - column mapping
1874 
1875    Level: advanced
1876 
1877    Concepts: matrices^local to global mapping blocked
1878    Concepts: local to global mapping^for matrices, blocked
1879 
1880 .seealso:  MatSetValuesBlockedLocal(), MatGetLocalToGlobalMapping()
1881 @*/
1882 PetscErrorCode  MatGetLocalToGlobalMappingBlock(Mat A,ISLocalToGlobalMapping *rmapping,ISLocalToGlobalMapping *cmapping)
1883 {
1884   PetscFunctionBegin;
1885   PetscValidHeaderSpecific(A,MAT_CLASSID,1);
1886   PetscValidType(A,1);
1887   if (rmapping) PetscValidPointer(rmapping,2);
1888   if (cmapping) PetscValidPointer(cmapping,3);
1889   if (rmapping) *rmapping = A->rmap->bmapping;
1890   if (cmapping) *cmapping = A->cmap->bmapping;
1891   PetscFunctionReturn(0);
1892 }
1893 
1894 #undef __FUNCT__
1895 #define __FUNCT__ "MatSetValuesLocal"
1896 /*@
1897    MatSetValuesLocal - Inserts or adds values into certain locations of a matrix,
1898    using a local ordering of the nodes.
1899 
1900    Not Collective
1901 
1902    Input Parameters:
1903 +  x - the matrix
1904 .  nrow, irow - number of rows and their local indices
1905 .  ncol, icol - number of columns and their local indices
1906 .  y -  a logically two-dimensional array of values
1907 -  addv - either INSERT_VALUES or ADD_VALUES, where
1908    ADD_VALUES adds values to any existing entries, and
1909    INSERT_VALUES replaces existing entries with new values
1910 
1911    Notes:
1912    If you create the matrix yourself (that is not with a call to DMCreateMatrix()) then you MUST call MatXXXXSetPreallocation() or
1913       MatSetUp() before using this routine
1914 
1915    If you create the matrix yourself (that is not with a call to DMCreateMatrix()) then you MUST call MatSetLocalToGlobalMapping() before using this routine
1916 
1917    Calls to MatSetValuesLocal() with the INSERT_VALUES and ADD_VALUES
1918    options cannot be mixed without intervening calls to the assembly
1919    routines.
1920 
1921    These values may be cached, so MatAssemblyBegin() and MatAssemblyEnd()
1922    MUST be called after all calls to MatSetValuesLocal() have been completed.
1923 
1924    Level: intermediate
1925 
1926    Concepts: matrices^putting entries in with local numbering
1927 
1928 .seealso:  MatAssemblyBegin(), MatAssemblyEnd(), MatSetValues(), MatSetLocalToGlobalMapping(),
1929            MatSetValueLocal()
1930 @*/
1931 PetscErrorCode  MatSetValuesLocal(Mat mat,PetscInt nrow,const PetscInt irow[],PetscInt ncol,const PetscInt icol[],const PetscScalar y[],InsertMode addv)
1932 {
1933   PetscErrorCode ierr;
1934 
1935   PetscFunctionBegin;
1936   PetscValidHeaderSpecific(mat,MAT_CLASSID,1);
1937   PetscValidType(mat,1);
1938   MatCheckPreallocated(mat,1);
1939   if (!nrow || !ncol) PetscFunctionReturn(0); /* no values to insert */
1940   PetscValidIntPointer(irow,3);
1941   PetscValidIntPointer(icol,5);
1942   PetscValidScalarPointer(y,6);
1943   if (mat->insertmode == NOT_SET_VALUES) {
1944     mat->insertmode = addv;
1945   }
1946 #if defined(PETSC_USE_DEBUG)
1947   else if (mat->insertmode != addv) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONGSTATE,"Cannot mix add values and insert values");
1948   if (mat->factortype) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONGSTATE,"Not for factored matrix");
1949   if (!mat->ops->setvalueslocal && !mat->ops->setvalues) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_SUP,"Mat type %s",((PetscObject)mat)->type_name);
1950 #endif
1951 
1952   if (mat->assembled) {
1953     mat->was_assembled = PETSC_TRUE;
1954     mat->assembled     = PETSC_FALSE;
1955   }
1956   ierr = PetscLogEventBegin(MAT_SetValues,mat,0,0,0);CHKERRQ(ierr);
1957   if (mat->ops->setvalueslocal) {
1958     ierr = (*mat->ops->setvalueslocal)(mat,nrow,irow,ncol,icol,y,addv);CHKERRQ(ierr);
1959   } else {
1960     PetscInt buf[8192],*bufr=0,*bufc=0,*irowm,*icolm;
1961     if ((nrow+ncol) <= (PetscInt)(sizeof(buf)/sizeof(PetscInt))) {
1962       irowm = buf; icolm = buf+nrow;
1963     } else {
1964       ierr  = PetscMalloc2(nrow,PetscInt,&bufr,ncol,PetscInt,&bufc);CHKERRQ(ierr);
1965       irowm = bufr; icolm = bufc;
1966     }
1967     ierr = ISLocalToGlobalMappingApply(mat->rmap->mapping,nrow,irow,irowm);CHKERRQ(ierr);
1968     ierr = ISLocalToGlobalMappingApply(mat->cmap->mapping,ncol,icol,icolm);CHKERRQ(ierr);
1969     ierr = MatSetValues(mat,nrow,irowm,ncol,icolm,y,addv);CHKERRQ(ierr);
1970     ierr = PetscFree2(bufr,bufc);CHKERRQ(ierr);
1971   }
1972   ierr = PetscLogEventEnd(MAT_SetValues,mat,0,0,0);CHKERRQ(ierr);
1973 #if defined(PETSC_HAVE_CUSP)
1974   if (mat->valid_GPU_matrix != PETSC_CUSP_UNALLOCATED) {
1975     mat->valid_GPU_matrix = PETSC_CUSP_CPU;
1976   }
1977 #endif
1978   PetscFunctionReturn(0);
1979 }
1980 
1981 #undef __FUNCT__
1982 #define __FUNCT__ "MatSetValuesBlockedLocal"
1983 /*@
1984    MatSetValuesBlockedLocal - Inserts or adds values into certain locations of a matrix,
1985    using a local ordering of the nodes a block at a time.
1986 
1987    Not Collective
1988 
1989    Input Parameters:
1990 +  x - the matrix
1991 .  nrow, irow - number of rows and their local indices
1992 .  ncol, icol - number of columns and their local indices
1993 .  y -  a logically two-dimensional array of values
1994 -  addv - either INSERT_VALUES or ADD_VALUES, where
1995    ADD_VALUES adds values to any existing entries, and
1996    INSERT_VALUES replaces existing entries with new values
1997 
1998    Notes:
1999    If you create the matrix yourself (that is not with a call to DMCreateMatrix()) then you MUST call MatXXXXSetPreallocation() or
2000       MatSetUp() before using this routine
2001 
2002    If you create the matrix yourself (that is not with a call to DMCreateMatrix()) then you MUST call MatSetBlockSize() and MatSetLocalToGlobalMappingBlock()
2003       before using this routineBefore calling MatSetValuesLocal(), the user must first set the
2004 
2005    Calls to MatSetValuesBlockedLocal() with the INSERT_VALUES and ADD_VALUES
2006    options cannot be mixed without intervening calls to the assembly
2007    routines.
2008 
2009    These values may be cached, so MatAssemblyBegin() and MatAssemblyEnd()
2010    MUST be called after all calls to MatSetValuesBlockedLocal() have been completed.
2011 
2012    Level: intermediate
2013 
2014    Concepts: matrices^putting blocked values in with local numbering
2015 
2016 .seealso:  MatSetBlockSize(), MatSetLocalToGlobalMappingBlock(), MatAssemblyBegin(), MatAssemblyEnd(),
2017            MatSetValuesLocal(), MatSetLocalToGlobalMappingBlock(), MatSetValuesBlocked()
2018 @*/
2019 PetscErrorCode  MatSetValuesBlockedLocal(Mat mat,PetscInt nrow,const PetscInt irow[],PetscInt ncol,const PetscInt icol[],const PetscScalar y[],InsertMode addv)
2020 {
2021   PetscErrorCode ierr;
2022 
2023   PetscFunctionBegin;
2024   PetscValidHeaderSpecific(mat,MAT_CLASSID,1);
2025   PetscValidType(mat,1);
2026   MatCheckPreallocated(mat,1);
2027   if (!nrow || !ncol) PetscFunctionReturn(0); /* no values to insert */
2028   PetscValidIntPointer(irow,3);
2029   PetscValidIntPointer(icol,5);
2030   PetscValidScalarPointer(y,6);
2031   if (mat->insertmode == NOT_SET_VALUES) {
2032     mat->insertmode = addv;
2033   }
2034 #if defined(PETSC_USE_DEBUG)
2035   else if (mat->insertmode != addv) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONGSTATE,"Cannot mix add values and insert values");
2036   if (mat->factortype) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONGSTATE,"Not for factored matrix");
2037   if (!mat->ops->setvaluesblockedlocal && !mat->ops->setvaluesblocked && !mat->ops->setvalueslocal && !mat->ops->setvalues) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_SUP,"Mat type %s",((PetscObject)mat)->type_name);
2038 #endif
2039 
2040   if (mat->assembled) {
2041     mat->was_assembled = PETSC_TRUE;
2042     mat->assembled     = PETSC_FALSE;
2043   }
2044   ierr = PetscLogEventBegin(MAT_SetValues,mat,0,0,0);CHKERRQ(ierr);
2045   if (mat->ops->setvaluesblockedlocal) {
2046     ierr = (*mat->ops->setvaluesblockedlocal)(mat,nrow,irow,ncol,icol,y,addv);CHKERRQ(ierr);
2047   } else {
2048     PetscInt buf[8192],*bufr=0,*bufc=0,*irowm,*icolm;
2049     if (mat->rmap->bmapping && mat->cmap->bmapping) {
2050       if ((nrow+ncol) <= (PetscInt)(sizeof(buf)/sizeof(PetscInt))) {
2051         irowm = buf; icolm = buf + nrow;
2052       } else {
2053         ierr  = PetscMalloc2(nrow,PetscInt,&bufr,ncol,PetscInt,&bufc);CHKERRQ(ierr);
2054         irowm = bufr; icolm = bufc;
2055       }
2056       ierr = ISLocalToGlobalMappingApply(mat->rmap->bmapping,nrow,irow,irowm);CHKERRQ(ierr);
2057       ierr = ISLocalToGlobalMappingApply(mat->cmap->bmapping,ncol,icol,icolm);CHKERRQ(ierr);
2058       ierr = MatSetValuesBlocked(mat,nrow,irowm,ncol,icolm,y,addv);CHKERRQ(ierr);
2059       ierr = PetscFree2(bufr,bufc);CHKERRQ(ierr);
2060     } else {
2061       PetscInt i,j,bs=mat->rmap->bs;
2062       if ((nrow+ncol)*bs <=(PetscInt)(sizeof(buf)/sizeof(PetscInt))) {
2063         irowm = buf; icolm = buf + nrow;
2064       } else {
2065         ierr  = PetscMalloc2(nrow*bs,PetscInt,&bufr,ncol*bs,PetscInt,&bufc);CHKERRQ(ierr);
2066         irowm = bufr; icolm = bufc;
2067       }
2068       for (i=0; i<nrow; i++) {
2069         for (j=0; j<bs; j++) irowm[i*bs+j] = irow[i]*bs+j;
2070       }
2071       for (i=0; i<ncol; i++) {
2072         for (j=0; j<bs; j++) icolm[i*bs+j] = icol[i]*bs+j;
2073       }
2074       ierr = MatSetValuesLocal(mat,nrow*bs,irowm,ncol*bs,icolm,y,addv);CHKERRQ(ierr);
2075       ierr = PetscFree2(bufr,bufc);CHKERRQ(ierr);
2076     }
2077   }
2078   ierr = PetscLogEventEnd(MAT_SetValues,mat,0,0,0);CHKERRQ(ierr);
2079 #if defined(PETSC_HAVE_CUSP)
2080   if (mat->valid_GPU_matrix != PETSC_CUSP_UNALLOCATED) {
2081     mat->valid_GPU_matrix = PETSC_CUSP_CPU;
2082   }
2083 #endif
2084   PetscFunctionReturn(0);
2085 }
2086 
2087 #undef __FUNCT__
2088 #define __FUNCT__ "MatMultDiagonalBlock"
2089 /*@
2090    MatMultDiagonalBlock - Computes the matrix-vector product, y = Dx. Where D is defined by the inode or block structure of the diagonal
2091 
2092    Collective on Mat and Vec
2093 
2094    Input Parameters:
2095 +  mat - the matrix
2096 -  x   - the vector to be multiplied
2097 
2098    Output Parameters:
2099 .  y - the result
2100 
2101    Notes:
2102    The vectors x and y cannot be the same.  I.e., one cannot
2103    call MatMult(A,y,y).
2104 
2105    Level: developer
2106 
2107    Concepts: matrix-vector product
2108 
2109 .seealso: MatMultTranspose(), MatMultAdd(), MatMultTransposeAdd()
2110 @*/
2111 PetscErrorCode  MatMultDiagonalBlock(Mat mat,Vec x,Vec y)
2112 {
2113   PetscErrorCode ierr;
2114 
2115   PetscFunctionBegin;
2116   PetscValidHeaderSpecific(mat,MAT_CLASSID,1);
2117   PetscValidType(mat,1);
2118   PetscValidHeaderSpecific(x,VEC_CLASSID,2);
2119   PetscValidHeaderSpecific(y,VEC_CLASSID,3);
2120 
2121   if (!mat->assembled) SETERRQ(PetscObjectComm((PetscObject)mat),PETSC_ERR_ARG_WRONGSTATE,"Not for unassembled matrix");
2122   if (mat->factortype) SETERRQ(PetscObjectComm((PetscObject)mat),PETSC_ERR_ARG_WRONGSTATE,"Not for factored matrix");
2123   if (x == y) SETERRQ(PetscObjectComm((PetscObject)mat),PETSC_ERR_ARG_WRONGSTATE,"x and y must be different vectors");
2124   MatCheckPreallocated(mat,1);
2125 
2126   if (!mat->ops->multdiagonalblock) SETERRQ(PetscObjectComm((PetscObject)mat),PETSC_ERR_SUP,"This matrix type does not have a multiply defined");
2127   ierr = (*mat->ops->multdiagonalblock)(mat,x,y);CHKERRQ(ierr);
2128   ierr = PetscObjectStateIncrease((PetscObject)y);CHKERRQ(ierr);
2129   PetscFunctionReturn(0);
2130 }
2131 
2132 /* --------------------------------------------------------*/
2133 #undef __FUNCT__
2134 #define __FUNCT__ "MatMult"
2135 /*@
2136    MatMult - Computes the matrix-vector product, y = Ax.
2137 
2138    Neighbor-wise Collective on Mat and Vec
2139 
2140    Input Parameters:
2141 +  mat - the matrix
2142 -  x   - the vector to be multiplied
2143 
2144    Output Parameters:
2145 .  y - the result
2146 
2147    Notes:
2148    The vectors x and y cannot be the same.  I.e., one cannot
2149    call MatMult(A,y,y).
2150 
2151    Level: beginner
2152 
2153    Concepts: matrix-vector product
2154 
2155 .seealso: MatMultTranspose(), MatMultAdd(), MatMultTransposeAdd()
2156 @*/
2157 PetscErrorCode  MatMult(Mat mat,Vec x,Vec y)
2158 {
2159   PetscErrorCode ierr;
2160 
2161   PetscFunctionBegin;
2162   PetscValidHeaderSpecific(mat,MAT_CLASSID,1);
2163   PetscValidType(mat,1);
2164   PetscValidHeaderSpecific(x,VEC_CLASSID,2);
2165   PetscValidHeaderSpecific(y,VEC_CLASSID,3);
2166   if (!mat->assembled) SETERRQ(PetscObjectComm((PetscObject)mat),PETSC_ERR_ARG_WRONGSTATE,"Not for unassembled matrix");
2167   if (mat->factortype) SETERRQ(PetscObjectComm((PetscObject)mat),PETSC_ERR_ARG_WRONGSTATE,"Not for factored matrix");
2168   if (x == y) SETERRQ(PetscObjectComm((PetscObject)mat),PETSC_ERR_ARG_WRONGSTATE,"x and y must be different vectors");
2169 #if !defined(PETSC_HAVE_CONSTRAINTS)
2170   if (mat->cmap->N != x->map->N) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_ARG_SIZ,"Mat mat,Vec x: global dim %D %D",mat->cmap->N,x->map->N);
2171   if (mat->rmap->N != y->map->N) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_ARG_SIZ,"Mat mat,Vec y: global dim %D %D",mat->rmap->N,y->map->N);
2172   if (mat->rmap->n != y->map->n) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_ARG_SIZ,"Mat mat,Vec y: local dim %D %D",mat->rmap->n,y->map->n);
2173 #endif
2174   ierr = VecValidValues(x,2,PETSC_TRUE);CHKERRQ(ierr);
2175   MatCheckPreallocated(mat,1);
2176 
2177   if (!mat->ops->mult) SETERRQ(PetscObjectComm((PetscObject)mat),PETSC_ERR_SUP,"This matrix type does not have a multiply defined");
2178   ierr = PetscLogEventBegin(MAT_Mult,mat,x,y,0);CHKERRQ(ierr);
2179   ierr = (*mat->ops->mult)(mat,x,y);CHKERRQ(ierr);
2180   ierr = PetscLogEventEnd(MAT_Mult,mat,x,y,0);CHKERRQ(ierr);
2181   ierr = VecValidValues(y,3,PETSC_FALSE);CHKERRQ(ierr);
2182   PetscFunctionReturn(0);
2183 }
2184 
2185 #undef __FUNCT__
2186 #define __FUNCT__ "MatMultTranspose"
2187 /*@
2188    MatMultTranspose - Computes matrix transpose times a vector.
2189 
2190    Neighbor-wise Collective on Mat and Vec
2191 
2192    Input Parameters:
2193 +  mat - the matrix
2194 -  x   - the vector to be multilplied
2195 
2196    Output Parameters:
2197 .  y - the result
2198 
2199    Notes:
2200    The vectors x and y cannot be the same.  I.e., one cannot
2201    call MatMultTranspose(A,y,y).
2202 
2203    For complex numbers this does NOT compute the Hermitian (complex conjugate) transpose multiple,
2204    use MatMultHermitianTranspose()
2205 
2206    Level: beginner
2207 
2208    Concepts: matrix vector product^transpose
2209 
2210 .seealso: MatMult(), MatMultAdd(), MatMultTransposeAdd(), MatMultHermitianTranspose(), MatTranspose()
2211 @*/
2212 PetscErrorCode  MatMultTranspose(Mat mat,Vec x,Vec y)
2213 {
2214   PetscErrorCode ierr;
2215 
2216   PetscFunctionBegin;
2217   PetscValidHeaderSpecific(mat,MAT_CLASSID,1);
2218   PetscValidType(mat,1);
2219   PetscValidHeaderSpecific(x,VEC_CLASSID,2);
2220   PetscValidHeaderSpecific(y,VEC_CLASSID,3);
2221 
2222   if (!mat->assembled) SETERRQ(PetscObjectComm((PetscObject)mat),PETSC_ERR_ARG_WRONGSTATE,"Not for unassembled matrix");
2223   if (mat->factortype) SETERRQ(PetscObjectComm((PetscObject)mat),PETSC_ERR_ARG_WRONGSTATE,"Not for factored matrix");
2224   if (x == y) SETERRQ(PetscObjectComm((PetscObject)mat),PETSC_ERR_ARG_WRONGSTATE,"x and y must be different vectors");
2225 #if !defined(PETSC_HAVE_CONSTRAINTS)
2226   if (mat->rmap->N != x->map->N) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_ARG_SIZ,"Mat mat,Vec x: global dim %D %D",mat->rmap->N,x->map->N);
2227   if (mat->cmap->N != y->map->N) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_ARG_SIZ,"Mat mat,Vec y: global dim %D %D",mat->cmap->N,y->map->N);
2228 #endif
2229   ierr = VecValidValues(x,2,PETSC_TRUE);CHKERRQ(ierr);
2230   MatCheckPreallocated(mat,1);
2231 
2232   if (!mat->ops->multtranspose) SETERRQ(PetscObjectComm((PetscObject)mat),PETSC_ERR_SUP,"This matrix type does not have a multiply tranpose defined");
2233   ierr = PetscLogEventBegin(MAT_MultTranspose,mat,x,y,0);CHKERRQ(ierr);
2234   ierr = (*mat->ops->multtranspose)(mat,x,y);CHKERRQ(ierr);
2235   ierr = PetscLogEventEnd(MAT_MultTranspose,mat,x,y,0);CHKERRQ(ierr);
2236   ierr = PetscObjectStateIncrease((PetscObject)y);CHKERRQ(ierr);
2237   ierr = VecValidValues(y,3,PETSC_FALSE);CHKERRQ(ierr);
2238   PetscFunctionReturn(0);
2239 }
2240 
2241 #undef __FUNCT__
2242 #define __FUNCT__ "MatMultHermitianTranspose"
2243 /*@
2244    MatMultHermitianTranspose - Computes matrix Hermitian transpose times a vector.
2245 
2246    Neighbor-wise Collective on Mat and Vec
2247 
2248    Input Parameters:
2249 +  mat - the matrix
2250 -  x   - the vector to be multilplied
2251 
2252    Output Parameters:
2253 .  y - the result
2254 
2255    Notes:
2256    The vectors x and y cannot be the same.  I.e., one cannot
2257    call MatMultHermitianTranspose(A,y,y).
2258 
2259    Also called the conjugate transpose, complex conjugate transpose, or adjoint.
2260 
2261    For real numbers MatMultTranspose() and MatMultHermitianTranspose() are identical.
2262 
2263    Level: beginner
2264 
2265    Concepts: matrix vector product^transpose
2266 
2267 .seealso: MatMult(), MatMultAdd(), MatMultHermitianTransposeAdd(), MatMultTranspose()
2268 @*/
2269 PetscErrorCode  MatMultHermitianTranspose(Mat mat,Vec x,Vec y)
2270 {
2271   PetscErrorCode ierr;
2272   Vec            w;
2273 
2274   PetscFunctionBegin;
2275   PetscValidHeaderSpecific(mat,MAT_CLASSID,1);
2276   PetscValidType(mat,1);
2277   PetscValidHeaderSpecific(x,VEC_CLASSID,2);
2278   PetscValidHeaderSpecific(y,VEC_CLASSID,3);
2279 
2280   if (!mat->assembled) SETERRQ(PetscObjectComm((PetscObject)mat),PETSC_ERR_ARG_WRONGSTATE,"Not for unassembled matrix");
2281   if (mat->factortype) SETERRQ(PetscObjectComm((PetscObject)mat),PETSC_ERR_ARG_WRONGSTATE,"Not for factored matrix");
2282   if (x == y) SETERRQ(PetscObjectComm((PetscObject)mat),PETSC_ERR_ARG_WRONGSTATE,"x and y must be different vectors");
2283 #if !defined(PETSC_HAVE_CONSTRAINTS)
2284   if (mat->rmap->N != x->map->N) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_ARG_SIZ,"Mat mat,Vec x: global dim %D %D",mat->rmap->N,x->map->N);
2285   if (mat->cmap->N != y->map->N) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_ARG_SIZ,"Mat mat,Vec y: global dim %D %D",mat->cmap->N,y->map->N);
2286 #endif
2287   MatCheckPreallocated(mat,1);
2288 
2289   ierr = PetscLogEventBegin(MAT_MultHermitianTranspose,mat,x,y,0);CHKERRQ(ierr);
2290   if (mat->ops->multhermitiantranspose) {
2291     ierr = (*mat->ops->multhermitiantranspose)(mat,x,y);CHKERRQ(ierr);
2292   } else {
2293     ierr = VecDuplicate(x,&w);CHKERRQ(ierr);
2294     ierr = VecCopy(x,w);CHKERRQ(ierr);
2295     ierr = VecConjugate(w);CHKERRQ(ierr);
2296     ierr = MatMultTranspose(mat,w,y);CHKERRQ(ierr);
2297     ierr = VecDestroy(&w);CHKERRQ(ierr);
2298     ierr = VecConjugate(y);CHKERRQ(ierr);
2299   }
2300   ierr = PetscLogEventEnd(MAT_MultHermitianTranspose,mat,x,y,0);CHKERRQ(ierr);
2301   ierr = PetscObjectStateIncrease((PetscObject)y);CHKERRQ(ierr);
2302   PetscFunctionReturn(0);
2303 }
2304 
2305 #undef __FUNCT__
2306 #define __FUNCT__ "MatMultAdd"
2307 /*@
2308     MatMultAdd -  Computes v3 = v2 + A * v1.
2309 
2310     Neighbor-wise Collective on Mat and Vec
2311 
2312     Input Parameters:
2313 +   mat - the matrix
2314 -   v1, v2 - the vectors
2315 
2316     Output Parameters:
2317 .   v3 - the result
2318 
2319     Notes:
2320     The vectors v1 and v3 cannot be the same.  I.e., one cannot
2321     call MatMultAdd(A,v1,v2,v1).
2322 
2323     Level: beginner
2324 
2325     Concepts: matrix vector product^addition
2326 
2327 .seealso: MatMultTranspose(), MatMult(), MatMultTransposeAdd()
2328 @*/
2329 PetscErrorCode  MatMultAdd(Mat mat,Vec v1,Vec v2,Vec v3)
2330 {
2331   PetscErrorCode ierr;
2332 
2333   PetscFunctionBegin;
2334   PetscValidHeaderSpecific(mat,MAT_CLASSID,1);
2335   PetscValidType(mat,1);
2336   PetscValidHeaderSpecific(v1,VEC_CLASSID,2);
2337   PetscValidHeaderSpecific(v2,VEC_CLASSID,3);
2338   PetscValidHeaderSpecific(v3,VEC_CLASSID,4);
2339 
2340   if (!mat->assembled) SETERRQ(PetscObjectComm((PetscObject)mat),PETSC_ERR_ARG_WRONGSTATE,"Not for unassembled matrix");
2341   if (mat->factortype) SETERRQ(PetscObjectComm((PetscObject)mat),PETSC_ERR_ARG_WRONGSTATE,"Not for factored matrix");
2342   if (mat->cmap->N != v1->map->N) SETERRQ2(PetscObjectComm((PetscObject)mat),PETSC_ERR_ARG_SIZ,"Mat mat,Vec v1: global dim %D %D",mat->cmap->N,v1->map->N);
2343   /* if (mat->rmap->N != v2->map->N) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_ARG_SIZ,"Mat mat,Vec v2: global dim %D %D",mat->rmap->N,v2->map->N);
2344      if (mat->rmap->N != v3->map->N) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_ARG_SIZ,"Mat mat,Vec v3: global dim %D %D",mat->rmap->N,v3->map->N); */
2345   if (mat->rmap->n != v3->map->n) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_ARG_SIZ,"Mat mat,Vec v3: local dim %D %D",mat->rmap->n,v3->map->n);
2346   if (mat->rmap->n != v2->map->n) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_ARG_SIZ,"Mat mat,Vec v2: local dim %D %D",mat->rmap->n,v2->map->n);
2347   if (v1 == v3) SETERRQ(PetscObjectComm((PetscObject)mat),PETSC_ERR_ARG_IDN,"v1 and v3 must be different vectors");
2348   MatCheckPreallocated(mat,1);
2349 
2350   if (!mat->ops->multadd) SETERRQ1(PetscObjectComm((PetscObject)mat),PETSC_ERR_SUP,"No MatMultAdd() for matrix type '%s'",((PetscObject)mat)->type_name);
2351   ierr = PetscLogEventBegin(MAT_MultAdd,mat,v1,v2,v3);CHKERRQ(ierr);
2352   ierr = (*mat->ops->multadd)(mat,v1,v2,v3);CHKERRQ(ierr);
2353   ierr = PetscLogEventEnd(MAT_MultAdd,mat,v1,v2,v3);CHKERRQ(ierr);
2354   ierr = PetscObjectStateIncrease((PetscObject)v3);CHKERRQ(ierr);
2355   PetscFunctionReturn(0);
2356 }
2357 
2358 #undef __FUNCT__
2359 #define __FUNCT__ "MatMultTransposeAdd"
2360 /*@
2361    MatMultTransposeAdd - Computes v3 = v2 + A' * v1.
2362 
2363    Neighbor-wise Collective on Mat and Vec
2364 
2365    Input Parameters:
2366 +  mat - the matrix
2367 -  v1, v2 - the vectors
2368 
2369    Output Parameters:
2370 .  v3 - the result
2371 
2372    Notes:
2373    The vectors v1 and v3 cannot be the same.  I.e., one cannot
2374    call MatMultTransposeAdd(A,v1,v2,v1).
2375 
2376    Level: beginner
2377 
2378    Concepts: matrix vector product^transpose and addition
2379 
2380 .seealso: MatMultTranspose(), MatMultAdd(), MatMult()
2381 @*/
2382 PetscErrorCode  MatMultTransposeAdd(Mat mat,Vec v1,Vec v2,Vec v3)
2383 {
2384   PetscErrorCode ierr;
2385 
2386   PetscFunctionBegin;
2387   PetscValidHeaderSpecific(mat,MAT_CLASSID,1);
2388   PetscValidType(mat,1);
2389   PetscValidHeaderSpecific(v1,VEC_CLASSID,2);
2390   PetscValidHeaderSpecific(v2,VEC_CLASSID,3);
2391   PetscValidHeaderSpecific(v3,VEC_CLASSID,4);
2392 
2393   if (!mat->assembled) SETERRQ(PetscObjectComm((PetscObject)mat),PETSC_ERR_ARG_WRONGSTATE,"Not for unassembled matrix");
2394   if (mat->factortype) SETERRQ(PetscObjectComm((PetscObject)mat),PETSC_ERR_ARG_WRONGSTATE,"Not for factored matrix");
2395   if (!mat->ops->multtransposeadd) SETERRQ1(PetscObjectComm((PetscObject)mat),PETSC_ERR_SUP,"Mat type %s",((PetscObject)mat)->type_name);
2396   if (v1 == v3) SETERRQ(PetscObjectComm((PetscObject)mat),PETSC_ERR_ARG_IDN,"v1 and v3 must be different vectors");
2397   if (mat->rmap->N != v1->map->N) SETERRQ2(PetscObjectComm((PetscObject)mat),PETSC_ERR_ARG_SIZ,"Mat mat,Vec v1: global dim %D %D",mat->rmap->N,v1->map->N);
2398   if (mat->cmap->N != v2->map->N) SETERRQ2(PetscObjectComm((PetscObject)mat),PETSC_ERR_ARG_SIZ,"Mat mat,Vec v2: global dim %D %D",mat->cmap->N,v2->map->N);
2399   if (mat->cmap->N != v3->map->N) SETERRQ2(PetscObjectComm((PetscObject)mat),PETSC_ERR_ARG_SIZ,"Mat mat,Vec v3: global dim %D %D",mat->cmap->N,v3->map->N);
2400   MatCheckPreallocated(mat,1);
2401 
2402   ierr = PetscLogEventBegin(MAT_MultTransposeAdd,mat,v1,v2,v3);CHKERRQ(ierr);
2403   ierr = (*mat->ops->multtransposeadd)(mat,v1,v2,v3);CHKERRQ(ierr);
2404   ierr = PetscLogEventEnd(MAT_MultTransposeAdd,mat,v1,v2,v3);CHKERRQ(ierr);
2405   ierr = PetscObjectStateIncrease((PetscObject)v3);CHKERRQ(ierr);
2406   PetscFunctionReturn(0);
2407 }
2408 
2409 #undef __FUNCT__
2410 #define __FUNCT__ "MatMultHermitianTransposeAdd"
2411 /*@
2412    MatMultHermitianTransposeAdd - Computes v3 = v2 + A^H * v1.
2413 
2414    Neighbor-wise Collective on Mat and Vec
2415 
2416    Input Parameters:
2417 +  mat - the matrix
2418 -  v1, v2 - the vectors
2419 
2420    Output Parameters:
2421 .  v3 - the result
2422 
2423    Notes:
2424    The vectors v1 and v3 cannot be the same.  I.e., one cannot
2425    call MatMultHermitianTransposeAdd(A,v1,v2,v1).
2426 
2427    Level: beginner
2428 
2429    Concepts: matrix vector product^transpose and addition
2430 
2431 .seealso: MatMultHermitianTranspose(), MatMultTranspose(), MatMultAdd(), MatMult()
2432 @*/
2433 PetscErrorCode  MatMultHermitianTransposeAdd(Mat mat,Vec v1,Vec v2,Vec v3)
2434 {
2435   PetscErrorCode ierr;
2436 
2437   PetscFunctionBegin;
2438   PetscValidHeaderSpecific(mat,MAT_CLASSID,1);
2439   PetscValidType(mat,1);
2440   PetscValidHeaderSpecific(v1,VEC_CLASSID,2);
2441   PetscValidHeaderSpecific(v2,VEC_CLASSID,3);
2442   PetscValidHeaderSpecific(v3,VEC_CLASSID,4);
2443 
2444   if (!mat->assembled) SETERRQ(PetscObjectComm((PetscObject)mat),PETSC_ERR_ARG_WRONGSTATE,"Not for unassembled matrix");
2445   if (mat->factortype) SETERRQ(PetscObjectComm((PetscObject)mat),PETSC_ERR_ARG_WRONGSTATE,"Not for factored matrix");
2446   if (!mat->ops->multhermitiantransposeadd) SETERRQ1(PetscObjectComm((PetscObject)mat),PETSC_ERR_SUP,"Mat type %s",((PetscObject)mat)->type_name);
2447   if (v1 == v3) SETERRQ(PetscObjectComm((PetscObject)mat),PETSC_ERR_ARG_IDN,"v1 and v3 must be different vectors");
2448   if (mat->rmap->N != v1->map->N) SETERRQ2(PetscObjectComm((PetscObject)mat),PETSC_ERR_ARG_SIZ,"Mat mat,Vec v1: global dim %D %D",mat->rmap->N,v1->map->N);
2449   if (mat->cmap->N != v2->map->N) SETERRQ2(PetscObjectComm((PetscObject)mat),PETSC_ERR_ARG_SIZ,"Mat mat,Vec v2: global dim %D %D",mat->cmap->N,v2->map->N);
2450   if (mat->cmap->N != v3->map->N) SETERRQ2(PetscObjectComm((PetscObject)mat),PETSC_ERR_ARG_SIZ,"Mat mat,Vec v3: global dim %D %D",mat->cmap->N,v3->map->N);
2451   MatCheckPreallocated(mat,1);
2452 
2453   ierr = PetscLogEventBegin(MAT_MultHermitianTransposeAdd,mat,v1,v2,v3);CHKERRQ(ierr);
2454   ierr = (*mat->ops->multhermitiantransposeadd)(mat,v1,v2,v3);CHKERRQ(ierr);
2455   ierr = PetscLogEventEnd(MAT_MultHermitianTransposeAdd,mat,v1,v2,v3);CHKERRQ(ierr);
2456   ierr = PetscObjectStateIncrease((PetscObject)v3);CHKERRQ(ierr);
2457   PetscFunctionReturn(0);
2458 }
2459 
2460 #undef __FUNCT__
2461 #define __FUNCT__ "MatMultConstrained"
2462 /*@
2463    MatMultConstrained - The inner multiplication routine for a
2464    constrained matrix P^T A P.
2465 
2466    Neighbor-wise Collective on Mat and Vec
2467 
2468    Input Parameters:
2469 +  mat - the matrix
2470 -  x   - the vector to be multilplied
2471 
2472    Output Parameters:
2473 .  y - the result
2474 
2475    Notes:
2476    The vectors x and y cannot be the same.  I.e., one cannot
2477    call MatMult(A,y,y).
2478 
2479    Level: beginner
2480 
2481 .keywords: matrix, multiply, matrix-vector product, constraint
2482 .seealso: MatMult(), MatMultTranspose(), MatMultAdd(), MatMultTransposeAdd()
2483 @*/
2484 PetscErrorCode  MatMultConstrained(Mat mat,Vec x,Vec y)
2485 {
2486   PetscErrorCode ierr;
2487 
2488   PetscFunctionBegin;
2489   PetscValidHeaderSpecific(mat,MAT_CLASSID,1);
2490   PetscValidHeaderSpecific(x,VEC_CLASSID,2);
2491   PetscValidHeaderSpecific(y,VEC_CLASSID,3);
2492   if (!mat->assembled) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONGSTATE,"Not for unassembled matrix");
2493   if (mat->factortype) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONGSTATE,"Not for factored matrix");
2494   if (x == y) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONGSTATE,"x and y must be different vectors");
2495   if (mat->cmap->N != x->map->N) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_ARG_SIZ,"Mat mat,Vec x: global dim %D %D",mat->cmap->N,x->map->N);
2496   if (mat->rmap->N != y->map->N) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_ARG_SIZ,"Mat mat,Vec y: global dim %D %D",mat->rmap->N,y->map->N);
2497   if (mat->rmap->n != y->map->n) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_ARG_SIZ,"Mat mat,Vec y: local dim %D %D",mat->rmap->n,y->map->n);
2498 
2499   ierr = PetscLogEventBegin(MAT_MultConstrained,mat,x,y,0);CHKERRQ(ierr);
2500   ierr = (*mat->ops->multconstrained)(mat,x,y);CHKERRQ(ierr);
2501   ierr = PetscLogEventEnd(MAT_MultConstrained,mat,x,y,0);CHKERRQ(ierr);
2502   ierr = PetscObjectStateIncrease((PetscObject)y);CHKERRQ(ierr);
2503   PetscFunctionReturn(0);
2504 }
2505 
2506 #undef __FUNCT__
2507 #define __FUNCT__ "MatMultTransposeConstrained"
2508 /*@
2509    MatMultTransposeConstrained - The inner multiplication routine for a
2510    constrained matrix P^T A^T P.
2511 
2512    Neighbor-wise Collective on Mat and Vec
2513 
2514    Input Parameters:
2515 +  mat - the matrix
2516 -  x   - the vector to be multilplied
2517 
2518    Output Parameters:
2519 .  y - the result
2520 
2521    Notes:
2522    The vectors x and y cannot be the same.  I.e., one cannot
2523    call MatMult(A,y,y).
2524 
2525    Level: beginner
2526 
2527 .keywords: matrix, multiply, matrix-vector product, constraint
2528 .seealso: MatMult(), MatMultTranspose(), MatMultAdd(), MatMultTransposeAdd()
2529 @*/
2530 PetscErrorCode  MatMultTransposeConstrained(Mat mat,Vec x,Vec y)
2531 {
2532   PetscErrorCode ierr;
2533 
2534   PetscFunctionBegin;
2535   PetscValidHeaderSpecific(mat,MAT_CLASSID,1);
2536   PetscValidHeaderSpecific(x,VEC_CLASSID,2);
2537   PetscValidHeaderSpecific(y,VEC_CLASSID,3);
2538   if (!mat->assembled) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONGSTATE,"Not for unassembled matrix");
2539   if (mat->factortype) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONGSTATE,"Not for factored matrix");
2540   if (x == y) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONGSTATE,"x and y must be different vectors");
2541   if (mat->rmap->N != x->map->N) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_ARG_SIZ,"Mat mat,Vec x: global dim %D %D",mat->cmap->N,x->map->N);
2542   if (mat->cmap->N != y->map->N) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_ARG_SIZ,"Mat mat,Vec y: global dim %D %D",mat->rmap->N,y->map->N);
2543 
2544   ierr = PetscLogEventBegin(MAT_MultConstrained,mat,x,y,0);CHKERRQ(ierr);
2545   ierr = (*mat->ops->multtransposeconstrained)(mat,x,y);CHKERRQ(ierr);
2546   ierr = PetscLogEventEnd(MAT_MultConstrained,mat,x,y,0);CHKERRQ(ierr);
2547   ierr = PetscObjectStateIncrease((PetscObject)y);CHKERRQ(ierr);
2548   PetscFunctionReturn(0);
2549 }
2550 
2551 #undef __FUNCT__
2552 #define __FUNCT__ "MatGetFactorType"
2553 /*@C
2554    MatGetFactorType - gets the type of factorization it is
2555 
2556    Note Collective
2557    as the flag
2558 
2559    Input Parameters:
2560 .  mat - the matrix
2561 
2562    Output Parameters:
2563 .  t - the type, one of MAT_FACTOR_NONE, MAT_FACTOR_LU, MAT_FACTOR_CHOLESKY, MAT_FACTOR_ILU, MAT_FACTOR_ICC,MAT_FACTOR_ILUDT
2564 
2565     Level: intermediate
2566 
2567 .seealso:    MatFactorType, MatGetFactor()
2568 @*/
2569 PetscErrorCode  MatGetFactorType(Mat mat,MatFactorType *t)
2570 {
2571   PetscFunctionBegin;
2572   PetscValidHeaderSpecific(mat,MAT_CLASSID,1);
2573   PetscValidType(mat,1);
2574   *t = mat->factortype;
2575   PetscFunctionReturn(0);
2576 }
2577 
2578 /* ------------------------------------------------------------*/
2579 #undef __FUNCT__
2580 #define __FUNCT__ "MatGetInfo"
2581 /*@C
2582    MatGetInfo - Returns information about matrix storage (number of
2583    nonzeros, memory, etc.).
2584 
2585    Collective on Mat if MAT_GLOBAL_MAX or MAT_GLOBAL_SUM is used as the flag
2586 
2587    Input Parameters:
2588 .  mat - the matrix
2589 
2590    Output Parameters:
2591 +  flag - flag indicating the type of parameters to be returned
2592    (MAT_LOCAL - local matrix, MAT_GLOBAL_MAX - maximum over all processors,
2593    MAT_GLOBAL_SUM - sum over all processors)
2594 -  info - matrix information context
2595 
2596    Notes:
2597    The MatInfo context contains a variety of matrix data, including
2598    number of nonzeros allocated and used, number of mallocs during
2599    matrix assembly, etc.  Additional information for factored matrices
2600    is provided (such as the fill ratio, number of mallocs during
2601    factorization, etc.).  Much of this info is printed to PETSC_STDOUT
2602    when using the runtime options
2603 $       -info -mat_view ::ascii_info
2604 
2605    Example for C/C++ Users:
2606    See the file ${PETSC_DIR}/include/petscmat.h for a complete list of
2607    data within the MatInfo context.  For example,
2608 .vb
2609       MatInfo info;
2610       Mat     A;
2611       double  mal, nz_a, nz_u;
2612 
2613       MatGetInfo(A,MAT_LOCAL,&info);
2614       mal  = info.mallocs;
2615       nz_a = info.nz_allocated;
2616 .ve
2617 
2618    Example for Fortran Users:
2619    Fortran users should declare info as a double precision
2620    array of dimension MAT_INFO_SIZE, and then extract the parameters
2621    of interest.  See the file ${PETSC_DIR}/include/finclude/petscmat.h
2622    a complete list of parameter names.
2623 .vb
2624       double  precision info(MAT_INFO_SIZE)
2625       double  precision mal, nz_a
2626       Mat     A
2627       integer ierr
2628 
2629       call MatGetInfo(A,MAT_LOCAL,info,ierr)
2630       mal = info(MAT_INFO_MALLOCS)
2631       nz_a = info(MAT_INFO_NZ_ALLOCATED)
2632 .ve
2633 
2634     Level: intermediate
2635 
2636     Concepts: matrices^getting information on
2637 
2638     Developer Note: fortran interface is not autogenerated as the f90
2639     interface defintion cannot be generated correctly [due to MatInfo]
2640 
2641 .seealso: MatStashGetInfo()
2642 
2643 @*/
2644 PetscErrorCode  MatGetInfo(Mat mat,MatInfoType flag,MatInfo *info)
2645 {
2646   PetscErrorCode ierr;
2647 
2648   PetscFunctionBegin;
2649   PetscValidHeaderSpecific(mat,MAT_CLASSID,1);
2650   PetscValidType(mat,1);
2651   PetscValidPointer(info,3);
2652   if (!mat->ops->getinfo) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_SUP,"Mat type %s",((PetscObject)mat)->type_name);
2653   MatCheckPreallocated(mat,1);
2654   ierr = (*mat->ops->getinfo)(mat,flag,info);CHKERRQ(ierr);
2655   PetscFunctionReturn(0);
2656 }
2657 
2658 /* ----------------------------------------------------------*/
2659 
2660 #undef __FUNCT__
2661 #define __FUNCT__ "MatLUFactor"
2662 /*@C
2663    MatLUFactor - Performs in-place LU factorization of matrix.
2664 
2665    Collective on Mat
2666 
2667    Input Parameters:
2668 +  mat - the matrix
2669 .  row - row permutation
2670 .  col - column permutation
2671 -  info - options for factorization, includes
2672 $          fill - expected fill as ratio of original fill.
2673 $          dtcol - pivot tolerance (0 no pivot, 1 full column pivoting)
2674 $                   Run with the option -info to determine an optimal value to use
2675 
2676    Notes:
2677    Most users should employ the simplified KSP interface for linear solvers
2678    instead of working directly with matrix algebra routines such as this.
2679    See, e.g., KSPCreate().
2680 
2681    This changes the state of the matrix to a factored matrix; it cannot be used
2682    for example with MatSetValues() unless one first calls MatSetUnfactored().
2683 
2684    Level: developer
2685 
2686    Concepts: matrices^LU factorization
2687 
2688 .seealso: MatLUFactorSymbolic(), MatLUFactorNumeric(), MatCholeskyFactor(),
2689           MatGetOrdering(), MatSetUnfactored(), MatFactorInfo, MatGetFactor()
2690 
2691     Developer Note: fortran interface is not autogenerated as the f90
2692     interface defintion cannot be generated correctly [due to MatFactorInfo]
2693 
2694 @*/
2695 PetscErrorCode  MatLUFactor(Mat mat,IS row,IS col,const MatFactorInfo *info)
2696 {
2697   PetscErrorCode ierr;
2698   MatFactorInfo  tinfo;
2699 
2700   PetscFunctionBegin;
2701   PetscValidHeaderSpecific(mat,MAT_CLASSID,1);
2702   if (row) PetscValidHeaderSpecific(row,IS_CLASSID,2);
2703   if (col) PetscValidHeaderSpecific(col,IS_CLASSID,3);
2704   if (info) PetscValidPointer(info,4);
2705   PetscValidType(mat,1);
2706   if (!mat->assembled) SETERRQ(PetscObjectComm((PetscObject)mat),PETSC_ERR_ARG_WRONGSTATE,"Not for unassembled matrix");
2707   if (mat->factortype) SETERRQ(PetscObjectComm((PetscObject)mat),PETSC_ERR_ARG_WRONGSTATE,"Not for factored matrix");
2708   if (!mat->ops->lufactor) SETERRQ1(PetscObjectComm((PetscObject)mat),PETSC_ERR_SUP,"Mat type %s",((PetscObject)mat)->type_name);
2709   MatCheckPreallocated(mat,1);
2710   if (!info) {
2711     ierr = MatFactorInfoInitialize(&tinfo);CHKERRQ(ierr);
2712     info = &tinfo;
2713   }
2714 
2715   ierr = PetscLogEventBegin(MAT_LUFactor,mat,row,col,0);CHKERRQ(ierr);
2716   ierr = (*mat->ops->lufactor)(mat,row,col,info);CHKERRQ(ierr);
2717   ierr = PetscLogEventEnd(MAT_LUFactor,mat,row,col,0);CHKERRQ(ierr);
2718   ierr = PetscObjectStateIncrease((PetscObject)mat);CHKERRQ(ierr);
2719   PetscFunctionReturn(0);
2720 }
2721 
2722 #undef __FUNCT__
2723 #define __FUNCT__ "MatILUFactor"
2724 /*@C
2725    MatILUFactor - Performs in-place ILU factorization of matrix.
2726 
2727    Collective on Mat
2728 
2729    Input Parameters:
2730 +  mat - the matrix
2731 .  row - row permutation
2732 .  col - column permutation
2733 -  info - structure containing
2734 $      levels - number of levels of fill.
2735 $      expected fill - as ratio of original fill.
2736 $      1 or 0 - indicating force fill on diagonal (improves robustness for matrices
2737                 missing diagonal entries)
2738 
2739    Notes:
2740    Probably really in-place only when level of fill is zero, otherwise allocates
2741    new space to store factored matrix and deletes previous memory.
2742 
2743    Most users should employ the simplified KSP interface for linear solvers
2744    instead of working directly with matrix algebra routines such as this.
2745    See, e.g., KSPCreate().
2746 
2747    Level: developer
2748 
2749    Concepts: matrices^ILU factorization
2750 
2751 .seealso: MatILUFactorSymbolic(), MatLUFactorNumeric(), MatCholeskyFactor(), MatFactorInfo
2752 
2753     Developer Note: fortran interface is not autogenerated as the f90
2754     interface defintion cannot be generated correctly [due to MatFactorInfo]
2755 
2756 @*/
2757 PetscErrorCode  MatILUFactor(Mat mat,IS row,IS col,const MatFactorInfo *info)
2758 {
2759   PetscErrorCode ierr;
2760 
2761   PetscFunctionBegin;
2762   PetscValidHeaderSpecific(mat,MAT_CLASSID,1);
2763   if (row) PetscValidHeaderSpecific(row,IS_CLASSID,2);
2764   if (col) PetscValidHeaderSpecific(col,IS_CLASSID,3);
2765   PetscValidPointer(info,4);
2766   PetscValidType(mat,1);
2767   if (mat->rmap->N != mat->cmap->N) SETERRQ(PetscObjectComm((PetscObject)mat),PETSC_ERR_ARG_WRONG,"matrix must be square");
2768   if (!mat->assembled) SETERRQ(PetscObjectComm((PetscObject)mat),PETSC_ERR_ARG_WRONGSTATE,"Not for unassembled matrix");
2769   if (mat->factortype) SETERRQ(PetscObjectComm((PetscObject)mat),PETSC_ERR_ARG_WRONGSTATE,"Not for factored matrix");
2770   if (!mat->ops->ilufactor) SETERRQ1(PetscObjectComm((PetscObject)mat),PETSC_ERR_SUP,"Mat type %s",((PetscObject)mat)->type_name);
2771   MatCheckPreallocated(mat,1);
2772 
2773   ierr = PetscLogEventBegin(MAT_ILUFactor,mat,row,col,0);CHKERRQ(ierr);
2774   ierr = (*mat->ops->ilufactor)(mat,row,col,info);CHKERRQ(ierr);
2775   ierr = PetscLogEventEnd(MAT_ILUFactor,mat,row,col,0);CHKERRQ(ierr);
2776   ierr = PetscObjectStateIncrease((PetscObject)mat);CHKERRQ(ierr);
2777   PetscFunctionReturn(0);
2778 }
2779 
2780 #undef __FUNCT__
2781 #define __FUNCT__ "MatLUFactorSymbolic"
2782 /*@C
2783    MatLUFactorSymbolic - Performs symbolic LU factorization of matrix.
2784    Call this routine before calling MatLUFactorNumeric().
2785 
2786    Collective on Mat
2787 
2788    Input Parameters:
2789 +  fact - the factor matrix obtained with MatGetFactor()
2790 .  mat - the matrix
2791 .  row, col - row and column permutations
2792 -  info - options for factorization, includes
2793 $          fill - expected fill as ratio of original fill.
2794 $          dtcol - pivot tolerance (0 no pivot, 1 full column pivoting)
2795 $                   Run with the option -info to determine an optimal value to use
2796 
2797 
2798    Notes:
2799    See the <a href="../../docs/manual.pdf">users manual</a> for additional information about
2800    choosing the fill factor for better efficiency.
2801 
2802    Most users should employ the simplified KSP interface for linear solvers
2803    instead of working directly with matrix algebra routines such as this.
2804    See, e.g., KSPCreate().
2805 
2806    Level: developer
2807 
2808    Concepts: matrices^LU symbolic factorization
2809 
2810 .seealso: MatLUFactor(), MatLUFactorNumeric(), MatCholeskyFactor(), MatFactorInfo
2811 
2812     Developer Note: fortran interface is not autogenerated as the f90
2813     interface defintion cannot be generated correctly [due to MatFactorInfo]
2814 
2815 @*/
2816 PetscErrorCode  MatLUFactorSymbolic(Mat fact,Mat mat,IS row,IS col,const MatFactorInfo *info)
2817 {
2818   PetscErrorCode ierr;
2819 
2820   PetscFunctionBegin;
2821   PetscValidHeaderSpecific(mat,MAT_CLASSID,1);
2822   if (row) PetscValidHeaderSpecific(row,IS_CLASSID,2);
2823   if (col) PetscValidHeaderSpecific(col,IS_CLASSID,3);
2824   if (info) PetscValidPointer(info,4);
2825   PetscValidType(mat,1);
2826   PetscValidPointer(fact,5);
2827   if (!mat->assembled) SETERRQ(PetscObjectComm((PetscObject)mat),PETSC_ERR_ARG_WRONGSTATE,"Not for unassembled matrix");
2828   if (mat->factortype) SETERRQ(PetscObjectComm((PetscObject)mat),PETSC_ERR_ARG_WRONGSTATE,"Not for factored matrix");
2829   if (!(fact)->ops->lufactorsymbolic) {
2830     const MatSolverPackage spackage;
2831     ierr = MatFactorGetSolverPackage(fact,&spackage);CHKERRQ(ierr);
2832     SETERRQ2(PetscObjectComm((PetscObject)mat),PETSC_ERR_SUP,"Matrix type %s symbolic LU using solver package %s",((PetscObject)mat)->type_name,spackage);
2833   }
2834   MatCheckPreallocated(mat,2);
2835 
2836   ierr = PetscLogEventBegin(MAT_LUFactorSymbolic,mat,row,col,0);CHKERRQ(ierr);
2837   ierr = (fact->ops->lufactorsymbolic)(fact,mat,row,col,info);CHKERRQ(ierr);
2838   ierr = PetscLogEventEnd(MAT_LUFactorSymbolic,mat,row,col,0);CHKERRQ(ierr);
2839   ierr = PetscObjectStateIncrease((PetscObject)fact);CHKERRQ(ierr);
2840   PetscFunctionReturn(0);
2841 }
2842 
2843 #undef __FUNCT__
2844 #define __FUNCT__ "MatLUFactorNumeric"
2845 /*@C
2846    MatLUFactorNumeric - Performs numeric LU factorization of a matrix.
2847    Call this routine after first calling MatLUFactorSymbolic().
2848 
2849    Collective on Mat
2850 
2851    Input Parameters:
2852 +  fact - the factor matrix obtained with MatGetFactor()
2853 .  mat - the matrix
2854 -  info - options for factorization
2855 
2856    Notes:
2857    See MatLUFactor() for in-place factorization.  See
2858    MatCholeskyFactorNumeric() for the symmetric, positive definite case.
2859 
2860    Most users should employ the simplified KSP interface for linear solvers
2861    instead of working directly with matrix algebra routines such as this.
2862    See, e.g., KSPCreate().
2863 
2864    Level: developer
2865 
2866    Concepts: matrices^LU numeric factorization
2867 
2868 .seealso: MatLUFactorSymbolic(), MatLUFactor(), MatCholeskyFactor()
2869 
2870     Developer Note: fortran interface is not autogenerated as the f90
2871     interface defintion cannot be generated correctly [due to MatFactorInfo]
2872 
2873 @*/
2874 PetscErrorCode  MatLUFactorNumeric(Mat fact,Mat mat,const MatFactorInfo *info)
2875 {
2876   PetscErrorCode ierr;
2877 
2878   PetscFunctionBegin;
2879   PetscValidHeaderSpecific(mat,MAT_CLASSID,1);
2880   PetscValidType(mat,1);
2881   PetscValidPointer(fact,2);
2882   PetscValidHeaderSpecific(fact,MAT_CLASSID,2);
2883   if (!mat->assembled) SETERRQ(PetscObjectComm((PetscObject)mat),PETSC_ERR_ARG_WRONGSTATE,"Not for unassembled matrix");
2884   if (mat->rmap->N != (fact)->rmap->N || mat->cmap->N != (fact)->cmap->N) SETERRQ4(PetscObjectComm((PetscObject)mat),PETSC_ERR_ARG_SIZ,"Mat mat,Mat fact: global dimensions are different %D should = %D %D should = %D",mat->rmap->N,(fact)->rmap->N,mat->cmap->N,(fact)->cmap->N);
2885 
2886   if (!(fact)->ops->lufactornumeric) SETERRQ1(PetscObjectComm((PetscObject)mat),PETSC_ERR_SUP,"Mat type %s numeric LU",((PetscObject)mat)->type_name);
2887   MatCheckPreallocated(mat,2);
2888   ierr = PetscLogEventBegin(MAT_LUFactorNumeric,mat,fact,0,0);CHKERRQ(ierr);
2889   ierr = (fact->ops->lufactornumeric)(fact,mat,info);CHKERRQ(ierr);
2890   ierr = PetscLogEventEnd(MAT_LUFactorNumeric,mat,fact,0,0);CHKERRQ(ierr);
2891 
2892   if (fact->viewonassembly) {
2893     ierr = PetscViewerPushFormat(fact->viewonassembly,fact->viewformatonassembly);CHKERRQ(ierr);
2894     ierr = MatView(fact,fact->viewonassembly);CHKERRQ(ierr);
2895     ierr = PetscViewerPopFormat(fact->viewonassembly);CHKERRQ(ierr);
2896   }
2897   ierr = PetscObjectStateIncrease((PetscObject)fact);CHKERRQ(ierr);
2898   PetscFunctionReturn(0);
2899 }
2900 
2901 #undef __FUNCT__
2902 #define __FUNCT__ "MatCholeskyFactor"
2903 /*@C
2904    MatCholeskyFactor - Performs in-place Cholesky factorization of a
2905    symmetric matrix.
2906 
2907    Collective on Mat
2908 
2909    Input Parameters:
2910 +  mat - the matrix
2911 .  perm - row and column permutations
2912 -  f - expected fill as ratio of original fill
2913 
2914    Notes:
2915    See MatLUFactor() for the nonsymmetric case.  See also
2916    MatCholeskyFactorSymbolic(), and MatCholeskyFactorNumeric().
2917 
2918    Most users should employ the simplified KSP interface for linear solvers
2919    instead of working directly with matrix algebra routines such as this.
2920    See, e.g., KSPCreate().
2921 
2922    Level: developer
2923 
2924    Concepts: matrices^Cholesky factorization
2925 
2926 .seealso: MatLUFactor(), MatCholeskyFactorSymbolic(), MatCholeskyFactorNumeric()
2927           MatGetOrdering()
2928 
2929     Developer Note: fortran interface is not autogenerated as the f90
2930     interface defintion cannot be generated correctly [due to MatFactorInfo]
2931 
2932 @*/
2933 PetscErrorCode  MatCholeskyFactor(Mat mat,IS perm,const MatFactorInfo *info)
2934 {
2935   PetscErrorCode ierr;
2936 
2937   PetscFunctionBegin;
2938   PetscValidHeaderSpecific(mat,MAT_CLASSID,1);
2939   PetscValidType(mat,1);
2940   if (perm) PetscValidHeaderSpecific(perm,IS_CLASSID,2);
2941   if (info) PetscValidPointer(info,3);
2942   if (mat->rmap->N != mat->cmap->N) SETERRQ(PetscObjectComm((PetscObject)mat),PETSC_ERR_ARG_WRONG,"Matrix must be square");
2943   if (!mat->assembled) SETERRQ(PetscObjectComm((PetscObject)mat),PETSC_ERR_ARG_WRONGSTATE,"Not for unassembled matrix");
2944   if (mat->factortype) SETERRQ(PetscObjectComm((PetscObject)mat),PETSC_ERR_ARG_WRONGSTATE,"Not for factored matrix");
2945   if (!mat->ops->choleskyfactor) SETERRQ1(PetscObjectComm((PetscObject)mat),PETSC_ERR_SUP,"Mat type %s",((PetscObject)mat)->type_name);
2946   MatCheckPreallocated(mat,1);
2947 
2948   ierr = PetscLogEventBegin(MAT_CholeskyFactor,mat,perm,0,0);CHKERRQ(ierr);
2949   ierr = (*mat->ops->choleskyfactor)(mat,perm,info);CHKERRQ(ierr);
2950   ierr = PetscLogEventEnd(MAT_CholeskyFactor,mat,perm,0,0);CHKERRQ(ierr);
2951   ierr = PetscObjectStateIncrease((PetscObject)mat);CHKERRQ(ierr);
2952   PetscFunctionReturn(0);
2953 }
2954 
2955 #undef __FUNCT__
2956 #define __FUNCT__ "MatCholeskyFactorSymbolic"
2957 /*@C
2958    MatCholeskyFactorSymbolic - Performs symbolic Cholesky factorization
2959    of a symmetric matrix.
2960 
2961    Collective on Mat
2962 
2963    Input Parameters:
2964 +  fact - the factor matrix obtained with MatGetFactor()
2965 .  mat - the matrix
2966 .  perm - row and column permutations
2967 -  info - options for factorization, includes
2968 $          fill - expected fill as ratio of original fill.
2969 $          dtcol - pivot tolerance (0 no pivot, 1 full column pivoting)
2970 $                   Run with the option -info to determine an optimal value to use
2971 
2972    Notes:
2973    See MatLUFactorSymbolic() for the nonsymmetric case.  See also
2974    MatCholeskyFactor() and MatCholeskyFactorNumeric().
2975 
2976    Most users should employ the simplified KSP interface for linear solvers
2977    instead of working directly with matrix algebra routines such as this.
2978    See, e.g., KSPCreate().
2979 
2980    Level: developer
2981 
2982    Concepts: matrices^Cholesky symbolic factorization
2983 
2984 .seealso: MatLUFactorSymbolic(), MatCholeskyFactor(), MatCholeskyFactorNumeric()
2985           MatGetOrdering()
2986 
2987     Developer Note: fortran interface is not autogenerated as the f90
2988     interface defintion cannot be generated correctly [due to MatFactorInfo]
2989 
2990 @*/
2991 PetscErrorCode  MatCholeskyFactorSymbolic(Mat fact,Mat mat,IS perm,const MatFactorInfo *info)
2992 {
2993   PetscErrorCode ierr;
2994 
2995   PetscFunctionBegin;
2996   PetscValidHeaderSpecific(mat,MAT_CLASSID,1);
2997   PetscValidType(mat,1);
2998   if (perm) PetscValidHeaderSpecific(perm,IS_CLASSID,2);
2999   if (info) PetscValidPointer(info,3);
3000   PetscValidPointer(fact,4);
3001   if (mat->rmap->N != mat->cmap->N) SETERRQ(PetscObjectComm((PetscObject)mat),PETSC_ERR_ARG_WRONG,"Matrix must be square");
3002   if (!mat->assembled) SETERRQ(PetscObjectComm((PetscObject)mat),PETSC_ERR_ARG_WRONGSTATE,"Not for unassembled matrix");
3003   if (mat->factortype) SETERRQ(PetscObjectComm((PetscObject)mat),PETSC_ERR_ARG_WRONGSTATE,"Not for factored matrix");
3004   if (!(fact)->ops->choleskyfactorsymbolic) {
3005     const MatSolverPackage spackage;
3006     ierr = MatFactorGetSolverPackage(fact,&spackage);CHKERRQ(ierr);
3007     SETERRQ2(PetscObjectComm((PetscObject)mat),PETSC_ERR_SUP,"Mat type %s symbolic factor Cholesky using solver package %s",((PetscObject)mat)->type_name,spackage);
3008   }
3009   MatCheckPreallocated(mat,2);
3010 
3011   ierr = PetscLogEventBegin(MAT_CholeskyFactorSymbolic,mat,perm,0,0);CHKERRQ(ierr);
3012   ierr = (fact->ops->choleskyfactorsymbolic)(fact,mat,perm,info);CHKERRQ(ierr);
3013   ierr = PetscLogEventEnd(MAT_CholeskyFactorSymbolic,mat,perm,0,0);CHKERRQ(ierr);
3014   ierr = PetscObjectStateIncrease((PetscObject)fact);CHKERRQ(ierr);
3015   PetscFunctionReturn(0);
3016 }
3017 
3018 #undef __FUNCT__
3019 #define __FUNCT__ "MatCholeskyFactorNumeric"
3020 /*@C
3021    MatCholeskyFactorNumeric - Performs numeric Cholesky factorization
3022    of a symmetric matrix. Call this routine after first calling
3023    MatCholeskyFactorSymbolic().
3024 
3025    Collective on Mat
3026 
3027    Input Parameters:
3028 +  fact - the factor matrix obtained with MatGetFactor()
3029 .  mat - the initial matrix
3030 .  info - options for factorization
3031 -  fact - the symbolic factor of mat
3032 
3033 
3034    Notes:
3035    Most users should employ the simplified KSP interface for linear solvers
3036    instead of working directly with matrix algebra routines such as this.
3037    See, e.g., KSPCreate().
3038 
3039    Level: developer
3040 
3041    Concepts: matrices^Cholesky numeric factorization
3042 
3043 .seealso: MatCholeskyFactorSymbolic(), MatCholeskyFactor(), MatLUFactorNumeric()
3044 
3045     Developer Note: fortran interface is not autogenerated as the f90
3046     interface defintion cannot be generated correctly [due to MatFactorInfo]
3047 
3048 @*/
3049 PetscErrorCode  MatCholeskyFactorNumeric(Mat fact,Mat mat,const MatFactorInfo *info)
3050 {
3051   PetscErrorCode ierr;
3052 
3053   PetscFunctionBegin;
3054   PetscValidHeaderSpecific(mat,MAT_CLASSID,1);
3055   PetscValidType(mat,1);
3056   PetscValidPointer(fact,2);
3057   PetscValidHeaderSpecific(fact,MAT_CLASSID,2);
3058   if (!mat->assembled) SETERRQ(PetscObjectComm((PetscObject)mat),PETSC_ERR_ARG_WRONGSTATE,"Not for unassembled matrix");
3059   if (!(fact)->ops->choleskyfactornumeric) SETERRQ1(PetscObjectComm((PetscObject)mat),PETSC_ERR_SUP,"Mat type %s numeric factor Cholesky",((PetscObject)mat)->type_name);
3060   if (mat->rmap->N != (fact)->rmap->N || mat->cmap->N != (fact)->cmap->N) SETERRQ4(PetscObjectComm((PetscObject)mat),PETSC_ERR_ARG_SIZ,"Mat mat,Mat fact: global dim %D should = %D %D should = %D",mat->rmap->N,(fact)->rmap->N,mat->cmap->N,(fact)->cmap->N);
3061   MatCheckPreallocated(mat,2);
3062 
3063   ierr = PetscLogEventBegin(MAT_CholeskyFactorNumeric,mat,fact,0,0);CHKERRQ(ierr);
3064   ierr = (fact->ops->choleskyfactornumeric)(fact,mat,info);CHKERRQ(ierr);
3065   ierr = PetscLogEventEnd(MAT_CholeskyFactorNumeric,mat,fact,0,0);CHKERRQ(ierr);
3066 
3067   if (fact->viewonassembly) {
3068     ierr = PetscViewerPushFormat(fact->viewonassembly,fact->viewformatonassembly);CHKERRQ(ierr);
3069     ierr = MatView(fact,fact->viewonassembly);CHKERRQ(ierr);
3070     ierr = PetscViewerPopFormat(fact->viewonassembly);CHKERRQ(ierr);
3071   }
3072   ierr = PetscObjectStateIncrease((PetscObject)fact);CHKERRQ(ierr);
3073   PetscFunctionReturn(0);
3074 }
3075 
3076 /* ----------------------------------------------------------------*/
3077 #undef __FUNCT__
3078 #define __FUNCT__ "MatSolve"
3079 /*@
3080    MatSolve - Solves A x = b, given a factored matrix.
3081 
3082    Neighbor-wise Collective on Mat and Vec
3083 
3084    Input Parameters:
3085 +  mat - the factored matrix
3086 -  b - the right-hand-side vector
3087 
3088    Output Parameter:
3089 .  x - the result vector
3090 
3091    Notes:
3092    The vectors b and x cannot be the same.  I.e., one cannot
3093    call MatSolve(A,x,x).
3094 
3095    Notes:
3096    Most users should employ the simplified KSP interface for linear solvers
3097    instead of working directly with matrix algebra routines such as this.
3098    See, e.g., KSPCreate().
3099 
3100    Level: developer
3101 
3102    Concepts: matrices^triangular solves
3103 
3104 .seealso: MatSolveAdd(), MatSolveTranspose(), MatSolveTransposeAdd()
3105 @*/
3106 PetscErrorCode  MatSolve(Mat mat,Vec b,Vec x)
3107 {
3108   PetscErrorCode ierr;
3109 
3110   PetscFunctionBegin;
3111   PetscValidHeaderSpecific(mat,MAT_CLASSID,1);
3112   PetscValidType(mat,1);
3113   PetscValidHeaderSpecific(b,VEC_CLASSID,2);
3114   PetscValidHeaderSpecific(x,VEC_CLASSID,3);
3115   PetscCheckSameComm(mat,1,b,2);
3116   PetscCheckSameComm(mat,1,x,3);
3117   if (x == b) SETERRQ(PetscObjectComm((PetscObject)mat),PETSC_ERR_ARG_IDN,"x and b must be different vectors");
3118   if (!mat->factortype) SETERRQ(PetscObjectComm((PetscObject)mat),PETSC_ERR_ARG_WRONGSTATE,"Unfactored matrix");
3119   if (mat->cmap->N != x->map->N) SETERRQ2(PetscObjectComm((PetscObject)mat),PETSC_ERR_ARG_SIZ,"Mat mat,Vec x: global dim %D %D",mat->cmap->N,x->map->N);
3120   if (mat->rmap->N != b->map->N) SETERRQ2(PetscObjectComm((PetscObject)mat),PETSC_ERR_ARG_SIZ,"Mat mat,Vec b: global dim %D %D",mat->rmap->N,b->map->N);
3121   if (mat->rmap->n != b->map->n) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_ARG_SIZ,"Mat mat,Vec b: local dim %D %D",mat->rmap->n,b->map->n);
3122   if (!mat->rmap->N && !mat->cmap->N) PetscFunctionReturn(0);
3123   if (!mat->ops->solve) SETERRQ1(PetscObjectComm((PetscObject)mat),PETSC_ERR_SUP,"Mat type %s",((PetscObject)mat)->type_name);
3124   MatCheckPreallocated(mat,1);
3125 
3126   ierr = PetscLogEventBegin(MAT_Solve,mat,b,x,0);CHKERRQ(ierr);
3127   ierr = (*mat->ops->solve)(mat,b,x);CHKERRQ(ierr);
3128   ierr = PetscLogEventEnd(MAT_Solve,mat,b,x,0);CHKERRQ(ierr);
3129   ierr = PetscObjectStateIncrease((PetscObject)x);CHKERRQ(ierr);
3130   PetscFunctionReturn(0);
3131 }
3132 
3133 #undef __FUNCT__
3134 #define __FUNCT__ "MatMatSolve_Basic"
3135 PetscErrorCode  MatMatSolve_Basic(Mat A,Mat B,Mat X)
3136 {
3137   PetscErrorCode ierr;
3138   Vec            b,x;
3139   PetscInt       m,N,i;
3140   PetscScalar    *bb,*xx;
3141   PetscBool      flg;
3142 
3143   PetscFunctionBegin;
3144   ierr = PetscObjectTypeCompareAny((PetscObject)B,&flg,MATSEQDENSE,MATMPIDENSE,NULL);CHKERRQ(ierr);
3145   if (!flg) SETERRQ(PetscObjectComm((PetscObject)A),PETSC_ERR_ARG_WRONG,"Matrix B must be MATDENSE matrix");
3146   ierr = PetscObjectTypeCompareAny((PetscObject)X,&flg,MATSEQDENSE,MATMPIDENSE,NULL);CHKERRQ(ierr);
3147   if (!flg) SETERRQ(PetscObjectComm((PetscObject)A),PETSC_ERR_ARG_WRONG,"Matrix X must be MATDENSE matrix");
3148 
3149   ierr = MatDenseGetArray(B,&bb);CHKERRQ(ierr);
3150   ierr = MatDenseGetArray(X,&xx);CHKERRQ(ierr);
3151   ierr = MatGetLocalSize(B,&m,NULL);CHKERRQ(ierr);  /* number local rows */
3152   ierr = MatGetSize(B,NULL,&N);CHKERRQ(ierr);       /* total columns in dense matrix */
3153   ierr = MatGetVecs(A,&x,&b);CHKERRQ(ierr);
3154   for (i=0; i<N; i++) {
3155     ierr = VecPlaceArray(b,bb + i*m);CHKERRQ(ierr);
3156     ierr = VecPlaceArray(x,xx + i*m);CHKERRQ(ierr);
3157     ierr = MatSolve(A,b,x);CHKERRQ(ierr);
3158     ierr = VecResetArray(x);CHKERRQ(ierr);
3159     ierr = VecResetArray(b);CHKERRQ(ierr);
3160   }
3161   ierr = VecDestroy(&b);CHKERRQ(ierr);
3162   ierr = VecDestroy(&x);CHKERRQ(ierr);
3163   ierr = MatDenseRestoreArray(B,&bb);CHKERRQ(ierr);
3164   ierr = MatDenseRestoreArray(X,&xx);CHKERRQ(ierr);
3165   PetscFunctionReturn(0);
3166 }
3167 
3168 #undef __FUNCT__
3169 #define __FUNCT__ "MatMatSolve"
3170 /*@
3171    MatMatSolve - Solves A X = B, given a factored matrix.
3172 
3173    Neighbor-wise Collective on Mat
3174 
3175    Input Parameters:
3176 +  mat - the factored matrix
3177 -  B - the right-hand-side matrix  (dense matrix)
3178 
3179    Output Parameter:
3180 .  X - the result matrix (dense matrix)
3181 
3182    Notes:
3183    The matrices b and x cannot be the same.  I.e., one cannot
3184    call MatMatSolve(A,x,x).
3185 
3186    Notes:
3187    Most users should usually employ the simplified KSP interface for linear solvers
3188    instead of working directly with matrix algebra routines such as this.
3189    See, e.g., KSPCreate(). However KSP can only solve for one vector (column of X)
3190    at a time.
3191 
3192    When using SuperLU_Dist as a parallel solver PETSc will use the SuperLU_Dist functionality to solve multiple right hand sides simultaneously. For MUMPS
3193    it calls a separate solve for each right hand side since MUMPS does not yet support distributed right hand sides.
3194 
3195    Since the resulting matrix X must always be dense we do not support sparse representation of the matrix B.
3196 
3197    Level: developer
3198 
3199    Concepts: matrices^triangular solves
3200 
3201 .seealso: MatMatSolveAdd(), MatMatSolveTranspose(), MatMatSolveTransposeAdd(), MatLUFactor(), MatCholeskyFactor()
3202 @*/
3203 PetscErrorCode  MatMatSolve(Mat A,Mat B,Mat X)
3204 {
3205   PetscErrorCode ierr;
3206 
3207   PetscFunctionBegin;
3208   PetscValidHeaderSpecific(A,MAT_CLASSID,1);
3209   PetscValidType(A,1);
3210   PetscValidHeaderSpecific(B,MAT_CLASSID,2);
3211   PetscValidHeaderSpecific(X,MAT_CLASSID,3);
3212   PetscCheckSameComm(A,1,B,2);
3213   PetscCheckSameComm(A,1,X,3);
3214   if (X == B) SETERRQ(PetscObjectComm((PetscObject)A),PETSC_ERR_ARG_IDN,"X and B must be different matrices");
3215   if (!A->factortype) SETERRQ(PetscObjectComm((PetscObject)A),PETSC_ERR_ARG_WRONGSTATE,"Unfactored matrix");
3216   if (A->cmap->N != X->rmap->N) SETERRQ2(PetscObjectComm((PetscObject)A),PETSC_ERR_ARG_SIZ,"Mat A,Mat X: global dim %D %D",A->cmap->N,X->rmap->N);
3217   if (A->rmap->N != B->rmap->N) SETERRQ2(PetscObjectComm((PetscObject)A),PETSC_ERR_ARG_SIZ,"Mat A,Mat B: global dim %D %D",A->rmap->N,B->rmap->N);
3218   if (A->rmap->n != B->rmap->n) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_ARG_SIZ,"Mat A,Mat B: local dim %D %D",A->rmap->n,B->rmap->n);
3219   if (!A->rmap->N && !A->cmap->N) PetscFunctionReturn(0);
3220   MatCheckPreallocated(A,1);
3221 
3222   ierr = PetscLogEventBegin(MAT_MatSolve,A,B,X,0);CHKERRQ(ierr);
3223   if (!A->ops->matsolve) {
3224     ierr = PetscInfo1(A,"Mat type %s using basic MatMatSolve",((PetscObject)A)->type_name);CHKERRQ(ierr);
3225     ierr = MatMatSolve_Basic(A,B,X);CHKERRQ(ierr);
3226   } else {
3227     ierr = (*A->ops->matsolve)(A,B,X);CHKERRQ(ierr);
3228   }
3229   ierr = PetscLogEventEnd(MAT_MatSolve,A,B,X,0);CHKERRQ(ierr);
3230   ierr = PetscObjectStateIncrease((PetscObject)X);CHKERRQ(ierr);
3231   PetscFunctionReturn(0);
3232 }
3233 
3234 
3235 #undef __FUNCT__
3236 #define __FUNCT__ "MatForwardSolve"
3237 /*@
3238    MatForwardSolve - Solves L x = b, given a factored matrix, A = LU, or
3239                             U^T*D^(1/2) x = b, given a factored symmetric matrix, A = U^T*D*U,
3240 
3241    Neighbor-wise Collective on Mat and Vec
3242 
3243    Input Parameters:
3244 +  mat - the factored matrix
3245 -  b - the right-hand-side vector
3246 
3247    Output Parameter:
3248 .  x - the result vector
3249 
3250    Notes:
3251    MatSolve() should be used for most applications, as it performs
3252    a forward solve followed by a backward solve.
3253 
3254    The vectors b and x cannot be the same,  i.e., one cannot
3255    call MatForwardSolve(A,x,x).
3256 
3257    For matrix in seqsbaij format with block size larger than 1,
3258    the diagonal blocks are not implemented as D = D^(1/2) * D^(1/2) yet.
3259    MatForwardSolve() solves U^T*D y = b, and
3260    MatBackwardSolve() solves U x = y.
3261    Thus they do not provide a symmetric preconditioner.
3262 
3263    Most users should employ the simplified KSP interface for linear solvers
3264    instead of working directly with matrix algebra routines such as this.
3265    See, e.g., KSPCreate().
3266 
3267    Level: developer
3268 
3269    Concepts: matrices^forward solves
3270 
3271 .seealso: MatSolve(), MatBackwardSolve()
3272 @*/
3273 PetscErrorCode  MatForwardSolve(Mat mat,Vec b,Vec x)
3274 {
3275   PetscErrorCode ierr;
3276 
3277   PetscFunctionBegin;
3278   PetscValidHeaderSpecific(mat,MAT_CLASSID,1);
3279   PetscValidType(mat,1);
3280   PetscValidHeaderSpecific(b,VEC_CLASSID,2);
3281   PetscValidHeaderSpecific(x,VEC_CLASSID,3);
3282   PetscCheckSameComm(mat,1,b,2);
3283   PetscCheckSameComm(mat,1,x,3);
3284   if (x == b) SETERRQ(PetscObjectComm((PetscObject)mat),PETSC_ERR_ARG_IDN,"x and b must be different vectors");
3285   if (!mat->factortype) SETERRQ(PetscObjectComm((PetscObject)mat),PETSC_ERR_ARG_WRONGSTATE,"Unfactored matrix");
3286   if (!mat->ops->forwardsolve) SETERRQ1(PetscObjectComm((PetscObject)mat),PETSC_ERR_SUP,"Mat type %s",((PetscObject)mat)->type_name);
3287   if (mat->cmap->N != x->map->N) SETERRQ2(PetscObjectComm((PetscObject)mat),PETSC_ERR_ARG_SIZ,"Mat mat,Vec x: global dim %D %D",mat->cmap->N,x->map->N);
3288   if (mat->rmap->N != b->map->N) SETERRQ2(PetscObjectComm((PetscObject)mat),PETSC_ERR_ARG_SIZ,"Mat mat,Vec b: global dim %D %D",mat->rmap->N,b->map->N);
3289   if (mat->rmap->n != b->map->n) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_ARG_SIZ,"Mat mat,Vec b: local dim %D %D",mat->rmap->n,b->map->n);
3290   MatCheckPreallocated(mat,1);
3291   ierr = PetscLogEventBegin(MAT_ForwardSolve,mat,b,x,0);CHKERRQ(ierr);
3292   ierr = (*mat->ops->forwardsolve)(mat,b,x);CHKERRQ(ierr);
3293   ierr = PetscLogEventEnd(MAT_ForwardSolve,mat,b,x,0);CHKERRQ(ierr);
3294   ierr = PetscObjectStateIncrease((PetscObject)x);CHKERRQ(ierr);
3295   PetscFunctionReturn(0);
3296 }
3297 
3298 #undef __FUNCT__
3299 #define __FUNCT__ "MatBackwardSolve"
3300 /*@
3301    MatBackwardSolve - Solves U x = b, given a factored matrix, A = LU.
3302                              D^(1/2) U x = b, given a factored symmetric matrix, A = U^T*D*U,
3303 
3304    Neighbor-wise Collective on Mat and Vec
3305 
3306    Input Parameters:
3307 +  mat - the factored matrix
3308 -  b - the right-hand-side vector
3309 
3310    Output Parameter:
3311 .  x - the result vector
3312 
3313    Notes:
3314    MatSolve() should be used for most applications, as it performs
3315    a forward solve followed by a backward solve.
3316 
3317    The vectors b and x cannot be the same.  I.e., one cannot
3318    call MatBackwardSolve(A,x,x).
3319 
3320    For matrix in seqsbaij format with block size larger than 1,
3321    the diagonal blocks are not implemented as D = D^(1/2) * D^(1/2) yet.
3322    MatForwardSolve() solves U^T*D y = b, and
3323    MatBackwardSolve() solves U x = y.
3324    Thus they do not provide a symmetric preconditioner.
3325 
3326    Most users should employ the simplified KSP interface for linear solvers
3327    instead of working directly with matrix algebra routines such as this.
3328    See, e.g., KSPCreate().
3329 
3330    Level: developer
3331 
3332    Concepts: matrices^backward solves
3333 
3334 .seealso: MatSolve(), MatForwardSolve()
3335 @*/
3336 PetscErrorCode  MatBackwardSolve(Mat mat,Vec b,Vec x)
3337 {
3338   PetscErrorCode ierr;
3339 
3340   PetscFunctionBegin;
3341   PetscValidHeaderSpecific(mat,MAT_CLASSID,1);
3342   PetscValidType(mat,1);
3343   PetscValidHeaderSpecific(b,VEC_CLASSID,2);
3344   PetscValidHeaderSpecific(x,VEC_CLASSID,3);
3345   PetscCheckSameComm(mat,1,b,2);
3346   PetscCheckSameComm(mat,1,x,3);
3347   if (x == b) SETERRQ(PetscObjectComm((PetscObject)mat),PETSC_ERR_ARG_IDN,"x and b must be different vectors");
3348   if (!mat->factortype) SETERRQ(PetscObjectComm((PetscObject)mat),PETSC_ERR_ARG_WRONGSTATE,"Unfactored matrix");
3349   if (!mat->ops->backwardsolve) SETERRQ1(PetscObjectComm((PetscObject)mat),PETSC_ERR_SUP,"Mat type %s",((PetscObject)mat)->type_name);
3350   if (mat->cmap->N != x->map->N) SETERRQ2(PetscObjectComm((PetscObject)mat),PETSC_ERR_ARG_SIZ,"Mat mat,Vec x: global dim %D %D",mat->cmap->N,x->map->N);
3351   if (mat->rmap->N != b->map->N) SETERRQ2(PetscObjectComm((PetscObject)mat),PETSC_ERR_ARG_SIZ,"Mat mat,Vec b: global dim %D %D",mat->rmap->N,b->map->N);
3352   if (mat->rmap->n != b->map->n) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_ARG_SIZ,"Mat mat,Vec b: local dim %D %D",mat->rmap->n,b->map->n);
3353   MatCheckPreallocated(mat,1);
3354 
3355   ierr = PetscLogEventBegin(MAT_BackwardSolve,mat,b,x,0);CHKERRQ(ierr);
3356   ierr = (*mat->ops->backwardsolve)(mat,b,x);CHKERRQ(ierr);
3357   ierr = PetscLogEventEnd(MAT_BackwardSolve,mat,b,x,0);CHKERRQ(ierr);
3358   ierr = PetscObjectStateIncrease((PetscObject)x);CHKERRQ(ierr);
3359   PetscFunctionReturn(0);
3360 }
3361 
3362 #undef __FUNCT__
3363 #define __FUNCT__ "MatSolveAdd"
3364 /*@
3365    MatSolveAdd - Computes x = y + inv(A)*b, given a factored matrix.
3366 
3367    Neighbor-wise Collective on Mat and Vec
3368 
3369    Input Parameters:
3370 +  mat - the factored matrix
3371 .  b - the right-hand-side vector
3372 -  y - the vector to be added to
3373 
3374    Output Parameter:
3375 .  x - the result vector
3376 
3377    Notes:
3378    The vectors b and x cannot be the same.  I.e., one cannot
3379    call MatSolveAdd(A,x,y,x).
3380 
3381    Most users should employ the simplified KSP interface for linear solvers
3382    instead of working directly with matrix algebra routines such as this.
3383    See, e.g., KSPCreate().
3384 
3385    Level: developer
3386 
3387    Concepts: matrices^triangular solves
3388 
3389 .seealso: MatSolve(), MatSolveTranspose(), MatSolveTransposeAdd()
3390 @*/
3391 PetscErrorCode  MatSolveAdd(Mat mat,Vec b,Vec y,Vec x)
3392 {
3393   PetscScalar    one = 1.0;
3394   Vec            tmp;
3395   PetscErrorCode ierr;
3396 
3397   PetscFunctionBegin;
3398   PetscValidHeaderSpecific(mat,MAT_CLASSID,1);
3399   PetscValidType(mat,1);
3400   PetscValidHeaderSpecific(y,VEC_CLASSID,2);
3401   PetscValidHeaderSpecific(b,VEC_CLASSID,3);
3402   PetscValidHeaderSpecific(x,VEC_CLASSID,4);
3403   PetscCheckSameComm(mat,1,b,2);
3404   PetscCheckSameComm(mat,1,y,2);
3405   PetscCheckSameComm(mat,1,x,3);
3406   if (x == b) SETERRQ(PetscObjectComm((PetscObject)mat),PETSC_ERR_ARG_IDN,"x and b must be different vectors");
3407   if (!mat->factortype) SETERRQ(PetscObjectComm((PetscObject)mat),PETSC_ERR_ARG_WRONGSTATE,"Unfactored matrix");
3408   if (mat->cmap->N != x->map->N) SETERRQ2(PetscObjectComm((PetscObject)mat),PETSC_ERR_ARG_SIZ,"Mat mat,Vec x: global dim %D %D",mat->cmap->N,x->map->N);
3409   if (mat->rmap->N != b->map->N) SETERRQ2(PetscObjectComm((PetscObject)mat),PETSC_ERR_ARG_SIZ,"Mat mat,Vec b: global dim %D %D",mat->rmap->N,b->map->N);
3410   if (mat->rmap->N != y->map->N) SETERRQ2(PetscObjectComm((PetscObject)mat),PETSC_ERR_ARG_SIZ,"Mat mat,Vec y: global dim %D %D",mat->rmap->N,y->map->N);
3411   if (mat->rmap->n != b->map->n) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_ARG_SIZ,"Mat mat,Vec b: local dim %D %D",mat->rmap->n,b->map->n);
3412   if (x->map->n != y->map->n) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_ARG_SIZ,"Vec x,Vec y: local dim %D %D",x->map->n,y->map->n);
3413   MatCheckPreallocated(mat,1);
3414 
3415   ierr = PetscLogEventBegin(MAT_SolveAdd,mat,b,x,y);CHKERRQ(ierr);
3416   if (mat->ops->solveadd) {
3417     ierr = (*mat->ops->solveadd)(mat,b,y,x);CHKERRQ(ierr);
3418   } else {
3419     /* do the solve then the add manually */
3420     if (x != y) {
3421       ierr = MatSolve(mat,b,x);CHKERRQ(ierr);
3422       ierr = VecAXPY(x,one,y);CHKERRQ(ierr);
3423     } else {
3424       ierr = VecDuplicate(x,&tmp);CHKERRQ(ierr);
3425       ierr = PetscLogObjectParent(mat,tmp);CHKERRQ(ierr);
3426       ierr = VecCopy(x,tmp);CHKERRQ(ierr);
3427       ierr = MatSolve(mat,b,x);CHKERRQ(ierr);
3428       ierr = VecAXPY(x,one,tmp);CHKERRQ(ierr);
3429       ierr = VecDestroy(&tmp);CHKERRQ(ierr);
3430     }
3431   }
3432   ierr = PetscLogEventEnd(MAT_SolveAdd,mat,b,x,y);CHKERRQ(ierr);
3433   ierr = PetscObjectStateIncrease((PetscObject)x);CHKERRQ(ierr);
3434   PetscFunctionReturn(0);
3435 }
3436 
3437 #undef __FUNCT__
3438 #define __FUNCT__ "MatSolveTranspose"
3439 /*@
3440    MatSolveTranspose - Solves A' x = b, given a factored matrix.
3441 
3442    Neighbor-wise Collective on Mat and Vec
3443 
3444    Input Parameters:
3445 +  mat - the factored matrix
3446 -  b - the right-hand-side vector
3447 
3448    Output Parameter:
3449 .  x - the result vector
3450 
3451    Notes:
3452    The vectors b and x cannot be the same.  I.e., one cannot
3453    call MatSolveTranspose(A,x,x).
3454 
3455    Most users should employ the simplified KSP interface for linear solvers
3456    instead of working directly with matrix algebra routines such as this.
3457    See, e.g., KSPCreate().
3458 
3459    Level: developer
3460 
3461    Concepts: matrices^triangular solves
3462 
3463 .seealso: MatSolve(), MatSolveAdd(), MatSolveTransposeAdd()
3464 @*/
3465 PetscErrorCode  MatSolveTranspose(Mat mat,Vec b,Vec x)
3466 {
3467   PetscErrorCode ierr;
3468 
3469   PetscFunctionBegin;
3470   PetscValidHeaderSpecific(mat,MAT_CLASSID,1);
3471   PetscValidType(mat,1);
3472   PetscValidHeaderSpecific(b,VEC_CLASSID,2);
3473   PetscValidHeaderSpecific(x,VEC_CLASSID,3);
3474   PetscCheckSameComm(mat,1,b,2);
3475   PetscCheckSameComm(mat,1,x,3);
3476   if (!mat->factortype) SETERRQ(PetscObjectComm((PetscObject)mat),PETSC_ERR_ARG_WRONGSTATE,"Unfactored matrix");
3477   if (x == b) SETERRQ(PetscObjectComm((PetscObject)mat),PETSC_ERR_ARG_IDN,"x and b must be different vectors");
3478   if (!mat->ops->solvetranspose) SETERRQ1(PetscObjectComm((PetscObject)mat),PETSC_ERR_SUP,"Matrix type %s",((PetscObject)mat)->type_name);
3479   if (mat->rmap->N != x->map->N) SETERRQ2(PetscObjectComm((PetscObject)mat),PETSC_ERR_ARG_SIZ,"Mat mat,Vec x: global dim %D %D",mat->rmap->N,x->map->N);
3480   if (mat->cmap->N != b->map->N) SETERRQ2(PetscObjectComm((PetscObject)mat),PETSC_ERR_ARG_SIZ,"Mat mat,Vec b: global dim %D %D",mat->cmap->N,b->map->N);
3481   MatCheckPreallocated(mat,1);
3482   ierr = PetscLogEventBegin(MAT_SolveTranspose,mat,b,x,0);CHKERRQ(ierr);
3483   ierr = (*mat->ops->solvetranspose)(mat,b,x);CHKERRQ(ierr);
3484   ierr = PetscLogEventEnd(MAT_SolveTranspose,mat,b,x,0);CHKERRQ(ierr);
3485   ierr = PetscObjectStateIncrease((PetscObject)x);CHKERRQ(ierr);
3486   PetscFunctionReturn(0);
3487 }
3488 
3489 #undef __FUNCT__
3490 #define __FUNCT__ "MatSolveTransposeAdd"
3491 /*@
3492    MatSolveTransposeAdd - Computes x = y + inv(Transpose(A)) b, given a
3493                       factored matrix.
3494 
3495    Neighbor-wise Collective on Mat and Vec
3496 
3497    Input Parameters:
3498 +  mat - the factored matrix
3499 .  b - the right-hand-side vector
3500 -  y - the vector to be added to
3501 
3502    Output Parameter:
3503 .  x - the result vector
3504 
3505    Notes:
3506    The vectors b and x cannot be the same.  I.e., one cannot
3507    call MatSolveTransposeAdd(A,x,y,x).
3508 
3509    Most users should employ the simplified KSP interface for linear solvers
3510    instead of working directly with matrix algebra routines such as this.
3511    See, e.g., KSPCreate().
3512 
3513    Level: developer
3514 
3515    Concepts: matrices^triangular solves
3516 
3517 .seealso: MatSolve(), MatSolveAdd(), MatSolveTranspose()
3518 @*/
3519 PetscErrorCode  MatSolveTransposeAdd(Mat mat,Vec b,Vec y,Vec x)
3520 {
3521   PetscScalar    one = 1.0;
3522   PetscErrorCode ierr;
3523   Vec            tmp;
3524 
3525   PetscFunctionBegin;
3526   PetscValidHeaderSpecific(mat,MAT_CLASSID,1);
3527   PetscValidType(mat,1);
3528   PetscValidHeaderSpecific(y,VEC_CLASSID,2);
3529   PetscValidHeaderSpecific(b,VEC_CLASSID,3);
3530   PetscValidHeaderSpecific(x,VEC_CLASSID,4);
3531   PetscCheckSameComm(mat,1,b,2);
3532   PetscCheckSameComm(mat,1,y,3);
3533   PetscCheckSameComm(mat,1,x,4);
3534   if (x == b) SETERRQ(PetscObjectComm((PetscObject)mat),PETSC_ERR_ARG_IDN,"x and b must be different vectors");
3535   if (!mat->factortype) SETERRQ(PetscObjectComm((PetscObject)mat),PETSC_ERR_ARG_WRONGSTATE,"Unfactored matrix");
3536   if (mat->rmap->N != x->map->N) SETERRQ2(PetscObjectComm((PetscObject)mat),PETSC_ERR_ARG_SIZ,"Mat mat,Vec x: global dim %D %D",mat->rmap->N,x->map->N);
3537   if (mat->cmap->N != b->map->N) SETERRQ2(PetscObjectComm((PetscObject)mat),PETSC_ERR_ARG_SIZ,"Mat mat,Vec b: global dim %D %D",mat->cmap->N,b->map->N);
3538   if (mat->cmap->N != y->map->N) SETERRQ2(PetscObjectComm((PetscObject)mat),PETSC_ERR_ARG_SIZ,"Mat mat,Vec y: global dim %D %D",mat->cmap->N,y->map->N);
3539   if (x->map->n != y->map->n) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_ARG_SIZ,"Vec x,Vec y: local dim %D %D",x->map->n,y->map->n);
3540   MatCheckPreallocated(mat,1);
3541 
3542   ierr = PetscLogEventBegin(MAT_SolveTransposeAdd,mat,b,x,y);CHKERRQ(ierr);
3543   if (mat->ops->solvetransposeadd) {
3544     ierr = (*mat->ops->solvetransposeadd)(mat,b,y,x);CHKERRQ(ierr);
3545   } else {
3546     /* do the solve then the add manually */
3547     if (x != y) {
3548       ierr = MatSolveTranspose(mat,b,x);CHKERRQ(ierr);
3549       ierr = VecAXPY(x,one,y);CHKERRQ(ierr);
3550     } else {
3551       ierr = VecDuplicate(x,&tmp);CHKERRQ(ierr);
3552       ierr = PetscLogObjectParent(mat,tmp);CHKERRQ(ierr);
3553       ierr = VecCopy(x,tmp);CHKERRQ(ierr);
3554       ierr = MatSolveTranspose(mat,b,x);CHKERRQ(ierr);
3555       ierr = VecAXPY(x,one,tmp);CHKERRQ(ierr);
3556       ierr = VecDestroy(&tmp);CHKERRQ(ierr);
3557     }
3558   }
3559   ierr = PetscLogEventEnd(MAT_SolveTransposeAdd,mat,b,x,y);CHKERRQ(ierr);
3560   ierr = PetscObjectStateIncrease((PetscObject)x);CHKERRQ(ierr);
3561   PetscFunctionReturn(0);
3562 }
3563 /* ----------------------------------------------------------------*/
3564 
3565 #undef __FUNCT__
3566 #define __FUNCT__ "MatSOR"
3567 /*@
3568    MatSOR - Computes relaxation (SOR, Gauss-Seidel) sweeps.
3569 
3570    Neighbor-wise Collective on Mat and Vec
3571 
3572    Input Parameters:
3573 +  mat - the matrix
3574 .  b - the right hand side
3575 .  omega - the relaxation factor
3576 .  flag - flag indicating the type of SOR (see below)
3577 .  shift -  diagonal shift
3578 .  its - the number of iterations
3579 -  lits - the number of local iterations
3580 
3581    Output Parameters:
3582 .  x - the solution (can contain an initial guess, use option SOR_ZERO_INITIAL_GUESS to indicate no guess)
3583 
3584    SOR Flags:
3585 .     SOR_FORWARD_SWEEP - forward SOR
3586 .     SOR_BACKWARD_SWEEP - backward SOR
3587 .     SOR_SYMMETRIC_SWEEP - SSOR (symmetric SOR)
3588 .     SOR_LOCAL_FORWARD_SWEEP - local forward SOR
3589 .     SOR_LOCAL_BACKWARD_SWEEP - local forward SOR
3590 .     SOR_LOCAL_SYMMETRIC_SWEEP - local SSOR
3591 .     SOR_APPLY_UPPER, SOR_APPLY_LOWER - applies
3592          upper/lower triangular part of matrix to
3593          vector (with omega)
3594 .     SOR_ZERO_INITIAL_GUESS - zero initial guess
3595 
3596    Notes:
3597    SOR_LOCAL_FORWARD_SWEEP, SOR_LOCAL_BACKWARD_SWEEP, and
3598    SOR_LOCAL_SYMMETRIC_SWEEP perform separate independent smoothings
3599    on each processor.
3600 
3601    Application programmers will not generally use MatSOR() directly,
3602    but instead will employ the KSP/PC interface.
3603 
3604    Notes: for BAIJ, SBAIJ, and AIJ matrices with Inodes this does a block SOR smoothing, otherwise it does a pointwise smoothing
3605 
3606    Notes for Advanced Users:
3607    The flags are implemented as bitwise inclusive or operations.
3608    For example, use (SOR_ZERO_INITIAL_GUESS | SOR_SYMMETRIC_SWEEP)
3609    to specify a zero initial guess for SSOR.
3610 
3611    Most users should employ the simplified KSP interface for linear solvers
3612    instead of working directly with matrix algebra routines such as this.
3613    See, e.g., KSPCreate().
3614 
3615    Vectors x and b CANNOT be the same
3616 
3617    Developer Note: We should add block SOR support for AIJ matrices with block size set to great than one and no inodes
3618 
3619    Level: developer
3620 
3621    Concepts: matrices^relaxation
3622    Concepts: matrices^SOR
3623    Concepts: matrices^Gauss-Seidel
3624 
3625 @*/
3626 PetscErrorCode  MatSOR(Mat mat,Vec b,PetscReal omega,MatSORType flag,PetscReal shift,PetscInt its,PetscInt lits,Vec x)
3627 {
3628   PetscErrorCode ierr;
3629 
3630   PetscFunctionBegin;
3631   PetscValidHeaderSpecific(mat,MAT_CLASSID,1);
3632   PetscValidType(mat,1);
3633   PetscValidHeaderSpecific(b,VEC_CLASSID,2);
3634   PetscValidHeaderSpecific(x,VEC_CLASSID,8);
3635   PetscCheckSameComm(mat,1,b,2);
3636   PetscCheckSameComm(mat,1,x,8);
3637   if (!mat->ops->sor) SETERRQ1(PetscObjectComm((PetscObject)mat),PETSC_ERR_SUP,"Mat type %s",((PetscObject)mat)->type_name);
3638   if (!mat->assembled) SETERRQ(PetscObjectComm((PetscObject)mat),PETSC_ERR_ARG_WRONGSTATE,"Not for unassembled matrix");
3639   if (mat->factortype) SETERRQ(PetscObjectComm((PetscObject)mat),PETSC_ERR_ARG_WRONGSTATE,"Not for factored matrix");
3640   if (mat->cmap->N != x->map->N) SETERRQ2(PetscObjectComm((PetscObject)mat),PETSC_ERR_ARG_SIZ,"Mat mat,Vec x: global dim %D %D",mat->cmap->N,x->map->N);
3641   if (mat->rmap->N != b->map->N) SETERRQ2(PetscObjectComm((PetscObject)mat),PETSC_ERR_ARG_SIZ,"Mat mat,Vec b: global dim %D %D",mat->rmap->N,b->map->N);
3642   if (mat->rmap->n != b->map->n) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_ARG_SIZ,"Mat mat,Vec b: local dim %D %D",mat->rmap->n,b->map->n);
3643   if (its <= 0) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONG,"Relaxation requires global its %D positive",its);
3644   if (lits <= 0) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONG,"Relaxation requires local its %D positive",lits);
3645   if (b == x) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_IDN,"b and x vector cannot be the same");
3646 
3647   MatCheckPreallocated(mat,1);
3648   ierr = PetscLogEventBegin(MAT_SOR,mat,b,x,0);CHKERRQ(ierr);
3649   ierr =(*mat->ops->sor)(mat,b,omega,flag,shift,its,lits,x);CHKERRQ(ierr);
3650   ierr = PetscLogEventEnd(MAT_SOR,mat,b,x,0);CHKERRQ(ierr);
3651   ierr = PetscObjectStateIncrease((PetscObject)x);CHKERRQ(ierr);
3652   PetscFunctionReturn(0);
3653 }
3654 
3655 #undef __FUNCT__
3656 #define __FUNCT__ "MatCopy_Basic"
3657 /*
3658       Default matrix copy routine.
3659 */
3660 PetscErrorCode MatCopy_Basic(Mat A,Mat B,MatStructure str)
3661 {
3662   PetscErrorCode    ierr;
3663   PetscInt          i,rstart = 0,rend = 0,nz;
3664   const PetscInt    *cwork;
3665   const PetscScalar *vwork;
3666 
3667   PetscFunctionBegin;
3668   if (B->assembled) {
3669     ierr = MatZeroEntries(B);CHKERRQ(ierr);
3670   }
3671   ierr = MatGetOwnershipRange(A,&rstart,&rend);CHKERRQ(ierr);
3672   for (i=rstart; i<rend; i++) {
3673     ierr = MatGetRow(A,i,&nz,&cwork,&vwork);CHKERRQ(ierr);
3674     ierr = MatSetValues(B,1,&i,nz,cwork,vwork,INSERT_VALUES);CHKERRQ(ierr);
3675     ierr = MatRestoreRow(A,i,&nz,&cwork,&vwork);CHKERRQ(ierr);
3676   }
3677   ierr = MatAssemblyBegin(B,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
3678   ierr = MatAssemblyEnd(B,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
3679   ierr = PetscObjectStateIncrease((PetscObject)B);CHKERRQ(ierr);
3680   PetscFunctionReturn(0);
3681 }
3682 
3683 #undef __FUNCT__
3684 #define __FUNCT__ "MatCopy"
3685 /*@
3686    MatCopy - Copys a matrix to another matrix.
3687 
3688    Collective on Mat
3689 
3690    Input Parameters:
3691 +  A - the matrix
3692 -  str - SAME_NONZERO_PATTERN or DIFFERENT_NONZERO_PATTERN
3693 
3694    Output Parameter:
3695 .  B - where the copy is put
3696 
3697    Notes:
3698    If you use SAME_NONZERO_PATTERN then the two matrices had better have the
3699    same nonzero pattern or the routine will crash.
3700 
3701    MatCopy() copies the matrix entries of a matrix to another existing
3702    matrix (after first zeroing the second matrix).  A related routine is
3703    MatConvert(), which first creates a new matrix and then copies the data.
3704 
3705    Level: intermediate
3706 
3707    Concepts: matrices^copying
3708 
3709 .seealso: MatConvert(), MatDuplicate()
3710 
3711 @*/
3712 PetscErrorCode  MatCopy(Mat A,Mat B,MatStructure str)
3713 {
3714   PetscErrorCode ierr;
3715   PetscInt       i;
3716 
3717   PetscFunctionBegin;
3718   PetscValidHeaderSpecific(A,MAT_CLASSID,1);
3719   PetscValidHeaderSpecific(B,MAT_CLASSID,2);
3720   PetscValidType(A,1);
3721   PetscValidType(B,2);
3722   PetscCheckSameComm(A,1,B,2);
3723   MatCheckPreallocated(B,2);
3724   if (!A->assembled) SETERRQ(PetscObjectComm((PetscObject)A),PETSC_ERR_ARG_WRONGSTATE,"Not for unassembled matrix");
3725   if (A->factortype) SETERRQ(PetscObjectComm((PetscObject)A),PETSC_ERR_ARG_WRONGSTATE,"Not for factored matrix");
3726   if (A->rmap->N != B->rmap->N || A->cmap->N != B->cmap->N) SETERRQ4(PetscObjectComm((PetscObject)A),PETSC_ERR_ARG_SIZ,"Mat A,Mat B: global dim (%D,%D) (%D,%D)",A->rmap->N,B->rmap->N,A->cmap->N,B->cmap->N);
3727   MatCheckPreallocated(A,1);
3728 
3729   ierr = PetscLogEventBegin(MAT_Copy,A,B,0,0);CHKERRQ(ierr);
3730   if (A->ops->copy) {
3731     ierr = (*A->ops->copy)(A,B,str);CHKERRQ(ierr);
3732   } else { /* generic conversion */
3733     ierr = MatCopy_Basic(A,B,str);CHKERRQ(ierr);
3734   }
3735 
3736   B->stencil.dim = A->stencil.dim;
3737   B->stencil.noc = A->stencil.noc;
3738   for (i=0; i<=A->stencil.dim; i++) {
3739     B->stencil.dims[i]   = A->stencil.dims[i];
3740     B->stencil.starts[i] = A->stencil.starts[i];
3741   }
3742 
3743   ierr = PetscLogEventEnd(MAT_Copy,A,B,0,0);CHKERRQ(ierr);
3744   ierr = PetscObjectStateIncrease((PetscObject)B);CHKERRQ(ierr);
3745   PetscFunctionReturn(0);
3746 }
3747 
3748 #undef __FUNCT__
3749 #define __FUNCT__ "MatConvert"
3750 /*@C
3751    MatConvert - Converts a matrix to another matrix, either of the same
3752    or different type.
3753 
3754    Collective on Mat
3755 
3756    Input Parameters:
3757 +  mat - the matrix
3758 .  newtype - new matrix type.  Use MATSAME to create a new matrix of the
3759    same type as the original matrix.
3760 -  reuse - denotes if the destination matrix is to be created or reused.  Currently
3761    MAT_REUSE_MATRIX is only supported for inplace conversion, otherwise use
3762    MAT_INITIAL_MATRIX.
3763 
3764    Output Parameter:
3765 .  M - pointer to place new matrix
3766 
3767    Notes:
3768    MatConvert() first creates a new matrix and then copies the data from
3769    the first matrix.  A related routine is MatCopy(), which copies the matrix
3770    entries of one matrix to another already existing matrix context.
3771 
3772    Cannot be used to convert a sequential matrix to parallel or parallel to sequential,
3773    the MPI communicator of the generated matrix is always the same as the communicator
3774    of the input matrix.
3775 
3776    Level: intermediate
3777 
3778    Concepts: matrices^converting between storage formats
3779 
3780 .seealso: MatCopy(), MatDuplicate()
3781 @*/
3782 PetscErrorCode  MatConvert(Mat mat, MatType newtype,MatReuse reuse,Mat *M)
3783 {
3784   PetscErrorCode ierr;
3785   PetscBool      sametype,issame,flg;
3786   char           convname[256],mtype[256];
3787   Mat            B;
3788 
3789   PetscFunctionBegin;
3790   PetscValidHeaderSpecific(mat,MAT_CLASSID,1);
3791   PetscValidType(mat,1);
3792   PetscValidPointer(M,3);
3793   if (!mat->assembled) SETERRQ(PetscObjectComm((PetscObject)mat),PETSC_ERR_ARG_WRONGSTATE,"Not for unassembled matrix");
3794   if (mat->factortype) SETERRQ(PetscObjectComm((PetscObject)mat),PETSC_ERR_ARG_WRONGSTATE,"Not for factored matrix");
3795   MatCheckPreallocated(mat,1);
3796   ierr = MatSetOption(mat,MAT_NEW_NONZERO_LOCATION_ERR,PETSC_FALSE);CHKERRQ(ierr);
3797 
3798   ierr = PetscOptionsGetString(((PetscObject)mat)->prefix,"-matconvert_type",mtype,256,&flg);CHKERRQ(ierr);
3799   if (flg) {
3800     newtype = mtype;
3801   }
3802   ierr = PetscObjectTypeCompare((PetscObject)mat,newtype,&sametype);CHKERRQ(ierr);
3803   ierr = PetscStrcmp(newtype,"same",&issame);CHKERRQ(ierr);
3804   if ((reuse == MAT_REUSE_MATRIX) && (mat != *M)) SETERRQ(PetscObjectComm((PetscObject)mat),PETSC_ERR_SUP,"MAT_REUSE_MATRIX only supported for in-place conversion currently");
3805 
3806   if ((reuse == MAT_REUSE_MATRIX) && (issame || sametype)) PetscFunctionReturn(0);
3807 
3808   if ((sametype || issame) && (reuse==MAT_INITIAL_MATRIX) && mat->ops->duplicate) {
3809     ierr = (*mat->ops->duplicate)(mat,MAT_COPY_VALUES,M);CHKERRQ(ierr);
3810   } else {
3811     PetscErrorCode (*conv)(Mat, MatType,MatReuse,Mat*)=NULL;
3812     const char     *prefix[3] = {"seq","mpi",""};
3813     PetscInt       i;
3814     /*
3815        Order of precedence:
3816        1) See if a specialized converter is known to the current matrix.
3817        2) See if a specialized converter is known to the desired matrix class.
3818        3) See if a good general converter is registered for the desired class
3819           (as of 6/27/03 only MATMPIADJ falls into this category).
3820        4) See if a good general converter is known for the current matrix.
3821        5) Use a really basic converter.
3822     */
3823 
3824     /* 1) See if a specialized converter is known to the current matrix and the desired class */
3825     for (i=0; i<3; i++) {
3826       ierr = PetscStrcpy(convname,"MatConvert_");CHKERRQ(ierr);
3827       ierr = PetscStrcat(convname,((PetscObject)mat)->type_name);CHKERRQ(ierr);
3828       ierr = PetscStrcat(convname,"_");CHKERRQ(ierr);
3829       ierr = PetscStrcat(convname,prefix[i]);CHKERRQ(ierr);
3830       ierr = PetscStrcat(convname,issame ? ((PetscObject)mat)->type_name : newtype);CHKERRQ(ierr);
3831       ierr = PetscStrcat(convname,"_C");CHKERRQ(ierr);
3832       ierr = PetscObjectQueryFunction((PetscObject)mat,convname,&conv);CHKERRQ(ierr);
3833       if (conv) goto foundconv;
3834     }
3835 
3836     /* 2)  See if a specialized converter is known to the desired matrix class. */
3837     ierr = MatCreate(PetscObjectComm((PetscObject)mat),&B);CHKERRQ(ierr);
3838     ierr = MatSetSizes(B,mat->rmap->n,mat->cmap->n,mat->rmap->N,mat->cmap->N);CHKERRQ(ierr);
3839     ierr = MatSetType(B,newtype);CHKERRQ(ierr);
3840     for (i=0; i<3; i++) {
3841       ierr = PetscStrcpy(convname,"MatConvert_");CHKERRQ(ierr);
3842       ierr = PetscStrcat(convname,((PetscObject)mat)->type_name);CHKERRQ(ierr);
3843       ierr = PetscStrcat(convname,"_");CHKERRQ(ierr);
3844       ierr = PetscStrcat(convname,prefix[i]);CHKERRQ(ierr);
3845       ierr = PetscStrcat(convname,newtype);CHKERRQ(ierr);
3846       ierr = PetscStrcat(convname,"_C");CHKERRQ(ierr);
3847       ierr = PetscObjectQueryFunction((PetscObject)B,convname,&conv);CHKERRQ(ierr);
3848       if (conv) {
3849         ierr = MatDestroy(&B);CHKERRQ(ierr);
3850         goto foundconv;
3851       }
3852     }
3853 
3854     /* 3) See if a good general converter is registered for the desired class */
3855     conv = B->ops->convertfrom;
3856     ierr = MatDestroy(&B);CHKERRQ(ierr);
3857     if (conv) goto foundconv;
3858 
3859     /* 4) See if a good general converter is known for the current matrix */
3860     if (mat->ops->convert) {
3861       conv = mat->ops->convert;
3862     }
3863     if (conv) goto foundconv;
3864 
3865     /* 5) Use a really basic converter. */
3866     conv = MatConvert_Basic;
3867 
3868 foundconv:
3869     ierr = PetscLogEventBegin(MAT_Convert,mat,0,0,0);CHKERRQ(ierr);
3870     ierr = (*conv)(mat,newtype,reuse,M);CHKERRQ(ierr);
3871     ierr = PetscLogEventEnd(MAT_Convert,mat,0,0,0);CHKERRQ(ierr);
3872   }
3873   ierr = PetscObjectStateIncrease((PetscObject)*M);CHKERRQ(ierr);
3874 
3875   /* Copy Mat options */
3876   if (mat->symmetric) {ierr = MatSetOption(*M,MAT_SYMMETRIC,PETSC_TRUE);CHKERRQ(ierr);}
3877   if (mat->hermitian) {ierr = MatSetOption(*M,MAT_HERMITIAN,PETSC_TRUE);CHKERRQ(ierr);}
3878   PetscFunctionReturn(0);
3879 }
3880 
3881 #undef __FUNCT__
3882 #define __FUNCT__ "MatFactorGetSolverPackage"
3883 /*@C
3884    MatFactorGetSolverPackage - Returns name of the package providing the factorization routines
3885 
3886    Not Collective
3887 
3888    Input Parameter:
3889 .  mat - the matrix, must be a factored matrix
3890 
3891    Output Parameter:
3892 .   type - the string name of the package (do not free this string)
3893 
3894    Notes:
3895       In Fortran you pass in a empty string and the package name will be copied into it.
3896     (Make sure the string is long enough)
3897 
3898    Level: intermediate
3899 
3900 .seealso: MatCopy(), MatDuplicate(), MatGetFactorAvailable(), MatGetFactor()
3901 @*/
3902 PetscErrorCode  MatFactorGetSolverPackage(Mat mat, const MatSolverPackage *type)
3903 {
3904   PetscErrorCode ierr, (*conv)(Mat,const MatSolverPackage*);
3905 
3906   PetscFunctionBegin;
3907   PetscValidHeaderSpecific(mat,MAT_CLASSID,1);
3908   PetscValidType(mat,1);
3909   if (!mat->factortype) SETERRQ(PetscObjectComm((PetscObject)mat),PETSC_ERR_ARG_WRONGSTATE,"Only for factored matrix");
3910   ierr = PetscObjectQueryFunction((PetscObject)mat,"MatFactorGetSolverPackage_C",&conv);CHKERRQ(ierr);
3911   if (!conv) {
3912     *type = MATSOLVERPETSC;
3913   } else {
3914     ierr = (*conv)(mat,type);CHKERRQ(ierr);
3915   }
3916   PetscFunctionReturn(0);
3917 }
3918 
3919 #undef __FUNCT__
3920 #define __FUNCT__ "MatGetFactor"
3921 /*@C
3922    MatGetFactor - Returns a matrix suitable to calls to MatXXFactorSymbolic()
3923 
3924    Collective on Mat
3925 
3926    Input Parameters:
3927 +  mat - the matrix
3928 .  type - name of solver type, for example, superlu, petsc (to use PETSc's default)
3929 -  ftype - factor type, MAT_FACTOR_LU, MAT_FACTOR_CHOLESKY, MAT_FACTOR_ICC, MAT_FACTOR_ILU,
3930 
3931    Output Parameters:
3932 .  f - the factor matrix used with MatXXFactorSymbolic() calls
3933 
3934    Notes:
3935       Some PETSc matrix formats have alternative solvers available that are contained in alternative packages
3936      such as pastix, superlu, mumps etc.
3937 
3938       PETSc must have been ./configure to use the external solver, using the option --download-package
3939 
3940    Level: intermediate
3941 
3942 .seealso: MatCopy(), MatDuplicate(), MatGetFactorAvailable()
3943 @*/
3944 PetscErrorCode  MatGetFactor(Mat mat, const MatSolverPackage type,MatFactorType ftype,Mat *f)
3945 {
3946   PetscErrorCode ierr,(*conv)(Mat,MatFactorType,Mat*);
3947   char           convname[256];
3948 
3949   PetscFunctionBegin;
3950   PetscValidHeaderSpecific(mat,MAT_CLASSID,1);
3951   PetscValidType(mat,1);
3952 
3953   if (mat->factortype) SETERRQ(PetscObjectComm((PetscObject)mat),PETSC_ERR_ARG_WRONGSTATE,"Not for factored matrix");
3954   MatCheckPreallocated(mat,1);
3955 
3956   ierr = PetscStrcpy(convname,"MatGetFactor_");CHKERRQ(ierr);
3957   ierr = PetscStrcat(convname,type);CHKERRQ(ierr);
3958   ierr = PetscStrcat(convname,"_C");CHKERRQ(ierr);
3959   ierr = PetscObjectQueryFunction((PetscObject)mat,convname,&conv);CHKERRQ(ierr);
3960   if (!conv) {
3961     PetscBool flag;
3962     MPI_Comm  comm;
3963 
3964     ierr = PetscObjectGetComm((PetscObject)mat,&comm);CHKERRQ(ierr);
3965     ierr = PetscStrcasecmp(MATSOLVERPETSC,type,&flag);CHKERRQ(ierr);
3966     if (flag) SETERRQ2(comm,PETSC_ERR_SUP,"Matrix format %s does not have a built-in PETSc %s",((PetscObject)mat)->type_name,MatFactorTypes[ftype]);
3967     else SETERRQ4(comm,PETSC_ERR_SUP,"Matrix format %s does not have a solver package %s for %s. Perhaps you must ./configure with --download-%s",((PetscObject)mat)->type_name,type,MatFactorTypes[ftype],type);
3968   }
3969   ierr = (*conv)(mat,ftype,f);CHKERRQ(ierr);
3970   PetscFunctionReturn(0);
3971 }
3972 
3973 #undef __FUNCT__
3974 #define __FUNCT__ "MatGetFactorAvailable"
3975 /*@C
3976    MatGetFactorAvailable - Returns a a flag if matrix supports particular package and factor type
3977 
3978    Not Collective
3979 
3980    Input Parameters:
3981 +  mat - the matrix
3982 .  type - name of solver type, for example, superlu, petsc (to use PETSc's default)
3983 -  ftype - factor type, MAT_FACTOR_LU, MAT_FACTOR_CHOLESKY, MAT_FACTOR_ICC, MAT_FACTOR_ILU,
3984 
3985    Output Parameter:
3986 .    flg - PETSC_TRUE if the factorization is available
3987 
3988    Notes:
3989       Some PETSc matrix formats have alternative solvers available that are contained in alternative packages
3990      such as pastix, superlu, mumps etc.
3991 
3992       PETSc must have been ./configure to use the external solver, using the option --download-package
3993 
3994    Level: intermediate
3995 
3996 .seealso: MatCopy(), MatDuplicate(), MatGetFactor()
3997 @*/
3998 PetscErrorCode  MatGetFactorAvailable(Mat mat, const MatSolverPackage type,MatFactorType ftype,PetscBool  *flg)
3999 {
4000   PetscErrorCode ierr, (*conv)(Mat,MatFactorType,PetscBool*);
4001   char           convname[256];
4002 
4003   PetscFunctionBegin;
4004   PetscValidHeaderSpecific(mat,MAT_CLASSID,1);
4005   PetscValidType(mat,1);
4006 
4007   if (mat->factortype) SETERRQ(PetscObjectComm((PetscObject)mat),PETSC_ERR_ARG_WRONGSTATE,"Not for factored matrix");
4008   MatCheckPreallocated(mat,1);
4009 
4010   ierr = PetscStrcpy(convname,"MatGetFactorAvailable_");CHKERRQ(ierr);
4011   ierr = PetscStrcat(convname,type);CHKERRQ(ierr);
4012   ierr = PetscStrcat(convname,"_C");CHKERRQ(ierr);
4013   ierr = PetscObjectQueryFunction((PetscObject)mat,convname,&conv);CHKERRQ(ierr);
4014   if (!conv) {
4015     *flg = PETSC_FALSE;
4016   } else {
4017     ierr = (*conv)(mat,ftype,flg);CHKERRQ(ierr);
4018   }
4019   PetscFunctionReturn(0);
4020 }
4021 
4022 
4023 #undef __FUNCT__
4024 #define __FUNCT__ "MatDuplicate"
4025 /*@
4026    MatDuplicate - Duplicates a matrix including the non-zero structure.
4027 
4028    Collective on Mat
4029 
4030    Input Parameters:
4031 +  mat - the matrix
4032 -  op - either MAT_DO_NOT_COPY_VALUES or MAT_COPY_VALUES, cause it to copy the numerical values in the matrix
4033         MAT_SHARE_NONZERO_PATTERN to share the nonzero patterns with the previous matrix and not copy them.
4034 
4035    Output Parameter:
4036 .  M - pointer to place new matrix
4037 
4038    Level: intermediate
4039 
4040    Concepts: matrices^duplicating
4041 
4042     Notes: You cannot change the nonzero pattern for the parent or child matrix if you use MAT_SHARE_NONZERO_PATTERN.
4043 
4044 .seealso: MatCopy(), MatConvert()
4045 @*/
4046 PetscErrorCode  MatDuplicate(Mat mat,MatDuplicateOption op,Mat *M)
4047 {
4048   PetscErrorCode ierr;
4049   Mat            B;
4050   PetscInt       i;
4051 
4052   PetscFunctionBegin;
4053   PetscValidHeaderSpecific(mat,MAT_CLASSID,1);
4054   PetscValidType(mat,1);
4055   PetscValidPointer(M,3);
4056   if (!mat->assembled) SETERRQ(PetscObjectComm((PetscObject)mat),PETSC_ERR_ARG_WRONGSTATE,"Not for unassembled matrix");
4057   if (mat->factortype) SETERRQ(PetscObjectComm((PetscObject)mat),PETSC_ERR_ARG_WRONGSTATE,"Not for factored matrix");
4058   MatCheckPreallocated(mat,1);
4059 
4060   *M = 0;
4061   if (!mat->ops->duplicate) SETERRQ(PetscObjectComm((PetscObject)mat),PETSC_ERR_SUP,"Not written for this matrix type");
4062   ierr = PetscLogEventBegin(MAT_Convert,mat,0,0,0);CHKERRQ(ierr);
4063   ierr = (*mat->ops->duplicate)(mat,op,M);CHKERRQ(ierr);
4064   B    = *M;
4065 
4066   B->stencil.dim = mat->stencil.dim;
4067   B->stencil.noc = mat->stencil.noc;
4068   for (i=0; i<=mat->stencil.dim; i++) {
4069     B->stencil.dims[i]   = mat->stencil.dims[i];
4070     B->stencil.starts[i] = mat->stencil.starts[i];
4071   }
4072 
4073   B->nooffproczerorows = mat->nooffproczerorows;
4074   B->nooffprocentries  = mat->nooffprocentries;
4075 
4076   ierr = PetscLogEventEnd(MAT_Convert,mat,0,0,0);CHKERRQ(ierr);
4077   ierr = PetscObjectStateIncrease((PetscObject)B);CHKERRQ(ierr);
4078   PetscFunctionReturn(0);
4079 }
4080 
4081 #undef __FUNCT__
4082 #define __FUNCT__ "MatGetDiagonal"
4083 /*@
4084    MatGetDiagonal - Gets the diagonal of a matrix.
4085 
4086    Logically Collective on Mat and Vec
4087 
4088    Input Parameters:
4089 +  mat - the matrix
4090 -  v - the vector for storing the diagonal
4091 
4092    Output Parameter:
4093 .  v - the diagonal of the matrix
4094 
4095    Level: intermediate
4096 
4097    Note:
4098    Currently only correct in parallel for square matrices.
4099 
4100    Concepts: matrices^accessing diagonals
4101 
4102 .seealso: MatGetRow(), MatGetSubMatrices(), MatGetSubmatrix(), MatGetRowMaxAbs()
4103 @*/
4104 PetscErrorCode  MatGetDiagonal(Mat mat,Vec v)
4105 {
4106   PetscErrorCode ierr;
4107 
4108   PetscFunctionBegin;
4109   PetscValidHeaderSpecific(mat,MAT_CLASSID,1);
4110   PetscValidType(mat,1);
4111   PetscValidHeaderSpecific(v,VEC_CLASSID,2);
4112   if (!mat->assembled) SETERRQ(PetscObjectComm((PetscObject)mat),PETSC_ERR_ARG_WRONGSTATE,"Not for unassembled matrix");
4113   if (!mat->ops->getdiagonal) SETERRQ1(PetscObjectComm((PetscObject)mat),PETSC_ERR_SUP,"Mat type %s",((PetscObject)mat)->type_name);
4114   MatCheckPreallocated(mat,1);
4115 
4116   ierr = (*mat->ops->getdiagonal)(mat,v);CHKERRQ(ierr);
4117   ierr = PetscObjectStateIncrease((PetscObject)v);CHKERRQ(ierr);
4118   PetscFunctionReturn(0);
4119 }
4120 
4121 #undef __FUNCT__
4122 #define __FUNCT__ "MatGetRowMin"
4123 /*@
4124    MatGetRowMin - Gets the minimum value (of the real part) of each
4125         row of the matrix
4126 
4127    Logically Collective on Mat and Vec
4128 
4129    Input Parameters:
4130 .  mat - the matrix
4131 
4132    Output Parameter:
4133 +  v - the vector for storing the maximums
4134 -  idx - the indices of the column found for each row (optional)
4135 
4136    Level: intermediate
4137 
4138    Notes: The result of this call are the same as if one converted the matrix to dense format
4139       and found the minimum value in each row (i.e. the implicit zeros are counted as zeros).
4140 
4141     This code is only implemented for a couple of matrix formats.
4142 
4143    Concepts: matrices^getting row maximums
4144 
4145 .seealso: MatGetDiagonal(), MatGetSubMatrices(), MatGetSubmatrix(), MatGetRowMaxAbs(),
4146           MatGetRowMax()
4147 @*/
4148 PetscErrorCode  MatGetRowMin(Mat mat,Vec v,PetscInt idx[])
4149 {
4150   PetscErrorCode ierr;
4151 
4152   PetscFunctionBegin;
4153   PetscValidHeaderSpecific(mat,MAT_CLASSID,1);
4154   PetscValidType(mat,1);
4155   PetscValidHeaderSpecific(v,VEC_CLASSID,2);
4156   if (!mat->assembled) SETERRQ(PetscObjectComm((PetscObject)mat),PETSC_ERR_ARG_WRONGSTATE,"Not for unassembled matrix");
4157   if (!mat->ops->getrowmax) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_SUP,"Mat type %s",((PetscObject)mat)->type_name);
4158   MatCheckPreallocated(mat,1);
4159 
4160   ierr = (*mat->ops->getrowmin)(mat,v,idx);CHKERRQ(ierr);
4161   ierr = PetscObjectStateIncrease((PetscObject)v);CHKERRQ(ierr);
4162   PetscFunctionReturn(0);
4163 }
4164 
4165 #undef __FUNCT__
4166 #define __FUNCT__ "MatGetRowMinAbs"
4167 /*@
4168    MatGetRowMinAbs - Gets the minimum value (in absolute value) of each
4169         row of the matrix
4170 
4171    Logically Collective on Mat and Vec
4172 
4173    Input Parameters:
4174 .  mat - the matrix
4175 
4176    Output Parameter:
4177 +  v - the vector for storing the minimums
4178 -  idx - the indices of the column found for each row (or NULL if not needed)
4179 
4180    Level: intermediate
4181 
4182    Notes: if a row is completely empty or has only 0.0 values then the idx[] value for that
4183     row is 0 (the first column).
4184 
4185     This code is only implemented for a couple of matrix formats.
4186 
4187    Concepts: matrices^getting row maximums
4188 
4189 .seealso: MatGetDiagonal(), MatGetSubMatrices(), MatGetSubmatrix(), MatGetRowMax(), MatGetRowMaxAbs(), MatGetRowMin()
4190 @*/
4191 PetscErrorCode  MatGetRowMinAbs(Mat mat,Vec v,PetscInt idx[])
4192 {
4193   PetscErrorCode ierr;
4194 
4195   PetscFunctionBegin;
4196   PetscValidHeaderSpecific(mat,MAT_CLASSID,1);
4197   PetscValidType(mat,1);
4198   PetscValidHeaderSpecific(v,VEC_CLASSID,2);
4199   if (!mat->assembled) SETERRQ(PetscObjectComm((PetscObject)mat),PETSC_ERR_ARG_WRONGSTATE,"Not for unassembled matrix");
4200   if (!mat->ops->getrowminabs) SETERRQ1(PetscObjectComm((PetscObject)mat),PETSC_ERR_SUP,"Mat type %s",((PetscObject)mat)->type_name);
4201   MatCheckPreallocated(mat,1);
4202   if (idx) {ierr = PetscMemzero(idx,mat->rmap->n*sizeof(PetscInt));CHKERRQ(ierr);}
4203 
4204   ierr = (*mat->ops->getrowminabs)(mat,v,idx);CHKERRQ(ierr);
4205   ierr = PetscObjectStateIncrease((PetscObject)v);CHKERRQ(ierr);
4206   PetscFunctionReturn(0);
4207 }
4208 
4209 #undef __FUNCT__
4210 #define __FUNCT__ "MatGetRowMax"
4211 /*@
4212    MatGetRowMax - Gets the maximum value (of the real part) of each
4213         row of the matrix
4214 
4215    Logically Collective on Mat and Vec
4216 
4217    Input Parameters:
4218 .  mat - the matrix
4219 
4220    Output Parameter:
4221 +  v - the vector for storing the maximums
4222 -  idx - the indices of the column found for each row (optional)
4223 
4224    Level: intermediate
4225 
4226    Notes: The result of this call are the same as if one converted the matrix to dense format
4227       and found the minimum value in each row (i.e. the implicit zeros are counted as zeros).
4228 
4229     This code is only implemented for a couple of matrix formats.
4230 
4231    Concepts: matrices^getting row maximums
4232 
4233 .seealso: MatGetDiagonal(), MatGetSubMatrices(), MatGetSubmatrix(), MatGetRowMaxAbs(), MatGetRowMin()
4234 @*/
4235 PetscErrorCode  MatGetRowMax(Mat mat,Vec v,PetscInt idx[])
4236 {
4237   PetscErrorCode ierr;
4238 
4239   PetscFunctionBegin;
4240   PetscValidHeaderSpecific(mat,MAT_CLASSID,1);
4241   PetscValidType(mat,1);
4242   PetscValidHeaderSpecific(v,VEC_CLASSID,2);
4243   if (!mat->assembled) SETERRQ(PetscObjectComm((PetscObject)mat),PETSC_ERR_ARG_WRONGSTATE,"Not for unassembled matrix");
4244   if (!mat->ops->getrowmax) SETERRQ1(PetscObjectComm((PetscObject)mat),PETSC_ERR_SUP,"Mat type %s",((PetscObject)mat)->type_name);
4245   MatCheckPreallocated(mat,1);
4246 
4247   ierr = (*mat->ops->getrowmax)(mat,v,idx);CHKERRQ(ierr);
4248   ierr = PetscObjectStateIncrease((PetscObject)v);CHKERRQ(ierr);
4249   PetscFunctionReturn(0);
4250 }
4251 
4252 #undef __FUNCT__
4253 #define __FUNCT__ "MatGetRowMaxAbs"
4254 /*@
4255    MatGetRowMaxAbs - Gets the maximum value (in absolute value) of each
4256         row of the matrix
4257 
4258    Logically Collective on Mat and Vec
4259 
4260    Input Parameters:
4261 .  mat - the matrix
4262 
4263    Output Parameter:
4264 +  v - the vector for storing the maximums
4265 -  idx - the indices of the column found for each row (or NULL if not needed)
4266 
4267    Level: intermediate
4268 
4269    Notes: if a row is completely empty or has only 0.0 values then the idx[] value for that
4270     row is 0 (the first column).
4271 
4272     This code is only implemented for a couple of matrix formats.
4273 
4274    Concepts: matrices^getting row maximums
4275 
4276 .seealso: MatGetDiagonal(), MatGetSubMatrices(), MatGetSubmatrix(), MatGetRowMax(), MatGetRowMin()
4277 @*/
4278 PetscErrorCode  MatGetRowMaxAbs(Mat mat,Vec v,PetscInt idx[])
4279 {
4280   PetscErrorCode ierr;
4281 
4282   PetscFunctionBegin;
4283   PetscValidHeaderSpecific(mat,MAT_CLASSID,1);
4284   PetscValidType(mat,1);
4285   PetscValidHeaderSpecific(v,VEC_CLASSID,2);
4286   if (!mat->assembled) SETERRQ(PetscObjectComm((PetscObject)mat),PETSC_ERR_ARG_WRONGSTATE,"Not for unassembled matrix");
4287   if (!mat->ops->getrowmaxabs) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_SUP,"Mat type %s",((PetscObject)mat)->type_name);
4288   MatCheckPreallocated(mat,1);
4289   if (idx) {ierr = PetscMemzero(idx,mat->rmap->n*sizeof(PetscInt));CHKERRQ(ierr);}
4290 
4291   ierr = (*mat->ops->getrowmaxabs)(mat,v,idx);CHKERRQ(ierr);
4292   ierr = PetscObjectStateIncrease((PetscObject)v);CHKERRQ(ierr);
4293   PetscFunctionReturn(0);
4294 }
4295 
4296 #undef __FUNCT__
4297 #define __FUNCT__ "MatGetRowSum"
4298 /*@
4299    MatGetRowSum - Gets the sum of each row of the matrix
4300 
4301    Logically Collective on Mat and Vec
4302 
4303    Input Parameters:
4304 .  mat - the matrix
4305 
4306    Output Parameter:
4307 .  v - the vector for storing the sum of rows
4308 
4309    Level: intermediate
4310 
4311    Notes: This code is slow since it is not currently specialized for different formats
4312 
4313    Concepts: matrices^getting row sums
4314 
4315 .seealso: MatGetDiagonal(), MatGetSubMatrices(), MatGetSubmatrix(), MatGetRowMax(), MatGetRowMin()
4316 @*/
4317 PetscErrorCode  MatGetRowSum(Mat mat, Vec v)
4318 {
4319   PetscInt       start = 0, end = 0, row;
4320   PetscScalar    *array;
4321   PetscErrorCode ierr;
4322 
4323   PetscFunctionBegin;
4324   PetscValidHeaderSpecific(mat,MAT_CLASSID,1);
4325   PetscValidType(mat,1);
4326   PetscValidHeaderSpecific(v,VEC_CLASSID,2);
4327   if (!mat->assembled) SETERRQ(PetscObjectComm((PetscObject)mat),PETSC_ERR_ARG_WRONGSTATE,"Not for unassembled matrix");
4328   MatCheckPreallocated(mat,1);
4329   ierr = MatGetOwnershipRange(mat, &start, &end);CHKERRQ(ierr);
4330   ierr = VecGetArray(v, &array);CHKERRQ(ierr);
4331   for (row = start; row < end; ++row) {
4332     PetscInt          ncols, col;
4333     const PetscInt    *cols;
4334     const PetscScalar *vals;
4335 
4336     array[row - start] = 0.0;
4337 
4338     ierr = MatGetRow(mat, row, &ncols, &cols, &vals);CHKERRQ(ierr);
4339     for (col = 0; col < ncols; col++) {
4340       array[row - start] += vals[col];
4341     }
4342     ierr = MatRestoreRow(mat, row, &ncols, &cols, &vals);CHKERRQ(ierr);
4343   }
4344   ierr = VecRestoreArray(v, &array);CHKERRQ(ierr);
4345   ierr = PetscObjectStateIncrease((PetscObject) v);CHKERRQ(ierr);
4346   PetscFunctionReturn(0);
4347 }
4348 
4349 #undef __FUNCT__
4350 #define __FUNCT__ "MatTranspose"
4351 /*@
4352    MatTranspose - Computes an in-place or out-of-place transpose of a matrix.
4353 
4354    Collective on Mat
4355 
4356    Input Parameter:
4357 +  mat - the matrix to transpose
4358 -  reuse - either MAT_INITIAL_MATRIX or MAT_REUSE_MATRIX
4359 
4360    Output Parameters:
4361 .  B - the transpose
4362 
4363    Notes:
4364      If you  pass in &mat for B the transpose will be done in place, for example MatTranspose(mat,MAT_REUSE_MATRIX,&mat);
4365 
4366      Consider using MatCreateTranspose() instead if you only need a matrix that behaves like the transpose, but don't need the storage to be changed.
4367 
4368    Level: intermediate
4369 
4370    Concepts: matrices^transposing
4371 
4372 .seealso: MatMultTranspose(), MatMultTransposeAdd(), MatIsTranspose(), MatReuse
4373 @*/
4374 PetscErrorCode  MatTranspose(Mat mat,MatReuse reuse,Mat *B)
4375 {
4376   PetscErrorCode ierr;
4377 
4378   PetscFunctionBegin;
4379   PetscValidHeaderSpecific(mat,MAT_CLASSID,1);
4380   PetscValidType(mat,1);
4381   if (!mat->assembled) SETERRQ(PetscObjectComm((PetscObject)mat),PETSC_ERR_ARG_WRONGSTATE,"Not for unassembled matrix");
4382   if (mat->factortype) SETERRQ(PetscObjectComm((PetscObject)mat),PETSC_ERR_ARG_WRONGSTATE,"Not for factored matrix");
4383   if (!mat->ops->transpose) SETERRQ1(PetscObjectComm((PetscObject)mat),PETSC_ERR_SUP,"Mat type %s",((PetscObject)mat)->type_name);
4384   MatCheckPreallocated(mat,1);
4385 
4386   ierr = PetscLogEventBegin(MAT_Transpose,mat,0,0,0);CHKERRQ(ierr);
4387   ierr = (*mat->ops->transpose)(mat,reuse,B);CHKERRQ(ierr);
4388   ierr = PetscLogEventEnd(MAT_Transpose,mat,0,0,0);CHKERRQ(ierr);
4389   if (B) {ierr = PetscObjectStateIncrease((PetscObject)*B);CHKERRQ(ierr);}
4390   PetscFunctionReturn(0);
4391 }
4392 
4393 #undef __FUNCT__
4394 #define __FUNCT__ "MatIsTranspose"
4395 /*@
4396    MatIsTranspose - Test whether a matrix is another one's transpose,
4397         or its own, in which case it tests symmetry.
4398 
4399    Collective on Mat
4400 
4401    Input Parameter:
4402 +  A - the matrix to test
4403 -  B - the matrix to test against, this can equal the first parameter
4404 
4405    Output Parameters:
4406 .  flg - the result
4407 
4408    Notes:
4409    Only available for SeqAIJ/MPIAIJ matrices. The sequential algorithm
4410    has a running time of the order of the number of nonzeros; the parallel
4411    test involves parallel copies of the block-offdiagonal parts of the matrix.
4412 
4413    Level: intermediate
4414 
4415    Concepts: matrices^transposing, matrix^symmetry
4416 
4417 .seealso: MatTranspose(), MatIsSymmetric(), MatIsHermitian()
4418 @*/
4419 PetscErrorCode  MatIsTranspose(Mat A,Mat B,PetscReal tol,PetscBool  *flg)
4420 {
4421   PetscErrorCode ierr,(*f)(Mat,Mat,PetscReal,PetscBool*),(*g)(Mat,Mat,PetscReal,PetscBool*);
4422 
4423   PetscFunctionBegin;
4424   PetscValidHeaderSpecific(A,MAT_CLASSID,1);
4425   PetscValidHeaderSpecific(B,MAT_CLASSID,2);
4426   PetscValidPointer(flg,3);
4427   ierr = PetscObjectQueryFunction((PetscObject)A,"MatIsTranspose_C",&f);CHKERRQ(ierr);
4428   ierr = PetscObjectQueryFunction((PetscObject)B,"MatIsTranspose_C",&g);CHKERRQ(ierr);
4429   *flg = PETSC_FALSE;
4430   if (f && g) {
4431     if (f == g) {
4432       ierr = (*f)(A,B,tol,flg);CHKERRQ(ierr);
4433     } else SETERRQ(PetscObjectComm((PetscObject)A),PETSC_ERR_ARG_NOTSAMETYPE,"Matrices do not have the same comparator for symmetry test");
4434   } else {
4435     MatType mattype;
4436     if (!f) {
4437       ierr = MatGetType(A,&mattype);CHKERRQ(ierr);
4438     } else {
4439       ierr = MatGetType(B,&mattype);CHKERRQ(ierr);
4440     }
4441     SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_SUP,"Matrix of type <%s> does not support checking for transpose",mattype);
4442   }
4443   PetscFunctionReturn(0);
4444 }
4445 
4446 #undef __FUNCT__
4447 #define __FUNCT__ "MatHermitianTranspose"
4448 /*@
4449    MatHermitianTranspose - Computes an in-place or out-of-place transpose of a matrix in complex conjugate.
4450 
4451    Collective on Mat
4452 
4453    Input Parameter:
4454 +  mat - the matrix to transpose and complex conjugate
4455 -  reuse - store the transpose matrix in the provided B
4456 
4457    Output Parameters:
4458 .  B - the Hermitian
4459 
4460    Notes:
4461      If you  pass in &mat for B the Hermitian will be done in place
4462 
4463    Level: intermediate
4464 
4465    Concepts: matrices^transposing, complex conjugatex
4466 
4467 .seealso: MatTranspose(), MatMultTranspose(), MatMultTransposeAdd(), MatIsTranspose(), MatReuse
4468 @*/
4469 PetscErrorCode  MatHermitianTranspose(Mat mat,MatReuse reuse,Mat *B)
4470 {
4471   PetscErrorCode ierr;
4472 
4473   PetscFunctionBegin;
4474   ierr = MatTranspose(mat,reuse,B);CHKERRQ(ierr);
4475 #if defined(PETSC_USE_COMPLEX)
4476   ierr = MatConjugate(*B);CHKERRQ(ierr);
4477 #endif
4478   PetscFunctionReturn(0);
4479 }
4480 
4481 #undef __FUNCT__
4482 #define __FUNCT__ "MatIsHermitianTranspose"
4483 /*@
4484    MatIsHermitianTranspose - Test whether a matrix is another one's Hermitian transpose,
4485 
4486    Collective on Mat
4487 
4488    Input Parameter:
4489 +  A - the matrix to test
4490 -  B - the matrix to test against, this can equal the first parameter
4491 
4492    Output Parameters:
4493 .  flg - the result
4494 
4495    Notes:
4496    Only available for SeqAIJ/MPIAIJ matrices. The sequential algorithm
4497    has a running time of the order of the number of nonzeros; the parallel
4498    test involves parallel copies of the block-offdiagonal parts of the matrix.
4499 
4500    Level: intermediate
4501 
4502    Concepts: matrices^transposing, matrix^symmetry
4503 
4504 .seealso: MatTranspose(), MatIsSymmetric(), MatIsHermitian(), MatIsTranspose()
4505 @*/
4506 PetscErrorCode  MatIsHermitianTranspose(Mat A,Mat B,PetscReal tol,PetscBool  *flg)
4507 {
4508   PetscErrorCode ierr,(*f)(Mat,Mat,PetscReal,PetscBool*),(*g)(Mat,Mat,PetscReal,PetscBool*);
4509 
4510   PetscFunctionBegin;
4511   PetscValidHeaderSpecific(A,MAT_CLASSID,1);
4512   PetscValidHeaderSpecific(B,MAT_CLASSID,2);
4513   PetscValidPointer(flg,3);
4514   ierr = PetscObjectQueryFunction((PetscObject)A,"MatIsHermitianTranspose_C",&f);CHKERRQ(ierr);
4515   ierr = PetscObjectQueryFunction((PetscObject)B,"MatIsHermitianTranspose_C",&g);CHKERRQ(ierr);
4516   if (f && g) {
4517     if (f==g) {
4518       ierr = (*f)(A,B,tol,flg);CHKERRQ(ierr);
4519     } else SETERRQ(PetscObjectComm((PetscObject)A),PETSC_ERR_ARG_NOTSAMETYPE,"Matrices do not have the same comparator for Hermitian test");
4520   }
4521   PetscFunctionReturn(0);
4522 }
4523 
4524 #undef __FUNCT__
4525 #define __FUNCT__ "MatPermute"
4526 /*@
4527    MatPermute - Creates a new matrix with rows and columns permuted from the
4528    original.
4529 
4530    Collective on Mat
4531 
4532    Input Parameters:
4533 +  mat - the matrix to permute
4534 .  row - row permutation, each processor supplies only the permutation for its rows
4535 -  col - column permutation, each processor supplies only the permutation for its columns
4536 
4537    Output Parameters:
4538 .  B - the permuted matrix
4539 
4540    Level: advanced
4541 
4542    Note:
4543    The index sets map from row/col of permuted matrix to row/col of original matrix.
4544    The index sets should be on the same communicator as Mat and have the same local sizes.
4545 
4546    Concepts: matrices^permuting
4547 
4548 .seealso: MatGetOrdering(), ISAllGather()
4549 
4550 @*/
4551 PetscErrorCode  MatPermute(Mat mat,IS row,IS col,Mat *B)
4552 {
4553   PetscErrorCode ierr;
4554 
4555   PetscFunctionBegin;
4556   PetscValidHeaderSpecific(mat,MAT_CLASSID,1);
4557   PetscValidType(mat,1);
4558   PetscValidHeaderSpecific(row,IS_CLASSID,2);
4559   PetscValidHeaderSpecific(col,IS_CLASSID,3);
4560   PetscValidPointer(B,4);
4561   if (!mat->assembled) SETERRQ(PetscObjectComm((PetscObject)mat),PETSC_ERR_ARG_WRONGSTATE,"Not for unassembled matrix");
4562   if (mat->factortype) SETERRQ(PetscObjectComm((PetscObject)mat),PETSC_ERR_ARG_WRONGSTATE,"Not for factored matrix");
4563   if (!mat->ops->permute) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_SUP,"MatPermute not available for Mat type %s",((PetscObject)mat)->type_name);
4564   MatCheckPreallocated(mat,1);
4565 
4566   ierr = (*mat->ops->permute)(mat,row,col,B);CHKERRQ(ierr);
4567   ierr = PetscObjectStateIncrease((PetscObject)*B);CHKERRQ(ierr);
4568   PetscFunctionReturn(0);
4569 }
4570 
4571 #undef __FUNCT__
4572 #define __FUNCT__ "MatEqual"
4573 /*@
4574    MatEqual - Compares two matrices.
4575 
4576    Collective on Mat
4577 
4578    Input Parameters:
4579 +  A - the first matrix
4580 -  B - the second matrix
4581 
4582    Output Parameter:
4583 .  flg - PETSC_TRUE if the matrices are equal; PETSC_FALSE otherwise.
4584 
4585    Level: intermediate
4586 
4587    Concepts: matrices^equality between
4588 @*/
4589 PetscErrorCode  MatEqual(Mat A,Mat B,PetscBool  *flg)
4590 {
4591   PetscErrorCode ierr;
4592 
4593   PetscFunctionBegin;
4594   PetscValidHeaderSpecific(A,MAT_CLASSID,1);
4595   PetscValidHeaderSpecific(B,MAT_CLASSID,2);
4596   PetscValidType(A,1);
4597   PetscValidType(B,2);
4598   PetscValidIntPointer(flg,3);
4599   PetscCheckSameComm(A,1,B,2);
4600   MatCheckPreallocated(B,2);
4601   if (!A->assembled) SETERRQ(PetscObjectComm((PetscObject)A),PETSC_ERR_ARG_WRONGSTATE,"Not for unassembled matrix");
4602   if (!B->assembled) SETERRQ(PetscObjectComm((PetscObject)A),PETSC_ERR_ARG_WRONGSTATE,"Not for unassembled matrix");
4603   if (A->rmap->N != B->rmap->N || A->cmap->N != B->cmap->N) SETERRQ4(PetscObjectComm((PetscObject)A),PETSC_ERR_ARG_SIZ,"Mat A,Mat B: global dim %D %D %D %D",A->rmap->N,B->rmap->N,A->cmap->N,B->cmap->N);
4604   if (!A->ops->equal) SETERRQ1(PetscObjectComm((PetscObject)A),PETSC_ERR_SUP,"Mat type %s",((PetscObject)A)->type_name);
4605   if (!B->ops->equal) SETERRQ1(PetscObjectComm((PetscObject)A),PETSC_ERR_SUP,"Mat type %s",((PetscObject)B)->type_name);
4606   if (A->ops->equal != B->ops->equal) SETERRQ2(PetscObjectComm((PetscObject)A),PETSC_ERR_ARG_INCOMP,"A is type: %s\nB is type: %s",((PetscObject)A)->type_name,((PetscObject)B)->type_name);
4607   MatCheckPreallocated(A,1);
4608 
4609   ierr = (*A->ops->equal)(A,B,flg);CHKERRQ(ierr);
4610   PetscFunctionReturn(0);
4611 }
4612 
4613 #undef __FUNCT__
4614 #define __FUNCT__ "MatDiagonalScale"
4615 /*@
4616    MatDiagonalScale - Scales a matrix on the left and right by diagonal
4617    matrices that are stored as vectors.  Either of the two scaling
4618    matrices can be NULL.
4619 
4620    Collective on Mat
4621 
4622    Input Parameters:
4623 +  mat - the matrix to be scaled
4624 .  l - the left scaling vector (or NULL)
4625 -  r - the right scaling vector (or NULL)
4626 
4627    Notes:
4628    MatDiagonalScale() computes A = LAR, where
4629    L = a diagonal matrix (stored as a vector), R = a diagonal matrix (stored as a vector)
4630    The L scales the rows of the matrix, the R scales the columns of the matrix.
4631 
4632    Level: intermediate
4633 
4634    Concepts: matrices^diagonal scaling
4635    Concepts: diagonal scaling of matrices
4636 
4637 .seealso: MatScale()
4638 @*/
4639 PetscErrorCode  MatDiagonalScale(Mat mat,Vec l,Vec r)
4640 {
4641   PetscErrorCode ierr;
4642 
4643   PetscFunctionBegin;
4644   PetscValidHeaderSpecific(mat,MAT_CLASSID,1);
4645   PetscValidType(mat,1);
4646   if (!mat->ops->diagonalscale) SETERRQ1(PetscObjectComm((PetscObject)mat),PETSC_ERR_SUP,"Mat type %s",((PetscObject)mat)->type_name);
4647   if (l) {PetscValidHeaderSpecific(l,VEC_CLASSID,2);PetscCheckSameComm(mat,1,l,2);}
4648   if (r) {PetscValidHeaderSpecific(r,VEC_CLASSID,3);PetscCheckSameComm(mat,1,r,3);}
4649   if (!mat->assembled) SETERRQ(PetscObjectComm((PetscObject)mat),PETSC_ERR_ARG_WRONGSTATE,"Not for unassembled matrix");
4650   if (mat->factortype) SETERRQ(PetscObjectComm((PetscObject)mat),PETSC_ERR_ARG_WRONGSTATE,"Not for factored matrix");
4651   MatCheckPreallocated(mat,1);
4652 
4653   ierr = PetscLogEventBegin(MAT_Scale,mat,0,0,0);CHKERRQ(ierr);
4654   ierr = (*mat->ops->diagonalscale)(mat,l,r);CHKERRQ(ierr);
4655   ierr = PetscLogEventEnd(MAT_Scale,mat,0,0,0);CHKERRQ(ierr);
4656   ierr = PetscObjectStateIncrease((PetscObject)mat);CHKERRQ(ierr);
4657 #if defined(PETSC_HAVE_CUSP)
4658   if (mat->valid_GPU_matrix != PETSC_CUSP_UNALLOCATED) {
4659     mat->valid_GPU_matrix = PETSC_CUSP_CPU;
4660   }
4661 #endif
4662   PetscFunctionReturn(0);
4663 }
4664 
4665 #undef __FUNCT__
4666 #define __FUNCT__ "MatScale"
4667 /*@
4668     MatScale - Scales all elements of a matrix by a given number.
4669 
4670     Logically Collective on Mat
4671 
4672     Input Parameters:
4673 +   mat - the matrix to be scaled
4674 -   a  - the scaling value
4675 
4676     Output Parameter:
4677 .   mat - the scaled matrix
4678 
4679     Level: intermediate
4680 
4681     Concepts: matrices^scaling all entries
4682 
4683 .seealso: MatDiagonalScale()
4684 @*/
4685 PetscErrorCode  MatScale(Mat mat,PetscScalar a)
4686 {
4687   PetscErrorCode ierr;
4688 
4689   PetscFunctionBegin;
4690   PetscValidHeaderSpecific(mat,MAT_CLASSID,1);
4691   PetscValidType(mat,1);
4692   if (a != (PetscScalar)1.0 && !mat->ops->scale) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_SUP,"Mat type %s",((PetscObject)mat)->type_name);
4693   if (!mat->assembled) SETERRQ(PetscObjectComm((PetscObject)mat),PETSC_ERR_ARG_WRONGSTATE,"Not for unassembled matrix");
4694   if (mat->factortype) SETERRQ(PetscObjectComm((PetscObject)mat),PETSC_ERR_ARG_WRONGSTATE,"Not for factored matrix");
4695   PetscValidLogicalCollectiveScalar(mat,a,2);
4696   MatCheckPreallocated(mat,1);
4697 
4698   ierr = PetscLogEventBegin(MAT_Scale,mat,0,0,0);CHKERRQ(ierr);
4699   if (a != (PetscScalar)1.0) {
4700     ierr = (*mat->ops->scale)(mat,a);CHKERRQ(ierr);
4701     ierr = PetscObjectStateIncrease((PetscObject)mat);CHKERRQ(ierr);
4702   }
4703   ierr = PetscLogEventEnd(MAT_Scale,mat,0,0,0);CHKERRQ(ierr);
4704 #if defined(PETSC_HAVE_CUSP)
4705   if (mat->valid_GPU_matrix != PETSC_CUSP_UNALLOCATED) {
4706     mat->valid_GPU_matrix = PETSC_CUSP_CPU;
4707   }
4708 #endif
4709   PetscFunctionReturn(0);
4710 }
4711 
4712 #undef __FUNCT__
4713 #define __FUNCT__ "MatNorm"
4714 /*@
4715    MatNorm - Calculates various norms of a matrix.
4716 
4717    Collective on Mat
4718 
4719    Input Parameters:
4720 +  mat - the matrix
4721 -  type - the type of norm, NORM_1, NORM_FROBENIUS, NORM_INFINITY
4722 
4723    Output Parameters:
4724 .  nrm - the resulting norm
4725 
4726    Level: intermediate
4727 
4728    Concepts: matrices^norm
4729    Concepts: norm^of matrix
4730 @*/
4731 PetscErrorCode  MatNorm(Mat mat,NormType type,PetscReal *nrm)
4732 {
4733   PetscErrorCode ierr;
4734 
4735   PetscFunctionBegin;
4736   PetscValidHeaderSpecific(mat,MAT_CLASSID,1);
4737   PetscValidType(mat,1);
4738   PetscValidScalarPointer(nrm,3);
4739 
4740   if (!mat->assembled) SETERRQ(PetscObjectComm((PetscObject)mat),PETSC_ERR_ARG_WRONGSTATE,"Not for unassembled matrix");
4741   if (mat->factortype) SETERRQ(PetscObjectComm((PetscObject)mat),PETSC_ERR_ARG_WRONGSTATE,"Not for factored matrix");
4742   if (!mat->ops->norm) SETERRQ1(PetscObjectComm((PetscObject)mat),PETSC_ERR_SUP,"Mat type %s",((PetscObject)mat)->type_name);
4743   MatCheckPreallocated(mat,1);
4744 
4745   ierr = (*mat->ops->norm)(mat,type,nrm);CHKERRQ(ierr);
4746   PetscFunctionReturn(0);
4747 }
4748 
4749 /*
4750      This variable is used to prevent counting of MatAssemblyBegin() that
4751    are called from within a MatAssemblyEnd().
4752 */
4753 static PetscInt MatAssemblyEnd_InUse = 0;
4754 #undef __FUNCT__
4755 #define __FUNCT__ "MatAssemblyBegin"
4756 /*@
4757    MatAssemblyBegin - Begins assembling the matrix.  This routine should
4758    be called after completing all calls to MatSetValues().
4759 
4760    Collective on Mat
4761 
4762    Input Parameters:
4763 +  mat - the matrix
4764 -  type - type of assembly, either MAT_FLUSH_ASSEMBLY or MAT_FINAL_ASSEMBLY
4765 
4766    Notes:
4767    MatSetValues() generally caches the values.  The matrix is ready to
4768    use only after MatAssemblyBegin() and MatAssemblyEnd() have been called.
4769    Use MAT_FLUSH_ASSEMBLY when switching between ADD_VALUES and INSERT_VALUES
4770    in MatSetValues(); use MAT_FINAL_ASSEMBLY for the final assembly before
4771    using the matrix.
4772 
4773    ALL processes that share a matrix MUST call MatAssemblyBegin() and MatAssemblyEnd() the SAME NUMBER of times, and each time with the
4774    same flag of MAT_FLUSH_ASSEMBLY or MAT_FINAL_ASSEMBLY for all processes. Thus you CANNOT locally change from ADD_VALUES to INSERT_VALUES, that is
4775    a global collective operation requring all processes that share the matrix.
4776 
4777    Space for preallocated nonzeros that is not filled by a call to MatSetValues() or a related routine are compressed
4778    out by assembly. If you intend to use that extra space on a subsequent assembly, be sure to insert explicit zeros
4779    before MAT_FINAL_ASSEMBLY so the space is not compressed out.
4780 
4781    Level: beginner
4782 
4783    Concepts: matrices^assembling
4784 
4785 .seealso: MatAssemblyEnd(), MatSetValues(), MatAssembled()
4786 @*/
4787 PetscErrorCode  MatAssemblyBegin(Mat mat,MatAssemblyType type)
4788 {
4789   PetscErrorCode ierr;
4790 
4791   PetscFunctionBegin;
4792   PetscValidHeaderSpecific(mat,MAT_CLASSID,1);
4793   PetscValidType(mat,1);
4794   MatCheckPreallocated(mat,1);
4795   if (mat->factortype) SETERRQ(PetscObjectComm((PetscObject)mat),PETSC_ERR_ARG_WRONGSTATE,"Not for factored matrix.\nDid you forget to call MatSetUnfactored()?");
4796   if (mat->assembled) {
4797     mat->was_assembled = PETSC_TRUE;
4798     mat->assembled     = PETSC_FALSE;
4799   }
4800   if (!MatAssemblyEnd_InUse) {
4801     ierr = PetscLogEventBegin(MAT_AssemblyBegin,mat,0,0,0);CHKERRQ(ierr);
4802     if (mat->ops->assemblybegin) {ierr = (*mat->ops->assemblybegin)(mat,type);CHKERRQ(ierr);}
4803     ierr = PetscLogEventEnd(MAT_AssemblyBegin,mat,0,0,0);CHKERRQ(ierr);
4804   } else if (mat->ops->assemblybegin) {
4805     ierr = (*mat->ops->assemblybegin)(mat,type);CHKERRQ(ierr);
4806   }
4807   PetscFunctionReturn(0);
4808 }
4809 
4810 #undef __FUNCT__
4811 #define __FUNCT__ "MatAssembled"
4812 /*@
4813    MatAssembled - Indicates if a matrix has been assembled and is ready for
4814      use; for example, in matrix-vector product.
4815 
4816    Not Collective
4817 
4818    Input Parameter:
4819 .  mat - the matrix
4820 
4821    Output Parameter:
4822 .  assembled - PETSC_TRUE or PETSC_FALSE
4823 
4824    Level: advanced
4825 
4826    Concepts: matrices^assembled?
4827 
4828 .seealso: MatAssemblyEnd(), MatSetValues(), MatAssemblyBegin()
4829 @*/
4830 PetscErrorCode  MatAssembled(Mat mat,PetscBool  *assembled)
4831 {
4832   PetscFunctionBegin;
4833   PetscValidHeaderSpecific(mat,MAT_CLASSID,1);
4834   PetscValidType(mat,1);
4835   PetscValidPointer(assembled,2);
4836   *assembled = mat->assembled;
4837   PetscFunctionReturn(0);
4838 }
4839 
4840 #undef __FUNCT__
4841 #define __FUNCT__ "MatViewFromOptions"
4842 /*
4843     Processes command line options to determine if/how a matrix
4844   is to be viewed. Called by MatAssemblyEnd() and MatLoad().
4845 */
4846 PetscErrorCode MatViewFromOptions(Mat mat,const char prefix[],const char optionname[])
4847 {
4848   PetscErrorCode    ierr;
4849   PetscViewer       viewer;
4850   PetscBool         flg;
4851   static PetscBool  incall = PETSC_FALSE;
4852   PetscViewerFormat format;
4853 
4854   PetscFunctionBegin;
4855   if (incall) PetscFunctionReturn(0);
4856   incall = PETSC_TRUE;
4857   if (prefix) {
4858     ierr   = PetscOptionsGetViewer(PetscObjectComm((PetscObject)mat),prefix,optionname,&viewer,&format,&flg);CHKERRQ(ierr);
4859   } else {
4860     ierr   = PetscOptionsGetViewer(PetscObjectComm((PetscObject)mat),((PetscObject)mat)->prefix,optionname,&viewer,&format,&flg);CHKERRQ(ierr);
4861   }
4862   if (flg) {
4863     ierr = PetscViewerPushFormat(viewer,format);CHKERRQ(ierr);
4864     ierr = MatView(mat,viewer);CHKERRQ(ierr);
4865     ierr = PetscViewerPopFormat(viewer);CHKERRQ(ierr);
4866     ierr = PetscViewerDestroy(&viewer);CHKERRQ(ierr);
4867   }
4868   incall = PETSC_FALSE;
4869   PetscFunctionReturn(0);
4870 }
4871 
4872 #undef __FUNCT__
4873 #define __FUNCT__ "MatAssemblyEnd"
4874 /*@
4875    MatAssemblyEnd - Completes assembling the matrix.  This routine should
4876    be called after MatAssemblyBegin().
4877 
4878    Collective on Mat
4879 
4880    Input Parameters:
4881 +  mat - the matrix
4882 -  type - type of assembly, either MAT_FLUSH_ASSEMBLY or MAT_FINAL_ASSEMBLY
4883 
4884    Options Database Keys:
4885 +  -mat_view ::ascii_info - Prints info on matrix at conclusion of MatEndAssembly()
4886 .  -mat_view ::ascii_info_detail - Prints more detailed info
4887 .  -mat_view - Prints matrix in ASCII format
4888 .  -mat_view ::ascii_matlab - Prints matrix in Matlab format
4889 .  -mat_view draw - PetscDraws nonzero structure of matrix, using MatView() and PetscDrawOpenX().
4890 .  -display <name> - Sets display name (default is host)
4891 .  -draw_pause <sec> - Sets number of seconds to pause after display
4892 .  -mat_view socket - Sends matrix to socket, can be accessed from Matlab (See the <a href="../../docs/manual.pdf">users manual</a>)
4893 .  -viewer_socket_machine <machine>
4894 .  -viewer_socket_port <port>
4895 .  -mat_view binary - save matrix to file in binary format
4896 -  -viewer_binary_filename <name>
4897 
4898    Notes:
4899    MatSetValues() generally caches the values.  The matrix is ready to
4900    use only after MatAssemblyBegin() and MatAssemblyEnd() have been called.
4901    Use MAT_FLUSH_ASSEMBLY when switching between ADD_VALUES and INSERT_VALUES
4902    in MatSetValues(); use MAT_FINAL_ASSEMBLY for the final assembly before
4903    using the matrix.
4904 
4905    Space for preallocated nonzeros that is not filled by a call to MatSetValues() or a related routine are compressed
4906    out by assembly. If you intend to use that extra space on a subsequent assembly, be sure to insert explicit zeros
4907    before MAT_FINAL_ASSEMBLY so the space is not compressed out.
4908 
4909    Level: beginner
4910 
4911 .seealso: MatAssemblyBegin(), MatSetValues(), PetscDrawOpenX(), PetscDrawCreate(), MatView(), MatAssembled(), PetscViewerSocketOpen()
4912 @*/
4913 PetscErrorCode  MatAssemblyEnd(Mat mat,MatAssemblyType type)
4914 {
4915   PetscErrorCode  ierr;
4916   static PetscInt inassm = 0;
4917   PetscBool       flg    = PETSC_FALSE;
4918 
4919   PetscFunctionBegin;
4920   PetscValidHeaderSpecific(mat,MAT_CLASSID,1);
4921   PetscValidType(mat,1);
4922 
4923   inassm++;
4924   MatAssemblyEnd_InUse++;
4925   if (MatAssemblyEnd_InUse == 1) { /* Do the logging only the first time through */
4926     ierr = PetscLogEventBegin(MAT_AssemblyEnd,mat,0,0,0);CHKERRQ(ierr);
4927     if (mat->ops->assemblyend) {
4928       ierr = (*mat->ops->assemblyend)(mat,type);CHKERRQ(ierr);
4929     }
4930     ierr = PetscLogEventEnd(MAT_AssemblyEnd,mat,0,0,0);CHKERRQ(ierr);
4931   } else if (mat->ops->assemblyend) {
4932     ierr = (*mat->ops->assemblyend)(mat,type);CHKERRQ(ierr);
4933   }
4934 
4935   /* Flush assembly is not a true assembly */
4936   if (type != MAT_FLUSH_ASSEMBLY) {
4937     mat->assembled = PETSC_TRUE; mat->num_ass++;
4938   }
4939   mat->insertmode = NOT_SET_VALUES;
4940   MatAssemblyEnd_InUse--;
4941   ierr = PetscObjectStateIncrease((PetscObject)mat);CHKERRQ(ierr);
4942   if (!mat->symmetric_eternal) {
4943     mat->symmetric_set              = PETSC_FALSE;
4944     mat->hermitian_set              = PETSC_FALSE;
4945     mat->structurally_symmetric_set = PETSC_FALSE;
4946   }
4947 #if defined(PETSC_HAVE_CUSP)
4948   if (mat->valid_GPU_matrix != PETSC_CUSP_UNALLOCATED) {
4949     mat->valid_GPU_matrix = PETSC_CUSP_CPU;
4950   }
4951 #endif
4952   if (inassm == 1 && type != MAT_FLUSH_ASSEMBLY) {
4953     if (mat->viewonassembly) {
4954       ierr = PetscViewerPushFormat(mat->viewonassembly,mat->viewformatonassembly);CHKERRQ(ierr);
4955       ierr = MatView(mat,mat->viewonassembly);CHKERRQ(ierr);
4956       ierr = PetscViewerPopFormat(mat->viewonassembly);CHKERRQ(ierr);
4957     }
4958 
4959     if (mat->checksymmetryonassembly) {
4960       ierr = MatIsSymmetric(mat,mat->checksymmetrytol,&flg);CHKERRQ(ierr);
4961       if (flg) {
4962         ierr = PetscPrintf(PetscObjectComm((PetscObject)mat),"Matrix is symmetric (tolerance %G)\n",mat->checksymmetrytol);CHKERRQ(ierr);
4963       } else {
4964         ierr = PetscPrintf(PetscObjectComm((PetscObject)mat),"Matrix is not symmetric (tolerance %G)\n",mat->checksymmetrytol);CHKERRQ(ierr);
4965       }
4966     }
4967     if (mat->nullsp && mat->checknullspaceonassembly) {
4968       ierr = MatNullSpaceTest(mat->nullsp,mat,NULL);CHKERRQ(ierr);
4969     }
4970   }
4971   inassm--;
4972   PetscFunctionReturn(0);
4973 }
4974 
4975 #undef __FUNCT__
4976 #define __FUNCT__ "MatSetOption"
4977 /*@
4978    MatSetOption - Sets a parameter option for a matrix. Some options
4979    may be specific to certain storage formats.  Some options
4980    determine how values will be inserted (or added). Sorted,
4981    row-oriented input will generally assemble the fastest. The default
4982    is row-oriented.
4983 
4984    Logically Collective on Mat for certain operations, such as MAT_SPD, not collective for MAT_ROW_ORIENTED, see MatOption
4985 
4986    Input Parameters:
4987 +  mat - the matrix
4988 .  option - the option, one of those listed below (and possibly others),
4989 -  flg - turn the option on (PETSC_TRUE) or off (PETSC_FALSE)
4990 
4991   Options Describing Matrix Structure:
4992 +    MAT_SPD - symmetric positive definite
4993 -    MAT_SYMMETRIC - symmetric in terms of both structure and value
4994 .    MAT_HERMITIAN - transpose is the complex conjugation
4995 .    MAT_STRUCTURALLY_SYMMETRIC - symmetric nonzero structure
4996 -    MAT_SYMMETRY_ETERNAL - if you would like the symmetry/Hermitian flag
4997                             you set to be kept with all future use of the matrix
4998                             including after MatAssemblyBegin/End() which could
4999                             potentially change the symmetry structure, i.e. you
5000                             KNOW the matrix will ALWAYS have the property you set.
5001 
5002 
5003    Options For Use with MatSetValues():
5004    Insert a logically dense subblock, which can be
5005 .    MAT_ROW_ORIENTED - row-oriented (default)
5006 
5007    Note these options reflect the data you pass in with MatSetValues(); it has
5008    nothing to do with how the data is stored internally in the matrix
5009    data structure.
5010 
5011    When (re)assembling a matrix, we can restrict the input for
5012    efficiency/debugging purposes.  These options include
5013 +    MAT_NEW_NONZERO_LOCATIONS - additional insertions will be
5014         allowed if they generate a new nonzero
5015 .    MAT_NEW_DIAGONALS - new diagonals will be allowed (for block diagonal format only)
5016 .    MAT_IGNORE_OFF_PROC_ENTRIES - drops off-processor entries
5017 .    MAT_NEW_NONZERO_LOCATION_ERR - generates an error for new matrix entry
5018 .    MAT_USE_HASH_TABLE - uses a hash table to speed up matrix assembly
5019 +    MAT_NO_OFF_PROC_ENTRIES - you know each process will only set values for its own rows, will generate an error if
5020         any process sets values for another process. This avoids all reductions in the MatAssembly routines and thus improves
5021         performance for very large process counts.
5022 
5023    Notes:
5024    Some options are relevant only for particular matrix types and
5025    are thus ignored by others.  Other options are not supported by
5026    certain matrix types and will generate an error message if set.
5027 
5028    If using a Fortran 77 module to compute a matrix, one may need to
5029    use the column-oriented option (or convert to the row-oriented
5030    format).
5031 
5032    MAT_NEW_NONZERO_LOCATIONS set to PETSC_FALSE indicates that any add or insertion
5033    that would generate a new entry in the nonzero structure is instead
5034    ignored.  Thus, if memory has not alredy been allocated for this particular
5035    data, then the insertion is ignored. For dense matrices, in which
5036    the entire array is allocated, no entries are ever ignored.
5037    Set after the first MatAssemblyEnd()
5038 
5039    MAT_NEW_NONZERO_LOCATION_ERR indicates that any add or insertion
5040    that would generate a new entry in the nonzero structure instead produces
5041    an error. (Currently supported for AIJ and BAIJ formats only.)
5042    This is a useful flag when using SAME_NONZERO_PATTERN in calling
5043    KSPSetOperators() to ensure that the nonzero pattern truely does
5044    remain unchanged. Set after the first MatAssemblyEnd()
5045 
5046    MAT_NEW_NONZERO_ALLOCATION_ERR indicates that any add or insertion
5047    that would generate a new entry that has not been preallocated will
5048    instead produce an error. (Currently supported for AIJ and BAIJ formats
5049    only.) This is a useful flag when debugging matrix memory preallocation.
5050 
5051    MAT_IGNORE_OFF_PROC_ENTRIES indicates entries destined for
5052    other processors should be dropped, rather than stashed.
5053    This is useful if you know that the "owning" processor is also
5054    always generating the correct matrix entries, so that PETSc need
5055    not transfer duplicate entries generated on another processor.
5056 
5057    MAT_USE_HASH_TABLE indicates that a hash table be used to improve the
5058    searches during matrix assembly. When this flag is set, the hash table
5059    is created during the first Matrix Assembly. This hash table is
5060    used the next time through, during MatSetVaules()/MatSetVaulesBlocked()
5061    to improve the searching of indices. MAT_NEW_NONZERO_LOCATIONS flag
5062    should be used with MAT_USE_HASH_TABLE flag. This option is currently
5063    supported by MATMPIBAIJ format only.
5064 
5065    MAT_KEEP_NONZERO_PATTERN indicates when MatZeroRows() is called the zeroed entries
5066    are kept in the nonzero structure
5067 
5068    MAT_IGNORE_ZERO_ENTRIES - for AIJ/IS matrices this will stop zero values from creating
5069    a zero location in the matrix
5070 
5071    MAT_USE_INODES - indicates using inode version of the code - works with AIJ and
5072    ROWBS matrix types
5073 
5074    MAT_NO_OFF_PROC_ZERO_ROWS - you know each process will only zero its own rows. This avoids all reductions in the
5075         zero row routines and thus improves performance for very large process counts.
5076 
5077    MAT_IGNORE_LOWER_TRIANGULAR - For SBAIJ matrices will ignore any insertions you make in the lower triangular
5078         part of the matrix (since they should match the upper triangular part).
5079 
5080    Notes: Can only be called after MatSetSizes() and MatSetType() have been set.
5081 
5082    Level: intermediate
5083 
5084    Concepts: matrices^setting options
5085 
5086 .seealso:  MatOption, Mat
5087 
5088 @*/
5089 PetscErrorCode  MatSetOption(Mat mat,MatOption op,PetscBool flg)
5090 {
5091   PetscErrorCode ierr;
5092 
5093   PetscFunctionBegin;
5094   PetscValidHeaderSpecific(mat,MAT_CLASSID,1);
5095   PetscValidType(mat,1);
5096   if (op > 0) PetscValidLogicalCollectiveEnum(mat,op,2);
5097   PetscValidLogicalCollectiveBool(mat,flg,3);
5098 
5099   if (((int) op) <= MAT_OPTION_MIN || ((int) op) >= MAT_OPTION_MAX) SETERRQ1(PetscObjectComm((PetscObject)mat),PETSC_ERR_ARG_OUTOFRANGE,"Options %d is out of range",(int)op);
5100   if (!((PetscObject)mat)->type_name) SETERRQ(PetscObjectComm((PetscObject)mat),PETSC_ERR_ARG_TYPENOTSET,"Cannot set options until type and size have been set, see MatSetType() and MatSetSizes()");
5101 
5102   switch (op) {
5103   case MAT_NO_OFF_PROC_ENTRIES:
5104     mat->nooffprocentries = flg;
5105     PetscFunctionReturn(0);
5106     break;
5107   case MAT_NO_OFF_PROC_ZERO_ROWS:
5108     mat->nooffproczerorows = flg;
5109     PetscFunctionReturn(0);
5110     break;
5111   case MAT_SPD:
5112     mat->spd_set = PETSC_TRUE;
5113     mat->spd     = flg;
5114     if (flg) {
5115       mat->symmetric                  = PETSC_TRUE;
5116       mat->structurally_symmetric     = PETSC_TRUE;
5117       mat->symmetric_set              = PETSC_TRUE;
5118       mat->structurally_symmetric_set = PETSC_TRUE;
5119     }
5120     break;
5121   case MAT_SYMMETRIC:
5122     mat->symmetric = flg;
5123     if (flg) mat->structurally_symmetric = PETSC_TRUE;
5124     mat->symmetric_set              = PETSC_TRUE;
5125     mat->structurally_symmetric_set = flg;
5126     break;
5127   case MAT_HERMITIAN:
5128     mat->hermitian = flg;
5129     if (flg) mat->structurally_symmetric = PETSC_TRUE;
5130     mat->hermitian_set              = PETSC_TRUE;
5131     mat->structurally_symmetric_set = flg;
5132     break;
5133   case MAT_STRUCTURALLY_SYMMETRIC:
5134     mat->structurally_symmetric     = flg;
5135     mat->structurally_symmetric_set = PETSC_TRUE;
5136     break;
5137   case MAT_SYMMETRY_ETERNAL:
5138     mat->symmetric_eternal = flg;
5139     break;
5140   default:
5141     break;
5142   }
5143   if (mat->ops->setoption) {
5144     ierr = (*mat->ops->setoption)(mat,op,flg);CHKERRQ(ierr);
5145   }
5146   PetscFunctionReturn(0);
5147 }
5148 
5149 #undef __FUNCT__
5150 #define __FUNCT__ "MatZeroEntries"
5151 /*@
5152    MatZeroEntries - Zeros all entries of a matrix.  For sparse matrices
5153    this routine retains the old nonzero structure.
5154 
5155    Logically Collective on Mat
5156 
5157    Input Parameters:
5158 .  mat - the matrix
5159 
5160    Level: intermediate
5161 
5162    Notes: If the matrix was not preallocated then a default, likely poor preallocation will be set in the matrix, so this should be called after the preallocation phase.
5163    See the Performance chapter of the users manual for information on preallocating matrices.
5164 
5165    Concepts: matrices^zeroing
5166 
5167 .seealso: MatZeroRows()
5168 @*/
5169 PetscErrorCode  MatZeroEntries(Mat mat)
5170 {
5171   PetscErrorCode ierr;
5172 
5173   PetscFunctionBegin;
5174   PetscValidHeaderSpecific(mat,MAT_CLASSID,1);
5175   PetscValidType(mat,1);
5176   if (mat->factortype) SETERRQ(PetscObjectComm((PetscObject)mat),PETSC_ERR_ARG_WRONGSTATE,"Not for factored matrix");
5177   if (mat->insertmode != NOT_SET_VALUES) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONGSTATE,"Not for matrices where you have set values but not yet assembled");
5178   if (!mat->ops->zeroentries) SETERRQ1(PetscObjectComm((PetscObject)mat),PETSC_ERR_SUP,"Mat type %s",((PetscObject)mat)->type_name);
5179   MatCheckPreallocated(mat,1);
5180 
5181   ierr = PetscLogEventBegin(MAT_ZeroEntries,mat,0,0,0);CHKERRQ(ierr);
5182   ierr = (*mat->ops->zeroentries)(mat);CHKERRQ(ierr);
5183   ierr = PetscLogEventEnd(MAT_ZeroEntries,mat,0,0,0);CHKERRQ(ierr);
5184   ierr = PetscObjectStateIncrease((PetscObject)mat);CHKERRQ(ierr);
5185 #if defined(PETSC_HAVE_CUSP)
5186   if (mat->valid_GPU_matrix != PETSC_CUSP_UNALLOCATED) {
5187     mat->valid_GPU_matrix = PETSC_CUSP_CPU;
5188   }
5189 #endif
5190   PetscFunctionReturn(0);
5191 }
5192 
5193 #undef __FUNCT__
5194 #define __FUNCT__ "MatZeroRowsColumns"
5195 /*@C
5196    MatZeroRowsColumns - Zeros all entries (except possibly the main diagonal)
5197    of a set of rows and columns of a matrix.
5198 
5199    Collective on Mat
5200 
5201    Input Parameters:
5202 +  mat - the matrix
5203 .  numRows - the number of rows to remove
5204 .  rows - the global row indices
5205 .  diag - value put in all diagonals of eliminated rows (0.0 will even eliminate diagonal entry)
5206 .  x - optional vector of solutions for zeroed rows (other entries in vector are not used)
5207 -  b - optional vector of right hand side, that will be adjusted by provided solution
5208 
5209    Notes:
5210    This does not change the nonzero structure of the matrix, it merely zeros those entries in the matrix.
5211 
5212    The user can set a value in the diagonal entry (or for the AIJ and
5213    row formats can optionally remove the main diagonal entry from the
5214    nonzero structure as well, by passing 0.0 as the final argument).
5215 
5216    For the parallel case, all processes that share the matrix (i.e.,
5217    those in the communicator used for matrix creation) MUST call this
5218    routine, regardless of whether any rows being zeroed are owned by
5219    them.
5220 
5221    Each processor can indicate any rows in the entire matrix to be zeroed (i.e. each process does NOT have to
5222    list only rows local to itself).
5223 
5224    The option MAT_NO_OFF_PROC_ZERO_ROWS does not apply to this routine.
5225 
5226    Level: intermediate
5227 
5228    Concepts: matrices^zeroing rows
5229 
5230 .seealso: MatZeroRowsIS(), MatZeroRowsStencil(), MatZeroEntries(), MatZeroRowsLocal(), MatSetOption(), MatZeroRowsColumnsIS()
5231 @*/
5232 PetscErrorCode  MatZeroRowsColumns(Mat mat,PetscInt numRows,const PetscInt rows[],PetscScalar diag,Vec x,Vec b)
5233 {
5234   PetscErrorCode ierr;
5235 
5236   PetscFunctionBegin;
5237   PetscValidHeaderSpecific(mat,MAT_CLASSID,1);
5238   PetscValidType(mat,1);
5239   if (numRows) PetscValidIntPointer(rows,3);
5240   if (!mat->assembled) SETERRQ(PetscObjectComm((PetscObject)mat),PETSC_ERR_ARG_WRONGSTATE,"Not for unassembled matrix");
5241   if (mat->factortype) SETERRQ(PetscObjectComm((PetscObject)mat),PETSC_ERR_ARG_WRONGSTATE,"Not for factored matrix");
5242   if (!mat->ops->zerorowscolumns) SETERRQ1(PetscObjectComm((PetscObject)mat),PETSC_ERR_SUP,"Mat type %s",((PetscObject)mat)->type_name);
5243   MatCheckPreallocated(mat,1);
5244 
5245   ierr = (*mat->ops->zerorowscolumns)(mat,numRows,rows,diag,x,b);CHKERRQ(ierr);
5246   if (mat->viewonassembly) {
5247     ierr = PetscViewerPushFormat(mat->viewonassembly,mat->viewformatonassembly);CHKERRQ(ierr);
5248     ierr = MatView(mat,mat->viewonassembly);CHKERRQ(ierr);
5249     ierr = PetscViewerPopFormat(mat->viewonassembly);CHKERRQ(ierr);
5250   }
5251   ierr = PetscObjectStateIncrease((PetscObject)mat);CHKERRQ(ierr);
5252 #if defined(PETSC_HAVE_CUSP)
5253   if (mat->valid_GPU_matrix != PETSC_CUSP_UNALLOCATED) {
5254     mat->valid_GPU_matrix = PETSC_CUSP_CPU;
5255   }
5256 #endif
5257   PetscFunctionReturn(0);
5258 }
5259 
5260 #undef __FUNCT__
5261 #define __FUNCT__ "MatZeroRowsColumnsIS"
5262 /*@C
5263    MatZeroRowsColumnsIS - Zeros all entries (except possibly the main diagonal)
5264    of a set of rows and columns of a matrix.
5265 
5266    Collective on Mat
5267 
5268    Input Parameters:
5269 +  mat - the matrix
5270 .  is - the rows to zero
5271 .  diag - value put in all diagonals of eliminated rows (0.0 will even eliminate diagonal entry)
5272 .  x - optional vector of solutions for zeroed rows (other entries in vector are not used)
5273 -  b - optional vector of right hand side, that will be adjusted by provided solution
5274 
5275    Notes:
5276    This does not change the nonzero structure of the matrix, it merely zeros those entries in the matrix.
5277 
5278    The user can set a value in the diagonal entry (or for the AIJ and
5279    row formats can optionally remove the main diagonal entry from the
5280    nonzero structure as well, by passing 0.0 as the final argument).
5281 
5282    For the parallel case, all processes that share the matrix (i.e.,
5283    those in the communicator used for matrix creation) MUST call this
5284    routine, regardless of whether any rows being zeroed are owned by
5285    them.
5286 
5287    Each processor can indicate any rows in the entire matrix to be zeroed (i.e. each process does NOT have to
5288    list only rows local to itself).
5289 
5290    The option MAT_NO_OFF_PROC_ZERO_ROWS does not apply to this routine.
5291 
5292    Level: intermediate
5293 
5294    Concepts: matrices^zeroing rows
5295 
5296 .seealso: MatZeroRowsIS(), MatZeroRowsStencil(), MatZeroEntries(), MatZeroRowsLocal(), MatSetOption(), MatZeroRowsColumns()
5297 @*/
5298 PetscErrorCode  MatZeroRowsColumnsIS(Mat mat,IS is,PetscScalar diag,Vec x,Vec b)
5299 {
5300   PetscErrorCode ierr;
5301   PetscInt       numRows;
5302   const PetscInt *rows;
5303 
5304   PetscFunctionBegin;
5305   PetscValidHeaderSpecific(mat,MAT_CLASSID,1);
5306   PetscValidHeaderSpecific(is,IS_CLASSID,2);
5307   PetscValidType(mat,1);
5308   PetscValidType(is,2);
5309   ierr = ISGetLocalSize(is,&numRows);CHKERRQ(ierr);
5310   ierr = ISGetIndices(is,&rows);CHKERRQ(ierr);
5311   ierr = MatZeroRowsColumns(mat,numRows,rows,diag,x,b);CHKERRQ(ierr);
5312   ierr = ISRestoreIndices(is,&rows);CHKERRQ(ierr);
5313   PetscFunctionReturn(0);
5314 }
5315 
5316 #undef __FUNCT__
5317 #define __FUNCT__ "MatZeroRows"
5318 /*@C
5319    MatZeroRows - Zeros all entries (except possibly the main diagonal)
5320    of a set of rows of a matrix.
5321 
5322    Collective on Mat
5323 
5324    Input Parameters:
5325 +  mat - the matrix
5326 .  numRows - the number of rows to remove
5327 .  rows - the global row indices
5328 .  diag - value put in all diagonals of eliminated rows (0.0 will even eliminate diagonal entry)
5329 .  x - optional vector of solutions for zeroed rows (other entries in vector are not used)
5330 -  b - optional vector of right hand side, that will be adjusted by provided solution
5331 
5332    Notes:
5333    For the AIJ and BAIJ matrix formats this removes the old nonzero structure,
5334    but does not release memory.  For the dense and block diagonal
5335    formats this does not alter the nonzero structure.
5336 
5337    If the option MatSetOption(mat,MAT_KEEP_NONZERO_PATTERN,PETSC_TRUE) the nonzero structure
5338    of the matrix is not changed (even for AIJ and BAIJ matrices) the values are
5339    merely zeroed.
5340 
5341    The user can set a value in the diagonal entry (or for the AIJ and
5342    row formats can optionally remove the main diagonal entry from the
5343    nonzero structure as well, by passing 0.0 as the final argument).
5344 
5345    For the parallel case, all processes that share the matrix (i.e.,
5346    those in the communicator used for matrix creation) MUST call this
5347    routine, regardless of whether any rows being zeroed are owned by
5348    them.
5349 
5350    Each processor can indicate any rows in the entire matrix to be zeroed (i.e. each process does NOT have to
5351    list only rows local to itself).
5352 
5353    You can call MatSetOption(mat,MAT_NO_OFF_PROC_ZERO_ROWS,PETSC_TRUE) if each process indicates only rows it
5354    owns that are to be zeroed. This saves a global synchronization in the implementation.
5355 
5356    Level: intermediate
5357 
5358    Concepts: matrices^zeroing rows
5359 
5360 .seealso: MatZeroRowsIS(), MatZeroRowsStencil(), MatZeroEntries(), MatZeroRowsLocal(), MatSetOption()
5361 @*/
5362 PetscErrorCode  MatZeroRows(Mat mat,PetscInt numRows,const PetscInt rows[],PetscScalar diag,Vec x,Vec b)
5363 {
5364   PetscErrorCode ierr;
5365 
5366   PetscFunctionBegin;
5367   PetscValidHeaderSpecific(mat,MAT_CLASSID,1);
5368   PetscValidType(mat,1);
5369   if (numRows) PetscValidIntPointer(rows,3);
5370   if (!mat->assembled) SETERRQ(PetscObjectComm((PetscObject)mat),PETSC_ERR_ARG_WRONGSTATE,"Not for unassembled matrix");
5371   if (mat->factortype) SETERRQ(PetscObjectComm((PetscObject)mat),PETSC_ERR_ARG_WRONGSTATE,"Not for factored matrix");
5372   if (!mat->ops->zerorows) SETERRQ1(PetscObjectComm((PetscObject)mat),PETSC_ERR_SUP,"Mat type %s",((PetscObject)mat)->type_name);
5373   MatCheckPreallocated(mat,1);
5374 
5375   ierr = (*mat->ops->zerorows)(mat,numRows,rows,diag,x,b);CHKERRQ(ierr);
5376   if (mat->viewonassembly) {
5377     ierr = PetscViewerPushFormat(mat->viewonassembly,mat->viewformatonassembly);CHKERRQ(ierr);
5378     ierr = MatView(mat,mat->viewonassembly);CHKERRQ(ierr);
5379     ierr = PetscViewerPopFormat(mat->viewonassembly);CHKERRQ(ierr);
5380   }
5381   ierr = PetscObjectStateIncrease((PetscObject)mat);CHKERRQ(ierr);
5382 #if defined(PETSC_HAVE_CUSP)
5383   if (mat->valid_GPU_matrix != PETSC_CUSP_UNALLOCATED) {
5384     mat->valid_GPU_matrix = PETSC_CUSP_CPU;
5385   }
5386 #endif
5387   PetscFunctionReturn(0);
5388 }
5389 
5390 #undef __FUNCT__
5391 #define __FUNCT__ "MatZeroRowsIS"
5392 /*@C
5393    MatZeroRowsIS - Zeros all entries (except possibly the main diagonal)
5394    of a set of rows of a matrix.
5395 
5396    Collective on Mat
5397 
5398    Input Parameters:
5399 +  mat - the matrix
5400 .  is - index set of rows to remove
5401 .  diag - value put in all diagonals of eliminated rows
5402 .  x - optional vector of solutions for zeroed rows (other entries in vector are not used)
5403 -  b - optional vector of right hand side, that will be adjusted by provided solution
5404 
5405    Notes:
5406    For the AIJ and BAIJ matrix formats this removes the old nonzero structure,
5407    but does not release memory.  For the dense and block diagonal
5408    formats this does not alter the nonzero structure.
5409 
5410    If the option MatSetOption(mat,MAT_KEEP_NONZERO_PATTERN,PETSC_TRUE) the nonzero structure
5411    of the matrix is not changed (even for AIJ and BAIJ matrices) the values are
5412    merely zeroed.
5413 
5414    The user can set a value in the diagonal entry (or for the AIJ and
5415    row formats can optionally remove the main diagonal entry from the
5416    nonzero structure as well, by passing 0.0 as the final argument).
5417 
5418    For the parallel case, all processes that share the matrix (i.e.,
5419    those in the communicator used for matrix creation) MUST call this
5420    routine, regardless of whether any rows being zeroed are owned by
5421    them.
5422 
5423    Each processor can indicate any rows in the entire matrix to be zeroed (i.e. each process does NOT have to
5424    list only rows local to itself).
5425 
5426    You can call MatSetOption(mat,MAT_NO_OFF_PROC_ZERO_ROWS,PETSC_TRUE) if each process indicates only rows it
5427    owns that are to be zeroed. This saves a global synchronization in the implementation.
5428 
5429    Level: intermediate
5430 
5431    Concepts: matrices^zeroing rows
5432 
5433 .seealso: MatZeroRows(), MatZeroRowsStencil(), MatZeroEntries(), MatZeroRowsLocal(), MatSetOption()
5434 @*/
5435 PetscErrorCode  MatZeroRowsIS(Mat mat,IS is,PetscScalar diag,Vec x,Vec b)
5436 {
5437   PetscInt       numRows;
5438   const PetscInt *rows;
5439   PetscErrorCode ierr;
5440 
5441   PetscFunctionBegin;
5442   PetscValidHeaderSpecific(mat,MAT_CLASSID,1);
5443   PetscValidType(mat,1);
5444   PetscValidHeaderSpecific(is,IS_CLASSID,2);
5445   ierr = ISGetLocalSize(is,&numRows);CHKERRQ(ierr);
5446   ierr = ISGetIndices(is,&rows);CHKERRQ(ierr);
5447   ierr = MatZeroRows(mat,numRows,rows,diag,x,b);CHKERRQ(ierr);
5448   ierr = ISRestoreIndices(is,&rows);CHKERRQ(ierr);
5449   PetscFunctionReturn(0);
5450 }
5451 
5452 #undef __FUNCT__
5453 #define __FUNCT__ "MatZeroRowsStencil"
5454 /*@C
5455    MatZeroRowsStencil - Zeros all entries (except possibly the main diagonal)
5456    of a set of rows of a matrix. These rows must be local to the process.
5457 
5458    Collective on Mat
5459 
5460    Input Parameters:
5461 +  mat - the matrix
5462 .  numRows - the number of rows to remove
5463 .  rows - the grid coordinates (and component number when dof > 1) for matrix rows
5464 .  diag - value put in all diagonals of eliminated rows (0.0 will even eliminate diagonal entry)
5465 .  x - optional vector of solutions for zeroed rows (other entries in vector are not used)
5466 -  b - optional vector of right hand side, that will be adjusted by provided solution
5467 
5468    Notes:
5469    For the AIJ and BAIJ matrix formats this removes the old nonzero structure,
5470    but does not release memory.  For the dense and block diagonal
5471    formats this does not alter the nonzero structure.
5472 
5473    If the option MatSetOption(mat,MAT_KEEP_NONZERO_PATTERN,PETSC_TRUE) the nonzero structure
5474    of the matrix is not changed (even for AIJ and BAIJ matrices) the values are
5475    merely zeroed.
5476 
5477    The user can set a value in the diagonal entry (or for the AIJ and
5478    row formats can optionally remove the main diagonal entry from the
5479    nonzero structure as well, by passing 0.0 as the final argument).
5480 
5481    For the parallel case, all processes that share the matrix (i.e.,
5482    those in the communicator used for matrix creation) MUST call this
5483    routine, regardless of whether any rows being zeroed are owned by
5484    them.
5485 
5486    Each processor can indicate any rows in the entire matrix to be zeroed (i.e. each process does NOT have to
5487    list only rows local to itself).
5488 
5489    The grid coordinates are across the entire grid, not just the local portion
5490 
5491    In Fortran idxm and idxn should be declared as
5492 $     MatStencil idxm(4,m)
5493    and the values inserted using
5494 $    idxm(MatStencil_i,1) = i
5495 $    idxm(MatStencil_j,1) = j
5496 $    idxm(MatStencil_k,1) = k
5497 $    idxm(MatStencil_c,1) = c
5498    etc
5499 
5500    For periodic boundary conditions use negative indices for values to the left (below 0; that are to be
5501    obtained by wrapping values from right edge). For values to the right of the last entry using that index plus one
5502    etc to obtain values that obtained by wrapping the values from the left edge. This does not work for anything but the
5503    DMDA_BOUNDARY_PERIODIC boundary type.
5504 
5505    For indices that don't mean anything for your case (like the k index when working in 2d) or the c index when you have
5506    a single value per point) you can skip filling those indices.
5507 
5508    Level: intermediate
5509 
5510    Concepts: matrices^zeroing rows
5511 
5512 .seealso: MatZeroRows(), MatZeroRowsIS(), MatZeroEntries(), MatZeroRowsLocal(), MatSetOption()
5513 @*/
5514 PetscErrorCode  MatZeroRowsStencil(Mat mat,PetscInt numRows,const MatStencil rows[],PetscScalar diag,Vec x,Vec b)
5515 {
5516   PetscInt       dim     = mat->stencil.dim;
5517   PetscInt       sdim    = dim - (1 - (PetscInt) mat->stencil.noc);
5518   PetscInt       *dims   = mat->stencil.dims+1;
5519   PetscInt       *starts = mat->stencil.starts;
5520   PetscInt       *dxm    = (PetscInt*) rows;
5521   PetscInt       *jdxm, i, j, tmp, numNewRows = 0;
5522   PetscErrorCode ierr;
5523 
5524   PetscFunctionBegin;
5525   PetscValidHeaderSpecific(mat,MAT_CLASSID,1);
5526   PetscValidType(mat,1);
5527   if (numRows) PetscValidIntPointer(rows,3);
5528 
5529   ierr = PetscMalloc(numRows*sizeof(PetscInt), &jdxm);CHKERRQ(ierr);
5530   for (i = 0; i < numRows; ++i) {
5531     /* Skip unused dimensions (they are ordered k, j, i, c) */
5532     for (j = 0; j < 3-sdim; ++j) dxm++;
5533     /* Local index in X dir */
5534     tmp = *dxm++ - starts[0];
5535     /* Loop over remaining dimensions */
5536     for (j = 0; j < dim-1; ++j) {
5537       /* If nonlocal, set index to be negative */
5538       if ((*dxm++ - starts[j+1]) < 0 || tmp < 0) tmp = PETSC_MIN_INT;
5539       /* Update local index */
5540       else tmp = tmp*dims[j] + *(dxm-1) - starts[j+1];
5541     }
5542     /* Skip component slot if necessary */
5543     if (mat->stencil.noc) dxm++;
5544     /* Local row number */
5545     if (tmp >= 0) {
5546       jdxm[numNewRows++] = tmp;
5547     }
5548   }
5549   ierr = MatZeroRowsLocal(mat,numNewRows,jdxm,diag,x,b);CHKERRQ(ierr);
5550   ierr = PetscFree(jdxm);CHKERRQ(ierr);
5551   PetscFunctionReturn(0);
5552 }
5553 
5554 #undef __FUNCT__
5555 #define __FUNCT__ "MatZeroRowsColumnsStencil"
5556 /*@C
5557    MatZeroRowsColumnsStencil - Zeros all row and column entries (except possibly the main diagonal)
5558    of a set of rows and columns of a matrix.
5559 
5560    Collective on Mat
5561 
5562    Input Parameters:
5563 +  mat - the matrix
5564 .  numRows - the number of rows/columns to remove
5565 .  rows - the grid coordinates (and component number when dof > 1) for matrix rows
5566 .  diag - value put in all diagonals of eliminated rows (0.0 will even eliminate diagonal entry)
5567 .  x - optional vector of solutions for zeroed rows (other entries in vector are not used)
5568 -  b - optional vector of right hand side, that will be adjusted by provided solution
5569 
5570    Notes:
5571    For the AIJ and BAIJ matrix formats this removes the old nonzero structure,
5572    but does not release memory.  For the dense and block diagonal
5573    formats this does not alter the nonzero structure.
5574 
5575    If the option MatSetOption(mat,MAT_KEEP_NONZERO_PATTERN,PETSC_TRUE) the nonzero structure
5576    of the matrix is not changed (even for AIJ and BAIJ matrices) the values are
5577    merely zeroed.
5578 
5579    The user can set a value in the diagonal entry (or for the AIJ and
5580    row formats can optionally remove the main diagonal entry from the
5581    nonzero structure as well, by passing 0.0 as the final argument).
5582 
5583    For the parallel case, all processes that share the matrix (i.e.,
5584    those in the communicator used for matrix creation) MUST call this
5585    routine, regardless of whether any rows being zeroed are owned by
5586    them.
5587 
5588    Each processor can indicate any rows in the entire matrix to be zeroed (i.e. each process does NOT have to
5589    list only rows local to itself, but the row/column numbers are given in local numbering).
5590 
5591    The grid coordinates are across the entire grid, not just the local portion
5592 
5593    In Fortran idxm and idxn should be declared as
5594 $     MatStencil idxm(4,m)
5595    and the values inserted using
5596 $    idxm(MatStencil_i,1) = i
5597 $    idxm(MatStencil_j,1) = j
5598 $    idxm(MatStencil_k,1) = k
5599 $    idxm(MatStencil_c,1) = c
5600    etc
5601 
5602    For periodic boundary conditions use negative indices for values to the left (below 0; that are to be
5603    obtained by wrapping values from right edge). For values to the right of the last entry using that index plus one
5604    etc to obtain values that obtained by wrapping the values from the left edge. This does not work for anything but the
5605    DMDA_BOUNDARY_PERIODIC boundary type.
5606 
5607    For indices that don't mean anything for your case (like the k index when working in 2d) or the c index when you have
5608    a single value per point) you can skip filling those indices.
5609 
5610    Level: intermediate
5611 
5612    Concepts: matrices^zeroing rows
5613 
5614 .seealso: MatZeroRows(), MatZeroRowsIS(), MatZeroEntries(), MatZeroRowsLocal(), MatSetOption()
5615 @*/
5616 PetscErrorCode  MatZeroRowsColumnsStencil(Mat mat,PetscInt numRows,const MatStencil rows[],PetscScalar diag,Vec x,Vec b)
5617 {
5618   PetscInt       dim     = mat->stencil.dim;
5619   PetscInt       sdim    = dim - (1 - (PetscInt) mat->stencil.noc);
5620   PetscInt       *dims   = mat->stencil.dims+1;
5621   PetscInt       *starts = mat->stencil.starts;
5622   PetscInt       *dxm    = (PetscInt*) rows;
5623   PetscInt       *jdxm, i, j, tmp, numNewRows = 0;
5624   PetscErrorCode ierr;
5625 
5626   PetscFunctionBegin;
5627   PetscValidHeaderSpecific(mat,MAT_CLASSID,1);
5628   PetscValidType(mat,1);
5629   if (numRows) PetscValidIntPointer(rows,3);
5630 
5631   ierr = PetscMalloc(numRows*sizeof(PetscInt), &jdxm);CHKERRQ(ierr);
5632   for (i = 0; i < numRows; ++i) {
5633     /* Skip unused dimensions (they are ordered k, j, i, c) */
5634     for (j = 0; j < 3-sdim; ++j) dxm++;
5635     /* Local index in X dir */
5636     tmp = *dxm++ - starts[0];
5637     /* Loop over remaining dimensions */
5638     for (j = 0; j < dim-1; ++j) {
5639       /* If nonlocal, set index to be negative */
5640       if ((*dxm++ - starts[j+1]) < 0 || tmp < 0) tmp = PETSC_MIN_INT;
5641       /* Update local index */
5642       else tmp = tmp*dims[j] + *(dxm-1) - starts[j+1];
5643     }
5644     /* Skip component slot if necessary */
5645     if (mat->stencil.noc) dxm++;
5646     /* Local row number */
5647     if (tmp >= 0) {
5648       jdxm[numNewRows++] = tmp;
5649     }
5650   }
5651   ierr = MatZeroRowsColumnsLocal(mat,numNewRows,jdxm,diag,x,b);CHKERRQ(ierr);
5652   ierr = PetscFree(jdxm);CHKERRQ(ierr);
5653   PetscFunctionReturn(0);
5654 }
5655 
5656 #undef __FUNCT__
5657 #define __FUNCT__ "MatZeroRowsLocal"
5658 /*@C
5659    MatZeroRowsLocal - Zeros all entries (except possibly the main diagonal)
5660    of a set of rows of a matrix; using local numbering of rows.
5661 
5662    Collective on Mat
5663 
5664    Input Parameters:
5665 +  mat - the matrix
5666 .  numRows - the number of rows to remove
5667 .  rows - the global row indices
5668 .  diag - value put in all diagonals of eliminated rows
5669 .  x - optional vector of solutions for zeroed rows (other entries in vector are not used)
5670 -  b - optional vector of right hand side, that will be adjusted by provided solution
5671 
5672    Notes:
5673    Before calling MatZeroRowsLocal(), the user must first set the
5674    local-to-global mapping by calling MatSetLocalToGlobalMapping().
5675 
5676    For the AIJ matrix formats this removes the old nonzero structure,
5677    but does not release memory.  For the dense and block diagonal
5678    formats this does not alter the nonzero structure.
5679 
5680    If the option MatSetOption(mat,MAT_KEEP_NONZERO_PATTERN,PETSC_TRUE) the nonzero structure
5681    of the matrix is not changed (even for AIJ and BAIJ matrices) the values are
5682    merely zeroed.
5683 
5684    The user can set a value in the diagonal entry (or for the AIJ and
5685    row formats can optionally remove the main diagonal entry from the
5686    nonzero structure as well, by passing 0.0 as the final argument).
5687 
5688    You can call MatSetOption(mat,MAT_NO_OFF_PROC_ZERO_ROWS,PETSC_TRUE) if each process indicates only rows it
5689    owns that are to be zeroed. This saves a global synchronization in the implementation.
5690 
5691    Level: intermediate
5692 
5693    Concepts: matrices^zeroing
5694 
5695 .seealso: MatZeroRows(), MatZeroRowsLocalIS(), MatZeroEntries(), MatZeroRows(), MatSetLocalToGlobalMapping
5696 @*/
5697 PetscErrorCode  MatZeroRowsLocal(Mat mat,PetscInt numRows,const PetscInt rows[],PetscScalar diag,Vec x,Vec b)
5698 {
5699   PetscErrorCode ierr;
5700   PetscMPIInt    size;
5701 
5702   PetscFunctionBegin;
5703   PetscValidHeaderSpecific(mat,MAT_CLASSID,1);
5704   PetscValidType(mat,1);
5705   if (numRows) PetscValidIntPointer(rows,3);
5706   if (!mat->assembled) SETERRQ(PetscObjectComm((PetscObject)mat),PETSC_ERR_ARG_WRONGSTATE,"Not for unassembled matrix");
5707   if (mat->factortype) SETERRQ(PetscObjectComm((PetscObject)mat),PETSC_ERR_ARG_WRONGSTATE,"Not for factored matrix");
5708   MatCheckPreallocated(mat,1);
5709 
5710   ierr = MPI_Comm_size(PetscObjectComm((PetscObject)mat),&size);CHKERRQ(ierr);
5711   if (mat->ops->zerorowslocal) {
5712     ierr = (*mat->ops->zerorowslocal)(mat,numRows,rows,diag,x,b);CHKERRQ(ierr);
5713   } else if (size == 1) {
5714     ierr = (*mat->ops->zerorows)(mat,numRows,rows,diag,x,b);CHKERRQ(ierr);
5715   } else {
5716     IS             is, newis;
5717     const PetscInt *newRows;
5718 
5719     if (!mat->rmap->mapping) SETERRQ(PetscObjectComm((PetscObject)mat),PETSC_ERR_ARG_WRONGSTATE,"Need to provide local to global mapping to matrix first");
5720     ierr = ISCreateGeneral(PETSC_COMM_SELF,numRows,rows,PETSC_COPY_VALUES,&is);CHKERRQ(ierr);
5721     ierr = ISLocalToGlobalMappingApplyIS(mat->rmap->mapping,is,&newis);CHKERRQ(ierr);
5722     ierr = ISGetIndices(newis,&newRows);CHKERRQ(ierr);
5723     ierr = (*mat->ops->zerorows)(mat,numRows,newRows,diag,x,b);CHKERRQ(ierr);
5724     ierr = ISRestoreIndices(newis,&newRows);CHKERRQ(ierr);
5725     ierr = ISDestroy(&newis);CHKERRQ(ierr);
5726     ierr = ISDestroy(&is);CHKERRQ(ierr);
5727   }
5728   ierr = PetscObjectStateIncrease((PetscObject)mat);CHKERRQ(ierr);
5729 #if defined(PETSC_HAVE_CUSP)
5730   if (mat->valid_GPU_matrix != PETSC_CUSP_UNALLOCATED) {
5731     mat->valid_GPU_matrix = PETSC_CUSP_CPU;
5732   }
5733 #endif
5734   PetscFunctionReturn(0);
5735 }
5736 
5737 #undef __FUNCT__
5738 #define __FUNCT__ "MatZeroRowsLocalIS"
5739 /*@C
5740    MatZeroRowsLocalIS - Zeros all entries (except possibly the main diagonal)
5741    of a set of rows of a matrix; using local numbering of rows.
5742 
5743    Collective on Mat
5744 
5745    Input Parameters:
5746 +  mat - the matrix
5747 .  is - index set of rows to remove
5748 .  diag - value put in all diagonals of eliminated rows
5749 .  x - optional vector of solutions for zeroed rows (other entries in vector are not used)
5750 -  b - optional vector of right hand side, that will be adjusted by provided solution
5751 
5752    Notes:
5753    Before calling MatZeroRowsLocalIS(), the user must first set the
5754    local-to-global mapping by calling MatSetLocalToGlobalMapping().
5755 
5756    For the AIJ matrix formats this removes the old nonzero structure,
5757    but does not release memory.  For the dense and block diagonal
5758    formats this does not alter the nonzero structure.
5759 
5760    If the option MatSetOption(mat,MAT_KEEP_NONZERO_PATTERN,PETSC_TRUE) the nonzero structure
5761    of the matrix is not changed (even for AIJ and BAIJ matrices) the values are
5762    merely zeroed.
5763 
5764    The user can set a value in the diagonal entry (or for the AIJ and
5765    row formats can optionally remove the main diagonal entry from the
5766    nonzero structure as well, by passing 0.0 as the final argument).
5767 
5768    You can call MatSetOption(mat,MAT_NO_OFF_PROC_ZERO_ROWS,PETSC_TRUE) if each process indicates only rows it
5769    owns that are to be zeroed. This saves a global synchronization in the implementation.
5770 
5771    Level: intermediate
5772 
5773    Concepts: matrices^zeroing
5774 
5775 .seealso: MatZeroRows(), MatZeroRowsLocal(), MatZeroEntries(), MatZeroRows(), MatSetLocalToGlobalMapping
5776 @*/
5777 PetscErrorCode  MatZeroRowsLocalIS(Mat mat,IS is,PetscScalar diag,Vec x,Vec b)
5778 {
5779   PetscErrorCode ierr;
5780   PetscInt       numRows;
5781   const PetscInt *rows;
5782 
5783   PetscFunctionBegin;
5784   PetscValidHeaderSpecific(mat,MAT_CLASSID,1);
5785   PetscValidType(mat,1);
5786   PetscValidHeaderSpecific(is,IS_CLASSID,2);
5787   if (!mat->assembled) SETERRQ(PetscObjectComm((PetscObject)mat),PETSC_ERR_ARG_WRONGSTATE,"Not for unassembled matrix");
5788   if (mat->factortype) SETERRQ(PetscObjectComm((PetscObject)mat),PETSC_ERR_ARG_WRONGSTATE,"Not for factored matrix");
5789   MatCheckPreallocated(mat,1);
5790 
5791   ierr = ISGetLocalSize(is,&numRows);CHKERRQ(ierr);
5792   ierr = ISGetIndices(is,&rows);CHKERRQ(ierr);
5793   ierr = MatZeroRowsLocal(mat,numRows,rows,diag,x,b);CHKERRQ(ierr);
5794   ierr = ISRestoreIndices(is,&rows);CHKERRQ(ierr);
5795   PetscFunctionReturn(0);
5796 }
5797 
5798 #undef __FUNCT__
5799 #define __FUNCT__ "MatZeroRowsColumnsLocal"
5800 /*@C
5801    MatZeroRowsColumnsLocal - Zeros all entries (except possibly the main diagonal)
5802    of a set of rows and columns of a matrix; using local numbering of rows.
5803 
5804    Collective on Mat
5805 
5806    Input Parameters:
5807 +  mat - the matrix
5808 .  numRows - the number of rows to remove
5809 .  rows - the global row indices
5810 .  diag - value put in all diagonals of eliminated rows
5811 .  x - optional vector of solutions for zeroed rows (other entries in vector are not used)
5812 -  b - optional vector of right hand side, that will be adjusted by provided solution
5813 
5814    Notes:
5815    Before calling MatZeroRowsColumnsLocal(), the user must first set the
5816    local-to-global mapping by calling MatSetLocalToGlobalMapping().
5817 
5818    The user can set a value in the diagonal entry (or for the AIJ and
5819    row formats can optionally remove the main diagonal entry from the
5820    nonzero structure as well, by passing 0.0 as the final argument).
5821 
5822    Level: intermediate
5823 
5824    Concepts: matrices^zeroing
5825 
5826 .seealso: MatZeroRows(), MatZeroRowsLocalIS(), MatZeroEntries(), MatZeroRows(), MatSetLocalToGlobalMapping
5827 @*/
5828 PetscErrorCode  MatZeroRowsColumnsLocal(Mat mat,PetscInt numRows,const PetscInt rows[],PetscScalar diag,Vec x,Vec b)
5829 {
5830   PetscErrorCode ierr;
5831   PetscMPIInt    size;
5832 
5833   PetscFunctionBegin;
5834   PetscValidHeaderSpecific(mat,MAT_CLASSID,1);
5835   PetscValidType(mat,1);
5836   if (numRows) PetscValidIntPointer(rows,3);
5837   if (!mat->assembled) SETERRQ(PetscObjectComm((PetscObject)mat),PETSC_ERR_ARG_WRONGSTATE,"Not for unassembled matrix");
5838   if (mat->factortype) SETERRQ(PetscObjectComm((PetscObject)mat),PETSC_ERR_ARG_WRONGSTATE,"Not for factored matrix");
5839   MatCheckPreallocated(mat,1);
5840 
5841   ierr = MPI_Comm_size(PetscObjectComm((PetscObject)mat),&size);CHKERRQ(ierr);
5842   if (size == 1) {
5843     ierr = (*mat->ops->zerorowscolumns)(mat,numRows,rows,diag,x,b);CHKERRQ(ierr);
5844   } else {
5845     IS             is, newis;
5846     const PetscInt *newRows;
5847 
5848     if (!mat->cmap->mapping) SETERRQ(PetscObjectComm((PetscObject)mat),PETSC_ERR_ARG_WRONGSTATE,"Need to provide local to global mapping to matrix first");
5849     ierr = ISCreateGeneral(PETSC_COMM_SELF,numRows,rows,PETSC_COPY_VALUES,&is);CHKERRQ(ierr);
5850     ierr = ISLocalToGlobalMappingApplyIS(mat->cmap->mapping,is,&newis);CHKERRQ(ierr);
5851     ierr = ISGetIndices(newis,&newRows);CHKERRQ(ierr);
5852     ierr = (*mat->ops->zerorowscolumns)(mat,numRows,newRows,diag,x,b);CHKERRQ(ierr);
5853     ierr = ISRestoreIndices(newis,&newRows);CHKERRQ(ierr);
5854     ierr = ISDestroy(&newis);CHKERRQ(ierr);
5855     ierr = ISDestroy(&is);CHKERRQ(ierr);
5856   }
5857   ierr = PetscObjectStateIncrease((PetscObject)mat);CHKERRQ(ierr);
5858 #if defined(PETSC_HAVE_CUSP)
5859   if (mat->valid_GPU_matrix != PETSC_CUSP_UNALLOCATED) {
5860     mat->valid_GPU_matrix = PETSC_CUSP_CPU;
5861   }
5862 #endif
5863   PetscFunctionReturn(0);
5864 }
5865 
5866 #undef __FUNCT__
5867 #define __FUNCT__ "MatZeroRowsColumnsLocalIS"
5868 /*@C
5869    MatZeroRowsColumnsLocalIS - Zeros all entries (except possibly the main diagonal)
5870    of a set of rows and columns of a matrix; using local numbering of rows.
5871 
5872    Collective on Mat
5873 
5874    Input Parameters:
5875 +  mat - the matrix
5876 .  is - index set of rows to remove
5877 .  diag - value put in all diagonals of eliminated rows
5878 .  x - optional vector of solutions for zeroed rows (other entries in vector are not used)
5879 -  b - optional vector of right hand side, that will be adjusted by provided solution
5880 
5881    Notes:
5882    Before calling MatZeroRowsColumnsLocalIS(), the user must first set the
5883    local-to-global mapping by calling MatSetLocalToGlobalMapping().
5884 
5885    The user can set a value in the diagonal entry (or for the AIJ and
5886    row formats can optionally remove the main diagonal entry from the
5887    nonzero structure as well, by passing 0.0 as the final argument).
5888 
5889    Level: intermediate
5890 
5891    Concepts: matrices^zeroing
5892 
5893 .seealso: MatZeroRows(), MatZeroRowsLocal(), MatZeroEntries(), MatZeroRows(), MatSetLocalToGlobalMapping
5894 @*/
5895 PetscErrorCode  MatZeroRowsColumnsLocalIS(Mat mat,IS is,PetscScalar diag,Vec x,Vec b)
5896 {
5897   PetscErrorCode ierr;
5898   PetscInt       numRows;
5899   const PetscInt *rows;
5900 
5901   PetscFunctionBegin;
5902   PetscValidHeaderSpecific(mat,MAT_CLASSID,1);
5903   PetscValidType(mat,1);
5904   PetscValidHeaderSpecific(is,IS_CLASSID,2);
5905   if (!mat->assembled) SETERRQ(PetscObjectComm((PetscObject)mat),PETSC_ERR_ARG_WRONGSTATE,"Not for unassembled matrix");
5906   if (mat->factortype) SETERRQ(PetscObjectComm((PetscObject)mat),PETSC_ERR_ARG_WRONGSTATE,"Not for factored matrix");
5907   MatCheckPreallocated(mat,1);
5908 
5909   ierr = ISGetLocalSize(is,&numRows);CHKERRQ(ierr);
5910   ierr = ISGetIndices(is,&rows);CHKERRQ(ierr);
5911   ierr = MatZeroRowsColumnsLocal(mat,numRows,rows,diag,x,b);CHKERRQ(ierr);
5912   ierr = ISRestoreIndices(is,&rows);CHKERRQ(ierr);
5913   PetscFunctionReturn(0);
5914 }
5915 
5916 #undef __FUNCT__
5917 #define __FUNCT__ "MatGetSize"
5918 /*@
5919    MatGetSize - Returns the numbers of rows and columns in a matrix.
5920 
5921    Not Collective
5922 
5923    Input Parameter:
5924 .  mat - the matrix
5925 
5926    Output Parameters:
5927 +  m - the number of global rows
5928 -  n - the number of global columns
5929 
5930    Note: both output parameters can be NULL on input.
5931 
5932    Level: beginner
5933 
5934    Concepts: matrices^size
5935 
5936 .seealso: MatGetLocalSize()
5937 @*/
5938 PetscErrorCode  MatGetSize(Mat mat,PetscInt *m,PetscInt *n)
5939 {
5940   PetscFunctionBegin;
5941   PetscValidHeaderSpecific(mat,MAT_CLASSID,1);
5942   if (m) *m = mat->rmap->N;
5943   if (n) *n = mat->cmap->N;
5944   PetscFunctionReturn(0);
5945 }
5946 
5947 #undef __FUNCT__
5948 #define __FUNCT__ "MatGetLocalSize"
5949 /*@
5950    MatGetLocalSize - Returns the number of rows and columns in a matrix
5951    stored locally.  This information may be implementation dependent, so
5952    use with care.
5953 
5954    Not Collective
5955 
5956    Input Parameters:
5957 .  mat - the matrix
5958 
5959    Output Parameters:
5960 +  m - the number of local rows
5961 -  n - the number of local columns
5962 
5963    Note: both output parameters can be NULL on input.
5964 
5965    Level: beginner
5966 
5967    Concepts: matrices^local size
5968 
5969 .seealso: MatGetSize()
5970 @*/
5971 PetscErrorCode  MatGetLocalSize(Mat mat,PetscInt *m,PetscInt *n)
5972 {
5973   PetscFunctionBegin;
5974   PetscValidHeaderSpecific(mat,MAT_CLASSID,1);
5975   if (m) PetscValidIntPointer(m,2);
5976   if (n) PetscValidIntPointer(n,3);
5977   if (m) *m = mat->rmap->n;
5978   if (n) *n = mat->cmap->n;
5979   PetscFunctionReturn(0);
5980 }
5981 
5982 #undef __FUNCT__
5983 #define __FUNCT__ "MatGetOwnershipRangeColumn"
5984 /*@
5985    MatGetOwnershipRangeColumn - Returns the range of matrix columns associated with rows of a vector one multiplies by that owned by
5986    this processor. (The columns of the "diagonal block")
5987 
5988    Not Collective, unless matrix has not been allocated, then collective on Mat
5989 
5990    Input Parameters:
5991 .  mat - the matrix
5992 
5993    Output Parameters:
5994 +  m - the global index of the first local column
5995 -  n - one more than the global index of the last local column
5996 
5997    Notes: both output parameters can be NULL on input.
5998 
5999    Level: developer
6000 
6001    Concepts: matrices^column ownership
6002 
6003 .seealso:  MatGetOwnershipRange(), MatGetOwnershipRanges(), MatGetOwnershipRangesColumn()
6004 
6005 @*/
6006 PetscErrorCode  MatGetOwnershipRangeColumn(Mat mat,PetscInt *m,PetscInt *n)
6007 {
6008   PetscFunctionBegin;
6009   PetscValidHeaderSpecific(mat,MAT_CLASSID,1);
6010   PetscValidType(mat,1);
6011   if (m) PetscValidIntPointer(m,2);
6012   if (n) PetscValidIntPointer(n,3);
6013   MatCheckPreallocated(mat,1);
6014   if (m) *m = mat->cmap->rstart;
6015   if (n) *n = mat->cmap->rend;
6016   PetscFunctionReturn(0);
6017 }
6018 
6019 #undef __FUNCT__
6020 #define __FUNCT__ "MatGetOwnershipRange"
6021 /*@
6022    MatGetOwnershipRange - Returns the range of matrix rows owned by
6023    this processor, assuming that the matrix is laid out with the first
6024    n1 rows on the first processor, the next n2 rows on the second, etc.
6025    For certain parallel layouts this range may not be well defined.
6026 
6027    Not Collective
6028 
6029    Input Parameters:
6030 .  mat - the matrix
6031 
6032    Output Parameters:
6033 +  m - the global index of the first local row
6034 -  n - one more than the global index of the last local row
6035 
6036    Note: Both output parameters can be NULL on input.
6037 $  This function requires that the matrix be preallocated. If you have not preallocated, consider using
6038 $    PetscSplitOwnership(MPI_Comm comm, PetscInt *n, PetscInt *N)
6039 $  and then MPI_Scan() to calculate prefix sums of the local sizes.
6040 
6041    Level: beginner
6042 
6043    Concepts: matrices^row ownership
6044 
6045 .seealso:   MatGetOwnershipRanges(), MatGetOwnershipRangeColumn(), MatGetOwnershipRangesColumn(), PetscSplitOwnership(), PetscSplitOwnershipBlock()
6046 
6047 @*/
6048 PetscErrorCode  MatGetOwnershipRange(Mat mat,PetscInt *m,PetscInt *n)
6049 {
6050   PetscFunctionBegin;
6051   PetscValidHeaderSpecific(mat,MAT_CLASSID,1);
6052   PetscValidType(mat,1);
6053   if (m) PetscValidIntPointer(m,2);
6054   if (n) PetscValidIntPointer(n,3);
6055   MatCheckPreallocated(mat,1);
6056   if (m) *m = mat->rmap->rstart;
6057   if (n) *n = mat->rmap->rend;
6058   PetscFunctionReturn(0);
6059 }
6060 
6061 #undef __FUNCT__
6062 #define __FUNCT__ "MatGetOwnershipRanges"
6063 /*@C
6064    MatGetOwnershipRanges - Returns the range of matrix rows owned by
6065    each process
6066 
6067    Not Collective, unless matrix has not been allocated, then collective on Mat
6068 
6069    Input Parameters:
6070 .  mat - the matrix
6071 
6072    Output Parameters:
6073 .  ranges - start of each processors portion plus one more then the total length at the end
6074 
6075    Level: beginner
6076 
6077    Concepts: matrices^row ownership
6078 
6079 .seealso:   MatGetOwnershipRange(), MatGetOwnershipRangeColumn(), MatGetOwnershipRangesColumn()
6080 
6081 @*/
6082 PetscErrorCode  MatGetOwnershipRanges(Mat mat,const PetscInt **ranges)
6083 {
6084   PetscErrorCode ierr;
6085 
6086   PetscFunctionBegin;
6087   PetscValidHeaderSpecific(mat,MAT_CLASSID,1);
6088   PetscValidType(mat,1);
6089   MatCheckPreallocated(mat,1);
6090   ierr = PetscLayoutGetRanges(mat->rmap,ranges);CHKERRQ(ierr);
6091   PetscFunctionReturn(0);
6092 }
6093 
6094 #undef __FUNCT__
6095 #define __FUNCT__ "MatGetOwnershipRangesColumn"
6096 /*@C
6097    MatGetOwnershipRangesColumn - Returns the range of matrix columns associated with rows of a vector one multiplies by that owned by
6098    this processor. (The columns of the "diagonal blocks" for each process)
6099 
6100    Not Collective, unless matrix has not been allocated, then collective on Mat
6101 
6102    Input Parameters:
6103 .  mat - the matrix
6104 
6105    Output Parameters:
6106 .  ranges - start of each processors portion plus one more then the total length at the end
6107 
6108    Level: beginner
6109 
6110    Concepts: matrices^column ownership
6111 
6112 .seealso:   MatGetOwnershipRange(), MatGetOwnershipRangeColumn(), MatGetOwnershipRanges()
6113 
6114 @*/
6115 PetscErrorCode  MatGetOwnershipRangesColumn(Mat mat,const PetscInt **ranges)
6116 {
6117   PetscErrorCode ierr;
6118 
6119   PetscFunctionBegin;
6120   PetscValidHeaderSpecific(mat,MAT_CLASSID,1);
6121   PetscValidType(mat,1);
6122   MatCheckPreallocated(mat,1);
6123   ierr = PetscLayoutGetRanges(mat->cmap,ranges);CHKERRQ(ierr);
6124   PetscFunctionReturn(0);
6125 }
6126 
6127 #undef __FUNCT__
6128 #define __FUNCT__ "MatGetOwnershipIS"
6129 /*@C
6130    MatGetOwnershipIS - Get row and column ownership as index sets
6131 
6132    Not Collective
6133 
6134    Input Arguments:
6135 .  A - matrix of type Elemental
6136 
6137    Output Arguments:
6138 +  rows - rows in which this process owns elements
6139 .  cols - columns in which this process owns elements
6140 
6141    Level: intermediate
6142 
6143 .seealso: MatGetOwnershipRange(), MatGetOwnershipRangeColumn(), MatSetValues(), MATELEMENTAL, MatSetValues()
6144 @*/
6145 PetscErrorCode MatGetOwnershipIS(Mat A,IS *rows,IS *cols)
6146 {
6147   PetscErrorCode ierr,(*f)(Mat,IS*,IS*);
6148 
6149   PetscFunctionBegin;
6150   MatCheckPreallocated(A,1);
6151   ierr = PetscObjectQueryFunction((PetscObject)A,"MatGetOwnershipIS_C",&f);CHKERRQ(ierr);
6152   if (f) {
6153     ierr = (*f)(A,rows,cols);CHKERRQ(ierr);
6154   } else {   /* Create a standard row-based partition, each process is responsible for ALL columns in their row block */
6155     if (rows) {ierr = ISCreateStride(PETSC_COMM_SELF,A->rmap->n,A->rmap->rstart,1,rows);CHKERRQ(ierr);}
6156     if (cols) {ierr = ISCreateStride(PETSC_COMM_SELF,A->cmap->N,0,1,cols);CHKERRQ(ierr);}
6157   }
6158   PetscFunctionReturn(0);
6159 }
6160 
6161 #undef __FUNCT__
6162 #define __FUNCT__ "MatILUFactorSymbolic"
6163 /*@C
6164    MatILUFactorSymbolic - Performs symbolic ILU factorization of a matrix.
6165    Uses levels of fill only, not drop tolerance. Use MatLUFactorNumeric()
6166    to complete the factorization.
6167 
6168    Collective on Mat
6169 
6170    Input Parameters:
6171 +  mat - the matrix
6172 .  row - row permutation
6173 .  column - column permutation
6174 -  info - structure containing
6175 $      levels - number of levels of fill.
6176 $      expected fill - as ratio of original fill.
6177 $      1 or 0 - indicating force fill on diagonal (improves robustness for matrices
6178                 missing diagonal entries)
6179 
6180    Output Parameters:
6181 .  fact - new matrix that has been symbolically factored
6182 
6183    Notes:
6184    See the <a href="../../docs/manual.pdf">users manual</a>  for additional information about
6185    choosing the fill factor for better efficiency.
6186 
6187    Most users should employ the simplified KSP interface for linear solvers
6188    instead of working directly with matrix algebra routines such as this.
6189    See, e.g., KSPCreate().
6190 
6191    Level: developer
6192 
6193   Concepts: matrices^symbolic LU factorization
6194   Concepts: matrices^factorization
6195   Concepts: LU^symbolic factorization
6196 
6197 .seealso: MatLUFactorSymbolic(), MatLUFactorNumeric(), MatCholeskyFactor()
6198           MatGetOrdering(), MatFactorInfo
6199 
6200     Developer Note: fortran interface is not autogenerated as the f90
6201     interface defintion cannot be generated correctly [due to MatFactorInfo]
6202 
6203 @*/
6204 PetscErrorCode  MatILUFactorSymbolic(Mat fact,Mat mat,IS row,IS col,const MatFactorInfo *info)
6205 {
6206   PetscErrorCode ierr;
6207 
6208   PetscFunctionBegin;
6209   PetscValidHeaderSpecific(mat,MAT_CLASSID,1);
6210   PetscValidType(mat,1);
6211   PetscValidHeaderSpecific(row,IS_CLASSID,2);
6212   PetscValidHeaderSpecific(col,IS_CLASSID,3);
6213   PetscValidPointer(info,4);
6214   PetscValidPointer(fact,5);
6215   if (info->levels < 0) SETERRQ1(PetscObjectComm((PetscObject)mat),PETSC_ERR_ARG_OUTOFRANGE,"Levels of fill negative %D",(PetscInt)info->levels);
6216   if (info->fill < 1.0) SETERRQ1(PetscObjectComm((PetscObject)mat),PETSC_ERR_ARG_OUTOFRANGE,"Expected fill less than 1.0 %G",info->fill);
6217   if (!(fact)->ops->ilufactorsymbolic) {
6218     const MatSolverPackage spackage;
6219     ierr = MatFactorGetSolverPackage(fact,&spackage);CHKERRQ(ierr);
6220     SETERRQ2(PetscObjectComm((PetscObject)mat),PETSC_ERR_SUP,"Matrix type %s symbolic ILU using solver package %s",((PetscObject)mat)->type_name,spackage);
6221   }
6222   if (!mat->assembled) SETERRQ(PetscObjectComm((PetscObject)mat),PETSC_ERR_ARG_WRONGSTATE,"Not for unassembled matrix");
6223   if (mat->factortype) SETERRQ(PetscObjectComm((PetscObject)mat),PETSC_ERR_ARG_WRONGSTATE,"Not for factored matrix");
6224   MatCheckPreallocated(mat,2);
6225 
6226   ierr = PetscLogEventBegin(MAT_ILUFactorSymbolic,mat,row,col,0);CHKERRQ(ierr);
6227   ierr = (fact->ops->ilufactorsymbolic)(fact,mat,row,col,info);CHKERRQ(ierr);
6228   ierr = PetscLogEventEnd(MAT_ILUFactorSymbolic,mat,row,col,0);CHKERRQ(ierr);
6229   PetscFunctionReturn(0);
6230 }
6231 
6232 #undef __FUNCT__
6233 #define __FUNCT__ "MatICCFactorSymbolic"
6234 /*@C
6235    MatICCFactorSymbolic - Performs symbolic incomplete
6236    Cholesky factorization for a symmetric matrix.  Use
6237    MatCholeskyFactorNumeric() to complete the factorization.
6238 
6239    Collective on Mat
6240 
6241    Input Parameters:
6242 +  mat - the matrix
6243 .  perm - row and column permutation
6244 -  info - structure containing
6245 $      levels - number of levels of fill.
6246 $      expected fill - as ratio of original fill.
6247 
6248    Output Parameter:
6249 .  fact - the factored matrix
6250 
6251    Notes:
6252    Most users should employ the KSP interface for linear solvers
6253    instead of working directly with matrix algebra routines such as this.
6254    See, e.g., KSPCreate().
6255 
6256    Level: developer
6257 
6258   Concepts: matrices^symbolic incomplete Cholesky factorization
6259   Concepts: matrices^factorization
6260   Concepts: Cholsky^symbolic factorization
6261 
6262 .seealso: MatCholeskyFactorNumeric(), MatCholeskyFactor(), MatFactorInfo
6263 
6264     Developer Note: fortran interface is not autogenerated as the f90
6265     interface defintion cannot be generated correctly [due to MatFactorInfo]
6266 
6267 @*/
6268 PetscErrorCode  MatICCFactorSymbolic(Mat fact,Mat mat,IS perm,const MatFactorInfo *info)
6269 {
6270   PetscErrorCode ierr;
6271 
6272   PetscFunctionBegin;
6273   PetscValidHeaderSpecific(mat,MAT_CLASSID,1);
6274   PetscValidType(mat,1);
6275   PetscValidHeaderSpecific(perm,IS_CLASSID,2);
6276   PetscValidPointer(info,3);
6277   PetscValidPointer(fact,4);
6278   if (mat->factortype) SETERRQ(PetscObjectComm((PetscObject)mat),PETSC_ERR_ARG_WRONGSTATE,"Not for factored matrix");
6279   if (info->levels < 0) SETERRQ1(PetscObjectComm((PetscObject)mat),PETSC_ERR_ARG_OUTOFRANGE,"Levels negative %D",(PetscInt) info->levels);
6280   if (info->fill < 1.0) SETERRQ1(PetscObjectComm((PetscObject)mat),PETSC_ERR_ARG_OUTOFRANGE,"Expected fill less than 1.0 %G",info->fill);
6281   if (!(fact)->ops->iccfactorsymbolic) {
6282     const MatSolverPackage spackage;
6283     ierr = MatFactorGetSolverPackage(fact,&spackage);CHKERRQ(ierr);
6284     SETERRQ2(PetscObjectComm((PetscObject)mat),PETSC_ERR_SUP,"Matrix type %s symbolic ICC using solver package %s",((PetscObject)mat)->type_name,spackage);
6285   }
6286   if (!mat->assembled) SETERRQ(PetscObjectComm((PetscObject)mat),PETSC_ERR_ARG_WRONGSTATE,"Not for unassembled matrix");
6287   MatCheckPreallocated(mat,2);
6288 
6289   ierr = PetscLogEventBegin(MAT_ICCFactorSymbolic,mat,perm,0,0);CHKERRQ(ierr);
6290   ierr = (fact->ops->iccfactorsymbolic)(fact,mat,perm,info);CHKERRQ(ierr);
6291   ierr = PetscLogEventEnd(MAT_ICCFactorSymbolic,mat,perm,0,0);CHKERRQ(ierr);
6292   PetscFunctionReturn(0);
6293 }
6294 
6295 #undef __FUNCT__
6296 #define __FUNCT__ "MatGetSubMatrices"
6297 /*@C
6298    MatGetSubMatrices - Extracts several submatrices from a matrix. If submat
6299    points to an array of valid matrices, they may be reused to store the new
6300    submatrices.
6301 
6302    Collective on Mat
6303 
6304    Input Parameters:
6305 +  mat - the matrix
6306 .  n   - the number of submatrixes to be extracted (on this processor, may be zero)
6307 .  irow, icol - index sets of rows and columns to extract (must be sorted)
6308 -  scall - either MAT_INITIAL_MATRIX or MAT_REUSE_MATRIX
6309 
6310    Output Parameter:
6311 .  submat - the array of submatrices
6312 
6313    Notes:
6314    MatGetSubMatrices() can extract ONLY sequential submatrices
6315    (from both sequential and parallel matrices). Use MatGetSubMatrix()
6316    to extract a parallel submatrix.
6317 
6318    Currently both row and column indices must be sorted to guarantee
6319    correctness with all matrix types.
6320 
6321    When extracting submatrices from a parallel matrix, each processor can
6322    form a different submatrix by setting the rows and columns of its
6323    individual index sets according to the local submatrix desired.
6324 
6325    When finished using the submatrices, the user should destroy
6326    them with MatDestroyMatrices().
6327 
6328    MAT_REUSE_MATRIX can only be used when the nonzero structure of the
6329    original matrix has not changed from that last call to MatGetSubMatrices().
6330 
6331    This routine creates the matrices in submat; you should NOT create them before
6332    calling it. It also allocates the array of matrix pointers submat.
6333 
6334    For BAIJ matrices the index sets must respect the block structure, that is if they
6335    request one row/column in a block, they must request all rows/columns that are in
6336    that block. For example, if the block size is 2 you cannot request just row 0 and
6337    column 0.
6338 
6339    Fortran Note:
6340    The Fortran interface is slightly different from that given below; it
6341    requires one to pass in  as submat a Mat (integer) array of size at least m.
6342 
6343    Level: advanced
6344 
6345    Concepts: matrices^accessing submatrices
6346    Concepts: submatrices
6347 
6348 .seealso: MatDestroyMatrices(), MatGetSubMatrix(), MatGetRow(), MatGetDiagonal(), MatReuse
6349 @*/
6350 PetscErrorCode  MatGetSubMatrices(Mat mat,PetscInt n,const IS irow[],const IS icol[],MatReuse scall,Mat *submat[])
6351 {
6352   PetscErrorCode ierr;
6353   PetscInt       i;
6354   PetscBool      eq;
6355 
6356   PetscFunctionBegin;
6357   PetscValidHeaderSpecific(mat,MAT_CLASSID,1);
6358   PetscValidType(mat,1);
6359   if (n) {
6360     PetscValidPointer(irow,3);
6361     PetscValidHeaderSpecific(*irow,IS_CLASSID,3);
6362     PetscValidPointer(icol,4);
6363     PetscValidHeaderSpecific(*icol,IS_CLASSID,4);
6364   }
6365   PetscValidPointer(submat,6);
6366   if (n && scall == MAT_REUSE_MATRIX) {
6367     PetscValidPointer(*submat,6);
6368     PetscValidHeaderSpecific(**submat,MAT_CLASSID,6);
6369   }
6370   if (!mat->ops->getsubmatrices) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_SUP,"Mat type %s",((PetscObject)mat)->type_name);
6371   if (!mat->assembled) SETERRQ(PetscObjectComm((PetscObject)mat),PETSC_ERR_ARG_WRONGSTATE,"Not for unassembled matrix");
6372   if (mat->factortype) SETERRQ(PetscObjectComm((PetscObject)mat),PETSC_ERR_ARG_WRONGSTATE,"Not for factored matrix");
6373   MatCheckPreallocated(mat,1);
6374 
6375   ierr = PetscLogEventBegin(MAT_GetSubMatrices,mat,0,0,0);CHKERRQ(ierr);
6376   ierr = (*mat->ops->getsubmatrices)(mat,n,irow,icol,scall,submat);CHKERRQ(ierr);
6377   ierr = PetscLogEventEnd(MAT_GetSubMatrices,mat,0,0,0);CHKERRQ(ierr);
6378   for (i=0; i<n; i++) {
6379     if (mat->symmetric || mat->structurally_symmetric || mat->hermitian) {
6380       ierr = ISEqual(irow[i],icol[i],&eq);CHKERRQ(ierr);
6381       if (eq) {
6382         if (mat->symmetric) {
6383           ierr = MatSetOption((*submat)[i],MAT_SYMMETRIC,PETSC_TRUE);CHKERRQ(ierr);
6384         } else if (mat->hermitian) {
6385           ierr = MatSetOption((*submat)[i],MAT_HERMITIAN,PETSC_TRUE);CHKERRQ(ierr);
6386         } else if (mat->structurally_symmetric) {
6387           ierr = MatSetOption((*submat)[i],MAT_STRUCTURALLY_SYMMETRIC,PETSC_TRUE);CHKERRQ(ierr);
6388         }
6389       }
6390     }
6391   }
6392   PetscFunctionReturn(0);
6393 }
6394 
6395 #undef __FUNCT__
6396 #define __FUNCT__ "MatGetSubMatricesParallel"
6397 PetscErrorCode  MatGetSubMatricesParallel(Mat mat,PetscInt n,const IS irow[],const IS icol[],MatReuse scall,Mat *submat[])
6398 {
6399   PetscErrorCode ierr;
6400   PetscInt       i;
6401   PetscBool      eq;
6402 
6403   PetscFunctionBegin;
6404   PetscValidHeaderSpecific(mat,MAT_CLASSID,1);
6405   PetscValidType(mat,1);
6406   if (n) {
6407     PetscValidPointer(irow,3);
6408     PetscValidHeaderSpecific(*irow,IS_CLASSID,3);
6409     PetscValidPointer(icol,4);
6410     PetscValidHeaderSpecific(*icol,IS_CLASSID,4);
6411   }
6412   PetscValidPointer(submat,6);
6413   if (n && scall == MAT_REUSE_MATRIX) {
6414     PetscValidPointer(*submat,6);
6415     PetscValidHeaderSpecific(**submat,MAT_CLASSID,6);
6416   }
6417   if (!mat->ops->getsubmatricesparallel) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_SUP,"Mat type %s",((PetscObject)mat)->type_name);
6418   if (!mat->assembled) SETERRQ(PetscObjectComm((PetscObject)mat),PETSC_ERR_ARG_WRONGSTATE,"Not for unassembled matrix");
6419   if (mat->factortype) SETERRQ(PetscObjectComm((PetscObject)mat),PETSC_ERR_ARG_WRONGSTATE,"Not for factored matrix");
6420   MatCheckPreallocated(mat,1);
6421 
6422   ierr = PetscLogEventBegin(MAT_GetSubMatrices,mat,0,0,0);CHKERRQ(ierr);
6423   ierr = (*mat->ops->getsubmatricesparallel)(mat,n,irow,icol,scall,submat);CHKERRQ(ierr);
6424   ierr = PetscLogEventEnd(MAT_GetSubMatrices,mat,0,0,0);CHKERRQ(ierr);
6425   for (i=0; i<n; i++) {
6426     if (mat->symmetric || mat->structurally_symmetric || mat->hermitian) {
6427       ierr = ISEqual(irow[i],icol[i],&eq);CHKERRQ(ierr);
6428       if (eq) {
6429         if (mat->symmetric) {
6430           ierr = MatSetOption((*submat)[i],MAT_SYMMETRIC,PETSC_TRUE);CHKERRQ(ierr);
6431         } else if (mat->hermitian) {
6432           ierr = MatSetOption((*submat)[i],MAT_HERMITIAN,PETSC_TRUE);CHKERRQ(ierr);
6433         } else if (mat->structurally_symmetric) {
6434           ierr = MatSetOption((*submat)[i],MAT_STRUCTURALLY_SYMMETRIC,PETSC_TRUE);CHKERRQ(ierr);
6435         }
6436       }
6437     }
6438   }
6439   PetscFunctionReturn(0);
6440 }
6441 
6442 #undef __FUNCT__
6443 #define __FUNCT__ "MatDestroyMatrices"
6444 /*@C
6445    MatDestroyMatrices - Destroys a set of matrices obtained with MatGetSubMatrices().
6446 
6447    Collective on Mat
6448 
6449    Input Parameters:
6450 +  n - the number of local matrices
6451 -  mat - the matrices (note that this is a pointer to the array of matrices, just to match the calling
6452                        sequence of MatGetSubMatrices())
6453 
6454    Level: advanced
6455 
6456     Notes: Frees not only the matrices, but also the array that contains the matrices
6457            In Fortran will not free the array.
6458 
6459 .seealso: MatGetSubMatrices()
6460 @*/
6461 PetscErrorCode  MatDestroyMatrices(PetscInt n,Mat *mat[])
6462 {
6463   PetscErrorCode ierr;
6464   PetscInt       i;
6465 
6466   PetscFunctionBegin;
6467   if (!*mat) PetscFunctionReturn(0);
6468   if (n < 0) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Trying to destroy negative number of matrices %D",n);
6469   PetscValidPointer(mat,2);
6470   for (i=0; i<n; i++) {
6471     ierr = MatDestroy(&(*mat)[i]);CHKERRQ(ierr);
6472   }
6473   /* memory is allocated even if n = 0 */
6474   ierr = PetscFree(*mat);CHKERRQ(ierr);
6475   *mat = NULL;
6476   PetscFunctionReturn(0);
6477 }
6478 
6479 #undef __FUNCT__
6480 #define __FUNCT__ "MatGetSeqNonzeroStructure"
6481 /*@C
6482    MatGetSeqNonzeroStructure - Extracts the sequential nonzero structure from a matrix.
6483 
6484    Collective on Mat
6485 
6486    Input Parameters:
6487 .  mat - the matrix
6488 
6489    Output Parameter:
6490 .  matstruct - the sequential matrix with the nonzero structure of mat
6491 
6492   Level: intermediate
6493 
6494 .seealso: MatDestroySeqNonzeroStructure(), MatGetSubMatrices(), MatDestroyMatrices()
6495 @*/
6496 PetscErrorCode  MatGetSeqNonzeroStructure(Mat mat,Mat *matstruct)
6497 {
6498   PetscErrorCode ierr;
6499 
6500   PetscFunctionBegin;
6501   PetscValidHeaderSpecific(mat,MAT_CLASSID,1);
6502   PetscValidPointer(matstruct,2);
6503 
6504   PetscValidType(mat,1);
6505   if (mat->factortype) SETERRQ(PetscObjectComm((PetscObject)mat),PETSC_ERR_ARG_WRONGSTATE,"Not for factored matrix");
6506   MatCheckPreallocated(mat,1);
6507 
6508   if (!mat->ops->getseqnonzerostructure) SETERRQ1(PetscObjectComm((PetscObject)mat),PETSC_ERR_SUP,"Not for matrix type %s\n",((PetscObject)mat)->type_name);
6509   ierr = PetscLogEventBegin(MAT_GetSeqNonzeroStructure,mat,0,0,0);CHKERRQ(ierr);
6510   ierr = (*mat->ops->getseqnonzerostructure)(mat,matstruct);CHKERRQ(ierr);
6511   ierr = PetscLogEventEnd(MAT_GetSeqNonzeroStructure,mat,0,0,0);CHKERRQ(ierr);
6512   PetscFunctionReturn(0);
6513 }
6514 
6515 #undef __FUNCT__
6516 #define __FUNCT__ "MatDestroySeqNonzeroStructure"
6517 /*@C
6518    MatDestroySeqNonzeroStructure - Destroys matrix obtained with MatGetSeqNonzeroStructure().
6519 
6520    Collective on Mat
6521 
6522    Input Parameters:
6523 .  mat - the matrix (note that this is a pointer to the array of matrices, just to match the calling
6524                        sequence of MatGetSequentialNonzeroStructure())
6525 
6526    Level: advanced
6527 
6528     Notes: Frees not only the matrices, but also the array that contains the matrices
6529 
6530 .seealso: MatGetSeqNonzeroStructure()
6531 @*/
6532 PetscErrorCode  MatDestroySeqNonzeroStructure(Mat *mat)
6533 {
6534   PetscErrorCode ierr;
6535 
6536   PetscFunctionBegin;
6537   PetscValidPointer(mat,1);
6538   ierr = MatDestroy(mat);CHKERRQ(ierr);
6539   PetscFunctionReturn(0);
6540 }
6541 
6542 #undef __FUNCT__
6543 #define __FUNCT__ "MatIncreaseOverlap"
6544 /*@
6545    MatIncreaseOverlap - Given a set of submatrices indicated by index sets,
6546    replaces the index sets by larger ones that represent submatrices with
6547    additional overlap.
6548 
6549    Collective on Mat
6550 
6551    Input Parameters:
6552 +  mat - the matrix
6553 .  n   - the number of index sets
6554 .  is  - the array of index sets (these index sets will changed during the call)
6555 -  ov  - the additional overlap requested
6556 
6557    Level: developer
6558 
6559    Concepts: overlap
6560    Concepts: ASM^computing overlap
6561 
6562 .seealso: MatGetSubMatrices()
6563 @*/
6564 PetscErrorCode  MatIncreaseOverlap(Mat mat,PetscInt n,IS is[],PetscInt ov)
6565 {
6566   PetscErrorCode ierr;
6567 
6568   PetscFunctionBegin;
6569   PetscValidHeaderSpecific(mat,MAT_CLASSID,1);
6570   PetscValidType(mat,1);
6571   if (n < 0) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Must have one or more domains, you have %D",n);
6572   if (n) {
6573     PetscValidPointer(is,3);
6574     PetscValidHeaderSpecific(*is,IS_CLASSID,3);
6575   }
6576   if (!mat->assembled) SETERRQ(PetscObjectComm((PetscObject)mat),PETSC_ERR_ARG_WRONGSTATE,"Not for unassembled matrix");
6577   if (mat->factortype) SETERRQ(PetscObjectComm((PetscObject)mat),PETSC_ERR_ARG_WRONGSTATE,"Not for factored matrix");
6578   MatCheckPreallocated(mat,1);
6579 
6580   if (!ov) PetscFunctionReturn(0);
6581   if (!mat->ops->increaseoverlap) SETERRQ1(PetscObjectComm((PetscObject)mat),PETSC_ERR_SUP,"Mat type %s",((PetscObject)mat)->type_name);
6582   ierr = PetscLogEventBegin(MAT_IncreaseOverlap,mat,0,0,0);CHKERRQ(ierr);
6583   ierr = (*mat->ops->increaseoverlap)(mat,n,is,ov);CHKERRQ(ierr);
6584   ierr = PetscLogEventEnd(MAT_IncreaseOverlap,mat,0,0,0);CHKERRQ(ierr);
6585   PetscFunctionReturn(0);
6586 }
6587 
6588 #undef __FUNCT__
6589 #define __FUNCT__ "MatGetBlockSize"
6590 /*@
6591    MatGetBlockSize - Returns the matrix block size; useful especially for the
6592    block row and block diagonal formats.
6593 
6594    Not Collective
6595 
6596    Input Parameter:
6597 .  mat - the matrix
6598 
6599    Output Parameter:
6600 .  bs - block size
6601 
6602    Notes:
6603    Block row formats are MATSEQBAIJ, MATMPIBAIJ, MATSEQSBAIJ, MATMPISBAIJ
6604 
6605    Level: intermediate
6606 
6607    Concepts: matrices^block size
6608 
6609 .seealso: MatCreateSeqBAIJ(), MatCreateBAIJ(), MatGetBlockSizes()
6610 @*/
6611 PetscErrorCode  MatGetBlockSize(Mat mat,PetscInt *bs)
6612 {
6613   PetscFunctionBegin;
6614   PetscValidHeaderSpecific(mat,MAT_CLASSID,1);
6615   PetscValidType(mat,1);
6616   PetscValidIntPointer(bs,2);
6617   MatCheckPreallocated(mat,1);
6618   *bs = mat->rmap->bs;
6619   PetscFunctionReturn(0);
6620 }
6621 
6622 #undef __FUNCT__
6623 #define __FUNCT__ "MatGetBlockSizes"
6624 /*@
6625    MatGetBlockSizes - Returns the matrix block row and column sizes;
6626    useful especially for the block row and block diagonal formats.
6627 
6628    Not Collective
6629 
6630    Input Parameter:
6631 .  mat - the matrix
6632 
6633    Output Parameter:
6634 .  rbs - row block size
6635 .  cbs - coumn block size
6636 
6637    Notes:
6638    Block row formats are MATSEQBAIJ, MATMPIBAIJ, MATSEQSBAIJ, MATMPISBAIJ
6639 
6640    Level: intermediate
6641 
6642    Concepts: matrices^block size
6643 
6644 .seealso: MatCreateSeqBAIJ(), MatCreateBAIJ(), MatGetBlockSize()
6645 @*/
6646 PetscErrorCode  MatGetBlockSizes(Mat mat,PetscInt *rbs, PetscInt *cbs)
6647 {
6648   PetscFunctionBegin;
6649   PetscValidHeaderSpecific(mat,MAT_CLASSID,1);
6650   PetscValidType(mat,1);
6651   if (rbs) PetscValidIntPointer(rbs,2);
6652   if (cbs) PetscValidIntPointer(cbs,3);
6653   MatCheckPreallocated(mat,1);
6654   if (rbs) *rbs = mat->rmap->bs;
6655   if (cbs) *cbs = mat->cmap->bs;
6656   PetscFunctionReturn(0);
6657 }
6658 
6659 #undef __FUNCT__
6660 #define __FUNCT__ "MatSetBlockSize"
6661 /*@
6662    MatSetBlockSize - Sets the matrix block size.
6663 
6664    Logically Collective on Mat
6665 
6666    Input Parameters:
6667 +  mat - the matrix
6668 -  bs - block size
6669 
6670    Notes:
6671      This must be called before MatSetUp() or MatXXXSetPreallocation() (or will default to 1) and the block size cannot be changed later
6672 
6673    Level: intermediate
6674 
6675    Concepts: matrices^block size
6676 
6677 .seealso: MatCreateSeqBAIJ(), MatCreateBAIJ(), MatGetBlockSize()
6678 @*/
6679 PetscErrorCode  MatSetBlockSize(Mat mat,PetscInt bs)
6680 {
6681   PetscErrorCode ierr;
6682 
6683   PetscFunctionBegin;
6684   PetscValidHeaderSpecific(mat,MAT_CLASSID,1);
6685   PetscValidLogicalCollectiveInt(mat,bs,2);
6686   ierr = PetscLayoutSetBlockSize(mat->rmap,bs);CHKERRQ(ierr);
6687   ierr = PetscLayoutSetBlockSize(mat->cmap,bs);CHKERRQ(ierr);
6688   PetscFunctionReturn(0);
6689 }
6690 
6691 #undef __FUNCT__
6692 #define __FUNCT__ "MatSetBlockSizes"
6693 /*@
6694    MatSetBlockSizes - Sets the matrix block row and column sizes.
6695 
6696    Logically Collective on Mat
6697 
6698    Input Parameters:
6699 +  mat - the matrix
6700 -  rbs - row block size
6701 -  cbs - column block size
6702 
6703    Notes:
6704      This must be called before MatSetUp() or MatXXXSetPreallocation() (or will default to 1) and the block size cannot be changed later
6705 
6706    Level: intermediate
6707 
6708    Concepts: matrices^block size
6709 
6710 .seealso: MatCreateSeqBAIJ(), MatCreateBAIJ(), MatGetBlockSize()
6711 @*/
6712 PetscErrorCode  MatSetBlockSizes(Mat mat,PetscInt rbs,PetscInt cbs)
6713 {
6714   PetscErrorCode ierr;
6715 
6716   PetscFunctionBegin;
6717   PetscValidHeaderSpecific(mat,MAT_CLASSID,1);
6718   PetscValidLogicalCollectiveInt(mat,rbs,2);
6719   ierr = PetscLayoutSetBlockSize(mat->rmap,rbs);CHKERRQ(ierr);
6720   ierr = PetscLayoutSetBlockSize(mat->cmap,cbs);CHKERRQ(ierr);
6721   PetscFunctionReturn(0);
6722 }
6723 
6724 #undef __FUNCT__
6725 #define __FUNCT__ "MatGetRowIJ"
6726 /*@C
6727     MatGetRowIJ - Returns the compressed row storage i and j indices for sequential matrices.
6728 
6729    Collective on Mat
6730 
6731     Input Parameters:
6732 +   mat - the matrix
6733 .   shift -  0 or 1 indicating we want the indices starting at 0 or 1
6734 .   symmetric - PETSC_TRUE or PETSC_FALSE indicating the matrix data structure should be   symmetrized
6735 -   inodecompressed - PETSC_TRUE or PETSC_FALSE  indicating if the nonzero structure of the
6736                  inodes or the nonzero elements is wanted. For BAIJ matrices the compressed version is
6737                  always used.
6738 
6739     Output Parameters:
6740 +   n - number of rows in the (possibly compressed) matrix
6741 .   ia - the row pointers [of length n+1]
6742 .   ja - the column indices
6743 -   done - indicates if the routine actually worked and returned appropriate ia[] and ja[] arrays; callers
6744            are responsible for handling the case when done == PETSC_FALSE and ia and ja are not set
6745 
6746     Level: developer
6747 
6748     Notes: You CANNOT change any of the ia[] or ja[] values.
6749 
6750            Use MatRestoreRowIJ() when you are finished accessing the ia[] and ja[] values
6751 
6752     Fortran Node
6753 
6754            In Fortran use
6755 $           PetscInt ia(1), ja(1)
6756 $           PetscOffset iia, jja
6757 $      call MatGetRowIJ(mat,shift,symmetric,inodecompressed,n,ia,iia,ja,jja,done,ierr)
6758 $
6759 $          or
6760 $
6761 $           PetscScalar, pointer :: xx_v(:)
6762 $    call  MatGetRowIJF90(mat,shift,symmetric,inodecompressed,n,ia,ja,done,ierr)
6763 
6764 
6765        Acess the ith and jth entries via ia(iia + i) and ja(jja + j)
6766 
6767 .seealso: MatGetColumnIJ(), MatRestoreRowIJ(), MatSeqAIJGetArray()
6768 @*/
6769 PetscErrorCode MatGetRowIJ(Mat mat,PetscInt shift,PetscBool symmetric,PetscBool inodecompressed,PetscInt *n,const PetscInt *ia[],const PetscInt *ja[],PetscBool  *done)
6770 {
6771   PetscErrorCode ierr;
6772 
6773   PetscFunctionBegin;
6774   PetscValidHeaderSpecific(mat,MAT_CLASSID,1);
6775   PetscValidType(mat,1);
6776   PetscValidIntPointer(n,4);
6777   if (ia) PetscValidIntPointer(ia,5);
6778   if (ja) PetscValidIntPointer(ja,6);
6779   PetscValidIntPointer(done,7);
6780   MatCheckPreallocated(mat,1);
6781   if (!mat->ops->getrowij) *done = PETSC_FALSE;
6782   else {
6783     *done = PETSC_TRUE;
6784     ierr  = PetscLogEventBegin(MAT_GetRowIJ,mat,0,0,0);CHKERRQ(ierr);
6785     ierr  = (*mat->ops->getrowij)(mat,shift,symmetric,inodecompressed,n,ia,ja,done);CHKERRQ(ierr);
6786     ierr  = PetscLogEventEnd(MAT_GetRowIJ,mat,0,0,0);CHKERRQ(ierr);
6787   }
6788   PetscFunctionReturn(0);
6789 }
6790 
6791 #undef __FUNCT__
6792 #define __FUNCT__ "MatGetColumnIJ"
6793 /*@C
6794     MatGetColumnIJ - Returns the compressed column storage i and j indices for sequential matrices.
6795 
6796     Collective on Mat
6797 
6798     Input Parameters:
6799 +   mat - the matrix
6800 .   shift - 1 or zero indicating we want the indices starting at 0 or 1
6801 .   symmetric - PETSC_TRUE or PETSC_FALSE indicating the matrix data structure should be
6802                 symmetrized
6803 -   inodecompressed - PETSC_TRUE or PETSC_FALSE indicating if the nonzero structure of the
6804                  inodes or the nonzero elements is wanted. For BAIJ matrices the compressed version is
6805                  always used.
6806 
6807     Output Parameters:
6808 +   n - number of columns in the (possibly compressed) matrix
6809 .   ia - the column pointers
6810 .   ja - the row indices
6811 -   done - PETSC_TRUE or PETSC_FALSE, indicating whether the values have been returned
6812 
6813     Level: developer
6814 
6815 .seealso: MatGetRowIJ(), MatRestoreColumnIJ()
6816 @*/
6817 PetscErrorCode MatGetColumnIJ(Mat mat,PetscInt shift,PetscBool symmetric,PetscBool inodecompressed,PetscInt *n,const PetscInt *ia[],const PetscInt *ja[],PetscBool  *done)
6818 {
6819   PetscErrorCode ierr;
6820 
6821   PetscFunctionBegin;
6822   PetscValidHeaderSpecific(mat,MAT_CLASSID,1);
6823   PetscValidType(mat,1);
6824   PetscValidIntPointer(n,4);
6825   if (ia) PetscValidIntPointer(ia,5);
6826   if (ja) PetscValidIntPointer(ja,6);
6827   PetscValidIntPointer(done,7);
6828   MatCheckPreallocated(mat,1);
6829   if (!mat->ops->getcolumnij) *done = PETSC_FALSE;
6830   else {
6831     *done = PETSC_TRUE;
6832     ierr  = (*mat->ops->getcolumnij)(mat,shift,symmetric,inodecompressed,n,ia,ja,done);CHKERRQ(ierr);
6833   }
6834   PetscFunctionReturn(0);
6835 }
6836 
6837 #undef __FUNCT__
6838 #define __FUNCT__ "MatRestoreRowIJ"
6839 /*@C
6840     MatRestoreRowIJ - Call after you are completed with the ia,ja indices obtained with
6841     MatGetRowIJ().
6842 
6843     Collective on Mat
6844 
6845     Input Parameters:
6846 +   mat - the matrix
6847 .   shift - 1 or zero indicating we want the indices starting at 0 or 1
6848 .   symmetric - PETSC_TRUE or PETSC_FALSE indicating the matrix data structure should be
6849                 symmetrized
6850 -   inodecompressed -  PETSC_TRUE or PETSC_FALSE indicating if the nonzero structure of the
6851                  inodes or the nonzero elements is wanted. For BAIJ matrices the compressed version is
6852                  always used.
6853 
6854     Output Parameters:
6855 +   n - size of (possibly compressed) matrix
6856 .   ia - the row pointers
6857 .   ja - the column indices
6858 -   done - PETSC_TRUE or PETSC_FALSE indicated that the values have been returned
6859 
6860     Level: developer
6861 
6862 .seealso: MatGetRowIJ(), MatRestoreColumnIJ()
6863 @*/
6864 PetscErrorCode MatRestoreRowIJ(Mat mat,PetscInt shift,PetscBool symmetric,PetscBool inodecompressed,PetscInt *n,const PetscInt *ia[],const PetscInt *ja[],PetscBool  *done)
6865 {
6866   PetscErrorCode ierr;
6867 
6868   PetscFunctionBegin;
6869   PetscValidHeaderSpecific(mat,MAT_CLASSID,1);
6870   PetscValidType(mat,1);
6871   if (ia) PetscValidIntPointer(ia,5);
6872   if (ja) PetscValidIntPointer(ja,6);
6873   PetscValidIntPointer(done,7);
6874   MatCheckPreallocated(mat,1);
6875 
6876   if (!mat->ops->restorerowij) *done = PETSC_FALSE;
6877   else {
6878     *done = PETSC_TRUE;
6879     ierr  = (*mat->ops->restorerowij)(mat,shift,symmetric,inodecompressed,n,ia,ja,done);CHKERRQ(ierr);
6880     if (n)  *n = 0;
6881     if (ia) *ia = NULL;
6882     if (ja) *ja = NULL;
6883   }
6884   PetscFunctionReturn(0);
6885 }
6886 
6887 #undef __FUNCT__
6888 #define __FUNCT__ "MatRestoreColumnIJ"
6889 /*@C
6890     MatRestoreColumnIJ - Call after you are completed with the ia,ja indices obtained with
6891     MatGetColumnIJ().
6892 
6893     Collective on Mat
6894 
6895     Input Parameters:
6896 +   mat - the matrix
6897 .   shift - 1 or zero indicating we want the indices starting at 0 or 1
6898 -   symmetric - PETSC_TRUE or PETSC_FALSE indicating the matrix data structure should be
6899                 symmetrized
6900 -   inodecompressed - PETSC_TRUE or PETSC_FALSE indicating if the nonzero structure of the
6901                  inodes or the nonzero elements is wanted. For BAIJ matrices the compressed version is
6902                  always used.
6903 
6904     Output Parameters:
6905 +   n - size of (possibly compressed) matrix
6906 .   ia - the column pointers
6907 .   ja - the row indices
6908 -   done - PETSC_TRUE or PETSC_FALSE indicated that the values have been returned
6909 
6910     Level: developer
6911 
6912 .seealso: MatGetColumnIJ(), MatRestoreRowIJ()
6913 @*/
6914 PetscErrorCode MatRestoreColumnIJ(Mat mat,PetscInt shift,PetscBool symmetric,PetscBool inodecompressed,PetscInt *n,const PetscInt *ia[],const PetscInt *ja[],PetscBool  *done)
6915 {
6916   PetscErrorCode ierr;
6917 
6918   PetscFunctionBegin;
6919   PetscValidHeaderSpecific(mat,MAT_CLASSID,1);
6920   PetscValidType(mat,1);
6921   if (ia) PetscValidIntPointer(ia,5);
6922   if (ja) PetscValidIntPointer(ja,6);
6923   PetscValidIntPointer(done,7);
6924   MatCheckPreallocated(mat,1);
6925 
6926   if (!mat->ops->restorecolumnij) *done = PETSC_FALSE;
6927   else {
6928     *done = PETSC_TRUE;
6929     ierr  = (*mat->ops->restorecolumnij)(mat,shift,symmetric,inodecompressed,n,ia,ja,done);CHKERRQ(ierr);
6930     if (n)  *n = 0;
6931     if (ia) *ia = NULL;
6932     if (ja) *ja = NULL;
6933   }
6934   PetscFunctionReturn(0);
6935 }
6936 
6937 #undef __FUNCT__
6938 #define __FUNCT__ "MatColoringPatch"
6939 /*@C
6940     MatColoringPatch -Used inside matrix coloring routines that
6941     use MatGetRowIJ() and/or MatGetColumnIJ().
6942 
6943     Collective on Mat
6944 
6945     Input Parameters:
6946 +   mat - the matrix
6947 .   ncolors - max color value
6948 .   n   - number of entries in colorarray
6949 -   colorarray - array indicating color for each column
6950 
6951     Output Parameters:
6952 .   iscoloring - coloring generated using colorarray information
6953 
6954     Level: developer
6955 
6956 .seealso: MatGetRowIJ(), MatGetColumnIJ()
6957 
6958 @*/
6959 PetscErrorCode  MatColoringPatch(Mat mat,PetscInt ncolors,PetscInt n,ISColoringValue colorarray[],ISColoring *iscoloring)
6960 {
6961   PetscErrorCode ierr;
6962 
6963   PetscFunctionBegin;
6964   PetscValidHeaderSpecific(mat,MAT_CLASSID,1);
6965   PetscValidType(mat,1);
6966   PetscValidIntPointer(colorarray,4);
6967   PetscValidPointer(iscoloring,5);
6968   MatCheckPreallocated(mat,1);
6969 
6970   if (!mat->ops->coloringpatch) {
6971     ierr = ISColoringCreate(PetscObjectComm((PetscObject)mat),ncolors,n,colorarray,iscoloring);CHKERRQ(ierr);
6972   } else {
6973     ierr = (*mat->ops->coloringpatch)(mat,ncolors,n,colorarray,iscoloring);CHKERRQ(ierr);
6974   }
6975   PetscFunctionReturn(0);
6976 }
6977 
6978 
6979 #undef __FUNCT__
6980 #define __FUNCT__ "MatSetUnfactored"
6981 /*@
6982    MatSetUnfactored - Resets a factored matrix to be treated as unfactored.
6983 
6984    Logically Collective on Mat
6985 
6986    Input Parameter:
6987 .  mat - the factored matrix to be reset
6988 
6989    Notes:
6990    This routine should be used only with factored matrices formed by in-place
6991    factorization via ILU(0) (or by in-place LU factorization for the MATSEQDENSE
6992    format).  This option can save memory, for example, when solving nonlinear
6993    systems with a matrix-free Newton-Krylov method and a matrix-based, in-place
6994    ILU(0) preconditioner.
6995 
6996    Note that one can specify in-place ILU(0) factorization by calling
6997 .vb
6998      PCType(pc,PCILU);
6999      PCFactorSeUseInPlace(pc);
7000 .ve
7001    or by using the options -pc_type ilu -pc_factor_in_place
7002 
7003    In-place factorization ILU(0) can also be used as a local
7004    solver for the blocks within the block Jacobi or additive Schwarz
7005    methods (runtime option: -sub_pc_factor_in_place).  See the discussion
7006    of these preconditioners in the <a href="../../docs/manual.pdf#ch_pc">PC chapter of the users manual</a> for details on setting
7007    local solver options.
7008 
7009    Most users should employ the simplified KSP interface for linear solvers
7010    instead of working directly with matrix algebra routines such as this.
7011    See, e.g., KSPCreate().
7012 
7013    Level: developer
7014 
7015 .seealso: PCFactorSetUseInPlace()
7016 
7017    Concepts: matrices^unfactored
7018 
7019 @*/
7020 PetscErrorCode  MatSetUnfactored(Mat mat)
7021 {
7022   PetscErrorCode ierr;
7023 
7024   PetscFunctionBegin;
7025   PetscValidHeaderSpecific(mat,MAT_CLASSID,1);
7026   PetscValidType(mat,1);
7027   MatCheckPreallocated(mat,1);
7028   mat->factortype = MAT_FACTOR_NONE;
7029   if (!mat->ops->setunfactored) PetscFunctionReturn(0);
7030   ierr = (*mat->ops->setunfactored)(mat);CHKERRQ(ierr);
7031   PetscFunctionReturn(0);
7032 }
7033 
7034 /*MC
7035     MatDenseGetArrayF90 - Accesses a matrix array from Fortran90.
7036 
7037     Synopsis:
7038     MatDenseGetArrayF90(Mat x,{Scalar, pointer :: xx_v(:,:)},integer ierr)
7039 
7040     Not collective
7041 
7042     Input Parameter:
7043 .   x - matrix
7044 
7045     Output Parameters:
7046 +   xx_v - the Fortran90 pointer to the array
7047 -   ierr - error code
7048 
7049     Example of Usage:
7050 .vb
7051       PetscScalar, pointer xx_v(:,:)
7052       ....
7053       call MatDenseGetArrayF90(x,xx_v,ierr)
7054       a = xx_v(3)
7055       call MatDenseRestoreArrayF90(x,xx_v,ierr)
7056 .ve
7057 
7058     Level: advanced
7059 
7060 .seealso:  MatDenseRestoreArrayF90(), MatDenseGetArray(), MatDenseRestoreArray()
7061 
7062     Concepts: matrices^accessing array
7063 
7064 M*/
7065 
7066 /*MC
7067     MatDenseRestoreArrayF90 - Restores a matrix array that has been
7068     accessed with MatGetArrayF90().
7069 
7070     Synopsis:
7071     MatDenseRestoreArrayF90(Mat x,{Scalar, pointer :: xx_v(:)},integer ierr)
7072 
7073     Not collective
7074 
7075     Input Parameters:
7076 +   x - matrix
7077 -   xx_v - the Fortran90 pointer to the array
7078 
7079     Output Parameter:
7080 .   ierr - error code
7081 
7082     Example of Usage:
7083 .vb
7084        PetscScalar, pointer xx_v(:)
7085        ....
7086        call MatDenseGetArrayF90(x,xx_v,ierr)
7087        a = xx_v(3)
7088        call MatDenseRestoreArrayF90(x,xx_v,ierr)
7089 .ve
7090 
7091     Level: advanced
7092 
7093 .seealso:  MatDenseGetArrayF90(), MatDenseGetArray(), MatDenseRestoreArray()
7094 
7095 M*/
7096 
7097 
7098 #undef __FUNCT__
7099 #define __FUNCT__ "MatGetSubMatrix"
7100 /*@
7101     MatGetSubMatrix - Gets a single submatrix on the same number of processors
7102                       as the original matrix.
7103 
7104     Collective on Mat
7105 
7106     Input Parameters:
7107 +   mat - the original matrix
7108 .   isrow - parallel IS containing the rows this processor should obtain
7109 .   iscol - parallel IS containing all columns you wish to keep. Each process should list the columns that will be in IT's "diagonal part" in the new matrix.
7110 -   cll - either MAT_INITIAL_MATRIX or MAT_REUSE_MATRIX
7111 
7112     Output Parameter:
7113 .   newmat - the new submatrix, of the same type as the old
7114 
7115     Level: advanced
7116 
7117     Notes:
7118     The submatrix will be able to be multiplied with vectors using the same layout as iscol.
7119 
7120     The rows in isrow will be sorted into the same order as the original matrix on each process.
7121 
7122       The first time this is called you should use a cll of MAT_INITIAL_MATRIX,
7123    the MatGetSubMatrix() routine will create the newmat for you. Any additional calls
7124    to this routine with a mat of the same nonzero structure and with a call of MAT_REUSE_MATRIX
7125    will reuse the matrix generated the first time.  You should call MatDestroy() on newmat when
7126    you are finished using it.
7127 
7128     The communicator of the newly obtained matrix is ALWAYS the same as the communicator of
7129     the input matrix.
7130 
7131     If iscol is NULL then all columns are obtained (not supported in Fortran).
7132 
7133    Example usage:
7134    Consider the following 8x8 matrix with 34 non-zero values, that is
7135    assembled across 3 processors. Let's assume that proc0 owns 3 rows,
7136    proc1 owns 3 rows, proc2 owns 2 rows. This division can be shown
7137    as follows:
7138 
7139 .vb
7140             1  2  0  |  0  3  0  |  0  4
7141     Proc0   0  5  6  |  7  0  0  |  8  0
7142             9  0 10  | 11  0  0  | 12  0
7143     -------------------------------------
7144            13  0 14  | 15 16 17  |  0  0
7145     Proc1   0 18  0  | 19 20 21  |  0  0
7146             0  0  0  | 22 23  0  | 24  0
7147     -------------------------------------
7148     Proc2  25 26 27  |  0  0 28  | 29  0
7149            30  0  0  | 31 32 33  |  0 34
7150 .ve
7151 
7152     Suppose isrow = [0 1 | 4 | 6 7] and iscol = [1 2 | 3 4 5 | 6].  The resulting submatrix is
7153 
7154 .vb
7155             2  0  |  0  3  0  |  0
7156     Proc0   5  6  |  7  0  0  |  8
7157     -------------------------------
7158     Proc1  18  0  | 19 20 21  |  0
7159     -------------------------------
7160     Proc2  26 27  |  0  0 28  | 29
7161             0  0  | 31 32 33  |  0
7162 .ve
7163 
7164 
7165     Concepts: matrices^submatrices
7166 
7167 .seealso: MatGetSubMatrices()
7168 @*/
7169 PetscErrorCode  MatGetSubMatrix(Mat mat,IS isrow,IS iscol,MatReuse cll,Mat *newmat)
7170 {
7171   PetscErrorCode ierr;
7172   PetscMPIInt    size;
7173   Mat            *local;
7174   IS             iscoltmp;
7175 
7176   PetscFunctionBegin;
7177   PetscValidHeaderSpecific(mat,MAT_CLASSID,1);
7178   PetscValidHeaderSpecific(isrow,IS_CLASSID,2);
7179   if (iscol) PetscValidHeaderSpecific(iscol,IS_CLASSID,3);
7180   PetscValidPointer(newmat,5);
7181   if (cll == MAT_REUSE_MATRIX) PetscValidHeaderSpecific(*newmat,MAT_CLASSID,5);
7182   PetscValidType(mat,1);
7183   if (mat->factortype) SETERRQ(PetscObjectComm((PetscObject)mat),PETSC_ERR_ARG_WRONGSTATE,"Not for factored matrix");
7184   MatCheckPreallocated(mat,1);
7185   ierr = MPI_Comm_size(PetscObjectComm((PetscObject)mat),&size);CHKERRQ(ierr);
7186 
7187   if (!iscol) {
7188     ierr = ISCreateStride(PetscObjectComm((PetscObject)mat),mat->cmap->n,mat->cmap->rstart,1,&iscoltmp);CHKERRQ(ierr);
7189   } else {
7190     iscoltmp = iscol;
7191   }
7192 
7193   /* if original matrix is on just one processor then use submatrix generated */
7194   if (mat->ops->getsubmatrices && !mat->ops->getsubmatrix && size == 1 && cll == MAT_REUSE_MATRIX) {
7195     ierr = MatGetSubMatrices(mat,1,&isrow,&iscoltmp,MAT_REUSE_MATRIX,&newmat);CHKERRQ(ierr);
7196     if (!iscol) {ierr = ISDestroy(&iscoltmp);CHKERRQ(ierr);}
7197     PetscFunctionReturn(0);
7198   } else if (mat->ops->getsubmatrices && !mat->ops->getsubmatrix && size == 1) {
7199     ierr    = MatGetSubMatrices(mat,1,&isrow,&iscoltmp,MAT_INITIAL_MATRIX,&local);CHKERRQ(ierr);
7200     *newmat = *local;
7201     ierr    = PetscFree(local);CHKERRQ(ierr);
7202     if (!iscol) {ierr = ISDestroy(&iscoltmp);CHKERRQ(ierr);}
7203     PetscFunctionReturn(0);
7204   } else if (!mat->ops->getsubmatrix) {
7205     /* Create a new matrix type that implements the operation using the full matrix */
7206     switch (cll) {
7207     case MAT_INITIAL_MATRIX:
7208       ierr = MatCreateSubMatrix(mat,isrow,iscoltmp,newmat);CHKERRQ(ierr);
7209       break;
7210     case MAT_REUSE_MATRIX:
7211       ierr = MatSubMatrixUpdate(*newmat,mat,isrow,iscoltmp);CHKERRQ(ierr);
7212       break;
7213     default: SETERRQ(PetscObjectComm((PetscObject)mat),PETSC_ERR_ARG_OUTOFRANGE,"Invalid MatReuse, must be either MAT_INITIAL_MATRIX or MAT_REUSE_MATRIX");
7214     }
7215     if (!iscol) {ierr = ISDestroy(&iscoltmp);CHKERRQ(ierr);}
7216     PetscFunctionReturn(0);
7217   }
7218 
7219   if (!mat->ops->getsubmatrix) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_SUP,"Mat type %s",((PetscObject)mat)->type_name);
7220   ierr = (*mat->ops->getsubmatrix)(mat,isrow,iscoltmp,cll,newmat);CHKERRQ(ierr);
7221   if (!iscol) {ierr = ISDestroy(&iscoltmp);CHKERRQ(ierr);}
7222   if (*newmat && cll == MAT_INITIAL_MATRIX) {ierr = PetscObjectStateIncrease((PetscObject)*newmat);CHKERRQ(ierr);}
7223   PetscFunctionReturn(0);
7224 }
7225 
7226 #undef __FUNCT__
7227 #define __FUNCT__ "MatStashSetInitialSize"
7228 /*@
7229    MatStashSetInitialSize - sets the sizes of the matrix stash, that is
7230    used during the assembly process to store values that belong to
7231    other processors.
7232 
7233    Not Collective
7234 
7235    Input Parameters:
7236 +  mat   - the matrix
7237 .  size  - the initial size of the stash.
7238 -  bsize - the initial size of the block-stash(if used).
7239 
7240    Options Database Keys:
7241 +   -matstash_initial_size <size> or <size0,size1,...sizep-1>
7242 -   -matstash_block_initial_size <bsize>  or <bsize0,bsize1,...bsizep-1>
7243 
7244    Level: intermediate
7245 
7246    Notes:
7247      The block-stash is used for values set with MatSetValuesBlocked() while
7248      the stash is used for values set with MatSetValues()
7249 
7250      Run with the option -info and look for output of the form
7251      MatAssemblyBegin_MPIXXX:Stash has MM entries, uses nn mallocs.
7252      to determine the appropriate value, MM, to use for size and
7253      MatAssemblyBegin_MPIXXX:Block-Stash has BMM entries, uses nn mallocs.
7254      to determine the value, BMM to use for bsize
7255 
7256    Concepts: stash^setting matrix size
7257    Concepts: matrices^stash
7258 
7259 .seealso: MatAssemblyBegin(), MatAssemblyEnd(), Mat, MatStashGetInfo()
7260 
7261 @*/
7262 PetscErrorCode  MatStashSetInitialSize(Mat mat,PetscInt size, PetscInt bsize)
7263 {
7264   PetscErrorCode ierr;
7265 
7266   PetscFunctionBegin;
7267   PetscValidHeaderSpecific(mat,MAT_CLASSID,1);
7268   PetscValidType(mat,1);
7269   ierr = MatStashSetInitialSize_Private(&mat->stash,size);CHKERRQ(ierr);
7270   ierr = MatStashSetInitialSize_Private(&mat->bstash,bsize);CHKERRQ(ierr);
7271   PetscFunctionReturn(0);
7272 }
7273 
7274 #undef __FUNCT__
7275 #define __FUNCT__ "MatInterpolateAdd"
7276 /*@
7277    MatInterpolateAdd - w = y + A*x or A'*x depending on the shape of
7278      the matrix
7279 
7280    Neighbor-wise Collective on Mat
7281 
7282    Input Parameters:
7283 +  mat   - the matrix
7284 .  x,y - the vectors
7285 -  w - where the result is stored
7286 
7287    Level: intermediate
7288 
7289    Notes:
7290     w may be the same vector as y.
7291 
7292     This allows one to use either the restriction or interpolation (its transpose)
7293     matrix to do the interpolation
7294 
7295     Concepts: interpolation
7296 
7297 .seealso: MatMultAdd(), MatMultTransposeAdd(), MatRestrict()
7298 
7299 @*/
7300 PetscErrorCode  MatInterpolateAdd(Mat A,Vec x,Vec y,Vec w)
7301 {
7302   PetscErrorCode ierr;
7303   PetscInt       M,N,Ny;
7304 
7305   PetscFunctionBegin;
7306   PetscValidHeaderSpecific(A,MAT_CLASSID,1);
7307   PetscValidHeaderSpecific(x,VEC_CLASSID,2);
7308   PetscValidHeaderSpecific(y,VEC_CLASSID,3);
7309   PetscValidHeaderSpecific(w,VEC_CLASSID,4);
7310   PetscValidType(A,1);
7311   MatCheckPreallocated(A,1);
7312   ierr = MatGetSize(A,&M,&N);CHKERRQ(ierr);
7313   ierr = VecGetSize(y,&Ny);CHKERRQ(ierr);
7314   if (M == Ny) {
7315     ierr = MatMultAdd(A,x,y,w);CHKERRQ(ierr);
7316   } else {
7317     ierr = MatMultTransposeAdd(A,x,y,w);CHKERRQ(ierr);
7318   }
7319   PetscFunctionReturn(0);
7320 }
7321 
7322 #undef __FUNCT__
7323 #define __FUNCT__ "MatInterpolate"
7324 /*@
7325    MatInterpolate - y = A*x or A'*x depending on the shape of
7326      the matrix
7327 
7328    Neighbor-wise Collective on Mat
7329 
7330    Input Parameters:
7331 +  mat   - the matrix
7332 -  x,y - the vectors
7333 
7334    Level: intermediate
7335 
7336    Notes:
7337     This allows one to use either the restriction or interpolation (its transpose)
7338     matrix to do the interpolation
7339 
7340    Concepts: matrices^interpolation
7341 
7342 .seealso: MatMultAdd(), MatMultTransposeAdd(), MatRestrict()
7343 
7344 @*/
7345 PetscErrorCode  MatInterpolate(Mat A,Vec x,Vec y)
7346 {
7347   PetscErrorCode ierr;
7348   PetscInt       M,N,Ny;
7349 
7350   PetscFunctionBegin;
7351   PetscValidHeaderSpecific(A,MAT_CLASSID,1);
7352   PetscValidHeaderSpecific(x,VEC_CLASSID,2);
7353   PetscValidHeaderSpecific(y,VEC_CLASSID,3);
7354   PetscValidType(A,1);
7355   MatCheckPreallocated(A,1);
7356   ierr = MatGetSize(A,&M,&N);CHKERRQ(ierr);
7357   ierr = VecGetSize(y,&Ny);CHKERRQ(ierr);
7358   if (M == Ny) {
7359     ierr = MatMult(A,x,y);CHKERRQ(ierr);
7360   } else {
7361     ierr = MatMultTranspose(A,x,y);CHKERRQ(ierr);
7362   }
7363   PetscFunctionReturn(0);
7364 }
7365 
7366 #undef __FUNCT__
7367 #define __FUNCT__ "MatRestrict"
7368 /*@
7369    MatRestrict - y = A*x or A'*x
7370 
7371    Neighbor-wise Collective on Mat
7372 
7373    Input Parameters:
7374 +  mat   - the matrix
7375 -  x,y - the vectors
7376 
7377    Level: intermediate
7378 
7379    Notes:
7380     This allows one to use either the restriction or interpolation (its transpose)
7381     matrix to do the restriction
7382 
7383    Concepts: matrices^restriction
7384 
7385 .seealso: MatMultAdd(), MatMultTransposeAdd(), MatInterpolate()
7386 
7387 @*/
7388 PetscErrorCode  MatRestrict(Mat A,Vec x,Vec y)
7389 {
7390   PetscErrorCode ierr;
7391   PetscInt       M,N,Ny;
7392 
7393   PetscFunctionBegin;
7394   PetscValidHeaderSpecific(A,MAT_CLASSID,1);
7395   PetscValidHeaderSpecific(x,VEC_CLASSID,2);
7396   PetscValidHeaderSpecific(y,VEC_CLASSID,3);
7397   PetscValidType(A,1);
7398   MatCheckPreallocated(A,1);
7399 
7400   ierr = MatGetSize(A,&M,&N);CHKERRQ(ierr);
7401   ierr = VecGetSize(y,&Ny);CHKERRQ(ierr);
7402   if (M == Ny) {
7403     ierr = MatMult(A,x,y);CHKERRQ(ierr);
7404   } else {
7405     ierr = MatMultTranspose(A,x,y);CHKERRQ(ierr);
7406   }
7407   PetscFunctionReturn(0);
7408 }
7409 
7410 #undef __FUNCT__
7411 #define __FUNCT__ "MatGetNullSpace"
7412 /*@
7413    MatGetNullSpace - retrieves the null space to a matrix.
7414 
7415    Logically Collective on Mat and MatNullSpace
7416 
7417    Input Parameters:
7418 +  mat - the matrix
7419 -  nullsp - the null space object
7420 
7421    Level: developer
7422 
7423    Notes:
7424       This null space is used by solvers. Overwrites any previous null space that may have been attached
7425 
7426    Concepts: null space^attaching to matrix
7427 
7428 .seealso: MatCreate(), MatNullSpaceCreate(), MatSetNearNullSpace()
7429 @*/
7430 PetscErrorCode MatGetNullSpace(Mat mat, MatNullSpace *nullsp)
7431 {
7432   PetscFunctionBegin;
7433   PetscValidHeaderSpecific(mat,MAT_CLASSID,1);
7434   PetscValidType(mat,1);
7435   PetscValidPointer(nullsp,2);
7436   *nullsp = mat->nullsp;
7437   PetscFunctionReturn(0);
7438 }
7439 
7440 #undef __FUNCT__
7441 #define __FUNCT__ "MatSetNullSpace"
7442 /*@
7443    MatSetNullSpace - attaches a null space to a matrix.
7444         This null space will be removed from the resulting vector whenever
7445         MatMult() is called
7446 
7447    Logically Collective on Mat and MatNullSpace
7448 
7449    Input Parameters:
7450 +  mat - the matrix
7451 -  nullsp - the null space object
7452 
7453    Level: advanced
7454 
7455    Notes:
7456       This null space is used by solvers. Overwrites any previous null space that may have been attached
7457 
7458    Concepts: null space^attaching to matrix
7459 
7460 .seealso: MatCreate(), MatNullSpaceCreate(), MatSetNearNullSpace()
7461 @*/
7462 PetscErrorCode  MatSetNullSpace(Mat mat,MatNullSpace nullsp)
7463 {
7464   PetscErrorCode ierr;
7465 
7466   PetscFunctionBegin;
7467   PetscValidHeaderSpecific(mat,MAT_CLASSID,1);
7468   PetscValidType(mat,1);
7469   PetscValidHeaderSpecific(nullsp,MAT_NULLSPACE_CLASSID,2);
7470   MatCheckPreallocated(mat,1);
7471   ierr = PetscObjectReference((PetscObject)nullsp);CHKERRQ(ierr);
7472   ierr = MatNullSpaceDestroy(&mat->nullsp);CHKERRQ(ierr);
7473 
7474   mat->nullsp = nullsp;
7475   PetscFunctionReturn(0);
7476 }
7477 
7478 #undef __FUNCT__
7479 #define __FUNCT__ "MatSetNearNullSpace"
7480 /*@
7481    MatSetNearNullSpace - attaches a null space to a matrix.
7482         This null space will be used to provide near null space vectors to a multigrid preconditioner built from this matrix.
7483 
7484    Logically Collective on Mat and MatNullSpace
7485 
7486    Input Parameters:
7487 +  mat - the matrix
7488 -  nullsp - the null space object
7489 
7490    Level: advanced
7491 
7492    Notes:
7493       Overwrites any previous near null space that may have been attached
7494 
7495    Concepts: null space^attaching to matrix
7496 
7497 .seealso: MatCreate(), MatNullSpaceCreate(), MatSetNullSpace()
7498 @*/
7499 PetscErrorCode MatSetNearNullSpace(Mat mat,MatNullSpace nullsp)
7500 {
7501   PetscErrorCode ierr;
7502 
7503   PetscFunctionBegin;
7504   PetscValidHeaderSpecific(mat,MAT_CLASSID,1);
7505   PetscValidType(mat,1);
7506   PetscValidHeaderSpecific(nullsp,MAT_NULLSPACE_CLASSID,2);
7507   MatCheckPreallocated(mat,1);
7508   ierr = PetscObjectReference((PetscObject)nullsp);CHKERRQ(ierr);
7509   ierr = MatNullSpaceDestroy(&mat->nearnullsp);CHKERRQ(ierr);
7510 
7511   mat->nearnullsp = nullsp;
7512   PetscFunctionReturn(0);
7513 }
7514 
7515 #undef __FUNCT__
7516 #define __FUNCT__ "MatGetNearNullSpace"
7517 /*@
7518    MatGetNearNullSpace -Get null space attached with MatSetNearNullSpace()
7519 
7520    Not Collective
7521 
7522    Input Parameters:
7523 .  mat - the matrix
7524 
7525    Output Parameters:
7526 .  nullsp - the null space object, NULL if not set
7527 
7528    Level: developer
7529 
7530    Concepts: null space^attaching to matrix
7531 
7532 .seealso: MatSetNearNullSpace(), MatGetNullSpace()
7533 @*/
7534 PetscErrorCode MatGetNearNullSpace(Mat mat,MatNullSpace *nullsp)
7535 {
7536   PetscFunctionBegin;
7537   PetscValidHeaderSpecific(mat,MAT_CLASSID,1);
7538   PetscValidType(mat,1);
7539   PetscValidPointer(nullsp,2);
7540   MatCheckPreallocated(mat,1);
7541   *nullsp = mat->nearnullsp;
7542   PetscFunctionReturn(0);
7543 }
7544 
7545 #undef __FUNCT__
7546 #define __FUNCT__ "MatICCFactor"
7547 /*@C
7548    MatICCFactor - Performs in-place incomplete Cholesky factorization of matrix.
7549 
7550    Collective on Mat
7551 
7552    Input Parameters:
7553 +  mat - the matrix
7554 .  row - row/column permutation
7555 .  fill - expected fill factor >= 1.0
7556 -  level - level of fill, for ICC(k)
7557 
7558    Notes:
7559    Probably really in-place only when level of fill is zero, otherwise allocates
7560    new space to store factored matrix and deletes previous memory.
7561 
7562    Most users should employ the simplified KSP interface for linear solvers
7563    instead of working directly with matrix algebra routines such as this.
7564    See, e.g., KSPCreate().
7565 
7566    Level: developer
7567 
7568    Concepts: matrices^incomplete Cholesky factorization
7569    Concepts: Cholesky factorization
7570 
7571 .seealso: MatICCFactorSymbolic(), MatLUFactorNumeric(), MatCholeskyFactor()
7572 
7573     Developer Note: fortran interface is not autogenerated as the f90
7574     interface defintion cannot be generated correctly [due to MatFactorInfo]
7575 
7576 @*/
7577 PetscErrorCode  MatICCFactor(Mat mat,IS row,const MatFactorInfo *info)
7578 {
7579   PetscErrorCode ierr;
7580 
7581   PetscFunctionBegin;
7582   PetscValidHeaderSpecific(mat,MAT_CLASSID,1);
7583   PetscValidType(mat,1);
7584   if (row) PetscValidHeaderSpecific(row,IS_CLASSID,2);
7585   PetscValidPointer(info,3);
7586   if (mat->rmap->N != mat->cmap->N) SETERRQ(PetscObjectComm((PetscObject)mat),PETSC_ERR_ARG_WRONG,"matrix must be square");
7587   if (!mat->assembled) SETERRQ(PetscObjectComm((PetscObject)mat),PETSC_ERR_ARG_WRONGSTATE,"Not for unassembled matrix");
7588   if (mat->factortype) SETERRQ(PetscObjectComm((PetscObject)mat),PETSC_ERR_ARG_WRONGSTATE,"Not for factored matrix");
7589   if (!mat->ops->iccfactor) SETERRQ1(PetscObjectComm((PetscObject)mat),PETSC_ERR_SUP,"Mat type %s",((PetscObject)mat)->type_name);
7590   MatCheckPreallocated(mat,1);
7591   ierr = (*mat->ops->iccfactor)(mat,row,info);CHKERRQ(ierr);
7592   ierr = PetscObjectStateIncrease((PetscObject)mat);CHKERRQ(ierr);
7593   PetscFunctionReturn(0);
7594 }
7595 
7596 #undef __FUNCT__
7597 #define __FUNCT__ "MatSetValuesAdifor"
7598 /*@
7599    MatSetValuesAdifor - Sets values computed with automatic differentiation into a matrix.
7600 
7601    Not Collective
7602 
7603    Input Parameters:
7604 +  mat - the matrix
7605 .  nl - leading dimension of v
7606 -  v - the values compute with ADIFOR
7607 
7608    Level: developer
7609 
7610    Notes:
7611      Must call MatSetColoring() before using this routine. Also this matrix must already
7612      have its nonzero pattern determined.
7613 
7614 .seealso: MatSetOption(), MatAssemblyBegin(), MatAssemblyEnd(), MatSetValuesBlocked(), MatSetValuesLocal(),
7615           MatSetValues(), MatSetColoring()
7616 @*/
7617 PetscErrorCode  MatSetValuesAdifor(Mat mat,PetscInt nl,void *v)
7618 {
7619   PetscErrorCode ierr;
7620 
7621   PetscFunctionBegin;
7622   PetscValidHeaderSpecific(mat,MAT_CLASSID,1);
7623   PetscValidType(mat,1);
7624   PetscValidPointer(v,3);
7625 
7626   if (!mat->assembled) SETERRQ(PetscObjectComm((PetscObject)mat),PETSC_ERR_ARG_WRONGSTATE,"Matrix must be already assembled");
7627   ierr = PetscLogEventBegin(MAT_SetValues,mat,0,0,0);CHKERRQ(ierr);
7628   if (!mat->ops->setvaluesadifor) SETERRQ1(PetscObjectComm((PetscObject)mat),PETSC_ERR_SUP,"Mat type %s",((PetscObject)mat)->type_name);
7629   ierr = (*mat->ops->setvaluesadifor)(mat,nl,v);CHKERRQ(ierr);
7630   ierr = PetscLogEventEnd(MAT_SetValues,mat,0,0,0);CHKERRQ(ierr);
7631   ierr = PetscObjectStateIncrease((PetscObject)mat);CHKERRQ(ierr);
7632   PetscFunctionReturn(0);
7633 }
7634 
7635 #undef __FUNCT__
7636 #define __FUNCT__ "MatDiagonalScaleLocal"
7637 /*@
7638    MatDiagonalScaleLocal - Scales columns of a matrix given the scaling values including the
7639          ghosted ones.
7640 
7641    Not Collective
7642 
7643    Input Parameters:
7644 +  mat - the matrix
7645 -  diag = the diagonal values, including ghost ones
7646 
7647    Level: developer
7648 
7649    Notes: Works only for MPIAIJ and MPIBAIJ matrices
7650 
7651 .seealso: MatDiagonalScale()
7652 @*/
7653 PetscErrorCode  MatDiagonalScaleLocal(Mat mat,Vec diag)
7654 {
7655   PetscErrorCode ierr;
7656   PetscMPIInt    size;
7657 
7658   PetscFunctionBegin;
7659   PetscValidHeaderSpecific(mat,MAT_CLASSID,1);
7660   PetscValidHeaderSpecific(diag,VEC_CLASSID,2);
7661   PetscValidType(mat,1);
7662 
7663   if (!mat->assembled) SETERRQ(PetscObjectComm((PetscObject)mat),PETSC_ERR_ARG_WRONGSTATE,"Matrix must be already assembled");
7664   ierr = PetscLogEventBegin(MAT_Scale,mat,0,0,0);CHKERRQ(ierr);
7665   ierr = MPI_Comm_size(PetscObjectComm((PetscObject)mat),&size);CHKERRQ(ierr);
7666   if (size == 1) {
7667     PetscInt n,m;
7668     ierr = VecGetSize(diag,&n);CHKERRQ(ierr);
7669     ierr = MatGetSize(mat,0,&m);CHKERRQ(ierr);
7670     if (m == n) {
7671       ierr = MatDiagonalScale(mat,0,diag);CHKERRQ(ierr);
7672     } else SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"Only supported for sequential matrices when no ghost points/periodic conditions");
7673   } else {
7674     ierr = PetscUseMethod(mat,"MatDiagonalScaleLocal_C",(Mat,Vec),(mat,diag));CHKERRQ(ierr);
7675   }
7676   ierr = PetscLogEventEnd(MAT_Scale,mat,0,0,0);CHKERRQ(ierr);
7677   ierr = PetscObjectStateIncrease((PetscObject)mat);CHKERRQ(ierr);
7678   PetscFunctionReturn(0);
7679 }
7680 
7681 #undef __FUNCT__
7682 #define __FUNCT__ "MatGetInertia"
7683 /*@
7684    MatGetInertia - Gets the inertia from a factored matrix
7685 
7686    Collective on Mat
7687 
7688    Input Parameter:
7689 .  mat - the matrix
7690 
7691    Output Parameters:
7692 +   nneg - number of negative eigenvalues
7693 .   nzero - number of zero eigenvalues
7694 -   npos - number of positive eigenvalues
7695 
7696    Level: advanced
7697 
7698    Notes: Matrix must have been factored by MatCholeskyFactor()
7699 
7700 
7701 @*/
7702 PetscErrorCode  MatGetInertia(Mat mat,PetscInt *nneg,PetscInt *nzero,PetscInt *npos)
7703 {
7704   PetscErrorCode ierr;
7705 
7706   PetscFunctionBegin;
7707   PetscValidHeaderSpecific(mat,MAT_CLASSID,1);
7708   PetscValidType(mat,1);
7709   if (!mat->factortype) SETERRQ(PetscObjectComm((PetscObject)mat),PETSC_ERR_ARG_WRONGSTATE,"Unfactored matrix");
7710   if (!mat->assembled) SETERRQ(PetscObjectComm((PetscObject)mat),PETSC_ERR_ARG_WRONGSTATE,"Numeric factor mat is not assembled");
7711   if (!mat->ops->getinertia) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_SUP,"Mat type %s",((PetscObject)mat)->type_name);
7712   ierr = (*mat->ops->getinertia)(mat,nneg,nzero,npos);CHKERRQ(ierr);
7713   PetscFunctionReturn(0);
7714 }
7715 
7716 /* ----------------------------------------------------------------*/
7717 #undef __FUNCT__
7718 #define __FUNCT__ "MatSolves"
7719 /*@C
7720    MatSolves - Solves A x = b, given a factored matrix, for a collection of vectors
7721 
7722    Neighbor-wise Collective on Mat and Vecs
7723 
7724    Input Parameters:
7725 +  mat - the factored matrix
7726 -  b - the right-hand-side vectors
7727 
7728    Output Parameter:
7729 .  x - the result vectors
7730 
7731    Notes:
7732    The vectors b and x cannot be the same.  I.e., one cannot
7733    call MatSolves(A,x,x).
7734 
7735    Notes:
7736    Most users should employ the simplified KSP interface for linear solvers
7737    instead of working directly with matrix algebra routines such as this.
7738    See, e.g., KSPCreate().
7739 
7740    Level: developer
7741 
7742    Concepts: matrices^triangular solves
7743 
7744 .seealso: MatSolveAdd(), MatSolveTranspose(), MatSolveTransposeAdd(), MatSolve()
7745 @*/
7746 PetscErrorCode  MatSolves(Mat mat,Vecs b,Vecs x)
7747 {
7748   PetscErrorCode ierr;
7749 
7750   PetscFunctionBegin;
7751   PetscValidHeaderSpecific(mat,MAT_CLASSID,1);
7752   PetscValidType(mat,1);
7753   if (x == b) SETERRQ(PetscObjectComm((PetscObject)mat),PETSC_ERR_ARG_IDN,"x and b must be different vectors");
7754   if (!mat->factortype) SETERRQ(PetscObjectComm((PetscObject)mat),PETSC_ERR_ARG_WRONGSTATE,"Unfactored matrix");
7755   if (!mat->rmap->N && !mat->cmap->N) PetscFunctionReturn(0);
7756 
7757   if (!mat->ops->solves) SETERRQ1(PetscObjectComm((PetscObject)mat),PETSC_ERR_SUP,"Mat type %s",((PetscObject)mat)->type_name);
7758   MatCheckPreallocated(mat,1);
7759   ierr = PetscLogEventBegin(MAT_Solves,mat,0,0,0);CHKERRQ(ierr);
7760   ierr = (*mat->ops->solves)(mat,b,x);CHKERRQ(ierr);
7761   ierr = PetscLogEventEnd(MAT_Solves,mat,0,0,0);CHKERRQ(ierr);
7762   PetscFunctionReturn(0);
7763 }
7764 
7765 #undef __FUNCT__
7766 #define __FUNCT__ "MatIsSymmetric"
7767 /*@
7768    MatIsSymmetric - Test whether a matrix is symmetric
7769 
7770    Collective on Mat
7771 
7772    Input Parameter:
7773 +  A - the matrix to test
7774 -  tol - difference between value and its transpose less than this amount counts as equal (use 0.0 for exact transpose)
7775 
7776    Output Parameters:
7777 .  flg - the result
7778 
7779    Notes: For real numbers MatIsSymmetric() and MatIsHermitian() return identical results
7780 
7781    Level: intermediate
7782 
7783    Concepts: matrix^symmetry
7784 
7785 .seealso: MatTranspose(), MatIsTranspose(), MatIsHermitian(), MatIsStructurallySymmetric(), MatSetOption(), MatIsSymmetricKnown()
7786 @*/
7787 PetscErrorCode  MatIsSymmetric(Mat A,PetscReal tol,PetscBool  *flg)
7788 {
7789   PetscErrorCode ierr;
7790 
7791   PetscFunctionBegin;
7792   PetscValidHeaderSpecific(A,MAT_CLASSID,1);
7793   PetscValidPointer(flg,2);
7794 
7795   if (!A->symmetric_set) {
7796     if (!A->ops->issymmetric) {
7797       MatType mattype;
7798       ierr = MatGetType(A,&mattype);CHKERRQ(ierr);
7799       SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_SUP,"Matrix of type <%s> does not support checking for symmetric",mattype);
7800     }
7801     ierr = (*A->ops->issymmetric)(A,tol,flg);CHKERRQ(ierr);
7802     if (!tol) {
7803       A->symmetric_set = PETSC_TRUE;
7804       A->symmetric     = *flg;
7805       if (A->symmetric) {
7806         A->structurally_symmetric_set = PETSC_TRUE;
7807         A->structurally_symmetric     = PETSC_TRUE;
7808       }
7809     }
7810   } else if (A->symmetric) {
7811     *flg = PETSC_TRUE;
7812   } else if (!tol) {
7813     *flg = PETSC_FALSE;
7814   } else {
7815     if (!A->ops->issymmetric) {
7816       MatType mattype;
7817       ierr = MatGetType(A,&mattype);CHKERRQ(ierr);
7818       SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_SUP,"Matrix of type <%s> does not support checking for symmetric",mattype);
7819     }
7820     ierr = (*A->ops->issymmetric)(A,tol,flg);CHKERRQ(ierr);
7821   }
7822   PetscFunctionReturn(0);
7823 }
7824 
7825 #undef __FUNCT__
7826 #define __FUNCT__ "MatIsHermitian"
7827 /*@
7828    MatIsHermitian - Test whether a matrix is Hermitian
7829 
7830    Collective on Mat
7831 
7832    Input Parameter:
7833 +  A - the matrix to test
7834 -  tol - difference between value and its transpose less than this amount counts as equal (use 0.0 for exact Hermitian)
7835 
7836    Output Parameters:
7837 .  flg - the result
7838 
7839    Level: intermediate
7840 
7841    Concepts: matrix^symmetry
7842 
7843 .seealso: MatTranspose(), MatIsTranspose(), MatIsHermitian(), MatIsStructurallySymmetric(), MatSetOption(),
7844           MatIsSymmetricKnown(), MatIsSymmetric()
7845 @*/
7846 PetscErrorCode  MatIsHermitian(Mat A,PetscReal tol,PetscBool  *flg)
7847 {
7848   PetscErrorCode ierr;
7849 
7850   PetscFunctionBegin;
7851   PetscValidHeaderSpecific(A,MAT_CLASSID,1);
7852   PetscValidPointer(flg,2);
7853 
7854   if (!A->hermitian_set) {
7855     if (!A->ops->ishermitian) {
7856       MatType mattype;
7857       ierr = MatGetType(A,&mattype);CHKERRQ(ierr);
7858       SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_SUP,"Matrix of type <%s> does not support checking for hermitian",mattype);
7859     }
7860     ierr = (*A->ops->ishermitian)(A,tol,flg);CHKERRQ(ierr);
7861     if (!tol) {
7862       A->hermitian_set = PETSC_TRUE;
7863       A->hermitian     = *flg;
7864       if (A->hermitian) {
7865         A->structurally_symmetric_set = PETSC_TRUE;
7866         A->structurally_symmetric     = PETSC_TRUE;
7867       }
7868     }
7869   } else if (A->hermitian) {
7870     *flg = PETSC_TRUE;
7871   } else if (!tol) {
7872     *flg = PETSC_FALSE;
7873   } else {
7874     if (!A->ops->ishermitian) {
7875       MatType mattype;
7876       ierr = MatGetType(A,&mattype);CHKERRQ(ierr);
7877       SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_SUP,"Matrix of type <%s> does not support checking for hermitian",mattype);
7878     }
7879     ierr = (*A->ops->ishermitian)(A,tol,flg);CHKERRQ(ierr);
7880   }
7881   PetscFunctionReturn(0);
7882 }
7883 
7884 #undef __FUNCT__
7885 #define __FUNCT__ "MatIsSymmetricKnown"
7886 /*@
7887    MatIsSymmetricKnown - Checks the flag on the matrix to see if it is symmetric.
7888 
7889    Not Collective
7890 
7891    Input Parameter:
7892 .  A - the matrix to check
7893 
7894    Output Parameters:
7895 +  set - if the symmetric flag is set (this tells you if the next flag is valid)
7896 -  flg - the result
7897 
7898    Level: advanced
7899 
7900    Concepts: matrix^symmetry
7901 
7902    Note: Does not check the matrix values directly, so this may return unknown (set = PETSC_FALSE). Use MatIsSymmetric()
7903          if you want it explicitly checked
7904 
7905 .seealso: MatTranspose(), MatIsTranspose(), MatIsHermitian(), MatIsStructurallySymmetric(), MatSetOption(), MatIsSymmetric()
7906 @*/
7907 PetscErrorCode  MatIsSymmetricKnown(Mat A,PetscBool  *set,PetscBool  *flg)
7908 {
7909   PetscFunctionBegin;
7910   PetscValidHeaderSpecific(A,MAT_CLASSID,1);
7911   PetscValidPointer(set,2);
7912   PetscValidPointer(flg,3);
7913   if (A->symmetric_set) {
7914     *set = PETSC_TRUE;
7915     *flg = A->symmetric;
7916   } else {
7917     *set = PETSC_FALSE;
7918   }
7919   PetscFunctionReturn(0);
7920 }
7921 
7922 #undef __FUNCT__
7923 #define __FUNCT__ "MatIsHermitianKnown"
7924 /*@
7925    MatIsHermitianKnown - Checks the flag on the matrix to see if it is hermitian.
7926 
7927    Not Collective
7928 
7929    Input Parameter:
7930 .  A - the matrix to check
7931 
7932    Output Parameters:
7933 +  set - if the hermitian flag is set (this tells you if the next flag is valid)
7934 -  flg - the result
7935 
7936    Level: advanced
7937 
7938    Concepts: matrix^symmetry
7939 
7940    Note: Does not check the matrix values directly, so this may return unknown (set = PETSC_FALSE). Use MatIsHermitian()
7941          if you want it explicitly checked
7942 
7943 .seealso: MatTranspose(), MatIsTranspose(), MatIsHermitian(), MatIsStructurallySymmetric(), MatSetOption(), MatIsSymmetric()
7944 @*/
7945 PetscErrorCode  MatIsHermitianKnown(Mat A,PetscBool  *set,PetscBool  *flg)
7946 {
7947   PetscFunctionBegin;
7948   PetscValidHeaderSpecific(A,MAT_CLASSID,1);
7949   PetscValidPointer(set,2);
7950   PetscValidPointer(flg,3);
7951   if (A->hermitian_set) {
7952     *set = PETSC_TRUE;
7953     *flg = A->hermitian;
7954   } else {
7955     *set = PETSC_FALSE;
7956   }
7957   PetscFunctionReturn(0);
7958 }
7959 
7960 #undef __FUNCT__
7961 #define __FUNCT__ "MatIsStructurallySymmetric"
7962 /*@
7963    MatIsStructurallySymmetric - Test whether a matrix is structurally symmetric
7964 
7965    Collective on Mat
7966 
7967    Input Parameter:
7968 .  A - the matrix to test
7969 
7970    Output Parameters:
7971 .  flg - the result
7972 
7973    Level: intermediate
7974 
7975    Concepts: matrix^symmetry
7976 
7977 .seealso: MatTranspose(), MatIsTranspose(), MatIsHermitian(), MatIsSymmetric(), MatSetOption()
7978 @*/
7979 PetscErrorCode  MatIsStructurallySymmetric(Mat A,PetscBool  *flg)
7980 {
7981   PetscErrorCode ierr;
7982 
7983   PetscFunctionBegin;
7984   PetscValidHeaderSpecific(A,MAT_CLASSID,1);
7985   PetscValidPointer(flg,2);
7986   if (!A->structurally_symmetric_set) {
7987     if (!A->ops->isstructurallysymmetric) SETERRQ(PetscObjectComm((PetscObject)A),PETSC_ERR_SUP,"Matrix does not support checking for structural symmetric");
7988     ierr = (*A->ops->isstructurallysymmetric)(A,&A->structurally_symmetric);CHKERRQ(ierr);
7989 
7990     A->structurally_symmetric_set = PETSC_TRUE;
7991   }
7992   *flg = A->structurally_symmetric;
7993   PetscFunctionReturn(0);
7994 }
7995 
7996 #undef __FUNCT__
7997 #define __FUNCT__ "MatStashGetInfo"
7998 extern PetscErrorCode MatStashGetInfo_Private(MatStash*,PetscInt*,PetscInt*);
7999 /*@
8000    MatStashGetInfo - Gets how many values are currently in the matrix stash, i.e. need
8001        to be communicated to other processors during the MatAssemblyBegin/End() process
8002 
8003     Not collective
8004 
8005    Input Parameter:
8006 .   vec - the vector
8007 
8008    Output Parameters:
8009 +   nstash   - the size of the stash
8010 .   reallocs - the number of additional mallocs incurred.
8011 .   bnstash   - the size of the block stash
8012 -   breallocs - the number of additional mallocs incurred.in the block stash
8013 
8014    Level: advanced
8015 
8016 .seealso: MatAssemblyBegin(), MatAssemblyEnd(), Mat, MatStashSetInitialSize()
8017 
8018 @*/
8019 PetscErrorCode  MatStashGetInfo(Mat mat,PetscInt *nstash,PetscInt *reallocs,PetscInt *bnstash,PetscInt *breallocs)
8020 {
8021   PetscErrorCode ierr;
8022 
8023   PetscFunctionBegin;
8024   ierr = MatStashGetInfo_Private(&mat->stash,nstash,reallocs);CHKERRQ(ierr);
8025   ierr = MatStashGetInfo_Private(&mat->bstash,bnstash,breallocs);CHKERRQ(ierr);
8026   PetscFunctionReturn(0);
8027 }
8028 
8029 #undef __FUNCT__
8030 #define __FUNCT__ "MatGetVecs"
8031 /*@C
8032    MatGetVecs - Get vector(s) compatible with the matrix, i.e. with the same
8033      parallel layout
8034 
8035    Collective on Mat
8036 
8037    Input Parameter:
8038 .  mat - the matrix
8039 
8040    Output Parameter:
8041 +   right - (optional) vector that the matrix can be multiplied against
8042 -   left - (optional) vector that the matrix vector product can be stored in
8043 
8044   Level: advanced
8045 
8046 .seealso: MatCreate()
8047 @*/
8048 PetscErrorCode  MatGetVecs(Mat mat,Vec *right,Vec *left)
8049 {
8050   PetscErrorCode ierr;
8051 
8052   PetscFunctionBegin;
8053   PetscValidHeaderSpecific(mat,MAT_CLASSID,1);
8054   PetscValidType(mat,1);
8055   MatCheckPreallocated(mat,1);
8056   if (mat->ops->getvecs) {
8057     ierr = (*mat->ops->getvecs)(mat,right,left);CHKERRQ(ierr);
8058   } else {
8059     PetscMPIInt size;
8060     ierr = MPI_Comm_size(PetscObjectComm((PetscObject)mat), &size);CHKERRQ(ierr);
8061     if (right) {
8062       ierr = VecCreate(PetscObjectComm((PetscObject)mat),right);CHKERRQ(ierr);
8063       ierr = VecSetSizes(*right,mat->cmap->n,PETSC_DETERMINE);CHKERRQ(ierr);
8064       ierr = VecSetBlockSize(*right,mat->rmap->bs);CHKERRQ(ierr);
8065       ierr = VecSetType(*right,VECSTANDARD);CHKERRQ(ierr);
8066       ierr = PetscLayoutReference(mat->cmap,&(*right)->map);CHKERRQ(ierr);
8067     }
8068     if (left) {
8069       ierr = VecCreate(PetscObjectComm((PetscObject)mat),left);CHKERRQ(ierr);
8070       ierr = VecSetSizes(*left,mat->rmap->n,PETSC_DETERMINE);CHKERRQ(ierr);
8071       ierr = VecSetBlockSize(*left,mat->rmap->bs);CHKERRQ(ierr);
8072       ierr = VecSetType(*left,VECSTANDARD);CHKERRQ(ierr);
8073       ierr = PetscLayoutReference(mat->rmap,&(*left)->map);CHKERRQ(ierr);
8074     }
8075   }
8076   PetscFunctionReturn(0);
8077 }
8078 
8079 #undef __FUNCT__
8080 #define __FUNCT__ "MatFactorInfoInitialize"
8081 /*@C
8082    MatFactorInfoInitialize - Initializes a MatFactorInfo data structure
8083      with default values.
8084 
8085    Not Collective
8086 
8087    Input Parameters:
8088 .    info - the MatFactorInfo data structure
8089 
8090 
8091    Notes: The solvers are generally used through the KSP and PC objects, for example
8092           PCLU, PCILU, PCCHOLESKY, PCICC
8093 
8094    Level: developer
8095 
8096 .seealso: MatFactorInfo
8097 
8098     Developer Note: fortran interface is not autogenerated as the f90
8099     interface defintion cannot be generated correctly [due to MatFactorInfo]
8100 
8101 @*/
8102 
8103 PetscErrorCode  MatFactorInfoInitialize(MatFactorInfo *info)
8104 {
8105   PetscErrorCode ierr;
8106 
8107   PetscFunctionBegin;
8108   ierr = PetscMemzero(info,sizeof(MatFactorInfo));CHKERRQ(ierr);
8109   PetscFunctionReturn(0);
8110 }
8111 
8112 #undef __FUNCT__
8113 #define __FUNCT__ "MatPtAP"
8114 /*@
8115    MatPtAP - Creates the matrix product C = P^T * A * P
8116 
8117    Neighbor-wise Collective on Mat
8118 
8119    Input Parameters:
8120 +  A - the matrix
8121 .  P - the projection matrix
8122 .  scall - either MAT_INITIAL_MATRIX or MAT_REUSE_MATRIX
8123 -  fill - expected fill as ratio of nnz(C)/(nnz(A) + nnz(P))
8124 
8125    Output Parameters:
8126 .  C - the product matrix
8127 
8128    Notes:
8129    C will be created and must be destroyed by the user with MatDestroy().
8130 
8131    This routine is currently only implemented for pairs of AIJ matrices and classes
8132    which inherit from AIJ.
8133 
8134    Level: intermediate
8135 
8136 .seealso: MatPtAPSymbolic(), MatPtAPNumeric(), MatMatMult(), MatRARt()
8137 @*/
8138 PetscErrorCode  MatPtAP(Mat A,Mat P,MatReuse scall,PetscReal fill,Mat *C)
8139 {
8140   PetscErrorCode ierr;
8141   PetscErrorCode (*fA)(Mat,Mat,MatReuse,PetscReal,Mat*);
8142   PetscErrorCode (*fP)(Mat,Mat,MatReuse,PetscReal,Mat*);
8143   PetscErrorCode (*ptap)(Mat,Mat,MatReuse,PetscReal,Mat*)=NULL;
8144   PetscBool      viatranspose=PETSC_FALSE,viamatmatmatmult=PETSC_FALSE;
8145 
8146   PetscFunctionBegin;
8147   ierr = PetscOptionsGetBool(((PetscObject)A)->prefix,"-matptap_viatranspose",&viatranspose,NULL);CHKERRQ(ierr);
8148   ierr = PetscOptionsGetBool(((PetscObject)A)->prefix,"-matptap_viamatmatmatmult",&viamatmatmatmult,NULL);CHKERRQ(ierr);
8149 
8150   PetscValidHeaderSpecific(A,MAT_CLASSID,1);
8151   PetscValidType(A,1);
8152   MatCheckPreallocated(A,1);
8153   if (!A->assembled) SETERRQ(PetscObjectComm((PetscObject)A),PETSC_ERR_ARG_WRONGSTATE,"Not for unassembled matrix");
8154   if (A->factortype) SETERRQ(PetscObjectComm((PetscObject)A),PETSC_ERR_ARG_WRONGSTATE,"Not for factored matrix");
8155   PetscValidHeaderSpecific(P,MAT_CLASSID,2);
8156   PetscValidType(P,2);
8157   MatCheckPreallocated(P,2);
8158   if (!P->assembled) SETERRQ(PetscObjectComm((PetscObject)A),PETSC_ERR_ARG_WRONGSTATE,"Not for unassembled matrix");
8159   if (P->factortype) SETERRQ(PetscObjectComm((PetscObject)A),PETSC_ERR_ARG_WRONGSTATE,"Not for factored matrix");
8160 
8161   if (P->rmap->N!=A->cmap->N) SETERRQ2(PetscObjectComm((PetscObject)A),PETSC_ERR_ARG_SIZ,"Matrix dimensions are incompatible, %D != %D",P->rmap->N,A->cmap->N);
8162   if (fill < 1.0) SETERRQ1(PetscObjectComm((PetscObject)A),PETSC_ERR_ARG_SIZ,"Expected fill=%G must be >= 1.0",fill);
8163 
8164   if (scall == MAT_REUSE_MATRIX) {
8165     PetscValidPointer(*C,5);
8166     PetscValidHeaderSpecific(*C,MAT_CLASSID,5);
8167     if (viatranspose || viamatmatmatmult) {
8168       Mat Pt;
8169       ierr = MatTranspose(P,MAT_INITIAL_MATRIX,&Pt);CHKERRQ(ierr);
8170       if (viamatmatmatmult) {
8171         ierr = MatMatMatMult(Pt,A,P,scall,fill,C);CHKERRQ(ierr);
8172       } else {
8173         Mat AP;
8174         ierr = MatMatMult(A,P,MAT_INITIAL_MATRIX,fill,&AP);CHKERRQ(ierr);
8175         ierr = MatMatMult(Pt,AP,scall,fill,C);CHKERRQ(ierr);
8176         ierr = MatDestroy(&AP);CHKERRQ(ierr);
8177       }
8178       ierr = MatDestroy(&Pt);CHKERRQ(ierr);
8179     } else {
8180       ierr = PetscLogEventBegin(MAT_PtAPNumeric,A,P,0,0);CHKERRQ(ierr);
8181       ierr = (*(*C)->ops->ptapnumeric)(A,P,*C);CHKERRQ(ierr);
8182       ierr = PetscLogEventEnd(MAT_PtAPNumeric,A,P,0,0);CHKERRQ(ierr);
8183     }
8184     PetscFunctionReturn(0);
8185   }
8186 
8187   if (fill == PETSC_DEFAULT || fill == PETSC_DECIDE) fill = 2.0;
8188   if (fill < 1.0) SETERRQ1(PetscObjectComm((PetscObject)A),PETSC_ERR_ARG_SIZ,"Expected fill=%G must be >= 1.0",fill);
8189 
8190   fA = A->ops->ptap;
8191   fP = P->ops->ptap;
8192   if (fP == fA) {
8193     if (!fA) SETERRQ1(PetscObjectComm((PetscObject)A),PETSC_ERR_SUP,"MatPtAP not supported for A of type %s",((PetscObject)A)->type_name);
8194     ptap = fA;
8195   } else {
8196     /* dispatch based on the type of A and P from their PetscObject's PetscFunctionLists. */
8197     char ptapname[256];
8198     ierr = PetscStrcpy(ptapname,"MatPtAP_");CHKERRQ(ierr);
8199     ierr = PetscStrcat(ptapname,((PetscObject)A)->type_name);CHKERRQ(ierr);
8200     ierr = PetscStrcat(ptapname,"_");CHKERRQ(ierr);
8201     ierr = PetscStrcat(ptapname,((PetscObject)P)->type_name);CHKERRQ(ierr);
8202     ierr = PetscStrcat(ptapname,"_C");CHKERRQ(ierr); /* e.g., ptapname = "MatPtAP_seqdense_seqaij_C" */
8203     ierr = PetscObjectQueryFunction((PetscObject)P,ptapname,&ptap);CHKERRQ(ierr);
8204     if (!ptap) SETERRQ2(PetscObjectComm((PetscObject)A),PETSC_ERR_ARG_INCOMP,"MatPtAP requires A, %s, to be compatible with P, %s",((PetscObject)A)->type_name,((PetscObject)P)->type_name);
8205   }
8206 
8207   if (viatranspose || viamatmatmatmult) {
8208     Mat Pt;
8209     ierr = MatTranspose(P,MAT_INITIAL_MATRIX,&Pt);CHKERRQ(ierr);
8210     if (viamatmatmatmult) {
8211       ierr = MatMatMatMult(Pt,A,P,scall,fill,C);CHKERRQ(ierr);
8212       ierr = PetscInfo(*C,"MatPtAP via MatMatMatMult\n");CHKERRQ(ierr);
8213     } else {
8214       Mat AP;
8215       ierr = MatMatMult(A,P,MAT_INITIAL_MATRIX,fill,&AP);CHKERRQ(ierr);
8216       ierr = MatMatMult(Pt,AP,scall,fill,C);CHKERRQ(ierr);
8217       ierr = MatDestroy(&AP);CHKERRQ(ierr);
8218       ierr = PetscInfo(*C,"MatPtAP via MatTranspose and MatMatMult\n");CHKERRQ(ierr);
8219     }
8220     ierr = MatDestroy(&Pt);CHKERRQ(ierr);
8221   } else {
8222     ierr = PetscLogEventBegin(MAT_PtAP,A,P,0,0);CHKERRQ(ierr);
8223     ierr = (*ptap)(A,P,scall,fill,C);CHKERRQ(ierr);
8224     ierr = PetscLogEventEnd(MAT_PtAP,A,P,0,0);CHKERRQ(ierr);
8225   }
8226   PetscFunctionReturn(0);
8227 }
8228 
8229 #undef __FUNCT__
8230 #define __FUNCT__ "MatPtAPNumeric"
8231 /*@
8232    MatPtAPNumeric - Computes the matrix product C = P^T * A * P
8233 
8234    Neighbor-wise Collective on Mat
8235 
8236    Input Parameters:
8237 +  A - the matrix
8238 -  P - the projection matrix
8239 
8240    Output Parameters:
8241 .  C - the product matrix
8242 
8243    Notes:
8244    C must have been created by calling MatPtAPSymbolic and must be destroyed by
8245    the user using MatDeatroy().
8246 
8247    This routine is currently only implemented for pairs of AIJ matrices and classes
8248    which inherit from AIJ.  C will be of type MATAIJ.
8249 
8250    Level: intermediate
8251 
8252 .seealso: MatPtAP(), MatPtAPSymbolic(), MatMatMultNumeric()
8253 @*/
8254 PetscErrorCode  MatPtAPNumeric(Mat A,Mat P,Mat C)
8255 {
8256   PetscErrorCode ierr;
8257 
8258   PetscFunctionBegin;
8259   PetscValidHeaderSpecific(A,MAT_CLASSID,1);
8260   PetscValidType(A,1);
8261   if (!A->assembled) SETERRQ(PetscObjectComm((PetscObject)A),PETSC_ERR_ARG_WRONGSTATE,"Not for unassembled matrix");
8262   if (A->factortype) SETERRQ(PetscObjectComm((PetscObject)A),PETSC_ERR_ARG_WRONGSTATE,"Not for factored matrix");
8263   PetscValidHeaderSpecific(P,MAT_CLASSID,2);
8264   PetscValidType(P,2);
8265   MatCheckPreallocated(P,2);
8266   if (!P->assembled) SETERRQ(PetscObjectComm((PetscObject)A),PETSC_ERR_ARG_WRONGSTATE,"Not for unassembled matrix");
8267   if (P->factortype) SETERRQ(PetscObjectComm((PetscObject)A),PETSC_ERR_ARG_WRONGSTATE,"Not for factored matrix");
8268   PetscValidHeaderSpecific(C,MAT_CLASSID,3);
8269   PetscValidType(C,3);
8270   MatCheckPreallocated(C,3);
8271   if (C->factortype) SETERRQ(PetscObjectComm((PetscObject)A),PETSC_ERR_ARG_WRONGSTATE,"Not for factored matrix");
8272   if (P->cmap->N!=C->rmap->N) SETERRQ2(PetscObjectComm((PetscObject)A),PETSC_ERR_ARG_SIZ,"Matrix dimensions are incompatible, %D != %D",P->cmap->N,C->rmap->N);
8273   if (P->rmap->N!=A->cmap->N) SETERRQ2(PetscObjectComm((PetscObject)A),PETSC_ERR_ARG_SIZ,"Matrix dimensions are incompatible, %D != %D",P->rmap->N,A->cmap->N);
8274   if (A->rmap->N!=A->cmap->N) SETERRQ2(PetscObjectComm((PetscObject)A),PETSC_ERR_ARG_SIZ,"Matrix 'A' must be square, %D != %D",A->rmap->N,A->cmap->N);
8275   if (P->cmap->N!=C->cmap->N) SETERRQ2(PetscObjectComm((PetscObject)A),PETSC_ERR_ARG_SIZ,"Matrix dimensions are incompatible, %D != %D",P->cmap->N,C->cmap->N);
8276   MatCheckPreallocated(A,1);
8277 
8278   ierr = PetscLogEventBegin(MAT_PtAPNumeric,A,P,0,0);CHKERRQ(ierr);
8279   ierr = (*C->ops->ptapnumeric)(A,P,C);CHKERRQ(ierr);
8280   ierr = PetscLogEventEnd(MAT_PtAPNumeric,A,P,0,0);CHKERRQ(ierr);
8281   PetscFunctionReturn(0);
8282 }
8283 
8284 #undef __FUNCT__
8285 #define __FUNCT__ "MatPtAPSymbolic"
8286 /*@
8287    MatPtAPSymbolic - Creates the (i,j) structure of the matrix product C = P^T * A * P
8288 
8289    Neighbor-wise Collective on Mat
8290 
8291    Input Parameters:
8292 +  A - the matrix
8293 -  P - the projection matrix
8294 
8295    Output Parameters:
8296 .  C - the (i,j) structure of the product matrix
8297 
8298    Notes:
8299    C will be created and must be destroyed by the user with MatDestroy().
8300 
8301    This routine is currently only implemented for pairs of SeqAIJ matrices and classes
8302    which inherit from SeqAIJ.  C will be of type MATSEQAIJ.  The product is computed using
8303    this (i,j) structure by calling MatPtAPNumeric().
8304 
8305    Level: intermediate
8306 
8307 .seealso: MatPtAP(), MatPtAPNumeric(), MatMatMultSymbolic()
8308 @*/
8309 PetscErrorCode  MatPtAPSymbolic(Mat A,Mat P,PetscReal fill,Mat *C)
8310 {
8311   PetscErrorCode ierr;
8312 
8313   PetscFunctionBegin;
8314   PetscValidHeaderSpecific(A,MAT_CLASSID,1);
8315   PetscValidType(A,1);
8316   if (!A->assembled) SETERRQ(PetscObjectComm((PetscObject)A),PETSC_ERR_ARG_WRONGSTATE,"Not for unassembled matrix");
8317   if (A->factortype) SETERRQ(PetscObjectComm((PetscObject)A),PETSC_ERR_ARG_WRONGSTATE,"Not for factored matrix");
8318   if (fill <1.0) SETERRQ1(PetscObjectComm((PetscObject)A),PETSC_ERR_ARG_SIZ,"Expected fill=%G must be >= 1.0",fill);
8319   PetscValidHeaderSpecific(P,MAT_CLASSID,2);
8320   PetscValidType(P,2);
8321   MatCheckPreallocated(P,2);
8322   if (!P->assembled) SETERRQ(PetscObjectComm((PetscObject)A),PETSC_ERR_ARG_WRONGSTATE,"Not for unassembled matrix");
8323   if (P->factortype) SETERRQ(PetscObjectComm((PetscObject)A),PETSC_ERR_ARG_WRONGSTATE,"Not for factored matrix");
8324   PetscValidPointer(C,3);
8325 
8326   if (P->rmap->N!=A->cmap->N) SETERRQ2(PetscObjectComm((PetscObject)A),PETSC_ERR_ARG_SIZ,"Matrix dimensions are incompatible, %D != %D",P->rmap->N,A->cmap->N);
8327   if (A->rmap->N!=A->cmap->N) SETERRQ2(PetscObjectComm((PetscObject)A),PETSC_ERR_ARG_SIZ,"Matrix 'A' must be square, %D != %D",A->rmap->N,A->cmap->N);
8328   MatCheckPreallocated(A,1);
8329   ierr = PetscLogEventBegin(MAT_PtAPSymbolic,A,P,0,0);CHKERRQ(ierr);
8330   ierr = (*A->ops->ptapsymbolic)(A,P,fill,C);CHKERRQ(ierr);
8331   ierr = PetscLogEventEnd(MAT_PtAPSymbolic,A,P,0,0);CHKERRQ(ierr);
8332 
8333   /* ierr = MatSetBlockSize(*C,A->rmap->bs);CHKERRQ(ierr); NO! this is not always true -ma */
8334   PetscFunctionReturn(0);
8335 }
8336 
8337 #undef __FUNCT__
8338 #define __FUNCT__ "MatRARt"
8339 /*@
8340    MatRARt - Creates the matrix product C = R * A * R^T
8341 
8342    Neighbor-wise Collective on Mat
8343 
8344    Input Parameters:
8345 +  A - the matrix
8346 .  R - the projection matrix
8347 .  scall - either MAT_INITIAL_MATRIX or MAT_REUSE_MATRIX
8348 -  fill - expected fill as ratio of nnz(C)/nnz(A)
8349 
8350    Output Parameters:
8351 .  C - the product matrix
8352 
8353    Notes:
8354    C will be created and must be destroyed by the user with MatDestroy().
8355 
8356    This routine is currently only implemented for pairs of AIJ matrices and classes
8357    which inherit from AIJ.
8358 
8359    Level: intermediate
8360 
8361 .seealso: MatRARtSymbolic(), MatRARtNumeric(), MatMatMult(), MatPtAP()
8362 @*/
8363 PetscErrorCode  MatRARt(Mat A,Mat R,MatReuse scall,PetscReal fill,Mat *C)
8364 {
8365   PetscErrorCode ierr;
8366 
8367   PetscFunctionBegin;
8368   PetscValidHeaderSpecific(A,MAT_CLASSID,1);
8369   PetscValidType(A,1);
8370   if (!A->assembled) SETERRQ(PetscObjectComm((PetscObject)A),PETSC_ERR_ARG_WRONGSTATE,"Not for unassembled matrix");
8371   if (A->factortype) SETERRQ(PetscObjectComm((PetscObject)A),PETSC_ERR_ARG_WRONGSTATE,"Not for factored matrix");
8372   PetscValidHeaderSpecific(R,MAT_CLASSID,2);
8373   PetscValidType(R,2);
8374   MatCheckPreallocated(R,2);
8375   if (!R->assembled) SETERRQ(PetscObjectComm((PetscObject)A),PETSC_ERR_ARG_WRONGSTATE,"Not for unassembled matrix");
8376   if (R->factortype) SETERRQ(PetscObjectComm((PetscObject)A),PETSC_ERR_ARG_WRONGSTATE,"Not for factored matrix");
8377   PetscValidPointer(C,3);
8378   if (R->cmap->N!=A->rmap->N) SETERRQ2(PetscObjectComm((PetscObject)R),PETSC_ERR_ARG_SIZ,"Matrix dimensions are incompatible, %D != %D",R->cmap->N,A->rmap->N);
8379   if (fill < 1.0) SETERRQ1(PetscObjectComm((PetscObject)A),PETSC_ERR_ARG_SIZ,"Expected fill=%G must be >= 1.0",fill);
8380   MatCheckPreallocated(A,1);
8381 
8382   if (!A->ops->rart) {
8383     MatType mattype;
8384     ierr = MatGetType(A,&mattype);CHKERRQ(ierr);
8385     SETERRQ1(PetscObjectComm((PetscObject)A),PETSC_ERR_SUP,"Matrix of type <%s> does not support RARt",mattype);
8386   }
8387   ierr = PetscLogEventBegin(MAT_RARt,A,R,0,0);CHKERRQ(ierr);
8388   ierr = (*A->ops->rart)(A,R,scall,fill,C);CHKERRQ(ierr);
8389   ierr = PetscLogEventEnd(MAT_RARt,A,R,0,0);CHKERRQ(ierr);
8390   PetscFunctionReturn(0);
8391 }
8392 
8393 #undef __FUNCT__
8394 #define __FUNCT__ "MatRARtNumeric"
8395 /*@
8396    MatRARtNumeric - Computes the matrix product C = R * A * R^T
8397 
8398    Neighbor-wise Collective on Mat
8399 
8400    Input Parameters:
8401 +  A - the matrix
8402 -  R - the projection matrix
8403 
8404    Output Parameters:
8405 .  C - the product matrix
8406 
8407    Notes:
8408    C must have been created by calling MatRARtSymbolic and must be destroyed by
8409    the user using MatDeatroy().
8410 
8411    This routine is currently only implemented for pairs of AIJ matrices and classes
8412    which inherit from AIJ.  C will be of type MATAIJ.
8413 
8414    Level: intermediate
8415 
8416 .seealso: MatRARt(), MatRARtSymbolic(), MatMatMultNumeric()
8417 @*/
8418 PetscErrorCode  MatRARtNumeric(Mat A,Mat R,Mat C)
8419 {
8420   PetscErrorCode ierr;
8421 
8422   PetscFunctionBegin;
8423   PetscValidHeaderSpecific(A,MAT_CLASSID,1);
8424   PetscValidType(A,1);
8425   if (!A->assembled) SETERRQ(PetscObjectComm((PetscObject)A),PETSC_ERR_ARG_WRONGSTATE,"Not for unassembled matrix");
8426   if (A->factortype) SETERRQ(PetscObjectComm((PetscObject)A),PETSC_ERR_ARG_WRONGSTATE,"Not for factored matrix");
8427   PetscValidHeaderSpecific(R,MAT_CLASSID,2);
8428   PetscValidType(R,2);
8429   MatCheckPreallocated(R,2);
8430   if (!R->assembled) SETERRQ(PetscObjectComm((PetscObject)A),PETSC_ERR_ARG_WRONGSTATE,"Not for unassembled matrix");
8431   if (R->factortype) SETERRQ(PetscObjectComm((PetscObject)A),PETSC_ERR_ARG_WRONGSTATE,"Not for factored matrix");
8432   PetscValidHeaderSpecific(C,MAT_CLASSID,3);
8433   PetscValidType(C,3);
8434   MatCheckPreallocated(C,3);
8435   if (C->factortype) SETERRQ(PetscObjectComm((PetscObject)A),PETSC_ERR_ARG_WRONGSTATE,"Not for factored matrix");
8436   if (R->rmap->N!=C->rmap->N) SETERRQ2(PetscObjectComm((PetscObject)A),PETSC_ERR_ARG_SIZ,"Matrix dimensions are incompatible, %D != %D",R->rmap->N,C->rmap->N);
8437   if (R->cmap->N!=A->rmap->N) SETERRQ2(PetscObjectComm((PetscObject)A),PETSC_ERR_ARG_SIZ,"Matrix dimensions are incompatible, %D != %D",R->cmap->N,A->rmap->N);
8438   if (A->rmap->N!=A->cmap->N) SETERRQ2(PetscObjectComm((PetscObject)A),PETSC_ERR_ARG_SIZ,"Matrix 'A' must be square, %D != %D",A->rmap->N,A->cmap->N);
8439   if (R->rmap->N!=C->cmap->N) SETERRQ2(PetscObjectComm((PetscObject)A),PETSC_ERR_ARG_SIZ,"Matrix dimensions are incompatible, %D != %D",R->rmap->N,C->cmap->N);
8440   MatCheckPreallocated(A,1);
8441 
8442   ierr = PetscLogEventBegin(MAT_RARtNumeric,A,R,0,0);CHKERRQ(ierr);
8443   ierr = (*A->ops->rartnumeric)(A,R,C);CHKERRQ(ierr);
8444   ierr = PetscLogEventEnd(MAT_RARtNumeric,A,R,0,0);CHKERRQ(ierr);
8445   PetscFunctionReturn(0);
8446 }
8447 
8448 #undef __FUNCT__
8449 #define __FUNCT__ "MatRARtSymbolic"
8450 /*@
8451    MatRARtSymbolic - Creates the (i,j) structure of the matrix product C = R * A * R^T
8452 
8453    Neighbor-wise Collective on Mat
8454 
8455    Input Parameters:
8456 +  A - the matrix
8457 -  R - the projection matrix
8458 
8459    Output Parameters:
8460 .  C - the (i,j) structure of the product matrix
8461 
8462    Notes:
8463    C will be created and must be destroyed by the user with MatDestroy().
8464 
8465    This routine is currently only implemented for pairs of SeqAIJ matrices and classes
8466    which inherit from SeqAIJ.  C will be of type MATSEQAIJ.  The product is computed using
8467    this (i,j) structure by calling MatRARtNumeric().
8468 
8469    Level: intermediate
8470 
8471 .seealso: MatRARt(), MatRARtNumeric(), MatMatMultSymbolic()
8472 @*/
8473 PetscErrorCode  MatRARtSymbolic(Mat A,Mat R,PetscReal fill,Mat *C)
8474 {
8475   PetscErrorCode ierr;
8476 
8477   PetscFunctionBegin;
8478   PetscValidHeaderSpecific(A,MAT_CLASSID,1);
8479   PetscValidType(A,1);
8480   if (!A->assembled) SETERRQ(PetscObjectComm((PetscObject)A),PETSC_ERR_ARG_WRONGSTATE,"Not for unassembled matrix");
8481   if (A->factortype) SETERRQ(PetscObjectComm((PetscObject)A),PETSC_ERR_ARG_WRONGSTATE,"Not for factored matrix");
8482   if (fill <1.0) SETERRQ1(PetscObjectComm((PetscObject)A),PETSC_ERR_ARG_SIZ,"Expected fill=%G must be >= 1.0",fill);
8483   PetscValidHeaderSpecific(R,MAT_CLASSID,2);
8484   PetscValidType(R,2);
8485   MatCheckPreallocated(R,2);
8486   if (!R->assembled) SETERRQ(PetscObjectComm((PetscObject)A),PETSC_ERR_ARG_WRONGSTATE,"Not for unassembled matrix");
8487   if (R->factortype) SETERRQ(PetscObjectComm((PetscObject)A),PETSC_ERR_ARG_WRONGSTATE,"Not for factored matrix");
8488   PetscValidPointer(C,3);
8489 
8490   if (R->cmap->N!=A->rmap->N) SETERRQ2(PetscObjectComm((PetscObject)A),PETSC_ERR_ARG_SIZ,"Matrix dimensions are incompatible, %D != %D",R->cmap->N,A->rmap->N);
8491   if (A->rmap->N!=A->cmap->N) SETERRQ2(PetscObjectComm((PetscObject)A),PETSC_ERR_ARG_SIZ,"Matrix 'A' must be square, %D != %D",A->rmap->N,A->cmap->N);
8492   MatCheckPreallocated(A,1);
8493   ierr = PetscLogEventBegin(MAT_RARtSymbolic,A,R,0,0);CHKERRQ(ierr);
8494   ierr = (*A->ops->rartsymbolic)(A,R,fill,C);CHKERRQ(ierr);
8495   ierr = PetscLogEventEnd(MAT_RARtSymbolic,A,R,0,0);CHKERRQ(ierr);
8496 
8497   ierr = MatSetBlockSize(*C,A->rmap->bs);CHKERRQ(ierr);
8498   PetscFunctionReturn(0);
8499 }
8500 
8501 #undef __FUNCT__
8502 #define __FUNCT__ "MatMatMult"
8503 /*@
8504    MatMatMult - Performs Matrix-Matrix Multiplication C=A*B.
8505 
8506    Neighbor-wise Collective on Mat
8507 
8508    Input Parameters:
8509 +  A - the left matrix
8510 .  B - the right matrix
8511 .  scall - either MAT_INITIAL_MATRIX or MAT_REUSE_MATRIX
8512 -  fill - expected fill as ratio of nnz(C)/(nnz(A) + nnz(B)), use PETSC_DEFAULT if you do not have a good estimate
8513           if the result is a dense matrix this is irrelevent
8514 
8515    Output Parameters:
8516 .  C - the product matrix
8517 
8518    Notes:
8519    Unless scall is MAT_REUSE_MATRIX C will be created.
8520 
8521    MAT_REUSE_MATRIX can only be used if the matrices A and B have the same nonzero pattern as in the previous call
8522 
8523    To determine the correct fill value, run with -info and search for the string "Fill ratio" to see the value
8524    actually needed.
8525 
8526    If you have many matrices with the same non-zero structure to multiply, you
8527    should either
8528 $   1) use MAT_REUSE_MATRIX in all calls but the first or
8529 $   2) call MatMatMultSymbolic() once and then MatMatMultNumeric() for each product needed
8530 
8531    Level: intermediate
8532 
8533 .seealso: MatMatMultSymbolic(), MatMatMultNumeric(), MatTransposeMatMult(),  MatMatTransposeMult(), MatPtAP()
8534 @*/
8535 PetscErrorCode  MatMatMult(Mat A,Mat B,MatReuse scall,PetscReal fill,Mat *C)
8536 {
8537   PetscErrorCode ierr;
8538   PetscErrorCode (*fA)(Mat,Mat,MatReuse,PetscReal,Mat*);
8539   PetscErrorCode (*fB)(Mat,Mat,MatReuse,PetscReal,Mat*);
8540   PetscErrorCode (*mult)(Mat,Mat,MatReuse,PetscReal,Mat*)=NULL;
8541 
8542   PetscFunctionBegin;
8543   PetscValidHeaderSpecific(A,MAT_CLASSID,1);
8544   PetscValidType(A,1);
8545   MatCheckPreallocated(A,1);
8546   if (!A->assembled) SETERRQ(PetscObjectComm((PetscObject)A),PETSC_ERR_ARG_WRONGSTATE,"Not for unassembled matrix");
8547   if (A->factortype) SETERRQ(PetscObjectComm((PetscObject)A),PETSC_ERR_ARG_WRONGSTATE,"Not for factored matrix");
8548   PetscValidHeaderSpecific(B,MAT_CLASSID,2);
8549   PetscValidType(B,2);
8550   MatCheckPreallocated(B,2);
8551   if (!B->assembled) SETERRQ(PetscObjectComm((PetscObject)A),PETSC_ERR_ARG_WRONGSTATE,"Not for unassembled matrix");
8552   if (B->factortype) SETERRQ(PetscObjectComm((PetscObject)A),PETSC_ERR_ARG_WRONGSTATE,"Not for factored matrix");
8553   PetscValidPointer(C,3);
8554   if (B->rmap->N!=A->cmap->N) SETERRQ2(PetscObjectComm((PetscObject)A),PETSC_ERR_ARG_SIZ,"Matrix dimensions are incompatible, %D != %D",B->rmap->N,A->cmap->N);
8555   if (scall == MAT_REUSE_MATRIX) {
8556     PetscValidPointer(*C,5);
8557     PetscValidHeaderSpecific(*C,MAT_CLASSID,5);
8558     ierr = PetscLogEventBegin(MAT_MatMult,A,B,0,0);CHKERRQ(ierr);
8559     ierr = PetscLogEventBegin(MAT_MatMultNumeric,A,B,0,0);CHKERRQ(ierr);
8560     ierr = (*(*C)->ops->matmultnumeric)(A,B,*C);CHKERRQ(ierr);
8561     ierr = PetscLogEventEnd(MAT_MatMultNumeric,A,B,0,0);CHKERRQ(ierr);
8562     ierr = PetscLogEventEnd(MAT_MatMult,A,B,0,0);CHKERRQ(ierr);
8563     PetscFunctionReturn(0);
8564   }
8565   if (fill == PETSC_DEFAULT || fill == PETSC_DECIDE) fill = 2.0;
8566   if (fill < 1.0) SETERRQ1(PetscObjectComm((PetscObject)A),PETSC_ERR_ARG_SIZ,"Expected fill=%G must be >= 1.0",fill);
8567 
8568   fA = A->ops->matmult;
8569   fB = B->ops->matmult;
8570   if (fB == fA) {
8571     if (!fB) SETERRQ1(PetscObjectComm((PetscObject)A),PETSC_ERR_SUP,"MatMatMult not supported for B of type %s",((PetscObject)B)->type_name);
8572     mult = fB;
8573   } else {
8574     /* dispatch based on the type of A and B from their PetscObject's PetscFunctionLists. */
8575     char multname[256];
8576     ierr = PetscStrcpy(multname,"MatMatMult_");CHKERRQ(ierr);
8577     ierr = PetscStrcat(multname,((PetscObject)A)->type_name);CHKERRQ(ierr);
8578     ierr = PetscStrcat(multname,"_");CHKERRQ(ierr);
8579     ierr = PetscStrcat(multname,((PetscObject)B)->type_name);CHKERRQ(ierr);
8580     ierr = PetscStrcat(multname,"_C");CHKERRQ(ierr); /* e.g., multname = "MatMatMult_seqdense_seqaij_C" */
8581     ierr = PetscObjectQueryFunction((PetscObject)B,multname,&mult);CHKERRQ(ierr);
8582     if (!mult) SETERRQ2(PetscObjectComm((PetscObject)A),PETSC_ERR_ARG_INCOMP,"MatMatMult requires A, %s, to be compatible with B, %s",((PetscObject)A)->type_name,((PetscObject)B)->type_name);
8583   }
8584   ierr = PetscLogEventBegin(MAT_MatMult,A,B,0,0);CHKERRQ(ierr);
8585   ierr = (*mult)(A,B,scall,fill,C);CHKERRQ(ierr);
8586   ierr = PetscLogEventEnd(MAT_MatMult,A,B,0,0);CHKERRQ(ierr);
8587   PetscFunctionReturn(0);
8588 }
8589 
8590 #undef __FUNCT__
8591 #define __FUNCT__ "MatMatMultSymbolic"
8592 /*@
8593    MatMatMultSymbolic - Performs construction, preallocation, and computes the ij structure
8594    of the matrix-matrix product C=A*B.  Call this routine before calling MatMatMultNumeric().
8595 
8596    Neighbor-wise Collective on Mat
8597 
8598    Input Parameters:
8599 +  A - the left matrix
8600 .  B - the right matrix
8601 -  fill - expected fill as ratio of nnz(C)/(nnz(A) + nnz(B)), use PETSC_DEFAULT if you do not have a good estimate,
8602       if C is a dense matrix this is irrelevent
8603 
8604    Output Parameters:
8605 .  C - the product matrix
8606 
8607    Notes:
8608    Unless scall is MAT_REUSE_MATRIX C will be created.
8609 
8610    To determine the correct fill value, run with -info and search for the string "Fill ratio" to see the value
8611    actually needed.
8612 
8613    This routine is currently implemented for
8614     - pairs of AIJ matrices and classes which inherit from AIJ, C will be of type AIJ
8615     - pairs of AIJ (A) and Dense (B) matrix, C will be of type Dense.
8616     - pairs of Dense (A) and AIJ (B) matrix, C will be of type Dense.
8617 
8618    Level: intermediate
8619 
8620    Developers Note: There are ways to estimate the number of nonzeros in the resulting product, see for example, http://arxiv.org/abs/1006.4173
8621      We should incorporate them into PETSc.
8622 
8623 .seealso: MatMatMult(), MatMatMultNumeric()
8624 @*/
8625 PetscErrorCode  MatMatMultSymbolic(Mat A,Mat B,PetscReal fill,Mat *C)
8626 {
8627   PetscErrorCode ierr;
8628   PetscErrorCode (*Asymbolic)(Mat,Mat,PetscReal,Mat*);
8629   PetscErrorCode (*Bsymbolic)(Mat,Mat,PetscReal,Mat*);
8630   PetscErrorCode (*symbolic)(Mat,Mat,PetscReal,Mat*)=NULL;
8631 
8632   PetscFunctionBegin;
8633   PetscValidHeaderSpecific(A,MAT_CLASSID,1);
8634   PetscValidType(A,1);
8635   if (!A->assembled) SETERRQ(PetscObjectComm((PetscObject)A),PETSC_ERR_ARG_WRONGSTATE,"Not for unassembled matrix");
8636   if (A->factortype) SETERRQ(PetscObjectComm((PetscObject)A),PETSC_ERR_ARG_WRONGSTATE,"Not for factored matrix");
8637 
8638   PetscValidHeaderSpecific(B,MAT_CLASSID,2);
8639   PetscValidType(B,2);
8640   MatCheckPreallocated(B,2);
8641   if (!B->assembled) SETERRQ(PetscObjectComm((PetscObject)A),PETSC_ERR_ARG_WRONGSTATE,"Not for unassembled matrix");
8642   if (B->factortype) SETERRQ(PetscObjectComm((PetscObject)A),PETSC_ERR_ARG_WRONGSTATE,"Not for factored matrix");
8643   PetscValidPointer(C,3);
8644 
8645   if (B->rmap->N!=A->cmap->N) SETERRQ2(PetscObjectComm((PetscObject)A),PETSC_ERR_ARG_SIZ,"Matrix dimensions are incompatible, %D != %D",B->rmap->N,A->cmap->N);
8646   if (fill == PETSC_DEFAULT) fill = 2.0;
8647   if (fill < 1.0) SETERRQ1(PetscObjectComm((PetscObject)A),PETSC_ERR_ARG_SIZ,"Expected fill=%G must be > 1.0",fill);
8648   MatCheckPreallocated(A,1);
8649 
8650   Asymbolic = A->ops->matmultsymbolic;
8651   Bsymbolic = B->ops->matmultsymbolic;
8652   if (Asymbolic == Bsymbolic) {
8653     if (!Bsymbolic) SETERRQ1(PetscObjectComm((PetscObject)A),PETSC_ERR_SUP,"C=A*B not implemented for B of type %s",((PetscObject)B)->type_name);
8654     symbolic = Bsymbolic;
8655   } else { /* dispatch based on the type of A and B */
8656     char symbolicname[256];
8657     ierr = PetscStrcpy(symbolicname,"MatMatMultSymbolic_");CHKERRQ(ierr);
8658     ierr = PetscStrcat(symbolicname,((PetscObject)A)->type_name);CHKERRQ(ierr);
8659     ierr = PetscStrcat(symbolicname,"_");CHKERRQ(ierr);
8660     ierr = PetscStrcat(symbolicname,((PetscObject)B)->type_name);CHKERRQ(ierr);
8661     ierr = PetscStrcat(symbolicname,"_C");CHKERRQ(ierr);
8662     ierr = PetscObjectQueryFunction((PetscObject)B,symbolicname,&symbolic);CHKERRQ(ierr);
8663     if (!symbolic) SETERRQ2(PetscObjectComm((PetscObject)A),PETSC_ERR_ARG_INCOMP,"MatMatMultSymbolic requires A, %s, to be compatible with B, %s",((PetscObject)A)->type_name,((PetscObject)B)->type_name);
8664   }
8665   ierr = PetscLogEventBegin(MAT_MatMultSymbolic,A,B,0,0);CHKERRQ(ierr);
8666   ierr = (*symbolic)(A,B,fill,C);CHKERRQ(ierr);
8667   ierr = PetscLogEventEnd(MAT_MatMultSymbolic,A,B,0,0);CHKERRQ(ierr);
8668   PetscFunctionReturn(0);
8669 }
8670 
8671 #undef __FUNCT__
8672 #define __FUNCT__ "MatMatMultNumeric"
8673 /*@
8674    MatMatMultNumeric - Performs the numeric matrix-matrix product.
8675    Call this routine after first calling MatMatMultSymbolic().
8676 
8677    Neighbor-wise Collective on Mat
8678 
8679    Input Parameters:
8680 +  A - the left matrix
8681 -  B - the right matrix
8682 
8683    Output Parameters:
8684 .  C - the product matrix, which was created by from MatMatMultSymbolic() or a call to MatMatMult().
8685 
8686    Notes:
8687    C must have been created with MatMatMultSymbolic().
8688 
8689    This routine is currently implemented for
8690     - pairs of AIJ matrices and classes which inherit from AIJ, C will be of type MATAIJ.
8691     - pairs of AIJ (A) and Dense (B) matrix, C will be of type Dense.
8692     - pairs of Dense (A) and AIJ (B) matrix, C will be of type Dense.
8693 
8694    Level: intermediate
8695 
8696 .seealso: MatMatMult(), MatMatMultSymbolic()
8697 @*/
8698 PetscErrorCode  MatMatMultNumeric(Mat A,Mat B,Mat C)
8699 {
8700   PetscErrorCode ierr;
8701 
8702   PetscFunctionBegin;
8703   ierr = MatMatMult(A,B,MAT_REUSE_MATRIX,0.0,&C);CHKERRQ(ierr);
8704   PetscFunctionReturn(0);
8705 }
8706 
8707 #undef __FUNCT__
8708 #define __FUNCT__ "MatMatTransposeMult"
8709 /*@
8710    MatMatTransposeMult - Performs Matrix-Matrix Multiplication C=A*B^T.
8711 
8712    Neighbor-wise Collective on Mat
8713 
8714    Input Parameters:
8715 +  A - the left matrix
8716 .  B - the right matrix
8717 .  scall - either MAT_INITIAL_MATRIX or MAT_REUSE_MATRIX
8718 -  fill - expected fill as ratio of nnz(C)/(nnz(A) + nnz(B)), use PETSC_DEFAULT if not known
8719 
8720    Output Parameters:
8721 .  C - the product matrix
8722 
8723    Notes:
8724    C will be created if MAT_INITIAL_MATRIX and must be destroyed by the user with MatDestroy().
8725 
8726    MAT_REUSE_MATRIX can only be used if the matrices A and B have the same nonzero pattern as in the previous call
8727 
8728   To determine the correct fill value, run with -info and search for the string "Fill ratio" to see the value
8729    actually needed.
8730 
8731    This routine is currently only implemented for pairs of SeqAIJ matrices.  C will be of type MATSEQAIJ.
8732 
8733    Level: intermediate
8734 
8735 .seealso: MatMatTransposeMultSymbolic(), MatMatTransposeMultNumeric(), MatMatMult(), MatTransposeMatMult() MatPtAP()
8736 @*/
8737 PetscErrorCode  MatMatTransposeMult(Mat A,Mat B,MatReuse scall,PetscReal fill,Mat *C)
8738 {
8739   PetscErrorCode ierr;
8740   PetscErrorCode (*fA)(Mat,Mat,MatReuse,PetscReal,Mat*);
8741   PetscErrorCode (*fB)(Mat,Mat,MatReuse,PetscReal,Mat*);
8742 
8743   PetscFunctionBegin;
8744   PetscValidHeaderSpecific(A,MAT_CLASSID,1);
8745   PetscValidType(A,1);
8746   if (!A->assembled) SETERRQ(PetscObjectComm((PetscObject)A),PETSC_ERR_ARG_WRONGSTATE,"Not for unassembled matrix");
8747   if (A->factortype) SETERRQ(PetscObjectComm((PetscObject)A),PETSC_ERR_ARG_WRONGSTATE,"Not for factored matrix");
8748   PetscValidHeaderSpecific(B,MAT_CLASSID,2);
8749   PetscValidType(B,2);
8750   MatCheckPreallocated(B,2);
8751   if (!B->assembled) SETERRQ(PetscObjectComm((PetscObject)A),PETSC_ERR_ARG_WRONGSTATE,"Not for unassembled matrix");
8752   if (B->factortype) SETERRQ(PetscObjectComm((PetscObject)A),PETSC_ERR_ARG_WRONGSTATE,"Not for factored matrix");
8753   PetscValidPointer(C,3);
8754   if (B->cmap->N!=A->cmap->N) SETERRQ2(PetscObjectComm((PetscObject)A),PETSC_ERR_ARG_SIZ,"Matrix dimensions are incompatible, AN %D != BN %D",A->cmap->N,B->cmap->N);
8755   if (fill == PETSC_DEFAULT || fill == PETSC_DECIDE) fill = 2.0;
8756   if (fill < 1.0) SETERRQ1(PetscObjectComm((PetscObject)A),PETSC_ERR_ARG_SIZ,"Expected fill=%G must be > 1.0",fill);
8757   MatCheckPreallocated(A,1);
8758 
8759   fA = A->ops->mattransposemult;
8760   if (!fA) SETERRQ1(PetscObjectComm((PetscObject)A),PETSC_ERR_SUP,"MatMatTransposeMult not supported for A of type %s",((PetscObject)A)->type_name);
8761   fB = B->ops->mattransposemult;
8762   if (!fB) SETERRQ1(PetscObjectComm((PetscObject)A),PETSC_ERR_SUP,"MatMatTransposeMult not supported for B of type %s",((PetscObject)B)->type_name);
8763   if (fB!=fA) SETERRQ2(PetscObjectComm((PetscObject)A),PETSC_ERR_ARG_INCOMP,"MatMatTransposeMult requires A, %s, to be compatible with B, %s",((PetscObject)A)->type_name,((PetscObject)B)->type_name);
8764 
8765   ierr = PetscLogEventBegin(MAT_MatTransposeMult,A,B,0,0);CHKERRQ(ierr);
8766   if (scall == MAT_INITIAL_MATRIX) {
8767     ierr = PetscLogEventBegin(MAT_MatTransposeMultSymbolic,A,B,0,0);CHKERRQ(ierr);
8768     ierr = (*A->ops->mattransposemultsymbolic)(A,B,fill,C);CHKERRQ(ierr);
8769     ierr = PetscLogEventEnd(MAT_MatTransposeMultSymbolic,A,B,0,0);CHKERRQ(ierr);
8770   }
8771   ierr = PetscLogEventBegin(MAT_MatTransposeMultNumeric,A,B,0,0);CHKERRQ(ierr);
8772   ierr = (*A->ops->mattransposemultnumeric)(A,B,*C);CHKERRQ(ierr);
8773   ierr = PetscLogEventEnd(MAT_MatTransposeMultNumeric,A,B,0,0);CHKERRQ(ierr);
8774   ierr = PetscLogEventEnd(MAT_MatTransposeMult,A,B,0,0);CHKERRQ(ierr);
8775   PetscFunctionReturn(0);
8776 }
8777 
8778 #undef __FUNCT__
8779 #define __FUNCT__ "MatTransposeMatMult"
8780 /*@
8781    MatTransposeMatMult - Performs Matrix-Matrix Multiplication C=A^T*B.
8782 
8783    Neighbor-wise Collective on Mat
8784 
8785    Input Parameters:
8786 +  A - the left matrix
8787 .  B - the right matrix
8788 .  scall - either MAT_INITIAL_MATRIX or MAT_REUSE_MATRIX
8789 -  fill - expected fill as ratio of nnz(C)/(nnz(A) + nnz(B)), use PETSC_DEFAULT if not known
8790 
8791    Output Parameters:
8792 .  C - the product matrix
8793 
8794    Notes:
8795    C will be created if MAT_INITIAL_MATRIX and must be destroyed by the user with MatDestroy().
8796 
8797    MAT_REUSE_MATRIX can only be used if the matrices A and B have the same nonzero pattern as in the previous call
8798 
8799   To determine the correct fill value, run with -info and search for the string "Fill ratio" to see the value
8800    actually needed.
8801 
8802    This routine is currently implemented for pairs of AIJ matrices and pairs of SeqDense matrices and classes
8803    which inherit from SeqAIJ.  C will be of same type as the input matrices.
8804 
8805    Level: intermediate
8806 
8807 .seealso: MatTransposeMatMultSymbolic(), MatTransposeMatMultNumeric(), MatMatMult(), MatMatTransposeMult(), MatPtAP()
8808 @*/
8809 PetscErrorCode  MatTransposeMatMult(Mat A,Mat B,MatReuse scall,PetscReal fill,Mat *C)
8810 {
8811   PetscErrorCode ierr;
8812   PetscErrorCode (*fA)(Mat,Mat,MatReuse,PetscReal,Mat*);
8813   PetscErrorCode (*fB)(Mat,Mat,MatReuse,PetscReal,Mat*);
8814   PetscErrorCode (*transposematmult)(Mat,Mat,MatReuse,PetscReal,Mat*) = NULL;
8815 
8816   PetscFunctionBegin;
8817   PetscValidHeaderSpecific(A,MAT_CLASSID,1);
8818   PetscValidType(A,1);
8819   if (!A->assembled) SETERRQ(PetscObjectComm((PetscObject)A),PETSC_ERR_ARG_WRONGSTATE,"Not for unassembled matrix");
8820   if (A->factortype) SETERRQ(PetscObjectComm((PetscObject)A),PETSC_ERR_ARG_WRONGSTATE,"Not for factored matrix");
8821   PetscValidHeaderSpecific(B,MAT_CLASSID,2);
8822   PetscValidType(B,2);
8823   MatCheckPreallocated(B,2);
8824   if (!B->assembled) SETERRQ(PetscObjectComm((PetscObject)A),PETSC_ERR_ARG_WRONGSTATE,"Not for unassembled matrix");
8825   if (B->factortype) SETERRQ(PetscObjectComm((PetscObject)A),PETSC_ERR_ARG_WRONGSTATE,"Not for factored matrix");
8826   PetscValidPointer(C,3);
8827   if (B->rmap->N!=A->rmap->N) SETERRQ2(PetscObjectComm((PetscObject)A),PETSC_ERR_ARG_SIZ,"Matrix dimensions are incompatible, %D != %D",B->rmap->N,A->rmap->N);
8828   if (fill == PETSC_DEFAULT || fill == PETSC_DECIDE) fill = 2.0;
8829   if (fill < 1.0) SETERRQ1(PetscObjectComm((PetscObject)A),PETSC_ERR_ARG_SIZ,"Expected fill=%G must be > 1.0",fill);
8830   MatCheckPreallocated(A,1);
8831 
8832   fA = A->ops->transposematmult;
8833   if (!fA) SETERRQ1(PetscObjectComm((PetscObject)A),PETSC_ERR_SUP,"MatTransposeMatMult not supported for A of type %s",((PetscObject)A)->type_name);
8834   fB = B->ops->transposematmult;
8835   if (!fB) SETERRQ1(PetscObjectComm((PetscObject)A),PETSC_ERR_SUP,"MatTransposeMatMult not supported for B of type %s",((PetscObject)B)->type_name);
8836   if (fB==fA) {
8837     transposematmult = fA;
8838   }
8839   if (!transposematmult) SETERRQ2(PetscObjectComm((PetscObject)A),PETSC_ERR_ARG_INCOMP,"MatTransposeMatMult requires A, %s, to be compatible with B, %s",((PetscObject)A)->type_name,((PetscObject)B)->type_name);
8840   ierr = PetscLogEventBegin(MAT_TransposeMatMult,A,B,0,0);CHKERRQ(ierr);
8841   ierr = (*transposematmult)(A,B,scall,fill,C);CHKERRQ(ierr);
8842   ierr = PetscLogEventEnd(MAT_TransposeMatMult,A,B,0,0);CHKERRQ(ierr);
8843   PetscFunctionReturn(0);
8844 }
8845 
8846 #undef __FUNCT__
8847 #define __FUNCT__ "MatMatMatMult"
8848 /*@
8849    MatMatMatMult - Performs Matrix-Matrix-Matrix Multiplication D=A*B*C.
8850 
8851    Neighbor-wise Collective on Mat
8852 
8853    Input Parameters:
8854 +  A - the left matrix
8855 .  B - the middle matrix
8856 .  C - the right matrix
8857 .  scall - either MAT_INITIAL_MATRIX or MAT_REUSE_MATRIX
8858 -  fill - expected fill as ratio of nnz(D)/(nnz(A) + nnz(B)+nnz(C)), use PETSC_DEFAULT if you do not have a good estimate
8859           if the result is a dense matrix this is irrelevent
8860 
8861    Output Parameters:
8862 .  D - the product matrix
8863 
8864    Notes:
8865    Unless scall is MAT_REUSE_MATRIX D will be created.
8866 
8867    MAT_REUSE_MATRIX can only be used if the matrices A, B and C have the same nonzero pattern as in the previous call
8868 
8869    To determine the correct fill value, run with -info and search for the string "Fill ratio" to see the value
8870    actually needed.
8871 
8872    If you have many matrices with the same non-zero structure to multiply, you
8873    should either
8874 $   1) use MAT_REUSE_MATRIX in all calls but the first or
8875 $   2) call MatMatMatMultSymbolic() once and then MatMatMatMultNumeric() for each product needed
8876 
8877    Level: intermediate
8878 
8879 .seealso: MatMatMult, MatPtAP()
8880 @*/
8881 PetscErrorCode  MatMatMatMult(Mat A,Mat B,Mat C,MatReuse scall,PetscReal fill,Mat *D)
8882 {
8883   PetscErrorCode ierr;
8884   PetscErrorCode (*fA)(Mat,Mat,Mat,MatReuse,PetscReal,Mat*);
8885   PetscErrorCode (*fB)(Mat,Mat,Mat,MatReuse,PetscReal,Mat*);
8886   PetscErrorCode (*fC)(Mat,Mat,Mat,MatReuse,PetscReal,Mat*);
8887   PetscErrorCode (*mult)(Mat,Mat,Mat,MatReuse,PetscReal,Mat*)=NULL;
8888 
8889   PetscFunctionBegin;
8890   PetscValidHeaderSpecific(A,MAT_CLASSID,1);
8891   PetscValidType(A,1);
8892   MatCheckPreallocated(A,1);
8893   if (!A->assembled) SETERRQ(PetscObjectComm((PetscObject)A),PETSC_ERR_ARG_WRONGSTATE,"Not for unassembled matrix");
8894   if (A->factortype) SETERRQ(PetscObjectComm((PetscObject)A),PETSC_ERR_ARG_WRONGSTATE,"Not for factored matrix");
8895   PetscValidHeaderSpecific(B,MAT_CLASSID,2);
8896   PetscValidType(B,2);
8897   MatCheckPreallocated(B,2);
8898   if (!B->assembled) SETERRQ(PetscObjectComm((PetscObject)B),PETSC_ERR_ARG_WRONGSTATE,"Not for unassembled matrix");
8899   if (B->factortype) SETERRQ(PetscObjectComm((PetscObject)B),PETSC_ERR_ARG_WRONGSTATE,"Not for factored matrix");
8900   PetscValidHeaderSpecific(C,MAT_CLASSID,3);
8901   PetscValidPointer(C,3);
8902   MatCheckPreallocated(C,3);
8903   if (!C->assembled) SETERRQ(PetscObjectComm((PetscObject)C),PETSC_ERR_ARG_WRONGSTATE,"Not for unassembled matrix");
8904   if (C->factortype) SETERRQ(PetscObjectComm((PetscObject)C),PETSC_ERR_ARG_WRONGSTATE,"Not for factored matrix");
8905   if (B->rmap->N!=A->cmap->N) SETERRQ2(PetscObjectComm((PetscObject)B),PETSC_ERR_ARG_SIZ,"Matrix dimensions are incompatible, %D != %D",B->rmap->N,A->cmap->N);
8906   if (C->rmap->N!=B->cmap->N) SETERRQ2(PetscObjectComm((PetscObject)C),PETSC_ERR_ARG_SIZ,"Matrix dimensions are incompatible, %D != %D",C->rmap->N,B->cmap->N);
8907   if (scall == MAT_REUSE_MATRIX) {
8908     PetscValidPointer(*D,6);
8909     PetscValidHeaderSpecific(*D,MAT_CLASSID,6);
8910     ierr = PetscLogEventBegin(MAT_MatMatMult,A,B,0,0);CHKERRQ(ierr);
8911     ierr = (*(*D)->ops->matmatmult)(A,B,C,scall,fill,D);CHKERRQ(ierr);
8912     ierr = PetscLogEventEnd(MAT_MatMatMult,A,B,0,0);CHKERRQ(ierr);
8913     PetscFunctionReturn(0);
8914   }
8915   if (fill == PETSC_DEFAULT || fill == PETSC_DECIDE) fill = 2.0;
8916   if (fill < 1.0) SETERRQ1(PetscObjectComm((PetscObject)A),PETSC_ERR_ARG_SIZ,"Expected fill=%G must be >= 1.0",fill);
8917 
8918   fA = A->ops->matmatmult;
8919   fB = B->ops->matmatmult;
8920   fC = C->ops->matmatmult;
8921   if (fA == fB && fA == fC) {
8922     if (!fA) SETERRQ1(PetscObjectComm((PetscObject)A),PETSC_ERR_SUP,"MatMatMatMult not supported for A of type %s",((PetscObject)A)->type_name);
8923     mult = fA;
8924   } else {
8925     /* dispatch based on the type of A, B and C from their PetscObject's PetscFunctionLists. */
8926     char multname[256];
8927     ierr = PetscStrcpy(multname,"MatMatMatMult_");CHKERRQ(ierr);
8928     ierr = PetscStrcat(multname,((PetscObject)A)->type_name);CHKERRQ(ierr);
8929     ierr = PetscStrcat(multname,"_");CHKERRQ(ierr);
8930     ierr = PetscStrcat(multname,((PetscObject)B)->type_name);CHKERRQ(ierr);
8931     ierr = PetscStrcat(multname,"_");CHKERRQ(ierr);
8932     ierr = PetscStrcat(multname,((PetscObject)C)->type_name);CHKERRQ(ierr);
8933     ierr = PetscStrcat(multname,"_C");CHKERRQ(ierr);
8934     ierr = PetscObjectQueryFunction((PetscObject)B,multname,&mult);CHKERRQ(ierr);
8935     if (!mult) SETERRQ3(PetscObjectComm((PetscObject)A),PETSC_ERR_ARG_INCOMP,"MatMatMatMult requires A, %s, to be compatible with B, %s, C, %s",((PetscObject)A)->type_name,((PetscObject)B)->type_name,((PetscObject)C)->type_name);
8936   }
8937   ierr = PetscLogEventBegin(MAT_MatMatMult,A,B,0,0);CHKERRQ(ierr);
8938   ierr = (*mult)(A,B,C,scall,fill,D);CHKERRQ(ierr);
8939   ierr = PetscLogEventEnd(MAT_MatMatMult,A,B,0,0);CHKERRQ(ierr);
8940   PetscFunctionReturn(0);
8941 }
8942 
8943 #undef __FUNCT__
8944 #define __FUNCT__ "MatGetRedundantMatrix"
8945 /*@C
8946    MatGetRedundantMatrix - Create redundant matrices and put them into processors of subcommunicators.
8947 
8948    Collective on Mat
8949 
8950    Input Parameters:
8951 +  mat - the matrix
8952 .  nsubcomm - the number of subcommunicators (= number of redundant parallel or sequential matrices)
8953 .  subcomm - MPI communicator split from the communicator where mat resides in
8954 .  mlocal_red - number of local rows of the redundant matrix
8955 -  reuse - either MAT_INITIAL_MATRIX or MAT_REUSE_MATRIX
8956 
8957    Output Parameter:
8958 .  matredundant - redundant matrix
8959 
8960    Notes:
8961    MAT_REUSE_MATRIX can only be used when the nonzero structure of the
8962    original matrix has not changed from that last call to MatGetRedundantMatrix().
8963 
8964    This routine creates the duplicated matrices in subcommunicators; you should NOT create them before
8965    calling it.
8966 
8967    Only MPIAIJ matrix is supported.
8968 
8969    Level: advanced
8970 
8971    Concepts: subcommunicator
8972    Concepts: duplicate matrix
8973 
8974 .seealso: MatDestroy()
8975 @*/
8976 PetscErrorCode  MatGetRedundantMatrix(Mat mat,PetscInt nsubcomm,MPI_Comm subcomm,PetscInt mlocal_red,MatReuse reuse,Mat *matredundant)
8977 {
8978   PetscErrorCode ierr;
8979 
8980   PetscFunctionBegin;
8981   PetscValidHeaderSpecific(mat,MAT_CLASSID,1);
8982   if (nsubcomm && reuse == MAT_REUSE_MATRIX) {
8983     PetscValidPointer(*matredundant,6);
8984     PetscValidHeaderSpecific(*matredundant,MAT_CLASSID,6);
8985   }
8986   if (!mat->ops->getredundantmatrix) SETERRQ1(PetscObjectComm((PetscObject)mat),PETSC_ERR_SUP,"Mat type %s",((PetscObject)mat)->type_name);
8987   if (!mat->assembled) SETERRQ(PetscObjectComm((PetscObject)mat),PETSC_ERR_ARG_WRONGSTATE,"Not for unassembled matrix");
8988   if (mat->factortype) SETERRQ(PetscObjectComm((PetscObject)mat),PETSC_ERR_ARG_WRONGSTATE,"Not for factored matrix");
8989   MatCheckPreallocated(mat,1);
8990 
8991   ierr = PetscLogEventBegin(MAT_GetRedundantMatrix,mat,0,0,0);CHKERRQ(ierr);
8992   ierr = (*mat->ops->getredundantmatrix)(mat,nsubcomm,subcomm,mlocal_red,reuse,matredundant);CHKERRQ(ierr);
8993   ierr = PetscLogEventEnd(MAT_GetRedundantMatrix,mat,0,0,0);CHKERRQ(ierr);
8994   PetscFunctionReturn(0);
8995 }
8996 
8997 #undef __FUNCT__
8998 #define __FUNCT__ "MatGetMultiProcBlock"
8999 /*@C
9000    MatGetMultiProcBlock - Create multiple [bjacobi] 'parallel submatrices' from
9001    a given 'mat' object. Each submatrix can span multiple procs.
9002 
9003    Collective on Mat
9004 
9005    Input Parameters:
9006 +  mat - the matrix
9007 .  subcomm - the subcommunicator obtained by com_split(comm)
9008 -  scall - either MAT_INITIAL_MATRIX or MAT_REUSE_MATRIX
9009 
9010    Output Parameter:
9011 .  subMat - 'parallel submatrices each spans a given subcomm
9012 
9013   Notes:
9014   The submatrix partition across processors is dicated by 'subComm' a
9015   communicator obtained by com_split(comm). The comm_split
9016   is not restriced to be grouped with consequitive original ranks.
9017 
9018   Due the comm_split() usage, the parallel layout of the submatrices
9019   map directly to the layout of the original matrix [wrt the local
9020   row,col partitioning]. So the original 'DiagonalMat' naturally maps
9021   into the 'DiagonalMat' of the subMat, hence it is used directly from
9022   the subMat. However the offDiagMat looses some columns - and this is
9023   reconstructed with MatSetValues()
9024 
9025   Level: advanced
9026 
9027   Concepts: subcommunicator
9028   Concepts: submatrices
9029 
9030 .seealso: MatGetSubMatrices()
9031 @*/
9032 PetscErrorCode   MatGetMultiProcBlock(Mat mat, MPI_Comm subComm, MatReuse scall,Mat *subMat)
9033 {
9034   PetscErrorCode ierr;
9035   PetscMPIInt    commsize,subCommSize;
9036 
9037   PetscFunctionBegin;
9038   ierr = MPI_Comm_size(PetscObjectComm((PetscObject)mat),&commsize);CHKERRQ(ierr);
9039   ierr = MPI_Comm_size(subComm,&subCommSize);CHKERRQ(ierr);
9040   if (subCommSize > commsize) SETERRQ2(PetscObjectComm((PetscObject)mat),PETSC_ERR_ARG_OUTOFRANGE,"CommSize %D < SubCommZize %D",commsize,subCommSize);
9041 
9042   ierr = PetscLogEventBegin(MAT_GetMultiProcBlock,mat,0,0,0);CHKERRQ(ierr);
9043   ierr = (*mat->ops->getmultiprocblock)(mat,subComm,scall,subMat);CHKERRQ(ierr);
9044   ierr = PetscLogEventEnd(MAT_GetMultiProcBlock,mat,0,0,0);CHKERRQ(ierr);
9045   PetscFunctionReturn(0);
9046 }
9047 
9048 #undef __FUNCT__
9049 #define __FUNCT__ "MatGetLocalSubMatrix"
9050 /*@
9051    MatGetLocalSubMatrix - Gets a reference to a submatrix specified in local numbering
9052 
9053    Not Collective
9054 
9055    Input Arguments:
9056    mat - matrix to extract local submatrix from
9057    isrow - local row indices for submatrix
9058    iscol - local column indices for submatrix
9059 
9060    Output Arguments:
9061    submat - the submatrix
9062 
9063    Level: intermediate
9064 
9065    Notes:
9066    The submat should be returned with MatRestoreLocalSubMatrix().
9067 
9068    Depending on the format of mat, the returned submat may not implement MatMult().  Its communicator may be
9069    the same as mat, it may be PETSC_COMM_SELF, or some other subcomm of mat's.
9070 
9071    The submat always implements MatSetValuesLocal().  If isrow and iscol have the same block size, then
9072    MatSetValuesBlockedLocal() will also be implemented.
9073 
9074 .seealso: MatRestoreLocalSubMatrix(), MatCreateLocalRef()
9075 @*/
9076 PetscErrorCode  MatGetLocalSubMatrix(Mat mat,IS isrow,IS iscol,Mat *submat)
9077 {
9078   PetscErrorCode ierr;
9079 
9080   PetscFunctionBegin;
9081   PetscValidHeaderSpecific(mat,MAT_CLASSID,1);
9082   PetscValidHeaderSpecific(isrow,IS_CLASSID,2);
9083   PetscValidHeaderSpecific(iscol,IS_CLASSID,3);
9084   PetscCheckSameComm(isrow,2,iscol,3);
9085   PetscValidPointer(submat,4);
9086 
9087   if (mat->ops->getlocalsubmatrix) {
9088     ierr = (*mat->ops->getlocalsubmatrix)(mat,isrow,iscol,submat);CHKERRQ(ierr);
9089   } else {
9090     ierr = MatCreateLocalRef(mat,isrow,iscol,submat);CHKERRQ(ierr);
9091   }
9092   PetscFunctionReturn(0);
9093 }
9094 
9095 #undef __FUNCT__
9096 #define __FUNCT__ "MatRestoreLocalSubMatrix"
9097 /*@
9098    MatRestoreLocalSubMatrix - Restores a reference to a submatrix specified in local numbering
9099 
9100    Not Collective
9101 
9102    Input Arguments:
9103    mat - matrix to extract local submatrix from
9104    isrow - local row indices for submatrix
9105    iscol - local column indices for submatrix
9106    submat - the submatrix
9107 
9108    Level: intermediate
9109 
9110 .seealso: MatGetLocalSubMatrix()
9111 @*/
9112 PetscErrorCode  MatRestoreLocalSubMatrix(Mat mat,IS isrow,IS iscol,Mat *submat)
9113 {
9114   PetscErrorCode ierr;
9115 
9116   PetscFunctionBegin;
9117   PetscValidHeaderSpecific(mat,MAT_CLASSID,1);
9118   PetscValidHeaderSpecific(isrow,IS_CLASSID,2);
9119   PetscValidHeaderSpecific(iscol,IS_CLASSID,3);
9120   PetscCheckSameComm(isrow,2,iscol,3);
9121   PetscValidPointer(submat,4);
9122   if (*submat) {
9123     PetscValidHeaderSpecific(*submat,MAT_CLASSID,4);
9124   }
9125 
9126   if (mat->ops->restorelocalsubmatrix) {
9127     ierr = (*mat->ops->restorelocalsubmatrix)(mat,isrow,iscol,submat);CHKERRQ(ierr);
9128   } else {
9129     ierr = MatDestroy(submat);CHKERRQ(ierr);
9130   }
9131   *submat = NULL;
9132   PetscFunctionReturn(0);
9133 }
9134 
9135 /* --------------------------------------------------------*/
9136 #undef __FUNCT__
9137 #define __FUNCT__ "MatFindZeroDiagonals"
9138 /*@
9139    MatFindZeroDiagonals - Finds all the rows of a matrix that have zero or no entry in the matrix
9140 
9141    Collective on Mat
9142 
9143    Input Parameter:
9144 .  mat - the matrix
9145 
9146    Output Parameter:
9147 .  is - if any rows have zero diagonals this contains the list of them
9148 
9149    Level: developer
9150 
9151    Concepts: matrix-vector product
9152 
9153 .seealso: MatMultTranspose(), MatMultAdd(), MatMultTransposeAdd()
9154 @*/
9155 PetscErrorCode  MatFindZeroDiagonals(Mat mat,IS *is)
9156 {
9157   PetscErrorCode ierr;
9158 
9159   PetscFunctionBegin;
9160   PetscValidHeaderSpecific(mat,MAT_CLASSID,1);
9161   PetscValidType(mat,1);
9162   if (!mat->assembled) SETERRQ(PetscObjectComm((PetscObject)mat),PETSC_ERR_ARG_WRONGSTATE,"Not for unassembled matrix");
9163   if (mat->factortype) SETERRQ(PetscObjectComm((PetscObject)mat),PETSC_ERR_ARG_WRONGSTATE,"Not for factored matrix");
9164 
9165   if (!mat->ops->findzerodiagonals) SETERRQ(PetscObjectComm((PetscObject)mat),PETSC_ERR_SUP,"This matrix type does not have a find zero diagonals defined");
9166   ierr = (*mat->ops->findzerodiagonals)(mat,is);CHKERRQ(ierr);
9167   PetscFunctionReturn(0);
9168 }
9169 
9170 #undef __FUNCT__
9171 #define __FUNCT__ "MatInvertBlockDiagonal"
9172 /*@C
9173   MatInvertBlockDiagonal - Inverts the block diagonal entries.
9174 
9175   Collective on Mat
9176 
9177   Input Parameters:
9178 . mat - the matrix
9179 
9180   Output Parameters:
9181 . values - the block inverses in column major order (FORTRAN-like)
9182 
9183    Note:
9184    This routine is not available from Fortran.
9185 
9186   Level: advanced
9187 @*/
9188 PetscErrorCode MatInvertBlockDiagonal(Mat mat,const PetscScalar **values)
9189 {
9190   PetscErrorCode ierr;
9191 
9192   PetscFunctionBegin;
9193   PetscValidHeaderSpecific(mat,MAT_CLASSID,1);
9194   if (!mat->assembled) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONGSTATE,"Not for unassembled matrix");
9195   if (mat->factortype) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONGSTATE,"Not for factored matrix");
9196   if (!mat->ops->invertblockdiagonal) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"Not supported");
9197   ierr = (*mat->ops->invertblockdiagonal)(mat,values);CHKERRQ(ierr);
9198   PetscFunctionReturn(0);
9199 }
9200 
9201 #undef __FUNCT__
9202 #define __FUNCT__ "MatTransposeColoringDestroy"
9203 /*@C
9204     MatTransposeColoringDestroy - Destroys a coloring context for matrix product C=A*B^T that was created
9205     via MatTransposeColoringCreate().
9206 
9207     Collective on MatTransposeColoring
9208 
9209     Input Parameter:
9210 .   c - coloring context
9211 
9212     Level: intermediate
9213 
9214 .seealso: MatTransposeColoringCreate()
9215 @*/
9216 PetscErrorCode  MatTransposeColoringDestroy(MatTransposeColoring *c)
9217 {
9218   PetscErrorCode       ierr;
9219   MatTransposeColoring matcolor=*c;
9220 
9221   PetscFunctionBegin;
9222   if (!matcolor) PetscFunctionReturn(0);
9223   if (--((PetscObject)matcolor)->refct > 0) {matcolor = 0; PetscFunctionReturn(0);}
9224 
9225   ierr = PetscFree(matcolor->ncolumns);CHKERRQ(ierr);
9226   ierr = PetscFree(matcolor->nrows);CHKERRQ(ierr);
9227   ierr = PetscFree(matcolor->colorforrow);CHKERRQ(ierr);
9228   ierr = PetscFree2(matcolor->rows,matcolor->columnsforspidx);CHKERRQ(ierr);
9229   ierr = PetscFree(matcolor->colorforcol);CHKERRQ(ierr);
9230   ierr = PetscFree(matcolor->columns);CHKERRQ(ierr);
9231   ierr = PetscHeaderDestroy(c);CHKERRQ(ierr);
9232   PetscFunctionReturn(0);
9233 }
9234 
9235 #undef __FUNCT__
9236 #define __FUNCT__ "MatTransColoringApplySpToDen"
9237 /*@C
9238     MatTransColoringApplySpToDen - Given a symbolic matrix product C=A*B^T for which
9239     a MatTransposeColoring context has been created, computes a dense B^T by Apply
9240     MatTransposeColoring to sparse B.
9241 
9242     Collective on MatTransposeColoring
9243 
9244     Input Parameters:
9245 +   B - sparse matrix B
9246 .   Btdense - symbolic dense matrix B^T
9247 -   coloring - coloring context created with MatTransposeColoringCreate()
9248 
9249     Output Parameter:
9250 .   Btdense - dense matrix B^T
9251 
9252     Options Database Keys:
9253 +    -mat_transpose_coloring_view - Activates basic viewing or coloring
9254 .    -mat_transpose_coloring_view_draw - Activates drawing of coloring
9255 -    -mat_transpose_coloring_view_info - Activates viewing of coloring info
9256 
9257     Level: intermediate
9258 
9259 .seealso: MatTransposeColoringCreate(), MatTransposeColoringDestroy()
9260 
9261 .keywords: coloring
9262 @*/
9263 PetscErrorCode MatTransColoringApplySpToDen(MatTransposeColoring coloring,Mat B,Mat Btdense)
9264 {
9265   PetscErrorCode ierr;
9266 
9267   PetscFunctionBegin;
9268   PetscValidHeaderSpecific(B,MAT_CLASSID,1);
9269   PetscValidHeaderSpecific(Btdense,MAT_CLASSID,2);
9270   PetscValidHeaderSpecific(coloring,MAT_TRANSPOSECOLORING_CLASSID,3);
9271 
9272   if (!B->ops->transcoloringapplysptoden) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_SUP,"Not supported for this matrix type %s",((PetscObject)B)->type_name);
9273   ierr = (B->ops->transcoloringapplysptoden)(coloring,B,Btdense);CHKERRQ(ierr);
9274   PetscFunctionReturn(0);
9275 }
9276 
9277 #undef __FUNCT__
9278 #define __FUNCT__ "MatTransColoringApplyDenToSp"
9279 /*@C
9280     MatTransColoringApplyDenToSp - Given a symbolic matrix product Csp=A*B^T for which
9281     a MatTransposeColoring context has been created and a dense matrix Cden=A*Btdense
9282     in which Btdens is obtained from MatTransColoringApplySpToDen(), recover sparse matrix
9283     Csp from Cden.
9284 
9285     Collective on MatTransposeColoring
9286 
9287     Input Parameters:
9288 +   coloring - coloring context created with MatTransposeColoringCreate()
9289 -   Cden - matrix product of a sparse matrix and a dense matrix Btdense
9290 
9291     Output Parameter:
9292 .   Csp - sparse matrix
9293 
9294     Options Database Keys:
9295 +    -mat_multtranspose_coloring_view - Activates basic viewing or coloring
9296 .    -mat_multtranspose_coloring_view_draw - Activates drawing of coloring
9297 -    -mat_multtranspose_coloring_view_info - Activates viewing of coloring info
9298 
9299     Level: intermediate
9300 
9301 .seealso: MatTransposeColoringCreate(), MatTransposeColoringDestroy(), MatTransColoringApplySpToDen()
9302 
9303 .keywords: coloring
9304 @*/
9305 PetscErrorCode MatTransColoringApplyDenToSp(MatTransposeColoring matcoloring,Mat Cden,Mat Csp)
9306 {
9307   PetscErrorCode ierr;
9308 
9309   PetscFunctionBegin;
9310   PetscValidHeaderSpecific(matcoloring,MAT_TRANSPOSECOLORING_CLASSID,1);
9311   PetscValidHeaderSpecific(Cden,MAT_CLASSID,2);
9312   PetscValidHeaderSpecific(Csp,MAT_CLASSID,3);
9313 
9314   if (!Csp->ops->transcoloringapplydentosp) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_SUP,"Not supported for this matrix type %s",((PetscObject)Csp)->type_name);
9315   ierr = (Csp->ops->transcoloringapplydentosp)(matcoloring,Cden,Csp);CHKERRQ(ierr);
9316   PetscFunctionReturn(0);
9317 }
9318 
9319 #undef __FUNCT__
9320 #define __FUNCT__ "MatTransposeColoringCreate"
9321 /*@C
9322    MatTransposeColoringCreate - Creates a matrix coloring context for matrix product C=A*B^T.
9323 
9324    Collective on Mat
9325 
9326    Input Parameters:
9327 +  mat - the matrix product C
9328 -  iscoloring - the coloring of the matrix; usually obtained with MatGetColoring() or DMCreateColoring()
9329 
9330     Output Parameter:
9331 .   color - the new coloring context
9332 
9333     Level: intermediate
9334 
9335 .seealso: MatTransposeColoringDestroy(), MatTransposeColoringSetFromOptions(), MatTransColoringApplySpToDen(),
9336            MatTransColoringApplyDenToSp(), MatTransposeColoringView(),
9337 @*/
9338 PetscErrorCode  MatTransposeColoringCreate(Mat mat,ISColoring iscoloring,MatTransposeColoring *color)
9339 {
9340   MatTransposeColoring c;
9341   MPI_Comm             comm;
9342   PetscErrorCode       ierr;
9343 
9344   PetscFunctionBegin;
9345   ierr = PetscLogEventBegin(MAT_TransposeColoringCreate,mat,0,0,0);CHKERRQ(ierr);
9346   ierr = PetscObjectGetComm((PetscObject)mat,&comm);CHKERRQ(ierr);
9347   ierr = PetscHeaderCreate(c,_p_MatTransposeColoring,int,MAT_TRANSPOSECOLORING_CLASSID,"MatTransposeColoring","Matrix product C=A*B^T via coloring","Mat",comm,MatTransposeColoringDestroy,0);CHKERRQ(ierr);
9348 
9349   c->ctype = iscoloring->ctype;
9350   if (mat->ops->transposecoloringcreate) {
9351     ierr = (*mat->ops->transposecoloringcreate)(mat,iscoloring,c);CHKERRQ(ierr);
9352   } else SETERRQ(PetscObjectComm((PetscObject)mat),PETSC_ERR_SUP,"Code not yet written for this matrix type");
9353 
9354   *color = c;
9355   ierr   = PetscLogEventEnd(MAT_TransposeColoringCreate,mat,0,0,0);CHKERRQ(ierr);
9356   PetscFunctionReturn(0);
9357 }
9358