| Line | Branch | Exec | Source |
|---|---|---|---|
| 1 | /* | ||
| 2 | - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - | ||
| 3 | SLEPc - Scalable Library for Eigenvalue Problem Computations | ||
| 4 | Copyright (c) 2002-, Universitat Politecnica de Valencia, Spain | ||
| 5 | |||
| 6 | This file is part of SLEPc. | ||
| 7 | SLEPc is distributed under a 2-clause BSD license (see LICENSE). | ||
| 8 | - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - | ||
| 9 | */ | ||
| 10 | |||
| 11 | #pragma once | ||
| 12 | |||
| 13 | #include <slepcst.h> | ||
| 14 | #include <slepc/private/slepcimpl.h> | ||
| 15 | |||
| 16 | /* SUBMANSEC = ST */ | ||
| 17 | |||
| 18 | SLEPC_EXTERN PetscBool STRegisterAllCalled; | ||
| 19 | SLEPC_EXTERN PetscErrorCode STRegisterAll(void); | ||
| 20 | SLEPC_EXTERN PetscLogEvent ST_SetUp,ST_ComputeOperator,ST_Apply,ST_ApplyTranspose,ST_ApplyHermitianTranspose,ST_MatSetUp,ST_MatMult,ST_MatMultTranspose,ST_MatSolve,ST_MatSolveTranspose; | ||
| 21 | |||
| 22 | typedef struct _STOps *STOps; | ||
| 23 | |||
| 24 | struct _STOps { | ||
| 25 | PetscErrorCode (*apply)(ST,Vec,Vec); | ||
| 26 | PetscErrorCode (*applymat)(ST,Mat,Mat); | ||
| 27 | PetscErrorCode (*applytrans)(ST,Vec,Vec); | ||
| 28 | PetscErrorCode (*applyhermtrans)(ST,Vec,Vec); | ||
| 29 | PetscErrorCode (*backtransform)(ST,PetscInt,PetscScalar*,PetscScalar*); | ||
| 30 | PetscErrorCode (*setshift)(ST,PetscScalar); | ||
| 31 | PetscErrorCode (*getbilinearform)(ST,Mat*); | ||
| 32 | PetscErrorCode (*setup)(ST); | ||
| 33 | PetscErrorCode (*computeoperator)(ST); | ||
| 34 | PetscErrorCode (*setfromoptions)(ST,PetscOptionItems); | ||
| 35 | PetscErrorCode (*postsolve)(ST); | ||
| 36 | PetscErrorCode (*destroy)(ST); | ||
| 37 | PetscErrorCode (*reset)(ST); | ||
| 38 | PetscErrorCode (*view)(ST,PetscViewer); | ||
| 39 | PetscErrorCode (*checknullspace)(ST,BV); | ||
| 40 | PetscErrorCode (*setdefaultksp)(ST); | ||
| 41 | }; | ||
| 42 | |||
| 43 | /* | ||
| 44 | 'Updated' state means STSetUp must be called because matrices have been | ||
| 45 | modified, but the pattern is the same (hence reuse symbolic factorization) | ||
| 46 | */ | ||
| 47 | typedef enum { ST_STATE_INITIAL, | ||
| 48 | ST_STATE_SETUP, | ||
| 49 | ST_STATE_UPDATED } STStateType; | ||
| 50 | |||
| 51 | struct _p_ST { | ||
| 52 | PETSCHEADER(struct _STOps); | ||
| 53 | /*------------------------- User parameters --------------------------*/ | ||
| 54 | Mat *A; /* matrices that define the eigensystem */ | ||
| 55 | PetscInt nmat; /* number of user-provided matrices */ | ||
| 56 | PetscScalar sigma; /* value of the shift */ | ||
| 57 | PetscScalar defsigma; /* default value of the shift */ | ||
| 58 | STMatMode matmode; /* how the transformation matrix is handled */ | ||
| 59 | MatStructure str; /* whether matrices have the same pattern or not */ | ||
| 60 | PetscBool transform; /* whether transformed matrices are computed */ | ||
| 61 | PetscBool structured; /* whether the operator is structured or not */ | ||
| 62 | Vec D; /* diagonal matrix for balancing */ | ||
| 63 | Mat Pmat; /* user-provided preconditioner matrix */ | ||
| 64 | PetscBool Pmat_set; /* whether the user provided a preconditioner matrix or not */ | ||
| 65 | Mat *Psplit; /* matrices for the split preconditioner */ | ||
| 66 | PetscInt nsplit; /* number of split preconditioner matrices */ | ||
| 67 | MatStructure strp; /* pattern of split preconditioner matrices */ | ||
| 68 | |||
| 69 | /*------------------------- Misc data --------------------------*/ | ||
| 70 | KSP ksp; /* linear solver used in some ST's */ | ||
| 71 | PetscBool usesksp; /* whether the KSP object is used or not */ | ||
| 72 | PetscInt nwork; /* number of work vectors */ | ||
| 73 | Vec *work; /* work vectors */ | ||
| 74 | Vec wb; /* balancing requires an extra work vector */ | ||
| 75 | Vec wht; /* extra work vector for hermitian transpose apply */ | ||
| 76 | STStateType state; /* initial -> setup -> with updated matrices */ | ||
| 77 | PetscObjectState *Astate; /* matrix state (to identify the original matrices) */ | ||
| 78 | Mat *T; /* matrices resulting from transformation */ | ||
| 79 | Mat Op; /* shell matrix for operator = alpha*D*inv(P)*M*inv(D) */ | ||
| 80 | PetscBool opseized; /* whether Op has been seized by user */ | ||
| 81 | PetscBool opready; /* whether Op is up-to-date or need be computed */ | ||
| 82 | Mat P; /* matrix from which preconditioner is built */ | ||
| 83 | Mat M; /* matrix corresponding to the non-inverted part of the operator */ | ||
| 84 | PetscBool sigma_set; /* whether the user provided the shift or not */ | ||
| 85 | PetscBool asymm; /* the user matrices are all symmetric */ | ||
| 86 | PetscBool aherm; /* the user matrices are all hermitian */ | ||
| 87 | void *data; | ||
| 88 | }; | ||
| 89 | |||
| 90 | /* | ||
| 91 | Macros to test valid ST arguments | ||
| 92 | */ | ||
| 93 | #if !defined(PETSC_USE_DEBUG) | ||
| 94 | |||
| 95 | #define STCheckMatrices(h,arg) do {(void)(h);} while (0) | ||
| 96 | #define STCheckNotSeized(h,arg) do {(void)(h);} while (0) | ||
| 97 | |||
| 98 | #else | ||
| 99 | |||
| 100 | #define STCheckMatrices(h,arg) \ | ||
| 101 | do { \ | ||
| 102 | PetscCheck((h)->A,PetscObjectComm((PetscObject)(h)),PETSC_ERR_ARG_WRONGSTATE,"ST matrices have not been set: Parameter #%d",arg); \ | ||
| 103 | } while (0) | ||
| 104 | #define STCheckNotSeized(h,arg) \ | ||
| 105 | do { \ | ||
| 106 | PetscCheck(!(h)->opseized,PetscObjectComm((PetscObject)(h)),PETSC_ERR_ARG_WRONGSTATE,"Must call STRestoreOperator() first: Parameter #%d",arg); \ | ||
| 107 | } while (0) | ||
| 108 | |||
| 109 | #endif | ||
| 110 | |||
| 111 | SLEPC_INTERN PetscErrorCode STGetBilinearForm_Default(ST,Mat*); | ||
| 112 | SLEPC_INTERN PetscErrorCode STCheckNullSpace_Default(ST,BV); | ||
| 113 | SLEPC_INTERN PetscErrorCode STMatShellCreate(ST,PetscScalar,PetscInt,PetscInt*,PetscScalar*,Mat*); | ||
| 114 | SLEPC_INTERN PetscErrorCode STMatShellShift(Mat,PetscScalar); | ||
| 115 | SLEPC_INTERN PetscErrorCode STCheckFactorPackage(ST); | ||
| 116 | SLEPC_INTERN PetscErrorCode STMatMAXPY_Private(ST,PetscScalar,PetscScalar,PetscInt,PetscScalar*,PetscBool,PetscBool,Mat*); | ||
| 117 | SLEPC_INTERN PetscErrorCode STCoeffs_Monomial(ST,PetscScalar*); | ||
| 118 | SLEPC_INTERN PetscErrorCode STSetDefaultKSP(ST); | ||
| 119 | SLEPC_INTERN PetscErrorCode STSetDefaultKSP_Default(ST); | ||
| 120 | SLEPC_INTERN PetscErrorCode STIsInjective_Shell(ST,PetscBool*); | ||
| 121 | SLEPC_INTERN PetscErrorCode STComputeOperator(ST); | ||
| 122 | SLEPC_INTERN PetscErrorCode STGetOperator_Private(ST,Mat*); | ||
| 123 | SLEPC_INTERN PetscErrorCode STApply_Generic(ST,Vec,Vec); | ||
| 124 | SLEPC_INTERN PetscErrorCode STApplyMat_Generic(ST,Mat,Mat); | ||
| 125 | SLEPC_INTERN PetscErrorCode STApplyTranspose_Generic(ST,Vec,Vec); | ||
| 126 | SLEPC_INTERN PetscErrorCode STApplyHermitianTranspose_Generic(ST,Vec,Vec); | ||
| 127 | |||
| 128 | /* | ||
| 129 | ST_KSPSetOperators - Sets the KSP matrices | ||
| 130 | */ | ||
| 131 | 12538 | static inline PetscErrorCode ST_KSPSetOperators(ST st,Mat A,Mat B) | |
| 132 | { | ||
| 133 | 12538 | const char *prefix; | |
| 134 | |||
| 135 |
1/2✓ Branch 0 taken 10 times.
✗ Branch 1 not taken.
|
12538 | PetscFunctionBegin; |
| 136 |
6/8✓ Branch 0 taken 20 times.
✓ Branch 1 taken 50 times.
✓ Branch 2 taken 4 times.
✓ Branch 3 taken 16 times.
✓ Branch 4 taken 4 times.
✗ Branch 5 not taken.
✗ Branch 6 not taken.
✓ Branch 7 taken 4 times.
|
12538 | if (!st->ksp) PetscCall(STGetKSP(st,&st->ksp)); |
| 137 |
4/6✓ Branch 0 taken 10 times.
✓ Branch 1 taken 40 times.
✓ Branch 2 taken 10 times.
✗ Branch 3 not taken.
✗ Branch 4 not taken.
✓ Branch 5 taken 10 times.
|
12538 | PetscCall(STCheckFactorPackage(st)); |
| 138 |
4/6✓ Branch 0 taken 10 times.
✓ Branch 1 taken 40 times.
✓ Branch 2 taken 10 times.
✗ Branch 3 not taken.
✗ Branch 4 not taken.
✓ Branch 5 taken 10 times.
|
12538 | PetscCall(KSPSetOperators(st->ksp,A,B)); |
| 139 |
4/6✓ Branch 0 taken 10 times.
✓ Branch 1 taken 40 times.
✓ Branch 2 taken 10 times.
✗ Branch 3 not taken.
✗ Branch 4 not taken.
✓ Branch 5 taken 10 times.
|
12538 | PetscCall(MatGetOptionsPrefix(B,&prefix)); |
| 140 |
2/2✓ Branch 0 taken 50 times.
✓ Branch 1 taken 40 times.
|
12538 | if (!prefix) { |
| 141 | /* set Mat prefix to be the same as KSP to enable setting command-line options (e.g. MUMPS) | ||
| 142 | only applies if the Mat has no user-defined prefix */ | ||
| 143 |
4/6✓ Branch 0 taken 10 times.
✓ Branch 1 taken 40 times.
✓ Branch 2 taken 10 times.
✗ Branch 3 not taken.
✗ Branch 4 not taken.
✓ Branch 5 taken 10 times.
|
5625 | PetscCall(KSPGetOptionsPrefix(st->ksp,&prefix)); |
| 144 |
4/6✓ Branch 0 taken 10 times.
✓ Branch 1 taken 40 times.
✓ Branch 2 taken 10 times.
✗ Branch 3 not taken.
✗ Branch 4 not taken.
✓ Branch 5 taken 10 times.
|
5625 | PetscCall(MatSetOptionsPrefix(B,prefix)); |
| 145 | } | ||
| 146 |
6/12✓ Branch 0 taken 10 times.
✗ Branch 1 not taken.
✗ Branch 2 not taken.
✓ Branch 3 taken 10 times.
✓ Branch 4 taken 10 times.
✗ Branch 5 not taken.
✓ Branch 6 taken 10 times.
✗ Branch 7 not taken.
✓ Branch 8 taken 10 times.
✗ Branch 9 not taken.
✗ Branch 10 not taken.
✓ Branch 11 taken 10 times.
|
2509 | PetscFunctionReturn(PETSC_SUCCESS); |
| 147 | } | ||
| 148 |