1: #pragma once

  3: #include <petsc/private/snesimpl.h>

  5: /*  Data structure for the Nonlinear GMRES method.  */
  6: typedef struct {
  7:   /* Solver parameters and counters */
  8:   PetscInt    msize;            /* maximum size of krylov space */
  9:   PetscInt    restart_it;       /* number of iterations the restart conditions persist before restart */
 10:   PetscViewer monitor;          /* debugging output for NGMRES */
 11:   PetscInt    restart_periodic; /* number of iterations to restart after */

 13:   SNESNGMRESRestartType restart_type;
 14:   SNESNGMRESSelectType  select_type;

 16:   /* History and subspace data */
 17:   Vec       *Fdot;   /* residual history -- length msize */
 18:   Vec       *Xdot;   /* solution history -- length msize */
 19:   PetscReal *fnorms; /* the residual norm history  */
 20:   PetscReal *xnorms; /* the solution norm history */

 22:   /* General minimization problem context */
 23:   PetscScalar *h;    /* the constraint matrix */
 24:   PetscScalar *beta; /* rhs for the minimization problem */
 25:   PetscScalar *xi;   /* the dot-product of the current and previous res. */

 27:   /* Line searches */
 28:   SNESLineSearch additive_linesearch; /* Line search for the additive variant */
 29:   PetscReal      descent_ls_test;     /* restart GMRES when X_A -X_M is not a descent direction */

 31:   /* Selection constants */
 32:   PetscBool candidate;       /* use candidate storage approach */
 33:   PetscBool approxfunc;      /* approximate the function rather than recomputing it */
 34:   PetscReal gammaA;          /* Criterion A residual tolerance */
 35:   PetscReal epsilonB;        /* Criterion B difference tolerance */
 36:   PetscReal deltaB;          /* Criterion B residual tolerance */
 37:   PetscReal gammaC;          /* Restart residual tolerance */
 38:   PetscBool restart_fm_rise; /* Restart on F_M residual increase */

 40:   PetscReal andersonBeta; /* Relaxation parameter for Anderson Mixing */

 42:   /* Least squares minimization solve context */
 43:   PetscScalar *q;     /* the matrix formed as q_ij = (rdot_i, rdot_j) */
 44:   PetscBLASInt m;     /* matrix dimension */
 45:   PetscBLASInt n;     /* matrix dimension */
 46:   PetscBLASInt nrhs;  /* the number of right-hand sides */
 47:   PetscBLASInt lda;   /* the padded matrix dimension */
 48:   PetscBLASInt ldb;   /* the padded vector dimension */
 49:   PetscReal   *s;     /* the singular values */
 50:   PetscReal    rcond; /* the exit condition */
 51:   PetscBLASInt rank;  /* the effective rank */
 52:   PetscScalar *work;  /* the work vector */
 53:   PetscReal   *rwork; /* the real work vector used for complex */
 54:   PetscBLASInt lwork; /* the size of the work vector */
 55:   PetscBLASInt info;  /* the output condition */

 57:   PetscBool setup_called; /* indicates whether SNESSetUp_NGMRES() has been called  */
 58: } SNES_NGMRES;

 60: #define H(i, j) ngmres->h[i * ngmres->msize + j]
 61: #define Q(i, j) ngmres->q[i * ngmres->msize + j]

 63: /* private functions that are shared components of the methods */
 64: PETSC_INTERN PetscErrorCode SNESNGMRESUpdateSubspace_Private(SNES, PetscInt, PetscInt, Vec, PetscReal, Vec);
 65: PETSC_INTERN PetscErrorCode SNESNGMRESFormCombinedSolution_Private(SNES, PetscInt, PetscInt, Vec, Vec, PetscReal, Vec, Vec, Vec);
 66: PETSC_INTERN PetscErrorCode SNESNGMRESNorms_Private(SNES, PetscInt, Vec, Vec, Vec, Vec, Vec, Vec, Vec, PetscReal *, PetscReal *, PetscReal *, PetscReal *, PetscReal *, PetscReal *, PetscReal *, PetscReal *);
 67: PETSC_INTERN PetscErrorCode SNESNGMRESSelect_Private(SNES, PetscInt, Vec, Vec, PetscReal, PetscReal, PetscReal, PetscReal, Vec, Vec, PetscReal, PetscReal, PetscReal, PetscReal, PetscReal, PetscReal, PetscReal, Vec, Vec, Vec, PetscReal *, PetscReal *, PetscReal *);
 68: PETSC_INTERN PetscErrorCode SNESNGMRESSelectRestart_Private(SNES, PetscInt, PetscReal, PetscReal, PetscReal, PetscReal, PetscReal, PetscReal, PetscBool *);

 70: PETSC_INTERN PetscErrorCode SNESNGMRESGetAdditiveLineSearch_Private(SNES, SNESLineSearch *);
 71: PETSC_INTERN PetscErrorCode SNESDestroy_NGMRES(SNES);
 72: PETSC_INTERN PetscErrorCode SNESReset_NGMRES(SNES);
 73: PETSC_INTERN PetscErrorCode SNESSetUp_NGMRES(SNES);
 74: PETSC_INTERN PetscErrorCode SNESView_NGMRES(SNES, PetscViewer);