xref: /petsc/src/snes/impls/vi/vi.c (revision 98d129c30f3ee9fdddc40fdbc5a989b7be64f888)
1 #include <petsc/private/snesimpl.h> /*I "petscsnes.h" I*/
2 #include <petscdm.h>
3 
4 /*@C
5   SNESVISetComputeVariableBounds - Sets a function that is called to compute the bounds on variable for
6   (differential) variable inequalities.
7 
8   Input Parameters:
9 + snes    - the `SNES` context
10 - compute - function that computes the bounds
11 
12   Calling sequence of `compute`:
13 + snes   - the `SNES` context
14 . lower  - vector to hold lower bounds
15 - higher - vector to hold upper bounds
16 
17   Level: advanced
18 
19   Notes:
20   Problems with bound constraints can be solved with the reduced space, `SNESVINEWTONRSLS`, and semi-smooth `SNESVINEWTONSSLS` solvers.
21 
22   For entries with no bounds you can set `PETSC_NINFINITY` or `PETSC_INFINITY`
23 
24   You may use `SNESVISetVariableBounds()` to provide the bounds once if they will never change
25 
26   If you have associated a `DM` with the `SNES` and provided a function to the `DM` via `DMSetVariableBounds()` that will be used automatically
27   to provide the bounds and you need not use this function.
28 
29 .seealso: [](sec_vi), `SNES`, `SNESVISetVariableBounds()`, `DMSetVariableBounds()`, `SNESSetFunctionDomainError()`, `SNESSetJacobianDomainError()`, `SNESVINEWTONRSLS`, `SNESVINEWTONSSLS`,
30           `SNESSetType()`, `PETSC_NINFINITY`, `PETSC_INFINITY`
31 @*/
32 PetscErrorCode SNESVISetComputeVariableBounds(SNES snes, PetscErrorCode (*compute)(SNES snes, Vec lower, Vec higher))
33 {
34   PetscErrorCode (*f)(SNES, PetscErrorCode (*)(SNES, Vec, Vec));
35 
36   PetscFunctionBegin;
37   PetscValidHeaderSpecific(snes, SNES_CLASSID, 1);
38   PetscCall(PetscObjectQueryFunction((PetscObject)snes, "SNESVISetComputeVariableBounds_C", &f));
39   if (f) PetscUseMethod(snes, "SNESVISetComputeVariableBounds_C", (SNES, PetscErrorCode(*)(SNES, Vec, Vec)), (snes, compute));
40   else PetscCall(SNESVISetComputeVariableBounds_VI(snes, compute));
41   PetscFunctionReturn(PETSC_SUCCESS);
42 }
43 
44 PetscErrorCode SNESVISetComputeVariableBounds_VI(SNES snes, SNESVIComputeVariableBoundsFn *compute)
45 {
46   PetscFunctionBegin;
47   snes->ops->computevariablebounds = compute;
48   PetscFunctionReturn(PETSC_SUCCESS);
49 }
50 
51 static PetscErrorCode SNESVIMonitorResidual(SNES snes, PetscInt its, PetscReal fgnorm, void *dummy)
52 {
53   Vec         X, F, Finactive;
54   IS          isactive;
55   PetscViewer viewer = (PetscViewer)dummy;
56 
57   PetscFunctionBegin;
58   PetscCall(SNESGetFunction(snes, &F, NULL, NULL));
59   PetscCall(SNESGetSolution(snes, &X));
60   PetscCall(SNESVIGetActiveSetIS(snes, X, F, &isactive));
61   PetscCall(VecDuplicate(F, &Finactive));
62   PetscCall(VecCopy(F, Finactive));
63   PetscCall(VecISSet(Finactive, isactive, 0.0));
64   PetscCall(ISDestroy(&isactive));
65   PetscCall(VecView(Finactive, viewer));
66   PetscCall(VecDestroy(&Finactive));
67   PetscFunctionReturn(PETSC_SUCCESS);
68 }
69 
70 static PetscErrorCode SNESMonitorVI(SNES snes, PetscInt its, PetscReal fgnorm, void *dummy)
71 {
72   PetscViewer        viewer = (PetscViewer)dummy;
73   const PetscScalar *x, *xl, *xu, *f;
74   PetscInt           i, n, act[2] = {0, 0}, fact[2], N;
75   /* Number of components that actually hit the bounds (c.f. active variables) */
76   PetscInt  act_bound[2] = {0, 0}, fact_bound[2];
77   PetscReal rnorm, fnorm, zerotolerance = snes->vizerotolerance;
78   double    tmp;
79 
80   PetscFunctionBegin;
81   PetscValidHeaderSpecific(viewer, PETSC_VIEWER_CLASSID, 4);
82   PetscCall(VecGetLocalSize(snes->vec_sol, &n));
83   PetscCall(VecGetSize(snes->vec_sol, &N));
84   PetscCall(VecGetArrayRead(snes->xl, &xl));
85   PetscCall(VecGetArrayRead(snes->xu, &xu));
86   PetscCall(VecGetArrayRead(snes->vec_sol, &x));
87   PetscCall(VecGetArrayRead(snes->vec_func, &f));
88 
89   rnorm = 0.0;
90   for (i = 0; i < n; i++) {
91     if ((PetscRealPart(x[i]) > PetscRealPart(xl[i]) + zerotolerance || (PetscRealPart(f[i]) <= 0.0)) && ((PetscRealPart(x[i]) < PetscRealPart(xu[i]) - zerotolerance) || PetscRealPart(f[i]) >= 0.0)) rnorm += PetscRealPart(PetscConj(f[i]) * f[i]);
92     else if (PetscRealPart(x[i]) <= PetscRealPart(xl[i]) + zerotolerance && PetscRealPart(f[i]) > 0.0) act[0]++;
93     else if (PetscRealPart(x[i]) >= PetscRealPart(xu[i]) - zerotolerance && PetscRealPart(f[i]) < 0.0) act[1]++;
94     else SETERRQ(PetscObjectComm((PetscObject)snes), PETSC_ERR_PLIB, "Can never get here");
95   }
96 
97   for (i = 0; i < n; i++) {
98     if (PetscRealPart(x[i]) <= PetscRealPart(xl[i]) + zerotolerance) act_bound[0]++;
99     else if (PetscRealPart(x[i]) >= PetscRealPart(xu[i]) - zerotolerance) act_bound[1]++;
100   }
101   PetscCall(VecRestoreArrayRead(snes->vec_func, &f));
102   PetscCall(VecRestoreArrayRead(snes->xl, &xl));
103   PetscCall(VecRestoreArrayRead(snes->xu, &xu));
104   PetscCall(VecRestoreArrayRead(snes->vec_sol, &x));
105   PetscCall(MPIU_Allreduce(&rnorm, &fnorm, 1, MPIU_REAL, MPIU_SUM, PetscObjectComm((PetscObject)snes)));
106   PetscCall(MPIU_Allreduce(act, fact, 2, MPIU_INT, MPI_SUM, PetscObjectComm((PetscObject)snes)));
107   PetscCall(MPIU_Allreduce(act_bound, fact_bound, 2, MPIU_INT, MPI_SUM, PetscObjectComm((PetscObject)snes)));
108   fnorm = PetscSqrtReal(fnorm);
109 
110   PetscCall(PetscViewerASCIIAddTab(viewer, ((PetscObject)snes)->tablevel));
111   if (snes->ntruebounds) tmp = ((double)(fact[0] + fact[1])) / ((double)snes->ntruebounds);
112   else tmp = 0.0;
113   PetscCall(PetscViewerASCIIPrintf(viewer, "%3" PetscInt_FMT " SNES VI Function norm %g Active lower constraints %" PetscInt_FMT "/%" PetscInt_FMT " upper constraints %" PetscInt_FMT "/%" PetscInt_FMT " Percent of total %g Percent of bounded %g\n", its, (double)fnorm, fact[0], fact_bound[0], fact[1], fact_bound[1], ((double)(fact[0] + fact[1])) / ((double)N), tmp));
114 
115   PetscCall(PetscViewerASCIISubtractTab(viewer, ((PetscObject)snes)->tablevel));
116   PetscFunctionReturn(PETSC_SUCCESS);
117 }
118 
119 /*
120      Checks if J^T F = 0 which implies we've found a local minimum of the norm of the function,
121     || F(u) ||_2 but not a zero, F(u) = 0. In the case when one cannot compute J^T F we use the fact that
122     0 = (J^T F)^T W = F^T J W iff W not in the null space of J. Thanks for Jorge More
123     for this trick. One assumes that the probability that W is in the null space of J is very, very small.
124 */
125 PetscErrorCode SNESVICheckLocalMin_Private(SNES snes, Mat A, Vec F, Vec W, PetscReal fnorm, PetscBool *ismin)
126 {
127   PetscReal a1;
128   PetscBool hastranspose;
129 
130   PetscFunctionBegin;
131   *ismin = PETSC_FALSE;
132   PetscCall(MatHasOperation(A, MATOP_MULT_TRANSPOSE, &hastranspose));
133   if (hastranspose) {
134     /* Compute || J^T F|| */
135     PetscCall(MatMultTranspose(A, F, W));
136     PetscCall(VecNorm(W, NORM_2, &a1));
137     PetscCall(PetscInfo(snes, "|| J^T F|| %g near zero implies found a local minimum\n", (double)(a1 / fnorm)));
138     if (a1 / fnorm < 1.e-4) *ismin = PETSC_TRUE;
139   } else {
140     Vec         work;
141     PetscScalar result;
142     PetscReal   wnorm;
143 
144     PetscCall(VecSetRandom(W, NULL));
145     PetscCall(VecNorm(W, NORM_2, &wnorm));
146     PetscCall(VecDuplicate(W, &work));
147     PetscCall(MatMult(A, W, work));
148     PetscCall(VecDot(F, work, &result));
149     PetscCall(VecDestroy(&work));
150     a1 = PetscAbsScalar(result) / (fnorm * wnorm);
151     PetscCall(PetscInfo(snes, "(F^T J random)/(|| F ||*||random|| %g near zero implies found a local minimum\n", (double)a1));
152     if (a1 < 1.e-4) *ismin = PETSC_TRUE;
153   }
154   PetscFunctionReturn(PETSC_SUCCESS);
155 }
156 
157 /*
158   SNESConvergedDefault_VI - Checks the convergence of the semismooth newton algorithm.
159 
160   Notes:
161   The convergence criterion currently implemented is
162   merit < abstol
163   merit < rtol*merit_initial
164 */
165 PetscErrorCode SNESConvergedDefault_VI(SNES snes, PetscInt it, PetscReal xnorm, PetscReal gradnorm, PetscReal fnorm, SNESConvergedReason *reason, void *dummy)
166 {
167   PetscFunctionBegin;
168   PetscValidHeaderSpecific(snes, SNES_CLASSID, 1);
169   PetscAssertPointer(reason, 6);
170 
171   *reason = SNES_CONVERGED_ITERATING;
172 
173   if (!it) {
174     /* set parameter for default relative tolerance convergence test */
175     snes->ttol = fnorm * snes->rtol;
176   }
177   if (fnorm != fnorm) {
178     PetscCall(PetscInfo(snes, "Failed to converged, function norm is NaN\n"));
179     *reason = SNES_DIVERGED_FNORM_NAN;
180   } else if (fnorm < snes->abstol && (it || !snes->forceiteration)) {
181     PetscCall(PetscInfo(snes, "Converged due to function norm %g < %g\n", (double)fnorm, (double)snes->abstol));
182     *reason = SNES_CONVERGED_FNORM_ABS;
183   } else if (snes->nfuncs >= snes->max_funcs && snes->max_funcs >= 0) {
184     PetscCall(PetscInfo(snes, "Exceeded maximum number of function evaluations: %" PetscInt_FMT " > %" PetscInt_FMT "\n", snes->nfuncs, snes->max_funcs));
185     *reason = SNES_DIVERGED_FUNCTION_COUNT;
186   }
187 
188   if (it && !*reason) {
189     if (fnorm < snes->ttol) {
190       PetscCall(PetscInfo(snes, "Converged due to function norm %g < %g (relative tolerance)\n", (double)fnorm, (double)snes->ttol));
191       *reason = SNES_CONVERGED_FNORM_RELATIVE;
192     }
193   }
194   PetscFunctionReturn(PETSC_SUCCESS);
195 }
196 
197 /*
198    SNESVIProjectOntoBounds - Projects X onto the feasible region so that Xl[i] <= X[i] <= Xu[i] for i = 1...n.
199 
200    Input Parameters:
201 .  SNES - nonlinear solver context
202 
203    Output Parameters:
204 .  X - Bound projected X
205 
206 */
207 
208 PetscErrorCode SNESVIProjectOntoBounds(SNES snes, Vec X)
209 {
210   const PetscScalar *xl, *xu;
211   PetscScalar       *x;
212   PetscInt           i, n;
213 
214   PetscFunctionBegin;
215   PetscCall(VecGetLocalSize(X, &n));
216   PetscCall(VecGetArray(X, &x));
217   PetscCall(VecGetArrayRead(snes->xl, &xl));
218   PetscCall(VecGetArrayRead(snes->xu, &xu));
219 
220   for (i = 0; i < n; i++) {
221     if (PetscRealPart(x[i]) < PetscRealPart(xl[i])) x[i] = xl[i];
222     else if (PetscRealPart(x[i]) > PetscRealPart(xu[i])) x[i] = xu[i];
223   }
224   PetscCall(VecRestoreArray(X, &x));
225   PetscCall(VecRestoreArrayRead(snes->xl, &xl));
226   PetscCall(VecRestoreArrayRead(snes->xu, &xu));
227   PetscFunctionReturn(PETSC_SUCCESS);
228 }
229 
230 /*@
231   SNESVIGetActiveSetIS - Gets the global indices for the active set variables
232 
233   Input Parameters:
234 + snes - the `SNES` context
235 . X    - the `snes` solution vector
236 - F    - the nonlinear function vector
237 
238   Output Parameter:
239 . ISact - active set index set
240 
241   Level: developer
242 
243 .seealso: [](ch_snes), `SNES`, `SNESVINEWTONRSLS`, `SNESVINEWTONSSLS`
244 @*/
245 PetscErrorCode SNESVIGetActiveSetIS(SNES snes, Vec X, Vec F, IS *ISact)
246 {
247   Vec                Xl = snes->xl, Xu = snes->xu;
248   const PetscScalar *x, *f, *xl, *xu;
249   PetscInt          *idx_act, i, nlocal, nloc_isact = 0, ilow, ihigh, i1 = 0;
250   PetscReal          zerotolerance = snes->vizerotolerance;
251 
252   PetscFunctionBegin;
253   PetscCall(VecGetLocalSize(X, &nlocal));
254   PetscCall(VecGetOwnershipRange(X, &ilow, &ihigh));
255   PetscCall(VecGetArrayRead(X, &x));
256   PetscCall(VecGetArrayRead(Xl, &xl));
257   PetscCall(VecGetArrayRead(Xu, &xu));
258   PetscCall(VecGetArrayRead(F, &f));
259   /* Compute active set size */
260   for (i = 0; i < nlocal; i++) {
261     if (!((PetscRealPart(x[i]) > PetscRealPart(xl[i]) + zerotolerance || (PetscRealPart(f[i]) <= 0.0)) && ((PetscRealPart(x[i]) < PetscRealPart(xu[i]) - zerotolerance) || PetscRealPart(f[i]) >= 0.0))) nloc_isact++;
262   }
263 
264   PetscCall(PetscMalloc1(nloc_isact, &idx_act));
265 
266   /* Set active set indices */
267   for (i = 0; i < nlocal; i++) {
268     if (!((PetscRealPart(x[i]) > PetscRealPart(xl[i]) + zerotolerance || (PetscRealPart(f[i]) <= 0.0)) && ((PetscRealPart(x[i]) < PetscRealPart(xu[i]) - zerotolerance) || PetscRealPart(f[i]) >= 0.0))) idx_act[i1++] = ilow + i;
269   }
270 
271   /* Create active set IS */
272   PetscCall(ISCreateGeneral(PetscObjectComm((PetscObject)snes), nloc_isact, idx_act, PETSC_OWN_POINTER, ISact));
273 
274   PetscCall(VecRestoreArrayRead(X, &x));
275   PetscCall(VecRestoreArrayRead(Xl, &xl));
276   PetscCall(VecRestoreArrayRead(Xu, &xu));
277   PetscCall(VecRestoreArrayRead(F, &f));
278   PetscFunctionReturn(PETSC_SUCCESS);
279 }
280 
281 PetscErrorCode SNESVIComputeInactiveSetFnorm(SNES snes, Vec F, Vec X, PetscReal *fnorm)
282 {
283   const PetscScalar *x, *xl, *xu, *f;
284   PetscInt           i, n;
285   PetscReal          rnorm, zerotolerance = snes->vizerotolerance;
286 
287   PetscFunctionBegin;
288   PetscCall(VecGetLocalSize(X, &n));
289   PetscCall(VecGetArrayRead(snes->xl, &xl));
290   PetscCall(VecGetArrayRead(snes->xu, &xu));
291   PetscCall(VecGetArrayRead(X, &x));
292   PetscCall(VecGetArrayRead(F, &f));
293   rnorm = 0.0;
294   for (i = 0; i < n; i++) {
295     if ((PetscRealPart(x[i]) > PetscRealPart(xl[i]) + zerotolerance || (PetscRealPart(f[i]) <= 0.0)) && ((PetscRealPart(x[i]) < PetscRealPart(xu[i]) - zerotolerance) || PetscRealPart(f[i]) >= 0.0)) rnorm += PetscRealPart(PetscConj(f[i]) * f[i]);
296   }
297   PetscCall(VecRestoreArrayRead(F, &f));
298   PetscCall(VecRestoreArrayRead(snes->xl, &xl));
299   PetscCall(VecRestoreArrayRead(snes->xu, &xu));
300   PetscCall(VecRestoreArrayRead(X, &x));
301   PetscCall(MPIU_Allreduce(&rnorm, fnorm, 1, MPIU_REAL, MPIU_SUM, PetscObjectComm((PetscObject)snes)));
302   *fnorm = PetscSqrtReal(*fnorm);
303   PetscFunctionReturn(PETSC_SUCCESS);
304 }
305 
306 static PetscErrorCode SNESVIDMComputeVariableBounds(SNES snes, Vec xl, Vec xu)
307 {
308   PetscFunctionBegin;
309   PetscCall(DMComputeVariableBounds(snes->dm, xl, xu));
310   PetscFunctionReturn(PETSC_SUCCESS);
311 }
312 
313 /*
314    SNESSetUp_VI - Does setup common to all VI solvers -- basically makes sure bounds have been properly set up
315    of the SNESVI nonlinear solver.
316 
317    Input Parameter:
318 .  snes - the SNES context
319 
320    Application Interface Routine: SNESSetUp()
321 
322    Notes:
323    For basic use of the SNES solvers, the user need not explicitly call
324    SNESSetUp(), since these actions will automatically occur during
325    the call to SNESSolve().
326  */
327 PetscErrorCode SNESSetUp_VI(SNES snes)
328 {
329   PetscInt i_start[3], i_end[3];
330 
331   PetscFunctionBegin;
332   PetscCall(SNESSetWorkVecs(snes, 1));
333   PetscCall(SNESSetUpMatrices(snes));
334 
335   if (!snes->ops->computevariablebounds && snes->dm) {
336     PetscBool flag;
337     PetscCall(DMHasVariableBounds(snes->dm, &flag));
338     if (flag) snes->ops->computevariablebounds = SNESVIDMComputeVariableBounds;
339   }
340   if (!snes->usersetbounds) {
341     if (snes->ops->computevariablebounds) {
342       if (!snes->xl) PetscCall(VecDuplicate(snes->vec_sol, &snes->xl));
343       if (!snes->xu) PetscCall(VecDuplicate(snes->vec_sol, &snes->xu));
344       PetscUseTypeMethod(snes, computevariablebounds, snes->xl, snes->xu);
345     } else if (!snes->xl && !snes->xu) {
346       /* If the lower and upper bound on variables are not set, set it to -Inf and Inf */
347       PetscCall(VecDuplicate(snes->vec_sol, &snes->xl));
348       PetscCall(VecSet(snes->xl, PETSC_NINFINITY));
349       PetscCall(VecDuplicate(snes->vec_sol, &snes->xu));
350       PetscCall(VecSet(snes->xu, PETSC_INFINITY));
351     } else {
352       /* Check if lower bound, upper bound and solution vector distribution across the processors is identical */
353       PetscCall(VecGetOwnershipRange(snes->vec_sol, i_start, i_end));
354       PetscCall(VecGetOwnershipRange(snes->xl, i_start + 1, i_end + 1));
355       PetscCall(VecGetOwnershipRange(snes->xu, i_start + 2, i_end + 2));
356       if ((i_start[0] != i_start[1]) || (i_start[0] != i_start[2]) || (i_end[0] != i_end[1]) || (i_end[0] != i_end[2]))
357         SETERRQ(PETSC_COMM_SELF, PETSC_ERR_ARG_SIZ, "Distribution of lower bound, upper bound and the solution vector should be identical across all the processors.");
358     }
359   }
360   PetscFunctionReturn(PETSC_SUCCESS);
361 }
362 PetscErrorCode SNESReset_VI(SNES snes)
363 {
364   PetscFunctionBegin;
365   PetscCall(VecDestroy(&snes->xl));
366   PetscCall(VecDestroy(&snes->xu));
367   snes->usersetbounds = PETSC_FALSE;
368   PetscFunctionReturn(PETSC_SUCCESS);
369 }
370 
371 /*
372    SNESDestroy_VI - Destroys the private SNES_VI context that was created
373    with SNESCreate_VI().
374 
375    Input Parameter:
376 .  snes - the SNES context
377 
378    Application Interface Routine: SNESDestroy()
379  */
380 PetscErrorCode SNESDestroy_VI(SNES snes)
381 {
382   PetscFunctionBegin;
383   PetscCall(PetscFree(snes->data));
384 
385   /* clear composed functions */
386   PetscCall(PetscObjectComposeFunction((PetscObject)snes, "SNESVISetVariableBounds_C", NULL));
387   PetscCall(PetscObjectComposeFunction((PetscObject)snes, "SNESVISetComputeVariableBounds_C", NULL));
388   PetscFunctionReturn(PETSC_SUCCESS);
389 }
390 
391 /*@
392   SNESVISetVariableBounds - Sets the lower and upper bounds for the solution vector. `xl` <= x <= `xu`. This allows solving
393   (differential) variable inequalities.
394 
395   Input Parameters:
396 + snes - the `SNES` context.
397 . xl   - lower bound.
398 - xu   - upper bound.
399 
400   Level: advanced
401 
402   Notes:
403   If this routine is not called then the lower and upper bounds are set to
404   `PETSC_NINFINITY` and `PETSC_INFINITY` respectively during `SNESSetUp()`.
405 
406   Problems with bound constraints can be solved with the reduced space, `SNESVINEWTONRSLS` or semi-smooth `SNESVINEWTONSSLS` solvers.
407 
408   For particular components that have no bounds you can use `PETSC_NINFINITY` or `PETSC_INFINITY`
409 
410   `SNESVISetComputeVariableBounds()` can be used to provide a function that computes the bounds. This should be used if you are using, for example, grid
411   sequencing and need bounds set for a variety of vectors
412 
413 .seealso: [](sec_vi), `SNES`, `SNESVIGetVariableBounds()`, `SNESVISetComputeVariableBounds()`, `SNESSetFunctionDomainError()`, `SNESSetJacobianDomainError()`, `SNESVINEWTONRSLS`, `SNESVINEWTONSSLS`, `SNESSetType()`, `PETSC_NINFINITY`, `PETSC_INFINITY`
414 @*/
415 PetscErrorCode SNESVISetVariableBounds(SNES snes, Vec xl, Vec xu)
416 {
417   PetscErrorCode (*f)(SNES, Vec, Vec);
418 
419   PetscFunctionBegin;
420   PetscValidHeaderSpecific(snes, SNES_CLASSID, 1);
421   PetscValidHeaderSpecific(xl, VEC_CLASSID, 2);
422   PetscValidHeaderSpecific(xu, VEC_CLASSID, 3);
423   PetscCall(PetscObjectQueryFunction((PetscObject)snes, "SNESVISetVariableBounds_C", &f));
424   if (f) PetscUseMethod(snes, "SNESVISetVariableBounds_C", (SNES, Vec, Vec), (snes, xl, xu));
425   else PetscCall(SNESVISetVariableBounds_VI(snes, xl, xu));
426   snes->usersetbounds = PETSC_TRUE;
427   PetscFunctionReturn(PETSC_SUCCESS);
428 }
429 
430 PetscErrorCode SNESVISetVariableBounds_VI(SNES snes, Vec xl, Vec xu)
431 {
432   const PetscScalar *xxl, *xxu;
433   PetscInt           i, n, cnt = 0;
434 
435   PetscFunctionBegin;
436   PetscCall(SNESGetFunction(snes, &snes->vec_func, NULL, NULL));
437   PetscCheck(snes->vec_func, PETSC_COMM_SELF, PETSC_ERR_ARG_WRONGSTATE, "Must call SNESSetFunction() or SNESSetDM() first");
438   {
439     PetscInt xlN, xuN, N;
440     PetscCall(VecGetSize(xl, &xlN));
441     PetscCall(VecGetSize(xu, &xuN));
442     PetscCall(VecGetSize(snes->vec_func, &N));
443     PetscCheck(xlN == N, PETSC_COMM_SELF, PETSC_ERR_ARG_INCOMP, "Incompatible vector lengths lower bound = %" PetscInt_FMT " solution vector = %" PetscInt_FMT, xlN, N);
444     PetscCheck(xuN == N, PETSC_COMM_SELF, PETSC_ERR_ARG_INCOMP, "Incompatible vector lengths: upper bound = %" PetscInt_FMT " solution vector = %" PetscInt_FMT, xuN, N);
445   }
446   PetscCall(PetscObjectReference((PetscObject)xl));
447   PetscCall(PetscObjectReference((PetscObject)xu));
448   PetscCall(VecDestroy(&snes->xl));
449   PetscCall(VecDestroy(&snes->xu));
450   snes->xl = xl;
451   snes->xu = xu;
452   PetscCall(VecGetLocalSize(xl, &n));
453   PetscCall(VecGetArrayRead(xl, &xxl));
454   PetscCall(VecGetArrayRead(xu, &xxu));
455   for (i = 0; i < n; i++) cnt += ((xxl[i] != PETSC_NINFINITY) || (xxu[i] != PETSC_INFINITY));
456 
457   PetscCall(MPIU_Allreduce(&cnt, &snes->ntruebounds, 1, MPIU_INT, MPI_SUM, PetscObjectComm((PetscObject)snes)));
458   PetscCall(VecRestoreArrayRead(xl, &xxl));
459   PetscCall(VecRestoreArrayRead(xu, &xxu));
460   PetscFunctionReturn(PETSC_SUCCESS);
461 }
462 
463 /*@
464   SNESVIGetVariableBounds - Gets the lower and upper bounds for the solution vector. `xl` <= x <= `xu`. These are used in solving
465   (differential) variable inequalities.
466 
467   Input Parameters:
468 + snes - the `SNES` context.
469 . xl   - lower bound (may be `NULL`)
470 - xu   - upper bound (may be `NULL`)
471 
472   Level: advanced
473 
474   Note:
475   These vectors are owned by the `SNESVI` and should not be destroyed by the caller
476 
477 .seealso: [](sec_vi), `SNES`, `SNESVISetVariableBounds()`, `SNESVISetComputeVariableBounds()`, `SNESSetFunctionDomainError()`, `SNESSetJacobianDomainError()`, `SNESVINEWTONRSLS`, `SNESVINEWTONSSLS`, `SNESSetType()`, `PETSC_NINFINITY`, `PETSC_INFINITY`
478 @*/
479 PetscErrorCode SNESVIGetVariableBounds(SNES snes, Vec *xl, Vec *xu)
480 {
481   PetscFunctionBegin;
482   PetscCheck(snes->usersetbounds, PETSC_COMM_SELF, PETSC_ERR_ARG_WRONGSTATE, "Must set SNESVI bounds before calling SNESVIGetVariableBounds()");
483   if (xl) *xl = snes->xl;
484   if (xu) *xu = snes->xu;
485   PetscFunctionReturn(PETSC_SUCCESS);
486 }
487 
488 PetscErrorCode SNESSetFromOptions_VI(SNES snes, PetscOptionItems *PetscOptionsObject)
489 {
490   PetscBool flg = PETSC_FALSE;
491 
492   PetscFunctionBegin;
493   PetscOptionsHeadBegin(PetscOptionsObject, "SNES VI options");
494   PetscCall(PetscOptionsReal("-snes_vi_zero_tolerance", "Tolerance for considering x[] value to be on a bound", "None", snes->vizerotolerance, &snes->vizerotolerance, NULL));
495   PetscCall(PetscOptionsBool("-snes_vi_monitor", "Monitor all non-active variables", "SNESMonitorResidual", flg, &flg, NULL));
496   if (flg) PetscCall(SNESMonitorSet(snes, SNESMonitorVI, PETSC_VIEWER_STDOUT_(PetscObjectComm((PetscObject)snes)), NULL));
497   flg = PETSC_FALSE;
498   PetscCall(PetscOptionsBool("-snes_vi_monitor_residual", "Monitor residual all non-active variables; using zero for active constraints", "SNESMonitorVIResidual", flg, &flg, NULL));
499   if (flg) PetscCall(SNESMonitorSet(snes, SNESVIMonitorResidual, PETSC_VIEWER_DRAW_(PetscObjectComm((PetscObject)snes)), NULL));
500   PetscOptionsHeadEnd();
501   PetscFunctionReturn(PETSC_SUCCESS);
502 }
503