Date: 2024-09-25T09:33:55+00:00

Actual source code: ex21.c
slepc-3.21.2 2024-09-25
Report Typos and Errors

      1: /*
      2:    - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
      3:    SLEPc - Scalable Library for Eigenvalue Problem Computations
      4:    Copyright (c) 2002-, Universitat Politecnica de Valencia, Spain

      6:    This file is part of SLEPc.
      7:    SLEPc is distributed under a 2-clause BSD license (see LICENSE).
      8:    - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
      9: */

     11: static char help[] = "Simple 1-D nonlinear eigenproblem (matrix-free version).\n\n"
     12:   "The command line options are:\n"
     13:   "  -n <n>, where <n> = number of grid subdivisions\n\n";

     15: /*
     16:    Solve 1-D PDE
     17:             -u'' = lambda*u
     18:    on [0,1] subject to
     19:             u(0)=0, u'(1)=u(1)*lambda*kappa/(kappa-lambda)
     20: */

     22: #include <slepcnep.h>

     24: /*
     25:    User-defined routines
     26: */
     27: PetscErrorCode FormFunction(NEP,PetscScalar,Mat,Mat,void*);
     28: PetscErrorCode FormJacobian(NEP,PetscScalar,Mat,void*);

     30: /*
     31:    Matrix operations and context
     32: */
     33: PetscErrorCode MatMult_Fun(Mat,Vec,Vec);
     34: PetscErrorCode MatGetDiagonal_Fun(Mat,Vec);
     35: PetscErrorCode MatDestroy_Fun(Mat);
     36: PetscErrorCode MatDuplicate_Fun(Mat,MatDuplicateOption,Mat*);
     37: PetscErrorCode MatMult_Jac(Mat,Vec,Vec);
     38: PetscErrorCode MatGetDiagonal_Jac(Mat,Vec);
     39: PetscErrorCode MatDestroy_Jac(Mat);

     41: typedef struct {
     42:   PetscScalar lambda,kappa;
     43:   PetscReal   h;
     44:   PetscMPIInt next,prev;
     45: } MatCtx;

     47: /*
     48:    User-defined application context
     49: */
     50: typedef struct {
     51:   PetscScalar kappa;   /* ratio between stiffness of spring and attached mass */
     52:   PetscReal   h;       /* mesh spacing */
     53: } ApplicationCtx;

     55: int main(int argc,char **argv)
     56: {
     57:   NEP            nep;             /* nonlinear eigensolver context */
     58:   Mat            F,J;             /* Function and Jacobian matrices */
     59:   ApplicationCtx ctx;             /* user-defined context */
     60:   MatCtx         *ctxF,*ctxJ;     /* contexts for shell matrices */
     61:   PetscInt       n=128,nev;
     62:   KSP            ksp;
     63:   PC             pc;
     64:   PetscMPIInt    rank,size;
     65:   PetscBool      terse;

     67:   PetscFunctionBeginUser;
     68:   PetscCall(SlepcInitialize(&argc,&argv,(char*)0,help));
     69:   PetscCallMPI(MPI_Comm_size(PETSC_COMM_WORLD,&size));
     70:   PetscCallMPI(MPI_Comm_rank(PETSC_COMM_WORLD,&rank));
     71:   PetscCall(PetscOptionsGetInt(NULL,NULL,"-n",&n,NULL));
     72:   PetscCall(PetscPrintf(PETSC_COMM_WORLD,"\n1-D Nonlinear Eigenproblem, n=%" PetscInt_FMT "\n\n",n));
     73:   ctx.h = 1.0/(PetscReal)n;
     74:   ctx.kappa = 1.0;

     76:   /* - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
     77:      Create nonlinear eigensolver context
     78:      - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - */

     80:   PetscCall(NEPCreate(PETSC_COMM_WORLD,&nep));

     82:   /* - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
     83:      Create matrix data structure; set Function evaluation routine
     84:      - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - */

     86:   PetscCall(PetscNew(&ctxF));
     87:   ctxF->h = ctx.h;
     88:   ctxF->kappa = ctx.kappa;
     89:   ctxF->next = rank==size-1? MPI_PROC_NULL: rank+1;
     90:   ctxF->prev = rank==0? MPI_PROC_NULL: rank-1;

     92:   PetscCall(MatCreateShell(PETSC_COMM_WORLD,PETSC_DECIDE,PETSC_DECIDE,n,n,(void*)ctxF,&F));
     93:   PetscCall(MatShellSetOperation(F,MATOP_MULT,(void(*)(void))MatMult_Fun));
     94:   PetscCall(MatShellSetOperation(F,MATOP_GET_DIAGONAL,(void(*)(void))MatGetDiagonal_Fun));
     95:   PetscCall(MatShellSetOperation(F,MATOP_DESTROY,(void(*)(void))MatDestroy_Fun));
     96:   PetscCall(MatShellSetOperation(F,MATOP_DUPLICATE,(void(*)(void))MatDuplicate_Fun));

     98:   /*
     99:      Set Function matrix data structure and default Function evaluation
    100:      routine
    101:   */
    102:   PetscCall(NEPSetFunction(nep,F,F,FormFunction,NULL));

    104:   /* - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
    105:      Create matrix data structure; set Jacobian evaluation routine
    106:      - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - */

    108:   PetscCall(PetscNew(&ctxJ));
    109:   ctxJ->h = ctx.h;
    110:   ctxJ->kappa = ctx.kappa;
    111:   ctxJ->next = rank==size-1? MPI_PROC_NULL: rank+1;
    112:   ctxJ->prev = rank==0? MPI_PROC_NULL: rank-1;

    114:   PetscCall(MatCreateShell(PETSC_COMM_WORLD,PETSC_DECIDE,PETSC_DECIDE,n,n,(void*)ctxJ,&J));
    115:   PetscCall(MatShellSetOperation(J,MATOP_MULT,(void(*)(void))MatMult_Jac));
    116:   PetscCall(MatShellSetOperation(J,MATOP_GET_DIAGONAL,(void(*)(void))MatGetDiagonal_Jac));
    117:   PetscCall(MatShellSetOperation(J,MATOP_DESTROY,(void(*)(void))MatDestroy_Jac));

    119:   /*
    120:      Set Jacobian matrix data structure and default Jacobian evaluation
    121:      routine
    122:   */
    123:   PetscCall(NEPSetJacobian(nep,J,FormJacobian,NULL));

    125:   /* - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
    126:      Customize nonlinear solver; set runtime options
    127:      - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - */

    129:   PetscCall(NEPSetType(nep,NEPRII));
    130:   PetscCall(NEPRIISetLagPreconditioner(nep,0));
    131:   PetscCall(NEPRIIGetKSP(nep,&ksp));
    132:   PetscCall(KSPSetType(ksp,KSPBCGS));
    133:   PetscCall(KSPGetPC(ksp,&pc));
    134:   PetscCall(PCSetType(pc,PCJACOBI));

    136:   /*
    137:      Set solver parameters at runtime
    138:   */
    139:   PetscCall(NEPSetFromOptions(nep));

    141:   /* - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
    142:                       Solve the eigensystem
    143:      - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - */

    145:   PetscCall(NEPSolve(nep));
    146:   PetscCall(NEPGetDimensions(nep,&nev,NULL,NULL));
    147:   PetscCall(PetscPrintf(PETSC_COMM_WORLD," Number of requested eigenvalues: %" PetscInt_FMT "\n",nev));

    149:   /* - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
    150:                     Display solution and clean up
    151:      - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - */

    153:   /* show detailed info unless -terse option is given by user */
    154:   PetscCall(PetscOptionsHasName(NULL,NULL,"-terse",&terse));
    155:   if (terse) PetscCall(NEPErrorView(nep,NEP_ERROR_RELATIVE,NULL));
    156:   else {
    157:     PetscCall(PetscViewerPushFormat(PETSC_VIEWER_STDOUT_WORLD,PETSC_VIEWER_ASCII_INFO_DETAIL));
    158:     PetscCall(NEPConvergedReasonView(nep,PETSC_VIEWER_STDOUT_WORLD));
    159:     PetscCall(NEPErrorView(nep,NEP_ERROR_RELATIVE,PETSC_VIEWER_STDOUT_WORLD));
    160:     PetscCall(PetscViewerPopFormat(PETSC_VIEWER_STDOUT_WORLD));
    161:   }
    162:   PetscCall(NEPDestroy(&nep));
    163:   PetscCall(MatDestroy(&F));
    164:   PetscCall(MatDestroy(&J));
    165:   PetscCall(SlepcFinalize());
    166:   return 0;
    167: }

    169: /* ------------------------------------------------------------------- */
    170: /*
    171:    FormFunction - Computes Function matrix  T(lambda)

    173:    Input Parameters:
    174: .  nep    - the NEP context
    175: .  lambda - real part of the scalar argument
    176: .  ctx    - optional user-defined context, as set by NEPSetFunction()

    178:    Output Parameters:
    179: .  fun - Function matrix
    180: .  B   - optionally different preconditioning matrix
    181: */
    182: PetscErrorCode FormFunction(NEP nep,PetscScalar lambda,Mat fun,Mat B,void *ctx)
    183: {
    184:   MatCtx         *ctxF;

    186:   PetscFunctionBeginUser;
    187:   PetscCall(MatShellGetContext(fun,&ctxF));
    188:   ctxF->lambda = lambda;
    189:   PetscFunctionReturn(PETSC_SUCCESS);
    190: }

    192: /* ------------------------------------------------------------------- */
    193: /*
    194:    FormJacobian - Computes Jacobian matrix  T'(lambda)

    196:    Input Parameters:
    197: .  nep    - the NEP context
    198: .  lambda - real part of the scalar argument
    199: .  ctx    - optional user-defined context, as set by NEPSetJacobian()

    201:    Output Parameters:
    202: .  jac - Jacobian matrix
    203: */
    204: PetscErrorCode FormJacobian(NEP nep,PetscScalar lambda,Mat jac,void *ctx)
    205: {
    206:   MatCtx         *ctxJ;

    208:   PetscFunctionBeginUser;
    209:   PetscCall(MatShellGetContext(jac,&ctxJ));
    210:   ctxJ->lambda = lambda;
    211:   PetscFunctionReturn(PETSC_SUCCESS);
    212: }

    214: /* ------------------------------------------------------------------- */
    215: PetscErrorCode MatMult_Fun(Mat A,Vec x,Vec y)
    216: {
    217:   MatCtx            *ctx;
    218:   PetscInt          i,n,N;
    219:   const PetscScalar *px;
    220:   PetscScalar       *py,c,d,de,oe,upper=0.0,lower=0.0;
    221:   PetscReal         h;
    222:   MPI_Comm          comm;

    224:   PetscFunctionBeginUser;
    225:   PetscCall(MatShellGetContext(A,&ctx));
    226:   PetscCall(VecGetArrayRead(x,&px));
    227:   PetscCall(VecGetArray(y,&py));
    228:   PetscCall(VecGetSize(x,&N));
    229:   PetscCall(VecGetLocalSize(x,&n));

    231:   PetscCall(PetscObjectGetComm((PetscObject)A,&comm));
    232:   PetscCallMPI(MPI_Sendrecv(px,1,MPIU_SCALAR,ctx->prev,0,&lower,1,MPIU_SCALAR,ctx->next,0,comm,MPI_STATUS_IGNORE));
    233:   PetscCallMPI(MPI_Sendrecv(px+n-1,1,MPIU_SCALAR,ctx->next,0,&upper,1,MPIU_SCALAR,ctx->prev,0,comm,MPI_STATUS_IGNORE));

    235:   h = ctx->h;
    236:   c = ctx->kappa/(ctx->lambda-ctx->kappa);
    237:   d = N;
    238:   de = 2.0*(d-ctx->lambda*h/3.0);   /* diagonal entry */
    239:   oe = -d-ctx->lambda*h/6.0;        /* offdiagonal entry */
    240:   py[0] = oe*upper + de*px[0] + oe*px[1];
    241:   for (i=1;i<n-1;i++) py[i] = oe*px[i-1] +de*px[i] + oe*px[i+1];
    242:   if (ctx->next==MPI_PROC_NULL) de = d-ctx->lambda*h/3.0+ctx->lambda*c;   /* diagonal entry of last row */
    243:   py[n-1] = oe*px[n-2] + de*px[n-1] + oe*lower;

    245:   PetscCall(VecRestoreArrayRead(x,&px));
    246:   PetscCall(VecRestoreArray(y,&py));
    247:   PetscFunctionReturn(PETSC_SUCCESS);
    248: }

    250: /* ------------------------------------------------------------------- */
    251: PetscErrorCode MatGetDiagonal_Fun(Mat A,Vec diag)
    252: {
    253:   MatCtx         *ctx;
    254:   PetscInt       n,N;
    255:   PetscScalar    *pd,c,d;
    256:   PetscReal      h;

    258:   PetscFunctionBeginUser;
    259:   PetscCall(MatShellGetContext(A,&ctx));
    260:   PetscCall(VecGetSize(diag,&N));
    261:   PetscCall(VecGetLocalSize(diag,&n));
    262:   h = ctx->h;
    263:   c = ctx->kappa/(ctx->lambda-ctx->kappa);
    264:   d = N;
    265:   PetscCall(VecSet(diag,2.0*(d-ctx->lambda*h/3.0)));
    266:   PetscCall(VecGetArray(diag,&pd));
    267:   pd[n-1] = d-ctx->lambda*h/3.0+ctx->lambda*c;
    268:   PetscCall(VecRestoreArray(diag,&pd));
    269:   PetscFunctionReturn(PETSC_SUCCESS);
    270: }

    272: /* ------------------------------------------------------------------- */
    273: PetscErrorCode MatDestroy_Fun(Mat A)
    274: {
    275:   MatCtx         *ctx;

    277:   PetscFunctionBegin;
    278:   PetscCall(MatShellGetContext(A,&ctx));
    279:   PetscCall(PetscFree(ctx));
    280:   PetscFunctionReturn(PETSC_SUCCESS);
    281: }

    283: /* ------------------------------------------------------------------- */
    284: PetscErrorCode MatDuplicate_Fun(Mat A,MatDuplicateOption op,Mat *B)
    285: {
    286:   MatCtx         *actx,*bctx;
    287:   PetscInt       m,n,M,N;
    288:   MPI_Comm       comm;

    290:   PetscFunctionBegin;
    291:   PetscCall(MatShellGetContext(A,&actx));
    292:   PetscCall(MatGetSize(A,&M,&N));
    293:   PetscCall(MatGetLocalSize(A,&m,&n));

    295:   PetscCall(PetscNew(&bctx));
    296:   bctx->h      = actx->h;
    297:   bctx->kappa  = actx->kappa;
    298:   bctx->lambda = actx->lambda;
    299:   bctx->next   = actx->next;
    300:   bctx->prev   = actx->prev;

    302:   PetscCall(PetscObjectGetComm((PetscObject)A,&comm));
    303:   PetscCall(MatCreateShell(comm,m,n,M,N,(void*)bctx,B));
    304:   PetscCall(MatShellSetOperation(*B,MATOP_MULT,(void(*)(void))MatMult_Fun));
    305:   PetscCall(MatShellSetOperation(*B,MATOP_GET_DIAGONAL,(void(*)(void))MatGetDiagonal_Fun));
    306:   PetscCall(MatShellSetOperation(*B,MATOP_DESTROY,(void(*)(void))MatDestroy_Fun));
    307:   PetscCall(MatShellSetOperation(*B,MATOP_DUPLICATE,(void(*)(void))MatDuplicate_Fun));
    308:   PetscFunctionReturn(PETSC_SUCCESS);
    309: }

    311: /* ------------------------------------------------------------------- */
    312: PetscErrorCode MatMult_Jac(Mat A,Vec x,Vec y)
    313: {
    314:   MatCtx            *ctx;
    315:   PetscInt          i,n;
    316:   const PetscScalar *px;
    317:   PetscScalar       *py,c,de,oe,upper=0.0,lower=0.0;
    318:   PetscReal         h;
    319:   MPI_Comm          comm;

    321:   PetscFunctionBeginUser;
    322:   PetscCall(MatShellGetContext(A,&ctx));
    323:   PetscCall(VecGetArrayRead(x,&px));
    324:   PetscCall(VecGetArray(y,&py));
    325:   PetscCall(VecGetLocalSize(x,&n));

    327:   PetscCall(PetscObjectGetComm((PetscObject)A,&comm));
    328:   PetscCallMPI(MPI_Sendrecv(px,1,MPIU_SCALAR,ctx->prev,0,&lower,1,MPIU_SCALAR,ctx->next,0,comm,MPI_STATUS_IGNORE));
    329:   PetscCallMPI(MPI_Sendrecv(px+n-1,1,MPIU_SCALAR,ctx->next,0,&upper,1,MPIU_SCALAR,ctx->prev,0,comm,MPI_STATUS_IGNORE));

    331:   h = ctx->h;
    332:   c = ctx->kappa/(ctx->lambda-ctx->kappa);
    333:   de = -2.0*h/3.0;    /* diagonal entry */
    334:   oe = -h/6.0;        /* offdiagonal entry */
    335:   py[0] = oe*upper + de*px[0] + oe*px[1];
    336:   for (i=1;i<n-1;i++) py[i] = oe*px[i-1] +de*px[i] + oe*px[i+1];
    337:   if (ctx->next==MPI_PROC_NULL) de = -h/3.0-c*c;    /* diagonal entry of last row */
    338:   py[n-1] = oe*px[n-2] + de*px[n-1] + oe*lower;

    340:   PetscCall(VecRestoreArrayRead(x,&px));
    341:   PetscCall(VecRestoreArray(y,&py));
    342:   PetscFunctionReturn(PETSC_SUCCESS);
    343: }

    345: /* ------------------------------------------------------------------- */
    346: PetscErrorCode MatGetDiagonal_Jac(Mat A,Vec diag)
    347: {
    348:   MatCtx         *ctx;
    349:   PetscInt       n;
    350:   PetscScalar    *pd,c;
    351:   PetscReal      h;

    353:   PetscFunctionBeginUser;
    354:   PetscCall(MatShellGetContext(A,&ctx));
    355:   PetscCall(VecGetLocalSize(diag,&n));
    356:   h = ctx->h;
    357:   c = ctx->kappa/(ctx->lambda-ctx->kappa);
    358:   PetscCall(VecSet(diag,-2.0*h/3.0));
    359:   PetscCall(VecGetArray(diag,&pd));
    360:   pd[n-1] = -h/3.0-c*c;
    361:   PetscCall(VecRestoreArray(diag,&pd));
    362:   PetscFunctionReturn(PETSC_SUCCESS);
    363: }

    365: /* ------------------------------------------------------------------- */
    366: PetscErrorCode MatDestroy_Jac(Mat A)
    367: {
    368:   MatCtx         *ctx;

    370:   PetscFunctionBegin;
    371:   PetscCall(MatShellGetContext(A,&ctx));
    372:   PetscCall(PetscFree(ctx));
    373:   PetscFunctionReturn(PETSC_SUCCESS);
    374: }

    376: /*TEST

    378:    testset:
    379:       nsize: {{1 2}}
    380:       args: -terse
    381:       requires: !single
    382:       output_file: output/ex21_1.out
    383:       filter: sed -e "s/[+-]0\.0*i//g" -e "s/+0i//g"
    384:       test:
    385:          suffix: 1_rii
    386:          args: -nep_type rii -nep_target 4
    387:       test:
    388:          suffix: 1_slp
    389:          args: -nep_type slp -nep_slp_pc_type jacobi -nep_slp_ksp_type bcgs -nep_target 10

    391: TEST*/