source: issm/trunk-jpl/externalpackages/petsc-dev/src/include/petsc-private/daimpl.h@ 11896

Last change on this file since 11896 was 11896, checked in by habbalf, 13 years ago

petsc-dev : Petsc development code in external packages. Mercurial updates
File size: 5.9 KB
Line 
1/*
2 Distributed arrays - communication tools for parallel, rectangular grids.
3*/
4
5#if !defined(_DAIMPL_H)
6#define _DAIMPL_H
7
8#include <petscdmda.h>
9#include "petsc-private/dmimpl.h"
10
11typedef struct {
12 PetscInt M,N,P; /* array dimensions */
13 PetscInt m,n,p; /* processor layout */
14 PetscInt w; /* degrees of freedom per node */
15 PetscInt s; /* stencil width */
16 PetscInt xs,xe,ys,ye,zs,ze; /* range of local values */
17 PetscInt Xs,Xe,Ys,Ye,Zs,Ze; /* range including ghost values
18 values above already scaled by w */
19 PetscInt *idx,Nl; /* local to global map */
20 PetscInt base; /* global number of 1st local node */
21 DMDABoundaryType bx,by,bz; /* indicates type of ghost nodes at boundary */
22 VecScatter gtol,ltog,ltol; /* scatters, see below for details */
23 DMDAStencilType stencil_type; /* stencil, either box or star */
24 PetscInt dim; /* DMDA dimension (1,2, or 3) */
25 DMDAInterpolationType interptype;
26
27 PetscInt nlocal,Nlocal; /* local size of local vector and global vector */
28
29 AO ao; /* application ordering context */
30
31 Vec coordinates; /* coordinates (x,y,z) of local nodes, not including ghosts*/
32 DM da_coordinates; /* da for getting ghost values of coordinates */
33 Vec ghosted_coordinates;/* coordinates with ghost nodes */
34 char **fieldname; /* names of individual components in vectors */
35
36 PetscInt *lx,*ly,*lz; /* number of nodes in each partition block along 3 axis */
37 Vec natural; /* global vector for storing items in natural order */
38 VecScatter gton; /* vector scatter from global to natural */
39 PetscMPIInt *neighbors; /* ranks of all neighbors and self */
40
41 ISColoring localcoloring; /* set by DMCreateColoring() */
42 ISColoring ghostedcoloring;
43
44 DMDAElementType elementtype;
45 PetscInt ne; /* number of elements */
46 PetscInt *e; /* the elements */
47
48 PetscInt refine_x,refine_y,refine_z; /* ratio used in refining */
49
50#define DMDA_MAX_AD_ARRAYS 2 /* work arrays for holding derivative type data, via DMDAGetAdicArray() */
51 void *adarrayin[DMDA_MAX_AD_ARRAYS],*adarrayout[DMDA_MAX_AD_ARRAYS];
52 void *adarrayghostedin[DMDA_MAX_AD_ARRAYS],*adarrayghostedout[DMDA_MAX_AD_ARRAYS];
53 void *adstartin[DMDA_MAX_AD_ARRAYS],*adstartout[DMDA_MAX_AD_ARRAYS];
54 void *adstartghostedin[DMDA_MAX_AD_ARRAYS],*adstartghostedout[DMDA_MAX_AD_ARRAYS];
55 PetscInt tdof,ghostedtdof;
56
57 /* work arrays for holding derivative type data, via DMDAGetAdicMFArray() */
58 void *admfarrayin[DMDA_MAX_AD_ARRAYS],*admfarrayout[DMDA_MAX_AD_ARRAYS];
59 void *admfarrayghostedin[DMDA_MAX_AD_ARRAYS],*admfarrayghostedout[DMDA_MAX_AD_ARRAYS];
60 void *admfstartin[DMDA_MAX_AD_ARRAYS],*admfstartout[DMDA_MAX_AD_ARRAYS];
61 void *admfstartghostedin[DMDA_MAX_AD_ARRAYS],*admfstartghostedout[DMDA_MAX_AD_ARRAYS];
62
63#define DMDA_MAX_WORK_ARRAYS 2 /* work arrays for holding work via DMDAGetArray() */
64 void *arrayin[DMDA_MAX_WORK_ARRAYS],*arrayout[DMDA_MAX_WORK_ARRAYS];
65 void *arrayghostedin[DMDA_MAX_WORK_ARRAYS],*arrayghostedout[DMDA_MAX_WORK_ARRAYS];
66 void *startin[DMDA_MAX_WORK_ARRAYS],*startout[DMDA_MAX_WORK_ARRAYS];
67 void *startghostedin[DMDA_MAX_WORK_ARRAYS],*startghostedout[DMDA_MAX_WORK_ARRAYS];
68
69 DMDALocalFunction1 lf;
70 DMDALocalFunction1 lj;
71 DMDALocalFunction1 adic_lf;
72 DMDALocalFunction1 adicmf_lf;
73 DMDALocalFunction1 adifor_lf;
74 DMDALocalFunction1 adiformf_lf;
75
76 PetscErrorCode (*lfi)(DMDALocalInfo*,MatStencil*,void*,PetscScalar*,void*);
77 PetscErrorCode (*adic_lfi)(DMDALocalInfo*,MatStencil*,void*,void*,void*);
78 PetscErrorCode (*adicmf_lfi)(DMDALocalInfo*,MatStencil*,void*,void*,void*);
79 PetscErrorCode (*lfib)(DMDALocalInfo*,MatStencil*,void*,PetscScalar*,void*);
80 PetscErrorCode (*adic_lfib)(DMDALocalInfo*,MatStencil*,void*,void*,void*);
81 PetscErrorCode (*adicmf_lfib)(DMDALocalInfo*,MatStencil*,void*,void*,void*);
82
83 /* used by DMDASetBlockFills() */
84 PetscInt *ofill,*dfill;
85
86 /* used by DMDASetMatPreallocateOnly() */
87 PetscBool prealloc_only;
88} DM_DA;
89
90/*
91 Vectors:
92 Global has on each processor the interior degrees of freedom and
93 no ghost points. This vector is what the solvers usually see.
94 Local has on each processor the ghost points as well. This is
95 what code to calculate Jacobians, etc. usually sees.
96 Vector scatters:
97 gtol - Global representation to local
98 ltog - Local representation to global (involves no communication)
99 ltol - Local representation to local representation, updates the
100 ghostpoint values in the second vector from (correct) interior
101 values in the first vector. This is good for explicit
102 nearest neighbor timestepping.
103*/
104
105EXTERN_C_BEGIN
106extern PetscErrorCode VecView_MPI_DA(Vec,PetscViewer);
107extern PetscErrorCode VecLoad_Default_DA(Vec, PetscViewer);
108EXTERN_C_END
109extern PetscErrorCode DMView_DA_Private(DM);
110extern PetscErrorCode DMView_DA_Matlab(DM,PetscViewer);
111extern PetscErrorCode DMView_DA_Binary(DM,PetscViewer);
112extern PetscErrorCode DMView_DA_VTK(DM,PetscViewer);
113extern PetscErrorCode DMDAVTKWriteAll(PetscObject,PetscViewer);
114
115extern PetscLogEvent DMDA_LocalADFunction;
116
117#endif
Note: See TracBrowser for help on using the repository browser.