Actual source code: pcgamgimpl.h
1: #ifndef PETSC_PCGAMGIMPL_H
2: #define PETSC_PCGAMGIMPL_H
3: #include <petscksp.h>
4: #include <petsc/private/pcimpl.h>
5: #include <petsc/private/pcmgimpl.h>
6: #include <petscmatcoarsen.h>
8: struct _PCGAMGOps {
9: PetscErrorCode (*creategraph)(PC, Mat, Mat *);
10: PetscErrorCode (*coarsen)(PC, Mat *, PetscCoarsenData **);
11: PetscErrorCode (*prolongator)(PC, Mat, Mat, PetscCoarsenData *, Mat *);
12: PetscErrorCode (*optprolongator)(PC, Mat, Mat *);
13: PetscErrorCode (*createlevel)(PC, Mat, PetscInt, Mat *, Mat *, PetscMPIInt *, IS *, PetscBool);
14: PetscErrorCode (*createdefaultdata)(PC, Mat); /* for data methods that have a default (SA) */
15: PetscErrorCode (*setfromoptions)(PC, PetscOptionItems *);
16: PetscErrorCode (*destroy)(PC);
17: PetscErrorCode (*view)(PC, PetscViewer);
18: };
19: /* Private context for the GAMG preconditioner */
20: typedef struct gamg_TAG {
21: PCGAMGType type;
22: PetscInt Nlevels;
23: PetscBool repart;
24: PetscBool reuse_prol;
25: PetscBool use_aggs_in_asm;
26: PetscBool use_parallel_coarse_grid_solver;
27: PCGAMGLayoutType layout_type;
28: PetscBool cpu_pin_coarse_grids;
29: PetscInt min_eq_proc;
30: PetscInt coarse_eq_limit;
31: PetscReal threshold_scale;
32: PetscReal threshold[PETSC_MG_MAXLEVELS]; /* common quatity to many AMG methods so keep it up here */
33: PetscInt level_reduction_factors[PETSC_MG_MAXLEVELS];
34: PetscInt current_level; /* stash construction state */
35: /* these 4 are all related to the method data and should be in the subctx */
36: PetscInt data_sz; /* nloc*data_rows*data_cols */
37: PetscInt data_cell_rows;
38: PetscInt data_cell_cols;
39: PetscInt orig_data_cell_rows;
40: PetscInt orig_data_cell_cols;
41: PetscReal *data; /* [data_sz] blocked vector of vertex data on fine grid (coordinates/nullspace) */
42: PetscReal *orig_data; /* cache data */
44: struct _PCGAMGOps *ops;
45: char *gamg_type_name;
47: void *subctx;
49: PetscBool use_sa_esteig;
50: PetscReal emin, emax;
51: } PC_GAMG;
53: PetscErrorCode PCReset_MG(PC);
55: /* hooks create derivied classes */
56: PetscErrorCode PCCreateGAMG_GEO(PC);
57: PetscErrorCode PCCreateGAMG_AGG(PC);
58: PetscErrorCode PCCreateGAMG_Classical(PC);
60: PetscErrorCode PCDestroy_GAMG(PC);
62: /* helper methods */
63: PetscErrorCode PCGAMGGetDataWithGhosts(Mat, PetscInt, PetscReal[], PetscInt *, PetscReal **);
65: enum tag {
66: GAMG_SETUP = 0,
67: GAMG_MESH,
68: GAMG_MATRIX,
69: GAMG_GRAPH,
70: GAMG_COARSEN,
71: GAMG_SQUARE,
72: GAMG_MIS,
73: GAMG_PROL,
74: GAMG_PROLA,
75: GAMG_PROLB,
76: GAMG_OPT,
77: GAMG_OPTSM,
78: GAMG_LEVEL,
79: GAMG_PTAP,
80: GAMG_REDUCE,
81: GAMG_REPART,
82: SET13,
83: SET14,
84: SET15,
85: GAMG_NUM_SET
86: };
87: PETSC_EXTERN PetscLogEvent petsc_gamg_setup_events[GAMG_NUM_SET];
88: PETSC_EXTERN PetscLogEvent petsc_gamg_setup_matmat_events[PETSC_MG_MAXLEVELS][3];
90: typedef struct _PCGAMGHashTable {
91: PetscInt *table;
92: PetscInt *data;
93: PetscInt size;
94: } PCGAMGHashTable;
96: PETSC_INTERN PetscErrorCode PCGAMGHashTableCreate(PetscInt, PCGAMGHashTable *);
97: PETSC_INTERN PetscErrorCode PCGAMGHashTableDestroy(PCGAMGHashTable *);
98: PETSC_INTERN PetscErrorCode PCGAMGHashTableAdd(PCGAMGHashTable *, PetscInt, PetscInt);
100: #define GAMG_HASH(key) (PetscInt)((((PetscInt64)7) * (PetscInt64)key) % (PetscInt64)a_tab->size)
101: static inline PetscErrorCode PCGAMGHashTableFind(PCGAMGHashTable *a_tab, PetscInt a_key, PetscInt *a_data)
102: {
103: PetscInt kk, idx;
106: for (kk = 0, idx = GAMG_HASH(a_key); kk < a_tab->size; kk++, idx = (idx == (a_tab->size - 1)) ? 0 : idx + 1) {
107: if (a_tab->table[idx] == a_key) {
108: *a_data = a_tab->data[idx];
109: break;
110: } else if (a_tab->table[idx] == -1) {
111: /* not here */
112: *a_data = -1;
113: break;
114: }
115: }
117: return 0;
118: }
120: #endif // PETSC_PCGAMGIMPL_H