xref: /petsc/src/ksp/pc/impls/ml/ml.c (revision e0262f4811d840cab04ffcc3e1242610bd23b780)
1ab718edeSHong Zhang 
25582bec1SHong Zhang /*
32dccc152SHong Zhang    Provides an interface to the ML smoothed Aggregation
47ffd031bSHong Zhang    Note: Something non-obvious breaks -pc_mg_type ADDITIVE for parallel runs
57ffd031bSHong Zhang                                     Jed Brown, see [PETSC #18321, #18449].
65582bec1SHong Zhang */
7c6db04a5SJed Brown #include <private/pcimpl.h>   /*I "petscpc.h" I*/
8c6db04a5SJed Brown #include <../src/ksp/pc/impls/mg/mgimpl.h>                    /*I "petscpcmg.h" I*/
9c6db04a5SJed Brown #include <../src/mat/impls/aij/seq/aij.h>
10c6db04a5SJed Brown #include <../src/mat/impls/aij/mpi/mpiaij.h>
11cb5d8e9eSHong Zhang 
125582bec1SHong Zhang #include <math.h>
132cf39c26SSatish Balay EXTERN_C_BEGIN
1468210224SSatish Balay /* HAVE_CONFIG_H flag is required by ML include files */
1568210224SSatish Balay #if !defined(HAVE_CONFIG_H)
1668210224SSatish Balay #define HAVE_CONFIG_H
1768210224SSatish Balay #endif
18c6db04a5SJed Brown #include <ml_include.h>
195582bec1SHong Zhang EXTERN_C_END
205582bec1SHong Zhang 
215582bec1SHong Zhang /* The context (data structure) at each grid level */
225582bec1SHong Zhang typedef struct {
235582bec1SHong Zhang   Vec        x,b,r;           /* global vectors */
245582bec1SHong Zhang   Mat        A,P,R;
255582bec1SHong Zhang   KSP        ksp;
265582bec1SHong Zhang } GridCtx;
275582bec1SHong Zhang 
285582bec1SHong Zhang /* The context used to input PETSc matrix into ML at fine grid */
295582bec1SHong Zhang typedef struct {
30573998d7SHong Zhang   Mat          A;      /* Petsc matrix in aij format */
31573998d7SHong Zhang   Mat          Aloc;   /* local portion of A to be used by ML */
3224a42b14SHong Zhang   Vec          x,y;
335582bec1SHong Zhang   ML_Operator  *mlmat;
345582bec1SHong Zhang   PetscScalar  *pwork; /* tmp array used by PetscML_comm() */
355582bec1SHong Zhang } FineGridCtx;
365582bec1SHong Zhang 
375582bec1SHong Zhang /* The context associates a ML matrix with a PETSc shell matrix */
385582bec1SHong Zhang typedef struct {
395582bec1SHong Zhang   Mat          A;       /* PETSc shell matrix associated with mlmat */
405582bec1SHong Zhang   ML_Operator  *mlmat;  /* ML matrix assorciated with A */
4167d6f150SMatthew G Knepley   Vec          y, work;
425582bec1SHong Zhang } Mat_MLShell;
435582bec1SHong Zhang 
445582bec1SHong Zhang /* Private context for the ML preconditioner */
455582bec1SHong Zhang typedef struct {
465582bec1SHong Zhang   ML             *ml_object;
475582bec1SHong Zhang   ML_Aggregate   *agg_object;
485582bec1SHong Zhang   GridCtx        *gridctx;
495582bec1SHong Zhang   FineGridCtx    *PetscMLdata;
50b5c8bdf8SJed Brown   PetscInt       Nlevels,MaxNlevels,MaxCoarseSize,CoarsenScheme,EnergyMinimization;
51b5c8bdf8SJed Brown   PetscReal      Threshold,DampingFactor,EnergyMinimizationDropTol;
52ace3abfcSBarry Smith   PetscBool      SpectralNormScheme_Anorm,BlockScaling,EnergyMinimizationCheap,Symmetrize,OldHierarchy,KeepAggInfo,Reusable;
53573998d7SHong Zhang   PetscMPIInt    size; /* size of communicator for pc->pmat */
545582bec1SHong Zhang } PC_ML;
5541ca0015SHong Zhang 
566562c4e1SBarry Smith #undef __FUNCT__
576562c4e1SBarry Smith #define __FUNCT__ "PetscML_getrow"
586562c4e1SBarry Smith static int PetscML_getrow(ML_Operator *ML_data, int N_requested_rows, int requested_rows[],int allocated_space, int columns[], double values[], int row_lengths[])
596562c4e1SBarry Smith {
606562c4e1SBarry Smith   PetscErrorCode ierr;
616562c4e1SBarry Smith   PetscInt       m,i,j,k=0,row,*aj;
626562c4e1SBarry Smith   PetscScalar    *aa;
636562c4e1SBarry Smith   FineGridCtx    *ml=(FineGridCtx*)ML_Get_MyGetrowData(ML_data);
646562c4e1SBarry Smith   Mat_SeqAIJ     *a = (Mat_SeqAIJ*)ml->Aloc->data;
655582bec1SHong Zhang 
666562c4e1SBarry Smith 
676562c4e1SBarry Smith   ierr = MatGetSize(ml->Aloc,&m,PETSC_NULL); if (ierr) return(0);
686562c4e1SBarry Smith   for (i = 0; i<N_requested_rows; i++) {
696562c4e1SBarry Smith     row   = requested_rows[i];
706562c4e1SBarry Smith     row_lengths[i] = a->ilen[row];
716562c4e1SBarry Smith     if (allocated_space < k+row_lengths[i]) return(0);
726562c4e1SBarry Smith     if ( (row >= 0) || (row <= (m-1)) ) {
736562c4e1SBarry Smith       aj = a->j + a->i[row];
746562c4e1SBarry Smith       aa = a->a + a->i[row];
756562c4e1SBarry Smith       for (j=0; j<row_lengths[i]; j++){
766562c4e1SBarry Smith         columns[k]  = aj[j];
776562c4e1SBarry Smith         values[k++] = aa[j];
786562c4e1SBarry Smith       }
796562c4e1SBarry Smith     }
806562c4e1SBarry Smith   }
816562c4e1SBarry Smith   return(1);
826562c4e1SBarry Smith }
836562c4e1SBarry Smith 
846562c4e1SBarry Smith #undef __FUNCT__
856562c4e1SBarry Smith #define __FUNCT__ "PetscML_comm"
866562c4e1SBarry Smith static PetscErrorCode PetscML_comm(double p[],void *ML_data)
876562c4e1SBarry Smith {
886562c4e1SBarry Smith   PetscErrorCode ierr;
896562c4e1SBarry Smith   FineGridCtx    *ml=(FineGridCtx*)ML_data;
906562c4e1SBarry Smith   Mat            A=ml->A;
916562c4e1SBarry Smith   Mat_MPIAIJ     *a = (Mat_MPIAIJ*)A->data;
926562c4e1SBarry Smith   PetscMPIInt    size;
936562c4e1SBarry Smith   PetscInt       i,in_length=A->rmap->n,out_length=ml->Aloc->cmap->n;
946562c4e1SBarry Smith   PetscScalar    *array;
956562c4e1SBarry Smith 
966562c4e1SBarry Smith   PetscFunctionBegin;
976562c4e1SBarry Smith   ierr = MPI_Comm_size(((PetscObject)A)->comm,&size);CHKERRQ(ierr);
986562c4e1SBarry Smith   if (size == 1) return 0;
996562c4e1SBarry Smith 
1006562c4e1SBarry Smith   ierr = VecPlaceArray(ml->y,p);CHKERRQ(ierr);
1016562c4e1SBarry Smith   ierr = VecScatterBegin(a->Mvctx,ml->y,a->lvec,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr);
1026562c4e1SBarry Smith   ierr = VecScatterEnd(a->Mvctx,ml->y,a->lvec,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr);
1036562c4e1SBarry Smith   ierr = VecResetArray(ml->y);CHKERRQ(ierr);
1046562c4e1SBarry Smith   ierr = VecGetArray(a->lvec,&array);CHKERRQ(ierr);
1056562c4e1SBarry Smith   for (i=in_length; i<out_length; i++){
1066562c4e1SBarry Smith     p[i] = array[i-in_length];
1076562c4e1SBarry Smith   }
1086562c4e1SBarry Smith   ierr = VecRestoreArray(a->lvec,&array);CHKERRQ(ierr);
1096562c4e1SBarry Smith   PetscFunctionReturn(0);
1106562c4e1SBarry Smith }
1116562c4e1SBarry Smith 
1126562c4e1SBarry Smith #undef __FUNCT__
1136562c4e1SBarry Smith #define __FUNCT__ "PetscML_matvec"
1146562c4e1SBarry Smith static int PetscML_matvec(ML_Operator *ML_data,int in_length,double p[],int out_length,double ap[])
1156562c4e1SBarry Smith {
1166562c4e1SBarry Smith   PetscErrorCode ierr;
1176562c4e1SBarry Smith   FineGridCtx    *ml=(FineGridCtx*)ML_Get_MyMatvecData(ML_data);
1186562c4e1SBarry Smith   Mat            A=ml->A, Aloc=ml->Aloc;
1196562c4e1SBarry Smith   PetscMPIInt    size;
1206562c4e1SBarry Smith   PetscScalar    *pwork=ml->pwork;
1216562c4e1SBarry Smith   PetscInt       i;
1226562c4e1SBarry Smith 
1236562c4e1SBarry Smith   PetscFunctionBegin;
1246562c4e1SBarry Smith   ierr = MPI_Comm_size(((PetscObject)A)->comm,&size);CHKERRQ(ierr);
1256562c4e1SBarry Smith   if (size == 1){
1266562c4e1SBarry Smith     ierr = VecPlaceArray(ml->x,p);CHKERRQ(ierr);
1276562c4e1SBarry Smith   } else {
1286562c4e1SBarry Smith     for (i=0; i<in_length; i++) pwork[i] = p[i];
1296562c4e1SBarry Smith     PetscML_comm(pwork,ml);
1306562c4e1SBarry Smith     ierr = VecPlaceArray(ml->x,pwork);CHKERRQ(ierr);
1316562c4e1SBarry Smith   }
1326562c4e1SBarry Smith   ierr = VecPlaceArray(ml->y,ap);CHKERRQ(ierr);
1336562c4e1SBarry Smith   ierr = MatMult(Aloc,ml->x,ml->y);CHKERRQ(ierr);
1346562c4e1SBarry Smith   ierr = VecResetArray(ml->x);CHKERRQ(ierr);
1356562c4e1SBarry Smith   ierr = VecResetArray(ml->y);CHKERRQ(ierr);
1366562c4e1SBarry Smith   PetscFunctionReturn(0);
1376562c4e1SBarry Smith }
1386562c4e1SBarry Smith 
1396562c4e1SBarry Smith #undef __FUNCT__
1406562c4e1SBarry Smith #define __FUNCT__ "MatMult_ML"
1416562c4e1SBarry Smith static PetscErrorCode MatMult_ML(Mat A,Vec x,Vec y)
1426562c4e1SBarry Smith {
1436562c4e1SBarry Smith   PetscErrorCode   ierr;
1446562c4e1SBarry Smith   Mat_MLShell      *shell;
1456562c4e1SBarry Smith   PetscScalar      *xarray,*yarray;
1466562c4e1SBarry Smith   PetscInt         x_length,y_length;
1476562c4e1SBarry Smith 
1486562c4e1SBarry Smith   PetscFunctionBegin;
1496562c4e1SBarry Smith   ierr = MatShellGetContext(A,(void **)&shell);CHKERRQ(ierr);
1506562c4e1SBarry Smith   ierr = VecGetArray(x,&xarray);CHKERRQ(ierr);
1516562c4e1SBarry Smith   ierr = VecGetArray(y,&yarray);CHKERRQ(ierr);
1526562c4e1SBarry Smith   x_length = shell->mlmat->invec_leng;
1536562c4e1SBarry Smith   y_length = shell->mlmat->outvec_leng;
1546562c4e1SBarry Smith   ML_Operator_Apply(shell->mlmat,x_length,xarray,y_length,yarray);
1556562c4e1SBarry Smith   ierr = VecRestoreArray(x,&xarray);CHKERRQ(ierr);
1566562c4e1SBarry Smith   ierr = VecRestoreArray(y,&yarray);CHKERRQ(ierr);
1576562c4e1SBarry Smith   PetscFunctionReturn(0);
1586562c4e1SBarry Smith }
1596562c4e1SBarry Smith 
1606562c4e1SBarry Smith #undef __FUNCT__
1616562c4e1SBarry Smith #define __FUNCT__ "MatMultAdd_ML"
16267d6f150SMatthew G Knepley /* Computes y = w + A * x
16367d6f150SMatthew G Knepley    It is possible that w == y, but not x == y
16467d6f150SMatthew G Knepley */
1656562c4e1SBarry Smith static PetscErrorCode MatMultAdd_ML(Mat A,Vec x,Vec w,Vec y)
1666562c4e1SBarry Smith {
1676562c4e1SBarry Smith   Mat_MLShell   *shell;
1686562c4e1SBarry Smith   PetscScalar   *xarray,*yarray;
1696562c4e1SBarry Smith   PetscInt       x_length,y_length;
17067d6f150SMatthew G Knepley   PetscErrorCode ierr;
1716562c4e1SBarry Smith 
1726562c4e1SBarry Smith   PetscFunctionBegin;
1736562c4e1SBarry Smith   ierr = MatShellGetContext(A, (void **) &shell);CHKERRQ(ierr);
17467d6f150SMatthew G Knepley   if (y == w) {
17567d6f150SMatthew G Knepley     if (!shell->work) {
17667d6f150SMatthew G Knepley       ierr = VecDuplicate(y, &shell->work);CHKERRQ(ierr);
17767d6f150SMatthew G Knepley     }
17867d6f150SMatthew G Knepley     ierr = VecGetArray(x,           &xarray);CHKERRQ(ierr);
17967d6f150SMatthew G Knepley     ierr = VecGetArray(shell->work, &yarray);CHKERRQ(ierr);
18067d6f150SMatthew G Knepley     x_length = shell->mlmat->invec_leng;
18167d6f150SMatthew G Knepley     y_length = shell->mlmat->outvec_leng;
18267d6f150SMatthew G Knepley     ML_Operator_Apply(shell->mlmat, x_length, xarray, y_length, yarray);
18367d6f150SMatthew G Knepley     ierr = VecRestoreArray(x,           &xarray);CHKERRQ(ierr);
18467d6f150SMatthew G Knepley     ierr = VecRestoreArray(shell->work, &yarray);CHKERRQ(ierr);
1853ba3408dSMatthew G Knepley     ierr = VecAXPY(y, 1.0, shell->work);CHKERRQ(ierr);
18667d6f150SMatthew G Knepley   } else {
1876562c4e1SBarry Smith     ierr = VecGetArray(x, &xarray);CHKERRQ(ierr);
1886562c4e1SBarry Smith     ierr = VecGetArray(y, &yarray);CHKERRQ(ierr);
1896562c4e1SBarry Smith     x_length = shell->mlmat->invec_leng;
1906562c4e1SBarry Smith     y_length = shell->mlmat->outvec_leng;
1916562c4e1SBarry Smith     ML_Operator_Apply(shell->mlmat, x_length, xarray, y_length, yarray);
1926562c4e1SBarry Smith     ierr = VecRestoreArray(x, &xarray);CHKERRQ(ierr);
1936562c4e1SBarry Smith     ierr = VecRestoreArray(y, &yarray);CHKERRQ(ierr);
1946562c4e1SBarry Smith     ierr = VecAXPY(y, 1.0, w);CHKERRQ(ierr);
19567d6f150SMatthew G Knepley   }
1966562c4e1SBarry Smith   PetscFunctionReturn(0);
1976562c4e1SBarry Smith }
1986562c4e1SBarry Smith 
1996562c4e1SBarry Smith /* newtype is ignored because "ml" is not listed under Petsc MatType */
2006562c4e1SBarry Smith #undef __FUNCT__
2016562c4e1SBarry Smith #define __FUNCT__ "MatConvert_MPIAIJ_ML"
2026562c4e1SBarry Smith static PetscErrorCode MatConvert_MPIAIJ_ML(Mat A,MatType newtype,MatReuse scall,Mat *Aloc)
2036562c4e1SBarry Smith {
2046562c4e1SBarry Smith   PetscErrorCode  ierr;
2056562c4e1SBarry Smith   Mat_MPIAIJ      *mpimat=(Mat_MPIAIJ*)A->data;
2066562c4e1SBarry Smith   Mat_SeqAIJ      *mat,*a=(Mat_SeqAIJ*)(mpimat->A)->data,*b=(Mat_SeqAIJ*)(mpimat->B)->data;
2076562c4e1SBarry Smith   PetscInt        *ai=a->i,*aj=a->j,*bi=b->i,*bj=b->j;
2086562c4e1SBarry Smith   PetscScalar     *aa=a->a,*ba=b->a,*ca;
2096562c4e1SBarry Smith   PetscInt        am=A->rmap->n,an=A->cmap->n,i,j,k;
2106562c4e1SBarry Smith   PetscInt        *ci,*cj,ncols;
2116562c4e1SBarry Smith 
2126562c4e1SBarry Smith   PetscFunctionBegin;
213e32f2f54SBarry Smith   if (am != an) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONG,"A must have a square diagonal portion, am: %d != an: %d",am,an);
2146562c4e1SBarry Smith 
2156562c4e1SBarry Smith   if (scall == MAT_INITIAL_MATRIX){
2166562c4e1SBarry Smith     ierr = PetscMalloc((1+am)*sizeof(PetscInt),&ci);CHKERRQ(ierr);
2176562c4e1SBarry Smith     ci[0] = 0;
2186562c4e1SBarry Smith     for (i=0; i<am; i++){
2196562c4e1SBarry Smith       ci[i+1] = ci[i] + (ai[i+1] - ai[i]) + (bi[i+1] - bi[i]);
2206562c4e1SBarry Smith     }
2216562c4e1SBarry Smith     ierr = PetscMalloc((1+ci[am])*sizeof(PetscInt),&cj);CHKERRQ(ierr);
2226562c4e1SBarry Smith     ierr = PetscMalloc((1+ci[am])*sizeof(PetscScalar),&ca);CHKERRQ(ierr);
2236562c4e1SBarry Smith 
2246562c4e1SBarry Smith     k = 0;
2256562c4e1SBarry Smith     for (i=0; i<am; i++){
2266562c4e1SBarry Smith       /* diagonal portion of A */
2276562c4e1SBarry Smith       ncols = ai[i+1] - ai[i];
2286562c4e1SBarry Smith       for (j=0; j<ncols; j++) {
2296562c4e1SBarry Smith         cj[k]   = *aj++;
2306562c4e1SBarry Smith         ca[k++] = *aa++;
2316562c4e1SBarry Smith       }
2326562c4e1SBarry Smith       /* off-diagonal portion of A */
2336562c4e1SBarry Smith       ncols = bi[i+1] - bi[i];
2346562c4e1SBarry Smith       for (j=0; j<ncols; j++) {
2356562c4e1SBarry Smith         cj[k]   = an + (*bj); bj++;
2366562c4e1SBarry Smith         ca[k++] = *ba++;
2376562c4e1SBarry Smith       }
2386562c4e1SBarry Smith     }
239e32f2f54SBarry Smith     if (k != ci[am]) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONG,"k: %d != ci[am]: %d",k,ci[am]);
2406562c4e1SBarry Smith 
2416562c4e1SBarry Smith     /* put together the new matrix */
2426562c4e1SBarry Smith     an = mpimat->A->cmap->n+mpimat->B->cmap->n;
2436562c4e1SBarry Smith     ierr = MatCreateSeqAIJWithArrays(PETSC_COMM_SELF,am,an,ci,cj,ca,Aloc);CHKERRQ(ierr);
2446562c4e1SBarry Smith 
2456562c4e1SBarry Smith     /* MatCreateSeqAIJWithArrays flags matrix so PETSc doesn't free the user's arrays. */
2466562c4e1SBarry Smith     /* Since these are PETSc arrays, change flags to free them as necessary. */
2476562c4e1SBarry Smith     mat = (Mat_SeqAIJ*)(*Aloc)->data;
2486562c4e1SBarry Smith     mat->free_a       = PETSC_TRUE;
2496562c4e1SBarry Smith     mat->free_ij      = PETSC_TRUE;
2506562c4e1SBarry Smith 
2516562c4e1SBarry Smith     mat->nonew    = 0;
2526562c4e1SBarry Smith   } else if (scall == MAT_REUSE_MATRIX){
2536562c4e1SBarry Smith     mat=(Mat_SeqAIJ*)(*Aloc)->data;
2546562c4e1SBarry Smith     ci = mat->i; cj = mat->j; ca = mat->a;
2556562c4e1SBarry Smith     for (i=0; i<am; i++) {
2566562c4e1SBarry Smith       /* diagonal portion of A */
2576562c4e1SBarry Smith       ncols = ai[i+1] - ai[i];
2586562c4e1SBarry Smith       for (j=0; j<ncols; j++) *ca++ = *aa++;
2596562c4e1SBarry Smith       /* off-diagonal portion of A */
2606562c4e1SBarry Smith       ncols = bi[i+1] - bi[i];
2616562c4e1SBarry Smith       for (j=0; j<ncols; j++) *ca++ = *ba++;
2626562c4e1SBarry Smith     }
2636562c4e1SBarry Smith   } else {
2640005702aSJed Brown     SETERRQ1(((PetscObject)A)->comm,PETSC_ERR_ARG_WRONG,"Invalid MatReuse %d",(int)scall);
2656562c4e1SBarry Smith   }
2666562c4e1SBarry Smith   PetscFunctionReturn(0);
2676562c4e1SBarry Smith }
2686562c4e1SBarry Smith 
2696562c4e1SBarry Smith extern PetscErrorCode MatDestroy_Shell(Mat);
2706562c4e1SBarry Smith #undef __FUNCT__
2716562c4e1SBarry Smith #define __FUNCT__ "MatDestroy_ML"
2726562c4e1SBarry Smith static PetscErrorCode MatDestroy_ML(Mat A)
2736562c4e1SBarry Smith {
2746562c4e1SBarry Smith   PetscErrorCode ierr;
2756562c4e1SBarry Smith   Mat_MLShell    *shell;
2766562c4e1SBarry Smith 
2776562c4e1SBarry Smith   PetscFunctionBegin;
2786562c4e1SBarry Smith   ierr = MatShellGetContext(A,(void **)&shell);CHKERRQ(ierr);
2796562c4e1SBarry Smith   ierr = VecDestroy(shell->y);CHKERRQ(ierr);
28067d6f150SMatthew G Knepley   if (shell->work) {ierr = VecDestroy(shell->work);CHKERRQ(ierr);}
2816562c4e1SBarry Smith   ierr = PetscFree(shell);CHKERRQ(ierr);
2826562c4e1SBarry Smith   ierr = MatDestroy_Shell(A);CHKERRQ(ierr);
2836562c4e1SBarry Smith   ierr = PetscObjectChangeTypeName((PetscObject)A,0);CHKERRQ(ierr);
2846562c4e1SBarry Smith   PetscFunctionReturn(0);
2856562c4e1SBarry Smith }
2866562c4e1SBarry Smith 
2876562c4e1SBarry Smith #undef __FUNCT__
2886562c4e1SBarry Smith #define __FUNCT__ "MatWrapML_SeqAIJ"
2896562c4e1SBarry Smith static PetscErrorCode MatWrapML_SeqAIJ(ML_Operator *mlmat,MatReuse reuse,Mat *newmat)
2906562c4e1SBarry Smith {
2916562c4e1SBarry Smith   struct ML_CSR_MSRdata *matdata = (struct ML_CSR_MSRdata *)mlmat->data;
2926562c4e1SBarry Smith   PetscErrorCode        ierr;
2936562c4e1SBarry Smith   PetscInt              m=mlmat->outvec_leng,n=mlmat->invec_leng,*nnz,nz_max;
2946562c4e1SBarry Smith   PetscInt              *ml_cols=matdata->columns,*ml_rowptr=matdata->rowptr,*aj,i,j,k;
2956562c4e1SBarry Smith   PetscScalar           *ml_vals=matdata->values,*aa;
2966562c4e1SBarry Smith 
2976562c4e1SBarry Smith   PetscFunctionBegin;
298e7e72b3dSBarry Smith   if (!mlmat->getrow) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_NULL,"mlmat->getrow = NULL");
2996562c4e1SBarry Smith   if (m != n){ /* ML Pmat and Rmat are in CSR format. Pass array pointers into SeqAIJ matrix */
3006562c4e1SBarry Smith     if (reuse){
3016562c4e1SBarry Smith       Mat_SeqAIJ  *aij= (Mat_SeqAIJ*)(*newmat)->data;
3026562c4e1SBarry Smith       aij->i = ml_rowptr;
3036562c4e1SBarry Smith       aij->j = ml_cols;
3046562c4e1SBarry Smith       aij->a = ml_vals;
3056562c4e1SBarry Smith     } else {
3066562c4e1SBarry Smith       /* sort ml_cols and ml_vals */
3076562c4e1SBarry Smith       ierr = PetscMalloc((m+1)*sizeof(PetscInt),&nnz);
3086562c4e1SBarry Smith       for (i=0; i<m; i++) {
3096562c4e1SBarry Smith         nnz[i] = ml_rowptr[i+1] - ml_rowptr[i];
3106562c4e1SBarry Smith       }
3116562c4e1SBarry Smith       aj = ml_cols; aa = ml_vals;
3126562c4e1SBarry Smith       for (i=0; i<m; i++){
3136562c4e1SBarry Smith         ierr = PetscSortIntWithScalarArray(nnz[i],aj,aa);CHKERRQ(ierr);
3146562c4e1SBarry Smith         aj += nnz[i]; aa += nnz[i];
3156562c4e1SBarry Smith       }
3166562c4e1SBarry Smith       ierr = MatCreateSeqAIJWithArrays(PETSC_COMM_SELF,m,n,ml_rowptr,ml_cols,ml_vals,newmat);CHKERRQ(ierr);
3176562c4e1SBarry Smith       ierr = PetscFree(nnz);CHKERRQ(ierr);
3186562c4e1SBarry Smith     }
3196562c4e1SBarry Smith     PetscFunctionReturn(0);
3206562c4e1SBarry Smith   }
3216562c4e1SBarry Smith 
3226562c4e1SBarry Smith   /* ML Amat is in MSR format. Copy its data into SeqAIJ matrix */
3236562c4e1SBarry Smith   ierr = MatCreate(PETSC_COMM_SELF,newmat);CHKERRQ(ierr);
3246562c4e1SBarry Smith   ierr = MatSetSizes(*newmat,m,n,PETSC_DECIDE,PETSC_DECIDE);CHKERRQ(ierr);
3256562c4e1SBarry Smith   ierr = MatSetType(*newmat,MATSEQAIJ);CHKERRQ(ierr);
3266562c4e1SBarry Smith 
3276562c4e1SBarry Smith   ierr = PetscMalloc((m+1)*sizeof(PetscInt),&nnz);
3286562c4e1SBarry Smith   nz_max = 1;
3296562c4e1SBarry Smith   for (i=0; i<m; i++) {
3306562c4e1SBarry Smith     nnz[i] = ml_cols[i+1] - ml_cols[i] + 1;
3316562c4e1SBarry Smith     if (nnz[i] > nz_max) nz_max += nnz[i];
3326562c4e1SBarry Smith   }
3336562c4e1SBarry Smith 
3346562c4e1SBarry Smith   ierr = MatSeqAIJSetPreallocation(*newmat,0,nnz);CHKERRQ(ierr);
3356562c4e1SBarry Smith   ierr = PetscMalloc2(nz_max,PetscScalar,&aa,nz_max,PetscInt,&aj);CHKERRQ(ierr);
3366562c4e1SBarry Smith   for (i=0; i<m; i++){
3376562c4e1SBarry Smith     k = 0;
3386562c4e1SBarry Smith     /* diagonal entry */
3396562c4e1SBarry Smith     aj[k] = i; aa[k++] = ml_vals[i];
3406562c4e1SBarry Smith     /* off diagonal entries */
3416562c4e1SBarry Smith     for (j=ml_cols[i]; j<ml_cols[i+1]; j++){
3426562c4e1SBarry Smith       aj[k] = ml_cols[j]; aa[k++] = ml_vals[j];
3436562c4e1SBarry Smith     }
3446562c4e1SBarry Smith     /* sort aj and aa */
3456562c4e1SBarry Smith     ierr = PetscSortIntWithScalarArray(nnz[i],aj,aa);CHKERRQ(ierr);
3466562c4e1SBarry Smith     ierr = MatSetValues(*newmat,1,&i,nnz[i],aj,aa,INSERT_VALUES);CHKERRQ(ierr);
3476562c4e1SBarry Smith   }
3486562c4e1SBarry Smith   ierr = MatAssemblyBegin(*newmat,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
3496562c4e1SBarry Smith   ierr = MatAssemblyEnd(*newmat,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
3506562c4e1SBarry Smith 
3516562c4e1SBarry Smith   ierr = PetscFree2(aa,aj);CHKERRQ(ierr);
3526562c4e1SBarry Smith   ierr = PetscFree(nnz);CHKERRQ(ierr);
3536562c4e1SBarry Smith   PetscFunctionReturn(0);
3546562c4e1SBarry Smith }
3556562c4e1SBarry Smith 
3566562c4e1SBarry Smith #undef __FUNCT__
3576562c4e1SBarry Smith #define __FUNCT__ "MatWrapML_SHELL"
3586562c4e1SBarry Smith static PetscErrorCode MatWrapML_SHELL(ML_Operator *mlmat,MatReuse reuse,Mat *newmat)
3596562c4e1SBarry Smith {
3606562c4e1SBarry Smith   PetscErrorCode ierr;
3616562c4e1SBarry Smith   PetscInt       m,n;
3626562c4e1SBarry Smith   ML_Comm        *MLcomm;
3636562c4e1SBarry Smith   Mat_MLShell    *shellctx;
3646562c4e1SBarry Smith 
3656562c4e1SBarry Smith   PetscFunctionBegin;
3666562c4e1SBarry Smith   m = mlmat->outvec_leng;
3676562c4e1SBarry Smith   n = mlmat->invec_leng;
3686562c4e1SBarry Smith   if (!m || !n){
3696562c4e1SBarry Smith     newmat = PETSC_NULL;
3706562c4e1SBarry Smith     PetscFunctionReturn(0);
3716562c4e1SBarry Smith   }
3726562c4e1SBarry Smith 
3736562c4e1SBarry Smith   if (reuse){
3746562c4e1SBarry Smith     ierr = MatShellGetContext(*newmat,(void **)&shellctx);CHKERRQ(ierr);
3756562c4e1SBarry Smith     shellctx->mlmat = mlmat;
3766562c4e1SBarry Smith     PetscFunctionReturn(0);
3776562c4e1SBarry Smith   }
3786562c4e1SBarry Smith 
3796562c4e1SBarry Smith   MLcomm = mlmat->comm;
3806562c4e1SBarry Smith   ierr = PetscNew(Mat_MLShell,&shellctx);CHKERRQ(ierr);
3816562c4e1SBarry Smith   ierr = MatCreateShell(MLcomm->USR_comm,m,n,PETSC_DETERMINE,PETSC_DETERMINE,shellctx,newmat);CHKERRQ(ierr);
3826562c4e1SBarry Smith   ierr = MatShellSetOperation(*newmat,MATOP_MULT,(void(*)(void))MatMult_ML);CHKERRQ(ierr);
3836562c4e1SBarry Smith   ierr = MatShellSetOperation(*newmat,MATOP_MULT_ADD,(void(*)(void))MatMultAdd_ML);CHKERRQ(ierr);
3846562c4e1SBarry Smith   shellctx->A         = *newmat;
3856562c4e1SBarry Smith   shellctx->mlmat     = mlmat;
38667d6f150SMatthew G Knepley   shellctx->work      = PETSC_NULL;
3876562c4e1SBarry Smith   ierr = VecCreate(PETSC_COMM_WORLD,&shellctx->y);CHKERRQ(ierr);
3886562c4e1SBarry Smith   ierr = VecSetSizes(shellctx->y,m,PETSC_DECIDE);CHKERRQ(ierr);
3896562c4e1SBarry Smith   ierr = VecSetFromOptions(shellctx->y);CHKERRQ(ierr);
3906562c4e1SBarry Smith   (*newmat)->ops->destroy = MatDestroy_ML;
3916562c4e1SBarry Smith   PetscFunctionReturn(0);
3926562c4e1SBarry Smith }
3936562c4e1SBarry Smith 
3946562c4e1SBarry Smith #undef __FUNCT__
3956562c4e1SBarry Smith #define __FUNCT__ "MatWrapML_MPIAIJ"
3966562c4e1SBarry Smith static PetscErrorCode MatWrapML_MPIAIJ(ML_Operator *mlmat,Mat *newmat)
3976562c4e1SBarry Smith {
3986562c4e1SBarry Smith   struct ML_CSR_MSRdata *matdata = (struct ML_CSR_MSRdata *)mlmat->data;
3996562c4e1SBarry Smith   PetscInt              *ml_cols=matdata->columns,*aj;
4006562c4e1SBarry Smith   PetscScalar           *ml_vals=matdata->values,*aa;
4016562c4e1SBarry Smith   PetscErrorCode        ierr;
4026562c4e1SBarry Smith   PetscInt              i,j,k,*gordering;
4036562c4e1SBarry Smith   PetscInt              m=mlmat->outvec_leng,n,*nnzA,*nnzB,*nnz,nz_max,row;
4046562c4e1SBarry Smith   Mat                   A;
4056562c4e1SBarry Smith 
4066562c4e1SBarry Smith   PetscFunctionBegin;
407e7e72b3dSBarry Smith   if (!mlmat->getrow) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_NULL,"mlmat->getrow = NULL");
4086562c4e1SBarry Smith   n = mlmat->invec_leng;
409e32f2f54SBarry Smith   if (m != n) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"m %d must equal to n %d",m,n);
4106562c4e1SBarry Smith 
4116562c4e1SBarry Smith   ierr = MatCreate(mlmat->comm->USR_comm,&A);CHKERRQ(ierr);
4126562c4e1SBarry Smith   ierr = MatSetSizes(A,m,n,PETSC_DECIDE,PETSC_DECIDE);CHKERRQ(ierr);
4136562c4e1SBarry Smith   ierr = MatSetType(A,MATMPIAIJ);CHKERRQ(ierr);
4146562c4e1SBarry Smith   ierr = PetscMalloc3(m,PetscInt,&nnzA,m,PetscInt,&nnzB,m,PetscInt,&nnz);CHKERRQ(ierr);
4156562c4e1SBarry Smith 
4166562c4e1SBarry Smith   nz_max = 0;
4176562c4e1SBarry Smith   for (i=0; i<m; i++){
4186562c4e1SBarry Smith     nnz[i] = ml_cols[i+1] - ml_cols[i] + 1;
4196562c4e1SBarry Smith     if (nz_max < nnz[i]) nz_max = nnz[i];
4206562c4e1SBarry Smith     nnzA[i] = 1; /* diag */
4216562c4e1SBarry Smith     for (j=ml_cols[i]; j<ml_cols[i+1]; j++){
4226562c4e1SBarry Smith       if (ml_cols[j] < m) nnzA[i]++;
4236562c4e1SBarry Smith     }
4246562c4e1SBarry Smith     nnzB[i] = nnz[i] - nnzA[i];
4256562c4e1SBarry Smith   }
4266562c4e1SBarry Smith   ierr = MatMPIAIJSetPreallocation(A,0,nnzA,0,nnzB);CHKERRQ(ierr);
4276562c4e1SBarry Smith 
4286562c4e1SBarry Smith   /* insert mat values -- remap row and column indices */
4296562c4e1SBarry Smith   nz_max++;
4306562c4e1SBarry Smith   ierr = PetscMalloc2(nz_max,PetscScalar,&aa,nz_max,PetscInt,&aj);CHKERRQ(ierr);
4316562c4e1SBarry Smith   /* create global row numbering for a ML_Operator */
4326562c4e1SBarry Smith   ML_build_global_numbering(mlmat,&gordering,"rows");
4336562c4e1SBarry Smith   for (i=0; i<m; i++){
4346562c4e1SBarry Smith     row = gordering[i];
4356562c4e1SBarry Smith     k = 0;
4366562c4e1SBarry Smith     /* diagonal entry */
4376562c4e1SBarry Smith     aj[k] = row; aa[k++] = ml_vals[i];
4386562c4e1SBarry Smith     /* off diagonal entries */
4396562c4e1SBarry Smith     for (j=ml_cols[i]; j<ml_cols[i+1]; j++){
4406562c4e1SBarry Smith       aj[k] = gordering[ml_cols[j]]; aa[k++] = ml_vals[j];
4416562c4e1SBarry Smith     }
4426562c4e1SBarry Smith     ierr = MatSetValues(A,1,&row,nnz[i],aj,aa,INSERT_VALUES);CHKERRQ(ierr);
4436562c4e1SBarry Smith   }
444eb4736aaSBarry Smith   ML_free(gordering);
4456562c4e1SBarry Smith   ierr = MatAssemblyBegin(A,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
4466562c4e1SBarry Smith   ierr = MatAssemblyEnd(A,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
4476562c4e1SBarry Smith   *newmat = A;
4486562c4e1SBarry Smith 
4496562c4e1SBarry Smith   ierr = PetscFree3(nnzA,nnzB,nnz);
4506562c4e1SBarry Smith   ierr = PetscFree2(aa,aj);CHKERRQ(ierr);
4516562c4e1SBarry Smith   PetscFunctionReturn(0);
4526562c4e1SBarry Smith }
4536562c4e1SBarry Smith 
4546562c4e1SBarry Smith /* -----------------------------------------------------------------------------*/
45501da6913SBarry Smith #undef __FUNCT__
456a06653b4SBarry Smith #define __FUNCT__ "PCReset_ML"
45716336fedSMatthew G Knepley PetscErrorCode PCReset_ML(PC pc)
45801da6913SBarry Smith {
45901da6913SBarry Smith   PetscErrorCode  ierr;
460*e0262f48SMatthew G Knepley   PC_MG           *mg    = (PC_MG *) pc->data;
461*e0262f48SMatthew G Knepley   PC_ML           *pc_ml = (PC_ML *) mg->innerctx;
46201da6913SBarry Smith   PetscInt        level,fine_level=pc_ml->Nlevels-1;
46301da6913SBarry Smith 
46401da6913SBarry Smith   PetscFunctionBegin;
46501da6913SBarry Smith   ML_Aggregate_Destroy(&pc_ml->agg_object);
46601da6913SBarry Smith   ML_Destroy(&pc_ml->ml_object);
46701da6913SBarry Smith 
46801da6913SBarry Smith   if (pc_ml->PetscMLdata) {
46901da6913SBarry Smith     ierr = PetscFree(pc_ml->PetscMLdata->pwork);CHKERRQ(ierr);
47001da6913SBarry Smith     if (pc_ml->size > 1)      {ierr = MatDestroy(pc_ml->PetscMLdata->Aloc);CHKERRQ(ierr);}
47101da6913SBarry Smith     if (pc_ml->PetscMLdata->x){ierr = VecDestroy(pc_ml->PetscMLdata->x);CHKERRQ(ierr);}
47201da6913SBarry Smith     if (pc_ml->PetscMLdata->y){ierr = VecDestroy(pc_ml->PetscMLdata->y);CHKERRQ(ierr);}
47301da6913SBarry Smith   }
47401da6913SBarry Smith   ierr = PetscFree(pc_ml->PetscMLdata);CHKERRQ(ierr);
47501da6913SBarry Smith 
47601da6913SBarry Smith   for (level=0; level<fine_level; level++){
47701da6913SBarry Smith     if (pc_ml->gridctx[level].A){ierr = MatDestroy(pc_ml->gridctx[level].A);CHKERRQ(ierr);}
47801da6913SBarry Smith     if (pc_ml->gridctx[level].P){ierr = MatDestroy(pc_ml->gridctx[level].P);CHKERRQ(ierr);}
47901da6913SBarry Smith     if (pc_ml->gridctx[level].R){ierr = MatDestroy(pc_ml->gridctx[level].R);CHKERRQ(ierr);}
48001da6913SBarry Smith     if (pc_ml->gridctx[level].x){ierr = VecDestroy(pc_ml->gridctx[level].x);CHKERRQ(ierr);}
48101da6913SBarry Smith     if (pc_ml->gridctx[level].b){ierr = VecDestroy(pc_ml->gridctx[level].b);CHKERRQ(ierr);}
48201da6913SBarry Smith     if (pc_ml->gridctx[level+1].r){ierr = VecDestroy(pc_ml->gridctx[level+1].r);CHKERRQ(ierr);}
48301da6913SBarry Smith   }
48401da6913SBarry Smith   ierr = PetscFree(pc_ml->gridctx);CHKERRQ(ierr);
48501da6913SBarry Smith   PetscFunctionReturn(0);
48601da6913SBarry Smith }
4875582bec1SHong Zhang /* -------------------------------------------------------------------------- */
4885582bec1SHong Zhang /*
4895582bec1SHong Zhang    PCSetUp_ML - Prepares for the use of the ML preconditioner
4905582bec1SHong Zhang                     by setting data structures and options.
4915582bec1SHong Zhang 
4925582bec1SHong Zhang    Input Parameter:
4935582bec1SHong Zhang .  pc - the preconditioner context
4945582bec1SHong Zhang 
4955582bec1SHong Zhang    Application Interface Routine: PCSetUp()
4965582bec1SHong Zhang 
4975582bec1SHong Zhang    Notes:
4985582bec1SHong Zhang    The interface routine PCSetUp() is not usually called directly by
4995582bec1SHong Zhang    the user, but instead is called by PCApply() if necessary.
5005582bec1SHong Zhang */
5016ca4d86aSHong Zhang extern PetscErrorCode PCSetFromOptions_MG(PC);
502a06653b4SBarry Smith extern PetscErrorCode PCReset_MG(PC);
503c07bf074SBarry Smith 
5045582bec1SHong Zhang #undef __FUNCT__
5055582bec1SHong Zhang #define __FUNCT__ "PCSetUp_ML"
5066ca4d86aSHong Zhang PetscErrorCode PCSetUp_ML(PC pc)
5075582bec1SHong Zhang {
5085582bec1SHong Zhang   PetscErrorCode  ierr;
509eef31507SHong Zhang   PetscMPIInt     size;
5105582bec1SHong Zhang   FineGridCtx     *PetscMLdata;
5115582bec1SHong Zhang   ML              *ml_object;
5125582bec1SHong Zhang   ML_Aggregate    *agg_object;
5135582bec1SHong Zhang   ML_Operator     *mlmat;
5144f8eab3cSJed Brown   PetscInt        nlocal_allcols,Nlevels,mllevel,level,level1,m,fine_level,bs;
5155582bec1SHong Zhang   Mat             A,Aloc;
5165582bec1SHong Zhang   GridCtx         *gridctx;
51701da6913SBarry Smith   PC_MG           *mg = (PC_MG*)pc->data;
51801da6913SBarry Smith   PC_ML           *pc_ml = (PC_ML*)mg->innerctx;
519ace3abfcSBarry Smith   PetscBool       isSeq, isMPI;
520c07bf074SBarry Smith   KSP             smoother;
521c07bf074SBarry Smith   PC              subpc;
5225582bec1SHong Zhang 
5235582bec1SHong Zhang   PetscFunctionBegin;
524573998d7SHong Zhang   if (pc->setupcalled){
525c07bf074SBarry Smith     /* since ML can change the size of vectors/matrices at any level we must destroy everything */
52616336fedSMatthew G Knepley     ierr = PCReset_ML(pc);CHKERRQ(ierr);
527a06653b4SBarry Smith     ierr = PCReset_MG(pc);CHKERRQ(ierr);
528573998d7SHong Zhang   }
529573998d7SHong Zhang 
5305582bec1SHong Zhang   /* setup special features of PCML */
5315582bec1SHong Zhang   /*--------------------------------*/
5325582bec1SHong Zhang   /* covert A to Aloc to be used by ML at fine grid */
5335582bec1SHong Zhang   A = pc->pmat;
5347adad957SLisandro Dalcin   ierr = MPI_Comm_size(((PetscObject)A)->comm,&size);CHKERRQ(ierr);
5355582bec1SHong Zhang   pc_ml->size = size;
536864b637dSMatthew Knepley   ierr = PetscTypeCompare((PetscObject) A, MATSEQAIJ, &isSeq);CHKERRQ(ierr);
537864b637dSMatthew Knepley   ierr = PetscTypeCompare((PetscObject) A, MATMPIAIJ, &isMPI);CHKERRQ(ierr);
538864b637dSMatthew Knepley   if (isMPI){
539db571536SBarry Smith     ierr = MatConvert_MPIAIJ_ML(A,PETSC_NULL,MAT_INITIAL_MATRIX,&Aloc);CHKERRQ(ierr);
540864b637dSMatthew Knepley   } else if (isSeq) {
5415582bec1SHong Zhang     Aloc = A;
542e7e72b3dSBarry Smith   } else SETERRQ(((PetscObject)pc)->comm,PETSC_ERR_ARG_WRONG, "Invalid matrix type for ML. ML can only handle AIJ matrices.");
5435582bec1SHong Zhang 
5445582bec1SHong Zhang   /* create and initialize struct 'PetscMLdata' */
54538f2d2fdSLisandro Dalcin   ierr = PetscNewLog(pc,FineGridCtx,&PetscMLdata);CHKERRQ(ierr);
5465582bec1SHong Zhang   pc_ml->PetscMLdata = PetscMLdata;
547d0f46423SBarry Smith   ierr = PetscMalloc((Aloc->cmap->n+1)*sizeof(PetscScalar),&PetscMLdata->pwork);CHKERRQ(ierr);
5485582bec1SHong Zhang 
54924a42b14SHong Zhang   ierr = VecCreate(PETSC_COMM_SELF,&PetscMLdata->x);CHKERRQ(ierr);
550d0f46423SBarry Smith   ierr = VecSetSizes(PetscMLdata->x,Aloc->cmap->n,Aloc->cmap->n);CHKERRQ(ierr);
55124a42b14SHong Zhang   ierr = VecSetType(PetscMLdata->x,VECSEQ);CHKERRQ(ierr);
55224a42b14SHong Zhang 
55324a42b14SHong Zhang   ierr = VecCreate(PETSC_COMM_SELF,&PetscMLdata->y);CHKERRQ(ierr);
554d0f46423SBarry Smith   ierr = VecSetSizes(PetscMLdata->y,A->rmap->n,PETSC_DECIDE);CHKERRQ(ierr);
55524a42b14SHong Zhang   ierr = VecSetType(PetscMLdata->y,VECSEQ);CHKERRQ(ierr);
556573998d7SHong Zhang   PetscMLdata->A    = A;
557573998d7SHong Zhang   PetscMLdata->Aloc = Aloc;
55824a42b14SHong Zhang 
5595582bec1SHong Zhang   /* create ML discretization matrix at fine grid */
56045cf47abSHong Zhang   /* ML requires input of fine-grid matrix. It determines nlevels. */
5615582bec1SHong Zhang   ierr = MatGetSize(Aloc,&m,&nlocal_allcols);CHKERRQ(ierr);
5624f8eab3cSJed Brown   ierr = MatGetBlockSize(A,&bs);CHKERRQ(ierr);
5635582bec1SHong Zhang   ML_Create(&ml_object,pc_ml->MaxNlevels);
564ead7dcbeSHong Zhang   ML_Comm_Set_UsrComm(ml_object->comm,((PetscObject)A)->comm);
565573998d7SHong Zhang   pc_ml->ml_object = ml_object;
5665582bec1SHong Zhang   ML_Init_Amatrix(ml_object,0,m,m,PetscMLdata);
5675582bec1SHong Zhang   ML_Set_Amatrix_Getrow(ml_object,0,PetscML_getrow,PetscML_comm,nlocal_allcols);
5685582bec1SHong Zhang   ML_Set_Amatrix_Matvec(ml_object,0,PetscML_matvec);
5695582bec1SHong Zhang 
570b5c8bdf8SJed Brown   ML_Set_Symmetrize(ml_object,pc_ml->Symmetrize ? ML_YES : ML_NO);
571b5c8bdf8SJed Brown 
5725582bec1SHong Zhang   /* aggregation */
5735582bec1SHong Zhang   ML_Aggregate_Create(&agg_object);
574573998d7SHong Zhang   pc_ml->agg_object = agg_object;
575573998d7SHong Zhang 
5764f8eab3cSJed Brown   ML_Aggregate_Set_NullSpace(agg_object,bs,bs,0,0);CHKERRQ(ierr);
5775582bec1SHong Zhang   ML_Aggregate_Set_MaxCoarseSize(agg_object,pc_ml->MaxCoarseSize);
5785582bec1SHong Zhang   /* set options */
5795582bec1SHong Zhang   switch (pc_ml->CoarsenScheme) {
5805582bec1SHong Zhang   case 1:
5815582bec1SHong Zhang     ML_Aggregate_Set_CoarsenScheme_Coupled(agg_object);break;
5825582bec1SHong Zhang   case 2:
5835582bec1SHong Zhang     ML_Aggregate_Set_CoarsenScheme_MIS(agg_object);break;
5845582bec1SHong Zhang   case 3:
5855582bec1SHong Zhang     ML_Aggregate_Set_CoarsenScheme_METIS(agg_object);break;
5865582bec1SHong Zhang   }
5875582bec1SHong Zhang   ML_Aggregate_Set_Threshold(agg_object,pc_ml->Threshold);
5885582bec1SHong Zhang   ML_Aggregate_Set_DampingFactor(agg_object,pc_ml->DampingFactor);
5895582bec1SHong Zhang   if (pc_ml->SpectralNormScheme_Anorm){
5907ffd031bSHong Zhang     ML_Set_SpectralNormScheme_Anorm(ml_object);
5915582bec1SHong Zhang   }
592b5c8bdf8SJed Brown   agg_object->keep_agg_information      = (int)pc_ml->KeepAggInfo;
593b5c8bdf8SJed Brown   agg_object->keep_P_tentative          = (int)pc_ml->Reusable;
594b5c8bdf8SJed Brown   agg_object->block_scaled_SA           = (int)pc_ml->BlockScaling;
595b5c8bdf8SJed Brown   agg_object->minimizing_energy         = (int)pc_ml->EnergyMinimization;
596b5c8bdf8SJed Brown   agg_object->minimizing_energy_droptol = (double)pc_ml->EnergyMinimizationDropTol;
597b5c8bdf8SJed Brown   agg_object->cheap_minimizing_energy   = (int)pc_ml->EnergyMinimizationCheap;
5985582bec1SHong Zhang 
599b5c8bdf8SJed Brown   if (pc_ml->OldHierarchy) {
6005582bec1SHong Zhang     Nlevels = ML_Gen_MGHierarchy_UsingAggregation(ml_object,0,ML_INCREASING,agg_object);
601b5c8bdf8SJed Brown   } else {
602b5c8bdf8SJed Brown     Nlevels = ML_Gen_MultiLevelHierarchy_UsingAggregation(ml_object,0,ML_INCREASING,agg_object);
603b5c8bdf8SJed Brown   }
60465e19b50SBarry Smith   if (Nlevels<=0) SETERRQ1(((PetscObject)pc)->comm,PETSC_ERR_ARG_OUTOFRANGE,"Nlevels %d must > 0",Nlevels);
605573998d7SHong Zhang   pc_ml->Nlevels = Nlevels;
606aa85bbbfSHong Zhang   fine_level = Nlevels - 1;
607c07bf074SBarry Smith 
60897177400SBarry Smith   ierr = PCMGSetLevels(pc,Nlevels,PETSC_NULL);CHKERRQ(ierr);
609aa85bbbfSHong Zhang   /* set default smoothers */
610aa85bbbfSHong Zhang   for (level=1; level<=fine_level; level++){
611aa85bbbfSHong Zhang     if (size == 1){
612aa85bbbfSHong Zhang       ierr = PCMGGetSmoother(pc,level,&smoother);CHKERRQ(ierr);
613aa85bbbfSHong Zhang       ierr = KSPSetType(smoother,KSPRICHARDSON);CHKERRQ(ierr);
614aa85bbbfSHong Zhang       ierr = KSPGetPC(smoother,&subpc);CHKERRQ(ierr);
615aa85bbbfSHong Zhang       ierr = PCSetType(subpc,PCSOR);CHKERRQ(ierr);
616aa85bbbfSHong Zhang     } else {
617aa85bbbfSHong Zhang       ierr = PCMGGetSmoother(pc,level,&smoother);CHKERRQ(ierr);
618aa85bbbfSHong Zhang       ierr = KSPSetType(smoother,KSPRICHARDSON);CHKERRQ(ierr);
619aa85bbbfSHong Zhang       ierr = KSPGetPC(smoother,&subpc);CHKERRQ(ierr);
620aa85bbbfSHong Zhang       ierr = PCSetType(subpc,PCSOR);CHKERRQ(ierr);
621aa85bbbfSHong Zhang     }
622aa85bbbfSHong Zhang   }
62397177400SBarry Smith   ierr = PCSetFromOptions_MG(pc);CHKERRQ(ierr); /* should be called in PCSetFromOptions_ML(), but cannot be called prior to PCMGSetLevels() */
6245582bec1SHong Zhang 
6255582bec1SHong Zhang   ierr = PetscMalloc(Nlevels*sizeof(GridCtx),&gridctx);CHKERRQ(ierr);
6265582bec1SHong Zhang   pc_ml->gridctx = gridctx;
6275582bec1SHong Zhang 
6285582bec1SHong Zhang   /* wrap ML matrices by PETSc shell matrices at coarsened grids.
6295582bec1SHong Zhang      Level 0 is the finest grid for ML, but coarsest for PETSc! */
630e14861a4SHong Zhang   gridctx[fine_level].A = A;
631573998d7SHong Zhang 
632e14861a4SHong Zhang   level = fine_level - 1;
633ab718edeSHong Zhang   if (size == 1){ /* convert ML P, R and A into seqaij format */
6345582bec1SHong Zhang     for (mllevel=1; mllevel<Nlevels; mllevel++){
635e14861a4SHong Zhang       mlmat = &(ml_object->Pmat[mllevel]);
636db571536SBarry Smith       ierr  = MatWrapML_SeqAIJ(mlmat,MAT_INITIAL_MATRIX,&gridctx[level].P);CHKERRQ(ierr);
637e14861a4SHong Zhang       mlmat = &(ml_object->Rmat[mllevel-1]);
638db571536SBarry Smith       ierr  = MatWrapML_SeqAIJ(mlmat,MAT_INITIAL_MATRIX,&gridctx[level].R);CHKERRQ(ierr);
639573998d7SHong Zhang 
640573998d7SHong Zhang       mlmat = &(ml_object->Amat[mllevel]);
641573998d7SHong Zhang       ierr  = MatWrapML_SeqAIJ(mlmat,MAT_INITIAL_MATRIX,&gridctx[level].A);CHKERRQ(ierr);
6425582bec1SHong Zhang       level--;
6435582bec1SHong Zhang     }
644ab718edeSHong Zhang   } else { /* convert ML P and R into shell format, ML A into mpiaij format */
6455582bec1SHong Zhang     for (mllevel=1; mllevel<Nlevels; mllevel++){
6465582bec1SHong Zhang       mlmat  = &(ml_object->Pmat[mllevel]);
647db571536SBarry Smith       ierr = MatWrapML_SHELL(mlmat,MAT_INITIAL_MATRIX,&gridctx[level].P);CHKERRQ(ierr);
648ab718edeSHong Zhang       mlmat  = &(ml_object->Rmat[mllevel-1]);
649db571536SBarry Smith       ierr = MatWrapML_SHELL(mlmat,MAT_INITIAL_MATRIX,&gridctx[level].R);CHKERRQ(ierr);
650573998d7SHong Zhang 
6515582bec1SHong Zhang       mlmat  = &(ml_object->Amat[mllevel]);
652eef31507SHong Zhang       ierr = MatWrapML_MPIAIJ(mlmat,&gridctx[level].A);CHKERRQ(ierr);
6535582bec1SHong Zhang       level--;
6545582bec1SHong Zhang     }
6555582bec1SHong Zhang   }
6565582bec1SHong Zhang 
657573998d7SHong Zhang   /* create vectors and ksp at all levels */
658ac346b81SHong Zhang   for (level=0; level<fine_level; level++){
659573998d7SHong Zhang     level1 = level + 1;
660e64afeacSLisandro Dalcin     ierr = VecCreate(((PetscObject)gridctx[level].A)->comm,&gridctx[level].x);CHKERRQ(ierr);
661d0f46423SBarry Smith     ierr = VecSetSizes(gridctx[level].x,gridctx[level].A->cmap->n,PETSC_DECIDE);CHKERRQ(ierr);
6625582bec1SHong Zhang     ierr = VecSetType(gridctx[level].x,VECMPI);CHKERRQ(ierr);
66397177400SBarry Smith     ierr = PCMGSetX(pc,level,gridctx[level].x);CHKERRQ(ierr);
6645582bec1SHong Zhang 
665e64afeacSLisandro Dalcin     ierr = VecCreate(((PetscObject)gridctx[level].A)->comm,&gridctx[level].b);CHKERRQ(ierr);
666d0f46423SBarry Smith     ierr = VecSetSizes(gridctx[level].b,gridctx[level].A->rmap->n,PETSC_DECIDE);CHKERRQ(ierr);
6675582bec1SHong Zhang     ierr = VecSetType(gridctx[level].b,VECMPI);CHKERRQ(ierr);
66897177400SBarry Smith     ierr = PCMGSetRhs(pc,level,gridctx[level].b);CHKERRQ(ierr);
669ac346b81SHong Zhang 
670e64afeacSLisandro Dalcin     ierr = VecCreate(((PetscObject)gridctx[level1].A)->comm,&gridctx[level1].r);CHKERRQ(ierr);
671d0f46423SBarry Smith     ierr = VecSetSizes(gridctx[level1].r,gridctx[level1].A->rmap->n,PETSC_DECIDE);CHKERRQ(ierr);
672ac346b81SHong Zhang     ierr = VecSetType(gridctx[level1].r,VECMPI);CHKERRQ(ierr);
67397177400SBarry Smith     ierr = PCMGSetR(pc,level1,gridctx[level1].r);CHKERRQ(ierr);
674ac346b81SHong Zhang 
6755582bec1SHong Zhang     if (level == 0){
67697177400SBarry Smith       ierr = PCMGGetCoarseSolve(pc,&gridctx[level].ksp);CHKERRQ(ierr);
6775582bec1SHong Zhang     } else {
67897177400SBarry Smith       ierr = PCMGGetSmoother(pc,level,&gridctx[level].ksp);CHKERRQ(ierr);
679573998d7SHong Zhang     }
680573998d7SHong Zhang   }
681573998d7SHong Zhang   ierr = PCMGGetSmoother(pc,fine_level,&gridctx[fine_level].ksp);CHKERRQ(ierr);
682573998d7SHong Zhang 
683573998d7SHong Zhang   /* create coarse level and the interpolation between the levels */
684573998d7SHong Zhang   for (level=0; level<fine_level; level++){
685573998d7SHong Zhang     level1 = level + 1;
686aea2a34eSBarry Smith     ierr = PCMGSetInterpolation(pc,level1,gridctx[level].P);CHKERRQ(ierr);
687573998d7SHong Zhang     ierr = PCMGSetRestriction(pc,level1,gridctx[level].R);CHKERRQ(ierr);
688573998d7SHong Zhang     if (level > 0){
68997177400SBarry Smith       ierr = PCMGSetResidual(pc,level,PCMGDefaultResidual,gridctx[level].A);CHKERRQ(ierr);
6905582bec1SHong Zhang     }
6915582bec1SHong Zhang     ierr = KSPSetOperators(gridctx[level].ksp,gridctx[level].A,gridctx[level].A,DIFFERENT_NONZERO_PATTERN);CHKERRQ(ierr);
6925582bec1SHong Zhang   }
69397177400SBarry Smith   ierr = PCMGSetResidual(pc,fine_level,PCMGDefaultResidual,gridctx[fine_level].A);CHKERRQ(ierr);
694ac346b81SHong Zhang   ierr = KSPSetOperators(gridctx[fine_level].ksp,gridctx[level].A,gridctx[fine_level].A,DIFFERENT_NONZERO_PATTERN);CHKERRQ(ierr);
6955582bec1SHong Zhang 
696c07bf074SBarry Smith   /* setupcalled is set to 0 so that MG is setup from scratch */
697c07bf074SBarry Smith   pc->setupcalled = 0;
6983751b4bdSBarry Smith   ierr = PCSetUp_MG(pc);CHKERRQ(ierr);
6995582bec1SHong Zhang   PetscFunctionReturn(0);
7005582bec1SHong Zhang }
7015582bec1SHong Zhang 
7025582bec1SHong Zhang /* -------------------------------------------------------------------------- */
7035582bec1SHong Zhang /*
7045582bec1SHong Zhang    PCDestroy_ML - Destroys the private context for the ML preconditioner
7055582bec1SHong Zhang    that was created with PCCreate_ML().
7065582bec1SHong Zhang 
7075582bec1SHong Zhang    Input Parameter:
7085582bec1SHong Zhang .  pc - the preconditioner context
7095582bec1SHong Zhang 
7105582bec1SHong Zhang    Application Interface Routine: PCDestroy()
7115582bec1SHong Zhang */
7125582bec1SHong Zhang #undef __FUNCT__
7135582bec1SHong Zhang #define __FUNCT__ "PCDestroy_ML"
7146ca4d86aSHong Zhang PetscErrorCode PCDestroy_ML(PC pc)
7155582bec1SHong Zhang {
7165582bec1SHong Zhang   PetscErrorCode  ierr;
71701da6913SBarry Smith   PC_MG           *mg = (PC_MG*)pc->data;
71801da6913SBarry Smith   PC_ML           *pc_ml= (PC_ML*)mg->innerctx;
7195582bec1SHong Zhang 
7205582bec1SHong Zhang   PetscFunctionBegin;
72116336fedSMatthew G Knepley   ierr = PCReset_ML(pc);CHKERRQ(ierr);
72201da6913SBarry Smith   ierr = PetscFree(pc_ml);CHKERRQ(ierr);
72301da6913SBarry Smith   ierr = PCDestroy_MG(pc);CHKERRQ(ierr);
7245582bec1SHong Zhang   PetscFunctionReturn(0);
7255582bec1SHong Zhang }
7265582bec1SHong Zhang 
7275582bec1SHong Zhang #undef __FUNCT__
7285582bec1SHong Zhang #define __FUNCT__ "PCSetFromOptions_ML"
7296ca4d86aSHong Zhang PetscErrorCode PCSetFromOptions_ML(PC pc)
7305582bec1SHong Zhang {
7315582bec1SHong Zhang   PetscErrorCode  ierr;
7323751b4bdSBarry Smith   PetscInt        indx,PrintLevel;
7335582bec1SHong Zhang   const char      *scheme[] = {"Uncoupled","Coupled","MIS","METIS"};
73401da6913SBarry Smith   PC_MG           *mg = (PC_MG*)pc->data;
73501da6913SBarry Smith   PC_ML           *pc_ml = (PC_ML*)mg->innerctx;
736b5c8bdf8SJed Brown   PetscMPIInt     size;
73788ff4cc7SJed Brown   MPI_Comm        comm = ((PetscObject)pc)->comm;
7385582bec1SHong Zhang 
7395582bec1SHong Zhang   PetscFunctionBegin;
74088ff4cc7SJed Brown   ierr = MPI_Comm_size(comm,&size);CHKERRQ(ierr);
7415582bec1SHong Zhang   ierr = PetscOptionsHead("ML options");CHKERRQ(ierr);
7425582bec1SHong Zhang   PrintLevel    = 0;
7435582bec1SHong Zhang   indx          = 0;
7445582bec1SHong Zhang   ierr = PetscOptionsInt("-pc_ml_PrintLevel","Print level","ML_Set_PrintLevel",PrintLevel,&PrintLevel,PETSC_NULL);CHKERRQ(ierr);
7455582bec1SHong Zhang   ML_Set_PrintLevel(PrintLevel);
746573998d7SHong Zhang   ierr = PetscOptionsInt("-pc_ml_maxNlevels","Maximum number of levels","None",pc_ml->MaxNlevels,&pc_ml->MaxNlevels,PETSC_NULL);CHKERRQ(ierr);
747573998d7SHong Zhang   ierr = PetscOptionsInt("-pc_ml_maxCoarseSize","Maximum coarsest mesh size","ML_Aggregate_Set_MaxCoarseSize",pc_ml->MaxCoarseSize,&pc_ml->MaxCoarseSize,PETSC_NULL);CHKERRQ(ierr);
7483751b4bdSBarry Smith   ierr = PetscOptionsEList("-pc_ml_CoarsenScheme","Aggregate Coarsen Scheme","ML_Aggregate_Set_CoarsenScheme_*",scheme,4,scheme[0],&indx,PETSC_NULL);CHKERRQ(ierr);
7495582bec1SHong Zhang   pc_ml->CoarsenScheme = indx;
750573998d7SHong Zhang   ierr = PetscOptionsReal("-pc_ml_DampingFactor","P damping factor","ML_Aggregate_Set_DampingFactor",pc_ml->DampingFactor,&pc_ml->DampingFactor,PETSC_NULL);CHKERRQ(ierr);
751573998d7SHong Zhang   ierr = PetscOptionsReal("-pc_ml_Threshold","Smoother drop tol","ML_Aggregate_Set_Threshold",pc_ml->Threshold,&pc_ml->Threshold,PETSC_NULL);CHKERRQ(ierr);
752acfcf0e5SJed Brown   ierr = PetscOptionsBool("-pc_ml_SpectralNormScheme_Anorm","Method used for estimating spectral radius","ML_Set_SpectralNormScheme_Anorm",pc_ml->SpectralNormScheme_Anorm,&pc_ml->SpectralNormScheme_Anorm,PETSC_NULL);CHKERRQ(ierr);
753acfcf0e5SJed Brown   ierr = PetscOptionsBool("-pc_ml_Symmetrize","Symmetrize aggregation","ML_Set_Symmetrize",pc_ml->Symmetrize,&pc_ml->Symmetrize,PETSC_NULL);CHKERRQ(ierr);
754acfcf0e5SJed Brown   ierr = PetscOptionsBool("-pc_ml_BlockScaling","Scale all dofs at each node together","None",pc_ml->BlockScaling,&pc_ml->BlockScaling,PETSC_NULL);CHKERRQ(ierr);
755b5c8bdf8SJed Brown   ierr = PetscOptionsInt("-pc_ml_EnergyMinimization","Energy minimization norm type (0=no minimization; see ML manual for 1,2,3; -1 and 4 undocumented)","None",pc_ml->EnergyMinimization,&pc_ml->EnergyMinimization,PETSC_NULL);CHKERRQ(ierr);
756b5c8bdf8SJed Brown   /*
757b5c8bdf8SJed Brown     The following checks a number of conditions.  If we let this stuff slip by, then ML's error handling will take over.
758b5c8bdf8SJed Brown     This is suboptimal because it amounts to calling exit(1) so we check for the most common conditions.
759b5c8bdf8SJed Brown 
760b5c8bdf8SJed Brown     We also try to set some sane defaults when energy minimization is activated, otherwise it's hard to find a working
761b5c8bdf8SJed Brown     combination of options and ML's exit(1) explanations don't help matters.
762b5c8bdf8SJed Brown   */
76388ff4cc7SJed Brown   if (pc_ml->EnergyMinimization < -1 || pc_ml->EnergyMinimization > 4) SETERRQ(comm,PETSC_ERR_ARG_OUTOFRANGE,"EnergyMinimization must be in range -1..4");
76488ff4cc7SJed Brown   if (pc_ml->EnergyMinimization == 4 && size > 1) SETERRQ(comm,PETSC_ERR_SUP,"Energy minimization type 4 does not work in parallel");
765b5c8bdf8SJed Brown   if (pc_ml->EnergyMinimization == 4) {ierr = PetscInfo(pc,"Mandel's energy minimization scheme is experimental and broken in ML-6.2");CHKERRQ(ierr);}
766b5c8bdf8SJed Brown   if (pc_ml->EnergyMinimization) {
767b5c8bdf8SJed Brown     ierr = PetscOptionsReal("-pc_ml_EnergyMinimizationDropTol","Energy minimization drop tolerance","None",pc_ml->EnergyMinimizationDropTol,&pc_ml->EnergyMinimizationDropTol,PETSC_NULL);CHKERRQ(ierr);
768b5c8bdf8SJed Brown   }
769b5c8bdf8SJed Brown   if (pc_ml->EnergyMinimization == 2) {
770b5c8bdf8SJed Brown     /* According to ml_MultiLevelPreconditioner.cpp, this option is only meaningful for norm type (2) */
771acfcf0e5SJed Brown     ierr = PetscOptionsBool("-pc_ml_EnergyMinimizationCheap","Use cheaper variant of norm type 2","None",pc_ml->EnergyMinimizationCheap,&pc_ml->EnergyMinimizationCheap,PETSC_NULL);CHKERRQ(ierr);
772b5c8bdf8SJed Brown   }
773b5c8bdf8SJed Brown   /* energy minimization sometimes breaks if this is turned off, the more classical stuff should be okay without it */
774b5c8bdf8SJed Brown   if (pc_ml->EnergyMinimization) pc_ml->KeepAggInfo = PETSC_TRUE;
775acfcf0e5SJed Brown   ierr = PetscOptionsBool("-pc_ml_KeepAggInfo","Allows the preconditioner to be reused, or auxilliary matrices to be generated","None",pc_ml->KeepAggInfo,&pc_ml->KeepAggInfo,PETSC_NULL);CHKERRQ(ierr);
776b5c8bdf8SJed Brown   /* Option (-1) doesn't work at all (calls exit(1)) if the tentative restriction operator isn't stored. */
777b5c8bdf8SJed Brown   if (pc_ml->EnergyMinimization == -1) pc_ml->Reusable = PETSC_TRUE;
778acfcf0e5SJed Brown   ierr = PetscOptionsBool("-pc_ml_Reusable","Store intermedaiate data structures so that the multilevel hierarchy is reusable","None",pc_ml->Reusable,&pc_ml->Reusable,PETSC_NULL);CHKERRQ(ierr);
779b5c8bdf8SJed Brown   /*
780b5c8bdf8SJed Brown     ML's C API is severely underdocumented and lacks significant functionality.  The C++ API calls
781b5c8bdf8SJed Brown     ML_Gen_MultiLevelHierarchy_UsingAggregation() which is a modified copy (!?) of the documented function
782b5c8bdf8SJed Brown     ML_Gen_MGHierarchy_UsingAggregation().  This modification, however, does not provide a strict superset of the
783b5c8bdf8SJed Brown     functionality in the old function, so some users may still want to use it.  Note that many options are ignored in
784b5c8bdf8SJed Brown     this context, but ML doesn't provide a way to find out which ones.
785b5c8bdf8SJed Brown    */
786acfcf0e5SJed Brown   ierr = PetscOptionsBool("-pc_ml_OldHierarchy","Use old routine to generate hierarchy","None",pc_ml->OldHierarchy,&pc_ml->OldHierarchy,PETSC_NULL);CHKERRQ(ierr);
7875582bec1SHong Zhang   ierr = PetscOptionsTail();CHKERRQ(ierr);
7885582bec1SHong Zhang   PetscFunctionReturn(0);
7895582bec1SHong Zhang }
7905582bec1SHong Zhang 
7915582bec1SHong Zhang /* -------------------------------------------------------------------------- */
7925582bec1SHong Zhang /*
7935582bec1SHong Zhang    PCCreate_ML - Creates a ML preconditioner context, PC_ML,
7945582bec1SHong Zhang    and sets this as the private data within the generic preconditioning
7955582bec1SHong Zhang    context, PC, that was created within PCCreate().
7965582bec1SHong Zhang 
7975582bec1SHong Zhang    Input Parameter:
7985582bec1SHong Zhang .  pc - the preconditioner context
7995582bec1SHong Zhang 
8005582bec1SHong Zhang    Application Interface Routine: PCCreate()
8015582bec1SHong Zhang */
8025582bec1SHong Zhang 
8035582bec1SHong Zhang /*MC
8041e5ab15bSHong Zhang      PCML - Use algebraic multigrid preconditioning. This preconditioner requires you provide
8055582bec1SHong Zhang        fine grid discretization matrix. The coarser grid matrices and restriction/interpolation
8066ca4d86aSHong Zhang        operators are computed by ML, with the matrices coverted to PETSc matrices in aij format
8076ca4d86aSHong Zhang        and the restriction/interpolation operators wrapped as PETSc shell matrices.
8085582bec1SHong Zhang 
8096ca4d86aSHong Zhang    Options Database Key:
8106ca4d86aSHong Zhang    Multigrid options(inherited)
8116ca4d86aSHong Zhang +  -pc_mg_cycles <1>: 1 for V cycle, 2 for W-cycle (MGSetCycles)
8126ca4d86aSHong Zhang .  -pc_mg_smoothup <1>: Number of post-smoothing steps (MGSetNumberSmoothUp)
8136ca4d86aSHong Zhang .  -pc_mg_smoothdown <1>: Number of pre-smoothing steps (MGSetNumberSmoothDown)
814ccd75124SBarry Smith    -pc_mg_type <multiplicative>: (one of) additive multiplicative full cascade kascade
81551f519a2SBarry Smith    ML options:
816ccd75124SBarry Smith .  -pc_ml_PrintLevel <0>: Print level (ML_Set_PrintLevel)
8176ca4d86aSHong Zhang .  -pc_ml_maxNlevels <10>: Maximum number of levels (None)
8186ca4d86aSHong Zhang .  -pc_ml_maxCoarseSize <1>: Maximum coarsest mesh size (ML_Aggregate_Set_MaxCoarseSize)
819f41ab451SVictor Eijkhout .  -pc_ml_CoarsenScheme <Uncoupled>: (one of) Uncoupled Coupled MIS METIS
8206ca4d86aSHong Zhang .  -pc_ml_DampingFactor <1.33333>: P damping factor (ML_Aggregate_Set_DampingFactor)
8216ca4d86aSHong Zhang .  -pc_ml_Threshold <0>: Smoother drop tol (ML_Aggregate_Set_Threshold)
8227ffd031bSHong Zhang -  -pc_ml_SpectralNormScheme_Anorm <false>: Method used for estimating spectral radius (ML_Set_SpectralNormScheme_Anorm)
8235582bec1SHong Zhang 
8245582bec1SHong Zhang    Level: intermediate
8255582bec1SHong Zhang 
8265582bec1SHong Zhang   Concepts: multigrid
8275582bec1SHong Zhang 
8285582bec1SHong Zhang .seealso:  PCCreate(), PCSetType(), PCType (for list of available types), PC, PCMGType,
82997177400SBarry Smith            PCMGSetLevels(), PCMGGetLevels(), PCMGSetType(), MPSetCycles(), PCMGSetNumberSmoothDown(),
83097177400SBarry Smith            PCMGSetNumberSmoothUp(), PCMGGetCoarseSolve(), PCMGSetResidual(), PCMGSetInterpolation(),
83197177400SBarry Smith            PCMGSetRestriction(), PCMGGetSmoother(), PCMGGetSmootherUp(), PCMGGetSmootherDown(),
83297177400SBarry Smith            PCMGSetCyclesOnLevel(), PCMGSetRhs(), PCMGSetX(), PCMGSetR()
8335582bec1SHong Zhang M*/
8345582bec1SHong Zhang 
8355582bec1SHong Zhang EXTERN_C_BEGIN
8365582bec1SHong Zhang #undef __FUNCT__
8375582bec1SHong Zhang #define __FUNCT__ "PCCreate_ML"
8387087cfbeSBarry Smith PetscErrorCode  PCCreate_ML(PC pc)
8395582bec1SHong Zhang {
8405582bec1SHong Zhang   PetscErrorCode  ierr;
8415582bec1SHong Zhang   PC_ML           *pc_ml;
84201da6913SBarry Smith   PC_MG           *mg;
8435582bec1SHong Zhang 
8445582bec1SHong Zhang   PetscFunctionBegin;
845573998d7SHong Zhang   /* PCML is an inherited class of PCMG. Initialize pc as PCMG */
846c9e1c140SHong Zhang   ierr = PetscObjectChangeTypeName((PetscObject)pc,PCML);CHKERRQ(ierr);
8475582bec1SHong Zhang   ierr = PCSetType(pc,PCMG);CHKERRQ(ierr); /* calls PCCreate_MG() and MGCreate_Private() */
8485582bec1SHong Zhang 
8495582bec1SHong Zhang   /* create a supporting struct and attach it to pc */
85038f2d2fdSLisandro Dalcin   ierr = PetscNewLog(pc,PC_ML,&pc_ml);CHKERRQ(ierr);
85101da6913SBarry Smith   mg = (PC_MG*)pc->data;
85201da6913SBarry Smith   mg->innerctx = pc_ml;
8535582bec1SHong Zhang 
854573998d7SHong Zhang   pc_ml->ml_object     = 0;
855573998d7SHong Zhang   pc_ml->agg_object    = 0;
856573998d7SHong Zhang   pc_ml->gridctx       = 0;
857573998d7SHong Zhang   pc_ml->PetscMLdata   = 0;
858573998d7SHong Zhang   pc_ml->Nlevels       = -1;
859573998d7SHong Zhang   pc_ml->MaxNlevels    = 10;
860573998d7SHong Zhang   pc_ml->MaxCoarseSize = 1;
8613751b4bdSBarry Smith   pc_ml->CoarsenScheme = 1;
862573998d7SHong Zhang   pc_ml->Threshold     = 0.0;
863573998d7SHong Zhang   pc_ml->DampingFactor = 4.0/3.0;
864573998d7SHong Zhang   pc_ml->SpectralNormScheme_Anorm = PETSC_FALSE;
865573998d7SHong Zhang   pc_ml->size          = 0;
866573998d7SHong Zhang 
8675582bec1SHong Zhang   /* overwrite the pointers of PCMG by the functions of PCML */
8685582bec1SHong Zhang   pc->ops->setfromoptions = PCSetFromOptions_ML;
8695582bec1SHong Zhang   pc->ops->setup          = PCSetUp_ML;
870a06653b4SBarry Smith   pc->ops->reset          = PCReset_ML;
8715582bec1SHong Zhang   pc->ops->destroy        = PCDestroy_ML;
8725582bec1SHong Zhang   PetscFunctionReturn(0);
8735582bec1SHong Zhang }
8745582bec1SHong Zhang EXTERN_C_END
875