Actual source code: dmimpl.h
2: #if !defined(_DMIMPL_H)
3: #define _DMIMPL_H
5: #include <petscdm.h>
6: #ifdef PETSC_HAVE_LIBCEED
7: #include <petscdmceed.h>
8: #endif
9: #include <petsc/private/petscimpl.h>
10: #include <petsc/private/petscdsimpl.h>
11: #include <petsc/private/sectionimpl.h>
13: PETSC_EXTERN PetscBool DMRegisterAllCalled;
14: PETSC_EXTERN PetscErrorCode DMRegisterAll(void);
15: typedef PetscErrorCode (*NullSpaceFunc)(DM dm, PetscInt origField, PetscInt field, MatNullSpace *nullSpace);
17: typedef struct _PetscHashAuxKey
18: {
19: DMLabel label;
20: PetscInt value;
21: } PetscHashAuxKey;
23: #define PetscHashAuxKeyHash(key) PetscHashCombine(PetscHashPointer((key).label),PetscHashInt((key).value))
25: #define PetscHashAuxKeyEqual(k1,k2) (((k1).label == (k2).label) ? ((k1).value == (k2).value) : 0)
27: PETSC_HASH_MAP(HMapAux, PetscHashAuxKey, Vec, PetscHashAuxKeyHash, PetscHashAuxKeyEqual, NULL)
29: typedef struct _DMOps *DMOps;
30: struct _DMOps {
31: PetscErrorCode (*view)(DM,PetscViewer);
32: PetscErrorCode (*load)(DM,PetscViewer);
33: PetscErrorCode (*clone)(DM,DM*);
34: PetscErrorCode (*setfromoptions)(PetscOptionItems*,DM);
35: PetscErrorCode (*setup)(DM);
36: PetscErrorCode (*createlocalsection)(DM);
37: PetscErrorCode (*createdefaultconstraints)(DM);
38: PetscErrorCode (*createglobalvector)(DM,Vec*);
39: PetscErrorCode (*createlocalvector)(DM,Vec*);
40: PetscErrorCode (*getlocaltoglobalmapping)(DM);
41: PetscErrorCode (*createfieldis)(DM,PetscInt*,char***,IS**);
42: PetscErrorCode (*createcoordinatedm)(DM,DM*);
43: PetscErrorCode (*createcoordinatefield)(DM,DMField*);
45: PetscErrorCode (*getcoloring)(DM,ISColoringType,ISColoring*);
46: PetscErrorCode (*creatematrix)(DM, Mat*);
47: PetscErrorCode (*createinterpolation)(DM,DM,Mat*,Vec*);
48: PetscErrorCode (*createrestriction)(DM,DM,Mat*);
49: PetscErrorCode (*createmassmatrix)(DM,DM,Mat*);
50: PetscErrorCode (*hascreateinjection)(DM,PetscBool*);
51: PetscErrorCode (*createinjection)(DM,DM,Mat*);
53: PetscErrorCode (*refine)(DM,MPI_Comm,DM*);
54: PetscErrorCode (*coarsen)(DM,MPI_Comm,DM*);
55: PetscErrorCode (*refinehierarchy)(DM,PetscInt,DM*);
56: PetscErrorCode (*coarsenhierarchy)(DM,PetscInt,DM*);
57: PetscErrorCode (*adaptlabel)(DM,DMLabel,DM*);
58: PetscErrorCode (*adaptmetric)(DM,Vec,DMLabel,DM*);
60: PetscErrorCode (*globaltolocalbegin)(DM,Vec,InsertMode,Vec);
61: PetscErrorCode (*globaltolocalend)(DM,Vec,InsertMode,Vec);
62: PetscErrorCode (*localtoglobalbegin)(DM,Vec,InsertMode,Vec);
63: PetscErrorCode (*localtoglobalend)(DM,Vec,InsertMode,Vec);
64: PetscErrorCode (*localtolocalbegin)(DM,Vec,InsertMode,Vec);
65: PetscErrorCode (*localtolocalend)(DM,Vec,InsertMode,Vec);
67: PetscErrorCode (*destroy)(DM);
69: PetscErrorCode (*computevariablebounds)(DM,Vec,Vec);
71: PetscErrorCode (*createsubdm)(DM,PetscInt,const PetscInt*,IS*,DM*);
72: PetscErrorCode (*createsuperdm)(DM*,PetscInt,IS**,DM*);
73: PetscErrorCode (*createfielddecomposition)(DM,PetscInt*,char***,IS**,DM**);
74: PetscErrorCode (*createdomaindecomposition)(DM,PetscInt*,char***,IS**,IS**,DM**);
75: PetscErrorCode (*createddscatters)(DM,PetscInt,DM*,VecScatter**,VecScatter**,VecScatter**);
77: PetscErrorCode (*getdimpoints)(DM,PetscInt,PetscInt*,PetscInt*);
78: PetscErrorCode (*locatepoints)(DM,Vec,DMPointLocationType,PetscSF);
79: PetscErrorCode (*getneighbors)(DM,PetscInt*,const PetscMPIInt**);
80: PetscErrorCode (*getboundingbox)(DM,PetscReal*,PetscReal*);
81: PetscErrorCode (*getlocalboundingbox)(DM,PetscReal*,PetscReal*);
82: PetscErrorCode (*locatepointssubdomain)(DM,Vec,PetscMPIInt**);
84: PetscErrorCode (*projectfunctionlocal)(DM,PetscReal,PetscErrorCode(**)(PetscInt,PetscReal,const PetscReal[],PetscInt,PetscScalar *,void *),void **,InsertMode,Vec);
85: PetscErrorCode (*projectfunctionlabellocal)(DM,PetscReal,DMLabel,PetscInt,const PetscInt[],PetscInt,const PetscInt[],PetscErrorCode(**)(PetscInt,PetscReal,const PetscReal[],PetscInt,PetscScalar *,void *),void **,InsertMode,Vec);
86: PetscErrorCode (*projectfieldlocal)(DM,PetscReal,Vec,void(**)(PetscInt,PetscInt,PetscInt,const PetscInt[],const PetscInt[],const PetscScalar[],const PetscScalar[],const PetscScalar[],const PetscInt[],const PetscInt[],const PetscScalar[],const PetscScalar[],const PetscScalar[],PetscReal,const PetscReal[],PetscInt,const PetscScalar[],PetscScalar[]),InsertMode,Vec);
87: PetscErrorCode (*projectfieldlabellocal)(DM,PetscReal,DMLabel,PetscInt,const PetscInt[],PetscInt,const PetscInt[],Vec,void(**)(PetscInt,PetscInt,PetscInt,const PetscInt[],const PetscInt[],const PetscScalar[],const PetscScalar[],const PetscScalar[],const PetscInt[],const PetscInt[],const PetscScalar[],const PetscScalar[],const PetscScalar[],PetscReal,const PetscReal[],PetscInt,const PetscScalar[],PetscScalar[]),InsertMode,Vec);
88: PetscErrorCode (*projectbdfieldlabellocal)(DM,PetscReal,DMLabel,PetscInt,const PetscInt[],PetscInt,const PetscInt[],Vec,void(**)(PetscInt,PetscInt,PetscInt,const PetscInt[],const PetscInt[],const PetscScalar[],const PetscScalar[],const PetscScalar[],const PetscInt[],const PetscInt[],const PetscScalar[],const PetscScalar[],const PetscScalar[],PetscReal,const PetscReal[],const PetscReal[],PetscInt,const PetscScalar[],PetscScalar[]),InsertMode,Vec);
89: PetscErrorCode (*computel2diff)(DM,PetscReal,PetscErrorCode(**)(PetscInt, PetscReal,const PetscReal [], PetscInt, PetscScalar *, void *), void **, Vec, PetscReal *);
90: PetscErrorCode (*computel2gradientdiff)(DM,PetscReal,PetscErrorCode(**)(PetscInt,PetscReal,const PetscReal [],const PetscReal[],PetscInt, PetscScalar *,void *),void **,Vec,const PetscReal[],PetscReal *);
91: PetscErrorCode (*computel2fielddiff)(DM,PetscReal,PetscErrorCode(**)(PetscInt, PetscReal,const PetscReal [], PetscInt, PetscScalar *, void *), void **, Vec, PetscReal *);
93: PetscErrorCode (*getcompatibility)(DM,DM,PetscBool*,PetscBool*);
94: };
96: PETSC_EXTERN PetscErrorCode DMLocalizeCoordinate_Internal(DM, PetscInt, const PetscScalar[], const PetscScalar[], PetscScalar[]);
97: PETSC_EXTERN PetscErrorCode DMLocalizeCoordinateReal_Internal(DM, PetscInt, const PetscReal[], const PetscReal[], PetscReal[]);
98: PETSC_EXTERN PetscErrorCode DMLocalizeAddCoordinate_Internal(DM, PetscInt, const PetscScalar[], const PetscScalar[], PetscScalar[]);
100: typedef struct _DMCoarsenHookLink *DMCoarsenHookLink;
101: struct _DMCoarsenHookLink {
102: PetscErrorCode (*coarsenhook)(DM,DM,void*); /* Run once, when coarse DM is created */
103: PetscErrorCode (*restricthook)(DM,Mat,Vec,Mat,DM,void*); /* Run each time a new problem is restricted to a coarse grid */
104: void *ctx;
105: DMCoarsenHookLink next;
106: };
108: typedef struct _DMRefineHookLink *DMRefineHookLink;
109: struct _DMRefineHookLink {
110: PetscErrorCode (*refinehook)(DM,DM,void*); /* Run once, when a fine DM is created */
111: PetscErrorCode (*interphook)(DM,Mat,DM,void*); /* Run each time a new problem is interpolated to a fine grid */
112: void *ctx;
113: DMRefineHookLink next;
114: };
116: typedef struct _DMSubDomainHookLink *DMSubDomainHookLink;
117: struct _DMSubDomainHookLink {
118: PetscErrorCode (*ddhook)(DM,DM,void*);
119: PetscErrorCode (*restricthook)(DM,VecScatter,VecScatter,DM,void*);
120: void *ctx;
121: DMSubDomainHookLink next;
122: };
124: typedef struct _DMGlobalToLocalHookLink *DMGlobalToLocalHookLink;
125: struct _DMGlobalToLocalHookLink {
126: PetscErrorCode (*beginhook)(DM,Vec,InsertMode,Vec,void*);
127: PetscErrorCode (*endhook)(DM,Vec,InsertMode,Vec,void*);
128: void *ctx;
129: DMGlobalToLocalHookLink next;
130: };
132: typedef struct _DMLocalToGlobalHookLink *DMLocalToGlobalHookLink;
133: struct _DMLocalToGlobalHookLink {
134: PetscErrorCode (*beginhook)(DM,Vec,InsertMode,Vec,void*);
135: PetscErrorCode (*endhook)(DM,Vec,InsertMode,Vec,void*);
136: void *ctx;
137: DMLocalToGlobalHookLink next;
138: };
140: typedef enum {DMVEC_STATUS_IN,DMVEC_STATUS_OUT} DMVecStatus;
141: typedef struct _DMNamedVecLink *DMNamedVecLink;
142: struct _DMNamedVecLink {
143: Vec X;
144: char *name;
145: DMVecStatus status;
146: DMNamedVecLink next;
147: };
149: typedef struct _DMWorkLink *DMWorkLink;
150: struct _DMWorkLink {
151: size_t bytes;
152: void *mem;
153: DMWorkLink next;
154: };
156: #define DM_MAX_WORK_VECTORS 100 /* work vectors available to users via DMGetGlobalVector(), DMGetLocalVector() */
158: struct _n_DMLabelLink {
159: DMLabel label;
160: PetscBool output;
161: struct _n_DMLabelLink *next;
162: };
163: typedef struct _n_DMLabelLink *DMLabelLink;
165: typedef struct _n_Boundary *DMBoundary;
167: struct _n_Boundary {
168: DSBoundary dsboundary;
169: DMLabel label;
170: DMBoundary next;
171: };
173: typedef struct _n_Field {
174: PetscObject disc; /* Field discretization, or a PetscContainer with the field name */
175: DMLabel label; /* Label defining the domain of definition of the field */
176: PetscBool adjacency[2]; /* Flags for defining variable influence (adjacency) for each field [use cone() or support() first, use the transitive closure] */
177: PetscBool avoidTensor; /* Flag to avoid defining field over tensor cells */
178: } RegionField;
180: typedef struct _n_Space {
181: PetscDS ds; /* Approximation space in this domain */
182: DMLabel label; /* Label defining the domain of definition of the discretization */
183: IS fields; /* Map from DS field numbers to original field numbers in the DM */
184: } DMSpace;
186: struct _p_UniversalLabel {
187: DMLabel label; /* The universal label */
188: PetscInt Nl; /* Number of labels encoded */
189: char **names; /* The label names */
190: PetscInt *indices; /* The original indices in the input DM */
191: PetscInt Nv; /* Total number of values in all the labels */
192: PetscInt *bits; /* Starting bit for values of each label */
193: PetscInt *masks; /* Masks to pull out label value bits for each label */
194: PetscInt *offsets; /* Starting offset for label values for each label */
195: PetscInt *values; /* Original label values before renumbering */
196: };
198: PETSC_INTERN PetscErrorCode DMDestroyLabelLinkList_Internal(DM);
200: #define MAXDMMONITORS 5
202: struct _p_DM {
203: PETSCHEADER(struct _DMOps);
204: Vec localin[DM_MAX_WORK_VECTORS],localout[DM_MAX_WORK_VECTORS];
205: Vec globalin[DM_MAX_WORK_VECTORS],globalout[DM_MAX_WORK_VECTORS];
206: DMNamedVecLink namedglobal;
207: DMNamedVecLink namedlocal;
208: DMWorkLink workin,workout;
209: DMLabelLink labels; /* Linked list of labels */
210: DMLabel depthLabel; /* Optimized access to depth label */
211: DMLabel celltypeLabel; /* Optimized access to celltype label */
212: void *ctx; /* a user context */
213: PetscErrorCode (*ctxdestroy)(void**);
214: ISColoringType coloringtype;
215: MatFDColoring fd;
216: VecType vectype; /* type of vector created with DMCreateLocalVector() and DMCreateGlobalVector() */
217: MatType mattype; /* type of matrix created with DMCreateMatrix() */
218: PetscInt bs;
219: ISLocalToGlobalMapping ltogmap;
220: PetscBool prealloc_only; /* Flag indicating the DMCreateMatrix() should only preallocate, not fill the matrix */
221: PetscBool structure_only; /* Flag indicating the DMCreateMatrix() create matrix structure without values */
222: PetscInt levelup,leveldown; /* if the DM has been obtained by refining (or coarsening) this indicates how many times that process has been used to generate this DM */
223: PetscBool setupcalled; /* Indicates that the DM has been set up, methods that modify a DM such that a fresh setup is required should reset this flag */
224: PetscBool setfromoptionscalled;
225: void *data;
226: /* Hierarchy / Submeshes */
227: DM coarseMesh;
228: DM fineMesh;
229: DMCoarsenHookLink coarsenhook; /* For transfering auxiliary problem data to coarser grids */
230: DMRefineHookLink refinehook;
231: DMSubDomainHookLink subdomainhook;
232: DMGlobalToLocalHookLink gtolhook;
233: DMLocalToGlobalHookLink ltoghook;
234: /* Topology */
235: PetscInt dim; /* The topological dimension */
236: /* Auxiliary data */
237: PetscHMapAux auxData; /* Auxiliary DM and Vec for region denoted by the key */
238: /* Flexible communication */
239: PetscSF sfMigration; /* SF for point distribution created during distribution */
240: PetscSF sf; /* SF for parallel point overlap */
241: PetscSF sectionSF; /* SF for parallel dof overlap using section */
242: PetscSF sfNatural; /* SF mapping to the "natural" ordering */
243: PetscBool useNatural; /* Create the natural SF */
244: /* Allows a non-standard data layout */
245: PetscBool adjacency[2]; /* [use cone() or support() first, use the transitive closure] */
246: PetscSection localSection; /* Layout for local vectors */
247: PetscSection globalSection; /* Layout for global vectors */
248: PetscLayout map;
249: /* Constraints */
250: PetscSection defaultConstraintSection;
251: Mat defaultConstraintMat;
252: /* Basis transformation */
253: DM transformDM; /* Layout for basis transformation */
254: Vec transform; /* Basis transformation matrices */
255: void *transformCtx; /* Basis transformation context */
256: PetscErrorCode (*transformSetUp)(DM, void *);
257: PetscErrorCode (*transformDestroy)(DM, void *);
258: PetscErrorCode (*transformGetMatrix)(DM, const PetscReal[], PetscBool, const PetscScalar **, void *);
259: /* Coordinates */
260: PetscInt dimEmbed; /* The dimension of the embedding space */
261: DM coordinateDM; /* Layout for coordinates */
262: Vec coordinates; /* Coordinate values in global vector */
263: Vec coordinatesLocal; /* Coordinate values in local vector */
264: PetscBool periodic; /* Is the DM periodic? */
265: DMField coordinateField; /* Coordinates as an abstract field */
266: PetscReal *L, *maxCell; /* Size of periodic box and max cell size for determining periodicity */
267: DMBoundaryType *bdtype; /* Indicates type of topological boundary */
268: /* Null spaces -- of course I should make this have a variable number of fields */
269: NullSpaceFunc nullspaceConstructors[10];
270: NullSpaceFunc nearnullspaceConstructors[10];
271: /* Fields are represented by objects */
272: PetscInt Nf; /* Number of fields defined on the total domain */
273: RegionField *fields; /* Array of discretization fields with regions of validity */
274: DMBoundary boundary; /* List of boundary conditions */
275: PetscInt Nds; /* Number of discrete systems defined on the total domain */
276: DMSpace *probs; /* Array of discrete systems */
277: /* Output structures */
278: DM dmBC; /* The DM with boundary conditions in the global DM */
279: PetscInt outputSequenceNum; /* The current sequence number for output */
280: PetscReal outputSequenceVal; /* The current sequence value for output */
281: PetscErrorCode (*monitor[MAXDMMONITORS])(DM, void *);
282: PetscErrorCode (*monitordestroy[MAXDMMONITORS])(void **);
283: void *monitorcontext[MAXDMMONITORS];
284: PetscInt numbermonitors;
286: PetscObject dmksp,dmsnes,dmts;
287: #ifdef PETSC_HAVE_LIBCEED
288: Ceed ceed; /* LibCEED context */
289: CeedElemRestriction ceedERestrict; /* Map from the local vector (Lvector) to the cells (Evector) */
290: #endif
291: };
293: PETSC_EXTERN PetscLogEvent DM_Convert;
294: PETSC_EXTERN PetscLogEvent DM_GlobalToLocal;
295: PETSC_EXTERN PetscLogEvent DM_LocalToGlobal;
296: PETSC_EXTERN PetscLogEvent DM_LocatePoints;
297: PETSC_EXTERN PetscLogEvent DM_Coarsen;
298: PETSC_EXTERN PetscLogEvent DM_Refine;
299: PETSC_EXTERN PetscLogEvent DM_CreateInterpolation;
300: PETSC_EXTERN PetscLogEvent DM_CreateRestriction;
301: PETSC_EXTERN PetscLogEvent DM_CreateInjection;
302: PETSC_EXTERN PetscLogEvent DM_CreateMatrix;
303: PETSC_EXTERN PetscLogEvent DM_Load;
304: PETSC_EXTERN PetscLogEvent DM_AdaptInterpolator;
306: PETSC_EXTERN PetscErrorCode DMCreateGlobalVector_Section_Private(DM,Vec*);
307: PETSC_EXTERN PetscErrorCode DMCreateLocalVector_Section_Private(DM,Vec*);
309: PETSC_EXTERN PetscErrorCode DMView_GLVis(DM,PetscViewer,PetscErrorCode(*)(DM,PetscViewer));
311: /*
313: Composite Vectors
315: Single global representation
316: Individual global representations
317: Single local representation
318: Individual local representations
320: Subsets of individual as a single????? Do we handle this by having DMComposite inside composite??????
322: DM da_u, da_v, da_p
324: DM dm_velocities
326: DM dm
328: DMDACreate(,&da_u);
329: DMDACreate(,&da_v);
330: DMCompositeCreate(,&dm_velocities);
331: DMCompositeAddDM(dm_velocities,(DM)du);
332: DMCompositeAddDM(dm_velocities,(DM)dv);
334: DMDACreate(,&da_p);
335: DMCompositeCreate(,&dm_velocities);
336: DMCompositeAddDM(dm,(DM)dm_velocities);
337: DMCompositeAddDM(dm,(DM)dm_p);
339: Access parts of composite vectors (Composite only)
340: ---------------------------------
341: DMCompositeGetAccess - access the global vector as subvectors and array (for redundant arrays)
342: ADD for local vector -
344: Element access
345: --------------
346: From global vectors
347: -DAVecGetArray - for DMDA
348: -VecGetArray - for DMSliced
349: ADD for DMComposite??? maybe
351: From individual vector
352: -DAVecGetArray - for DMDA
353: -VecGetArray -for sliced
354: ADD for DMComposite??? maybe
356: From single local vector
357: ADD * single local vector as arrays?
359: Communication
360: -------------
361: DMGlobalToLocal - global vector to single local vector
363: DMCompositeScatter/Gather - direct to individual local vectors and arrays CHANGE name closer to GlobalToLocal?
365: Obtaining vectors
366: -----------------
367: DMCreateGlobal/Local
368: DMGetGlobal/Local
369: DMCompositeGetLocalVectors - gives individual local work vectors and arrays
371: ????? individual global vectors ????
373: */
375: #if defined(PETSC_HAVE_HDF5)
376: PETSC_EXTERN PetscErrorCode DMSequenceLoad_HDF5_Internal(DM, const char *, PetscInt, PetscScalar *, PetscViewer);
377: #endif
379: PETSC_STATIC_INLINE PetscErrorCode DMGetLocalOffset_Private(DM dm, PetscInt point, PetscInt *start, PetscInt *end)
380: {
382: #if defined(PETSC_USE_DEBUG)
383: {
384: PetscInt dof;
386: *start = *end = 0; /* Silence overzealous compiler warning */
387: if (!dm->localSection) SETERRQ(PetscObjectComm((PetscObject) dm), PETSC_ERR_ARG_WRONG, "DM must have a local section, see DMSetLocalSection()");
388: PetscSectionGetOffset(dm->localSection, point, start);
389: PetscSectionGetDof(dm->localSection, point, &dof);
390: *end = *start + dof;
391: }
392: #else
393: {
394: const PetscSection s = dm->localSection;
395: *start = s->atlasOff[point - s->pStart];
396: *end = *start + s->atlasDof[point - s->pStart];
397: }
398: #endif
399: return(0);
400: }
402: PETSC_STATIC_INLINE PetscErrorCode DMGetLocalFieldOffset_Private(DM dm, PetscInt point, PetscInt field, PetscInt *start, PetscInt *end)
403: {
405: #if defined(PETSC_USE_DEBUG)
406: {
407: PetscInt dof;
409: *start = *end = 0; /* Silence overzealous compiler warning */
410: if (!dm->localSection) SETERRQ(PetscObjectComm((PetscObject) dm), PETSC_ERR_ARG_WRONG, "DM must have a local section, see DMSetLocalSection()");
411: PetscSectionGetFieldOffset(dm->localSection, point, field, start);
412: PetscSectionGetFieldDof(dm->localSection, point, field, &dof);
413: *end = *start + dof;
414: }
415: #else
416: {
417: const PetscSection s = dm->localSection->field[field];
418: *start = s->atlasOff[point - s->pStart];
419: *end = *start + s->atlasDof[point - s->pStart];
420: }
421: #endif
422: return(0);
423: }
425: PETSC_STATIC_INLINE PetscErrorCode DMGetGlobalOffset_Private(DM dm, PetscInt point, PetscInt *start, PetscInt *end)
426: {
428: #if defined(PETSC_USE_DEBUG)
429: {
431: PetscInt dof,cdof;
432: *start = *end = 0; /* Silence overzealous compiler warning */
433: if (!dm->localSection) SETERRQ(PetscObjectComm((PetscObject) dm), PETSC_ERR_ARG_WRONG, "DM must have a local section, see DMSetLocalSection()");
434: if (!dm->globalSection) SETERRQ(PetscObjectComm((PetscObject) dm), PETSC_ERR_ARG_WRONG, "DM must have a global section. It will be created automatically by DMGetGlobalSection()");
435: PetscSectionGetOffset(dm->globalSection, point, start);
436: PetscSectionGetDof(dm->globalSection, point, &dof);
437: PetscSectionGetConstraintDof(dm->globalSection, point, &cdof);
438: *end = *start + dof - cdof + (dof < 0 ? 1 : 0);
439: }
440: #else
441: {
442: const PetscSection s = dm->globalSection;
443: const PetscInt dof = s->atlasDof[point - s->pStart];
444: const PetscInt cdof = s->bc ? s->bc->atlasDof[point - s->bc->pStart] : 0;
445: *start = s->atlasOff[point - s->pStart];
446: *end = *start + dof - cdof + (dof < 0 ? 1 : 0);
447: }
448: #endif
449: return(0);
450: }
452: PETSC_STATIC_INLINE PetscErrorCode DMGetGlobalFieldOffset_Private(DM dm, PetscInt point, PetscInt field, PetscInt *start, PetscInt *end)
453: {
455: #if defined(PETSC_USE_DEBUG)
456: {
457: PetscInt loff, lfoff, fdof, fcdof, ffcdof, f;
459: *start = *end = 0; /* Silence overzealous compiler warning */
460: if (!dm->localSection) SETERRQ(PetscObjectComm((PetscObject) dm), PETSC_ERR_ARG_WRONG, "DM must have a local section, see DMSetLocalSection()");
461: if (!dm->globalSection) SETERRQ(PetscObjectComm((PetscObject) dm), PETSC_ERR_ARG_WRONG, "DM must have a global section. It will be crated automatically by DMGetGlobalSection()");
462: PetscSectionGetOffset(dm->globalSection, point, start);
463: PetscSectionGetOffset(dm->localSection, point, &loff);
464: PetscSectionGetFieldOffset(dm->localSection, point, field, &lfoff);
465: PetscSectionGetFieldDof(dm->localSection, point, field, &fdof);
466: PetscSectionGetFieldConstraintDof(dm->localSection, point, field, &fcdof);
467: *start = *start < 0 ? *start - (lfoff-loff) : *start + lfoff-loff;
468: for (f = 0; f < field; ++f) {
469: PetscSectionGetFieldConstraintDof(dm->localSection, point, f, &ffcdof);
470: *start = *start < 0 ? *start + ffcdof : *start - ffcdof;
471: }
472: *end = *start < 0 ? *start - (fdof-fcdof) : *start + fdof-fcdof;
473: }
474: #else
475: {
476: const PetscSection s = dm->localSection;
477: const PetscSection fs = dm->localSection->field[field];
478: const PetscSection gs = dm->globalSection;
479: const PetscInt loff = s->atlasOff[point - s->pStart];
480: const PetscInt goff = gs->atlasOff[point - s->pStart];
481: const PetscInt lfoff = fs->atlasOff[point - s->pStart];
482: const PetscInt fdof = fs->atlasDof[point - s->pStart];
483: const PetscInt fcdof = fs->bc ? fs->bc->atlasDof[point - fs->bc->pStart] : 0;
484: PetscInt ffcdof = 0, f;
486: for (f = 0; f < field; ++f) {
487: const PetscSection ffs = dm->localSection->field[f];
488: ffcdof += ffs->bc ? ffs->bc->atlasDof[point - ffs->bc->pStart] : 0;
489: }
490: *start = goff + (goff < 0 ? loff-lfoff + ffcdof : lfoff-loff - ffcdof);
491: *end = *start < 0 ? *start - (fdof-fcdof) : *start + fdof-fcdof;
492: }
493: #endif
494: return(0);
495: }
497: PETSC_EXTERN PetscErrorCode DMGetBasisTransformDM_Internal(DM, DM *);
498: PETSC_EXTERN PetscErrorCode DMGetBasisTransformVec_Internal(DM, Vec *);
499: PETSC_INTERN PetscErrorCode DMConstructBasisTransform_Internal(DM);
501: PETSC_INTERN PetscErrorCode DMGetLocalBoundingIndices_DMDA(DM, PetscReal[], PetscReal[]);
502: PETSC_INTERN PetscErrorCode DMSetField_Internal(DM, PetscInt, DMLabel, PetscObject);
504: PETSC_INTERN PetscErrorCode DMSetLabelValue_Fast(DM, DMLabel*, const char[], PetscInt, PetscInt);
506: PETSC_EXTERN PetscErrorCode DMUniversalLabelCreate(DM, DMUniversalLabel *);
507: PETSC_EXTERN PetscErrorCode DMUniversalLabelDestroy(DMUniversalLabel *);
508: PETSC_EXTERN PetscErrorCode DMUniversalLabelGetLabel(DMUniversalLabel, DMLabel *);
509: PETSC_EXTERN PetscErrorCode DMUniversalLabelCreateLabels(DMUniversalLabel, PetscBool, DM);
510: PETSC_EXTERN PetscErrorCode DMUniversalLabelSetLabelValue(DMUniversalLabel, DM, PetscBool, PetscInt, PetscInt);
512: #endif