xref: /petsc/src/mat/interface/matrix.c (revision 36763ca0ee4da147d6889d2a25ad2e0432c9db62)
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/isimpl.h>
8 #include <petsc/private/vecimpl.h>
9 
10 /* Logging support */
11 PetscClassId MAT_CLASSID;
12 PetscClassId MAT_COLORING_CLASSID;
13 PetscClassId MAT_FDCOLORING_CLASSID;
14 PetscClassId MAT_TRANSPOSECOLORING_CLASSID;
15 
16 PetscLogEvent MAT_Mult, MAT_Mults, MAT_MultConstrained, MAT_MultAdd, MAT_MultTranspose;
17 PetscLogEvent MAT_MultTransposeConstrained, MAT_MultTransposeAdd, MAT_Solve, MAT_Solves, MAT_SolveAdd, MAT_SolveTranspose, MAT_MatSolve,MAT_MatTrSolve;
18 PetscLogEvent MAT_SolveTransposeAdd, MAT_SOR, MAT_ForwardSolve, MAT_BackwardSolve, MAT_LUFactor, MAT_LUFactorSymbolic;
19 PetscLogEvent MAT_LUFactorNumeric, MAT_CholeskyFactor, MAT_CholeskyFactorSymbolic, MAT_CholeskyFactorNumeric, MAT_ILUFactor;
20 PetscLogEvent MAT_ILUFactorSymbolic, MAT_ICCFactorSymbolic, MAT_Copy, MAT_Convert, MAT_Scale, MAT_AssemblyBegin;
21 PetscLogEvent MAT_AssemblyEnd, MAT_SetValues, MAT_GetValues, MAT_GetRow, MAT_GetRowIJ, MAT_CreateSubMats, MAT_GetOrdering, MAT_RedundantMat, MAT_GetSeqNonzeroStructure;
22 PetscLogEvent MAT_IncreaseOverlap, MAT_Partitioning, MAT_PartitioningND, MAT_Coarsen, MAT_ZeroEntries, MAT_Load, MAT_View, MAT_AXPY, MAT_FDColoringCreate;
23 PetscLogEvent MAT_FDColoringSetUp, MAT_FDColoringApply,MAT_Transpose,MAT_FDColoringFunction, MAT_CreateSubMat;
24 PetscLogEvent MAT_TransposeColoringCreate;
25 PetscLogEvent MAT_MatMult, MAT_MatMultSymbolic, MAT_MatMultNumeric;
26 PetscLogEvent MAT_PtAP, MAT_PtAPSymbolic, MAT_PtAPNumeric,MAT_RARt, MAT_RARtSymbolic, MAT_RARtNumeric;
27 PetscLogEvent MAT_MatTransposeMult, MAT_MatTransposeMultSymbolic, MAT_MatTransposeMultNumeric;
28 PetscLogEvent MAT_TransposeMatMult, MAT_TransposeMatMultSymbolic, MAT_TransposeMatMultNumeric;
29 PetscLogEvent MAT_MatMatMult, MAT_MatMatMultSymbolic, MAT_MatMatMultNumeric;
30 PetscLogEvent MAT_MultHermitianTranspose,MAT_MultHermitianTransposeAdd;
31 PetscLogEvent MAT_Getsymtranspose, MAT_Getsymtransreduced, MAT_Transpose_SeqAIJ, MAT_GetBrowsOfAcols;
32 PetscLogEvent MAT_GetBrowsOfAocols, MAT_Getlocalmat, MAT_Getlocalmatcondensed, MAT_Seqstompi, MAT_Seqstompinum, MAT_Seqstompisym;
33 PetscLogEvent MAT_Applypapt, MAT_Applypapt_numeric, MAT_Applypapt_symbolic, MAT_GetSequentialNonzeroStructure;
34 PetscLogEvent MAT_GetMultiProcBlock;
35 PetscLogEvent MAT_CUSPARSECopyToGPU, MAT_SetValuesBatch;
36 PetscLogEvent MAT_ViennaCLCopyToGPU;
37 PetscLogEvent MAT_Merge,MAT_Residual,MAT_SetRandom;
38 PetscLogEvent MATCOLORING_Apply,MATCOLORING_Comm,MATCOLORING_Local,MATCOLORING_ISCreate,MATCOLORING_SetUp,MATCOLORING_Weights;
39 
40 const char *const MatFactorTypes[] = {"NONE","LU","CHOLESKY","ILU","ICC","ILUDT","MatFactorType","MAT_FACTOR_",0};
41 
42 /*@
43    MatSetRandom - Sets all components of a matrix to random numbers. For sparse matrices that have been preallocated it randomly selects appropriate locations
44 
45    Logically Collective on Mat
46 
47    Input Parameters:
48 +  x  - the matrix
49 -  rctx - the random number context, formed by PetscRandomCreate(), or NULL and
50           it will create one internally.
51 
52    Output Parameter:
53 .  x  - the matrix
54 
55    Example of Usage:
56 .vb
57      PetscRandomCreate(PETSC_COMM_WORLD,&rctx);
58      MatSetRandom(x,rctx);
59      PetscRandomDestroy(rctx);
60 .ve
61 
62    Level: intermediate
63 
64    Concepts: matrix^setting to random
65    Concepts: random^matrix
66 
67 .seealso: MatZeroEntries(), MatSetValues(), PetscRandomCreate(), PetscRandomDestroy()
68 @*/
69 PetscErrorCode MatSetRandom(Mat x,PetscRandom rctx)
70 {
71   PetscErrorCode ierr;
72   PetscRandom    randObj = NULL;
73 
74   PetscFunctionBegin;
75   PetscValidHeaderSpecific(x,MAT_CLASSID,1);
76   if (rctx) PetscValidHeaderSpecific(rctx,PETSC_RANDOM_CLASSID,2);
77   PetscValidType(x,1);
78 
79   if (!x->ops->setrandom) SETERRQ1(PetscObjectComm((PetscObject)x),PETSC_ERR_SUP,"Mat type %s",((PetscObject)x)->type_name);
80 
81   if (!rctx) {
82     MPI_Comm comm;
83     ierr = PetscObjectGetComm((PetscObject)x,&comm);CHKERRQ(ierr);
84     ierr = PetscRandomCreate(comm,&randObj);CHKERRQ(ierr);
85     ierr = PetscRandomSetFromOptions(randObj);CHKERRQ(ierr);
86     rctx = randObj;
87   }
88 
89   ierr = PetscLogEventBegin(MAT_SetRandom,x,rctx,0,0);CHKERRQ(ierr);
90   ierr = (*x->ops->setrandom)(x,rctx);CHKERRQ(ierr);
91   ierr = PetscLogEventEnd(MAT_SetRandom,x,rctx,0,0);CHKERRQ(ierr);
92 
93   ierr = MatAssemblyBegin(x, MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
94   ierr = MatAssemblyEnd(x, MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
95   ierr = PetscRandomDestroy(&randObj);CHKERRQ(ierr);
96   PetscFunctionReturn(0);
97 }
98 
99 /*@
100    MatFactorGetErrorZeroPivot - returns the pivot value that was determined to be zero and the row it occurred in
101 
102    Logically Collective on Mat
103 
104    Input Parameters:
105 .  mat - the factored matrix
106 
107    Output Parameter:
108 +  pivot - the pivot value computed
109 -  row - the row that the zero pivot occurred. Note that this row must be interpreted carefully due to row reorderings and which processes
110          the share the matrix
111 
112    Level: advanced
113 
114    Notes:
115     This routine does not work for factorizations done with external packages.
116    This routine should only be called if MatGetFactorError() returns a value of MAT_FACTOR_NUMERIC_ZEROPIVOT
117 
118    This can be called on non-factored matrices that come from, for example, matrices used in SOR.
119 
120 .seealso: MatZeroEntries(), MatFactor(), MatGetFactor(), MatFactorSymbolic(), MatFactorClearError(), MatFactorGetErrorZeroPivot()
121 @*/
122 PetscErrorCode MatFactorGetErrorZeroPivot(Mat mat,PetscReal *pivot,PetscInt *row)
123 {
124   PetscFunctionBegin;
125   PetscValidHeaderSpecific(mat,MAT_CLASSID,1);
126   *pivot = mat->factorerror_zeropivot_value;
127   *row   = mat->factorerror_zeropivot_row;
128   PetscFunctionReturn(0);
129 }
130 
131 /*@
132    MatFactorGetError - gets the error code from a factorization
133 
134    Logically Collective on Mat
135 
136    Input Parameters:
137 .  mat - the factored matrix
138 
139    Output Parameter:
140 .  err  - the error code
141 
142    Level: advanced
143 
144    Notes:
145     This can be called on non-factored matrices that come from, for example, matrices used in SOR.
146 
147 .seealso: MatZeroEntries(), MatFactor(), MatGetFactor(), MatFactorSymbolic(), MatFactorClearError(), MatFactorGetErrorZeroPivot()
148 @*/
149 PetscErrorCode MatFactorGetError(Mat mat,MatFactorError *err)
150 {
151   PetscFunctionBegin;
152   PetscValidHeaderSpecific(mat,MAT_CLASSID,1);
153   *err = mat->factorerrortype;
154   PetscFunctionReturn(0);
155 }
156 
157 /*@
158    MatFactorClearError - clears the error code in a factorization
159 
160    Logically Collective on Mat
161 
162    Input Parameter:
163 .  mat - the factored matrix
164 
165    Level: developer
166 
167    Notes:
168     This can be called on non-factored matrices that come from, for example, matrices used in SOR.
169 
170 .seealso: MatZeroEntries(), MatFactor(), MatGetFactor(), MatFactorSymbolic(), MatFactorGetError(), MatFactorGetErrorZeroPivot()
171 @*/
172 PetscErrorCode MatFactorClearError(Mat mat)
173 {
174   PetscFunctionBegin;
175   PetscValidHeaderSpecific(mat,MAT_CLASSID,1);
176   mat->factorerrortype             = MAT_FACTOR_NOERROR;
177   mat->factorerror_zeropivot_value = 0.0;
178   mat->factorerror_zeropivot_row   = 0;
179   PetscFunctionReturn(0);
180 }
181 
182 PETSC_INTERN PetscErrorCode MatFindNonzeroRowsOrCols_Basic(Mat mat,PetscBool cols,PetscReal tol,IS *nonzero)
183 {
184   PetscErrorCode    ierr;
185   Vec               r,l;
186   const PetscScalar *al;
187   PetscInt          i,nz,gnz,N,n;
188 
189   PetscFunctionBegin;
190   ierr = MatCreateVecs(mat,&r,&l);CHKERRQ(ierr);
191   if (!cols) { /* nonzero rows */
192     ierr = MatGetSize(mat,&N,NULL);CHKERRQ(ierr);
193     ierr = MatGetLocalSize(mat,&n,NULL);CHKERRQ(ierr);
194     ierr = VecSet(l,0.0);CHKERRQ(ierr);
195     ierr = VecSetRandom(r,NULL);CHKERRQ(ierr);
196     ierr = MatMult(mat,r,l);CHKERRQ(ierr);
197     ierr = VecGetArrayRead(l,&al);CHKERRQ(ierr);
198   } else { /* nonzero columns */
199     ierr = MatGetSize(mat,NULL,&N);CHKERRQ(ierr);
200     ierr = MatGetLocalSize(mat,NULL,&n);CHKERRQ(ierr);
201     ierr = VecSet(r,0.0);CHKERRQ(ierr);
202     ierr = VecSetRandom(l,NULL);CHKERRQ(ierr);
203     ierr = MatMultTranspose(mat,l,r);CHKERRQ(ierr);
204     ierr = VecGetArrayRead(r,&al);CHKERRQ(ierr);
205   }
206   if (tol <= 0.0) { for (i=0,nz=0;i<n;i++) if (al[i] != 0.0) nz++; }
207   else { for (i=0,nz=0;i<n;i++) if (PetscAbsScalar(al[i]) > tol) nz++; }
208   ierr = MPIU_Allreduce(&nz,&gnz,1,MPIU_INT,MPI_SUM,PetscObjectComm((PetscObject)mat));CHKERRQ(ierr);
209   if (gnz != N) {
210     PetscInt *nzr;
211     ierr = PetscMalloc1(nz,&nzr);CHKERRQ(ierr);
212     if (nz) {
213       if (tol < 0) { for (i=0,nz=0;i<n;i++) if (al[i] != 0.0) nzr[nz++] = i; }
214       else { for (i=0,nz=0;i<n;i++) if (PetscAbsScalar(al[i]) > tol) nzr[nz++] = i; }
215     }
216     ierr = ISCreateGeneral(PetscObjectComm((PetscObject)mat),nz,nzr,PETSC_OWN_POINTER,nonzero);CHKERRQ(ierr);
217   } else *nonzero = NULL;
218   if (!cols) { /* nonzero rows */
219     ierr = VecRestoreArrayRead(l,&al);CHKERRQ(ierr);
220   } else {
221     ierr = VecRestoreArrayRead(r,&al);CHKERRQ(ierr);
222   }
223   ierr = VecDestroy(&l);CHKERRQ(ierr);
224   ierr = VecDestroy(&r);CHKERRQ(ierr);
225   PetscFunctionReturn(0);
226 }
227 
228 /*@
229       MatFindNonzeroRows - Locate all rows that are not completely zero in the matrix
230 
231   Input Parameter:
232 .    A  - the matrix
233 
234   Output Parameter:
235 .    keptrows - the rows that are not completely zero
236 
237   Notes:
238     keptrows is set to NULL if all rows are nonzero.
239 
240   Level: intermediate
241 
242  @*/
243 PetscErrorCode MatFindNonzeroRows(Mat mat,IS *keptrows)
244 {
245   PetscErrorCode ierr;
246 
247   PetscFunctionBegin;
248   PetscValidHeaderSpecific(mat,MAT_CLASSID,1);
249   PetscValidType(mat,1);
250   PetscValidPointer(keptrows,2);
251   if (!mat->assembled) SETERRQ(PetscObjectComm((PetscObject)mat),PETSC_ERR_ARG_WRONGSTATE,"Not for unassembled matrix");
252   if (mat->factortype) SETERRQ(PetscObjectComm((PetscObject)mat),PETSC_ERR_ARG_WRONGSTATE,"Not for factored matrix");
253   if (!mat->ops->findnonzerorows) {
254     ierr = MatFindNonzeroRowsOrCols_Basic(mat,PETSC_FALSE,0.0,keptrows);CHKERRQ(ierr);
255   } else {
256     ierr = (*mat->ops->findnonzerorows)(mat,keptrows);CHKERRQ(ierr);
257   }
258   PetscFunctionReturn(0);
259 }
260 
261 /*@
262       MatFindZeroRows - Locate all rows that are completely zero in the matrix
263 
264   Input Parameter:
265 .    A  - the matrix
266 
267   Output Parameter:
268 .    zerorows - the rows that are completely zero
269 
270   Notes:
271     zerorows is set to NULL if no rows are zero.
272 
273   Level: intermediate
274 
275  @*/
276 PetscErrorCode MatFindZeroRows(Mat mat,IS *zerorows)
277 {
278   PetscErrorCode ierr;
279   IS keptrows;
280   PetscInt m, n;
281 
282   PetscValidHeaderSpecific(mat,MAT_CLASSID,1);
283   PetscValidType(mat,1);
284 
285   ierr = MatFindNonzeroRows(mat, &keptrows);CHKERRQ(ierr);
286   /* MatFindNonzeroRows sets keptrows to NULL if there are no zero rows.
287      In keeping with this convention, we set zerorows to NULL if there are no zero
288      rows. */
289   if (keptrows == NULL) {
290     *zerorows = NULL;
291   } else {
292     ierr = MatGetOwnershipRange(mat,&m,&n);CHKERRQ(ierr);
293     ierr = ISComplement(keptrows,m,n,zerorows);CHKERRQ(ierr);
294     ierr = ISDestroy(&keptrows);CHKERRQ(ierr);
295   }
296   PetscFunctionReturn(0);
297 }
298 
299 /*@
300    MatGetDiagonalBlock - Returns the part of the matrix associated with the on-process coupling
301 
302    Not Collective
303 
304    Input Parameters:
305 .   A - the matrix
306 
307    Output Parameters:
308 .   a - the diagonal part (which is a SEQUENTIAL matrix)
309 
310    Notes:
311     see the manual page for MatCreateAIJ() for more information on the "diagonal part" of the matrix.
312           Use caution, as the reference count on the returned matrix is not incremented and it is used as
313 	  part of the containing MPI Mat's normal operation.
314 
315    Level: advanced
316 
317 @*/
318 PetscErrorCode MatGetDiagonalBlock(Mat A,Mat *a)
319 {
320   PetscErrorCode ierr;
321 
322   PetscFunctionBegin;
323   PetscValidHeaderSpecific(A,MAT_CLASSID,1);
324   PetscValidType(A,1);
325   PetscValidPointer(a,3);
326   if (A->factortype) SETERRQ(PetscObjectComm((PetscObject)A),PETSC_ERR_ARG_WRONGSTATE,"Not for factored matrix");
327   if (!A->ops->getdiagonalblock) {
328     PetscMPIInt size;
329     ierr = MPI_Comm_size(PetscObjectComm((PetscObject)A),&size);CHKERRQ(ierr);
330     if (size == 1) {
331       *a = A;
332       PetscFunctionReturn(0);
333     } else SETERRQ(PetscObjectComm((PetscObject)A),PETSC_ERR_SUP,"Not coded for this matrix type");
334   }
335   ierr = (*A->ops->getdiagonalblock)(A,a);CHKERRQ(ierr);
336   PetscFunctionReturn(0);
337 }
338 
339 /*@
340    MatGetTrace - Gets the trace of a matrix. The sum of the diagonal entries.
341 
342    Collective on Mat
343 
344    Input Parameters:
345 .  mat - the matrix
346 
347    Output Parameter:
348 .   trace - the sum of the diagonal entries
349 
350    Level: advanced
351 
352 @*/
353 PetscErrorCode MatGetTrace(Mat mat,PetscScalar *trace)
354 {
355   PetscErrorCode ierr;
356   Vec            diag;
357 
358   PetscFunctionBegin;
359   ierr = MatCreateVecs(mat,&diag,NULL);CHKERRQ(ierr);
360   ierr = MatGetDiagonal(mat,diag);CHKERRQ(ierr);
361   ierr = VecSum(diag,trace);CHKERRQ(ierr);
362   ierr = VecDestroy(&diag);CHKERRQ(ierr);
363   PetscFunctionReturn(0);
364 }
365 
366 /*@
367    MatRealPart - Zeros out the imaginary part of the matrix
368 
369    Logically Collective on Mat
370 
371    Input Parameters:
372 .  mat - the matrix
373 
374    Level: advanced
375 
376 
377 .seealso: MatImaginaryPart()
378 @*/
379 PetscErrorCode MatRealPart(Mat mat)
380 {
381   PetscErrorCode ierr;
382 
383   PetscFunctionBegin;
384   PetscValidHeaderSpecific(mat,MAT_CLASSID,1);
385   PetscValidType(mat,1);
386   if (!mat->assembled) SETERRQ(PetscObjectComm((PetscObject)mat),PETSC_ERR_ARG_WRONGSTATE,"Not for unassembled matrix");
387   if (mat->factortype) SETERRQ(PetscObjectComm((PetscObject)mat),PETSC_ERR_ARG_WRONGSTATE,"Not for factored matrix");
388   if (!mat->ops->realpart) SETERRQ1(PetscObjectComm((PetscObject)mat),PETSC_ERR_SUP,"Mat type %s",((PetscObject)mat)->type_name);
389   MatCheckPreallocated(mat,1);
390   ierr = (*mat->ops->realpart)(mat);CHKERRQ(ierr);
391 #if defined(PETSC_HAVE_VIENNACL) || defined(PETSC_HAVE_CUDA)
392   if (mat->valid_GPU_matrix != PETSC_OFFLOAD_UNALLOCATED) {
393     mat->valid_GPU_matrix = PETSC_OFFLOAD_CPU;
394   }
395 #endif
396   PetscFunctionReturn(0);
397 }
398 
399 /*@C
400    MatGetGhosts - Get the global index of all ghost nodes defined by the sparse matrix
401 
402    Collective on Mat
403 
404    Input Parameter:
405 .  mat - the matrix
406 
407    Output Parameters:
408 +   nghosts - number of ghosts (note for BAIJ matrices there is one ghost for each block)
409 -   ghosts - the global indices of the ghost points
410 
411    Notes:
412     the nghosts and ghosts are suitable to pass into VecCreateGhost()
413 
414    Level: advanced
415 
416 @*/
417 PetscErrorCode MatGetGhosts(Mat mat,PetscInt *nghosts,const PetscInt *ghosts[])
418 {
419   PetscErrorCode ierr;
420 
421   PetscFunctionBegin;
422   PetscValidHeaderSpecific(mat,MAT_CLASSID,1);
423   PetscValidType(mat,1);
424   if (!mat->assembled) SETERRQ(PetscObjectComm((PetscObject)mat),PETSC_ERR_ARG_WRONGSTATE,"Not for unassembled matrix");
425   if (mat->factortype) SETERRQ(PetscObjectComm((PetscObject)mat),PETSC_ERR_ARG_WRONGSTATE,"Not for factored matrix");
426   if (!mat->ops->getghosts) {
427     if (nghosts) *nghosts = 0;
428     if (ghosts) *ghosts = 0;
429   } else {
430     ierr = (*mat->ops->getghosts)(mat,nghosts,ghosts);CHKERRQ(ierr);
431   }
432   PetscFunctionReturn(0);
433 }
434 
435 
436 /*@
437    MatImaginaryPart - Moves the imaginary part of the matrix to the real part and zeros the imaginary part
438 
439    Logically Collective on Mat
440 
441    Input Parameters:
442 .  mat - the matrix
443 
444    Level: advanced
445 
446 
447 .seealso: MatRealPart()
448 @*/
449 PetscErrorCode MatImaginaryPart(Mat mat)
450 {
451   PetscErrorCode ierr;
452 
453   PetscFunctionBegin;
454   PetscValidHeaderSpecific(mat,MAT_CLASSID,1);
455   PetscValidType(mat,1);
456   if (!mat->assembled) SETERRQ(PetscObjectComm((PetscObject)mat),PETSC_ERR_ARG_WRONGSTATE,"Not for unassembled matrix");
457   if (mat->factortype) SETERRQ(PetscObjectComm((PetscObject)mat),PETSC_ERR_ARG_WRONGSTATE,"Not for factored matrix");
458   if (!mat->ops->imaginarypart) SETERRQ1(PetscObjectComm((PetscObject)mat),PETSC_ERR_SUP,"Mat type %s",((PetscObject)mat)->type_name);
459   MatCheckPreallocated(mat,1);
460   ierr = (*mat->ops->imaginarypart)(mat);CHKERRQ(ierr);
461 #if defined(PETSC_HAVE_VIENNACL) || defined(PETSC_HAVE_CUDA)
462   if (mat->valid_GPU_matrix != PETSC_OFFLOAD_UNALLOCATED) {
463     mat->valid_GPU_matrix = PETSC_OFFLOAD_CPU;
464   }
465 #endif
466   PetscFunctionReturn(0);
467 }
468 
469 /*@
470    MatMissingDiagonal - Determine if sparse matrix is missing a diagonal entry (or block entry for BAIJ matrices)
471 
472    Not Collective
473 
474    Input Parameter:
475 .  mat - the matrix
476 
477    Output Parameters:
478 +  missing - is any diagonal missing
479 -  dd - first diagonal entry that is missing (optional) on this process
480 
481    Level: advanced
482 
483 
484 .seealso: MatRealPart()
485 @*/
486 PetscErrorCode MatMissingDiagonal(Mat mat,PetscBool *missing,PetscInt *dd)
487 {
488   PetscErrorCode ierr;
489 
490   PetscFunctionBegin;
491   PetscValidHeaderSpecific(mat,MAT_CLASSID,1);
492   PetscValidType(mat,1);
493   if (!mat->assembled) SETERRQ(PetscObjectComm((PetscObject)mat),PETSC_ERR_ARG_WRONGSTATE,"Not for unassembled matrix");
494   if (mat->factortype) SETERRQ(PetscObjectComm((PetscObject)mat),PETSC_ERR_ARG_WRONGSTATE,"Not for factored matrix");
495   if (!mat->ops->missingdiagonal) SETERRQ1(PetscObjectComm((PetscObject)mat),PETSC_ERR_SUP,"Mat type %s",((PetscObject)mat)->type_name);
496   ierr = (*mat->ops->missingdiagonal)(mat,missing,dd);CHKERRQ(ierr);
497   PetscFunctionReturn(0);
498 }
499 
500 /*@C
501    MatGetRow - Gets a row of a matrix.  You MUST call MatRestoreRow()
502    for each row that you get to ensure that your application does
503    not bleed memory.
504 
505    Not Collective
506 
507    Input Parameters:
508 +  mat - the matrix
509 -  row - the row to get
510 
511    Output Parameters:
512 +  ncols -  if not NULL, the number of nonzeros in the row
513 .  cols - if not NULL, the column numbers
514 -  vals - if not NULL, the values
515 
516    Notes:
517    This routine is provided for people who need to have direct access
518    to the structure of a matrix.  We hope that we provide enough
519    high-level matrix routines that few users will need it.
520 
521    MatGetRow() always returns 0-based column indices, regardless of
522    whether the internal representation is 0-based (default) or 1-based.
523 
524    For better efficiency, set cols and/or vals to NULL if you do
525    not wish to extract these quantities.
526 
527    The user can only examine the values extracted with MatGetRow();
528    the values cannot be altered.  To change the matrix entries, one
529    must use MatSetValues().
530 
531    You can only have one call to MatGetRow() outstanding for a particular
532    matrix at a time, per processor. MatGetRow() can only obtain rows
533    associated with the given processor, it cannot get rows from the
534    other processors; for that we suggest using MatCreateSubMatrices(), then
535    MatGetRow() on the submatrix. The row index passed to MatGetRow()
536    is in the global number of rows.
537 
538    Fortran Notes:
539    The calling sequence from Fortran is
540 .vb
541    MatGetRow(matrix,row,ncols,cols,values,ierr)
542          Mat     matrix (input)
543          integer row    (input)
544          integer ncols  (output)
545          integer cols(maxcols) (output)
546          double precision (or double complex) values(maxcols) output
547 .ve
548    where maxcols >= maximum nonzeros in any row of the matrix.
549 
550 
551    Caution:
552    Do not try to change the contents of the output arrays (cols and vals).
553    In some cases, this may corrupt the matrix.
554 
555    Level: advanced
556 
557    Concepts: matrices^row access
558 
559 .seealso: MatRestoreRow(), MatSetValues(), MatGetValues(), MatCreateSubMatrices(), MatGetDiagonal()
560 @*/
561 PetscErrorCode MatGetRow(Mat mat,PetscInt row,PetscInt *ncols,const PetscInt *cols[],const PetscScalar *vals[])
562 {
563   PetscErrorCode ierr;
564   PetscInt       incols;
565 
566   PetscFunctionBegin;
567   PetscValidHeaderSpecific(mat,MAT_CLASSID,1);
568   PetscValidType(mat,1);
569   if (!mat->assembled) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONGSTATE,"Not for unassembled matrix");
570   if (mat->factortype) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONGSTATE,"Not for factored matrix");
571   if (!mat->ops->getrow) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_SUP,"Mat type %s",((PetscObject)mat)->type_name);
572   MatCheckPreallocated(mat,1);
573   ierr = PetscLogEventBegin(MAT_GetRow,mat,0,0,0);CHKERRQ(ierr);
574   ierr = (*mat->ops->getrow)(mat,row,&incols,(PetscInt**)cols,(PetscScalar**)vals);CHKERRQ(ierr);
575   if (ncols) *ncols = incols;
576   ierr = PetscLogEventEnd(MAT_GetRow,mat,0,0,0);CHKERRQ(ierr);
577   PetscFunctionReturn(0);
578 }
579 
580 /*@
581    MatConjugate - replaces the matrix values with their complex conjugates
582 
583    Logically Collective on Mat
584 
585    Input Parameters:
586 .  mat - the matrix
587 
588    Level: advanced
589 
590 .seealso:  VecConjugate()
591 @*/
592 PetscErrorCode MatConjugate(Mat mat)
593 {
594 #if defined(PETSC_USE_COMPLEX)
595   PetscErrorCode ierr;
596 
597   PetscFunctionBegin;
598   PetscValidHeaderSpecific(mat,MAT_CLASSID,1);
599   if (!mat->assembled) SETERRQ(PetscObjectComm((PetscObject)mat),PETSC_ERR_ARG_WRONGSTATE,"Not for unassembled matrix");
600   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");
601   ierr = (*mat->ops->conjugate)(mat);CHKERRQ(ierr);
602 #if defined(PETSC_HAVE_VIENNACL) || defined(PETSC_HAVE_CUDA)
603   if (mat->valid_GPU_matrix != PETSC_OFFLOAD_UNALLOCATED) {
604     mat->valid_GPU_matrix = PETSC_OFFLOAD_CPU;
605   }
606 #endif
607   PetscFunctionReturn(0);
608 #else
609   return 0;
610 #endif
611 }
612 
613 /*@C
614    MatRestoreRow - Frees any temporary space allocated by MatGetRow().
615 
616    Not Collective
617 
618    Input Parameters:
619 +  mat - the matrix
620 .  row - the row to get
621 .  ncols, cols - the number of nonzeros and their columns
622 -  vals - if nonzero the column values
623 
624    Notes:
625    This routine should be called after you have finished examining the entries.
626 
627    This routine zeros out ncols, cols, and vals. This is to prevent accidental
628    us of the array after it has been restored. If you pass NULL, it will
629    not zero the pointers.  Use of cols or vals after MatRestoreRow is invalid.
630 
631    Fortran Notes:
632    The calling sequence from Fortran is
633 .vb
634    MatRestoreRow(matrix,row,ncols,cols,values,ierr)
635       Mat     matrix (input)
636       integer row    (input)
637       integer ncols  (output)
638       integer cols(maxcols) (output)
639       double precision (or double complex) values(maxcols) output
640 .ve
641    Where maxcols >= maximum nonzeros in any row of the matrix.
642 
643    In Fortran MatRestoreRow() MUST be called after MatGetRow()
644    before another call to MatGetRow() can be made.
645 
646    Level: advanced
647 
648 .seealso:  MatGetRow()
649 @*/
650 PetscErrorCode MatRestoreRow(Mat mat,PetscInt row,PetscInt *ncols,const PetscInt *cols[],const PetscScalar *vals[])
651 {
652   PetscErrorCode ierr;
653 
654   PetscFunctionBegin;
655   PetscValidHeaderSpecific(mat,MAT_CLASSID,1);
656   if (ncols) PetscValidIntPointer(ncols,3);
657   if (!mat->assembled) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONGSTATE,"Not for unassembled matrix");
658   if (!mat->ops->restorerow) PetscFunctionReturn(0);
659   ierr = (*mat->ops->restorerow)(mat,row,ncols,(PetscInt **)cols,(PetscScalar **)vals);CHKERRQ(ierr);
660   if (ncols) *ncols = 0;
661   if (cols)  *cols = NULL;
662   if (vals)  *vals = NULL;
663   PetscFunctionReturn(0);
664 }
665 
666 /*@
667    MatGetRowUpperTriangular - Sets a flag to enable calls to MatGetRow() for matrix in MATSBAIJ format.
668    You should call MatRestoreRowUpperTriangular() after calling MatGetRow/MatRestoreRow() to disable the flag.
669 
670    Not Collective
671 
672    Input Parameters:
673 +  mat - the matrix
674 
675    Notes:
676    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.
677 
678    Level: advanced
679 
680    Concepts: matrices^row access
681 
682 .seealso: MatRestoreRowRowUpperTriangular()
683 @*/
684 PetscErrorCode MatGetRowUpperTriangular(Mat mat)
685 {
686   PetscErrorCode ierr;
687 
688   PetscFunctionBegin;
689   PetscValidHeaderSpecific(mat,MAT_CLASSID,1);
690   PetscValidType(mat,1);
691   if (!mat->assembled) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONGSTATE,"Not for unassembled matrix");
692   if (mat->factortype) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONGSTATE,"Not for factored matrix");
693   if (!mat->ops->getrowuppertriangular) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_SUP,"Mat type %s",((PetscObject)mat)->type_name);
694   MatCheckPreallocated(mat,1);
695   ierr = (*mat->ops->getrowuppertriangular)(mat);CHKERRQ(ierr);
696   PetscFunctionReturn(0);
697 }
698 
699 /*@
700    MatRestoreRowUpperTriangular - Disable calls to MatGetRow() for matrix in MATSBAIJ format.
701 
702    Not Collective
703 
704    Input Parameters:
705 +  mat - the matrix
706 
707    Notes:
708    This routine should be called after you have finished MatGetRow/MatRestoreRow().
709 
710 
711    Level: advanced
712 
713 .seealso:  MatGetRowUpperTriangular()
714 @*/
715 PetscErrorCode MatRestoreRowUpperTriangular(Mat mat)
716 {
717   PetscErrorCode ierr;
718 
719   PetscFunctionBegin;
720   PetscValidHeaderSpecific(mat,MAT_CLASSID,1);
721   if (!mat->assembled) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONGSTATE,"Not for unassembled matrix");
722   if (!mat->ops->restorerowuppertriangular) PetscFunctionReturn(0);
723   ierr = (*mat->ops->restorerowuppertriangular)(mat);CHKERRQ(ierr);
724   PetscFunctionReturn(0);
725 }
726 
727 /*@C
728    MatSetOptionsPrefix - Sets the prefix used for searching for all
729    Mat options in the database.
730 
731    Logically Collective on Mat
732 
733    Input Parameter:
734 +  A - the Mat context
735 -  prefix - the prefix to prepend to all option names
736 
737    Notes:
738    A hyphen (-) must NOT be given at the beginning of the prefix name.
739    The first character of all runtime options is AUTOMATICALLY the hyphen.
740 
741    Level: advanced
742 
743 .keywords: Mat, set, options, prefix, database
744 
745 .seealso: MatSetFromOptions()
746 @*/
747 PetscErrorCode MatSetOptionsPrefix(Mat A,const char prefix[])
748 {
749   PetscErrorCode ierr;
750 
751   PetscFunctionBegin;
752   PetscValidHeaderSpecific(A,MAT_CLASSID,1);
753   ierr = PetscObjectSetOptionsPrefix((PetscObject)A,prefix);CHKERRQ(ierr);
754   PetscFunctionReturn(0);
755 }
756 
757 /*@C
758    MatAppendOptionsPrefix - Appends to the prefix used for searching for all
759    Mat options in the database.
760 
761    Logically Collective on Mat
762 
763    Input Parameters:
764 +  A - the Mat context
765 -  prefix - the prefix to prepend to all option names
766 
767    Notes:
768    A hyphen (-) must NOT be given at the beginning of the prefix name.
769    The first character of all runtime options is AUTOMATICALLY the hyphen.
770 
771    Level: advanced
772 
773 .keywords: Mat, append, options, prefix, database
774 
775 .seealso: MatGetOptionsPrefix()
776 @*/
777 PetscErrorCode MatAppendOptionsPrefix(Mat A,const char prefix[])
778 {
779   PetscErrorCode ierr;
780 
781   PetscFunctionBegin;
782   PetscValidHeaderSpecific(A,MAT_CLASSID,1);
783   ierr = PetscObjectAppendOptionsPrefix((PetscObject)A,prefix);CHKERRQ(ierr);
784   PetscFunctionReturn(0);
785 }
786 
787 /*@C
788    MatGetOptionsPrefix - Sets the prefix used for searching for all
789    Mat options in the database.
790 
791    Not Collective
792 
793    Input Parameter:
794 .  A - the Mat context
795 
796    Output Parameter:
797 .  prefix - pointer to the prefix string used
798 
799    Notes:
800     On the fortran side, the user should pass in a string 'prefix' of
801    sufficient length to hold the prefix.
802 
803    Level: advanced
804 
805 .keywords: Mat, get, options, prefix, database
806 
807 .seealso: MatAppendOptionsPrefix()
808 @*/
809 PetscErrorCode MatGetOptionsPrefix(Mat A,const char *prefix[])
810 {
811   PetscErrorCode ierr;
812 
813   PetscFunctionBegin;
814   PetscValidHeaderSpecific(A,MAT_CLASSID,1);
815   ierr = PetscObjectGetOptionsPrefix((PetscObject)A,prefix);CHKERRQ(ierr);
816   PetscFunctionReturn(0);
817 }
818 
819 /*@
820    MatResetPreallocation - Reset mat to use the original nonzero pattern provided by users.
821 
822    Collective on Mat
823 
824    Input Parameters:
825 .  A - the Mat context
826 
827    Notes:
828    The allocated memory will be shrunk after calling MatAssembly with MAT_FINAL_ASSEMBLY. Users can reset the preallocation to access the original memory.
829    Currently support MPIAIJ and SEQAIJ.
830 
831    Level: beginner
832 
833 .keywords: Mat, ResetPreallocation
834 
835 .seealso: MatSeqAIJSetPreallocation(), MatMPIAIJSetPreallocation(), MatXAIJSetPreallocation()
836 @*/
837 PetscErrorCode MatResetPreallocation(Mat A)
838 {
839   PetscErrorCode ierr;
840 
841   PetscFunctionBegin;
842   PetscValidHeaderSpecific(A,MAT_CLASSID,1);
843   PetscValidType(A,1);
844   ierr = PetscUseMethod(A,"MatResetPreallocation_C",(Mat),(A));CHKERRQ(ierr);
845   PetscFunctionReturn(0);
846 }
847 
848 
849 /*@
850    MatSetUp - Sets up the internal matrix data structures for the later use.
851 
852    Collective on Mat
853 
854    Input Parameters:
855 .  A - the Mat context
856 
857    Notes:
858    If the user has not set preallocation for this matrix then a default preallocation that is likely to be inefficient is used.
859 
860    If a suitable preallocation routine is used, this function does not need to be called.
861 
862    See the Performance chapter of the PETSc users manual for how to preallocate matrices
863 
864    Level: beginner
865 
866 .keywords: Mat, setup
867 
868 .seealso: MatCreate(), MatDestroy()
869 @*/
870 PetscErrorCode MatSetUp(Mat A)
871 {
872   PetscMPIInt    size;
873   PetscErrorCode ierr;
874 
875   PetscFunctionBegin;
876   PetscValidHeaderSpecific(A,MAT_CLASSID,1);
877   if (!((PetscObject)A)->type_name) {
878     ierr = MPI_Comm_size(PetscObjectComm((PetscObject)A), &size);CHKERRQ(ierr);
879     if (size == 1) {
880       ierr = MatSetType(A, MATSEQAIJ);CHKERRQ(ierr);
881     } else {
882       ierr = MatSetType(A, MATMPIAIJ);CHKERRQ(ierr);
883     }
884   }
885   if (!A->preallocated && A->ops->setup) {
886     ierr = PetscInfo(A,"Warning not preallocating matrix storage\n");CHKERRQ(ierr);
887     ierr = (*A->ops->setup)(A);CHKERRQ(ierr);
888   }
889   ierr = PetscLayoutSetUp(A->rmap);CHKERRQ(ierr);
890   ierr = PetscLayoutSetUp(A->cmap);CHKERRQ(ierr);
891   A->preallocated = PETSC_TRUE;
892   PetscFunctionReturn(0);
893 }
894 
895 #if defined(PETSC_HAVE_SAWS)
896 #include <petscviewersaws.h>
897 #endif
898 /*@C
899    MatView - Visualizes a matrix object.
900 
901    Collective on Mat
902 
903    Input Parameters:
904 +  mat - the matrix
905 -  viewer - visualization context
906 
907   Notes:
908   The available visualization contexts include
909 +    PETSC_VIEWER_STDOUT_SELF - for sequential matrices
910 .    PETSC_VIEWER_STDOUT_WORLD - for parallel matrices created on PETSC_COMM_WORLD
911 .    PETSC_VIEWER_STDOUT_(comm) - for matrices created on MPI communicator comm
912 -     PETSC_VIEWER_DRAW_WORLD - graphical display of nonzero structure
913 
914    The user can open alternative visualization contexts with
915 +    PetscViewerASCIIOpen() - Outputs matrix to a specified file
916 .    PetscViewerBinaryOpen() - Outputs matrix in binary to a
917          specified file; corresponding input uses MatLoad()
918 .    PetscViewerDrawOpen() - Outputs nonzero matrix structure to
919          an X window display
920 -    PetscViewerSocketOpen() - Outputs matrix to Socket viewer.
921          Currently only the sequential dense and AIJ
922          matrix types support the Socket viewer.
923 
924    The user can call PetscViewerPushFormat() to specify the output
925    format of ASCII printed objects (when using PETSC_VIEWER_STDOUT_SELF,
926    PETSC_VIEWER_STDOUT_WORLD and PetscViewerASCIIOpen).  Available formats include
927 +    PETSC_VIEWER_DEFAULT - default, prints matrix contents
928 .    PETSC_VIEWER_ASCII_MATLAB - prints matrix contents in Matlab format
929 .    PETSC_VIEWER_ASCII_DENSE - prints entire matrix including zeros
930 .    PETSC_VIEWER_ASCII_COMMON - prints matrix contents, using a sparse
931          format common among all matrix types
932 .    PETSC_VIEWER_ASCII_IMPL - prints matrix contents, using an implementation-specific
933          format (which is in many cases the same as the default)
934 .    PETSC_VIEWER_ASCII_INFO - prints basic information about the matrix
935          size and structure (not the matrix entries)
936 -    PETSC_VIEWER_ASCII_INFO_DETAIL - prints more detailed information about
937          the matrix structure
938 
939    Options Database Keys:
940 +  -mat_view ::ascii_info - Prints info on matrix at conclusion of MatAssemblyEnd()
941 .  -mat_view ::ascii_info_detail - Prints more detailed info
942 .  -mat_view - Prints matrix in ASCII format
943 .  -mat_view ::ascii_matlab - Prints matrix in Matlab format
944 .  -mat_view draw - PetscDraws nonzero structure of matrix, using MatView() and PetscDrawOpenX().
945 .  -display <name> - Sets display name (default is host)
946 .  -draw_pause <sec> - Sets number of seconds to pause after display
947 .  -mat_view socket - Sends matrix to socket, can be accessed from Matlab (see Users-Manual: ch_matlab for details)
948 .  -viewer_socket_machine <machine> -
949 .  -viewer_socket_port <port> -
950 .  -mat_view binary - save matrix to file in binary format
951 -  -viewer_binary_filename <name> -
952    Level: beginner
953 
954    Notes:
955     See the manual page for MatLoad() for the exact format of the binary file when the binary
956       viewer is used.
957 
958       See share/petsc/matlab/PetscBinaryRead.m for a Matlab code that can read in the binary file when the binary
959       viewer is used.
960 
961       One can use '-mat_view draw -draw_pause -1' to pause the graphical display of matrix nonzero structure,
962       and then use the following mouse functions.
963 + left mouse: zoom in
964 . middle mouse: zoom out
965 - right mouse: continue with the simulation
966 
967    Concepts: matrices^viewing
968    Concepts: matrices^plotting
969    Concepts: matrices^printing
970 
971 .seealso: PetscViewerPushFormat(), PetscViewerASCIIOpen(), PetscViewerDrawOpen(),
972           PetscViewerSocketOpen(), PetscViewerBinaryOpen(), MatLoad()
973 @*/
974 PetscErrorCode MatView(Mat mat,PetscViewer viewer)
975 {
976   PetscErrorCode    ierr;
977   PetscInt          rows,cols,rbs,cbs;
978   PetscBool         iascii,ibinary;
979   PetscViewerFormat format;
980   PetscMPIInt       size;
981 #if defined(PETSC_HAVE_SAWS)
982   PetscBool         issaws;
983 #endif
984 
985   PetscFunctionBegin;
986   PetscValidHeaderSpecific(mat,MAT_CLASSID,1);
987   PetscValidType(mat,1);
988   if (!viewer) {
989     ierr = PetscViewerASCIIGetStdout(PetscObjectComm((PetscObject)mat),&viewer);CHKERRQ(ierr);
990   }
991   PetscValidHeaderSpecific(viewer,PETSC_VIEWER_CLASSID,2);
992   PetscCheckSameComm(mat,1,viewer,2);
993   MatCheckPreallocated(mat,1);
994   ierr = PetscViewerGetFormat(viewer,&format);CHKERRQ(ierr);
995   ierr = MPI_Comm_size(PetscObjectComm((PetscObject)mat),&size);CHKERRQ(ierr);
996   if (size == 1 && format == PETSC_VIEWER_LOAD_BALANCE) PetscFunctionReturn(0);
997   ierr = PetscObjectTypeCompare((PetscObject)viewer,PETSCVIEWERBINARY,&ibinary);CHKERRQ(ierr);
998   if (ibinary) {
999     PetscBool mpiio;
1000     ierr = PetscViewerBinaryGetUseMPIIO(viewer,&mpiio);CHKERRQ(ierr);
1001     if (mpiio) SETERRQ(PetscObjectComm((PetscObject)viewer),PETSC_ERR_SUP,"PETSc matrix viewers do not support using MPI-IO, turn off that flag");
1002   }
1003 
1004   ierr = PetscLogEventBegin(MAT_View,mat,viewer,0,0);CHKERRQ(ierr);
1005   ierr = PetscObjectTypeCompare((PetscObject)viewer,PETSCVIEWERASCII,&iascii);CHKERRQ(ierr);
1006   if ((!iascii || (format != PETSC_VIEWER_ASCII_INFO || format == PETSC_VIEWER_ASCII_INFO_DETAIL)) && mat->factortype) {
1007     SETERRQ(PetscObjectComm((PetscObject)mat),PETSC_ERR_ARG_WRONGSTATE,"No viewers for factored matrix except ASCII info or info_detailed");
1008   }
1009 
1010 #if defined(PETSC_HAVE_SAWS)
1011   ierr = PetscObjectTypeCompare((PetscObject)viewer,PETSCVIEWERSAWS,&issaws);CHKERRQ(ierr);
1012 #endif
1013   if (iascii) {
1014     if (!mat->assembled) SETERRQ(PetscObjectComm((PetscObject)mat),PETSC_ERR_ORDER,"Must call MatAssemblyBegin/End() before viewing matrix");
1015     ierr = PetscObjectPrintClassNamePrefixType((PetscObject)mat,viewer);CHKERRQ(ierr);
1016     if (format == PETSC_VIEWER_ASCII_INFO || format == PETSC_VIEWER_ASCII_INFO_DETAIL) {
1017       MatNullSpace nullsp,transnullsp;
1018 
1019       ierr = PetscViewerASCIIPushTab(viewer);CHKERRQ(ierr);
1020       ierr = MatGetSize(mat,&rows,&cols);CHKERRQ(ierr);
1021       ierr = MatGetBlockSizes(mat,&rbs,&cbs);CHKERRQ(ierr);
1022       if (rbs != 1 || cbs != 1) {
1023         if (rbs != cbs) {ierr = PetscViewerASCIIPrintf(viewer,"rows=%D, cols=%D, rbs=%D, cbs = %D\n",rows,cols,rbs,cbs);CHKERRQ(ierr);}
1024         else            {ierr = PetscViewerASCIIPrintf(viewer,"rows=%D, cols=%D, bs=%D\n",rows,cols,rbs);CHKERRQ(ierr);}
1025       } else {
1026         ierr = PetscViewerASCIIPrintf(viewer,"rows=%D, cols=%D\n",rows,cols);CHKERRQ(ierr);
1027       }
1028       if (mat->factortype) {
1029         MatSolverType solver;
1030         ierr = MatFactorGetSolverType(mat,&solver);CHKERRQ(ierr);
1031         ierr = PetscViewerASCIIPrintf(viewer,"package used to perform factorization: %s\n",solver);CHKERRQ(ierr);
1032       }
1033       if (mat->ops->getinfo) {
1034         MatInfo info;
1035         ierr = MatGetInfo(mat,MAT_GLOBAL_SUM,&info);CHKERRQ(ierr);
1036         ierr = PetscViewerASCIIPrintf(viewer,"total: nonzeros=%.f, allocated nonzeros=%.f\n",info.nz_used,info.nz_allocated);CHKERRQ(ierr);
1037         ierr = PetscViewerASCIIPrintf(viewer,"total number of mallocs used during MatSetValues calls =%D\n",(PetscInt)info.mallocs);CHKERRQ(ierr);
1038       }
1039       ierr = MatGetNullSpace(mat,&nullsp);CHKERRQ(ierr);
1040       ierr = MatGetTransposeNullSpace(mat,&transnullsp);CHKERRQ(ierr);
1041       if (nullsp) {ierr = PetscViewerASCIIPrintf(viewer,"  has attached null space\n");CHKERRQ(ierr);}
1042       if (transnullsp && transnullsp != nullsp) {ierr = PetscViewerASCIIPrintf(viewer,"  has attached transposed null space\n");CHKERRQ(ierr);}
1043       ierr = MatGetNearNullSpace(mat,&nullsp);CHKERRQ(ierr);
1044       if (nullsp) {ierr = PetscViewerASCIIPrintf(viewer,"  has attached near null space\n");CHKERRQ(ierr);}
1045     }
1046 #if defined(PETSC_HAVE_SAWS)
1047   } else if (issaws) {
1048     PetscMPIInt rank;
1049 
1050     ierr = PetscObjectName((PetscObject)mat);CHKERRQ(ierr);
1051     ierr = MPI_Comm_rank(PETSC_COMM_WORLD,&rank);CHKERRQ(ierr);
1052     if (!((PetscObject)mat)->amsmem && !rank) {
1053       ierr = PetscObjectViewSAWs((PetscObject)mat,viewer);CHKERRQ(ierr);
1054     }
1055 #endif
1056   }
1057   if ((format == PETSC_VIEWER_NATIVE || format == PETSC_VIEWER_LOAD_BALANCE) && mat->ops->viewnative) {
1058     ierr = PetscViewerASCIIPushTab(viewer);CHKERRQ(ierr);
1059     ierr = (*mat->ops->viewnative)(mat,viewer);CHKERRQ(ierr);
1060     ierr = PetscViewerASCIIPopTab(viewer);CHKERRQ(ierr);
1061   } else if (mat->ops->view) {
1062     ierr = PetscViewerASCIIPushTab(viewer);CHKERRQ(ierr);
1063     ierr = (*mat->ops->view)(mat,viewer);CHKERRQ(ierr);
1064     ierr = PetscViewerASCIIPopTab(viewer);CHKERRQ(ierr);
1065   }
1066   if (iascii) {
1067     ierr = PetscViewerGetFormat(viewer,&format);CHKERRQ(ierr);
1068     if (format == PETSC_VIEWER_ASCII_INFO || format == PETSC_VIEWER_ASCII_INFO_DETAIL) {
1069       ierr = PetscViewerASCIIPopTab(viewer);CHKERRQ(ierr);
1070     }
1071   }
1072   ierr = PetscLogEventEnd(MAT_View,mat,viewer,0,0);CHKERRQ(ierr);
1073   PetscFunctionReturn(0);
1074 }
1075 
1076 #if defined(PETSC_USE_DEBUG)
1077 #include <../src/sys/totalview/tv_data_display.h>
1078 PETSC_UNUSED static int TV_display_type(const struct _p_Mat *mat)
1079 {
1080   TV_add_row("Local rows", "int", &mat->rmap->n);
1081   TV_add_row("Local columns", "int", &mat->cmap->n);
1082   TV_add_row("Global rows", "int", &mat->rmap->N);
1083   TV_add_row("Global columns", "int", &mat->cmap->N);
1084   TV_add_row("Typename", TV_ascii_string_type, ((PetscObject)mat)->type_name);
1085   return TV_format_OK;
1086 }
1087 #endif
1088 
1089 /*@C
1090    MatLoad - Loads a matrix that has been stored in binary/HDF5 format
1091    with MatView().  The matrix format is determined from the options database.
1092    Generates a parallel MPI matrix if the communicator has more than one
1093    processor.  The default matrix type is AIJ.
1094 
1095    Collective on PetscViewer
1096 
1097    Input Parameters:
1098 +  newmat - the newly loaded matrix, this needs to have been created with MatCreate()
1099             or some related function before a call to MatLoad()
1100 -  viewer - binary/HDF5 file viewer
1101 
1102    Options Database Keys:
1103    Used with block matrix formats (MATSEQBAIJ,  ...) to specify
1104    block size
1105 .    -matload_block_size <bs>
1106 
1107    Level: beginner
1108 
1109    Notes:
1110    If the Mat type has not yet been given then MATAIJ is used, call MatSetFromOptions() on the
1111    Mat before calling this routine if you wish to set it from the options database.
1112 
1113    MatLoad() automatically loads into the options database any options
1114    given in the file filename.info where filename is the name of the file
1115    that was passed to the PetscViewerBinaryOpen(). The options in the info
1116    file will be ignored if you use the -viewer_binary_skip_info option.
1117 
1118    If the type or size of newmat is not set before a call to MatLoad, PETSc
1119    sets the default matrix type AIJ and sets the local and global sizes.
1120    If type and/or size is already set, then the same are used.
1121 
1122    In parallel, each processor can load a subset of rows (or the
1123    entire matrix).  This routine is especially useful when a large
1124    matrix is stored on disk and only part of it is desired on each
1125    processor.  For example, a parallel solver may access only some of
1126    the rows from each processor.  The algorithm used here reads
1127    relatively small blocks of data rather than reading the entire
1128    matrix and then subsetting it.
1129 
1130    Viewer's PetscViewerType must be either PETSCVIEWERBINARY or PETSCVIEWERHDF5.
1131    Such viewer can be created using PetscViewerBinaryOpen()/PetscViewerHDF5Open(),
1132    or the sequence like
1133 $    PetscViewer v;
1134 $    PetscViewerCreate(PETSC_COMM_WORLD,&v);
1135 $    PetscViewerSetType(v,PETSCVIEWERBINARY);
1136 $    PetscViewerSetFromOptions(v);
1137 $    PetscViewerFileSetMode(v,FILE_MODE_READ);
1138 $    PetscViewerFileSetName(v,"datafile");
1139    The optional PetscViewerSetFromOptions() call allows to override PetscViewerSetType() using option
1140 $ -viewer_type {binary,hdf5}
1141 
1142    See the example src/ksp/ksp/examples/tutorials/ex27.c with the first approach,
1143    and src/mat/examples/tutorials/ex10.c with the second approach.
1144 
1145    Notes about the PETSc binary format:
1146    In case of PETSCVIEWERBINARY, a native PETSc binary format is used. Each of the blocks
1147    is read onto rank 0 and then shipped to its destination rank, one after another.
1148    Multiple objects, both matrices and vectors, can be stored within the same file.
1149    Their PetscObject name is ignored; they are loaded in the order of their storage.
1150 
1151    Most users should not need to know the details of the binary storage
1152    format, since MatLoad() and MatView() completely hide these details.
1153    But for anyone who's interested, the standard binary matrix storage
1154    format is
1155 
1156 $    int    MAT_FILE_CLASSID
1157 $    int    number of rows
1158 $    int    number of columns
1159 $    int    total number of nonzeros
1160 $    int    *number nonzeros in each row
1161 $    int    *column indices of all nonzeros (starting index is zero)
1162 $    PetscScalar *values of all nonzeros
1163 
1164    PETSc automatically does the byte swapping for
1165 machines that store the bytes reversed, e.g.  DEC alpha, freebsd,
1166 linux, Windows and the paragon; thus if you write your own binary
1167 read/write routines you have to swap the bytes; see PetscBinaryRead()
1168 and PetscBinaryWrite() to see how this may be done.
1169 
1170    Notes about the HDF5 (MATLAB MAT-File Version 7.3) format:
1171    In case of PETSCVIEWERHDF5, a parallel HDF5 reader is used.
1172    Each processor's chunk is loaded independently by its owning rank.
1173    Multiple objects, both matrices and vectors, can be stored within the same file.
1174    They are looked up by their PetscObject name.
1175 
1176    As the MATLAB MAT-File Version 7.3 format is also a HDF5 flavor, we decided to use
1177    by default the same structure and naming of the AIJ arrays and column count
1178    (see PetscViewerHDF5SetAIJNames())
1179    within the HDF5 file. This means that a MAT file saved with -v7.3 flag, e.g.
1180 $    save example.mat A b -v7.3
1181    can be directly read by this routine (see Reference 1 for details).
1182    Note that depending on your MATLAB version, this format might be a default,
1183    otherwise you can set it as default in Preferences.
1184 
1185    Unless -nocompression flag is used to save the file in MATLAB,
1186    PETSc must be configured with ZLIB package.
1187 
1188    Current HDF5 limitations:
1189    This reader currently supports only real MATSEQAIJ and MATMPIAIJ matrices.
1190 
1191    MatView() is not yet implemented.
1192 
1193    References:
1194 1. MATLAB(R) Documentation, manual page of save(), https://www.mathworks.com/help/matlab/ref/save.html#btox10b-1-version
1195 
1196 .keywords: matrix, load, binary, input, HDF5
1197 
1198 .seealso: PetscViewerBinaryOpen(), PetscViewerSetType(), PetscViewerHDF5SetAIJNames(), MatView(), VecLoad()
1199 
1200  @*/
1201 PetscErrorCode MatLoad(Mat newmat,PetscViewer viewer)
1202 {
1203   PetscErrorCode ierr;
1204   PetscBool      flg;
1205 
1206   PetscFunctionBegin;
1207   PetscValidHeaderSpecific(newmat,MAT_CLASSID,1);
1208   PetscValidHeaderSpecific(viewer,PETSC_VIEWER_CLASSID,2);
1209 
1210   if (!((PetscObject)newmat)->type_name) {
1211     ierr = MatSetType(newmat,MATAIJ);CHKERRQ(ierr);
1212   }
1213 
1214   flg  = PETSC_FALSE;
1215   ierr = PetscOptionsGetBool(((PetscObject)newmat)->options,((PetscObject)newmat)->prefix,"-matload_symmetric",&flg,NULL);CHKERRQ(ierr);
1216   if (flg) {
1217     ierr = MatSetOption(newmat,MAT_SYMMETRIC,PETSC_TRUE);CHKERRQ(ierr);
1218     ierr = MatSetOption(newmat,MAT_SYMMETRY_ETERNAL,PETSC_TRUE);CHKERRQ(ierr);
1219   }
1220   flg  = PETSC_FALSE;
1221   ierr = PetscOptionsGetBool(((PetscObject)newmat)->options,((PetscObject)newmat)->prefix,"-matload_spd",&flg,NULL);CHKERRQ(ierr);
1222   if (flg) {
1223     ierr = MatSetOption(newmat,MAT_SPD,PETSC_TRUE);CHKERRQ(ierr);
1224   }
1225 
1226   if (!newmat->ops->load) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"MatLoad is not supported for type");
1227   ierr = PetscLogEventBegin(MAT_Load,viewer,0,0,0);CHKERRQ(ierr);
1228   ierr = (*newmat->ops->load)(newmat,viewer);CHKERRQ(ierr);
1229   ierr = PetscLogEventEnd(MAT_Load,viewer,0,0,0);CHKERRQ(ierr);
1230   PetscFunctionReturn(0);
1231 }
1232 
1233 PetscErrorCode MatDestroy_Redundant(Mat_Redundant **redundant)
1234 {
1235   PetscErrorCode ierr;
1236   Mat_Redundant  *redund = *redundant;
1237   PetscInt       i;
1238 
1239   PetscFunctionBegin;
1240   if (redund){
1241     if (redund->matseq) { /* via MatCreateSubMatrices()  */
1242       ierr = ISDestroy(&redund->isrow);CHKERRQ(ierr);
1243       ierr = ISDestroy(&redund->iscol);CHKERRQ(ierr);
1244       ierr = MatDestroySubMatrices(1,&redund->matseq);CHKERRQ(ierr);
1245     } else {
1246       ierr = PetscFree2(redund->send_rank,redund->recv_rank);CHKERRQ(ierr);
1247       ierr = PetscFree(redund->sbuf_j);CHKERRQ(ierr);
1248       ierr = PetscFree(redund->sbuf_a);CHKERRQ(ierr);
1249       for (i=0; i<redund->nrecvs; i++) {
1250         ierr = PetscFree(redund->rbuf_j[i]);CHKERRQ(ierr);
1251         ierr = PetscFree(redund->rbuf_a[i]);CHKERRQ(ierr);
1252       }
1253       ierr = PetscFree4(redund->sbuf_nz,redund->rbuf_nz,redund->rbuf_j,redund->rbuf_a);CHKERRQ(ierr);
1254     }
1255 
1256     if (redund->subcomm) {
1257       ierr = PetscCommDestroy(&redund->subcomm);CHKERRQ(ierr);
1258     }
1259     ierr = PetscFree(redund);CHKERRQ(ierr);
1260   }
1261   PetscFunctionReturn(0);
1262 }
1263 
1264 /*@
1265    MatDestroy - Frees space taken by a matrix.
1266 
1267    Collective on Mat
1268 
1269    Input Parameter:
1270 .  A - the matrix
1271 
1272    Level: beginner
1273 
1274 @*/
1275 PetscErrorCode MatDestroy(Mat *A)
1276 {
1277   PetscErrorCode ierr;
1278 
1279   PetscFunctionBegin;
1280   if (!*A) PetscFunctionReturn(0);
1281   PetscValidHeaderSpecific(*A,MAT_CLASSID,1);
1282   if (--((PetscObject)(*A))->refct > 0) {*A = NULL; PetscFunctionReturn(0);}
1283 
1284   /* if memory was published with SAWs then destroy it */
1285   ierr = PetscObjectSAWsViewOff((PetscObject)*A);CHKERRQ(ierr);
1286   if ((*A)->ops->destroy) {
1287     ierr = (*(*A)->ops->destroy)(*A);CHKERRQ(ierr);
1288   }
1289 
1290   ierr = PetscFree((*A)->defaultvectype);CHKERRQ(ierr);
1291   ierr = PetscFree((*A)->bsizes);CHKERRQ(ierr);
1292   ierr = PetscFree((*A)->solvertype);CHKERRQ(ierr);
1293   ierr = MatDestroy_Redundant(&(*A)->redundant);CHKERRQ(ierr);
1294   ierr = MatNullSpaceDestroy(&(*A)->nullsp);CHKERRQ(ierr);
1295   ierr = MatNullSpaceDestroy(&(*A)->transnullsp);CHKERRQ(ierr);
1296   ierr = MatNullSpaceDestroy(&(*A)->nearnullsp);CHKERRQ(ierr);
1297   ierr = MatDestroy(&(*A)->schur);CHKERRQ(ierr);
1298   ierr = PetscLayoutDestroy(&(*A)->rmap);CHKERRQ(ierr);
1299   ierr = PetscLayoutDestroy(&(*A)->cmap);CHKERRQ(ierr);
1300   ierr = PetscHeaderDestroy(A);CHKERRQ(ierr);
1301   PetscFunctionReturn(0);
1302 }
1303 
1304 /*@C
1305    MatSetValues - Inserts or adds a block of values into a matrix.
1306    These values may be cached, so MatAssemblyBegin() and MatAssemblyEnd()
1307    MUST be called after all calls to MatSetValues() have been completed.
1308 
1309    Not Collective
1310 
1311    Input Parameters:
1312 +  mat - the matrix
1313 .  v - a logically two-dimensional array of values
1314 .  m, idxm - the number of rows and their global indices
1315 .  n, idxn - the number of columns and their global indices
1316 -  addv - either ADD_VALUES or INSERT_VALUES, where
1317    ADD_VALUES adds values to any existing entries, and
1318    INSERT_VALUES replaces existing entries with new values
1319 
1320    Notes:
1321    If you create the matrix yourself (that is not with a call to DMCreateMatrix()) then you MUST call MatXXXXSetPreallocation() or
1322       MatSetUp() before using this routine
1323 
1324    By default the values, v, are row-oriented. See MatSetOption() for other options.
1325 
1326    Calls to MatSetValues() with the INSERT_VALUES and ADD_VALUES
1327    options cannot be mixed without intervening calls to the assembly
1328    routines.
1329 
1330    MatSetValues() uses 0-based row and column numbers in Fortran
1331    as well as in C.
1332 
1333    Negative indices may be passed in idxm and idxn, these rows and columns are
1334    simply ignored. This allows easily inserting element stiffness matrices
1335    with homogeneous Dirchlet boundary conditions that you don't want represented
1336    in the matrix.
1337 
1338    Efficiency Alert:
1339    The routine MatSetValuesBlocked() may offer much better efficiency
1340    for users of block sparse formats (MATSEQBAIJ and MATMPIBAIJ).
1341 
1342    Level: beginner
1343 
1344    Developer Notes:
1345     This is labeled with C so does not automatically generate Fortran stubs and interfaces
1346                     because it requires multiple Fortran interfaces depending on which arguments are scalar or arrays.
1347 
1348    Concepts: matrices^putting entries in
1349 
1350 .seealso: MatSetOption(), MatAssemblyBegin(), MatAssemblyEnd(), MatSetValuesBlocked(), MatSetValuesLocal(),
1351           InsertMode, INSERT_VALUES, ADD_VALUES
1352 @*/
1353 PetscErrorCode MatSetValues(Mat mat,PetscInt m,const PetscInt idxm[],PetscInt n,const PetscInt idxn[],const PetscScalar v[],InsertMode addv)
1354 {
1355   PetscErrorCode ierr;
1356 #if defined(PETSC_USE_DEBUG)
1357   PetscInt       i,j;
1358 #endif
1359 
1360   PetscFunctionBeginHot;
1361   PetscValidHeaderSpecific(mat,MAT_CLASSID,1);
1362   PetscValidType(mat,1);
1363   if (!m || !n) PetscFunctionReturn(0); /* no values to insert */
1364   PetscValidIntPointer(idxm,3);
1365   PetscValidIntPointer(idxn,5);
1366   PetscValidScalarPointer(v,6);
1367   MatCheckPreallocated(mat,1);
1368   if (mat->insertmode == NOT_SET_VALUES) {
1369     mat->insertmode = addv;
1370   }
1371 #if defined(PETSC_USE_DEBUG)
1372   else if (mat->insertmode != addv) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONGSTATE,"Cannot mix add values and insert values");
1373   if (mat->factortype) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONGSTATE,"Not for factored matrix");
1374   if (!mat->ops->setvalues) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_SUP,"Mat type %s",((PetscObject)mat)->type_name);
1375 
1376   for (i=0; i<m; i++) {
1377     for (j=0; j<n; j++) {
1378       if (mat->erroriffailure && PetscIsInfOrNanScalar(v[i*n+j]))
1379 #if defined(PETSC_USE_COMPLEX)
1380         SETERRQ4(PETSC_COMM_SELF,PETSC_ERR_FP,"Inserting %g+ig at matrix entry (%D,%D)",(double)PetscRealPart(v[i*n+j]),(double)PetscImaginaryPart(v[i*n+j]),idxm[i],idxn[j]);
1381 #else
1382         SETERRQ3(PETSC_COMM_SELF,PETSC_ERR_FP,"Inserting %g at matrix entry (%D,%D)",(double)v[i*n+j],idxm[i],idxn[j]);
1383 #endif
1384     }
1385   }
1386 #endif
1387 
1388   if (mat->assembled) {
1389     mat->was_assembled = PETSC_TRUE;
1390     mat->assembled     = PETSC_FALSE;
1391   }
1392   ierr = PetscLogEventBegin(MAT_SetValues,mat,0,0,0);CHKERRQ(ierr);
1393   ierr = (*mat->ops->setvalues)(mat,m,idxm,n,idxn,v,addv);CHKERRQ(ierr);
1394   ierr = PetscLogEventEnd(MAT_SetValues,mat,0,0,0);CHKERRQ(ierr);
1395 #if defined(PETSC_HAVE_VIENNACL) || defined(PETSC_HAVE_CUDA)
1396   if (mat->valid_GPU_matrix != PETSC_OFFLOAD_UNALLOCATED) {
1397     mat->valid_GPU_matrix = PETSC_OFFLOAD_CPU;
1398   }
1399 #endif
1400   PetscFunctionReturn(0);
1401 }
1402 
1403 
1404 /*@
1405    MatSetValuesRowLocal - Inserts a row (block row for BAIJ matrices) of nonzero
1406         values into a matrix
1407 
1408    Not Collective
1409 
1410    Input Parameters:
1411 +  mat - the matrix
1412 .  row - the (block) row to set
1413 -  v - a logically two-dimensional array of values
1414 
1415    Notes:
1416    By the values, v, are column-oriented (for the block version) and sorted
1417 
1418    All the nonzeros in the row must be provided
1419 
1420    The matrix must have previously had its column indices set
1421 
1422    The row must belong to this process
1423 
1424    Level: intermediate
1425 
1426    Concepts: matrices^putting entries in
1427 
1428 .seealso: MatSetOption(), MatAssemblyBegin(), MatAssemblyEnd(), MatSetValuesBlocked(), MatSetValuesLocal(),
1429           InsertMode, INSERT_VALUES, ADD_VALUES, MatSetValues(), MatSetValuesRow(), MatSetLocalToGlobalMapping()
1430 @*/
1431 PetscErrorCode MatSetValuesRowLocal(Mat mat,PetscInt row,const PetscScalar v[])
1432 {
1433   PetscErrorCode ierr;
1434   PetscInt       globalrow;
1435 
1436   PetscFunctionBegin;
1437   PetscValidHeaderSpecific(mat,MAT_CLASSID,1);
1438   PetscValidType(mat,1);
1439   PetscValidScalarPointer(v,2);
1440   ierr = ISLocalToGlobalMappingApply(mat->rmap->mapping,1,&row,&globalrow);CHKERRQ(ierr);
1441   ierr = MatSetValuesRow(mat,globalrow,v);CHKERRQ(ierr);
1442 #if defined(PETSC_HAVE_VIENNACL) || defined(PETSC_HAVE_CUDA)
1443   if (mat->valid_GPU_matrix != PETSC_OFFLOAD_UNALLOCATED) {
1444     mat->valid_GPU_matrix = PETSC_OFFLOAD_CPU;
1445   }
1446 #endif
1447   PetscFunctionReturn(0);
1448 }
1449 
1450 /*@
1451    MatSetValuesRow - Inserts a row (block row for BAIJ matrices) of nonzero
1452         values into a matrix
1453 
1454    Not Collective
1455 
1456    Input Parameters:
1457 +  mat - the matrix
1458 .  row - the (block) row to set
1459 -  v - a logically two-dimensional (column major) array of values for  block matrices with blocksize larger than one, otherwise a one dimensional array of values
1460 
1461    Notes:
1462    The values, v, are column-oriented for the block version.
1463 
1464    All the nonzeros in the row must be provided
1465 
1466    THE MATRIX MUST HAVE PREVIOUSLY HAD ITS COLUMN INDICES SET. IT IS RARE THAT THIS ROUTINE IS USED, usually MatSetValues() is used.
1467 
1468    The row must belong to this process
1469 
1470    Level: advanced
1471 
1472    Concepts: matrices^putting entries in
1473 
1474 .seealso: MatSetOption(), MatAssemblyBegin(), MatAssemblyEnd(), MatSetValuesBlocked(), MatSetValuesLocal(),
1475           InsertMode, INSERT_VALUES, ADD_VALUES, MatSetValues()
1476 @*/
1477 PetscErrorCode MatSetValuesRow(Mat mat,PetscInt row,const PetscScalar v[])
1478 {
1479   PetscErrorCode ierr;
1480 
1481   PetscFunctionBeginHot;
1482   PetscValidHeaderSpecific(mat,MAT_CLASSID,1);
1483   PetscValidType(mat,1);
1484   MatCheckPreallocated(mat,1);
1485   PetscValidScalarPointer(v,2);
1486 #if defined(PETSC_USE_DEBUG)
1487   if (mat->insertmode == ADD_VALUES) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONGSTATE,"Cannot mix add and insert values");
1488   if (mat->factortype) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONGSTATE,"Not for factored matrix");
1489 #endif
1490   mat->insertmode = INSERT_VALUES;
1491 
1492   if (mat->assembled) {
1493     mat->was_assembled = PETSC_TRUE;
1494     mat->assembled     = PETSC_FALSE;
1495   }
1496   ierr = PetscLogEventBegin(MAT_SetValues,mat,0,0,0);CHKERRQ(ierr);
1497   if (!mat->ops->setvaluesrow) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_SUP,"Mat type %s",((PetscObject)mat)->type_name);
1498   ierr = (*mat->ops->setvaluesrow)(mat,row,v);CHKERRQ(ierr);
1499   ierr = PetscLogEventEnd(MAT_SetValues,mat,0,0,0);CHKERRQ(ierr);
1500 #if defined(PETSC_HAVE_VIENNACL) || defined(PETSC_HAVE_CUDA)
1501   if (mat->valid_GPU_matrix != PETSC_OFFLOAD_UNALLOCATED) {
1502     mat->valid_GPU_matrix = PETSC_OFFLOAD_CPU;
1503   }
1504 #endif
1505   PetscFunctionReturn(0);
1506 }
1507 
1508 /*@
1509    MatSetValuesStencil - Inserts or adds a block of values into a matrix.
1510      Using structured grid indexing
1511 
1512    Not Collective
1513 
1514    Input Parameters:
1515 +  mat - the matrix
1516 .  m - number of rows being entered
1517 .  idxm - grid coordinates (and component number when dof > 1) for matrix rows being entered
1518 .  n - number of columns being entered
1519 .  idxn - grid coordinates (and component number when dof > 1) for matrix columns being entered
1520 .  v - a logically two-dimensional array of values
1521 -  addv - either ADD_VALUES or INSERT_VALUES, where
1522    ADD_VALUES adds values to any existing entries, and
1523    INSERT_VALUES replaces existing entries with new values
1524 
1525    Notes:
1526    By default the values, v, are row-oriented.  See MatSetOption() for other options.
1527 
1528    Calls to MatSetValuesStencil() with the INSERT_VALUES and ADD_VALUES
1529    options cannot be mixed without intervening calls to the assembly
1530    routines.
1531 
1532    The grid coordinates are across the entire grid, not just the local portion
1533 
1534    MatSetValuesStencil() uses 0-based row and column numbers in Fortran
1535    as well as in C.
1536 
1537    For setting/accessing vector values via array coordinates you can use the DMDAVecGetArray() routine
1538 
1539    In order to use this routine you must either obtain the matrix with DMCreateMatrix()
1540    or call MatSetLocalToGlobalMapping() and MatSetStencil() first.
1541 
1542    The columns and rows in the stencil passed in MUST be contained within the
1543    ghost region of the given process as set with DMDACreateXXX() or MatSetStencil(). For example,
1544    if you create a DMDA with an overlap of one grid level and on a particular process its first
1545    local nonghost x logical coordinate is 6 (so its first ghost x logical coordinate is 5) the
1546    first i index you can use in your column and row indices in MatSetStencil() is 5.
1547 
1548    In Fortran idxm and idxn should be declared as
1549 $     MatStencil idxm(4,m),idxn(4,n)
1550    and the values inserted using
1551 $    idxm(MatStencil_i,1) = i
1552 $    idxm(MatStencil_j,1) = j
1553 $    idxm(MatStencil_k,1) = k
1554 $    idxm(MatStencil_c,1) = c
1555    etc
1556 
1557    For periodic boundary conditions use negative indices for values to the left (below 0; that are to be
1558    obtained by wrapping values from right edge). For values to the right of the last entry using that index plus one
1559    etc to obtain values that obtained by wrapping the values from the left edge. This does not work for anything but the
1560    DM_BOUNDARY_PERIODIC boundary type.
1561 
1562    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
1563    a single value per point) you can skip filling those indices.
1564 
1565    Inspired by the structured grid interface to the HYPRE package
1566    (http://www.llnl.gov/CASC/hypre)
1567 
1568    Efficiency Alert:
1569    The routine MatSetValuesBlockedStencil() may offer much better efficiency
1570    for users of block sparse formats (MATSEQBAIJ and MATMPIBAIJ).
1571 
1572    Level: beginner
1573 
1574    Concepts: matrices^putting entries in
1575 
1576 .seealso: MatSetOption(), MatAssemblyBegin(), MatAssemblyEnd(), MatSetValuesBlocked(), MatSetValuesLocal()
1577           MatSetValues(), MatSetValuesBlockedStencil(), MatSetStencil(), DMCreateMatrix(), DMDAVecGetArray(), MatStencil
1578 @*/
1579 PetscErrorCode MatSetValuesStencil(Mat mat,PetscInt m,const MatStencil idxm[],PetscInt n,const MatStencil idxn[],const PetscScalar v[],InsertMode addv)
1580 {
1581   PetscErrorCode ierr;
1582   PetscInt       buf[8192],*bufm=0,*bufn=0,*jdxm,*jdxn;
1583   PetscInt       j,i,dim = mat->stencil.dim,*dims = mat->stencil.dims+1,tmp;
1584   PetscInt       *starts = mat->stencil.starts,*dxm = (PetscInt*)idxm,*dxn = (PetscInt*)idxn,sdim = dim - (1 - (PetscInt)mat->stencil.noc);
1585 
1586   PetscFunctionBegin;
1587   if (!m || !n) PetscFunctionReturn(0); /* no values to insert */
1588   PetscValidHeaderSpecific(mat,MAT_CLASSID,1);
1589   PetscValidType(mat,1);
1590   PetscValidIntPointer(idxm,3);
1591   PetscValidIntPointer(idxn,5);
1592   PetscValidScalarPointer(v,6);
1593 
1594   if ((m+n) <= (PetscInt)(sizeof(buf)/sizeof(PetscInt))) {
1595     jdxm = buf; jdxn = buf+m;
1596   } else {
1597     ierr = PetscMalloc2(m,&bufm,n,&bufn);CHKERRQ(ierr);
1598     jdxm = bufm; jdxn = bufn;
1599   }
1600   for (i=0; i<m; i++) {
1601     for (j=0; j<3-sdim; j++) dxm++;
1602     tmp = *dxm++ - starts[0];
1603     for (j=0; j<dim-1; j++) {
1604       if ((*dxm++ - starts[j+1]) < 0 || tmp < 0) tmp = -1;
1605       else                                       tmp = tmp*dims[j] + *(dxm-1) - starts[j+1];
1606     }
1607     if (mat->stencil.noc) dxm++;
1608     jdxm[i] = tmp;
1609   }
1610   for (i=0; i<n; i++) {
1611     for (j=0; j<3-sdim; j++) dxn++;
1612     tmp = *dxn++ - starts[0];
1613     for (j=0; j<dim-1; j++) {
1614       if ((*dxn++ - starts[j+1]) < 0 || tmp < 0) tmp = -1;
1615       else                                       tmp = tmp*dims[j] + *(dxn-1) - starts[j+1];
1616     }
1617     if (mat->stencil.noc) dxn++;
1618     jdxn[i] = tmp;
1619   }
1620   ierr = MatSetValuesLocal(mat,m,jdxm,n,jdxn,v,addv);CHKERRQ(ierr);
1621   ierr = PetscFree2(bufm,bufn);CHKERRQ(ierr);
1622   PetscFunctionReturn(0);
1623 }
1624 
1625 /*@
1626    MatSetValuesBlockedStencil - Inserts or adds a block of values into a matrix.
1627      Using structured grid indexing
1628 
1629    Not Collective
1630 
1631    Input Parameters:
1632 +  mat - the matrix
1633 .  m - number of rows being entered
1634 .  idxm - grid coordinates for matrix rows being entered
1635 .  n - number of columns being entered
1636 .  idxn - grid coordinates for matrix columns being entered
1637 .  v - a logically two-dimensional array of values
1638 -  addv - either ADD_VALUES or INSERT_VALUES, where
1639    ADD_VALUES adds values to any existing entries, and
1640    INSERT_VALUES replaces existing entries with new values
1641 
1642    Notes:
1643    By default the values, v, are row-oriented and unsorted.
1644    See MatSetOption() for other options.
1645 
1646    Calls to MatSetValuesBlockedStencil() with the INSERT_VALUES and ADD_VALUES
1647    options cannot be mixed without intervening calls to the assembly
1648    routines.
1649 
1650    The grid coordinates are across the entire grid, not just the local portion
1651 
1652    MatSetValuesBlockedStencil() uses 0-based row and column numbers in Fortran
1653    as well as in C.
1654 
1655    For setting/accessing vector values via array coordinates you can use the DMDAVecGetArray() routine
1656 
1657    In order to use this routine you must either obtain the matrix with DMCreateMatrix()
1658    or call MatSetBlockSize(), MatSetLocalToGlobalMapping() and MatSetStencil() first.
1659 
1660    The columns and rows in the stencil passed in MUST be contained within the
1661    ghost region of the given process as set with DMDACreateXXX() or MatSetStencil(). For example,
1662    if you create a DMDA with an overlap of one grid level and on a particular process its first
1663    local nonghost x logical coordinate is 6 (so its first ghost x logical coordinate is 5) the
1664    first i index you can use in your column and row indices in MatSetStencil() is 5.
1665 
1666    In Fortran idxm and idxn should be declared as
1667 $     MatStencil idxm(4,m),idxn(4,n)
1668    and the values inserted using
1669 $    idxm(MatStencil_i,1) = i
1670 $    idxm(MatStencil_j,1) = j
1671 $    idxm(MatStencil_k,1) = k
1672    etc
1673 
1674    Negative indices may be passed in idxm and idxn, these rows and columns are
1675    simply ignored. This allows easily inserting element stiffness matrices
1676    with homogeneous Dirchlet boundary conditions that you don't want represented
1677    in the matrix.
1678 
1679    Inspired by the structured grid interface to the HYPRE package
1680    (http://www.llnl.gov/CASC/hypre)
1681 
1682    Level: beginner
1683 
1684    Concepts: matrices^putting entries in
1685 
1686 .seealso: MatSetOption(), MatAssemblyBegin(), MatAssemblyEnd(), MatSetValuesBlocked(), MatSetValuesLocal()
1687           MatSetValues(), MatSetValuesStencil(), MatSetStencil(), DMCreateMatrix(), DMDAVecGetArray(), MatStencil,
1688           MatSetBlockSize(), MatSetLocalToGlobalMapping()
1689 @*/
1690 PetscErrorCode MatSetValuesBlockedStencil(Mat mat,PetscInt m,const MatStencil idxm[],PetscInt n,const MatStencil idxn[],const PetscScalar v[],InsertMode addv)
1691 {
1692   PetscErrorCode ierr;
1693   PetscInt       buf[8192],*bufm=0,*bufn=0,*jdxm,*jdxn;
1694   PetscInt       j,i,dim = mat->stencil.dim,*dims = mat->stencil.dims+1,tmp;
1695   PetscInt       *starts = mat->stencil.starts,*dxm = (PetscInt*)idxm,*dxn = (PetscInt*)idxn,sdim = dim - (1 - (PetscInt)mat->stencil.noc);
1696 
1697   PetscFunctionBegin;
1698   if (!m || !n) PetscFunctionReturn(0); /* no values to insert */
1699   PetscValidHeaderSpecific(mat,MAT_CLASSID,1);
1700   PetscValidType(mat,1);
1701   PetscValidIntPointer(idxm,3);
1702   PetscValidIntPointer(idxn,5);
1703   PetscValidScalarPointer(v,6);
1704 
1705   if ((m+n) <= (PetscInt)(sizeof(buf)/sizeof(PetscInt))) {
1706     jdxm = buf; jdxn = buf+m;
1707   } else {
1708     ierr = PetscMalloc2(m,&bufm,n,&bufn);CHKERRQ(ierr);
1709     jdxm = bufm; jdxn = bufn;
1710   }
1711   for (i=0; i<m; i++) {
1712     for (j=0; j<3-sdim; j++) dxm++;
1713     tmp = *dxm++ - starts[0];
1714     for (j=0; j<sdim-1; j++) {
1715       if ((*dxm++ - starts[j+1]) < 0 || tmp < 0) tmp = -1;
1716       else                                       tmp = tmp*dims[j] + *(dxm-1) - starts[j+1];
1717     }
1718     dxm++;
1719     jdxm[i] = tmp;
1720   }
1721   for (i=0; i<n; i++) {
1722     for (j=0; j<3-sdim; j++) dxn++;
1723     tmp = *dxn++ - starts[0];
1724     for (j=0; j<sdim-1; j++) {
1725       if ((*dxn++ - starts[j+1]) < 0 || tmp < 0) tmp = -1;
1726       else                                       tmp = tmp*dims[j] + *(dxn-1) - starts[j+1];
1727     }
1728     dxn++;
1729     jdxn[i] = tmp;
1730   }
1731   ierr = MatSetValuesBlockedLocal(mat,m,jdxm,n,jdxn,v,addv);CHKERRQ(ierr);
1732   ierr = PetscFree2(bufm,bufn);CHKERRQ(ierr);
1733 #if defined(PETSC_HAVE_VIENNACL) || defined(PETSC_HAVE_CUDA)
1734   if (mat->valid_GPU_matrix != PETSC_OFFLOAD_UNALLOCATED) {
1735     mat->valid_GPU_matrix = PETSC_OFFLOAD_CPU;
1736   }
1737 #endif
1738   PetscFunctionReturn(0);
1739 }
1740 
1741 /*@
1742    MatSetStencil - Sets the grid information for setting values into a matrix via
1743         MatSetValuesStencil()
1744 
1745    Not Collective
1746 
1747    Input Parameters:
1748 +  mat - the matrix
1749 .  dim - dimension of the grid 1, 2, or 3
1750 .  dims - number of grid points in x, y, and z direction, including ghost points on your processor
1751 .  starts - starting point of ghost nodes on your processor in x, y, and z direction
1752 -  dof - number of degrees of freedom per node
1753 
1754 
1755    Inspired by the structured grid interface to the HYPRE package
1756    (www.llnl.gov/CASC/hyper)
1757 
1758    For matrices generated with DMCreateMatrix() this routine is automatically called and so not needed by the
1759    user.
1760 
1761    Level: beginner
1762 
1763    Concepts: matrices^putting entries in
1764 
1765 .seealso: MatSetOption(), MatAssemblyBegin(), MatAssemblyEnd(), MatSetValuesBlocked(), MatSetValuesLocal()
1766           MatSetValues(), MatSetValuesBlockedStencil(), MatSetValuesStencil()
1767 @*/
1768 PetscErrorCode MatSetStencil(Mat mat,PetscInt dim,const PetscInt dims[],const PetscInt starts[],PetscInt dof)
1769 {
1770   PetscInt i;
1771 
1772   PetscFunctionBegin;
1773   PetscValidHeaderSpecific(mat,MAT_CLASSID,1);
1774   PetscValidIntPointer(dims,3);
1775   PetscValidIntPointer(starts,4);
1776 
1777   mat->stencil.dim = dim + (dof > 1);
1778   for (i=0; i<dim; i++) {
1779     mat->stencil.dims[i]   = dims[dim-i-1];      /* copy the values in backwards */
1780     mat->stencil.starts[i] = starts[dim-i-1];
1781   }
1782   mat->stencil.dims[dim]   = dof;
1783   mat->stencil.starts[dim] = 0;
1784   mat->stencil.noc         = (PetscBool)(dof == 1);
1785   PetscFunctionReturn(0);
1786 }
1787 
1788 /*@C
1789    MatSetValuesBlocked - Inserts or adds a block of values into a matrix.
1790 
1791    Not Collective
1792 
1793    Input Parameters:
1794 +  mat - the matrix
1795 .  v - a logically two-dimensional array of values
1796 .  m, idxm - the number of block rows and their global block indices
1797 .  n, idxn - the number of block columns and their global block indices
1798 -  addv - either ADD_VALUES or INSERT_VALUES, where
1799    ADD_VALUES adds values to any existing entries, and
1800    INSERT_VALUES replaces existing entries with new values
1801 
1802    Notes:
1803    If you create the matrix yourself (that is not with a call to DMCreateMatrix()) then you MUST call
1804    MatXXXXSetPreallocation() or MatSetUp() before using this routine.
1805 
1806    The m and n count the NUMBER of blocks in the row direction and column direction,
1807    NOT the total number of rows/columns; for example, if the block size is 2 and
1808    you are passing in values for rows 2,3,4,5  then m would be 2 (not 4).
1809    The values in idxm would be 1 2; that is the first index for each block divided by
1810    the block size.
1811 
1812    Note that you must call MatSetBlockSize() when constructing this matrix (before
1813    preallocating it).
1814 
1815    By default the values, v, are row-oriented, so the layout of
1816    v is the same as for MatSetValues(). See MatSetOption() for other options.
1817 
1818    Calls to MatSetValuesBlocked() with the INSERT_VALUES and ADD_VALUES
1819    options cannot be mixed without intervening calls to the assembly
1820    routines.
1821 
1822    MatSetValuesBlocked() uses 0-based row and column numbers in Fortran
1823    as well as in C.
1824 
1825    Negative indices may be passed in idxm and idxn, these rows and columns are
1826    simply ignored. This allows easily inserting element stiffness matrices
1827    with homogeneous Dirchlet boundary conditions that you don't want represented
1828    in the matrix.
1829 
1830    Each time an entry is set within a sparse matrix via MatSetValues(),
1831    internal searching must be done to determine where to place the
1832    data in the matrix storage space.  By instead inserting blocks of
1833    entries via MatSetValuesBlocked(), the overhead of matrix assembly is
1834    reduced.
1835 
1836    Example:
1837 $   Suppose m=n=2 and block size(bs) = 2 The array is
1838 $
1839 $   1  2  | 3  4
1840 $   5  6  | 7  8
1841 $   - - - | - - -
1842 $   9  10 | 11 12
1843 $   13 14 | 15 16
1844 $
1845 $   v[] should be passed in like
1846 $   v[] = [1,2,3,4,5,6,7,8,9,10,11,12,13,14,15,16]
1847 $
1848 $  If you are not using row oriented storage of v (that is you called MatSetOption(mat,MAT_ROW_ORIENTED,PETSC_FALSE)) then
1849 $   v[] = [1,5,9,13,2,6,10,14,3,7,11,15,4,8,12,16]
1850 
1851    Level: intermediate
1852 
1853    Concepts: matrices^putting entries in blocked
1854 
1855 .seealso: MatSetBlockSize(), MatSetOption(), MatAssemblyBegin(), MatAssemblyEnd(), MatSetValues(), MatSetValuesBlockedLocal()
1856 @*/
1857 PetscErrorCode MatSetValuesBlocked(Mat mat,PetscInt m,const PetscInt idxm[],PetscInt n,const PetscInt idxn[],const PetscScalar v[],InsertMode addv)
1858 {
1859   PetscErrorCode ierr;
1860 
1861   PetscFunctionBeginHot;
1862   PetscValidHeaderSpecific(mat,MAT_CLASSID,1);
1863   PetscValidType(mat,1);
1864   if (!m || !n) PetscFunctionReturn(0); /* no values to insert */
1865   PetscValidIntPointer(idxm,3);
1866   PetscValidIntPointer(idxn,5);
1867   PetscValidScalarPointer(v,6);
1868   MatCheckPreallocated(mat,1);
1869   if (mat->insertmode == NOT_SET_VALUES) {
1870     mat->insertmode = addv;
1871   }
1872 #if defined(PETSC_USE_DEBUG)
1873   else if (mat->insertmode != addv) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONGSTATE,"Cannot mix add values and insert values");
1874   if (mat->factortype) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONGSTATE,"Not for factored matrix");
1875   if (!mat->ops->setvaluesblocked && !mat->ops->setvalues) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_SUP,"Mat type %s",((PetscObject)mat)->type_name);
1876 #endif
1877 
1878   if (mat->assembled) {
1879     mat->was_assembled = PETSC_TRUE;
1880     mat->assembled     = PETSC_FALSE;
1881   }
1882   ierr = PetscLogEventBegin(MAT_SetValues,mat,0,0,0);CHKERRQ(ierr);
1883   if (mat->ops->setvaluesblocked) {
1884     ierr = (*mat->ops->setvaluesblocked)(mat,m,idxm,n,idxn,v,addv);CHKERRQ(ierr);
1885   } else {
1886     PetscInt buf[8192],*bufr=0,*bufc=0,*iidxm,*iidxn;
1887     PetscInt i,j,bs,cbs;
1888     ierr = MatGetBlockSizes(mat,&bs,&cbs);CHKERRQ(ierr);
1889     if (m*bs+n*cbs <= (PetscInt)(sizeof(buf)/sizeof(PetscInt))) {
1890       iidxm = buf; iidxn = buf + m*bs;
1891     } else {
1892       ierr  = PetscMalloc2(m*bs,&bufr,n*cbs,&bufc);CHKERRQ(ierr);
1893       iidxm = bufr; iidxn = bufc;
1894     }
1895     for (i=0; i<m; i++) {
1896       for (j=0; j<bs; j++) {
1897         iidxm[i*bs+j] = bs*idxm[i] + j;
1898       }
1899     }
1900     for (i=0; i<n; i++) {
1901       for (j=0; j<cbs; j++) {
1902         iidxn[i*cbs+j] = cbs*idxn[i] + j;
1903       }
1904     }
1905     ierr = MatSetValues(mat,m*bs,iidxm,n*cbs,iidxn,v,addv);CHKERRQ(ierr);
1906     ierr = PetscFree2(bufr,bufc);CHKERRQ(ierr);
1907   }
1908   ierr = PetscLogEventEnd(MAT_SetValues,mat,0,0,0);CHKERRQ(ierr);
1909 #if defined(PETSC_HAVE_VIENNACL) || defined(PETSC_HAVE_CUDA)
1910   if (mat->valid_GPU_matrix != PETSC_OFFLOAD_UNALLOCATED) {
1911     mat->valid_GPU_matrix = PETSC_OFFLOAD_CPU;
1912   }
1913 #endif
1914   PetscFunctionReturn(0);
1915 }
1916 
1917 /*@
1918    MatGetValues - Gets a block of values from a matrix.
1919 
1920    Not Collective; currently only returns a local block
1921 
1922    Input Parameters:
1923 +  mat - the matrix
1924 .  v - a logically two-dimensional array for storing the values
1925 .  m, idxm - the number of rows and their global indices
1926 -  n, idxn - the number of columns and their global indices
1927 
1928    Notes:
1929    The user must allocate space (m*n PetscScalars) for the values, v.
1930    The values, v, are then returned in a row-oriented format,
1931    analogous to that used by default in MatSetValues().
1932 
1933    MatGetValues() uses 0-based row and column numbers in
1934    Fortran as well as in C.
1935 
1936    MatGetValues() requires that the matrix has been assembled
1937    with MatAssemblyBegin()/MatAssemblyEnd().  Thus, calls to
1938    MatSetValues() and MatGetValues() CANNOT be made in succession
1939    without intermediate matrix assembly.
1940 
1941    Negative row or column indices will be ignored and those locations in v[] will be
1942    left unchanged.
1943 
1944    Level: advanced
1945 
1946    Concepts: matrices^accessing values
1947 
1948 .seealso: MatGetRow(), MatCreateSubMatrices(), MatSetValues()
1949 @*/
1950 PetscErrorCode MatGetValues(Mat mat,PetscInt m,const PetscInt idxm[],PetscInt n,const PetscInt idxn[],PetscScalar v[])
1951 {
1952   PetscErrorCode ierr;
1953 
1954   PetscFunctionBegin;
1955   PetscValidHeaderSpecific(mat,MAT_CLASSID,1);
1956   PetscValidType(mat,1);
1957   if (!m || !n) PetscFunctionReturn(0);
1958   PetscValidIntPointer(idxm,3);
1959   PetscValidIntPointer(idxn,5);
1960   PetscValidScalarPointer(v,6);
1961   if (!mat->assembled) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONGSTATE,"Not for unassembled matrix");
1962   if (mat->factortype) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONGSTATE,"Not for factored matrix");
1963   if (!mat->ops->getvalues) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_SUP,"Mat type %s",((PetscObject)mat)->type_name);
1964   MatCheckPreallocated(mat,1);
1965 
1966   ierr = PetscLogEventBegin(MAT_GetValues,mat,0,0,0);CHKERRQ(ierr);
1967   ierr = (*mat->ops->getvalues)(mat,m,idxm,n,idxn,v);CHKERRQ(ierr);
1968   ierr = PetscLogEventEnd(MAT_GetValues,mat,0,0,0);CHKERRQ(ierr);
1969   PetscFunctionReturn(0);
1970 }
1971 
1972 /*@
1973   MatSetValuesBatch - Adds (ADD_VALUES) many blocks of values into a matrix at once. The blocks must all be square and
1974   the same size. Currently, this can only be called once and creates the given matrix.
1975 
1976   Not Collective
1977 
1978   Input Parameters:
1979 + mat - the matrix
1980 . nb - the number of blocks
1981 . bs - the number of rows (and columns) in each block
1982 . rows - a concatenation of the rows for each block
1983 - v - a concatenation of logically two-dimensional arrays of values
1984 
1985   Notes:
1986   In the future, we will extend this routine to handle rectangular blocks, and to allow multiple calls for a given matrix.
1987 
1988   Level: advanced
1989 
1990   Concepts: matrices^putting entries in
1991 
1992 .seealso: MatSetOption(), MatAssemblyBegin(), MatAssemblyEnd(), MatSetValuesBlocked(), MatSetValuesLocal(),
1993           InsertMode, INSERT_VALUES, ADD_VALUES, MatSetValues()
1994 @*/
1995 PetscErrorCode MatSetValuesBatch(Mat mat, PetscInt nb, PetscInt bs, PetscInt rows[], const PetscScalar v[])
1996 {
1997   PetscErrorCode ierr;
1998 
1999   PetscFunctionBegin;
2000   PetscValidHeaderSpecific(mat,MAT_CLASSID,1);
2001   PetscValidType(mat,1);
2002   PetscValidScalarPointer(rows,4);
2003   PetscValidScalarPointer(v,5);
2004 #if defined(PETSC_USE_DEBUG)
2005   if (mat->factortype) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONGSTATE,"Not for factored matrix");
2006 #endif
2007 
2008   ierr = PetscLogEventBegin(MAT_SetValuesBatch,mat,0,0,0);CHKERRQ(ierr);
2009   if (mat->ops->setvaluesbatch) {
2010     ierr = (*mat->ops->setvaluesbatch)(mat,nb,bs,rows,v);CHKERRQ(ierr);
2011   } else {
2012     PetscInt b;
2013     for (b = 0; b < nb; ++b) {
2014       ierr = MatSetValues(mat, bs, &rows[b*bs], bs, &rows[b*bs], &v[b*bs*bs], ADD_VALUES);CHKERRQ(ierr);
2015     }
2016   }
2017   ierr = PetscLogEventEnd(MAT_SetValuesBatch,mat,0,0,0);CHKERRQ(ierr);
2018   PetscFunctionReturn(0);
2019 }
2020 
2021 /*@
2022    MatSetLocalToGlobalMapping - Sets a local-to-global numbering for use by
2023    the routine MatSetValuesLocal() to allow users to insert matrix entries
2024    using a local (per-processor) numbering.
2025 
2026    Not Collective
2027 
2028    Input Parameters:
2029 +  x - the matrix
2030 .  rmapping - row mapping created with ISLocalToGlobalMappingCreate()   or ISLocalToGlobalMappingCreateIS()
2031 - cmapping - column mapping
2032 
2033    Level: intermediate
2034 
2035    Concepts: matrices^local to global mapping
2036    Concepts: local to global mapping^for matrices
2037 
2038 .seealso:  MatAssemblyBegin(), MatAssemblyEnd(), MatSetValues(), MatSetValuesLocal()
2039 @*/
2040 PetscErrorCode MatSetLocalToGlobalMapping(Mat x,ISLocalToGlobalMapping rmapping,ISLocalToGlobalMapping cmapping)
2041 {
2042   PetscErrorCode ierr;
2043 
2044   PetscFunctionBegin;
2045   PetscValidHeaderSpecific(x,MAT_CLASSID,1);
2046   PetscValidType(x,1);
2047   PetscValidHeaderSpecific(rmapping,IS_LTOGM_CLASSID,2);
2048   PetscValidHeaderSpecific(cmapping,IS_LTOGM_CLASSID,3);
2049 
2050   if (x->ops->setlocaltoglobalmapping) {
2051     ierr = (*x->ops->setlocaltoglobalmapping)(x,rmapping,cmapping);CHKERRQ(ierr);
2052   } else {
2053     ierr = PetscLayoutSetISLocalToGlobalMapping(x->rmap,rmapping);CHKERRQ(ierr);
2054     ierr = PetscLayoutSetISLocalToGlobalMapping(x->cmap,cmapping);CHKERRQ(ierr);
2055   }
2056   PetscFunctionReturn(0);
2057 }
2058 
2059 
2060 /*@
2061    MatGetLocalToGlobalMapping - Gets the local-to-global numbering set by MatSetLocalToGlobalMapping()
2062 
2063    Not Collective
2064 
2065    Input Parameters:
2066 .  A - the matrix
2067 
2068    Output Parameters:
2069 + rmapping - row mapping
2070 - cmapping - column mapping
2071 
2072    Level: advanced
2073 
2074    Concepts: matrices^local to global mapping
2075    Concepts: local to global mapping^for matrices
2076 
2077 .seealso:  MatSetValuesLocal()
2078 @*/
2079 PetscErrorCode MatGetLocalToGlobalMapping(Mat A,ISLocalToGlobalMapping *rmapping,ISLocalToGlobalMapping *cmapping)
2080 {
2081   PetscFunctionBegin;
2082   PetscValidHeaderSpecific(A,MAT_CLASSID,1);
2083   PetscValidType(A,1);
2084   if (rmapping) PetscValidPointer(rmapping,2);
2085   if (cmapping) PetscValidPointer(cmapping,3);
2086   if (rmapping) *rmapping = A->rmap->mapping;
2087   if (cmapping) *cmapping = A->cmap->mapping;
2088   PetscFunctionReturn(0);
2089 }
2090 
2091 /*@
2092    MatGetLayouts - Gets the PetscLayout objects for rows and columns
2093 
2094    Not Collective
2095 
2096    Input Parameters:
2097 .  A - the matrix
2098 
2099    Output Parameters:
2100 + rmap - row layout
2101 - cmap - column layout
2102 
2103    Level: advanced
2104 
2105 .seealso:  MatCreateVecs(), MatGetLocalToGlobalMapping()
2106 @*/
2107 PetscErrorCode MatGetLayouts(Mat A,PetscLayout *rmap,PetscLayout *cmap)
2108 {
2109   PetscFunctionBegin;
2110   PetscValidHeaderSpecific(A,MAT_CLASSID,1);
2111   PetscValidType(A,1);
2112   if (rmap) PetscValidPointer(rmap,2);
2113   if (cmap) PetscValidPointer(cmap,3);
2114   if (rmap) *rmap = A->rmap;
2115   if (cmap) *cmap = A->cmap;
2116   PetscFunctionReturn(0);
2117 }
2118 
2119 /*@C
2120    MatSetValuesLocal - Inserts or adds values into certain locations of a matrix,
2121    using a local ordering of the nodes.
2122 
2123    Not Collective
2124 
2125    Input Parameters:
2126 +  mat - the matrix
2127 .  nrow, irow - number of rows and their local indices
2128 .  ncol, icol - number of columns and their local indices
2129 .  y -  a logically two-dimensional array of values
2130 -  addv - either INSERT_VALUES or ADD_VALUES, where
2131    ADD_VALUES adds values to any existing entries, and
2132    INSERT_VALUES replaces existing entries with new values
2133 
2134    Notes:
2135    If you create the matrix yourself (that is not with a call to DMCreateMatrix()) then you MUST call MatXXXXSetPreallocation() or
2136       MatSetUp() before using this routine
2137 
2138    If you create the matrix yourself (that is not with a call to DMCreateMatrix()) then you MUST call MatSetLocalToGlobalMapping() before using this routine
2139 
2140    Calls to MatSetValuesLocal() with the INSERT_VALUES and ADD_VALUES
2141    options cannot be mixed without intervening calls to the assembly
2142    routines.
2143 
2144    These values may be cached, so MatAssemblyBegin() and MatAssemblyEnd()
2145    MUST be called after all calls to MatSetValuesLocal() have been completed.
2146 
2147    Level: intermediate
2148 
2149    Concepts: matrices^putting entries in with local numbering
2150 
2151    Developer Notes:
2152     This is labeled with C so does not automatically generate Fortran stubs and interfaces
2153                     because it requires multiple Fortran interfaces depending on which arguments are scalar or arrays.
2154 
2155 .seealso:  MatAssemblyBegin(), MatAssemblyEnd(), MatSetValues(), MatSetLocalToGlobalMapping(),
2156            MatSetValueLocal()
2157 @*/
2158 PetscErrorCode MatSetValuesLocal(Mat mat,PetscInt nrow,const PetscInt irow[],PetscInt ncol,const PetscInt icol[],const PetscScalar y[],InsertMode addv)
2159 {
2160   PetscErrorCode ierr;
2161 
2162   PetscFunctionBeginHot;
2163   PetscValidHeaderSpecific(mat,MAT_CLASSID,1);
2164   PetscValidType(mat,1);
2165   MatCheckPreallocated(mat,1);
2166   if (!nrow || !ncol) PetscFunctionReturn(0); /* no values to insert */
2167   PetscValidIntPointer(irow,3);
2168   PetscValidIntPointer(icol,5);
2169   PetscValidScalarPointer(y,6);
2170   if (mat->insertmode == NOT_SET_VALUES) {
2171     mat->insertmode = addv;
2172   }
2173 #if defined(PETSC_USE_DEBUG)
2174   else if (mat->insertmode != addv) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONGSTATE,"Cannot mix add values and insert values");
2175   if (mat->factortype) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONGSTATE,"Not for factored matrix");
2176   if (!mat->ops->setvalueslocal && !mat->ops->setvalues) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_SUP,"Mat type %s",((PetscObject)mat)->type_name);
2177 #endif
2178 
2179   if (mat->assembled) {
2180     mat->was_assembled = PETSC_TRUE;
2181     mat->assembled     = PETSC_FALSE;
2182   }
2183   ierr = PetscLogEventBegin(MAT_SetValues,mat,0,0,0);CHKERRQ(ierr);
2184   if (mat->ops->setvalueslocal) {
2185     ierr = (*mat->ops->setvalueslocal)(mat,nrow,irow,ncol,icol,y,addv);CHKERRQ(ierr);
2186   } else {
2187     PetscInt buf[8192],*bufr=0,*bufc=0,*irowm,*icolm;
2188     if ((nrow+ncol) <= (PetscInt)(sizeof(buf)/sizeof(PetscInt))) {
2189       irowm = buf; icolm = buf+nrow;
2190     } else {
2191       ierr  = PetscMalloc2(nrow,&bufr,ncol,&bufc);CHKERRQ(ierr);
2192       irowm = bufr; icolm = bufc;
2193     }
2194     ierr = ISLocalToGlobalMappingApply(mat->rmap->mapping,nrow,irow,irowm);CHKERRQ(ierr);
2195     ierr = ISLocalToGlobalMappingApply(mat->cmap->mapping,ncol,icol,icolm);CHKERRQ(ierr);
2196     ierr = MatSetValues(mat,nrow,irowm,ncol,icolm,y,addv);CHKERRQ(ierr);
2197     ierr = PetscFree2(bufr,bufc);CHKERRQ(ierr);
2198   }
2199   ierr = PetscLogEventEnd(MAT_SetValues,mat,0,0,0);CHKERRQ(ierr);
2200 #if defined(PETSC_HAVE_VIENNACL) || defined(PETSC_HAVE_CUDA)
2201   if (mat->valid_GPU_matrix != PETSC_OFFLOAD_UNALLOCATED) {
2202     mat->valid_GPU_matrix = PETSC_OFFLOAD_CPU;
2203   }
2204 #endif
2205   PetscFunctionReturn(0);
2206 }
2207 
2208 /*@C
2209    MatSetValuesBlockedLocal - Inserts or adds values into certain locations of a matrix,
2210    using a local ordering of the nodes a block at a time.
2211 
2212    Not Collective
2213 
2214    Input Parameters:
2215 +  x - the matrix
2216 .  nrow, irow - number of rows and their local indices
2217 .  ncol, icol - number of columns and their local indices
2218 .  y -  a logically two-dimensional array of values
2219 -  addv - either INSERT_VALUES or ADD_VALUES, where
2220    ADD_VALUES adds values to any existing entries, and
2221    INSERT_VALUES replaces existing entries with new values
2222 
2223    Notes:
2224    If you create the matrix yourself (that is not with a call to DMCreateMatrix()) then you MUST call MatXXXXSetPreallocation() or
2225       MatSetUp() before using this routine
2226 
2227    If you create the matrix yourself (that is not with a call to DMCreateMatrix()) then you MUST call MatSetBlockSize() and MatSetLocalToGlobalMapping()
2228       before using this routineBefore calling MatSetValuesLocal(), the user must first set the
2229 
2230    Calls to MatSetValuesBlockedLocal() with the INSERT_VALUES and ADD_VALUES
2231    options cannot be mixed without intervening calls to the assembly
2232    routines.
2233 
2234    These values may be cached, so MatAssemblyBegin() and MatAssemblyEnd()
2235    MUST be called after all calls to MatSetValuesBlockedLocal() have been completed.
2236 
2237    Level: intermediate
2238 
2239    Developer Notes:
2240     This is labeled with C so does not automatically generate Fortran stubs and interfaces
2241                     because it requires multiple Fortran interfaces depending on which arguments are scalar or arrays.
2242 
2243    Concepts: matrices^putting blocked values in with local numbering
2244 
2245 .seealso:  MatSetBlockSize(), MatSetLocalToGlobalMapping(), MatAssemblyBegin(), MatAssemblyEnd(),
2246            MatSetValuesLocal(),  MatSetValuesBlocked()
2247 @*/
2248 PetscErrorCode MatSetValuesBlockedLocal(Mat mat,PetscInt nrow,const PetscInt irow[],PetscInt ncol,const PetscInt icol[],const PetscScalar y[],InsertMode addv)
2249 {
2250   PetscErrorCode ierr;
2251 
2252   PetscFunctionBeginHot;
2253   PetscValidHeaderSpecific(mat,MAT_CLASSID,1);
2254   PetscValidType(mat,1);
2255   MatCheckPreallocated(mat,1);
2256   if (!nrow || !ncol) PetscFunctionReturn(0); /* no values to insert */
2257   PetscValidIntPointer(irow,3);
2258   PetscValidIntPointer(icol,5);
2259   PetscValidScalarPointer(y,6);
2260   if (mat->insertmode == NOT_SET_VALUES) {
2261     mat->insertmode = addv;
2262   }
2263 #if defined(PETSC_USE_DEBUG)
2264   else if (mat->insertmode != addv) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONGSTATE,"Cannot mix add values and insert values");
2265   if (mat->factortype) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONGSTATE,"Not for factored matrix");
2266   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);
2267 #endif
2268 
2269   if (mat->assembled) {
2270     mat->was_assembled = PETSC_TRUE;
2271     mat->assembled     = PETSC_FALSE;
2272   }
2273   ierr = PetscLogEventBegin(MAT_SetValues,mat,0,0,0);CHKERRQ(ierr);
2274   if (mat->ops->setvaluesblockedlocal) {
2275     ierr = (*mat->ops->setvaluesblockedlocal)(mat,nrow,irow,ncol,icol,y,addv);CHKERRQ(ierr);
2276   } else {
2277     PetscInt buf[8192],*bufr=0,*bufc=0,*irowm,*icolm;
2278     if ((nrow+ncol) <= (PetscInt)(sizeof(buf)/sizeof(PetscInt))) {
2279       irowm = buf; icolm = buf + nrow;
2280     } else {
2281       ierr  = PetscMalloc2(nrow,&bufr,ncol,&bufc);CHKERRQ(ierr);
2282       irowm = bufr; icolm = bufc;
2283     }
2284     ierr = ISLocalToGlobalMappingApplyBlock(mat->rmap->mapping,nrow,irow,irowm);CHKERRQ(ierr);
2285     ierr = ISLocalToGlobalMappingApplyBlock(mat->cmap->mapping,ncol,icol,icolm);CHKERRQ(ierr);
2286     ierr = MatSetValuesBlocked(mat,nrow,irowm,ncol,icolm,y,addv);CHKERRQ(ierr);
2287     ierr = PetscFree2(bufr,bufc);CHKERRQ(ierr);
2288   }
2289   ierr = PetscLogEventEnd(MAT_SetValues,mat,0,0,0);CHKERRQ(ierr);
2290 #if defined(PETSC_HAVE_VIENNACL) || defined(PETSC_HAVE_CUDA)
2291   if (mat->valid_GPU_matrix != PETSC_OFFLOAD_UNALLOCATED) {
2292     mat->valid_GPU_matrix = PETSC_OFFLOAD_CPU;
2293   }
2294 #endif
2295   PetscFunctionReturn(0);
2296 }
2297 
2298 /*@
2299    MatMultDiagonalBlock - Computes the matrix-vector product, y = Dx. Where D is defined by the inode or block structure of the diagonal
2300 
2301    Collective on Mat and Vec
2302 
2303    Input Parameters:
2304 +  mat - the matrix
2305 -  x   - the vector to be multiplied
2306 
2307    Output Parameters:
2308 .  y - the result
2309 
2310    Notes:
2311    The vectors x and y cannot be the same.  I.e., one cannot
2312    call MatMult(A,y,y).
2313 
2314    Level: developer
2315 
2316    Concepts: matrix-vector product
2317 
2318 .seealso: MatMultTranspose(), MatMultAdd(), MatMultTransposeAdd()
2319 @*/
2320 PetscErrorCode MatMultDiagonalBlock(Mat mat,Vec x,Vec y)
2321 {
2322   PetscErrorCode ierr;
2323 
2324   PetscFunctionBegin;
2325   PetscValidHeaderSpecific(mat,MAT_CLASSID,1);
2326   PetscValidType(mat,1);
2327   PetscValidHeaderSpecific(x,VEC_CLASSID,2);
2328   PetscValidHeaderSpecific(y,VEC_CLASSID,3);
2329 
2330   if (!mat->assembled) SETERRQ(PetscObjectComm((PetscObject)mat),PETSC_ERR_ARG_WRONGSTATE,"Not for unassembled matrix");
2331   if (mat->factortype) SETERRQ(PetscObjectComm((PetscObject)mat),PETSC_ERR_ARG_WRONGSTATE,"Not for factored matrix");
2332   if (x == y) SETERRQ(PetscObjectComm((PetscObject)mat),PETSC_ERR_ARG_WRONGSTATE,"x and y must be different vectors");
2333   MatCheckPreallocated(mat,1);
2334 
2335   if (!mat->ops->multdiagonalblock) SETERRQ(PetscObjectComm((PetscObject)mat),PETSC_ERR_SUP,"This matrix type does not have a multiply defined");
2336   ierr = (*mat->ops->multdiagonalblock)(mat,x,y);CHKERRQ(ierr);
2337   ierr = PetscObjectStateIncrease((PetscObject)y);CHKERRQ(ierr);
2338   PetscFunctionReturn(0);
2339 }
2340 
2341 /* --------------------------------------------------------*/
2342 /*@
2343    MatMult - Computes the matrix-vector product, y = Ax.
2344 
2345    Neighbor-wise Collective on Mat and Vec
2346 
2347    Input Parameters:
2348 +  mat - the matrix
2349 -  x   - the vector to be multiplied
2350 
2351    Output Parameters:
2352 .  y - the result
2353 
2354    Notes:
2355    The vectors x and y cannot be the same.  I.e., one cannot
2356    call MatMult(A,y,y).
2357 
2358    Level: beginner
2359 
2360    Concepts: matrix-vector product
2361 
2362 .seealso: MatMultTranspose(), MatMultAdd(), MatMultTransposeAdd()
2363 @*/
2364 PetscErrorCode MatMult(Mat mat,Vec x,Vec y)
2365 {
2366   PetscErrorCode ierr;
2367 
2368   PetscFunctionBegin;
2369   PetscValidHeaderSpecific(mat,MAT_CLASSID,1);
2370   PetscValidType(mat,1);
2371   PetscValidHeaderSpecific(x,VEC_CLASSID,2);
2372   PetscValidHeaderSpecific(y,VEC_CLASSID,3);
2373   if (!mat->assembled) SETERRQ(PetscObjectComm((PetscObject)mat),PETSC_ERR_ARG_WRONGSTATE,"Not for unassembled matrix");
2374   if (mat->factortype) SETERRQ(PetscObjectComm((PetscObject)mat),PETSC_ERR_ARG_WRONGSTATE,"Not for factored matrix");
2375   if (x == y) SETERRQ(PetscObjectComm((PetscObject)mat),PETSC_ERR_ARG_WRONGSTATE,"x and y must be different vectors");
2376 #if !defined(PETSC_HAVE_CONSTRAINTS)
2377   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);
2378   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);
2379   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);
2380 #endif
2381   ierr = VecSetErrorIfLocked(y,3);CHKERRQ(ierr);
2382   if (mat->erroriffailure) {ierr = VecValidValues(x,2,PETSC_TRUE);CHKERRQ(ierr);}
2383   MatCheckPreallocated(mat,1);
2384 
2385   ierr = VecLockReadPush(x);CHKERRQ(ierr);
2386   if (!mat->ops->mult) SETERRQ(PetscObjectComm((PetscObject)mat),PETSC_ERR_SUP,"This matrix type does not have a multiply defined");
2387   ierr = PetscLogEventBegin(MAT_Mult,mat,x,y,0);CHKERRQ(ierr);
2388   ierr = (*mat->ops->mult)(mat,x,y);CHKERRQ(ierr);
2389   ierr = PetscLogEventEnd(MAT_Mult,mat,x,y,0);CHKERRQ(ierr);
2390   if (mat->erroriffailure) {ierr = VecValidValues(y,3,PETSC_FALSE);CHKERRQ(ierr);}
2391   ierr = VecLockReadPop(x);CHKERRQ(ierr);
2392   PetscFunctionReturn(0);
2393 }
2394 
2395 /*@
2396    MatMultTranspose - Computes matrix transpose times a vector y = A^T * x.
2397 
2398    Neighbor-wise Collective on Mat and Vec
2399 
2400    Input Parameters:
2401 +  mat - the matrix
2402 -  x   - the vector to be multiplied
2403 
2404    Output Parameters:
2405 .  y - the result
2406 
2407    Notes:
2408    The vectors x and y cannot be the same.  I.e., one cannot
2409    call MatMultTranspose(A,y,y).
2410 
2411    For complex numbers this does NOT compute the Hermitian (complex conjugate) transpose multiple,
2412    use MatMultHermitianTranspose()
2413 
2414    Level: beginner
2415 
2416    Concepts: matrix vector product^transpose
2417 
2418 .seealso: MatMult(), MatMultAdd(), MatMultTransposeAdd(), MatMultHermitianTranspose(), MatTranspose()
2419 @*/
2420 PetscErrorCode MatMultTranspose(Mat mat,Vec x,Vec y)
2421 {
2422   PetscErrorCode ierr;
2423 
2424   PetscFunctionBegin;
2425   PetscValidHeaderSpecific(mat,MAT_CLASSID,1);
2426   PetscValidType(mat,1);
2427   PetscValidHeaderSpecific(x,VEC_CLASSID,2);
2428   PetscValidHeaderSpecific(y,VEC_CLASSID,3);
2429 
2430   if (!mat->assembled) SETERRQ(PetscObjectComm((PetscObject)mat),PETSC_ERR_ARG_WRONGSTATE,"Not for unassembled matrix");
2431   if (mat->factortype) SETERRQ(PetscObjectComm((PetscObject)mat),PETSC_ERR_ARG_WRONGSTATE,"Not for factored matrix");
2432   if (x == y) SETERRQ(PetscObjectComm((PetscObject)mat),PETSC_ERR_ARG_WRONGSTATE,"x and y must be different vectors");
2433 #if !defined(PETSC_HAVE_CONSTRAINTS)
2434   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);
2435   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);
2436 #endif
2437   if (mat->erroriffailure) {ierr = VecValidValues(x,2,PETSC_TRUE);CHKERRQ(ierr);}
2438   MatCheckPreallocated(mat,1);
2439 
2440   if (!mat->ops->multtranspose) SETERRQ(PetscObjectComm((PetscObject)mat),PETSC_ERR_SUP,"This matrix type does not have a multiply transpose defined");
2441   ierr = PetscLogEventBegin(MAT_MultTranspose,mat,x,y,0);CHKERRQ(ierr);
2442   ierr = VecLockReadPush(x);CHKERRQ(ierr);
2443   ierr = (*mat->ops->multtranspose)(mat,x,y);CHKERRQ(ierr);
2444   ierr = VecLockReadPop(x);CHKERRQ(ierr);
2445   ierr = PetscLogEventEnd(MAT_MultTranspose,mat,x,y,0);CHKERRQ(ierr);
2446   ierr = PetscObjectStateIncrease((PetscObject)y);CHKERRQ(ierr);
2447   if (mat->erroriffailure) {ierr = VecValidValues(y,3,PETSC_FALSE);CHKERRQ(ierr);}
2448   PetscFunctionReturn(0);
2449 }
2450 
2451 /*@
2452    MatMultHermitianTranspose - Computes matrix Hermitian transpose times a vector.
2453 
2454    Neighbor-wise Collective on Mat and Vec
2455 
2456    Input Parameters:
2457 +  mat - the matrix
2458 -  x   - the vector to be multilplied
2459 
2460    Output Parameters:
2461 .  y - the result
2462 
2463    Notes:
2464    The vectors x and y cannot be the same.  I.e., one cannot
2465    call MatMultHermitianTranspose(A,y,y).
2466 
2467    Also called the conjugate transpose, complex conjugate transpose, or adjoint.
2468 
2469    For real numbers MatMultTranspose() and MatMultHermitianTranspose() are identical.
2470 
2471    Level: beginner
2472 
2473    Concepts: matrix vector product^transpose
2474 
2475 .seealso: MatMult(), MatMultAdd(), MatMultHermitianTransposeAdd(), MatMultTranspose()
2476 @*/
2477 PetscErrorCode MatMultHermitianTranspose(Mat mat,Vec x,Vec y)
2478 {
2479   PetscErrorCode ierr;
2480   Vec            w;
2481 
2482   PetscFunctionBegin;
2483   PetscValidHeaderSpecific(mat,MAT_CLASSID,1);
2484   PetscValidType(mat,1);
2485   PetscValidHeaderSpecific(x,VEC_CLASSID,2);
2486   PetscValidHeaderSpecific(y,VEC_CLASSID,3);
2487 
2488   if (!mat->assembled) SETERRQ(PetscObjectComm((PetscObject)mat),PETSC_ERR_ARG_WRONGSTATE,"Not for unassembled matrix");
2489   if (mat->factortype) SETERRQ(PetscObjectComm((PetscObject)mat),PETSC_ERR_ARG_WRONGSTATE,"Not for factored matrix");
2490   if (x == y) SETERRQ(PetscObjectComm((PetscObject)mat),PETSC_ERR_ARG_WRONGSTATE,"x and y must be different vectors");
2491 #if !defined(PETSC_HAVE_CONSTRAINTS)
2492   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);
2493   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);
2494 #endif
2495   MatCheckPreallocated(mat,1);
2496 
2497   ierr = PetscLogEventBegin(MAT_MultHermitianTranspose,mat,x,y,0);CHKERRQ(ierr);
2498   if (mat->ops->multhermitiantranspose) {
2499     ierr = VecLockReadPush(x);CHKERRQ(ierr);
2500     ierr = (*mat->ops->multhermitiantranspose)(mat,x,y);CHKERRQ(ierr);
2501     ierr = VecLockReadPop(x);CHKERRQ(ierr);
2502   } else {
2503     ierr = VecDuplicate(x,&w);CHKERRQ(ierr);
2504     ierr = VecCopy(x,w);CHKERRQ(ierr);
2505     ierr = VecConjugate(w);CHKERRQ(ierr);
2506     ierr = MatMultTranspose(mat,w,y);CHKERRQ(ierr);
2507     ierr = VecDestroy(&w);CHKERRQ(ierr);
2508     ierr = VecConjugate(y);CHKERRQ(ierr);
2509   }
2510   ierr = PetscLogEventEnd(MAT_MultHermitianTranspose,mat,x,y,0);CHKERRQ(ierr);
2511   ierr = PetscObjectStateIncrease((PetscObject)y);CHKERRQ(ierr);
2512   PetscFunctionReturn(0);
2513 }
2514 
2515 /*@
2516     MatMultAdd -  Computes v3 = v2 + A * v1.
2517 
2518     Neighbor-wise Collective on Mat and Vec
2519 
2520     Input Parameters:
2521 +   mat - the matrix
2522 -   v1, v2 - the vectors
2523 
2524     Output Parameters:
2525 .   v3 - the result
2526 
2527     Notes:
2528     The vectors v1 and v3 cannot be the same.  I.e., one cannot
2529     call MatMultAdd(A,v1,v2,v1).
2530 
2531     Level: beginner
2532 
2533     Concepts: matrix vector product^addition
2534 
2535 .seealso: MatMultTranspose(), MatMult(), MatMultTransposeAdd()
2536 @*/
2537 PetscErrorCode MatMultAdd(Mat mat,Vec v1,Vec v2,Vec v3)
2538 {
2539   PetscErrorCode ierr;
2540 
2541   PetscFunctionBegin;
2542   PetscValidHeaderSpecific(mat,MAT_CLASSID,1);
2543   PetscValidType(mat,1);
2544   PetscValidHeaderSpecific(v1,VEC_CLASSID,2);
2545   PetscValidHeaderSpecific(v2,VEC_CLASSID,3);
2546   PetscValidHeaderSpecific(v3,VEC_CLASSID,4);
2547 
2548   if (!mat->assembled) SETERRQ(PetscObjectComm((PetscObject)mat),PETSC_ERR_ARG_WRONGSTATE,"Not for unassembled matrix");
2549   if (mat->factortype) SETERRQ(PetscObjectComm((PetscObject)mat),PETSC_ERR_ARG_WRONGSTATE,"Not for factored matrix");
2550   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);
2551   /* 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);
2552      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); */
2553   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);
2554   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);
2555   if (v1 == v3) SETERRQ(PetscObjectComm((PetscObject)mat),PETSC_ERR_ARG_IDN,"v1 and v3 must be different vectors");
2556   MatCheckPreallocated(mat,1);
2557 
2558   if (!mat->ops->multadd) SETERRQ1(PetscObjectComm((PetscObject)mat),PETSC_ERR_SUP,"No MatMultAdd() for matrix type '%s'",((PetscObject)mat)->type_name);
2559   ierr = PetscLogEventBegin(MAT_MultAdd,mat,v1,v2,v3);CHKERRQ(ierr);
2560   ierr = VecLockReadPush(v1);CHKERRQ(ierr);
2561   ierr = (*mat->ops->multadd)(mat,v1,v2,v3);CHKERRQ(ierr);
2562   ierr = VecLockReadPop(v1);CHKERRQ(ierr);
2563   ierr = PetscLogEventEnd(MAT_MultAdd,mat,v1,v2,v3);CHKERRQ(ierr);
2564   ierr = PetscObjectStateIncrease((PetscObject)v3);CHKERRQ(ierr);
2565   PetscFunctionReturn(0);
2566 }
2567 
2568 /*@
2569    MatMultTransposeAdd - Computes v3 = v2 + A' * v1.
2570 
2571    Neighbor-wise Collective on Mat and Vec
2572 
2573    Input Parameters:
2574 +  mat - the matrix
2575 -  v1, v2 - the vectors
2576 
2577    Output Parameters:
2578 .  v3 - the result
2579 
2580    Notes:
2581    The vectors v1 and v3 cannot be the same.  I.e., one cannot
2582    call MatMultTransposeAdd(A,v1,v2,v1).
2583 
2584    Level: beginner
2585 
2586    Concepts: matrix vector product^transpose and addition
2587 
2588 .seealso: MatMultTranspose(), MatMultAdd(), MatMult()
2589 @*/
2590 PetscErrorCode MatMultTransposeAdd(Mat mat,Vec v1,Vec v2,Vec v3)
2591 {
2592   PetscErrorCode ierr;
2593 
2594   PetscFunctionBegin;
2595   PetscValidHeaderSpecific(mat,MAT_CLASSID,1);
2596   PetscValidType(mat,1);
2597   PetscValidHeaderSpecific(v1,VEC_CLASSID,2);
2598   PetscValidHeaderSpecific(v2,VEC_CLASSID,3);
2599   PetscValidHeaderSpecific(v3,VEC_CLASSID,4);
2600 
2601   if (!mat->assembled) SETERRQ(PetscObjectComm((PetscObject)mat),PETSC_ERR_ARG_WRONGSTATE,"Not for unassembled matrix");
2602   if (mat->factortype) SETERRQ(PetscObjectComm((PetscObject)mat),PETSC_ERR_ARG_WRONGSTATE,"Not for factored matrix");
2603   if (!mat->ops->multtransposeadd) SETERRQ1(PetscObjectComm((PetscObject)mat),PETSC_ERR_SUP,"Mat type %s",((PetscObject)mat)->type_name);
2604   if (v1 == v3) SETERRQ(PetscObjectComm((PetscObject)mat),PETSC_ERR_ARG_IDN,"v1 and v3 must be different vectors");
2605   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);
2606   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);
2607   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);
2608   MatCheckPreallocated(mat,1);
2609 
2610   ierr = PetscLogEventBegin(MAT_MultTransposeAdd,mat,v1,v2,v3);CHKERRQ(ierr);
2611   ierr = VecLockReadPush(v1);CHKERRQ(ierr);
2612   ierr = (*mat->ops->multtransposeadd)(mat,v1,v2,v3);CHKERRQ(ierr);
2613   ierr = VecLockReadPop(v1);CHKERRQ(ierr);
2614   ierr = PetscLogEventEnd(MAT_MultTransposeAdd,mat,v1,v2,v3);CHKERRQ(ierr);
2615   ierr = PetscObjectStateIncrease((PetscObject)v3);CHKERRQ(ierr);
2616   PetscFunctionReturn(0);
2617 }
2618 
2619 /*@
2620    MatMultHermitianTransposeAdd - Computes v3 = v2 + A^H * v1.
2621 
2622    Neighbor-wise Collective on Mat and Vec
2623 
2624    Input Parameters:
2625 +  mat - the matrix
2626 -  v1, v2 - the vectors
2627 
2628    Output Parameters:
2629 .  v3 - the result
2630 
2631    Notes:
2632    The vectors v1 and v3 cannot be the same.  I.e., one cannot
2633    call MatMultHermitianTransposeAdd(A,v1,v2,v1).
2634 
2635    Level: beginner
2636 
2637    Concepts: matrix vector product^transpose and addition
2638 
2639 .seealso: MatMultHermitianTranspose(), MatMultTranspose(), MatMultAdd(), MatMult()
2640 @*/
2641 PetscErrorCode MatMultHermitianTransposeAdd(Mat mat,Vec v1,Vec v2,Vec v3)
2642 {
2643   PetscErrorCode ierr;
2644 
2645   PetscFunctionBegin;
2646   PetscValidHeaderSpecific(mat,MAT_CLASSID,1);
2647   PetscValidType(mat,1);
2648   PetscValidHeaderSpecific(v1,VEC_CLASSID,2);
2649   PetscValidHeaderSpecific(v2,VEC_CLASSID,3);
2650   PetscValidHeaderSpecific(v3,VEC_CLASSID,4);
2651 
2652   if (!mat->assembled) SETERRQ(PetscObjectComm((PetscObject)mat),PETSC_ERR_ARG_WRONGSTATE,"Not for unassembled matrix");
2653   if (mat->factortype) SETERRQ(PetscObjectComm((PetscObject)mat),PETSC_ERR_ARG_WRONGSTATE,"Not for factored matrix");
2654   if (v1 == v3) SETERRQ(PetscObjectComm((PetscObject)mat),PETSC_ERR_ARG_IDN,"v1 and v3 must be different vectors");
2655   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);
2656   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);
2657   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);
2658   MatCheckPreallocated(mat,1);
2659 
2660   ierr = PetscLogEventBegin(MAT_MultHermitianTransposeAdd,mat,v1,v2,v3);CHKERRQ(ierr);
2661   ierr = VecLockReadPush(v1);CHKERRQ(ierr);
2662   if (mat->ops->multhermitiantransposeadd) {
2663     ierr = (*mat->ops->multhermitiantransposeadd)(mat,v1,v2,v3);CHKERRQ(ierr);
2664   } else {
2665     Vec w,z;
2666     ierr = VecDuplicate(v1,&w);CHKERRQ(ierr);
2667     ierr = VecCopy(v1,w);CHKERRQ(ierr);
2668     ierr = VecConjugate(w);CHKERRQ(ierr);
2669     ierr = VecDuplicate(v3,&z);CHKERRQ(ierr);
2670     ierr = MatMultTranspose(mat,w,z);CHKERRQ(ierr);
2671     ierr = VecDestroy(&w);CHKERRQ(ierr);
2672     ierr = VecConjugate(z);CHKERRQ(ierr);
2673     if (v2 != v3) {
2674       ierr = VecWAXPY(v3,1.0,v2,z);CHKERRQ(ierr);
2675     } else {
2676       ierr = VecAXPY(v3,1.0,z);CHKERRQ(ierr);
2677     }
2678     ierr = VecDestroy(&z);CHKERRQ(ierr);
2679   }
2680   ierr = VecLockReadPop(v1);CHKERRQ(ierr);
2681   ierr = PetscLogEventEnd(MAT_MultHermitianTransposeAdd,mat,v1,v2,v3);CHKERRQ(ierr);
2682   ierr = PetscObjectStateIncrease((PetscObject)v3);CHKERRQ(ierr);
2683   PetscFunctionReturn(0);
2684 }
2685 
2686 /*@
2687    MatMultConstrained - The inner multiplication routine for a
2688    constrained matrix P^T A P.
2689 
2690    Neighbor-wise Collective on Mat and Vec
2691 
2692    Input Parameters:
2693 +  mat - the matrix
2694 -  x   - the vector to be multilplied
2695 
2696    Output Parameters:
2697 .  y - the result
2698 
2699    Notes:
2700    The vectors x and y cannot be the same.  I.e., one cannot
2701    call MatMult(A,y,y).
2702 
2703    Level: beginner
2704 
2705 .keywords: matrix, multiply, matrix-vector product, constraint
2706 .seealso: MatMult(), MatMultTranspose(), MatMultAdd(), MatMultTransposeAdd()
2707 @*/
2708 PetscErrorCode MatMultConstrained(Mat mat,Vec x,Vec y)
2709 {
2710   PetscErrorCode ierr;
2711 
2712   PetscFunctionBegin;
2713   PetscValidHeaderSpecific(mat,MAT_CLASSID,1);
2714   PetscValidHeaderSpecific(x,VEC_CLASSID,2);
2715   PetscValidHeaderSpecific(y,VEC_CLASSID,3);
2716   if (!mat->assembled) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONGSTATE,"Not for unassembled matrix");
2717   if (mat->factortype) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONGSTATE,"Not for factored matrix");
2718   if (x == y) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONGSTATE,"x and y must be different vectors");
2719   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);
2720   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);
2721   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);
2722 
2723   ierr = PetscLogEventBegin(MAT_MultConstrained,mat,x,y,0);CHKERRQ(ierr);
2724   ierr = VecLockReadPush(x);CHKERRQ(ierr);
2725   ierr = (*mat->ops->multconstrained)(mat,x,y);CHKERRQ(ierr);
2726   ierr = VecLockReadPop(x);CHKERRQ(ierr);
2727   ierr = PetscLogEventEnd(MAT_MultConstrained,mat,x,y,0);CHKERRQ(ierr);
2728   ierr = PetscObjectStateIncrease((PetscObject)y);CHKERRQ(ierr);
2729   PetscFunctionReturn(0);
2730 }
2731 
2732 /*@
2733    MatMultTransposeConstrained - The inner multiplication routine for a
2734    constrained matrix P^T A^T P.
2735 
2736    Neighbor-wise Collective on Mat and Vec
2737 
2738    Input Parameters:
2739 +  mat - the matrix
2740 -  x   - the vector to be multilplied
2741 
2742    Output Parameters:
2743 .  y - the result
2744 
2745    Notes:
2746    The vectors x and y cannot be the same.  I.e., one cannot
2747    call MatMult(A,y,y).
2748 
2749    Level: beginner
2750 
2751 .keywords: matrix, multiply, matrix-vector product, constraint
2752 .seealso: MatMult(), MatMultTranspose(), MatMultAdd(), MatMultTransposeAdd()
2753 @*/
2754 PetscErrorCode MatMultTransposeConstrained(Mat mat,Vec x,Vec y)
2755 {
2756   PetscErrorCode ierr;
2757 
2758   PetscFunctionBegin;
2759   PetscValidHeaderSpecific(mat,MAT_CLASSID,1);
2760   PetscValidHeaderSpecific(x,VEC_CLASSID,2);
2761   PetscValidHeaderSpecific(y,VEC_CLASSID,3);
2762   if (!mat->assembled) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONGSTATE,"Not for unassembled matrix");
2763   if (mat->factortype) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONGSTATE,"Not for factored matrix");
2764   if (x == y) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONGSTATE,"x and y must be different vectors");
2765   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);
2766   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);
2767 
2768   ierr = PetscLogEventBegin(MAT_MultConstrained,mat,x,y,0);CHKERRQ(ierr);
2769   ierr = (*mat->ops->multtransposeconstrained)(mat,x,y);CHKERRQ(ierr);
2770   ierr = PetscLogEventEnd(MAT_MultConstrained,mat,x,y,0);CHKERRQ(ierr);
2771   ierr = PetscObjectStateIncrease((PetscObject)y);CHKERRQ(ierr);
2772   PetscFunctionReturn(0);
2773 }
2774 
2775 /*@C
2776    MatGetFactorType - gets the type of factorization it is
2777 
2778    Not Collective
2779 
2780    Input Parameters:
2781 .  mat - the matrix
2782 
2783    Output Parameters:
2784 .  t - the type, one of MAT_FACTOR_NONE, MAT_FACTOR_LU, MAT_FACTOR_CHOLESKY, MAT_FACTOR_ILU, MAT_FACTOR_ICC,MAT_FACTOR_ILUDT
2785 
2786    Level: intermediate
2787 
2788 .seealso: MatFactorType, MatGetFactor(), MatSetFactorType()
2789 @*/
2790 PetscErrorCode MatGetFactorType(Mat mat,MatFactorType *t)
2791 {
2792   PetscFunctionBegin;
2793   PetscValidHeaderSpecific(mat,MAT_CLASSID,1);
2794   PetscValidType(mat,1);
2795   PetscValidPointer(t,2);
2796   *t = mat->factortype;
2797   PetscFunctionReturn(0);
2798 }
2799 
2800 /*@C
2801    MatSetFactorType - sets the type of factorization it is
2802 
2803    Logically Collective on Mat
2804 
2805    Input Parameters:
2806 +  mat - the matrix
2807 -  t - the type, one of MAT_FACTOR_NONE, MAT_FACTOR_LU, MAT_FACTOR_CHOLESKY, MAT_FACTOR_ILU, MAT_FACTOR_ICC,MAT_FACTOR_ILUDT
2808 
2809    Level: intermediate
2810 
2811 .seealso: MatFactorType, MatGetFactor(), MatGetFactorType()
2812 @*/
2813 PetscErrorCode MatSetFactorType(Mat mat, MatFactorType t)
2814 {
2815   PetscFunctionBegin;
2816   PetscValidHeaderSpecific(mat,MAT_CLASSID,1);
2817   PetscValidType(mat,1);
2818   mat->factortype = t;
2819   PetscFunctionReturn(0);
2820 }
2821 
2822 /* ------------------------------------------------------------*/
2823 /*@C
2824    MatGetInfo - Returns information about matrix storage (number of
2825    nonzeros, memory, etc.).
2826 
2827    Collective on Mat if MAT_GLOBAL_MAX or MAT_GLOBAL_SUM is used as the flag
2828 
2829    Input Parameters:
2830 .  mat - the matrix
2831 
2832    Output Parameters:
2833 +  flag - flag indicating the type of parameters to be returned
2834    (MAT_LOCAL - local matrix, MAT_GLOBAL_MAX - maximum over all processors,
2835    MAT_GLOBAL_SUM - sum over all processors)
2836 -  info - matrix information context
2837 
2838    Notes:
2839    The MatInfo context contains a variety of matrix data, including
2840    number of nonzeros allocated and used, number of mallocs during
2841    matrix assembly, etc.  Additional information for factored matrices
2842    is provided (such as the fill ratio, number of mallocs during
2843    factorization, etc.).  Much of this info is printed to PETSC_STDOUT
2844    when using the runtime options
2845 $       -info -mat_view ::ascii_info
2846 
2847    Example for C/C++ Users:
2848    See the file ${PETSC_DIR}/include/petscmat.h for a complete list of
2849    data within the MatInfo context.  For example,
2850 .vb
2851       MatInfo info;
2852       Mat     A;
2853       double  mal, nz_a, nz_u;
2854 
2855       MatGetInfo(A,MAT_LOCAL,&info);
2856       mal  = info.mallocs;
2857       nz_a = info.nz_allocated;
2858 .ve
2859 
2860    Example for Fortran Users:
2861    Fortran users should declare info as a double precision
2862    array of dimension MAT_INFO_SIZE, and then extract the parameters
2863    of interest.  See the file ${PETSC_DIR}/include/petsc/finclude/petscmat.h
2864    a complete list of parameter names.
2865 .vb
2866       double  precision info(MAT_INFO_SIZE)
2867       double  precision mal, nz_a
2868       Mat     A
2869       integer ierr
2870 
2871       call MatGetInfo(A,MAT_LOCAL,info,ierr)
2872       mal = info(MAT_INFO_MALLOCS)
2873       nz_a = info(MAT_INFO_NZ_ALLOCATED)
2874 .ve
2875 
2876     Level: intermediate
2877 
2878     Concepts: matrices^getting information on
2879 
2880     Developer Note: fortran interface is not autogenerated as the f90
2881     interface defintion cannot be generated correctly [due to MatInfo]
2882 
2883 .seealso: MatStashGetInfo()
2884 
2885 @*/
2886 PetscErrorCode MatGetInfo(Mat mat,MatInfoType flag,MatInfo *info)
2887 {
2888   PetscErrorCode ierr;
2889 
2890   PetscFunctionBegin;
2891   PetscValidHeaderSpecific(mat,MAT_CLASSID,1);
2892   PetscValidType(mat,1);
2893   PetscValidPointer(info,3);
2894   if (!mat->ops->getinfo) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_SUP,"Mat type %s",((PetscObject)mat)->type_name);
2895   MatCheckPreallocated(mat,1);
2896   ierr = (*mat->ops->getinfo)(mat,flag,info);CHKERRQ(ierr);
2897   PetscFunctionReturn(0);
2898 }
2899 
2900 /*
2901    This is used by external packages where it is not easy to get the info from the actual
2902    matrix factorization.
2903 */
2904 PetscErrorCode MatGetInfo_External(Mat A,MatInfoType flag,MatInfo *info)
2905 {
2906   PetscErrorCode ierr;
2907 
2908   PetscFunctionBegin;
2909   ierr = PetscMemzero(info,sizeof(MatInfo));CHKERRQ(ierr);
2910   PetscFunctionReturn(0);
2911 }
2912 
2913 /* ----------------------------------------------------------*/
2914 
2915 /*@C
2916    MatLUFactor - Performs in-place LU factorization of matrix.
2917 
2918    Collective on Mat
2919 
2920    Input Parameters:
2921 +  mat - the matrix
2922 .  row - row permutation
2923 .  col - column permutation
2924 -  info - options for factorization, includes
2925 $          fill - expected fill as ratio of original fill.
2926 $          dtcol - pivot tolerance (0 no pivot, 1 full column pivoting)
2927 $                   Run with the option -info to determine an optimal value to use
2928 
2929    Notes:
2930    Most users should employ the simplified KSP interface for linear solvers
2931    instead of working directly with matrix algebra routines such as this.
2932    See, e.g., KSPCreate().
2933 
2934    This changes the state of the matrix to a factored matrix; it cannot be used
2935    for example with MatSetValues() unless one first calls MatSetUnfactored().
2936 
2937    Level: developer
2938 
2939    Concepts: matrices^LU factorization
2940 
2941 .seealso: MatLUFactorSymbolic(), MatLUFactorNumeric(), MatCholeskyFactor(),
2942           MatGetOrdering(), MatSetUnfactored(), MatFactorInfo, MatGetFactor()
2943 
2944     Developer Note: fortran interface is not autogenerated as the f90
2945     interface defintion cannot be generated correctly [due to MatFactorInfo]
2946 
2947 @*/
2948 PetscErrorCode MatLUFactor(Mat mat,IS row,IS col,const MatFactorInfo *info)
2949 {
2950   PetscErrorCode ierr;
2951   MatFactorInfo  tinfo;
2952 
2953   PetscFunctionBegin;
2954   PetscValidHeaderSpecific(mat,MAT_CLASSID,1);
2955   if (row) PetscValidHeaderSpecific(row,IS_CLASSID,2);
2956   if (col) PetscValidHeaderSpecific(col,IS_CLASSID,3);
2957   if (info) PetscValidPointer(info,4);
2958   PetscValidType(mat,1);
2959   if (!mat->assembled) SETERRQ(PetscObjectComm((PetscObject)mat),PETSC_ERR_ARG_WRONGSTATE,"Not for unassembled matrix");
2960   if (mat->factortype) SETERRQ(PetscObjectComm((PetscObject)mat),PETSC_ERR_ARG_WRONGSTATE,"Not for factored matrix");
2961   if (!mat->ops->lufactor) SETERRQ1(PetscObjectComm((PetscObject)mat),PETSC_ERR_SUP,"Mat type %s",((PetscObject)mat)->type_name);
2962   MatCheckPreallocated(mat,1);
2963   if (!info) {
2964     ierr = MatFactorInfoInitialize(&tinfo);CHKERRQ(ierr);
2965     info = &tinfo;
2966   }
2967 
2968   ierr = PetscLogEventBegin(MAT_LUFactor,mat,row,col,0);CHKERRQ(ierr);
2969   ierr = (*mat->ops->lufactor)(mat,row,col,info);CHKERRQ(ierr);
2970   ierr = PetscLogEventEnd(MAT_LUFactor,mat,row,col,0);CHKERRQ(ierr);
2971   ierr = PetscObjectStateIncrease((PetscObject)mat);CHKERRQ(ierr);
2972   PetscFunctionReturn(0);
2973 }
2974 
2975 /*@C
2976    MatILUFactor - Performs in-place ILU factorization of matrix.
2977 
2978    Collective on Mat
2979 
2980    Input Parameters:
2981 +  mat - the matrix
2982 .  row - row permutation
2983 .  col - column permutation
2984 -  info - structure containing
2985 $      levels - number of levels of fill.
2986 $      expected fill - as ratio of original fill.
2987 $      1 or 0 - indicating force fill on diagonal (improves robustness for matrices
2988                 missing diagonal entries)
2989 
2990    Notes:
2991    Probably really in-place only when level of fill is zero, otherwise allocates
2992    new space to store factored matrix and deletes previous memory.
2993 
2994    Most users should employ the simplified KSP interface for linear solvers
2995    instead of working directly with matrix algebra routines such as this.
2996    See, e.g., KSPCreate().
2997 
2998    Level: developer
2999 
3000    Concepts: matrices^ILU factorization
3001 
3002 .seealso: MatILUFactorSymbolic(), MatLUFactorNumeric(), MatCholeskyFactor(), MatFactorInfo
3003 
3004     Developer Note: fortran interface is not autogenerated as the f90
3005     interface defintion cannot be generated correctly [due to MatFactorInfo]
3006 
3007 @*/
3008 PetscErrorCode MatILUFactor(Mat mat,IS row,IS col,const MatFactorInfo *info)
3009 {
3010   PetscErrorCode ierr;
3011 
3012   PetscFunctionBegin;
3013   PetscValidHeaderSpecific(mat,MAT_CLASSID,1);
3014   if (row) PetscValidHeaderSpecific(row,IS_CLASSID,2);
3015   if (col) PetscValidHeaderSpecific(col,IS_CLASSID,3);
3016   PetscValidPointer(info,4);
3017   PetscValidType(mat,1);
3018   if (mat->rmap->N != mat->cmap->N) SETERRQ(PetscObjectComm((PetscObject)mat),PETSC_ERR_ARG_WRONG,"matrix must be square");
3019   if (!mat->assembled) SETERRQ(PetscObjectComm((PetscObject)mat),PETSC_ERR_ARG_WRONGSTATE,"Not for unassembled matrix");
3020   if (mat->factortype) SETERRQ(PetscObjectComm((PetscObject)mat),PETSC_ERR_ARG_WRONGSTATE,"Not for factored matrix");
3021   if (!mat->ops->ilufactor) SETERRQ1(PetscObjectComm((PetscObject)mat),PETSC_ERR_SUP,"Mat type %s",((PetscObject)mat)->type_name);
3022   MatCheckPreallocated(mat,1);
3023 
3024   ierr = PetscLogEventBegin(MAT_ILUFactor,mat,row,col,0);CHKERRQ(ierr);
3025   ierr = (*mat->ops->ilufactor)(mat,row,col,info);CHKERRQ(ierr);
3026   ierr = PetscLogEventEnd(MAT_ILUFactor,mat,row,col,0);CHKERRQ(ierr);
3027   ierr = PetscObjectStateIncrease((PetscObject)mat);CHKERRQ(ierr);
3028   PetscFunctionReturn(0);
3029 }
3030 
3031 /*@C
3032    MatLUFactorSymbolic - Performs symbolic LU factorization of matrix.
3033    Call this routine before calling MatLUFactorNumeric().
3034 
3035    Collective on Mat
3036 
3037    Input Parameters:
3038 +  fact - the factor matrix obtained with MatGetFactor()
3039 .  mat - the matrix
3040 .  row, col - row and column permutations
3041 -  info - options for factorization, includes
3042 $          fill - expected fill as ratio of original fill.
3043 $          dtcol - pivot tolerance (0 no pivot, 1 full column pivoting)
3044 $                   Run with the option -info to determine an optimal value to use
3045 
3046 
3047    Notes:
3048     See Users-Manual: ch_mat for additional information about choosing the fill factor for better efficiency.
3049 
3050    Most users should employ the simplified KSP interface for linear solvers
3051    instead of working directly with matrix algebra routines such as this.
3052    See, e.g., KSPCreate().
3053 
3054    Level: developer
3055 
3056    Concepts: matrices^LU symbolic factorization
3057 
3058 .seealso: MatLUFactor(), MatLUFactorNumeric(), MatCholeskyFactor(), MatFactorInfo, MatFactorInfoInitialize()
3059 
3060     Developer Note: fortran interface is not autogenerated as the f90
3061     interface defintion cannot be generated correctly [due to MatFactorInfo]
3062 
3063 @*/
3064 PetscErrorCode MatLUFactorSymbolic(Mat fact,Mat mat,IS row,IS col,const MatFactorInfo *info)
3065 {
3066   PetscErrorCode ierr;
3067 
3068   PetscFunctionBegin;
3069   PetscValidHeaderSpecific(mat,MAT_CLASSID,1);
3070   if (row) PetscValidHeaderSpecific(row,IS_CLASSID,2);
3071   if (col) PetscValidHeaderSpecific(col,IS_CLASSID,3);
3072   if (info) PetscValidPointer(info,4);
3073   PetscValidType(mat,1);
3074   PetscValidPointer(fact,5);
3075   if (!mat->assembled) SETERRQ(PetscObjectComm((PetscObject)mat),PETSC_ERR_ARG_WRONGSTATE,"Not for unassembled matrix");
3076   if (mat->factortype) SETERRQ(PetscObjectComm((PetscObject)mat),PETSC_ERR_ARG_WRONGSTATE,"Not for factored matrix");
3077   if (!(fact)->ops->lufactorsymbolic) {
3078     MatSolverType spackage;
3079     ierr = MatFactorGetSolverType(fact,&spackage);CHKERRQ(ierr);
3080     SETERRQ2(PetscObjectComm((PetscObject)mat),PETSC_ERR_SUP,"Matrix type %s symbolic LU using solver package %s",((PetscObject)mat)->type_name,spackage);
3081   }
3082   MatCheckPreallocated(mat,2);
3083 
3084   ierr = PetscLogEventBegin(MAT_LUFactorSymbolic,mat,row,col,0);CHKERRQ(ierr);
3085   ierr = (fact->ops->lufactorsymbolic)(fact,mat,row,col,info);CHKERRQ(ierr);
3086   ierr = PetscLogEventEnd(MAT_LUFactorSymbolic,mat,row,col,0);CHKERRQ(ierr);
3087   ierr = PetscObjectStateIncrease((PetscObject)fact);CHKERRQ(ierr);
3088   PetscFunctionReturn(0);
3089 }
3090 
3091 /*@C
3092    MatLUFactorNumeric - Performs numeric LU factorization of a matrix.
3093    Call this routine after first calling MatLUFactorSymbolic().
3094 
3095    Collective on Mat
3096 
3097    Input Parameters:
3098 +  fact - the factor matrix obtained with MatGetFactor()
3099 .  mat - the matrix
3100 -  info - options for factorization
3101 
3102    Notes:
3103    See MatLUFactor() for in-place factorization.  See
3104    MatCholeskyFactorNumeric() for the symmetric, positive definite case.
3105 
3106    Most users should employ the simplified KSP interface for linear solvers
3107    instead of working directly with matrix algebra routines such as this.
3108    See, e.g., KSPCreate().
3109 
3110    Level: developer
3111 
3112    Concepts: matrices^LU numeric factorization
3113 
3114 .seealso: MatLUFactorSymbolic(), MatLUFactor(), MatCholeskyFactor()
3115 
3116     Developer Note: fortran interface is not autogenerated as the f90
3117     interface defintion cannot be generated correctly [due to MatFactorInfo]
3118 
3119 @*/
3120 PetscErrorCode MatLUFactorNumeric(Mat fact,Mat mat,const MatFactorInfo *info)
3121 {
3122   PetscErrorCode ierr;
3123 
3124   PetscFunctionBegin;
3125   PetscValidHeaderSpecific(mat,MAT_CLASSID,1);
3126   PetscValidType(mat,1);
3127   PetscValidPointer(fact,2);
3128   PetscValidHeaderSpecific(fact,MAT_CLASSID,2);
3129   if (!mat->assembled) SETERRQ(PetscObjectComm((PetscObject)mat),PETSC_ERR_ARG_WRONGSTATE,"Not for unassembled matrix");
3130   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);
3131 
3132   if (!(fact)->ops->lufactornumeric) SETERRQ1(PetscObjectComm((PetscObject)mat),PETSC_ERR_SUP,"Mat type %s numeric LU",((PetscObject)mat)->type_name);
3133   MatCheckPreallocated(mat,2);
3134   ierr = PetscLogEventBegin(MAT_LUFactorNumeric,mat,fact,0,0);CHKERRQ(ierr);
3135   ierr = (fact->ops->lufactornumeric)(fact,mat,info);CHKERRQ(ierr);
3136   ierr = PetscLogEventEnd(MAT_LUFactorNumeric,mat,fact,0,0);CHKERRQ(ierr);
3137   ierr = MatViewFromOptions(fact,NULL,"-mat_factor_view");CHKERRQ(ierr);
3138   ierr = PetscObjectStateIncrease((PetscObject)fact);CHKERRQ(ierr);
3139   PetscFunctionReturn(0);
3140 }
3141 
3142 /*@C
3143    MatCholeskyFactor - Performs in-place Cholesky factorization of a
3144    symmetric matrix.
3145 
3146    Collective on Mat
3147 
3148    Input Parameters:
3149 +  mat - the matrix
3150 .  perm - row and column permutations
3151 -  f - expected fill as ratio of original fill
3152 
3153    Notes:
3154    See MatLUFactor() for the nonsymmetric case.  See also
3155    MatCholeskyFactorSymbolic(), and MatCholeskyFactorNumeric().
3156 
3157    Most users should employ the simplified KSP interface for linear solvers
3158    instead of working directly with matrix algebra routines such as this.
3159    See, e.g., KSPCreate().
3160 
3161    Level: developer
3162 
3163    Concepts: matrices^Cholesky factorization
3164 
3165 .seealso: MatLUFactor(), MatCholeskyFactorSymbolic(), MatCholeskyFactorNumeric()
3166           MatGetOrdering()
3167 
3168     Developer Note: fortran interface is not autogenerated as the f90
3169     interface defintion cannot be generated correctly [due to MatFactorInfo]
3170 
3171 @*/
3172 PetscErrorCode MatCholeskyFactor(Mat mat,IS perm,const MatFactorInfo *info)
3173 {
3174   PetscErrorCode ierr;
3175 
3176   PetscFunctionBegin;
3177   PetscValidHeaderSpecific(mat,MAT_CLASSID,1);
3178   PetscValidType(mat,1);
3179   if (perm) PetscValidHeaderSpecific(perm,IS_CLASSID,2);
3180   if (info) PetscValidPointer(info,3);
3181   if (mat->rmap->N != mat->cmap->N) SETERRQ(PetscObjectComm((PetscObject)mat),PETSC_ERR_ARG_WRONG,"Matrix must be square");
3182   if (!mat->assembled) SETERRQ(PetscObjectComm((PetscObject)mat),PETSC_ERR_ARG_WRONGSTATE,"Not for unassembled matrix");
3183   if (mat->factortype) SETERRQ(PetscObjectComm((PetscObject)mat),PETSC_ERR_ARG_WRONGSTATE,"Not for factored matrix");
3184   if (!mat->ops->choleskyfactor) SETERRQ1(PetscObjectComm((PetscObject)mat),PETSC_ERR_SUP,"In-place factorization for Mat type %s is not supported, try out-of-place factorization. See MatCholeskyFactorSymbolic/Numeric",((PetscObject)mat)->type_name);
3185   MatCheckPreallocated(mat,1);
3186 
3187   ierr = PetscLogEventBegin(MAT_CholeskyFactor,mat,perm,0,0);CHKERRQ(ierr);
3188   ierr = (*mat->ops->choleskyfactor)(mat,perm,info);CHKERRQ(ierr);
3189   ierr = PetscLogEventEnd(MAT_CholeskyFactor,mat,perm,0,0);CHKERRQ(ierr);
3190   ierr = PetscObjectStateIncrease((PetscObject)mat);CHKERRQ(ierr);
3191   PetscFunctionReturn(0);
3192 }
3193 
3194 /*@C
3195    MatCholeskyFactorSymbolic - Performs symbolic Cholesky factorization
3196    of a symmetric matrix.
3197 
3198    Collective on Mat
3199 
3200    Input Parameters:
3201 +  fact - the factor matrix obtained with MatGetFactor()
3202 .  mat - the matrix
3203 .  perm - row and column permutations
3204 -  info - options for factorization, includes
3205 $          fill - expected fill as ratio of original fill.
3206 $          dtcol - pivot tolerance (0 no pivot, 1 full column pivoting)
3207 $                   Run with the option -info to determine an optimal value to use
3208 
3209    Notes:
3210    See MatLUFactorSymbolic() for the nonsymmetric case.  See also
3211    MatCholeskyFactor() and MatCholeskyFactorNumeric().
3212 
3213    Most users should employ the simplified KSP interface for linear solvers
3214    instead of working directly with matrix algebra routines such as this.
3215    See, e.g., KSPCreate().
3216 
3217    Level: developer
3218 
3219    Concepts: matrices^Cholesky symbolic factorization
3220 
3221 .seealso: MatLUFactorSymbolic(), MatCholeskyFactor(), MatCholeskyFactorNumeric()
3222           MatGetOrdering()
3223 
3224     Developer Note: fortran interface is not autogenerated as the f90
3225     interface defintion cannot be generated correctly [due to MatFactorInfo]
3226 
3227 @*/
3228 PetscErrorCode MatCholeskyFactorSymbolic(Mat fact,Mat mat,IS perm,const MatFactorInfo *info)
3229 {
3230   PetscErrorCode ierr;
3231 
3232   PetscFunctionBegin;
3233   PetscValidHeaderSpecific(mat,MAT_CLASSID,1);
3234   PetscValidType(mat,1);
3235   if (perm) PetscValidHeaderSpecific(perm,IS_CLASSID,2);
3236   if (info) PetscValidPointer(info,3);
3237   PetscValidPointer(fact,4);
3238   if (mat->rmap->N != mat->cmap->N) SETERRQ(PetscObjectComm((PetscObject)mat),PETSC_ERR_ARG_WRONG,"Matrix must be square");
3239   if (!mat->assembled) SETERRQ(PetscObjectComm((PetscObject)mat),PETSC_ERR_ARG_WRONGSTATE,"Not for unassembled matrix");
3240   if (mat->factortype) SETERRQ(PetscObjectComm((PetscObject)mat),PETSC_ERR_ARG_WRONGSTATE,"Not for factored matrix");
3241   if (!(fact)->ops->choleskyfactorsymbolic) {
3242     MatSolverType spackage;
3243     ierr = MatFactorGetSolverType(fact,&spackage);CHKERRQ(ierr);
3244     SETERRQ2(PetscObjectComm((PetscObject)mat),PETSC_ERR_SUP,"Mat type %s symbolic factor Cholesky using solver package %s",((PetscObject)mat)->type_name,spackage);
3245   }
3246   MatCheckPreallocated(mat,2);
3247 
3248   ierr = PetscLogEventBegin(MAT_CholeskyFactorSymbolic,mat,perm,0,0);CHKERRQ(ierr);
3249   ierr = (fact->ops->choleskyfactorsymbolic)(fact,mat,perm,info);CHKERRQ(ierr);
3250   ierr = PetscLogEventEnd(MAT_CholeskyFactorSymbolic,mat,perm,0,0);CHKERRQ(ierr);
3251   ierr = PetscObjectStateIncrease((PetscObject)fact);CHKERRQ(ierr);
3252   PetscFunctionReturn(0);
3253 }
3254 
3255 /*@C
3256    MatCholeskyFactorNumeric - Performs numeric Cholesky factorization
3257    of a symmetric matrix. Call this routine after first calling
3258    MatCholeskyFactorSymbolic().
3259 
3260    Collective on Mat
3261 
3262    Input Parameters:
3263 +  fact - the factor matrix obtained with MatGetFactor()
3264 .  mat - the initial matrix
3265 .  info - options for factorization
3266 -  fact - the symbolic factor of mat
3267 
3268 
3269    Notes:
3270    Most users should employ the simplified KSP interface for linear solvers
3271    instead of working directly with matrix algebra routines such as this.
3272    See, e.g., KSPCreate().
3273 
3274    Level: developer
3275 
3276    Concepts: matrices^Cholesky numeric factorization
3277 
3278 .seealso: MatCholeskyFactorSymbolic(), MatCholeskyFactor(), MatLUFactorNumeric()
3279 
3280     Developer Note: fortran interface is not autogenerated as the f90
3281     interface defintion cannot be generated correctly [due to MatFactorInfo]
3282 
3283 @*/
3284 PetscErrorCode MatCholeskyFactorNumeric(Mat fact,Mat mat,const MatFactorInfo *info)
3285 {
3286   PetscErrorCode ierr;
3287 
3288   PetscFunctionBegin;
3289   PetscValidHeaderSpecific(mat,MAT_CLASSID,1);
3290   PetscValidType(mat,1);
3291   PetscValidPointer(fact,2);
3292   PetscValidHeaderSpecific(fact,MAT_CLASSID,2);
3293   if (!mat->assembled) SETERRQ(PetscObjectComm((PetscObject)mat),PETSC_ERR_ARG_WRONGSTATE,"Not for unassembled matrix");
3294   if (!(fact)->ops->choleskyfactornumeric) SETERRQ1(PetscObjectComm((PetscObject)mat),PETSC_ERR_SUP,"Mat type %s numeric factor Cholesky",((PetscObject)mat)->type_name);
3295   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);
3296   MatCheckPreallocated(mat,2);
3297 
3298   ierr = PetscLogEventBegin(MAT_CholeskyFactorNumeric,mat,fact,0,0);CHKERRQ(ierr);
3299   ierr = (fact->ops->choleskyfactornumeric)(fact,mat,info);CHKERRQ(ierr);
3300   ierr = PetscLogEventEnd(MAT_CholeskyFactorNumeric,mat,fact,0,0);CHKERRQ(ierr);
3301   ierr = MatViewFromOptions(fact,NULL,"-mat_factor_view");CHKERRQ(ierr);
3302   ierr = PetscObjectStateIncrease((PetscObject)fact);CHKERRQ(ierr);
3303   PetscFunctionReturn(0);
3304 }
3305 
3306 /* ----------------------------------------------------------------*/
3307 /*@
3308    MatSolve - Solves A x = b, given a factored matrix.
3309 
3310    Neighbor-wise Collective on Mat and Vec
3311 
3312    Input Parameters:
3313 +  mat - the factored matrix
3314 -  b - the right-hand-side vector
3315 
3316    Output Parameter:
3317 .  x - the result vector
3318 
3319    Notes:
3320    The vectors b and x cannot be the same.  I.e., one cannot
3321    call MatSolve(A,x,x).
3322 
3323    Notes:
3324    Most users should employ the simplified KSP interface for linear solvers
3325    instead of working directly with matrix algebra routines such as this.
3326    See, e.g., KSPCreate().
3327 
3328    Level: developer
3329 
3330    Concepts: matrices^triangular solves
3331 
3332 .seealso: MatSolveAdd(), MatSolveTranspose(), MatSolveTransposeAdd()
3333 @*/
3334 PetscErrorCode MatSolve(Mat mat,Vec b,Vec x)
3335 {
3336   PetscErrorCode ierr;
3337 
3338   PetscFunctionBegin;
3339   PetscValidHeaderSpecific(mat,MAT_CLASSID,1);
3340   PetscValidType(mat,1);
3341   PetscValidHeaderSpecific(b,VEC_CLASSID,2);
3342   PetscValidHeaderSpecific(x,VEC_CLASSID,3);
3343   PetscCheckSameComm(mat,1,b,2);
3344   PetscCheckSameComm(mat,1,x,3);
3345   if (x == b) SETERRQ(PetscObjectComm((PetscObject)mat),PETSC_ERR_ARG_IDN,"x and b must be different vectors");
3346   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);
3347   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);
3348   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);
3349   if (!mat->rmap->N && !mat->cmap->N) PetscFunctionReturn(0);
3350   if (!mat->ops->solve) SETERRQ1(PetscObjectComm((PetscObject)mat),PETSC_ERR_SUP,"Mat type %s",((PetscObject)mat)->type_name);
3351   MatCheckPreallocated(mat,1);
3352 
3353   ierr = PetscLogEventBegin(MAT_Solve,mat,b,x,0);CHKERRQ(ierr);
3354   if (mat->factorerrortype) {
3355     ierr = PetscInfo1(mat,"MatFactorError %D\n",mat->factorerrortype);CHKERRQ(ierr);
3356     ierr = VecSetInf(x);CHKERRQ(ierr);
3357   } else {
3358     if (!mat->ops->solve) SETERRQ1(PetscObjectComm((PetscObject)mat),PETSC_ERR_SUP,"Mat type %s",((PetscObject)mat)->type_name);
3359     ierr = (*mat->ops->solve)(mat,b,x);CHKERRQ(ierr);
3360   }
3361   ierr = PetscLogEventEnd(MAT_Solve,mat,b,x,0);CHKERRQ(ierr);
3362   ierr = PetscObjectStateIncrease((PetscObject)x);CHKERRQ(ierr);
3363   PetscFunctionReturn(0);
3364 }
3365 
3366 static PetscErrorCode MatMatSolve_Basic(Mat A,Mat B,Mat X, PetscBool trans)
3367 {
3368   PetscErrorCode ierr;
3369   Vec            b,x;
3370   PetscInt       m,N,i;
3371   PetscScalar    *bb,*xx;
3372   PetscBool      flg;
3373 
3374   PetscFunctionBegin;
3375   ierr = PetscObjectTypeCompareAny((PetscObject)B,&flg,MATSEQDENSE,MATMPIDENSE,NULL);CHKERRQ(ierr);
3376   if (!flg) SETERRQ(PetscObjectComm((PetscObject)A),PETSC_ERR_ARG_WRONG,"Matrix B must be MATDENSE matrix");
3377   ierr = PetscObjectTypeCompareAny((PetscObject)X,&flg,MATSEQDENSE,MATMPIDENSE,NULL);CHKERRQ(ierr);
3378   if (!flg) SETERRQ(PetscObjectComm((PetscObject)A),PETSC_ERR_ARG_WRONG,"Matrix X must be MATDENSE matrix");
3379 
3380   ierr = MatDenseGetArray(B,&bb);CHKERRQ(ierr);
3381   ierr = MatDenseGetArray(X,&xx);CHKERRQ(ierr);
3382   ierr = MatGetLocalSize(B,&m,NULL);CHKERRQ(ierr);  /* number local rows */
3383   ierr = MatGetSize(B,NULL,&N);CHKERRQ(ierr);       /* total columns in dense matrix */
3384   ierr = MatCreateVecs(A,&x,&b);CHKERRQ(ierr);
3385   for (i=0; i<N; i++) {
3386     ierr = VecPlaceArray(b,bb + i*m);CHKERRQ(ierr);
3387     ierr = VecPlaceArray(x,xx + i*m);CHKERRQ(ierr);
3388     if (trans) {
3389       ierr = MatSolveTranspose(A,b,x);CHKERRQ(ierr);
3390     } else {
3391       ierr = MatSolve(A,b,x);CHKERRQ(ierr);
3392     }
3393     ierr = VecResetArray(x);CHKERRQ(ierr);
3394     ierr = VecResetArray(b);CHKERRQ(ierr);
3395   }
3396   ierr = VecDestroy(&b);CHKERRQ(ierr);
3397   ierr = VecDestroy(&x);CHKERRQ(ierr);
3398   ierr = MatDenseRestoreArray(B,&bb);CHKERRQ(ierr);
3399   ierr = MatDenseRestoreArray(X,&xx);CHKERRQ(ierr);
3400   PetscFunctionReturn(0);
3401 }
3402 
3403 /*@
3404    MatMatSolve - Solves A X = B, given a factored matrix.
3405 
3406    Neighbor-wise Collective on Mat
3407 
3408    Input Parameters:
3409 +  A - the factored matrix
3410 -  B - the right-hand-side matrix  (dense matrix)
3411 
3412    Output Parameter:
3413 .  X - the result matrix (dense matrix)
3414 
3415    Notes:
3416    The matrices b and x cannot be the same.  I.e., one cannot
3417    call MatMatSolve(A,x,x).
3418 
3419    Notes:
3420    Most users should usually employ the simplified KSP interface for linear solvers
3421    instead of working directly with matrix algebra routines such as this.
3422    See, e.g., KSPCreate(). However KSP can only solve for one vector (column of X)
3423    at a time.
3424 
3425    When using SuperLU_Dist as a parallel solver PETSc will use the SuperLU_Dist functionality to solve multiple right hand sides simultaneously. For MUMPS
3426    it calls a separate solve for each right hand side since MUMPS does not yet support distributed right hand sides.
3427 
3428    Since the resulting matrix X must always be dense we do not support sparse representation of the matrix B.
3429 
3430    Level: developer
3431 
3432    Concepts: matrices^triangular solves
3433 
3434 .seealso: MatMatSolveTranspose(), MatLUFactor(), MatCholeskyFactor()
3435 @*/
3436 PetscErrorCode MatMatSolve(Mat A,Mat B,Mat X)
3437 {
3438   PetscErrorCode ierr;
3439 
3440   PetscFunctionBegin;
3441   PetscValidHeaderSpecific(A,MAT_CLASSID,1);
3442   PetscValidType(A,1);
3443   PetscValidHeaderSpecific(B,MAT_CLASSID,2);
3444   PetscValidHeaderSpecific(X,MAT_CLASSID,3);
3445   PetscCheckSameComm(A,1,B,2);
3446   PetscCheckSameComm(A,1,X,3);
3447   if (X == B) SETERRQ(PetscObjectComm((PetscObject)A),PETSC_ERR_ARG_IDN,"X and B must be different matrices");
3448   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);
3449   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);
3450   if (X->cmap->N < B->cmap->N) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_SIZ,"Solution matrix must have same number of columns as rhs matrix");
3451   if (!A->rmap->N && !A->cmap->N) PetscFunctionReturn(0);
3452   if (!A->factortype) SETERRQ(PetscObjectComm((PetscObject)A),PETSC_ERR_ARG_WRONGSTATE,"Unfactored matrix");
3453   MatCheckPreallocated(A,1);
3454 
3455   ierr = PetscLogEventBegin(MAT_MatSolve,A,B,X,0);CHKERRQ(ierr);
3456   if (!A->ops->matsolve) {
3457     ierr = PetscInfo1(A,"Mat type %s using basic MatMatSolve\n",((PetscObject)A)->type_name);CHKERRQ(ierr);
3458     ierr = MatMatSolve_Basic(A,B,X,PETSC_FALSE);CHKERRQ(ierr);
3459   } else {
3460     ierr = (*A->ops->matsolve)(A,B,X);CHKERRQ(ierr);
3461   }
3462   ierr = PetscLogEventEnd(MAT_MatSolve,A,B,X,0);CHKERRQ(ierr);
3463   ierr = PetscObjectStateIncrease((PetscObject)X);CHKERRQ(ierr);
3464   PetscFunctionReturn(0);
3465 }
3466 
3467 /*@
3468    MatMatSolveTranspose - Solves A^T X = B, given a factored matrix.
3469 
3470    Neighbor-wise Collective on Mat
3471 
3472    Input Parameters:
3473 +  A - the factored matrix
3474 -  B - the right-hand-side matrix  (dense matrix)
3475 
3476    Output Parameter:
3477 .  X - the result matrix (dense matrix)
3478 
3479    Notes:
3480    The matrices B and X cannot be the same.  I.e., one cannot
3481    call MatMatSolveTranspose(A,X,X).
3482 
3483    Notes:
3484    Most users should usually employ the simplified KSP interface for linear solvers
3485    instead of working directly with matrix algebra routines such as this.
3486    See, e.g., KSPCreate(). However KSP can only solve for one vector (column of X)
3487    at a time.
3488 
3489    When using SuperLU_Dist or MUMPS as a parallel solver, PETSc will use their functionality to solve multiple right hand sides simultaneously.
3490 
3491    Level: developer
3492 
3493    Concepts: matrices^triangular solves
3494 
3495 .seealso: MatMatSolve(), MatLUFactor(), MatCholeskyFactor()
3496 @*/
3497 PetscErrorCode MatMatSolveTranspose(Mat A,Mat B,Mat X)
3498 {
3499   PetscErrorCode ierr;
3500 
3501   PetscFunctionBegin;
3502   PetscValidHeaderSpecific(A,MAT_CLASSID,1);
3503   PetscValidType(A,1);
3504   PetscValidHeaderSpecific(B,MAT_CLASSID,2);
3505   PetscValidHeaderSpecific(X,MAT_CLASSID,3);
3506   PetscCheckSameComm(A,1,B,2);
3507   PetscCheckSameComm(A,1,X,3);
3508   if (X == B) SETERRQ(PetscObjectComm((PetscObject)A),PETSC_ERR_ARG_IDN,"X and B must be different matrices");
3509   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);
3510   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);
3511   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);
3512   if (X->cmap->N < B->cmap->N) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_SIZ,"Solution matrix must have same number of columns as rhs matrix");
3513   if (!A->rmap->N && !A->cmap->N) PetscFunctionReturn(0);
3514   if (!A->factortype) SETERRQ(PetscObjectComm((PetscObject)A),PETSC_ERR_ARG_WRONGSTATE,"Unfactored matrix");
3515   MatCheckPreallocated(A,1);
3516 
3517   ierr = PetscLogEventBegin(MAT_MatSolve,A,B,X,0);CHKERRQ(ierr);
3518   if (!A->ops->matsolvetranspose) {
3519     ierr = PetscInfo1(A,"Mat type %s using basic MatMatSolveTranspose\n",((PetscObject)A)->type_name);CHKERRQ(ierr);
3520     ierr = MatMatSolve_Basic(A,B,X,PETSC_TRUE);CHKERRQ(ierr);
3521   } else {
3522     ierr = (*A->ops->matsolvetranspose)(A,B,X);CHKERRQ(ierr);
3523   }
3524   ierr = PetscLogEventEnd(MAT_MatSolve,A,B,X,0);CHKERRQ(ierr);
3525   ierr = PetscObjectStateIncrease((PetscObject)X);CHKERRQ(ierr);
3526   PetscFunctionReturn(0);
3527 }
3528 
3529 /*@
3530    MatMatTransposeSolve - Solves A X = B^T, given a factored matrix.
3531 
3532    Neighbor-wise Collective on Mat
3533 
3534    Input Parameters:
3535 +  A - the factored matrix
3536 -  Bt - the transpose of right-hand-side matrix
3537 
3538    Output Parameter:
3539 .  X - the result matrix (dense matrix)
3540 
3541    Notes:
3542    Most users should usually employ the simplified KSP interface for linear solvers
3543    instead of working directly with matrix algebra routines such as this.
3544    See, e.g., KSPCreate(). However KSP can only solve for one vector (column of X)
3545    at a time.
3546 
3547    For MUMPS, it only supports centralized sparse compressed column format on the host processor for right hand side matrix. User must create B^T in sparse compressed row format on the host processor and call MatMatTransposeSolve() to implement MUMPS' MatMatSolve().
3548 
3549    Level: developer
3550 
3551    Concepts: matrices^triangular solves
3552 
3553 .seealso: MatMatSolve(), MatMatSolveTranspose(), MatLUFactor(), MatCholeskyFactor()
3554 @*/
3555 PetscErrorCode MatMatTransposeSolve(Mat A,Mat Bt,Mat X)
3556 {
3557   PetscErrorCode ierr;
3558 
3559   PetscFunctionBegin;
3560   PetscValidHeaderSpecific(A,MAT_CLASSID,1);
3561   PetscValidType(A,1);
3562   PetscValidHeaderSpecific(Bt,MAT_CLASSID,2);
3563   PetscValidHeaderSpecific(X,MAT_CLASSID,3);
3564   PetscCheckSameComm(A,1,Bt,2);
3565   PetscCheckSameComm(A,1,X,3);
3566 
3567   if (X == Bt) SETERRQ(PetscObjectComm((PetscObject)A),PETSC_ERR_ARG_IDN,"X and B must be different matrices");
3568   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);
3569   if (A->rmap->N != Bt->cmap->N) SETERRQ2(PetscObjectComm((PetscObject)A),PETSC_ERR_ARG_SIZ,"Mat A,Mat Bt: global dim %D %D",A->rmap->N,Bt->cmap->N);
3570   if (X->cmap->N < Bt->rmap->N) SETERRQ(PetscObjectComm((PetscObject)X),PETSC_ERR_ARG_SIZ,"Solution matrix must have same number of columns as row number of the rhs matrix");
3571   if (!A->rmap->N && !A->cmap->N) PetscFunctionReturn(0);
3572   if (!A->factortype) SETERRQ(PetscObjectComm((PetscObject)A),PETSC_ERR_ARG_WRONGSTATE,"Unfactored matrix");
3573   MatCheckPreallocated(A,1);
3574 
3575   if (!A->ops->mattransposesolve) SETERRQ1(PetscObjectComm((PetscObject)A),PETSC_ERR_SUP,"Mat type %s",((PetscObject)A)->type_name);
3576   ierr = PetscLogEventBegin(MAT_MatTrSolve,A,Bt,X,0);CHKERRQ(ierr);
3577   ierr = (*A->ops->mattransposesolve)(A,Bt,X);CHKERRQ(ierr);
3578   ierr = PetscLogEventEnd(MAT_MatTrSolve,A,Bt,X,0);CHKERRQ(ierr);
3579   ierr = PetscObjectStateIncrease((PetscObject)X);CHKERRQ(ierr);
3580   PetscFunctionReturn(0);
3581 }
3582 
3583 /*@
3584    MatForwardSolve - Solves L x = b, given a factored matrix, A = LU, or
3585                             U^T*D^(1/2) x = b, given a factored symmetric matrix, A = U^T*D*U,
3586 
3587    Neighbor-wise Collective on Mat and Vec
3588 
3589    Input Parameters:
3590 +  mat - the factored matrix
3591 -  b - the right-hand-side vector
3592 
3593    Output Parameter:
3594 .  x - the result vector
3595 
3596    Notes:
3597    MatSolve() should be used for most applications, as it performs
3598    a forward solve followed by a backward solve.
3599 
3600    The vectors b and x cannot be the same,  i.e., one cannot
3601    call MatForwardSolve(A,x,x).
3602 
3603    For matrix in seqsbaij format with block size larger than 1,
3604    the diagonal blocks are not implemented as D = D^(1/2) * D^(1/2) yet.
3605    MatForwardSolve() solves U^T*D y = b, and
3606    MatBackwardSolve() solves U x = y.
3607    Thus they do not provide a symmetric preconditioner.
3608 
3609    Most users should employ the simplified KSP interface for linear solvers
3610    instead of working directly with matrix algebra routines such as this.
3611    See, e.g., KSPCreate().
3612 
3613    Level: developer
3614 
3615    Concepts: matrices^forward solves
3616 
3617 .seealso: MatSolve(), MatBackwardSolve()
3618 @*/
3619 PetscErrorCode MatForwardSolve(Mat mat,Vec b,Vec x)
3620 {
3621   PetscErrorCode ierr;
3622 
3623   PetscFunctionBegin;
3624   PetscValidHeaderSpecific(mat,MAT_CLASSID,1);
3625   PetscValidType(mat,1);
3626   PetscValidHeaderSpecific(b,VEC_CLASSID,2);
3627   PetscValidHeaderSpecific(x,VEC_CLASSID,3);
3628   PetscCheckSameComm(mat,1,b,2);
3629   PetscCheckSameComm(mat,1,x,3);
3630   if (x == b) SETERRQ(PetscObjectComm((PetscObject)mat),PETSC_ERR_ARG_IDN,"x and b must be different vectors");
3631   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);
3632   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);
3633   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);
3634   if (!mat->rmap->N && !mat->cmap->N) PetscFunctionReturn(0);
3635   if (!mat->factortype) SETERRQ(PetscObjectComm((PetscObject)mat),PETSC_ERR_ARG_WRONGSTATE,"Unfactored matrix");
3636   MatCheckPreallocated(mat,1);
3637 
3638   if (!mat->ops->forwardsolve) SETERRQ1(PetscObjectComm((PetscObject)mat),PETSC_ERR_SUP,"Mat type %s",((PetscObject)mat)->type_name);
3639   ierr = PetscLogEventBegin(MAT_ForwardSolve,mat,b,x,0);CHKERRQ(ierr);
3640   ierr = (*mat->ops->forwardsolve)(mat,b,x);CHKERRQ(ierr);
3641   ierr = PetscLogEventEnd(MAT_ForwardSolve,mat,b,x,0);CHKERRQ(ierr);
3642   ierr = PetscObjectStateIncrease((PetscObject)x);CHKERRQ(ierr);
3643   PetscFunctionReturn(0);
3644 }
3645 
3646 /*@
3647    MatBackwardSolve - Solves U x = b, given a factored matrix, A = LU.
3648                              D^(1/2) U x = b, given a factored symmetric matrix, A = U^T*D*U,
3649 
3650    Neighbor-wise Collective on Mat and Vec
3651 
3652    Input Parameters:
3653 +  mat - the factored matrix
3654 -  b - the right-hand-side vector
3655 
3656    Output Parameter:
3657 .  x - the result vector
3658 
3659    Notes:
3660    MatSolve() should be used for most applications, as it performs
3661    a forward solve followed by a backward solve.
3662 
3663    The vectors b and x cannot be the same.  I.e., one cannot
3664    call MatBackwardSolve(A,x,x).
3665 
3666    For matrix in seqsbaij format with block size larger than 1,
3667    the diagonal blocks are not implemented as D = D^(1/2) * D^(1/2) yet.
3668    MatForwardSolve() solves U^T*D y = b, and
3669    MatBackwardSolve() solves U x = y.
3670    Thus they do not provide a symmetric preconditioner.
3671 
3672    Most users should employ the simplified KSP interface for linear solvers
3673    instead of working directly with matrix algebra routines such as this.
3674    See, e.g., KSPCreate().
3675 
3676    Level: developer
3677 
3678    Concepts: matrices^backward solves
3679 
3680 .seealso: MatSolve(), MatForwardSolve()
3681 @*/
3682 PetscErrorCode MatBackwardSolve(Mat mat,Vec b,Vec x)
3683 {
3684   PetscErrorCode ierr;
3685 
3686   PetscFunctionBegin;
3687   PetscValidHeaderSpecific(mat,MAT_CLASSID,1);
3688   PetscValidType(mat,1);
3689   PetscValidHeaderSpecific(b,VEC_CLASSID,2);
3690   PetscValidHeaderSpecific(x,VEC_CLASSID,3);
3691   PetscCheckSameComm(mat,1,b,2);
3692   PetscCheckSameComm(mat,1,x,3);
3693   if (x == b) SETERRQ(PetscObjectComm((PetscObject)mat),PETSC_ERR_ARG_IDN,"x and b must be different vectors");
3694   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);
3695   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);
3696   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);
3697   if (!mat->rmap->N && !mat->cmap->N) PetscFunctionReturn(0);
3698   if (!mat->factortype) SETERRQ(PetscObjectComm((PetscObject)mat),PETSC_ERR_ARG_WRONGSTATE,"Unfactored matrix");
3699   MatCheckPreallocated(mat,1);
3700 
3701   if (!mat->ops->backwardsolve) SETERRQ1(PetscObjectComm((PetscObject)mat),PETSC_ERR_SUP,"Mat type %s",((PetscObject)mat)->type_name);
3702   ierr = PetscLogEventBegin(MAT_BackwardSolve,mat,b,x,0);CHKERRQ(ierr);
3703   ierr = (*mat->ops->backwardsolve)(mat,b,x);CHKERRQ(ierr);
3704   ierr = PetscLogEventEnd(MAT_BackwardSolve,mat,b,x,0);CHKERRQ(ierr);
3705   ierr = PetscObjectStateIncrease((PetscObject)x);CHKERRQ(ierr);
3706   PetscFunctionReturn(0);
3707 }
3708 
3709 /*@
3710    MatSolveAdd - Computes x = y + inv(A)*b, given a factored matrix.
3711 
3712    Neighbor-wise Collective on Mat and Vec
3713 
3714    Input Parameters:
3715 +  mat - the factored matrix
3716 .  b - the right-hand-side vector
3717 -  y - the vector to be added to
3718 
3719    Output Parameter:
3720 .  x - the result vector
3721 
3722    Notes:
3723    The vectors b and x cannot be the same.  I.e., one cannot
3724    call MatSolveAdd(A,x,y,x).
3725 
3726    Most users should employ the simplified KSP interface for linear solvers
3727    instead of working directly with matrix algebra routines such as this.
3728    See, e.g., KSPCreate().
3729 
3730    Level: developer
3731 
3732    Concepts: matrices^triangular solves
3733 
3734 .seealso: MatSolve(), MatSolveTranspose(), MatSolveTransposeAdd()
3735 @*/
3736 PetscErrorCode MatSolveAdd(Mat mat,Vec b,Vec y,Vec x)
3737 {
3738   PetscScalar    one = 1.0;
3739   Vec            tmp;
3740   PetscErrorCode ierr;
3741 
3742   PetscFunctionBegin;
3743   PetscValidHeaderSpecific(mat,MAT_CLASSID,1);
3744   PetscValidType(mat,1);
3745   PetscValidHeaderSpecific(y,VEC_CLASSID,2);
3746   PetscValidHeaderSpecific(b,VEC_CLASSID,3);
3747   PetscValidHeaderSpecific(x,VEC_CLASSID,4);
3748   PetscCheckSameComm(mat,1,b,2);
3749   PetscCheckSameComm(mat,1,y,2);
3750   PetscCheckSameComm(mat,1,x,3);
3751   if (x == b) SETERRQ(PetscObjectComm((PetscObject)mat),PETSC_ERR_ARG_IDN,"x and b must be different vectors");
3752   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);
3753   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);
3754   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);
3755   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);
3756   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);
3757   if (!mat->rmap->N && !mat->cmap->N) PetscFunctionReturn(0);
3758   if (!mat->factortype) SETERRQ(PetscObjectComm((PetscObject)mat),PETSC_ERR_ARG_WRONGSTATE,"Unfactored matrix");
3759   MatCheckPreallocated(mat,1);
3760 
3761   ierr = PetscLogEventBegin(MAT_SolveAdd,mat,b,x,y);CHKERRQ(ierr);
3762   if (mat->ops->solveadd) {
3763     ierr = (*mat->ops->solveadd)(mat,b,y,x);CHKERRQ(ierr);
3764   } else {
3765     /* do the solve then the add manually */
3766     if (x != y) {
3767       ierr = MatSolve(mat,b,x);CHKERRQ(ierr);
3768       ierr = VecAXPY(x,one,y);CHKERRQ(ierr);
3769     } else {
3770       ierr = VecDuplicate(x,&tmp);CHKERRQ(ierr);
3771       ierr = PetscLogObjectParent((PetscObject)mat,(PetscObject)tmp);CHKERRQ(ierr);
3772       ierr = VecCopy(x,tmp);CHKERRQ(ierr);
3773       ierr = MatSolve(mat,b,x);CHKERRQ(ierr);
3774       ierr = VecAXPY(x,one,tmp);CHKERRQ(ierr);
3775       ierr = VecDestroy(&tmp);CHKERRQ(ierr);
3776     }
3777   }
3778   ierr = PetscLogEventEnd(MAT_SolveAdd,mat,b,x,y);CHKERRQ(ierr);
3779   ierr = PetscObjectStateIncrease((PetscObject)x);CHKERRQ(ierr);
3780   PetscFunctionReturn(0);
3781 }
3782 
3783 /*@
3784    MatSolveTranspose - Solves A' x = b, given a factored matrix.
3785 
3786    Neighbor-wise Collective on Mat and Vec
3787 
3788    Input Parameters:
3789 +  mat - the factored matrix
3790 -  b - the right-hand-side vector
3791 
3792    Output Parameter:
3793 .  x - the result vector
3794 
3795    Notes:
3796    The vectors b and x cannot be the same.  I.e., one cannot
3797    call MatSolveTranspose(A,x,x).
3798 
3799    Most users should employ the simplified KSP interface for linear solvers
3800    instead of working directly with matrix algebra routines such as this.
3801    See, e.g., KSPCreate().
3802 
3803    Level: developer
3804 
3805    Concepts: matrices^triangular solves
3806 
3807 .seealso: MatSolve(), MatSolveAdd(), MatSolveTransposeAdd()
3808 @*/
3809 PetscErrorCode MatSolveTranspose(Mat mat,Vec b,Vec x)
3810 {
3811   PetscErrorCode ierr;
3812 
3813   PetscFunctionBegin;
3814   PetscValidHeaderSpecific(mat,MAT_CLASSID,1);
3815   PetscValidType(mat,1);
3816   PetscValidHeaderSpecific(b,VEC_CLASSID,2);
3817   PetscValidHeaderSpecific(x,VEC_CLASSID,3);
3818   PetscCheckSameComm(mat,1,b,2);
3819   PetscCheckSameComm(mat,1,x,3);
3820   if (x == b) SETERRQ(PetscObjectComm((PetscObject)mat),PETSC_ERR_ARG_IDN,"x and b must be different vectors");
3821   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);
3822   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);
3823   if (!mat->rmap->N && !mat->cmap->N) PetscFunctionReturn(0);
3824   if (!mat->factortype) SETERRQ(PetscObjectComm((PetscObject)mat),PETSC_ERR_ARG_WRONGSTATE,"Unfactored matrix");
3825   MatCheckPreallocated(mat,1);
3826   ierr = PetscLogEventBegin(MAT_SolveTranspose,mat,b,x,0);CHKERRQ(ierr);
3827   if (mat->factorerrortype) {
3828     ierr = PetscInfo1(mat,"MatFactorError %D\n",mat->factorerrortype);CHKERRQ(ierr);
3829     ierr = VecSetInf(x);CHKERRQ(ierr);
3830   } else {
3831     if (!mat->ops->solvetranspose) SETERRQ1(PetscObjectComm((PetscObject)mat),PETSC_ERR_SUP,"Matrix type %s",((PetscObject)mat)->type_name);
3832     ierr = (*mat->ops->solvetranspose)(mat,b,x);CHKERRQ(ierr);
3833   }
3834   ierr = PetscLogEventEnd(MAT_SolveTranspose,mat,b,x,0);CHKERRQ(ierr);
3835   ierr = PetscObjectStateIncrease((PetscObject)x);CHKERRQ(ierr);
3836   PetscFunctionReturn(0);
3837 }
3838 
3839 /*@
3840    MatSolveTransposeAdd - Computes x = y + inv(Transpose(A)) b, given a
3841                       factored matrix.
3842 
3843    Neighbor-wise Collective on Mat and Vec
3844 
3845    Input Parameters:
3846 +  mat - the factored matrix
3847 .  b - the right-hand-side vector
3848 -  y - the vector to be added to
3849 
3850    Output Parameter:
3851 .  x - the result vector
3852 
3853    Notes:
3854    The vectors b and x cannot be the same.  I.e., one cannot
3855    call MatSolveTransposeAdd(A,x,y,x).
3856 
3857    Most users should employ the simplified KSP interface for linear solvers
3858    instead of working directly with matrix algebra routines such as this.
3859    See, e.g., KSPCreate().
3860 
3861    Level: developer
3862 
3863    Concepts: matrices^triangular solves
3864 
3865 .seealso: MatSolve(), MatSolveAdd(), MatSolveTranspose()
3866 @*/
3867 PetscErrorCode MatSolveTransposeAdd(Mat mat,Vec b,Vec y,Vec x)
3868 {
3869   PetscScalar    one = 1.0;
3870   PetscErrorCode ierr;
3871   Vec            tmp;
3872 
3873   PetscFunctionBegin;
3874   PetscValidHeaderSpecific(mat,MAT_CLASSID,1);
3875   PetscValidType(mat,1);
3876   PetscValidHeaderSpecific(y,VEC_CLASSID,2);
3877   PetscValidHeaderSpecific(b,VEC_CLASSID,3);
3878   PetscValidHeaderSpecific(x,VEC_CLASSID,4);
3879   PetscCheckSameComm(mat,1,b,2);
3880   PetscCheckSameComm(mat,1,y,3);
3881   PetscCheckSameComm(mat,1,x,4);
3882   if (x == b) SETERRQ(PetscObjectComm((PetscObject)mat),PETSC_ERR_ARG_IDN,"x and b must be different vectors");
3883   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);
3884   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);
3885   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);
3886   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);
3887   if (!mat->rmap->N && !mat->cmap->N) PetscFunctionReturn(0);
3888   if (!mat->factortype) SETERRQ(PetscObjectComm((PetscObject)mat),PETSC_ERR_ARG_WRONGSTATE,"Unfactored matrix");
3889   MatCheckPreallocated(mat,1);
3890 
3891   ierr = PetscLogEventBegin(MAT_SolveTransposeAdd,mat,b,x,y);CHKERRQ(ierr);
3892   if (mat->ops->solvetransposeadd) {
3893     if (mat->factorerrortype) {
3894       ierr = PetscInfo1(mat,"MatFactorError %D\n",mat->factorerrortype);CHKERRQ(ierr);
3895       ierr = VecSetInf(x);CHKERRQ(ierr);
3896     } else {
3897       ierr = (*mat->ops->solvetransposeadd)(mat,b,y,x);CHKERRQ(ierr);
3898     }
3899   } else {
3900     /* do the solve then the add manually */
3901     if (x != y) {
3902       ierr = MatSolveTranspose(mat,b,x);CHKERRQ(ierr);
3903       ierr = VecAXPY(x,one,y);CHKERRQ(ierr);
3904     } else {
3905       ierr = VecDuplicate(x,&tmp);CHKERRQ(ierr);
3906       ierr = PetscLogObjectParent((PetscObject)mat,(PetscObject)tmp);CHKERRQ(ierr);
3907       ierr = VecCopy(x,tmp);CHKERRQ(ierr);
3908       ierr = MatSolveTranspose(mat,b,x);CHKERRQ(ierr);
3909       ierr = VecAXPY(x,one,tmp);CHKERRQ(ierr);
3910       ierr = VecDestroy(&tmp);CHKERRQ(ierr);
3911     }
3912   }
3913   ierr = PetscLogEventEnd(MAT_SolveTransposeAdd,mat,b,x,y);CHKERRQ(ierr);
3914   ierr = PetscObjectStateIncrease((PetscObject)x);CHKERRQ(ierr);
3915   PetscFunctionReturn(0);
3916 }
3917 /* ----------------------------------------------------------------*/
3918 
3919 /*@
3920    MatSOR - Computes relaxation (SOR, Gauss-Seidel) sweeps.
3921 
3922    Neighbor-wise Collective on Mat and Vec
3923 
3924    Input Parameters:
3925 +  mat - the matrix
3926 .  b - the right hand side
3927 .  omega - the relaxation factor
3928 .  flag - flag indicating the type of SOR (see below)
3929 .  shift -  diagonal shift
3930 .  its - the number of iterations
3931 -  lits - the number of local iterations
3932 
3933    Output Parameters:
3934 .  x - the solution (can contain an initial guess, use option SOR_ZERO_INITIAL_GUESS to indicate no guess)
3935 
3936    SOR Flags:
3937 .     SOR_FORWARD_SWEEP - forward SOR
3938 .     SOR_BACKWARD_SWEEP - backward SOR
3939 .     SOR_SYMMETRIC_SWEEP - SSOR (symmetric SOR)
3940 .     SOR_LOCAL_FORWARD_SWEEP - local forward SOR
3941 .     SOR_LOCAL_BACKWARD_SWEEP - local forward SOR
3942 .     SOR_LOCAL_SYMMETRIC_SWEEP - local SSOR
3943 .     SOR_APPLY_UPPER, SOR_APPLY_LOWER - applies
3944          upper/lower triangular part of matrix to
3945          vector (with omega)
3946 .     SOR_ZERO_INITIAL_GUESS - zero initial guess
3947 
3948    Notes:
3949    SOR_LOCAL_FORWARD_SWEEP, SOR_LOCAL_BACKWARD_SWEEP, and
3950    SOR_LOCAL_SYMMETRIC_SWEEP perform separate independent smoothings
3951    on each processor.
3952 
3953    Application programmers will not generally use MatSOR() directly,
3954    but instead will employ the KSP/PC interface.
3955 
3956    Notes:
3957     for BAIJ, SBAIJ, and AIJ matrices with Inodes this does a block SOR smoothing, otherwise it does a pointwise smoothing
3958 
3959    Notes for Advanced Users:
3960    The flags are implemented as bitwise inclusive or operations.
3961    For example, use (SOR_ZERO_INITIAL_GUESS | SOR_SYMMETRIC_SWEEP)
3962    to specify a zero initial guess for SSOR.
3963 
3964    Most users should employ the simplified KSP interface for linear solvers
3965    instead of working directly with matrix algebra routines such as this.
3966    See, e.g., KSPCreate().
3967 
3968    Vectors x and b CANNOT be the same
3969 
3970    Developer Note: We should add block SOR support for AIJ matrices with block size set to great than one and no inodes
3971 
3972    Level: developer
3973 
3974    Concepts: matrices^relaxation
3975    Concepts: matrices^SOR
3976    Concepts: matrices^Gauss-Seidel
3977 
3978 @*/
3979 PetscErrorCode MatSOR(Mat mat,Vec b,PetscReal omega,MatSORType flag,PetscReal shift,PetscInt its,PetscInt lits,Vec x)
3980 {
3981   PetscErrorCode ierr;
3982 
3983   PetscFunctionBegin;
3984   PetscValidHeaderSpecific(mat,MAT_CLASSID,1);
3985   PetscValidType(mat,1);
3986   PetscValidHeaderSpecific(b,VEC_CLASSID,2);
3987   PetscValidHeaderSpecific(x,VEC_CLASSID,8);
3988   PetscCheckSameComm(mat,1,b,2);
3989   PetscCheckSameComm(mat,1,x,8);
3990   if (!mat->ops->sor) SETERRQ1(PetscObjectComm((PetscObject)mat),PETSC_ERR_SUP,"Mat type %s",((PetscObject)mat)->type_name);
3991   if (!mat->assembled) SETERRQ(PetscObjectComm((PetscObject)mat),PETSC_ERR_ARG_WRONGSTATE,"Not for unassembled matrix");
3992   if (mat->factortype) SETERRQ(PetscObjectComm((PetscObject)mat),PETSC_ERR_ARG_WRONGSTATE,"Not for factored matrix");
3993   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);
3994   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);
3995   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);
3996   if (its <= 0) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONG,"Relaxation requires global its %D positive",its);
3997   if (lits <= 0) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONG,"Relaxation requires local its %D positive",lits);
3998   if (b == x) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_IDN,"b and x vector cannot be the same");
3999 
4000   MatCheckPreallocated(mat,1);
4001   ierr = PetscLogEventBegin(MAT_SOR,mat,b,x,0);CHKERRQ(ierr);
4002   ierr =(*mat->ops->sor)(mat,b,omega,flag,shift,its,lits,x);CHKERRQ(ierr);
4003   ierr = PetscLogEventEnd(MAT_SOR,mat,b,x,0);CHKERRQ(ierr);
4004   ierr = PetscObjectStateIncrease((PetscObject)x);CHKERRQ(ierr);
4005   PetscFunctionReturn(0);
4006 }
4007 
4008 /*
4009       Default matrix copy routine.
4010 */
4011 PetscErrorCode MatCopy_Basic(Mat A,Mat B,MatStructure str)
4012 {
4013   PetscErrorCode    ierr;
4014   PetscInt          i,rstart = 0,rend = 0,nz;
4015   const PetscInt    *cwork;
4016   const PetscScalar *vwork;
4017 
4018   PetscFunctionBegin;
4019   if (B->assembled) {
4020     ierr = MatZeroEntries(B);CHKERRQ(ierr);
4021   }
4022   ierr = MatGetOwnershipRange(A,&rstart,&rend);CHKERRQ(ierr);
4023   for (i=rstart; i<rend; i++) {
4024     ierr = MatGetRow(A,i,&nz,&cwork,&vwork);CHKERRQ(ierr);
4025     ierr = MatSetValues(B,1,&i,nz,cwork,vwork,INSERT_VALUES);CHKERRQ(ierr);
4026     ierr = MatRestoreRow(A,i,&nz,&cwork,&vwork);CHKERRQ(ierr);
4027   }
4028   ierr = MatAssemblyBegin(B,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
4029   ierr = MatAssemblyEnd(B,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
4030   PetscFunctionReturn(0);
4031 }
4032 
4033 /*@
4034    MatCopy - Copys a matrix to another matrix.
4035 
4036    Collective on Mat
4037 
4038    Input Parameters:
4039 +  A - the matrix
4040 -  str - SAME_NONZERO_PATTERN or DIFFERENT_NONZERO_PATTERN
4041 
4042    Output Parameter:
4043 .  B - where the copy is put
4044 
4045    Notes:
4046    If you use SAME_NONZERO_PATTERN then the two matrices had better have the
4047    same nonzero pattern or the routine will crash.
4048 
4049    MatCopy() copies the matrix entries of a matrix to another existing
4050    matrix (after first zeroing the second matrix).  A related routine is
4051    MatConvert(), which first creates a new matrix and then copies the data.
4052 
4053    Level: intermediate
4054 
4055    Concepts: matrices^copying
4056 
4057 .seealso: MatConvert(), MatDuplicate()
4058 
4059 @*/
4060 PetscErrorCode MatCopy(Mat A,Mat B,MatStructure str)
4061 {
4062   PetscErrorCode ierr;
4063   PetscInt       i;
4064 
4065   PetscFunctionBegin;
4066   PetscValidHeaderSpecific(A,MAT_CLASSID,1);
4067   PetscValidHeaderSpecific(B,MAT_CLASSID,2);
4068   PetscValidType(A,1);
4069   PetscValidType(B,2);
4070   PetscCheckSameComm(A,1,B,2);
4071   MatCheckPreallocated(B,2);
4072   if (!A->assembled) SETERRQ(PetscObjectComm((PetscObject)A),PETSC_ERR_ARG_WRONGSTATE,"Not for unassembled matrix");
4073   if (A->factortype) SETERRQ(PetscObjectComm((PetscObject)A),PETSC_ERR_ARG_WRONGSTATE,"Not for factored matrix");
4074   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);
4075   MatCheckPreallocated(A,1);
4076   if (A == B) PetscFunctionReturn(0);
4077 
4078   ierr = PetscLogEventBegin(MAT_Copy,A,B,0,0);CHKERRQ(ierr);
4079   if (A->ops->copy) {
4080     ierr = (*A->ops->copy)(A,B,str);CHKERRQ(ierr);
4081   } else { /* generic conversion */
4082     ierr = MatCopy_Basic(A,B,str);CHKERRQ(ierr);
4083   }
4084 
4085   B->stencil.dim = A->stencil.dim;
4086   B->stencil.noc = A->stencil.noc;
4087   for (i=0; i<=A->stencil.dim; i++) {
4088     B->stencil.dims[i]   = A->stencil.dims[i];
4089     B->stencil.starts[i] = A->stencil.starts[i];
4090   }
4091 
4092   ierr = PetscLogEventEnd(MAT_Copy,A,B,0,0);CHKERRQ(ierr);
4093   ierr = PetscObjectStateIncrease((PetscObject)B);CHKERRQ(ierr);
4094   PetscFunctionReturn(0);
4095 }
4096 
4097 /*@C
4098    MatConvert - Converts a matrix to another matrix, either of the same
4099    or different type.
4100 
4101    Collective on Mat
4102 
4103    Input Parameters:
4104 +  mat - the matrix
4105 .  newtype - new matrix type.  Use MATSAME to create a new matrix of the
4106    same type as the original matrix.
4107 -  reuse - denotes if the destination matrix is to be created or reused.
4108    Use MAT_INPLACE_MATRIX for inplace conversion (that is when you want the input mat to be changed to contain the matrix in the new format), otherwise use
4109    MAT_INITIAL_MATRIX or MAT_REUSE_MATRIX (can only be used after the first call was made with MAT_INITIAL_MATRIX, causes the matrix space in M to be reused).
4110 
4111    Output Parameter:
4112 .  M - pointer to place new matrix
4113 
4114    Notes:
4115    MatConvert() first creates a new matrix and then copies the data from
4116    the first matrix.  A related routine is MatCopy(), which copies the matrix
4117    entries of one matrix to another already existing matrix context.
4118 
4119    Cannot be used to convert a sequential matrix to parallel or parallel to sequential,
4120    the MPI communicator of the generated matrix is always the same as the communicator
4121    of the input matrix.
4122 
4123    Level: intermediate
4124 
4125    Concepts: matrices^converting between storage formats
4126 
4127 .seealso: MatCopy(), MatDuplicate()
4128 @*/
4129 PetscErrorCode MatConvert(Mat mat, MatType newtype,MatReuse reuse,Mat *M)
4130 {
4131   PetscErrorCode ierr;
4132   PetscBool      sametype,issame,flg;
4133   char           convname[256],mtype[256];
4134   Mat            B;
4135 
4136   PetscFunctionBegin;
4137   PetscValidHeaderSpecific(mat,MAT_CLASSID,1);
4138   PetscValidType(mat,1);
4139   PetscValidPointer(M,3);
4140   if (!mat->assembled) SETERRQ(PetscObjectComm((PetscObject)mat),PETSC_ERR_ARG_WRONGSTATE,"Not for unassembled matrix");
4141   if (mat->factortype) SETERRQ(PetscObjectComm((PetscObject)mat),PETSC_ERR_ARG_WRONGSTATE,"Not for factored matrix");
4142   MatCheckPreallocated(mat,1);
4143 
4144   ierr = PetscOptionsGetString(((PetscObject)mat)->options,((PetscObject)mat)->prefix,"-matconvert_type",mtype,256,&flg);CHKERRQ(ierr);
4145   if (flg) {
4146     newtype = mtype;
4147   }
4148   ierr = PetscObjectTypeCompare((PetscObject)mat,newtype,&sametype);CHKERRQ(ierr);
4149   ierr = PetscStrcmp(newtype,"same",&issame);CHKERRQ(ierr);
4150   if ((reuse == MAT_INPLACE_MATRIX) && (mat != *M)) SETERRQ(PetscObjectComm((PetscObject)mat),PETSC_ERR_SUP,"MAT_INPLACE_MATRIX requires same input and output matrix");
4151   if ((reuse == MAT_REUSE_MATRIX) && (mat == *M)) SETERRQ(PetscObjectComm((PetscObject)mat),PETSC_ERR_SUP,"MAT_REUSE_MATRIX means reuse matrix in final argument, perhaps you mean MAT_INPLACE_MATRIX");
4152 
4153   if ((reuse == MAT_INPLACE_MATRIX) && (issame || sametype)) PetscFunctionReturn(0);
4154 
4155   if ((sametype || issame) && (reuse==MAT_INITIAL_MATRIX) && mat->ops->duplicate) {
4156     ierr = (*mat->ops->duplicate)(mat,MAT_COPY_VALUES,M);CHKERRQ(ierr);
4157   } else {
4158     PetscErrorCode (*conv)(Mat, MatType,MatReuse,Mat*)=NULL;
4159     const char     *prefix[3] = {"seq","mpi",""};
4160     PetscInt       i;
4161     /*
4162        Order of precedence:
4163        0) See if newtype is a superclass of the current matrix.
4164        1) See if a specialized converter is known to the current matrix.
4165        2) See if a specialized converter is known to the desired matrix class.
4166        3) See if a good general converter is registered for the desired class
4167           (as of 6/27/03 only MATMPIADJ falls into this category).
4168        4) See if a good general converter is known for the current matrix.
4169        5) Use a really basic converter.
4170     */
4171 
4172     /* 0) See if newtype is a superclass of the current matrix.
4173           i.e mat is mpiaij and newtype is aij */
4174     for (i=0; i<2; i++) {
4175       ierr = PetscStrncpy(convname,prefix[i],sizeof(convname));CHKERRQ(ierr);
4176       ierr = PetscStrlcat(convname,newtype,sizeof(convname));CHKERRQ(ierr);
4177       ierr = PetscStrcmp(convname,((PetscObject)mat)->type_name,&flg);CHKERRQ(ierr);
4178       if (flg) {
4179         if (reuse == MAT_INPLACE_MATRIX) {
4180           PetscFunctionReturn(0);
4181         } else if (reuse == MAT_INITIAL_MATRIX && mat->ops->duplicate) {
4182           ierr = (*mat->ops->duplicate)(mat,MAT_COPY_VALUES,M);CHKERRQ(ierr);
4183           PetscFunctionReturn(0);
4184         } else if (reuse == MAT_REUSE_MATRIX && mat->ops->copy) {
4185           ierr = MatCopy(mat,*M,SAME_NONZERO_PATTERN);CHKERRQ(ierr);
4186           PetscFunctionReturn(0);
4187         }
4188       }
4189     }
4190     /* 1) See if a specialized converter is known to the current matrix and the desired class */
4191     for (i=0; i<3; i++) {
4192       ierr = PetscStrncpy(convname,"MatConvert_",sizeof(convname));CHKERRQ(ierr);
4193       ierr = PetscStrlcat(convname,((PetscObject)mat)->type_name,sizeof(convname));CHKERRQ(ierr);
4194       ierr = PetscStrlcat(convname,"_",sizeof(convname));CHKERRQ(ierr);
4195       ierr = PetscStrlcat(convname,prefix[i],sizeof(convname));CHKERRQ(ierr);
4196       ierr = PetscStrlcat(convname,issame ? ((PetscObject)mat)->type_name : newtype,sizeof(convname));CHKERRQ(ierr);
4197       ierr = PetscStrlcat(convname,"_C",sizeof(convname));CHKERRQ(ierr);
4198       ierr = PetscObjectQueryFunction((PetscObject)mat,convname,&conv);CHKERRQ(ierr);
4199       if (conv) goto foundconv;
4200     }
4201 
4202     /* 2)  See if a specialized converter is known to the desired matrix class. */
4203     ierr = MatCreate(PetscObjectComm((PetscObject)mat),&B);CHKERRQ(ierr);
4204     ierr = MatSetSizes(B,mat->rmap->n,mat->cmap->n,mat->rmap->N,mat->cmap->N);CHKERRQ(ierr);
4205     ierr = MatSetType(B,newtype);CHKERRQ(ierr);
4206     for (i=0; i<3; i++) {
4207       ierr = PetscStrncpy(convname,"MatConvert_",sizeof(convname));CHKERRQ(ierr);
4208       ierr = PetscStrlcat(convname,((PetscObject)mat)->type_name,sizeof(convname));CHKERRQ(ierr);
4209       ierr = PetscStrlcat(convname,"_",sizeof(convname));CHKERRQ(ierr);
4210       ierr = PetscStrlcat(convname,prefix[i],sizeof(convname));CHKERRQ(ierr);
4211       ierr = PetscStrlcat(convname,newtype,sizeof(convname));CHKERRQ(ierr);
4212       ierr = PetscStrlcat(convname,"_C",sizeof(convname));CHKERRQ(ierr);
4213       ierr = PetscObjectQueryFunction((PetscObject)B,convname,&conv);CHKERRQ(ierr);
4214       if (conv) {
4215         ierr = MatDestroy(&B);CHKERRQ(ierr);
4216         goto foundconv;
4217       }
4218     }
4219 
4220     /* 3) See if a good general converter is registered for the desired class */
4221     conv = B->ops->convertfrom;
4222     ierr = MatDestroy(&B);CHKERRQ(ierr);
4223     if (conv) goto foundconv;
4224 
4225     /* 4) See if a good general converter is known for the current matrix */
4226     if (mat->ops->convert) {
4227       conv = mat->ops->convert;
4228     }
4229     if (conv) goto foundconv;
4230 
4231     /* 5) Use a really basic converter. */
4232     conv = MatConvert_Basic;
4233 
4234 foundconv:
4235     ierr = PetscLogEventBegin(MAT_Convert,mat,0,0,0);CHKERRQ(ierr);
4236     ierr = (*conv)(mat,newtype,reuse,M);CHKERRQ(ierr);
4237     if (mat->rmap->mapping && mat->cmap->mapping && !(*M)->rmap->mapping && !(*M)->cmap->mapping) {
4238       /* the block sizes must be same if the mappings are copied over */
4239       (*M)->rmap->bs = mat->rmap->bs;
4240       (*M)->cmap->bs = mat->cmap->bs;
4241       ierr = PetscObjectReference((PetscObject)mat->rmap->mapping);CHKERRQ(ierr);
4242       ierr = PetscObjectReference((PetscObject)mat->cmap->mapping);CHKERRQ(ierr);
4243       (*M)->rmap->mapping = mat->rmap->mapping;
4244       (*M)->cmap->mapping = mat->cmap->mapping;
4245     }
4246     (*M)->stencil.dim = mat->stencil.dim;
4247     (*M)->stencil.noc = mat->stencil.noc;
4248     for (i=0; i<=mat->stencil.dim; i++) {
4249       (*M)->stencil.dims[i]   = mat->stencil.dims[i];
4250       (*M)->stencil.starts[i] = mat->stencil.starts[i];
4251     }
4252     ierr = PetscLogEventEnd(MAT_Convert,mat,0,0,0);CHKERRQ(ierr);
4253   }
4254   ierr = PetscObjectStateIncrease((PetscObject)*M);CHKERRQ(ierr);
4255 
4256   /* Copy Mat options */
4257   if (mat->symmetric) {ierr = MatSetOption(*M,MAT_SYMMETRIC,PETSC_TRUE);CHKERRQ(ierr);}
4258   if (mat->hermitian) {ierr = MatSetOption(*M,MAT_HERMITIAN,PETSC_TRUE);CHKERRQ(ierr);}
4259   PetscFunctionReturn(0);
4260 }
4261 
4262 /*@C
4263    MatFactorGetSolverType - Returns name of the package providing the factorization routines
4264 
4265    Not Collective
4266 
4267    Input Parameter:
4268 .  mat - the matrix, must be a factored matrix
4269 
4270    Output Parameter:
4271 .   type - the string name of the package (do not free this string)
4272 
4273    Notes:
4274       In Fortran you pass in a empty string and the package name will be copied into it.
4275     (Make sure the string is long enough)
4276 
4277    Level: intermediate
4278 
4279 .seealso: MatCopy(), MatDuplicate(), MatGetFactorAvailable(), MatGetFactor()
4280 @*/
4281 PetscErrorCode MatFactorGetSolverType(Mat mat, MatSolverType *type)
4282 {
4283   PetscErrorCode ierr, (*conv)(Mat,MatSolverType*);
4284 
4285   PetscFunctionBegin;
4286   PetscValidHeaderSpecific(mat,MAT_CLASSID,1);
4287   PetscValidType(mat,1);
4288   if (!mat->factortype) SETERRQ(PetscObjectComm((PetscObject)mat),PETSC_ERR_ARG_WRONGSTATE,"Only for factored matrix");
4289   ierr = PetscObjectQueryFunction((PetscObject)mat,"MatFactorGetSolverType_C",&conv);CHKERRQ(ierr);
4290   if (!conv) {
4291     *type = MATSOLVERPETSC;
4292   } else {
4293     ierr = (*conv)(mat,type);CHKERRQ(ierr);
4294   }
4295   PetscFunctionReturn(0);
4296 }
4297 
4298 typedef struct _MatSolverTypeForSpecifcType* MatSolverTypeForSpecifcType;
4299 struct _MatSolverTypeForSpecifcType {
4300   MatType                        mtype;
4301   PetscErrorCode                 (*getfactor[4])(Mat,MatFactorType,Mat*);
4302   MatSolverTypeForSpecifcType next;
4303 };
4304 
4305 typedef struct _MatSolverTypeHolder* MatSolverTypeHolder;
4306 struct _MatSolverTypeHolder {
4307   char                           *name;
4308   MatSolverTypeForSpecifcType handlers;
4309   MatSolverTypeHolder         next;
4310 };
4311 
4312 static MatSolverTypeHolder MatSolverTypeHolders = NULL;
4313 
4314 /*@C
4315    MatSolvePackageRegister - Registers a MatSolverType that works for a particular matrix type
4316 
4317    Input Parameters:
4318 +    package - name of the package, for example petsc or superlu
4319 .    mtype - the matrix type that works with this package
4320 .    ftype - the type of factorization supported by the package
4321 -    getfactor - routine that will create the factored matrix ready to be used
4322 
4323     Level: intermediate
4324 
4325 .seealso: MatCopy(), MatDuplicate(), MatGetFactorAvailable()
4326 @*/
4327 PetscErrorCode MatSolverTypeRegister(MatSolverType package,MatType mtype,MatFactorType ftype,PetscErrorCode (*getfactor)(Mat,MatFactorType,Mat*))
4328 {
4329   PetscErrorCode              ierr;
4330   MatSolverTypeHolder         next = MatSolverTypeHolders,prev;
4331   PetscBool                   flg;
4332   MatSolverTypeForSpecifcType inext,iprev = NULL;
4333 
4334   PetscFunctionBegin;
4335   ierr = MatInitializePackage();CHKERRQ(ierr);
4336   if (!next) {
4337     ierr = PetscNew(&MatSolverTypeHolders);CHKERRQ(ierr);
4338     ierr = PetscStrallocpy(package,&MatSolverTypeHolders->name);CHKERRQ(ierr);
4339     ierr = PetscNew(&MatSolverTypeHolders->handlers);CHKERRQ(ierr);
4340     ierr = PetscStrallocpy(mtype,(char **)&MatSolverTypeHolders->handlers->mtype);CHKERRQ(ierr);
4341     MatSolverTypeHolders->handlers->getfactor[(int)ftype-1] = getfactor;
4342     PetscFunctionReturn(0);
4343   }
4344   while (next) {
4345     ierr = PetscStrcasecmp(package,next->name,&flg);CHKERRQ(ierr);
4346     if (flg) {
4347       if (!next->handlers) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_PLIB,"MatSolverTypeHolder is missing handlers");
4348       inext = next->handlers;
4349       while (inext) {
4350         ierr = PetscStrcasecmp(mtype,inext->mtype,&flg);CHKERRQ(ierr);
4351         if (flg) {
4352           inext->getfactor[(int)ftype-1] = getfactor;
4353           PetscFunctionReturn(0);
4354         }
4355         iprev = inext;
4356         inext = inext->next;
4357       }
4358       ierr = PetscNew(&iprev->next);CHKERRQ(ierr);
4359       ierr = PetscStrallocpy(mtype,(char **)&iprev->next->mtype);CHKERRQ(ierr);
4360       iprev->next->getfactor[(int)ftype-1] = getfactor;
4361       PetscFunctionReturn(0);
4362     }
4363     prev = next;
4364     next = next->next;
4365   }
4366   ierr = PetscNew(&prev->next);CHKERRQ(ierr);
4367   ierr = PetscStrallocpy(package,&prev->next->name);CHKERRQ(ierr);
4368   ierr = PetscNew(&prev->next->handlers);CHKERRQ(ierr);
4369   ierr = PetscStrallocpy(mtype,(char **)&prev->next->handlers->mtype);CHKERRQ(ierr);
4370   prev->next->handlers->getfactor[(int)ftype-1] = getfactor;
4371   PetscFunctionReturn(0);
4372 }
4373 
4374 /*@C
4375    MatSolvePackageGet - Get's the function that creates the factor matrix if it exist
4376 
4377    Input Parameters:
4378 +    package - name of the package, for example petsc or superlu
4379 .    ftype - the type of factorization supported by the package
4380 -    mtype - the matrix type that works with this package
4381 
4382    Output Parameters:
4383 +   foundpackage - PETSC_TRUE if the package was registered
4384 .   foundmtype - PETSC_TRUE if the package supports the requested mtype
4385 -   getfactor - routine that will create the factored matrix ready to be used or NULL if not found
4386 
4387     Level: intermediate
4388 
4389 .seealso: MatCopy(), MatDuplicate(), MatGetFactorAvailable()
4390 @*/
4391 PetscErrorCode MatSolverTypeGet(MatSolverType package,MatType mtype,MatFactorType ftype,PetscBool *foundpackage,PetscBool *foundmtype,PetscErrorCode (**getfactor)(Mat,MatFactorType,Mat*))
4392 {
4393   PetscErrorCode                 ierr;
4394   MatSolverTypeHolder         next = MatSolverTypeHolders;
4395   PetscBool                      flg;
4396   MatSolverTypeForSpecifcType inext;
4397 
4398   PetscFunctionBegin;
4399   if (foundpackage) *foundpackage = PETSC_FALSE;
4400   if (foundmtype)   *foundmtype   = PETSC_FALSE;
4401   if (getfactor)    *getfactor    = NULL;
4402 
4403   if (package) {
4404     while (next) {
4405       ierr = PetscStrcasecmp(package,next->name,&flg);CHKERRQ(ierr);
4406       if (flg) {
4407         if (foundpackage) *foundpackage = PETSC_TRUE;
4408         inext = next->handlers;
4409         while (inext) {
4410           ierr = PetscStrbeginswith(mtype,inext->mtype,&flg);CHKERRQ(ierr);
4411           if (flg) {
4412             if (foundmtype) *foundmtype = PETSC_TRUE;
4413             if (getfactor)  *getfactor  = inext->getfactor[(int)ftype-1];
4414             PetscFunctionReturn(0);
4415           }
4416           inext = inext->next;
4417         }
4418       }
4419       next = next->next;
4420     }
4421   } else {
4422     while (next) {
4423       inext = next->handlers;
4424       while (inext) {
4425         ierr = PetscStrbeginswith(mtype,inext->mtype,&flg);CHKERRQ(ierr);
4426         if (flg && inext->getfactor[(int)ftype-1]) {
4427           if (foundpackage) *foundpackage = PETSC_TRUE;
4428           if (foundmtype)   *foundmtype   = PETSC_TRUE;
4429           if (getfactor)    *getfactor    = inext->getfactor[(int)ftype-1];
4430           PetscFunctionReturn(0);
4431         }
4432         inext = inext->next;
4433       }
4434       next = next->next;
4435     }
4436   }
4437   PetscFunctionReturn(0);
4438 }
4439 
4440 PetscErrorCode MatSolverTypeDestroy(void)
4441 {
4442   PetscErrorCode              ierr;
4443   MatSolverTypeHolder         next = MatSolverTypeHolders,prev;
4444   MatSolverTypeForSpecifcType inext,iprev;
4445 
4446   PetscFunctionBegin;
4447   while (next) {
4448     ierr = PetscFree(next->name);CHKERRQ(ierr);
4449     inext = next->handlers;
4450     while (inext) {
4451       ierr = PetscFree(inext->mtype);CHKERRQ(ierr);
4452       iprev = inext;
4453       inext = inext->next;
4454       ierr = PetscFree(iprev);CHKERRQ(ierr);
4455     }
4456     prev = next;
4457     next = next->next;
4458     ierr = PetscFree(prev);CHKERRQ(ierr);
4459   }
4460   MatSolverTypeHolders = NULL;
4461   PetscFunctionReturn(0);
4462 }
4463 
4464 /*@C
4465    MatGetFactor - Returns a matrix suitable to calls to MatXXFactorSymbolic()
4466 
4467    Collective on Mat
4468 
4469    Input Parameters:
4470 +  mat - the matrix
4471 .  type - name of solver type, for example, superlu, petsc (to use PETSc's default)
4472 -  ftype - factor type, MAT_FACTOR_LU, MAT_FACTOR_CHOLESKY, MAT_FACTOR_ICC, MAT_FACTOR_ILU,
4473 
4474    Output Parameters:
4475 .  f - the factor matrix used with MatXXFactorSymbolic() calls
4476 
4477    Notes:
4478       Some PETSc matrix formats have alternative solvers available that are contained in alternative packages
4479      such as pastix, superlu, mumps etc.
4480 
4481       PETSc must have been ./configure to use the external solver, using the option --download-package
4482 
4483    Level: intermediate
4484 
4485 .seealso: MatCopy(), MatDuplicate(), MatGetFactorAvailable()
4486 @*/
4487 PetscErrorCode MatGetFactor(Mat mat, MatSolverType type,MatFactorType ftype,Mat *f)
4488 {
4489   PetscErrorCode ierr,(*conv)(Mat,MatFactorType,Mat*);
4490   PetscBool      foundpackage,foundmtype;
4491 
4492   PetscFunctionBegin;
4493   PetscValidHeaderSpecific(mat,MAT_CLASSID,1);
4494   PetscValidType(mat,1);
4495 
4496   if (mat->factortype) SETERRQ(PetscObjectComm((PetscObject)mat),PETSC_ERR_ARG_WRONGSTATE,"Not for factored matrix");
4497   MatCheckPreallocated(mat,1);
4498 
4499   ierr = MatSolverTypeGet(type,((PetscObject)mat)->type_name,ftype,&foundpackage,&foundmtype,&conv);CHKERRQ(ierr);
4500   if (!foundpackage) {
4501     if (type) {
4502       SETERRQ2(PetscObjectComm((PetscObject)mat),PETSC_ERR_MISSING_FACTOR,"Could not locate solver package %s. Perhaps you must ./configure with --download-%s",type,type);
4503     } else {
4504       SETERRQ(PetscObjectComm((PetscObject)mat),PETSC_ERR_MISSING_FACTOR,"Could not locate a solver package. Perhaps you must ./configure with --download-<package>");
4505     }
4506   }
4507 
4508   if (!foundmtype) SETERRQ2(PetscObjectComm((PetscObject)mat),PETSC_ERR_MISSING_FACTOR,"MatSolverType %s does not support matrix type %s",type,((PetscObject)mat)->type_name);
4509   if (!conv) SETERRQ3(PetscObjectComm((PetscObject)mat),PETSC_ERR_MISSING_FACTOR,"MatSolverType %s does not support factorization type %s for  matrix type %s",type,MatFactorTypes[ftype],((PetscObject)mat)->type_name);
4510 
4511 #if defined(PETSC_USE_COMPLEX)
4512   if (mat->hermitian && !mat->symmetric && (ftype == MAT_FACTOR_CHOLESKY||ftype == MAT_FACTOR_ICC)) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"Hermitian CHOLESKY or ICC Factor is not supported");
4513 #endif
4514 
4515   ierr = (*conv)(mat,ftype,f);CHKERRQ(ierr);
4516   PetscFunctionReturn(0);
4517 }
4518 
4519 /*@C
4520    MatGetFactorAvailable - Returns a a flag if matrix supports particular package and factor type
4521 
4522    Not Collective
4523 
4524    Input Parameters:
4525 +  mat - the matrix
4526 .  type - name of solver type, for example, superlu, petsc (to use PETSc's default)
4527 -  ftype - factor type, MAT_FACTOR_LU, MAT_FACTOR_CHOLESKY, MAT_FACTOR_ICC, MAT_FACTOR_ILU,
4528 
4529    Output Parameter:
4530 .    flg - PETSC_TRUE if the factorization is available
4531 
4532    Notes:
4533       Some PETSc matrix formats have alternative solvers available that are contained in alternative packages
4534      such as pastix, superlu, mumps etc.
4535 
4536       PETSc must have been ./configure to use the external solver, using the option --download-package
4537 
4538    Level: intermediate
4539 
4540 .seealso: MatCopy(), MatDuplicate(), MatGetFactor()
4541 @*/
4542 PetscErrorCode MatGetFactorAvailable(Mat mat, MatSolverType type,MatFactorType ftype,PetscBool  *flg)
4543 {
4544   PetscErrorCode ierr, (*gconv)(Mat,MatFactorType,Mat*);
4545 
4546   PetscFunctionBegin;
4547   PetscValidHeaderSpecific(mat,MAT_CLASSID,1);
4548   PetscValidType(mat,1);
4549 
4550   if (mat->factortype) SETERRQ(PetscObjectComm((PetscObject)mat),PETSC_ERR_ARG_WRONGSTATE,"Not for factored matrix");
4551   MatCheckPreallocated(mat,1);
4552 
4553   *flg = PETSC_FALSE;
4554   ierr = MatSolverTypeGet(type,((PetscObject)mat)->type_name,ftype,NULL,NULL,&gconv);CHKERRQ(ierr);
4555   if (gconv) {
4556     *flg = PETSC_TRUE;
4557   }
4558   PetscFunctionReturn(0);
4559 }
4560 
4561 #include <petscdmtypes.h>
4562 
4563 /*@
4564    MatDuplicate - Duplicates a matrix including the non-zero structure.
4565 
4566    Collective on Mat
4567 
4568    Input Parameters:
4569 +  mat - the matrix
4570 -  op - One of MAT_DO_NOT_COPY_VALUES, MAT_COPY_VALUES, or MAT_SHARE_NONZERO_PATTERN.
4571         See the manual page for MatDuplicateOption for an explanation of these options.
4572 
4573    Output Parameter:
4574 .  M - pointer to place new matrix
4575 
4576    Level: intermediate
4577 
4578    Concepts: matrices^duplicating
4579 
4580    Notes:
4581     You cannot change the nonzero pattern for the parent or child matrix if you use MAT_SHARE_NONZERO_PATTERN.
4582     When original mat is a product of matrix operation, e.g., an output of MatMatMult() or MatCreateSubMatrix(), only the simple matrix data structure of mat is duplicated and the internal data structures created for the reuse of previous matrix operations are not duplicated. User should not use MatDuplicate() to create new matrix M if M is intended to be reused as the product of matrix operation.
4583 
4584 .seealso: MatCopy(), MatConvert(), MatDuplicateOption
4585 @*/
4586 PetscErrorCode MatDuplicate(Mat mat,MatDuplicateOption op,Mat *M)
4587 {
4588   PetscErrorCode ierr;
4589   Mat            B;
4590   PetscInt       i;
4591   DM             dm;
4592   void           (*viewf)(void);
4593 
4594   PetscFunctionBegin;
4595   PetscValidHeaderSpecific(mat,MAT_CLASSID,1);
4596   PetscValidType(mat,1);
4597   PetscValidPointer(M,3);
4598   if (op == MAT_COPY_VALUES && !mat->assembled) SETERRQ(PetscObjectComm((PetscObject)mat),PETSC_ERR_ARG_WRONGSTATE,"MAT_COPY_VALUES not allowed for unassembled matrix");
4599   if (mat->factortype) SETERRQ(PetscObjectComm((PetscObject)mat),PETSC_ERR_ARG_WRONGSTATE,"Not for factored matrix");
4600   MatCheckPreallocated(mat,1);
4601 
4602   *M = 0;
4603   if (!mat->ops->duplicate) SETERRQ(PetscObjectComm((PetscObject)mat),PETSC_ERR_SUP,"Not written for this matrix type");
4604   ierr = PetscLogEventBegin(MAT_Convert,mat,0,0,0);CHKERRQ(ierr);
4605   ierr = (*mat->ops->duplicate)(mat,op,M);CHKERRQ(ierr);
4606   B    = *M;
4607 
4608   ierr = MatGetOperation(mat,MATOP_VIEW,&viewf);CHKERRQ(ierr);
4609   if (viewf) {
4610     ierr = MatSetOperation(B,MATOP_VIEW,viewf);CHKERRQ(ierr);
4611   }
4612 
4613   B->stencil.dim = mat->stencil.dim;
4614   B->stencil.noc = mat->stencil.noc;
4615   for (i=0; i<=mat->stencil.dim; i++) {
4616     B->stencil.dims[i]   = mat->stencil.dims[i];
4617     B->stencil.starts[i] = mat->stencil.starts[i];
4618   }
4619 
4620   B->nooffproczerorows = mat->nooffproczerorows;
4621   B->nooffprocentries  = mat->nooffprocentries;
4622 
4623   ierr = PetscObjectQuery((PetscObject) mat, "__PETSc_dm", (PetscObject*) &dm);CHKERRQ(ierr);
4624   if (dm) {
4625     ierr = PetscObjectCompose((PetscObject) B, "__PETSc_dm", (PetscObject) dm);CHKERRQ(ierr);
4626   }
4627   ierr = PetscLogEventEnd(MAT_Convert,mat,0,0,0);CHKERRQ(ierr);
4628   ierr = PetscObjectStateIncrease((PetscObject)B);CHKERRQ(ierr);
4629   PetscFunctionReturn(0);
4630 }
4631 
4632 /*@
4633    MatGetDiagonal - Gets the diagonal of a matrix.
4634 
4635    Logically Collective on Mat and Vec
4636 
4637    Input Parameters:
4638 +  mat - the matrix
4639 -  v - the vector for storing the diagonal
4640 
4641    Output Parameter:
4642 .  v - the diagonal of the matrix
4643 
4644    Level: intermediate
4645 
4646    Note:
4647    Currently only correct in parallel for square matrices.
4648 
4649    Concepts: matrices^accessing diagonals
4650 
4651 .seealso: MatGetRow(), MatCreateSubMatrices(), MatCreateSubMatrix(), MatGetRowMaxAbs()
4652 @*/
4653 PetscErrorCode MatGetDiagonal(Mat mat,Vec v)
4654 {
4655   PetscErrorCode ierr;
4656 
4657   PetscFunctionBegin;
4658   PetscValidHeaderSpecific(mat,MAT_CLASSID,1);
4659   PetscValidType(mat,1);
4660   PetscValidHeaderSpecific(v,VEC_CLASSID,2);
4661   if (!mat->assembled) SETERRQ(PetscObjectComm((PetscObject)mat),PETSC_ERR_ARG_WRONGSTATE,"Not for unassembled matrix");
4662   if (!mat->ops->getdiagonal) SETERRQ1(PetscObjectComm((PetscObject)mat),PETSC_ERR_SUP,"Mat type %s",((PetscObject)mat)->type_name);
4663   MatCheckPreallocated(mat,1);
4664 
4665   ierr = (*mat->ops->getdiagonal)(mat,v);CHKERRQ(ierr);
4666   ierr = PetscObjectStateIncrease((PetscObject)v);CHKERRQ(ierr);
4667   PetscFunctionReturn(0);
4668 }
4669 
4670 /*@C
4671    MatGetRowMin - Gets the minimum value (of the real part) of each
4672         row of the matrix
4673 
4674    Logically Collective on Mat and Vec
4675 
4676    Input Parameters:
4677 .  mat - the matrix
4678 
4679    Output Parameter:
4680 +  v - the vector for storing the maximums
4681 -  idx - the indices of the column found for each row (optional)
4682 
4683    Level: intermediate
4684 
4685    Notes:
4686     The result of this call are the same as if one converted the matrix to dense format
4687       and found the minimum value in each row (i.e. the implicit zeros are counted as zeros).
4688 
4689     This code is only implemented for a couple of matrix formats.
4690 
4691    Concepts: matrices^getting row maximums
4692 
4693 .seealso: MatGetDiagonal(), MatCreateSubMatrices(), MatCreateSubMatrix(), MatGetRowMaxAbs(),
4694           MatGetRowMax()
4695 @*/
4696 PetscErrorCode MatGetRowMin(Mat mat,Vec v,PetscInt idx[])
4697 {
4698   PetscErrorCode ierr;
4699 
4700   PetscFunctionBegin;
4701   PetscValidHeaderSpecific(mat,MAT_CLASSID,1);
4702   PetscValidType(mat,1);
4703   PetscValidHeaderSpecific(v,VEC_CLASSID,2);
4704   if (!mat->assembled) SETERRQ(PetscObjectComm((PetscObject)mat),PETSC_ERR_ARG_WRONGSTATE,"Not for unassembled matrix");
4705   if (!mat->ops->getrowmax) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_SUP,"Mat type %s",((PetscObject)mat)->type_name);
4706   MatCheckPreallocated(mat,1);
4707 
4708   ierr = (*mat->ops->getrowmin)(mat,v,idx);CHKERRQ(ierr);
4709   ierr = PetscObjectStateIncrease((PetscObject)v);CHKERRQ(ierr);
4710   PetscFunctionReturn(0);
4711 }
4712 
4713 /*@C
4714    MatGetRowMinAbs - Gets the minimum value (in absolute value) of each
4715         row of the matrix
4716 
4717    Logically Collective on Mat and Vec
4718 
4719    Input Parameters:
4720 .  mat - the matrix
4721 
4722    Output Parameter:
4723 +  v - the vector for storing the minimums
4724 -  idx - the indices of the column found for each row (or NULL if not needed)
4725 
4726    Level: intermediate
4727 
4728    Notes:
4729     if a row is completely empty or has only 0.0 values then the idx[] value for that
4730     row is 0 (the first column).
4731 
4732     This code is only implemented for a couple of matrix formats.
4733 
4734    Concepts: matrices^getting row maximums
4735 
4736 .seealso: MatGetDiagonal(), MatCreateSubMatrices(), MatCreateSubMatrix(), MatGetRowMax(), MatGetRowMaxAbs(), MatGetRowMin()
4737 @*/
4738 PetscErrorCode MatGetRowMinAbs(Mat mat,Vec v,PetscInt idx[])
4739 {
4740   PetscErrorCode ierr;
4741 
4742   PetscFunctionBegin;
4743   PetscValidHeaderSpecific(mat,MAT_CLASSID,1);
4744   PetscValidType(mat,1);
4745   PetscValidHeaderSpecific(v,VEC_CLASSID,2);
4746   if (!mat->assembled) SETERRQ(PetscObjectComm((PetscObject)mat),PETSC_ERR_ARG_WRONGSTATE,"Not for unassembled matrix");
4747   if (!mat->ops->getrowminabs) SETERRQ1(PetscObjectComm((PetscObject)mat),PETSC_ERR_SUP,"Mat type %s",((PetscObject)mat)->type_name);
4748   MatCheckPreallocated(mat,1);
4749   if (idx) {ierr = PetscMemzero(idx,mat->rmap->n*sizeof(PetscInt));CHKERRQ(ierr);}
4750 
4751   ierr = (*mat->ops->getrowminabs)(mat,v,idx);CHKERRQ(ierr);
4752   ierr = PetscObjectStateIncrease((PetscObject)v);CHKERRQ(ierr);
4753   PetscFunctionReturn(0);
4754 }
4755 
4756 /*@C
4757    MatGetRowMax - Gets the maximum value (of the real part) of each
4758         row of the matrix
4759 
4760    Logically Collective on Mat and Vec
4761 
4762    Input Parameters:
4763 .  mat - the matrix
4764 
4765    Output Parameter:
4766 +  v - the vector for storing the maximums
4767 -  idx - the indices of the column found for each row (optional)
4768 
4769    Level: intermediate
4770 
4771    Notes:
4772     The result of this call are the same as if one converted the matrix to dense format
4773       and found the minimum value in each row (i.e. the implicit zeros are counted as zeros).
4774 
4775     This code is only implemented for a couple of matrix formats.
4776 
4777    Concepts: matrices^getting row maximums
4778 
4779 .seealso: MatGetDiagonal(), MatCreateSubMatrices(), MatCreateSubMatrix(), MatGetRowMaxAbs(), MatGetRowMin()
4780 @*/
4781 PetscErrorCode MatGetRowMax(Mat mat,Vec v,PetscInt idx[])
4782 {
4783   PetscErrorCode ierr;
4784 
4785   PetscFunctionBegin;
4786   PetscValidHeaderSpecific(mat,MAT_CLASSID,1);
4787   PetscValidType(mat,1);
4788   PetscValidHeaderSpecific(v,VEC_CLASSID,2);
4789   if (!mat->assembled) SETERRQ(PetscObjectComm((PetscObject)mat),PETSC_ERR_ARG_WRONGSTATE,"Not for unassembled matrix");
4790   if (!mat->ops->getrowmax) SETERRQ1(PetscObjectComm((PetscObject)mat),PETSC_ERR_SUP,"Mat type %s",((PetscObject)mat)->type_name);
4791   MatCheckPreallocated(mat,1);
4792 
4793   ierr = (*mat->ops->getrowmax)(mat,v,idx);CHKERRQ(ierr);
4794   ierr = PetscObjectStateIncrease((PetscObject)v);CHKERRQ(ierr);
4795   PetscFunctionReturn(0);
4796 }
4797 
4798 /*@C
4799    MatGetRowMaxAbs - Gets the maximum value (in absolute value) of each
4800         row of the matrix
4801 
4802    Logically Collective on Mat and Vec
4803 
4804    Input Parameters:
4805 .  mat - the matrix
4806 
4807    Output Parameter:
4808 +  v - the vector for storing the maximums
4809 -  idx - the indices of the column found for each row (or NULL if not needed)
4810 
4811    Level: intermediate
4812 
4813    Notes:
4814     if a row is completely empty or has only 0.0 values then the idx[] value for that
4815     row is 0 (the first column).
4816 
4817     This code is only implemented for a couple of matrix formats.
4818 
4819    Concepts: matrices^getting row maximums
4820 
4821 .seealso: MatGetDiagonal(), MatCreateSubMatrices(), MatCreateSubMatrix(), MatGetRowMax(), MatGetRowMin()
4822 @*/
4823 PetscErrorCode MatGetRowMaxAbs(Mat mat,Vec v,PetscInt idx[])
4824 {
4825   PetscErrorCode ierr;
4826 
4827   PetscFunctionBegin;
4828   PetscValidHeaderSpecific(mat,MAT_CLASSID,1);
4829   PetscValidType(mat,1);
4830   PetscValidHeaderSpecific(v,VEC_CLASSID,2);
4831   if (!mat->assembled) SETERRQ(PetscObjectComm((PetscObject)mat),PETSC_ERR_ARG_WRONGSTATE,"Not for unassembled matrix");
4832   if (!mat->ops->getrowmaxabs) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_SUP,"Mat type %s",((PetscObject)mat)->type_name);
4833   MatCheckPreallocated(mat,1);
4834   if (idx) {ierr = PetscMemzero(idx,mat->rmap->n*sizeof(PetscInt));CHKERRQ(ierr);}
4835 
4836   ierr = (*mat->ops->getrowmaxabs)(mat,v,idx);CHKERRQ(ierr);
4837   ierr = PetscObjectStateIncrease((PetscObject)v);CHKERRQ(ierr);
4838   PetscFunctionReturn(0);
4839 }
4840 
4841 /*@
4842    MatGetRowSum - Gets the sum of each row of the matrix
4843 
4844    Logically or Neighborhood Collective on Mat and Vec
4845 
4846    Input Parameters:
4847 .  mat - the matrix
4848 
4849    Output Parameter:
4850 .  v - the vector for storing the sum of rows
4851 
4852    Level: intermediate
4853 
4854    Notes:
4855     This code is slow since it is not currently specialized for different formats
4856 
4857    Concepts: matrices^getting row sums
4858 
4859 .seealso: MatGetDiagonal(), MatCreateSubMatrices(), MatCreateSubMatrix(), MatGetRowMax(), MatGetRowMin()
4860 @*/
4861 PetscErrorCode MatGetRowSum(Mat mat, Vec v)
4862 {
4863   Vec            ones;
4864   PetscErrorCode ierr;
4865 
4866   PetscFunctionBegin;
4867   PetscValidHeaderSpecific(mat,MAT_CLASSID,1);
4868   PetscValidType(mat,1);
4869   PetscValidHeaderSpecific(v,VEC_CLASSID,2);
4870   if (!mat->assembled) SETERRQ(PetscObjectComm((PetscObject)mat),PETSC_ERR_ARG_WRONGSTATE,"Not for unassembled matrix");
4871   MatCheckPreallocated(mat,1);
4872   ierr = MatCreateVecs(mat,&ones,NULL);CHKERRQ(ierr);
4873   ierr = VecSet(ones,1.);CHKERRQ(ierr);
4874   ierr = MatMult(mat,ones,v);CHKERRQ(ierr);
4875   ierr = VecDestroy(&ones);CHKERRQ(ierr);
4876   PetscFunctionReturn(0);
4877 }
4878 
4879 /*@
4880    MatTranspose - Computes an in-place or out-of-place transpose of a matrix.
4881 
4882    Collective on Mat
4883 
4884    Input Parameter:
4885 +  mat - the matrix to transpose
4886 -  reuse - either MAT_INITIAL_MATRIX, MAT_REUSE_MATRIX, or MAT_INPLACE_MATRIX
4887 
4888    Output Parameters:
4889 .  B - the transpose
4890 
4891    Notes:
4892      If you use MAT_INPLACE_MATRIX then you must pass in &mat for B
4893 
4894      MAT_REUSE_MATRIX causes the B matrix from a previous call to this function with MAT_INITIAL_MATRIX to be used
4895 
4896      Consider using MatCreateTranspose() instead if you only need a matrix that behaves like the transpose, but don't need the storage to be changed.
4897 
4898    Level: intermediate
4899 
4900    Concepts: matrices^transposing
4901 
4902 .seealso: MatMultTranspose(), MatMultTransposeAdd(), MatIsTranspose(), MatReuse
4903 @*/
4904 PetscErrorCode MatTranspose(Mat mat,MatReuse reuse,Mat *B)
4905 {
4906   PetscErrorCode ierr;
4907 
4908   PetscFunctionBegin;
4909   PetscValidHeaderSpecific(mat,MAT_CLASSID,1);
4910   PetscValidType(mat,1);
4911   if (!mat->assembled) SETERRQ(PetscObjectComm((PetscObject)mat),PETSC_ERR_ARG_WRONGSTATE,"Not for unassembled matrix");
4912   if (mat->factortype) SETERRQ(PetscObjectComm((PetscObject)mat),PETSC_ERR_ARG_WRONGSTATE,"Not for factored matrix");
4913   if (!mat->ops->transpose) SETERRQ1(PetscObjectComm((PetscObject)mat),PETSC_ERR_SUP,"Mat type %s",((PetscObject)mat)->type_name);
4914   if (reuse == MAT_INPLACE_MATRIX && mat != *B) SETERRQ(PetscObjectComm((PetscObject)mat),PETSC_ERR_SUP,"MAT_INPLACE_MATRIX requires last matrix to match first");
4915   if (reuse == MAT_REUSE_MATRIX && mat == *B) SETERRQ(PetscObjectComm((PetscObject)mat),PETSC_ERR_SUP,"Perhaps you mean MAT_INPLACE_MATRIX");
4916   MatCheckPreallocated(mat,1);
4917 
4918   ierr = PetscLogEventBegin(MAT_Transpose,mat,0,0,0);CHKERRQ(ierr);
4919   ierr = (*mat->ops->transpose)(mat,reuse,B);CHKERRQ(ierr);
4920   ierr = PetscLogEventEnd(MAT_Transpose,mat,0,0,0);CHKERRQ(ierr);
4921   if (B) {ierr = PetscObjectStateIncrease((PetscObject)*B);CHKERRQ(ierr);}
4922   PetscFunctionReturn(0);
4923 }
4924 
4925 /*@
4926    MatIsTranspose - Test whether a matrix is another one's transpose,
4927         or its own, in which case it tests symmetry.
4928 
4929    Collective on Mat
4930 
4931    Input Parameter:
4932 +  A - the matrix to test
4933 -  B - the matrix to test against, this can equal the first parameter
4934 
4935    Output Parameters:
4936 .  flg - the result
4937 
4938    Notes:
4939    Only available for SeqAIJ/MPIAIJ matrices. The sequential algorithm
4940    has a running time of the order of the number of nonzeros; the parallel
4941    test involves parallel copies of the block-offdiagonal parts of the matrix.
4942 
4943    Level: intermediate
4944 
4945    Concepts: matrices^transposing, matrix^symmetry
4946 
4947 .seealso: MatTranspose(), MatIsSymmetric(), MatIsHermitian()
4948 @*/
4949 PetscErrorCode MatIsTranspose(Mat A,Mat B,PetscReal tol,PetscBool  *flg)
4950 {
4951   PetscErrorCode ierr,(*f)(Mat,Mat,PetscReal,PetscBool*),(*g)(Mat,Mat,PetscReal,PetscBool*);
4952 
4953   PetscFunctionBegin;
4954   PetscValidHeaderSpecific(A,MAT_CLASSID,1);
4955   PetscValidHeaderSpecific(B,MAT_CLASSID,2);
4956   PetscValidPointer(flg,3);
4957   ierr = PetscObjectQueryFunction((PetscObject)A,"MatIsTranspose_C",&f);CHKERRQ(ierr);
4958   ierr = PetscObjectQueryFunction((PetscObject)B,"MatIsTranspose_C",&g);CHKERRQ(ierr);
4959   *flg = PETSC_FALSE;
4960   if (f && g) {
4961     if (f == g) {
4962       ierr = (*f)(A,B,tol,flg);CHKERRQ(ierr);
4963     } else SETERRQ(PetscObjectComm((PetscObject)A),PETSC_ERR_ARG_NOTSAMETYPE,"Matrices do not have the same comparator for symmetry test");
4964   } else {
4965     MatType mattype;
4966     if (!f) {
4967       ierr = MatGetType(A,&mattype);CHKERRQ(ierr);
4968     } else {
4969       ierr = MatGetType(B,&mattype);CHKERRQ(ierr);
4970     }
4971     SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_SUP,"Matrix of type <%s> does not support checking for transpose",mattype);
4972   }
4973   PetscFunctionReturn(0);
4974 }
4975 
4976 /*@
4977    MatHermitianTranspose - Computes an in-place or out-of-place transpose of a matrix in complex conjugate.
4978 
4979    Collective on Mat
4980 
4981    Input Parameter:
4982 +  mat - the matrix to transpose and complex conjugate
4983 -  reuse - MAT_INITIAL_MATRIX to create a new matrix, MAT_INPLACE_MATRIX to reuse the first argument to store the transpose
4984 
4985    Output Parameters:
4986 .  B - the Hermitian
4987 
4988    Level: intermediate
4989 
4990    Concepts: matrices^transposing, complex conjugatex
4991 
4992 .seealso: MatTranspose(), MatMultTranspose(), MatMultTransposeAdd(), MatIsTranspose(), MatReuse
4993 @*/
4994 PetscErrorCode MatHermitianTranspose(Mat mat,MatReuse reuse,Mat *B)
4995 {
4996   PetscErrorCode ierr;
4997 
4998   PetscFunctionBegin;
4999   ierr = MatTranspose(mat,reuse,B);CHKERRQ(ierr);
5000 #if defined(PETSC_USE_COMPLEX)
5001   ierr = MatConjugate(*B);CHKERRQ(ierr);
5002 #endif
5003   PetscFunctionReturn(0);
5004 }
5005 
5006 /*@
5007    MatIsHermitianTranspose - Test whether a matrix is another one's Hermitian transpose,
5008 
5009    Collective on Mat
5010 
5011    Input Parameter:
5012 +  A - the matrix to test
5013 -  B - the matrix to test against, this can equal the first parameter
5014 
5015    Output Parameters:
5016 .  flg - the result
5017 
5018    Notes:
5019    Only available for SeqAIJ/MPIAIJ matrices. The sequential algorithm
5020    has a running time of the order of the number of nonzeros; the parallel
5021    test involves parallel copies of the block-offdiagonal parts of the matrix.
5022 
5023    Level: intermediate
5024 
5025    Concepts: matrices^transposing, matrix^symmetry
5026 
5027 .seealso: MatTranspose(), MatIsSymmetric(), MatIsHermitian(), MatIsTranspose()
5028 @*/
5029 PetscErrorCode MatIsHermitianTranspose(Mat A,Mat B,PetscReal tol,PetscBool  *flg)
5030 {
5031   PetscErrorCode ierr,(*f)(Mat,Mat,PetscReal,PetscBool*),(*g)(Mat,Mat,PetscReal,PetscBool*);
5032 
5033   PetscFunctionBegin;
5034   PetscValidHeaderSpecific(A,MAT_CLASSID,1);
5035   PetscValidHeaderSpecific(B,MAT_CLASSID,2);
5036   PetscValidPointer(flg,3);
5037   ierr = PetscObjectQueryFunction((PetscObject)A,"MatIsHermitianTranspose_C",&f);CHKERRQ(ierr);
5038   ierr = PetscObjectQueryFunction((PetscObject)B,"MatIsHermitianTranspose_C",&g);CHKERRQ(ierr);
5039   if (f && g) {
5040     if (f==g) {
5041       ierr = (*f)(A,B,tol,flg);CHKERRQ(ierr);
5042     } else SETERRQ(PetscObjectComm((PetscObject)A),PETSC_ERR_ARG_NOTSAMETYPE,"Matrices do not have the same comparator for Hermitian test");
5043   }
5044   PetscFunctionReturn(0);
5045 }
5046 
5047 /*@
5048    MatPermute - Creates a new matrix with rows and columns permuted from the
5049    original.
5050 
5051    Collective on Mat
5052 
5053    Input Parameters:
5054 +  mat - the matrix to permute
5055 .  row - row permutation, each processor supplies only the permutation for its rows
5056 -  col - column permutation, each processor supplies only the permutation for its columns
5057 
5058    Output Parameters:
5059 .  B - the permuted matrix
5060 
5061    Level: advanced
5062 
5063    Note:
5064    The index sets map from row/col of permuted matrix to row/col of original matrix.
5065    The index sets should be on the same communicator as Mat and have the same local sizes.
5066 
5067    Concepts: matrices^permuting
5068 
5069 .seealso: MatGetOrdering(), ISAllGather()
5070 
5071 @*/
5072 PetscErrorCode MatPermute(Mat mat,IS row,IS col,Mat *B)
5073 {
5074   PetscErrorCode ierr;
5075 
5076   PetscFunctionBegin;
5077   PetscValidHeaderSpecific(mat,MAT_CLASSID,1);
5078   PetscValidType(mat,1);
5079   PetscValidHeaderSpecific(row,IS_CLASSID,2);
5080   PetscValidHeaderSpecific(col,IS_CLASSID,3);
5081   PetscValidPointer(B,4);
5082   if (!mat->assembled) SETERRQ(PetscObjectComm((PetscObject)mat),PETSC_ERR_ARG_WRONGSTATE,"Not for unassembled matrix");
5083   if (mat->factortype) SETERRQ(PetscObjectComm((PetscObject)mat),PETSC_ERR_ARG_WRONGSTATE,"Not for factored matrix");
5084   if (!mat->ops->permute) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_SUP,"MatPermute not available for Mat type %s",((PetscObject)mat)->type_name);
5085   MatCheckPreallocated(mat,1);
5086 
5087   ierr = (*mat->ops->permute)(mat,row,col,B);CHKERRQ(ierr);
5088   ierr = PetscObjectStateIncrease((PetscObject)*B);CHKERRQ(ierr);
5089   PetscFunctionReturn(0);
5090 }
5091 
5092 /*@
5093    MatEqual - Compares two matrices.
5094 
5095    Collective on Mat
5096 
5097    Input Parameters:
5098 +  A - the first matrix
5099 -  B - the second matrix
5100 
5101    Output Parameter:
5102 .  flg - PETSC_TRUE if the matrices are equal; PETSC_FALSE otherwise.
5103 
5104    Level: intermediate
5105 
5106    Concepts: matrices^equality between
5107 @*/
5108 PetscErrorCode MatEqual(Mat A,Mat B,PetscBool  *flg)
5109 {
5110   PetscErrorCode ierr;
5111 
5112   PetscFunctionBegin;
5113   PetscValidHeaderSpecific(A,MAT_CLASSID,1);
5114   PetscValidHeaderSpecific(B,MAT_CLASSID,2);
5115   PetscValidType(A,1);
5116   PetscValidType(B,2);
5117   PetscValidIntPointer(flg,3);
5118   PetscCheckSameComm(A,1,B,2);
5119   MatCheckPreallocated(B,2);
5120   if (!A->assembled) SETERRQ(PetscObjectComm((PetscObject)A),PETSC_ERR_ARG_WRONGSTATE,"Not for unassembled matrix");
5121   if (!B->assembled) SETERRQ(PetscObjectComm((PetscObject)A),PETSC_ERR_ARG_WRONGSTATE,"Not for unassembled matrix");
5122   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);
5123   if (!A->ops->equal) SETERRQ1(PetscObjectComm((PetscObject)A),PETSC_ERR_SUP,"Mat type %s",((PetscObject)A)->type_name);
5124   if (!B->ops->equal) SETERRQ1(PetscObjectComm((PetscObject)A),PETSC_ERR_SUP,"Mat type %s",((PetscObject)B)->type_name);
5125   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);
5126   MatCheckPreallocated(A,1);
5127 
5128   ierr = (*A->ops->equal)(A,B,flg);CHKERRQ(ierr);
5129   PetscFunctionReturn(0);
5130 }
5131 
5132 /*@
5133    MatDiagonalScale - Scales a matrix on the left and right by diagonal
5134    matrices that are stored as vectors.  Either of the two scaling
5135    matrices can be NULL.
5136 
5137    Collective on Mat
5138 
5139    Input Parameters:
5140 +  mat - the matrix to be scaled
5141 .  l - the left scaling vector (or NULL)
5142 -  r - the right scaling vector (or NULL)
5143 
5144    Notes:
5145    MatDiagonalScale() computes A = LAR, where
5146    L = a diagonal matrix (stored as a vector), R = a diagonal matrix (stored as a vector)
5147    The L scales the rows of the matrix, the R scales the columns of the matrix.
5148 
5149    Level: intermediate
5150 
5151    Concepts: matrices^diagonal scaling
5152    Concepts: diagonal scaling of matrices
5153 
5154 .seealso: MatScale(), MatShift(), MatDiagonalSet()
5155 @*/
5156 PetscErrorCode MatDiagonalScale(Mat mat,Vec l,Vec r)
5157 {
5158   PetscErrorCode ierr;
5159 
5160   PetscFunctionBegin;
5161   PetscValidHeaderSpecific(mat,MAT_CLASSID,1);
5162   PetscValidType(mat,1);
5163   if (!mat->ops->diagonalscale) SETERRQ1(PetscObjectComm((PetscObject)mat),PETSC_ERR_SUP,"Mat type %s",((PetscObject)mat)->type_name);
5164   if (l) {PetscValidHeaderSpecific(l,VEC_CLASSID,2);PetscCheckSameComm(mat,1,l,2);}
5165   if (r) {PetscValidHeaderSpecific(r,VEC_CLASSID,3);PetscCheckSameComm(mat,1,r,3);}
5166   if (!mat->assembled) SETERRQ(PetscObjectComm((PetscObject)mat),PETSC_ERR_ARG_WRONGSTATE,"Not for unassembled matrix");
5167   if (mat->factortype) SETERRQ(PetscObjectComm((PetscObject)mat),PETSC_ERR_ARG_WRONGSTATE,"Not for factored matrix");
5168   MatCheckPreallocated(mat,1);
5169 
5170   ierr = PetscLogEventBegin(MAT_Scale,mat,0,0,0);CHKERRQ(ierr);
5171   ierr = (*mat->ops->diagonalscale)(mat,l,r);CHKERRQ(ierr);
5172   ierr = PetscLogEventEnd(MAT_Scale,mat,0,0,0);CHKERRQ(ierr);
5173   ierr = PetscObjectStateIncrease((PetscObject)mat);CHKERRQ(ierr);
5174 #if defined(PETSC_HAVE_VIENNACL) || defined(PETSC_HAVE_CUDA)
5175   if (mat->valid_GPU_matrix != PETSC_OFFLOAD_UNALLOCATED) {
5176     mat->valid_GPU_matrix = PETSC_OFFLOAD_CPU;
5177   }
5178 #endif
5179   PetscFunctionReturn(0);
5180 }
5181 
5182 /*@
5183     MatScale - Scales all elements of a matrix by a given number.
5184 
5185     Logically Collective on Mat
5186 
5187     Input Parameters:
5188 +   mat - the matrix to be scaled
5189 -   a  - the scaling value
5190 
5191     Output Parameter:
5192 .   mat - the scaled matrix
5193 
5194     Level: intermediate
5195 
5196     Concepts: matrices^scaling all entries
5197 
5198 .seealso: MatDiagonalScale()
5199 @*/
5200 PetscErrorCode MatScale(Mat mat,PetscScalar a)
5201 {
5202   PetscErrorCode ierr;
5203 
5204   PetscFunctionBegin;
5205   PetscValidHeaderSpecific(mat,MAT_CLASSID,1);
5206   PetscValidType(mat,1);
5207   if (a != (PetscScalar)1.0 && !mat->ops->scale) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_SUP,"Mat type %s",((PetscObject)mat)->type_name);
5208   if (!mat->assembled) SETERRQ(PetscObjectComm((PetscObject)mat),PETSC_ERR_ARG_WRONGSTATE,"Not for unassembled matrix");
5209   if (mat->factortype) SETERRQ(PetscObjectComm((PetscObject)mat),PETSC_ERR_ARG_WRONGSTATE,"Not for factored matrix");
5210   PetscValidLogicalCollectiveScalar(mat,a,2);
5211   MatCheckPreallocated(mat,1);
5212 
5213   ierr = PetscLogEventBegin(MAT_Scale,mat,0,0,0);CHKERRQ(ierr);
5214   if (a != (PetscScalar)1.0) {
5215     ierr = (*mat->ops->scale)(mat,a);CHKERRQ(ierr);
5216     ierr = PetscObjectStateIncrease((PetscObject)mat);CHKERRQ(ierr);
5217 #if defined(PETSC_HAVE_VIENNACL) || defined(PETSC_HAVE_CUDA)
5218     if (mat->valid_GPU_matrix != PETSC_OFFLOAD_UNALLOCATED) {
5219       mat->valid_GPU_matrix = PETSC_OFFLOAD_CPU;
5220     }
5221 #endif
5222   }
5223   ierr = PetscLogEventEnd(MAT_Scale,mat,0,0,0);CHKERRQ(ierr);
5224   PetscFunctionReturn(0);
5225 }
5226 
5227 /*@
5228    MatNorm - Calculates various norms of a matrix.
5229 
5230    Collective on Mat
5231 
5232    Input Parameters:
5233 +  mat - the matrix
5234 -  type - the type of norm, NORM_1, NORM_FROBENIUS, NORM_INFINITY
5235 
5236    Output Parameters:
5237 .  nrm - the resulting norm
5238 
5239    Level: intermediate
5240 
5241    Concepts: matrices^norm
5242    Concepts: norm^of matrix
5243 @*/
5244 PetscErrorCode MatNorm(Mat mat,NormType type,PetscReal *nrm)
5245 {
5246   PetscErrorCode ierr;
5247 
5248   PetscFunctionBegin;
5249   PetscValidHeaderSpecific(mat,MAT_CLASSID,1);
5250   PetscValidType(mat,1);
5251   PetscValidScalarPointer(nrm,3);
5252 
5253   if (!mat->assembled) SETERRQ(PetscObjectComm((PetscObject)mat),PETSC_ERR_ARG_WRONGSTATE,"Not for unassembled matrix");
5254   if (mat->factortype) SETERRQ(PetscObjectComm((PetscObject)mat),PETSC_ERR_ARG_WRONGSTATE,"Not for factored matrix");
5255   if (!mat->ops->norm) SETERRQ1(PetscObjectComm((PetscObject)mat),PETSC_ERR_SUP,"Mat type %s",((PetscObject)mat)->type_name);
5256   MatCheckPreallocated(mat,1);
5257 
5258   ierr = (*mat->ops->norm)(mat,type,nrm);CHKERRQ(ierr);
5259   PetscFunctionReturn(0);
5260 }
5261 
5262 /*
5263      This variable is used to prevent counting of MatAssemblyBegin() that
5264    are called from within a MatAssemblyEnd().
5265 */
5266 static PetscInt MatAssemblyEnd_InUse = 0;
5267 /*@
5268    MatAssemblyBegin - Begins assembling the matrix.  This routine should
5269    be called after completing all calls to MatSetValues().
5270 
5271    Collective on Mat
5272 
5273    Input Parameters:
5274 +  mat - the matrix
5275 -  type - type of assembly, either MAT_FLUSH_ASSEMBLY or MAT_FINAL_ASSEMBLY
5276 
5277    Notes:
5278    MatSetValues() generally caches the values.  The matrix is ready to
5279    use only after MatAssemblyBegin() and MatAssemblyEnd() have been called.
5280    Use MAT_FLUSH_ASSEMBLY when switching between ADD_VALUES and INSERT_VALUES
5281    in MatSetValues(); use MAT_FINAL_ASSEMBLY for the final assembly before
5282    using the matrix.
5283 
5284    ALL processes that share a matrix MUST call MatAssemblyBegin() and MatAssemblyEnd() the SAME NUMBER of times, and each time with the
5285    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
5286    a global collective operation requring all processes that share the matrix.
5287 
5288    Space for preallocated nonzeros that is not filled by a call to MatSetValues() or a related routine are compressed
5289    out by assembly. If you intend to use that extra space on a subsequent assembly, be sure to insert explicit zeros
5290    before MAT_FINAL_ASSEMBLY so the space is not compressed out.
5291 
5292    Level: beginner
5293 
5294    Concepts: matrices^assembling
5295 
5296 .seealso: MatAssemblyEnd(), MatSetValues(), MatAssembled()
5297 @*/
5298 PetscErrorCode MatAssemblyBegin(Mat mat,MatAssemblyType type)
5299 {
5300   PetscErrorCode ierr;
5301 
5302   PetscFunctionBegin;
5303   PetscValidHeaderSpecific(mat,MAT_CLASSID,1);
5304   PetscValidType(mat,1);
5305   MatCheckPreallocated(mat,1);
5306   if (mat->factortype) SETERRQ(PetscObjectComm((PetscObject)mat),PETSC_ERR_ARG_WRONGSTATE,"Not for factored matrix.\nDid you forget to call MatSetUnfactored()?");
5307   if (mat->assembled) {
5308     mat->was_assembled = PETSC_TRUE;
5309     mat->assembled     = PETSC_FALSE;
5310   }
5311   if (!MatAssemblyEnd_InUse) {
5312     ierr = PetscLogEventBegin(MAT_AssemblyBegin,mat,0,0,0);CHKERRQ(ierr);
5313     if (mat->ops->assemblybegin) {ierr = (*mat->ops->assemblybegin)(mat,type);CHKERRQ(ierr);}
5314     ierr = PetscLogEventEnd(MAT_AssemblyBegin,mat,0,0,0);CHKERRQ(ierr);
5315   } else if (mat->ops->assemblybegin) {
5316     ierr = (*mat->ops->assemblybegin)(mat,type);CHKERRQ(ierr);
5317   }
5318   PetscFunctionReturn(0);
5319 }
5320 
5321 /*@
5322    MatAssembled - Indicates if a matrix has been assembled and is ready for
5323      use; for example, in matrix-vector product.
5324 
5325    Not Collective
5326 
5327    Input Parameter:
5328 .  mat - the matrix
5329 
5330    Output Parameter:
5331 .  assembled - PETSC_TRUE or PETSC_FALSE
5332 
5333    Level: advanced
5334 
5335    Concepts: matrices^assembled?
5336 
5337 .seealso: MatAssemblyEnd(), MatSetValues(), MatAssemblyBegin()
5338 @*/
5339 PetscErrorCode MatAssembled(Mat mat,PetscBool  *assembled)
5340 {
5341   PetscFunctionBegin;
5342   PetscValidHeaderSpecific(mat,MAT_CLASSID,1);
5343   PetscValidPointer(assembled,2);
5344   *assembled = mat->assembled;
5345   PetscFunctionReturn(0);
5346 }
5347 
5348 /*@
5349    MatAssemblyEnd - Completes assembling the matrix.  This routine should
5350    be called after MatAssemblyBegin().
5351 
5352    Collective on Mat
5353 
5354    Input Parameters:
5355 +  mat - the matrix
5356 -  type - type of assembly, either MAT_FLUSH_ASSEMBLY or MAT_FINAL_ASSEMBLY
5357 
5358    Options Database Keys:
5359 +  -mat_view ::ascii_info - Prints info on matrix at conclusion of MatEndAssembly()
5360 .  -mat_view ::ascii_info_detail - Prints more detailed info
5361 .  -mat_view - Prints matrix in ASCII format
5362 .  -mat_view ::ascii_matlab - Prints matrix in Matlab format
5363 .  -mat_view draw - PetscDraws nonzero structure of matrix, using MatView() and PetscDrawOpenX().
5364 .  -display <name> - Sets display name (default is host)
5365 .  -draw_pause <sec> - Sets number of seconds to pause after display
5366 .  -mat_view socket - Sends matrix to socket, can be accessed from Matlab (See Users-Manual: ch_matlab )
5367 .  -viewer_socket_machine <machine> - Machine to use for socket
5368 .  -viewer_socket_port <port> - Port number to use for socket
5369 -  -mat_view binary:filename[:append] - Save matrix to file in binary format
5370 
5371    Notes:
5372    MatSetValues() generally caches the values.  The matrix is ready to
5373    use only after MatAssemblyBegin() and MatAssemblyEnd() have been called.
5374    Use MAT_FLUSH_ASSEMBLY when switching between ADD_VALUES and INSERT_VALUES
5375    in MatSetValues(); use MAT_FINAL_ASSEMBLY for the final assembly before
5376    using the matrix.
5377 
5378    Space for preallocated nonzeros that is not filled by a call to MatSetValues() or a related routine are compressed
5379    out by assembly. If you intend to use that extra space on a subsequent assembly, be sure to insert explicit zeros
5380    before MAT_FINAL_ASSEMBLY so the space is not compressed out.
5381 
5382    Level: beginner
5383 
5384 .seealso: MatAssemblyBegin(), MatSetValues(), PetscDrawOpenX(), PetscDrawCreate(), MatView(), MatAssembled(), PetscViewerSocketOpen()
5385 @*/
5386 PetscErrorCode MatAssemblyEnd(Mat mat,MatAssemblyType type)
5387 {
5388   PetscErrorCode  ierr;
5389   static PetscInt inassm = 0;
5390   PetscBool       flg    = PETSC_FALSE;
5391 
5392   PetscFunctionBegin;
5393   PetscValidHeaderSpecific(mat,MAT_CLASSID,1);
5394   PetscValidType(mat,1);
5395 
5396   inassm++;
5397   MatAssemblyEnd_InUse++;
5398   if (MatAssemblyEnd_InUse == 1) { /* Do the logging only the first time through */
5399     ierr = PetscLogEventBegin(MAT_AssemblyEnd,mat,0,0,0);CHKERRQ(ierr);
5400     if (mat->ops->assemblyend) {
5401       ierr = (*mat->ops->assemblyend)(mat,type);CHKERRQ(ierr);
5402     }
5403     ierr = PetscLogEventEnd(MAT_AssemblyEnd,mat,0,0,0);CHKERRQ(ierr);
5404   } else if (mat->ops->assemblyend) {
5405     ierr = (*mat->ops->assemblyend)(mat,type);CHKERRQ(ierr);
5406   }
5407 
5408   /* Flush assembly is not a true assembly */
5409   if (type != MAT_FLUSH_ASSEMBLY) {
5410     mat->assembled = PETSC_TRUE; mat->num_ass++;
5411   }
5412   mat->insertmode = NOT_SET_VALUES;
5413   MatAssemblyEnd_InUse--;
5414   ierr = PetscObjectStateIncrease((PetscObject)mat);CHKERRQ(ierr);
5415   if (!mat->symmetric_eternal) {
5416     mat->symmetric_set              = PETSC_FALSE;
5417     mat->hermitian_set              = PETSC_FALSE;
5418     mat->structurally_symmetric_set = PETSC_FALSE;
5419   }
5420 #if defined(PETSC_HAVE_VIENNACL) || defined(PETSC_HAVE_CUDA)
5421   if (mat->valid_GPU_matrix != PETSC_OFFLOAD_UNALLOCATED) {
5422     mat->valid_GPU_matrix = PETSC_OFFLOAD_CPU;
5423   }
5424 #endif
5425   if (inassm == 1 && type != MAT_FLUSH_ASSEMBLY) {
5426     ierr = MatViewFromOptions(mat,NULL,"-mat_view");CHKERRQ(ierr);
5427 
5428     if (mat->checksymmetryonassembly) {
5429       ierr = MatIsSymmetric(mat,mat->checksymmetrytol,&flg);CHKERRQ(ierr);
5430       if (flg) {
5431         ierr = PetscPrintf(PetscObjectComm((PetscObject)mat),"Matrix is symmetric (tolerance %g)\n",(double)mat->checksymmetrytol);CHKERRQ(ierr);
5432       } else {
5433         ierr = PetscPrintf(PetscObjectComm((PetscObject)mat),"Matrix is not symmetric (tolerance %g)\n",(double)mat->checksymmetrytol);CHKERRQ(ierr);
5434       }
5435     }
5436     if (mat->nullsp && mat->checknullspaceonassembly) {
5437       ierr = MatNullSpaceTest(mat->nullsp,mat,NULL);CHKERRQ(ierr);
5438     }
5439   }
5440   inassm--;
5441   PetscFunctionReturn(0);
5442 }
5443 
5444 /*@
5445    MatSetOption - Sets a parameter option for a matrix. Some options
5446    may be specific to certain storage formats.  Some options
5447    determine how values will be inserted (or added). Sorted,
5448    row-oriented input will generally assemble the fastest. The default
5449    is row-oriented.
5450 
5451    Logically Collective on Mat for certain operations, such as MAT_SPD, not collective for MAT_ROW_ORIENTED, see MatOption
5452 
5453    Input Parameters:
5454 +  mat - the matrix
5455 .  option - the option, one of those listed below (and possibly others),
5456 -  flg - turn the option on (PETSC_TRUE) or off (PETSC_FALSE)
5457 
5458   Options Describing Matrix Structure:
5459 +    MAT_SPD - symmetric positive definite
5460 .    MAT_SYMMETRIC - symmetric in terms of both structure and value
5461 .    MAT_HERMITIAN - transpose is the complex conjugation
5462 .    MAT_STRUCTURALLY_SYMMETRIC - symmetric nonzero structure
5463 -    MAT_SYMMETRY_ETERNAL - if you would like the symmetry/Hermitian flag
5464                             you set to be kept with all future use of the matrix
5465                             including after MatAssemblyBegin/End() which could
5466                             potentially change the symmetry structure, i.e. you
5467                             KNOW the matrix will ALWAYS have the property you set.
5468 
5469 
5470    Options For Use with MatSetValues():
5471    Insert a logically dense subblock, which can be
5472 .    MAT_ROW_ORIENTED - row-oriented (default)
5473 
5474    Note these options reflect the data you pass in with MatSetValues(); it has
5475    nothing to do with how the data is stored internally in the matrix
5476    data structure.
5477 
5478    When (re)assembling a matrix, we can restrict the input for
5479    efficiency/debugging purposes.  These options include:
5480 +    MAT_NEW_NONZERO_LOCATIONS - additional insertions will be allowed if they generate a new nonzero (slow)
5481 .    MAT_NEW_DIAGONALS - new diagonals will be allowed (for block diagonal format only)
5482 .    MAT_IGNORE_OFF_PROC_ENTRIES - drops off-processor entries
5483 .    MAT_NEW_NONZERO_LOCATION_ERR - generates an error for new matrix entry
5484 .    MAT_USE_HASH_TABLE - uses a hash table to speed up matrix assembly
5485 .    MAT_NO_OFF_PROC_ENTRIES - you know each process will only set values for its own rows, will generate an error if
5486         any process sets values for another process. This avoids all reductions in the MatAssembly routines and thus improves
5487         performance for very large process counts.
5488 -    MAT_SUBSET_OFF_PROC_ENTRIES - you know that the first assembly after setting this flag will set a superset
5489         of the off-process entries required for all subsequent assemblies. This avoids a rendezvous step in the MatAssembly
5490         functions, instead sending only neighbor messages.
5491 
5492    Notes:
5493    Except for MAT_UNUSED_NONZERO_LOCATION_ERR and  MAT_ROW_ORIENTED all processes that share the matrix must pass the same value in flg!
5494 
5495    Some options are relevant only for particular matrix types and
5496    are thus ignored by others.  Other options are not supported by
5497    certain matrix types and will generate an error message if set.
5498 
5499    If using a Fortran 77 module to compute a matrix, one may need to
5500    use the column-oriented option (or convert to the row-oriented
5501    format).
5502 
5503    MAT_NEW_NONZERO_LOCATIONS set to PETSC_FALSE indicates that any add or insertion
5504    that would generate a new entry in the nonzero structure is instead
5505    ignored.  Thus, if memory has not alredy been allocated for this particular
5506    data, then the insertion is ignored. For dense matrices, in which
5507    the entire array is allocated, no entries are ever ignored.
5508    Set after the first MatAssemblyEnd(). If this option is set then the MatAssemblyBegin/End() processes has one less global reduction
5509 
5510    MAT_NEW_NONZERO_LOCATION_ERR set to PETSC_TRUE indicates that any add or insertion
5511    that would generate a new entry in the nonzero structure instead produces
5512    an error. (Currently supported for AIJ and BAIJ formats only.) If this option is set then the MatAssemblyBegin/End() processes has one less global reduction
5513 
5514    MAT_NEW_NONZERO_ALLOCATION_ERR set to PETSC_TRUE indicates that any add or insertion
5515    that would generate a new entry that has not been preallocated will
5516    instead produce an error. (Currently supported for AIJ and BAIJ formats
5517    only.) This is a useful flag when debugging matrix memory preallocation.
5518    If this option is set then the MatAssemblyBegin/End() processes has one less global reduction
5519 
5520    MAT_IGNORE_OFF_PROC_ENTRIES set to PETSC_TRUE indicates entries destined for
5521    other processors should be dropped, rather than stashed.
5522    This is useful if you know that the "owning" processor is also
5523    always generating the correct matrix entries, so that PETSc need
5524    not transfer duplicate entries generated on another processor.
5525 
5526    MAT_USE_HASH_TABLE indicates that a hash table be used to improve the
5527    searches during matrix assembly. When this flag is set, the hash table
5528    is created during the first Matrix Assembly. This hash table is
5529    used the next time through, during MatSetVaules()/MatSetVaulesBlocked()
5530    to improve the searching of indices. MAT_NEW_NONZERO_LOCATIONS flag
5531    should be used with MAT_USE_HASH_TABLE flag. This option is currently
5532    supported by MATMPIBAIJ format only.
5533 
5534    MAT_KEEP_NONZERO_PATTERN indicates when MatZeroRows() is called the zeroed entries
5535    are kept in the nonzero structure
5536 
5537    MAT_IGNORE_ZERO_ENTRIES - for AIJ/IS matrices this will stop zero values from creating
5538    a zero location in the matrix
5539 
5540    MAT_USE_INODES - indicates using inode version of the code - works with AIJ matrix types
5541 
5542    MAT_NO_OFF_PROC_ZERO_ROWS - you know each process will only zero its own rows. This avoids all reductions in the
5543         zero row routines and thus improves performance for very large process counts.
5544 
5545    MAT_IGNORE_LOWER_TRIANGULAR - For SBAIJ matrices will ignore any insertions you make in the lower triangular
5546         part of the matrix (since they should match the upper triangular part).
5547 
5548    Notes:
5549     Can only be called after MatSetSizes() and MatSetType() have been set.
5550 
5551    Level: intermediate
5552 
5553    Concepts: matrices^setting options
5554 
5555 .seealso:  MatOption, Mat
5556 
5557 @*/
5558 PetscErrorCode MatSetOption(Mat mat,MatOption op,PetscBool flg)
5559 {
5560   PetscErrorCode ierr;
5561 
5562   PetscFunctionBegin;
5563   PetscValidHeaderSpecific(mat,MAT_CLASSID,1);
5564   PetscValidType(mat,1);
5565   if (op > 0) {
5566     PetscValidLogicalCollectiveEnum(mat,op,2);
5567     PetscValidLogicalCollectiveBool(mat,flg,3);
5568   }
5569 
5570   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);
5571   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()");
5572 
5573   switch (op) {
5574   case MAT_NO_OFF_PROC_ENTRIES:
5575     mat->nooffprocentries = flg;
5576     PetscFunctionReturn(0);
5577     break;
5578   case MAT_SUBSET_OFF_PROC_ENTRIES:
5579     mat->subsetoffprocentries = flg;
5580     PetscFunctionReturn(0);
5581   case MAT_NO_OFF_PROC_ZERO_ROWS:
5582     mat->nooffproczerorows = flg;
5583     PetscFunctionReturn(0);
5584     break;
5585   case MAT_SPD:
5586     mat->spd_set = PETSC_TRUE;
5587     mat->spd     = flg;
5588     if (flg) {
5589       mat->symmetric                  = PETSC_TRUE;
5590       mat->structurally_symmetric     = PETSC_TRUE;
5591       mat->symmetric_set              = PETSC_TRUE;
5592       mat->structurally_symmetric_set = PETSC_TRUE;
5593     }
5594     break;
5595   case MAT_SYMMETRIC:
5596     mat->symmetric = flg;
5597     if (flg) mat->structurally_symmetric = PETSC_TRUE;
5598     mat->symmetric_set              = PETSC_TRUE;
5599     mat->structurally_symmetric_set = flg;
5600 #if !defined(PETSC_USE_COMPLEX)
5601     mat->hermitian     = flg;
5602     mat->hermitian_set = PETSC_TRUE;
5603 #endif
5604     break;
5605   case MAT_HERMITIAN:
5606     mat->hermitian = flg;
5607     if (flg) mat->structurally_symmetric = PETSC_TRUE;
5608     mat->hermitian_set              = PETSC_TRUE;
5609     mat->structurally_symmetric_set = flg;
5610 #if !defined(PETSC_USE_COMPLEX)
5611     mat->symmetric     = flg;
5612     mat->symmetric_set = PETSC_TRUE;
5613 #endif
5614     break;
5615   case MAT_STRUCTURALLY_SYMMETRIC:
5616     mat->structurally_symmetric     = flg;
5617     mat->structurally_symmetric_set = PETSC_TRUE;
5618     break;
5619   case MAT_SYMMETRY_ETERNAL:
5620     mat->symmetric_eternal = flg;
5621     break;
5622   case MAT_STRUCTURE_ONLY:
5623     mat->structure_only = flg;
5624     break;
5625   default:
5626     break;
5627   }
5628   if (mat->ops->setoption) {
5629     ierr = (*mat->ops->setoption)(mat,op,flg);CHKERRQ(ierr);
5630   }
5631   PetscFunctionReturn(0);
5632 }
5633 
5634 /*@
5635    MatGetOption - Gets a parameter option that has been set for a matrix.
5636 
5637    Logically Collective on Mat for certain operations, such as MAT_SPD, not collective for MAT_ROW_ORIENTED, see MatOption
5638 
5639    Input Parameters:
5640 +  mat - the matrix
5641 -  option - the option, this only responds to certain options, check the code for which ones
5642 
5643    Output Parameter:
5644 .  flg - turn the option on (PETSC_TRUE) or off (PETSC_FALSE)
5645 
5646     Notes:
5647     Can only be called after MatSetSizes() and MatSetType() have been set.
5648 
5649    Level: intermediate
5650 
5651    Concepts: matrices^setting options
5652 
5653 .seealso:  MatOption, MatSetOption()
5654 
5655 @*/
5656 PetscErrorCode MatGetOption(Mat mat,MatOption op,PetscBool *flg)
5657 {
5658   PetscFunctionBegin;
5659   PetscValidHeaderSpecific(mat,MAT_CLASSID,1);
5660   PetscValidType(mat,1);
5661 
5662   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);
5663   if (!((PetscObject)mat)->type_name) SETERRQ(PetscObjectComm((PetscObject)mat),PETSC_ERR_ARG_TYPENOTSET,"Cannot get options until type and size have been set, see MatSetType() and MatSetSizes()");
5664 
5665   switch (op) {
5666   case MAT_NO_OFF_PROC_ENTRIES:
5667     *flg = mat->nooffprocentries;
5668     break;
5669   case MAT_NO_OFF_PROC_ZERO_ROWS:
5670     *flg = mat->nooffproczerorows;
5671     break;
5672   case MAT_SYMMETRIC:
5673     *flg = mat->symmetric;
5674     break;
5675   case MAT_HERMITIAN:
5676     *flg = mat->hermitian;
5677     break;
5678   case MAT_STRUCTURALLY_SYMMETRIC:
5679     *flg = mat->structurally_symmetric;
5680     break;
5681   case MAT_SYMMETRY_ETERNAL:
5682     *flg = mat->symmetric_eternal;
5683     break;
5684   case MAT_SPD:
5685     *flg = mat->spd;
5686     break;
5687   default:
5688     break;
5689   }
5690   PetscFunctionReturn(0);
5691 }
5692 
5693 /*@
5694    MatZeroEntries - Zeros all entries of a matrix.  For sparse matrices
5695    this routine retains the old nonzero structure.
5696 
5697    Logically Collective on Mat
5698 
5699    Input Parameters:
5700 .  mat - the matrix
5701 
5702    Level: intermediate
5703 
5704    Notes:
5705     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.
5706    See the Performance chapter of the users manual for information on preallocating matrices.
5707 
5708    Concepts: matrices^zeroing
5709 
5710 .seealso: MatZeroRows()
5711 @*/
5712 PetscErrorCode MatZeroEntries(Mat mat)
5713 {
5714   PetscErrorCode ierr;
5715 
5716   PetscFunctionBegin;
5717   PetscValidHeaderSpecific(mat,MAT_CLASSID,1);
5718   PetscValidType(mat,1);
5719   if (mat->factortype) SETERRQ(PetscObjectComm((PetscObject)mat),PETSC_ERR_ARG_WRONGSTATE,"Not for factored matrix");
5720   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");
5721   if (!mat->ops->zeroentries) SETERRQ1(PetscObjectComm((PetscObject)mat),PETSC_ERR_SUP,"Mat type %s",((PetscObject)mat)->type_name);
5722   MatCheckPreallocated(mat,1);
5723 
5724   ierr = PetscLogEventBegin(MAT_ZeroEntries,mat,0,0,0);CHKERRQ(ierr);
5725   ierr = (*mat->ops->zeroentries)(mat);CHKERRQ(ierr);
5726   ierr = PetscLogEventEnd(MAT_ZeroEntries,mat,0,0,0);CHKERRQ(ierr);
5727   ierr = PetscObjectStateIncrease((PetscObject)mat);CHKERRQ(ierr);
5728 #if defined(PETSC_HAVE_VIENNACL) || defined(PETSC_HAVE_CUDA)
5729   if (mat->valid_GPU_matrix != PETSC_OFFLOAD_UNALLOCATED) {
5730     mat->valid_GPU_matrix = PETSC_OFFLOAD_CPU;
5731   }
5732 #endif
5733   PetscFunctionReturn(0);
5734 }
5735 
5736 /*@
5737    MatZeroRowsColumns - Zeros all entries (except possibly the main diagonal)
5738    of a set of rows and columns of a matrix.
5739 
5740    Collective on Mat
5741 
5742    Input Parameters:
5743 +  mat - the matrix
5744 .  numRows - the number of rows to remove
5745 .  rows - the global row indices
5746 .  diag - value put in all diagonals of eliminated rows (0.0 will even eliminate diagonal entry)
5747 .  x - optional vector of solutions for zeroed rows (other entries in vector are not used)
5748 -  b - optional vector of right hand side, that will be adjusted by provided solution
5749 
5750    Notes:
5751    This does not change the nonzero structure of the matrix, it merely zeros those entries in the matrix.
5752 
5753    The user can set a value in the diagonal entry (or for the AIJ and
5754    row formats can optionally remove the main diagonal entry from the
5755    nonzero structure as well, by passing 0.0 as the final argument).
5756 
5757    For the parallel case, all processes that share the matrix (i.e.,
5758    those in the communicator used for matrix creation) MUST call this
5759    routine, regardless of whether any rows being zeroed are owned by
5760    them.
5761 
5762    Each processor can indicate any rows in the entire matrix to be zeroed (i.e. each process does NOT have to
5763    list only rows local to itself).
5764 
5765    The option MAT_NO_OFF_PROC_ZERO_ROWS does not apply to this routine.
5766 
5767    Level: intermediate
5768 
5769    Concepts: matrices^zeroing rows
5770 
5771 .seealso: MatZeroRowsIS(), MatZeroRows(), MatZeroRowsLocalIS(), MatZeroRowsStencil(), MatZeroEntries(), MatZeroRowsLocal(), MatSetOption(),
5772           MatZeroRowsColumnsLocal(), MatZeroRowsColumnsLocalIS(), MatZeroRowsColumnsIS(), MatZeroRowsColumnsStencil()
5773 @*/
5774 PetscErrorCode MatZeroRowsColumns(Mat mat,PetscInt numRows,const PetscInt rows[],PetscScalar diag,Vec x,Vec b)
5775 {
5776   PetscErrorCode ierr;
5777 
5778   PetscFunctionBegin;
5779   PetscValidHeaderSpecific(mat,MAT_CLASSID,1);
5780   PetscValidType(mat,1);
5781   if (numRows) PetscValidIntPointer(rows,3);
5782   if (!mat->assembled) SETERRQ(PetscObjectComm((PetscObject)mat),PETSC_ERR_ARG_WRONGSTATE,"Not for unassembled matrix");
5783   if (mat->factortype) SETERRQ(PetscObjectComm((PetscObject)mat),PETSC_ERR_ARG_WRONGSTATE,"Not for factored matrix");
5784   if (!mat->ops->zerorowscolumns) SETERRQ1(PetscObjectComm((PetscObject)mat),PETSC_ERR_SUP,"Mat type %s",((PetscObject)mat)->type_name);
5785   MatCheckPreallocated(mat,1);
5786 
5787   ierr = (*mat->ops->zerorowscolumns)(mat,numRows,rows,diag,x,b);CHKERRQ(ierr);
5788   ierr = MatViewFromOptions(mat,NULL,"-mat_view");CHKERRQ(ierr);
5789   ierr = PetscObjectStateIncrease((PetscObject)mat);CHKERRQ(ierr);
5790 #if defined(PETSC_HAVE_VIENNACL) || defined(PETSC_HAVE_CUDA)
5791   if (mat->valid_GPU_matrix != PETSC_OFFLOAD_UNALLOCATED) {
5792     mat->valid_GPU_matrix = PETSC_OFFLOAD_CPU;
5793   }
5794 #endif
5795   PetscFunctionReturn(0);
5796 }
5797 
5798 /*@
5799    MatZeroRowsColumnsIS - Zeros all entries (except possibly the main diagonal)
5800    of a set of rows and columns of a matrix.
5801 
5802    Collective on Mat
5803 
5804    Input Parameters:
5805 +  mat - the matrix
5806 .  is - the rows to zero
5807 .  diag - value put in all diagonals of eliminated rows (0.0 will even eliminate diagonal entry)
5808 .  x - optional vector of solutions for zeroed rows (other entries in vector are not used)
5809 -  b - optional vector of right hand side, that will be adjusted by provided solution
5810 
5811    Notes:
5812    This does not change the nonzero structure of the matrix, it merely zeros those entries in the matrix.
5813 
5814    The user can set a value in the diagonal entry (or for the AIJ and
5815    row formats can optionally remove the main diagonal entry from the
5816    nonzero structure as well, by passing 0.0 as the final argument).
5817 
5818    For the parallel case, all processes that share the matrix (i.e.,
5819    those in the communicator used for matrix creation) MUST call this
5820    routine, regardless of whether any rows being zeroed are owned by
5821    them.
5822 
5823    Each processor can indicate any rows in the entire matrix to be zeroed (i.e. each process does NOT have to
5824    list only rows local to itself).
5825 
5826    The option MAT_NO_OFF_PROC_ZERO_ROWS does not apply to this routine.
5827 
5828    Level: intermediate
5829 
5830    Concepts: matrices^zeroing rows
5831 
5832 .seealso: MatZeroRowsIS(), MatZeroRowsColumns(), MatZeroRowsLocalIS(), MatZeroRowsStencil(), MatZeroEntries(), MatZeroRowsLocal(), MatSetOption(),
5833           MatZeroRowsColumnsLocal(), MatZeroRowsColumnsLocalIS(), MatZeroRows(), MatZeroRowsColumnsStencil()
5834 @*/
5835 PetscErrorCode MatZeroRowsColumnsIS(Mat mat,IS is,PetscScalar diag,Vec x,Vec b)
5836 {
5837   PetscErrorCode ierr;
5838   PetscInt       numRows;
5839   const PetscInt *rows;
5840 
5841   PetscFunctionBegin;
5842   PetscValidHeaderSpecific(mat,MAT_CLASSID,1);
5843   PetscValidHeaderSpecific(is,IS_CLASSID,2);
5844   PetscValidType(mat,1);
5845   PetscValidType(is,2);
5846   ierr = ISGetLocalSize(is,&numRows);CHKERRQ(ierr);
5847   ierr = ISGetIndices(is,&rows);CHKERRQ(ierr);
5848   ierr = MatZeroRowsColumns(mat,numRows,rows,diag,x,b);CHKERRQ(ierr);
5849   ierr = ISRestoreIndices(is,&rows);CHKERRQ(ierr);
5850   PetscFunctionReturn(0);
5851 }
5852 
5853 /*@
5854    MatZeroRows - Zeros all entries (except possibly the main diagonal)
5855    of a set of rows of a matrix.
5856 
5857    Collective on Mat
5858 
5859    Input Parameters:
5860 +  mat - the matrix
5861 .  numRows - the number of rows to remove
5862 .  rows - the global row indices
5863 .  diag - value put in all diagonals of eliminated rows (0.0 will even eliminate diagonal entry)
5864 .  x - optional vector of solutions for zeroed rows (other entries in vector are not used)
5865 -  b - optional vector of right hand side, that will be adjusted by provided solution
5866 
5867    Notes:
5868    For the AIJ and BAIJ matrix formats this removes the old nonzero structure,
5869    but does not release memory.  For the dense and block diagonal
5870    formats this does not alter the nonzero structure.
5871 
5872    If the option MatSetOption(mat,MAT_KEEP_NONZERO_PATTERN,PETSC_TRUE) the nonzero structure
5873    of the matrix is not changed (even for AIJ and BAIJ matrices) the values are
5874    merely zeroed.
5875 
5876    The user can set a value in the diagonal entry (or for the AIJ and
5877    row formats can optionally remove the main diagonal entry from the
5878    nonzero structure as well, by passing 0.0 as the final argument).
5879 
5880    For the parallel case, all processes that share the matrix (i.e.,
5881    those in the communicator used for matrix creation) MUST call this
5882    routine, regardless of whether any rows being zeroed are owned by
5883    them.
5884 
5885    Each processor can indicate any rows in the entire matrix to be zeroed (i.e. each process does NOT have to
5886    list only rows local to itself).
5887 
5888    You can call MatSetOption(mat,MAT_NO_OFF_PROC_ZERO_ROWS,PETSC_TRUE) if each process indicates only rows it
5889    owns that are to be zeroed. This saves a global synchronization in the implementation.
5890 
5891    Level: intermediate
5892 
5893    Concepts: matrices^zeroing rows
5894 
5895 .seealso: MatZeroRowsIS(), MatZeroRowsColumns(), MatZeroRowsLocalIS(), MatZeroRowsStencil(), MatZeroEntries(), MatZeroRowsLocal(), MatSetOption(),
5896           MatZeroRowsColumnsLocal(), MatZeroRowsColumnsLocalIS(), MatZeroRowsColumnsIS(), MatZeroRowsColumnsStencil()
5897 @*/
5898 PetscErrorCode MatZeroRows(Mat mat,PetscInt numRows,const PetscInt rows[],PetscScalar diag,Vec x,Vec b)
5899 {
5900   PetscErrorCode ierr;
5901 
5902   PetscFunctionBegin;
5903   PetscValidHeaderSpecific(mat,MAT_CLASSID,1);
5904   PetscValidType(mat,1);
5905   if (numRows) PetscValidIntPointer(rows,3);
5906   if (!mat->assembled) SETERRQ(PetscObjectComm((PetscObject)mat),PETSC_ERR_ARG_WRONGSTATE,"Not for unassembled matrix");
5907   if (mat->factortype) SETERRQ(PetscObjectComm((PetscObject)mat),PETSC_ERR_ARG_WRONGSTATE,"Not for factored matrix");
5908   if (!mat->ops->zerorows) SETERRQ1(PetscObjectComm((PetscObject)mat),PETSC_ERR_SUP,"Mat type %s",((PetscObject)mat)->type_name);
5909   MatCheckPreallocated(mat,1);
5910 
5911   ierr = (*mat->ops->zerorows)(mat,numRows,rows,diag,x,b);CHKERRQ(ierr);
5912   ierr = MatViewFromOptions(mat,NULL,"-mat_view");CHKERRQ(ierr);
5913   ierr = PetscObjectStateIncrease((PetscObject)mat);CHKERRQ(ierr);
5914 #if defined(PETSC_HAVE_VIENNACL) || defined(PETSC_HAVE_CUDA)
5915   if (mat->valid_GPU_matrix != PETSC_OFFLOAD_UNALLOCATED) {
5916     mat->valid_GPU_matrix = PETSC_OFFLOAD_CPU;
5917   }
5918 #endif
5919   PetscFunctionReturn(0);
5920 }
5921 
5922 /*@
5923    MatZeroRowsIS - Zeros all entries (except possibly the main diagonal)
5924    of a set of rows of a matrix.
5925 
5926    Collective on Mat
5927 
5928    Input Parameters:
5929 +  mat - the matrix
5930 .  is - index set of rows to remove
5931 .  diag - value put in all diagonals of eliminated rows
5932 .  x - optional vector of solutions for zeroed rows (other entries in vector are not used)
5933 -  b - optional vector of right hand side, that will be adjusted by provided solution
5934 
5935    Notes:
5936    For the AIJ and BAIJ matrix formats this removes the old nonzero structure,
5937    but does not release memory.  For the dense and block diagonal
5938    formats this does not alter the nonzero structure.
5939 
5940    If the option MatSetOption(mat,MAT_KEEP_NONZERO_PATTERN,PETSC_TRUE) the nonzero structure
5941    of the matrix is not changed (even for AIJ and BAIJ matrices) the values are
5942    merely zeroed.
5943 
5944    The user can set a value in the diagonal entry (or for the AIJ and
5945    row formats can optionally remove the main diagonal entry from the
5946    nonzero structure as well, by passing 0.0 as the final argument).
5947 
5948    For the parallel case, all processes that share the matrix (i.e.,
5949    those in the communicator used for matrix creation) MUST call this
5950    routine, regardless of whether any rows being zeroed are owned by
5951    them.
5952 
5953    Each processor can indicate any rows in the entire matrix to be zeroed (i.e. each process does NOT have to
5954    list only rows local to itself).
5955 
5956    You can call MatSetOption(mat,MAT_NO_OFF_PROC_ZERO_ROWS,PETSC_TRUE) if each process indicates only rows it
5957    owns that are to be zeroed. This saves a global synchronization in the implementation.
5958 
5959    Level: intermediate
5960 
5961    Concepts: matrices^zeroing rows
5962 
5963 .seealso: MatZeroRows(), MatZeroRowsColumns(), MatZeroRowsLocalIS(), MatZeroRowsStencil(), MatZeroEntries(), MatZeroRowsLocal(), MatSetOption(),
5964           MatZeroRowsColumnsLocal(), MatZeroRowsColumnsLocalIS(), MatZeroRowsColumnsIS(), MatZeroRowsColumnsStencil()
5965 @*/
5966 PetscErrorCode MatZeroRowsIS(Mat mat,IS is,PetscScalar diag,Vec x,Vec b)
5967 {
5968   PetscInt       numRows;
5969   const PetscInt *rows;
5970   PetscErrorCode ierr;
5971 
5972   PetscFunctionBegin;
5973   PetscValidHeaderSpecific(mat,MAT_CLASSID,1);
5974   PetscValidType(mat,1);
5975   PetscValidHeaderSpecific(is,IS_CLASSID,2);
5976   ierr = ISGetLocalSize(is,&numRows);CHKERRQ(ierr);
5977   ierr = ISGetIndices(is,&rows);CHKERRQ(ierr);
5978   ierr = MatZeroRows(mat,numRows,rows,diag,x,b);CHKERRQ(ierr);
5979   ierr = ISRestoreIndices(is,&rows);CHKERRQ(ierr);
5980   PetscFunctionReturn(0);
5981 }
5982 
5983 /*@
5984    MatZeroRowsStencil - Zeros all entries (except possibly the main diagonal)
5985    of a set of rows of a matrix. These rows must be local to the process.
5986 
5987    Collective on Mat
5988 
5989    Input Parameters:
5990 +  mat - the matrix
5991 .  numRows - the number of rows to remove
5992 .  rows - the grid coordinates (and component number when dof > 1) for matrix rows
5993 .  diag - value put in all diagonals of eliminated rows (0.0 will even eliminate diagonal entry)
5994 .  x - optional vector of solutions for zeroed rows (other entries in vector are not used)
5995 -  b - optional vector of right hand side, that will be adjusted by provided solution
5996 
5997    Notes:
5998    For the AIJ and BAIJ matrix formats this removes the old nonzero structure,
5999    but does not release memory.  For the dense and block diagonal
6000    formats this does not alter the nonzero structure.
6001 
6002    If the option MatSetOption(mat,MAT_KEEP_NONZERO_PATTERN,PETSC_TRUE) the nonzero structure
6003    of the matrix is not changed (even for AIJ and BAIJ matrices) the values are
6004    merely zeroed.
6005 
6006    The user can set a value in the diagonal entry (or for the AIJ and
6007    row formats can optionally remove the main diagonal entry from the
6008    nonzero structure as well, by passing 0.0 as the final argument).
6009 
6010    For the parallel case, all processes that share the matrix (i.e.,
6011    those in the communicator used for matrix creation) MUST call this
6012    routine, regardless of whether any rows being zeroed are owned by
6013    them.
6014 
6015    Each processor can indicate any rows in the entire matrix to be zeroed (i.e. each process does NOT have to
6016    list only rows local to itself).
6017 
6018    The grid coordinates are across the entire grid, not just the local portion
6019 
6020    In Fortran idxm and idxn should be declared as
6021 $     MatStencil idxm(4,m)
6022    and the values inserted using
6023 $    idxm(MatStencil_i,1) = i
6024 $    idxm(MatStencil_j,1) = j
6025 $    idxm(MatStencil_k,1) = k
6026 $    idxm(MatStencil_c,1) = c
6027    etc
6028 
6029    For periodic boundary conditions use negative indices for values to the left (below 0; that are to be
6030    obtained by wrapping values from right edge). For values to the right of the last entry using that index plus one
6031    etc to obtain values that obtained by wrapping the values from the left edge. This does not work for anything but the
6032    DM_BOUNDARY_PERIODIC boundary type.
6033 
6034    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
6035    a single value per point) you can skip filling those indices.
6036 
6037    Level: intermediate
6038 
6039    Concepts: matrices^zeroing rows
6040 
6041 .seealso: MatZeroRowsIS(), MatZeroRowsColumns(), MatZeroRowsLocalIS(), MatZeroRowsl(), MatZeroEntries(), MatZeroRowsLocal(), MatSetOption(),
6042           MatZeroRowsColumnsLocal(), MatZeroRowsColumnsLocalIS(), MatZeroRowsColumnsIS(), MatZeroRowsColumnsStencil()
6043 @*/
6044 PetscErrorCode MatZeroRowsStencil(Mat mat,PetscInt numRows,const MatStencil rows[],PetscScalar diag,Vec x,Vec b)
6045 {
6046   PetscInt       dim     = mat->stencil.dim;
6047   PetscInt       sdim    = dim - (1 - (PetscInt) mat->stencil.noc);
6048   PetscInt       *dims   = mat->stencil.dims+1;
6049   PetscInt       *starts = mat->stencil.starts;
6050   PetscInt       *dxm    = (PetscInt*) rows;
6051   PetscInt       *jdxm, i, j, tmp, numNewRows = 0;
6052   PetscErrorCode ierr;
6053 
6054   PetscFunctionBegin;
6055   PetscValidHeaderSpecific(mat,MAT_CLASSID,1);
6056   PetscValidType(mat,1);
6057   if (numRows) PetscValidIntPointer(rows,3);
6058 
6059   ierr = PetscMalloc1(numRows, &jdxm);CHKERRQ(ierr);
6060   for (i = 0; i < numRows; ++i) {
6061     /* Skip unused dimensions (they are ordered k, j, i, c) */
6062     for (j = 0; j < 3-sdim; ++j) dxm++;
6063     /* Local index in X dir */
6064     tmp = *dxm++ - starts[0];
6065     /* Loop over remaining dimensions */
6066     for (j = 0; j < dim-1; ++j) {
6067       /* If nonlocal, set index to be negative */
6068       if ((*dxm++ - starts[j+1]) < 0 || tmp < 0) tmp = PETSC_MIN_INT;
6069       /* Update local index */
6070       else tmp = tmp*dims[j] + *(dxm-1) - starts[j+1];
6071     }
6072     /* Skip component slot if necessary */
6073     if (mat->stencil.noc) dxm++;
6074     /* Local row number */
6075     if (tmp >= 0) {
6076       jdxm[numNewRows++] = tmp;
6077     }
6078   }
6079   ierr = MatZeroRowsLocal(mat,numNewRows,jdxm,diag,x,b);CHKERRQ(ierr);
6080   ierr = PetscFree(jdxm);CHKERRQ(ierr);
6081   PetscFunctionReturn(0);
6082 }
6083 
6084 /*@
6085    MatZeroRowsColumnsStencil - Zeros all row and column entries (except possibly the main diagonal)
6086    of a set of rows and columns of a matrix.
6087 
6088    Collective on Mat
6089 
6090    Input Parameters:
6091 +  mat - the matrix
6092 .  numRows - the number of rows/columns to remove
6093 .  rows - the grid coordinates (and component number when dof > 1) for matrix rows
6094 .  diag - value put in all diagonals of eliminated rows (0.0 will even eliminate diagonal entry)
6095 .  x - optional vector of solutions for zeroed rows (other entries in vector are not used)
6096 -  b - optional vector of right hand side, that will be adjusted by provided solution
6097 
6098    Notes:
6099    For the AIJ and BAIJ matrix formats this removes the old nonzero structure,
6100    but does not release memory.  For the dense and block diagonal
6101    formats this does not alter the nonzero structure.
6102 
6103    If the option MatSetOption(mat,MAT_KEEP_NONZERO_PATTERN,PETSC_TRUE) the nonzero structure
6104    of the matrix is not changed (even for AIJ and BAIJ matrices) the values are
6105    merely zeroed.
6106 
6107    The user can set a value in the diagonal entry (or for the AIJ and
6108    row formats can optionally remove the main diagonal entry from the
6109    nonzero structure as well, by passing 0.0 as the final argument).
6110 
6111    For the parallel case, all processes that share the matrix (i.e.,
6112    those in the communicator used for matrix creation) MUST call this
6113    routine, regardless of whether any rows being zeroed are owned by
6114    them.
6115 
6116    Each processor can indicate any rows in the entire matrix to be zeroed (i.e. each process does NOT have to
6117    list only rows local to itself, but the row/column numbers are given in local numbering).
6118 
6119    The grid coordinates are across the entire grid, not just the local portion
6120 
6121    In Fortran idxm and idxn should be declared as
6122 $     MatStencil idxm(4,m)
6123    and the values inserted using
6124 $    idxm(MatStencil_i,1) = i
6125 $    idxm(MatStencil_j,1) = j
6126 $    idxm(MatStencil_k,1) = k
6127 $    idxm(MatStencil_c,1) = c
6128    etc
6129 
6130    For periodic boundary conditions use negative indices for values to the left (below 0; that are to be
6131    obtained by wrapping values from right edge). For values to the right of the last entry using that index plus one
6132    etc to obtain values that obtained by wrapping the values from the left edge. This does not work for anything but the
6133    DM_BOUNDARY_PERIODIC boundary type.
6134 
6135    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
6136    a single value per point) you can skip filling those indices.
6137 
6138    Level: intermediate
6139 
6140    Concepts: matrices^zeroing rows
6141 
6142 .seealso: MatZeroRowsIS(), MatZeroRowsColumns(), MatZeroRowsLocalIS(), MatZeroRowsStencil(), MatZeroEntries(), MatZeroRowsLocal(), MatSetOption(),
6143           MatZeroRowsColumnsLocal(), MatZeroRowsColumnsLocalIS(), MatZeroRowsColumnsIS(), MatZeroRows()
6144 @*/
6145 PetscErrorCode MatZeroRowsColumnsStencil(Mat mat,PetscInt numRows,const MatStencil rows[],PetscScalar diag,Vec x,Vec b)
6146 {
6147   PetscInt       dim     = mat->stencil.dim;
6148   PetscInt       sdim    = dim - (1 - (PetscInt) mat->stencil.noc);
6149   PetscInt       *dims   = mat->stencil.dims+1;
6150   PetscInt       *starts = mat->stencil.starts;
6151   PetscInt       *dxm    = (PetscInt*) rows;
6152   PetscInt       *jdxm, i, j, tmp, numNewRows = 0;
6153   PetscErrorCode ierr;
6154 
6155   PetscFunctionBegin;
6156   PetscValidHeaderSpecific(mat,MAT_CLASSID,1);
6157   PetscValidType(mat,1);
6158   if (numRows) PetscValidIntPointer(rows,3);
6159 
6160   ierr = PetscMalloc1(numRows, &jdxm);CHKERRQ(ierr);
6161   for (i = 0; i < numRows; ++i) {
6162     /* Skip unused dimensions (they are ordered k, j, i, c) */
6163     for (j = 0; j < 3-sdim; ++j) dxm++;
6164     /* Local index in X dir */
6165     tmp = *dxm++ - starts[0];
6166     /* Loop over remaining dimensions */
6167     for (j = 0; j < dim-1; ++j) {
6168       /* If nonlocal, set index to be negative */
6169       if ((*dxm++ - starts[j+1]) < 0 || tmp < 0) tmp = PETSC_MIN_INT;
6170       /* Update local index */
6171       else tmp = tmp*dims[j] + *(dxm-1) - starts[j+1];
6172     }
6173     /* Skip component slot if necessary */
6174     if (mat->stencil.noc) dxm++;
6175     /* Local row number */
6176     if (tmp >= 0) {
6177       jdxm[numNewRows++] = tmp;
6178     }
6179   }
6180   ierr = MatZeroRowsColumnsLocal(mat,numNewRows,jdxm,diag,x,b);CHKERRQ(ierr);
6181   ierr = PetscFree(jdxm);CHKERRQ(ierr);
6182   PetscFunctionReturn(0);
6183 }
6184 
6185 /*@C
6186    MatZeroRowsLocal - Zeros all entries (except possibly the main diagonal)
6187    of a set of rows of a matrix; using local numbering of rows.
6188 
6189    Collective on Mat
6190 
6191    Input Parameters:
6192 +  mat - the matrix
6193 .  numRows - the number of rows to remove
6194 .  rows - the global row indices
6195 .  diag - value put in all diagonals of eliminated rows
6196 .  x - optional vector of solutions for zeroed rows (other entries in vector are not used)
6197 -  b - optional vector of right hand side, that will be adjusted by provided solution
6198 
6199    Notes:
6200    Before calling MatZeroRowsLocal(), the user must first set the
6201    local-to-global mapping by calling MatSetLocalToGlobalMapping().
6202 
6203    For the AIJ matrix formats this removes the old nonzero structure,
6204    but does not release memory.  For the dense and block diagonal
6205    formats this does not alter the nonzero structure.
6206 
6207    If the option MatSetOption(mat,MAT_KEEP_NONZERO_PATTERN,PETSC_TRUE) the nonzero structure
6208    of the matrix is not changed (even for AIJ and BAIJ matrices) the values are
6209    merely zeroed.
6210 
6211    The user can set a value in the diagonal entry (or for the AIJ and
6212    row formats can optionally remove the main diagonal entry from the
6213    nonzero structure as well, by passing 0.0 as the final argument).
6214 
6215    You can call MatSetOption(mat,MAT_NO_OFF_PROC_ZERO_ROWS,PETSC_TRUE) if each process indicates only rows it
6216    owns that are to be zeroed. This saves a global synchronization in the implementation.
6217 
6218    Level: intermediate
6219 
6220    Concepts: matrices^zeroing
6221 
6222 .seealso: MatZeroRowsIS(), MatZeroRowsColumns(), MatZeroRowsLocalIS(), MatZeroRowsStencil(), MatZeroEntries(), MatZeroRows(), MatSetOption(),
6223           MatZeroRowsColumnsLocal(), MatZeroRowsColumnsLocalIS(), MatZeroRowsColumnsIS(), MatZeroRowsColumnsStencil()
6224 @*/
6225 PetscErrorCode MatZeroRowsLocal(Mat mat,PetscInt numRows,const PetscInt rows[],PetscScalar diag,Vec x,Vec b)
6226 {
6227   PetscErrorCode ierr;
6228 
6229   PetscFunctionBegin;
6230   PetscValidHeaderSpecific(mat,MAT_CLASSID,1);
6231   PetscValidType(mat,1);
6232   if (numRows) PetscValidIntPointer(rows,3);
6233   if (!mat->assembled) SETERRQ(PetscObjectComm((PetscObject)mat),PETSC_ERR_ARG_WRONGSTATE,"Not for unassembled matrix");
6234   if (mat->factortype) SETERRQ(PetscObjectComm((PetscObject)mat),PETSC_ERR_ARG_WRONGSTATE,"Not for factored matrix");
6235   MatCheckPreallocated(mat,1);
6236 
6237   if (mat->ops->zerorowslocal) {
6238     ierr = (*mat->ops->zerorowslocal)(mat,numRows,rows,diag,x,b);CHKERRQ(ierr);
6239   } else {
6240     IS             is, newis;
6241     const PetscInt *newRows;
6242 
6243     if (!mat->rmap->mapping) SETERRQ(PetscObjectComm((PetscObject)mat),PETSC_ERR_ARG_WRONGSTATE,"Need to provide local to global mapping to matrix first");
6244     ierr = ISCreateGeneral(PETSC_COMM_SELF,numRows,rows,PETSC_COPY_VALUES,&is);CHKERRQ(ierr);
6245     ierr = ISLocalToGlobalMappingApplyIS(mat->rmap->mapping,is,&newis);CHKERRQ(ierr);
6246     ierr = ISGetIndices(newis,&newRows);CHKERRQ(ierr);
6247     ierr = (*mat->ops->zerorows)(mat,numRows,newRows,diag,x,b);CHKERRQ(ierr);
6248     ierr = ISRestoreIndices(newis,&newRows);CHKERRQ(ierr);
6249     ierr = ISDestroy(&newis);CHKERRQ(ierr);
6250     ierr = ISDestroy(&is);CHKERRQ(ierr);
6251   }
6252   ierr = PetscObjectStateIncrease((PetscObject)mat);CHKERRQ(ierr);
6253 #if defined(PETSC_HAVE_VIENNACL) || defined(PETSC_HAVE_CUDA)
6254   if (mat->valid_GPU_matrix != PETSC_OFFLOAD_UNALLOCATED) {
6255     mat->valid_GPU_matrix = PETSC_OFFLOAD_CPU;
6256   }
6257 #endif
6258   PetscFunctionReturn(0);
6259 }
6260 
6261 /*@
6262    MatZeroRowsLocalIS - Zeros all entries (except possibly the main diagonal)
6263    of a set of rows of a matrix; using local numbering of rows.
6264 
6265    Collective on Mat
6266 
6267    Input Parameters:
6268 +  mat - the matrix
6269 .  is - index set of rows to remove
6270 .  diag - value put in all diagonals of eliminated rows
6271 .  x - optional vector of solutions for zeroed rows (other entries in vector are not used)
6272 -  b - optional vector of right hand side, that will be adjusted by provided solution
6273 
6274    Notes:
6275    Before calling MatZeroRowsLocalIS(), the user must first set the
6276    local-to-global mapping by calling MatSetLocalToGlobalMapping().
6277 
6278    For the AIJ matrix formats this removes the old nonzero structure,
6279    but does not release memory.  For the dense and block diagonal
6280    formats this does not alter the nonzero structure.
6281 
6282    If the option MatSetOption(mat,MAT_KEEP_NONZERO_PATTERN,PETSC_TRUE) the nonzero structure
6283    of the matrix is not changed (even for AIJ and BAIJ matrices) the values are
6284    merely zeroed.
6285 
6286    The user can set a value in the diagonal entry (or for the AIJ and
6287    row formats can optionally remove the main diagonal entry from the
6288    nonzero structure as well, by passing 0.0 as the final argument).
6289 
6290    You can call MatSetOption(mat,MAT_NO_OFF_PROC_ZERO_ROWS,PETSC_TRUE) if each process indicates only rows it
6291    owns that are to be zeroed. This saves a global synchronization in the implementation.
6292 
6293    Level: intermediate
6294 
6295    Concepts: matrices^zeroing
6296 
6297 .seealso: MatZeroRowsIS(), MatZeroRowsColumns(), MatZeroRows(), MatZeroRowsStencil(), MatZeroEntries(), MatZeroRowsLocal(), MatSetOption(),
6298           MatZeroRowsColumnsLocal(), MatZeroRowsColumnsLocalIS(), MatZeroRowsColumnsIS(), MatZeroRowsColumnsStencil()
6299 @*/
6300 PetscErrorCode MatZeroRowsLocalIS(Mat mat,IS is,PetscScalar diag,Vec x,Vec b)
6301 {
6302   PetscErrorCode ierr;
6303   PetscInt       numRows;
6304   const PetscInt *rows;
6305 
6306   PetscFunctionBegin;
6307   PetscValidHeaderSpecific(mat,MAT_CLASSID,1);
6308   PetscValidType(mat,1);
6309   PetscValidHeaderSpecific(is,IS_CLASSID,2);
6310   if (!mat->assembled) SETERRQ(PetscObjectComm((PetscObject)mat),PETSC_ERR_ARG_WRONGSTATE,"Not for unassembled matrix");
6311   if (mat->factortype) SETERRQ(PetscObjectComm((PetscObject)mat),PETSC_ERR_ARG_WRONGSTATE,"Not for factored matrix");
6312   MatCheckPreallocated(mat,1);
6313 
6314   ierr = ISGetLocalSize(is,&numRows);CHKERRQ(ierr);
6315   ierr = ISGetIndices(is,&rows);CHKERRQ(ierr);
6316   ierr = MatZeroRowsLocal(mat,numRows,rows,diag,x,b);CHKERRQ(ierr);
6317   ierr = ISRestoreIndices(is,&rows);CHKERRQ(ierr);
6318   PetscFunctionReturn(0);
6319 }
6320 
6321 /*@
6322    MatZeroRowsColumnsLocal - Zeros all entries (except possibly the main diagonal)
6323    of a set of rows and columns of a matrix; using local numbering of rows.
6324 
6325    Collective on Mat
6326 
6327    Input Parameters:
6328 +  mat - the matrix
6329 .  numRows - the number of rows to remove
6330 .  rows - the global row indices
6331 .  diag - value put in all diagonals of eliminated rows
6332 .  x - optional vector of solutions for zeroed rows (other entries in vector are not used)
6333 -  b - optional vector of right hand side, that will be adjusted by provided solution
6334 
6335    Notes:
6336    Before calling MatZeroRowsColumnsLocal(), the user must first set the
6337    local-to-global mapping by calling MatSetLocalToGlobalMapping().
6338 
6339    The user can set a value in the diagonal entry (or for the AIJ and
6340    row formats can optionally remove the main diagonal entry from the
6341    nonzero structure as well, by passing 0.0 as the final argument).
6342 
6343    Level: intermediate
6344 
6345    Concepts: matrices^zeroing
6346 
6347 .seealso: MatZeroRowsIS(), MatZeroRowsColumns(), MatZeroRowsLocalIS(), MatZeroRowsStencil(), MatZeroEntries(), MatZeroRowsLocal(), MatSetOption(),
6348           MatZeroRows(), MatZeroRowsColumnsLocalIS(), MatZeroRowsColumnsIS(), MatZeroRowsColumnsStencil()
6349 @*/
6350 PetscErrorCode MatZeroRowsColumnsLocal(Mat mat,PetscInt numRows,const PetscInt rows[],PetscScalar diag,Vec x,Vec b)
6351 {
6352   PetscErrorCode ierr;
6353   IS             is, newis;
6354   const PetscInt *newRows;
6355 
6356   PetscFunctionBegin;
6357   PetscValidHeaderSpecific(mat,MAT_CLASSID,1);
6358   PetscValidType(mat,1);
6359   if (numRows) PetscValidIntPointer(rows,3);
6360   if (!mat->assembled) SETERRQ(PetscObjectComm((PetscObject)mat),PETSC_ERR_ARG_WRONGSTATE,"Not for unassembled matrix");
6361   if (mat->factortype) SETERRQ(PetscObjectComm((PetscObject)mat),PETSC_ERR_ARG_WRONGSTATE,"Not for factored matrix");
6362   MatCheckPreallocated(mat,1);
6363 
6364   if (!mat->cmap->mapping) SETERRQ(PetscObjectComm((PetscObject)mat),PETSC_ERR_ARG_WRONGSTATE,"Need to provide local to global mapping to matrix first");
6365   ierr = ISCreateGeneral(PETSC_COMM_SELF,numRows,rows,PETSC_COPY_VALUES,&is);CHKERRQ(ierr);
6366   ierr = ISLocalToGlobalMappingApplyIS(mat->cmap->mapping,is,&newis);CHKERRQ(ierr);
6367   ierr = ISGetIndices(newis,&newRows);CHKERRQ(ierr);
6368   ierr = (*mat->ops->zerorowscolumns)(mat,numRows,newRows,diag,x,b);CHKERRQ(ierr);
6369   ierr = ISRestoreIndices(newis,&newRows);CHKERRQ(ierr);
6370   ierr = ISDestroy(&newis);CHKERRQ(ierr);
6371   ierr = ISDestroy(&is);CHKERRQ(ierr);
6372   ierr = PetscObjectStateIncrease((PetscObject)mat);CHKERRQ(ierr);
6373 #if defined(PETSC_HAVE_VIENNACL) || defined(PETSC_HAVE_CUDA)
6374   if (mat->valid_GPU_matrix != PETSC_OFFLOAD_UNALLOCATED) {
6375     mat->valid_GPU_matrix = PETSC_OFFLOAD_CPU;
6376   }
6377 #endif
6378   PetscFunctionReturn(0);
6379 }
6380 
6381 /*@
6382    MatZeroRowsColumnsLocalIS - Zeros all entries (except possibly the main diagonal)
6383    of a set of rows and columns of a matrix; using local numbering of rows.
6384 
6385    Collective on Mat
6386 
6387    Input Parameters:
6388 +  mat - the matrix
6389 .  is - index set of rows to remove
6390 .  diag - value put in all diagonals of eliminated rows
6391 .  x - optional vector of solutions for zeroed rows (other entries in vector are not used)
6392 -  b - optional vector of right hand side, that will be adjusted by provided solution
6393 
6394    Notes:
6395    Before calling MatZeroRowsColumnsLocalIS(), the user must first set the
6396    local-to-global mapping by calling MatSetLocalToGlobalMapping().
6397 
6398    The user can set a value in the diagonal entry (or for the AIJ and
6399    row formats can optionally remove the main diagonal entry from the
6400    nonzero structure as well, by passing 0.0 as the final argument).
6401 
6402    Level: intermediate
6403 
6404    Concepts: matrices^zeroing
6405 
6406 .seealso: MatZeroRowsIS(), MatZeroRowsColumns(), MatZeroRowsLocalIS(), MatZeroRowsStencil(), MatZeroEntries(), MatZeroRowsLocal(), MatSetOption(),
6407           MatZeroRowsColumnsLocal(), MatZeroRows(), MatZeroRowsColumnsIS(), MatZeroRowsColumnsStencil()
6408 @*/
6409 PetscErrorCode MatZeroRowsColumnsLocalIS(Mat mat,IS is,PetscScalar diag,Vec x,Vec b)
6410 {
6411   PetscErrorCode ierr;
6412   PetscInt       numRows;
6413   const PetscInt *rows;
6414 
6415   PetscFunctionBegin;
6416   PetscValidHeaderSpecific(mat,MAT_CLASSID,1);
6417   PetscValidType(mat,1);
6418   PetscValidHeaderSpecific(is,IS_CLASSID,2);
6419   if (!mat->assembled) SETERRQ(PetscObjectComm((PetscObject)mat),PETSC_ERR_ARG_WRONGSTATE,"Not for unassembled matrix");
6420   if (mat->factortype) SETERRQ(PetscObjectComm((PetscObject)mat),PETSC_ERR_ARG_WRONGSTATE,"Not for factored matrix");
6421   MatCheckPreallocated(mat,1);
6422 
6423   ierr = ISGetLocalSize(is,&numRows);CHKERRQ(ierr);
6424   ierr = ISGetIndices(is,&rows);CHKERRQ(ierr);
6425   ierr = MatZeroRowsColumnsLocal(mat,numRows,rows,diag,x,b);CHKERRQ(ierr);
6426   ierr = ISRestoreIndices(is,&rows);CHKERRQ(ierr);
6427   PetscFunctionReturn(0);
6428 }
6429 
6430 /*@C
6431    MatGetSize - Returns the numbers of rows and columns in a matrix.
6432 
6433    Not Collective
6434 
6435    Input Parameter:
6436 .  mat - the matrix
6437 
6438    Output Parameters:
6439 +  m - the number of global rows
6440 -  n - the number of global columns
6441 
6442    Note: both output parameters can be NULL on input.
6443 
6444    Level: beginner
6445 
6446    Concepts: matrices^size
6447 
6448 .seealso: MatGetLocalSize()
6449 @*/
6450 PetscErrorCode MatGetSize(Mat mat,PetscInt *m,PetscInt *n)
6451 {
6452   PetscFunctionBegin;
6453   PetscValidHeaderSpecific(mat,MAT_CLASSID,1);
6454   if (m) *m = mat->rmap->N;
6455   if (n) *n = mat->cmap->N;
6456   PetscFunctionReturn(0);
6457 }
6458 
6459 /*@C
6460    MatGetLocalSize - Returns the number of rows and columns in a matrix
6461    stored locally.  This information may be implementation dependent, so
6462    use with care.
6463 
6464    Not Collective
6465 
6466    Input Parameters:
6467 .  mat - the matrix
6468 
6469    Output Parameters:
6470 +  m - the number of local rows
6471 -  n - the number of local columns
6472 
6473    Note: both output parameters can be NULL on input.
6474 
6475    Level: beginner
6476 
6477    Concepts: matrices^local size
6478 
6479 .seealso: MatGetSize()
6480 @*/
6481 PetscErrorCode MatGetLocalSize(Mat mat,PetscInt *m,PetscInt *n)
6482 {
6483   PetscFunctionBegin;
6484   PetscValidHeaderSpecific(mat,MAT_CLASSID,1);
6485   if (m) PetscValidIntPointer(m,2);
6486   if (n) PetscValidIntPointer(n,3);
6487   if (m) *m = mat->rmap->n;
6488   if (n) *n = mat->cmap->n;
6489   PetscFunctionReturn(0);
6490 }
6491 
6492 /*@C
6493    MatGetOwnershipRangeColumn - Returns the range of matrix columns associated with rows of a vector one multiplies by that owned by
6494    this processor. (The columns of the "diagonal block")
6495 
6496    Not Collective, unless matrix has not been allocated, then collective on Mat
6497 
6498    Input Parameters:
6499 .  mat - the matrix
6500 
6501    Output Parameters:
6502 +  m - the global index of the first local column
6503 -  n - one more than the global index of the last local column
6504 
6505    Notes:
6506     both output parameters can be NULL on input.
6507 
6508    Level: developer
6509 
6510    Concepts: matrices^column ownership
6511 
6512 .seealso:  MatGetOwnershipRange(), MatGetOwnershipRanges(), MatGetOwnershipRangesColumn()
6513 
6514 @*/
6515 PetscErrorCode MatGetOwnershipRangeColumn(Mat mat,PetscInt *m,PetscInt *n)
6516 {
6517   PetscFunctionBegin;
6518   PetscValidHeaderSpecific(mat,MAT_CLASSID,1);
6519   PetscValidType(mat,1);
6520   if (m) PetscValidIntPointer(m,2);
6521   if (n) PetscValidIntPointer(n,3);
6522   MatCheckPreallocated(mat,1);
6523   if (m) *m = mat->cmap->rstart;
6524   if (n) *n = mat->cmap->rend;
6525   PetscFunctionReturn(0);
6526 }
6527 
6528 /*@C
6529    MatGetOwnershipRange - Returns the range of matrix rows owned by
6530    this processor, assuming that the matrix is laid out with the first
6531    n1 rows on the first processor, the next n2 rows on the second, etc.
6532    For certain parallel layouts this range may not be well defined.
6533 
6534    Not Collective
6535 
6536    Input Parameters:
6537 .  mat - the matrix
6538 
6539    Output Parameters:
6540 +  m - the global index of the first local row
6541 -  n - one more than the global index of the last local row
6542 
6543    Note: Both output parameters can be NULL on input.
6544 $  This function requires that the matrix be preallocated. If you have not preallocated, consider using
6545 $    PetscSplitOwnership(MPI_Comm comm, PetscInt *n, PetscInt *N)
6546 $  and then MPI_Scan() to calculate prefix sums of the local sizes.
6547 
6548    Level: beginner
6549 
6550    Concepts: matrices^row ownership
6551 
6552 .seealso:   MatGetOwnershipRanges(), MatGetOwnershipRangeColumn(), MatGetOwnershipRangesColumn(), PetscSplitOwnership(), PetscSplitOwnershipBlock()
6553 
6554 @*/
6555 PetscErrorCode MatGetOwnershipRange(Mat mat,PetscInt *m,PetscInt *n)
6556 {
6557   PetscFunctionBegin;
6558   PetscValidHeaderSpecific(mat,MAT_CLASSID,1);
6559   PetscValidType(mat,1);
6560   if (m) PetscValidIntPointer(m,2);
6561   if (n) PetscValidIntPointer(n,3);
6562   MatCheckPreallocated(mat,1);
6563   if (m) *m = mat->rmap->rstart;
6564   if (n) *n = mat->rmap->rend;
6565   PetscFunctionReturn(0);
6566 }
6567 
6568 /*@C
6569    MatGetOwnershipRanges - Returns the range of matrix rows owned by
6570    each process
6571 
6572    Not Collective, unless matrix has not been allocated, then collective on Mat
6573 
6574    Input Parameters:
6575 .  mat - the matrix
6576 
6577    Output Parameters:
6578 .  ranges - start of each processors portion plus one more than the total length at the end
6579 
6580    Level: beginner
6581 
6582    Concepts: matrices^row ownership
6583 
6584 .seealso:   MatGetOwnershipRange(), MatGetOwnershipRangeColumn(), MatGetOwnershipRangesColumn()
6585 
6586 @*/
6587 PetscErrorCode MatGetOwnershipRanges(Mat mat,const PetscInt **ranges)
6588 {
6589   PetscErrorCode ierr;
6590 
6591   PetscFunctionBegin;
6592   PetscValidHeaderSpecific(mat,MAT_CLASSID,1);
6593   PetscValidType(mat,1);
6594   MatCheckPreallocated(mat,1);
6595   ierr = PetscLayoutGetRanges(mat->rmap,ranges);CHKERRQ(ierr);
6596   PetscFunctionReturn(0);
6597 }
6598 
6599 /*@C
6600    MatGetOwnershipRangesColumn - Returns the range of matrix columns associated with rows of a vector one multiplies by that owned by
6601    this processor. (The columns of the "diagonal blocks" for each process)
6602 
6603    Not Collective, unless matrix has not been allocated, then collective on Mat
6604 
6605    Input Parameters:
6606 .  mat - the matrix
6607 
6608    Output Parameters:
6609 .  ranges - start of each processors portion plus one more then the total length at the end
6610 
6611    Level: beginner
6612 
6613    Concepts: matrices^column ownership
6614 
6615 .seealso:   MatGetOwnershipRange(), MatGetOwnershipRangeColumn(), MatGetOwnershipRanges()
6616 
6617 @*/
6618 PetscErrorCode MatGetOwnershipRangesColumn(Mat mat,const PetscInt **ranges)
6619 {
6620   PetscErrorCode ierr;
6621 
6622   PetscFunctionBegin;
6623   PetscValidHeaderSpecific(mat,MAT_CLASSID,1);
6624   PetscValidType(mat,1);
6625   MatCheckPreallocated(mat,1);
6626   ierr = PetscLayoutGetRanges(mat->cmap,ranges);CHKERRQ(ierr);
6627   PetscFunctionReturn(0);
6628 }
6629 
6630 /*@C
6631    MatGetOwnershipIS - Get row and column ownership as index sets
6632 
6633    Not Collective
6634 
6635    Input Arguments:
6636 .  A - matrix of type Elemental
6637 
6638    Output Arguments:
6639 +  rows - rows in which this process owns elements
6640 .  cols - columns in which this process owns elements
6641 
6642    Level: intermediate
6643 
6644 .seealso: MatGetOwnershipRange(), MatGetOwnershipRangeColumn(), MatSetValues(), MATELEMENTAL
6645 @*/
6646 PetscErrorCode MatGetOwnershipIS(Mat A,IS *rows,IS *cols)
6647 {
6648   PetscErrorCode ierr,(*f)(Mat,IS*,IS*);
6649 
6650   PetscFunctionBegin;
6651   MatCheckPreallocated(A,1);
6652   ierr = PetscObjectQueryFunction((PetscObject)A,"MatGetOwnershipIS_C",&f);CHKERRQ(ierr);
6653   if (f) {
6654     ierr = (*f)(A,rows,cols);CHKERRQ(ierr);
6655   } else {   /* Create a standard row-based partition, each process is responsible for ALL columns in their row block */
6656     if (rows) {ierr = ISCreateStride(PETSC_COMM_SELF,A->rmap->n,A->rmap->rstart,1,rows);CHKERRQ(ierr);}
6657     if (cols) {ierr = ISCreateStride(PETSC_COMM_SELF,A->cmap->N,0,1,cols);CHKERRQ(ierr);}
6658   }
6659   PetscFunctionReturn(0);
6660 }
6661 
6662 /*@C
6663    MatILUFactorSymbolic - Performs symbolic ILU factorization of a matrix.
6664    Uses levels of fill only, not drop tolerance. Use MatLUFactorNumeric()
6665    to complete the factorization.
6666 
6667    Collective on Mat
6668 
6669    Input Parameters:
6670 +  mat - the matrix
6671 .  row - row permutation
6672 .  column - column permutation
6673 -  info - structure containing
6674 $      levels - number of levels of fill.
6675 $      expected fill - as ratio of original fill.
6676 $      1 or 0 - indicating force fill on diagonal (improves robustness for matrices
6677                 missing diagonal entries)
6678 
6679    Output Parameters:
6680 .  fact - new matrix that has been symbolically factored
6681 
6682    Notes:
6683     See Users-Manual: ch_mat for additional information about choosing the fill factor for better efficiency.
6684 
6685    Most users should employ the simplified KSP interface for linear solvers
6686    instead of working directly with matrix algebra routines such as this.
6687    See, e.g., KSPCreate().
6688 
6689    Level: developer
6690 
6691   Concepts: matrices^symbolic LU factorization
6692   Concepts: matrices^factorization
6693   Concepts: LU^symbolic factorization
6694 
6695 .seealso: MatLUFactorSymbolic(), MatLUFactorNumeric(), MatCholeskyFactor()
6696           MatGetOrdering(), MatFactorInfo
6697 
6698     Note: this uses the definition of level of fill as in Y. Saad, 2003
6699 
6700     Developer Note: fortran interface is not autogenerated as the f90
6701     interface defintion cannot be generated correctly [due to MatFactorInfo]
6702 
6703    References:
6704      Y. Saad, Iterative methods for sparse linear systems Philadelphia: Society for Industrial and Applied Mathematics, 2003
6705 @*/
6706 PetscErrorCode MatILUFactorSymbolic(Mat fact,Mat mat,IS row,IS col,const MatFactorInfo *info)
6707 {
6708   PetscErrorCode ierr;
6709 
6710   PetscFunctionBegin;
6711   PetscValidHeaderSpecific(mat,MAT_CLASSID,1);
6712   PetscValidType(mat,1);
6713   PetscValidHeaderSpecific(row,IS_CLASSID,2);
6714   PetscValidHeaderSpecific(col,IS_CLASSID,3);
6715   PetscValidPointer(info,4);
6716   PetscValidPointer(fact,5);
6717   if (info->levels < 0) SETERRQ1(PetscObjectComm((PetscObject)mat),PETSC_ERR_ARG_OUTOFRANGE,"Levels of fill negative %D",(PetscInt)info->levels);
6718   if (info->fill < 1.0) SETERRQ1(PetscObjectComm((PetscObject)mat),PETSC_ERR_ARG_OUTOFRANGE,"Expected fill less than 1.0 %g",(double)info->fill);
6719   if (!(fact)->ops->ilufactorsymbolic) {
6720     MatSolverType spackage;
6721     ierr = MatFactorGetSolverType(fact,&spackage);CHKERRQ(ierr);
6722     SETERRQ2(PetscObjectComm((PetscObject)mat),PETSC_ERR_SUP,"Matrix type %s symbolic ILU using solver package %s",((PetscObject)mat)->type_name,spackage);
6723   }
6724   if (!mat->assembled) SETERRQ(PetscObjectComm((PetscObject)mat),PETSC_ERR_ARG_WRONGSTATE,"Not for unassembled matrix");
6725   if (mat->factortype) SETERRQ(PetscObjectComm((PetscObject)mat),PETSC_ERR_ARG_WRONGSTATE,"Not for factored matrix");
6726   MatCheckPreallocated(mat,2);
6727 
6728   ierr = PetscLogEventBegin(MAT_ILUFactorSymbolic,mat,row,col,0);CHKERRQ(ierr);
6729   ierr = (fact->ops->ilufactorsymbolic)(fact,mat,row,col,info);CHKERRQ(ierr);
6730   ierr = PetscLogEventEnd(MAT_ILUFactorSymbolic,mat,row,col,0);CHKERRQ(ierr);
6731   PetscFunctionReturn(0);
6732 }
6733 
6734 /*@C
6735    MatICCFactorSymbolic - Performs symbolic incomplete
6736    Cholesky factorization for a symmetric matrix.  Use
6737    MatCholeskyFactorNumeric() to complete the factorization.
6738 
6739    Collective on Mat
6740 
6741    Input Parameters:
6742 +  mat - the matrix
6743 .  perm - row and column permutation
6744 -  info - structure containing
6745 $      levels - number of levels of fill.
6746 $      expected fill - as ratio of original fill.
6747 
6748    Output Parameter:
6749 .  fact - the factored matrix
6750 
6751    Notes:
6752    Most users should employ the KSP interface for linear solvers
6753    instead of working directly with matrix algebra routines such as this.
6754    See, e.g., KSPCreate().
6755 
6756    Level: developer
6757 
6758   Concepts: matrices^symbolic incomplete Cholesky factorization
6759   Concepts: matrices^factorization
6760   Concepts: Cholsky^symbolic factorization
6761 
6762 .seealso: MatCholeskyFactorNumeric(), MatCholeskyFactor(), MatFactorInfo
6763 
6764     Note: this uses the definition of level of fill as in Y. Saad, 2003
6765 
6766     Developer Note: fortran interface is not autogenerated as the f90
6767     interface defintion cannot be generated correctly [due to MatFactorInfo]
6768 
6769    References:
6770      Y. Saad, Iterative methods for sparse linear systems Philadelphia: Society for Industrial and Applied Mathematics, 2003
6771 @*/
6772 PetscErrorCode MatICCFactorSymbolic(Mat fact,Mat mat,IS perm,const MatFactorInfo *info)
6773 {
6774   PetscErrorCode ierr;
6775 
6776   PetscFunctionBegin;
6777   PetscValidHeaderSpecific(mat,MAT_CLASSID,1);
6778   PetscValidType(mat,1);
6779   PetscValidHeaderSpecific(perm,IS_CLASSID,2);
6780   PetscValidPointer(info,3);
6781   PetscValidPointer(fact,4);
6782   if (mat->factortype) SETERRQ(PetscObjectComm((PetscObject)mat),PETSC_ERR_ARG_WRONGSTATE,"Not for factored matrix");
6783   if (info->levels < 0) SETERRQ1(PetscObjectComm((PetscObject)mat),PETSC_ERR_ARG_OUTOFRANGE,"Levels negative %D",(PetscInt) info->levels);
6784   if (info->fill < 1.0) SETERRQ1(PetscObjectComm((PetscObject)mat),PETSC_ERR_ARG_OUTOFRANGE,"Expected fill less than 1.0 %g",(double)info->fill);
6785   if (!(fact)->ops->iccfactorsymbolic) {
6786     MatSolverType spackage;
6787     ierr = MatFactorGetSolverType(fact,&spackage);CHKERRQ(ierr);
6788     SETERRQ2(PetscObjectComm((PetscObject)mat),PETSC_ERR_SUP,"Matrix type %s symbolic ICC using solver package %s",((PetscObject)mat)->type_name,spackage);
6789   }
6790   if (!mat->assembled) SETERRQ(PetscObjectComm((PetscObject)mat),PETSC_ERR_ARG_WRONGSTATE,"Not for unassembled matrix");
6791   MatCheckPreallocated(mat,2);
6792 
6793   ierr = PetscLogEventBegin(MAT_ICCFactorSymbolic,mat,perm,0,0);CHKERRQ(ierr);
6794   ierr = (fact->ops->iccfactorsymbolic)(fact,mat,perm,info);CHKERRQ(ierr);
6795   ierr = PetscLogEventEnd(MAT_ICCFactorSymbolic,mat,perm,0,0);CHKERRQ(ierr);
6796   PetscFunctionReturn(0);
6797 }
6798 
6799 /*@C
6800    MatCreateSubMatrices - Extracts several submatrices from a matrix. If submat
6801    points to an array of valid matrices, they may be reused to store the new
6802    submatrices.
6803 
6804    Collective on Mat
6805 
6806    Input Parameters:
6807 +  mat - the matrix
6808 .  n   - the number of submatrixes to be extracted (on this processor, may be zero)
6809 .  irow, icol - index sets of rows and columns to extract
6810 -  scall - either MAT_INITIAL_MATRIX or MAT_REUSE_MATRIX
6811 
6812    Output Parameter:
6813 .  submat - the array of submatrices
6814 
6815    Notes:
6816    MatCreateSubMatrices() can extract ONLY sequential submatrices
6817    (from both sequential and parallel matrices). Use MatCreateSubMatrix()
6818    to extract a parallel submatrix.
6819 
6820    Some matrix types place restrictions on the row and column
6821    indices, such as that they be sorted or that they be equal to each other.
6822 
6823    The index sets may not have duplicate entries.
6824 
6825    When extracting submatrices from a parallel matrix, each processor can
6826    form a different submatrix by setting the rows and columns of its
6827    individual index sets according to the local submatrix desired.
6828 
6829    When finished using the submatrices, the user should destroy
6830    them with MatDestroySubMatrices().
6831 
6832    MAT_REUSE_MATRIX can only be used when the nonzero structure of the
6833    original matrix has not changed from that last call to MatCreateSubMatrices().
6834 
6835    This routine creates the matrices in submat; you should NOT create them before
6836    calling it. It also allocates the array of matrix pointers submat.
6837 
6838    For BAIJ matrices the index sets must respect the block structure, that is if they
6839    request one row/column in a block, they must request all rows/columns that are in
6840    that block. For example, if the block size is 2 you cannot request just row 0 and
6841    column 0.
6842 
6843    Fortran Note:
6844    The Fortran interface is slightly different from that given below; it
6845    requires one to pass in  as submat a Mat (integer) array of size at least n+1.
6846 
6847    Level: advanced
6848 
6849    Concepts: matrices^accessing submatrices
6850    Concepts: submatrices
6851 
6852 .seealso: MatDestroySubMatrices(), MatCreateSubMatrix(), MatGetRow(), MatGetDiagonal(), MatReuse
6853 @*/
6854 PetscErrorCode MatCreateSubMatrices(Mat mat,PetscInt n,const IS irow[],const IS icol[],MatReuse scall,Mat *submat[])
6855 {
6856   PetscErrorCode ierr;
6857   PetscInt       i;
6858   PetscBool      eq;
6859 
6860   PetscFunctionBegin;
6861   PetscValidHeaderSpecific(mat,MAT_CLASSID,1);
6862   PetscValidType(mat,1);
6863   if (n) {
6864     PetscValidPointer(irow,3);
6865     PetscValidHeaderSpecific(*irow,IS_CLASSID,3);
6866     PetscValidPointer(icol,4);
6867     PetscValidHeaderSpecific(*icol,IS_CLASSID,4);
6868   }
6869   PetscValidPointer(submat,6);
6870   if (n && scall == MAT_REUSE_MATRIX) {
6871     PetscValidPointer(*submat,6);
6872     PetscValidHeaderSpecific(**submat,MAT_CLASSID,6);
6873   }
6874   if (!mat->ops->createsubmatrices) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_SUP,"Mat type %s",((PetscObject)mat)->type_name);
6875   if (!mat->assembled) SETERRQ(PetscObjectComm((PetscObject)mat),PETSC_ERR_ARG_WRONGSTATE,"Not for unassembled matrix");
6876   if (mat->factortype) SETERRQ(PetscObjectComm((PetscObject)mat),PETSC_ERR_ARG_WRONGSTATE,"Not for factored matrix");
6877   MatCheckPreallocated(mat,1);
6878 
6879   ierr = PetscLogEventBegin(MAT_CreateSubMats,mat,0,0,0);CHKERRQ(ierr);
6880   ierr = (*mat->ops->createsubmatrices)(mat,n,irow,icol,scall,submat);CHKERRQ(ierr);
6881   ierr = PetscLogEventEnd(MAT_CreateSubMats,mat,0,0,0);CHKERRQ(ierr);
6882   for (i=0; i<n; i++) {
6883     (*submat)[i]->factortype = MAT_FACTOR_NONE;  /* in case in place factorization was previously done on submatrix */
6884     if (mat->symmetric || mat->structurally_symmetric || mat->hermitian) {
6885       ierr = ISEqual(irow[i],icol[i],&eq);CHKERRQ(ierr);
6886       if (eq) {
6887         if (mat->symmetric) {
6888           ierr = MatSetOption((*submat)[i],MAT_SYMMETRIC,PETSC_TRUE);CHKERRQ(ierr);
6889         } else if (mat->hermitian) {
6890           ierr = MatSetOption((*submat)[i],MAT_HERMITIAN,PETSC_TRUE);CHKERRQ(ierr);
6891         } else if (mat->structurally_symmetric) {
6892           ierr = MatSetOption((*submat)[i],MAT_STRUCTURALLY_SYMMETRIC,PETSC_TRUE);CHKERRQ(ierr);
6893         }
6894       }
6895     }
6896   }
6897   PetscFunctionReturn(0);
6898 }
6899 
6900 /*@C
6901    MatCreateSubMatricesMPI - Extracts MPI submatrices across a sub communicator of mat (by pairs of IS that may live on subcomms).
6902 
6903    Collective on Mat
6904 
6905    Input Parameters:
6906 +  mat - the matrix
6907 .  n   - the number of submatrixes to be extracted
6908 .  irow, icol - index sets of rows and columns to extract
6909 -  scall - either MAT_INITIAL_MATRIX or MAT_REUSE_MATRIX
6910 
6911    Output Parameter:
6912 .  submat - the array of submatrices
6913 
6914    Level: advanced
6915 
6916    Concepts: matrices^accessing submatrices
6917    Concepts: submatrices
6918 
6919 .seealso: MatCreateSubMatrices(), MatCreateSubMatrix(), MatGetRow(), MatGetDiagonal(), MatReuse
6920 @*/
6921 PetscErrorCode MatCreateSubMatricesMPI(Mat mat,PetscInt n,const IS irow[],const IS icol[],MatReuse scall,Mat *submat[])
6922 {
6923   PetscErrorCode ierr;
6924   PetscInt       i;
6925   PetscBool      eq;
6926 
6927   PetscFunctionBegin;
6928   PetscValidHeaderSpecific(mat,MAT_CLASSID,1);
6929   PetscValidType(mat,1);
6930   if (n) {
6931     PetscValidPointer(irow,3);
6932     PetscValidHeaderSpecific(*irow,IS_CLASSID,3);
6933     PetscValidPointer(icol,4);
6934     PetscValidHeaderSpecific(*icol,IS_CLASSID,4);
6935   }
6936   PetscValidPointer(submat,6);
6937   if (n && scall == MAT_REUSE_MATRIX) {
6938     PetscValidPointer(*submat,6);
6939     PetscValidHeaderSpecific(**submat,MAT_CLASSID,6);
6940   }
6941   if (!mat->ops->createsubmatricesmpi) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_SUP,"Mat type %s",((PetscObject)mat)->type_name);
6942   if (!mat->assembled) SETERRQ(PetscObjectComm((PetscObject)mat),PETSC_ERR_ARG_WRONGSTATE,"Not for unassembled matrix");
6943   if (mat->factortype) SETERRQ(PetscObjectComm((PetscObject)mat),PETSC_ERR_ARG_WRONGSTATE,"Not for factored matrix");
6944   MatCheckPreallocated(mat,1);
6945 
6946   ierr = PetscLogEventBegin(MAT_CreateSubMats,mat,0,0,0);CHKERRQ(ierr);
6947   ierr = (*mat->ops->createsubmatricesmpi)(mat,n,irow,icol,scall,submat);CHKERRQ(ierr);
6948   ierr = PetscLogEventEnd(MAT_CreateSubMats,mat,0,0,0);CHKERRQ(ierr);
6949   for (i=0; i<n; i++) {
6950     if (mat->symmetric || mat->structurally_symmetric || mat->hermitian) {
6951       ierr = ISEqual(irow[i],icol[i],&eq);CHKERRQ(ierr);
6952       if (eq) {
6953         if (mat->symmetric) {
6954           ierr = MatSetOption((*submat)[i],MAT_SYMMETRIC,PETSC_TRUE);CHKERRQ(ierr);
6955         } else if (mat->hermitian) {
6956           ierr = MatSetOption((*submat)[i],MAT_HERMITIAN,PETSC_TRUE);CHKERRQ(ierr);
6957         } else if (mat->structurally_symmetric) {
6958           ierr = MatSetOption((*submat)[i],MAT_STRUCTURALLY_SYMMETRIC,PETSC_TRUE);CHKERRQ(ierr);
6959         }
6960       }
6961     }
6962   }
6963   PetscFunctionReturn(0);
6964 }
6965 
6966 /*@C
6967    MatDestroyMatrices - Destroys an array of matrices.
6968 
6969    Collective on Mat
6970 
6971    Input Parameters:
6972 +  n - the number of local matrices
6973 -  mat - the matrices (note that this is a pointer to the array of matrices)
6974 
6975    Level: advanced
6976 
6977     Notes:
6978     Frees not only the matrices, but also the array that contains the matrices
6979            In Fortran will not free the array.
6980 
6981 .seealso: MatCreateSubMatrices() MatDestroySubMatrices()
6982 @*/
6983 PetscErrorCode MatDestroyMatrices(PetscInt n,Mat *mat[])
6984 {
6985   PetscErrorCode ierr;
6986   PetscInt       i;
6987 
6988   PetscFunctionBegin;
6989   if (!*mat) PetscFunctionReturn(0);
6990   if (n < 0) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Trying to destroy negative number of matrices %D",n);
6991   PetscValidPointer(mat,2);
6992 
6993   for (i=0; i<n; i++) {
6994     ierr = MatDestroy(&(*mat)[i]);CHKERRQ(ierr);
6995   }
6996 
6997   /* memory is allocated even if n = 0 */
6998   ierr = PetscFree(*mat);CHKERRQ(ierr);
6999   PetscFunctionReturn(0);
7000 }
7001 
7002 /*@C
7003    MatDestroySubMatrices - Destroys a set of matrices obtained with MatCreateSubMatrices().
7004 
7005    Collective on Mat
7006 
7007    Input Parameters:
7008 +  n - the number of local matrices
7009 -  mat - the matrices (note that this is a pointer to the array of matrices, just to match the calling
7010                        sequence of MatCreateSubMatrices())
7011 
7012    Level: advanced
7013 
7014     Notes:
7015     Frees not only the matrices, but also the array that contains the matrices
7016            In Fortran will not free the array.
7017 
7018 .seealso: MatCreateSubMatrices()
7019 @*/
7020 PetscErrorCode MatDestroySubMatrices(PetscInt n,Mat *mat[])
7021 {
7022   PetscErrorCode ierr;
7023   Mat            mat0;
7024 
7025   PetscFunctionBegin;
7026   if (!*mat) PetscFunctionReturn(0);
7027   /* mat[] is an array of length n+1, see MatCreateSubMatrices_xxx() */
7028   if (n < 0) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Trying to destroy negative number of matrices %D",n);
7029   PetscValidPointer(mat,2);
7030 
7031   mat0 = (*mat)[0];
7032   if (mat0 && mat0->ops->destroysubmatrices) {
7033     ierr = (mat0->ops->destroysubmatrices)(n,mat);CHKERRQ(ierr);
7034   } else {
7035     ierr = MatDestroyMatrices(n,mat);CHKERRQ(ierr);
7036   }
7037   PetscFunctionReturn(0);
7038 }
7039 
7040 /*@C
7041    MatGetSeqNonzeroStructure - Extracts the sequential nonzero structure from a matrix.
7042 
7043    Collective on Mat
7044 
7045    Input Parameters:
7046 .  mat - the matrix
7047 
7048    Output Parameter:
7049 .  matstruct - the sequential matrix with the nonzero structure of mat
7050 
7051   Level: intermediate
7052 
7053 .seealso: MatDestroySeqNonzeroStructure(), MatCreateSubMatrices(), MatDestroyMatrices()
7054 @*/
7055 PetscErrorCode MatGetSeqNonzeroStructure(Mat mat,Mat *matstruct)
7056 {
7057   PetscErrorCode ierr;
7058 
7059   PetscFunctionBegin;
7060   PetscValidHeaderSpecific(mat,MAT_CLASSID,1);
7061   PetscValidPointer(matstruct,2);
7062 
7063   PetscValidType(mat,1);
7064   if (mat->factortype) SETERRQ(PetscObjectComm((PetscObject)mat),PETSC_ERR_ARG_WRONGSTATE,"Not for factored matrix");
7065   MatCheckPreallocated(mat,1);
7066 
7067   if (!mat->ops->getseqnonzerostructure) SETERRQ1(PetscObjectComm((PetscObject)mat),PETSC_ERR_SUP,"Not for matrix type %s\n",((PetscObject)mat)->type_name);
7068   ierr = PetscLogEventBegin(MAT_GetSeqNonzeroStructure,mat,0,0,0);CHKERRQ(ierr);
7069   ierr = (*mat->ops->getseqnonzerostructure)(mat,matstruct);CHKERRQ(ierr);
7070   ierr = PetscLogEventEnd(MAT_GetSeqNonzeroStructure,mat,0,0,0);CHKERRQ(ierr);
7071   PetscFunctionReturn(0);
7072 }
7073 
7074 /*@C
7075    MatDestroySeqNonzeroStructure - Destroys matrix obtained with MatGetSeqNonzeroStructure().
7076 
7077    Collective on Mat
7078 
7079    Input Parameters:
7080 .  mat - the matrix (note that this is a pointer to the array of matrices, just to match the calling
7081                        sequence of MatGetSequentialNonzeroStructure())
7082 
7083    Level: advanced
7084 
7085     Notes:
7086     Frees not only the matrices, but also the array that contains the matrices
7087 
7088 .seealso: MatGetSeqNonzeroStructure()
7089 @*/
7090 PetscErrorCode MatDestroySeqNonzeroStructure(Mat *mat)
7091 {
7092   PetscErrorCode ierr;
7093 
7094   PetscFunctionBegin;
7095   PetscValidPointer(mat,1);
7096   ierr = MatDestroy(mat);CHKERRQ(ierr);
7097   PetscFunctionReturn(0);
7098 }
7099 
7100 /*@
7101    MatIncreaseOverlap - Given a set of submatrices indicated by index sets,
7102    replaces the index sets by larger ones that represent submatrices with
7103    additional overlap.
7104 
7105    Collective on Mat
7106 
7107    Input Parameters:
7108 +  mat - the matrix
7109 .  n   - the number of index sets
7110 .  is  - the array of index sets (these index sets will changed during the call)
7111 -  ov  - the additional overlap requested
7112 
7113    Options Database:
7114 .  -mat_increase_overlap_scalable - use a scalable algorithm to compute the overlap (supported by MPIAIJ matrix)
7115 
7116    Level: developer
7117 
7118    Concepts: overlap
7119    Concepts: ASM^computing overlap
7120 
7121 .seealso: MatCreateSubMatrices()
7122 @*/
7123 PetscErrorCode MatIncreaseOverlap(Mat mat,PetscInt n,IS is[],PetscInt ov)
7124 {
7125   PetscErrorCode ierr;
7126 
7127   PetscFunctionBegin;
7128   PetscValidHeaderSpecific(mat,MAT_CLASSID,1);
7129   PetscValidType(mat,1);
7130   if (n < 0) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Must have one or more domains, you have %D",n);
7131   if (n) {
7132     PetscValidPointer(is,3);
7133     PetscValidHeaderSpecific(*is,IS_CLASSID,3);
7134   }
7135   if (!mat->assembled) SETERRQ(PetscObjectComm((PetscObject)mat),PETSC_ERR_ARG_WRONGSTATE,"Not for unassembled matrix");
7136   if (mat->factortype) SETERRQ(PetscObjectComm((PetscObject)mat),PETSC_ERR_ARG_WRONGSTATE,"Not for factored matrix");
7137   MatCheckPreallocated(mat,1);
7138 
7139   if (!ov) PetscFunctionReturn(0);
7140   if (!mat->ops->increaseoverlap) SETERRQ1(PetscObjectComm((PetscObject)mat),PETSC_ERR_SUP,"Mat type %s",((PetscObject)mat)->type_name);
7141   ierr = PetscLogEventBegin(MAT_IncreaseOverlap,mat,0,0,0);CHKERRQ(ierr);
7142   ierr = (*mat->ops->increaseoverlap)(mat,n,is,ov);CHKERRQ(ierr);
7143   ierr = PetscLogEventEnd(MAT_IncreaseOverlap,mat,0,0,0);CHKERRQ(ierr);
7144   PetscFunctionReturn(0);
7145 }
7146 
7147 
7148 PetscErrorCode MatIncreaseOverlapSplit_Single(Mat,IS*,PetscInt);
7149 
7150 /*@
7151    MatIncreaseOverlapSplit - Given a set of submatrices indicated by index sets across
7152    a sub communicator, replaces the index sets by larger ones that represent submatrices with
7153    additional overlap.
7154 
7155    Collective on Mat
7156 
7157    Input Parameters:
7158 +  mat - the matrix
7159 .  n   - the number of index sets
7160 .  is  - the array of index sets (these index sets will changed during the call)
7161 -  ov  - the additional overlap requested
7162 
7163    Options Database:
7164 .  -mat_increase_overlap_scalable - use a scalable algorithm to compute the overlap (supported by MPIAIJ matrix)
7165 
7166    Level: developer
7167 
7168    Concepts: overlap
7169    Concepts: ASM^computing overlap
7170 
7171 .seealso: MatCreateSubMatrices()
7172 @*/
7173 PetscErrorCode MatIncreaseOverlapSplit(Mat mat,PetscInt n,IS is[],PetscInt ov)
7174 {
7175   PetscInt       i;
7176   PetscErrorCode ierr;
7177 
7178   PetscFunctionBegin;
7179   PetscValidHeaderSpecific(mat,MAT_CLASSID,1);
7180   PetscValidType(mat,1);
7181   if (n < 0) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Must have one or more domains, you have %D",n);
7182   if (n) {
7183     PetscValidPointer(is,3);
7184     PetscValidHeaderSpecific(*is,IS_CLASSID,3);
7185   }
7186   if (!mat->assembled) SETERRQ(PetscObjectComm((PetscObject)mat),PETSC_ERR_ARG_WRONGSTATE,"Not for unassembled matrix");
7187   if (mat->factortype) SETERRQ(PetscObjectComm((PetscObject)mat),PETSC_ERR_ARG_WRONGSTATE,"Not for factored matrix");
7188   MatCheckPreallocated(mat,1);
7189   if (!ov) PetscFunctionReturn(0);
7190   ierr = PetscLogEventBegin(MAT_IncreaseOverlap,mat,0,0,0);CHKERRQ(ierr);
7191   for(i=0; i<n; i++){
7192 	ierr =  MatIncreaseOverlapSplit_Single(mat,&is[i],ov);CHKERRQ(ierr);
7193   }
7194   ierr = PetscLogEventEnd(MAT_IncreaseOverlap,mat,0,0,0);CHKERRQ(ierr);
7195   PetscFunctionReturn(0);
7196 }
7197 
7198 
7199 
7200 
7201 /*@
7202    MatGetBlockSize - Returns the matrix block size.
7203 
7204    Not Collective
7205 
7206    Input Parameter:
7207 .  mat - the matrix
7208 
7209    Output Parameter:
7210 .  bs - block size
7211 
7212    Notes:
7213     Block row formats are MATSEQBAIJ, MATMPIBAIJ, MATSEQSBAIJ, MATMPISBAIJ. These formats ALWAYS have square block storage in the matrix.
7214 
7215    If the block size has not been set yet this routine returns 1.
7216 
7217    Level: intermediate
7218 
7219    Concepts: matrices^block size
7220 
7221 .seealso: MatCreateSeqBAIJ(), MatCreateBAIJ(), MatGetBlockSizes()
7222 @*/
7223 PetscErrorCode MatGetBlockSize(Mat mat,PetscInt *bs)
7224 {
7225   PetscFunctionBegin;
7226   PetscValidHeaderSpecific(mat,MAT_CLASSID,1);
7227   PetscValidIntPointer(bs,2);
7228   *bs = PetscAbs(mat->rmap->bs);
7229   PetscFunctionReturn(0);
7230 }
7231 
7232 /*@
7233    MatGetBlockSizes - Returns the matrix block row and column sizes.
7234 
7235    Not Collective
7236 
7237    Input Parameter:
7238 .  mat - the matrix
7239 
7240    Output Parameter:
7241 .  rbs - row block size
7242 .  cbs - column block size
7243 
7244    Notes:
7245     Block row formats are MATSEQBAIJ, MATMPIBAIJ, MATSEQSBAIJ, MATMPISBAIJ. These formats ALWAYS have square block storage in the matrix.
7246     If you pass a different block size for the columns than the rows, the row block size determines the square block storage.
7247 
7248    If a block size has not been set yet this routine returns 1.
7249 
7250    Level: intermediate
7251 
7252    Concepts: matrices^block size
7253 
7254 .seealso: MatCreateSeqBAIJ(), MatCreateBAIJ(), MatGetBlockSize(), MatSetBlockSize(), MatSetBlockSizes()
7255 @*/
7256 PetscErrorCode MatGetBlockSizes(Mat mat,PetscInt *rbs, PetscInt *cbs)
7257 {
7258   PetscFunctionBegin;
7259   PetscValidHeaderSpecific(mat,MAT_CLASSID,1);
7260   if (rbs) PetscValidIntPointer(rbs,2);
7261   if (cbs) PetscValidIntPointer(cbs,3);
7262   if (rbs) *rbs = PetscAbs(mat->rmap->bs);
7263   if (cbs) *cbs = PetscAbs(mat->cmap->bs);
7264   PetscFunctionReturn(0);
7265 }
7266 
7267 /*@
7268    MatSetBlockSize - Sets the matrix block size.
7269 
7270    Logically Collective on Mat
7271 
7272    Input Parameters:
7273 +  mat - the matrix
7274 -  bs - block size
7275 
7276    Notes:
7277     Block row formats are MATSEQBAIJ, MATMPIBAIJ, MATSEQSBAIJ, MATMPISBAIJ. These formats ALWAYS have square block storage in the matrix.
7278     This must be called before MatSetUp() or MatXXXSetPreallocation() (or will default to 1) and the block size cannot be changed later.
7279 
7280     For MATMPIAIJ and MATSEQAIJ matrix formats, this function can be called at a later stage, provided that the specified block size
7281     is compatible with the matrix local sizes.
7282 
7283    Level: intermediate
7284 
7285    Concepts: matrices^block size
7286 
7287 .seealso: MatCreateSeqBAIJ(), MatCreateBAIJ(), MatGetBlockSize(), MatSetBlockSizes(), MatGetBlockSizes()
7288 @*/
7289 PetscErrorCode MatSetBlockSize(Mat mat,PetscInt bs)
7290 {
7291   PetscErrorCode ierr;
7292 
7293   PetscFunctionBegin;
7294   PetscValidHeaderSpecific(mat,MAT_CLASSID,1);
7295   PetscValidLogicalCollectiveInt(mat,bs,2);
7296   ierr = MatSetBlockSizes(mat,bs,bs);CHKERRQ(ierr);
7297   PetscFunctionReturn(0);
7298 }
7299 
7300 /*@
7301    MatSetVariableBlockSizes - Sets a diagonal blocks of the matrix that need not be of the same size
7302 
7303    Logically Collective on Mat
7304 
7305    Input Parameters:
7306 +  mat - the matrix
7307 .  nblocks - the number of blocks on this process
7308 -  bsizes - the block sizes
7309 
7310    Notes:
7311     Currently used by PCVPBJACOBI for SeqAIJ matrices
7312 
7313    Level: intermediate
7314 
7315    Concepts: matrices^block size
7316 
7317 .seealso: MatCreateSeqBAIJ(), MatCreateBAIJ(), MatGetBlockSize(), MatSetBlockSizes(), MatGetBlockSizes(), MatGetVariableBlockSizes()
7318 @*/
7319 PetscErrorCode MatSetVariableBlockSizes(Mat mat,PetscInt nblocks,PetscInt *bsizes)
7320 {
7321   PetscErrorCode ierr;
7322   PetscInt       i,ncnt = 0, nlocal;
7323 
7324   PetscFunctionBegin;
7325   PetscValidHeaderSpecific(mat,MAT_CLASSID,1);
7326   if (nblocks < 0) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_SIZ,"Number of local blocks must be great than or equal to zero");
7327   ierr = MatGetLocalSize(mat,&nlocal,NULL);CHKERRQ(ierr);
7328   for (i=0; i<nblocks; i++) ncnt += bsizes[i];
7329   if (ncnt != nlocal) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_ARG_SIZ,"Sum of local block sizes %D does not equal local size of matrix %D",ncnt,nlocal);
7330   ierr = PetscFree(mat->bsizes);CHKERRQ(ierr);
7331   mat->nblocks = nblocks;
7332   ierr = PetscMalloc1(nblocks,&mat->bsizes);CHKERRQ(ierr);
7333   ierr = PetscMemcpy(mat->bsizes,bsizes,nblocks*sizeof(PetscInt));CHKERRQ(ierr);
7334   PetscFunctionReturn(0);
7335 }
7336 
7337 /*@C
7338    MatGetVariableBlockSizes - Gets a diagonal blocks of the matrix that need not be of the same size
7339 
7340    Logically Collective on Mat
7341 
7342    Input Parameters:
7343 .  mat - the matrix
7344 
7345    Output Parameters:
7346 +  nblocks - the number of blocks on this process
7347 -  bsizes - the block sizes
7348 
7349    Notes: Currently not supported from Fortran
7350 
7351    Level: intermediate
7352 
7353    Concepts: matrices^block size
7354 
7355 .seealso: MatCreateSeqBAIJ(), MatCreateBAIJ(), MatGetBlockSize(), MatSetBlockSizes(), MatGetBlockSizes(), MatSetVariableBlockSizes()
7356 @*/
7357 PetscErrorCode MatGetVariableBlockSizes(Mat mat,PetscInt *nblocks,const PetscInt **bsizes)
7358 {
7359   PetscFunctionBegin;
7360   PetscValidHeaderSpecific(mat,MAT_CLASSID,1);
7361   *nblocks = mat->nblocks;
7362   *bsizes  = mat->bsizes;
7363   PetscFunctionReturn(0);
7364 }
7365 
7366 /*@
7367    MatSetBlockSizes - Sets the matrix block row and column sizes.
7368 
7369    Logically Collective on Mat
7370 
7371    Input Parameters:
7372 +  mat - the matrix
7373 -  rbs - row block size
7374 -  cbs - column block size
7375 
7376    Notes:
7377     Block row formats are MATSEQBAIJ, MATMPIBAIJ, MATSEQSBAIJ, MATMPISBAIJ. These formats ALWAYS have square block storage in the matrix.
7378     If you pass a different block size for the columns than the rows, the row block size determines the square block storage.
7379     This must be called before MatSetUp() or MatXXXSetPreallocation() (or will default to 1) and the block size cannot be changed later
7380 
7381     For MATMPIAIJ and MATSEQAIJ matrix formats, this function can be called at a later stage, provided that the specified block sizes
7382     are compatible with the matrix local sizes.
7383 
7384     The row and column block size determine the blocksize of the "row" and "column" vectors returned by MatCreateVecs().
7385 
7386    Level: intermediate
7387 
7388    Concepts: matrices^block size
7389 
7390 .seealso: MatCreateSeqBAIJ(), MatCreateBAIJ(), MatGetBlockSize(), MatSetBlockSize(), MatGetBlockSizes()
7391 @*/
7392 PetscErrorCode MatSetBlockSizes(Mat mat,PetscInt rbs,PetscInt cbs)
7393 {
7394   PetscErrorCode ierr;
7395 
7396   PetscFunctionBegin;
7397   PetscValidHeaderSpecific(mat,MAT_CLASSID,1);
7398   PetscValidLogicalCollectiveInt(mat,rbs,2);
7399   PetscValidLogicalCollectiveInt(mat,cbs,3);
7400   if (mat->ops->setblocksizes) {
7401     ierr = (*mat->ops->setblocksizes)(mat,rbs,cbs);CHKERRQ(ierr);
7402   }
7403   if (mat->rmap->refcnt) {
7404     ISLocalToGlobalMapping l2g = NULL;
7405     PetscLayout            nmap = NULL;
7406 
7407     ierr = PetscLayoutDuplicate(mat->rmap,&nmap);CHKERRQ(ierr);
7408     if (mat->rmap->mapping) {
7409       ierr = ISLocalToGlobalMappingDuplicate(mat->rmap->mapping,&l2g);CHKERRQ(ierr);
7410     }
7411     ierr = PetscLayoutDestroy(&mat->rmap);CHKERRQ(ierr);
7412     mat->rmap = nmap;
7413     mat->rmap->mapping = l2g;
7414   }
7415   if (mat->cmap->refcnt) {
7416     ISLocalToGlobalMapping l2g = NULL;
7417     PetscLayout            nmap = NULL;
7418 
7419     ierr = PetscLayoutDuplicate(mat->cmap,&nmap);CHKERRQ(ierr);
7420     if (mat->cmap->mapping) {
7421       ierr = ISLocalToGlobalMappingDuplicate(mat->cmap->mapping,&l2g);CHKERRQ(ierr);
7422     }
7423     ierr = PetscLayoutDestroy(&mat->cmap);CHKERRQ(ierr);
7424     mat->cmap = nmap;
7425     mat->cmap->mapping = l2g;
7426   }
7427   ierr = PetscLayoutSetBlockSize(mat->rmap,rbs);CHKERRQ(ierr);
7428   ierr = PetscLayoutSetBlockSize(mat->cmap,cbs);CHKERRQ(ierr);
7429   PetscFunctionReturn(0);
7430 }
7431 
7432 /*@
7433    MatSetBlockSizesFromMats - Sets the matrix block row and column sizes to match a pair of matrices
7434 
7435    Logically Collective on Mat
7436 
7437    Input Parameters:
7438 +  mat - the matrix
7439 .  fromRow - matrix from which to copy row block size
7440 -  fromCol - matrix from which to copy column block size (can be same as fromRow)
7441 
7442    Level: developer
7443 
7444    Concepts: matrices^block size
7445 
7446 .seealso: MatCreateSeqBAIJ(), MatCreateBAIJ(), MatGetBlockSize(), MatSetBlockSizes()
7447 @*/
7448 PetscErrorCode MatSetBlockSizesFromMats(Mat mat,Mat fromRow,Mat fromCol)
7449 {
7450   PetscErrorCode ierr;
7451 
7452   PetscFunctionBegin;
7453   PetscValidHeaderSpecific(mat,MAT_CLASSID,1);
7454   PetscValidHeaderSpecific(fromRow,MAT_CLASSID,2);
7455   PetscValidHeaderSpecific(fromCol,MAT_CLASSID,3);
7456   if (fromRow->rmap->bs > 0) {ierr = PetscLayoutSetBlockSize(mat->rmap,fromRow->rmap->bs);CHKERRQ(ierr);}
7457   if (fromCol->cmap->bs > 0) {ierr = PetscLayoutSetBlockSize(mat->cmap,fromCol->cmap->bs);CHKERRQ(ierr);}
7458   PetscFunctionReturn(0);
7459 }
7460 
7461 /*@
7462    MatResidual - Default routine to calculate the residual.
7463 
7464    Collective on Mat and Vec
7465 
7466    Input Parameters:
7467 +  mat - the matrix
7468 .  b   - the right-hand-side
7469 -  x   - the approximate solution
7470 
7471    Output Parameter:
7472 .  r - location to store the residual
7473 
7474    Level: developer
7475 
7476 .keywords: MG, default, multigrid, residual
7477 
7478 .seealso: PCMGSetResidual()
7479 @*/
7480 PetscErrorCode MatResidual(Mat mat,Vec b,Vec x,Vec r)
7481 {
7482   PetscErrorCode ierr;
7483 
7484   PetscFunctionBegin;
7485   PetscValidHeaderSpecific(mat,MAT_CLASSID,1);
7486   PetscValidHeaderSpecific(b,VEC_CLASSID,2);
7487   PetscValidHeaderSpecific(x,VEC_CLASSID,3);
7488   PetscValidHeaderSpecific(r,VEC_CLASSID,4);
7489   PetscValidType(mat,1);
7490   MatCheckPreallocated(mat,1);
7491   ierr  = PetscLogEventBegin(MAT_Residual,mat,0,0,0);CHKERRQ(ierr);
7492   if (!mat->ops->residual) {
7493     ierr = MatMult(mat,x,r);CHKERRQ(ierr);
7494     ierr = VecAYPX(r,-1.0,b);CHKERRQ(ierr);
7495   } else {
7496     ierr  = (*mat->ops->residual)(mat,b,x,r);CHKERRQ(ierr);
7497   }
7498   ierr  = PetscLogEventEnd(MAT_Residual,mat,0,0,0);CHKERRQ(ierr);
7499   PetscFunctionReturn(0);
7500 }
7501 
7502 /*@C
7503     MatGetRowIJ - Returns the compressed row storage i and j indices for sequential matrices.
7504 
7505    Collective on Mat
7506 
7507     Input Parameters:
7508 +   mat - the matrix
7509 .   shift -  0 or 1 indicating we want the indices starting at 0 or 1
7510 .   symmetric - PETSC_TRUE or PETSC_FALSE indicating the matrix data structure should be   symmetrized
7511 -   inodecompressed - PETSC_TRUE or PETSC_FALSE  indicating if the nonzero structure of the
7512                  inodes or the nonzero elements is wanted. For BAIJ matrices the compressed version is
7513                  always used.
7514 
7515     Output Parameters:
7516 +   n - number of rows in the (possibly compressed) matrix
7517 .   ia - the row pointers; that is ia[0] = 0, ia[row] = ia[row-1] + number of elements in that row of the matrix
7518 .   ja - the column indices
7519 -   done - indicates if the routine actually worked and returned appropriate ia[] and ja[] arrays; callers
7520            are responsible for handling the case when done == PETSC_FALSE and ia and ja are not set
7521 
7522     Level: developer
7523 
7524     Notes:
7525     You CANNOT change any of the ia[] or ja[] values.
7526 
7527     Use MatRestoreRowIJ() when you are finished accessing the ia[] and ja[] values.
7528 
7529     Fortran Notes:
7530     In Fortran use
7531 $
7532 $      PetscInt ia(1), ja(1)
7533 $      PetscOffset iia, jja
7534 $      call MatGetRowIJ(mat,shift,symmetric,inodecompressed,n,ia,iia,ja,jja,done,ierr)
7535 $      ! Access the ith and jth entries via ia(iia + i) and ja(jja + j)
7536 
7537      or
7538 $
7539 $    PetscInt, pointer :: ia(:),ja(:)
7540 $    call MatGetRowIJF90(mat,shift,symmetric,inodecompressed,n,ia,ja,done,ierr)
7541 $    ! Access the ith and jth entries via ia(i) and ja(j)
7542 
7543 .seealso: MatGetColumnIJ(), MatRestoreRowIJ(), MatSeqAIJGetArray()
7544 @*/
7545 PetscErrorCode MatGetRowIJ(Mat mat,PetscInt shift,PetscBool symmetric,PetscBool inodecompressed,PetscInt *n,const PetscInt *ia[],const PetscInt *ja[],PetscBool  *done)
7546 {
7547   PetscErrorCode ierr;
7548 
7549   PetscFunctionBegin;
7550   PetscValidHeaderSpecific(mat,MAT_CLASSID,1);
7551   PetscValidType(mat,1);
7552   PetscValidIntPointer(n,5);
7553   if (ia) PetscValidIntPointer(ia,6);
7554   if (ja) PetscValidIntPointer(ja,7);
7555   PetscValidIntPointer(done,8);
7556   MatCheckPreallocated(mat,1);
7557   if (!mat->ops->getrowij) *done = PETSC_FALSE;
7558   else {
7559     *done = PETSC_TRUE;
7560     ierr  = PetscLogEventBegin(MAT_GetRowIJ,mat,0,0,0);CHKERRQ(ierr);
7561     ierr  = (*mat->ops->getrowij)(mat,shift,symmetric,inodecompressed,n,ia,ja,done);CHKERRQ(ierr);
7562     ierr  = PetscLogEventEnd(MAT_GetRowIJ,mat,0,0,0);CHKERRQ(ierr);
7563   }
7564   PetscFunctionReturn(0);
7565 }
7566 
7567 /*@C
7568     MatGetColumnIJ - Returns the compressed column storage i and j indices for sequential matrices.
7569 
7570     Collective on Mat
7571 
7572     Input Parameters:
7573 +   mat - the matrix
7574 .   shift - 1 or zero indicating we want the indices starting at 0 or 1
7575 .   symmetric - PETSC_TRUE or PETSC_FALSE indicating the matrix data structure should be
7576                 symmetrized
7577 .   inodecompressed - PETSC_TRUE or PETSC_FALSE indicating if the nonzero structure of the
7578                  inodes or the nonzero elements is wanted. For BAIJ matrices the compressed version is
7579                  always used.
7580 .   n - number of columns in the (possibly compressed) matrix
7581 .   ia - the column pointers; that is ia[0] = 0, ia[col] = i[col-1] + number of elements in that col of the matrix
7582 -   ja - the row indices
7583 
7584     Output Parameters:
7585 .   done - PETSC_TRUE or PETSC_FALSE, indicating whether the values have been returned
7586 
7587     Level: developer
7588 
7589 .seealso: MatGetRowIJ(), MatRestoreColumnIJ()
7590 @*/
7591 PetscErrorCode MatGetColumnIJ(Mat mat,PetscInt shift,PetscBool symmetric,PetscBool inodecompressed,PetscInt *n,const PetscInt *ia[],const PetscInt *ja[],PetscBool  *done)
7592 {
7593   PetscErrorCode ierr;
7594 
7595   PetscFunctionBegin;
7596   PetscValidHeaderSpecific(mat,MAT_CLASSID,1);
7597   PetscValidType(mat,1);
7598   PetscValidIntPointer(n,4);
7599   if (ia) PetscValidIntPointer(ia,5);
7600   if (ja) PetscValidIntPointer(ja,6);
7601   PetscValidIntPointer(done,7);
7602   MatCheckPreallocated(mat,1);
7603   if (!mat->ops->getcolumnij) *done = PETSC_FALSE;
7604   else {
7605     *done = PETSC_TRUE;
7606     ierr  = (*mat->ops->getcolumnij)(mat,shift,symmetric,inodecompressed,n,ia,ja,done);CHKERRQ(ierr);
7607   }
7608   PetscFunctionReturn(0);
7609 }
7610 
7611 /*@C
7612     MatRestoreRowIJ - Call after you are completed with the ia,ja indices obtained with
7613     MatGetRowIJ().
7614 
7615     Collective on Mat
7616 
7617     Input Parameters:
7618 +   mat - the matrix
7619 .   shift - 1 or zero indicating we want the indices starting at 0 or 1
7620 .   symmetric - PETSC_TRUE or PETSC_FALSE indicating the matrix data structure should be
7621                 symmetrized
7622 .   inodecompressed -  PETSC_TRUE or PETSC_FALSE indicating if the nonzero structure of the
7623                  inodes or the nonzero elements is wanted. For BAIJ matrices the compressed version is
7624                  always used.
7625 .   n - size of (possibly compressed) matrix
7626 .   ia - the row pointers
7627 -   ja - the column indices
7628 
7629     Output Parameters:
7630 .   done - PETSC_TRUE or PETSC_FALSE indicated that the values have been returned
7631 
7632     Note:
7633     This routine zeros out n, ia, and ja. This is to prevent accidental
7634     us of the array after it has been restored. If you pass NULL, it will
7635     not zero the pointers.  Use of ia or ja after MatRestoreRowIJ() is invalid.
7636 
7637     Level: developer
7638 
7639 .seealso: MatGetRowIJ(), MatRestoreColumnIJ()
7640 @*/
7641 PetscErrorCode MatRestoreRowIJ(Mat mat,PetscInt shift,PetscBool symmetric,PetscBool inodecompressed,PetscInt *n,const PetscInt *ia[],const PetscInt *ja[],PetscBool  *done)
7642 {
7643   PetscErrorCode ierr;
7644 
7645   PetscFunctionBegin;
7646   PetscValidHeaderSpecific(mat,MAT_CLASSID,1);
7647   PetscValidType(mat,1);
7648   if (ia) PetscValidIntPointer(ia,6);
7649   if (ja) PetscValidIntPointer(ja,7);
7650   PetscValidIntPointer(done,8);
7651   MatCheckPreallocated(mat,1);
7652 
7653   if (!mat->ops->restorerowij) *done = PETSC_FALSE;
7654   else {
7655     *done = PETSC_TRUE;
7656     ierr  = (*mat->ops->restorerowij)(mat,shift,symmetric,inodecompressed,n,ia,ja,done);CHKERRQ(ierr);
7657     if (n)  *n = 0;
7658     if (ia) *ia = NULL;
7659     if (ja) *ja = NULL;
7660   }
7661   PetscFunctionReturn(0);
7662 }
7663 
7664 /*@C
7665     MatRestoreColumnIJ - Call after you are completed with the ia,ja indices obtained with
7666     MatGetColumnIJ().
7667 
7668     Collective on Mat
7669 
7670     Input Parameters:
7671 +   mat - the matrix
7672 .   shift - 1 or zero indicating we want the indices starting at 0 or 1
7673 -   symmetric - PETSC_TRUE or PETSC_FALSE indicating the matrix data structure should be
7674                 symmetrized
7675 -   inodecompressed - PETSC_TRUE or PETSC_FALSE indicating if the nonzero structure of the
7676                  inodes or the nonzero elements is wanted. For BAIJ matrices the compressed version is
7677                  always used.
7678 
7679     Output Parameters:
7680 +   n - size of (possibly compressed) matrix
7681 .   ia - the column pointers
7682 .   ja - the row indices
7683 -   done - PETSC_TRUE or PETSC_FALSE indicated that the values have been returned
7684 
7685     Level: developer
7686 
7687 .seealso: MatGetColumnIJ(), MatRestoreRowIJ()
7688 @*/
7689 PetscErrorCode MatRestoreColumnIJ(Mat mat,PetscInt shift,PetscBool symmetric,PetscBool inodecompressed,PetscInt *n,const PetscInt *ia[],const PetscInt *ja[],PetscBool  *done)
7690 {
7691   PetscErrorCode ierr;
7692 
7693   PetscFunctionBegin;
7694   PetscValidHeaderSpecific(mat,MAT_CLASSID,1);
7695   PetscValidType(mat,1);
7696   if (ia) PetscValidIntPointer(ia,5);
7697   if (ja) PetscValidIntPointer(ja,6);
7698   PetscValidIntPointer(done,7);
7699   MatCheckPreallocated(mat,1);
7700 
7701   if (!mat->ops->restorecolumnij) *done = PETSC_FALSE;
7702   else {
7703     *done = PETSC_TRUE;
7704     ierr  = (*mat->ops->restorecolumnij)(mat,shift,symmetric,inodecompressed,n,ia,ja,done);CHKERRQ(ierr);
7705     if (n)  *n = 0;
7706     if (ia) *ia = NULL;
7707     if (ja) *ja = NULL;
7708   }
7709   PetscFunctionReturn(0);
7710 }
7711 
7712 /*@C
7713     MatColoringPatch -Used inside matrix coloring routines that
7714     use MatGetRowIJ() and/or MatGetColumnIJ().
7715 
7716     Collective on Mat
7717 
7718     Input Parameters:
7719 +   mat - the matrix
7720 .   ncolors - max color value
7721 .   n   - number of entries in colorarray
7722 -   colorarray - array indicating color for each column
7723 
7724     Output Parameters:
7725 .   iscoloring - coloring generated using colorarray information
7726 
7727     Level: developer
7728 
7729 .seealso: MatGetRowIJ(), MatGetColumnIJ()
7730 
7731 @*/
7732 PetscErrorCode MatColoringPatch(Mat mat,PetscInt ncolors,PetscInt n,ISColoringValue colorarray[],ISColoring *iscoloring)
7733 {
7734   PetscErrorCode ierr;
7735 
7736   PetscFunctionBegin;
7737   PetscValidHeaderSpecific(mat,MAT_CLASSID,1);
7738   PetscValidType(mat,1);
7739   PetscValidIntPointer(colorarray,4);
7740   PetscValidPointer(iscoloring,5);
7741   MatCheckPreallocated(mat,1);
7742 
7743   if (!mat->ops->coloringpatch) {
7744     ierr = ISColoringCreate(PetscObjectComm((PetscObject)mat),ncolors,n,colorarray,PETSC_OWN_POINTER,iscoloring);CHKERRQ(ierr);
7745   } else {
7746     ierr = (*mat->ops->coloringpatch)(mat,ncolors,n,colorarray,iscoloring);CHKERRQ(ierr);
7747   }
7748   PetscFunctionReturn(0);
7749 }
7750 
7751 
7752 /*@
7753    MatSetUnfactored - Resets a factored matrix to be treated as unfactored.
7754 
7755    Logically Collective on Mat
7756 
7757    Input Parameter:
7758 .  mat - the factored matrix to be reset
7759 
7760    Notes:
7761    This routine should be used only with factored matrices formed by in-place
7762    factorization via ILU(0) (or by in-place LU factorization for the MATSEQDENSE
7763    format).  This option can save memory, for example, when solving nonlinear
7764    systems with a matrix-free Newton-Krylov method and a matrix-based, in-place
7765    ILU(0) preconditioner.
7766 
7767    Note that one can specify in-place ILU(0) factorization by calling
7768 .vb
7769      PCType(pc,PCILU);
7770      PCFactorSeUseInPlace(pc);
7771 .ve
7772    or by using the options -pc_type ilu -pc_factor_in_place
7773 
7774    In-place factorization ILU(0) can also be used as a local
7775    solver for the blocks within the block Jacobi or additive Schwarz
7776    methods (runtime option: -sub_pc_factor_in_place).  See Users-Manual: ch_pc
7777    for details on setting local solver options.
7778 
7779    Most users should employ the simplified KSP interface for linear solvers
7780    instead of working directly with matrix algebra routines such as this.
7781    See, e.g., KSPCreate().
7782 
7783    Level: developer
7784 
7785 .seealso: PCFactorSetUseInPlace(), PCFactorGetUseInPlace()
7786 
7787    Concepts: matrices^unfactored
7788 
7789 @*/
7790 PetscErrorCode MatSetUnfactored(Mat mat)
7791 {
7792   PetscErrorCode ierr;
7793 
7794   PetscFunctionBegin;
7795   PetscValidHeaderSpecific(mat,MAT_CLASSID,1);
7796   PetscValidType(mat,1);
7797   MatCheckPreallocated(mat,1);
7798   mat->factortype = MAT_FACTOR_NONE;
7799   if (!mat->ops->setunfactored) PetscFunctionReturn(0);
7800   ierr = (*mat->ops->setunfactored)(mat);CHKERRQ(ierr);
7801   PetscFunctionReturn(0);
7802 }
7803 
7804 /*MC
7805     MatDenseGetArrayF90 - Accesses a matrix array from Fortran90.
7806 
7807     Synopsis:
7808     MatDenseGetArrayF90(Mat x,{Scalar, pointer :: xx_v(:,:)},integer ierr)
7809 
7810     Not collective
7811 
7812     Input Parameter:
7813 .   x - matrix
7814 
7815     Output Parameters:
7816 +   xx_v - the Fortran90 pointer to the array
7817 -   ierr - error code
7818 
7819     Example of Usage:
7820 .vb
7821       PetscScalar, pointer xx_v(:,:)
7822       ....
7823       call MatDenseGetArrayF90(x,xx_v,ierr)
7824       a = xx_v(3)
7825       call MatDenseRestoreArrayF90(x,xx_v,ierr)
7826 .ve
7827 
7828     Level: advanced
7829 
7830 .seealso:  MatDenseRestoreArrayF90(), MatDenseGetArray(), MatDenseRestoreArray(), MatSeqAIJGetArrayF90()
7831 
7832     Concepts: matrices^accessing array
7833 
7834 M*/
7835 
7836 /*MC
7837     MatDenseRestoreArrayF90 - Restores a matrix array that has been
7838     accessed with MatDenseGetArrayF90().
7839 
7840     Synopsis:
7841     MatDenseRestoreArrayF90(Mat x,{Scalar, pointer :: xx_v(:,:)},integer ierr)
7842 
7843     Not collective
7844 
7845     Input Parameters:
7846 +   x - matrix
7847 -   xx_v - the Fortran90 pointer to the array
7848 
7849     Output Parameter:
7850 .   ierr - error code
7851 
7852     Example of Usage:
7853 .vb
7854        PetscScalar, pointer xx_v(:,:)
7855        ....
7856        call MatDenseGetArrayF90(x,xx_v,ierr)
7857        a = xx_v(3)
7858        call MatDenseRestoreArrayF90(x,xx_v,ierr)
7859 .ve
7860 
7861     Level: advanced
7862 
7863 .seealso:  MatDenseGetArrayF90(), MatDenseGetArray(), MatDenseRestoreArray(), MatSeqAIJRestoreArrayF90()
7864 
7865 M*/
7866 
7867 
7868 /*MC
7869     MatSeqAIJGetArrayF90 - Accesses a matrix array from Fortran90.
7870 
7871     Synopsis:
7872     MatSeqAIJGetArrayF90(Mat x,{Scalar, pointer :: xx_v(:)},integer ierr)
7873 
7874     Not collective
7875 
7876     Input Parameter:
7877 .   x - matrix
7878 
7879     Output Parameters:
7880 +   xx_v - the Fortran90 pointer to the array
7881 -   ierr - error code
7882 
7883     Example of Usage:
7884 .vb
7885       PetscScalar, pointer xx_v(:)
7886       ....
7887       call MatSeqAIJGetArrayF90(x,xx_v,ierr)
7888       a = xx_v(3)
7889       call MatSeqAIJRestoreArrayF90(x,xx_v,ierr)
7890 .ve
7891 
7892     Level: advanced
7893 
7894 .seealso:  MatSeqAIJRestoreArrayF90(), MatSeqAIJGetArray(), MatSeqAIJRestoreArray(), MatDenseGetArrayF90()
7895 
7896     Concepts: matrices^accessing array
7897 
7898 M*/
7899 
7900 /*MC
7901     MatSeqAIJRestoreArrayF90 - Restores a matrix array that has been
7902     accessed with MatSeqAIJGetArrayF90().
7903 
7904     Synopsis:
7905     MatSeqAIJRestoreArrayF90(Mat x,{Scalar, pointer :: xx_v(:)},integer ierr)
7906 
7907     Not collective
7908 
7909     Input Parameters:
7910 +   x - matrix
7911 -   xx_v - the Fortran90 pointer to the array
7912 
7913     Output Parameter:
7914 .   ierr - error code
7915 
7916     Example of Usage:
7917 .vb
7918        PetscScalar, pointer xx_v(:)
7919        ....
7920        call MatSeqAIJGetArrayF90(x,xx_v,ierr)
7921        a = xx_v(3)
7922        call MatSeqAIJRestoreArrayF90(x,xx_v,ierr)
7923 .ve
7924 
7925     Level: advanced
7926 
7927 .seealso:  MatSeqAIJGetArrayF90(), MatSeqAIJGetArray(), MatSeqAIJRestoreArray(), MatDenseRestoreArrayF90()
7928 
7929 M*/
7930 
7931 
7932 /*@
7933     MatCreateSubMatrix - Gets a single submatrix on the same number of processors
7934                       as the original matrix.
7935 
7936     Collective on Mat
7937 
7938     Input Parameters:
7939 +   mat - the original matrix
7940 .   isrow - parallel IS containing the rows this processor should obtain
7941 .   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.
7942 -   cll - either MAT_INITIAL_MATRIX or MAT_REUSE_MATRIX
7943 
7944     Output Parameter:
7945 .   newmat - the new submatrix, of the same type as the old
7946 
7947     Level: advanced
7948 
7949     Notes:
7950     The submatrix will be able to be multiplied with vectors using the same layout as iscol.
7951 
7952     Some matrix types place restrictions on the row and column indices, such
7953     as that they be sorted or that they be equal to each other.
7954 
7955     The index sets may not have duplicate entries.
7956 
7957       The first time this is called you should use a cll of MAT_INITIAL_MATRIX,
7958    the MatCreateSubMatrix() routine will create the newmat for you. Any additional calls
7959    to this routine with a mat of the same nonzero structure and with a call of MAT_REUSE_MATRIX
7960    will reuse the matrix generated the first time.  You should call MatDestroy() on newmat when
7961    you are finished using it.
7962 
7963     The communicator of the newly obtained matrix is ALWAYS the same as the communicator of
7964     the input matrix.
7965 
7966     If iscol is NULL then all columns are obtained (not supported in Fortran).
7967 
7968    Example usage:
7969    Consider the following 8x8 matrix with 34 non-zero values, that is
7970    assembled across 3 processors. Let's assume that proc0 owns 3 rows,
7971    proc1 owns 3 rows, proc2 owns 2 rows. This division can be shown
7972    as follows:
7973 
7974 .vb
7975             1  2  0  |  0  3  0  |  0  4
7976     Proc0   0  5  6  |  7  0  0  |  8  0
7977             9  0 10  | 11  0  0  | 12  0
7978     -------------------------------------
7979            13  0 14  | 15 16 17  |  0  0
7980     Proc1   0 18  0  | 19 20 21  |  0  0
7981             0  0  0  | 22 23  0  | 24  0
7982     -------------------------------------
7983     Proc2  25 26 27  |  0  0 28  | 29  0
7984            30  0  0  | 31 32 33  |  0 34
7985 .ve
7986 
7987     Suppose isrow = [0 1 | 4 | 6 7] and iscol = [1 2 | 3 4 5 | 6].  The resulting submatrix is
7988 
7989 .vb
7990             2  0  |  0  3  0  |  0
7991     Proc0   5  6  |  7  0  0  |  8
7992     -------------------------------
7993     Proc1  18  0  | 19 20 21  |  0
7994     -------------------------------
7995     Proc2  26 27  |  0  0 28  | 29
7996             0  0  | 31 32 33  |  0
7997 .ve
7998 
7999 
8000     Concepts: matrices^submatrices
8001 
8002 .seealso: MatCreateSubMatrices()
8003 @*/
8004 PetscErrorCode MatCreateSubMatrix(Mat mat,IS isrow,IS iscol,MatReuse cll,Mat *newmat)
8005 {
8006   PetscErrorCode ierr;
8007   PetscMPIInt    size;
8008   Mat            *local;
8009   IS             iscoltmp;
8010 
8011   PetscFunctionBegin;
8012   PetscValidHeaderSpecific(mat,MAT_CLASSID,1);
8013   PetscValidHeaderSpecific(isrow,IS_CLASSID,2);
8014   if (iscol) PetscValidHeaderSpecific(iscol,IS_CLASSID,3);
8015   PetscValidPointer(newmat,5);
8016   if (cll == MAT_REUSE_MATRIX) PetscValidHeaderSpecific(*newmat,MAT_CLASSID,5);
8017   PetscValidType(mat,1);
8018   if (mat->factortype) SETERRQ(PetscObjectComm((PetscObject)mat),PETSC_ERR_ARG_WRONGSTATE,"Not for factored matrix");
8019   if (cll == MAT_IGNORE_MATRIX) SETERRQ(PetscObjectComm((PetscObject)mat),PETSC_ERR_ARG_WRONGSTATE,"Cannot use MAT_IGNORE_MATRIX");
8020 
8021   MatCheckPreallocated(mat,1);
8022   ierr = MPI_Comm_size(PetscObjectComm((PetscObject)mat),&size);CHKERRQ(ierr);
8023 
8024   if (!iscol || isrow == iscol) {
8025     PetscBool   stride;
8026     PetscMPIInt grabentirematrix = 0,grab;
8027     ierr = PetscObjectTypeCompare((PetscObject)isrow,ISSTRIDE,&stride);CHKERRQ(ierr);
8028     if (stride) {
8029       PetscInt first,step,n,rstart,rend;
8030       ierr = ISStrideGetInfo(isrow,&first,&step);CHKERRQ(ierr);
8031       if (step == 1) {
8032         ierr = MatGetOwnershipRange(mat,&rstart,&rend);CHKERRQ(ierr);
8033         if (rstart == first) {
8034           ierr = ISGetLocalSize(isrow,&n);CHKERRQ(ierr);
8035           if (n == rend-rstart) {
8036             grabentirematrix = 1;
8037           }
8038         }
8039       }
8040     }
8041     ierr = MPIU_Allreduce(&grabentirematrix,&grab,1,MPI_INT,MPI_MIN,PetscObjectComm((PetscObject)mat));CHKERRQ(ierr);
8042     if (grab) {
8043       ierr = PetscInfo(mat,"Getting entire matrix as submatrix\n");CHKERRQ(ierr);
8044       if (cll == MAT_INITIAL_MATRIX) {
8045         *newmat = mat;
8046         ierr    = PetscObjectReference((PetscObject)mat);CHKERRQ(ierr);
8047       }
8048       PetscFunctionReturn(0);
8049     }
8050   }
8051 
8052   if (!iscol) {
8053     ierr = ISCreateStride(PetscObjectComm((PetscObject)mat),mat->cmap->n,mat->cmap->rstart,1,&iscoltmp);CHKERRQ(ierr);
8054   } else {
8055     iscoltmp = iscol;
8056   }
8057 
8058   /* if original matrix is on just one processor then use submatrix generated */
8059   if (mat->ops->createsubmatrices && !mat->ops->createsubmatrix && size == 1 && cll == MAT_REUSE_MATRIX) {
8060     ierr = MatCreateSubMatrices(mat,1,&isrow,&iscoltmp,MAT_REUSE_MATRIX,&newmat);CHKERRQ(ierr);
8061     goto setproperties;
8062   } else if (mat->ops->createsubmatrices && !mat->ops->createsubmatrix && size == 1) {
8063     ierr    = MatCreateSubMatrices(mat,1,&isrow,&iscoltmp,MAT_INITIAL_MATRIX,&local);CHKERRQ(ierr);
8064     *newmat = *local;
8065     ierr    = PetscFree(local);CHKERRQ(ierr);
8066     goto setproperties;
8067   } else if (!mat->ops->createsubmatrix) {
8068     /* Create a new matrix type that implements the operation using the full matrix */
8069     ierr = PetscLogEventBegin(MAT_CreateSubMat,mat,0,0,0);CHKERRQ(ierr);
8070     switch (cll) {
8071     case MAT_INITIAL_MATRIX:
8072       ierr = MatCreateSubMatrixVirtual(mat,isrow,iscoltmp,newmat);CHKERRQ(ierr);
8073       break;
8074     case MAT_REUSE_MATRIX:
8075       ierr = MatSubMatrixVirtualUpdate(*newmat,mat,isrow,iscoltmp);CHKERRQ(ierr);
8076       break;
8077     default: SETERRQ(PetscObjectComm((PetscObject)mat),PETSC_ERR_ARG_OUTOFRANGE,"Invalid MatReuse, must be either MAT_INITIAL_MATRIX or MAT_REUSE_MATRIX");
8078     }
8079     ierr = PetscLogEventEnd(MAT_CreateSubMat,mat,0,0,0);CHKERRQ(ierr);
8080     goto setproperties;
8081   }
8082 
8083   if (!mat->ops->createsubmatrix) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_SUP,"Mat type %s",((PetscObject)mat)->type_name);
8084   ierr = PetscLogEventBegin(MAT_CreateSubMat,mat,0,0,0);CHKERRQ(ierr);
8085   ierr = (*mat->ops->createsubmatrix)(mat,isrow,iscoltmp,cll,newmat);CHKERRQ(ierr);
8086   ierr = PetscLogEventEnd(MAT_CreateSubMat,mat,0,0,0);CHKERRQ(ierr);
8087 
8088   /* Propagate symmetry information for diagonal blocks */
8089 setproperties:
8090   if (isrow == iscoltmp) {
8091     if (mat->symmetric_set && mat->symmetric) {
8092       ierr = MatSetOption(*newmat,MAT_SYMMETRIC,PETSC_TRUE);CHKERRQ(ierr);
8093     }
8094     if (mat->structurally_symmetric_set && mat->structurally_symmetric) {
8095       ierr = MatSetOption(*newmat,MAT_STRUCTURALLY_SYMMETRIC,PETSC_TRUE);CHKERRQ(ierr);
8096     }
8097     if (mat->hermitian_set && mat->hermitian) {
8098       ierr = MatSetOption(*newmat,MAT_HERMITIAN,PETSC_TRUE);CHKERRQ(ierr);
8099     }
8100     if (mat->spd_set && mat->spd) {
8101       ierr = MatSetOption(*newmat,MAT_SPD,PETSC_TRUE);CHKERRQ(ierr);
8102     }
8103   }
8104 
8105   if (!iscol) {ierr = ISDestroy(&iscoltmp);CHKERRQ(ierr);}
8106   if (*newmat && cll == MAT_INITIAL_MATRIX) {ierr = PetscObjectStateIncrease((PetscObject)*newmat);CHKERRQ(ierr);}
8107   PetscFunctionReturn(0);
8108 }
8109 
8110 /*@
8111    MatStashSetInitialSize - sets the sizes of the matrix stash, that is
8112    used during the assembly process to store values that belong to
8113    other processors.
8114 
8115    Not Collective
8116 
8117    Input Parameters:
8118 +  mat   - the matrix
8119 .  size  - the initial size of the stash.
8120 -  bsize - the initial size of the block-stash(if used).
8121 
8122    Options Database Keys:
8123 +   -matstash_initial_size <size> or <size0,size1,...sizep-1>
8124 -   -matstash_block_initial_size <bsize>  or <bsize0,bsize1,...bsizep-1>
8125 
8126    Level: intermediate
8127 
8128    Notes:
8129      The block-stash is used for values set with MatSetValuesBlocked() while
8130      the stash is used for values set with MatSetValues()
8131 
8132      Run with the option -info and look for output of the form
8133      MatAssemblyBegin_MPIXXX:Stash has MM entries, uses nn mallocs.
8134      to determine the appropriate value, MM, to use for size and
8135      MatAssemblyBegin_MPIXXX:Block-Stash has BMM entries, uses nn mallocs.
8136      to determine the value, BMM to use for bsize
8137 
8138    Concepts: stash^setting matrix size
8139    Concepts: matrices^stash
8140 
8141 .seealso: MatAssemblyBegin(), MatAssemblyEnd(), Mat, MatStashGetInfo()
8142 
8143 @*/
8144 PetscErrorCode MatStashSetInitialSize(Mat mat,PetscInt size, PetscInt bsize)
8145 {
8146   PetscErrorCode ierr;
8147 
8148   PetscFunctionBegin;
8149   PetscValidHeaderSpecific(mat,MAT_CLASSID,1);
8150   PetscValidType(mat,1);
8151   ierr = MatStashSetInitialSize_Private(&mat->stash,size);CHKERRQ(ierr);
8152   ierr = MatStashSetInitialSize_Private(&mat->bstash,bsize);CHKERRQ(ierr);
8153   PetscFunctionReturn(0);
8154 }
8155 
8156 /*@
8157    MatInterpolateAdd - w = y + A*x or A'*x depending on the shape of
8158      the matrix
8159 
8160    Neighbor-wise Collective on Mat
8161 
8162    Input Parameters:
8163 +  mat   - the matrix
8164 .  x,y - the vectors
8165 -  w - where the result is stored
8166 
8167    Level: intermediate
8168 
8169    Notes:
8170     w may be the same vector as y.
8171 
8172     This allows one to use either the restriction or interpolation (its transpose)
8173     matrix to do the interpolation
8174 
8175     Concepts: interpolation
8176 
8177 .seealso: MatMultAdd(), MatMultTransposeAdd(), MatRestrict()
8178 
8179 @*/
8180 PetscErrorCode MatInterpolateAdd(Mat A,Vec x,Vec y,Vec w)
8181 {
8182   PetscErrorCode ierr;
8183   PetscInt       M,N,Ny;
8184 
8185   PetscFunctionBegin;
8186   PetscValidHeaderSpecific(A,MAT_CLASSID,1);
8187   PetscValidHeaderSpecific(x,VEC_CLASSID,2);
8188   PetscValidHeaderSpecific(y,VEC_CLASSID,3);
8189   PetscValidHeaderSpecific(w,VEC_CLASSID,4);
8190   PetscValidType(A,1);
8191   MatCheckPreallocated(A,1);
8192   ierr = MatGetSize(A,&M,&N);CHKERRQ(ierr);
8193   ierr = VecGetSize(y,&Ny);CHKERRQ(ierr);
8194   if (M == Ny) {
8195     ierr = MatMultAdd(A,x,y,w);CHKERRQ(ierr);
8196   } else {
8197     ierr = MatMultTransposeAdd(A,x,y,w);CHKERRQ(ierr);
8198   }
8199   PetscFunctionReturn(0);
8200 }
8201 
8202 /*@
8203    MatInterpolate - y = A*x or A'*x depending on the shape of
8204      the matrix
8205 
8206    Neighbor-wise Collective on Mat
8207 
8208    Input Parameters:
8209 +  mat   - the matrix
8210 -  x,y - the vectors
8211 
8212    Level: intermediate
8213 
8214    Notes:
8215     This allows one to use either the restriction or interpolation (its transpose)
8216     matrix to do the interpolation
8217 
8218    Concepts: matrices^interpolation
8219 
8220 .seealso: MatMultAdd(), MatMultTransposeAdd(), MatRestrict()
8221 
8222 @*/
8223 PetscErrorCode MatInterpolate(Mat A,Vec x,Vec y)
8224 {
8225   PetscErrorCode ierr;
8226   PetscInt       M,N,Ny;
8227 
8228   PetscFunctionBegin;
8229   PetscValidHeaderSpecific(A,MAT_CLASSID,1);
8230   PetscValidHeaderSpecific(x,VEC_CLASSID,2);
8231   PetscValidHeaderSpecific(y,VEC_CLASSID,3);
8232   PetscValidType(A,1);
8233   MatCheckPreallocated(A,1);
8234   ierr = MatGetSize(A,&M,&N);CHKERRQ(ierr);
8235   ierr = VecGetSize(y,&Ny);CHKERRQ(ierr);
8236   if (M == Ny) {
8237     ierr = MatMult(A,x,y);CHKERRQ(ierr);
8238   } else {
8239     ierr = MatMultTranspose(A,x,y);CHKERRQ(ierr);
8240   }
8241   PetscFunctionReturn(0);
8242 }
8243 
8244 /*@
8245    MatRestrict - y = A*x or A'*x
8246 
8247    Neighbor-wise Collective on Mat
8248 
8249    Input Parameters:
8250 +  mat   - the matrix
8251 -  x,y - the vectors
8252 
8253    Level: intermediate
8254 
8255    Notes:
8256     This allows one to use either the restriction or interpolation (its transpose)
8257     matrix to do the restriction
8258 
8259    Concepts: matrices^restriction
8260 
8261 .seealso: MatMultAdd(), MatMultTransposeAdd(), MatInterpolate()
8262 
8263 @*/
8264 PetscErrorCode MatRestrict(Mat A,Vec x,Vec y)
8265 {
8266   PetscErrorCode ierr;
8267   PetscInt       M,N,Ny;
8268 
8269   PetscFunctionBegin;
8270   PetscValidHeaderSpecific(A,MAT_CLASSID,1);
8271   PetscValidHeaderSpecific(x,VEC_CLASSID,2);
8272   PetscValidHeaderSpecific(y,VEC_CLASSID,3);
8273   PetscValidType(A,1);
8274   MatCheckPreallocated(A,1);
8275 
8276   ierr = MatGetSize(A,&M,&N);CHKERRQ(ierr);
8277   ierr = VecGetSize(y,&Ny);CHKERRQ(ierr);
8278   if (M == Ny) {
8279     ierr = MatMult(A,x,y);CHKERRQ(ierr);
8280   } else {
8281     ierr = MatMultTranspose(A,x,y);CHKERRQ(ierr);
8282   }
8283   PetscFunctionReturn(0);
8284 }
8285 
8286 /*@
8287    MatGetNullSpace - retrieves the null space of a matrix.
8288 
8289    Logically Collective on Mat and MatNullSpace
8290 
8291    Input Parameters:
8292 +  mat - the matrix
8293 -  nullsp - the null space object
8294 
8295    Level: developer
8296 
8297    Concepts: null space^attaching to matrix
8298 
8299 .seealso: MatCreate(), MatNullSpaceCreate(), MatSetNearNullSpace(), MatSetNullSpace()
8300 @*/
8301 PetscErrorCode MatGetNullSpace(Mat mat, MatNullSpace *nullsp)
8302 {
8303   PetscFunctionBegin;
8304   PetscValidHeaderSpecific(mat,MAT_CLASSID,1);
8305   PetscValidPointer(nullsp,2);
8306   *nullsp = (mat->symmetric_set && mat->symmetric && !mat->nullsp) ? mat->transnullsp : mat->nullsp;
8307   PetscFunctionReturn(0);
8308 }
8309 
8310 /*@
8311    MatSetNullSpace - attaches a null space to a matrix.
8312 
8313    Logically Collective on Mat and MatNullSpace
8314 
8315    Input Parameters:
8316 +  mat - the matrix
8317 -  nullsp - the null space object
8318 
8319    Level: advanced
8320 
8321    Notes:
8322       This null space is used by the linear solvers. Overwrites any previous null space that may have been attached
8323 
8324       For inconsistent singular systems (linear systems where the right hand side is not in the range of the operator) you also likely should
8325       call MatSetTransposeNullSpace(). This allows the linear system to be solved in a least squares sense.
8326 
8327       You can remove the null space by calling this routine with an nullsp of NULL
8328 
8329 
8330       The fundamental theorem of linear algebra (Gilbert Strang, Introduction to Applied Mathematics, page 72) states that
8331    the domain of a matrix A (from R^n to R^m (m rows, n columns) R^n = the direct sum of the null space of A, n(A), + the range of A^T, R(A^T).
8332    Similarly R^m = direct sum n(A^T) + R(A).  Hence the linear system A x = b has a solution only if b in R(A) (or correspondingly b is orthogonal to
8333    n(A^T)) and if x is a solution then x + alpha n(A) is a solution for any alpha. The minimum norm solution is orthogonal to n(A). For problems without a solution
8334    the solution that minimizes the norm of the residual (the least squares solution) can be obtained by solving A x = \hat{b} where \hat{b} is b orthogonalized to the n(A^T).
8335 
8336       Krylov solvers can produce the minimal norm solution to the least squares problem by utilizing MatNullSpaceRemove().
8337 
8338     If the matrix is known to be symmetric because it is an SBAIJ matrix or one as called MatSetOption(mat,MAT_SYMMETRIC or MAT_SYMMETRIC_ETERNAL,PETSC_TRUE); this
8339     routine also automatically calls MatSetTransposeNullSpace().
8340 
8341    Concepts: null space^attaching to matrix
8342 
8343 .seealso: MatCreate(), MatNullSpaceCreate(), MatSetNearNullSpace(), MatGetNullSpace(), MatSetTransposeNullSpace(), MatGetTransposeNullSpace(), MatNullSpaceRemove()
8344 @*/
8345 PetscErrorCode MatSetNullSpace(Mat mat,MatNullSpace nullsp)
8346 {
8347   PetscErrorCode ierr;
8348 
8349   PetscFunctionBegin;
8350   PetscValidHeaderSpecific(mat,MAT_CLASSID,1);
8351   if (nullsp) PetscValidHeaderSpecific(nullsp,MAT_NULLSPACE_CLASSID,2);
8352   if (nullsp) {ierr = PetscObjectReference((PetscObject)nullsp);CHKERRQ(ierr);}
8353   ierr = MatNullSpaceDestroy(&mat->nullsp);CHKERRQ(ierr);
8354   mat->nullsp = nullsp;
8355   if (mat->symmetric_set && mat->symmetric) {
8356     ierr = MatSetTransposeNullSpace(mat,nullsp);CHKERRQ(ierr);
8357   }
8358   PetscFunctionReturn(0);
8359 }
8360 
8361 /*@
8362    MatGetTransposeNullSpace - retrieves the null space of the transpose of a matrix.
8363 
8364    Logically Collective on Mat and MatNullSpace
8365 
8366    Input Parameters:
8367 +  mat - the matrix
8368 -  nullsp - the null space object
8369 
8370    Level: developer
8371 
8372    Concepts: null space^attaching to matrix
8373 
8374 .seealso: MatCreate(), MatNullSpaceCreate(), MatSetNearNullSpace(), MatSetTransposeNullSpace(), MatSetNullSpace(), MatGetNullSpace()
8375 @*/
8376 PetscErrorCode MatGetTransposeNullSpace(Mat mat, MatNullSpace *nullsp)
8377 {
8378   PetscFunctionBegin;
8379   PetscValidHeaderSpecific(mat,MAT_CLASSID,1);
8380   PetscValidType(mat,1);
8381   PetscValidPointer(nullsp,2);
8382   *nullsp = (mat->symmetric_set && mat->symmetric && !mat->transnullsp) ? mat->nullsp : mat->transnullsp;
8383   PetscFunctionReturn(0);
8384 }
8385 
8386 /*@
8387    MatSetTransposeNullSpace - attaches a null space to a matrix.
8388 
8389    Logically Collective on Mat and MatNullSpace
8390 
8391    Input Parameters:
8392 +  mat - the matrix
8393 -  nullsp - the null space object
8394 
8395    Level: advanced
8396 
8397    Notes:
8398       For inconsistent singular systems (linear systems where the right hand side is not in the range of the operator) this allows the linear system to be solved in a least squares sense.
8399       You must also call MatSetNullSpace()
8400 
8401 
8402       The fundamental theorem of linear algebra (Gilbert Strang, Introduction to Applied Mathematics, page 72) states that
8403    the domain of a matrix A (from R^n to R^m (m rows, n columns) R^n = the direct sum of the null space of A, n(A), + the range of A^T, R(A^T).
8404    Similarly R^m = direct sum n(A^T) + R(A).  Hence the linear system A x = b has a solution only if b in R(A) (or correspondingly b is orthogonal to
8405    n(A^T)) and if x is a solution then x + alpha n(A) is a solution for any alpha. The minimum norm solution is orthogonal to n(A). For problems without a solution
8406    the solution that minimizes the norm of the residual (the least squares solution) can be obtained by solving A x = \hat{b} where \hat{b} is b orthogonalized to the n(A^T).
8407 
8408       Krylov solvers can produce the minimal norm solution to the least squares problem by utilizing MatNullSpaceRemove().
8409 
8410    Concepts: null space^attaching to matrix
8411 
8412 .seealso: MatCreate(), MatNullSpaceCreate(), MatSetNearNullSpace(), MatGetNullSpace(), MatSetNullSpace(), MatGetTransposeNullSpace(), MatNullSpaceRemove()
8413 @*/
8414 PetscErrorCode MatSetTransposeNullSpace(Mat mat,MatNullSpace nullsp)
8415 {
8416   PetscErrorCode ierr;
8417 
8418   PetscFunctionBegin;
8419   PetscValidHeaderSpecific(mat,MAT_CLASSID,1);
8420   if (nullsp) PetscValidHeaderSpecific(nullsp,MAT_NULLSPACE_CLASSID,2);
8421   if (nullsp) {ierr = PetscObjectReference((PetscObject)nullsp);CHKERRQ(ierr);}
8422   ierr = MatNullSpaceDestroy(&mat->transnullsp);CHKERRQ(ierr);
8423   mat->transnullsp = nullsp;
8424   PetscFunctionReturn(0);
8425 }
8426 
8427 /*@
8428    MatSetNearNullSpace - attaches a null space to a matrix, which is often the null space (rigid body modes) of the operator without boundary conditions
8429         This null space will be used to provide near null space vectors to a multigrid preconditioner built from this matrix.
8430 
8431    Logically Collective on Mat and MatNullSpace
8432 
8433    Input Parameters:
8434 +  mat - the matrix
8435 -  nullsp - the null space object
8436 
8437    Level: advanced
8438 
8439    Notes:
8440       Overwrites any previous near null space that may have been attached
8441 
8442       You can remove the null space by calling this routine with an nullsp of NULL
8443 
8444    Concepts: null space^attaching to matrix
8445 
8446 .seealso: MatCreate(), MatNullSpaceCreate(), MatSetNullSpace(), MatNullSpaceCreateRigidBody(), MatGetNearNullSpace()
8447 @*/
8448 PetscErrorCode MatSetNearNullSpace(Mat mat,MatNullSpace nullsp)
8449 {
8450   PetscErrorCode ierr;
8451 
8452   PetscFunctionBegin;
8453   PetscValidHeaderSpecific(mat,MAT_CLASSID,1);
8454   PetscValidType(mat,1);
8455   if (nullsp) PetscValidHeaderSpecific(nullsp,MAT_NULLSPACE_CLASSID,2);
8456   MatCheckPreallocated(mat,1);
8457   if (nullsp) {ierr = PetscObjectReference((PetscObject)nullsp);CHKERRQ(ierr);}
8458   ierr = MatNullSpaceDestroy(&mat->nearnullsp);CHKERRQ(ierr);
8459   mat->nearnullsp = nullsp;
8460   PetscFunctionReturn(0);
8461 }
8462 
8463 /*@
8464    MatGetNearNullSpace -Get null space attached with MatSetNearNullSpace()
8465 
8466    Not Collective
8467 
8468    Input Parameters:
8469 .  mat - the matrix
8470 
8471    Output Parameters:
8472 .  nullsp - the null space object, NULL if not set
8473 
8474    Level: developer
8475 
8476    Concepts: null space^attaching to matrix
8477 
8478 .seealso: MatSetNearNullSpace(), MatGetNullSpace(), MatNullSpaceCreate()
8479 @*/
8480 PetscErrorCode MatGetNearNullSpace(Mat mat,MatNullSpace *nullsp)
8481 {
8482   PetscFunctionBegin;
8483   PetscValidHeaderSpecific(mat,MAT_CLASSID,1);
8484   PetscValidType(mat,1);
8485   PetscValidPointer(nullsp,2);
8486   MatCheckPreallocated(mat,1);
8487   *nullsp = mat->nearnullsp;
8488   PetscFunctionReturn(0);
8489 }
8490 
8491 /*@C
8492    MatICCFactor - Performs in-place incomplete Cholesky factorization of matrix.
8493 
8494    Collective on Mat
8495 
8496    Input Parameters:
8497 +  mat - the matrix
8498 .  row - row/column permutation
8499 .  fill - expected fill factor >= 1.0
8500 -  level - level of fill, for ICC(k)
8501 
8502    Notes:
8503    Probably really in-place only when level of fill is zero, otherwise allocates
8504    new space to store factored matrix and deletes previous memory.
8505 
8506    Most users should employ the simplified KSP interface for linear solvers
8507    instead of working directly with matrix algebra routines such as this.
8508    See, e.g., KSPCreate().
8509 
8510    Level: developer
8511 
8512    Concepts: matrices^incomplete Cholesky factorization
8513    Concepts: Cholesky factorization
8514 
8515 .seealso: MatICCFactorSymbolic(), MatLUFactorNumeric(), MatCholeskyFactor()
8516 
8517     Developer Note: fortran interface is not autogenerated as the f90
8518     interface defintion cannot be generated correctly [due to MatFactorInfo]
8519 
8520 @*/
8521 PetscErrorCode MatICCFactor(Mat mat,IS row,const MatFactorInfo *info)
8522 {
8523   PetscErrorCode ierr;
8524 
8525   PetscFunctionBegin;
8526   PetscValidHeaderSpecific(mat,MAT_CLASSID,1);
8527   PetscValidType(mat,1);
8528   if (row) PetscValidHeaderSpecific(row,IS_CLASSID,2);
8529   PetscValidPointer(info,3);
8530   if (mat->rmap->N != mat->cmap->N) SETERRQ(PetscObjectComm((PetscObject)mat),PETSC_ERR_ARG_WRONG,"matrix must be square");
8531   if (!mat->assembled) SETERRQ(PetscObjectComm((PetscObject)mat),PETSC_ERR_ARG_WRONGSTATE,"Not for unassembled matrix");
8532   if (mat->factortype) SETERRQ(PetscObjectComm((PetscObject)mat),PETSC_ERR_ARG_WRONGSTATE,"Not for factored matrix");
8533   if (!mat->ops->iccfactor) SETERRQ1(PetscObjectComm((PetscObject)mat),PETSC_ERR_SUP,"Mat type %s",((PetscObject)mat)->type_name);
8534   MatCheckPreallocated(mat,1);
8535   ierr = (*mat->ops->iccfactor)(mat,row,info);CHKERRQ(ierr);
8536   ierr = PetscObjectStateIncrease((PetscObject)mat);CHKERRQ(ierr);
8537   PetscFunctionReturn(0);
8538 }
8539 
8540 /*@
8541    MatDiagonalScaleLocal - Scales columns of a matrix given the scaling values including the
8542          ghosted ones.
8543 
8544    Not Collective
8545 
8546    Input Parameters:
8547 +  mat - the matrix
8548 -  diag = the diagonal values, including ghost ones
8549 
8550    Level: developer
8551 
8552    Notes:
8553     Works only for MPIAIJ and MPIBAIJ matrices
8554 
8555 .seealso: MatDiagonalScale()
8556 @*/
8557 PetscErrorCode MatDiagonalScaleLocal(Mat mat,Vec diag)
8558 {
8559   PetscErrorCode ierr;
8560   PetscMPIInt    size;
8561 
8562   PetscFunctionBegin;
8563   PetscValidHeaderSpecific(mat,MAT_CLASSID,1);
8564   PetscValidHeaderSpecific(diag,VEC_CLASSID,2);
8565   PetscValidType(mat,1);
8566 
8567   if (!mat->assembled) SETERRQ(PetscObjectComm((PetscObject)mat),PETSC_ERR_ARG_WRONGSTATE,"Matrix must be already assembled");
8568   ierr = PetscLogEventBegin(MAT_Scale,mat,0,0,0);CHKERRQ(ierr);
8569   ierr = MPI_Comm_size(PetscObjectComm((PetscObject)mat),&size);CHKERRQ(ierr);
8570   if (size == 1) {
8571     PetscInt n,m;
8572     ierr = VecGetSize(diag,&n);CHKERRQ(ierr);
8573     ierr = MatGetSize(mat,0,&m);CHKERRQ(ierr);
8574     if (m == n) {
8575       ierr = MatDiagonalScale(mat,0,diag);CHKERRQ(ierr);
8576     } else SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"Only supported for sequential matrices when no ghost points/periodic conditions");
8577   } else {
8578     ierr = PetscUseMethod(mat,"MatDiagonalScaleLocal_C",(Mat,Vec),(mat,diag));CHKERRQ(ierr);
8579   }
8580   ierr = PetscLogEventEnd(MAT_Scale,mat,0,0,0);CHKERRQ(ierr);
8581   ierr = PetscObjectStateIncrease((PetscObject)mat);CHKERRQ(ierr);
8582   PetscFunctionReturn(0);
8583 }
8584 
8585 /*@
8586    MatGetInertia - Gets the inertia from a factored matrix
8587 
8588    Collective on Mat
8589 
8590    Input Parameter:
8591 .  mat - the matrix
8592 
8593    Output Parameters:
8594 +   nneg - number of negative eigenvalues
8595 .   nzero - number of zero eigenvalues
8596 -   npos - number of positive eigenvalues
8597 
8598    Level: advanced
8599 
8600    Notes:
8601     Matrix must have been factored by MatCholeskyFactor()
8602 
8603 
8604 @*/
8605 PetscErrorCode MatGetInertia(Mat mat,PetscInt *nneg,PetscInt *nzero,PetscInt *npos)
8606 {
8607   PetscErrorCode ierr;
8608 
8609   PetscFunctionBegin;
8610   PetscValidHeaderSpecific(mat,MAT_CLASSID,1);
8611   PetscValidType(mat,1);
8612   if (!mat->factortype) SETERRQ(PetscObjectComm((PetscObject)mat),PETSC_ERR_ARG_WRONGSTATE,"Unfactored matrix");
8613   if (!mat->assembled) SETERRQ(PetscObjectComm((PetscObject)mat),PETSC_ERR_ARG_WRONGSTATE,"Numeric factor mat is not assembled");
8614   if (!mat->ops->getinertia) SETERRQ1(PetscObjectComm((PetscObject)mat),PETSC_ERR_SUP,"Mat type %s",((PetscObject)mat)->type_name);
8615   ierr = (*mat->ops->getinertia)(mat,nneg,nzero,npos);CHKERRQ(ierr);
8616   PetscFunctionReturn(0);
8617 }
8618 
8619 /* ----------------------------------------------------------------*/
8620 /*@C
8621    MatSolves - Solves A x = b, given a factored matrix, for a collection of vectors
8622 
8623    Neighbor-wise Collective on Mat and Vecs
8624 
8625    Input Parameters:
8626 +  mat - the factored matrix
8627 -  b - the right-hand-side vectors
8628 
8629    Output Parameter:
8630 .  x - the result vectors
8631 
8632    Notes:
8633    The vectors b and x cannot be the same.  I.e., one cannot
8634    call MatSolves(A,x,x).
8635 
8636    Notes:
8637    Most users should employ the simplified KSP interface for linear solvers
8638    instead of working directly with matrix algebra routines such as this.
8639    See, e.g., KSPCreate().
8640 
8641    Level: developer
8642 
8643    Concepts: matrices^triangular solves
8644 
8645 .seealso: MatSolveAdd(), MatSolveTranspose(), MatSolveTransposeAdd(), MatSolve()
8646 @*/
8647 PetscErrorCode MatSolves(Mat mat,Vecs b,Vecs x)
8648 {
8649   PetscErrorCode ierr;
8650 
8651   PetscFunctionBegin;
8652   PetscValidHeaderSpecific(mat,MAT_CLASSID,1);
8653   PetscValidType(mat,1);
8654   if (x == b) SETERRQ(PetscObjectComm((PetscObject)mat),PETSC_ERR_ARG_IDN,"x and b must be different vectors");
8655   if (!mat->factortype) SETERRQ(PetscObjectComm((PetscObject)mat),PETSC_ERR_ARG_WRONGSTATE,"Unfactored matrix");
8656   if (!mat->rmap->N && !mat->cmap->N) PetscFunctionReturn(0);
8657 
8658   if (!mat->ops->solves) SETERRQ1(PetscObjectComm((PetscObject)mat),PETSC_ERR_SUP,"Mat type %s",((PetscObject)mat)->type_name);
8659   MatCheckPreallocated(mat,1);
8660   ierr = PetscLogEventBegin(MAT_Solves,mat,0,0,0);CHKERRQ(ierr);
8661   ierr = (*mat->ops->solves)(mat,b,x);CHKERRQ(ierr);
8662   ierr = PetscLogEventEnd(MAT_Solves,mat,0,0,0);CHKERRQ(ierr);
8663   PetscFunctionReturn(0);
8664 }
8665 
8666 /*@
8667    MatIsSymmetric - Test whether a matrix is symmetric
8668 
8669    Collective on Mat
8670 
8671    Input Parameter:
8672 +  A - the matrix to test
8673 -  tol - difference between value and its transpose less than this amount counts as equal (use 0.0 for exact transpose)
8674 
8675    Output Parameters:
8676 .  flg - the result
8677 
8678    Notes:
8679     For real numbers MatIsSymmetric() and MatIsHermitian() return identical results
8680 
8681    Level: intermediate
8682 
8683    Concepts: matrix^symmetry
8684 
8685 .seealso: MatTranspose(), MatIsTranspose(), MatIsHermitian(), MatIsStructurallySymmetric(), MatSetOption(), MatIsSymmetricKnown()
8686 @*/
8687 PetscErrorCode MatIsSymmetric(Mat A,PetscReal tol,PetscBool  *flg)
8688 {
8689   PetscErrorCode ierr;
8690 
8691   PetscFunctionBegin;
8692   PetscValidHeaderSpecific(A,MAT_CLASSID,1);
8693   PetscValidPointer(flg,2);
8694 
8695   if (!A->symmetric_set) {
8696     if (!A->ops->issymmetric) {
8697       MatType mattype;
8698       ierr = MatGetType(A,&mattype);CHKERRQ(ierr);
8699       SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_SUP,"Matrix of type <%s> does not support checking for symmetric",mattype);
8700     }
8701     ierr = (*A->ops->issymmetric)(A,tol,flg);CHKERRQ(ierr);
8702     if (!tol) {
8703       A->symmetric_set = PETSC_TRUE;
8704       A->symmetric     = *flg;
8705       if (A->symmetric) {
8706         A->structurally_symmetric_set = PETSC_TRUE;
8707         A->structurally_symmetric     = PETSC_TRUE;
8708       }
8709     }
8710   } else if (A->symmetric) {
8711     *flg = PETSC_TRUE;
8712   } else if (!tol) {
8713     *flg = PETSC_FALSE;
8714   } else {
8715     if (!A->ops->issymmetric) {
8716       MatType mattype;
8717       ierr = MatGetType(A,&mattype);CHKERRQ(ierr);
8718       SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_SUP,"Matrix of type <%s> does not support checking for symmetric",mattype);
8719     }
8720     ierr = (*A->ops->issymmetric)(A,tol,flg);CHKERRQ(ierr);
8721   }
8722   PetscFunctionReturn(0);
8723 }
8724 
8725 /*@
8726    MatIsHermitian - Test whether a matrix is Hermitian
8727 
8728    Collective on Mat
8729 
8730    Input Parameter:
8731 +  A - the matrix to test
8732 -  tol - difference between value and its transpose less than this amount counts as equal (use 0.0 for exact Hermitian)
8733 
8734    Output Parameters:
8735 .  flg - the result
8736 
8737    Level: intermediate
8738 
8739    Concepts: matrix^symmetry
8740 
8741 .seealso: MatTranspose(), MatIsTranspose(), MatIsHermitian(), MatIsStructurallySymmetric(), MatSetOption(),
8742           MatIsSymmetricKnown(), MatIsSymmetric()
8743 @*/
8744 PetscErrorCode MatIsHermitian(Mat A,PetscReal tol,PetscBool  *flg)
8745 {
8746   PetscErrorCode ierr;
8747 
8748   PetscFunctionBegin;
8749   PetscValidHeaderSpecific(A,MAT_CLASSID,1);
8750   PetscValidPointer(flg,2);
8751 
8752   if (!A->hermitian_set) {
8753     if (!A->ops->ishermitian) {
8754       MatType mattype;
8755       ierr = MatGetType(A,&mattype);CHKERRQ(ierr);
8756       SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_SUP,"Matrix of type <%s> does not support checking for hermitian",mattype);
8757     }
8758     ierr = (*A->ops->ishermitian)(A,tol,flg);CHKERRQ(ierr);
8759     if (!tol) {
8760       A->hermitian_set = PETSC_TRUE;
8761       A->hermitian     = *flg;
8762       if (A->hermitian) {
8763         A->structurally_symmetric_set = PETSC_TRUE;
8764         A->structurally_symmetric     = PETSC_TRUE;
8765       }
8766     }
8767   } else if (A->hermitian) {
8768     *flg = PETSC_TRUE;
8769   } else if (!tol) {
8770     *flg = PETSC_FALSE;
8771   } else {
8772     if (!A->ops->ishermitian) {
8773       MatType mattype;
8774       ierr = MatGetType(A,&mattype);CHKERRQ(ierr);
8775       SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_SUP,"Matrix of type <%s> does not support checking for hermitian",mattype);
8776     }
8777     ierr = (*A->ops->ishermitian)(A,tol,flg);CHKERRQ(ierr);
8778   }
8779   PetscFunctionReturn(0);
8780 }
8781 
8782 /*@
8783    MatIsSymmetricKnown - Checks the flag on the matrix to see if it is symmetric.
8784 
8785    Not Collective
8786 
8787    Input Parameter:
8788 .  A - the matrix to check
8789 
8790    Output Parameters:
8791 +  set - if the symmetric flag is set (this tells you if the next flag is valid)
8792 -  flg - the result
8793 
8794    Level: advanced
8795 
8796    Concepts: matrix^symmetry
8797 
8798    Note: Does not check the matrix values directly, so this may return unknown (set = PETSC_FALSE). Use MatIsSymmetric()
8799          if you want it explicitly checked
8800 
8801 .seealso: MatTranspose(), MatIsTranspose(), MatIsHermitian(), MatIsStructurallySymmetric(), MatSetOption(), MatIsSymmetric()
8802 @*/
8803 PetscErrorCode MatIsSymmetricKnown(Mat A,PetscBool  *set,PetscBool  *flg)
8804 {
8805   PetscFunctionBegin;
8806   PetscValidHeaderSpecific(A,MAT_CLASSID,1);
8807   PetscValidPointer(set,2);
8808   PetscValidPointer(flg,3);
8809   if (A->symmetric_set) {
8810     *set = PETSC_TRUE;
8811     *flg = A->symmetric;
8812   } else {
8813     *set = PETSC_FALSE;
8814   }
8815   PetscFunctionReturn(0);
8816 }
8817 
8818 /*@
8819    MatIsHermitianKnown - Checks the flag on the matrix to see if it is hermitian.
8820 
8821    Not Collective
8822 
8823    Input Parameter:
8824 .  A - the matrix to check
8825 
8826    Output Parameters:
8827 +  set - if the hermitian flag is set (this tells you if the next flag is valid)
8828 -  flg - the result
8829 
8830    Level: advanced
8831 
8832    Concepts: matrix^symmetry
8833 
8834    Note: Does not check the matrix values directly, so this may return unknown (set = PETSC_FALSE). Use MatIsHermitian()
8835          if you want it explicitly checked
8836 
8837 .seealso: MatTranspose(), MatIsTranspose(), MatIsHermitian(), MatIsStructurallySymmetric(), MatSetOption(), MatIsSymmetric()
8838 @*/
8839 PetscErrorCode MatIsHermitianKnown(Mat A,PetscBool  *set,PetscBool  *flg)
8840 {
8841   PetscFunctionBegin;
8842   PetscValidHeaderSpecific(A,MAT_CLASSID,1);
8843   PetscValidPointer(set,2);
8844   PetscValidPointer(flg,3);
8845   if (A->hermitian_set) {
8846     *set = PETSC_TRUE;
8847     *flg = A->hermitian;
8848   } else {
8849     *set = PETSC_FALSE;
8850   }
8851   PetscFunctionReturn(0);
8852 }
8853 
8854 /*@
8855    MatIsStructurallySymmetric - Test whether a matrix is structurally symmetric
8856 
8857    Collective on Mat
8858 
8859    Input Parameter:
8860 .  A - the matrix to test
8861 
8862    Output Parameters:
8863 .  flg - the result
8864 
8865    Level: intermediate
8866 
8867    Concepts: matrix^symmetry
8868 
8869 .seealso: MatTranspose(), MatIsTranspose(), MatIsHermitian(), MatIsSymmetric(), MatSetOption()
8870 @*/
8871 PetscErrorCode MatIsStructurallySymmetric(Mat A,PetscBool  *flg)
8872 {
8873   PetscErrorCode ierr;
8874 
8875   PetscFunctionBegin;
8876   PetscValidHeaderSpecific(A,MAT_CLASSID,1);
8877   PetscValidPointer(flg,2);
8878   if (!A->structurally_symmetric_set) {
8879     if (!A->ops->isstructurallysymmetric) SETERRQ(PetscObjectComm((PetscObject)A),PETSC_ERR_SUP,"Matrix does not support checking for structural symmetric");
8880     ierr = (*A->ops->isstructurallysymmetric)(A,&A->structurally_symmetric);CHKERRQ(ierr);
8881 
8882     A->structurally_symmetric_set = PETSC_TRUE;
8883   }
8884   *flg = A->structurally_symmetric;
8885   PetscFunctionReturn(0);
8886 }
8887 
8888 /*@
8889    MatStashGetInfo - Gets how many values are currently in the matrix stash, i.e. need
8890        to be communicated to other processors during the MatAssemblyBegin/End() process
8891 
8892     Not collective
8893 
8894    Input Parameter:
8895 .   vec - the vector
8896 
8897    Output Parameters:
8898 +   nstash   - the size of the stash
8899 .   reallocs - the number of additional mallocs incurred.
8900 .   bnstash   - the size of the block stash
8901 -   breallocs - the number of additional mallocs incurred.in the block stash
8902 
8903    Level: advanced
8904 
8905 .seealso: MatAssemblyBegin(), MatAssemblyEnd(), Mat, MatStashSetInitialSize()
8906 
8907 @*/
8908 PetscErrorCode MatStashGetInfo(Mat mat,PetscInt *nstash,PetscInt *reallocs,PetscInt *bnstash,PetscInt *breallocs)
8909 {
8910   PetscErrorCode ierr;
8911 
8912   PetscFunctionBegin;
8913   ierr = MatStashGetInfo_Private(&mat->stash,nstash,reallocs);CHKERRQ(ierr);
8914   ierr = MatStashGetInfo_Private(&mat->bstash,bnstash,breallocs);CHKERRQ(ierr);
8915   PetscFunctionReturn(0);
8916 }
8917 
8918 /*@C
8919    MatCreateVecs - Get vector(s) compatible with the matrix, i.e. with the same
8920      parallel layout
8921 
8922    Collective on Mat
8923 
8924    Input Parameter:
8925 .  mat - the matrix
8926 
8927    Output Parameter:
8928 +   right - (optional) vector that the matrix can be multiplied against
8929 -   left - (optional) vector that the matrix vector product can be stored in
8930 
8931    Notes:
8932     The blocksize of the returned vectors is determined by the row and column block sizes set with MatSetBlockSizes() or the single blocksize (same for both) set by MatSetBlockSize().
8933 
8934   Notes:
8935     These are new vectors which are not owned by the Mat, they should be destroyed in VecDestroy() when no longer needed
8936 
8937   Level: advanced
8938 
8939 .seealso: MatCreate(), VecDestroy()
8940 @*/
8941 PetscErrorCode MatCreateVecs(Mat mat,Vec *right,Vec *left)
8942 {
8943   PetscErrorCode ierr;
8944 
8945   PetscFunctionBegin;
8946   PetscValidHeaderSpecific(mat,MAT_CLASSID,1);
8947   PetscValidType(mat,1);
8948   if (mat->ops->getvecs) {
8949     ierr = (*mat->ops->getvecs)(mat,right,left);CHKERRQ(ierr);
8950   } else {
8951     PetscInt rbs,cbs;
8952     ierr = MatGetBlockSizes(mat,&rbs,&cbs);CHKERRQ(ierr);
8953     if (right) {
8954       if (mat->cmap->n < 0) SETERRQ(PetscObjectComm((PetscObject)mat),PETSC_ERR_ARG_WRONGSTATE,"PetscLayout for columns not yet setup");
8955       ierr = VecCreate(PetscObjectComm((PetscObject)mat),right);CHKERRQ(ierr);
8956       ierr = VecSetSizes(*right,mat->cmap->n,PETSC_DETERMINE);CHKERRQ(ierr);
8957       ierr = VecSetBlockSize(*right,cbs);CHKERRQ(ierr);
8958       ierr = VecSetType(*right,mat->defaultvectype);CHKERRQ(ierr);
8959       ierr = PetscLayoutReference(mat->cmap,&(*right)->map);CHKERRQ(ierr);
8960     }
8961     if (left) {
8962       if (mat->rmap->n < 0) SETERRQ(PetscObjectComm((PetscObject)mat),PETSC_ERR_ARG_WRONGSTATE,"PetscLayout for rows not yet setup");
8963       ierr = VecCreate(PetscObjectComm((PetscObject)mat),left);CHKERRQ(ierr);
8964       ierr = VecSetSizes(*left,mat->rmap->n,PETSC_DETERMINE);CHKERRQ(ierr);
8965       ierr = VecSetBlockSize(*left,rbs);CHKERRQ(ierr);
8966       ierr = VecSetType(*left,mat->defaultvectype);CHKERRQ(ierr);
8967       ierr = PetscLayoutReference(mat->rmap,&(*left)->map);CHKERRQ(ierr);
8968     }
8969   }
8970   PetscFunctionReturn(0);
8971 }
8972 
8973 /*@C
8974    MatFactorInfoInitialize - Initializes a MatFactorInfo data structure
8975      with default values.
8976 
8977    Not Collective
8978 
8979    Input Parameters:
8980 .    info - the MatFactorInfo data structure
8981 
8982 
8983    Notes:
8984     The solvers are generally used through the KSP and PC objects, for example
8985           PCLU, PCILU, PCCHOLESKY, PCICC
8986 
8987    Level: developer
8988 
8989 .seealso: MatFactorInfo
8990 
8991     Developer Note: fortran interface is not autogenerated as the f90
8992     interface defintion cannot be generated correctly [due to MatFactorInfo]
8993 
8994 @*/
8995 
8996 PetscErrorCode MatFactorInfoInitialize(MatFactorInfo *info)
8997 {
8998   PetscErrorCode ierr;
8999 
9000   PetscFunctionBegin;
9001   ierr = PetscMemzero(info,sizeof(MatFactorInfo));CHKERRQ(ierr);
9002   PetscFunctionReturn(0);
9003 }
9004 
9005 /*@
9006    MatFactorSetSchurIS - Set indices corresponding to the Schur complement you wish to have computed
9007 
9008    Collective on Mat
9009 
9010    Input Parameters:
9011 +  mat - the factored matrix
9012 -  is - the index set defining the Schur indices (0-based)
9013 
9014    Notes:
9015     Call MatFactorSolveSchurComplement() or MatFactorSolveSchurComplementTranspose() after this call to solve a Schur complement system.
9016 
9017    You can call MatFactorGetSchurComplement() or MatFactorCreateSchurComplement() after this call.
9018 
9019    Level: developer
9020 
9021    Concepts:
9022 
9023 .seealso: MatGetFactor(), MatFactorGetSchurComplement(), MatFactorRestoreSchurComplement(), MatFactorCreateSchurComplement(), MatFactorSolveSchurComplement(),
9024           MatFactorSolveSchurComplementTranspose(), MatFactorSolveSchurComplement()
9025 
9026 @*/
9027 PetscErrorCode MatFactorSetSchurIS(Mat mat,IS is)
9028 {
9029   PetscErrorCode ierr,(*f)(Mat,IS);
9030 
9031   PetscFunctionBegin;
9032   PetscValidType(mat,1);
9033   PetscValidHeaderSpecific(mat,MAT_CLASSID,1);
9034   PetscValidType(is,2);
9035   PetscValidHeaderSpecific(is,IS_CLASSID,2);
9036   PetscCheckSameComm(mat,1,is,2);
9037   if (!mat->factortype) SETERRQ(PetscObjectComm((PetscObject)mat),PETSC_ERR_ARG_WRONGSTATE,"Only for factored matrix");
9038   ierr = PetscObjectQueryFunction((PetscObject)mat,"MatFactorSetSchurIS_C",&f);CHKERRQ(ierr);
9039   if (!f) SETERRQ(PetscObjectComm((PetscObject)mat),PETSC_ERR_SUP,"The selected MatSolverType does not support Schur complement computation. You should use MATSOLVERMUMPS or MATSOLVERMKL_PARDISO");
9040   if (mat->schur) {
9041     ierr = MatDestroy(&mat->schur);CHKERRQ(ierr);
9042   }
9043   ierr = (*f)(mat,is);CHKERRQ(ierr);
9044   if (!mat->schur) SETERRQ(PetscObjectComm((PetscObject)mat),PETSC_ERR_PLIB,"Schur complement has not been created");
9045   ierr = MatFactorSetUpInPlaceSchur_Private(mat);CHKERRQ(ierr);
9046   PetscFunctionReturn(0);
9047 }
9048 
9049 /*@
9050   MatFactorCreateSchurComplement - Create a Schur complement matrix object using Schur data computed during the factorization step
9051 
9052    Logically Collective on Mat
9053 
9054    Input Parameters:
9055 +  F - the factored matrix obtained by calling MatGetFactor() from PETSc-MUMPS interface
9056 .  S - location where to return the Schur complement, can be NULL
9057 -  status - the status of the Schur complement matrix, can be NULL
9058 
9059    Notes:
9060    You must call MatFactorSetSchurIS() before calling this routine.
9061 
9062    The routine provides a copy of the Schur matrix stored within the solver data structures.
9063    The caller must destroy the object when it is no longer needed.
9064    If MatFactorInvertSchurComplement() has been called, the routine gets back the inverse.
9065 
9066    Use MatFactorGetSchurComplement() to get access to the Schur complement matrix inside the factored matrix instead of making a copy of it (which this function does)
9067 
9068    Developer Notes:
9069     The reason this routine exists is because the representation of the Schur complement within the factor matrix may be different than a standard PETSc
9070    matrix representation and we normally do not want to use the time or memory to make a copy as a regular PETSc matrix.
9071 
9072    See MatCreateSchurComplement() or MatGetSchurComplement() for ways to create virtual or approximate Schur complements.
9073 
9074    Level: advanced
9075 
9076    References:
9077 
9078 .seealso: MatGetFactor(), MatFactorSetSchurIS(), MatFactorGetSchurComplement(), MatFactorSchurStatus
9079 @*/
9080 PetscErrorCode MatFactorCreateSchurComplement(Mat F,Mat* S,MatFactorSchurStatus* status)
9081 {
9082   PetscErrorCode ierr;
9083 
9084   PetscFunctionBegin;
9085   PetscValidHeaderSpecific(F,MAT_CLASSID,1);
9086   if (S) PetscValidPointer(S,2);
9087   if (status) PetscValidPointer(status,3);
9088   if (S) {
9089     PetscErrorCode (*f)(Mat,Mat*);
9090 
9091     ierr = PetscObjectQueryFunction((PetscObject)F,"MatFactorCreateSchurComplement_C",&f);CHKERRQ(ierr);
9092     if (f) {
9093       ierr = (*f)(F,S);CHKERRQ(ierr);
9094     } else {
9095       ierr = MatDuplicate(F->schur,MAT_COPY_VALUES,S);CHKERRQ(ierr);
9096     }
9097   }
9098   if (status) *status = F->schur_status;
9099   PetscFunctionReturn(0);
9100 }
9101 
9102 /*@
9103   MatFactorGetSchurComplement - Gets access to a Schur complement matrix using the current Schur data within a factored matrix
9104 
9105    Logically Collective on Mat
9106 
9107    Input Parameters:
9108 +  F - the factored matrix obtained by calling MatGetFactor()
9109 .  *S - location where to return the Schur complement, can be NULL
9110 -  status - the status of the Schur complement matrix, can be NULL
9111 
9112    Notes:
9113    You must call MatFactorSetSchurIS() before calling this routine.
9114 
9115    Schur complement mode is currently implemented for sequential matrices.
9116    The routine returns a the Schur Complement stored within the data strutures of the solver.
9117    If MatFactorInvertSchurComplement() has previously been called, the returned matrix is actually the inverse of the Schur complement.
9118    The returned matrix should not be destroyed; the caller should call MatFactorRestoreSchurComplement() when the object is no longer needed.
9119 
9120    Use MatFactorCreateSchurComplement() to create a copy of the Schur complement matrix that is within a factored matrix
9121 
9122    See MatCreateSchurComplement() or MatGetSchurComplement() for ways to create virtual or approximate Schur complements.
9123 
9124    Level: advanced
9125 
9126    References:
9127 
9128 .seealso: MatGetFactor(), MatFactorSetSchurIS(), MatFactorRestoreSchurComplement(), MatFactorCreateSchurComplement(), MatFactorSchurStatus
9129 @*/
9130 PetscErrorCode MatFactorGetSchurComplement(Mat F,Mat* S,MatFactorSchurStatus* status)
9131 {
9132   PetscFunctionBegin;
9133   PetscValidHeaderSpecific(F,MAT_CLASSID,1);
9134   if (S) PetscValidPointer(S,2);
9135   if (status) PetscValidPointer(status,3);
9136   if (S) *S = F->schur;
9137   if (status) *status = F->schur_status;
9138   PetscFunctionReturn(0);
9139 }
9140 
9141 /*@
9142   MatFactorRestoreSchurComplement - Restore the Schur complement matrix object obtained from a call to MatFactorGetSchurComplement
9143 
9144    Logically Collective on Mat
9145 
9146    Input Parameters:
9147 +  F - the factored matrix obtained by calling MatGetFactor()
9148 .  *S - location where the Schur complement is stored
9149 -  status - the status of the Schur complement matrix (see MatFactorSchurStatus)
9150 
9151    Notes:
9152 
9153    Level: advanced
9154 
9155    References:
9156 
9157 .seealso: MatGetFactor(), MatFactorSetSchurIS(), MatFactorRestoreSchurComplement(), MatFactorCreateSchurComplement(), MatFactorSchurStatus
9158 @*/
9159 PetscErrorCode MatFactorRestoreSchurComplement(Mat F,Mat* S,MatFactorSchurStatus status)
9160 {
9161   PetscErrorCode ierr;
9162 
9163   PetscFunctionBegin;
9164   PetscValidHeaderSpecific(F,MAT_CLASSID,1);
9165   if (S) {
9166     PetscValidHeaderSpecific(*S,MAT_CLASSID,2);
9167     *S = NULL;
9168   }
9169   F->schur_status = status;
9170   ierr = MatFactorUpdateSchurStatus_Private(F);CHKERRQ(ierr);
9171   PetscFunctionReturn(0);
9172 }
9173 
9174 /*@
9175   MatFactorSolveSchurComplementTranspose - Solve the transpose of the Schur complement system computed during the factorization step
9176 
9177    Logically Collective on Mat
9178 
9179    Input Parameters:
9180 +  F - the factored matrix obtained by calling MatGetFactor()
9181 .  rhs - location where the right hand side of the Schur complement system is stored
9182 -  sol - location where the solution of the Schur complement system has to be returned
9183 
9184    Notes:
9185    The sizes of the vectors should match the size of the Schur complement
9186 
9187    Must be called after MatFactorSetSchurIS()
9188 
9189    Level: advanced
9190 
9191    References:
9192 
9193 .seealso: MatGetFactor(), MatFactorSetSchurIS(), MatFactorSolveSchurComplement()
9194 @*/
9195 PetscErrorCode MatFactorSolveSchurComplementTranspose(Mat F, Vec rhs, Vec sol)
9196 {
9197   PetscErrorCode ierr;
9198 
9199   PetscFunctionBegin;
9200   PetscValidType(F,1);
9201   PetscValidType(rhs,2);
9202   PetscValidType(sol,3);
9203   PetscValidHeaderSpecific(F,MAT_CLASSID,1);
9204   PetscValidHeaderSpecific(rhs,VEC_CLASSID,2);
9205   PetscValidHeaderSpecific(sol,VEC_CLASSID,3);
9206   PetscCheckSameComm(F,1,rhs,2);
9207   PetscCheckSameComm(F,1,sol,3);
9208   ierr = MatFactorFactorizeSchurComplement(F);CHKERRQ(ierr);
9209   switch (F->schur_status) {
9210   case MAT_FACTOR_SCHUR_FACTORED:
9211     ierr = MatSolveTranspose(F->schur,rhs,sol);CHKERRQ(ierr);
9212     break;
9213   case MAT_FACTOR_SCHUR_INVERTED:
9214     ierr = MatMultTranspose(F->schur,rhs,sol);CHKERRQ(ierr);
9215     break;
9216   default:
9217     SETERRQ1(PetscObjectComm((PetscObject)F),PETSC_ERR_SUP,"Unhandled MatFactorSchurStatus %D",F->schur_status);
9218     break;
9219   }
9220   PetscFunctionReturn(0);
9221 }
9222 
9223 /*@
9224   MatFactorSolveSchurComplement - Solve the Schur complement system computed during the factorization step
9225 
9226    Logically Collective on Mat
9227 
9228    Input Parameters:
9229 +  F - the factored matrix obtained by calling MatGetFactor()
9230 .  rhs - location where the right hand side of the Schur complement system is stored
9231 -  sol - location where the solution of the Schur complement system has to be returned
9232 
9233    Notes:
9234    The sizes of the vectors should match the size of the Schur complement
9235 
9236    Must be called after MatFactorSetSchurIS()
9237 
9238    Level: advanced
9239 
9240    References:
9241 
9242 .seealso: MatGetFactor(), MatFactorSetSchurIS(), MatFactorSolveSchurComplementTranspose()
9243 @*/
9244 PetscErrorCode MatFactorSolveSchurComplement(Mat F, Vec rhs, Vec sol)
9245 {
9246   PetscErrorCode ierr;
9247 
9248   PetscFunctionBegin;
9249   PetscValidType(F,1);
9250   PetscValidType(rhs,2);
9251   PetscValidType(sol,3);
9252   PetscValidHeaderSpecific(F,MAT_CLASSID,1);
9253   PetscValidHeaderSpecific(rhs,VEC_CLASSID,2);
9254   PetscValidHeaderSpecific(sol,VEC_CLASSID,3);
9255   PetscCheckSameComm(F,1,rhs,2);
9256   PetscCheckSameComm(F,1,sol,3);
9257   ierr = MatFactorFactorizeSchurComplement(F);CHKERRQ(ierr);
9258   switch (F->schur_status) {
9259   case MAT_FACTOR_SCHUR_FACTORED:
9260     ierr = MatSolve(F->schur,rhs,sol);CHKERRQ(ierr);
9261     break;
9262   case MAT_FACTOR_SCHUR_INVERTED:
9263     ierr = MatMult(F->schur,rhs,sol);CHKERRQ(ierr);
9264     break;
9265   default:
9266     SETERRQ1(PetscObjectComm((PetscObject)F),PETSC_ERR_SUP,"Unhandled MatFactorSchurStatus %D",F->schur_status);
9267     break;
9268   }
9269   PetscFunctionReturn(0);
9270 }
9271 
9272 /*@
9273   MatFactorInvertSchurComplement - Invert the Schur complement matrix computed during the factorization step
9274 
9275    Logically Collective on Mat
9276 
9277    Input Parameters:
9278 +  F - the factored matrix obtained by calling MatGetFactor()
9279 
9280    Notes:
9281     Must be called after MatFactorSetSchurIS().
9282 
9283    Call MatFactorGetSchurComplement() or  MatFactorCreateSchurComplement() AFTER this call to actually compute the inverse and get access to it.
9284 
9285    Level: advanced
9286 
9287    References:
9288 
9289 .seealso: MatGetFactor(), MatFactorSetSchurIS(), MatFactorGetSchurComplement(), MatFactorCreateSchurComplement()
9290 @*/
9291 PetscErrorCode MatFactorInvertSchurComplement(Mat F)
9292 {
9293   PetscErrorCode ierr;
9294 
9295   PetscFunctionBegin;
9296   PetscValidType(F,1);
9297   PetscValidHeaderSpecific(F,MAT_CLASSID,1);
9298   if (F->schur_status == MAT_FACTOR_SCHUR_INVERTED) PetscFunctionReturn(0);
9299   ierr = MatFactorFactorizeSchurComplement(F);CHKERRQ(ierr);
9300   ierr = MatFactorInvertSchurComplement_Private(F);CHKERRQ(ierr);
9301   F->schur_status = MAT_FACTOR_SCHUR_INVERTED;
9302   PetscFunctionReturn(0);
9303 }
9304 
9305 /*@
9306   MatFactorFactorizeSchurComplement - Factorize the Schur complement matrix computed during the factorization step
9307 
9308    Logically Collective on Mat
9309 
9310    Input Parameters:
9311 +  F - the factored matrix obtained by calling MatGetFactor()
9312 
9313    Notes:
9314     Must be called after MatFactorSetSchurIS().
9315 
9316    Level: advanced
9317 
9318    References:
9319 
9320 .seealso: MatGetFactor(), MatFactorSetSchurIS(), MatFactorInvertSchurComplement()
9321 @*/
9322 PetscErrorCode MatFactorFactorizeSchurComplement(Mat F)
9323 {
9324   PetscErrorCode ierr;
9325 
9326   PetscFunctionBegin;
9327   PetscValidType(F,1);
9328   PetscValidHeaderSpecific(F,MAT_CLASSID,1);
9329   if (F->schur_status == MAT_FACTOR_SCHUR_INVERTED || F->schur_status == MAT_FACTOR_SCHUR_FACTORED) PetscFunctionReturn(0);
9330   ierr = MatFactorFactorizeSchurComplement_Private(F);CHKERRQ(ierr);
9331   F->schur_status = MAT_FACTOR_SCHUR_FACTORED;
9332   PetscFunctionReturn(0);
9333 }
9334 
9335 /*@
9336    MatPtAP - Creates the matrix product C = P^T * A * P
9337 
9338    Neighbor-wise Collective on Mat
9339 
9340    Input Parameters:
9341 +  A - the matrix
9342 .  P - the projection matrix
9343 .  scall - either MAT_INITIAL_MATRIX or MAT_REUSE_MATRIX
9344 -  fill - expected fill as ratio of nnz(C)/(nnz(A) + nnz(P)), use PETSC_DEFAULT if you do not have a good estimate
9345           if the result is a dense matrix this is irrelevent
9346 
9347    Output Parameters:
9348 .  C - the product matrix
9349 
9350    Notes:
9351    C will be created and must be destroyed by the user with MatDestroy().
9352 
9353    This routine is currently only implemented for pairs of sequential dense matrices, AIJ matrices and classes
9354    which inherit from AIJ.
9355 
9356    Level: intermediate
9357 
9358 .seealso: MatPtAPSymbolic(), MatPtAPNumeric(), MatMatMult(), MatRARt()
9359 @*/
9360 PetscErrorCode MatPtAP(Mat A,Mat P,MatReuse scall,PetscReal fill,Mat *C)
9361 {
9362   PetscErrorCode ierr;
9363   PetscErrorCode (*fA)(Mat,Mat,MatReuse,PetscReal,Mat*);
9364   PetscErrorCode (*fP)(Mat,Mat,MatReuse,PetscReal,Mat*);
9365   PetscErrorCode (*ptap)(Mat,Mat,MatReuse,PetscReal,Mat*)=NULL;
9366   PetscBool      sametype;
9367 
9368   PetscFunctionBegin;
9369   PetscValidHeaderSpecific(A,MAT_CLASSID,1);
9370   PetscValidType(A,1);
9371   MatCheckPreallocated(A,1);
9372   if (scall == MAT_INPLACE_MATRIX) SETERRQ(PetscObjectComm((PetscObject)A),PETSC_ERR_SUP,"Inplace product not supported");
9373   if (!A->assembled) SETERRQ(PetscObjectComm((PetscObject)A),PETSC_ERR_ARG_WRONGSTATE,"Not for unassembled matrix");
9374   if (A->factortype) SETERRQ(PetscObjectComm((PetscObject)A),PETSC_ERR_ARG_WRONGSTATE,"Not for factored matrix");
9375   PetscValidHeaderSpecific(P,MAT_CLASSID,2);
9376   PetscValidType(P,2);
9377   MatCheckPreallocated(P,2);
9378   if (!P->assembled) SETERRQ(PetscObjectComm((PetscObject)A),PETSC_ERR_ARG_WRONGSTATE,"Not for unassembled matrix");
9379   if (P->factortype) SETERRQ(PetscObjectComm((PetscObject)A),PETSC_ERR_ARG_WRONGSTATE,"Not for factored matrix");
9380 
9381   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);
9382   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);
9383   if (fill == PETSC_DEFAULT || fill == PETSC_DECIDE) fill = 2.0;
9384   if (fill < 1.0) SETERRQ1(PetscObjectComm((PetscObject)A),PETSC_ERR_ARG_SIZ,"Expected fill=%g must be >= 1.0",(double)fill);
9385 
9386   if (scall == MAT_REUSE_MATRIX) {
9387     PetscValidPointer(*C,5);
9388     PetscValidHeaderSpecific(*C,MAT_CLASSID,5);
9389 
9390     if (!(*C)->ops->ptapnumeric) SETERRQ(PetscObjectComm((PetscObject)C),PETSC_ERR_ARG_WRONGSTATE,"MatPtAPNumeric implementation is missing. You cannot use MAT_REUSE_MATRIX");
9391     ierr = PetscLogEventBegin(MAT_PtAP,A,P,0,0);CHKERRQ(ierr);
9392     ierr = PetscLogEventBegin(MAT_PtAPNumeric,A,P,0,0);CHKERRQ(ierr);
9393     ierr = (*(*C)->ops->ptapnumeric)(A,P,*C);CHKERRQ(ierr);
9394     ierr = PetscLogEventEnd(MAT_PtAPNumeric,A,P,0,0);CHKERRQ(ierr);
9395     ierr = PetscLogEventEnd(MAT_PtAP,A,P,0,0);CHKERRQ(ierr);
9396     PetscFunctionReturn(0);
9397   }
9398 
9399   if (fill == PETSC_DEFAULT || fill == PETSC_DECIDE) fill = 2.0;
9400   if (fill < 1.0) SETERRQ1(PetscObjectComm((PetscObject)A),PETSC_ERR_ARG_SIZ,"Expected fill=%g must be >= 1.0",(double)fill);
9401 
9402   fA = A->ops->ptap;
9403   fP = P->ops->ptap;
9404   ierr = PetscStrcmp(((PetscObject)A)->type_name,((PetscObject)P)->type_name,&sametype);CHKERRQ(ierr);
9405   if (fP == fA && sametype) {
9406     if (!fA) SETERRQ1(PetscObjectComm((PetscObject)A),PETSC_ERR_SUP,"MatPtAP not supported for A of type %s",((PetscObject)A)->type_name);
9407     ptap = fA;
9408   } else {
9409     /* dispatch based on the type of A and P from their PetscObject's PetscFunctionLists. */
9410     char ptapname[256];
9411     ierr = PetscStrncpy(ptapname,"MatPtAP_",sizeof(ptapname));CHKERRQ(ierr);
9412     ierr = PetscStrlcat(ptapname,((PetscObject)A)->type_name,sizeof(ptapname));CHKERRQ(ierr);
9413     ierr = PetscStrlcat(ptapname,"_",sizeof(ptapname));CHKERRQ(ierr);
9414     ierr = PetscStrlcat(ptapname,((PetscObject)P)->type_name,sizeof(ptapname));CHKERRQ(ierr);
9415     ierr = PetscStrlcat(ptapname,"_C",sizeof(ptapname));CHKERRQ(ierr); /* e.g., ptapname = "MatPtAP_seqdense_seqaij_C" */
9416     ierr = PetscObjectQueryFunction((PetscObject)P,ptapname,&ptap);CHKERRQ(ierr);
9417     if (!ptap) SETERRQ3(PetscObjectComm((PetscObject)A),PETSC_ERR_ARG_INCOMP,"MatPtAP requires A, %s, to be compatible with P, %s (Misses composed function %s)",((PetscObject)A)->type_name,((PetscObject)P)->type_name,ptapname);
9418   }
9419 
9420   ierr = PetscLogEventBegin(MAT_PtAP,A,P,0,0);CHKERRQ(ierr);
9421   ierr = (*ptap)(A,P,scall,fill,C);CHKERRQ(ierr);
9422   ierr = PetscLogEventEnd(MAT_PtAP,A,P,0,0);CHKERRQ(ierr);
9423   if (A->symmetric_set && A->symmetric) {
9424     ierr = MatSetOption(*C,MAT_SYMMETRIC,PETSC_TRUE);CHKERRQ(ierr);
9425   }
9426   PetscFunctionReturn(0);
9427 }
9428 
9429 /*@
9430    MatPtAPNumeric - Computes the matrix product C = P^T * A * P
9431 
9432    Neighbor-wise Collective on Mat
9433 
9434    Input Parameters:
9435 +  A - the matrix
9436 -  P - the projection matrix
9437 
9438    Output Parameters:
9439 .  C - the product matrix
9440 
9441    Notes:
9442    C must have been created by calling MatPtAPSymbolic and must be destroyed by
9443    the user using MatDeatroy().
9444 
9445    This routine is currently only implemented for pairs of AIJ matrices and classes
9446    which inherit from AIJ.  C will be of type MATAIJ.
9447 
9448    Level: intermediate
9449 
9450 .seealso: MatPtAP(), MatPtAPSymbolic(), MatMatMultNumeric()
9451 @*/
9452 PetscErrorCode MatPtAPNumeric(Mat A,Mat P,Mat C)
9453 {
9454   PetscErrorCode ierr;
9455 
9456   PetscFunctionBegin;
9457   PetscValidHeaderSpecific(A,MAT_CLASSID,1);
9458   PetscValidType(A,1);
9459   if (!A->assembled) SETERRQ(PetscObjectComm((PetscObject)A),PETSC_ERR_ARG_WRONGSTATE,"Not for unassembled matrix");
9460   if (A->factortype) SETERRQ(PetscObjectComm((PetscObject)A),PETSC_ERR_ARG_WRONGSTATE,"Not for factored matrix");
9461   PetscValidHeaderSpecific(P,MAT_CLASSID,2);
9462   PetscValidType(P,2);
9463   MatCheckPreallocated(P,2);
9464   if (!P->assembled) SETERRQ(PetscObjectComm((PetscObject)A),PETSC_ERR_ARG_WRONGSTATE,"Not for unassembled matrix");
9465   if (P->factortype) SETERRQ(PetscObjectComm((PetscObject)A),PETSC_ERR_ARG_WRONGSTATE,"Not for factored matrix");
9466   PetscValidHeaderSpecific(C,MAT_CLASSID,3);
9467   PetscValidType(C,3);
9468   MatCheckPreallocated(C,3);
9469   if (C->factortype) SETERRQ(PetscObjectComm((PetscObject)A),PETSC_ERR_ARG_WRONGSTATE,"Not for factored matrix");
9470   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);
9471   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);
9472   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);
9473   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);
9474   MatCheckPreallocated(A,1);
9475 
9476   if (!C->ops->ptapnumeric) SETERRQ(PetscObjectComm((PetscObject)C),PETSC_ERR_ARG_WRONGSTATE,"MatPtAPNumeric implementation is missing. You should call MatPtAPSymbolic first");
9477   ierr = PetscLogEventBegin(MAT_PtAPNumeric,A,P,0,0);CHKERRQ(ierr);
9478   ierr = (*C->ops->ptapnumeric)(A,P,C);CHKERRQ(ierr);
9479   ierr = PetscLogEventEnd(MAT_PtAPNumeric,A,P,0,0);CHKERRQ(ierr);
9480   PetscFunctionReturn(0);
9481 }
9482 
9483 /*@
9484    MatPtAPSymbolic - Creates the (i,j) structure of the matrix product C = P^T * A * P
9485 
9486    Neighbor-wise Collective on Mat
9487 
9488    Input Parameters:
9489 +  A - the matrix
9490 -  P - the projection matrix
9491 
9492    Output Parameters:
9493 .  C - the (i,j) structure of the product matrix
9494 
9495    Notes:
9496    C will be created and must be destroyed by the user with MatDestroy().
9497 
9498    This routine is currently only implemented for pairs of SeqAIJ matrices and classes
9499    which inherit from SeqAIJ.  C will be of type MATSEQAIJ.  The product is computed using
9500    this (i,j) structure by calling MatPtAPNumeric().
9501 
9502    Level: intermediate
9503 
9504 .seealso: MatPtAP(), MatPtAPNumeric(), MatMatMultSymbolic()
9505 @*/
9506 PetscErrorCode MatPtAPSymbolic(Mat A,Mat P,PetscReal fill,Mat *C)
9507 {
9508   PetscErrorCode ierr;
9509 
9510   PetscFunctionBegin;
9511   PetscValidHeaderSpecific(A,MAT_CLASSID,1);
9512   PetscValidType(A,1);
9513   if (!A->assembled) SETERRQ(PetscObjectComm((PetscObject)A),PETSC_ERR_ARG_WRONGSTATE,"Not for unassembled matrix");
9514   if (A->factortype) SETERRQ(PetscObjectComm((PetscObject)A),PETSC_ERR_ARG_WRONGSTATE,"Not for factored matrix");
9515   if (fill <1.0) SETERRQ1(PetscObjectComm((PetscObject)A),PETSC_ERR_ARG_SIZ,"Expected fill=%g must be >= 1.0",(double)fill);
9516   PetscValidHeaderSpecific(P,MAT_CLASSID,2);
9517   PetscValidType(P,2);
9518   MatCheckPreallocated(P,2);
9519   if (!P->assembled) SETERRQ(PetscObjectComm((PetscObject)A),PETSC_ERR_ARG_WRONGSTATE,"Not for unassembled matrix");
9520   if (P->factortype) SETERRQ(PetscObjectComm((PetscObject)A),PETSC_ERR_ARG_WRONGSTATE,"Not for factored matrix");
9521   PetscValidPointer(C,3);
9522 
9523   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);
9524   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);
9525   MatCheckPreallocated(A,1);
9526 
9527   if (!A->ops->ptapsymbolic) SETERRQ1(PetscObjectComm((PetscObject)A),PETSC_ERR_SUP,"MatType %s",((PetscObject)A)->type_name);
9528   ierr = PetscLogEventBegin(MAT_PtAPSymbolic,A,P,0,0);CHKERRQ(ierr);
9529   ierr = (*A->ops->ptapsymbolic)(A,P,fill,C);CHKERRQ(ierr);
9530   ierr = PetscLogEventEnd(MAT_PtAPSymbolic,A,P,0,0);CHKERRQ(ierr);
9531 
9532   /* ierr = MatSetBlockSize(*C,A->rmap->bs);CHKERRQ(ierr); NO! this is not always true -ma */
9533   PetscFunctionReturn(0);
9534 }
9535 
9536 /*@
9537    MatRARt - Creates the matrix product C = R * A * R^T
9538 
9539    Neighbor-wise Collective on Mat
9540 
9541    Input Parameters:
9542 +  A - the matrix
9543 .  R - the projection matrix
9544 .  scall - either MAT_INITIAL_MATRIX or MAT_REUSE_MATRIX
9545 -  fill - expected fill as ratio of nnz(C)/nnz(A), use PETSC_DEFAULT if you do not have a good estimate
9546           if the result is a dense matrix this is irrelevent
9547 
9548    Output Parameters:
9549 .  C - the product matrix
9550 
9551    Notes:
9552    C will be created and must be destroyed by the user with MatDestroy().
9553 
9554    This routine is currently only implemented for pairs of AIJ matrices and classes
9555    which inherit from AIJ. Due to PETSc sparse matrix block row distribution among processes,
9556    parallel MatRARt is implemented via explicit transpose of R, which could be very expensive.
9557    We recommend using MatPtAP().
9558 
9559    Level: intermediate
9560 
9561 .seealso: MatRARtSymbolic(), MatRARtNumeric(), MatMatMult(), MatPtAP()
9562 @*/
9563 PetscErrorCode MatRARt(Mat A,Mat R,MatReuse scall,PetscReal fill,Mat *C)
9564 {
9565   PetscErrorCode ierr;
9566 
9567   PetscFunctionBegin;
9568   PetscValidHeaderSpecific(A,MAT_CLASSID,1);
9569   PetscValidType(A,1);
9570   if (scall == MAT_INPLACE_MATRIX) SETERRQ(PetscObjectComm((PetscObject)A),PETSC_ERR_SUP,"Inplace product not supported");
9571   if (!A->assembled) SETERRQ(PetscObjectComm((PetscObject)A),PETSC_ERR_ARG_WRONGSTATE,"Not for unassembled matrix");
9572   if (A->factortype) SETERRQ(PetscObjectComm((PetscObject)A),PETSC_ERR_ARG_WRONGSTATE,"Not for factored matrix");
9573   PetscValidHeaderSpecific(R,MAT_CLASSID,2);
9574   PetscValidType(R,2);
9575   MatCheckPreallocated(R,2);
9576   if (!R->assembled) SETERRQ(PetscObjectComm((PetscObject)A),PETSC_ERR_ARG_WRONGSTATE,"Not for unassembled matrix");
9577   if (R->factortype) SETERRQ(PetscObjectComm((PetscObject)A),PETSC_ERR_ARG_WRONGSTATE,"Not for factored matrix");
9578   PetscValidPointer(C,3);
9579   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);
9580 
9581   if (fill == PETSC_DEFAULT || fill == PETSC_DECIDE) fill = 2.0;
9582   if (fill < 1.0) SETERRQ1(PetscObjectComm((PetscObject)A),PETSC_ERR_ARG_SIZ,"Expected fill=%g must be >= 1.0",(double)fill);
9583   MatCheckPreallocated(A,1);
9584 
9585   if (!A->ops->rart) {
9586     Mat Rt;
9587     ierr = MatTranspose(R,MAT_INITIAL_MATRIX,&Rt);CHKERRQ(ierr);
9588     ierr = MatMatMatMult(R,A,Rt,scall,fill,C);CHKERRQ(ierr);
9589     ierr = MatDestroy(&Rt);CHKERRQ(ierr);
9590     PetscFunctionReturn(0);
9591   }
9592   ierr = PetscLogEventBegin(MAT_RARt,A,R,0,0);CHKERRQ(ierr);
9593   ierr = (*A->ops->rart)(A,R,scall,fill,C);CHKERRQ(ierr);
9594   ierr = PetscLogEventEnd(MAT_RARt,A,R,0,0);CHKERRQ(ierr);
9595   PetscFunctionReturn(0);
9596 }
9597 
9598 /*@
9599    MatRARtNumeric - Computes the matrix product C = R * A * R^T
9600 
9601    Neighbor-wise Collective on Mat
9602 
9603    Input Parameters:
9604 +  A - the matrix
9605 -  R - the projection matrix
9606 
9607    Output Parameters:
9608 .  C - the product matrix
9609 
9610    Notes:
9611    C must have been created by calling MatRARtSymbolic and must be destroyed by
9612    the user using MatDestroy().
9613 
9614    This routine is currently only implemented for pairs of AIJ matrices and classes
9615    which inherit from AIJ.  C will be of type MATAIJ.
9616 
9617    Level: intermediate
9618 
9619 .seealso: MatRARt(), MatRARtSymbolic(), MatMatMultNumeric()
9620 @*/
9621 PetscErrorCode MatRARtNumeric(Mat A,Mat R,Mat C)
9622 {
9623   PetscErrorCode ierr;
9624 
9625   PetscFunctionBegin;
9626   PetscValidHeaderSpecific(A,MAT_CLASSID,1);
9627   PetscValidType(A,1);
9628   if (!A->assembled) SETERRQ(PetscObjectComm((PetscObject)A),PETSC_ERR_ARG_WRONGSTATE,"Not for unassembled matrix");
9629   if (A->factortype) SETERRQ(PetscObjectComm((PetscObject)A),PETSC_ERR_ARG_WRONGSTATE,"Not for factored matrix");
9630   PetscValidHeaderSpecific(R,MAT_CLASSID,2);
9631   PetscValidType(R,2);
9632   MatCheckPreallocated(R,2);
9633   if (!R->assembled) SETERRQ(PetscObjectComm((PetscObject)A),PETSC_ERR_ARG_WRONGSTATE,"Not for unassembled matrix");
9634   if (R->factortype) SETERRQ(PetscObjectComm((PetscObject)A),PETSC_ERR_ARG_WRONGSTATE,"Not for factored matrix");
9635   PetscValidHeaderSpecific(C,MAT_CLASSID,3);
9636   PetscValidType(C,3);
9637   MatCheckPreallocated(C,3);
9638   if (C->factortype) SETERRQ(PetscObjectComm((PetscObject)A),PETSC_ERR_ARG_WRONGSTATE,"Not for factored matrix");
9639   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);
9640   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);
9641   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);
9642   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);
9643   MatCheckPreallocated(A,1);
9644 
9645   ierr = PetscLogEventBegin(MAT_RARtNumeric,A,R,0,0);CHKERRQ(ierr);
9646   ierr = (*A->ops->rartnumeric)(A,R,C);CHKERRQ(ierr);
9647   ierr = PetscLogEventEnd(MAT_RARtNumeric,A,R,0,0);CHKERRQ(ierr);
9648   PetscFunctionReturn(0);
9649 }
9650 
9651 /*@
9652    MatRARtSymbolic - Creates the (i,j) structure of the matrix product C = R * A * R^T
9653 
9654    Neighbor-wise Collective on Mat
9655 
9656    Input Parameters:
9657 +  A - the matrix
9658 -  R - the projection matrix
9659 
9660    Output Parameters:
9661 .  C - the (i,j) structure of the product matrix
9662 
9663    Notes:
9664    C will be created and must be destroyed by the user with MatDestroy().
9665 
9666    This routine is currently only implemented for pairs of SeqAIJ matrices and classes
9667    which inherit from SeqAIJ.  C will be of type MATSEQAIJ.  The product is computed using
9668    this (i,j) structure by calling MatRARtNumeric().
9669 
9670    Level: intermediate
9671 
9672 .seealso: MatRARt(), MatRARtNumeric(), MatMatMultSymbolic()
9673 @*/
9674 PetscErrorCode MatRARtSymbolic(Mat A,Mat R,PetscReal fill,Mat *C)
9675 {
9676   PetscErrorCode ierr;
9677 
9678   PetscFunctionBegin;
9679   PetscValidHeaderSpecific(A,MAT_CLASSID,1);
9680   PetscValidType(A,1);
9681   if (!A->assembled) SETERRQ(PetscObjectComm((PetscObject)A),PETSC_ERR_ARG_WRONGSTATE,"Not for unassembled matrix");
9682   if (A->factortype) SETERRQ(PetscObjectComm((PetscObject)A),PETSC_ERR_ARG_WRONGSTATE,"Not for factored matrix");
9683   if (fill <1.0) SETERRQ1(PetscObjectComm((PetscObject)A),PETSC_ERR_ARG_SIZ,"Expected fill=%g must be >= 1.0",(double)fill);
9684   PetscValidHeaderSpecific(R,MAT_CLASSID,2);
9685   PetscValidType(R,2);
9686   MatCheckPreallocated(R,2);
9687   if (!R->assembled) SETERRQ(PetscObjectComm((PetscObject)A),PETSC_ERR_ARG_WRONGSTATE,"Not for unassembled matrix");
9688   if (R->factortype) SETERRQ(PetscObjectComm((PetscObject)A),PETSC_ERR_ARG_WRONGSTATE,"Not for factored matrix");
9689   PetscValidPointer(C,3);
9690 
9691   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);
9692   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);
9693   MatCheckPreallocated(A,1);
9694   ierr = PetscLogEventBegin(MAT_RARtSymbolic,A,R,0,0);CHKERRQ(ierr);
9695   ierr = (*A->ops->rartsymbolic)(A,R,fill,C);CHKERRQ(ierr);
9696   ierr = PetscLogEventEnd(MAT_RARtSymbolic,A,R,0,0);CHKERRQ(ierr);
9697 
9698   ierr = MatSetBlockSizes(*C,PetscAbs(R->rmap->bs),PetscAbs(R->rmap->bs));CHKERRQ(ierr);
9699   PetscFunctionReturn(0);
9700 }
9701 
9702 /*@
9703    MatMatMult - Performs Matrix-Matrix Multiplication C=A*B.
9704 
9705    Neighbor-wise Collective on Mat
9706 
9707    Input Parameters:
9708 +  A - the left matrix
9709 .  B - the right matrix
9710 .  scall - either MAT_INITIAL_MATRIX or MAT_REUSE_MATRIX
9711 -  fill - expected fill as ratio of nnz(C)/(nnz(A) + nnz(B)), use PETSC_DEFAULT if you do not have a good estimate
9712           if the result is a dense matrix this is irrelevent
9713 
9714    Output Parameters:
9715 .  C - the product matrix
9716 
9717    Notes:
9718    Unless scall is MAT_REUSE_MATRIX C will be created.
9719 
9720    MAT_REUSE_MATRIX can only be used if the matrices A and B have the same nonzero pattern as in the previous call and C was obtained from a previous
9721    call to this function with either MAT_INITIAL_MATRIX or MatMatMultSymbolic()
9722 
9723    To determine the correct fill value, run with -info and search for the string "Fill ratio" to see the value
9724    actually needed.
9725 
9726    If you have many matrices with the same non-zero structure to multiply, you
9727    should either
9728 $   1) use MAT_REUSE_MATRIX in all calls but the first or
9729 $   2) call MatMatMultSymbolic() once and then MatMatMultNumeric() for each product needed
9730    In the special case where matrix B (and hence C) are dense you can create the correctly sized matrix C yourself and then call this routine
9731    with MAT_REUSE_MATRIX, rather than first having MatMatMult() create it for you. You can NEVER do this if the matrix C is sparse.
9732 
9733    Level: intermediate
9734 
9735 .seealso: MatMatMultSymbolic(), MatMatMultNumeric(), MatTransposeMatMult(),  MatMatTransposeMult(), MatPtAP()
9736 @*/
9737 PetscErrorCode MatMatMult(Mat A,Mat B,MatReuse scall,PetscReal fill,Mat *C)
9738 {
9739   PetscErrorCode ierr;
9740   PetscErrorCode (*fA)(Mat,Mat,MatReuse,PetscReal,Mat*);
9741   PetscErrorCode (*fB)(Mat,Mat,MatReuse,PetscReal,Mat*);
9742   PetscErrorCode (*mult)(Mat,Mat,MatReuse,PetscReal,Mat*)=NULL;
9743 
9744   PetscFunctionBegin;
9745   PetscValidHeaderSpecific(A,MAT_CLASSID,1);
9746   PetscValidType(A,1);
9747   MatCheckPreallocated(A,1);
9748   if (!A->assembled) SETERRQ(PetscObjectComm((PetscObject)A),PETSC_ERR_ARG_WRONGSTATE,"Not for unassembled matrix");
9749   if (A->factortype) SETERRQ(PetscObjectComm((PetscObject)A),PETSC_ERR_ARG_WRONGSTATE,"Not for factored matrix");
9750   PetscValidHeaderSpecific(B,MAT_CLASSID,2);
9751   PetscValidType(B,2);
9752   MatCheckPreallocated(B,2);
9753   if (!B->assembled) SETERRQ(PetscObjectComm((PetscObject)A),PETSC_ERR_ARG_WRONGSTATE,"Not for unassembled matrix");
9754   if (B->factortype) SETERRQ(PetscObjectComm((PetscObject)A),PETSC_ERR_ARG_WRONGSTATE,"Not for factored matrix");
9755   PetscValidPointer(C,3);
9756   if (scall == MAT_INPLACE_MATRIX) SETERRQ(PetscObjectComm((PetscObject)A),PETSC_ERR_SUP,"Inplace product not supported");
9757   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);
9758   if (scall == MAT_REUSE_MATRIX) {
9759     PetscValidPointer(*C,5);
9760     PetscValidHeaderSpecific(*C,MAT_CLASSID,5);
9761     ierr = PetscLogEventBegin(MAT_MatMult,A,B,0,0);CHKERRQ(ierr);
9762     ierr = PetscLogEventBegin(MAT_MatMultNumeric,A,B,0,0);CHKERRQ(ierr);
9763     ierr = (*(*C)->ops->matmultnumeric)(A,B,*C);CHKERRQ(ierr);
9764     ierr = PetscLogEventEnd(MAT_MatMultNumeric,A,B,0,0);CHKERRQ(ierr);
9765     ierr = PetscLogEventEnd(MAT_MatMult,A,B,0,0);CHKERRQ(ierr);
9766     PetscFunctionReturn(0);
9767   }
9768   if (fill == PETSC_DEFAULT || fill == PETSC_DECIDE) fill = 2.0;
9769   if (fill < 1.0) SETERRQ1(PetscObjectComm((PetscObject)A),PETSC_ERR_ARG_SIZ,"Expected fill=%g must be >= 1.0",(double)fill);
9770 
9771   fA = A->ops->matmult;
9772   fB = B->ops->matmult;
9773   if (fB == fA) {
9774     if (!fB) SETERRQ1(PetscObjectComm((PetscObject)A),PETSC_ERR_SUP,"MatMatMult not supported for B of type %s",((PetscObject)B)->type_name);
9775     mult = fB;
9776   } else {
9777     /* dispatch based on the type of A and B from their PetscObject's PetscFunctionLists. */
9778     char multname[256];
9779     ierr = PetscStrncpy(multname,"MatMatMult_",sizeof(multname));CHKERRQ(ierr);
9780     ierr = PetscStrlcat(multname,((PetscObject)A)->type_name,sizeof(multname));CHKERRQ(ierr);
9781     ierr = PetscStrlcat(multname,"_",sizeof(multname));CHKERRQ(ierr);
9782     ierr = PetscStrlcat(multname,((PetscObject)B)->type_name,sizeof(multname));CHKERRQ(ierr);
9783     ierr = PetscStrlcat(multname,"_C",sizeof(multname));CHKERRQ(ierr); /* e.g., multname = "MatMatMult_seqdense_seqaij_C" */
9784     ierr = PetscObjectQueryFunction((PetscObject)B,multname,&mult);CHKERRQ(ierr);
9785     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);
9786   }
9787   ierr = PetscLogEventBegin(MAT_MatMult,A,B,0,0);CHKERRQ(ierr);
9788   ierr = (*mult)(A,B,scall,fill,C);CHKERRQ(ierr);
9789   ierr = PetscLogEventEnd(MAT_MatMult,A,B,0,0);CHKERRQ(ierr);
9790   PetscFunctionReturn(0);
9791 }
9792 
9793 /*@
9794    MatMatMultSymbolic - Performs construction, preallocation, and computes the ij structure
9795    of the matrix-matrix product C=A*B.  Call this routine before calling MatMatMultNumeric().
9796 
9797    Neighbor-wise Collective on Mat
9798 
9799    Input Parameters:
9800 +  A - the left matrix
9801 .  B - the right matrix
9802 -  fill - expected fill as ratio of nnz(C)/(nnz(A) + nnz(B)), use PETSC_DEFAULT if you do not have a good estimate,
9803       if C is a dense matrix this is irrelevent
9804 
9805    Output Parameters:
9806 .  C - the product matrix
9807 
9808    Notes:
9809    Unless scall is MAT_REUSE_MATRIX C will be created.
9810 
9811    To determine the correct fill value, run with -info and search for the string "Fill ratio" to see the value
9812    actually needed.
9813 
9814    This routine is currently implemented for
9815     - pairs of AIJ matrices and classes which inherit from AIJ, C will be of type AIJ
9816     - pairs of AIJ (A) and Dense (B) matrix, C will be of type Dense.
9817     - pairs of Dense (A) and AIJ (B) matrix, C will be of type Dense.
9818 
9819    Level: intermediate
9820 
9821    Developers Note: There are ways to estimate the number of nonzeros in the resulting product, see for example, http://arxiv.org/abs/1006.4173
9822      We should incorporate them into PETSc.
9823 
9824 .seealso: MatMatMult(), MatMatMultNumeric()
9825 @*/
9826 PetscErrorCode MatMatMultSymbolic(Mat A,Mat B,PetscReal fill,Mat *C)
9827 {
9828   PetscErrorCode ierr;
9829   PetscErrorCode (*Asymbolic)(Mat,Mat,PetscReal,Mat*);
9830   PetscErrorCode (*Bsymbolic)(Mat,Mat,PetscReal,Mat*);
9831   PetscErrorCode (*symbolic)(Mat,Mat,PetscReal,Mat*)=NULL;
9832 
9833   PetscFunctionBegin;
9834   PetscValidHeaderSpecific(A,MAT_CLASSID,1);
9835   PetscValidType(A,1);
9836   if (!A->assembled) SETERRQ(PetscObjectComm((PetscObject)A),PETSC_ERR_ARG_WRONGSTATE,"Not for unassembled matrix");
9837   if (A->factortype) SETERRQ(PetscObjectComm((PetscObject)A),PETSC_ERR_ARG_WRONGSTATE,"Not for factored matrix");
9838 
9839   PetscValidHeaderSpecific(B,MAT_CLASSID,2);
9840   PetscValidType(B,2);
9841   MatCheckPreallocated(B,2);
9842   if (!B->assembled) SETERRQ(PetscObjectComm((PetscObject)A),PETSC_ERR_ARG_WRONGSTATE,"Not for unassembled matrix");
9843   if (B->factortype) SETERRQ(PetscObjectComm((PetscObject)A),PETSC_ERR_ARG_WRONGSTATE,"Not for factored matrix");
9844   PetscValidPointer(C,3);
9845 
9846   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);
9847   if (fill == PETSC_DEFAULT) fill = 2.0;
9848   if (fill < 1.0) SETERRQ1(PetscObjectComm((PetscObject)A),PETSC_ERR_ARG_SIZ,"Expected fill=%g must be > 1.0",(double)fill);
9849   MatCheckPreallocated(A,1);
9850 
9851   Asymbolic = A->ops->matmultsymbolic;
9852   Bsymbolic = B->ops->matmultsymbolic;
9853   if (Asymbolic == Bsymbolic) {
9854     if (!Bsymbolic) SETERRQ1(PetscObjectComm((PetscObject)A),PETSC_ERR_SUP,"C=A*B not implemented for B of type %s",((PetscObject)B)->type_name);
9855     symbolic = Bsymbolic;
9856   } else { /* dispatch based on the type of A and B */
9857     char symbolicname[256];
9858     ierr = PetscStrncpy(symbolicname,"MatMatMultSymbolic_",sizeof(symbolicname));CHKERRQ(ierr);
9859     ierr = PetscStrlcat(symbolicname,((PetscObject)A)->type_name,sizeof(symbolicname));CHKERRQ(ierr);
9860     ierr = PetscStrlcat(symbolicname,"_",sizeof(symbolicname));CHKERRQ(ierr);
9861     ierr = PetscStrlcat(symbolicname,((PetscObject)B)->type_name,sizeof(symbolicname));CHKERRQ(ierr);
9862     ierr = PetscStrlcat(symbolicname,"_C",sizeof(symbolicname));CHKERRQ(ierr);
9863     ierr = PetscObjectQueryFunction((PetscObject)B,symbolicname,&symbolic);CHKERRQ(ierr);
9864     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);
9865   }
9866   ierr = PetscLogEventBegin(MAT_MatMultSymbolic,A,B,0,0);CHKERRQ(ierr);
9867   ierr = (*symbolic)(A,B,fill,C);CHKERRQ(ierr);
9868   ierr = PetscLogEventEnd(MAT_MatMultSymbolic,A,B,0,0);CHKERRQ(ierr);
9869   PetscFunctionReturn(0);
9870 }
9871 
9872 /*@
9873    MatMatMultNumeric - Performs the numeric matrix-matrix product.
9874    Call this routine after first calling MatMatMultSymbolic().
9875 
9876    Neighbor-wise Collective on Mat
9877 
9878    Input Parameters:
9879 +  A - the left matrix
9880 -  B - the right matrix
9881 
9882    Output Parameters:
9883 .  C - the product matrix, which was created by from MatMatMultSymbolic() or a call to MatMatMult().
9884 
9885    Notes:
9886    C must have been created with MatMatMultSymbolic().
9887 
9888    This routine is currently implemented for
9889     - pairs of AIJ matrices and classes which inherit from AIJ, C will be of type MATAIJ.
9890     - pairs of AIJ (A) and Dense (B) matrix, C will be of type Dense.
9891     - pairs of Dense (A) and AIJ (B) matrix, C will be of type Dense.
9892 
9893    Level: intermediate
9894 
9895 .seealso: MatMatMult(), MatMatMultSymbolic()
9896 @*/
9897 PetscErrorCode MatMatMultNumeric(Mat A,Mat B,Mat C)
9898 {
9899   PetscErrorCode ierr;
9900 
9901   PetscFunctionBegin;
9902   ierr = MatMatMult(A,B,MAT_REUSE_MATRIX,0.0,&C);CHKERRQ(ierr);
9903   PetscFunctionReturn(0);
9904 }
9905 
9906 /*@
9907    MatMatTransposeMult - Performs Matrix-Matrix Multiplication C=A*B^T.
9908 
9909    Neighbor-wise Collective on Mat
9910 
9911    Input Parameters:
9912 +  A - the left matrix
9913 .  B - the right matrix
9914 .  scall - either MAT_INITIAL_MATRIX or MAT_REUSE_MATRIX
9915 -  fill - expected fill as ratio of nnz(C)/(nnz(A) + nnz(B)), use PETSC_DEFAULT if not known
9916 
9917    Output Parameters:
9918 .  C - the product matrix
9919 
9920    Notes:
9921    C will be created if MAT_INITIAL_MATRIX and must be destroyed by the user with MatDestroy().
9922 
9923    MAT_REUSE_MATRIX can only be used if the matrices A and B have the same nonzero pattern as in the previous call
9924 
9925   To determine the correct fill value, run with -info and search for the string "Fill ratio" to see the value
9926    actually needed.
9927 
9928    This routine is currently only implemented for pairs of SeqAIJ matrices, for the SeqDense class,
9929    and for pairs of MPIDense matrices.
9930 
9931    Options Database Keys:
9932 +  -matmattransmult_mpidense_mpidense_via {allgatherv,cyclic} - Choose between algorthims for MPIDense matrices: the
9933                                                                 first redundantly copies the transposed B matrix on each process and requiers O(log P) communication complexity;
9934                                                                 the second never stores more than one portion of the B matrix at a time by requires O(P) communication complexity.
9935 
9936    Level: intermediate
9937 
9938 .seealso: MatMatTransposeMultSymbolic(), MatMatTransposeMultNumeric(), MatMatMult(), MatTransposeMatMult() MatPtAP()
9939 @*/
9940 PetscErrorCode MatMatTransposeMult(Mat A,Mat B,MatReuse scall,PetscReal fill,Mat *C)
9941 {
9942   PetscErrorCode ierr;
9943   PetscErrorCode (*fA)(Mat,Mat,MatReuse,PetscReal,Mat*);
9944   PetscErrorCode (*fB)(Mat,Mat,MatReuse,PetscReal,Mat*);
9945 
9946   PetscFunctionBegin;
9947   PetscValidHeaderSpecific(A,MAT_CLASSID,1);
9948   PetscValidType(A,1);
9949   if (scall == MAT_INPLACE_MATRIX) SETERRQ(PetscObjectComm((PetscObject)A),PETSC_ERR_SUP,"Inplace product not supported");
9950   if (!A->assembled) SETERRQ(PetscObjectComm((PetscObject)A),PETSC_ERR_ARG_WRONGSTATE,"Not for unassembled matrix");
9951   if (A->factortype) SETERRQ(PetscObjectComm((PetscObject)A),PETSC_ERR_ARG_WRONGSTATE,"Not for factored matrix");
9952   PetscValidHeaderSpecific(B,MAT_CLASSID,2);
9953   PetscValidType(B,2);
9954   MatCheckPreallocated(B,2);
9955   if (!B->assembled) SETERRQ(PetscObjectComm((PetscObject)A),PETSC_ERR_ARG_WRONGSTATE,"Not for unassembled matrix");
9956   if (B->factortype) SETERRQ(PetscObjectComm((PetscObject)A),PETSC_ERR_ARG_WRONGSTATE,"Not for factored matrix");
9957   PetscValidPointer(C,3);
9958   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);
9959   if (fill == PETSC_DEFAULT || fill == PETSC_DECIDE) fill = 2.0;
9960   if (fill < 1.0) SETERRQ1(PetscObjectComm((PetscObject)A),PETSC_ERR_ARG_SIZ,"Expected fill=%g must be > 1.0",(double)fill);
9961   MatCheckPreallocated(A,1);
9962 
9963   fA = A->ops->mattransposemult;
9964   if (!fA) SETERRQ1(PetscObjectComm((PetscObject)A),PETSC_ERR_SUP,"MatMatTransposeMult not supported for A of type %s",((PetscObject)A)->type_name);
9965   fB = B->ops->mattransposemult;
9966   if (!fB) SETERRQ1(PetscObjectComm((PetscObject)A),PETSC_ERR_SUP,"MatMatTransposeMult not supported for B of type %s",((PetscObject)B)->type_name);
9967   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);
9968 
9969   ierr = PetscLogEventBegin(MAT_MatTransposeMult,A,B,0,0);CHKERRQ(ierr);
9970   if (scall == MAT_INITIAL_MATRIX) {
9971     ierr = PetscLogEventBegin(MAT_MatTransposeMultSymbolic,A,B,0,0);CHKERRQ(ierr);
9972     ierr = (*A->ops->mattransposemultsymbolic)(A,B,fill,C);CHKERRQ(ierr);
9973     ierr = PetscLogEventEnd(MAT_MatTransposeMultSymbolic,A,B,0,0);CHKERRQ(ierr);
9974   }
9975   ierr = PetscLogEventBegin(MAT_MatTransposeMultNumeric,A,B,0,0);CHKERRQ(ierr);
9976   ierr = (*A->ops->mattransposemultnumeric)(A,B,*C);CHKERRQ(ierr);
9977   ierr = PetscLogEventEnd(MAT_MatTransposeMultNumeric,A,B,0,0);CHKERRQ(ierr);
9978   ierr = PetscLogEventEnd(MAT_MatTransposeMult,A,B,0,0);CHKERRQ(ierr);
9979   PetscFunctionReturn(0);
9980 }
9981 
9982 /*@
9983    MatTransposeMatMult - Performs Matrix-Matrix Multiplication C=A^T*B.
9984 
9985    Neighbor-wise Collective on Mat
9986 
9987    Input Parameters:
9988 +  A - the left matrix
9989 .  B - the right matrix
9990 .  scall - either MAT_INITIAL_MATRIX or MAT_REUSE_MATRIX
9991 -  fill - expected fill as ratio of nnz(C)/(nnz(A) + nnz(B)), use PETSC_DEFAULT if not known
9992 
9993    Output Parameters:
9994 .  C - the product matrix
9995 
9996    Notes:
9997    C will be created if MAT_INITIAL_MATRIX and must be destroyed by the user with MatDestroy().
9998 
9999    MAT_REUSE_MATRIX can only be used if the matrices A and B have the same nonzero pattern as in the previous call
10000 
10001   To determine the correct fill value, run with -info and search for the string "Fill ratio" to see the value
10002    actually needed.
10003 
10004    This routine is currently implemented for pairs of AIJ matrices and pairs of SeqDense matrices and classes
10005    which inherit from SeqAIJ.  C will be of same type as the input matrices.
10006 
10007    Level: intermediate
10008 
10009 .seealso: MatTransposeMatMultSymbolic(), MatTransposeMatMultNumeric(), MatMatMult(), MatMatTransposeMult(), MatPtAP()
10010 @*/
10011 PetscErrorCode MatTransposeMatMult(Mat A,Mat B,MatReuse scall,PetscReal fill,Mat *C)
10012 {
10013   PetscErrorCode ierr;
10014   PetscErrorCode (*fA)(Mat,Mat,MatReuse,PetscReal,Mat*);
10015   PetscErrorCode (*fB)(Mat,Mat,MatReuse,PetscReal,Mat*);
10016   PetscErrorCode (*transposematmult)(Mat,Mat,MatReuse,PetscReal,Mat*) = NULL;
10017 
10018   PetscFunctionBegin;
10019   PetscValidHeaderSpecific(A,MAT_CLASSID,1);
10020   PetscValidType(A,1);
10021   if (scall == MAT_INPLACE_MATRIX) SETERRQ(PetscObjectComm((PetscObject)A),PETSC_ERR_SUP,"Inplace product not supported");
10022   if (!A->assembled) SETERRQ(PetscObjectComm((PetscObject)A),PETSC_ERR_ARG_WRONGSTATE,"Not for unassembled matrix");
10023   if (A->factortype) SETERRQ(PetscObjectComm((PetscObject)A),PETSC_ERR_ARG_WRONGSTATE,"Not for factored matrix");
10024   PetscValidHeaderSpecific(B,MAT_CLASSID,2);
10025   PetscValidType(B,2);
10026   MatCheckPreallocated(B,2);
10027   if (!B->assembled) SETERRQ(PetscObjectComm((PetscObject)A),PETSC_ERR_ARG_WRONGSTATE,"Not for unassembled matrix");
10028   if (B->factortype) SETERRQ(PetscObjectComm((PetscObject)A),PETSC_ERR_ARG_WRONGSTATE,"Not for factored matrix");
10029   PetscValidPointer(C,3);
10030   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);
10031   if (fill == PETSC_DEFAULT || fill == PETSC_DECIDE) fill = 2.0;
10032   if (fill < 1.0) SETERRQ1(PetscObjectComm((PetscObject)A),PETSC_ERR_ARG_SIZ,"Expected fill=%g must be > 1.0",(double)fill);
10033   MatCheckPreallocated(A,1);
10034 
10035   fA = A->ops->transposematmult;
10036   fB = B->ops->transposematmult;
10037   if (fB==fA) {
10038     if (!fA) SETERRQ1(PetscObjectComm((PetscObject)A),PETSC_ERR_SUP,"MatTransposeMatMult not supported for A of type %s",((PetscObject)A)->type_name);
10039     transposematmult = fA;
10040   } else {
10041     /* dispatch based on the type of A and B from their PetscObject's PetscFunctionLists. */
10042     char multname[256];
10043     ierr = PetscStrncpy(multname,"MatTransposeMatMult_",sizeof(multname));CHKERRQ(ierr);
10044     ierr = PetscStrlcat(multname,((PetscObject)A)->type_name,sizeof(multname));CHKERRQ(ierr);
10045     ierr = PetscStrlcat(multname,"_",sizeof(multname));CHKERRQ(ierr);
10046     ierr = PetscStrlcat(multname,((PetscObject)B)->type_name,sizeof(multname));CHKERRQ(ierr);
10047     ierr = PetscStrlcat(multname,"_C",sizeof(multname));CHKERRQ(ierr); /* e.g., multname = "MatMatMult_seqdense_seqaij_C" */
10048     ierr = PetscObjectQueryFunction((PetscObject)B,multname,&transposematmult);CHKERRQ(ierr);
10049     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);
10050   }
10051   ierr = PetscLogEventBegin(MAT_TransposeMatMult,A,B,0,0);CHKERRQ(ierr);
10052   ierr = (*transposematmult)(A,B,scall,fill,C);CHKERRQ(ierr);
10053   ierr = PetscLogEventEnd(MAT_TransposeMatMult,A,B,0,0);CHKERRQ(ierr);
10054   PetscFunctionReturn(0);
10055 }
10056 
10057 /*@
10058    MatMatMatMult - Performs Matrix-Matrix-Matrix Multiplication D=A*B*C.
10059 
10060    Neighbor-wise Collective on Mat
10061 
10062    Input Parameters:
10063 +  A - the left matrix
10064 .  B - the middle matrix
10065 .  C - the right matrix
10066 .  scall - either MAT_INITIAL_MATRIX or MAT_REUSE_MATRIX
10067 -  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
10068           if the result is a dense matrix this is irrelevent
10069 
10070    Output Parameters:
10071 .  D - the product matrix
10072 
10073    Notes:
10074    Unless scall is MAT_REUSE_MATRIX D will be created.
10075 
10076    MAT_REUSE_MATRIX can only be used if the matrices A, B and C have the same nonzero pattern as in the previous call
10077 
10078    To determine the correct fill value, run with -info and search for the string "Fill ratio" to see the value
10079    actually needed.
10080 
10081    If you have many matrices with the same non-zero structure to multiply, you
10082    should use MAT_REUSE_MATRIX in all calls but the first or
10083 
10084    Level: intermediate
10085 
10086 .seealso: MatMatMult, MatPtAP()
10087 @*/
10088 PetscErrorCode MatMatMatMult(Mat A,Mat B,Mat C,MatReuse scall,PetscReal fill,Mat *D)
10089 {
10090   PetscErrorCode ierr;
10091   PetscErrorCode (*fA)(Mat,Mat,Mat,MatReuse,PetscReal,Mat*);
10092   PetscErrorCode (*fB)(Mat,Mat,Mat,MatReuse,PetscReal,Mat*);
10093   PetscErrorCode (*fC)(Mat,Mat,Mat,MatReuse,PetscReal,Mat*);
10094   PetscErrorCode (*mult)(Mat,Mat,Mat,MatReuse,PetscReal,Mat*)=NULL;
10095 
10096   PetscFunctionBegin;
10097   PetscValidHeaderSpecific(A,MAT_CLASSID,1);
10098   PetscValidType(A,1);
10099   MatCheckPreallocated(A,1);
10100   if (scall == MAT_INPLACE_MATRIX) SETERRQ(PetscObjectComm((PetscObject)A),PETSC_ERR_SUP,"Inplace product not supported");
10101   if (!A->assembled) SETERRQ(PetscObjectComm((PetscObject)A),PETSC_ERR_ARG_WRONGSTATE,"Not for unassembled matrix");
10102   if (A->factortype) SETERRQ(PetscObjectComm((PetscObject)A),PETSC_ERR_ARG_WRONGSTATE,"Not for factored matrix");
10103   PetscValidHeaderSpecific(B,MAT_CLASSID,2);
10104   PetscValidType(B,2);
10105   MatCheckPreallocated(B,2);
10106   if (!B->assembled) SETERRQ(PetscObjectComm((PetscObject)B),PETSC_ERR_ARG_WRONGSTATE,"Not for unassembled matrix");
10107   if (B->factortype) SETERRQ(PetscObjectComm((PetscObject)B),PETSC_ERR_ARG_WRONGSTATE,"Not for factored matrix");
10108   PetscValidHeaderSpecific(C,MAT_CLASSID,3);
10109   PetscValidPointer(C,3);
10110   MatCheckPreallocated(C,3);
10111   if (!C->assembled) SETERRQ(PetscObjectComm((PetscObject)C),PETSC_ERR_ARG_WRONGSTATE,"Not for unassembled matrix");
10112   if (C->factortype) SETERRQ(PetscObjectComm((PetscObject)C),PETSC_ERR_ARG_WRONGSTATE,"Not for factored matrix");
10113   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);
10114   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);
10115   if (scall == MAT_REUSE_MATRIX) {
10116     PetscValidPointer(*D,6);
10117     PetscValidHeaderSpecific(*D,MAT_CLASSID,6);
10118     ierr = PetscLogEventBegin(MAT_MatMatMult,A,B,0,0);CHKERRQ(ierr);
10119     ierr = (*(*D)->ops->matmatmult)(A,B,C,scall,fill,D);CHKERRQ(ierr);
10120     ierr = PetscLogEventEnd(MAT_MatMatMult,A,B,0,0);CHKERRQ(ierr);
10121     PetscFunctionReturn(0);
10122   }
10123   if (fill == PETSC_DEFAULT || fill == PETSC_DECIDE) fill = 2.0;
10124   if (fill < 1.0) SETERRQ1(PetscObjectComm((PetscObject)A),PETSC_ERR_ARG_SIZ,"Expected fill=%g must be >= 1.0",(double)fill);
10125 
10126   fA = A->ops->matmatmult;
10127   fB = B->ops->matmatmult;
10128   fC = C->ops->matmatmult;
10129   if (fA == fB && fA == fC) {
10130     if (!fA) SETERRQ1(PetscObjectComm((PetscObject)A),PETSC_ERR_SUP,"MatMatMatMult not supported for A of type %s",((PetscObject)A)->type_name);
10131     mult = fA;
10132   } else {
10133     /* dispatch based on the type of A, B and C from their PetscObject's PetscFunctionLists. */
10134     char multname[256];
10135     ierr = PetscStrncpy(multname,"MatMatMatMult_",sizeof(multname));CHKERRQ(ierr);
10136     ierr = PetscStrlcat(multname,((PetscObject)A)->type_name,sizeof(multname));CHKERRQ(ierr);
10137     ierr = PetscStrlcat(multname,"_",sizeof(multname));CHKERRQ(ierr);
10138     ierr = PetscStrlcat(multname,((PetscObject)B)->type_name,sizeof(multname));CHKERRQ(ierr);
10139     ierr = PetscStrlcat(multname,"_",sizeof(multname));CHKERRQ(ierr);
10140     ierr = PetscStrlcat(multname,((PetscObject)C)->type_name,sizeof(multname));CHKERRQ(ierr);
10141     ierr = PetscStrlcat(multname,"_C",sizeof(multname));CHKERRQ(ierr);
10142     ierr = PetscObjectQueryFunction((PetscObject)B,multname,&mult);CHKERRQ(ierr);
10143     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);
10144   }
10145   ierr = PetscLogEventBegin(MAT_MatMatMult,A,B,0,0);CHKERRQ(ierr);
10146   ierr = (*mult)(A,B,C,scall,fill,D);CHKERRQ(ierr);
10147   ierr = PetscLogEventEnd(MAT_MatMatMult,A,B,0,0);CHKERRQ(ierr);
10148   PetscFunctionReturn(0);
10149 }
10150 
10151 /*@
10152    MatCreateRedundantMatrix - Create redundant matrices and put them into processors of subcommunicators.
10153 
10154    Collective on Mat
10155 
10156    Input Parameters:
10157 +  mat - the matrix
10158 .  nsubcomm - the number of subcommunicators (= number of redundant parallel or sequential matrices)
10159 .  subcomm - MPI communicator split from the communicator where mat resides in (or MPI_COMM_NULL if nsubcomm is used)
10160 -  reuse - either MAT_INITIAL_MATRIX or MAT_REUSE_MATRIX
10161 
10162    Output Parameter:
10163 .  matredundant - redundant matrix
10164 
10165    Notes:
10166    MAT_REUSE_MATRIX can only be used when the nonzero structure of the
10167    original matrix has not changed from that last call to MatCreateRedundantMatrix().
10168 
10169    This routine creates the duplicated matrices in subcommunicators; you should NOT create them before
10170    calling it.
10171 
10172    Level: advanced
10173 
10174    Concepts: subcommunicator
10175    Concepts: duplicate matrix
10176 
10177 .seealso: MatDestroy()
10178 @*/
10179 PetscErrorCode MatCreateRedundantMatrix(Mat mat,PetscInt nsubcomm,MPI_Comm subcomm,MatReuse reuse,Mat *matredundant)
10180 {
10181   PetscErrorCode ierr;
10182   MPI_Comm       comm;
10183   PetscMPIInt    size;
10184   PetscInt       mloc_sub,nloc_sub,rstart,rend,M=mat->rmap->N,N=mat->cmap->N,bs=mat->rmap->bs;
10185   Mat_Redundant  *redund=NULL;
10186   PetscSubcomm   psubcomm=NULL;
10187   MPI_Comm       subcomm_in=subcomm;
10188   Mat            *matseq;
10189   IS             isrow,iscol;
10190   PetscBool      newsubcomm=PETSC_FALSE;
10191 
10192   PetscFunctionBegin;
10193   PetscValidHeaderSpecific(mat,MAT_CLASSID,1);
10194   if (nsubcomm && reuse == MAT_REUSE_MATRIX) {
10195     PetscValidPointer(*matredundant,5);
10196     PetscValidHeaderSpecific(*matredundant,MAT_CLASSID,5);
10197   }
10198 
10199   ierr = MPI_Comm_size(PetscObjectComm((PetscObject)mat),&size);CHKERRQ(ierr);
10200   if (size == 1 || nsubcomm == 1) {
10201     if (reuse == MAT_INITIAL_MATRIX) {
10202       ierr = MatDuplicate(mat,MAT_COPY_VALUES,matredundant);CHKERRQ(ierr);
10203     } else {
10204       if (*matredundant == mat) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONG,"MAT_REUSE_MATRIX means reuse the matrix passed in as the final argument, not the original matrix");
10205       ierr = MatCopy(mat,*matredundant,SAME_NONZERO_PATTERN);CHKERRQ(ierr);
10206     }
10207     PetscFunctionReturn(0);
10208   }
10209 
10210   if (!mat->assembled) SETERRQ(PetscObjectComm((PetscObject)mat),PETSC_ERR_ARG_WRONGSTATE,"Not for unassembled matrix");
10211   if (mat->factortype) SETERRQ(PetscObjectComm((PetscObject)mat),PETSC_ERR_ARG_WRONGSTATE,"Not for factored matrix");
10212   MatCheckPreallocated(mat,1);
10213 
10214   ierr = PetscLogEventBegin(MAT_RedundantMat,mat,0,0,0);CHKERRQ(ierr);
10215   if (subcomm_in == MPI_COMM_NULL && reuse == MAT_INITIAL_MATRIX) { /* get subcomm if user does not provide subcomm */
10216     /* create psubcomm, then get subcomm */
10217     ierr = PetscObjectGetComm((PetscObject)mat,&comm);CHKERRQ(ierr);
10218     ierr = MPI_Comm_size(comm,&size);CHKERRQ(ierr);
10219     if (nsubcomm < 1 || nsubcomm > size) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_ARG_SIZ,"nsubcomm must between 1 and %D",size);
10220 
10221     ierr = PetscSubcommCreate(comm,&psubcomm);CHKERRQ(ierr);
10222     ierr = PetscSubcommSetNumber(psubcomm,nsubcomm);CHKERRQ(ierr);
10223     ierr = PetscSubcommSetType(psubcomm,PETSC_SUBCOMM_CONTIGUOUS);CHKERRQ(ierr);
10224     ierr = PetscSubcommSetFromOptions(psubcomm);CHKERRQ(ierr);
10225     ierr = PetscCommDuplicate(PetscSubcommChild(psubcomm),&subcomm,NULL);CHKERRQ(ierr);
10226     newsubcomm = PETSC_TRUE;
10227     ierr = PetscSubcommDestroy(&psubcomm);CHKERRQ(ierr);
10228   }
10229 
10230   /* get isrow, iscol and a local sequential matrix matseq[0] */
10231   if (reuse == MAT_INITIAL_MATRIX) {
10232     mloc_sub = PETSC_DECIDE;
10233     nloc_sub = PETSC_DECIDE;
10234     if (bs < 1) {
10235       ierr = PetscSplitOwnership(subcomm,&mloc_sub,&M);CHKERRQ(ierr);
10236       ierr = PetscSplitOwnership(subcomm,&nloc_sub,&N);CHKERRQ(ierr);
10237     } else {
10238       ierr = PetscSplitOwnershipBlock(subcomm,bs,&mloc_sub,&M);CHKERRQ(ierr);
10239       ierr = PetscSplitOwnershipBlock(subcomm,bs,&nloc_sub,&N);CHKERRQ(ierr);
10240     }
10241     ierr = MPI_Scan(&mloc_sub,&rend,1,MPIU_INT,MPI_SUM,subcomm);CHKERRQ(ierr);
10242     rstart = rend - mloc_sub;
10243     ierr = ISCreateStride(PETSC_COMM_SELF,mloc_sub,rstart,1,&isrow);CHKERRQ(ierr);
10244     ierr = ISCreateStride(PETSC_COMM_SELF,N,0,1,&iscol);CHKERRQ(ierr);
10245   } else { /* reuse == MAT_REUSE_MATRIX */
10246     if (*matredundant == mat) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONG,"MAT_REUSE_MATRIX means reuse the matrix passed in as the final argument, not the original matrix");
10247     /* retrieve subcomm */
10248     ierr = PetscObjectGetComm((PetscObject)(*matredundant),&subcomm);CHKERRQ(ierr);
10249     redund = (*matredundant)->redundant;
10250     isrow  = redund->isrow;
10251     iscol  = redund->iscol;
10252     matseq = redund->matseq;
10253   }
10254   ierr = MatCreateSubMatrices(mat,1,&isrow,&iscol,reuse,&matseq);CHKERRQ(ierr);
10255 
10256   /* get matredundant over subcomm */
10257   if (reuse == MAT_INITIAL_MATRIX) {
10258     ierr = MatCreateMPIMatConcatenateSeqMat(subcomm,matseq[0],nloc_sub,reuse,matredundant);CHKERRQ(ierr);
10259 
10260     /* create a supporting struct and attach it to C for reuse */
10261     ierr = PetscNewLog(*matredundant,&redund);CHKERRQ(ierr);
10262     (*matredundant)->redundant = redund;
10263     redund->isrow              = isrow;
10264     redund->iscol              = iscol;
10265     redund->matseq             = matseq;
10266     if (newsubcomm) {
10267       redund->subcomm          = subcomm;
10268     } else {
10269       redund->subcomm          = MPI_COMM_NULL;
10270     }
10271   } else {
10272     ierr = MatCreateMPIMatConcatenateSeqMat(subcomm,matseq[0],PETSC_DECIDE,reuse,matredundant);CHKERRQ(ierr);
10273   }
10274   ierr = PetscLogEventEnd(MAT_RedundantMat,mat,0,0,0);CHKERRQ(ierr);
10275   PetscFunctionReturn(0);
10276 }
10277 
10278 /*@C
10279    MatGetMultiProcBlock - Create multiple [bjacobi] 'parallel submatrices' from
10280    a given 'mat' object. Each submatrix can span multiple procs.
10281 
10282    Collective on Mat
10283 
10284    Input Parameters:
10285 +  mat - the matrix
10286 .  subcomm - the subcommunicator obtained by com_split(comm)
10287 -  scall - either MAT_INITIAL_MATRIX or MAT_REUSE_MATRIX
10288 
10289    Output Parameter:
10290 .  subMat - 'parallel submatrices each spans a given subcomm
10291 
10292   Notes:
10293   The submatrix partition across processors is dictated by 'subComm' a
10294   communicator obtained by com_split(comm). The comm_split
10295   is not restriced to be grouped with consecutive original ranks.
10296 
10297   Due the comm_split() usage, the parallel layout of the submatrices
10298   map directly to the layout of the original matrix [wrt the local
10299   row,col partitioning]. So the original 'DiagonalMat' naturally maps
10300   into the 'DiagonalMat' of the subMat, hence it is used directly from
10301   the subMat. However the offDiagMat looses some columns - and this is
10302   reconstructed with MatSetValues()
10303 
10304   Level: advanced
10305 
10306   Concepts: subcommunicator
10307   Concepts: submatrices
10308 
10309 .seealso: MatCreateSubMatrices()
10310 @*/
10311 PetscErrorCode   MatGetMultiProcBlock(Mat mat, MPI_Comm subComm, MatReuse scall,Mat *subMat)
10312 {
10313   PetscErrorCode ierr;
10314   PetscMPIInt    commsize,subCommSize;
10315 
10316   PetscFunctionBegin;
10317   ierr = MPI_Comm_size(PetscObjectComm((PetscObject)mat),&commsize);CHKERRQ(ierr);
10318   ierr = MPI_Comm_size(subComm,&subCommSize);CHKERRQ(ierr);
10319   if (subCommSize > commsize) SETERRQ2(PetscObjectComm((PetscObject)mat),PETSC_ERR_ARG_OUTOFRANGE,"CommSize %D < SubCommZize %D",commsize,subCommSize);
10320 
10321   if (scall == MAT_REUSE_MATRIX && *subMat == mat) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONG,"MAT_REUSE_MATRIX means reuse the matrix passed in as the final argument, not the original matrix");
10322   ierr = PetscLogEventBegin(MAT_GetMultiProcBlock,mat,0,0,0);CHKERRQ(ierr);
10323   ierr = (*mat->ops->getmultiprocblock)(mat,subComm,scall,subMat);CHKERRQ(ierr);
10324   ierr = PetscLogEventEnd(MAT_GetMultiProcBlock,mat,0,0,0);CHKERRQ(ierr);
10325   PetscFunctionReturn(0);
10326 }
10327 
10328 /*@
10329    MatGetLocalSubMatrix - Gets a reference to a submatrix specified in local numbering
10330 
10331    Not Collective
10332 
10333    Input Arguments:
10334    mat - matrix to extract local submatrix from
10335    isrow - local row indices for submatrix
10336    iscol - local column indices for submatrix
10337 
10338    Output Arguments:
10339    submat - the submatrix
10340 
10341    Level: intermediate
10342 
10343    Notes:
10344    The submat should be returned with MatRestoreLocalSubMatrix().
10345 
10346    Depending on the format of mat, the returned submat may not implement MatMult().  Its communicator may be
10347    the same as mat, it may be PETSC_COMM_SELF, or some other subcomm of mat's.
10348 
10349    The submat always implements MatSetValuesLocal().  If isrow and iscol have the same block size, then
10350    MatSetValuesBlockedLocal() will also be implemented.
10351 
10352    The mat must have had a ISLocalToGlobalMapping provided to it with MatSetLocalToGlobalMapping(). Note that
10353    matrices obtained with DMCreateMatrix() generally already have the local to global mapping provided.
10354 
10355 .seealso: MatRestoreLocalSubMatrix(), MatCreateLocalRef(), MatSetLocalToGlobalMapping()
10356 @*/
10357 PetscErrorCode MatGetLocalSubMatrix(Mat mat,IS isrow,IS iscol,Mat *submat)
10358 {
10359   PetscErrorCode ierr;
10360 
10361   PetscFunctionBegin;
10362   PetscValidHeaderSpecific(mat,MAT_CLASSID,1);
10363   PetscValidHeaderSpecific(isrow,IS_CLASSID,2);
10364   PetscValidHeaderSpecific(iscol,IS_CLASSID,3);
10365   PetscCheckSameComm(isrow,2,iscol,3);
10366   PetscValidPointer(submat,4);
10367   if (!mat->rmap->mapping) SETERRQ(PetscObjectComm((PetscObject)mat),PETSC_ERR_ARG_WRONGSTATE,"Matrix must have local to global mapping provided before this call");
10368 
10369   if (mat->ops->getlocalsubmatrix) {
10370     ierr = (*mat->ops->getlocalsubmatrix)(mat,isrow,iscol,submat);CHKERRQ(ierr);
10371   } else {
10372     ierr = MatCreateLocalRef(mat,isrow,iscol,submat);CHKERRQ(ierr);
10373   }
10374   PetscFunctionReturn(0);
10375 }
10376 
10377 /*@
10378    MatRestoreLocalSubMatrix - Restores a reference to a submatrix specified in local numbering
10379 
10380    Not Collective
10381 
10382    Input Arguments:
10383    mat - matrix to extract local submatrix from
10384    isrow - local row indices for submatrix
10385    iscol - local column indices for submatrix
10386    submat - the submatrix
10387 
10388    Level: intermediate
10389 
10390 .seealso: MatGetLocalSubMatrix()
10391 @*/
10392 PetscErrorCode MatRestoreLocalSubMatrix(Mat mat,IS isrow,IS iscol,Mat *submat)
10393 {
10394   PetscErrorCode ierr;
10395 
10396   PetscFunctionBegin;
10397   PetscValidHeaderSpecific(mat,MAT_CLASSID,1);
10398   PetscValidHeaderSpecific(isrow,IS_CLASSID,2);
10399   PetscValidHeaderSpecific(iscol,IS_CLASSID,3);
10400   PetscCheckSameComm(isrow,2,iscol,3);
10401   PetscValidPointer(submat,4);
10402   if (*submat) {
10403     PetscValidHeaderSpecific(*submat,MAT_CLASSID,4);
10404   }
10405 
10406   if (mat->ops->restorelocalsubmatrix) {
10407     ierr = (*mat->ops->restorelocalsubmatrix)(mat,isrow,iscol,submat);CHKERRQ(ierr);
10408   } else {
10409     ierr = MatDestroy(submat);CHKERRQ(ierr);
10410   }
10411   *submat = NULL;
10412   PetscFunctionReturn(0);
10413 }
10414 
10415 /* --------------------------------------------------------*/
10416 /*@
10417    MatFindZeroDiagonals - Finds all the rows of a matrix that have zero or no diagonal entry in the matrix
10418 
10419    Collective on Mat
10420 
10421    Input Parameter:
10422 .  mat - the matrix
10423 
10424    Output Parameter:
10425 .  is - if any rows have zero diagonals this contains the list of them
10426 
10427    Level: developer
10428 
10429    Concepts: matrix-vector product
10430 
10431 .seealso: MatMultTranspose(), MatMultAdd(), MatMultTransposeAdd()
10432 @*/
10433 PetscErrorCode MatFindZeroDiagonals(Mat mat,IS *is)
10434 {
10435   PetscErrorCode ierr;
10436 
10437   PetscFunctionBegin;
10438   PetscValidHeaderSpecific(mat,MAT_CLASSID,1);
10439   PetscValidType(mat,1);
10440   if (!mat->assembled) SETERRQ(PetscObjectComm((PetscObject)mat),PETSC_ERR_ARG_WRONGSTATE,"Not for unassembled matrix");
10441   if (mat->factortype) SETERRQ(PetscObjectComm((PetscObject)mat),PETSC_ERR_ARG_WRONGSTATE,"Not for factored matrix");
10442 
10443   if (!mat->ops->findzerodiagonals) {
10444     Vec                diag;
10445     const PetscScalar *a;
10446     PetscInt          *rows;
10447     PetscInt           rStart, rEnd, r, nrow = 0;
10448 
10449     ierr = MatCreateVecs(mat, &diag, NULL);CHKERRQ(ierr);
10450     ierr = MatGetDiagonal(mat, diag);CHKERRQ(ierr);
10451     ierr = MatGetOwnershipRange(mat, &rStart, &rEnd);CHKERRQ(ierr);
10452     ierr = VecGetArrayRead(diag, &a);CHKERRQ(ierr);
10453     for (r = 0; r < rEnd-rStart; ++r) if (a[r] == 0.0) ++nrow;
10454     ierr = PetscMalloc1(nrow, &rows);CHKERRQ(ierr);
10455     nrow = 0;
10456     for (r = 0; r < rEnd-rStart; ++r) if (a[r] == 0.0) rows[nrow++] = r+rStart;
10457     ierr = VecRestoreArrayRead(diag, &a);CHKERRQ(ierr);
10458     ierr = VecDestroy(&diag);CHKERRQ(ierr);
10459     ierr = ISCreateGeneral(PetscObjectComm((PetscObject) mat), nrow, rows, PETSC_OWN_POINTER, is);CHKERRQ(ierr);
10460   } else {
10461     ierr = (*mat->ops->findzerodiagonals)(mat, is);CHKERRQ(ierr);
10462   }
10463   PetscFunctionReturn(0);
10464 }
10465 
10466 /*@
10467    MatFindOffBlockDiagonalEntries - Finds all the rows of a matrix that have entries outside of the main diagonal block (defined by the matrix block size)
10468 
10469    Collective on Mat
10470 
10471    Input Parameter:
10472 .  mat - the matrix
10473 
10474    Output Parameter:
10475 .  is - contains the list of rows with off block diagonal entries
10476 
10477    Level: developer
10478 
10479    Concepts: matrix-vector product
10480 
10481 .seealso: MatMultTranspose(), MatMultAdd(), MatMultTransposeAdd()
10482 @*/
10483 PetscErrorCode MatFindOffBlockDiagonalEntries(Mat mat,IS *is)
10484 {
10485   PetscErrorCode ierr;
10486 
10487   PetscFunctionBegin;
10488   PetscValidHeaderSpecific(mat,MAT_CLASSID,1);
10489   PetscValidType(mat,1);
10490   if (!mat->assembled) SETERRQ(PetscObjectComm((PetscObject)mat),PETSC_ERR_ARG_WRONGSTATE,"Not for unassembled matrix");
10491   if (mat->factortype) SETERRQ(PetscObjectComm((PetscObject)mat),PETSC_ERR_ARG_WRONGSTATE,"Not for factored matrix");
10492 
10493   if (!mat->ops->findoffblockdiagonalentries) SETERRQ(PetscObjectComm((PetscObject)mat),PETSC_ERR_SUP,"This matrix type does not have a find off block diagonal entries defined");
10494   ierr = (*mat->ops->findoffblockdiagonalentries)(mat,is);CHKERRQ(ierr);
10495   PetscFunctionReturn(0);
10496 }
10497 
10498 /*@C
10499   MatInvertBlockDiagonal - Inverts the block diagonal entries.
10500 
10501   Collective on Mat
10502 
10503   Input Parameters:
10504 . mat - the matrix
10505 
10506   Output Parameters:
10507 . values - the block inverses in column major order (FORTRAN-like)
10508 
10509    Note:
10510    This routine is not available from Fortran.
10511 
10512   Level: advanced
10513 
10514 .seealso: MatInvertBockDiagonalMat
10515 @*/
10516 PetscErrorCode MatInvertBlockDiagonal(Mat mat,const PetscScalar **values)
10517 {
10518   PetscErrorCode ierr;
10519 
10520   PetscFunctionBegin;
10521   PetscValidHeaderSpecific(mat,MAT_CLASSID,1);
10522   if (!mat->assembled) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONGSTATE,"Not for unassembled matrix");
10523   if (mat->factortype) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONGSTATE,"Not for factored matrix");
10524   if (!mat->ops->invertblockdiagonal) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"Not supported");
10525   ierr = (*mat->ops->invertblockdiagonal)(mat,values);CHKERRQ(ierr);
10526   PetscFunctionReturn(0);
10527 }
10528 
10529 /*@C
10530   MatInvertVariableBlockDiagonal - Inverts the block diagonal entries.
10531 
10532   Collective on Mat
10533 
10534   Input Parameters:
10535 + mat - the matrix
10536 . nblocks - the number of blocks
10537 - bsizes - the size of each block
10538 
10539   Output Parameters:
10540 . values - the block inverses in column major order (FORTRAN-like)
10541 
10542    Note:
10543    This routine is not available from Fortran.
10544 
10545   Level: advanced
10546 
10547 .seealso: MatInvertBockDiagonal()
10548 @*/
10549 PetscErrorCode MatInvertVariableBlockDiagonal(Mat mat,PetscInt nblocks,const PetscInt *bsizes,PetscScalar *values)
10550 {
10551   PetscErrorCode ierr;
10552 
10553   PetscFunctionBegin;
10554   PetscValidHeaderSpecific(mat,MAT_CLASSID,1);
10555   if (!mat->assembled) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONGSTATE,"Not for unassembled matrix");
10556   if (mat->factortype) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONGSTATE,"Not for factored matrix");
10557   if (!mat->ops->invertvariableblockdiagonal) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"Not supported");
10558   ierr = (*mat->ops->invertvariableblockdiagonal)(mat,nblocks,bsizes,values);CHKERRQ(ierr);
10559   PetscFunctionReturn(0);
10560 }
10561 
10562 /*@
10563   MatInvertBlockDiagonalMat - set matrix C to be the inverted block diagonal of matrix A
10564 
10565   Collective on Mat
10566 
10567   Input Parameters:
10568 . A - the matrix
10569 
10570   Output Parameters:
10571 . C - matrix with inverted block diagonal of A.  This matrix should be created and may have its type set.
10572 
10573   Notes: the blocksize of the matrix is used to determine the blocks on the diagonal of C
10574 
10575   Level: advanced
10576 
10577 .seealso: MatInvertBockDiagonal()
10578 @*/
10579 PetscErrorCode MatInvertBlockDiagonalMat(Mat A,Mat C)
10580 {
10581   PetscErrorCode     ierr;
10582   const PetscScalar *vals;
10583   PetscInt          *dnnz;
10584   PetscInt           M,N,m,n,rstart,rend,bs,i,j;
10585 
10586   PetscFunctionBegin;
10587   ierr = MatInvertBlockDiagonal(A,&vals);CHKERRQ(ierr);
10588   ierr = MatGetBlockSize(A,&bs);CHKERRQ(ierr);
10589   ierr = MatGetSize(A,&M,&N);CHKERRQ(ierr);
10590   ierr = MatGetLocalSize(A,&m,&n);CHKERRQ(ierr);
10591   ierr = MatSetSizes(C,m,n,M,N);CHKERRQ(ierr);
10592   ierr = MatSetBlockSize(C,bs);CHKERRQ(ierr);
10593   ierr = PetscMalloc1(m/bs,&dnnz);CHKERRQ(ierr);
10594   for (j = 0; j < m/bs; j++) dnnz[j] = 1;
10595   ierr = MatXAIJSetPreallocation(C,bs,dnnz,NULL,NULL,NULL);CHKERRQ(ierr);
10596   ierr = PetscFree(dnnz);CHKERRQ(ierr);
10597   ierr = MatGetOwnershipRange(C,&rstart,&rend);CHKERRQ(ierr);
10598   ierr = MatSetOption(C,MAT_ROW_ORIENTED,PETSC_FALSE);CHKERRQ(ierr);
10599   for (i = rstart/bs; i < rend/bs; i++) {
10600     ierr = MatSetValuesBlocked(C,1,&i,1,&i,&vals[(i-rstart/bs)*bs*bs],INSERT_VALUES);CHKERRQ(ierr);
10601   }
10602   ierr = MatAssemblyBegin(C,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
10603   ierr = MatAssemblyEnd(C,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
10604   ierr = MatSetOption(C,MAT_ROW_ORIENTED,PETSC_TRUE);CHKERRQ(ierr);
10605   PetscFunctionReturn(0);
10606 }
10607 
10608 /*@C
10609     MatTransposeColoringDestroy - Destroys a coloring context for matrix product C=A*B^T that was created
10610     via MatTransposeColoringCreate().
10611 
10612     Collective on MatTransposeColoring
10613 
10614     Input Parameter:
10615 .   c - coloring context
10616 
10617     Level: intermediate
10618 
10619 .seealso: MatTransposeColoringCreate()
10620 @*/
10621 PetscErrorCode MatTransposeColoringDestroy(MatTransposeColoring *c)
10622 {
10623   PetscErrorCode       ierr;
10624   MatTransposeColoring matcolor=*c;
10625 
10626   PetscFunctionBegin;
10627   if (!matcolor) PetscFunctionReturn(0);
10628   if (--((PetscObject)matcolor)->refct > 0) {matcolor = 0; PetscFunctionReturn(0);}
10629 
10630   ierr = PetscFree3(matcolor->ncolumns,matcolor->nrows,matcolor->colorforrow);CHKERRQ(ierr);
10631   ierr = PetscFree(matcolor->rows);CHKERRQ(ierr);
10632   ierr = PetscFree(matcolor->den2sp);CHKERRQ(ierr);
10633   ierr = PetscFree(matcolor->colorforcol);CHKERRQ(ierr);
10634   ierr = PetscFree(matcolor->columns);CHKERRQ(ierr);
10635   if (matcolor->brows>0) {
10636     ierr = PetscFree(matcolor->lstart);CHKERRQ(ierr);
10637   }
10638   ierr = PetscHeaderDestroy(c);CHKERRQ(ierr);
10639   PetscFunctionReturn(0);
10640 }
10641 
10642 /*@C
10643     MatTransColoringApplySpToDen - Given a symbolic matrix product C=A*B^T for which
10644     a MatTransposeColoring context has been created, computes a dense B^T by Apply
10645     MatTransposeColoring to sparse B.
10646 
10647     Collective on MatTransposeColoring
10648 
10649     Input Parameters:
10650 +   B - sparse matrix B
10651 .   Btdense - symbolic dense matrix B^T
10652 -   coloring - coloring context created with MatTransposeColoringCreate()
10653 
10654     Output Parameter:
10655 .   Btdense - dense matrix B^T
10656 
10657     Level: advanced
10658 
10659      Notes:
10660     These are used internally for some implementations of MatRARt()
10661 
10662 .seealso: MatTransposeColoringCreate(), MatTransposeColoringDestroy(), MatTransColoringApplyDenToSp()
10663 
10664 .keywords: coloring
10665 @*/
10666 PetscErrorCode MatTransColoringApplySpToDen(MatTransposeColoring coloring,Mat B,Mat Btdense)
10667 {
10668   PetscErrorCode ierr;
10669 
10670   PetscFunctionBegin;
10671   PetscValidHeaderSpecific(B,MAT_CLASSID,1);
10672   PetscValidHeaderSpecific(Btdense,MAT_CLASSID,2);
10673   PetscValidHeaderSpecific(coloring,MAT_TRANSPOSECOLORING_CLASSID,3);
10674 
10675   if (!B->ops->transcoloringapplysptoden) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_SUP,"Not supported for this matrix type %s",((PetscObject)B)->type_name);
10676   ierr = (B->ops->transcoloringapplysptoden)(coloring,B,Btdense);CHKERRQ(ierr);
10677   PetscFunctionReturn(0);
10678 }
10679 
10680 /*@C
10681     MatTransColoringApplyDenToSp - Given a symbolic matrix product Csp=A*B^T for which
10682     a MatTransposeColoring context has been created and a dense matrix Cden=A*Btdense
10683     in which Btdens is obtained from MatTransColoringApplySpToDen(), recover sparse matrix
10684     Csp from Cden.
10685 
10686     Collective on MatTransposeColoring
10687 
10688     Input Parameters:
10689 +   coloring - coloring context created with MatTransposeColoringCreate()
10690 -   Cden - matrix product of a sparse matrix and a dense matrix Btdense
10691 
10692     Output Parameter:
10693 .   Csp - sparse matrix
10694 
10695     Level: advanced
10696 
10697      Notes:
10698     These are used internally for some implementations of MatRARt()
10699 
10700 .seealso: MatTransposeColoringCreate(), MatTransposeColoringDestroy(), MatTransColoringApplySpToDen()
10701 
10702 .keywords: coloring
10703 @*/
10704 PetscErrorCode MatTransColoringApplyDenToSp(MatTransposeColoring matcoloring,Mat Cden,Mat Csp)
10705 {
10706   PetscErrorCode ierr;
10707 
10708   PetscFunctionBegin;
10709   PetscValidHeaderSpecific(matcoloring,MAT_TRANSPOSECOLORING_CLASSID,1);
10710   PetscValidHeaderSpecific(Cden,MAT_CLASSID,2);
10711   PetscValidHeaderSpecific(Csp,MAT_CLASSID,3);
10712 
10713   if (!Csp->ops->transcoloringapplydentosp) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_SUP,"Not supported for this matrix type %s",((PetscObject)Csp)->type_name);
10714   ierr = (Csp->ops->transcoloringapplydentosp)(matcoloring,Cden,Csp);CHKERRQ(ierr);
10715   PetscFunctionReturn(0);
10716 }
10717 
10718 /*@C
10719    MatTransposeColoringCreate - Creates a matrix coloring context for matrix product C=A*B^T.
10720 
10721    Collective on Mat
10722 
10723    Input Parameters:
10724 +  mat - the matrix product C
10725 -  iscoloring - the coloring of the matrix; usually obtained with MatColoringCreate() or DMCreateColoring()
10726 
10727     Output Parameter:
10728 .   color - the new coloring context
10729 
10730     Level: intermediate
10731 
10732 .seealso: MatTransposeColoringDestroy(),  MatTransColoringApplySpToDen(),
10733            MatTransColoringApplyDenToSp()
10734 @*/
10735 PetscErrorCode MatTransposeColoringCreate(Mat mat,ISColoring iscoloring,MatTransposeColoring *color)
10736 {
10737   MatTransposeColoring c;
10738   MPI_Comm             comm;
10739   PetscErrorCode       ierr;
10740 
10741   PetscFunctionBegin;
10742   ierr = PetscLogEventBegin(MAT_TransposeColoringCreate,mat,0,0,0);CHKERRQ(ierr);
10743   ierr = PetscObjectGetComm((PetscObject)mat,&comm);CHKERRQ(ierr);
10744   ierr = PetscHeaderCreate(c,MAT_TRANSPOSECOLORING_CLASSID,"MatTransposeColoring","Matrix product C=A*B^T via coloring","Mat",comm,MatTransposeColoringDestroy,NULL);CHKERRQ(ierr);
10745 
10746   c->ctype = iscoloring->ctype;
10747   if (mat->ops->transposecoloringcreate) {
10748     ierr = (*mat->ops->transposecoloringcreate)(mat,iscoloring,c);CHKERRQ(ierr);
10749   } else SETERRQ(PetscObjectComm((PetscObject)mat),PETSC_ERR_SUP,"Code not yet written for this matrix type");
10750 
10751   *color = c;
10752   ierr   = PetscLogEventEnd(MAT_TransposeColoringCreate,mat,0,0,0);CHKERRQ(ierr);
10753   PetscFunctionReturn(0);
10754 }
10755 
10756 /*@
10757       MatGetNonzeroState - Returns a 64 bit integer representing the current state of nonzeros in the matrix. If the
10758         matrix has had no new nonzero locations added to the matrix since the previous call then the value will be the
10759         same, otherwise it will be larger
10760 
10761      Not Collective
10762 
10763   Input Parameter:
10764 .    A  - the matrix
10765 
10766   Output Parameter:
10767 .    state - the current state
10768 
10769   Notes:
10770     You can only compare states from two different calls to the SAME matrix, you cannot compare calls between
10771          different matrices
10772 
10773   Level: intermediate
10774 
10775 @*/
10776 PetscErrorCode MatGetNonzeroState(Mat mat,PetscObjectState *state)
10777 {
10778   PetscFunctionBegin;
10779   PetscValidHeaderSpecific(mat,MAT_CLASSID,1);
10780   *state = mat->nonzerostate;
10781   PetscFunctionReturn(0);
10782 }
10783 
10784 /*@
10785       MatCreateMPIMatConcatenateSeqMat - Creates a single large PETSc matrix by concatenating sequential
10786                  matrices from each processor
10787 
10788     Collective on MPI_Comm
10789 
10790    Input Parameters:
10791 +    comm - the communicators the parallel matrix will live on
10792 .    seqmat - the input sequential matrices
10793 .    n - number of local columns (or PETSC_DECIDE)
10794 -    reuse - either MAT_INITIAL_MATRIX or MAT_REUSE_MATRIX
10795 
10796    Output Parameter:
10797 .    mpimat - the parallel matrix generated
10798 
10799     Level: advanced
10800 
10801    Notes:
10802     The number of columns of the matrix in EACH processor MUST be the same.
10803 
10804 @*/
10805 PetscErrorCode MatCreateMPIMatConcatenateSeqMat(MPI_Comm comm,Mat seqmat,PetscInt n,MatReuse reuse,Mat *mpimat)
10806 {
10807   PetscErrorCode ierr;
10808 
10809   PetscFunctionBegin;
10810   if (!seqmat->ops->creatempimatconcatenateseqmat) SETERRQ1(PetscObjectComm((PetscObject)seqmat),PETSC_ERR_SUP,"Mat type %s",((PetscObject)seqmat)->type_name);
10811   if (reuse == MAT_REUSE_MATRIX && seqmat == *mpimat) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONG,"MAT_REUSE_MATRIX means reuse the matrix passed in as the final argument, not the original matrix");
10812 
10813   ierr = PetscLogEventBegin(MAT_Merge,seqmat,0,0,0);CHKERRQ(ierr);
10814   ierr = (*seqmat->ops->creatempimatconcatenateseqmat)(comm,seqmat,n,reuse,mpimat);CHKERRQ(ierr);
10815   ierr = PetscLogEventEnd(MAT_Merge,seqmat,0,0,0);CHKERRQ(ierr);
10816   PetscFunctionReturn(0);
10817 }
10818 
10819 /*@
10820      MatSubdomainsCreateCoalesce - Creates index subdomains by coalescing adjacent
10821                  ranks' ownership ranges.
10822 
10823     Collective on A
10824 
10825    Input Parameters:
10826 +    A   - the matrix to create subdomains from
10827 -    N   - requested number of subdomains
10828 
10829 
10830    Output Parameters:
10831 +    n   - number of subdomains resulting on this rank
10832 -    iss - IS list with indices of subdomains on this rank
10833 
10834     Level: advanced
10835 
10836     Notes:
10837     number of subdomains must be smaller than the communicator size
10838 @*/
10839 PetscErrorCode MatSubdomainsCreateCoalesce(Mat A,PetscInt N,PetscInt *n,IS *iss[])
10840 {
10841   MPI_Comm        comm,subcomm;
10842   PetscMPIInt     size,rank,color;
10843   PetscInt        rstart,rend,k;
10844   PetscErrorCode  ierr;
10845 
10846   PetscFunctionBegin;
10847   ierr = PetscObjectGetComm((PetscObject)A,&comm);CHKERRQ(ierr);
10848   ierr = MPI_Comm_size(comm,&size);CHKERRQ(ierr);
10849   ierr = MPI_Comm_rank(comm,&rank);CHKERRQ(ierr);
10850   if (N < 1 || N >= (PetscInt)size) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONG,"number of subdomains must be > 0 and < %D, got N = %D",size,N);
10851   *n = 1;
10852   k = ((PetscInt)size)/N + ((PetscInt)size%N>0); /* There are up to k ranks to a color */
10853   color = rank/k;
10854   ierr = MPI_Comm_split(comm,color,rank,&subcomm);CHKERRQ(ierr);
10855   ierr = PetscMalloc1(1,iss);CHKERRQ(ierr);
10856   ierr = MatGetOwnershipRange(A,&rstart,&rend);CHKERRQ(ierr);
10857   ierr = ISCreateStride(subcomm,rend-rstart,rstart,1,iss[0]);CHKERRQ(ierr);
10858   ierr = MPI_Comm_free(&subcomm);CHKERRQ(ierr);
10859   PetscFunctionReturn(0);
10860 }
10861 
10862 /*@
10863    MatGalerkin - Constructs the coarse grid problem via Galerkin projection.
10864 
10865    If the interpolation and restriction operators are the same, uses MatPtAP.
10866    If they are not the same, use MatMatMatMult.
10867 
10868    Once the coarse grid problem is constructed, correct for interpolation operators
10869    that are not of full rank, which can legitimately happen in the case of non-nested
10870    geometric multigrid.
10871 
10872    Input Parameters:
10873 +  restrct - restriction operator
10874 .  dA - fine grid matrix
10875 .  interpolate - interpolation operator
10876 .  reuse - either MAT_INITIAL_MATRIX or MAT_REUSE_MATRIX
10877 -  fill - expected fill, use PETSC_DEFAULT if you do not have a good estimate
10878 
10879    Output Parameters:
10880 .  A - the Galerkin coarse matrix
10881 
10882    Options Database Key:
10883 .  -pc_mg_galerkin <both,pmat,mat,none>
10884 
10885    Level: developer
10886 
10887 .keywords: MG, multigrid, Galerkin
10888 
10889 .seealso: MatPtAP(), MatMatMatMult()
10890 @*/
10891 PetscErrorCode  MatGalerkin(Mat restrct, Mat dA, Mat interpolate, MatReuse reuse, PetscReal fill, Mat *A)
10892 {
10893   PetscErrorCode ierr;
10894   IS             zerorows;
10895   Vec            diag;
10896 
10897   PetscFunctionBegin;
10898   if (reuse == MAT_INPLACE_MATRIX) SETERRQ(PetscObjectComm((PetscObject)A),PETSC_ERR_SUP,"Inplace product not supported");
10899   /* Construct the coarse grid matrix */
10900   if (interpolate == restrct) {
10901     ierr = MatPtAP(dA,interpolate,reuse,fill,A);CHKERRQ(ierr);
10902   } else {
10903     ierr = MatMatMatMult(restrct,dA,interpolate,reuse,fill,A);CHKERRQ(ierr);
10904   }
10905 
10906   /* If the interpolation matrix is not of full rank, A will have zero rows.
10907      This can legitimately happen in the case of non-nested geometric multigrid.
10908      In that event, we set the rows of the matrix to the rows of the identity,
10909      ignoring the equations (as the RHS will also be zero). */
10910 
10911   ierr = MatFindZeroRows(*A, &zerorows);CHKERRQ(ierr);
10912 
10913   if (zerorows != NULL) { /* if there are any zero rows */
10914     ierr = MatCreateVecs(*A, &diag, NULL);CHKERRQ(ierr);
10915     ierr = MatGetDiagonal(*A, diag);CHKERRQ(ierr);
10916     ierr = VecISSet(diag, zerorows, 1.0);CHKERRQ(ierr);
10917     ierr = MatDiagonalSet(*A, diag, INSERT_VALUES);CHKERRQ(ierr);
10918     ierr = VecDestroy(&diag);CHKERRQ(ierr);
10919     ierr = ISDestroy(&zerorows);CHKERRQ(ierr);
10920   }
10921   PetscFunctionReturn(0);
10922 }
10923 
10924 /*@C
10925     MatSetOperation - Allows user to set a matrix operation for any matrix type
10926 
10927    Logically Collective on Mat
10928 
10929     Input Parameters:
10930 +   mat - the matrix
10931 .   op - the name of the operation
10932 -   f - the function that provides the operation
10933 
10934    Level: developer
10935 
10936     Usage:
10937 $      extern PetscErrorCode usermult(Mat,Vec,Vec);
10938 $      ierr = MatCreateXXX(comm,...&A);
10939 $      ierr = MatSetOperation(A,MATOP_MULT,(void(*)(void))usermult);
10940 
10941     Notes:
10942     See the file include/petscmat.h for a complete list of matrix
10943     operations, which all have the form MATOP_<OPERATION>, where
10944     <OPERATION> is the name (in all capital letters) of the
10945     user interface routine (e.g., MatMult() -> MATOP_MULT).
10946 
10947     All user-provided functions (except for MATOP_DESTROY) should have the same calling
10948     sequence as the usual matrix interface routines, since they
10949     are intended to be accessed via the usual matrix interface
10950     routines, e.g.,
10951 $       MatMult(Mat,Vec,Vec) -> usermult(Mat,Vec,Vec)
10952 
10953     In particular each function MUST return an error code of 0 on success and
10954     nonzero on failure.
10955 
10956     This routine is distinct from MatShellSetOperation() in that it can be called on any matrix type.
10957 
10958 .keywords: matrix, set, operation
10959 
10960 .seealso: MatGetOperation(), MatCreateShell(), MatShellSetContext(), MatShellSetOperation()
10961 @*/
10962 PetscErrorCode MatSetOperation(Mat mat,MatOperation op,void (*f)(void))
10963 {
10964   PetscFunctionBegin;
10965   PetscValidHeaderSpecific(mat,MAT_CLASSID,1);
10966   if (op == MATOP_VIEW && !mat->ops->viewnative && f != (void (*)(void))(mat->ops->view)) {
10967     mat->ops->viewnative = mat->ops->view;
10968   }
10969   (((void(**)(void))mat->ops)[op]) = f;
10970   PetscFunctionReturn(0);
10971 }
10972 
10973 /*@C
10974     MatGetOperation - Gets a matrix operation for any matrix type.
10975 
10976     Not Collective
10977 
10978     Input Parameters:
10979 +   mat - the matrix
10980 -   op - the name of the operation
10981 
10982     Output Parameter:
10983 .   f - the function that provides the operation
10984 
10985     Level: developer
10986 
10987     Usage:
10988 $      PetscErrorCode (*usermult)(Mat,Vec,Vec);
10989 $      ierr = MatGetOperation(A,MATOP_MULT,(void(**)(void))&usermult);
10990 
10991     Notes:
10992     See the file include/petscmat.h for a complete list of matrix
10993     operations, which all have the form MATOP_<OPERATION>, where
10994     <OPERATION> is the name (in all capital letters) of the
10995     user interface routine (e.g., MatMult() -> MATOP_MULT).
10996 
10997     This routine is distinct from MatShellGetOperation() in that it can be called on any matrix type.
10998 
10999 .keywords: matrix, get, operation
11000 
11001 .seealso: MatSetOperation(), MatCreateShell(), MatShellGetContext(), MatShellGetOperation()
11002 @*/
11003 PetscErrorCode MatGetOperation(Mat mat,MatOperation op,void(**f)(void))
11004 {
11005   PetscFunctionBegin;
11006   PetscValidHeaderSpecific(mat,MAT_CLASSID,1);
11007   *f = (((void (**)(void))mat->ops)[op]);
11008   PetscFunctionReturn(0);
11009 }
11010 
11011 /*@
11012     MatHasOperation - Determines whether the given matrix supports the particular
11013     operation.
11014 
11015    Not Collective
11016 
11017    Input Parameters:
11018 +  mat - the matrix
11019 -  op - the operation, for example, MATOP_GET_DIAGONAL
11020 
11021    Output Parameter:
11022 .  has - either PETSC_TRUE or PETSC_FALSE
11023 
11024    Level: advanced
11025 
11026    Notes:
11027    See the file include/petscmat.h for a complete list of matrix
11028    operations, which all have the form MATOP_<OPERATION>, where
11029    <OPERATION> is the name (in all capital letters) of the
11030    user-level routine.  E.g., MatNorm() -> MATOP_NORM.
11031 
11032 .keywords: matrix, has, operation
11033 
11034 .seealso: MatCreateShell()
11035 @*/
11036 PetscErrorCode MatHasOperation(Mat mat,MatOperation op,PetscBool *has)
11037 {
11038   PetscErrorCode ierr;
11039 
11040   PetscFunctionBegin;
11041   PetscValidHeaderSpecific(mat,MAT_CLASSID,1);
11042   PetscValidType(mat,1);
11043   PetscValidPointer(has,3);
11044   if (mat->ops->hasoperation) {
11045     ierr = (*mat->ops->hasoperation)(mat,op,has);CHKERRQ(ierr);
11046   } else {
11047     if (((void**)mat->ops)[op]) *has =  PETSC_TRUE;
11048     else {
11049       *has = PETSC_FALSE;
11050       if (op == MATOP_CREATE_SUBMATRIX) {
11051         PetscMPIInt size;
11052 
11053         ierr = MPI_Comm_size(PetscObjectComm((PetscObject)mat),&size);CHKERRQ(ierr);
11054         if (size == 1) {
11055           ierr = MatHasOperation(mat,MATOP_CREATE_SUBMATRICES,has);CHKERRQ(ierr);
11056         }
11057       }
11058     }
11059   }
11060   PetscFunctionReturn(0);
11061 }
11062 
11063 /*@
11064     MatHasCongruentLayouts - Determines whether the rows and columns layouts
11065     of the matrix are congruent
11066 
11067    Collective on mat
11068 
11069    Input Parameters:
11070 .  mat - the matrix
11071 
11072    Output Parameter:
11073 .  cong - either PETSC_TRUE or PETSC_FALSE
11074 
11075    Level: beginner
11076 
11077    Notes:
11078 
11079 .keywords: matrix, has
11080 
11081 .seealso: MatCreate(), MatSetSizes()
11082 @*/
11083 PetscErrorCode MatHasCongruentLayouts(Mat mat,PetscBool *cong)
11084 {
11085   PetscErrorCode ierr;
11086 
11087   PetscFunctionBegin;
11088   PetscValidHeaderSpecific(mat,MAT_CLASSID,1);
11089   PetscValidType(mat,1);
11090   PetscValidPointer(cong,2);
11091   if (!mat->rmap || !mat->cmap) {
11092     *cong = mat->rmap == mat->cmap ? PETSC_TRUE : PETSC_FALSE;
11093     PetscFunctionReturn(0);
11094   }
11095   if (mat->congruentlayouts == PETSC_DECIDE) { /* first time we compare rows and cols layouts */
11096     ierr = PetscLayoutCompare(mat->rmap,mat->cmap,cong);CHKERRQ(ierr);
11097     if (*cong) mat->congruentlayouts = 1;
11098     else       mat->congruentlayouts = 0;
11099   } else *cong = mat->congruentlayouts ? PETSC_TRUE : PETSC_FALSE;
11100   PetscFunctionReturn(0);
11101 }
11102 
11103 /*@
11104     MatFreeIntermediateDataStructures - Free intermediate data structures created for reuse,
11105     e.g., matrx product of MatPtAP.
11106 
11107    Collective on mat
11108 
11109    Input Parameters:
11110 .  mat - the matrix
11111 
11112    Output Parameter:
11113 .  mat - the matrix with intermediate data structures released
11114 
11115    Level: advanced
11116 
11117    Notes:
11118 
11119 .keywords: matrix
11120 
11121 .seealso: MatPtAP(), MatMatMult()
11122 @*/
11123 PetscErrorCode MatFreeIntermediateDataStructures(Mat mat)
11124 {
11125   PetscErrorCode ierr;
11126 
11127   PetscFunctionBegin;
11128   PetscValidHeaderSpecific(mat,MAT_CLASSID,1);
11129   PetscValidType(mat,1);
11130   if (mat->ops->freeintermediatedatastructures) {
11131     ierr = (*mat->ops->freeintermediatedatastructures)(mat);CHKERRQ(ierr);
11132   }
11133   PetscFunctionReturn(0);
11134 }
11135