xref: /petsc/src/ksp/pc/impls/gamg/gamg.c (revision 03a628fea78312efee5eccebb643bd7972f1fa85)
15b89ad90SMark F. Adams /*
25b89ad90SMark F. Adams  GAMG geometric-algebric multiogrid PC - Mark Adams 2011
35b89ad90SMark F. Adams  */
45ef31b24SMark F. Adams #include <../src/ksp/pc/impls/gamg/gamg.h>
558471d46SMark F. Adams #include "private/matimpl.h"          /*I "petscmat.h" I*/
6f96513f1SMatthew G Knepley 
7b4fbaa2aSMark F. Adams #if defined PETSC_USE_LOG
8b4fbaa2aSMark F. Adams PetscLogEvent gamg_setup_events[NUM_SET];
9b4fbaa2aSMark F. Adams #endif
10b4fbaa2aSMark F. Adams #define GAMG_MAXLEVELS 30
11b4fbaa2aSMark F. Adams 
12b8fd24d8SMark F. Adams /*#define GAMG_STAGES*/
13b4fbaa2aSMark F. Adams #if (defined PETSC_USE_LOG && defined GAMG_STAGES)
14b4fbaa2aSMark F. Adams static PetscLogStage gamg_stages[GAMG_MAXLEVELS];
15b4fbaa2aSMark F. Adams #endif
16f96513f1SMatthew G Knepley 
175b89ad90SMark F. Adams /* Private context for the GAMG preconditioner */
185b89ad90SMark F. Adams typedef struct gamg_TAG {
195b89ad90SMark F. Adams   PetscInt       m_dim;
205b89ad90SMark F. Adams   PetscInt       m_Nlevels;
215b89ad90SMark F. Adams   PetscInt       m_data_sz;
22d3d6bff4SMark F. Adams   PetscInt       m_data_rows;
23d3d6bff4SMark F. Adams   PetscInt       m_data_cols;
24*03a628feSMark F. Adams   PetscInt       m_count;
25d3d6bff4SMark F. Adams   PetscBool      m_useSA;
265b89ad90SMark F. Adams   PetscReal     *m_data; /* blocked vector of vertex data on fine grid (coordinates) */
275b89ad90SMark F. Adams } PC_GAMG;
285b89ad90SMark F. Adams 
295b89ad90SMark F. Adams /* -------------------------------------------------------------------------- */
305b89ad90SMark F. Adams /*
315b89ad90SMark F. Adams    PCSetCoordinates_GAMG
325b89ad90SMark F. Adams 
335b89ad90SMark F. Adams    Input Parameter:
345b89ad90SMark F. Adams    .  pc - the preconditioner context
355b89ad90SMark F. Adams */
36a92563c5SMark F. Adams EXTERN_C_BEGIN
375b89ad90SMark F. Adams #undef __FUNCT__
385b89ad90SMark F. Adams #define __FUNCT__ "PCSetCoordinates_GAMG"
39eb07cef2SMark F. Adams PetscErrorCode PCSetCoordinates_GAMG( PC a_pc, PetscInt a_ndm, PetscReal *a_coords )
405b89ad90SMark F. Adams {
41eb07cef2SMark F. Adams   PC_MG          *mg = (PC_MG*)a_pc->data;
425b89ad90SMark F. Adams   PC_GAMG        *pc_gamg = (PC_GAMG*)mg->innerctx;
436c237d78SBarry Smith   PetscErrorCode ierr;
44d3d6bff4SMark F. Adams   PetscInt       arrsz,bs,my0,kk,ii,jj,nloc,Iend;
45038e3b61SMark F. Adams   Mat            Amat = a_pc->pmat;
46d3d6bff4SMark F. Adams   PetscBool      flag;
47785cba28SMark F. Adams   char           str[64];
485b89ad90SMark F. Adams 
495b89ad90SMark F. Adams   PetscFunctionBegin;
5058471d46SMark F. Adams   PetscValidHeaderSpecific( Amat, MAT_CLASSID, 1 );
51038e3b61SMark F. Adams   ierr  = MatGetBlockSize( Amat, &bs );               CHKERRQ( ierr );
52d3d6bff4SMark F. Adams   ierr  = MatGetOwnershipRange( Amat, &my0, &Iend ); CHKERRQ(ierr);
53d3d6bff4SMark F. Adams   nloc = (Iend-my0)/bs;
54d3d6bff4SMark F. Adams   if((Iend-my0)%bs!=0) SETERRQ1(((PetscObject)Amat)->comm,PETSC_ERR_ARG_WRONG, "Bad local size %d.",nloc);
55038e3b61SMark F. Adams 
56785cba28SMark F. Adams   ierr  = PetscOptionsGetString(PETSC_NULL,"-pc_gamg_type",str,64,&flag);    CHKERRQ( ierr );
57d3d6bff4SMark F. Adams   pc_gamg->m_useSA = (PetscBool)(flag && strcmp(str,"sa") == 0);
58038e3b61SMark F. Adams 
59d3d6bff4SMark F. Adams   pc_gamg->m_data_rows = 1;
60d3d6bff4SMark F. Adams   if(a_coords == 0) pc_gamg->m_useSA = PETSC_TRUE; /* use SA if no data */
61d3d6bff4SMark F. Adams   if( !pc_gamg->m_useSA ) pc_gamg->m_data_cols = a_ndm; /* coordinates */
62038e3b61SMark F. Adams   else{ /* SA: null space vectors */
63d3d6bff4SMark F. Adams     if(a_coords != 0 && bs==1 ) pc_gamg->m_data_cols = 1; /* scalar w/ coords and SA (not needed) */
64d3d6bff4SMark F. Adams     else if(a_coords != 0) pc_gamg->m_data_cols = (a_ndm==2 ? 3 : 6); /* elasticity */
65d3d6bff4SMark F. Adams     else pc_gamg->m_data_cols = bs; /* no data, force SA with constant null space vectors */
66d3d6bff4SMark F. Adams     pc_gamg->m_data_rows = bs;
67038e3b61SMark F. Adams   }
68d3d6bff4SMark F. Adams   arrsz = nloc*pc_gamg->m_data_rows*pc_gamg->m_data_cols;
695b89ad90SMark F. Adams 
70038e3b61SMark F. Adams   /* create data - syntactic sugar that should be refactored at some point */
716e3e101aSMark F. Adams   if (pc_gamg->m_data==0 || (pc_gamg->m_data_sz != arrsz)) {
725b89ad90SMark F. Adams     ierr = PetscFree( pc_gamg->m_data );  CHKERRQ(ierr);
73eb07cef2SMark F. Adams     ierr = PetscMalloc((arrsz+1)*sizeof(double), &pc_gamg->m_data ); CHKERRQ(ierr);
745b89ad90SMark F. Adams   }
75038e3b61SMark F. Adams   for(kk=0;kk<arrsz;kk++)pc_gamg->m_data[kk] = -999.;
76eb07cef2SMark F. Adams   pc_gamg->m_data[arrsz] = -99.;
77038e3b61SMark F. Adams   /* copy data in - column oriented */
78d3d6bff4SMark F. Adams   if( pc_gamg->m_useSA ) {
79d3d6bff4SMark F. Adams     const PetscInt M = Iend - my0;
80038e3b61SMark F. Adams     for(kk=0;kk<nloc;kk++){
81038e3b61SMark F. Adams       PetscReal *data = &pc_gamg->m_data[kk*bs];
82d3d6bff4SMark F. Adams       if( pc_gamg->m_data_cols==1 ) *data = 1.0;
83038e3b61SMark F. Adams       else {
84038e3b61SMark F. Adams         for(ii=0;ii<bs;ii++)
85038e3b61SMark F. Adams 	  for(jj=0;jj<bs;jj++)
86038e3b61SMark F. Adams 	    if(ii==jj)data[ii*M + jj] = 1.0; /* translational modes */
87038e3b61SMark F. Adams 	    else data[ii*M + jj] = 0.0;
88eb07cef2SMark F. Adams         if( a_coords != 0 ) {
89038e3b61SMark F. Adams           if( a_ndm == 2 ){ /* rotational modes */
90038e3b61SMark F. Adams             data += 2*M;
91038e3b61SMark F. Adams             data[0] = -a_coords[2*kk+1];
92038e3b61SMark F. Adams             data[1] =  a_coords[2*kk];
935b89ad90SMark F. Adams           }
94eb07cef2SMark F. Adams           else {
95038e3b61SMark F. Adams             data += 3*M;
96038e3b61SMark F. Adams             data[0] = 0.0;               data[M+0] =  a_coords[3*kk+2]; data[2*M+0] = -a_coords[3*kk+1];
97038e3b61SMark F. Adams             data[1] = -a_coords[3*kk+2]; data[M+1] = 0.0;               data[2*M+1] =  a_coords[3*kk];
98038e3b61SMark F. Adams             data[2] =  a_coords[3*kk+1]; data[M+2] = -a_coords[3*kk];   data[2*M+2] = 0.0;
99038e3b61SMark F. Adams           }
100eb07cef2SMark F. Adams         }
101eb07cef2SMark F. Adams       }
102eb07cef2SMark F. Adams     }
103eb07cef2SMark F. Adams   }
104eb07cef2SMark F. Adams   else {
105038e3b61SMark F. Adams     for( kk = 0 ; kk < nloc ; kk++ ){
106038e3b61SMark F. Adams       for( ii = 0 ; ii < a_ndm ; ii++ ) {
107038e3b61SMark F. Adams         pc_gamg->m_data[ii*nloc + kk] =  a_coords[kk*a_ndm + ii];
108eb07cef2SMark F. Adams       }
109eb07cef2SMark F. Adams     }
110038e3b61SMark F. Adams   }
111eb07cef2SMark F. Adams   assert(pc_gamg->m_data[arrsz] == -99.);
112038e3b61SMark F. Adams 
1135b89ad90SMark F. Adams   pc_gamg->m_data_sz = arrsz;
114eb07cef2SMark F. Adams   pc_gamg->m_dim = a_ndm;
1155b89ad90SMark F. Adams 
1165b89ad90SMark F. Adams   PetscFunctionReturn(0);
1175b89ad90SMark F. Adams }
118a92563c5SMark F. Adams EXTERN_C_END
1195b89ad90SMark F. Adams 
120d3d6bff4SMark F. Adams 
121d3d6bff4SMark F. Adams /* -----------------------------------------------------------------------------*/
122d3d6bff4SMark F. Adams #undef __FUNCT__
123d3d6bff4SMark F. Adams #define __FUNCT__ "PCReset_GAMG"
124d3d6bff4SMark F. Adams PetscErrorCode PCReset_GAMG(PC pc)
125d3d6bff4SMark F. Adams {
126d3d6bff4SMark F. Adams   PetscErrorCode  ierr;
127d3d6bff4SMark F. Adams   PC_MG           *mg = (PC_MG*)pc->data;
128d3d6bff4SMark F. Adams   PC_GAMG         *pc_gamg = (PC_GAMG*)mg->innerctx;
129d3d6bff4SMark F. Adams 
130d3d6bff4SMark F. Adams   PetscFunctionBegin;
13158471d46SMark F. Adams   if( pc_gamg->m_data != 0 ) { /* this should not happen, cleaned up in SetUp */
132d3d6bff4SMark F. Adams     ierr = PetscFree(pc_gamg->m_data); CHKERRQ(ierr);
13358471d46SMark F. Adams   }
134d3d6bff4SMark F. Adams   pc_gamg->m_data = 0; pc_gamg->m_data_sz = 0;
135d3d6bff4SMark F. Adams   PetscFunctionReturn(0);
136d3d6bff4SMark F. Adams }
137d3d6bff4SMark F. Adams 
1385b89ad90SMark F. Adams /* -------------------------------------------------------------------------- */
1395b89ad90SMark F. Adams /*
14011e60469SMark F. Adams    partitionLevel
1415b89ad90SMark F. Adams 
1425b89ad90SMark F. Adams    Input Parameter:
1433530afc2SMark F. Adams    . a_Amat_fine - matrix on this fine (k) level
144d3d6bff4SMark F. Adams    . a_ndata_rows - size of data to move (coarse grid)
145d3d6bff4SMark F. Adams    . a_ndata_cols - size of data to move (coarse grid)
1463530afc2SMark F. Adams    In/Output Parameter:
1473530afc2SMark F. Adams    . a_P_inout - prolongation operator to the next level (k-1)
148eb07cef2SMark F. Adams    . a_coarse_data - data that need to be moved
149afc97cdcSMark F. Adams    . a_nactive_proc - number of active procs
15011e60469SMark F. Adams    Output Parameter:
1513530afc2SMark F. Adams    . a_Amat_crs - coarse matrix that is created (k-1)
1525b89ad90SMark F. Adams */
1535cb416c2SMark F. Adams 
15458471d46SMark F. Adams #define MIN_EQ_PROC 600
155cc4ca5ccSMark F. Adams #define TOP_GRID_LIM 2*MIN_EQ_PROC /* this will happen anyway */
15622063be5SMark F. Adams 
1575b89ad90SMark F. Adams #undef __FUNCT__
15811e60469SMark F. Adams #define __FUNCT__ "partitionLevel"
1593530afc2SMark F. Adams PetscErrorCode partitionLevel( Mat a_Amat_fine,
160d3d6bff4SMark F. Adams                                PetscInt a_ndata_rows,
161d3d6bff4SMark F. Adams                                PetscInt a_ndata_cols,
162038e3b61SMark F. Adams 			       PetscInt a_cbs,
1633530afc2SMark F. Adams                                Mat *a_P_inout,
164eb07cef2SMark F. Adams                                PetscReal **a_coarse_data,
165afc97cdcSMark F. Adams                                PetscMPIInt *a_nactive_proc,
1663530afc2SMark F. Adams                                Mat *a_Amat_crs
16711e60469SMark F. Adams                                )
1685b89ad90SMark F. Adams {
1695b89ad90SMark F. Adams   PetscErrorCode   ierr;
170038e3b61SMark F. Adams   Mat              Cmat,Pnew,Pold=*a_P_inout;
17111e60469SMark F. Adams   IS               new_indices,isnum;
1723530afc2SMark F. Adams   MPI_Comm         wcomm = ((PetscObject)a_Amat_fine)->comm;
173fa31c87dSMark F. Adams   PetscMPIInt      mype,npe,new_npe,nactive;
174fa31c87dSMark F. Adams   PetscInt         neq,NN,Istart,Iend,Istart0,Iend0,ncrs_new,ncrs0;
175737a81a9SMark F. Adams   PetscBool        flag = PETSC_FALSE;
1765b89ad90SMark F. Adams 
1775b89ad90SMark F. Adams   PetscFunctionBegin;
17811e60469SMark F. Adams   ierr = MPI_Comm_rank( wcomm, &mype ); CHKERRQ(ierr);
17911e60469SMark F. Adams   ierr = MPI_Comm_size( wcomm, &npe );  CHKERRQ(ierr);
18011e60469SMark F. Adams   /* RAP */
181038e3b61SMark F. Adams   ierr = MatPtAP( a_Amat_fine, Pold, MAT_INITIAL_MATRIX, 2.0, &Cmat ); CHKERRQ(ierr);
182737a81a9SMark F. Adams 
183038e3b61SMark F. Adams   ierr = MatSetBlockSize( Cmat, a_cbs );      CHKERRQ(ierr);
184038e3b61SMark F. Adams   ierr = MatGetOwnershipRange( Cmat, &Istart0, &Iend0 ); CHKERRQ(ierr);
185038e3b61SMark F. Adams   ncrs0 = (Iend0-Istart0)/a_cbs; assert((Iend0-Istart0)%a_cbs == 0);
186038e3b61SMark F. Adams 
187737a81a9SMark F. Adams   ierr  = PetscOptionsHasName(PETSC_NULL,"-pc_gamg_avoid_repartitioning",&flag);
188737a81a9SMark F. Adams   CHKERRQ( ierr );
18922063be5SMark F. Adams   if( flag ) {
19022063be5SMark F. Adams     *a_Amat_crs = Cmat; /* output */
19122063be5SMark F. Adams   }
19222063be5SMark F. Adams   else {
19322063be5SMark F. Adams     /* Repartition Cmat_{k} and move colums of P^{k}_{k-1} and coordinates accordingly */
1945ef31b24SMark F. Adams     Mat              adj;
19522063be5SMark F. Adams     const PetscInt *idx,data_sz=a_ndata_rows*a_ndata_cols;
196b4fbaa2aSMark F. Adams     const PetscInt  stride0=ncrs0*a_ndata_rows;
19792a756f0SMark F. Adams     PetscInt        is_sz,*isnewproc_idx,ii,jj,kk,strideNew,*tidx;
198c9a0b8beSMark F. Adams     /* create sub communicator  */
199c9a0b8beSMark F. Adams     MPI_Comm        cm,new_comm;
200afc97cdcSMark F. Adams     MPI_Group       wg, g2;
201fd3c6afaSMark F. Adams     PetscInt       *counts,inpe;
202fd3c6afaSMark F. Adams     PetscMPIInt    *ranks;
20322063be5SMark F. Adams     IS              isscat;
20422063be5SMark F. Adams     PetscScalar    *array;
20522063be5SMark F. Adams     Vec             src_crd, dest_crd;
20622063be5SMark F. Adams     PetscReal      *data = *a_coarse_data;
20722063be5SMark F. Adams     VecScatter      vecscat;
208b4fbaa2aSMark F. Adams     IS  isnewproc;
209e33ef3b1SMark F. Adams 
21022063be5SMark F. Adams     /* get number of PEs to make active, reduce */
21122063be5SMark F. Adams     ierr = MatGetSize( Cmat, &neq, &NN );CHKERRQ(ierr);
21222063be5SMark F. Adams     new_npe = neq/MIN_EQ_PROC; /* hardwire min. number of eq/proc */
21322063be5SMark F. Adams     if( new_npe == 0 || neq < TOP_GRID_LIM ) new_npe = 1;
21422063be5SMark F. Adams     else if (new_npe >= *a_nactive_proc ) new_npe = *a_nactive_proc; /* no change, rare */
21522063be5SMark F. Adams 
21692a756f0SMark F. Adams     ierr = PetscMalloc( npe*sizeof(PetscMPIInt), &ranks ); CHKERRQ(ierr);
217fd3c6afaSMark F. Adams     ierr = PetscMalloc( npe*sizeof(PetscInt), &counts ); CHKERRQ(ierr);
21892a756f0SMark F. Adams 
219fd3c6afaSMark F. Adams     ierr = MPI_Allgather( &ncrs0, 1, MPIU_INT, counts, 1, MPIU_INT, wcomm ); CHKERRQ(ierr);
220afc97cdcSMark F. Adams     assert(counts[mype]==ncrs0);
221afc97cdcSMark F. Adams     /* count real active pes */
22222063be5SMark F. Adams     for( nactive = jj = 0 ; jj < npe ; jj++) {
223afc97cdcSMark F. Adams       if( counts[jj] != 0 ) {
22422063be5SMark F. Adams 	ranks[nactive++] = jj;
225afc97cdcSMark F. Adams       }
226afc97cdcSMark F. Adams     }
22722063be5SMark F. Adams     assert(nactive>=new_npe);
22822063be5SMark F. Adams 
22958471d46SMark F. Adams #ifdef VERBOSE
23022063be5SMark F. Adams     PetscPrintf(PETSC_COMM_WORLD,"\t[%d]%s npe (active): %d --> %d. new npe = %d, neq = %d\n",mype,__FUNCT__,*a_nactive_proc,nactive,new_npe,neq);
23158471d46SMark F. Adams #endif
23258471d46SMark F. Adams 
23322063be5SMark F. Adams     *a_nactive_proc = new_npe; /* output */
2342f03bc48SMark F. Adams 
235afc97cdcSMark F. Adams     ierr = MPI_Comm_group( wcomm, &wg ); CHKERRQ(ierr);
23622063be5SMark F. Adams     ierr = MPI_Group_incl( wg, nactive, ranks, &g2 ); CHKERRQ(ierr);
237afc97cdcSMark F. Adams     ierr = MPI_Comm_create( wcomm, g2, &cm ); CHKERRQ(ierr);
238b4fbaa2aSMark F. Adams 
239afc97cdcSMark F. Adams     if( cm != MPI_COMM_NULL ) {
240b4fbaa2aSMark F. Adams       assert(ncrs0 != 0);
241c9a0b8beSMark F. Adams       ierr = PetscCommDuplicate( cm, &new_comm, PETSC_NULL ); CHKERRQ(ierr);
242c9a0b8beSMark F. Adams       ierr = MPI_Comm_free( &cm );                             CHKERRQ(ierr);
243afc97cdcSMark F. Adams     }
244b4fbaa2aSMark F. Adams     else assert(ncrs0 == 0);
245b4fbaa2aSMark F. Adams 
246afc97cdcSMark F. Adams     ierr = MPI_Group_free( &wg );                            CHKERRQ(ierr);
247afc97cdcSMark F. Adams     ierr = MPI_Group_free( &g2 );                            CHKERRQ(ierr);
248c9a0b8beSMark F. Adams 
2495ef31b24SMark F. Adams     /* MatPartitioningApply call MatConvert, which is collective */
250b4fbaa2aSMark F. Adams #if defined PETSC_USE_LOG
251b4fbaa2aSMark F. Adams     ierr = PetscLogEventBegin(gamg_setup_events[SET12],0,0,0,0);CHKERRQ(ierr);
252b4fbaa2aSMark F. Adams #endif
253038e3b61SMark F. Adams     if( a_cbs == 1) {
254038e3b61SMark F. Adams       ierr = MatConvert( Cmat, MATMPIADJ, MAT_INITIAL_MATRIX, &adj );   CHKERRQ(ierr);
255eb07cef2SMark F. Adams     }
256eb07cef2SMark F. Adams     else{
257038e3b61SMark F. Adams       /* make a scalar matrix to partition */
258eb07cef2SMark F. Adams       Mat tMat;
25958471d46SMark F. Adams       PetscInt ncols,jj,Ii;
260b4fbaa2aSMark F. Adams       const PetscScalar *vals;
261b4fbaa2aSMark F. Adams       const PetscInt *idx;
26258471d46SMark F. Adams       PetscInt *d_nnz;
263b4fbaa2aSMark F. Adams 
26458471d46SMark F. Adams       ierr = PetscMalloc( ncrs0*sizeof(PetscInt), &d_nnz ); CHKERRQ(ierr);
26558471d46SMark F. Adams       for ( Ii = Istart0, jj = 0 ; Ii < Iend0 ; Ii += a_cbs, jj++ ) {
26658471d46SMark F. Adams         ierr = MatGetRow(Cmat,Ii,&ncols,0,0); CHKERRQ(ierr);
26758471d46SMark F. Adams         d_nnz[jj] = ncols/a_cbs;
26858471d46SMark F. Adams         if( d_nnz[jj] > ncrs0 ) d_nnz[jj] = ncrs0;
26958471d46SMark F. Adams         ierr = MatRestoreRow(Cmat,Ii,&ncols,0,0); CHKERRQ(ierr);
27058471d46SMark F. Adams       }
2716876a03eSMark F. Adams 
272eb07cef2SMark F. Adams       ierr = MatCreateMPIAIJ( wcomm, ncrs0, ncrs0,
273eb07cef2SMark F. Adams                               PETSC_DETERMINE, PETSC_DETERMINE,
27458471d46SMark F. Adams                               0, d_nnz, 0, d_nnz,
275eb07cef2SMark F. Adams                               &tMat );
2766876a03eSMark F. Adams       CHKERRQ(ierr);
27758471d46SMark F. Adams       ierr = PetscFree( d_nnz ); CHKERRQ(ierr);
278eb07cef2SMark F. Adams 
27922063be5SMark F. Adams       for ( ii = Istart0; ii < Iend0; ii++ ) {
28022063be5SMark F. Adams         PetscInt dest_row = ii/a_cbs;
28122063be5SMark F. Adams         ierr = MatGetRow(Cmat,ii,&ncols,&idx,&vals); CHKERRQ(ierr);
282eb07cef2SMark F. Adams         for( jj = 0 ; jj < ncols ; jj++ ){
283038e3b61SMark F. Adams           PetscInt dest_col = idx[jj]/a_cbs;
284eb07cef2SMark F. Adams           PetscScalar v = 1.0;
285eb07cef2SMark F. Adams           ierr = MatSetValues(tMat,1,&dest_row,1,&dest_col,&v,ADD_VALUES); CHKERRQ(ierr);
286eb07cef2SMark F. Adams         }
28722063be5SMark F. Adams         ierr = MatRestoreRow(Cmat,ii,&ncols,&idx,&vals); CHKERRQ(ierr);
288eb07cef2SMark F. Adams       }
289eb07cef2SMark F. Adams       ierr = MatAssemblyBegin(tMat,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
290eb07cef2SMark F. Adams       ierr = MatAssemblyEnd(tMat,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
291eb07cef2SMark F. Adams 
292b4fbaa2aSMark F. Adams       static int llev = 0;
293b4fbaa2aSMark F. Adams       if( llev++ == -1 ) {
294b4fbaa2aSMark F. Adams 	PetscViewer viewer; char fname[32];
295b4fbaa2aSMark F. Adams 	sprintf(fname,"part_mat_%d.mat",llev);
296b4fbaa2aSMark F. Adams 	PetscViewerBinaryOpen(wcomm,fname,FILE_MODE_WRITE,&viewer);
297b4fbaa2aSMark F. Adams 	ierr = MatView( tMat, viewer ); CHKERRQ(ierr);
298b4fbaa2aSMark F. Adams 	ierr = PetscViewerDestroy( &viewer );
299b4fbaa2aSMark F. Adams       }
300b4fbaa2aSMark F. Adams 
301eb07cef2SMark F. Adams       ierr = MatConvert( tMat, MATMPIADJ, MAT_INITIAL_MATRIX, &adj );   CHKERRQ(ierr);
302eb07cef2SMark F. Adams 
303eb07cef2SMark F. Adams       ierr = MatDestroy( &tMat );  CHKERRQ(ierr);
304eb07cef2SMark F. Adams     }
305afc97cdcSMark F. Adams     if( ncrs0 != 0 ){
306b4fbaa2aSMark F. Adams       const PetscInt *is_idx;
307b4fbaa2aSMark F. Adams       MatPartitioning  mpart;
3085ef31b24SMark F. Adams       /* hack to fix global data that pmetis.c uses in 'adj' */
30922063be5SMark F. Adams       for( nactive = jj = 0 ; jj < npe ; jj++ ) {
310afc97cdcSMark F. Adams 	if( counts[jj] != 0 ) {
31122063be5SMark F. Adams 	  adj->rmap->range[nactive++] = adj->rmap->range[jj];
312afc97cdcSMark F. Adams 	}
3135ef31b24SMark F. Adams       }
31422063be5SMark F. Adams       adj->rmap->range[nactive] = adj->rmap->range[npe];
3152f03bc48SMark F. Adams 
3165ef31b24SMark F. Adams       ierr = MatPartitioningCreate( new_comm, &mpart ); CHKERRQ(ierr);
3175ef31b24SMark F. Adams       ierr = MatPartitioningSetAdjacency( mpart, adj ); CHKERRQ(ierr);
31811e60469SMark F. Adams       ierr = MatPartitioningSetFromOptions( mpart );    CHKERRQ(ierr);
3195ef31b24SMark F. Adams       ierr = MatPartitioningSetNParts( mpart, new_npe );CHKERRQ(ierr);
32011e60469SMark F. Adams       ierr = MatPartitioningApply( mpart, &isnewproc ); CHKERRQ(ierr);
32111e60469SMark F. Adams       ierr = MatPartitioningDestroy( &mpart );          CHKERRQ(ierr);
32222063be5SMark F. Adams 
3235ef31b24SMark F. Adams       /* collect IS info */
3245ef31b24SMark F. Adams       ierr = ISGetLocalSize( isnewproc, &is_sz );       CHKERRQ(ierr);
325038e3b61SMark F. Adams       ierr = PetscMalloc( a_cbs*is_sz*sizeof(PetscInt), &isnewproc_idx ); CHKERRQ(ierr);
3265ef31b24SMark F. Adams       ierr = ISGetIndices( isnewproc, &is_idx );        CHKERRQ(ierr);
327b4fbaa2aSMark F. Adams       /* spread partitioning across machine - best way ??? */
328ecd57171SMark F. Adams       NN = 1; /*npe/new_npe;*/
329b4fbaa2aSMark F. Adams       for( kk = jj = 0 ; kk < is_sz ; kk++ ){
330afc97cdcSMark F. Adams         for( ii = 0 ; ii < a_cbs ; ii++, jj++ ) {
33122063be5SMark F. Adams           isnewproc_idx[jj] = is_idx[kk] * NN; /* distribution */
332eb07cef2SMark F. Adams         }
3335ef31b24SMark F. Adams       }
3345ef31b24SMark F. Adams       ierr = ISRestoreIndices( isnewproc, &is_idx );     CHKERRQ(ierr);
3355ef31b24SMark F. Adams       ierr = ISDestroy( &isnewproc );                    CHKERRQ(ierr);
3364624a670SBarry Smith       ierr = PetscCommDestroy( &new_comm );              CHKERRQ(ierr);
337b4fbaa2aSMark F. Adams 
338b4fbaa2aSMark F. Adams       is_sz *= a_cbs;
3395ef31b24SMark F. Adams     }
3405ef31b24SMark F. Adams     else{
3415ef31b24SMark F. Adams       isnewproc_idx = 0;
3425ef31b24SMark F. Adams       is_sz = 0;
3435ef31b24SMark F. Adams     }
3445ef31b24SMark F. Adams     ierr = MatDestroy( &adj );                       CHKERRQ(ierr);
3455ef31b24SMark F. Adams     ierr = ISCreateGeneral( wcomm, is_sz, isnewproc_idx, PETSC_COPY_VALUES, &isnewproc );
346afc97cdcSMark F. Adams     if( isnewproc_idx != 0 ) {
3475ef31b24SMark F. Adams       ierr = PetscFree( isnewproc_idx );  CHKERRQ(ierr);
3485ef31b24SMark F. Adams     }
349e33ef3b1SMark F. Adams 
35011e60469SMark F. Adams     /*
35111e60469SMark F. Adams      Create an index set from the isnewproc index set to indicate the mapping TO
35211e60469SMark F. Adams      */
35311e60469SMark F. Adams     ierr = ISPartitioningToNumbering( isnewproc, &isnum ); CHKERRQ(ierr);
35411e60469SMark F. Adams     /*
35511e60469SMark F. Adams      Determine how many elements are assigned to each processor
35611e60469SMark F. Adams      */
357fd3c6afaSMark F. Adams     inpe = npe;
358fd3c6afaSMark F. Adams     ierr = ISPartitioningCount( isnewproc, inpe, counts ); CHKERRQ(ierr);
35911e60469SMark F. Adams     ierr = ISDestroy( &isnewproc );                       CHKERRQ(ierr);
360038e3b61SMark F. Adams     ncrs_new = counts[mype]/a_cbs;
36122063be5SMark F. Adams     strideNew = ncrs_new*a_ndata_rows;
362b4fbaa2aSMark F. Adams #if defined PETSC_USE_LOG
363b4fbaa2aSMark F. Adams     ierr = PetscLogEventEnd(gamg_setup_events[SET12],0,0,0,0);   CHKERRQ(ierr);
364b4fbaa2aSMark F. Adams #endif
36522063be5SMark F. Adams     /* Create a vector to contain the newly ordered element information */
36611e60469SMark F. Adams     ierr = VecCreate( wcomm, &dest_crd );
367d3d6bff4SMark F. Adams     ierr = VecSetSizes( dest_crd, data_sz*ncrs_new, PETSC_DECIDE ); CHKERRQ(ierr);
36811e60469SMark F. Adams     ierr = VecSetFromOptions( dest_crd ); CHKERRQ(ierr); /*funny vector-get global options?*/
36911e60469SMark F. Adams     /*
370d3d6bff4SMark F. Adams       There are 'a_ndata_rows*a_ndata_cols' data items per node, (one can think of the vectors of having
371d3d6bff4SMark F. Adams       a block size of ...).  Note, ISs are expanded into equation space by 'a_cbs'.
37211e60469SMark F. Adams     */
37392a756f0SMark F. Adams     ierr = PetscMalloc( (ncrs0*data_sz)*sizeof(PetscInt), &tidx ); CHKERRQ(ierr);
37411e60469SMark F. Adams     ierr = ISGetIndices( isnum, &idx ); CHKERRQ(ierr);
375038e3b61SMark F. Adams     for(ii=0,jj=0; ii<ncrs0 ; ii++) {
376d3d6bff4SMark F. Adams       PetscInt id = idx[ii*a_cbs]/a_cbs; /* get node back */
377d3d6bff4SMark F. Adams       for( kk=0; kk<data_sz ; kk++, jj++) tidx[jj] = id*data_sz + kk;
37811e60469SMark F. Adams     }
379038e3b61SMark F. Adams     ierr = ISRestoreIndices( isnum, &idx ); CHKERRQ(ierr);
380d3d6bff4SMark F. Adams     ierr = ISCreateGeneral( wcomm, data_sz*ncrs0, tidx, PETSC_COPY_VALUES, &isscat );
38111e60469SMark F. Adams     CHKERRQ(ierr);
38292a756f0SMark F. Adams     ierr = PetscFree( tidx );  CHKERRQ(ierr);
38311e60469SMark F. Adams     /*
38411e60469SMark F. Adams       Create a vector to contain the original vertex information for each element
38511e60469SMark F. Adams     */
386d3d6bff4SMark F. Adams     ierr = VecCreateSeq( PETSC_COMM_SELF, data_sz*ncrs0, &src_crd ); CHKERRQ(ierr);
387d3d6bff4SMark F. Adams     for( jj=0; jj<a_ndata_cols ; jj++ ) {
388d3d6bff4SMark F. Adams       for( ii=0 ; ii<ncrs0 ; ii++) {
389d3d6bff4SMark F. Adams 	for( kk=0; kk<a_ndata_rows ; kk++ ) {
390d3d6bff4SMark F. Adams 	  PetscInt ix = ii*a_ndata_rows + kk + jj*stride0, jx = ii*data_sz + kk*a_ndata_cols + jj;
391d3d6bff4SMark F. Adams 	  ierr = VecSetValues( src_crd, 1, &jx, &data[ix], INSERT_VALUES );  CHKERRQ(ierr);
392d3d6bff4SMark F. Adams 	}
393038e3b61SMark F. Adams       }
394eb07cef2SMark F. Adams     }
395eb07cef2SMark F. Adams     ierr = VecAssemblyBegin(src_crd); CHKERRQ(ierr);
396eb07cef2SMark F. Adams     ierr = VecAssemblyEnd(src_crd); CHKERRQ(ierr);
39711e60469SMark F. Adams     /*
39811e60469SMark F. Adams       Scatter the element vertex information (still in the original vertex ordering)
39911e60469SMark F. Adams       to the correct processor
40011e60469SMark F. Adams     */
40111e60469SMark F. Adams     ierr = VecScatterCreate( src_crd, PETSC_NULL, dest_crd, isscat, &vecscat);
40211e60469SMark F. Adams     CHKERRQ(ierr);
40311e60469SMark F. Adams     ierr = ISDestroy( &isscat );       CHKERRQ(ierr);
40411e60469SMark F. Adams     ierr = VecScatterBegin(vecscat,src_crd,dest_crd,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr);
40511e60469SMark F. Adams     ierr = VecScatterEnd(vecscat,src_crd,dest_crd,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr);
40611e60469SMark F. Adams     ierr = VecScatterDestroy( &vecscat );       CHKERRQ(ierr);
40711e60469SMark F. Adams     ierr = VecDestroy( &src_crd );       CHKERRQ(ierr);
40811e60469SMark F. Adams     /*
40911e60469SMark F. Adams       Put the element vertex data into a new allocation of the gdata->ele
41011e60469SMark F. Adams     */
411eb07cef2SMark F. Adams     ierr = PetscFree( *a_coarse_data );    CHKERRQ(ierr);
412d3d6bff4SMark F. Adams     ierr = PetscMalloc( data_sz*ncrs_new*sizeof(PetscReal), a_coarse_data );    CHKERRQ(ierr);
413eb07cef2SMark F. Adams 
41411e60469SMark F. Adams     ierr = VecGetArray( dest_crd, &array );    CHKERRQ(ierr);
415eb07cef2SMark F. Adams     data = *a_coarse_data;
416d3d6bff4SMark F. Adams     for( jj=0; jj<a_ndata_cols ; jj++ ) {
417d3d6bff4SMark F. Adams       for( ii=0 ; ii<ncrs_new ; ii++) {
418d3d6bff4SMark F. Adams 	for( kk=0; kk<a_ndata_rows ; kk++ ) {
419d3d6bff4SMark F. Adams 	  PetscInt ix = ii*a_ndata_rows + kk + jj*strideNew, jx = ii*data_sz + kk*a_ndata_cols + jj;
420d3d6bff4SMark F. Adams 	  data[ix] = PetscRealPart(array[jx]);
421d3d6bff4SMark F. Adams 	  array[jx] = 1.e300;
422d3d6bff4SMark F. Adams 	}
423038e3b61SMark F. Adams       }
424038e3b61SMark F. Adams     }
42511e60469SMark F. Adams     ierr = VecRestoreArray( dest_crd, &array );    CHKERRQ(ierr);
42611e60469SMark F. Adams     ierr = VecDestroy( &dest_crd );    CHKERRQ(ierr);
42711e60469SMark F. Adams     /*
42811e60469SMark F. Adams       Invert for MatGetSubMatrix
42911e60469SMark F. Adams     */
430038e3b61SMark F. Adams     ierr = ISInvertPermutation( isnum, ncrs_new*a_cbs, &new_indices ); CHKERRQ(ierr);
43111e60469SMark F. Adams     ierr = ISSort( new_indices ); CHKERRQ(ierr); /* is this needed? */
43211e60469SMark F. Adams     ierr = ISDestroy( &isnum ); CHKERRQ(ierr);
43311e60469SMark F. Adams     /* A_crs output */
434038e3b61SMark F. Adams     ierr = MatGetSubMatrix( Cmat, new_indices, new_indices, MAT_INITIAL_MATRIX, a_Amat_crs );
43511e60469SMark F. Adams     CHKERRQ(ierr);
436eb07cef2SMark F. Adams 
437038e3b61SMark F. Adams     ierr = MatDestroy( &Cmat ); CHKERRQ(ierr);
438e33ef3b1SMark F. Adams     Cmat = *a_Amat_crs; /* output */
439038e3b61SMark F. Adams     ierr = MatSetBlockSize( Cmat, a_cbs );      CHKERRQ(ierr);
440eb07cef2SMark F. Adams 
44111e60469SMark F. Adams     /* prolongator */
44211e60469SMark F. Adams     ierr = MatGetOwnershipRange( Pold, &Istart, &Iend );    CHKERRQ(ierr);
44311e60469SMark F. Adams     {
44411e60469SMark F. Adams       IS findices;
44511e60469SMark F. Adams       ierr = ISCreateStride(wcomm,Iend-Istart,Istart,1,&findices);   CHKERRQ(ierr);
446d61a3a7dSMark F. Adams 
44711e60469SMark F. Adams       ierr = MatGetSubMatrix( Pold, findices, new_indices, MAT_INITIAL_MATRIX, &Pnew );
44811e60469SMark F. Adams       CHKERRQ(ierr);
449d61a3a7dSMark F. Adams 
45011e60469SMark F. Adams       ierr = ISDestroy( &findices ); CHKERRQ(ierr);
45111e60469SMark F. Adams     }
452d61a3a7dSMark F. Adams 
4533530afc2SMark F. Adams     ierr = MatDestroy( a_P_inout ); CHKERRQ(ierr);
4543530afc2SMark F. Adams     *a_P_inout = Pnew; /* output */
45592a756f0SMark F. Adams 
45611e60469SMark F. Adams     ierr = ISDestroy( &new_indices ); CHKERRQ(ierr);
45792a756f0SMark F. Adams     ierr = PetscFree( counts );  CHKERRQ(ierr);
45892a756f0SMark F. Adams     ierr = PetscFree( ranks );  CHKERRQ(ierr);
459e33ef3b1SMark F. Adams   }
4605b89ad90SMark F. Adams 
4615b89ad90SMark F. Adams   PetscFunctionReturn(0);
4625b89ad90SMark F. Adams }
4635b89ad90SMark F. Adams 
4645b89ad90SMark F. Adams /* -------------------------------------------------------------------------- */
4655b89ad90SMark F. Adams /*
4665b89ad90SMark F. Adams    PCSetUp_GAMG - Prepares for the use of the GAMG preconditioner
4675b89ad90SMark F. Adams                     by setting data structures and options.
4685b89ad90SMark F. Adams 
4695b89ad90SMark F. Adams    Input Parameter:
4705b89ad90SMark F. Adams .  pc - the preconditioner context
4715b89ad90SMark F. Adams 
4725b89ad90SMark F. Adams    Application Interface Routine: PCSetUp()
4735b89ad90SMark F. Adams 
4745b89ad90SMark F. Adams    Notes:
4755b89ad90SMark F. Adams    The interface routine PCSetUp() is not usually called directly by
4765b89ad90SMark F. Adams    the user, but instead is called by PCApply() if necessary.
4775b89ad90SMark F. Adams */
4785b89ad90SMark F. Adams #undef __FUNCT__
4795b89ad90SMark F. Adams #define __FUNCT__ "PCSetUp_GAMG"
480eb07cef2SMark F. Adams PetscErrorCode PCSetUp_GAMG( PC a_pc )
4815b89ad90SMark F. Adams {
4825b89ad90SMark F. Adams   PetscErrorCode  ierr;
483eb07cef2SMark F. Adams   PC_MG           *mg = (PC_MG*)a_pc->data;
4845b89ad90SMark F. Adams   PC_GAMG         *pc_gamg = (PC_GAMG*)mg->innerctx;
48558471d46SMark F. Adams   PC_MG_Levels   **mglevels = mg->levels;
486eb07cef2SMark F. Adams   Mat              Amat = a_pc->mat, Pmat = a_pc->pmat;
487d3d6bff4SMark F. Adams   PetscInt         fine_level, level, level1, M, N, bs, nloc, lidx, Istart, Iend;
488eb07cef2SMark F. Adams   MPI_Comm         wcomm = ((PetscObject)a_pc)->comm;
4893530afc2SMark F. Adams   PetscMPIInt      mype,npe,nactivepe;
490038e3b61SMark F. Adams   PetscBool        isOK;
491587fa25dSMark F. Adams   Mat              Aarr[GAMG_MAXLEVELS], Parr[GAMG_MAXLEVELS];
492587fa25dSMark F. Adams   PetscReal       *coarse_data = 0, *data, emaxs[GAMG_MAXLEVELS];
493737a81a9SMark F. Adams   MatInfo          info;
4945ef31b24SMark F. Adams 
4955b89ad90SMark F. Adams   PetscFunctionBegin;
496*03a628feSMark F. Adams   pc_gamg->m_count++;
49758471d46SMark F. Adams   if( a_pc->setupcalled > 0 ) {
498*03a628feSMark F. Adams     /* just do Galerkin grids */
49958471d46SMark F. Adams     Mat B,dA,dB;
50058471d46SMark F. Adams 
50158471d46SMark F. Adams     /* PCSetUp_MG seems to insists on setting this to GMRES */
50258471d46SMark F. Adams     ierr = KSPSetType( mglevels[0]->smoothd, KSPPREONLY ); CHKERRQ(ierr);
50358471d46SMark F. Adams 
50458471d46SMark F. Adams     /* currently only handle case where mat and pmat are the same on coarser levels */
50558471d46SMark F. Adams     ierr = KSPGetOperators(mglevels[pc_gamg->m_Nlevels-1]->smoothd,&dA,&dB,PETSC_NULL);CHKERRQ(ierr);
50658471d46SMark F. Adams     /* (re)set to get dirty flag */
50758471d46SMark F. Adams     ierr = KSPSetOperators(mglevels[pc_gamg->m_Nlevels-1]->smoothd,dA,dB,SAME_NONZERO_PATTERN);CHKERRQ(ierr);
50858471d46SMark F. Adams     ierr = KSPSetUp( mglevels[pc_gamg->m_Nlevels-1]->smoothd ); CHKERRQ(ierr);
50958471d46SMark F. Adams 
51058471d46SMark F. Adams     for (level=pc_gamg->m_Nlevels-2; level>-1; level--) {
51158471d46SMark F. Adams       ierr = KSPGetOperators(mglevels[level]->smoothd,PETSC_NULL,&B,PETSC_NULL);CHKERRQ(ierr);
512*03a628feSMark F. Adams       /* the first time through the matrix structure has changed from repartitioning */
513*03a628feSMark F. Adams       if( pc_gamg->m_count == 2 ) {
514*03a628feSMark F. Adams         ierr = MatDestroy( &B );  CHKERRQ(ierr);
515*03a628feSMark F. Adams         ierr = MatPtAP(dB,mglevels[level+1]->interpolate,MAT_INITIAL_MATRIX,1.0,&B);CHKERRQ(ierr);
516*03a628feSMark F. Adams         mglevels[level]->A = B;
517*03a628feSMark F. Adams       }
518*03a628feSMark F. Adams       else {
51958471d46SMark F. Adams         ierr = MatPtAP(dB,mglevels[level+1]->interpolate,MAT_REUSE_MATRIX,1.0,&B);CHKERRQ(ierr);
520*03a628feSMark F. Adams       }
52158471d46SMark F. Adams       ierr = KSPSetOperators(mglevels[level]->smoothd,B,B,SAME_NONZERO_PATTERN); CHKERRQ(ierr);
52258471d46SMark F. Adams       dB = B;
52358471d46SMark F. Adams       /* setup KSP/PC */
52458471d46SMark F. Adams       ierr = KSPSetUp( mglevels[level]->smoothd ); CHKERRQ(ierr);
52558471d46SMark F. Adams     }
52658471d46SMark F. Adams 
52758471d46SMark F. Adams #define PRINT_MATS !PETSC_TRUE
52858471d46SMark F. Adams     /* plot levels - A */
52958471d46SMark F. Adams     if( PRINT_MATS ) {
53058471d46SMark F. Adams       for (lidx=0, level=pc_gamg->m_Nlevels-1; level>0 ; level--,lidx++){
53158471d46SMark F. Adams         PetscViewer viewer;
53258471d46SMark F. Adams         char fname[32]; KSP smoother; Mat Tmat, TTm;
53358471d46SMark F. Adams         ierr = PCMGGetSmoother( a_pc, lidx, &smoother ); CHKERRQ(ierr);
53458471d46SMark F. Adams         ierr = KSPGetOperators( smoother, &Tmat, &TTm, 0 ); CHKERRQ(ierr);
535*03a628feSMark F. Adams         sprintf(fname,"Amat_%d_%d.m",(int)pc_gamg->m_count,(int)level);
53658471d46SMark F. Adams         ierr = PetscViewerASCIIOpen( wcomm, fname, &viewer );  CHKERRQ(ierr);
53758471d46SMark F. Adams         ierr = PetscViewerSetFormat( viewer, PETSC_VIEWER_ASCII_MATLAB);  CHKERRQ(ierr);
53858471d46SMark F. Adams         ierr = MatView( Tmat, viewer ); CHKERRQ(ierr);
53958471d46SMark F. Adams         ierr = PetscViewerDestroy( &viewer );
54058471d46SMark F. Adams       }
54158471d46SMark F. Adams     }
54258471d46SMark F. Adams 
543*03a628feSMark F. Adams     a_pc->setupcalled = 2;
544*03a628feSMark F. Adams 
54558471d46SMark F. Adams     PetscFunctionReturn(0);
546eb07cef2SMark F. Adams   }
547baa4e9faSMark F. Adams   ierr = MPI_Comm_rank(wcomm,&mype);CHKERRQ(ierr);
548baa4e9faSMark F. Adams   ierr = MPI_Comm_size(wcomm,&npe);CHKERRQ(ierr);
5495b89ad90SMark F. Adams   /* GAMG requires input of fine-grid matrix. It determines nlevels. */
5505b89ad90SMark F. Adams   ierr = MatGetBlockSize( Amat, &bs ); CHKERRQ(ierr);
5513530afc2SMark F. Adams   ierr = MatGetSize( Amat, &M, &N );CHKERRQ(ierr);
552eb07cef2SMark F. Adams   ierr = MatGetOwnershipRange( Amat, &Istart, &Iend ); CHKERRQ(ierr);
553eb07cef2SMark F. Adams   nloc = (Iend-Istart)/bs; assert((Iend-Istart)%bs == 0);
554eb07cef2SMark F. Adams 
555038e3b61SMark F. Adams   /* get data of not around */
556038e3b61SMark F. Adams   if( pc_gamg->m_data == 0 && nloc > 0 ) {
557038e3b61SMark F. Adams     ierr  = PCSetCoordinates_GAMG( a_pc, -1, 0 );    CHKERRQ( ierr );
558eb07cef2SMark F. Adams   }
559eb07cef2SMark F. Adams   data = pc_gamg->m_data;
560038e3b61SMark F. Adams 
561eb07cef2SMark F. Adams   /* Get A_i and R_i */
562737a81a9SMark F. Adams   ierr = MatGetInfo(Amat,MAT_GLOBAL_SUM,&info); CHKERRQ(ierr);
56358471d46SMark F. Adams #ifdef VERBOSE
564fd3c6afaSMark F. Adams   PetscPrintf(PETSC_COMM_WORLD,"\t[%d]%s level %d N=%ld, n data rows=%d, n data cols=%d, nnz/row (ave)=%d, np=%d\n",
565fd3c6afaSMark F. Adams 	      mype,__FUNCT__,0,(long long int)N,(int)pc_gamg->m_data_rows,(int)pc_gamg->m_data_cols,
566fd3c6afaSMark F. Adams 	      (int)(info.nz_used/(PetscReal)N),(int)npe);
56758471d46SMark F. Adams #endif
5688f4b7eb5SMark F. Adams   for ( level=0, Aarr[0] = Pmat, nactivepe = npe; /* hard wired stopping logic */
569785cba28SMark F. Adams         level < GAMG_MAXLEVELS-1 && (level==0 || M>TOP_GRID_LIM) && (npe==1 || nactivepe>1);
5700205a208SMark F. Adams         level++ ){
5715b89ad90SMark F. Adams     level1 = level + 1;
572b4fbaa2aSMark F. Adams #if (defined PETSC_USE_LOG && defined GAMG_STAGES)
573b4fbaa2aSMark F. Adams     ierr = PetscLogStagePush(gamg_stages[level]); CHKERRQ( ierr );
574b4fbaa2aSMark F. Adams #endif
575b4fbaa2aSMark F. Adams #if defined PETSC_USE_LOG
576b4fbaa2aSMark F. Adams     ierr = PetscLogEventBegin(gamg_setup_events[SET1],0,0,0,0);CHKERRQ(ierr);
577b4fbaa2aSMark F. Adams #endif
578d3d6bff4SMark F. Adams     ierr = createProlongation(Aarr[level], data, pc_gamg->m_dim, pc_gamg->m_data_cols, pc_gamg->m_useSA,
579785cba28SMark F. Adams                               level, &bs, &Parr[level1], &coarse_data, &isOK, &emaxs[level] );
5805b89ad90SMark F. Adams     CHKERRQ(ierr);
581d3d6bff4SMark F. Adams     ierr = PetscFree( data ); CHKERRQ( ierr );
582b4fbaa2aSMark F. Adams #if defined PETSC_USE_LOG
583b4fbaa2aSMark F. Adams     ierr = PetscLogEventEnd(gamg_setup_events[SET1],0,0,0,0);CHKERRQ(ierr);
584b4fbaa2aSMark F. Adams #endif
585baa4e9faSMark F. Adams     if(level==0) Aarr[0] = Amat; /* use Pmat for finest level setup, but use mat for solver */
586baa4e9faSMark F. Adams     if( isOK ) {
587b4fbaa2aSMark F. Adams #if defined PETSC_USE_LOG
588b4fbaa2aSMark F. Adams       ierr = PetscLogEventBegin(gamg_setup_events[SET2],0,0,0,0);CHKERRQ(ierr);
589b4fbaa2aSMark F. Adams #endif
590d3d6bff4SMark F. Adams       ierr = partitionLevel( Aarr[level], pc_gamg->m_useSA ? bs : 1, pc_gamg->m_data_cols, bs,
591eb07cef2SMark F. Adams                              &Parr[level1], &coarse_data, &nactivepe, &Aarr[level1] );
5923530afc2SMark F. Adams       CHKERRQ(ierr);
593b4fbaa2aSMark F. Adams #if defined PETSC_USE_LOG
594b4fbaa2aSMark F. Adams       ierr = PetscLogEventEnd(gamg_setup_events[SET2],0,0,0,0);CHKERRQ(ierr);
595b4fbaa2aSMark F. Adams #endif
5963530afc2SMark F. Adams       ierr = MatGetSize( Aarr[level1], &M, &N );CHKERRQ(ierr);
597737a81a9SMark F. Adams       ierr = MatGetInfo(Aarr[level1],MAT_GLOBAL_SUM,&info); CHKERRQ(ierr);
59858471d46SMark F. Adams #ifdef VERBOSE
59958471d46SMark F. Adams       PetscPrintf(PETSC_COMM_WORLD,"\t\t[%d]%s %d) N=%d, n data cols=%d, nnz/row (ave)=%d, %d active pes\n",
60058471d46SMark F. Adams 		  mype,__FUNCT__,(int)level1,(int)N,(int)pc_gamg->m_data_cols,
601fd3c6afaSMark F. Adams 		  (int)(info.nz_used/(PetscReal)N),(int)nactivepe);
60258471d46SMark F. Adams #endif
603e33ef3b1SMark F. Adams       /* coarse grids with SA can have zero row/cols from singleton aggregates */
60458471d46SMark F. Adams       /* aggregation method should gaurrentee this does not happen! */
605737a81a9SMark F. Adams 
60658471d46SMark F. Adams #ifdef VERBOSE
607e33ef3b1SMark F. Adams       if( PETSC_TRUE ){
608785cba28SMark F. Adams         Vec diag; PetscScalar *data_arr,v; PetscInt Istart,Iend,kk,nloceq,id;
609785cba28SMark F. Adams         v = 1.e-10; /* LU factor has hard wired numbers for small diags so this needs to match (yuk) */
610e33ef3b1SMark F. Adams         ierr = MatGetOwnershipRange(Aarr[level1], &Istart, &Iend); CHKERRQ(ierr);
611785cba28SMark F. Adams         nloceq = Iend-Istart;
612e33ef3b1SMark F. Adams         ierr = MatGetVecs( Aarr[level1], &diag, 0 );    CHKERRQ(ierr);
613e33ef3b1SMark F. Adams         ierr = MatGetDiagonal( Aarr[level1], diag );    CHKERRQ(ierr);
614e33ef3b1SMark F. Adams         ierr = VecGetArray( diag, &data_arr );   CHKERRQ(ierr);
615785cba28SMark F. Adams         for(kk=0;kk<nloceq;kk++){
616e33ef3b1SMark F. Adams           if(data_arr[kk]==0.0) {
617785cba28SMark F. Adams             id = kk + Istart;
618e33ef3b1SMark F. Adams             ierr = MatSetValues(Aarr[level1],1,&id,1,&id,&v,INSERT_VALUES);
619e33ef3b1SMark F. Adams             CHKERRQ(ierr);
620ecd57171SMark F. Adams             PetscPrintf(PETSC_COMM_SELF,"\t[%d]%s warning: added zero to diag (%d) on level %d \n",mype,__FUNCT__,id,level1);
621e33ef3b1SMark F. Adams           }
622e33ef3b1SMark F. Adams         }
623e33ef3b1SMark F. Adams         ierr = VecRestoreArray( diag, &data_arr ); CHKERRQ(ierr);
624e33ef3b1SMark F. Adams         ierr = VecDestroy( &diag );                CHKERRQ(ierr);
625e33ef3b1SMark F. Adams         ierr = MatAssemblyBegin(Aarr[level1],MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
626e33ef3b1SMark F. Adams         ierr = MatAssemblyEnd(Aarr[level1],MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
627e33ef3b1SMark F. Adams       }
62858471d46SMark F. Adams #endif
629baa4e9faSMark F. Adams     }
630baa4e9faSMark F. Adams     else{
631be544d3cSMark F. Adams       coarse_data = 0;
632baa4e9faSMark F. Adams       break;
633baa4e9faSMark F. Adams     }
634eb07cef2SMark F. Adams     data = coarse_data;
635b4fbaa2aSMark F. Adams 
636b4fbaa2aSMark F. Adams #if (defined PETSC_USE_LOG && defined GAMG_STAGES)
637b4fbaa2aSMark F. Adams     ierr = PetscLogStagePop(); CHKERRQ( ierr );
638b4fbaa2aSMark F. Adams #endif
6395b89ad90SMark F. Adams   }
640be544d3cSMark F. Adams   if( coarse_data ) {
641eb07cef2SMark F. Adams     ierr = PetscFree( coarse_data ); CHKERRQ( ierr );
642be544d3cSMark F. Adams   }
64358471d46SMark F. Adams #ifdef VERBOSE
644baa4e9faSMark F. Adams   PetscPrintf(PETSC_COMM_WORLD,"\t[%d]%s %d levels\n",0,__FUNCT__,level + 1);
64558471d46SMark F. Adams #endif
6465b89ad90SMark F. Adams   pc_gamg->m_data = 0; /* destroyed coordinate data */
6475b89ad90SMark F. Adams   pc_gamg->m_Nlevels = level + 1;
6485b89ad90SMark F. Adams   fine_level = level;
649eb07cef2SMark F. Adams   ierr = PCMGSetLevels(a_pc,pc_gamg->m_Nlevels,PETSC_NULL);CHKERRQ(ierr);
6505b89ad90SMark F. Adams 
651fc4362bfSMark F. Adams   /* set default smoothers */
652587fa25dSMark F. Adams   for ( lidx=1, level = pc_gamg->m_Nlevels-2;
653587fa25dSMark F. Adams         lidx <= fine_level;
654587fa25dSMark F. Adams         lidx++, level--) {
6555745e0f5SMark F. Adams     PetscReal emax, emin;
6565b89ad90SMark F. Adams     KSP smoother; PC subpc;
657587fa25dSMark F. Adams     ierr = PCMGGetSmoother( a_pc, lidx, &smoother ); CHKERRQ(ierr);
6585b89ad90SMark F. Adams     ierr = KSPSetType( smoother, KSPCHEBYCHEV );CHKERRQ(ierr);
659038e3b61SMark F. Adams     if( emaxs[level] > 0.0 ) emax=emaxs[level];
660587fa25dSMark F. Adams     else{ /* eigen estimate 'emax' */
661587fa25dSMark F. Adams       KSP eksp; Mat Lmat = Aarr[level];
662fc4362bfSMark F. Adams       Vec bb, xx; PC pc;
663038e3b61SMark F. Adams 
6645745e0f5SMark F. Adams       ierr = MatGetVecs( Lmat, &bb, 0 );         CHKERRQ(ierr);
6655745e0f5SMark F. Adams       ierr = MatGetVecs( Lmat, &xx, 0 );         CHKERRQ(ierr);
666fc4362bfSMark F. Adams       {
667fc4362bfSMark F. Adams 	PetscRandom    rctx;
668fc4362bfSMark F. Adams 	ierr = PetscRandomCreate(wcomm,&rctx);CHKERRQ(ierr);
669fc4362bfSMark F. Adams 	ierr = PetscRandomSetFromOptions(rctx);CHKERRQ(ierr);
670fc4362bfSMark F. Adams 	ierr = VecSetRandom(bb,rctx);CHKERRQ(ierr);
671fc4362bfSMark F. Adams 	ierr = PetscRandomDestroy( &rctx ); CHKERRQ(ierr);
6725b89ad90SMark F. Adams       }
673fc4362bfSMark F. Adams       ierr = KSPCreate(wcomm,&eksp);CHKERRQ(ierr);
674e33ef3b1SMark F. Adams       ierr = KSPSetType( eksp, KSPCG );                      CHKERRQ(ierr);
675fc4362bfSMark F. Adams       ierr = KSPSetInitialGuessNonzero( eksp, PETSC_FALSE ); CHKERRQ(ierr);
67658471d46SMark F. Adams       ierr = KSPSetOperators( eksp, Lmat, Lmat, SAME_NONZERO_PATTERN ); CHKERRQ( ierr );
677fc4362bfSMark F. Adams       ierr = KSPGetPC( eksp, &pc );CHKERRQ( ierr );
67858471d46SMark F. Adams       ierr = PCSetType( pc, PETSC_GAMG_SMOOTHER ); CHKERRQ(ierr); /* should be same as eigen estimates op. */
679038e3b61SMark F. Adams       ierr = KSPSetTolerances( eksp, PETSC_DEFAULT, PETSC_DEFAULT, PETSC_DEFAULT, 10 );
680fc4362bfSMark F. Adams       CHKERRQ(ierr);
681fc4362bfSMark F. Adams       ierr = KSPSetNormType( eksp, KSP_NORM_NONE );                 CHKERRQ(ierr);
682fc4362bfSMark F. Adams 
683fc4362bfSMark F. Adams       ierr = KSPSetComputeSingularValues( eksp,PETSC_TRUE ); CHKERRQ(ierr);
684fc4362bfSMark F. Adams       ierr = KSPSolve( eksp, bb, xx ); CHKERRQ(ierr);
685fc4362bfSMark F. Adams       ierr = KSPComputeExtremeSingularValues( eksp, &emax, &emin ); CHKERRQ(ierr);
686fc4362bfSMark F. Adams       ierr = VecDestroy( &xx );       CHKERRQ(ierr);
687fc4362bfSMark F. Adams       ierr = VecDestroy( &bb );       CHKERRQ(ierr);
688fc4362bfSMark F. Adams       ierr = KSPDestroy( &eksp );       CHKERRQ(ierr);
68958471d46SMark F. Adams #ifdef VERBOSE
6901fddbf69SMark F. Adams       PetscPrintf(PETSC_COMM_WORLD,"\t\t\t%s max eigen=%e min=%e PC=%s\n",__FUNCT__,emax,emin,PETSC_GAMG_SMOOTHER);
69158471d46SMark F. Adams #endif
692fc4362bfSMark F. Adams     }
693038e3b61SMark F. Adams     {
694038e3b61SMark F. Adams       PetscInt N1, N0, tt;
695038e3b61SMark F. Adams       ierr = MatGetSize( Aarr[level], &N1, &tt );         CHKERRQ(ierr);
696038e3b61SMark F. Adams       ierr = MatGetSize( Aarr[level+1], &N0, &tt );       CHKERRQ(ierr);
697785cba28SMark F. Adams       emin = 1.*emax/((PetscReal)N1/(PetscReal)N0); /* this should be about the coarsening rate */
698038e3b61SMark F. Adams       emax *= 1.05;
699038e3b61SMark F. Adams     }
700038e3b61SMark F. Adams 
70158471d46SMark F. Adams     ierr = KSPSetOperators( smoother, Aarr[level], Aarr[level], SAME_NONZERO_PATTERN );
702fc4362bfSMark F. Adams     ierr = KSPChebychevSetEigenvalues( smoother, emax, emin );CHKERRQ(ierr);
7030e1b4bd6SMark F. Adams     /* ierr = KSPSetTolerances(smoother,PETSC_DEFAULT,PETSC_DEFAULT,PETSC_DEFAULT,2); CHKERRQ(ierr); */
7045745e0f5SMark F. Adams     ierr = KSPGetPC( smoother, &subpc ); CHKERRQ(ierr);
705e33ef3b1SMark F. Adams     ierr = PCSetType( subpc, PETSC_GAMG_SMOOTHER ); CHKERRQ(ierr);
7065745e0f5SMark F. Adams     ierr = KSPSetNormType( smoother, KSP_NORM_NONE ); CHKERRQ(ierr);
7075745e0f5SMark F. Adams   }
7085745e0f5SMark F. Adams   {
709ecd57171SMark F. Adams     /* coarse grid */
710ecd57171SMark F. Adams     KSP smoother,*k2; PC subpc,pc2; PetscInt ii,first;
7115745e0f5SMark F. Adams     Mat Lmat = Aarr[pc_gamg->m_Nlevels-1];
712eb07cef2SMark F. Adams     ierr = PCMGGetSmoother( a_pc, 0, &smoother ); CHKERRQ(ierr);
71358471d46SMark F. Adams     ierr = KSPSetOperators( smoother, Lmat, Lmat, SAME_NONZERO_PATTERN ); CHKERRQ(ierr);
7145745e0f5SMark F. Adams     ierr = KSPSetNormType( smoother, KSP_NORM_NONE ); CHKERRQ(ierr);
715ecd57171SMark F. Adams     ierr = KSPGetPC( smoother, &subpc ); CHKERRQ(ierr);
716ecd57171SMark F. Adams     ierr = PCSetType( subpc, PCBJACOBI ); CHKERRQ(ierr);
717ecd57171SMark F. Adams     ierr = PCSetUp( subpc ); CHKERRQ(ierr);
718ecd57171SMark F. Adams     ierr = PCBJacobiGetSubKSP(subpc,&ii,&first,&k2);CHKERRQ(ierr);
719ecd57171SMark F. Adams     assert(ii==1);
720ecd57171SMark F. Adams     ierr = KSPGetPC(k2[0],&pc2);CHKERRQ(ierr);
721ecd57171SMark F. Adams     ierr = PCSetType( pc2, PCLU ); CHKERRQ(ierr);
722fc4362bfSMark F. Adams   }
723737a81a9SMark F. Adams 
724fc4362bfSMark F. Adams   /* should be called in PCSetFromOptions_GAMG(), but cannot be called prior to PCMGSetLevels() */
725eb07cef2SMark F. Adams   ierr = PCSetFromOptions_MG(a_pc); CHKERRQ(ierr);
7265b89ad90SMark F. Adams   {
7275b89ad90SMark F. Adams     PetscBool galerkin;
728eb07cef2SMark F. Adams     ierr = PCMGGetGalerkin( a_pc,  &galerkin); CHKERRQ(ierr);
7295b89ad90SMark F. Adams     if(galerkin){
7305b89ad90SMark F. Adams       SETERRQ(wcomm,PETSC_ERR_ARG_WRONG, "GAMG does galerkin manually so it must not be used in PC_MG.");
7315b89ad90SMark F. Adams     }
7325b89ad90SMark F. Adams   }
7335745e0f5SMark F. Adams 
73458471d46SMark F. Adams   /* plot levels - R/P */
73558471d46SMark F. Adams   if( PRINT_MATS ) {
73658471d46SMark F. Adams     for (level=pc_gamg->m_Nlevels-1;level>0;level--){
73758471d46SMark F. Adams       PetscViewer viewer;
73858471d46SMark F. Adams       char fname[32];
739*03a628feSMark F. Adams       sprintf(fname,"Pmat_%d_%d.m",(int)pc_gamg->m_count,(int)level);
74011e60469SMark F. Adams       ierr = PetscViewerASCIIOpen( wcomm, fname, &viewer );  CHKERRQ(ierr);
7415b89ad90SMark F. Adams       ierr = PetscViewerSetFormat( viewer, PETSC_VIEWER_ASCII_MATLAB);  CHKERRQ(ierr);
742038e3b61SMark F. Adams       ierr = MatView( Parr[level], viewer ); CHKERRQ(ierr);
7435b89ad90SMark F. Adams       ierr = PetscViewerDestroy( &viewer );
744*03a628feSMark F. Adams       sprintf(fname,"Amat_%d_%d.m",(int)pc_gamg->m_count,(int)level);
745e33ef3b1SMark F. Adams       ierr = PetscViewerASCIIOpen( wcomm, fname, &viewer );  CHKERRQ(ierr);
746e33ef3b1SMark F. Adams       ierr = PetscViewerSetFormat( viewer, PETSC_VIEWER_ASCII_MATLAB);  CHKERRQ(ierr);
747e33ef3b1SMark F. Adams       ierr = MatView( Aarr[level], viewer ); CHKERRQ(ierr);
748e33ef3b1SMark F. Adams       ierr = PetscViewerDestroy( &viewer );
7495b89ad90SMark F. Adams     }
75058471d46SMark F. Adams   }
75158471d46SMark F. Adams 
75258471d46SMark F. Adams   /* set interpolation between the levels, clean up */
75358471d46SMark F. Adams   for (lidx=0,level=pc_gamg->m_Nlevels-1;
75458471d46SMark F. Adams        lidx<fine_level;
75558471d46SMark F. Adams        lidx++, level--){
75658471d46SMark F. Adams     ierr = PCMGSetInterpolation( a_pc, lidx+1, Parr[level] );CHKERRQ(ierr);
757587fa25dSMark F. Adams     ierr = MatDestroy( &Parr[level] );  CHKERRQ(ierr);
758587fa25dSMark F. Adams     ierr = MatDestroy( &Aarr[level] );  CHKERRQ(ierr);
7595b89ad90SMark F. Adams   }
7605745e0f5SMark F. Adams 
7615b89ad90SMark F. Adams   /* setupcalled is set to 0 so that MG is setup from scratch */
762eb07cef2SMark F. Adams   a_pc->setupcalled = 0;
763eb07cef2SMark F. Adams   ierr = PCSetUp_MG( a_pc );CHKERRQ( ierr );
764*03a628feSMark F. Adams   a_pc->setupcalled = 1; /* use 1 as signal that this has not been re-setup */
76558471d46SMark F. Adams 
76658471d46SMark F. Adams   {
76758471d46SMark F. Adams     KSP smoother;  /* PCSetUp_MG seems to insists on setting this to GMRES on coarse grid */
76858471d46SMark F. Adams     ierr = PCMGGetSmoother( a_pc, 0, &smoother ); CHKERRQ(ierr);
76958471d46SMark F. Adams     ierr = KSPSetType( smoother, KSPPREONLY ); CHKERRQ(ierr);
77058471d46SMark F. Adams     ierr = KSPSetUp( smoother ); CHKERRQ(ierr);
77158471d46SMark F. Adams   }
772e33ef3b1SMark F. Adams 
7735b89ad90SMark F. Adams   PetscFunctionReturn(0);
7745b89ad90SMark F. Adams }
7755b89ad90SMark F. Adams 
776eb07cef2SMark F. Adams /* ------------------------------------------------------------------------- */
7775b89ad90SMark F. Adams /*
7785b89ad90SMark F. Adams    PCDestroy_GAMG - Destroys the private context for the GAMG preconditioner
7795b89ad90SMark F. Adams    that was created with PCCreate_GAMG().
7805b89ad90SMark F. Adams 
7815b89ad90SMark F. Adams    Input Parameter:
7825b89ad90SMark F. Adams .  pc - the preconditioner context
7835b89ad90SMark F. Adams 
7845b89ad90SMark F. Adams    Application Interface Routine: PCDestroy()
7855b89ad90SMark F. Adams */
7865b89ad90SMark F. Adams #undef __FUNCT__
7875b89ad90SMark F. Adams #define __FUNCT__ "PCDestroy_GAMG"
7885b89ad90SMark F. Adams PetscErrorCode PCDestroy_GAMG(PC pc)
7895b89ad90SMark F. Adams {
7905b89ad90SMark F. Adams   PetscErrorCode  ierr;
7915b89ad90SMark F. Adams   PC_MG           *mg = (PC_MG*)pc->data;
7925b89ad90SMark F. Adams   PC_GAMG         *pc_gamg= (PC_GAMG*)mg->innerctx;
7935b89ad90SMark F. Adams 
7945b89ad90SMark F. Adams   PetscFunctionBegin;
7955b89ad90SMark F. Adams   ierr = PCReset_GAMG(pc);CHKERRQ(ierr);
7965b89ad90SMark F. Adams   ierr = PetscFree(pc_gamg);CHKERRQ(ierr);
7975b89ad90SMark F. Adams   ierr = PCDestroy_MG(pc);CHKERRQ(ierr);
7985b89ad90SMark F. Adams   PetscFunctionReturn(0);
7995b89ad90SMark F. Adams }
8005b89ad90SMark F. Adams 
8015b89ad90SMark F. Adams #undef __FUNCT__
8025b89ad90SMark F. Adams #define __FUNCT__ "PCSetFromOptions_GAMG"
8035b89ad90SMark F. Adams PetscErrorCode PCSetFromOptions_GAMG(PC pc)
8045b89ad90SMark F. Adams {
8055b89ad90SMark F. Adams   /* PetscErrorCode  ierr; */
8065b89ad90SMark F. Adams   /* PC_MG           *mg = (PC_MG*)pc->data; */
8075b89ad90SMark F. Adams   /* PC_GAMG         *pc_gamg = (PC_GAMG*)mg->innerctx; */
8085b89ad90SMark F. Adams   /* MPI_Comm        comm = ((PetscObject)pc)->comm; */
8095b89ad90SMark F. Adams 
8105b89ad90SMark F. Adams   PetscFunctionBegin;
8115b89ad90SMark F. Adams   PetscFunctionReturn(0);
8125b89ad90SMark F. Adams }
8135b89ad90SMark F. Adams 
8145b89ad90SMark F. Adams /* -------------------------------------------------------------------------- */
8155b89ad90SMark F. Adams /*
8165b89ad90SMark F. Adams  PCCreate_GAMG - Creates a GAMG preconditioner context, PC_GAMG
8175b89ad90SMark F. Adams 
8185b89ad90SMark F. Adams    Input Parameter:
8195b89ad90SMark F. Adams .  pc - the preconditioner context
8205b89ad90SMark F. Adams 
8215b89ad90SMark F. Adams    Application Interface Routine: PCCreate()
8225b89ad90SMark F. Adams 
8235b89ad90SMark F. Adams   */
8245b89ad90SMark F. Adams  /* MC
8255b89ad90SMark F. Adams      PCGAMG - Use algebraic multigrid preconditioning. This preconditioner requires you provide
8265b89ad90SMark F. Adams        fine grid discretization matrix and coordinates on the fine grid.
8275b89ad90SMark F. Adams 
8285b89ad90SMark F. Adams    Options Database Key:
8295b89ad90SMark F. Adams    Multigrid options(inherited)
8305b89ad90SMark F. Adams +  -pc_mg_cycles <1>: 1 for V cycle, 2 for W-cycle (MGSetCycles)
8315b89ad90SMark F. Adams .  -pc_mg_smoothup <1>: Number of post-smoothing steps (MGSetNumberSmoothUp)
8325b89ad90SMark F. Adams .  -pc_mg_smoothdown <1>: Number of pre-smoothing steps (MGSetNumberSmoothDown)
8335b89ad90SMark F. Adams    -pc_mg_type <multiplicative>: (one of) additive multiplicative full cascade kascade
8345b89ad90SMark F. Adams    GAMG options:
8355b89ad90SMark F. Adams 
8365b89ad90SMark F. Adams    Level: intermediate
8375b89ad90SMark F. Adams   Concepts: multigrid
8385b89ad90SMark F. Adams 
8395b89ad90SMark F. Adams .seealso:  PCCreate(), PCSetType(), PCType (for list of available types), PC, PCMGType,
8405b89ad90SMark F. Adams            PCMGSetLevels(), PCMGGetLevels(), PCMGSetType(), MPSetCycles(), PCMGSetNumberSmoothDown(),
8415b89ad90SMark F. Adams            PCMGSetNumberSmoothUp(), PCMGGetCoarseSolve(), PCMGSetResidual(), PCMGSetInterpolation(),
8425b89ad90SMark F. Adams            PCMGSetRestriction(), PCMGGetSmoother(), PCMGGetSmootherUp(), PCMGGetSmootherDown(),
8435b89ad90SMark F. Adams            PCMGSetCyclesOnLevel(), PCMGSetRhs(), PCMGSetX(), PCMGSetR()
8445b89ad90SMark F. Adams M */
8455b89ad90SMark F. Adams 
8465b89ad90SMark F. Adams EXTERN_C_BEGIN
8475b89ad90SMark F. Adams #undef __FUNCT__
8485b89ad90SMark F. Adams #define __FUNCT__ "PCCreate_GAMG"
8495b89ad90SMark F. Adams PetscErrorCode  PCCreate_GAMG(PC pc)
8505b89ad90SMark F. Adams {
8515b89ad90SMark F. Adams   PetscErrorCode  ierr;
8525b89ad90SMark F. Adams   PC_GAMG         *pc_gamg;
8535b89ad90SMark F. Adams   PC_MG           *mg;
8545ef31b24SMark F. Adams   PetscClassId     cookie;
8555b89ad90SMark F. Adams 
8565b89ad90SMark F. Adams   PetscFunctionBegin;
8575b89ad90SMark F. Adams   /* PCGAMG is an inherited class of PCMG. Initialize pc as PCMG */
8585b89ad90SMark F. Adams   ierr = PCSetType(pc,PCMG);CHKERRQ(ierr); /* calls PCCreate_MG() and MGCreate_Private() */
8595b89ad90SMark F. Adams   ierr = PetscObjectChangeTypeName((PetscObject)pc,PCGAMG);CHKERRQ(ierr);
8605b89ad90SMark F. Adams 
8615b89ad90SMark F. Adams   /* create a supporting struct and attach it to pc */
8625b89ad90SMark F. Adams   ierr = PetscNewLog(pc,PC_GAMG,&pc_gamg);CHKERRQ(ierr);
863*03a628feSMark F. Adams   pc_gamg->m_data_sz = 0; pc_gamg->m_data = 0; pc_gamg->m_count = 0;
8645b89ad90SMark F. Adams   mg = (PC_MG*)pc->data;
8655b89ad90SMark F. Adams   mg->innerctx = pc_gamg;
8665b89ad90SMark F. Adams 
8675b89ad90SMark F. Adams   pc_gamg->m_Nlevels    = -1;
8685b89ad90SMark F. Adams 
8695b89ad90SMark F. Adams   /* overwrite the pointers of PCMG by the functions of PCGAMG */
8705b89ad90SMark F. Adams   pc->ops->setfromoptions = PCSetFromOptions_GAMG;
8715b89ad90SMark F. Adams   pc->ops->setup          = PCSetUp_GAMG;
8725b89ad90SMark F. Adams   pc->ops->reset          = PCReset_GAMG;
8735b89ad90SMark F. Adams   pc->ops->destroy        = PCDestroy_GAMG;
8745b89ad90SMark F. Adams 
8755b89ad90SMark F. Adams   ierr = PetscObjectComposeFunctionDynamic( (PetscObject)pc,
8765b89ad90SMark F. Adams 					    "PCSetCoordinates_C",
8775b89ad90SMark F. Adams 					    "PCSetCoordinates_GAMG",
8785b89ad90SMark F. Adams 					    PCSetCoordinates_GAMG);CHKERRQ(ierr);
879b4fbaa2aSMark F. Adams #if defined PETSC_USE_LOG
880785cba28SMark F. Adams   static int count = 0;
881785cba28SMark F. Adams   if( count++ == 0 ) {
8825ef31b24SMark F. Adams     PetscClassIdRegister("GAMG Setup",&cookie);
883b4fbaa2aSMark F. Adams     PetscLogEventRegister("GAMG: createProl", cookie, &gamg_setup_events[SET1]);
884b4fbaa2aSMark F. Adams     PetscLogEventRegister(" make graph", cookie, &gamg_setup_events[SET3]);
885b4fbaa2aSMark F. Adams     PetscLogEventRegister(" MIS/Agg", cookie, &gamg_setup_events[SET4]);
886b4fbaa2aSMark F. Adams     PetscLogEventRegister("  geo: growSupp", cookie, &gamg_setup_events[SET5]);
887b4fbaa2aSMark F. Adams     PetscLogEventRegister("  geo: triangle", cookie, &gamg_setup_events[SET6]);
888b4fbaa2aSMark F. Adams     PetscLogEventRegister("   search & set", cookie, &gamg_setup_events[FIND_V]);
889b4fbaa2aSMark F. Adams     PetscLogEventRegister("  SA: init", cookie, &gamg_setup_events[SET7]);
890b4fbaa2aSMark F. Adams     /* PetscLogEventRegister("  SA: frmProl0", cookie, &gamg_setup_events[SET8]); */
891b4fbaa2aSMark F. Adams     PetscLogEventRegister("  SA: smooth", cookie, &gamg_setup_events[SET9]);
892b4fbaa2aSMark F. Adams     PetscLogEventRegister("GAMG: partLevel", cookie, &gamg_setup_events[SET2]);
893b4fbaa2aSMark F. Adams     PetscLogEventRegister(" PL repartition", cookie, &gamg_setup_events[SET12]);
894b4fbaa2aSMark F. Adams     /* PetscLogEventRegister(" PL move data", cookie, &gamg_setup_events[SET13]); */
895b4fbaa2aSMark F. Adams     /* PetscLogEventRegister("GAMG: fix", cookie, &gamg_setup_events[SET10]); */
896b4fbaa2aSMark F. Adams     /* PetscLogEventRegister("GAMG: set levels", cookie, &gamg_setup_events[SET11]); */
8975ef31b24SMark F. Adams 
898b4fbaa2aSMark F. Adams     /* create timer stages */
899b4fbaa2aSMark F. Adams #if defined GAMG_STAGES
900b4fbaa2aSMark F. Adams     {
901b4fbaa2aSMark F. Adams       char str[32];
902b4fbaa2aSMark F. Adams       sprintf(str,"MG Level %d (finest)",0);
903b4fbaa2aSMark F. Adams       PetscLogStageRegister(str, &gamg_stages[0]);
904b4fbaa2aSMark F. Adams       PetscInt lidx;
905b4fbaa2aSMark F. Adams       for (lidx=1;lidx<9;lidx++){
906b4fbaa2aSMark F. Adams 	sprintf(str,"MG Level %d",lidx);
907b4fbaa2aSMark F. Adams 	PetscLogStageRegister(str, &gamg_stages[lidx]);
908b4fbaa2aSMark F. Adams       }
909b4fbaa2aSMark F. Adams     }
910b4fbaa2aSMark F. Adams #endif
911b4fbaa2aSMark F. Adams   }
912b4fbaa2aSMark F. Adams #endif
9135b89ad90SMark F. Adams   PetscFunctionReturn(0);
9145b89ad90SMark F. Adams }
9155b89ad90SMark F. Adams EXTERN_C_END
916