Line data    Source code

       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             :    SLEPc eigensolver: "davidson"
      12             : 
      13             :    Some utils
      14             : */
      15             : 
      16             : #include "davidson.h"
      17             : 
      18             : typedef struct {
      19             :   PC pc;
      20             : } dvdPCWrapper;
      21             : 
      22             : /*
      23             :   Configure the harmonics.
      24             :   switch (mode) {
      25             :   DVD_HARM_RR:    harmonic RR
      26             :   DVD_HARM_RRR:   relative harmonic RR
      27             :   DVD_HARM_REIGS: rightmost eigenvalues
      28             :   DVD_HARM_LEIGS: largest eigenvalues
      29             :   }
      30             :   fixedTarged, if true use the target instead of the best eigenvalue
      31             :   target, the fixed target to be used
      32             : */
      33             : typedef struct {
      34             :   PetscScalar Wa,Wb;       /* span{W} = span{Wa*AV - Wb*BV} */
      35             :   PetscScalar Pa,Pb;       /* H=W'*(Pa*AV - Pb*BV), G=W'*(Wa*AV - Wb*BV) */
      36             :   PetscBool   withTarget;
      37             :   HarmType_t  mode;
      38             : } dvdHarmonic;
      39             : 
      40             : typedef struct {
      41             :   Vec diagA, diagB;
      42             : } dvdJacobiPrecond;
      43             : 
      44          95 : static PetscErrorCode dvd_improvex_precond_d(dvdDashboard *d)
      45             : {
      46          95 :   dvdPCWrapper   *dvdpc = (dvdPCWrapper*)d->improvex_precond_data;
      47             : 
      48          95 :   PetscFunctionBegin;
      49             :   /* Free local data */
      50          95 :   PetscCall(PCDestroy(&dvdpc->pc));
      51          95 :   PetscCall(PetscFree(d->improvex_precond_data));
      52          95 :   PetscFunctionReturn(PETSC_SUCCESS);
      53             : }
      54             : 
      55       14176 : static PetscErrorCode dvd_static_precond_PC_0(dvdDashboard *d,PetscInt i,Vec x,Vec Px)
      56             : {
      57       14176 :   dvdPCWrapper   *dvdpc = (dvdPCWrapper*)d->improvex_precond_data;
      58             : 
      59       14176 :   PetscFunctionBegin;
      60       14176 :   PetscCall(PCApply(dvdpc->pc,x,Px));
      61       14176 :   PetscFunctionReturn(PETSC_SUCCESS);
      62             : }
      63             : 
      64             : /*
      65             :   Create a trivial preconditioner
      66             : */
      67        9145 : static PetscErrorCode dvd_precond_none(dvdDashboard *d,PetscInt i,Vec x,Vec Px)
      68             : {
      69        9145 :   PetscFunctionBegin;
      70        9145 :   PetscCall(VecCopy(x,Px));
      71        9145 :   PetscFunctionReturn(PETSC_SUCCESS);
      72             : }
      73             : 
      74             : /*
      75             :   Create a static preconditioner from a PC
      76             : */
      77         285 : PetscErrorCode dvd_static_precond_PC(dvdDashboard *d,dvdBlackboard *b,PC pc)
      78             : {
      79         285 :   dvdPCWrapper   *dvdpc;
      80         285 :   Mat            P;
      81         285 :   PetscBool      t0,t1,t2;
      82             : 
      83         285 :   PetscFunctionBegin;
      84             :   /* Setup the step */
      85         285 :   if (b->state >= DVD_STATE_CONF) {
      86             :     /* If the preconditioner is valid */
      87          95 :     if (pc) {
      88          95 :       PetscCall(PetscNew(&dvdpc));
      89          95 :       dvdpc->pc = pc;
      90          95 :       PetscCall(PetscObjectReference((PetscObject)pc));
      91          95 :       d->improvex_precond_data = dvdpc;
      92          95 :       d->improvex_precond = dvd_static_precond_PC_0;
      93             : 
      94             :       /* PC saves the matrix associated with the linear system, and it has to
      95             :          be initialize to a valid matrix */
      96          95 :       PetscCall(PCGetOperatorsSet(pc,NULL,&t0));
      97          95 :       PetscCall(PetscObjectTypeCompare((PetscObject)pc,PCNONE,&t1));
      98          95 :       PetscCall(PetscObjectTypeCompare((PetscObject)pc,PCSHELL,&t2));
      99          95 :       if (t0 && !t1) {
     100          41 :         PetscCall(PCGetOperators(pc,NULL,&P));
     101          41 :         PetscCall(PetscObjectReference((PetscObject)P));
     102          41 :         PetscCall(PCSetOperators(pc,P,P));
     103          41 :         PetscCall(PCSetReusePreconditioner(pc,PETSC_TRUE));
     104          41 :         PetscCall(MatDestroy(&P));
     105          54 :       } else if (t2) {
     106           0 :         PetscCall(PCSetOperators(pc,d->A,d->A));
     107           0 :         PetscCall(PCSetReusePreconditioner(pc,PETSC_TRUE));
     108             :       } else {
     109          54 :         d->improvex_precond = dvd_precond_none;
     110             :       }
     111             : 
     112          95 :       PetscCall(EPSDavidsonFLAdd(&d->destroyList,dvd_improvex_precond_d));
     113             : 
     114             :     /* Else, use no preconditioner */
     115           0 :     } else d->improvex_precond = dvd_precond_none;
     116             :   }
     117         285 :   PetscFunctionReturn(PETSC_SUCCESS);
     118             : }
     119             : 
     120          23 : static PetscErrorCode dvd_harm_d(dvdDashboard *d)
     121             : {
     122          23 :   PetscFunctionBegin;
     123             :   /* Free local data */
     124          23 :   PetscCall(PetscFree(d->calcpairs_W_data));
     125          23 :   PetscFunctionReturn(PETSC_SUCCESS);
     126             : }
     127             : 
     128          23 : static PetscErrorCode dvd_harm_transf(dvdHarmonic *dvdh,PetscScalar t)
     129             : {
     130          23 :   PetscFunctionBegin;
     131          23 :   switch (dvdh->mode) {
     132          13 :   case DVD_HARM_RR:    /* harmonic RR */
     133          13 :     dvdh->Wa = 1.0; dvdh->Wb = t;   dvdh->Pa = 0.0; dvdh->Pb = -1.0;
     134          13 :     break;
     135           0 :   case DVD_HARM_RRR:   /* relative harmonic RR */
     136           0 :     dvdh->Wa = 1.0; dvdh->Wb = t;   dvdh->Pa = 1.0; dvdh->Pb = 0.0;
     137           0 :     break;
     138           0 :   case DVD_HARM_REIGS: /* rightmost eigenvalues */
     139           0 :     dvdh->Wa = 1.0; dvdh->Wb = t;   dvdh->Pa = 1.0; dvdh->Pb = -PetscConj(t);
     140           0 :     break;
     141          10 :   case DVD_HARM_LEIGS: /* largest eigenvalues */
     142          10 :     dvdh->Wa = 0.0; dvdh->Wb = 1.0; dvdh->Pa = 1.0; dvdh->Pb = 0.0;
     143          10 :     break;
     144           0 :   case DVD_HARM_NONE:
     145             :   default:
     146           0 :     SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"Harmonic type not supported");
     147             :   }
     148             : 
     149             :   /* Check the transformation does not change the sign of the imaginary part */
     150             : #if !defined(PETSC_USE_COMPLEX)
     151          23 :   if (dvdh->Pb*dvdh->Wa - dvdh->Wb*dvdh->Pa < 0.0) {
     152          23 :     dvdh->Pa *= -1.0;
     153          23 :     dvdh->Pb *= -1.0;
     154             :   }
     155             : #endif
     156          23 :   PetscFunctionReturn(PETSC_SUCCESS);
     157             : }
     158             : 
     159         941 : static PetscErrorCode dvd_harm_updateW(dvdDashboard *d)
     160             : {
     161         941 :   dvdHarmonic    *data = (dvdHarmonic*)d->calcpairs_W_data;
     162         941 :   PetscInt       l,k;
     163         941 :   BV             BX = d->BX?d->BX:d->eps->V;
     164             : 
     165         941 :   PetscFunctionBegin;
     166             :   /* Update the target if it is necessary */
     167         941 :   if (!data->withTarget) PetscCall(dvd_harm_transf(data,d->eigr[0]));
     168             : 
     169             :   /* W(i) <- Wa*AV(i) - Wb*BV(i) */
     170         941 :   PetscCall(BVGetActiveColumns(d->eps->V,&l,&k));
     171         941 :   PetscAssert(k==l+d->V_new_s,PETSC_COMM_SELF,PETSC_ERR_PLIB,"Consistency broken");
     172         941 :   PetscCall(BVSetActiveColumns(d->W,l+d->V_new_s,l+d->V_new_e));
     173         941 :   PetscCall(BVSetActiveColumns(d->AX,l+d->V_new_s,l+d->V_new_e));
     174         941 :   PetscCall(BVSetActiveColumns(BX,l+d->V_new_s,l+d->V_new_e));
     175         941 :   PetscCall(BVCopy(d->AX,d->W));
     176         941 :   PetscCall(BVScale(d->W,data->Wa));
     177         941 :   PetscCall(BVMult(d->W,-data->Wb,1.0,BX,NULL));
     178         941 :   PetscCall(BVSetActiveColumns(d->W,l,k));
     179         941 :   PetscCall(BVSetActiveColumns(d->AX,l,k));
     180         941 :   PetscCall(BVSetActiveColumns(BX,l,k));
     181         941 :   PetscFunctionReturn(PETSC_SUCCESS);
     182             : }
     183             : 
     184         941 : static PetscErrorCode dvd_harm_proj(dvdDashboard *d)
     185             : {
     186         941 :   dvdHarmonic    *data = (dvdHarmonic*)d->calcpairs_W_data;
     187         941 :   PetscInt       i,j,l0,l,k,ld;
     188         941 :   PetscScalar    h,g,*H,*G;
     189             : 
     190         941 :   PetscFunctionBegin;
     191         941 :   PetscCall(BVGetActiveColumns(d->eps->V,&l0,&k));
     192         941 :   l = l0 + d->V_new_s;
     193         941 :   k = l0 + d->V_new_e;
     194         941 :   PetscCall(MatGetSize(d->H,&ld,NULL));
     195         941 :   PetscCall(MatDenseGetArray(d->H,&H));
     196         941 :   PetscCall(MatDenseGetArray(d->G,&G));
     197             :   /* [H G] <- [Pa*H - Pb*G, Wa*H - Wb*G] */
     198             :   /* Right part */
     199        2397 :   for (i=l;i<k;i++) {
     200       18389 :     for (j=l0;j<k;j++) {
     201       16933 :       h = H[ld*i+j];
     202       16933 :       g = G[ld*i+j];
     203       16933 :       H[ld*i+j] = data->Pa*h - data->Pb*g;
     204       16933 :       G[ld*i+j] = data->Wa*h - data->Wb*g;
     205             :     }
     206             :   }
     207             :   /* Left part */
     208       10527 :   for (i=l0;i<l;i++) {
     209       23573 :     for (j=l;j<k;j++) {
     210       13987 :       h = H[ld*i+j];
     211       13987 :       g = G[ld*i+j];
     212       13987 :       H[ld*i+j] = data->Pa*h - data->Pb*g;
     213       13987 :       G[ld*i+j] = data->Wa*h - data->Wb*g;
     214             :     }
     215             :   }
     216         941 :   PetscCall(MatDenseRestoreArray(d->H,&H));
     217         941 :   PetscCall(MatDenseRestoreArray(d->G,&G));
     218         941 :   PetscFunctionReturn(PETSC_SUCCESS);
     219             : }
     220             : 
     221          55 : PetscErrorCode dvd_harm_updateproj(dvdDashboard *d)
     222             : {
     223          55 :   dvdHarmonic    *data = (dvdHarmonic*)d->calcpairs_W_data;
     224          55 :   PetscInt       i,j,l,k,ld;
     225          55 :   PetscScalar    h,g,*H,*G;
     226             : 
     227          55 :   PetscFunctionBegin;
     228          55 :   PetscCall(BVGetActiveColumns(d->eps->V,&l,&k));
     229          55 :   k = l + d->V_tra_s;
     230          55 :   PetscCall(MatGetSize(d->H,&ld,NULL));
     231          55 :   PetscCall(MatDenseGetArray(d->H,&H));
     232          55 :   PetscCall(MatDenseGetArray(d->G,&G));
     233             :   /* [H G] <- [Pa*H - Pb*G, Wa*H - Wb*G] */
     234             :   /* Right part */
     235         112 :   for (i=l;i<k;i++) {
     236         133 :     for (j=0;j<l;j++) {
     237          76 :       h = H[ld*i+j];
     238          76 :       g = G[ld*i+j];
     239          76 :       H[ld*i+j] = data->Pa*h - data->Pb*g;
     240          76 :       G[ld*i+j] = data->Wa*h - data->Wb*g;
     241             :     }
     242             :   }
     243             :   /* Lower triangular part */
     244         129 :   for (i=0;i<l;i++) {
     245         150 :     for (j=l;j<k;j++) {
     246          76 :       h = H[ld*i+j];
     247          76 :       g = G[ld*i+j];
     248          76 :       H[ld*i+j] = data->Pa*h - data->Pb*g;
     249          76 :       G[ld*i+j] = data->Wa*h - data->Wb*g;
     250             :     }
     251             :   }
     252          55 :   PetscCall(MatDenseRestoreArray(d->H,&H));
     253          55 :   PetscCall(MatDenseRestoreArray(d->G,&G));
     254          55 :   PetscFunctionReturn(PETSC_SUCCESS);
     255             : }
     256             : 
     257       36912 : static PetscErrorCode dvd_harm_backtrans(dvdHarmonic *data,PetscScalar *ar,PetscScalar *ai)
     258             : {
     259       36912 :   PetscScalar xr;
     260             : #if !defined(PETSC_USE_COMPLEX)
     261       36912 :   PetscScalar xi, k;
     262             : #endif
     263             : 
     264       36912 :   PetscFunctionBegin;
     265       36912 :   xr = *ar;
     266             : #if !defined(PETSC_USE_COMPLEX)
     267       36912 :   xi = *ai;
     268       36912 :   if (PetscUnlikely(xi != 0.0)) {
     269        2416 :     k = (data->Pa - data->Wa*xr)*(data->Pa - data->Wa*xr) + data->Wa*data->Wa*xi*xi;
     270        2416 :     *ar = (data->Pb*data->Pa - (data->Pb*data->Wa + data->Wb*data->Pa)*xr + data->Wb*data->Wa*(xr*xr + xi*xi))/k;
     271        2416 :     *ai = (data->Pb*data->Wa - data->Wb*data->Pa)*xi/k;
     272             :   } else
     273             : #endif
     274       34496 :     *ar = (data->Pb - data->Wb*xr) / (data->Pa - data->Wa*xr);
     275       36912 :   PetscFunctionReturn(PETSC_SUCCESS);
     276             : }
     277             : 
     278       24627 : static PetscErrorCode dvd_harm_eig_backtrans(dvdDashboard *d,PetscScalar ar,PetscScalar ai,PetscScalar *br,PetscScalar *bi)
     279             : {
     280       24627 :   dvdHarmonic    *data = (dvdHarmonic*)d->calcpairs_W_data;
     281             : 
     282       24627 :   PetscFunctionBegin;
     283       24627 :   PetscCall(dvd_harm_backtrans(data,&ar,&ai));
     284       24627 :   *br = ar;
     285       24627 :   *bi = ai;
     286       24627 :   PetscFunctionReturn(PETSC_SUCCESS);
     287             : }
     288             : 
     289        1112 : static PetscErrorCode dvd_harm_eigs_trans(dvdDashboard *d)
     290             : {
     291        1112 :   dvdHarmonic    *data = (dvdHarmonic*)d->calcpairs_W_data;
     292        1112 :   PetscInt       i,l,k;
     293             : 
     294        1112 :   PetscFunctionBegin;
     295        1112 :   PetscCall(BVGetActiveColumns(d->eps->V,&l,&k));
     296       13397 :   for (i=0;i<k-l;i++) PetscCall(dvd_harm_backtrans(data,&d->eigr[i],&d->eigi[i]));
     297        1112 :   PetscFunctionReturn(PETSC_SUCCESS);
     298             : }
     299             : 
     300          69 : PetscErrorCode dvd_harm_conf(dvdDashboard *d,dvdBlackboard *b,HarmType_t mode,PetscBool fixedTarget,PetscScalar t)
     301             : {
     302          69 :   dvdHarmonic    *dvdh;
     303             : 
     304          69 :   PetscFunctionBegin;
     305             :   /* Set the problem to GNHEP:
     306             :      d->G maybe is upper triangular due to biorthogonality of V and W */
     307          69 :   d->sEP = d->sA = d->sB = 0;
     308             : 
     309             :   /* Setup the step */
     310          69 :   if (b->state >= DVD_STATE_CONF) {
     311          23 :     PetscCall(PetscNew(&dvdh));
     312          23 :     dvdh->withTarget = fixedTarget;
     313          23 :     dvdh->mode = mode;
     314          23 :     if (fixedTarget) PetscCall(dvd_harm_transf(dvdh, t));
     315          23 :     d->calcpairs_W_data = dvdh;
     316          23 :     d->calcpairs_W = dvd_harm_updateW;
     317          23 :     d->calcpairs_proj_trans = dvd_harm_proj;
     318          23 :     d->calcpairs_eigs_trans = dvd_harm_eigs_trans;
     319          23 :     d->calcpairs_eig_backtrans = dvd_harm_eig_backtrans;
     320             : 
     321          23 :     PetscCall(EPSDavidsonFLAdd(&d->destroyList,dvd_harm_d));
     322             :   }
     323          69 :   PetscFunctionReturn(PETSC_SUCCESS);
     324             : }