xref: /petsc/src/dm/interface/dm.c (revision 6801d45cb9d4653c0f00f6a72992dfe249a93cf2)
1 #include <petsc/private/dmimpl.h>           /*I      "petscdm.h"          I*/
2 #include <petsc/private/dmlabelimpl.h>      /*I      "petscdmlabel.h"     I*/
3 #include <petsc/private/petscdsimpl.h>      /*I      "petscds.h"     I*/
4 #include <petscdmplex.h>
5 #include <petscdmfield.h>
6 #include <petscsf.h>
7 #include <petscds.h>
8 
9 PetscClassId  DM_CLASSID;
10 PetscClassId  DMLABEL_CLASSID;
11 PetscLogEvent DM_Convert, DM_GlobalToLocal, DM_LocalToGlobal, DM_LocalToLocal, DM_LocatePoints, DM_Coarsen, DM_Refine, DM_CreateInterpolation, DM_CreateRestriction;
12 
13 const char *const DMBoundaryTypes[] = {"NONE","GHOSTED","MIRROR","PERIODIC","TWIST","DMBoundaryType","DM_BOUNDARY_",0};
14 
15 static PetscErrorCode DMHasCreateInjection_Default(DM dm, PetscBool *flg)
16 {
17   PetscFunctionBegin;
18   PetscValidHeaderSpecific(dm,DM_CLASSID,1);
19   PetscValidPointer(flg,2);
20   *flg = PETSC_FALSE;
21   PetscFunctionReturn(0);
22 }
23 
24 /*@
25   DMCreate - Creates an empty DM object. The type can then be set with DMSetType().
26 
27    If you never  call DMSetType()  it will generate an
28    error when you try to use the vector.
29 
30   Collective on MPI_Comm
31 
32   Input Parameter:
33 . comm - The communicator for the DM object
34 
35   Output Parameter:
36 . dm - The DM object
37 
38   Level: beginner
39 
40 .seealso: DMSetType(), DMDA, DMSLICED, DMCOMPOSITE, DMPLEX, DMMOAB, DMNETWORK
41 @*/
42 PetscErrorCode  DMCreate(MPI_Comm comm,DM *dm)
43 {
44   DM             v;
45   PetscDS        ds;
46   PetscErrorCode ierr;
47 
48   PetscFunctionBegin;
49   PetscValidPointer(dm,2);
50   *dm = NULL;
51   ierr = DMInitializePackage();CHKERRQ(ierr);
52 
53   ierr = PetscHeaderCreate(v, DM_CLASSID, "DM", "Distribution Manager", "DM", comm, DMDestroy, DMView);CHKERRQ(ierr);
54 
55   v->setupcalled              = PETSC_FALSE;
56   v->setfromoptionscalled     = PETSC_FALSE;
57   v->ltogmap                  = NULL;
58   v->bs                       = 1;
59   v->coloringtype             = IS_COLORING_GLOBAL;
60   ierr                        = PetscSFCreate(comm, &v->sf);CHKERRQ(ierr);
61   ierr                        = PetscSFCreate(comm, &v->defaultSF);CHKERRQ(ierr);
62   v->labels                   = NULL;
63   v->depthLabel               = NULL;
64   v->defaultSection           = NULL;
65   v->defaultGlobalSection     = NULL;
66   v->defaultConstraintSection = NULL;
67   v->defaultConstraintMat     = NULL;
68   v->L                        = NULL;
69   v->maxCell                  = NULL;
70   v->bdtype                   = NULL;
71   v->dimEmbed                 = PETSC_DEFAULT;
72   v->dim                      = PETSC_DETERMINE;
73   {
74     PetscInt i;
75     for (i = 0; i < 10; ++i) {
76       v->nullspaceConstructors[i] = NULL;
77       v->nearnullspaceConstructors[i] = NULL;
78     }
79   }
80   ierr = PetscDSCreate(PetscObjectComm((PetscObject) v), &ds);CHKERRQ(ierr);
81   ierr = DMSetRegionDS(v, NULL, ds);CHKERRQ(ierr);
82   ierr = PetscDSDestroy(&ds);CHKERRQ(ierr);
83   v->dmBC = NULL;
84   v->coarseMesh = NULL;
85   v->outputSequenceNum = -1;
86   v->outputSequenceVal = 0.0;
87   ierr = DMSetVecType(v,VECSTANDARD);CHKERRQ(ierr);
88   ierr = DMSetMatType(v,MATAIJ);CHKERRQ(ierr);
89   ierr = PetscNew(&(v->labels));CHKERRQ(ierr);
90   v->labels->refct = 1;
91 
92   v->ops->hascreateinjection = DMHasCreateInjection_Default;
93 
94   *dm = v;
95   PetscFunctionReturn(0);
96 }
97 
98 /*@
99   DMClone - Creates a DM object with the same topology as the original.
100 
101   Collective on MPI_Comm
102 
103   Input Parameter:
104 . dm - The original DM object
105 
106   Output Parameter:
107 . newdm  - The new DM object
108 
109   Level: beginner
110 
111 .keywords: DM, topology, create
112 @*/
113 PetscErrorCode DMClone(DM dm, DM *newdm)
114 {
115   PetscSF        sf;
116   Vec            coords;
117   void          *ctx;
118   PetscInt       dim, cdim;
119   PetscErrorCode ierr;
120 
121   PetscFunctionBegin;
122   PetscValidHeaderSpecific(dm, DM_CLASSID, 1);
123   PetscValidPointer(newdm,2);
124   ierr = DMCreate(PetscObjectComm((PetscObject) dm), newdm);CHKERRQ(ierr);
125   ierr = PetscFree((*newdm)->labels);CHKERRQ(ierr);
126   dm->labels->refct++;
127   (*newdm)->labels = dm->labels;
128   (*newdm)->depthLabel = dm->depthLabel;
129   (*newdm)->leveldown  = dm->leveldown;
130   (*newdm)->levelup    = dm->levelup;
131   ierr = DMGetDimension(dm, &dim);CHKERRQ(ierr);
132   ierr = DMSetDimension(*newdm, dim);CHKERRQ(ierr);
133   if (dm->ops->clone) {
134     ierr = (*dm->ops->clone)(dm, newdm);CHKERRQ(ierr);
135   }
136   (*newdm)->setupcalled = dm->setupcalled;
137   ierr = DMGetPointSF(dm, &sf);CHKERRQ(ierr);
138   ierr = DMSetPointSF(*newdm, sf);CHKERRQ(ierr);
139   ierr = DMGetApplicationContext(dm, &ctx);CHKERRQ(ierr);
140   ierr = DMSetApplicationContext(*newdm, ctx);CHKERRQ(ierr);
141   if (dm->coordinateDM) {
142     DM           ncdm;
143     PetscSection cs;
144     PetscInt     pEnd = -1, pEndMax = -1;
145 
146     ierr = DMGetSection(dm->coordinateDM, &cs);CHKERRQ(ierr);
147     if (cs) {ierr = PetscSectionGetChart(cs, NULL, &pEnd);CHKERRQ(ierr);}
148     ierr = MPI_Allreduce(&pEnd,&pEndMax,1,MPIU_INT,MPI_MAX,PetscObjectComm((PetscObject)dm));CHKERRQ(ierr);
149     if (pEndMax >= 0) {
150       ierr = DMClone(dm->coordinateDM, &ncdm);CHKERRQ(ierr);
151       ierr = DMSetSection(ncdm, cs);CHKERRQ(ierr);
152       ierr = DMSetCoordinateDM(*newdm, ncdm);CHKERRQ(ierr);
153       ierr = DMDestroy(&ncdm);CHKERRQ(ierr);
154     }
155   }
156   ierr = DMGetCoordinateDim(dm, &cdim);CHKERRQ(ierr);
157   ierr = DMSetCoordinateDim(*newdm, cdim);CHKERRQ(ierr);
158   ierr = DMGetCoordinatesLocal(dm, &coords);CHKERRQ(ierr);
159   if (coords) {
160     ierr = DMSetCoordinatesLocal(*newdm, coords);CHKERRQ(ierr);
161   } else {
162     ierr = DMGetCoordinates(dm, &coords);CHKERRQ(ierr);
163     if (coords) {ierr = DMSetCoordinates(*newdm, coords);CHKERRQ(ierr);}
164   }
165   {
166     PetscBool             isper;
167     const PetscReal      *maxCell, *L;
168     const DMBoundaryType *bd;
169     ierr = DMGetPeriodicity(dm, &isper, &maxCell, &L, &bd);CHKERRQ(ierr);
170     ierr = DMSetPeriodicity(*newdm, isper, maxCell,  L,  bd);CHKERRQ(ierr);
171   }
172   PetscFunctionReturn(0);
173 }
174 
175 /*@C
176        DMSetVecType - Sets the type of vector created with DMCreateLocalVector() and DMCreateGlobalVector()
177 
178    Logically Collective on DM
179 
180    Input Parameter:
181 +  da - initial distributed array
182 .  ctype - the vector type, currently either VECSTANDARD, VECCUDA, or VECVIENNACL
183 
184    Options Database:
185 .   -dm_vec_type ctype
186 
187    Level: intermediate
188 
189 .seealso: DMCreate(), DMDestroy(), DM, DMDAInterpolationType, VecType, DMGetVecType(), DMSetMatType(), DMGetMatType()
190 @*/
191 PetscErrorCode  DMSetVecType(DM da,VecType ctype)
192 {
193   PetscErrorCode ierr;
194 
195   PetscFunctionBegin;
196   PetscValidHeaderSpecific(da,DM_CLASSID,1);
197   ierr = PetscFree(da->vectype);CHKERRQ(ierr);
198   ierr = PetscStrallocpy(ctype,(char**)&da->vectype);CHKERRQ(ierr);
199   PetscFunctionReturn(0);
200 }
201 
202 /*@C
203        DMGetVecType - Gets the type of vector created with DMCreateLocalVector() and DMCreateGlobalVector()
204 
205    Logically Collective on DM
206 
207    Input Parameter:
208 .  da - initial distributed array
209 
210    Output Parameter:
211 .  ctype - the vector type
212 
213    Level: intermediate
214 
215 .seealso: DMCreate(), DMDestroy(), DM, DMDAInterpolationType, VecType, DMSetMatType(), DMGetMatType(), DMSetVecType()
216 @*/
217 PetscErrorCode  DMGetVecType(DM da,VecType *ctype)
218 {
219   PetscFunctionBegin;
220   PetscValidHeaderSpecific(da,DM_CLASSID,1);
221   *ctype = da->vectype;
222   PetscFunctionReturn(0);
223 }
224 
225 /*@
226   VecGetDM - Gets the DM defining the data layout of the vector
227 
228   Not collective
229 
230   Input Parameter:
231 . v - The Vec
232 
233   Output Parameter:
234 . dm - The DM
235 
236   Level: intermediate
237 
238 .seealso: VecSetDM(), DMGetLocalVector(), DMGetGlobalVector(), DMSetVecType()
239 @*/
240 PetscErrorCode VecGetDM(Vec v, DM *dm)
241 {
242   PetscErrorCode ierr;
243 
244   PetscFunctionBegin;
245   PetscValidHeaderSpecific(v,VEC_CLASSID,1);
246   PetscValidPointer(dm,2);
247   ierr = PetscObjectQuery((PetscObject) v, "__PETSc_dm", (PetscObject*) dm);CHKERRQ(ierr);
248   PetscFunctionReturn(0);
249 }
250 
251 /*@
252   VecSetDM - Sets the DM defining the data layout of the vector.
253 
254   Not collective
255 
256   Input Parameters:
257 + v - The Vec
258 - dm - The DM
259 
260   Note: This is NOT the same as DMCreateGlobalVector() since it does not change the view methods or perform other customization, but merely sets the DM member.
261 
262   Level: intermediate
263 
264 .seealso: VecGetDM(), DMGetLocalVector(), DMGetGlobalVector(), DMSetVecType()
265 @*/
266 PetscErrorCode VecSetDM(Vec v, DM dm)
267 {
268   PetscErrorCode ierr;
269 
270   PetscFunctionBegin;
271   PetscValidHeaderSpecific(v,VEC_CLASSID,1);
272   if (dm) PetscValidHeaderSpecific(dm,DM_CLASSID,2);
273   ierr = PetscObjectCompose((PetscObject) v, "__PETSc_dm", (PetscObject) dm);CHKERRQ(ierr);
274   PetscFunctionReturn(0);
275 }
276 
277 /*@C
278        DMSetISColoringType - Sets the type of coloring, global or local, that is created by the DM
279 
280    Logically Collective on DM
281 
282    Input Parameters:
283 +  dm - the DM context
284 -  ctype - the matrix type
285 
286    Options Database:
287 .   -dm_is_coloring_type - global or local
288 
289    Level: intermediate
290 
291 .seealso: DMDACreate1d(), DMDACreate2d(), DMDACreate3d(), DMCreateMatrix(), DMSetMatrixPreallocateOnly(), MatType, DMGetMatType(),
292           DMGetISColoringType()
293 @*/
294 PetscErrorCode  DMSetISColoringType(DM dm,ISColoringType ctype)
295 {
296   PetscFunctionBegin;
297   PetscValidHeaderSpecific(dm,DM_CLASSID,1);
298   dm->coloringtype = ctype;
299   PetscFunctionReturn(0);
300 }
301 
302 /*@C
303        DMGetISColoringType - Gets the type of coloring, global or local, that is created by the DM
304 
305    Logically Collective on DM
306 
307    Input Parameter:
308 .  dm - the DM context
309 
310    Output Parameter:
311 .  ctype - the matrix type
312 
313    Options Database:
314 .   -dm_is_coloring_type - global or local
315 
316    Level: intermediate
317 
318 .seealso: DMDACreate1d(), DMDACreate2d(), DMDACreate3d(), DMCreateMatrix(), DMSetMatrixPreallocateOnly(), MatType, DMGetMatType(),
319           DMGetISColoringType()
320 @*/
321 PetscErrorCode  DMGetISColoringType(DM dm,ISColoringType *ctype)
322 {
323   PetscFunctionBegin;
324   PetscValidHeaderSpecific(dm,DM_CLASSID,1);
325   *ctype = dm->coloringtype;
326   PetscFunctionReturn(0);
327 }
328 
329 /*@C
330        DMSetMatType - Sets the type of matrix created with DMCreateMatrix()
331 
332    Logically Collective on DM
333 
334    Input Parameters:
335 +  dm - the DM context
336 -  ctype - the matrix type
337 
338    Options Database:
339 .   -dm_mat_type ctype
340 
341    Level: intermediate
342 
343 .seealso: DMDACreate1d(), DMDACreate2d(), DMDACreate3d(), DMCreateMatrix(), DMSetMatrixPreallocateOnly(), MatType, DMGetMatType(), DMSetMatType(), DMGetMatType()
344 @*/
345 PetscErrorCode  DMSetMatType(DM dm,MatType ctype)
346 {
347   PetscErrorCode ierr;
348 
349   PetscFunctionBegin;
350   PetscValidHeaderSpecific(dm,DM_CLASSID,1);
351   ierr = PetscFree(dm->mattype);CHKERRQ(ierr);
352   ierr = PetscStrallocpy(ctype,(char**)&dm->mattype);CHKERRQ(ierr);
353   PetscFunctionReturn(0);
354 }
355 
356 /*@C
357        DMGetMatType - Gets the type of matrix created with DMCreateMatrix()
358 
359    Logically Collective on DM
360 
361    Input Parameter:
362 .  dm - the DM context
363 
364    Output Parameter:
365 .  ctype - the matrix type
366 
367    Options Database:
368 .   -dm_mat_type ctype
369 
370    Level: intermediate
371 
372 .seealso: DMDACreate1d(), DMDACreate2d(), DMDACreate3d(), DMCreateMatrix(), DMSetMatrixPreallocateOnly(), MatType, DMSetMatType(), DMSetMatType(), DMGetMatType()
373 @*/
374 PetscErrorCode  DMGetMatType(DM dm,MatType *ctype)
375 {
376   PetscFunctionBegin;
377   PetscValidHeaderSpecific(dm,DM_CLASSID,1);
378   *ctype = dm->mattype;
379   PetscFunctionReturn(0);
380 }
381 
382 /*@
383   MatGetDM - Gets the DM defining the data layout of the matrix
384 
385   Not collective
386 
387   Input Parameter:
388 . A - The Mat
389 
390   Output Parameter:
391 . dm - The DM
392 
393   Level: intermediate
394 
395   Developer Note: Since the Mat class doesn't know about the DM class the DM object is associated with
396                   the Mat through a PetscObjectCompose() operation
397 
398 .seealso: MatSetDM(), DMCreateMatrix(), DMSetMatType()
399 @*/
400 PetscErrorCode MatGetDM(Mat A, DM *dm)
401 {
402   PetscErrorCode ierr;
403 
404   PetscFunctionBegin;
405   PetscValidHeaderSpecific(A,MAT_CLASSID,1);
406   PetscValidPointer(dm,2);
407   ierr = PetscObjectQuery((PetscObject) A, "__PETSc_dm", (PetscObject*) dm);CHKERRQ(ierr);
408   PetscFunctionReturn(0);
409 }
410 
411 /*@
412   MatSetDM - Sets the DM defining the data layout of the matrix
413 
414   Not collective
415 
416   Input Parameters:
417 + A - The Mat
418 - dm - The DM
419 
420   Level: intermediate
421 
422   Developer Note: Since the Mat class doesn't know about the DM class the DM object is associated with
423                   the Mat through a PetscObjectCompose() operation
424 
425 
426 .seealso: MatGetDM(), DMCreateMatrix(), DMSetMatType()
427 @*/
428 PetscErrorCode MatSetDM(Mat A, DM dm)
429 {
430   PetscErrorCode ierr;
431 
432   PetscFunctionBegin;
433   PetscValidHeaderSpecific(A,MAT_CLASSID,1);
434   if (dm) PetscValidHeaderSpecific(dm,DM_CLASSID,2);
435   ierr = PetscObjectCompose((PetscObject) A, "__PETSc_dm", (PetscObject) dm);CHKERRQ(ierr);
436   PetscFunctionReturn(0);
437 }
438 
439 /*@C
440    DMSetOptionsPrefix - Sets the prefix used for searching for all
441    DM options in the database.
442 
443    Logically Collective on DM
444 
445    Input Parameter:
446 +  da - the DM context
447 -  prefix - the prefix to prepend to all option names
448 
449    Notes:
450    A hyphen (-) must NOT be given at the beginning of the prefix name.
451    The first character of all runtime options is AUTOMATICALLY the hyphen.
452 
453    Level: advanced
454 
455 .keywords: DM, set, options, prefix, database
456 
457 .seealso: DMSetFromOptions()
458 @*/
459 PetscErrorCode  DMSetOptionsPrefix(DM dm,const char prefix[])
460 {
461   PetscErrorCode ierr;
462 
463   PetscFunctionBegin;
464   PetscValidHeaderSpecific(dm,DM_CLASSID,1);
465   ierr = PetscObjectSetOptionsPrefix((PetscObject)dm,prefix);CHKERRQ(ierr);
466   if (dm->sf) {
467     ierr = PetscObjectSetOptionsPrefix((PetscObject)dm->sf,prefix);CHKERRQ(ierr);
468   }
469   if (dm->defaultSF) {
470     ierr = PetscObjectSetOptionsPrefix((PetscObject)dm->defaultSF,prefix);CHKERRQ(ierr);
471   }
472   PetscFunctionReturn(0);
473 }
474 
475 /*@C
476    DMAppendOptionsPrefix - Appends to the prefix used for searching for all
477    DM options in the database.
478 
479    Logically Collective on DM
480 
481    Input Parameters:
482 +  dm - the DM context
483 -  prefix - the prefix string to prepend to all DM option requests
484 
485    Notes:
486    A hyphen (-) must NOT be given at the beginning of the prefix name.
487    The first character of all runtime options is AUTOMATICALLY the hyphen.
488 
489    Level: advanced
490 
491 .keywords: DM, append, options, prefix, database
492 
493 .seealso: DMSetOptionsPrefix(), DMGetOptionsPrefix()
494 @*/
495 PetscErrorCode  DMAppendOptionsPrefix(DM dm,const char prefix[])
496 {
497   PetscErrorCode ierr;
498 
499   PetscFunctionBegin;
500   PetscValidHeaderSpecific(dm,DM_CLASSID,1);
501   ierr = PetscObjectAppendOptionsPrefix((PetscObject)dm,prefix);CHKERRQ(ierr);
502   PetscFunctionReturn(0);
503 }
504 
505 /*@C
506    DMGetOptionsPrefix - Gets the prefix used for searching for all
507    DM options in the database.
508 
509    Not Collective
510 
511    Input Parameters:
512 .  dm - the DM context
513 
514    Output Parameters:
515 .  prefix - pointer to the prefix string used is returned
516 
517    Notes:
518     On the fortran side, the user should pass in a string 'prefix' of
519    sufficient length to hold the prefix.
520 
521    Level: advanced
522 
523 .keywords: DM, set, options, prefix, database
524 
525 .seealso: DMSetOptionsPrefix(), DMAppendOptionsPrefix()
526 @*/
527 PetscErrorCode  DMGetOptionsPrefix(DM dm,const char *prefix[])
528 {
529   PetscErrorCode ierr;
530 
531   PetscFunctionBegin;
532   PetscValidHeaderSpecific(dm,DM_CLASSID,1);
533   ierr = PetscObjectGetOptionsPrefix((PetscObject)dm,prefix);CHKERRQ(ierr);
534   PetscFunctionReturn(0);
535 }
536 
537 static PetscErrorCode DMCountNonCyclicReferences(DM dm, PetscBool recurseCoarse, PetscBool recurseFine, PetscInt *ncrefct)
538 {
539   PetscInt i, refct = ((PetscObject) dm)->refct;
540   DMNamedVecLink nlink;
541   PetscErrorCode ierr;
542 
543   PetscFunctionBegin;
544   *ncrefct = 0;
545   /* count all the circular references of DM and its contained Vecs */
546   for (i=0; i<DM_MAX_WORK_VECTORS; i++) {
547     if (dm->localin[i])  refct--;
548     if (dm->globalin[i]) refct--;
549   }
550   for (nlink=dm->namedglobal; nlink; nlink=nlink->next) refct--;
551   for (nlink=dm->namedlocal; nlink; nlink=nlink->next) refct--;
552   if (dm->x) {
553     DM obj;
554     ierr = VecGetDM(dm->x, &obj);CHKERRQ(ierr);
555     if (obj == dm) refct--;
556   }
557   if (dm->coarseMesh && dm->coarseMesh->fineMesh == dm) {
558     refct--;
559     if (recurseCoarse) {
560       PetscInt coarseCount;
561 
562       ierr = DMCountNonCyclicReferences(dm->coarseMesh, PETSC_TRUE, PETSC_FALSE,&coarseCount);CHKERRQ(ierr);
563       refct += coarseCount;
564     }
565   }
566   if (dm->fineMesh && dm->fineMesh->coarseMesh == dm) {
567     refct--;
568     if (recurseFine) {
569       PetscInt fineCount;
570 
571       ierr = DMCountNonCyclicReferences(dm->fineMesh, PETSC_FALSE, PETSC_TRUE,&fineCount);CHKERRQ(ierr);
572       refct += fineCount;
573     }
574   }
575   *ncrefct = refct;
576   PetscFunctionReturn(0);
577 }
578 
579 PetscErrorCode DMDestroyLabelLinkList(DM dm)
580 {
581   PetscErrorCode ierr;
582 
583   PetscFunctionBegin;
584   if (!--(dm->labels->refct)) {
585     DMLabelLink next = dm->labels->next;
586 
587     /* destroy the labels */
588     while (next) {
589       DMLabelLink tmp = next->next;
590 
591       ierr = DMLabelDestroy(&next->label);CHKERRQ(ierr);
592       ierr = PetscFree(next);CHKERRQ(ierr);
593       next = tmp;
594     }
595     ierr = PetscFree(dm->labels);CHKERRQ(ierr);
596   }
597   PetscFunctionReturn(0);
598 }
599 
600 /*@
601     DMDestroy - Destroys a vector packer or DM.
602 
603     Collective on DM
604 
605     Input Parameter:
606 .   dm - the DM object to destroy
607 
608     Level: developer
609 
610 .seealso DMView(), DMCreateGlobalVector(), DMCreateInterpolation(), DMCreateColoring(), DMCreateMatrix()
611 
612 @*/
613 PetscErrorCode  DMDestroy(DM *dm)
614 {
615   PetscInt       i, cnt;
616   DMNamedVecLink nlink,nnext;
617   PetscErrorCode ierr;
618 
619   PetscFunctionBegin;
620   if (!*dm) PetscFunctionReturn(0);
621   PetscValidHeaderSpecific((*dm),DM_CLASSID,1);
622 
623   /* count all non-cyclic references in the doubly-linked list of coarse<->fine meshes */
624   ierr = DMCountNonCyclicReferences(*dm,PETSC_TRUE,PETSC_TRUE,&cnt);CHKERRQ(ierr);
625   --((PetscObject)(*dm))->refct;
626   if (--cnt > 0) {*dm = 0; PetscFunctionReturn(0);}
627   /*
628      Need this test because the dm references the vectors that
629      reference the dm, so destroying the dm calls destroy on the
630      vectors that cause another destroy on the dm
631   */
632   if (((PetscObject)(*dm))->refct < 0) PetscFunctionReturn(0);
633   ((PetscObject) (*dm))->refct = 0;
634   for (i=0; i<DM_MAX_WORK_VECTORS; i++) {
635     if ((*dm)->localout[i]) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONGSTATE,"Destroying a DM that has a local vector obtained with DMGetLocalVector()");
636     ierr = VecDestroy(&(*dm)->localin[i]);CHKERRQ(ierr);
637   }
638   nnext=(*dm)->namedglobal;
639   (*dm)->namedglobal = NULL;
640   for (nlink=nnext; nlink; nlink=nnext) { /* Destroy the named vectors */
641     nnext = nlink->next;
642     if (nlink->status != DMVEC_STATUS_IN) SETERRQ1(((PetscObject)*dm)->comm,PETSC_ERR_ARG_WRONGSTATE,"DM still has Vec named '%s' checked out",nlink->name);
643     ierr = PetscFree(nlink->name);CHKERRQ(ierr);
644     ierr = VecDestroy(&nlink->X);CHKERRQ(ierr);
645     ierr = PetscFree(nlink);CHKERRQ(ierr);
646   }
647   nnext=(*dm)->namedlocal;
648   (*dm)->namedlocal = NULL;
649   for (nlink=nnext; nlink; nlink=nnext) { /* Destroy the named local vectors */
650     nnext = nlink->next;
651     if (nlink->status != DMVEC_STATUS_IN) SETERRQ1(((PetscObject)*dm)->comm,PETSC_ERR_ARG_WRONGSTATE,"DM still has Vec named '%s' checked out",nlink->name);
652     ierr = PetscFree(nlink->name);CHKERRQ(ierr);
653     ierr = VecDestroy(&nlink->X);CHKERRQ(ierr);
654     ierr = PetscFree(nlink);CHKERRQ(ierr);
655   }
656 
657   /* Destroy the list of hooks */
658   {
659     DMCoarsenHookLink link,next;
660     for (link=(*dm)->coarsenhook; link; link=next) {
661       next = link->next;
662       ierr = PetscFree(link);CHKERRQ(ierr);
663     }
664     (*dm)->coarsenhook = NULL;
665   }
666   {
667     DMRefineHookLink link,next;
668     for (link=(*dm)->refinehook; link; link=next) {
669       next = link->next;
670       ierr = PetscFree(link);CHKERRQ(ierr);
671     }
672     (*dm)->refinehook = NULL;
673   }
674   {
675     DMSubDomainHookLink link,next;
676     for (link=(*dm)->subdomainhook; link; link=next) {
677       next = link->next;
678       ierr = PetscFree(link);CHKERRQ(ierr);
679     }
680     (*dm)->subdomainhook = NULL;
681   }
682   {
683     DMGlobalToLocalHookLink link,next;
684     for (link=(*dm)->gtolhook; link; link=next) {
685       next = link->next;
686       ierr = PetscFree(link);CHKERRQ(ierr);
687     }
688     (*dm)->gtolhook = NULL;
689   }
690   {
691     DMLocalToGlobalHookLink link,next;
692     for (link=(*dm)->ltoghook; link; link=next) {
693       next = link->next;
694       ierr = PetscFree(link);CHKERRQ(ierr);
695     }
696     (*dm)->ltoghook = NULL;
697   }
698   /* Destroy the work arrays */
699   {
700     DMWorkLink link,next;
701     if ((*dm)->workout) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONGSTATE,"Work array still checked out");
702     for (link=(*dm)->workin; link; link=next) {
703       next = link->next;
704       ierr = PetscFree(link->mem);CHKERRQ(ierr);
705       ierr = PetscFree(link);CHKERRQ(ierr);
706     }
707     (*dm)->workin = NULL;
708   }
709   if (!--((*dm)->labels->refct)) {
710     DMLabelLink next = (*dm)->labels->next;
711 
712     /* destroy the labels */
713     while (next) {
714       DMLabelLink tmp = next->next;
715 
716       ierr = DMLabelDestroy(&next->label);CHKERRQ(ierr);
717       ierr = PetscFree(next);CHKERRQ(ierr);
718       next = tmp;
719     }
720     ierr = PetscFree((*dm)->labels);CHKERRQ(ierr);
721   }
722   ierr = DMClearFields(*dm);CHKERRQ(ierr);
723   {
724     DMBoundary next = (*dm)->boundary;
725     while (next) {
726       DMBoundary b = next;
727 
728       next = b->next;
729       ierr = PetscFree(b);CHKERRQ(ierr);
730     }
731   }
732 
733   ierr = PetscObjectDestroy(&(*dm)->dmksp);CHKERRQ(ierr);
734   ierr = PetscObjectDestroy(&(*dm)->dmsnes);CHKERRQ(ierr);
735   ierr = PetscObjectDestroy(&(*dm)->dmts);CHKERRQ(ierr);
736 
737   if ((*dm)->ctx && (*dm)->ctxdestroy) {
738     ierr = (*(*dm)->ctxdestroy)(&(*dm)->ctx);CHKERRQ(ierr);
739   }
740   ierr = VecDestroy(&(*dm)->x);CHKERRQ(ierr);
741   ierr = MatFDColoringDestroy(&(*dm)->fd);CHKERRQ(ierr);
742   ierr = DMClearGlobalVectors(*dm);CHKERRQ(ierr);
743   ierr = ISLocalToGlobalMappingDestroy(&(*dm)->ltogmap);CHKERRQ(ierr);
744   ierr = PetscFree((*dm)->vectype);CHKERRQ(ierr);
745   ierr = PetscFree((*dm)->mattype);CHKERRQ(ierr);
746 
747   ierr = PetscSectionDestroy(&(*dm)->defaultSection);CHKERRQ(ierr);
748   ierr = PetscSectionDestroy(&(*dm)->defaultGlobalSection);CHKERRQ(ierr);
749   ierr = PetscLayoutDestroy(&(*dm)->map);CHKERRQ(ierr);
750   ierr = PetscSectionDestroy(&(*dm)->defaultConstraintSection);CHKERRQ(ierr);
751   ierr = MatDestroy(&(*dm)->defaultConstraintMat);CHKERRQ(ierr);
752   ierr = PetscSFDestroy(&(*dm)->sf);CHKERRQ(ierr);
753   ierr = PetscSFDestroy(&(*dm)->defaultSF);CHKERRQ(ierr);
754   if ((*dm)->useNatural) {
755     if ((*dm)->sfNatural) {
756       ierr = PetscSFDestroy(&(*dm)->sfNatural);CHKERRQ(ierr);
757     }
758     ierr = PetscObjectDereference((PetscObject) (*dm)->sfMigration);CHKERRQ(ierr);
759   }
760   if ((*dm)->coarseMesh && (*dm)->coarseMesh->fineMesh == *dm) {
761     ierr = DMSetFineDM((*dm)->coarseMesh,NULL);CHKERRQ(ierr);
762   }
763   ierr = DMDestroy(&(*dm)->coarseMesh);CHKERRQ(ierr);
764   if ((*dm)->fineMesh && (*dm)->fineMesh->coarseMesh == *dm) {
765     ierr = DMSetCoarseDM((*dm)->fineMesh,NULL);CHKERRQ(ierr);
766   }
767   ierr = DMDestroy(&(*dm)->fineMesh);CHKERRQ(ierr);
768   ierr = DMFieldDestroy(&(*dm)->coordinateField);CHKERRQ(ierr);
769   ierr = DMDestroy(&(*dm)->coordinateDM);CHKERRQ(ierr);
770   ierr = VecDestroy(&(*dm)->coordinates);CHKERRQ(ierr);
771   ierr = VecDestroy(&(*dm)->coordinatesLocal);CHKERRQ(ierr);
772   ierr = PetscFree3((*dm)->L,(*dm)->maxCell,(*dm)->bdtype);CHKERRQ(ierr);
773 
774   ierr = DMClearDS(*dm);CHKERRQ(ierr);
775   ierr = DMDestroy(&(*dm)->dmBC);CHKERRQ(ierr);
776   /* if memory was published with SAWs then destroy it */
777   ierr = PetscObjectSAWsViewOff((PetscObject)*dm);CHKERRQ(ierr);
778 
779   if ((*dm)->ops->destroy) {
780     ierr = (*(*dm)->ops->destroy)(*dm);CHKERRQ(ierr);
781   }
782   /* We do not destroy (*dm)->data here so that we can reference count backend objects */
783   ierr = PetscHeaderDestroy(dm);CHKERRQ(ierr);
784   PetscFunctionReturn(0);
785 }
786 
787 /*@
788     DMSetUp - sets up the data structures inside a DM object
789 
790     Collective on DM
791 
792     Input Parameter:
793 .   dm - the DM object to setup
794 
795     Level: developer
796 
797 .seealso DMView(), DMCreateGlobalVector(), DMCreateInterpolation(), DMCreateColoring(), DMCreateMatrix()
798 
799 @*/
800 PetscErrorCode  DMSetUp(DM dm)
801 {
802   PetscErrorCode ierr;
803 
804   PetscFunctionBegin;
805   PetscValidHeaderSpecific(dm,DM_CLASSID,1);
806   if (dm->setupcalled) PetscFunctionReturn(0);
807   if (dm->ops->setup) {
808     ierr = (*dm->ops->setup)(dm);CHKERRQ(ierr);
809   }
810   dm->setupcalled = PETSC_TRUE;
811   PetscFunctionReturn(0);
812 }
813 
814 /*@
815     DMSetFromOptions - sets parameters in a DM from the options database
816 
817     Collective on DM
818 
819     Input Parameter:
820 .   dm - the DM object to set options for
821 
822     Options Database:
823 +   -dm_preallocate_only - Only preallocate the matrix for DMCreateMatrix(), but do not fill it with zeros
824 .   -dm_vec_type <type>  - type of vector to create inside DM
825 .   -dm_mat_type <type>  - type of matrix to create inside DM
826 -   -dm_is_coloring_type - <global or local>
827 
828     Level: developer
829 
830 .seealso DMView(), DMCreateGlobalVector(), DMCreateInterpolation(), DMCreateColoring(), DMCreateMatrix()
831 
832 @*/
833 PetscErrorCode DMSetFromOptions(DM dm)
834 {
835   char           typeName[256];
836   PetscBool      flg;
837   PetscErrorCode ierr;
838 
839   PetscFunctionBegin;
840   PetscValidHeaderSpecific(dm,DM_CLASSID,1);
841   dm->setfromoptionscalled = PETSC_TRUE;
842   if (dm->sf) {ierr = PetscSFSetFromOptions(dm->sf);CHKERRQ(ierr);}
843   if (dm->defaultSF) {ierr = PetscSFSetFromOptions(dm->defaultSF);CHKERRQ(ierr);}
844   ierr = PetscObjectOptionsBegin((PetscObject)dm);CHKERRQ(ierr);
845   ierr = PetscOptionsBool("-dm_preallocate_only","only preallocate matrix, but do not set column indices","DMSetMatrixPreallocateOnly",dm->prealloc_only,&dm->prealloc_only,NULL);CHKERRQ(ierr);
846   ierr = PetscOptionsFList("-dm_vec_type","Vector type used for created vectors","DMSetVecType",VecList,dm->vectype,typeName,256,&flg);CHKERRQ(ierr);
847   if (flg) {
848     ierr = DMSetVecType(dm,typeName);CHKERRQ(ierr);
849   }
850   ierr = PetscOptionsFList("-dm_mat_type","Matrix type used for created matrices","DMSetMatType",MatList,dm->mattype ? dm->mattype : typeName,typeName,sizeof(typeName),&flg);CHKERRQ(ierr);
851   if (flg) {
852     ierr = DMSetMatType(dm,typeName);CHKERRQ(ierr);
853   }
854   ierr = PetscOptionsEnum("-dm_is_coloring_type","Global or local coloring of Jacobian","DMSetISColoringType",ISColoringTypes,(PetscEnum)dm->coloringtype,(PetscEnum*)&dm->coloringtype,NULL);CHKERRQ(ierr);
855   if (dm->ops->setfromoptions) {
856     ierr = (*dm->ops->setfromoptions)(PetscOptionsObject,dm);CHKERRQ(ierr);
857   }
858   /* process any options handlers added with PetscObjectAddOptionsHandler() */
859   ierr = PetscObjectProcessOptionsHandlers(PetscOptionsObject,(PetscObject) dm);CHKERRQ(ierr);
860   ierr = PetscOptionsEnd();CHKERRQ(ierr);
861   PetscFunctionReturn(0);
862 }
863 
864 /*@C
865     DMView - Views a DM
866 
867     Collective on DM
868 
869     Input Parameter:
870 +   dm - the DM object to view
871 -   v - the viewer
872 
873     Level: beginner
874 
875 .seealso DMDestroy(), DMCreateGlobalVector(), DMCreateInterpolation(), DMCreateColoring(), DMCreateMatrix()
876 
877 @*/
878 PetscErrorCode  DMView(DM dm,PetscViewer v)
879 {
880   PetscErrorCode    ierr;
881   PetscBool         isbinary;
882   PetscMPIInt       size;
883   PetscViewerFormat format;
884 
885   PetscFunctionBegin;
886   PetscValidHeaderSpecific(dm,DM_CLASSID,1);
887   if (!v) {
888     ierr = PetscViewerASCIIGetStdout(PetscObjectComm((PetscObject)dm),&v);CHKERRQ(ierr);
889   }
890   ierr = PetscViewerCheckWritable(v);CHKERRQ(ierr);
891   ierr = PetscViewerGetFormat(v,&format);CHKERRQ(ierr);
892   ierr = MPI_Comm_size(PetscObjectComm((PetscObject)dm),&size);CHKERRQ(ierr);
893   if (size == 1 && format == PETSC_VIEWER_LOAD_BALANCE) PetscFunctionReturn(0);
894   ierr = PetscObjectPrintClassNamePrefixType((PetscObject)dm,v);CHKERRQ(ierr);
895   ierr = PetscObjectTypeCompare((PetscObject)v,PETSCVIEWERBINARY,&isbinary);CHKERRQ(ierr);
896   if (isbinary) {
897     PetscInt classid = DM_FILE_CLASSID;
898     char     type[256];
899 
900     ierr = PetscViewerBinaryWrite(v,&classid,1,PETSC_INT,PETSC_FALSE);CHKERRQ(ierr);
901     ierr = PetscStrncpy(type,((PetscObject)dm)->type_name,256);CHKERRQ(ierr);
902     ierr = PetscViewerBinaryWrite(v,type,256,PETSC_CHAR,PETSC_FALSE);CHKERRQ(ierr);
903   }
904   if (dm->ops->view) {
905     ierr = (*dm->ops->view)(dm,v);CHKERRQ(ierr);
906   }
907   PetscFunctionReturn(0);
908 }
909 
910 /*@
911     DMCreateGlobalVector - Creates a global vector from a DM object
912 
913     Collective on DM
914 
915     Input Parameter:
916 .   dm - the DM object
917 
918     Output Parameter:
919 .   vec - the global vector
920 
921     Level: beginner
922 
923 .seealso DMDestroy(), DMView(), DMCreateInterpolation(), DMCreateColoring(), DMCreateMatrix()
924 
925 @*/
926 PetscErrorCode  DMCreateGlobalVector(DM dm,Vec *vec)
927 {
928   PetscErrorCode ierr;
929 
930   PetscFunctionBegin;
931   PetscValidHeaderSpecific(dm,DM_CLASSID,1);
932   ierr = (*dm->ops->createglobalvector)(dm,vec);CHKERRQ(ierr);
933   PetscFunctionReturn(0);
934 }
935 
936 /*@
937     DMCreateLocalVector - Creates a local vector from a DM object
938 
939     Not Collective
940 
941     Input Parameter:
942 .   dm - the DM object
943 
944     Output Parameter:
945 .   vec - the local vector
946 
947     Level: beginner
948 
949 .seealso DMDestroy(), DMView(), DMCreateInterpolation(), DMCreateColoring(), DMCreateMatrix()
950 
951 @*/
952 PetscErrorCode  DMCreateLocalVector(DM dm,Vec *vec)
953 {
954   PetscErrorCode ierr;
955 
956   PetscFunctionBegin;
957   PetscValidHeaderSpecific(dm,DM_CLASSID,1);
958   ierr = (*dm->ops->createlocalvector)(dm,vec);CHKERRQ(ierr);
959   PetscFunctionReturn(0);
960 }
961 
962 /*@
963    DMGetLocalToGlobalMapping - Accesses the local-to-global mapping in a DM.
964 
965    Collective on DM
966 
967    Input Parameter:
968 .  dm - the DM that provides the mapping
969 
970    Output Parameter:
971 .  ltog - the mapping
972 
973    Level: intermediate
974 
975    Notes:
976    This mapping can then be used by VecSetLocalToGlobalMapping() or
977    MatSetLocalToGlobalMapping().
978 
979 .seealso: DMCreateLocalVector()
980 @*/
981 PetscErrorCode DMGetLocalToGlobalMapping(DM dm,ISLocalToGlobalMapping *ltog)
982 {
983   PetscInt       bs = -1, bsLocal[2], bsMinMax[2];
984   PetscErrorCode ierr;
985 
986   PetscFunctionBegin;
987   PetscValidHeaderSpecific(dm,DM_CLASSID,1);
988   PetscValidPointer(ltog,2);
989   if (!dm->ltogmap) {
990     PetscSection section, sectionGlobal;
991 
992     ierr = DMGetSection(dm, &section);CHKERRQ(ierr);
993     if (section) {
994       const PetscInt *cdofs;
995       PetscInt       *ltog;
996       PetscInt        pStart, pEnd, n, p, k, l;
997 
998       ierr = DMGetGlobalSection(dm, &sectionGlobal);CHKERRQ(ierr);
999       ierr = PetscSectionGetChart(section, &pStart, &pEnd);CHKERRQ(ierr);
1000       ierr = PetscSectionGetStorageSize(section, &n);CHKERRQ(ierr);
1001       ierr = PetscMalloc1(n, &ltog);CHKERRQ(ierr); /* We want the local+overlap size */
1002       for (p = pStart, l = 0; p < pEnd; ++p) {
1003         PetscInt bdof, cdof, dof, off, c, cind = 0;
1004 
1005         /* Should probably use constrained dofs */
1006         ierr = PetscSectionGetDof(section, p, &dof);CHKERRQ(ierr);
1007         ierr = PetscSectionGetConstraintDof(section, p, &cdof);CHKERRQ(ierr);
1008         ierr = PetscSectionGetConstraintIndices(section, p, &cdofs);CHKERRQ(ierr);
1009         ierr = PetscSectionGetOffset(sectionGlobal, p, &off);CHKERRQ(ierr);
1010         /* If you have dofs, and constraints, and they are unequal, we set the blocksize to 1 */
1011         bdof = cdof && (dof-cdof) ? 1 : dof;
1012         if (dof) {
1013           if (bs < 0)          {bs = bdof;}
1014           else if (bs != bdof) {bs = 1;}
1015         }
1016         for (c = 0; c < dof; ++c, ++l) {
1017           if ((cind < cdof) && (c == cdofs[cind])) ltog[l] = off < 0 ? off-c : off+c;
1018           else                                     ltog[l] = (off < 0 ? -(off+1) : off) + c;
1019         }
1020       }
1021       /* Must have same blocksize on all procs (some might have no points) */
1022       bsLocal[0] = bs < 0 ? PETSC_MAX_INT : bs; bsLocal[1] = bs;
1023       ierr = PetscGlobalMinMaxInt(PetscObjectComm((PetscObject) dm), bsLocal, bsMinMax);CHKERRQ(ierr);
1024       if (bsMinMax[0] != bsMinMax[1]) {bs = 1;}
1025       else                            {bs = bsMinMax[0];}
1026       bs = bs < 0 ? 1 : bs;
1027       /* Must reduce indices by blocksize */
1028       if (bs > 1) {
1029         for (l = 0, k = 0; l < n; l += bs, ++k) ltog[k] = ltog[l]/bs;
1030         n /= bs;
1031       }
1032       ierr = ISLocalToGlobalMappingCreate(PetscObjectComm((PetscObject)dm), bs, n, ltog, PETSC_OWN_POINTER, &dm->ltogmap);CHKERRQ(ierr);
1033       ierr = PetscLogObjectParent((PetscObject)dm, (PetscObject)dm->ltogmap);CHKERRQ(ierr);
1034     } else {
1035       if (!dm->ops->getlocaltoglobalmapping) SETERRQ(PetscObjectComm((PetscObject)dm),PETSC_ERR_SUP,"DM can not create LocalToGlobalMapping");
1036       ierr = (*dm->ops->getlocaltoglobalmapping)(dm);CHKERRQ(ierr);
1037     }
1038   }
1039   *ltog = dm->ltogmap;
1040   PetscFunctionReturn(0);
1041 }
1042 
1043 /*@
1044    DMGetBlockSize - Gets the inherent block size associated with a DM
1045 
1046    Not Collective
1047 
1048    Input Parameter:
1049 .  dm - the DM with block structure
1050 
1051    Output Parameter:
1052 .  bs - the block size, 1 implies no exploitable block structure
1053 
1054    Level: intermediate
1055 
1056 .seealso: ISCreateBlock(), VecSetBlockSize(), MatSetBlockSize(), DMGetLocalToGlobalMapping()
1057 @*/
1058 PetscErrorCode  DMGetBlockSize(DM dm,PetscInt *bs)
1059 {
1060   PetscFunctionBegin;
1061   PetscValidHeaderSpecific(dm,DM_CLASSID,1);
1062   PetscValidPointer(bs,2);
1063   if (dm->bs < 1) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONGSTATE,"DM does not have enough information to provide a block size yet");
1064   *bs = dm->bs;
1065   PetscFunctionReturn(0);
1066 }
1067 
1068 /*@
1069     DMCreateInterpolation - Gets interpolation matrix between two DM objects
1070 
1071     Collective on DM
1072 
1073     Input Parameter:
1074 +   dm1 - the DM object
1075 -   dm2 - the second, finer DM object
1076 
1077     Output Parameter:
1078 +  mat - the interpolation
1079 -  vec - the scaling (optional)
1080 
1081     Level: developer
1082 
1083     Notes:
1084     For DMDA objects this only works for "uniform refinement", that is the refined mesh was obtained DMRefine() or the coarse mesh was obtained by
1085         DMCoarsen(). The coordinates set into the DMDA are completely ignored in computing the interpolation.
1086 
1087         For DMDA objects you can use this interpolation (more precisely the interpolation from the DMGetCoordinateDM()) to interpolate the mesh coordinate vectors
1088         EXCEPT in the periodic case where it does not make sense since the coordinate vectors are not periodic.
1089 
1090 
1091 .seealso DMDestroy(), DMView(), DMCreateGlobalVector(), DMCreateColoring(), DMCreateMatrix(), DMRefine(), DMCoarsen(), DMCreateRestriction()
1092 
1093 @*/
1094 PetscErrorCode  DMCreateInterpolation(DM dm1,DM dm2,Mat *mat,Vec *vec)
1095 {
1096   PetscErrorCode ierr;
1097 
1098   PetscFunctionBegin;
1099   PetscValidHeaderSpecific(dm1,DM_CLASSID,1);
1100   PetscValidHeaderSpecific(dm2,DM_CLASSID,2);
1101   PetscValidPointer(mat,3);
1102   if (!dm1->ops->createinterpolation) SETERRQ1(PetscObjectComm((PetscObject)dm1),PETSC_ERR_SUP,"DMCreateInterpolation not implemented for type %s",((PetscObject)dm1)->type_name);
1103   ierr = PetscLogEventBegin(DM_CreateInterpolation,dm1,dm2,0,0);CHKERRQ(ierr);
1104   ierr = (*dm1->ops->createinterpolation)(dm1,dm2,mat,vec);CHKERRQ(ierr);
1105   ierr = PetscLogEventEnd(DM_CreateInterpolation,dm1,dm2,0,0);CHKERRQ(ierr);
1106   PetscFunctionReturn(0);
1107 }
1108 
1109 /*@
1110     DMCreateRestriction - Gets restriction matrix between two DM objects
1111 
1112     Collective on DM
1113 
1114     Input Parameter:
1115 +   dm1 - the DM object
1116 -   dm2 - the second, finer DM object
1117 
1118     Output Parameter:
1119 .  mat - the restriction
1120 
1121 
1122     Level: developer
1123 
1124     Notes:
1125     For DMDA objects this only works for "uniform refinement", that is the refined mesh was obtained DMRefine() or the coarse mesh was obtained by
1126         DMCoarsen(). The coordinates set into the DMDA are completely ignored in computing the interpolation.
1127 
1128 
1129 .seealso DMDestroy(), DMView(), DMCreateGlobalVector(), DMCreateColoring(), DMCreateMatrix(), DMRefine(), DMCoarsen(), DMCreateInterpolation()
1130 
1131 @*/
1132 PetscErrorCode  DMCreateRestriction(DM dm1,DM dm2,Mat *mat)
1133 {
1134   PetscErrorCode ierr;
1135 
1136   PetscFunctionBegin;
1137   PetscValidHeaderSpecific(dm1,DM_CLASSID,1);
1138   PetscValidHeaderSpecific(dm2,DM_CLASSID,2);
1139   ierr = PetscLogEventBegin(DM_CreateRestriction,dm1,dm2,0,0);CHKERRQ(ierr);
1140   if (!dm1->ops->createrestriction) SETERRQ(PetscObjectComm((PetscObject)dm1),PETSC_ERR_SUP,"DMCreateRestriction not implemented for this type");
1141   ierr = (*dm1->ops->createrestriction)(dm1,dm2,mat);CHKERRQ(ierr);
1142   ierr = PetscLogEventEnd(DM_CreateRestriction,dm1,dm2,0,0);CHKERRQ(ierr);
1143   PetscFunctionReturn(0);
1144 }
1145 
1146 /*@
1147     DMCreateInjection - Gets injection matrix between two DM objects
1148 
1149     Collective on DM
1150 
1151     Input Parameter:
1152 +   dm1 - the DM object
1153 -   dm2 - the second, finer DM object
1154 
1155     Output Parameter:
1156 .   mat - the injection
1157 
1158     Level: developer
1159 
1160    Notes:
1161     For DMDA objects this only works for "uniform refinement", that is the refined mesh was obtained DMRefine() or the coarse mesh was obtained by
1162         DMCoarsen(). The coordinates set into the DMDA are completely ignored in computing the injection.
1163 
1164 .seealso DMDestroy(), DMView(), DMCreateGlobalVector(), DMCreateColoring(), DMCreateMatrix(), DMCreateInterpolation()
1165 
1166 @*/
1167 PetscErrorCode  DMCreateInjection(DM dm1,DM dm2,Mat *mat)
1168 {
1169   PetscErrorCode ierr;
1170 
1171   PetscFunctionBegin;
1172   PetscValidHeaderSpecific(dm1,DM_CLASSID,1);
1173   PetscValidHeaderSpecific(dm2,DM_CLASSID,2);
1174   if (!dm1->ops->getinjection) SETERRQ(PetscObjectComm((PetscObject)dm1),PETSC_ERR_SUP,"DMCreateInjection not implemented for this type");
1175   ierr = (*dm1->ops->getinjection)(dm1,dm2,mat);CHKERRQ(ierr);
1176   PetscFunctionReturn(0);
1177 }
1178 
1179 /*@
1180   DMCreateMassMatrix - Gets mass matrix between two DM objects, M_ij = \int \phi_i \psi_j
1181 
1182   Collective on DM
1183 
1184   Input Parameter:
1185 + dm1 - the DM object
1186 - dm2 - the second, finer DM object
1187 
1188   Output Parameter:
1189 . mat - the interpolation
1190 
1191   Level: developer
1192 
1193 .seealso DMCreateMatrix(), DMRefine(), DMCoarsen(), DMCreateRestriction(), DMCreateInterpolation(), DMCreateInjection()
1194 @*/
1195 PetscErrorCode DMCreateMassMatrix(DM dm1, DM dm2, Mat *mat)
1196 {
1197   PetscErrorCode ierr;
1198 
1199   PetscFunctionBegin;
1200   PetscValidHeaderSpecific(dm1, DM_CLASSID, 1);
1201   PetscValidHeaderSpecific(dm2, DM_CLASSID, 2);
1202   ierr = (*dm1->ops->createmassmatrix)(dm1, dm2, mat);CHKERRQ(ierr);
1203   PetscFunctionReturn(0);
1204 }
1205 
1206 /*@
1207     DMCreateColoring - Gets coloring for a DM
1208 
1209     Collective on DM
1210 
1211     Input Parameter:
1212 +   dm - the DM object
1213 -   ctype - IS_COLORING_LOCAL or IS_COLORING_GLOBAL
1214 
1215     Output Parameter:
1216 .   coloring - the coloring
1217 
1218     Level: developer
1219 
1220 .seealso DMDestroy(), DMView(), DMCreateGlobalVector(), DMCreateInterpolation(), DMCreateMatrix(), DMSetMatType()
1221 
1222 @*/
1223 PetscErrorCode  DMCreateColoring(DM dm,ISColoringType ctype,ISColoring *coloring)
1224 {
1225   PetscErrorCode ierr;
1226 
1227   PetscFunctionBegin;
1228   PetscValidHeaderSpecific(dm,DM_CLASSID,1);
1229   if (!dm->ops->getcoloring) SETERRQ(PetscObjectComm((PetscObject)dm),PETSC_ERR_SUP,"No coloring for this type of DM yet");
1230   ierr = (*dm->ops->getcoloring)(dm,ctype,coloring);CHKERRQ(ierr);
1231   PetscFunctionReturn(0);
1232 }
1233 
1234 /*@
1235     DMCreateMatrix - Gets empty Jacobian for a DM
1236 
1237     Collective on DM
1238 
1239     Input Parameter:
1240 .   dm - the DM object
1241 
1242     Output Parameter:
1243 .   mat - the empty Jacobian
1244 
1245     Level: beginner
1246 
1247     Notes:
1248     This properly preallocates the number of nonzeros in the sparse matrix so you
1249        do not need to do it yourself.
1250 
1251        By default it also sets the nonzero structure and puts in the zero entries. To prevent setting
1252        the nonzero pattern call DMSetMatrixPreallocateOnly()
1253 
1254        For structured grid problems, when you call MatView() on this matrix it is displayed using the global natural ordering, NOT in the ordering used
1255        internally by PETSc.
1256 
1257        For structured grid problems, in general it is easiest to use MatSetValuesStencil() or MatSetValuesLocal() to put values into the matrix because MatSetValues() requires
1258        the indices for the global numbering for DMDAs which is complicated.
1259 
1260 .seealso DMDestroy(), DMView(), DMCreateGlobalVector(), DMCreateInterpolation(), DMSetMatType()
1261 
1262 @*/
1263 PetscErrorCode  DMCreateMatrix(DM dm,Mat *mat)
1264 {
1265   PetscErrorCode ierr;
1266 
1267   PetscFunctionBegin;
1268   PetscValidHeaderSpecific(dm,DM_CLASSID,1);
1269   ierr = MatInitializePackage();CHKERRQ(ierr);
1270   PetscValidHeaderSpecific(dm,DM_CLASSID,1);
1271   PetscValidPointer(mat,3);
1272   ierr = (*dm->ops->creatematrix)(dm,mat);CHKERRQ(ierr);
1273   /* Handle nullspace and near nullspace */
1274   if (dm->Nf) {
1275     MatNullSpace nullSpace;
1276     PetscInt     Nf;
1277 
1278     ierr = DMGetNumFields(dm, &Nf);CHKERRQ(ierr);
1279     if (Nf == 1) {
1280       if (dm->nullspaceConstructors[0]) {
1281         ierr = (*dm->nullspaceConstructors[0])(dm, 0, &nullSpace);CHKERRQ(ierr);
1282         ierr = MatSetNullSpace(*mat, nullSpace);CHKERRQ(ierr);
1283         ierr = MatNullSpaceDestroy(&nullSpace);CHKERRQ(ierr);
1284       }
1285       if (dm->nearnullspaceConstructors[0]) {
1286         ierr = (*dm->nearnullspaceConstructors[0])(dm, 0, &nullSpace);CHKERRQ(ierr);
1287         ierr = MatSetNearNullSpace(*mat, nullSpace);CHKERRQ(ierr);
1288         ierr = MatNullSpaceDestroy(&nullSpace);CHKERRQ(ierr);
1289       }
1290     }
1291   }
1292   PetscFunctionReturn(0);
1293 }
1294 
1295 /*@
1296   DMSetMatrixPreallocateOnly - When DMCreateMatrix() is called the matrix will be properly
1297     preallocated but the nonzero structure and zero values will not be set.
1298 
1299   Logically Collective on DM
1300 
1301   Input Parameter:
1302 + dm - the DM
1303 - only - PETSC_TRUE if only want preallocation
1304 
1305   Level: developer
1306 .seealso DMCreateMatrix(), DMSetMatrixStructureOnly()
1307 @*/
1308 PetscErrorCode DMSetMatrixPreallocateOnly(DM dm, PetscBool only)
1309 {
1310   PetscFunctionBegin;
1311   PetscValidHeaderSpecific(dm,DM_CLASSID,1);
1312   dm->prealloc_only = only;
1313   PetscFunctionReturn(0);
1314 }
1315 
1316 /*@
1317   DMSetMatrixStructureOnly - When DMCreateMatrix() is called, the matrix structure will be created
1318     but the array for values will not be allocated.
1319 
1320   Logically Collective on DM
1321 
1322   Input Parameter:
1323 + dm - the DM
1324 - only - PETSC_TRUE if only want matrix stucture
1325 
1326   Level: developer
1327 .seealso DMCreateMatrix(), DMSetMatrixPreallocateOnly()
1328 @*/
1329 PetscErrorCode DMSetMatrixStructureOnly(DM dm, PetscBool only)
1330 {
1331   PetscFunctionBegin;
1332   PetscValidHeaderSpecific(dm,DM_CLASSID,1);
1333   dm->structure_only = only;
1334   PetscFunctionReturn(0);
1335 }
1336 
1337 /*@C
1338   DMGetWorkArray - Gets a work array guaranteed to be at least the input size, restore with DMRestoreWorkArray()
1339 
1340   Not Collective
1341 
1342   Input Parameters:
1343 + dm - the DM object
1344 . count - The minium size
1345 - dtype - MPI data type, often MPIU_REAL, MPIU_SCALAR, MPIU_INT)
1346 
1347   Output Parameter:
1348 . array - the work array
1349 
1350   Level: developer
1351 
1352 .seealso DMDestroy(), DMCreate()
1353 @*/
1354 PetscErrorCode DMGetWorkArray(DM dm,PetscInt count,MPI_Datatype dtype,void *mem)
1355 {
1356   PetscErrorCode ierr;
1357   DMWorkLink     link;
1358   PetscMPIInt    dsize;
1359 
1360   PetscFunctionBegin;
1361   PetscValidHeaderSpecific(dm,DM_CLASSID,1);
1362   PetscValidPointer(mem,4);
1363   if (dm->workin) {
1364     link       = dm->workin;
1365     dm->workin = dm->workin->next;
1366   } else {
1367     ierr = PetscNewLog(dm,&link);CHKERRQ(ierr);
1368   }
1369   ierr = MPI_Type_size(dtype,&dsize);CHKERRQ(ierr);
1370   if (((size_t)dsize*count) > link->bytes) {
1371     ierr        = PetscFree(link->mem);CHKERRQ(ierr);
1372     ierr        = PetscMalloc(dsize*count,&link->mem);CHKERRQ(ierr);
1373     link->bytes = dsize*count;
1374   }
1375   link->next   = dm->workout;
1376   dm->workout  = link;
1377   *(void**)mem = link->mem;
1378   PetscFunctionReturn(0);
1379 }
1380 
1381 /*@C
1382   DMRestoreWorkArray - Restores a work array guaranteed to be at least the input size, restore with DMRestoreWorkArray()
1383 
1384   Not Collective
1385 
1386   Input Parameters:
1387 + dm - the DM object
1388 . count - The minium size
1389 - dtype - MPI data type, often MPIU_REAL, MPIU_SCALAR, MPIU_INT
1390 
1391   Output Parameter:
1392 . array - the work array
1393 
1394   Level: developer
1395 
1396   Developer Notes:
1397     count and dtype are ignored, they are only needed for DMGetWorkArray()
1398 .seealso DMDestroy(), DMCreate()
1399 @*/
1400 PetscErrorCode DMRestoreWorkArray(DM dm,PetscInt count,MPI_Datatype dtype,void *mem)
1401 {
1402   DMWorkLink *p,link;
1403 
1404   PetscFunctionBegin;
1405   PetscValidHeaderSpecific(dm,DM_CLASSID,1);
1406   PetscValidPointer(mem,4);
1407   for (p=&dm->workout; (link=*p); p=&link->next) {
1408     if (link->mem == *(void**)mem) {
1409       *p           = link->next;
1410       link->next   = dm->workin;
1411       dm->workin   = link;
1412       *(void**)mem = NULL;
1413       PetscFunctionReturn(0);
1414     }
1415   }
1416   SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONGSTATE,"Array was not checked out");
1417 }
1418 
1419 PetscErrorCode DMSetNullSpaceConstructor(DM dm, PetscInt field, PetscErrorCode (*nullsp)(DM dm, PetscInt field, MatNullSpace *nullSpace))
1420 {
1421   PetscFunctionBegin;
1422   PetscValidHeaderSpecific(dm, DM_CLASSID, 1);
1423   if (field >= 10) SETERRQ1(PetscObjectComm((PetscObject)dm), PETSC_ERR_ARG_OUTOFRANGE, "Cannot handle %d >= 10 fields", field);
1424   dm->nullspaceConstructors[field] = nullsp;
1425   PetscFunctionReturn(0);
1426 }
1427 
1428 PetscErrorCode DMGetNullSpaceConstructor(DM dm, PetscInt field, PetscErrorCode (**nullsp)(DM dm, PetscInt field, MatNullSpace *nullSpace))
1429 {
1430   PetscFunctionBegin;
1431   PetscValidHeaderSpecific(dm, DM_CLASSID, 1);
1432   PetscValidPointer(nullsp, 3);
1433   if (field >= 10) SETERRQ1(PetscObjectComm((PetscObject)dm), PETSC_ERR_ARG_OUTOFRANGE, "Cannot handle %d >= 10 fields", field);
1434   *nullsp = dm->nullspaceConstructors[field];
1435   PetscFunctionReturn(0);
1436 }
1437 
1438 PetscErrorCode DMSetNearNullSpaceConstructor(DM dm, PetscInt field, PetscErrorCode (*nullsp)(DM dm, PetscInt field, MatNullSpace *nullSpace))
1439 {
1440   PetscFunctionBegin;
1441   PetscValidHeaderSpecific(dm, DM_CLASSID, 1);
1442   if (field >= 10) SETERRQ1(PetscObjectComm((PetscObject)dm), PETSC_ERR_ARG_OUTOFRANGE, "Cannot handle %d >= 10 fields", field);
1443   dm->nearnullspaceConstructors[field] = nullsp;
1444   PetscFunctionReturn(0);
1445 }
1446 
1447 PetscErrorCode DMGetNearNullSpaceConstructor(DM dm, PetscInt field, PetscErrorCode (**nullsp)(DM dm, PetscInt field, MatNullSpace *nullSpace))
1448 {
1449   PetscFunctionBegin;
1450   PetscValidHeaderSpecific(dm, DM_CLASSID, 1);
1451   PetscValidPointer(nullsp, 3);
1452   if (field >= 10) SETERRQ1(PetscObjectComm((PetscObject)dm), PETSC_ERR_ARG_OUTOFRANGE, "Cannot handle %d >= 10 fields", field);
1453   *nullsp = dm->nearnullspaceConstructors[field];
1454   PetscFunctionReturn(0);
1455 }
1456 
1457 /*@C
1458   DMCreateFieldIS - Creates a set of IS objects with the global indices of dofs for each field
1459 
1460   Not collective
1461 
1462   Input Parameter:
1463 . dm - the DM object
1464 
1465   Output Parameters:
1466 + numFields  - The number of fields (or NULL if not requested)
1467 . fieldNames - The name for each field (or NULL if not requested)
1468 - fields     - The global indices for each field (or NULL if not requested)
1469 
1470   Level: intermediate
1471 
1472   Notes:
1473   The user is responsible for freeing all requested arrays. In particular, every entry of names should be freed with
1474   PetscFree(), every entry of fields should be destroyed with ISDestroy(), and both arrays should be freed with
1475   PetscFree().
1476 
1477 .seealso DMDestroy(), DMView(), DMCreateInterpolation(), DMCreateColoring(), DMCreateMatrix()
1478 @*/
1479 PetscErrorCode DMCreateFieldIS(DM dm, PetscInt *numFields, char ***fieldNames, IS **fields)
1480 {
1481   PetscSection   section, sectionGlobal;
1482   PetscErrorCode ierr;
1483 
1484   PetscFunctionBegin;
1485   PetscValidHeaderSpecific(dm,DM_CLASSID,1);
1486   if (numFields) {
1487     PetscValidPointer(numFields,2);
1488     *numFields = 0;
1489   }
1490   if (fieldNames) {
1491     PetscValidPointer(fieldNames,3);
1492     *fieldNames = NULL;
1493   }
1494   if (fields) {
1495     PetscValidPointer(fields,4);
1496     *fields = NULL;
1497   }
1498   ierr = DMGetSection(dm, &section);CHKERRQ(ierr);
1499   if (section) {
1500     PetscInt *fieldSizes, *fieldNc, **fieldIndices;
1501     PetscInt nF, f, pStart, pEnd, p;
1502 
1503     ierr = DMGetGlobalSection(dm, &sectionGlobal);CHKERRQ(ierr);
1504     ierr = PetscSectionGetNumFields(section, &nF);CHKERRQ(ierr);
1505     ierr = PetscMalloc3(nF,&fieldSizes,nF,&fieldNc,nF,&fieldIndices);CHKERRQ(ierr);
1506     ierr = PetscSectionGetChart(sectionGlobal, &pStart, &pEnd);CHKERRQ(ierr);
1507     for (f = 0; f < nF; ++f) {
1508       fieldSizes[f] = 0;
1509       ierr          = PetscSectionGetFieldComponents(section, f, &fieldNc[f]);CHKERRQ(ierr);
1510     }
1511     for (p = pStart; p < pEnd; ++p) {
1512       PetscInt gdof;
1513 
1514       ierr = PetscSectionGetDof(sectionGlobal, p, &gdof);CHKERRQ(ierr);
1515       if (gdof > 0) {
1516         for (f = 0; f < nF; ++f) {
1517           PetscInt fdof, fcdof, fpdof;
1518 
1519           ierr  = PetscSectionGetFieldDof(section, p, f, &fdof);CHKERRQ(ierr);
1520           ierr  = PetscSectionGetFieldConstraintDof(section, p, f, &fcdof);CHKERRQ(ierr);
1521           fpdof = fdof-fcdof;
1522           if (fpdof && fpdof != fieldNc[f]) {
1523             /* Layout does not admit a pointwise block size */
1524             fieldNc[f] = 1;
1525           }
1526           fieldSizes[f] += fpdof;
1527         }
1528       }
1529     }
1530     for (f = 0; f < nF; ++f) {
1531       ierr          = PetscMalloc1(fieldSizes[f], &fieldIndices[f]);CHKERRQ(ierr);
1532       fieldSizes[f] = 0;
1533     }
1534     for (p = pStart; p < pEnd; ++p) {
1535       PetscInt gdof, goff;
1536 
1537       ierr = PetscSectionGetDof(sectionGlobal, p, &gdof);CHKERRQ(ierr);
1538       if (gdof > 0) {
1539         ierr = PetscSectionGetOffset(sectionGlobal, p, &goff);CHKERRQ(ierr);
1540         for (f = 0; f < nF; ++f) {
1541           PetscInt fdof, fcdof, fc;
1542 
1543           ierr = PetscSectionGetFieldDof(section, p, f, &fdof);CHKERRQ(ierr);
1544           ierr = PetscSectionGetFieldConstraintDof(section, p, f, &fcdof);CHKERRQ(ierr);
1545           for (fc = 0; fc < fdof-fcdof; ++fc, ++fieldSizes[f]) {
1546             fieldIndices[f][fieldSizes[f]] = goff++;
1547           }
1548         }
1549       }
1550     }
1551     if (numFields) *numFields = nF;
1552     if (fieldNames) {
1553       ierr = PetscMalloc1(nF, fieldNames);CHKERRQ(ierr);
1554       for (f = 0; f < nF; ++f) {
1555         const char *fieldName;
1556 
1557         ierr = PetscSectionGetFieldName(section, f, &fieldName);CHKERRQ(ierr);
1558         ierr = PetscStrallocpy(fieldName, (char**) &(*fieldNames)[f]);CHKERRQ(ierr);
1559       }
1560     }
1561     if (fields) {
1562       ierr = PetscMalloc1(nF, fields);CHKERRQ(ierr);
1563       for (f = 0; f < nF; ++f) {
1564         PetscInt bs, in[2], out[2];
1565 
1566         ierr  = ISCreateGeneral(PetscObjectComm((PetscObject)dm), fieldSizes[f], fieldIndices[f], PETSC_OWN_POINTER, &(*fields)[f]);CHKERRQ(ierr);
1567         in[0] = -fieldNc[f];
1568         in[1] = fieldNc[f];
1569         ierr  = MPIU_Allreduce(in, out, 2, MPIU_INT, MPI_MAX, PetscObjectComm((PetscObject)dm));CHKERRQ(ierr);
1570         bs    = (-out[0] == out[1]) ? out[1] : 1;
1571         ierr  = ISSetBlockSize((*fields)[f], bs);CHKERRQ(ierr);
1572       }
1573     }
1574     ierr = PetscFree3(fieldSizes,fieldNc,fieldIndices);CHKERRQ(ierr);
1575   } else if (dm->ops->createfieldis) {
1576     ierr = (*dm->ops->createfieldis)(dm, numFields, fieldNames, fields);CHKERRQ(ierr);
1577   }
1578   PetscFunctionReturn(0);
1579 }
1580 
1581 
1582 /*@C
1583   DMCreateFieldDecomposition - Returns a list of IS objects defining a decomposition of a problem into subproblems
1584                           corresponding to different fields: each IS contains the global indices of the dofs of the
1585                           corresponding field. The optional list of DMs define the DM for each subproblem.
1586                           Generalizes DMCreateFieldIS().
1587 
1588   Not collective
1589 
1590   Input Parameter:
1591 . dm - the DM object
1592 
1593   Output Parameters:
1594 + len       - The number of subproblems in the field decomposition (or NULL if not requested)
1595 . namelist  - The name for each field (or NULL if not requested)
1596 . islist    - The global indices for each field (or NULL if not requested)
1597 - dmlist    - The DMs for each field subproblem (or NULL, if not requested; if NULL is returned, no DMs are defined)
1598 
1599   Level: intermediate
1600 
1601   Notes:
1602   The user is responsible for freeing all requested arrays. In particular, every entry of names should be freed with
1603   PetscFree(), every entry of is should be destroyed with ISDestroy(), every entry of dm should be destroyed with DMDestroy(),
1604   and all of the arrays should be freed with PetscFree().
1605 
1606 .seealso DMDestroy(), DMView(), DMCreateInterpolation(), DMCreateColoring(), DMCreateMatrix(), DMCreateFieldIS()
1607 @*/
1608 PetscErrorCode DMCreateFieldDecomposition(DM dm, PetscInt *len, char ***namelist, IS **islist, DM **dmlist)
1609 {
1610   PetscErrorCode ierr;
1611 
1612   PetscFunctionBegin;
1613   PetscValidHeaderSpecific(dm,DM_CLASSID,1);
1614   if (len) {
1615     PetscValidPointer(len,2);
1616     *len = 0;
1617   }
1618   if (namelist) {
1619     PetscValidPointer(namelist,3);
1620     *namelist = 0;
1621   }
1622   if (islist) {
1623     PetscValidPointer(islist,4);
1624     *islist = 0;
1625   }
1626   if (dmlist) {
1627     PetscValidPointer(dmlist,5);
1628     *dmlist = 0;
1629   }
1630   /*
1631    Is it a good idea to apply the following check across all impls?
1632    Perhaps some impls can have a well-defined decomposition before DMSetUp?
1633    This, however, follows the general principle that accessors are not well-behaved until the object is set up.
1634    */
1635   if (!dm->setupcalled) SETERRQ(PetscObjectComm((PetscObject)dm),PETSC_ERR_ARG_WRONGSTATE, "Decomposition defined only after DMSetUp");
1636   if (!dm->ops->createfielddecomposition) {
1637     PetscSection section;
1638     PetscInt     numFields, f;
1639 
1640     ierr = DMGetSection(dm, &section);CHKERRQ(ierr);
1641     if (section) {ierr = PetscSectionGetNumFields(section, &numFields);CHKERRQ(ierr);}
1642     if (section && numFields && dm->ops->createsubdm) {
1643       if (len) *len = numFields;
1644       if (namelist) {ierr = PetscMalloc1(numFields,namelist);CHKERRQ(ierr);}
1645       if (islist)   {ierr = PetscMalloc1(numFields,islist);CHKERRQ(ierr);}
1646       if (dmlist)   {ierr = PetscMalloc1(numFields,dmlist);CHKERRQ(ierr);}
1647       for (f = 0; f < numFields; ++f) {
1648         const char *fieldName;
1649 
1650         ierr = DMCreateSubDM(dm, 1, &f, islist ? &(*islist)[f] : NULL, dmlist ? &(*dmlist)[f] : NULL);CHKERRQ(ierr);
1651         if (namelist) {
1652           ierr = PetscSectionGetFieldName(section, f, &fieldName);CHKERRQ(ierr);
1653           ierr = PetscStrallocpy(fieldName, (char**) &(*namelist)[f]);CHKERRQ(ierr);
1654         }
1655       }
1656     } else {
1657       ierr = DMCreateFieldIS(dm, len, namelist, islist);CHKERRQ(ierr);
1658       /* By default there are no DMs associated with subproblems. */
1659       if (dmlist) *dmlist = NULL;
1660     }
1661   } else {
1662     ierr = (*dm->ops->createfielddecomposition)(dm,len,namelist,islist,dmlist);CHKERRQ(ierr);
1663   }
1664   PetscFunctionReturn(0);
1665 }
1666 
1667 /*@
1668   DMCreateSubDM - Returns an IS and DM encapsulating a subproblem defined by the fields passed in.
1669                   The fields are defined by DMCreateFieldIS().
1670 
1671   Not collective
1672 
1673   Input Parameters:
1674 + dm        - The DM object
1675 . numFields - The number of fields in this subproblem
1676 - fields    - The field numbers of the selected fields
1677 
1678   Output Parameters:
1679 + is - The global indices for the subproblem
1680 - subdm - The DM for the subproblem
1681 
1682   Note: You need to call DMPlexSetMigrationSF() on the original DM if you want the Global-To-Natural map to be automatically constructed
1683 
1684   Level: intermediate
1685 
1686 .seealso DMPlexSetMigrationSF(), DMDestroy(), DMView(), DMCreateInterpolation(), DMCreateColoring(), DMCreateMatrix(), DMCreateFieldIS()
1687 @*/
1688 PetscErrorCode DMCreateSubDM(DM dm, PetscInt numFields, const PetscInt fields[], IS *is, DM *subdm)
1689 {
1690   PetscErrorCode ierr;
1691 
1692   PetscFunctionBegin;
1693   PetscValidHeaderSpecific(dm,DM_CLASSID,1);
1694   PetscValidPointer(fields,3);
1695   if (is) PetscValidPointer(is,4);
1696   if (subdm) PetscValidPointer(subdm,5);
1697   if (dm->ops->createsubdm) {
1698     ierr = (*dm->ops->createsubdm)(dm, numFields, fields, is, subdm);CHKERRQ(ierr);
1699   } else SETERRQ(PetscObjectComm((PetscObject)dm), PETSC_ERR_SUP, "This type has no DMCreateSubDM implementation defined");
1700   PetscFunctionReturn(0);
1701 }
1702 
1703 /*@C
1704   DMCreateSuperDM - Returns an arrays of ISes and DM encapsulating a superproblem defined by the DMs passed in.
1705 
1706   Not collective
1707 
1708   Input Parameter:
1709 + dms - The DM objects
1710 - len - The number of DMs
1711 
1712   Output Parameters:
1713 + is - The global indices for the subproblem, or NULL
1714 - superdm - The DM for the superproblem
1715 
1716   Note: You need to call DMPlexSetMigrationSF() on the original DM if you want the Global-To-Natural map to be automatically constructed
1717 
1718   Level: intermediate
1719 
1720 .seealso DMPlexSetMigrationSF(), DMDestroy(), DMView(), DMCreateInterpolation(), DMCreateColoring(), DMCreateMatrix(), DMCreateFieldIS()
1721 @*/
1722 PetscErrorCode DMCreateSuperDM(DM dms[], PetscInt len, IS **is, DM *superdm)
1723 {
1724   PetscInt       i;
1725   PetscErrorCode ierr;
1726 
1727   PetscFunctionBegin;
1728   PetscValidPointer(dms,1);
1729   for (i = 0; i < len; ++i) {PetscValidHeaderSpecific(dms[i],DM_CLASSID,1);}
1730   if (is) PetscValidPointer(is,3);
1731   PetscValidPointer(superdm,4);
1732   if (len < 0) SETERRQ1(PETSC_COMM_SELF, PETSC_ERR_ARG_OUTOFRANGE, "Number of DMs must be nonnegative: %D", len);
1733   if (len) {
1734     if (dms[0]->ops->createsuperdm) {ierr = (*dms[0]->ops->createsuperdm)(dms, len, is, superdm);CHKERRQ(ierr);}
1735     else SETERRQ(PetscObjectComm((PetscObject) dms[0]), PETSC_ERR_SUP, "This type has no DMCreateSuperDM implementation defined");
1736   }
1737   PetscFunctionReturn(0);
1738 }
1739 
1740 
1741 /*@C
1742   DMCreateDomainDecomposition - Returns lists of IS objects defining a decomposition of a problem into subproblems
1743                           corresponding to restrictions to pairs nested subdomains: each IS contains the global
1744                           indices of the dofs of the corresponding subdomains.  The inner subdomains conceptually
1745                           define a nonoverlapping covering, while outer subdomains can overlap.
1746                           The optional list of DMs define the DM for each subproblem.
1747 
1748   Not collective
1749 
1750   Input Parameter:
1751 . dm - the DM object
1752 
1753   Output Parameters:
1754 + len         - The number of subproblems in the domain decomposition (or NULL if not requested)
1755 . namelist    - The name for each subdomain (or NULL if not requested)
1756 . innerislist - The global indices for each inner subdomain (or NULL, if not requested)
1757 . outerislist - The global indices for each outer subdomain (or NULL, if not requested)
1758 - dmlist      - The DMs for each subdomain subproblem (or NULL, if not requested; if NULL is returned, no DMs are defined)
1759 
1760   Level: intermediate
1761 
1762   Notes:
1763   The user is responsible for freeing all requested arrays. In particular, every entry of names should be freed with
1764   PetscFree(), every entry of is should be destroyed with ISDestroy(), every entry of dm should be destroyed with DMDestroy(),
1765   and all of the arrays should be freed with PetscFree().
1766 
1767 .seealso DMDestroy(), DMView(), DMCreateInterpolation(), DMCreateColoring(), DMCreateMatrix(), DMCreateDomainDecompositionDM(), DMCreateFieldDecomposition()
1768 @*/
1769 PetscErrorCode DMCreateDomainDecomposition(DM dm, PetscInt *len, char ***namelist, IS **innerislist, IS **outerislist, DM **dmlist)
1770 {
1771   PetscErrorCode      ierr;
1772   DMSubDomainHookLink link;
1773   PetscInt            i,l;
1774 
1775   PetscFunctionBegin;
1776   PetscValidHeaderSpecific(dm,DM_CLASSID,1);
1777   if (len)           {PetscValidPointer(len,2);            *len         = 0;}
1778   if (namelist)      {PetscValidPointer(namelist,3);       *namelist    = NULL;}
1779   if (innerislist)   {PetscValidPointer(innerislist,4);    *innerislist = NULL;}
1780   if (outerislist)   {PetscValidPointer(outerislist,5);    *outerislist = NULL;}
1781   if (dmlist)        {PetscValidPointer(dmlist,6);         *dmlist      = NULL;}
1782   /*
1783    Is it a good idea to apply the following check across all impls?
1784    Perhaps some impls can have a well-defined decomposition before DMSetUp?
1785    This, however, follows the general principle that accessors are not well-behaved until the object is set up.
1786    */
1787   if (!dm->setupcalled) SETERRQ(PetscObjectComm((PetscObject)dm),PETSC_ERR_ARG_WRONGSTATE, "Decomposition defined only after DMSetUp");
1788   if (dm->ops->createdomaindecomposition) {
1789     ierr = (*dm->ops->createdomaindecomposition)(dm,&l,namelist,innerislist,outerislist,dmlist);CHKERRQ(ierr);
1790     /* copy subdomain hooks and context over to the subdomain DMs */
1791     if (dmlist && *dmlist) {
1792       for (i = 0; i < l; i++) {
1793         for (link=dm->subdomainhook; link; link=link->next) {
1794           if (link->ddhook) {ierr = (*link->ddhook)(dm,(*dmlist)[i],link->ctx);CHKERRQ(ierr);}
1795         }
1796         if (dm->ctx) (*dmlist)[i]->ctx = dm->ctx;
1797       }
1798     }
1799     if (len) *len = l;
1800   }
1801   PetscFunctionReturn(0);
1802 }
1803 
1804 
1805 /*@C
1806   DMCreateDomainDecompositionScatters - Returns scatters to the subdomain vectors from the global vector
1807 
1808   Not collective
1809 
1810   Input Parameters:
1811 + dm - the DM object
1812 . n  - the number of subdomain scatters
1813 - subdms - the local subdomains
1814 
1815   Output Parameters:
1816 + n     - the number of scatters returned
1817 . iscat - scatter from global vector to nonoverlapping global vector entries on subdomain
1818 . oscat - scatter from global vector to overlapping global vector entries on subdomain
1819 - gscat - scatter from global vector to local vector on subdomain (fills in ghosts)
1820 
1821   Notes:
1822     This is an alternative to the iis and ois arguments in DMCreateDomainDecomposition that allow for the solution
1823   of general nonlinear problems with overlapping subdomain methods.  While merely having index sets that enable subsets
1824   of the residual equations to be created is fine for linear problems, nonlinear problems require local assembly of
1825   solution and residual data.
1826 
1827   Level: developer
1828 
1829 .seealso DMDestroy(), DMView(), DMCreateInterpolation(), DMCreateColoring(), DMCreateMatrix(), DMCreateFieldIS()
1830 @*/
1831 PetscErrorCode DMCreateDomainDecompositionScatters(DM dm,PetscInt n,DM *subdms,VecScatter **iscat,VecScatter **oscat,VecScatter **gscat)
1832 {
1833   PetscErrorCode ierr;
1834 
1835   PetscFunctionBegin;
1836   PetscValidHeaderSpecific(dm,DM_CLASSID,1);
1837   PetscValidPointer(subdms,3);
1838   if (dm->ops->createddscatters) {
1839     ierr = (*dm->ops->createddscatters)(dm,n,subdms,iscat,oscat,gscat);CHKERRQ(ierr);
1840   } else SETERRQ(PetscObjectComm((PetscObject)dm), PETSC_ERR_SUP, "This type has no DMCreateDomainDecompositionScatter implementation defined");
1841   PetscFunctionReturn(0);
1842 }
1843 
1844 /*@
1845   DMRefine - Refines a DM object
1846 
1847   Collective on DM
1848 
1849   Input Parameter:
1850 + dm   - the DM object
1851 - comm - the communicator to contain the new DM object (or MPI_COMM_NULL)
1852 
1853   Output Parameter:
1854 . dmf - the refined DM, or NULL
1855 
1856   Note: If no refinement was done, the return value is NULL
1857 
1858   Level: developer
1859 
1860 .seealso DMCoarsen(), DMDestroy(), DMView(), DMCreateGlobalVector(), DMCreateInterpolation()
1861 @*/
1862 PetscErrorCode  DMRefine(DM dm,MPI_Comm comm,DM *dmf)
1863 {
1864   PetscErrorCode   ierr;
1865   DMRefineHookLink link;
1866 
1867   PetscFunctionBegin;
1868   PetscValidHeaderSpecific(dm,DM_CLASSID,1);
1869   ierr = PetscLogEventBegin(DM_Refine,dm,0,0,0);CHKERRQ(ierr);
1870   if (!dm->ops->refine) SETERRQ(PetscObjectComm((PetscObject)dm),PETSC_ERR_SUP,"This DM cannot refine");
1871   ierr = (*dm->ops->refine)(dm,comm,dmf);CHKERRQ(ierr);
1872   if (*dmf) {
1873     (*dmf)->ops->creatematrix = dm->ops->creatematrix;
1874 
1875     ierr = PetscObjectCopyFortranFunctionPointers((PetscObject)dm,(PetscObject)*dmf);CHKERRQ(ierr);
1876 
1877     (*dmf)->ctx       = dm->ctx;
1878     (*dmf)->leveldown = dm->leveldown;
1879     (*dmf)->levelup   = dm->levelup + 1;
1880 
1881     ierr = DMSetMatType(*dmf,dm->mattype);CHKERRQ(ierr);
1882     for (link=dm->refinehook; link; link=link->next) {
1883       if (link->refinehook) {
1884         ierr = (*link->refinehook)(dm,*dmf,link->ctx);CHKERRQ(ierr);
1885       }
1886     }
1887   }
1888   ierr = PetscLogEventEnd(DM_Refine,dm,0,0,0);CHKERRQ(ierr);
1889   PetscFunctionReturn(0);
1890 }
1891 
1892 /*@C
1893    DMRefineHookAdd - adds a callback to be run when interpolating a nonlinear problem to a finer grid
1894 
1895    Logically Collective
1896 
1897    Input Arguments:
1898 +  coarse - nonlinear solver context on which to run a hook when restricting to a coarser level
1899 .  refinehook - function to run when setting up a coarser level
1900 .  interphook - function to run to update data on finer levels (once per SNESSolve())
1901 -  ctx - [optional] user-defined context for provide data for the hooks (may be NULL)
1902 
1903    Calling sequence of refinehook:
1904 $    refinehook(DM coarse,DM fine,void *ctx);
1905 
1906 +  coarse - coarse level DM
1907 .  fine - fine level DM to interpolate problem to
1908 -  ctx - optional user-defined function context
1909 
1910    Calling sequence for interphook:
1911 $    interphook(DM coarse,Mat interp,DM fine,void *ctx)
1912 
1913 +  coarse - coarse level DM
1914 .  interp - matrix interpolating a coarse-level solution to the finer grid
1915 .  fine - fine level DM to update
1916 -  ctx - optional user-defined function context
1917 
1918    Level: advanced
1919 
1920    Notes:
1921    This function is only needed if auxiliary data needs to be passed to fine grids while grid sequencing
1922 
1923    If this function is called multiple times, the hooks will be run in the order they are added.
1924 
1925    This function is currently not available from Fortran.
1926 
1927 .seealso: DMCoarsenHookAdd(), SNESFASGetInterpolation(), SNESFASGetInjection(), PetscObjectCompose(), PetscContainerCreate()
1928 @*/
1929 PetscErrorCode DMRefineHookAdd(DM coarse,PetscErrorCode (*refinehook)(DM,DM,void*),PetscErrorCode (*interphook)(DM,Mat,DM,void*),void *ctx)
1930 {
1931   PetscErrorCode   ierr;
1932   DMRefineHookLink link,*p;
1933 
1934   PetscFunctionBegin;
1935   PetscValidHeaderSpecific(coarse,DM_CLASSID,1);
1936   for (p=&coarse->refinehook; *p; p=&(*p)->next) { /* Scan to the end of the current list of hooks */
1937     if ((*p)->refinehook == refinehook && (*p)->interphook == interphook && (*p)->ctx == ctx) PetscFunctionReturn(0);
1938   }
1939   ierr             = PetscNew(&link);CHKERRQ(ierr);
1940   link->refinehook = refinehook;
1941   link->interphook = interphook;
1942   link->ctx        = ctx;
1943   link->next       = NULL;
1944   *p               = link;
1945   PetscFunctionReturn(0);
1946 }
1947 
1948 /*@C
1949    DMRefineHookRemove - remove a callback from the list of hooks to be run when interpolating a nonlinear problem to a finer grid
1950 
1951    Logically Collective
1952 
1953    Input Arguments:
1954 +  coarse - nonlinear solver context on which to run a hook when restricting to a coarser level
1955 .  refinehook - function to run when setting up a coarser level
1956 .  interphook - function to run to update data on finer levels (once per SNESSolve())
1957 -  ctx - [optional] user-defined context for provide data for the hooks (may be NULL)
1958 
1959    Level: advanced
1960 
1961    Notes:
1962    This function does nothing if the hook is not in the list.
1963 
1964    This function is currently not available from Fortran.
1965 
1966 .seealso: DMCoarsenHookRemove(), SNESFASGetInterpolation(), SNESFASGetInjection(), PetscObjectCompose(), PetscContainerCreate()
1967 @*/
1968 PetscErrorCode DMRefineHookRemove(DM coarse,PetscErrorCode (*refinehook)(DM,DM,void*),PetscErrorCode (*interphook)(DM,Mat,DM,void*),void *ctx)
1969 {
1970   PetscErrorCode   ierr;
1971   DMRefineHookLink link,*p;
1972 
1973   PetscFunctionBegin;
1974   PetscValidHeaderSpecific(coarse,DM_CLASSID,1);
1975   for (p=&coarse->refinehook; *p; p=&(*p)->next) { /* Search the list of current hooks */
1976     if ((*p)->refinehook == refinehook && (*p)->interphook == interphook && (*p)->ctx == ctx) {
1977       link = *p;
1978       *p = link->next;
1979       ierr = PetscFree(link);CHKERRQ(ierr);
1980       break;
1981     }
1982   }
1983   PetscFunctionReturn(0);
1984 }
1985 
1986 /*@
1987    DMInterpolate - interpolates user-defined problem data to a finer DM by running hooks registered by DMRefineHookAdd()
1988 
1989    Collective if any hooks are
1990 
1991    Input Arguments:
1992 +  coarse - coarser DM to use as a base
1993 .  interp - interpolation matrix, apply using MatInterpolate()
1994 -  fine - finer DM to update
1995 
1996    Level: developer
1997 
1998 .seealso: DMRefineHookAdd(), MatInterpolate()
1999 @*/
2000 PetscErrorCode DMInterpolate(DM coarse,Mat interp,DM fine)
2001 {
2002   PetscErrorCode   ierr;
2003   DMRefineHookLink link;
2004 
2005   PetscFunctionBegin;
2006   for (link=fine->refinehook; link; link=link->next) {
2007     if (link->interphook) {
2008       ierr = (*link->interphook)(coarse,interp,fine,link->ctx);CHKERRQ(ierr);
2009     }
2010   }
2011   PetscFunctionReturn(0);
2012 }
2013 
2014 /*@
2015     DMGetRefineLevel - Get's the number of refinements that have generated this DM.
2016 
2017     Not Collective
2018 
2019     Input Parameter:
2020 .   dm - the DM object
2021 
2022     Output Parameter:
2023 .   level - number of refinements
2024 
2025     Level: developer
2026 
2027 .seealso DMCoarsen(), DMGetCoarsenLevel(), DMDestroy(), DMView(), DMCreateGlobalVector(), DMCreateInterpolation()
2028 
2029 @*/
2030 PetscErrorCode  DMGetRefineLevel(DM dm,PetscInt *level)
2031 {
2032   PetscFunctionBegin;
2033   PetscValidHeaderSpecific(dm,DM_CLASSID,1);
2034   *level = dm->levelup;
2035   PetscFunctionReturn(0);
2036 }
2037 
2038 /*@
2039     DMSetRefineLevel - Set's the number of refinements that have generated this DM.
2040 
2041     Not Collective
2042 
2043     Input Parameter:
2044 +   dm - the DM object
2045 -   level - number of refinements
2046 
2047     Level: advanced
2048 
2049     Notes:
2050     This value is used by PCMG to determine how many multigrid levels to use
2051 
2052 .seealso DMCoarsen(), DMGetCoarsenLevel(), DMDestroy(), DMView(), DMCreateGlobalVector(), DMCreateInterpolation()
2053 
2054 @*/
2055 PetscErrorCode  DMSetRefineLevel(DM dm,PetscInt level)
2056 {
2057   PetscFunctionBegin;
2058   PetscValidHeaderSpecific(dm,DM_CLASSID,1);
2059   dm->levelup = level;
2060   PetscFunctionReturn(0);
2061 }
2062 
2063 /*@C
2064    DMGlobalToLocalHookAdd - adds a callback to be run when global to local is called
2065 
2066    Logically Collective
2067 
2068    Input Arguments:
2069 +  dm - the DM
2070 .  beginhook - function to run at the beginning of DMGlobalToLocalBegin()
2071 .  endhook - function to run after DMGlobalToLocalEnd() has completed
2072 -  ctx - [optional] user-defined context for provide data for the hooks (may be NULL)
2073 
2074    Calling sequence for beginhook:
2075 $    beginhook(DM fine,VecScatter out,VecScatter in,DM coarse,void *ctx)
2076 
2077 +  dm - global DM
2078 .  g - global vector
2079 .  mode - mode
2080 .  l - local vector
2081 -  ctx - optional user-defined function context
2082 
2083 
2084    Calling sequence for endhook:
2085 $    endhook(DM fine,VecScatter out,VecScatter in,DM coarse,void *ctx)
2086 
2087 +  global - global DM
2088 -  ctx - optional user-defined function context
2089 
2090    Level: advanced
2091 
2092 .seealso: DMRefineHookAdd(), SNESFASGetInterpolation(), SNESFASGetInjection(), PetscObjectCompose(), PetscContainerCreate()
2093 @*/
2094 PetscErrorCode DMGlobalToLocalHookAdd(DM dm,PetscErrorCode (*beginhook)(DM,Vec,InsertMode,Vec,void*),PetscErrorCode (*endhook)(DM,Vec,InsertMode,Vec,void*),void *ctx)
2095 {
2096   PetscErrorCode          ierr;
2097   DMGlobalToLocalHookLink link,*p;
2098 
2099   PetscFunctionBegin;
2100   PetscValidHeaderSpecific(dm,DM_CLASSID,1);
2101   for (p=&dm->gtolhook; *p; p=&(*p)->next) {} /* Scan to the end of the current list of hooks */
2102   ierr            = PetscNew(&link);CHKERRQ(ierr);
2103   link->beginhook = beginhook;
2104   link->endhook   = endhook;
2105   link->ctx       = ctx;
2106   link->next      = NULL;
2107   *p              = link;
2108   PetscFunctionReturn(0);
2109 }
2110 
2111 static PetscErrorCode DMGlobalToLocalHook_Constraints(DM dm, Vec g, InsertMode mode, Vec l, void *ctx)
2112 {
2113   Mat cMat;
2114   Vec cVec;
2115   PetscSection section, cSec;
2116   PetscInt pStart, pEnd, p, dof;
2117   PetscErrorCode ierr;
2118 
2119   PetscFunctionBegin;
2120   PetscValidHeaderSpecific(dm, DM_CLASSID, 1);
2121   ierr = DMGetDefaultConstraints(dm,&cSec,&cMat);CHKERRQ(ierr);
2122   if (cMat && (mode == INSERT_VALUES || mode == INSERT_ALL_VALUES || mode == INSERT_BC_VALUES)) {
2123     PetscInt nRows;
2124 
2125     ierr = MatGetSize(cMat,&nRows,NULL);CHKERRQ(ierr);
2126     if (nRows <= 0) PetscFunctionReturn(0);
2127     ierr = DMGetSection(dm,&section);CHKERRQ(ierr);
2128     ierr = MatCreateVecs(cMat,NULL,&cVec);CHKERRQ(ierr);
2129     ierr = MatMult(cMat,l,cVec);CHKERRQ(ierr);
2130     ierr = PetscSectionGetChart(cSec,&pStart,&pEnd);CHKERRQ(ierr);
2131     for (p = pStart; p < pEnd; p++) {
2132       ierr = PetscSectionGetDof(cSec,p,&dof);CHKERRQ(ierr);
2133       if (dof) {
2134         PetscScalar *vals;
2135         ierr = VecGetValuesSection(cVec,cSec,p,&vals);CHKERRQ(ierr);
2136         ierr = VecSetValuesSection(l,section,p,vals,INSERT_ALL_VALUES);CHKERRQ(ierr);
2137       }
2138     }
2139     ierr = VecDestroy(&cVec);CHKERRQ(ierr);
2140   }
2141   PetscFunctionReturn(0);
2142 }
2143 
2144 /*@
2145     DMGlobalToLocal - update local vectors from global vector
2146 
2147     Neighbor-wise Collective on DM
2148 
2149     Input Parameters:
2150 +   dm - the DM object
2151 .   g - the global vector
2152 .   mode - INSERT_VALUES or ADD_VALUES
2153 -   l - the local vector
2154 
2155     Notes:
2156     The communication involved in this update can be overlapped with computation by using
2157     DMGlobalToLocalBegin() and DMGlobalToLocalEnd().
2158 
2159     Level: beginner
2160 
2161 .seealso DMCoarsen(), DMDestroy(), DMView(), DMCreateGlobalVector(), DMCreateInterpolation(), DMGlobalToLocalEnd(), DMLocalToGlobalBegin(), DMLocalToGlobal(), DMLocalToGlobalBegin(), DMLocalToGlobalEnd()
2162 
2163 @*/
2164 PetscErrorCode DMGlobalToLocal(DM dm,Vec g,InsertMode mode,Vec l)
2165 {
2166   PetscErrorCode ierr;
2167 
2168   PetscFunctionBegin;
2169   ierr = DMGlobalToLocalBegin(dm,g,mode,l);CHKERRQ(ierr);
2170   ierr = DMGlobalToLocalEnd(dm,g,mode,l);CHKERRQ(ierr);
2171   PetscFunctionReturn(0);
2172 }
2173 
2174 /*@
2175     DMGlobalToLocalBegin - Begins updating local vectors from global vector
2176 
2177     Neighbor-wise Collective on DM
2178 
2179     Input Parameters:
2180 +   dm - the DM object
2181 .   g - the global vector
2182 .   mode - INSERT_VALUES or ADD_VALUES
2183 -   l - the local vector
2184 
2185     Level: intermediate
2186 
2187 .seealso DMCoarsen(), DMDestroy(), DMView(), DMCreateGlobalVector(), DMCreateInterpolation(), DMGlobalToLocal(), DMGlobalToLocalEnd(), DMLocalToGlobalBegin(), DMLocalToGlobal(), DMLocalToGlobalBegin(), DMLocalToGlobalEnd()
2188 
2189 @*/
2190 PetscErrorCode  DMGlobalToLocalBegin(DM dm,Vec g,InsertMode mode,Vec l)
2191 {
2192   PetscSF                 sf;
2193   PetscErrorCode          ierr;
2194   DMGlobalToLocalHookLink link;
2195 
2196   PetscFunctionBegin;
2197   PetscValidHeaderSpecific(dm,DM_CLASSID,1);
2198   for (link=dm->gtolhook; link; link=link->next) {
2199     if (link->beginhook) {
2200       ierr = (*link->beginhook)(dm,g,mode,l,link->ctx);CHKERRQ(ierr);
2201     }
2202   }
2203   ierr = DMGetDefaultSF(dm, &sf);CHKERRQ(ierr);
2204   if (sf) {
2205     const PetscScalar *gArray;
2206     PetscScalar       *lArray;
2207 
2208     if (mode == ADD_VALUES) SETERRQ1(PetscObjectComm((PetscObject)dm), PETSC_ERR_ARG_OUTOFRANGE, "Invalid insertion mode %D", mode);
2209     ierr = VecGetArray(l, &lArray);CHKERRQ(ierr);
2210     ierr = VecGetArrayRead(g, &gArray);CHKERRQ(ierr);
2211     ierr = PetscSFBcastBegin(sf, MPIU_SCALAR, gArray, lArray);CHKERRQ(ierr);
2212     ierr = VecRestoreArray(l, &lArray);CHKERRQ(ierr);
2213     ierr = VecRestoreArrayRead(g, &gArray);CHKERRQ(ierr);
2214   } else {
2215     if (!dm->ops->globaltolocalbegin) SETERRQ1(PetscObjectComm((PetscObject)dm), PETSC_ERR_SUP, "Missing DMGlobalToLocalBegin() for type %s",((PetscObject)dm)->type_name);
2216     ierr = (*dm->ops->globaltolocalbegin)(dm,g,mode == INSERT_ALL_VALUES ? INSERT_VALUES : (mode == ADD_ALL_VALUES ? ADD_VALUES : mode),l);CHKERRQ(ierr);
2217   }
2218   PetscFunctionReturn(0);
2219 }
2220 
2221 /*@
2222     DMGlobalToLocalEnd - Ends updating local vectors from global vector
2223 
2224     Neighbor-wise Collective on DM
2225 
2226     Input Parameters:
2227 +   dm - the DM object
2228 .   g - the global vector
2229 .   mode - INSERT_VALUES or ADD_VALUES
2230 -   l - the local vector
2231 
2232     Level: intermediate
2233 
2234 .seealso DMCoarsen(), DMDestroy(), DMView(), DMCreateGlobalVector(), DMCreateInterpolation(), DMGlobalToLocal(), DMLocalToGlobalBegin(), DMLocalToGlobal(), DMLocalToGlobalBegin(), DMLocalToGlobalEnd()
2235 
2236 @*/
2237 PetscErrorCode  DMGlobalToLocalEnd(DM dm,Vec g,InsertMode mode,Vec l)
2238 {
2239   PetscSF                 sf;
2240   PetscErrorCode          ierr;
2241   const PetscScalar      *gArray;
2242   PetscScalar            *lArray;
2243   DMGlobalToLocalHookLink link;
2244 
2245   PetscFunctionBegin;
2246   PetscValidHeaderSpecific(dm,DM_CLASSID,1);
2247   ierr = DMGetDefaultSF(dm, &sf);CHKERRQ(ierr);
2248   if (sf) {
2249     if (mode == ADD_VALUES) SETERRQ1(PetscObjectComm((PetscObject)dm), PETSC_ERR_ARG_OUTOFRANGE, "Invalid insertion mode %D", mode);
2250 
2251     ierr = VecGetArray(l, &lArray);CHKERRQ(ierr);
2252     ierr = VecGetArrayRead(g, &gArray);CHKERRQ(ierr);
2253     ierr = PetscSFBcastEnd(sf, MPIU_SCALAR, gArray, lArray);CHKERRQ(ierr);
2254     ierr = VecRestoreArray(l, &lArray);CHKERRQ(ierr);
2255     ierr = VecRestoreArrayRead(g, &gArray);CHKERRQ(ierr);
2256   } else {
2257     if (!dm->ops->globaltolocalend) SETERRQ1(PetscObjectComm((PetscObject)dm), PETSC_ERR_SUP, "Missing DMGlobalToLocalEnd() for type %s",((PetscObject)dm)->type_name);
2258     ierr = (*dm->ops->globaltolocalend)(dm,g,mode == INSERT_ALL_VALUES ? INSERT_VALUES : (mode == ADD_ALL_VALUES ? ADD_VALUES : mode),l);CHKERRQ(ierr);
2259   }
2260   ierr = DMGlobalToLocalHook_Constraints(dm,g,mode,l,NULL);CHKERRQ(ierr);
2261   for (link=dm->gtolhook; link; link=link->next) {
2262     if (link->endhook) {ierr = (*link->endhook)(dm,g,mode,l,link->ctx);CHKERRQ(ierr);}
2263   }
2264   PetscFunctionReturn(0);
2265 }
2266 
2267 /*@C
2268    DMLocalToGlobalHookAdd - adds a callback to be run when a local to global is called
2269 
2270    Logically Collective
2271 
2272    Input Arguments:
2273 +  dm - the DM
2274 .  beginhook - function to run at the beginning of DMLocalToGlobalBegin()
2275 .  endhook - function to run after DMLocalToGlobalEnd() has completed
2276 -  ctx - [optional] user-defined context for provide data for the hooks (may be NULL)
2277 
2278    Calling sequence for beginhook:
2279 $    beginhook(DM fine,Vec l,InsertMode mode,Vec g,void *ctx)
2280 
2281 +  dm - global DM
2282 .  l - local vector
2283 .  mode - mode
2284 .  g - global vector
2285 -  ctx - optional user-defined function context
2286 
2287 
2288    Calling sequence for endhook:
2289 $    endhook(DM fine,Vec l,InsertMode mode,Vec g,void *ctx)
2290 
2291 +  global - global DM
2292 .  l - local vector
2293 .  mode - mode
2294 .  g - global vector
2295 -  ctx - optional user-defined function context
2296 
2297    Level: advanced
2298 
2299 .seealso: DMRefineHookAdd(), SNESFASGetInterpolation(), SNESFASGetInjection(), PetscObjectCompose(), PetscContainerCreate()
2300 @*/
2301 PetscErrorCode DMLocalToGlobalHookAdd(DM dm,PetscErrorCode (*beginhook)(DM,Vec,InsertMode,Vec,void*),PetscErrorCode (*endhook)(DM,Vec,InsertMode,Vec,void*),void *ctx)
2302 {
2303   PetscErrorCode          ierr;
2304   DMLocalToGlobalHookLink link,*p;
2305 
2306   PetscFunctionBegin;
2307   PetscValidHeaderSpecific(dm,DM_CLASSID,1);
2308   for (p=&dm->ltoghook; *p; p=&(*p)->next) {} /* Scan to the end of the current list of hooks */
2309   ierr            = PetscNew(&link);CHKERRQ(ierr);
2310   link->beginhook = beginhook;
2311   link->endhook   = endhook;
2312   link->ctx       = ctx;
2313   link->next      = NULL;
2314   *p              = link;
2315   PetscFunctionReturn(0);
2316 }
2317 
2318 static PetscErrorCode DMLocalToGlobalHook_Constraints(DM dm, Vec l, InsertMode mode, Vec g, void *ctx)
2319 {
2320   Mat cMat;
2321   Vec cVec;
2322   PetscSection section, cSec;
2323   PetscInt pStart, pEnd, p, dof;
2324   PetscErrorCode ierr;
2325 
2326   PetscFunctionBegin;
2327   PetscValidHeaderSpecific(dm, DM_CLASSID, 1);
2328   ierr = DMGetDefaultConstraints(dm,&cSec,&cMat);CHKERRQ(ierr);
2329   if (cMat && (mode == ADD_VALUES || mode == ADD_ALL_VALUES || mode == ADD_BC_VALUES)) {
2330     PetscInt nRows;
2331 
2332     ierr = MatGetSize(cMat,&nRows,NULL);CHKERRQ(ierr);
2333     if (nRows <= 0) PetscFunctionReturn(0);
2334     ierr = DMGetSection(dm,&section);CHKERRQ(ierr);
2335     ierr = MatCreateVecs(cMat,NULL,&cVec);CHKERRQ(ierr);
2336     ierr = PetscSectionGetChart(cSec,&pStart,&pEnd);CHKERRQ(ierr);
2337     for (p = pStart; p < pEnd; p++) {
2338       ierr = PetscSectionGetDof(cSec,p,&dof);CHKERRQ(ierr);
2339       if (dof) {
2340         PetscInt d;
2341         PetscScalar *vals;
2342         ierr = VecGetValuesSection(l,section,p,&vals);CHKERRQ(ierr);
2343         ierr = VecSetValuesSection(cVec,cSec,p,vals,mode);CHKERRQ(ierr);
2344         /* for this to be the true transpose, we have to zero the values that
2345          * we just extracted */
2346         for (d = 0; d < dof; d++) {
2347           vals[d] = 0.;
2348         }
2349       }
2350     }
2351     ierr = MatMultTransposeAdd(cMat,cVec,l,l);CHKERRQ(ierr);
2352     ierr = VecDestroy(&cVec);CHKERRQ(ierr);
2353   }
2354   PetscFunctionReturn(0);
2355 }
2356 /*@
2357     DMLocalToGlobal - updates global vectors from local vectors
2358 
2359     Neighbor-wise Collective on DM
2360 
2361     Input Parameters:
2362 +   dm - the DM object
2363 .   l - the local vector
2364 .   mode - if INSERT_VALUES then no parallel communication is used, if ADD_VALUES then all ghost points from the same base point accumulate into that base point.
2365 -   g - the global vector
2366 
2367     Notes:
2368     The communication involved in this update can be overlapped with computation by using
2369     DMLocalToGlobalBegin() and DMLocalToGlobalEnd().
2370 
2371     In the ADD_VALUES case you normally would zero the receiving vector before beginning this operation.
2372            INSERT_VALUES is not supported for DMDA; in that case simply compute the values directly into a global vector instead of a local one.
2373 
2374     Level: beginner
2375 
2376 .seealso DMLocalToGlobalBegin(), DMLocalToGlobalEnd(), DMCoarsen(), DMDestroy(), DMView(), DMCreateGlobalVector(), DMCreateInterpolation(), DMGlobalToLocal(), DMGlobalToLocalEnd(), DMGlobalToLocalBegin()
2377 
2378 @*/
2379 PetscErrorCode DMLocalToGlobal(DM dm,Vec l,InsertMode mode,Vec g)
2380 {
2381   PetscErrorCode ierr;
2382 
2383   PetscFunctionBegin;
2384   ierr = DMLocalToGlobalBegin(dm,l,mode,g);CHKERRQ(ierr);
2385   ierr = DMLocalToGlobalEnd(dm,l,mode,g);CHKERRQ(ierr);
2386   PetscFunctionReturn(0);
2387 }
2388 
2389 /*@
2390     DMLocalToGlobalBegin - begins updating global vectors from local vectors
2391 
2392     Neighbor-wise Collective on DM
2393 
2394     Input Parameters:
2395 +   dm - the DM object
2396 .   l - the local vector
2397 .   mode - if INSERT_VALUES then no parallel communication is used, if ADD_VALUES then all ghost points from the same base point accumulate into that base point.
2398 -   g - the global vector
2399 
2400     Notes:
2401     In the ADD_VALUES case you normally would zero the receiving vector before beginning this operation.
2402            INSERT_VALUES is not supported for DMDA, in that case simply compute the values directly into a global vector instead of a local one.
2403 
2404     Level: intermediate
2405 
2406 .seealso DMLocalToGlobal(), DMLocalToGlobalEnd(), DMCoarsen(), DMDestroy(), DMView(), DMCreateGlobalVector(), DMCreateInterpolation(), DMGlobalToLocal(), DMGlobalToLocalEnd(), DMGlobalToLocalBegin()
2407 
2408 @*/
2409 PetscErrorCode  DMLocalToGlobalBegin(DM dm,Vec l,InsertMode mode,Vec g)
2410 {
2411   PetscSF                 sf;
2412   PetscSection            s, gs;
2413   DMLocalToGlobalHookLink link;
2414   const PetscScalar      *lArray;
2415   PetscScalar            *gArray;
2416   PetscBool               isInsert;
2417   PetscErrorCode          ierr;
2418 
2419   PetscFunctionBegin;
2420   PetscValidHeaderSpecific(dm,DM_CLASSID,1);
2421   for (link=dm->ltoghook; link; link=link->next) {
2422     if (link->beginhook) {
2423       ierr = (*link->beginhook)(dm,l,mode,g,link->ctx);CHKERRQ(ierr);
2424     }
2425   }
2426   ierr = DMLocalToGlobalHook_Constraints(dm,l,mode,g,NULL);CHKERRQ(ierr);
2427   ierr = DMGetDefaultSF(dm, &sf);CHKERRQ(ierr);
2428   ierr = DMGetSection(dm, &s);CHKERRQ(ierr);
2429   switch (mode) {
2430   case INSERT_VALUES:
2431   case INSERT_ALL_VALUES:
2432   case INSERT_BC_VALUES:
2433     isInsert = PETSC_TRUE; break;
2434   case ADD_VALUES:
2435   case ADD_ALL_VALUES:
2436   case ADD_BC_VALUES:
2437     isInsert = PETSC_FALSE; break;
2438   default:
2439     SETERRQ1(PetscObjectComm((PetscObject) dm), PETSC_ERR_ARG_OUTOFRANGE, "Invalid insertion mode %D", mode);
2440   }
2441   if (sf && !isInsert) {
2442     ierr = VecGetArrayRead(l, &lArray);CHKERRQ(ierr);
2443     ierr = VecGetArray(g, &gArray);CHKERRQ(ierr);
2444     ierr = PetscSFReduceBegin(sf, MPIU_SCALAR, lArray, gArray, MPIU_SUM);CHKERRQ(ierr);
2445     ierr = VecRestoreArrayRead(l, &lArray);CHKERRQ(ierr);
2446     ierr = VecRestoreArray(g, &gArray);CHKERRQ(ierr);
2447   } else if (s && isInsert) {
2448     PetscInt gStart, pStart, pEnd, p;
2449 
2450     ierr = DMGetGlobalSection(dm, &gs);CHKERRQ(ierr);
2451     ierr = PetscSectionGetChart(s, &pStart, &pEnd);CHKERRQ(ierr);
2452     ierr = VecGetOwnershipRange(g, &gStart, NULL);CHKERRQ(ierr);
2453     ierr = VecGetArrayRead(l, &lArray);CHKERRQ(ierr);
2454     ierr = VecGetArray(g, &gArray);CHKERRQ(ierr);
2455     for (p = pStart; p < pEnd; ++p) {
2456       PetscInt dof, gdof, cdof, gcdof, off, goff, d, e;
2457 
2458       ierr = PetscSectionGetDof(s, p, &dof);CHKERRQ(ierr);
2459       ierr = PetscSectionGetDof(gs, p, &gdof);CHKERRQ(ierr);
2460       ierr = PetscSectionGetConstraintDof(s, p, &cdof);CHKERRQ(ierr);
2461       ierr = PetscSectionGetConstraintDof(gs, p, &gcdof);CHKERRQ(ierr);
2462       ierr = PetscSectionGetOffset(s, p, &off);CHKERRQ(ierr);
2463       ierr = PetscSectionGetOffset(gs, p, &goff);CHKERRQ(ierr);
2464       /* Ignore off-process data and points with no global data */
2465       if (!gdof || goff < 0) continue;
2466       if (dof != gdof) SETERRQ5(PETSC_COMM_SELF, PETSC_ERR_ARG_SIZ, "Inconsistent sizes, p: %d dof: %d gdof: %d cdof: %d gcdof: %d", p, dof, gdof, cdof, gcdof);
2467       /* If no constraints are enforced in the global vector */
2468       if (!gcdof) {
2469         for (d = 0; d < dof; ++d) gArray[goff-gStart+d] = lArray[off+d];
2470       /* If constraints are enforced in the global vector */
2471       } else if (cdof == gcdof) {
2472         const PetscInt *cdofs;
2473         PetscInt        cind = 0;
2474 
2475         ierr = PetscSectionGetConstraintIndices(s, p, &cdofs);CHKERRQ(ierr);
2476         for (d = 0, e = 0; d < dof; ++d) {
2477           if ((cind < cdof) && (d == cdofs[cind])) {++cind; continue;}
2478           gArray[goff-gStart+e++] = lArray[off+d];
2479         }
2480       } else SETERRQ5(PETSC_COMM_SELF, PETSC_ERR_ARG_SIZ, "Inconsistent sizes, p: %d dof: %d gdof: %d cdof: %d gcdof: %d", p, dof, gdof, cdof, gcdof);
2481     }
2482     ierr = VecRestoreArrayRead(l, &lArray);CHKERRQ(ierr);
2483     ierr = VecRestoreArray(g, &gArray);CHKERRQ(ierr);
2484   } else {
2485     ierr = (*dm->ops->localtoglobalbegin)(dm,l,mode == INSERT_ALL_VALUES ? INSERT_VALUES : (mode == ADD_ALL_VALUES ? ADD_VALUES : mode),g);CHKERRQ(ierr);
2486   }
2487   PetscFunctionReturn(0);
2488 }
2489 
2490 /*@
2491     DMLocalToGlobalEnd - updates global vectors from local vectors
2492 
2493     Neighbor-wise Collective on DM
2494 
2495     Input Parameters:
2496 +   dm - the DM object
2497 .   l - the local vector
2498 .   mode - INSERT_VALUES or ADD_VALUES
2499 -   g - the global vector
2500 
2501     Level: intermediate
2502 
2503 .seealso DMCoarsen(), DMDestroy(), DMView(), DMCreateGlobalVector(), DMCreateInterpolation(), DMGlobalToLocalEnd(), DMGlobalToLocalEnd()
2504 
2505 @*/
2506 PetscErrorCode  DMLocalToGlobalEnd(DM dm,Vec l,InsertMode mode,Vec g)
2507 {
2508   PetscSF                 sf;
2509   PetscSection            s;
2510   DMLocalToGlobalHookLink link;
2511   PetscBool               isInsert;
2512   PetscErrorCode          ierr;
2513 
2514   PetscFunctionBegin;
2515   PetscValidHeaderSpecific(dm,DM_CLASSID,1);
2516   ierr = DMGetDefaultSF(dm, &sf);CHKERRQ(ierr);
2517   ierr = DMGetSection(dm, &s);CHKERRQ(ierr);
2518   switch (mode) {
2519   case INSERT_VALUES:
2520   case INSERT_ALL_VALUES:
2521     isInsert = PETSC_TRUE; break;
2522   case ADD_VALUES:
2523   case ADD_ALL_VALUES:
2524     isInsert = PETSC_FALSE; break;
2525   default:
2526     SETERRQ1(PetscObjectComm((PetscObject) dm), PETSC_ERR_ARG_OUTOFRANGE, "Invalid insertion mode %D", mode);
2527   }
2528   if (sf && !isInsert) {
2529     const PetscScalar *lArray;
2530     PetscScalar       *gArray;
2531 
2532     ierr = VecGetArrayRead(l, &lArray);CHKERRQ(ierr);
2533     ierr = VecGetArray(g, &gArray);CHKERRQ(ierr);
2534     ierr = PetscSFReduceEnd(sf, MPIU_SCALAR, lArray, gArray, MPIU_SUM);CHKERRQ(ierr);
2535     ierr = VecRestoreArrayRead(l, &lArray);CHKERRQ(ierr);
2536     ierr = VecRestoreArray(g, &gArray);CHKERRQ(ierr);
2537   } else if (s && isInsert) {
2538   } else {
2539     ierr = (*dm->ops->localtoglobalend)(dm,l,mode == INSERT_ALL_VALUES ? INSERT_VALUES : (mode == ADD_ALL_VALUES ? ADD_VALUES : mode),g);CHKERRQ(ierr);
2540   }
2541   for (link=dm->ltoghook; link; link=link->next) {
2542     if (link->endhook) {ierr = (*link->endhook)(dm,g,mode,l,link->ctx);CHKERRQ(ierr);}
2543   }
2544   PetscFunctionReturn(0);
2545 }
2546 
2547 /*@
2548    DMLocalToLocalBegin - Maps from a local vector (including ghost points
2549    that contain irrelevant values) to another local vector where the ghost
2550    points in the second are set correctly. Must be followed by DMLocalToLocalEnd().
2551 
2552    Neighbor-wise Collective on DM and Vec
2553 
2554    Input Parameters:
2555 +  dm - the DM object
2556 .  g - the original local vector
2557 -  mode - one of INSERT_VALUES or ADD_VALUES
2558 
2559    Output Parameter:
2560 .  l  - the local vector with correct ghost values
2561 
2562    Level: intermediate
2563 
2564    Notes:
2565    The local vectors used here need not be the same as those
2566    obtained from DMCreateLocalVector(), BUT they
2567    must have the same parallel data layout; they could, for example, be
2568    obtained with VecDuplicate() from the DM originating vectors.
2569 
2570 .keywords: DM, local-to-local, begin
2571 .seealso DMCoarsen(), DMDestroy(), DMView(), DMCreateLocalVector(), DMCreateGlobalVector(), DMCreateInterpolation(), DMLocalToLocalEnd(), DMGlobalToLocalEnd(), DMLocalToGlobalBegin()
2572 
2573 @*/
2574 PetscErrorCode  DMLocalToLocalBegin(DM dm,Vec g,InsertMode mode,Vec l)
2575 {
2576   PetscErrorCode          ierr;
2577 
2578   PetscFunctionBegin;
2579   PetscValidHeaderSpecific(dm,DM_CLASSID,1);
2580   if (!dm->ops->localtolocalbegin) SETERRQ(PetscObjectComm((PetscObject)dm),PETSC_ERR_SUP,"This DM does not support local to local maps");
2581   ierr = (*dm->ops->localtolocalbegin)(dm,g,mode == INSERT_ALL_VALUES ? INSERT_VALUES : (mode == ADD_ALL_VALUES ? ADD_VALUES : mode),l);CHKERRQ(ierr);
2582   PetscFunctionReturn(0);
2583 }
2584 
2585 /*@
2586    DMLocalToLocalEnd - Maps from a local vector (including ghost points
2587    that contain irrelevant values) to another local vector where the ghost
2588    points in the second are set correctly. Must be preceded by DMLocalToLocalBegin().
2589 
2590    Neighbor-wise Collective on DM and Vec
2591 
2592    Input Parameters:
2593 +  da - the DM object
2594 .  g - the original local vector
2595 -  mode - one of INSERT_VALUES or ADD_VALUES
2596 
2597    Output Parameter:
2598 .  l  - the local vector with correct ghost values
2599 
2600    Level: intermediate
2601 
2602    Notes:
2603    The local vectors used here need not be the same as those
2604    obtained from DMCreateLocalVector(), BUT they
2605    must have the same parallel data layout; they could, for example, be
2606    obtained with VecDuplicate() from the DM originating vectors.
2607 
2608 .keywords: DM, local-to-local, end
2609 .seealso DMCoarsen(), DMDestroy(), DMView(), DMCreateLocalVector(), DMCreateGlobalVector(), DMCreateInterpolation(), DMLocalToLocalBegin(), DMGlobalToLocalEnd(), DMLocalToGlobalBegin()
2610 
2611 @*/
2612 PetscErrorCode  DMLocalToLocalEnd(DM dm,Vec g,InsertMode mode,Vec l)
2613 {
2614   PetscErrorCode          ierr;
2615 
2616   PetscFunctionBegin;
2617   PetscValidHeaderSpecific(dm,DM_CLASSID,1);
2618   if (!dm->ops->localtolocalend) SETERRQ(PetscObjectComm((PetscObject)dm),PETSC_ERR_SUP,"This DM does not support local to local maps");
2619   ierr = (*dm->ops->localtolocalend)(dm,g,mode == INSERT_ALL_VALUES ? INSERT_VALUES : (mode == ADD_ALL_VALUES ? ADD_VALUES : mode),l);CHKERRQ(ierr);
2620   PetscFunctionReturn(0);
2621 }
2622 
2623 
2624 /*@
2625     DMCoarsen - Coarsens a DM object
2626 
2627     Collective on DM
2628 
2629     Input Parameter:
2630 +   dm - the DM object
2631 -   comm - the communicator to contain the new DM object (or MPI_COMM_NULL)
2632 
2633     Output Parameter:
2634 .   dmc - the coarsened DM
2635 
2636     Level: developer
2637 
2638 .seealso DMRefine(), DMDestroy(), DMView(), DMCreateGlobalVector(), DMCreateInterpolation()
2639 
2640 @*/
2641 PetscErrorCode DMCoarsen(DM dm, MPI_Comm comm, DM *dmc)
2642 {
2643   PetscErrorCode    ierr;
2644   DMCoarsenHookLink link;
2645 
2646   PetscFunctionBegin;
2647   PetscValidHeaderSpecific(dm,DM_CLASSID,1);
2648   if (!dm->ops->coarsen) SETERRQ(PetscObjectComm((PetscObject)dm),PETSC_ERR_SUP,"This DM cannot coarsen");
2649   ierr = PetscLogEventBegin(DM_Coarsen,dm,0,0,0);CHKERRQ(ierr);
2650   ierr                      = (*dm->ops->coarsen)(dm, comm, dmc);CHKERRQ(ierr);
2651   if (!(*dmc)) SETERRQ(PETSC_COMM_SELF, PETSC_ERR_ARG_WRONG, "NULL coarse mesh produced");
2652   ierr = DMSetCoarseDM(dm,*dmc);CHKERRQ(ierr);
2653   (*dmc)->ops->creatematrix = dm->ops->creatematrix;
2654   ierr                      = PetscObjectCopyFortranFunctionPointers((PetscObject)dm,(PetscObject)*dmc);CHKERRQ(ierr);
2655   (*dmc)->ctx               = dm->ctx;
2656   (*dmc)->levelup           = dm->levelup;
2657   (*dmc)->leveldown         = dm->leveldown + 1;
2658   ierr                      = DMSetMatType(*dmc,dm->mattype);CHKERRQ(ierr);
2659   for (link=dm->coarsenhook; link; link=link->next) {
2660     if (link->coarsenhook) {ierr = (*link->coarsenhook)(dm,*dmc,link->ctx);CHKERRQ(ierr);}
2661   }
2662   ierr = PetscLogEventEnd(DM_Coarsen,dm,0,0,0);CHKERRQ(ierr);
2663   PetscFunctionReturn(0);
2664 }
2665 
2666 /*@C
2667    DMCoarsenHookAdd - adds a callback to be run when restricting a nonlinear problem to the coarse grid
2668 
2669    Logically Collective
2670 
2671    Input Arguments:
2672 +  fine - nonlinear solver context on which to run a hook when restricting to a coarser level
2673 .  coarsenhook - function to run when setting up a coarser level
2674 .  restricthook - function to run to update data on coarser levels (once per SNESSolve())
2675 -  ctx - [optional] user-defined context for provide data for the hooks (may be NULL)
2676 
2677    Calling sequence of coarsenhook:
2678 $    coarsenhook(DM fine,DM coarse,void *ctx);
2679 
2680 +  fine - fine level DM
2681 .  coarse - coarse level DM to restrict problem to
2682 -  ctx - optional user-defined function context
2683 
2684    Calling sequence for restricthook:
2685 $    restricthook(DM fine,Mat mrestrict,Vec rscale,Mat inject,DM coarse,void *ctx)
2686 
2687 +  fine - fine level DM
2688 .  mrestrict - matrix restricting a fine-level solution to the coarse grid
2689 .  rscale - scaling vector for restriction
2690 .  inject - matrix restricting by injection
2691 .  coarse - coarse level DM to update
2692 -  ctx - optional user-defined function context
2693 
2694    Level: advanced
2695 
2696    Notes:
2697    This function is only needed if auxiliary data needs to be set up on coarse grids.
2698 
2699    If this function is called multiple times, the hooks will be run in the order they are added.
2700 
2701    In order to compose with nonlinear preconditioning without duplicating storage, the hook should be implemented to
2702    extract the finest level information from its context (instead of from the SNES).
2703 
2704    This function is currently not available from Fortran.
2705 
2706 .seealso: DMCoarsenHookRemove(), DMRefineHookAdd(), SNESFASGetInterpolation(), SNESFASGetInjection(), PetscObjectCompose(), PetscContainerCreate()
2707 @*/
2708 PetscErrorCode DMCoarsenHookAdd(DM fine,PetscErrorCode (*coarsenhook)(DM,DM,void*),PetscErrorCode (*restricthook)(DM,Mat,Vec,Mat,DM,void*),void *ctx)
2709 {
2710   PetscErrorCode    ierr;
2711   DMCoarsenHookLink link,*p;
2712 
2713   PetscFunctionBegin;
2714   PetscValidHeaderSpecific(fine,DM_CLASSID,1);
2715   for (p=&fine->coarsenhook; *p; p=&(*p)->next) { /* Scan to the end of the current list of hooks */
2716     if ((*p)->coarsenhook == coarsenhook && (*p)->restricthook == restricthook && (*p)->ctx == ctx) PetscFunctionReturn(0);
2717   }
2718   ierr               = PetscNew(&link);CHKERRQ(ierr);
2719   link->coarsenhook  = coarsenhook;
2720   link->restricthook = restricthook;
2721   link->ctx          = ctx;
2722   link->next         = NULL;
2723   *p                 = link;
2724   PetscFunctionReturn(0);
2725 }
2726 
2727 /*@C
2728    DMCoarsenHookRemove - remove a callback from the list of hooks to be run when restricting a nonlinear problem to the coarse grid
2729 
2730    Logically Collective
2731 
2732    Input Arguments:
2733 +  fine - nonlinear solver context on which to run a hook when restricting to a coarser level
2734 .  coarsenhook - function to run when setting up a coarser level
2735 .  restricthook - function to run to update data on coarser levels (once per SNESSolve())
2736 -  ctx - [optional] user-defined context for provide data for the hooks (may be NULL)
2737 
2738    Level: advanced
2739 
2740    Notes:
2741    This function does nothing if the hook is not in the list.
2742 
2743    This function is currently not available from Fortran.
2744 
2745 .seealso: DMCoarsenHookAdd(), DMRefineHookAdd(), SNESFASGetInterpolation(), SNESFASGetInjection(), PetscObjectCompose(), PetscContainerCreate()
2746 @*/
2747 PetscErrorCode DMCoarsenHookRemove(DM fine,PetscErrorCode (*coarsenhook)(DM,DM,void*),PetscErrorCode (*restricthook)(DM,Mat,Vec,Mat,DM,void*),void *ctx)
2748 {
2749   PetscErrorCode    ierr;
2750   DMCoarsenHookLink link,*p;
2751 
2752   PetscFunctionBegin;
2753   PetscValidHeaderSpecific(fine,DM_CLASSID,1);
2754   for (p=&fine->coarsenhook; *p; p=&(*p)->next) { /* Search the list of current hooks */
2755     if ((*p)->coarsenhook == coarsenhook && (*p)->restricthook == restricthook && (*p)->ctx == ctx) {
2756       link = *p;
2757       *p = link->next;
2758       ierr = PetscFree(link);CHKERRQ(ierr);
2759       break;
2760     }
2761   }
2762   PetscFunctionReturn(0);
2763 }
2764 
2765 
2766 /*@
2767    DMRestrict - restricts user-defined problem data to a coarser DM by running hooks registered by DMCoarsenHookAdd()
2768 
2769    Collective if any hooks are
2770 
2771    Input Arguments:
2772 +  fine - finer DM to use as a base
2773 .  restrct - restriction matrix, apply using MatRestrict()
2774 .  rscale - scaling vector for restriction
2775 .  inject - injection matrix, also use MatRestrict()
2776 -  coarse - coarser DM to update
2777 
2778    Level: developer
2779 
2780 .seealso: DMCoarsenHookAdd(), MatRestrict()
2781 @*/
2782 PetscErrorCode DMRestrict(DM fine,Mat restrct,Vec rscale,Mat inject,DM coarse)
2783 {
2784   PetscErrorCode    ierr;
2785   DMCoarsenHookLink link;
2786 
2787   PetscFunctionBegin;
2788   for (link=fine->coarsenhook; link; link=link->next) {
2789     if (link->restricthook) {
2790       ierr = (*link->restricthook)(fine,restrct,rscale,inject,coarse,link->ctx);CHKERRQ(ierr);
2791     }
2792   }
2793   PetscFunctionReturn(0);
2794 }
2795 
2796 /*@C
2797    DMSubDomainHookAdd - adds a callback to be run when restricting a problem to the coarse grid
2798 
2799    Logically Collective
2800 
2801    Input Arguments:
2802 +  global - global DM
2803 .  ddhook - function to run to pass data to the decomposition DM upon its creation
2804 .  restricthook - function to run to update data on block solve (at the beginning of the block solve)
2805 -  ctx - [optional] user-defined context for provide data for the hooks (may be NULL)
2806 
2807 
2808    Calling sequence for ddhook:
2809 $    ddhook(DM global,DM block,void *ctx)
2810 
2811 +  global - global DM
2812 .  block  - block DM
2813 -  ctx - optional user-defined function context
2814 
2815    Calling sequence for restricthook:
2816 $    restricthook(DM global,VecScatter out,VecScatter in,DM block,void *ctx)
2817 
2818 +  global - global DM
2819 .  out    - scatter to the outer (with ghost and overlap points) block vector
2820 .  in     - scatter to block vector values only owned locally
2821 .  block  - block DM
2822 -  ctx - optional user-defined function context
2823 
2824    Level: advanced
2825 
2826    Notes:
2827    This function is only needed if auxiliary data needs to be set up on subdomain DMs.
2828 
2829    If this function is called multiple times, the hooks will be run in the order they are added.
2830 
2831    In order to compose with nonlinear preconditioning without duplicating storage, the hook should be implemented to
2832    extract the global information from its context (instead of from the SNES).
2833 
2834    This function is currently not available from Fortran.
2835 
2836 .seealso: DMRefineHookAdd(), SNESFASGetInterpolation(), SNESFASGetInjection(), PetscObjectCompose(), PetscContainerCreate()
2837 @*/
2838 PetscErrorCode DMSubDomainHookAdd(DM global,PetscErrorCode (*ddhook)(DM,DM,void*),PetscErrorCode (*restricthook)(DM,VecScatter,VecScatter,DM,void*),void *ctx)
2839 {
2840   PetscErrorCode      ierr;
2841   DMSubDomainHookLink link,*p;
2842 
2843   PetscFunctionBegin;
2844   PetscValidHeaderSpecific(global,DM_CLASSID,1);
2845   for (p=&global->subdomainhook; *p; p=&(*p)->next) { /* Scan to the end of the current list of hooks */
2846     if ((*p)->ddhook == ddhook && (*p)->restricthook == restricthook && (*p)->ctx == ctx) PetscFunctionReturn(0);
2847   }
2848   ierr               = PetscNew(&link);CHKERRQ(ierr);
2849   link->restricthook = restricthook;
2850   link->ddhook       = ddhook;
2851   link->ctx          = ctx;
2852   link->next         = NULL;
2853   *p                 = link;
2854   PetscFunctionReturn(0);
2855 }
2856 
2857 /*@C
2858    DMSubDomainHookRemove - remove a callback from the list to be run when restricting a problem to the coarse grid
2859 
2860    Logically Collective
2861 
2862    Input Arguments:
2863 +  global - global DM
2864 .  ddhook - function to run to pass data to the decomposition DM upon its creation
2865 .  restricthook - function to run to update data on block solve (at the beginning of the block solve)
2866 -  ctx - [optional] user-defined context for provide data for the hooks (may be NULL)
2867 
2868    Level: advanced
2869 
2870    Notes:
2871 
2872    This function is currently not available from Fortran.
2873 
2874 .seealso: DMSubDomainHookAdd(), SNESFASGetInterpolation(), SNESFASGetInjection(), PetscObjectCompose(), PetscContainerCreate()
2875 @*/
2876 PetscErrorCode DMSubDomainHookRemove(DM global,PetscErrorCode (*ddhook)(DM,DM,void*),PetscErrorCode (*restricthook)(DM,VecScatter,VecScatter,DM,void*),void *ctx)
2877 {
2878   PetscErrorCode      ierr;
2879   DMSubDomainHookLink link,*p;
2880 
2881   PetscFunctionBegin;
2882   PetscValidHeaderSpecific(global,DM_CLASSID,1);
2883   for (p=&global->subdomainhook; *p; p=&(*p)->next) { /* Search the list of current hooks */
2884     if ((*p)->ddhook == ddhook && (*p)->restricthook == restricthook && (*p)->ctx == ctx) {
2885       link = *p;
2886       *p = link->next;
2887       ierr = PetscFree(link);CHKERRQ(ierr);
2888       break;
2889     }
2890   }
2891   PetscFunctionReturn(0);
2892 }
2893 
2894 /*@
2895    DMSubDomainRestrict - restricts user-defined problem data to a block DM by running hooks registered by DMSubDomainHookAdd()
2896 
2897    Collective if any hooks are
2898 
2899    Input Arguments:
2900 +  fine - finer DM to use as a base
2901 .  oscatter - scatter from domain global vector filling subdomain global vector with overlap
2902 .  gscatter - scatter from domain global vector filling subdomain local vector with ghosts
2903 -  coarse - coarer DM to update
2904 
2905    Level: developer
2906 
2907 .seealso: DMCoarsenHookAdd(), MatRestrict()
2908 @*/
2909 PetscErrorCode DMSubDomainRestrict(DM global,VecScatter oscatter,VecScatter gscatter,DM subdm)
2910 {
2911   PetscErrorCode      ierr;
2912   DMSubDomainHookLink link;
2913 
2914   PetscFunctionBegin;
2915   for (link=global->subdomainhook; link; link=link->next) {
2916     if (link->restricthook) {
2917       ierr = (*link->restricthook)(global,oscatter,gscatter,subdm,link->ctx);CHKERRQ(ierr);
2918     }
2919   }
2920   PetscFunctionReturn(0);
2921 }
2922 
2923 /*@
2924     DMGetCoarsenLevel - Get's the number of coarsenings that have generated this DM.
2925 
2926     Not Collective
2927 
2928     Input Parameter:
2929 .   dm - the DM object
2930 
2931     Output Parameter:
2932 .   level - number of coarsenings
2933 
2934     Level: developer
2935 
2936 .seealso DMCoarsen(), DMGetRefineLevel(), DMDestroy(), DMView(), DMCreateGlobalVector(), DMCreateInterpolation()
2937 
2938 @*/
2939 PetscErrorCode  DMGetCoarsenLevel(DM dm,PetscInt *level)
2940 {
2941   PetscFunctionBegin;
2942   PetscValidHeaderSpecific(dm,DM_CLASSID,1);
2943   *level = dm->leveldown;
2944   PetscFunctionReturn(0);
2945 }
2946 
2947 /*@
2948     DMSetCoarsenLevel - Sets the number of coarsenings that have generated this DM.
2949 
2950     Not Collective
2951 
2952     Input Parameters:
2953 +   dm - the DM object
2954 -   level - number of coarsenings
2955 
2956     Level: developer
2957 
2958 .seealso DMCoarsen(), DMGetCoarsenLevel(), DMGetRefineLevel(), DMDestroy(), DMView(), DMCreateGlobalVector(), DMCreateInterpolation()
2959 @*/
2960 PetscErrorCode DMSetCoarsenLevel(DM dm,PetscInt level)
2961 {
2962   PetscFunctionBegin;
2963   PetscValidHeaderSpecific(dm,DM_CLASSID,1);
2964   dm->leveldown = level;
2965   PetscFunctionReturn(0);
2966 }
2967 
2968 
2969 
2970 /*@C
2971     DMRefineHierarchy - Refines a DM object, all levels at once
2972 
2973     Collective on DM
2974 
2975     Input Parameter:
2976 +   dm - the DM object
2977 -   nlevels - the number of levels of refinement
2978 
2979     Output Parameter:
2980 .   dmf - the refined DM hierarchy
2981 
2982     Level: developer
2983 
2984 .seealso DMCoarsenHierarchy(), DMDestroy(), DMView(), DMCreateGlobalVector(), DMCreateInterpolation()
2985 
2986 @*/
2987 PetscErrorCode  DMRefineHierarchy(DM dm,PetscInt nlevels,DM dmf[])
2988 {
2989   PetscErrorCode ierr;
2990 
2991   PetscFunctionBegin;
2992   PetscValidHeaderSpecific(dm,DM_CLASSID,1);
2993   if (nlevels < 0) SETERRQ(PetscObjectComm((PetscObject)dm),PETSC_ERR_ARG_OUTOFRANGE,"nlevels cannot be negative");
2994   if (nlevels == 0) PetscFunctionReturn(0);
2995   if (dm->ops->refinehierarchy) {
2996     ierr = (*dm->ops->refinehierarchy)(dm,nlevels,dmf);CHKERRQ(ierr);
2997   } else if (dm->ops->refine) {
2998     PetscInt i;
2999 
3000     ierr = DMRefine(dm,PetscObjectComm((PetscObject)dm),&dmf[0]);CHKERRQ(ierr);
3001     for (i=1; i<nlevels; i++) {
3002       ierr = DMRefine(dmf[i-1],PetscObjectComm((PetscObject)dm),&dmf[i]);CHKERRQ(ierr);
3003     }
3004   } else SETERRQ(PetscObjectComm((PetscObject)dm),PETSC_ERR_SUP,"No RefineHierarchy for this DM yet");
3005   PetscFunctionReturn(0);
3006 }
3007 
3008 /*@C
3009     DMCoarsenHierarchy - Coarsens a DM object, all levels at once
3010 
3011     Collective on DM
3012 
3013     Input Parameter:
3014 +   dm - the DM object
3015 -   nlevels - the number of levels of coarsening
3016 
3017     Output Parameter:
3018 .   dmc - the coarsened DM hierarchy
3019 
3020     Level: developer
3021 
3022 .seealso DMRefineHierarchy(), DMDestroy(), DMView(), DMCreateGlobalVector(), DMCreateInterpolation()
3023 
3024 @*/
3025 PetscErrorCode  DMCoarsenHierarchy(DM dm, PetscInt nlevels, DM dmc[])
3026 {
3027   PetscErrorCode ierr;
3028 
3029   PetscFunctionBegin;
3030   PetscValidHeaderSpecific(dm,DM_CLASSID,1);
3031   if (nlevels < 0) SETERRQ(PetscObjectComm((PetscObject)dm),PETSC_ERR_ARG_OUTOFRANGE,"nlevels cannot be negative");
3032   if (nlevels == 0) PetscFunctionReturn(0);
3033   PetscValidPointer(dmc,3);
3034   if (dm->ops->coarsenhierarchy) {
3035     ierr = (*dm->ops->coarsenhierarchy)(dm, nlevels, dmc);CHKERRQ(ierr);
3036   } else if (dm->ops->coarsen) {
3037     PetscInt i;
3038 
3039     ierr = DMCoarsen(dm,PetscObjectComm((PetscObject)dm),&dmc[0]);CHKERRQ(ierr);
3040     for (i=1; i<nlevels; i++) {
3041       ierr = DMCoarsen(dmc[i-1],PetscObjectComm((PetscObject)dm),&dmc[i]);CHKERRQ(ierr);
3042     }
3043   } else SETERRQ(PetscObjectComm((PetscObject)dm),PETSC_ERR_SUP,"No CoarsenHierarchy for this DM yet");
3044   PetscFunctionReturn(0);
3045 }
3046 
3047 /*@
3048    DMCreateAggregates - Gets the aggregates that map between
3049    grids associated with two DMs.
3050 
3051    Collective on DM
3052 
3053    Input Parameters:
3054 +  dmc - the coarse grid DM
3055 -  dmf - the fine grid DM
3056 
3057    Output Parameters:
3058 .  rest - the restriction matrix (transpose of the projection matrix)
3059 
3060    Level: intermediate
3061 
3062 .keywords: interpolation, restriction, multigrid
3063 
3064 .seealso: DMRefine(), DMCreateInjection(), DMCreateInterpolation()
3065 @*/
3066 PetscErrorCode  DMCreateAggregates(DM dmc, DM dmf, Mat *rest)
3067 {
3068   PetscErrorCode ierr;
3069 
3070   PetscFunctionBegin;
3071   PetscValidHeaderSpecific(dmc,DM_CLASSID,1);
3072   PetscValidHeaderSpecific(dmf,DM_CLASSID,2);
3073   ierr = (*dmc->ops->getaggregates)(dmc, dmf, rest);CHKERRQ(ierr);
3074   PetscFunctionReturn(0);
3075 }
3076 
3077 /*@C
3078     DMSetApplicationContextDestroy - Sets a user function that will be called to destroy the application context when the DM is destroyed
3079 
3080     Not Collective
3081 
3082     Input Parameters:
3083 +   dm - the DM object
3084 -   destroy - the destroy function
3085 
3086     Level: intermediate
3087 
3088 .seealso DMView(), DMCreateGlobalVector(), DMCreateInterpolation(), DMCreateColoring(), DMCreateMatrix(), DMGetApplicationContext()
3089 
3090 @*/
3091 PetscErrorCode  DMSetApplicationContextDestroy(DM dm,PetscErrorCode (*destroy)(void**))
3092 {
3093   PetscFunctionBegin;
3094   PetscValidHeaderSpecific(dm,DM_CLASSID,1);
3095   dm->ctxdestroy = destroy;
3096   PetscFunctionReturn(0);
3097 }
3098 
3099 /*@
3100     DMSetApplicationContext - Set a user context into a DM object
3101 
3102     Not Collective
3103 
3104     Input Parameters:
3105 +   dm - the DM object
3106 -   ctx - the user context
3107 
3108     Level: intermediate
3109 
3110 .seealso DMView(), DMCreateGlobalVector(), DMCreateInterpolation(), DMCreateColoring(), DMCreateMatrix(), DMGetApplicationContext()
3111 
3112 @*/
3113 PetscErrorCode  DMSetApplicationContext(DM dm,void *ctx)
3114 {
3115   PetscFunctionBegin;
3116   PetscValidHeaderSpecific(dm,DM_CLASSID,1);
3117   dm->ctx = ctx;
3118   PetscFunctionReturn(0);
3119 }
3120 
3121 /*@
3122     DMGetApplicationContext - Gets a user context from a DM object
3123 
3124     Not Collective
3125 
3126     Input Parameter:
3127 .   dm - the DM object
3128 
3129     Output Parameter:
3130 .   ctx - the user context
3131 
3132     Level: intermediate
3133 
3134 .seealso DMView(), DMCreateGlobalVector(), DMCreateInterpolation(), DMCreateColoring(), DMCreateMatrix(), DMGetApplicationContext()
3135 
3136 @*/
3137 PetscErrorCode  DMGetApplicationContext(DM dm,void *ctx)
3138 {
3139   PetscFunctionBegin;
3140   PetscValidHeaderSpecific(dm,DM_CLASSID,1);
3141   *(void**)ctx = dm->ctx;
3142   PetscFunctionReturn(0);
3143 }
3144 
3145 /*@C
3146     DMSetVariableBounds - sets a function to compute the lower and upper bound vectors for SNESVI.
3147 
3148     Logically Collective on DM
3149 
3150     Input Parameter:
3151 +   dm - the DM object
3152 -   f - the function that computes variable bounds used by SNESVI (use NULL to cancel a previous function that was set)
3153 
3154     Level: intermediate
3155 
3156 .seealso DMView(), DMCreateGlobalVector(), DMCreateInterpolation(), DMCreateColoring(), DMCreateMatrix(), DMGetApplicationContext(),
3157          DMSetJacobian()
3158 
3159 @*/
3160 PetscErrorCode  DMSetVariableBounds(DM dm,PetscErrorCode (*f)(DM,Vec,Vec))
3161 {
3162   PetscFunctionBegin;
3163   dm->ops->computevariablebounds = f;
3164   PetscFunctionReturn(0);
3165 }
3166 
3167 /*@
3168     DMHasVariableBounds - does the DM object have a variable bounds function?
3169 
3170     Not Collective
3171 
3172     Input Parameter:
3173 .   dm - the DM object to destroy
3174 
3175     Output Parameter:
3176 .   flg - PETSC_TRUE if the variable bounds function exists
3177 
3178     Level: developer
3179 
3180 .seealso DMView(), DMCreateGlobalVector(), DMCreateInterpolation(), DMCreateColoring(), DMCreateMatrix(), DMGetApplicationContext()
3181 
3182 @*/
3183 PetscErrorCode  DMHasVariableBounds(DM dm,PetscBool  *flg)
3184 {
3185   PetscFunctionBegin;
3186   *flg =  (dm->ops->computevariablebounds) ? PETSC_TRUE : PETSC_FALSE;
3187   PetscFunctionReturn(0);
3188 }
3189 
3190 /*@C
3191     DMComputeVariableBounds - compute variable bounds used by SNESVI.
3192 
3193     Logically Collective on DM
3194 
3195     Input Parameters:
3196 .   dm - the DM object
3197 
3198     Output parameters:
3199 +   xl - lower bound
3200 -   xu - upper bound
3201 
3202     Level: advanced
3203 
3204     Notes:
3205     This is generally not called by users. It calls the function provided by the user with DMSetVariableBounds()
3206 
3207 .seealso DMView(), DMCreateGlobalVector(), DMCreateInterpolation(), DMCreateColoring(), DMCreateMatrix(), DMGetApplicationContext()
3208 
3209 @*/
3210 PetscErrorCode  DMComputeVariableBounds(DM dm, Vec xl, Vec xu)
3211 {
3212   PetscErrorCode ierr;
3213 
3214   PetscFunctionBegin;
3215   PetscValidHeaderSpecific(xl,VEC_CLASSID,2);
3216   PetscValidHeaderSpecific(xu,VEC_CLASSID,2);
3217   if (dm->ops->computevariablebounds) {
3218     ierr = (*dm->ops->computevariablebounds)(dm, xl,xu);CHKERRQ(ierr);
3219   } else SETERRQ(PETSC_COMM_SELF, PETSC_ERR_ARG_WRONGSTATE, "This DM is incapable of computing variable bounds.");
3220   PetscFunctionReturn(0);
3221 }
3222 
3223 /*@
3224     DMHasColoring - does the DM object have a method of providing a coloring?
3225 
3226     Not Collective
3227 
3228     Input Parameter:
3229 .   dm - the DM object
3230 
3231     Output Parameter:
3232 .   flg - PETSC_TRUE if the DM has facilities for DMCreateColoring().
3233 
3234     Level: developer
3235 
3236 .seealso DMHasFunction(), DMCreateColoring()
3237 
3238 @*/
3239 PetscErrorCode  DMHasColoring(DM dm,PetscBool  *flg)
3240 {
3241   PetscFunctionBegin;
3242   *flg =  (dm->ops->getcoloring) ? PETSC_TRUE : PETSC_FALSE;
3243   PetscFunctionReturn(0);
3244 }
3245 
3246 /*@
3247     DMHasCreateRestriction - does the DM object have a method of providing a restriction?
3248 
3249     Not Collective
3250 
3251     Input Parameter:
3252 .   dm - the DM object
3253 
3254     Output Parameter:
3255 .   flg - PETSC_TRUE if the DM has facilities for DMCreateRestriction().
3256 
3257     Level: developer
3258 
3259 .seealso DMHasFunction(), DMCreateRestriction()
3260 
3261 @*/
3262 PetscErrorCode  DMHasCreateRestriction(DM dm,PetscBool  *flg)
3263 {
3264   PetscFunctionBegin;
3265   *flg =  (dm->ops->createrestriction) ? PETSC_TRUE : PETSC_FALSE;
3266   PetscFunctionReturn(0);
3267 }
3268 
3269 
3270 /*@
3271     DMHasCreateInjection - does the DM object have a method of providing an injection?
3272 
3273     Not Collective
3274 
3275     Input Parameter:
3276 .   dm - the DM object
3277 
3278     Output Parameter:
3279 .   flg - PETSC_TRUE if the DM has facilities for DMCreateInjection().
3280 
3281     Level: developer
3282 
3283 .seealso DMHasFunction(), DMCreateInjection()
3284 
3285 @*/
3286 PetscErrorCode  DMHasCreateInjection(DM dm,PetscBool  *flg)
3287 {
3288   PetscErrorCode ierr;
3289   PetscFunctionBegin;
3290   if (!dm->ops->hascreateinjection) SETERRQ(PetscObjectComm((PetscObject)dm),PETSC_ERR_SUP,"DMHasCreateInjection not implemented for this type");
3291   ierr = (*dm->ops->hascreateinjection)(dm,flg);CHKERRQ(ierr);
3292   PetscFunctionReturn(0);
3293 }
3294 
3295 /*@C
3296     DMSetVec - set the vector at which to compute residual, Jacobian and VI bounds, if the problem is nonlinear.
3297 
3298     Collective on DM
3299 
3300     Input Parameter:
3301 +   dm - the DM object
3302 -   x - location to compute residual and Jacobian, if NULL is passed to those routines; will be NULL for linear problems.
3303 
3304     Level: developer
3305 
3306 .seealso DMView(), DMCreateGlobalVector(), DMCreateInterpolation(), DMCreateColoring(), DMCreateMatrix(), DMGetApplicationContext()
3307 
3308 @*/
3309 PetscErrorCode  DMSetVec(DM dm,Vec x)
3310 {
3311   PetscErrorCode ierr;
3312 
3313   PetscFunctionBegin;
3314   if (x) {
3315     if (!dm->x) {
3316       ierr = DMCreateGlobalVector(dm,&dm->x);CHKERRQ(ierr);
3317     }
3318     ierr = VecCopy(x,dm->x);CHKERRQ(ierr);
3319   } else if (dm->x) {
3320     ierr = VecDestroy(&dm->x);CHKERRQ(ierr);
3321   }
3322   PetscFunctionReturn(0);
3323 }
3324 
3325 PetscFunctionList DMList              = NULL;
3326 PetscBool         DMRegisterAllCalled = PETSC_FALSE;
3327 
3328 /*@C
3329   DMSetType - Builds a DM, for a particular DM implementation.
3330 
3331   Collective on DM
3332 
3333   Input Parameters:
3334 + dm     - The DM object
3335 - method - The name of the DM type
3336 
3337   Options Database Key:
3338 . -dm_type <type> - Sets the DM type; use -help for a list of available types
3339 
3340   Notes:
3341   See "petsc/include/petscdm.h" for available DM types (for instance, DM1D, DM2D, or DM3D).
3342 
3343   Level: intermediate
3344 
3345 .keywords: DM, set, type
3346 .seealso: DMGetType(), DMCreate()
3347 @*/
3348 PetscErrorCode  DMSetType(DM dm, DMType method)
3349 {
3350   PetscErrorCode (*r)(DM);
3351   PetscBool      match;
3352   PetscErrorCode ierr;
3353 
3354   PetscFunctionBegin;
3355   PetscValidHeaderSpecific(dm, DM_CLASSID,1);
3356   ierr = PetscObjectTypeCompare((PetscObject) dm, method, &match);CHKERRQ(ierr);
3357   if (match) PetscFunctionReturn(0);
3358 
3359   ierr = DMRegisterAll();CHKERRQ(ierr);
3360   ierr = PetscFunctionListFind(DMList,method,&r);CHKERRQ(ierr);
3361   if (!r) SETERRQ1(PetscObjectComm((PetscObject)dm),PETSC_ERR_ARG_UNKNOWN_TYPE, "Unknown DM type: %s", method);
3362 
3363   if (dm->ops->destroy) {
3364     ierr             = (*dm->ops->destroy)(dm);CHKERRQ(ierr);
3365     dm->ops->destroy = NULL;
3366   }
3367   ierr = (*r)(dm);CHKERRQ(ierr);
3368   ierr = PetscObjectChangeTypeName((PetscObject)dm,method);CHKERRQ(ierr);
3369   PetscFunctionReturn(0);
3370 }
3371 
3372 /*@C
3373   DMGetType - Gets the DM type name (as a string) from the DM.
3374 
3375   Not Collective
3376 
3377   Input Parameter:
3378 . dm  - The DM
3379 
3380   Output Parameter:
3381 . type - The DM type name
3382 
3383   Level: intermediate
3384 
3385 .keywords: DM, get, type, name
3386 .seealso: DMSetType(), DMCreate()
3387 @*/
3388 PetscErrorCode  DMGetType(DM dm, DMType *type)
3389 {
3390   PetscErrorCode ierr;
3391 
3392   PetscFunctionBegin;
3393   PetscValidHeaderSpecific(dm, DM_CLASSID,1);
3394   PetscValidPointer(type,2);
3395   ierr = DMRegisterAll();CHKERRQ(ierr);
3396   *type = ((PetscObject)dm)->type_name;
3397   PetscFunctionReturn(0);
3398 }
3399 
3400 /*@C
3401   DMConvert - Converts a DM to another DM, either of the same or different type.
3402 
3403   Collective on DM
3404 
3405   Input Parameters:
3406 + dm - the DM
3407 - newtype - new DM type (use "same" for the same type)
3408 
3409   Output Parameter:
3410 . M - pointer to new DM
3411 
3412   Notes:
3413   Cannot be used to convert a sequential DM to parallel or parallel to sequential,
3414   the MPI communicator of the generated DM is always the same as the communicator
3415   of the input DM.
3416 
3417   Level: intermediate
3418 
3419 .seealso: DMCreate()
3420 @*/
3421 PetscErrorCode DMConvert(DM dm, DMType newtype, DM *M)
3422 {
3423   DM             B;
3424   char           convname[256];
3425   PetscBool      sametype/*, issame */;
3426   PetscErrorCode ierr;
3427 
3428   PetscFunctionBegin;
3429   PetscValidHeaderSpecific(dm,DM_CLASSID,1);
3430   PetscValidType(dm,1);
3431   PetscValidPointer(M,3);
3432   ierr = PetscObjectTypeCompare((PetscObject) dm, newtype, &sametype);CHKERRQ(ierr);
3433   /* ierr = PetscStrcmp(newtype, "same", &issame);CHKERRQ(ierr); */
3434   if (sametype) {
3435     *M   = dm;
3436     ierr = PetscObjectReference((PetscObject) dm);CHKERRQ(ierr);
3437     PetscFunctionReturn(0);
3438   } else {
3439     PetscErrorCode (*conv)(DM, DMType, DM*) = NULL;
3440 
3441     /*
3442        Order of precedence:
3443        1) See if a specialized converter is known to the current DM.
3444        2) See if a specialized converter is known to the desired DM class.
3445        3) See if a good general converter is registered for the desired class
3446        4) See if a good general converter is known for the current matrix.
3447        5) Use a really basic converter.
3448     */
3449 
3450     /* 1) See if a specialized converter is known to the current DM and the desired class */
3451     ierr = PetscStrncpy(convname,"DMConvert_",sizeof(convname));CHKERRQ(ierr);
3452     ierr = PetscStrlcat(convname,((PetscObject) dm)->type_name,sizeof(convname));CHKERRQ(ierr);
3453     ierr = PetscStrlcat(convname,"_",sizeof(convname));CHKERRQ(ierr);
3454     ierr = PetscStrlcat(convname,newtype,sizeof(convname));CHKERRQ(ierr);
3455     ierr = PetscStrlcat(convname,"_C",sizeof(convname));CHKERRQ(ierr);
3456     ierr = PetscObjectQueryFunction((PetscObject)dm,convname,&conv);CHKERRQ(ierr);
3457     if (conv) goto foundconv;
3458 
3459     /* 2)  See if a specialized converter is known to the desired DM class. */
3460     ierr = DMCreate(PetscObjectComm((PetscObject)dm), &B);CHKERRQ(ierr);
3461     ierr = DMSetType(B, newtype);CHKERRQ(ierr);
3462     ierr = PetscStrncpy(convname,"DMConvert_",sizeof(convname));CHKERRQ(ierr);
3463     ierr = PetscStrlcat(convname,((PetscObject) dm)->type_name,sizeof(convname));CHKERRQ(ierr);
3464     ierr = PetscStrlcat(convname,"_",sizeof(convname));CHKERRQ(ierr);
3465     ierr = PetscStrlcat(convname,newtype,sizeof(convname));CHKERRQ(ierr);
3466     ierr = PetscStrlcat(convname,"_C",sizeof(convname));CHKERRQ(ierr);
3467     ierr = PetscObjectQueryFunction((PetscObject)B,convname,&conv);CHKERRQ(ierr);
3468     if (conv) {
3469       ierr = DMDestroy(&B);CHKERRQ(ierr);
3470       goto foundconv;
3471     }
3472 
3473 #if 0
3474     /* 3) See if a good general converter is registered for the desired class */
3475     conv = B->ops->convertfrom;
3476     ierr = DMDestroy(&B);CHKERRQ(ierr);
3477     if (conv) goto foundconv;
3478 
3479     /* 4) See if a good general converter is known for the current matrix */
3480     if (dm->ops->convert) {
3481       conv = dm->ops->convert;
3482     }
3483     if (conv) goto foundconv;
3484 #endif
3485 
3486     /* 5) Use a really basic converter. */
3487     SETERRQ2(PetscObjectComm((PetscObject)dm), PETSC_ERR_SUP, "No conversion possible between DM types %s and %s", ((PetscObject) dm)->type_name, newtype);
3488 
3489 foundconv:
3490     ierr = PetscLogEventBegin(DM_Convert,dm,0,0,0);CHKERRQ(ierr);
3491     ierr = (*conv)(dm,newtype,M);CHKERRQ(ierr);
3492     /* Things that are independent of DM type: We should consult DMClone() here */
3493     {
3494       PetscBool             isper;
3495       const PetscReal      *maxCell, *L;
3496       const DMBoundaryType *bd;
3497       ierr = DMGetPeriodicity(dm, &isper, &maxCell, &L, &bd);CHKERRQ(ierr);
3498       ierr = DMSetPeriodicity(*M, isper, maxCell,  L,  bd);CHKERRQ(ierr);
3499     }
3500     ierr = PetscLogEventEnd(DM_Convert,dm,0,0,0);CHKERRQ(ierr);
3501   }
3502   ierr = PetscObjectStateIncrease((PetscObject) *M);CHKERRQ(ierr);
3503   PetscFunctionReturn(0);
3504 }
3505 
3506 /*--------------------------------------------------------------------------------------------------------------------*/
3507 
3508 /*@C
3509   DMRegister -  Adds a new DM component implementation
3510 
3511   Not Collective
3512 
3513   Input Parameters:
3514 + name        - The name of a new user-defined creation routine
3515 - create_func - The creation routine itself
3516 
3517   Notes:
3518   DMRegister() may be called multiple times to add several user-defined DMs
3519 
3520 
3521   Sample usage:
3522 .vb
3523     DMRegister("my_da", MyDMCreate);
3524 .ve
3525 
3526   Then, your DM type can be chosen with the procedural interface via
3527 .vb
3528     DMCreate(MPI_Comm, DM *);
3529     DMSetType(DM,"my_da");
3530 .ve
3531    or at runtime via the option
3532 .vb
3533     -da_type my_da
3534 .ve
3535 
3536   Level: advanced
3537 
3538 .keywords: DM, register
3539 .seealso: DMRegisterAll(), DMRegisterDestroy()
3540 
3541 @*/
3542 PetscErrorCode  DMRegister(const char sname[],PetscErrorCode (*function)(DM))
3543 {
3544   PetscErrorCode ierr;
3545 
3546   PetscFunctionBegin;
3547   ierr = DMInitializePackage();CHKERRQ(ierr);
3548   ierr = PetscFunctionListAdd(&DMList,sname,function);CHKERRQ(ierr);
3549   PetscFunctionReturn(0);
3550 }
3551 
3552 /*@C
3553   DMLoad - Loads a DM that has been stored in binary  with DMView().
3554 
3555   Collective on PetscViewer
3556 
3557   Input Parameters:
3558 + newdm - the newly loaded DM, this needs to have been created with DMCreate() or
3559            some related function before a call to DMLoad().
3560 - viewer - binary file viewer, obtained from PetscViewerBinaryOpen() or
3561            HDF5 file viewer, obtained from PetscViewerHDF5Open()
3562 
3563    Level: intermediate
3564 
3565   Notes:
3566    The type is determined by the data in the file, any type set into the DM before this call is ignored.
3567 
3568   Notes for advanced users:
3569   Most users should not need to know the details of the binary storage
3570   format, since DMLoad() and DMView() completely hide these details.
3571   But for anyone who's interested, the standard binary matrix storage
3572   format is
3573 .vb
3574      has not yet been determined
3575 .ve
3576 
3577 .seealso: PetscViewerBinaryOpen(), DMView(), MatLoad(), VecLoad()
3578 @*/
3579 PetscErrorCode  DMLoad(DM newdm, PetscViewer viewer)
3580 {
3581   PetscBool      isbinary, ishdf5;
3582   PetscErrorCode ierr;
3583 
3584   PetscFunctionBegin;
3585   PetscValidHeaderSpecific(newdm,DM_CLASSID,1);
3586   PetscValidHeaderSpecific(viewer,PETSC_VIEWER_CLASSID,2);
3587   ierr = PetscViewerCheckReadable(viewer);CHKERRQ(ierr);
3588   ierr = PetscObjectTypeCompare((PetscObject)viewer,PETSCVIEWERBINARY,&isbinary);CHKERRQ(ierr);
3589   ierr = PetscObjectTypeCompare((PetscObject)viewer,PETSCVIEWERHDF5,&ishdf5);CHKERRQ(ierr);
3590   if (isbinary) {
3591     PetscInt classid;
3592     char     type[256];
3593 
3594     ierr = PetscViewerBinaryRead(viewer,&classid,1,NULL,PETSC_INT);CHKERRQ(ierr);
3595     if (classid != DM_FILE_CLASSID) SETERRQ1(PetscObjectComm((PetscObject)newdm),PETSC_ERR_ARG_WRONG,"Not DM next in file, classid found %d",(int)classid);
3596     ierr = PetscViewerBinaryRead(viewer,type,256,NULL,PETSC_CHAR);CHKERRQ(ierr);
3597     ierr = DMSetType(newdm, type);CHKERRQ(ierr);
3598     if (newdm->ops->load) {ierr = (*newdm->ops->load)(newdm,viewer);CHKERRQ(ierr);}
3599   } else if (ishdf5) {
3600     if (newdm->ops->load) {ierr = (*newdm->ops->load)(newdm,viewer);CHKERRQ(ierr);}
3601   } else SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONG,"Invalid viewer; open viewer with PetscViewerBinaryOpen() or PetscViewerHDF5Open()");
3602   PetscFunctionReturn(0);
3603 }
3604 
3605 /******************************** FEM Support **********************************/
3606 
3607 PetscErrorCode DMPrintCellVector(PetscInt c, const char name[], PetscInt len, const PetscScalar x[])
3608 {
3609   PetscInt       f;
3610   PetscErrorCode ierr;
3611 
3612   PetscFunctionBegin;
3613   ierr = PetscPrintf(PETSC_COMM_SELF, "Cell %D Element %s\n", c, name);CHKERRQ(ierr);
3614   for (f = 0; f < len; ++f) {
3615     ierr = PetscPrintf(PETSC_COMM_SELF, "  | %g |\n", (double)PetscRealPart(x[f]));CHKERRQ(ierr);
3616   }
3617   PetscFunctionReturn(0);
3618 }
3619 
3620 PetscErrorCode DMPrintCellMatrix(PetscInt c, const char name[], PetscInt rows, PetscInt cols, const PetscScalar A[])
3621 {
3622   PetscInt       f, g;
3623   PetscErrorCode ierr;
3624 
3625   PetscFunctionBegin;
3626   ierr = PetscPrintf(PETSC_COMM_SELF, "Cell %D Element %s\n", c, name);CHKERRQ(ierr);
3627   for (f = 0; f < rows; ++f) {
3628     ierr = PetscPrintf(PETSC_COMM_SELF, "  |");CHKERRQ(ierr);
3629     for (g = 0; g < cols; ++g) {
3630       ierr = PetscPrintf(PETSC_COMM_SELF, " % 9.5g", PetscRealPart(A[f*cols+g]));CHKERRQ(ierr);
3631     }
3632     ierr = PetscPrintf(PETSC_COMM_SELF, " |\n");CHKERRQ(ierr);
3633   }
3634   PetscFunctionReturn(0);
3635 }
3636 
3637 PetscErrorCode DMPrintLocalVec(DM dm, const char name[], PetscReal tol, Vec X)
3638 {
3639   PetscInt          localSize, bs;
3640   PetscMPIInt       size;
3641   Vec               x, xglob;
3642   const PetscScalar *xarray;
3643   PetscErrorCode    ierr;
3644 
3645   PetscFunctionBegin;
3646   ierr = MPI_Comm_size(PetscObjectComm((PetscObject) dm),&size);CHKERRQ(ierr);
3647   ierr = VecDuplicate(X, &x);CHKERRQ(ierr);
3648   ierr = VecCopy(X, x);CHKERRQ(ierr);
3649   ierr = VecChop(x, tol);CHKERRQ(ierr);
3650   ierr = PetscPrintf(PetscObjectComm((PetscObject) dm),"%s:\n",name);CHKERRQ(ierr);
3651   if (size > 1) {
3652     ierr = VecGetLocalSize(x,&localSize);CHKERRQ(ierr);
3653     ierr = VecGetArrayRead(x,&xarray);CHKERRQ(ierr);
3654     ierr = VecGetBlockSize(x,&bs);CHKERRQ(ierr);
3655     ierr = VecCreateMPIWithArray(PetscObjectComm((PetscObject) dm),bs,localSize,PETSC_DETERMINE,xarray,&xglob);CHKERRQ(ierr);
3656   } else {
3657     xglob = x;
3658   }
3659   ierr = VecView(xglob,PETSC_VIEWER_STDOUT_(PetscObjectComm((PetscObject) dm)));CHKERRQ(ierr);
3660   if (size > 1) {
3661     ierr = VecDestroy(&xglob);CHKERRQ(ierr);
3662     ierr = VecRestoreArrayRead(x,&xarray);CHKERRQ(ierr);
3663   }
3664   ierr = VecDestroy(&x);CHKERRQ(ierr);
3665   PetscFunctionReturn(0);
3666 }
3667 
3668 /*@
3669   DMGetSection - Get the PetscSection encoding the local data layout for the DM.
3670 
3671   Input Parameter:
3672 . dm - The DM
3673 
3674   Output Parameter:
3675 . section - The PetscSection
3676 
3677   Options Database Keys:
3678 . -dm_petscsection_view - View the Section created by the DM
3679 
3680   Level: intermediate
3681 
3682   Note: This gets a borrowed reference, so the user should not destroy this PetscSection.
3683 
3684 .seealso: DMSetSection(), DMGetGlobalSection()
3685 @*/
3686 PetscErrorCode DMGetSection(DM dm, PetscSection *section)
3687 {
3688   PetscErrorCode ierr;
3689 
3690   PetscFunctionBegin;
3691   PetscValidHeaderSpecific(dm, DM_CLASSID, 1);
3692   PetscValidPointer(section, 2);
3693   if (!dm->defaultSection && dm->ops->createdefaultsection) {
3694     PetscInt d;
3695 
3696     if (dm->setfromoptionscalled) for (d = 0; d < dm->Nds; ++d) {ierr = PetscDSSetFromOptions(dm->probs[d].ds);CHKERRQ(ierr);}
3697     ierr = (*dm->ops->createdefaultsection)(dm);CHKERRQ(ierr);
3698     if (dm->defaultSection) {ierr = PetscObjectViewFromOptions((PetscObject) dm->defaultSection, NULL, "-dm_petscsection_view");CHKERRQ(ierr);}
3699   }
3700   *section = dm->defaultSection;
3701   PetscFunctionReturn(0);
3702 }
3703 
3704 /*@
3705   DMSetSection - Set the PetscSection encoding the local data layout for the DM.
3706 
3707   Input Parameters:
3708 + dm - The DM
3709 - section - The PetscSection
3710 
3711   Level: intermediate
3712 
3713   Note: Any existing Section will be destroyed
3714 
3715 .seealso: DMSetSection(), DMGetGlobalSection()
3716 @*/
3717 PetscErrorCode DMSetSection(DM dm, PetscSection section)
3718 {
3719   PetscInt       numFields = 0;
3720   PetscInt       f;
3721   PetscErrorCode ierr;
3722 
3723   PetscFunctionBegin;
3724   PetscValidHeaderSpecific(dm, DM_CLASSID, 1);
3725   if (section) {
3726     PetscValidHeaderSpecific(section,PETSC_SECTION_CLASSID,2);
3727     ierr = PetscObjectReference((PetscObject)section);CHKERRQ(ierr);
3728   }
3729   ierr = PetscSectionDestroy(&dm->defaultSection);CHKERRQ(ierr);
3730   dm->defaultSection = section;
3731   if (section) {ierr = PetscSectionGetNumFields(dm->defaultSection, &numFields);CHKERRQ(ierr);}
3732   if (numFields) {
3733     ierr = DMSetNumFields(dm, numFields);CHKERRQ(ierr);
3734     for (f = 0; f < numFields; ++f) {
3735       PetscObject disc;
3736       const char *name;
3737 
3738       ierr = PetscSectionGetFieldName(dm->defaultSection, f, &name);CHKERRQ(ierr);
3739       ierr = DMGetField(dm, f, NULL, &disc);CHKERRQ(ierr);
3740       ierr = PetscObjectSetName(disc, name);CHKERRQ(ierr);
3741     }
3742   }
3743   /* The global section will be rebuilt in the next call to DMGetGlobalSection(). */
3744   ierr = PetscSectionDestroy(&dm->defaultGlobalSection);CHKERRQ(ierr);
3745   PetscFunctionReturn(0);
3746 }
3747 
3748 /*@
3749   DMGetDefaultConstraints - Get the PetscSection and Mat the specify the local constraint interpolation. See DMSetDefaultConstraints() for a description of the purpose of constraint interpolation.
3750 
3751   not collective
3752 
3753   Input Parameter:
3754 . dm - The DM
3755 
3756   Output Parameter:
3757 + section - The PetscSection describing the range of the constraint matrix: relates rows of the constraint matrix to dofs of the default section.  Returns NULL if there are no local constraints.
3758 - mat - The Mat that interpolates local constraints: its width should be the layout size of the default section.  Returns NULL if there are no local constraints.
3759 
3760   Level: advanced
3761 
3762   Note: This gets borrowed references, so the user should not destroy the PetscSection or the Mat.
3763 
3764 .seealso: DMSetDefaultConstraints()
3765 @*/
3766 PetscErrorCode DMGetDefaultConstraints(DM dm, PetscSection *section, Mat *mat)
3767 {
3768   PetscErrorCode ierr;
3769 
3770   PetscFunctionBegin;
3771   PetscValidHeaderSpecific(dm, DM_CLASSID, 1);
3772   if (!dm->defaultConstraintSection && !dm->defaultConstraintMat && dm->ops->createdefaultconstraints) {ierr = (*dm->ops->createdefaultconstraints)(dm);CHKERRQ(ierr);}
3773   if (section) {*section = dm->defaultConstraintSection;}
3774   if (mat) {*mat = dm->defaultConstraintMat;}
3775   PetscFunctionReturn(0);
3776 }
3777 
3778 /*@
3779   DMSetDefaultConstraints - Set the PetscSection and Mat the specify the local constraint interpolation.
3780 
3781   If a constraint matrix is specified, then it is applied during DMGlobalToLocalEnd() when mode is INSERT_VALUES, INSERT_BC_VALUES, or INSERT_ALL_VALUES.  Without a constraint matrix, the local vector l returned by DMGlobalToLocalEnd() contains values that have been scattered from a global vector without modification; with a constraint matrix A, l is modified by computing c = A * l, l[s[i]] = c[i], where the scatter s is defined by the PetscSection returned by DMGetDefaultConstraintMatrix().
3782 
3783   If a constraint matrix is specified, then its adjoint is applied during DMLocalToGlobalBegin() when mode is ADD_VALUES, ADD_BC_VALUES, or ADD_ALL_VALUES.  Without a constraint matrix, the local vector l is accumulated into a global vector without modification; with a constraint matrix A, l is first modified by computing c[i] = l[s[i]], l[s[i]] = 0, l = l + A'*c, which is the adjoint of the operation described above.
3784 
3785   collective on dm
3786 
3787   Input Parameters:
3788 + dm - The DM
3789 + section - The PetscSection describing the range of the constraint matrix: relates rows of the constraint matrix to dofs of the default section.  Must have a local communicator (PETSC_COMM_SELF or derivative).
3790 - mat - The Mat that interpolates local constraints: its width should be the layout size of the default section:  NULL indicates no constraints.  Must have a local communicator (PETSC_COMM_SELF or derivative).
3791 
3792   Level: advanced
3793 
3794   Note: This increments the references of the PetscSection and the Mat, so they user can destroy them
3795 
3796 .seealso: DMGetDefaultConstraints()
3797 @*/
3798 PetscErrorCode DMSetDefaultConstraints(DM dm, PetscSection section, Mat mat)
3799 {
3800   PetscMPIInt result;
3801   PetscErrorCode ierr;
3802 
3803   PetscFunctionBegin;
3804   PetscValidHeaderSpecific(dm, DM_CLASSID, 1);
3805   if (section) {
3806     PetscValidHeaderSpecific(section,PETSC_SECTION_CLASSID,2);
3807     ierr = MPI_Comm_compare(PETSC_COMM_SELF,PetscObjectComm((PetscObject)section),&result);CHKERRQ(ierr);
3808     if (result != MPI_CONGRUENT && result != MPI_IDENT) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_NOTSAMECOMM,"constraint section must have local communicator");
3809   }
3810   if (mat) {
3811     PetscValidHeaderSpecific(mat,MAT_CLASSID,3);
3812     ierr = MPI_Comm_compare(PETSC_COMM_SELF,PetscObjectComm((PetscObject)mat),&result);CHKERRQ(ierr);
3813     if (result != MPI_CONGRUENT && result != MPI_IDENT) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_NOTSAMECOMM,"constraint matrix must have local communicator");
3814   }
3815   ierr = PetscObjectReference((PetscObject)section);CHKERRQ(ierr);
3816   ierr = PetscSectionDestroy(&dm->defaultConstraintSection);CHKERRQ(ierr);
3817   dm->defaultConstraintSection = section;
3818   ierr = PetscObjectReference((PetscObject)mat);CHKERRQ(ierr);
3819   ierr = MatDestroy(&dm->defaultConstraintMat);CHKERRQ(ierr);
3820   dm->defaultConstraintMat = mat;
3821   PetscFunctionReturn(0);
3822 }
3823 
3824 #if defined(PETSC_USE_DEBUG)
3825 /*
3826   DMDefaultSectionCheckConsistency - Check the consistentcy of the global and local sections.
3827 
3828   Input Parameters:
3829 + dm - The DM
3830 . localSection - PetscSection describing the local data layout
3831 - globalSection - PetscSection describing the global data layout
3832 
3833   Level: intermediate
3834 
3835 .seealso: DMGetDefaultSF(), DMSetDefaultSF()
3836 */
3837 static PetscErrorCode DMDefaultSectionCheckConsistency_Internal(DM dm, PetscSection localSection, PetscSection globalSection)
3838 {
3839   MPI_Comm        comm;
3840   PetscLayout     layout;
3841   const PetscInt *ranges;
3842   PetscInt        pStart, pEnd, p, nroots;
3843   PetscMPIInt     size, rank;
3844   PetscBool       valid = PETSC_TRUE, gvalid;
3845   PetscErrorCode  ierr;
3846 
3847   PetscFunctionBegin;
3848   ierr = PetscObjectGetComm((PetscObject)dm,&comm);CHKERRQ(ierr);
3849   PetscValidHeaderSpecific(dm, DM_CLASSID, 1);
3850   ierr = MPI_Comm_size(comm, &size);CHKERRQ(ierr);
3851   ierr = MPI_Comm_rank(comm, &rank);CHKERRQ(ierr);
3852   ierr = PetscSectionGetChart(globalSection, &pStart, &pEnd);CHKERRQ(ierr);
3853   ierr = PetscSectionGetConstrainedStorageSize(globalSection, &nroots);CHKERRQ(ierr);
3854   ierr = PetscLayoutCreate(comm, &layout);CHKERRQ(ierr);
3855   ierr = PetscLayoutSetBlockSize(layout, 1);CHKERRQ(ierr);
3856   ierr = PetscLayoutSetLocalSize(layout, nroots);CHKERRQ(ierr);
3857   ierr = PetscLayoutSetUp(layout);CHKERRQ(ierr);
3858   ierr = PetscLayoutGetRanges(layout, &ranges);CHKERRQ(ierr);
3859   for (p = pStart; p < pEnd; ++p) {
3860     PetscInt       dof, cdof, off, gdof, gcdof, goff, gsize, d;
3861 
3862     ierr = PetscSectionGetDof(localSection, p, &dof);CHKERRQ(ierr);
3863     ierr = PetscSectionGetOffset(localSection, p, &off);CHKERRQ(ierr);
3864     ierr = PetscSectionGetConstraintDof(localSection, p, &cdof);CHKERRQ(ierr);
3865     ierr = PetscSectionGetDof(globalSection, p, &gdof);CHKERRQ(ierr);
3866     ierr = PetscSectionGetConstraintDof(globalSection, p, &gcdof);CHKERRQ(ierr);
3867     ierr = PetscSectionGetOffset(globalSection, p, &goff);CHKERRQ(ierr);
3868     if (!gdof) continue; /* Censored point */
3869     if ((gdof < 0 ? -(gdof+1) : gdof) != dof) {ierr = PetscSynchronizedPrintf(comm, "[%d]Global dof %d for point %d not equal to local dof %d\n", rank, gdof, p, dof);CHKERRQ(ierr); valid = PETSC_FALSE;}
3870     if (gcdof && (gcdof != cdof)) {ierr = PetscSynchronizedPrintf(comm, "[%d]Global constraints %d for point %d not equal to local constraints %d\n", rank, gcdof, p, cdof);CHKERRQ(ierr); valid = PETSC_FALSE;}
3871     if (gdof < 0) {
3872       gsize = gdof < 0 ? -(gdof+1)-gcdof : gdof-gcdof;
3873       for (d = 0; d < gsize; ++d) {
3874         PetscInt offset = -(goff+1) + d, r;
3875 
3876         ierr = PetscFindInt(offset,size+1,ranges,&r);CHKERRQ(ierr);
3877         if (r < 0) r = -(r+2);
3878         if ((r < 0) || (r >= size)) {ierr = PetscSynchronizedPrintf(comm, "[%d]Point %d mapped to invalid process %d (%d, %d)\n", rank, p, r, gdof, goff);CHKERRQ(ierr); valid = PETSC_FALSE;break;}
3879       }
3880     }
3881   }
3882   ierr = PetscLayoutDestroy(&layout);CHKERRQ(ierr);
3883   ierr = PetscSynchronizedFlush(comm, NULL);CHKERRQ(ierr);
3884   ierr = MPIU_Allreduce(&valid, &gvalid, 1, MPIU_BOOL, MPI_LAND, comm);CHKERRQ(ierr);
3885   if (!gvalid) {
3886     ierr = DMView(dm, NULL);CHKERRQ(ierr);
3887     SETERRQ(comm, PETSC_ERR_ARG_WRONG, "Inconsistent local and global sections");
3888   }
3889   PetscFunctionReturn(0);
3890 }
3891 #endif
3892 
3893 /*@
3894   DMGetGlobalSection - Get the PetscSection encoding the global data layout for the DM.
3895 
3896   Collective on DM
3897 
3898   Input Parameter:
3899 . dm - The DM
3900 
3901   Output Parameter:
3902 . section - The PetscSection
3903 
3904   Level: intermediate
3905 
3906   Note: This gets a borrowed reference, so the user should not destroy this PetscSection.
3907 
3908 .seealso: DMSetSection(), DMGetSection()
3909 @*/
3910 PetscErrorCode DMGetGlobalSection(DM dm, PetscSection *section)
3911 {
3912   PetscErrorCode ierr;
3913 
3914   PetscFunctionBegin;
3915   PetscValidHeaderSpecific(dm, DM_CLASSID, 1);
3916   PetscValidPointer(section, 2);
3917   if (!dm->defaultGlobalSection) {
3918     PetscSection s;
3919 
3920     ierr = DMGetSection(dm, &s);CHKERRQ(ierr);
3921     if (!s)  SETERRQ(PetscObjectComm((PetscObject) dm), PETSC_ERR_ARG_WRONGSTATE, "DM must have a default PetscSection in order to create a global PetscSection");
3922     if (!dm->sf) SETERRQ(PetscObjectComm((PetscObject)dm), PETSC_ERR_ARG_WRONGSTATE, "DM must have a point PetscSF in order to create a global PetscSection");
3923     ierr = PetscSectionCreateGlobalSection(s, dm->sf, PETSC_FALSE, PETSC_FALSE, &dm->defaultGlobalSection);CHKERRQ(ierr);
3924     ierr = PetscLayoutDestroy(&dm->map);CHKERRQ(ierr);
3925     ierr = PetscSectionGetValueLayout(PetscObjectComm((PetscObject)dm), dm->defaultGlobalSection, &dm->map);CHKERRQ(ierr);
3926     ierr = PetscSectionViewFromOptions(dm->defaultGlobalSection, NULL, "-global_section_view");CHKERRQ(ierr);
3927   }
3928   *section = dm->defaultGlobalSection;
3929   PetscFunctionReturn(0);
3930 }
3931 
3932 /*@
3933   DMSetGlobalSection - Set the PetscSection encoding the global data layout for the DM.
3934 
3935   Input Parameters:
3936 + dm - The DM
3937 - section - The PetscSection, or NULL
3938 
3939   Level: intermediate
3940 
3941   Note: Any existing Section will be destroyed
3942 
3943 .seealso: DMGetGlobalSection(), DMSetSection()
3944 @*/
3945 PetscErrorCode DMSetGlobalSection(DM dm, PetscSection section)
3946 {
3947   PetscErrorCode ierr;
3948 
3949   PetscFunctionBegin;
3950   PetscValidHeaderSpecific(dm, DM_CLASSID, 1);
3951   if (section) PetscValidHeaderSpecific(section,PETSC_SECTION_CLASSID,2);
3952   ierr = PetscObjectReference((PetscObject)section);CHKERRQ(ierr);
3953   ierr = PetscSectionDestroy(&dm->defaultGlobalSection);CHKERRQ(ierr);
3954   dm->defaultGlobalSection = section;
3955 #if defined(PETSC_USE_DEBUG)
3956   if (section) {ierr = DMDefaultSectionCheckConsistency_Internal(dm, dm->defaultSection, section);CHKERRQ(ierr);}
3957 #endif
3958   PetscFunctionReturn(0);
3959 }
3960 
3961 /*@
3962   DMGetDefaultSF - Get the PetscSF encoding the parallel dof overlap for the DM. If it has not been set,
3963   it is created from the default PetscSection layouts in the DM.
3964 
3965   Input Parameter:
3966 . dm - The DM
3967 
3968   Output Parameter:
3969 . sf - The PetscSF
3970 
3971   Level: intermediate
3972 
3973   Note: This gets a borrowed reference, so the user should not destroy this PetscSF.
3974 
3975 .seealso: DMSetDefaultSF(), DMCreateDefaultSF()
3976 @*/
3977 PetscErrorCode DMGetDefaultSF(DM dm, PetscSF *sf)
3978 {
3979   PetscInt       nroots;
3980   PetscErrorCode ierr;
3981 
3982   PetscFunctionBegin;
3983   PetscValidHeaderSpecific(dm, DM_CLASSID, 1);
3984   PetscValidPointer(sf, 2);
3985   if (!dm->defaultSF) {
3986     ierr = PetscSFCreate(PetscObjectComm((PetscObject)dm),&dm->defaultSF);CHKERRQ(ierr);
3987   }
3988   ierr = PetscSFGetGraph(dm->defaultSF, &nroots, NULL, NULL, NULL);CHKERRQ(ierr);
3989   if (nroots < 0) {
3990     PetscSection section, gSection;
3991 
3992     ierr = DMGetSection(dm, &section);CHKERRQ(ierr);
3993     if (section) {
3994       ierr = DMGetGlobalSection(dm, &gSection);CHKERRQ(ierr);
3995       ierr = DMCreateDefaultSF(dm, section, gSection);CHKERRQ(ierr);
3996     } else {
3997       *sf = NULL;
3998       PetscFunctionReturn(0);
3999     }
4000   }
4001   *sf = dm->defaultSF;
4002   PetscFunctionReturn(0);
4003 }
4004 
4005 /*@
4006   DMSetDefaultSF - Set the PetscSF encoding the parallel dof overlap for the DM
4007 
4008   Input Parameters:
4009 + dm - The DM
4010 - sf - The PetscSF
4011 
4012   Level: intermediate
4013 
4014   Note: Any previous SF is destroyed
4015 
4016 .seealso: DMGetDefaultSF(), DMCreateDefaultSF()
4017 @*/
4018 PetscErrorCode DMSetDefaultSF(DM dm, PetscSF sf)
4019 {
4020   PetscErrorCode ierr;
4021 
4022   PetscFunctionBegin;
4023   PetscValidHeaderSpecific(dm, DM_CLASSID, 1);
4024   if (sf) {
4025     PetscValidHeaderSpecific(sf, PETSCSF_CLASSID, 2);
4026     ierr = PetscObjectReference((PetscObject) sf);CHKERRQ(ierr);
4027   }
4028   ierr          = PetscSFDestroy(&dm->defaultSF);CHKERRQ(ierr);
4029   dm->defaultSF = sf;
4030   PetscFunctionReturn(0);
4031 }
4032 
4033 /*@C
4034   DMCreateDefaultSF - Create the PetscSF encoding the parallel dof overlap for the DM based upon the PetscSections
4035   describing the data layout.
4036 
4037   Input Parameters:
4038 + dm - The DM
4039 . localSection - PetscSection describing the local data layout
4040 - globalSection - PetscSection describing the global data layout
4041 
4042   Level: intermediate
4043 
4044 .seealso: DMGetDefaultSF(), DMSetDefaultSF()
4045 @*/
4046 PetscErrorCode DMCreateDefaultSF(DM dm, PetscSection localSection, PetscSection globalSection)
4047 {
4048   MPI_Comm       comm;
4049   PetscLayout    layout;
4050   const PetscInt *ranges;
4051   PetscInt       *local;
4052   PetscSFNode    *remote;
4053   PetscInt       pStart, pEnd, p, nroots, nleaves = 0, l;
4054   PetscMPIInt    size, rank;
4055   PetscErrorCode ierr;
4056 
4057   PetscFunctionBegin;
4058   PetscValidHeaderSpecific(dm, DM_CLASSID, 1);
4059   ierr = PetscObjectGetComm((PetscObject)dm,&comm);CHKERRQ(ierr);
4060   ierr = MPI_Comm_size(comm, &size);CHKERRQ(ierr);
4061   ierr = MPI_Comm_rank(comm, &rank);CHKERRQ(ierr);
4062   ierr = PetscSectionGetChart(globalSection, &pStart, &pEnd);CHKERRQ(ierr);
4063   ierr = PetscSectionGetConstrainedStorageSize(globalSection, &nroots);CHKERRQ(ierr);
4064   ierr = PetscLayoutCreate(comm, &layout);CHKERRQ(ierr);
4065   ierr = PetscLayoutSetBlockSize(layout, 1);CHKERRQ(ierr);
4066   ierr = PetscLayoutSetLocalSize(layout, nroots);CHKERRQ(ierr);
4067   ierr = PetscLayoutSetUp(layout);CHKERRQ(ierr);
4068   ierr = PetscLayoutGetRanges(layout, &ranges);CHKERRQ(ierr);
4069   for (p = pStart; p < pEnd; ++p) {
4070     PetscInt gdof, gcdof;
4071 
4072     ierr = PetscSectionGetDof(globalSection, p, &gdof);CHKERRQ(ierr);
4073     ierr = PetscSectionGetConstraintDof(globalSection, p, &gcdof);CHKERRQ(ierr);
4074     if (gcdof > (gdof < 0 ? -(gdof+1) : gdof)) SETERRQ3(PETSC_COMM_SELF, PETSC_ERR_ARG_OUTOFRANGE, "Point %d has %d constraints > %d dof", p, gcdof, (gdof < 0 ? -(gdof+1) : gdof));
4075     nleaves += gdof < 0 ? -(gdof+1)-gcdof : gdof-gcdof;
4076   }
4077   ierr = PetscMalloc1(nleaves, &local);CHKERRQ(ierr);
4078   ierr = PetscMalloc1(nleaves, &remote);CHKERRQ(ierr);
4079   for (p = pStart, l = 0; p < pEnd; ++p) {
4080     const PetscInt *cind;
4081     PetscInt       dof, cdof, off, gdof, gcdof, goff, gsize, d, c;
4082 
4083     ierr = PetscSectionGetDof(localSection, p, &dof);CHKERRQ(ierr);
4084     ierr = PetscSectionGetOffset(localSection, p, &off);CHKERRQ(ierr);
4085     ierr = PetscSectionGetConstraintDof(localSection, p, &cdof);CHKERRQ(ierr);
4086     ierr = PetscSectionGetConstraintIndices(localSection, p, &cind);CHKERRQ(ierr);
4087     ierr = PetscSectionGetDof(globalSection, p, &gdof);CHKERRQ(ierr);
4088     ierr = PetscSectionGetConstraintDof(globalSection, p, &gcdof);CHKERRQ(ierr);
4089     ierr = PetscSectionGetOffset(globalSection, p, &goff);CHKERRQ(ierr);
4090     if (!gdof) continue; /* Censored point */
4091     gsize = gdof < 0 ? -(gdof+1)-gcdof : gdof-gcdof;
4092     if (gsize != dof-cdof) {
4093       if (gsize != dof) SETERRQ4(comm, PETSC_ERR_ARG_WRONG, "Global dof %d for point %d is neither the constrained size %d, nor the unconstrained %d", gsize, p, dof-cdof, dof);
4094       cdof = 0; /* Ignore constraints */
4095     }
4096     for (d = 0, c = 0; d < dof; ++d) {
4097       if ((c < cdof) && (cind[c] == d)) {++c; continue;}
4098       local[l+d-c] = off+d;
4099     }
4100     if (gdof < 0) {
4101       for (d = 0; d < gsize; ++d, ++l) {
4102         PetscInt offset = -(goff+1) + d, r;
4103 
4104         ierr = PetscFindInt(offset,size+1,ranges,&r);CHKERRQ(ierr);
4105         if (r < 0) r = -(r+2);
4106         if ((r < 0) || (r >= size)) SETERRQ4(PETSC_COMM_SELF, PETSC_ERR_ARG_OUTOFRANGE, "Point %d mapped to invalid process %d (%d, %d)", p, r, gdof, goff);
4107         remote[l].rank  = r;
4108         remote[l].index = offset - ranges[r];
4109       }
4110     } else {
4111       for (d = 0; d < gsize; ++d, ++l) {
4112         remote[l].rank  = rank;
4113         remote[l].index = goff+d - ranges[rank];
4114       }
4115     }
4116   }
4117   if (l != nleaves) SETERRQ2(comm, PETSC_ERR_PLIB, "Iteration error, l %d != nleaves %d", l, nleaves);
4118   ierr = PetscLayoutDestroy(&layout);CHKERRQ(ierr);
4119   ierr = PetscSFSetGraph(dm->defaultSF, nroots, nleaves, local, PETSC_OWN_POINTER, remote, PETSC_OWN_POINTER);CHKERRQ(ierr);
4120   PetscFunctionReturn(0);
4121 }
4122 
4123 /*@
4124   DMGetPointSF - Get the PetscSF encoding the parallel section point overlap for the DM.
4125 
4126   Input Parameter:
4127 . dm - The DM
4128 
4129   Output Parameter:
4130 . sf - The PetscSF
4131 
4132   Level: intermediate
4133 
4134   Note: This gets a borrowed reference, so the user should not destroy this PetscSF.
4135 
4136 .seealso: DMSetPointSF(), DMGetDefaultSF(), DMSetDefaultSF(), DMCreateDefaultSF()
4137 @*/
4138 PetscErrorCode DMGetPointSF(DM dm, PetscSF *sf)
4139 {
4140   PetscFunctionBegin;
4141   PetscValidHeaderSpecific(dm, DM_CLASSID, 1);
4142   PetscValidPointer(sf, 2);
4143   *sf = dm->sf;
4144   PetscFunctionReturn(0);
4145 }
4146 
4147 /*@
4148   DMSetPointSF - Set the PetscSF encoding the parallel section point overlap for the DM.
4149 
4150   Input Parameters:
4151 + dm - The DM
4152 - sf - The PetscSF
4153 
4154   Level: intermediate
4155 
4156 .seealso: DMGetPointSF(), DMGetDefaultSF(), DMSetDefaultSF(), DMCreateDefaultSF()
4157 @*/
4158 PetscErrorCode DMSetPointSF(DM dm, PetscSF sf)
4159 {
4160   PetscErrorCode ierr;
4161 
4162   PetscFunctionBegin;
4163   PetscValidHeaderSpecific(dm, DM_CLASSID, 1);
4164   if (sf) {
4165     PetscValidHeaderSpecific(sf, PETSCSF_CLASSID, 2);
4166     ierr = PetscObjectReference((PetscObject) sf);CHKERRQ(ierr);
4167   }
4168   ierr   = PetscSFDestroy(&dm->sf);CHKERRQ(ierr);
4169   dm->sf = sf;
4170   PetscFunctionReturn(0);
4171 }
4172 
4173 static PetscErrorCode DMFieldEnlarge_Static(DM dm, PetscInt NfNew)
4174 {
4175   RegionField   *tmpr;
4176   PetscInt       Nf = dm->Nf, f;
4177   PetscErrorCode ierr;
4178 
4179   PetscFunctionBegin;
4180   if (Nf >= NfNew) PetscFunctionReturn(0);
4181   ierr = PetscMalloc1(NfNew, &tmpr);CHKERRQ(ierr);
4182   for (f = 0; f < Nf; ++f) tmpr[f] = dm->fields[f];
4183   for (f = Nf; f < NfNew; ++f) {tmpr[f].disc = NULL; tmpr[f].label = NULL;}
4184   ierr = PetscFree(dm->fields);CHKERRQ(ierr);
4185   dm->Nf     = NfNew;
4186   dm->fields = tmpr;
4187   PetscFunctionReturn(0);
4188 }
4189 
4190 /*@
4191   DMClearFields - Remove all fields from the DM
4192 
4193   Logically collective on DM
4194 
4195   Input Parameter:
4196 . dm - The DM
4197 
4198   Level: intermediate
4199 
4200 .seealso: DMGetNumFields(), DMSetNumFields(), DMSetField()
4201 @*/
4202 PetscErrorCode DMClearFields(DM dm)
4203 {
4204   PetscInt       f;
4205   PetscErrorCode ierr;
4206 
4207   PetscFunctionBegin;
4208   PetscValidHeaderSpecific(dm, DM_CLASSID, 1);
4209   for (f = 0; f < dm->Nf; ++f) {
4210     ierr = PetscObjectDestroy(&dm->fields[f].disc);CHKERRQ(ierr);
4211     ierr = DMLabelDestroy(&dm->fields[f].label);CHKERRQ(ierr);
4212   }
4213   ierr = PetscFree(dm->fields);CHKERRQ(ierr);
4214   dm->fields = NULL;
4215   dm->Nf     = 0;
4216   PetscFunctionReturn(0);
4217 }
4218 
4219 /*@
4220   DMGetNumFields - Get the number of fields in the DM
4221 
4222   Not collective
4223 
4224   Input Parameter:
4225 . dm - The DM
4226 
4227   Output Parameter:
4228 . Nf - The number of fields
4229 
4230   Level: intermediate
4231 
4232 .seealso: DMSetNumFields(), DMSetField()
4233 @*/
4234 PetscErrorCode DMGetNumFields(DM dm, PetscInt *numFields)
4235 {
4236   PetscFunctionBegin;
4237   PetscValidHeaderSpecific(dm, DM_CLASSID, 1);
4238   PetscValidPointer(numFields, 2);
4239   *numFields = dm->Nf;
4240   PetscFunctionReturn(0);
4241 }
4242 
4243 /*@
4244   DMSetNumFields - Set the number of fields in the DM
4245 
4246   Logically collective on DM
4247 
4248   Input Parameters:
4249 + dm - The DM
4250 - Nf - The number of fields
4251 
4252   Level: intermediate
4253 
4254 .seealso: DMGetNumFields(), DMSetField()
4255 @*/
4256 PetscErrorCode DMSetNumFields(DM dm, PetscInt numFields)
4257 {
4258   PetscDS        ds;
4259   PetscInt       Nf, f;
4260   PetscErrorCode ierr;
4261 
4262   PetscFunctionBegin;
4263   PetscValidHeaderSpecific(dm, DM_CLASSID, 1);
4264   ierr = DMGetNumFields(dm, &Nf);CHKERRQ(ierr);
4265   for (f = Nf; f < numFields; ++f) {
4266     PetscContainer obj;
4267 
4268     ierr = PetscContainerCreate(PetscObjectComm((PetscObject) dm), &obj);CHKERRQ(ierr);
4269     ierr = DMAddField(dm, NULL, (PetscObject) obj);CHKERRQ(ierr);
4270     ierr = PetscContainerDestroy(&obj);CHKERRQ(ierr);
4271   }
4272   PetscFunctionReturn(0);
4273 }
4274 
4275 /*@
4276   DMGetField - Return the discretization object for a given DM field
4277 
4278   Not collective
4279 
4280   Input Parameters:
4281 + dm - The DM
4282 - f  - The field number
4283 
4284   Output Parameters:
4285 + label - The label indicating the support of the field, or NULL for the entire mesh
4286 - field - The discretization object
4287 
4288   Level: intermediate
4289 
4290 .seealso: DMAddField(), DMSetField()
4291 @*/
4292 PetscErrorCode DMGetField(DM dm, PetscInt f, DMLabel *label, PetscObject *field)
4293 {
4294   PetscFunctionBegin;
4295   PetscValidHeaderSpecific(dm, DM_CLASSID, 1);
4296   PetscValidPointer(field, 3);
4297   if ((f < 0) || (f >= dm->Nf)) SETERRQ2(PETSC_COMM_SELF, PETSC_ERR_ARG_OUTOFRANGE, "Field number %d must be in [0, %d)", f, dm->Nf);
4298   if (label) *label = dm->fields[f].label;
4299   if (field) *field = dm->fields[f].disc;
4300   PetscFunctionReturn(0);
4301 }
4302 
4303 /*@
4304   DMSetField - Set the discretization object for a given DM field
4305 
4306   Logically collective on DM
4307 
4308   Input Parameters:
4309 + dm    - The DM
4310 . f     - The field number
4311 . label - The label indicating the support of the field, or NULL for the entire mesh
4312 - field - The discretization object
4313 
4314   Level: intermediate
4315 
4316 .seealso: DMAddField(), DMGetField()
4317 @*/
4318 PetscErrorCode DMSetField(DM dm, PetscInt f, DMLabel label, PetscObject field)
4319 {
4320   PetscErrorCode ierr;
4321 
4322   PetscFunctionBegin;
4323   PetscValidHeaderSpecific(dm, DM_CLASSID, 1);
4324   if (label) PetscValidHeaderSpecific(label, DMLABEL_CLASSID, 3);
4325   PetscValidHeader(field, 4);
4326   if (f < 0) SETERRQ1(PETSC_COMM_SELF, PETSC_ERR_ARG_OUTOFRANGE, "Field number %d must be non-negative", f);
4327   ierr = DMFieldEnlarge_Static(dm, f+1);CHKERRQ(ierr);
4328   ierr = DMLabelDestroy(&dm->fields[f].label);CHKERRQ(ierr);
4329   ierr = PetscObjectDestroy(&dm->fields[f].disc);CHKERRQ(ierr);
4330   dm->fields[f].label = label;
4331   dm->fields[f].disc  = field;
4332   ierr = PetscObjectReference((PetscObject) label);CHKERRQ(ierr);
4333   ierr = PetscObjectReference((PetscObject) field);CHKERRQ(ierr);
4334   PetscFunctionReturn(0);
4335 }
4336 
4337 /*@
4338   DMAddField - Add the discretization object for the given DM field
4339 
4340   Logically collective on DM
4341 
4342   Input Parameters:
4343 + dm    - The DM
4344 . label - The label indicating the support of the field, or NULL for the entire mesh
4345 - field - The discretization object
4346 
4347   Level: intermediate
4348 
4349 .seealso: DMSetField(), DMGetField()
4350 @*/
4351 PetscErrorCode DMAddField(DM dm, DMLabel label, PetscObject field)
4352 {
4353   PetscInt       Nf = dm->Nf;
4354   PetscErrorCode ierr;
4355 
4356   PetscFunctionBegin;
4357   PetscValidHeaderSpecific(dm, DM_CLASSID, 1);
4358   if (label) PetscValidHeaderSpecific(label, DMLABEL_CLASSID, 3);
4359   PetscValidHeader(field, 3);
4360   ierr = DMFieldEnlarge_Static(dm, Nf+1);CHKERRQ(ierr);
4361   dm->fields[Nf].label = label;
4362   dm->fields[Nf].disc  = field;
4363   ierr = PetscObjectReference((PetscObject) label);CHKERRQ(ierr);
4364   ierr = PetscObjectReference((PetscObject) field);CHKERRQ(ierr);
4365   PetscFunctionReturn(0);
4366 }
4367 
4368 /*@
4369   DMCopyFields - Copy the discretizations for the DM into another DM
4370 
4371   Collective on DM
4372 
4373   Input Parameter:
4374 . dm - The DM
4375 
4376   Output Parameter:
4377 . newdm - The DM
4378 
4379   Level: advanced
4380 
4381 .seealso: DMGetField(), DMSetField(), DMAddField(), DMCopyDS(), DMGetDS(), DMGetCellDS()
4382 @*/
4383 PetscErrorCode DMCopyFields(DM dm, DM newdm)
4384 {
4385   PetscInt       Nf, f;
4386   PetscErrorCode ierr;
4387 
4388   PetscFunctionBegin;
4389   if (dm == newdm) PetscFunctionReturn(0);
4390   ierr = DMGetNumFields(dm, &Nf);CHKERRQ(ierr);
4391   ierr = DMClearFields(newdm);CHKERRQ(ierr);
4392   for (f = 0; f < Nf; ++f) {
4393     DMLabel     label;
4394     PetscObject field;
4395 
4396     ierr = DMGetField(dm, f, &label, &field);CHKERRQ(ierr);
4397     ierr = DMSetField(newdm, f, label, field);CHKERRQ(ierr);
4398   }
4399   PetscFunctionReturn(0);
4400 }
4401 
4402 static PetscErrorCode DMDSEnlarge_Static(DM dm, PetscInt NdsNew)
4403 {
4404   DMSpace       *tmpd;
4405   PetscInt       Nds = dm->Nds, s;
4406   PetscErrorCode ierr;
4407 
4408   PetscFunctionBegin;
4409   if (Nds >= NdsNew) PetscFunctionReturn(0);
4410   ierr = PetscMalloc1(NdsNew, &tmpd);CHKERRQ(ierr);
4411   for (s = 0; s < Nds; ++s) tmpd[s] = dm->probs[s];
4412   for (s = Nds; s < NdsNew; ++s) {tmpd[s].ds = NULL; tmpd[s].label = NULL;}
4413   ierr = PetscFree(dm->probs);CHKERRQ(ierr);
4414   dm->Nds   = NdsNew;
4415   dm->probs = tmpd;
4416   PetscFunctionReturn(0);
4417 }
4418 
4419 /*@
4420   DMGetNumDS - Get the number of discrete systems in the DM
4421 
4422   Not collective
4423 
4424   Input Parameter:
4425 . dm - The DM
4426 
4427   Output Parameter:
4428 . Nds - The number of PetscDS objects
4429 
4430   Level: intermediate
4431 
4432 .seealso: DMGetDS(), DMGetCellDS()
4433 @*/
4434 PetscErrorCode DMGetNumDS(DM dm, PetscInt *Nds)
4435 {
4436   PetscFunctionBegin;
4437   PetscValidHeaderSpecific(dm, DM_CLASSID, 1);
4438   PetscValidPointer(Nds, 2);
4439   *Nds = dm->Nds;
4440   PetscFunctionReturn(0);
4441 }
4442 
4443 /*@
4444   DMClearDS - Remove all discrete systems from the DM
4445 
4446   Logically collective on DM
4447 
4448   Input Parameter:
4449 . dm - The DM
4450 
4451   Level: intermediate
4452 
4453 .seealso: DMGetNumDS(), DMGetDS(), DMSetField()
4454 @*/
4455 PetscErrorCode DMClearDS(DM dm)
4456 {
4457   PetscInt       s;
4458   PetscErrorCode ierr;
4459 
4460   PetscFunctionBegin;
4461   PetscValidHeaderSpecific(dm, DM_CLASSID, 1);
4462   for (s = 0; s < dm->Nds; ++s) {
4463     ierr = PetscDSDestroy(&dm->probs[s].ds);CHKERRQ(ierr);
4464     ierr = DMLabelDestroy(&dm->probs[s].label);CHKERRQ(ierr);
4465   }
4466   ierr = PetscFree(dm->probs);CHKERRQ(ierr);
4467   dm->probs = NULL;
4468   dm->Nds   = 0;
4469   PetscFunctionReturn(0);
4470 }
4471 
4472 /*@
4473   DMGetDS - Get the default PetscDS
4474 
4475   Not collective
4476 
4477   Input Parameter:
4478 . dm    - The DM
4479 
4480   Output Parameter:
4481 . prob - The default PetscDS
4482 
4483   Level: intermediate
4484 
4485 .seealso: DMGetCellDS(), DMGetRegionDS()
4486 @*/
4487 PetscErrorCode DMGetDS(DM dm, PetscDS *prob)
4488 {
4489   PetscFunctionBeginHot;
4490   PetscValidHeaderSpecific(dm, DM_CLASSID, 1);
4491   PetscValidPointer(prob, 2);
4492   if (dm->Nds) *prob = dm->probs[0].ds;
4493   else         *prob = NULL;
4494   PetscFunctionReturn(0);
4495 }
4496 
4497 /*@
4498   DMGetCellDS - Get the PetscDS defined on a given cell
4499 
4500   Not collective
4501 
4502   Input Parameters:
4503 + dm    - The DM
4504 - point - Cell for the DS
4505 
4506   Output Parameter:
4507 . prob - The PetscDS defined on the given cell
4508 
4509   Level: developer
4510 
4511 .seealso: DMGetDS(), DMSetDS()
4512 @*/
4513 PetscErrorCode DMGetCellDS(DM dm, PetscInt point, PetscDS *prob)
4514 {
4515   PetscDS        probDef = NULL;
4516   PetscInt       s;
4517   PetscErrorCode ierr;
4518 
4519   PetscFunctionBeginHot;
4520   PetscValidHeaderSpecific(dm, DM_CLASSID, 1);
4521   PetscValidPointer(prob, 3);
4522   *prob = NULL;
4523   for (s = 0; s < dm->Nds; ++s) {
4524     PetscInt val;
4525 
4526     if (!dm->probs[s].label) {probDef = dm->probs[s].ds;}
4527     else {
4528       ierr = DMLabelGetValue(dm->probs[s].label, point, &val);CHKERRQ(ierr);
4529       if (val >= 0) {*prob = dm->probs[s].ds; break;}
4530     }
4531   }
4532   if (!*prob) *prob = probDef;
4533   PetscFunctionReturn(0);
4534 }
4535 
4536 /*@
4537   DMGetRegionDS - Get the PetscDS for a given mesh region, defined by a DMLabel
4538 
4539   Not collective
4540 
4541   Input Parameters:
4542 + dm    - The DM
4543 - label - The DMLabel defining the mesh region, or NULL for the entire mesh
4544 
4545   Output Parameter:
4546 . prob - The PetscDS defined on the given region
4547 
4548   Note: If the label is missing, this function returns an error
4549 
4550   Level: advanced
4551 
4552 .seealso: DMGetRegionNumDS(), DMSetRegionDS(), DMGetDS(), DMGetCellDS()
4553 @*/
4554 PetscErrorCode DMGetRegionDS(DM dm, DMLabel label, PetscDS *ds)
4555 {
4556   PetscInt Nds = dm->Nds, s;
4557 
4558   PetscFunctionBegin;
4559   PetscValidHeaderSpecific(dm, DM_CLASSID, 1);
4560   if (label) PetscValidHeaderSpecific(label, DMLABEL_CLASSID, 2);
4561   PetscValidPointer(ds, 3);
4562   *ds = NULL;
4563   for (s = 0; s < Nds; ++s) {
4564     if (dm->probs[s].label == label) {*ds = dm->probs[s].ds; PetscFunctionReturn(0);}
4565   }
4566   SETERRQ(PetscObjectComm((PetscObject) dm), PETSC_ERR_ARG_WRONG, "Label not found in DM");
4567 }
4568 
4569 /*@
4570   DMGetRegionNumDS - Get the PetscDS for a given mesh region, defined by the region number
4571 
4572   Not collective
4573 
4574   Input Parameters:
4575 + dm  - The DM
4576 - num - The region number, in [0, Nds)
4577 
4578   Output Parameters:
4579 + label - The region label
4580 - prob  - The PetscDS defined on the given region
4581 
4582   Level: advanced
4583 
4584 .seealso: DMGetRegionDS(), DMSetRegionDS(), DMGetDS(), DMGetCellDS()
4585 @*/
4586 PetscErrorCode DMGetRegionNumDS(DM dm, PetscInt num, DMLabel *label, PetscDS *ds)
4587 {
4588   PetscInt       Nds;
4589   PetscErrorCode ierr;
4590 
4591   PetscFunctionBegin;
4592   PetscValidHeaderSpecific(dm, DM_CLASSID, 1);
4593   ierr = DMGetNumDS(dm, &Nds);CHKERRQ(ierr);
4594   if ((num < 0) || (num >= Nds)) SETERRQ2(PETSC_COMM_SELF, PETSC_ERR_ARG_OUTOFRANGE, "Region number %D is not in [0, %D)", num, Nds);
4595   if (label) {
4596     PetscValidPointer(label, 3);
4597     *label = dm->probs[num].label;
4598   }
4599   if (ds) {
4600     PetscValidPointer(ds, 4);
4601     *ds = dm->probs[num].ds;
4602   }
4603   PetscFunctionReturn(0);
4604 }
4605 
4606 /*@
4607   DMSetRegionDS - Set the PetscDS for a given mesh region, defined by a DMLabel
4608 
4609   Collective on DM
4610 
4611   Input Parameters:
4612 + dm    - The DM
4613 . label - The DMLabel defining the mesh region, or NULL for the entire mesh
4614 - prob - The PetscDS defined on the given cell
4615 
4616   Note: If the label has a DS defined, it will be replaced. Otherwise, it will be added to the DM.
4617 
4618   Level: advanced
4619 
4620 .seealso: DMGetRegionDS(), DMGetDS(), DMGetCellDS()
4621 @*/
4622 PetscErrorCode DMSetRegionDS(DM dm, DMLabel label, PetscDS ds)
4623 {
4624   PetscInt       Nds = dm->Nds, s;
4625   PetscErrorCode ierr;
4626 
4627   PetscFunctionBegin;
4628   PetscValidHeaderSpecific(dm, DM_CLASSID, 1);
4629   if (label) PetscValidHeaderSpecific(label, DMLABEL_CLASSID, 2);
4630   PetscValidHeaderSpecific(ds, PETSCDS_CLASSID, 3);
4631   for (s = 0; s < Nds; ++s) {
4632     if (dm->probs[s].label == label) {
4633       dm->probs[s].ds = ds;
4634       PetscFunctionReturn(0);
4635     }
4636   }
4637   ierr = DMDSEnlarge_Static(dm, Nds+1);CHKERRQ(ierr);
4638   ierr = PetscObjectReference((PetscObject) label);CHKERRQ(ierr);
4639   ierr = PetscObjectReference((PetscObject) ds);CHKERRQ(ierr);
4640   if (!label) {
4641     /* Put the NULL label at the front, so it is returned as the default */
4642     for (s = Nds-1; s >=0; --s) dm->probs[s+1] = dm->probs[s];
4643     Nds = 0;
4644   }
4645   dm->probs[Nds].label = label;
4646   dm->probs[Nds].ds    = ds;
4647   PetscFunctionReturn(0);
4648 }
4649 
4650 /*@
4651   DMCreateDS - Create the discrete systems for the DM based upon the fields added to the DM
4652 
4653   Collective on DM
4654 
4655   Input Parameter:
4656 . dm - The DM
4657 
4658   Note: If the label has a DS defined, it will be replaced. Otherwise, it will be added to the DM.
4659 
4660   Level: intermediate
4661 
4662 .seealso: DMSetField, DMAddField(), DMGetDS(), DMGetCellDS(), DMGetRegionDS(), DMSetRegionDS()
4663 @*/
4664 PetscErrorCode DMCreateDS(DM dm)
4665 {
4666   MPI_Comm       comm;
4667   PetscDS        prob, probh = NULL;
4668   PetscInt       dimEmbed, f, s, field = 0, fieldh = 0;
4669   PetscBool      doSetup = PETSC_TRUE;
4670   PetscErrorCode ierr;
4671 
4672   PetscFunctionBegin;
4673   PetscValidHeaderSpecific(dm, DM_CLASSID, 1);
4674   if (!dm->fields) PetscFunctionReturn(0);
4675   /* Can only handle two label cases right now:
4676    1) NULL
4677    2) Hybrid cells
4678   */
4679   ierr = PetscObjectGetComm((PetscObject) dm, &comm);CHKERRQ(ierr);
4680   ierr = DMGetCoordinateDim(dm, &dimEmbed);CHKERRQ(ierr);
4681   /* Create default DS */
4682   ierr = DMGetRegionDS(dm, NULL, &prob);CHKERRQ(ierr);
4683   ierr = PetscDSSetCoordinateDimension(prob, dimEmbed);CHKERRQ(ierr);
4684   /* Optionally create hybrid DS */
4685   for (f = 0; f < dm->Nf; ++f) {
4686     DMLabel  label = dm->fields[f].label;
4687     PetscInt lStart, lEnd;
4688 
4689     if (label) {
4690       DM       plex;
4691       PetscInt depth, pMax[4];
4692 
4693       ierr = DMConvert(dm, DMPLEX, &plex);CHKERRQ(ierr);
4694       ierr = DMPlexGetDepth(plex, &depth);CHKERRQ(ierr);
4695       ierr = DMPlexGetHybridBounds(plex, depth >= 0 ? &pMax[depth] : NULL, depth>1 ? &pMax[depth-1] : NULL, depth>2 ? &pMax[1] : NULL, &pMax[0]);CHKERRQ(ierr);
4696       ierr = DMDestroy(&plex);CHKERRQ(ierr);
4697 
4698       ierr = DMLabelGetBounds(label, &lStart, &lEnd);CHKERRQ(ierr);
4699       if (lStart < pMax[depth]) SETERRQ(PETSC_COMM_SELF, PETSC_ERR_SUP, "Only support labels over hybrid cells right now");
4700       ierr = PetscDSCreate(comm, &probh);CHKERRQ(ierr);
4701       ierr = DMSetRegionDS(dm, label, probh);CHKERRQ(ierr);
4702       ierr = PetscDSSetHybrid(probh, PETSC_TRUE);CHKERRQ(ierr);
4703       ierr = PetscDSSetCoordinateDimension(probh, dimEmbed);CHKERRQ(ierr);
4704       break;
4705     }
4706   }
4707   /* Set fields in DSes */
4708   for (f = 0; f < dm->Nf; ++f) {
4709     DMLabel     label = dm->fields[f].label;
4710     PetscObject disc  = dm->fields[f].disc;
4711 
4712     if (!label) {
4713       ierr = PetscDSSetDiscretization(prob,  field++,  disc);CHKERRQ(ierr);
4714       if (probh) {
4715         PetscFE subfe;
4716 
4717         ierr = PetscFEGetHeightSubspace((PetscFE) disc, 1, &subfe);CHKERRQ(ierr);
4718         ierr = PetscDSSetDiscretization(probh, fieldh++, (PetscObject) subfe);CHKERRQ(ierr);
4719       }
4720     } else {
4721       ierr = PetscDSSetDiscretization(probh, fieldh++, disc);CHKERRQ(ierr);
4722     }
4723     /* We allow people to have placeholder fields and construct the Section by hand */
4724     {
4725       PetscClassId id;
4726 
4727       ierr = PetscObjectGetClassId(disc, &id);CHKERRQ(ierr);
4728       if ((id != PETSCFE_CLASSID) && (id != PETSCFV_CLASSID)) doSetup = PETSC_FALSE;
4729     }
4730   }
4731   ierr = PetscDSDestroy(&probh);CHKERRQ(ierr);
4732   /* Setup DSes */
4733   if (doSetup) {
4734     for (s = 0; s < dm->Nds; ++s) {ierr = PetscDSSetUp(dm->probs[s].ds);CHKERRQ(ierr);}
4735   }
4736   PetscFunctionReturn(0);
4737 }
4738 
4739 /*@
4740   DMCopyDS - Copy the discrete systems for the DM into another DM
4741 
4742   Collective on DM
4743 
4744   Input Parameter:
4745 . dm - The DM
4746 
4747   Output Parameter:
4748 . newdm - The DM
4749 
4750   Level: advanced
4751 
4752 .seealso: DMCopyFields(), DMAddField(), DMGetDS(), DMGetCellDS(), DMGetRegionDS(), DMSetRegionDS()
4753 @*/
4754 PetscErrorCode DMCopyDS(DM dm, DM newdm)
4755 {
4756   PetscInt       Nds, s;
4757   PetscErrorCode ierr;
4758 
4759   PetscFunctionBegin;
4760   if (dm == newdm) PetscFunctionReturn(0);
4761   ierr = DMGetNumDS(dm, &Nds);CHKERRQ(ierr);
4762   ierr = DMClearDS(newdm);CHKERRQ(ierr);
4763   for (s = 0; s < Nds; ++s) {
4764     DMLabel label;
4765     PetscDS ds;
4766 
4767     ierr = DMGetRegionNumDS(dm, s, &label, &ds);CHKERRQ(ierr);
4768     ierr = DMSetRegionDS(newdm, label, ds);CHKERRQ(ierr);
4769   }
4770   PetscFunctionReturn(0);
4771 }
4772 
4773 /*@
4774   DMCopyDisc - Copy the fields and discrete systems for the DM into another DM
4775 
4776   Collective on DM
4777 
4778   Input Parameter:
4779 . dm - The DM
4780 
4781   Output Parameter:
4782 . newdm - The DM
4783 
4784   Level: advanced
4785 
4786 .seealso: DMCopyFields(), DMCopyDS()
4787 @*/
4788 PetscErrorCode DMCopyDisc(DM dm, DM newdm)
4789 {
4790   PetscErrorCode ierr;
4791 
4792   PetscFunctionBegin;
4793   if (dm == newdm) PetscFunctionReturn(0);
4794   ierr = DMCopyFields(dm, newdm);CHKERRQ(ierr);
4795   ierr = DMCopyDS(dm, newdm);CHKERRQ(ierr);
4796   PetscFunctionReturn(0);
4797 }
4798 
4799 PetscErrorCode DMRestrictHook_Coordinates(DM dm,DM dmc,void *ctx)
4800 {
4801   DM dm_coord,dmc_coord;
4802   PetscErrorCode ierr;
4803   Vec coords,ccoords;
4804   Mat inject;
4805   PetscFunctionBegin;
4806   ierr = DMGetCoordinateDM(dm,&dm_coord);CHKERRQ(ierr);
4807   ierr = DMGetCoordinateDM(dmc,&dmc_coord);CHKERRQ(ierr);
4808   ierr = DMGetCoordinates(dm,&coords);CHKERRQ(ierr);
4809   ierr = DMGetCoordinates(dmc,&ccoords);CHKERRQ(ierr);
4810   if (coords && !ccoords) {
4811     ierr = DMCreateGlobalVector(dmc_coord,&ccoords);CHKERRQ(ierr);
4812     ierr = PetscObjectSetName((PetscObject)ccoords,"coordinates");CHKERRQ(ierr);
4813     ierr = DMCreateInjection(dmc_coord,dm_coord,&inject);CHKERRQ(ierr);
4814     ierr = MatRestrict(inject,coords,ccoords);CHKERRQ(ierr);
4815     ierr = MatDestroy(&inject);CHKERRQ(ierr);
4816     ierr = DMSetCoordinates(dmc,ccoords);CHKERRQ(ierr);
4817     ierr = VecDestroy(&ccoords);CHKERRQ(ierr);
4818   }
4819   PetscFunctionReturn(0);
4820 }
4821 
4822 static PetscErrorCode DMSubDomainHook_Coordinates(DM dm,DM subdm,void *ctx)
4823 {
4824   DM dm_coord,subdm_coord;
4825   PetscErrorCode ierr;
4826   Vec coords,ccoords,clcoords;
4827   VecScatter *scat_i,*scat_g;
4828   PetscFunctionBegin;
4829   ierr = DMGetCoordinateDM(dm,&dm_coord);CHKERRQ(ierr);
4830   ierr = DMGetCoordinateDM(subdm,&subdm_coord);CHKERRQ(ierr);
4831   ierr = DMGetCoordinates(dm,&coords);CHKERRQ(ierr);
4832   ierr = DMGetCoordinates(subdm,&ccoords);CHKERRQ(ierr);
4833   if (coords && !ccoords) {
4834     ierr = DMCreateGlobalVector(subdm_coord,&ccoords);CHKERRQ(ierr);
4835     ierr = PetscObjectSetName((PetscObject)ccoords,"coordinates");CHKERRQ(ierr);
4836     ierr = DMCreateLocalVector(subdm_coord,&clcoords);CHKERRQ(ierr);
4837     ierr = PetscObjectSetName((PetscObject)clcoords,"coordinates");CHKERRQ(ierr);
4838     ierr = DMCreateDomainDecompositionScatters(dm_coord,1,&subdm_coord,NULL,&scat_i,&scat_g);CHKERRQ(ierr);
4839     ierr = VecScatterBegin(scat_i[0],coords,ccoords,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr);
4840     ierr = VecScatterBegin(scat_g[0],coords,clcoords,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr);
4841     ierr = VecScatterEnd(scat_i[0],coords,ccoords,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr);
4842     ierr = VecScatterEnd(scat_g[0],coords,clcoords,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr);
4843     ierr = DMSetCoordinates(subdm,ccoords);CHKERRQ(ierr);
4844     ierr = DMSetCoordinatesLocal(subdm,clcoords);CHKERRQ(ierr);
4845     ierr = VecScatterDestroy(&scat_i[0]);CHKERRQ(ierr);
4846     ierr = VecScatterDestroy(&scat_g[0]);CHKERRQ(ierr);
4847     ierr = VecDestroy(&ccoords);CHKERRQ(ierr);
4848     ierr = VecDestroy(&clcoords);CHKERRQ(ierr);
4849     ierr = PetscFree(scat_i);CHKERRQ(ierr);
4850     ierr = PetscFree(scat_g);CHKERRQ(ierr);
4851   }
4852   PetscFunctionReturn(0);
4853 }
4854 
4855 /*@
4856   DMGetDimension - Return the topological dimension of the DM
4857 
4858   Not collective
4859 
4860   Input Parameter:
4861 . dm - The DM
4862 
4863   Output Parameter:
4864 . dim - The topological dimension
4865 
4866   Level: beginner
4867 
4868 .seealso: DMSetDimension(), DMCreate()
4869 @*/
4870 PetscErrorCode DMGetDimension(DM dm, PetscInt *dim)
4871 {
4872   PetscFunctionBegin;
4873   PetscValidHeaderSpecific(dm, DM_CLASSID, 1);
4874   PetscValidPointer(dim, 2);
4875   *dim = dm->dim;
4876   PetscFunctionReturn(0);
4877 }
4878 
4879 /*@
4880   DMSetDimension - Set the topological dimension of the DM
4881 
4882   Collective on dm
4883 
4884   Input Parameters:
4885 + dm - The DM
4886 - dim - The topological dimension
4887 
4888   Level: beginner
4889 
4890 .seealso: DMGetDimension(), DMCreate()
4891 @*/
4892 PetscErrorCode DMSetDimension(DM dm, PetscInt dim)
4893 {
4894   PetscDS        ds;
4895   PetscErrorCode ierr;
4896 
4897   PetscFunctionBegin;
4898   PetscValidHeaderSpecific(dm, DM_CLASSID, 1);
4899   PetscValidLogicalCollectiveInt(dm, dim, 2);
4900   dm->dim = dim;
4901   ierr = DMGetDS(dm, &ds);CHKERRQ(ierr);
4902   if (ds->dimEmbed < 0) {ierr = PetscDSSetCoordinateDimension(ds, dm->dim);CHKERRQ(ierr);}
4903   PetscFunctionReturn(0);
4904 }
4905 
4906 /*@
4907   DMGetDimPoints - Get the half-open interval for all points of a given dimension
4908 
4909   Collective on DM
4910 
4911   Input Parameters:
4912 + dm - the DM
4913 - dim - the dimension
4914 
4915   Output Parameters:
4916 + pStart - The first point of the given dimension
4917 . pEnd - The first point following points of the given dimension
4918 
4919   Note:
4920   The points are vertices in the Hasse diagram encoding the topology. This is explained in
4921   http://arxiv.org/abs/0908.4427. If no points exist of this dimension in the storage scheme,
4922   then the interval is empty.
4923 
4924   Level: intermediate
4925 
4926 .keywords: point, Hasse Diagram, dimension
4927 .seealso: DMPLEX, DMPlexGetDepthStratum(), DMPlexGetHeightStratum()
4928 @*/
4929 PetscErrorCode DMGetDimPoints(DM dm, PetscInt dim, PetscInt *pStart, PetscInt *pEnd)
4930 {
4931   PetscInt       d;
4932   PetscErrorCode ierr;
4933 
4934   PetscFunctionBegin;
4935   PetscValidHeaderSpecific(dm,DM_CLASSID,1);
4936   ierr = DMGetDimension(dm, &d);CHKERRQ(ierr);
4937   if ((dim < 0) || (dim > d)) SETERRQ2(PetscObjectComm((PetscObject) dm), PETSC_ERR_ARG_OUTOFRANGE, "Invalid dimension %d 1", dim, d);
4938   ierr = (*dm->ops->getdimpoints)(dm, dim, pStart, pEnd);CHKERRQ(ierr);
4939   PetscFunctionReturn(0);
4940 }
4941 
4942 /*@
4943   DMSetCoordinates - Sets into the DM a global vector that holds the coordinates
4944 
4945   Collective on DM
4946 
4947   Input Parameters:
4948 + dm - the DM
4949 - c - coordinate vector
4950 
4951   Notes:
4952   The coordinates do include those for ghost points, which are in the local vector.
4953 
4954   The vector c should be destroyed by the caller.
4955 
4956   Level: intermediate
4957 
4958 .keywords: distributed array, get, corners, nodes, local indices, coordinates
4959 .seealso: DMSetCoordinatesLocal(), DMGetCoordinates(), DMGetCoordinatesLocal(), DMGetCoordinateDM()
4960 @*/
4961 PetscErrorCode DMSetCoordinates(DM dm, Vec c)
4962 {
4963   PetscErrorCode ierr;
4964 
4965   PetscFunctionBegin;
4966   PetscValidHeaderSpecific(dm,DM_CLASSID,1);
4967   PetscValidHeaderSpecific(c,VEC_CLASSID,2);
4968   ierr            = PetscObjectReference((PetscObject) c);CHKERRQ(ierr);
4969   ierr            = VecDestroy(&dm->coordinates);CHKERRQ(ierr);
4970   dm->coordinates = c;
4971   ierr            = VecDestroy(&dm->coordinatesLocal);CHKERRQ(ierr);
4972   ierr            = DMCoarsenHookAdd(dm,DMRestrictHook_Coordinates,NULL,NULL);CHKERRQ(ierr);
4973   ierr            = DMSubDomainHookAdd(dm,DMSubDomainHook_Coordinates,NULL,NULL);CHKERRQ(ierr);
4974   PetscFunctionReturn(0);
4975 }
4976 
4977 /*@
4978   DMSetCoordinatesLocal - Sets into the DM a local vector that holds the coordinates
4979 
4980   Not collective
4981 
4982    Input Parameters:
4983 +  dm - the DM
4984 -  c - coordinate vector
4985 
4986   Notes:
4987   The coordinates of ghost points can be set using DMSetCoordinates()
4988   followed by DMGetCoordinatesLocal(). This is intended to enable the
4989   setting of ghost coordinates outside of the domain.
4990 
4991   The vector c should be destroyed by the caller.
4992 
4993   Level: intermediate
4994 
4995 .keywords: distributed array, get, corners, nodes, local indices, coordinates
4996 .seealso: DMGetCoordinatesLocal(), DMSetCoordinates(), DMGetCoordinates(), DMGetCoordinateDM()
4997 @*/
4998 PetscErrorCode DMSetCoordinatesLocal(DM dm, Vec c)
4999 {
5000   PetscErrorCode ierr;
5001 
5002   PetscFunctionBegin;
5003   PetscValidHeaderSpecific(dm,DM_CLASSID,1);
5004   PetscValidHeaderSpecific(c,VEC_CLASSID,2);
5005   ierr = PetscObjectReference((PetscObject) c);CHKERRQ(ierr);
5006   ierr = VecDestroy(&dm->coordinatesLocal);CHKERRQ(ierr);
5007 
5008   dm->coordinatesLocal = c;
5009 
5010   ierr = VecDestroy(&dm->coordinates);CHKERRQ(ierr);
5011   PetscFunctionReturn(0);
5012 }
5013 
5014 /*@
5015   DMGetCoordinates - Gets a global vector with the coordinates associated with the DM.
5016 
5017   Collective on DM
5018 
5019   Input Parameter:
5020 . dm - the DM
5021 
5022   Output Parameter:
5023 . c - global coordinate vector
5024 
5025   Note:
5026   This is a borrowed reference, so the user should NOT destroy this vector
5027 
5028   Each process has only the local coordinates (does NOT have the ghost coordinates).
5029 
5030   For DMDA, in two and three dimensions coordinates are interlaced (x_0,y_0,x_1,y_1,...)
5031   and (x_0,y_0,z_0,x_1,y_1,z_1...)
5032 
5033   Level: intermediate
5034 
5035 .keywords: distributed array, get, corners, nodes, local indices, coordinates
5036 .seealso: DMSetCoordinates(), DMGetCoordinatesLocal(), DMGetCoordinateDM()
5037 @*/
5038 PetscErrorCode DMGetCoordinates(DM dm, Vec *c)
5039 {
5040   PetscErrorCode ierr;
5041 
5042   PetscFunctionBegin;
5043   PetscValidHeaderSpecific(dm,DM_CLASSID,1);
5044   PetscValidPointer(c,2);
5045   if (!dm->coordinates && dm->coordinatesLocal) {
5046     DM cdm = NULL;
5047 
5048     ierr = DMGetCoordinateDM(dm, &cdm);CHKERRQ(ierr);
5049     ierr = DMCreateGlobalVector(cdm, &dm->coordinates);CHKERRQ(ierr);
5050     ierr = PetscObjectSetName((PetscObject) dm->coordinates, "coordinates");CHKERRQ(ierr);
5051     ierr = DMLocalToGlobalBegin(cdm, dm->coordinatesLocal, INSERT_VALUES, dm->coordinates);CHKERRQ(ierr);
5052     ierr = DMLocalToGlobalEnd(cdm, dm->coordinatesLocal, INSERT_VALUES, dm->coordinates);CHKERRQ(ierr);
5053   }
5054   *c = dm->coordinates;
5055   PetscFunctionReturn(0);
5056 }
5057 
5058 /*@
5059   DMGetCoordinatesLocalSetUp - Prepares a local vector of coordinates, so that DMGetCoordinatesLocalNoncollective() can be used as non-collective afterwards.
5060 
5061   Collective on DM
5062 
5063   Input Parameter:
5064 . dm - the DM
5065 
5066   Level: advanced
5067 
5068 .keywords: distributed array, get, corners, nodes, local indices, coordinates
5069 .seealso: DMGetCoordinatesLocalNoncollective()
5070 @*/
5071 PetscErrorCode DMGetCoordinatesLocalSetUp(DM dm)
5072 {
5073   DM cdm = NULL;
5074   PetscErrorCode ierr;
5075 
5076   PetscFunctionBegin;
5077   PetscValidHeaderSpecific(dm,DM_CLASSID,1);
5078   if (!dm->coordinatesLocal && dm->coordinates) {
5079     ierr = DMGetCoordinateDM(dm, &cdm);CHKERRQ(ierr);
5080     ierr = DMCreateLocalVector(cdm, &dm->coordinatesLocal);CHKERRQ(ierr);
5081     ierr = PetscObjectSetName((PetscObject) dm->coordinatesLocal, "coordinates");CHKERRQ(ierr);
5082     ierr = DMGlobalToLocalBegin(cdm, dm->coordinates, INSERT_VALUES, dm->coordinatesLocal);CHKERRQ(ierr);
5083     ierr = DMGlobalToLocalEnd(cdm, dm->coordinates, INSERT_VALUES, dm->coordinatesLocal);CHKERRQ(ierr);
5084   }
5085   PetscFunctionReturn(0);
5086 }
5087 
5088 /*@
5089   DMGetCoordinatesLocal - Gets a local vector with the coordinates associated with the DM.
5090 
5091   Collective on DM
5092 
5093   Input Parameter:
5094 . dm - the DM
5095 
5096   Output Parameter:
5097 . c - coordinate vector
5098 
5099   Note:
5100   This is a borrowed reference, so the user should NOT destroy this vector
5101 
5102   Each process has the local and ghost coordinates
5103 
5104   For DMDA, in two and three dimensions coordinates are interlaced (x_0,y_0,x_1,y_1,...)
5105   and (x_0,y_0,z_0,x_1,y_1,z_1...)
5106 
5107   Level: intermediate
5108 
5109 .keywords: distributed array, get, corners, nodes, local indices, coordinates
5110 .seealso: DMSetCoordinatesLocal(), DMGetCoordinates(), DMSetCoordinates(), DMGetCoordinateDM(), DMGetCoordinatesLocalNoncollective()
5111 @*/
5112 PetscErrorCode DMGetCoordinatesLocal(DM dm, Vec *c)
5113 {
5114   PetscErrorCode ierr;
5115 
5116   PetscFunctionBegin;
5117   PetscValidHeaderSpecific(dm,DM_CLASSID,1);
5118   PetscValidPointer(c,2);
5119   ierr = DMGetCoordinatesLocalSetUp(dm);CHKERRQ(ierr);
5120   *c = dm->coordinatesLocal;
5121   PetscFunctionReturn(0);
5122 }
5123 
5124 /*@
5125   DMGetCoordinatesLocalNoncollective - Non-collective version of DMGetCoordinatesLocal(). Fails if global coordinates have been set and DMGetCoordinatesLocalSetUp() not called.
5126 
5127   Not collective
5128 
5129   Input Parameter:
5130 . dm - the DM
5131 
5132   Output Parameter:
5133 . c - coordinate vector
5134 
5135   Level: advanced
5136 
5137 .keywords: distributed array, get, corners, nodes, local indices, coordinates
5138 .seealso: DMGetCoordinatesLocalSetUp(), DMGetCoordinatesLocal(), DMSetCoordinatesLocal(), DMGetCoordinates(), DMSetCoordinates(), DMGetCoordinateDM()
5139 @*/
5140 PetscErrorCode DMGetCoordinatesLocalNoncollective(DM dm, Vec *c)
5141 {
5142   PetscFunctionBegin;
5143   PetscValidHeaderSpecific(dm,DM_CLASSID,1);
5144   PetscValidPointer(c,2);
5145   if (!dm->coordinatesLocal && dm->coordinates) SETERRQ(PetscObjectComm((PetscObject)dm), PETSC_ERR_ARG_WRONGSTATE, "DMGetCoordinatesLocalSetUp() has not been called");
5146   *c = dm->coordinatesLocal;
5147   PetscFunctionReturn(0);
5148 }
5149 
5150 /*@
5151   DMGetCoordinatesLocalTuple - Gets a local vector with the coordinates of specified points and section describing its layout.
5152 
5153   Not collective
5154 
5155   Input Parameter:
5156 + dm - the DM
5157 - p - the IS of points whose coordinates will be returned
5158 
5159   Output Parameter:
5160 + pCoordSection - the PetscSection describing the layout of pCoord, i.e. each point corresponds to one point in p, and DOFs correspond to coordinates
5161 - pCoord - the Vec with coordinates of points in p
5162 
5163   Note:
5164   DMGetCoordinatesLocalSetUp() must be called first. This function employs DMGetCoordinatesLocalNoncollective() so it is not collective.
5165 
5166   This creates a new vector, so the user SHOULD destroy this vector
5167 
5168   Each process has the local and ghost coordinates
5169 
5170   For DMDA, in two and three dimensions coordinates are interlaced (x_0,y_0,x_1,y_1,...)
5171   and (x_0,y_0,z_0,x_1,y_1,z_1...)
5172 
5173   Level: advanced
5174 
5175 .keywords: distributed array, get, corners, nodes, local indices, coordinates
5176 .seealso: DMSetCoordinatesLocal(), DMGetCoordinatesLocal(), DMGetCoordinatesLocalNoncollective(), DMGetCoordinatesLocalSetUp(), DMGetCoordinates(), DMSetCoordinates(), DMGetCoordinateDM()
5177 @*/
5178 PetscErrorCode DMGetCoordinatesLocalTuple(DM dm, IS p, PetscSection *pCoordSection, Vec *pCoord)
5179 {
5180   PetscSection        cs, newcs;
5181   Vec                 coords;
5182   const PetscScalar   *arr;
5183   PetscScalar         *newarr=NULL;
5184   PetscInt            n;
5185   PetscErrorCode      ierr;
5186 
5187   PetscFunctionBegin;
5188   PetscValidHeaderSpecific(dm, DM_CLASSID, 1);
5189   PetscValidHeaderSpecific(p, IS_CLASSID, 2);
5190   if (pCoordSection) PetscValidPointer(pCoordSection, 3);
5191   if (pCoord) PetscValidPointer(pCoord, 4);
5192   if (!dm->coordinatesLocal) SETERRQ(PetscObjectComm((PetscObject)dm), PETSC_ERR_ARG_WRONGSTATE, "DMGetCoordinatesLocalSetUp() has not been called or coordinates not set");
5193   if (!dm->coordinateDM || !dm->coordinateDM->defaultSection) SETERRQ(PetscObjectComm((PetscObject)dm), PETSC_ERR_ARG_WRONGSTATE, "DM not supported");
5194   cs = dm->coordinateDM->defaultSection;
5195   coords = dm->coordinatesLocal;
5196   ierr = VecGetArrayRead(coords, &arr);CHKERRQ(ierr);
5197   ierr = PetscSectionExtractDofsFromArray(cs, MPIU_SCALAR, arr, p, &newcs, pCoord ? ((void**)&newarr) : NULL);CHKERRQ(ierr);
5198   ierr = VecRestoreArrayRead(coords, &arr);CHKERRQ(ierr);
5199   if (pCoord) {
5200     ierr = PetscSectionGetStorageSize(newcs, &n);CHKERRQ(ierr);
5201     /* set array in two steps to mimic PETSC_OWN_POINTER */
5202     ierr = VecCreateSeqWithArray(PetscObjectComm((PetscObject)p), 1, n, NULL, pCoord);CHKERRQ(ierr);
5203     ierr = VecReplaceArray(*pCoord, newarr);CHKERRQ(ierr);
5204   } else {
5205     ierr = PetscFree(newarr);CHKERRQ(ierr);
5206   }
5207   if (pCoordSection) {*pCoordSection = newcs;}
5208   else               {ierr = PetscSectionDestroy(&newcs);CHKERRQ(ierr);}
5209   PetscFunctionReturn(0);
5210 }
5211 
5212 PetscErrorCode DMGetCoordinateField(DM dm, DMField *field)
5213 {
5214   PetscErrorCode ierr;
5215 
5216   PetscFunctionBegin;
5217   PetscValidHeaderSpecific(dm,DM_CLASSID,1);
5218   PetscValidPointer(field,2);
5219   if (!dm->coordinateField) {
5220     if (dm->ops->createcoordinatefield) {
5221       ierr = (*dm->ops->createcoordinatefield)(dm,&dm->coordinateField);CHKERRQ(ierr);
5222     }
5223   }
5224   *field = dm->coordinateField;
5225   PetscFunctionReturn(0);
5226 }
5227 
5228 PetscErrorCode DMSetCoordinateField(DM dm, DMField field)
5229 {
5230   PetscErrorCode ierr;
5231 
5232   PetscFunctionBegin;
5233   PetscValidHeaderSpecific(dm,DM_CLASSID,1);
5234   if (field) PetscValidHeaderSpecific(field,DMFIELD_CLASSID,2);
5235   ierr = PetscObjectReference((PetscObject)field);CHKERRQ(ierr);
5236   ierr = DMFieldDestroy(&dm->coordinateField);CHKERRQ(ierr);
5237   dm->coordinateField = field;
5238   PetscFunctionReturn(0);
5239 }
5240 
5241 /*@
5242   DMGetCoordinateDM - Gets the DM that prescribes coordinate layout and scatters between global and local coordinates
5243 
5244   Collective on DM
5245 
5246   Input Parameter:
5247 . dm - the DM
5248 
5249   Output Parameter:
5250 . cdm - coordinate DM
5251 
5252   Level: intermediate
5253 
5254 .keywords: distributed array, get, corners, nodes, local indices, coordinates
5255 .seealso: DMSetCoordinateDM(), DMSetCoordinates(), DMSetCoordinatesLocal(), DMGetCoordinates(), DMGetCoordinatesLocal()
5256 @*/
5257 PetscErrorCode DMGetCoordinateDM(DM dm, DM *cdm)
5258 {
5259   PetscErrorCode ierr;
5260 
5261   PetscFunctionBegin;
5262   PetscValidHeaderSpecific(dm,DM_CLASSID,1);
5263   PetscValidPointer(cdm,2);
5264   if (!dm->coordinateDM) {
5265     DM cdm;
5266 
5267     if (!dm->ops->createcoordinatedm) SETERRQ(PetscObjectComm((PetscObject)dm), PETSC_ERR_SUP, "Unable to create coordinates for this DM");
5268     ierr = (*dm->ops->createcoordinatedm)(dm, &cdm);CHKERRQ(ierr);
5269     /* Just in case the DM sets the coordinate DM when creating it (DMP4est can do this, because it may not setup
5270      * until the call to CreateCoordinateDM) */
5271     ierr = DMDestroy(&dm->coordinateDM);CHKERRQ(ierr);
5272     dm->coordinateDM = cdm;
5273   }
5274   *cdm = dm->coordinateDM;
5275   PetscFunctionReturn(0);
5276 }
5277 
5278 /*@
5279   DMSetCoordinateDM - Sets the DM that prescribes coordinate layout and scatters between global and local coordinates
5280 
5281   Logically Collective on DM
5282 
5283   Input Parameters:
5284 + dm - the DM
5285 - cdm - coordinate DM
5286 
5287   Level: intermediate
5288 
5289 .keywords: distributed array, get, corners, nodes, local indices, coordinates
5290 .seealso: DMGetCoordinateDM(), DMSetCoordinates(), DMSetCoordinatesLocal(), DMGetCoordinates(), DMGetCoordinatesLocal()
5291 @*/
5292 PetscErrorCode DMSetCoordinateDM(DM dm, DM cdm)
5293 {
5294   PetscErrorCode ierr;
5295 
5296   PetscFunctionBegin;
5297   PetscValidHeaderSpecific(dm,DM_CLASSID,1);
5298   PetscValidHeaderSpecific(cdm,DM_CLASSID,2);
5299   ierr = PetscObjectReference((PetscObject)cdm);CHKERRQ(ierr);
5300   ierr = DMDestroy(&dm->coordinateDM);CHKERRQ(ierr);
5301   dm->coordinateDM = cdm;
5302   PetscFunctionReturn(0);
5303 }
5304 
5305 /*@
5306   DMGetCoordinateDim - Retrieve the dimension of embedding space for coordinate values.
5307 
5308   Not Collective
5309 
5310   Input Parameter:
5311 . dm - The DM object
5312 
5313   Output Parameter:
5314 . dim - The embedding dimension
5315 
5316   Level: intermediate
5317 
5318 .keywords: mesh, coordinates
5319 .seealso: DMSetCoordinateDim(), DMGetCoordinateSection(), DMGetCoordinateDM(), DMGetSection(), DMSetSection()
5320 @*/
5321 PetscErrorCode DMGetCoordinateDim(DM dm, PetscInt *dim)
5322 {
5323   PetscFunctionBegin;
5324   PetscValidHeaderSpecific(dm, DM_CLASSID, 1);
5325   PetscValidPointer(dim, 2);
5326   if (dm->dimEmbed == PETSC_DEFAULT) {
5327     dm->dimEmbed = dm->dim;
5328   }
5329   *dim = dm->dimEmbed;
5330   PetscFunctionReturn(0);
5331 }
5332 
5333 /*@
5334   DMSetCoordinateDim - Set the dimension of the embedding space for coordinate values.
5335 
5336   Not Collective
5337 
5338   Input Parameters:
5339 + dm  - The DM object
5340 - dim - The embedding dimension
5341 
5342   Level: intermediate
5343 
5344 .keywords: mesh, coordinates
5345 .seealso: DMGetCoordinateDim(), DMSetCoordinateSection(), DMGetCoordinateSection(), DMGetSection(), DMSetSection()
5346 @*/
5347 PetscErrorCode DMSetCoordinateDim(DM dm, PetscInt dim)
5348 {
5349   PetscDS        ds;
5350   PetscErrorCode ierr;
5351 
5352   PetscFunctionBegin;
5353   PetscValidHeaderSpecific(dm,DM_CLASSID,1);
5354   dm->dimEmbed = dim;
5355   ierr = DMGetDS(dm, &ds);CHKERRQ(ierr);
5356   ierr = PetscDSSetCoordinateDimension(ds, dim);CHKERRQ(ierr);
5357   PetscFunctionReturn(0);
5358 }
5359 
5360 /*@
5361   DMGetCoordinateSection - Retrieve the layout of coordinate values over the mesh.
5362 
5363   Collective on DM
5364 
5365   Input Parameter:
5366 . dm - The DM object
5367 
5368   Output Parameter:
5369 . section - The PetscSection object
5370 
5371   Level: intermediate
5372 
5373 .keywords: mesh, coordinates
5374 .seealso: DMGetCoordinateDM(), DMGetSection(), DMSetSection()
5375 @*/
5376 PetscErrorCode DMGetCoordinateSection(DM dm, PetscSection *section)
5377 {
5378   DM             cdm;
5379   PetscErrorCode ierr;
5380 
5381   PetscFunctionBegin;
5382   PetscValidHeaderSpecific(dm, DM_CLASSID, 1);
5383   PetscValidPointer(section, 2);
5384   ierr = DMGetCoordinateDM(dm, &cdm);CHKERRQ(ierr);
5385   ierr = DMGetSection(cdm, section);CHKERRQ(ierr);
5386   PetscFunctionReturn(0);
5387 }
5388 
5389 /*@
5390   DMSetCoordinateSection - Set the layout of coordinate values over the mesh.
5391 
5392   Not Collective
5393 
5394   Input Parameters:
5395 + dm      - The DM object
5396 . dim     - The embedding dimension, or PETSC_DETERMINE
5397 - section - The PetscSection object
5398 
5399   Level: intermediate
5400 
5401 .keywords: mesh, coordinates
5402 .seealso: DMGetCoordinateSection(), DMGetSection(), DMSetSection()
5403 @*/
5404 PetscErrorCode DMSetCoordinateSection(DM dm, PetscInt dim, PetscSection section)
5405 {
5406   DM             cdm;
5407   PetscErrorCode ierr;
5408 
5409   PetscFunctionBegin;
5410   PetscValidHeaderSpecific(dm,DM_CLASSID,1);
5411   PetscValidHeaderSpecific(section,PETSC_SECTION_CLASSID,3);
5412   ierr = DMGetCoordinateDM(dm, &cdm);CHKERRQ(ierr);
5413   ierr = DMSetSection(cdm, section);CHKERRQ(ierr);
5414   if (dim == PETSC_DETERMINE) {
5415     PetscInt d = PETSC_DEFAULT;
5416     PetscInt pStart, pEnd, vStart, vEnd, v, dd;
5417 
5418     ierr = PetscSectionGetChart(section, &pStart, &pEnd);CHKERRQ(ierr);
5419     ierr = DMGetDimPoints(dm, 0, &vStart, &vEnd);CHKERRQ(ierr);
5420     pStart = PetscMax(vStart, pStart);
5421     pEnd   = PetscMin(vEnd, pEnd);
5422     for (v = pStart; v < pEnd; ++v) {
5423       ierr = PetscSectionGetDof(section, v, &dd);CHKERRQ(ierr);
5424       if (dd) {d = dd; break;}
5425     }
5426     if (d >= 0) {ierr = DMSetCoordinateDim(dm, d);CHKERRQ(ierr);}
5427   }
5428   PetscFunctionReturn(0);
5429 }
5430 
5431 /*@C
5432   DMGetPeriodicity - Get the description of mesh periodicity
5433 
5434   Input Parameters:
5435 . dm      - The DM object
5436 
5437   Output Parameters:
5438 + per     - Whether the DM is periodic or not
5439 . maxCell - Over distances greater than this, we can assume a point has crossed over to another sheet, when trying to localize cell coordinates
5440 . L       - If we assume the mesh is a torus, this is the length of each coordinate
5441 - bd      - This describes the type of periodicity in each topological dimension
5442 
5443   Level: developer
5444 
5445 .seealso: DMGetPeriodicity()
5446 @*/
5447 PetscErrorCode DMGetPeriodicity(DM dm, PetscBool *per, const PetscReal **maxCell, const PetscReal **L, const DMBoundaryType **bd)
5448 {
5449   PetscFunctionBegin;
5450   PetscValidHeaderSpecific(dm,DM_CLASSID,1);
5451   if (per)     *per     = dm->periodic;
5452   if (L)       *L       = dm->L;
5453   if (maxCell) *maxCell = dm->maxCell;
5454   if (bd)      *bd      = dm->bdtype;
5455   PetscFunctionReturn(0);
5456 }
5457 
5458 /*@C
5459   DMSetPeriodicity - Set the description of mesh periodicity
5460 
5461   Input Parameters:
5462 + dm      - The DM object
5463 . per     - Whether the DM is periodic or not. If maxCell is not provided, coordinates need to be localized
5464 . maxCell - Over distances greater than this, we can assume a point has crossed over to another sheet, when trying to localize cell coordinates
5465 . L       - If we assume the mesh is a torus, this is the length of each coordinate
5466 - bd      - This describes the type of periodicity in each topological dimension
5467 
5468   Level: developer
5469 
5470 .seealso: DMGetPeriodicity()
5471 @*/
5472 PetscErrorCode DMSetPeriodicity(DM dm, PetscBool per, const PetscReal maxCell[], const PetscReal L[], const DMBoundaryType bd[])
5473 {
5474   PetscInt       dim, d;
5475   PetscErrorCode ierr;
5476 
5477   PetscFunctionBegin;
5478   PetscValidHeaderSpecific(dm,DM_CLASSID,1);
5479   PetscValidLogicalCollectiveBool(dm,per,2);
5480   if (maxCell) {
5481     PetscValidPointer(maxCell,3);
5482     PetscValidPointer(L,4);
5483     PetscValidPointer(bd,5);
5484   }
5485   ierr = PetscFree3(dm->L,dm->maxCell,dm->bdtype);CHKERRQ(ierr);
5486   ierr = DMGetDimension(dm, &dim);CHKERRQ(ierr);
5487   if (maxCell) {
5488     ierr = PetscMalloc3(dim,&dm->L,dim,&dm->maxCell,dim,&dm->bdtype);CHKERRQ(ierr);
5489     for (d = 0; d < dim; ++d) {dm->L[d] = L[d]; dm->maxCell[d] = maxCell[d]; dm->bdtype[d] = bd[d];}
5490   }
5491   dm->periodic = per;
5492   PetscFunctionReturn(0);
5493 }
5494 
5495 /*@
5496   DMLocalizeCoordinate - If a mesh is periodic (a torus with lengths L_i, some of which can be infinite), project the coordinate onto [0, L_i) in each dimension.
5497 
5498   Input Parameters:
5499 + dm     - The DM
5500 . in     - The input coordinate point (dim numbers)
5501 - endpoint - Include the endpoint L_i
5502 
5503   Output Parameter:
5504 . out - The localized coordinate point
5505 
5506   Level: developer
5507 
5508 .seealso: DMLocalizeCoordinates(), DMLocalizeAddCoordinate()
5509 @*/
5510 PetscErrorCode DMLocalizeCoordinate(DM dm, const PetscScalar in[], PetscBool endpoint, PetscScalar out[])
5511 {
5512   PetscInt       dim, d;
5513   PetscErrorCode ierr;
5514 
5515   PetscFunctionBegin;
5516   ierr = DMGetCoordinateDim(dm, &dim);CHKERRQ(ierr);
5517   if (!dm->maxCell) {
5518     for (d = 0; d < dim; ++d) out[d] = in[d];
5519   } else {
5520     if (endpoint) {
5521       for (d = 0; d < dim; ++d) {
5522         if ((PetscAbsReal(PetscRealPart(in[d])/dm->L[d] - PetscFloorReal(PetscRealPart(in[d])/dm->L[d])) < PETSC_SMALL) && (PetscRealPart(in[d])/dm->L[d] > PETSC_SMALL)) {
5523           out[d] = in[d] - dm->L[d]*(PetscFloorReal(PetscRealPart(in[d])/dm->L[d]) - 1);
5524         } else {
5525           out[d] = in[d] - dm->L[d]*PetscFloorReal(PetscRealPart(in[d])/dm->L[d]);
5526         }
5527       }
5528     } else {
5529       for (d = 0; d < dim; ++d) {
5530         out[d] = in[d] - dm->L[d]*PetscFloorReal(PetscRealPart(in[d])/dm->L[d]);
5531       }
5532     }
5533   }
5534   PetscFunctionReturn(0);
5535 }
5536 
5537 /*
5538   DMLocalizeCoordinate_Internal - If a mesh is periodic, and the input point is far from the anchor, pick the coordinate sheet of the torus which moves it closer.
5539 
5540   Input Parameters:
5541 + dm     - The DM
5542 . dim    - The spatial dimension
5543 . anchor - The anchor point, the input point can be no more than maxCell away from it
5544 - in     - The input coordinate point (dim numbers)
5545 
5546   Output Parameter:
5547 . out - The localized coordinate point
5548 
5549   Level: developer
5550 
5551   Note: This is meant to get a set of coordinates close to each other, as in a cell. The anchor is usually the one of the vertices on a containing cell
5552 
5553 .seealso: DMLocalizeCoordinates(), DMLocalizeAddCoordinate()
5554 */
5555 PetscErrorCode DMLocalizeCoordinate_Internal(DM dm, PetscInt dim, const PetscScalar anchor[], const PetscScalar in[], PetscScalar out[])
5556 {
5557   PetscInt d;
5558 
5559   PetscFunctionBegin;
5560   if (!dm->maxCell) {
5561     for (d = 0; d < dim; ++d) out[d] = in[d];
5562   } else {
5563     for (d = 0; d < dim; ++d) {
5564       if ((dm->bdtype[d] != DM_BOUNDARY_NONE) && (PetscAbsScalar(anchor[d] - in[d]) > dm->maxCell[d])) {
5565         out[d] = PetscRealPart(anchor[d]) > PetscRealPart(in[d]) ? dm->L[d] + in[d] : in[d] - dm->L[d];
5566       } else {
5567         out[d] = in[d];
5568       }
5569     }
5570   }
5571   PetscFunctionReturn(0);
5572 }
5573 PetscErrorCode DMLocalizeCoordinateReal_Internal(DM dm, PetscInt dim, const PetscReal anchor[], const PetscReal in[], PetscReal out[])
5574 {
5575   PetscInt d;
5576 
5577   PetscFunctionBegin;
5578   if (!dm->maxCell) {
5579     for (d = 0; d < dim; ++d) out[d] = in[d];
5580   } else {
5581     for (d = 0; d < dim; ++d) {
5582       if ((dm->bdtype[d] != DM_BOUNDARY_NONE) && (PetscAbsReal(anchor[d] - in[d]) > dm->maxCell[d])) {
5583         out[d] = anchor[d] > in[d] ? dm->L[d] + in[d] : in[d] - dm->L[d];
5584       } else {
5585         out[d] = in[d];
5586       }
5587     }
5588   }
5589   PetscFunctionReturn(0);
5590 }
5591 
5592 /*
5593   DMLocalizeAddCoordinate_Internal - If a mesh is periodic, and the input point is far from the anchor, pick the coordinate sheet of the torus which moves it closer.
5594 
5595   Input Parameters:
5596 + dm     - The DM
5597 . dim    - The spatial dimension
5598 . anchor - The anchor point, the input point can be no more than maxCell away from it
5599 . in     - The input coordinate delta (dim numbers)
5600 - out    - The input coordinate point (dim numbers)
5601 
5602   Output Parameter:
5603 . out    - The localized coordinate in + out
5604 
5605   Level: developer
5606 
5607   Note: This is meant to get a set of coordinates close to each other, as in a cell. The anchor is usually the one of the vertices on a containing cell
5608 
5609 .seealso: DMLocalizeCoordinates(), DMLocalizeCoordinate()
5610 */
5611 PetscErrorCode DMLocalizeAddCoordinate_Internal(DM dm, PetscInt dim, const PetscScalar anchor[], const PetscScalar in[], PetscScalar out[])
5612 {
5613   PetscInt d;
5614 
5615   PetscFunctionBegin;
5616   if (!dm->maxCell) {
5617     for (d = 0; d < dim; ++d) out[d] += in[d];
5618   } else {
5619     for (d = 0; d < dim; ++d) {
5620       if ((dm->bdtype[d] != DM_BOUNDARY_NONE) && (PetscAbsScalar(anchor[d] - in[d]) > dm->maxCell[d])) {
5621         out[d] += PetscRealPart(anchor[d]) > PetscRealPart(in[d]) ? dm->L[d] + in[d] : in[d] - dm->L[d];
5622       } else {
5623         out[d] += in[d];
5624       }
5625     }
5626   }
5627   PetscFunctionReturn(0);
5628 }
5629 
5630 /*@
5631   DMGetCoordinatesLocalizedLocal - Check if the DM coordinates have been localized for cells on this process
5632 
5633   Not collective
5634 
5635   Input Parameter:
5636 . dm - The DM
5637 
5638   Output Parameter:
5639   areLocalized - True if localized
5640 
5641   Level: developer
5642 
5643 .seealso: DMLocalizeCoordinates(), DMGetCoordinatesLocalized(), DMSetPeriodicity()
5644 @*/
5645 PetscErrorCode DMGetCoordinatesLocalizedLocal(DM dm,PetscBool *areLocalized)
5646 {
5647   DM             cdm;
5648   PetscSection   coordSection;
5649   PetscInt       cStart, cEnd, sStart, sEnd, c, dof;
5650   PetscBool      isPlex, alreadyLocalized;
5651   PetscErrorCode ierr;
5652 
5653   PetscFunctionBegin;
5654   PetscValidHeaderSpecific(dm, DM_CLASSID, 1);
5655   PetscValidPointer(areLocalized, 2);
5656   *areLocalized = PETSC_FALSE;
5657 
5658   /* We need some generic way of refering to cells/vertices */
5659   ierr = DMGetCoordinateDM(dm, &cdm);CHKERRQ(ierr);
5660   ierr = DMGetCoordinateSection(dm, &coordSection);CHKERRQ(ierr);
5661   ierr = PetscObjectTypeCompare((PetscObject) cdm, DMPLEX, &isPlex);CHKERRQ(ierr);
5662   if (!isPlex) SETERRQ(PetscObjectComm((PetscObject) cdm), PETSC_ERR_ARG_WRONG, "Coordinate localization requires a DMPLEX coordinate DM");
5663 
5664   ierr = DMPlexGetHeightStratum(cdm, 0, &cStart, &cEnd);CHKERRQ(ierr);
5665   ierr = PetscSectionGetChart(coordSection, &sStart, &sEnd);CHKERRQ(ierr);
5666   alreadyLocalized = PETSC_FALSE;
5667   for (c = cStart; c < cEnd; ++c) {
5668     if (c < sStart || c >= sEnd) continue;
5669     ierr = PetscSectionGetDof(coordSection, c, &dof);CHKERRQ(ierr);
5670     if (dof) { alreadyLocalized = PETSC_TRUE; break; }
5671   }
5672   *areLocalized = alreadyLocalized;
5673   PetscFunctionReturn(0);
5674 }
5675 
5676 /*@
5677   DMGetCoordinatesLocalized - Check if the DM coordinates have been localized for cells
5678 
5679   Collective on dm
5680 
5681   Input Parameter:
5682 . dm - The DM
5683 
5684   Output Parameter:
5685   areLocalized - True if localized
5686 
5687   Level: developer
5688 
5689 .seealso: DMLocalizeCoordinates(), DMSetPeriodicity(), DMGetCoordinatesLocalizedLocal()
5690 @*/
5691 PetscErrorCode DMGetCoordinatesLocalized(DM dm,PetscBool *areLocalized)
5692 {
5693   PetscBool      localized;
5694   PetscErrorCode ierr;
5695 
5696   PetscFunctionBegin;
5697   PetscValidHeaderSpecific(dm, DM_CLASSID, 1);
5698   PetscValidPointer(areLocalized, 2);
5699   ierr = DMGetCoordinatesLocalizedLocal(dm,&localized);CHKERRQ(ierr);
5700   ierr = MPIU_Allreduce(&localized,areLocalized,1,MPIU_BOOL,MPI_LOR,PetscObjectComm((PetscObject)dm));CHKERRQ(ierr);
5701   PetscFunctionReturn(0);
5702 }
5703 
5704 /*@
5705   DMLocalizeCoordinates - If a mesh is periodic, create local coordinates for cells having periodic faces
5706 
5707   Collective on dm
5708 
5709   Input Parameter:
5710 . dm - The DM
5711 
5712   Level: developer
5713 
5714 .seealso: DMSetPeriodicity(), DMLocalizeCoordinate(), DMLocalizeAddCoordinate()
5715 @*/
5716 PetscErrorCode DMLocalizeCoordinates(DM dm)
5717 {
5718   DM             cdm;
5719   PetscSection   coordSection, cSection;
5720   Vec            coordinates,  cVec;
5721   PetscScalar   *coords, *coords2, *anchor, *localized;
5722   PetscInt       Nc, vStart, vEnd, v, sStart, sEnd, newStart = PETSC_MAX_INT, newEnd = PETSC_MIN_INT, dof, d, off, off2, bs, coordSize;
5723   PetscBool      alreadyLocalized, alreadyLocalizedGlobal;
5724   PetscInt       maxHeight = 0, h;
5725   PetscInt       *pStart = NULL, *pEnd = NULL;
5726   PetscErrorCode ierr;
5727 
5728   PetscFunctionBegin;
5729   PetscValidHeaderSpecific(dm, DM_CLASSID, 1);
5730   if (!dm->periodic) PetscFunctionReturn(0);
5731   ierr = DMGetCoordinatesLocalized(dm, &alreadyLocalized);CHKERRQ(ierr);
5732   if (alreadyLocalized) PetscFunctionReturn(0);
5733 
5734   /* We need some generic way of refering to cells/vertices */
5735   ierr = DMGetCoordinateDM(dm, &cdm);CHKERRQ(ierr);
5736   {
5737     PetscBool isplex;
5738 
5739     ierr = PetscObjectTypeCompare((PetscObject) cdm, DMPLEX, &isplex);CHKERRQ(ierr);
5740     if (isplex) {
5741       ierr = DMPlexGetDepthStratum(cdm, 0, &vStart, &vEnd);CHKERRQ(ierr);
5742       ierr = DMPlexGetMaxProjectionHeight(cdm,&maxHeight);CHKERRQ(ierr);
5743       ierr = DMGetWorkArray(dm,2*(maxHeight + 1),MPIU_INT,&pStart);CHKERRQ(ierr);
5744       pEnd = &pStart[maxHeight + 1];
5745       newStart = vStart;
5746       newEnd   = vEnd;
5747       for (h = 0; h <= maxHeight; h++) {
5748         ierr = DMPlexGetHeightStratum(cdm, h, &pStart[h], &pEnd[h]);CHKERRQ(ierr);
5749         newStart = PetscMin(newStart,pStart[h]);
5750         newEnd   = PetscMax(newEnd,pEnd[h]);
5751       }
5752     } else SETERRQ(PetscObjectComm((PetscObject) cdm), PETSC_ERR_ARG_WRONG, "Coordinate localization requires a DMPLEX coordinate DM");
5753   }
5754   ierr = DMGetCoordinatesLocal(dm, &coordinates);CHKERRQ(ierr);
5755   if (!coordinates) SETERRQ(PetscObjectComm((PetscObject)dm),PETSC_ERR_SUP,"Missing local coordinates vector");
5756   ierr = DMGetCoordinateSection(dm, &coordSection);CHKERRQ(ierr);
5757   ierr = VecGetBlockSize(coordinates, &bs);CHKERRQ(ierr);
5758   ierr = PetscSectionGetChart(coordSection,&sStart,&sEnd);CHKERRQ(ierr);
5759 
5760   ierr = PetscSectionCreate(PetscObjectComm((PetscObject) dm), &cSection);CHKERRQ(ierr);
5761   ierr = PetscSectionSetNumFields(cSection, 1);CHKERRQ(ierr);
5762   ierr = PetscSectionGetFieldComponents(coordSection, 0, &Nc);CHKERRQ(ierr);
5763   ierr = PetscSectionSetFieldComponents(cSection, 0, Nc);CHKERRQ(ierr);
5764   ierr = PetscSectionSetChart(cSection, newStart, newEnd);CHKERRQ(ierr);
5765 
5766   ierr = DMGetWorkArray(dm, 2 * bs, MPIU_SCALAR, &anchor);CHKERRQ(ierr);
5767   localized = &anchor[bs];
5768   alreadyLocalized = alreadyLocalizedGlobal = PETSC_TRUE;
5769   for (h = 0; h <= maxHeight; h++) {
5770     PetscInt cStart = pStart[h], cEnd = pEnd[h], c;
5771 
5772     for (c = cStart; c < cEnd; ++c) {
5773       PetscScalar *cellCoords = NULL;
5774       PetscInt     b;
5775 
5776       if (c < sStart || c >= sEnd) alreadyLocalized = PETSC_FALSE;
5777       ierr = DMPlexVecGetClosure(cdm, coordSection, coordinates, c, &dof, &cellCoords);CHKERRQ(ierr);
5778       for (b = 0; b < bs; ++b) anchor[b] = cellCoords[b];
5779       for (d = 0; d < dof/bs; ++d) {
5780         ierr = DMLocalizeCoordinate_Internal(dm, bs, anchor, &cellCoords[d*bs], localized);CHKERRQ(ierr);
5781         for (b = 0; b < bs; b++) {
5782           if (cellCoords[d*bs + b] != localized[b]) break;
5783         }
5784         if (b < bs) break;
5785       }
5786       if (d < dof/bs) {
5787         if (c >= sStart && c < sEnd) {
5788           PetscInt cdof;
5789 
5790           ierr = PetscSectionGetDof(coordSection, c, &cdof);CHKERRQ(ierr);
5791           if (cdof != dof) alreadyLocalized = PETSC_FALSE;
5792         }
5793         ierr = PetscSectionSetDof(cSection, c, dof);CHKERRQ(ierr);
5794         ierr = PetscSectionSetFieldDof(cSection, c, 0, dof);CHKERRQ(ierr);
5795       }
5796       ierr = DMPlexVecRestoreClosure(cdm, coordSection, coordinates, c, &dof, &cellCoords);CHKERRQ(ierr);
5797     }
5798   }
5799   ierr = MPI_Allreduce(&alreadyLocalized,&alreadyLocalizedGlobal,1,MPIU_BOOL,MPI_LAND,PetscObjectComm((PetscObject)dm));CHKERRQ(ierr);
5800   if (alreadyLocalizedGlobal) {
5801     ierr = DMRestoreWorkArray(dm, 2 * bs, MPIU_SCALAR, &anchor);CHKERRQ(ierr);
5802     ierr = PetscSectionDestroy(&cSection);CHKERRQ(ierr);
5803     ierr = DMRestoreWorkArray(dm,2*(maxHeight + 1),MPIU_INT,&pStart);CHKERRQ(ierr);
5804     PetscFunctionReturn(0);
5805   }
5806   for (v = vStart; v < vEnd; ++v) {
5807     ierr = PetscSectionGetDof(coordSection, v, &dof);CHKERRQ(ierr);
5808     ierr = PetscSectionSetDof(cSection, v, dof);CHKERRQ(ierr);
5809     ierr = PetscSectionSetFieldDof(cSection, v, 0, dof);CHKERRQ(ierr);
5810   }
5811   ierr = PetscSectionSetUp(cSection);CHKERRQ(ierr);
5812   ierr = PetscSectionGetStorageSize(cSection, &coordSize);CHKERRQ(ierr);
5813   ierr = VecCreate(PETSC_COMM_SELF, &cVec);CHKERRQ(ierr);
5814   ierr = PetscObjectSetName((PetscObject)cVec,"coordinates");CHKERRQ(ierr);
5815   ierr = VecSetBlockSize(cVec, bs);CHKERRQ(ierr);
5816   ierr = VecSetSizes(cVec, coordSize, PETSC_DETERMINE);CHKERRQ(ierr);
5817   ierr = VecSetType(cVec, VECSTANDARD);CHKERRQ(ierr);
5818   ierr = VecGetArrayRead(coordinates, (const PetscScalar**)&coords);CHKERRQ(ierr);
5819   ierr = VecGetArray(cVec, &coords2);CHKERRQ(ierr);
5820   for (v = vStart; v < vEnd; ++v) {
5821     ierr = PetscSectionGetDof(coordSection, v, &dof);CHKERRQ(ierr);
5822     ierr = PetscSectionGetOffset(coordSection, v, &off);CHKERRQ(ierr);
5823     ierr = PetscSectionGetOffset(cSection,     v, &off2);CHKERRQ(ierr);
5824     for (d = 0; d < dof; ++d) coords2[off2+d] = coords[off+d];
5825   }
5826   for (h = 0; h <= maxHeight; h++) {
5827     PetscInt cStart = pStart[h], cEnd = pEnd[h], c;
5828 
5829     for (c = cStart; c < cEnd; ++c) {
5830       PetscScalar *cellCoords = NULL;
5831       PetscInt     b, cdof;
5832 
5833       ierr = PetscSectionGetDof(cSection,c,&cdof);CHKERRQ(ierr);
5834       if (!cdof) continue;
5835       ierr = DMPlexVecGetClosure(cdm, coordSection, coordinates, c, &dof, &cellCoords);CHKERRQ(ierr);
5836       ierr = PetscSectionGetOffset(cSection, c, &off2);CHKERRQ(ierr);
5837       for (b = 0; b < bs; ++b) anchor[b] = cellCoords[b];
5838       for (d = 0; d < dof/bs; ++d) {ierr = DMLocalizeCoordinate_Internal(dm, bs, anchor, &cellCoords[d*bs], &coords2[off2+d*bs]);CHKERRQ(ierr);}
5839       ierr = DMPlexVecRestoreClosure(cdm, coordSection, coordinates, c, &dof, &cellCoords);CHKERRQ(ierr);
5840     }
5841   }
5842   ierr = DMRestoreWorkArray(dm, 2 * bs, MPIU_SCALAR, &anchor);CHKERRQ(ierr);
5843   ierr = DMRestoreWorkArray(dm,2*(maxHeight + 1),MPIU_INT,&pStart);CHKERRQ(ierr);
5844   ierr = VecRestoreArrayRead(coordinates, (const PetscScalar**)&coords);CHKERRQ(ierr);
5845   ierr = VecRestoreArray(cVec, &coords2);CHKERRQ(ierr);
5846   ierr = DMSetCoordinateSection(dm, PETSC_DETERMINE, cSection);CHKERRQ(ierr);
5847   ierr = DMSetCoordinatesLocal(dm, cVec);CHKERRQ(ierr);
5848   ierr = VecDestroy(&cVec);CHKERRQ(ierr);
5849   ierr = PetscSectionDestroy(&cSection);CHKERRQ(ierr);
5850   PetscFunctionReturn(0);
5851 }
5852 
5853 /*@
5854   DMLocatePoints - Locate the points in v in the mesh and return a PetscSF of the containing cells
5855 
5856   Collective on Vec v (see explanation below)
5857 
5858   Input Parameters:
5859 + dm - The DM
5860 . v - The Vec of points
5861 . ltype - The type of point location, e.g. DM_POINTLOCATION_NONE or DM_POINTLOCATION_NEAREST
5862 - cells - Points to either NULL, or a PetscSF with guesses for which cells contain each point.
5863 
5864   Output Parameter:
5865 + v - The Vec of points, which now contains the nearest mesh points to the given points if DM_POINTLOCATION_NEAREST is used
5866 - cells - The PetscSF containing the ranks and local indices of the containing points.
5867 
5868 
5869   Level: developer
5870 
5871   Notes:
5872   To do a search of the local cells of the mesh, v should have PETSC_COMM_SELF as its communicator.
5873   To do a search of all the cells in the distributed mesh, v should have the same communicator as dm.
5874 
5875   If *cellSF is NULL on input, a PetscSF will be created.
5876   If *cellSF is not NULL on input, it should point to an existing PetscSF, whose graph will be used as initial guesses.
5877 
5878   An array that maps each point to its containing cell can be obtained with
5879 
5880 $    const PetscSFNode *cells;
5881 $    PetscInt           nFound;
5882 $    const PetscInt    *found;
5883 $
5884 $    PetscSFGetGraph(cellSF,NULL,&nFound,&found,&cells);
5885 
5886   Where cells[i].rank is the rank of the cell containing point found[i] (or i if found == NULL), and cells[i].index is
5887   the index of the cell in its rank's local numbering.
5888 
5889 .keywords: point location, mesh
5890 .seealso: DMSetCoordinates(), DMSetCoordinatesLocal(), DMGetCoordinates(), DMGetCoordinatesLocal(), DMPointLocationType
5891 @*/
5892 PetscErrorCode DMLocatePoints(DM dm, Vec v, DMPointLocationType ltype, PetscSF *cellSF)
5893 {
5894   PetscErrorCode ierr;
5895 
5896   PetscFunctionBegin;
5897   PetscValidHeaderSpecific(dm,DM_CLASSID,1);
5898   PetscValidHeaderSpecific(v,VEC_CLASSID,2);
5899   PetscValidPointer(cellSF,4);
5900   if (*cellSF) {
5901     PetscMPIInt result;
5902 
5903     PetscValidHeaderSpecific(*cellSF,PETSCSF_CLASSID,4);
5904     ierr = MPI_Comm_compare(PetscObjectComm((PetscObject)v),PetscObjectComm((PetscObject)*cellSF),&result);CHKERRQ(ierr);
5905     if (result != MPI_IDENT && result != MPI_CONGRUENT) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_INCOMP,"cellSF must have a communicator congruent to v's");
5906   } else {
5907     ierr = PetscSFCreate(PetscObjectComm((PetscObject)v),cellSF);CHKERRQ(ierr);
5908   }
5909   ierr = PetscLogEventBegin(DM_LocatePoints,dm,0,0,0);CHKERRQ(ierr);
5910   if (dm->ops->locatepoints) {
5911     ierr = (*dm->ops->locatepoints)(dm,v,ltype,*cellSF);CHKERRQ(ierr);
5912   } else SETERRQ(PetscObjectComm((PetscObject)dm), PETSC_ERR_SUP, "Point location not available for this DM");
5913   ierr = PetscLogEventEnd(DM_LocatePoints,dm,0,0,0);CHKERRQ(ierr);
5914   PetscFunctionReturn(0);
5915 }
5916 
5917 /*@
5918   DMGetOutputDM - Retrieve the DM associated with the layout for output
5919 
5920   Collective on dm
5921 
5922   Input Parameter:
5923 . dm - The original DM
5924 
5925   Output Parameter:
5926 . odm - The DM which provides the layout for output
5927 
5928   Level: intermediate
5929 
5930 .seealso: VecView(), DMGetGlobalSection()
5931 @*/
5932 PetscErrorCode DMGetOutputDM(DM dm, DM *odm)
5933 {
5934   PetscSection   section;
5935   PetscBool      hasConstraints, ghasConstraints;
5936   PetscErrorCode ierr;
5937 
5938   PetscFunctionBegin;
5939   PetscValidHeaderSpecific(dm,DM_CLASSID,1);
5940   PetscValidPointer(odm,2);
5941   ierr = DMGetSection(dm, &section);CHKERRQ(ierr);
5942   ierr = PetscSectionHasConstraints(section, &hasConstraints);CHKERRQ(ierr);
5943   ierr = MPI_Allreduce(&hasConstraints, &ghasConstraints, 1, MPIU_BOOL, MPI_LOR, PetscObjectComm((PetscObject) dm));CHKERRQ(ierr);
5944   if (!ghasConstraints) {
5945     *odm = dm;
5946     PetscFunctionReturn(0);
5947   }
5948   if (!dm->dmBC) {
5949     PetscSection newSection, gsection;
5950     PetscSF      sf;
5951 
5952     ierr = DMClone(dm, &dm->dmBC);CHKERRQ(ierr);
5953     ierr = DMCopyDisc(dm, dm->dmBC);CHKERRQ(ierr);
5954     ierr = PetscSectionClone(section, &newSection);CHKERRQ(ierr);
5955     ierr = DMSetSection(dm->dmBC, newSection);CHKERRQ(ierr);
5956     ierr = PetscSectionDestroy(&newSection);CHKERRQ(ierr);
5957     ierr = DMGetPointSF(dm->dmBC, &sf);CHKERRQ(ierr);
5958     ierr = PetscSectionCreateGlobalSection(section, sf, PETSC_TRUE, PETSC_FALSE, &gsection);CHKERRQ(ierr);
5959     ierr = DMSetGlobalSection(dm->dmBC, gsection);CHKERRQ(ierr);
5960     ierr = PetscSectionDestroy(&gsection);CHKERRQ(ierr);
5961   }
5962   *odm = dm->dmBC;
5963   PetscFunctionReturn(0);
5964 }
5965 
5966 /*@
5967   DMGetOutputSequenceNumber - Retrieve the sequence number/value for output
5968 
5969   Input Parameter:
5970 . dm - The original DM
5971 
5972   Output Parameters:
5973 + num - The output sequence number
5974 - val - The output sequence value
5975 
5976   Level: intermediate
5977 
5978   Note: This is intended for output that should appear in sequence, for instance
5979   a set of timesteps in an HDF5 file, or a set of realizations of a stochastic system.
5980 
5981 .seealso: VecView()
5982 @*/
5983 PetscErrorCode DMGetOutputSequenceNumber(DM dm, PetscInt *num, PetscReal *val)
5984 {
5985   PetscFunctionBegin;
5986   PetscValidHeaderSpecific(dm,DM_CLASSID,1);
5987   if (num) {PetscValidPointer(num,2); *num = dm->outputSequenceNum;}
5988   if (val) {PetscValidPointer(val,3);*val = dm->outputSequenceVal;}
5989   PetscFunctionReturn(0);
5990 }
5991 
5992 /*@
5993   DMSetOutputSequenceNumber - Set the sequence number/value for output
5994 
5995   Input Parameters:
5996 + dm - The original DM
5997 . num - The output sequence number
5998 - val - The output sequence value
5999 
6000   Level: intermediate
6001 
6002   Note: This is intended for output that should appear in sequence, for instance
6003   a set of timesteps in an HDF5 file, or a set of realizations of a stochastic system.
6004 
6005 .seealso: VecView()
6006 @*/
6007 PetscErrorCode DMSetOutputSequenceNumber(DM dm, PetscInt num, PetscReal val)
6008 {
6009   PetscFunctionBegin;
6010   PetscValidHeaderSpecific(dm,DM_CLASSID,1);
6011   dm->outputSequenceNum = num;
6012   dm->outputSequenceVal = val;
6013   PetscFunctionReturn(0);
6014 }
6015 
6016 /*@C
6017   DMOutputSequenceLoad - Retrieve the sequence value from a Viewer
6018 
6019   Input Parameters:
6020 + dm   - The original DM
6021 . name - The sequence name
6022 - num  - The output sequence number
6023 
6024   Output Parameter:
6025 . val  - The output sequence value
6026 
6027   Level: intermediate
6028 
6029   Note: This is intended for output that should appear in sequence, for instance
6030   a set of timesteps in an HDF5 file, or a set of realizations of a stochastic system.
6031 
6032 .seealso: DMGetOutputSequenceNumber(), DMSetOutputSequenceNumber(), VecView()
6033 @*/
6034 PetscErrorCode DMOutputSequenceLoad(DM dm, PetscViewer viewer, const char *name, PetscInt num, PetscReal *val)
6035 {
6036   PetscBool      ishdf5;
6037   PetscErrorCode ierr;
6038 
6039   PetscFunctionBegin;
6040   PetscValidHeaderSpecific(dm,DM_CLASSID,1);
6041   PetscValidHeaderSpecific(viewer,PETSC_VIEWER_CLASSID,2);
6042   PetscValidPointer(val,4);
6043   ierr = PetscObjectTypeCompare((PetscObject) viewer, PETSCVIEWERHDF5, &ishdf5);CHKERRQ(ierr);
6044   if (ishdf5) {
6045 #if defined(PETSC_HAVE_HDF5)
6046     PetscScalar value;
6047 
6048     ierr = DMSequenceLoad_HDF5_Internal(dm, name, num, &value, viewer);CHKERRQ(ierr);
6049     *val = PetscRealPart(value);
6050 #endif
6051   } else SETERRQ(PETSC_COMM_SELF, PETSC_ERR_ARG_WRONG, "Invalid viewer; open viewer with PetscViewerHDF5Open()");
6052   PetscFunctionReturn(0);
6053 }
6054 
6055 /*@
6056   DMGetUseNatural - Get the flag for creating a mapping to the natural order on distribution
6057 
6058   Not collective
6059 
6060   Input Parameter:
6061 . dm - The DM
6062 
6063   Output Parameter:
6064 . useNatural - The flag to build the mapping to a natural order during distribution
6065 
6066   Level: beginner
6067 
6068 .seealso: DMSetUseNatural(), DMCreate()
6069 @*/
6070 PetscErrorCode DMGetUseNatural(DM dm, PetscBool *useNatural)
6071 {
6072   PetscFunctionBegin;
6073   PetscValidHeaderSpecific(dm, DM_CLASSID, 1);
6074   PetscValidPointer(useNatural, 2);
6075   *useNatural = dm->useNatural;
6076   PetscFunctionReturn(0);
6077 }
6078 
6079 /*@
6080   DMSetUseNatural - Set the flag for creating a mapping to the natural order after distribution
6081 
6082   Collective on dm
6083 
6084   Input Parameters:
6085 + dm - The DM
6086 - useNatural - The flag to build the mapping to a natural order during distribution
6087 
6088   Note: This also causes the map to be build after DMCreateSubDM() and DMCreateSuperDM()
6089 
6090   Level: beginner
6091 
6092 .seealso: DMGetUseNatural(), DMCreate(), DMPlexDistribute(), DMCreateSubDM(), DMCreateSuperDM()
6093 @*/
6094 PetscErrorCode DMSetUseNatural(DM dm, PetscBool useNatural)
6095 {
6096   PetscFunctionBegin;
6097   PetscValidHeaderSpecific(dm, DM_CLASSID, 1);
6098   PetscValidLogicalCollectiveBool(dm, useNatural, 2);
6099   dm->useNatural = useNatural;
6100   PetscFunctionReturn(0);
6101 }
6102 
6103 
6104 /*@C
6105   DMCreateLabel - Create a label of the given name if it does not already exist
6106 
6107   Not Collective
6108 
6109   Input Parameters:
6110 + dm   - The DM object
6111 - name - The label name
6112 
6113   Level: intermediate
6114 
6115 .keywords: mesh
6116 .seealso: DMLabelCreate(), DMHasLabel(), DMGetLabelValue(), DMSetLabelValue(), DMGetStratumIS()
6117 @*/
6118 PetscErrorCode DMCreateLabel(DM dm, const char name[])
6119 {
6120   DMLabelLink    next  = dm->labels->next;
6121   PetscBool      flg   = PETSC_FALSE;
6122   const char    *lname;
6123   PetscErrorCode ierr;
6124 
6125   PetscFunctionBegin;
6126   PetscValidHeaderSpecific(dm, DM_CLASSID, 1);
6127   PetscValidCharPointer(name, 2);
6128   while (next) {
6129     ierr = PetscObjectGetName((PetscObject) next->label, &lname);CHKERRQ(ierr);
6130     ierr = PetscStrcmp(name, lname, &flg);CHKERRQ(ierr);
6131     if (flg) break;
6132     next = next->next;
6133   }
6134   if (!flg) {
6135     DMLabelLink tmpLabel;
6136 
6137     ierr = PetscCalloc1(1, &tmpLabel);CHKERRQ(ierr);
6138     ierr = DMLabelCreate(PETSC_COMM_SELF, name, &tmpLabel->label);CHKERRQ(ierr);
6139     tmpLabel->output = PETSC_TRUE;
6140     tmpLabel->next   = dm->labels->next;
6141     dm->labels->next = tmpLabel;
6142   }
6143   PetscFunctionReturn(0);
6144 }
6145 
6146 /*@C
6147   DMGetLabelValue - Get the value in a Sieve Label for the given point, with 0 as the default
6148 
6149   Not Collective
6150 
6151   Input Parameters:
6152 + dm   - The DM object
6153 . name - The label name
6154 - point - The mesh point
6155 
6156   Output Parameter:
6157 . value - The label value for this point, or -1 if the point is not in the label
6158 
6159   Level: beginner
6160 
6161 .keywords: mesh
6162 .seealso: DMLabelGetValue(), DMSetLabelValue(), DMGetStratumIS()
6163 @*/
6164 PetscErrorCode DMGetLabelValue(DM dm, const char name[], PetscInt point, PetscInt *value)
6165 {
6166   DMLabel        label;
6167   PetscErrorCode ierr;
6168 
6169   PetscFunctionBegin;
6170   PetscValidHeaderSpecific(dm, DM_CLASSID, 1);
6171   PetscValidCharPointer(name, 2);
6172   ierr = DMGetLabel(dm, name, &label);CHKERRQ(ierr);
6173   if (!label) SETERRQ1(PETSC_COMM_SELF, PETSC_ERR_ARG_WRONG, "No label named %s was found", name);
6174   ierr = DMLabelGetValue(label, point, value);CHKERRQ(ierr);
6175   PetscFunctionReturn(0);
6176 }
6177 
6178 /*@C
6179   DMSetLabelValue - Add a point to a Sieve Label with given value
6180 
6181   Not Collective
6182 
6183   Input Parameters:
6184 + dm   - The DM object
6185 . name - The label name
6186 . point - The mesh point
6187 - value - The label value for this point
6188 
6189   Output Parameter:
6190 
6191   Level: beginner
6192 
6193 .keywords: mesh
6194 .seealso: DMLabelSetValue(), DMGetStratumIS(), DMClearLabelValue()
6195 @*/
6196 PetscErrorCode DMSetLabelValue(DM dm, const char name[], PetscInt point, PetscInt value)
6197 {
6198   DMLabel        label;
6199   PetscErrorCode ierr;
6200 
6201   PetscFunctionBegin;
6202   PetscValidHeaderSpecific(dm, DM_CLASSID, 1);
6203   PetscValidCharPointer(name, 2);
6204   ierr = DMGetLabel(dm, name, &label);CHKERRQ(ierr);
6205   if (!label) {
6206     ierr = DMCreateLabel(dm, name);CHKERRQ(ierr);
6207     ierr = DMGetLabel(dm, name, &label);CHKERRQ(ierr);
6208   }
6209   ierr = DMLabelSetValue(label, point, value);CHKERRQ(ierr);
6210   PetscFunctionReturn(0);
6211 }
6212 
6213 /*@C
6214   DMClearLabelValue - Remove a point from a Sieve Label with given value
6215 
6216   Not Collective
6217 
6218   Input Parameters:
6219 + dm   - The DM object
6220 . name - The label name
6221 . point - The mesh point
6222 - value - The label value for this point
6223 
6224   Output Parameter:
6225 
6226   Level: beginner
6227 
6228 .keywords: mesh
6229 .seealso: DMLabelClearValue(), DMSetLabelValue(), DMGetStratumIS()
6230 @*/
6231 PetscErrorCode DMClearLabelValue(DM dm, const char name[], PetscInt point, PetscInt value)
6232 {
6233   DMLabel        label;
6234   PetscErrorCode ierr;
6235 
6236   PetscFunctionBegin;
6237   PetscValidHeaderSpecific(dm, DM_CLASSID, 1);
6238   PetscValidCharPointer(name, 2);
6239   ierr = DMGetLabel(dm, name, &label);CHKERRQ(ierr);
6240   if (!label) PetscFunctionReturn(0);
6241   ierr = DMLabelClearValue(label, point, value);CHKERRQ(ierr);
6242   PetscFunctionReturn(0);
6243 }
6244 
6245 /*@C
6246   DMGetLabelSize - Get the number of different integer ids in a Label
6247 
6248   Not Collective
6249 
6250   Input Parameters:
6251 + dm   - The DM object
6252 - name - The label name
6253 
6254   Output Parameter:
6255 . size - The number of different integer ids, or 0 if the label does not exist
6256 
6257   Level: beginner
6258 
6259 .keywords: mesh
6260 .seealso: DMLabelGetNumValues(), DMSetLabelValue()
6261 @*/
6262 PetscErrorCode DMGetLabelSize(DM dm, const char name[], PetscInt *size)
6263 {
6264   DMLabel        label;
6265   PetscErrorCode ierr;
6266 
6267   PetscFunctionBegin;
6268   PetscValidHeaderSpecific(dm, DM_CLASSID, 1);
6269   PetscValidCharPointer(name, 2);
6270   PetscValidPointer(size, 3);
6271   ierr  = DMGetLabel(dm, name, &label);CHKERRQ(ierr);
6272   *size = 0;
6273   if (!label) PetscFunctionReturn(0);
6274   ierr = DMLabelGetNumValues(label, size);CHKERRQ(ierr);
6275   PetscFunctionReturn(0);
6276 }
6277 
6278 /*@C
6279   DMGetLabelIdIS - Get the integer ids in a label
6280 
6281   Not Collective
6282 
6283   Input Parameters:
6284 + mesh - The DM object
6285 - name - The label name
6286 
6287   Output Parameter:
6288 . ids - The integer ids, or NULL if the label does not exist
6289 
6290   Level: beginner
6291 
6292 .keywords: mesh
6293 .seealso: DMLabelGetValueIS(), DMGetLabelSize()
6294 @*/
6295 PetscErrorCode DMGetLabelIdIS(DM dm, const char name[], IS *ids)
6296 {
6297   DMLabel        label;
6298   PetscErrorCode ierr;
6299 
6300   PetscFunctionBegin;
6301   PetscValidHeaderSpecific(dm, DM_CLASSID, 1);
6302   PetscValidCharPointer(name, 2);
6303   PetscValidPointer(ids, 3);
6304   ierr = DMGetLabel(dm, name, &label);CHKERRQ(ierr);
6305   *ids = NULL;
6306  if (label) {
6307     ierr = DMLabelGetValueIS(label, ids);CHKERRQ(ierr);
6308   } else {
6309     /* returning an empty IS */
6310     ierr = ISCreateGeneral(PETSC_COMM_SELF,0,NULL,PETSC_USE_POINTER,ids);CHKERRQ(ierr);
6311   }
6312   PetscFunctionReturn(0);
6313 }
6314 
6315 /*@C
6316   DMGetStratumSize - Get the number of points in a label stratum
6317 
6318   Not Collective
6319 
6320   Input Parameters:
6321 + dm - The DM object
6322 . name - The label name
6323 - value - The stratum value
6324 
6325   Output Parameter:
6326 . size - The stratum size
6327 
6328   Level: beginner
6329 
6330 .keywords: mesh
6331 .seealso: DMLabelGetStratumSize(), DMGetLabelSize(), DMGetLabelIds()
6332 @*/
6333 PetscErrorCode DMGetStratumSize(DM dm, const char name[], PetscInt value, PetscInt *size)
6334 {
6335   DMLabel        label;
6336   PetscErrorCode ierr;
6337 
6338   PetscFunctionBegin;
6339   PetscValidHeaderSpecific(dm, DM_CLASSID, 1);
6340   PetscValidCharPointer(name, 2);
6341   PetscValidPointer(size, 4);
6342   ierr  = DMGetLabel(dm, name, &label);CHKERRQ(ierr);
6343   *size = 0;
6344   if (!label) PetscFunctionReturn(0);
6345   ierr = DMLabelGetStratumSize(label, value, size);CHKERRQ(ierr);
6346   PetscFunctionReturn(0);
6347 }
6348 
6349 /*@C
6350   DMGetStratumIS - Get the points in a label stratum
6351 
6352   Not Collective
6353 
6354   Input Parameters:
6355 + dm - The DM object
6356 . name - The label name
6357 - value - The stratum value
6358 
6359   Output Parameter:
6360 . points - The stratum points, or NULL if the label does not exist or does not have that value
6361 
6362   Level: beginner
6363 
6364 .keywords: mesh
6365 .seealso: DMLabelGetStratumIS(), DMGetStratumSize()
6366 @*/
6367 PetscErrorCode DMGetStratumIS(DM dm, const char name[], PetscInt value, IS *points)
6368 {
6369   DMLabel        label;
6370   PetscErrorCode ierr;
6371 
6372   PetscFunctionBegin;
6373   PetscValidHeaderSpecific(dm, DM_CLASSID, 1);
6374   PetscValidCharPointer(name, 2);
6375   PetscValidPointer(points, 4);
6376   ierr    = DMGetLabel(dm, name, &label);CHKERRQ(ierr);
6377   *points = NULL;
6378   if (!label) PetscFunctionReturn(0);
6379   ierr = DMLabelGetStratumIS(label, value, points);CHKERRQ(ierr);
6380   PetscFunctionReturn(0);
6381 }
6382 
6383 /*@C
6384   DMSetStratumIS - Set the points in a label stratum
6385 
6386   Not Collective
6387 
6388   Input Parameters:
6389 + dm - The DM object
6390 . name - The label name
6391 . value - The stratum value
6392 - points - The stratum points
6393 
6394   Level: beginner
6395 
6396 .keywords: mesh
6397 .seealso: DMLabelSetStratumIS(), DMGetStratumSize()
6398 @*/
6399 PetscErrorCode DMSetStratumIS(DM dm, const char name[], PetscInt value, IS points)
6400 {
6401   DMLabel        label;
6402   PetscErrorCode ierr;
6403 
6404   PetscFunctionBegin;
6405   PetscValidHeaderSpecific(dm, DM_CLASSID, 1);
6406   PetscValidCharPointer(name, 2);
6407   PetscValidPointer(points, 4);
6408   ierr = DMGetLabel(dm, name, &label);CHKERRQ(ierr);
6409   if (!label) PetscFunctionReturn(0);
6410   ierr = DMLabelSetStratumIS(label, value, points);CHKERRQ(ierr);
6411   PetscFunctionReturn(0);
6412 }
6413 
6414 /*@C
6415   DMClearLabelStratum - Remove all points from a stratum from a Sieve Label
6416 
6417   Not Collective
6418 
6419   Input Parameters:
6420 + dm   - The DM object
6421 . name - The label name
6422 - value - The label value for this point
6423 
6424   Output Parameter:
6425 
6426   Level: beginner
6427 
6428 .keywords: mesh
6429 .seealso: DMLabelClearStratum(), DMSetLabelValue(), DMGetStratumIS(), DMClearLabelValue()
6430 @*/
6431 PetscErrorCode DMClearLabelStratum(DM dm, const char name[], PetscInt value)
6432 {
6433   DMLabel        label;
6434   PetscErrorCode ierr;
6435 
6436   PetscFunctionBegin;
6437   PetscValidHeaderSpecific(dm, DM_CLASSID, 1);
6438   PetscValidCharPointer(name, 2);
6439   ierr = DMGetLabel(dm, name, &label);CHKERRQ(ierr);
6440   if (!label) PetscFunctionReturn(0);
6441   ierr = DMLabelClearStratum(label, value);CHKERRQ(ierr);
6442   PetscFunctionReturn(0);
6443 }
6444 
6445 /*@
6446   DMGetNumLabels - Return the number of labels defined by the mesh
6447 
6448   Not Collective
6449 
6450   Input Parameter:
6451 . dm   - The DM object
6452 
6453   Output Parameter:
6454 . numLabels - the number of Labels
6455 
6456   Level: intermediate
6457 
6458 .keywords: mesh
6459 .seealso: DMGetLabelValue(), DMSetLabelValue(), DMGetStratumIS()
6460 @*/
6461 PetscErrorCode DMGetNumLabels(DM dm, PetscInt *numLabels)
6462 {
6463   DMLabelLink next = dm->labels->next;
6464   PetscInt  n    = 0;
6465 
6466   PetscFunctionBegin;
6467   PetscValidHeaderSpecific(dm, DM_CLASSID, 1);
6468   PetscValidPointer(numLabels, 2);
6469   while (next) {++n; next = next->next;}
6470   *numLabels = n;
6471   PetscFunctionReturn(0);
6472 }
6473 
6474 /*@C
6475   DMGetLabelName - Return the name of nth label
6476 
6477   Not Collective
6478 
6479   Input Parameters:
6480 + dm - The DM object
6481 - n  - the label number
6482 
6483   Output Parameter:
6484 . name - the label name
6485 
6486   Level: intermediate
6487 
6488 .keywords: mesh
6489 .seealso: DMGetLabelValue(), DMSetLabelValue(), DMGetStratumIS()
6490 @*/
6491 PetscErrorCode DMGetLabelName(DM dm, PetscInt n, const char **name)
6492 {
6493   DMLabelLink    next = dm->labels->next;
6494   PetscInt       l    = 0;
6495   PetscErrorCode ierr;
6496 
6497   PetscFunctionBegin;
6498   PetscValidHeaderSpecific(dm, DM_CLASSID, 1);
6499   PetscValidPointer(name, 3);
6500   while (next) {
6501     if (l == n) {
6502       ierr = PetscObjectGetName((PetscObject) next->label, name);CHKERRQ(ierr);
6503       PetscFunctionReturn(0);
6504     }
6505     ++l;
6506     next = next->next;
6507   }
6508   SETERRQ1(PETSC_COMM_SELF, PETSC_ERR_ARG_OUTOFRANGE, "Label %D does not exist in this DM", n);
6509 }
6510 
6511 /*@C
6512   DMHasLabel - Determine whether the mesh has a label of a given name
6513 
6514   Not Collective
6515 
6516   Input Parameters:
6517 + dm   - The DM object
6518 - name - The label name
6519 
6520   Output Parameter:
6521 . hasLabel - PETSC_TRUE if the label is present
6522 
6523   Level: intermediate
6524 
6525 .keywords: mesh
6526 .seealso: DMCreateLabel(), DMGetLabelValue(), DMSetLabelValue(), DMGetStratumIS()
6527 @*/
6528 PetscErrorCode DMHasLabel(DM dm, const char name[], PetscBool *hasLabel)
6529 {
6530   DMLabelLink    next = dm->labels->next;
6531   const char    *lname;
6532   PetscErrorCode ierr;
6533 
6534   PetscFunctionBegin;
6535   PetscValidHeaderSpecific(dm, DM_CLASSID, 1);
6536   PetscValidCharPointer(name, 2);
6537   PetscValidPointer(hasLabel, 3);
6538   *hasLabel = PETSC_FALSE;
6539   while (next) {
6540     ierr = PetscObjectGetName((PetscObject) next->label, &lname);CHKERRQ(ierr);
6541     ierr = PetscStrcmp(name, lname, hasLabel);CHKERRQ(ierr);
6542     if (*hasLabel) break;
6543     next = next->next;
6544   }
6545   PetscFunctionReturn(0);
6546 }
6547 
6548 /*@C
6549   DMGetLabel - Return the label of a given name, or NULL
6550 
6551   Not Collective
6552 
6553   Input Parameters:
6554 + dm   - The DM object
6555 - name - The label name
6556 
6557   Output Parameter:
6558 . label - The DMLabel, or NULL if the label is absent
6559 
6560   Level: intermediate
6561 
6562 .keywords: mesh
6563 .seealso: DMCreateLabel(), DMHasLabel(), DMGetLabelValue(), DMSetLabelValue(), DMGetStratumIS()
6564 @*/
6565 PetscErrorCode DMGetLabel(DM dm, const char name[], DMLabel *label)
6566 {
6567   DMLabelLink    next = dm->labels->next;
6568   PetscBool      hasLabel;
6569   const char    *lname;
6570   PetscErrorCode ierr;
6571 
6572   PetscFunctionBegin;
6573   PetscValidHeaderSpecific(dm, DM_CLASSID, 1);
6574   PetscValidCharPointer(name, 2);
6575   PetscValidPointer(label, 3);
6576   *label = NULL;
6577   while (next) {
6578     ierr = PetscObjectGetName((PetscObject) next->label, &lname);CHKERRQ(ierr);
6579     ierr = PetscStrcmp(name, lname, &hasLabel);CHKERRQ(ierr);
6580     if (hasLabel) {
6581       *label = next->label;
6582       break;
6583     }
6584     next = next->next;
6585   }
6586   PetscFunctionReturn(0);
6587 }
6588 
6589 /*@C
6590   DMGetLabelByNum - Return the nth label
6591 
6592   Not Collective
6593 
6594   Input Parameters:
6595 + dm - The DM object
6596 - n  - the label number
6597 
6598   Output Parameter:
6599 . label - the label
6600 
6601   Level: intermediate
6602 
6603 .keywords: mesh
6604 .seealso: DMGetLabelValue(), DMSetLabelValue(), DMGetStratumIS()
6605 @*/
6606 PetscErrorCode DMGetLabelByNum(DM dm, PetscInt n, DMLabel *label)
6607 {
6608   DMLabelLink next = dm->labels->next;
6609   PetscInt    l    = 0;
6610 
6611   PetscFunctionBegin;
6612   PetscValidHeaderSpecific(dm, DM_CLASSID, 1);
6613   PetscValidPointer(label, 3);
6614   while (next) {
6615     if (l == n) {
6616       *label = next->label;
6617       PetscFunctionReturn(0);
6618     }
6619     ++l;
6620     next = next->next;
6621   }
6622   SETERRQ1(PETSC_COMM_SELF, PETSC_ERR_ARG_OUTOFRANGE, "Label %D does not exist in this DM", n);
6623 }
6624 
6625 /*@C
6626   DMAddLabel - Add the label to this mesh
6627 
6628   Not Collective
6629 
6630   Input Parameters:
6631 + dm   - The DM object
6632 - label - The DMLabel
6633 
6634   Level: developer
6635 
6636 .keywords: mesh
6637 .seealso: DMCreateLabel(), DMHasLabel(), DMGetLabelValue(), DMSetLabelValue(), DMGetStratumIS()
6638 @*/
6639 PetscErrorCode DMAddLabel(DM dm, DMLabel label)
6640 {
6641   DMLabelLink    tmpLabel;
6642   PetscBool      hasLabel;
6643   const char    *lname;
6644   PetscErrorCode ierr;
6645 
6646   PetscFunctionBegin;
6647   PetscValidHeaderSpecific(dm, DM_CLASSID, 1);
6648   ierr = PetscObjectGetName((PetscObject) label, &lname);CHKERRQ(ierr);
6649   ierr = DMHasLabel(dm, lname, &hasLabel);CHKERRQ(ierr);
6650   if (hasLabel) SETERRQ1(PETSC_COMM_SELF, PETSC_ERR_ARG_OUTOFRANGE, "Label %s already exists in this DM", lname);
6651   ierr = PetscCalloc1(1, &tmpLabel);CHKERRQ(ierr);
6652   tmpLabel->label  = label;
6653   tmpLabel->output = PETSC_TRUE;
6654   tmpLabel->next   = dm->labels->next;
6655   dm->labels->next = tmpLabel;
6656   PetscFunctionReturn(0);
6657 }
6658 
6659 /*@C
6660   DMRemoveLabel - Remove the label from this mesh
6661 
6662   Not Collective
6663 
6664   Input Parameters:
6665 + dm   - The DM object
6666 - name - The label name
6667 
6668   Output Parameter:
6669 . label - The DMLabel, or NULL if the label is absent
6670 
6671   Level: developer
6672 
6673 .keywords: mesh
6674 .seealso: DMCreateLabel(), DMHasLabel(), DMGetLabelValue(), DMSetLabelValue(), DMGetStratumIS()
6675 @*/
6676 PetscErrorCode DMRemoveLabel(DM dm, const char name[], DMLabel *label)
6677 {
6678   DMLabelLink    next = dm->labels->next;
6679   DMLabelLink    last = NULL;
6680   PetscBool      hasLabel;
6681   const char    *lname;
6682   PetscErrorCode ierr;
6683 
6684   PetscFunctionBegin;
6685   PetscValidHeaderSpecific(dm, DM_CLASSID, 1);
6686   ierr   = DMHasLabel(dm, name, &hasLabel);CHKERRQ(ierr);
6687   *label = NULL;
6688   if (!hasLabel) PetscFunctionReturn(0);
6689   while (next) {
6690     ierr = PetscObjectGetName((PetscObject) next->label, &lname);CHKERRQ(ierr);
6691     ierr = PetscStrcmp(name, lname, &hasLabel);CHKERRQ(ierr);
6692     if (hasLabel) {
6693       if (last) last->next       = next->next;
6694       else      dm->labels->next = next->next;
6695       next->next = NULL;
6696       *label     = next->label;
6697       ierr = PetscStrcmp(name, "depth", &hasLabel);CHKERRQ(ierr);
6698       if (hasLabel) {
6699         dm->depthLabel = NULL;
6700       }
6701       ierr = PetscFree(next);CHKERRQ(ierr);
6702       break;
6703     }
6704     last = next;
6705     next = next->next;
6706   }
6707   PetscFunctionReturn(0);
6708 }
6709 
6710 /*@C
6711   DMGetLabelOutput - Get the output flag for a given label
6712 
6713   Not Collective
6714 
6715   Input Parameters:
6716 + dm   - The DM object
6717 - name - The label name
6718 
6719   Output Parameter:
6720 . output - The flag for output
6721 
6722   Level: developer
6723 
6724 .keywords: mesh
6725 .seealso: DMSetLabelOutput(), DMCreateLabel(), DMHasLabel(), DMGetLabelValue(), DMSetLabelValue(), DMGetStratumIS()
6726 @*/
6727 PetscErrorCode DMGetLabelOutput(DM dm, const char name[], PetscBool *output)
6728 {
6729   DMLabelLink    next = dm->labels->next;
6730   const char    *lname;
6731   PetscErrorCode ierr;
6732 
6733   PetscFunctionBegin;
6734   PetscValidHeaderSpecific(dm, DM_CLASSID, 1);
6735   PetscValidPointer(name, 2);
6736   PetscValidPointer(output, 3);
6737   while (next) {
6738     PetscBool flg;
6739 
6740     ierr = PetscObjectGetName((PetscObject) next->label, &lname);CHKERRQ(ierr);
6741     ierr = PetscStrcmp(name, lname, &flg);CHKERRQ(ierr);
6742     if (flg) {*output = next->output; PetscFunctionReturn(0);}
6743     next = next->next;
6744   }
6745   SETERRQ1(PETSC_COMM_SELF, PETSC_ERR_ARG_OUTOFRANGE, "No label named %s was present in this dm", name);
6746 }
6747 
6748 /*@C
6749   DMSetLabelOutput - Set the output flag for a given label
6750 
6751   Not Collective
6752 
6753   Input Parameters:
6754 + dm     - The DM object
6755 . name   - The label name
6756 - output - The flag for output
6757 
6758   Level: developer
6759 
6760 .keywords: mesh
6761 .seealso: DMGetLabelOutput(), DMCreateLabel(), DMHasLabel(), DMGetLabelValue(), DMSetLabelValue(), DMGetStratumIS()
6762 @*/
6763 PetscErrorCode DMSetLabelOutput(DM dm, const char name[], PetscBool output)
6764 {
6765   DMLabelLink    next = dm->labels->next;
6766   const char    *lname;
6767   PetscErrorCode ierr;
6768 
6769   PetscFunctionBegin;
6770   PetscValidHeaderSpecific(dm, DM_CLASSID, 1);
6771   PetscValidPointer(name, 2);
6772   while (next) {
6773     PetscBool flg;
6774 
6775     ierr = PetscObjectGetName((PetscObject) next->label, &lname);CHKERRQ(ierr);
6776     ierr = PetscStrcmp(name, lname, &flg);CHKERRQ(ierr);
6777     if (flg) {next->output = output; PetscFunctionReturn(0);}
6778     next = next->next;
6779   }
6780   SETERRQ1(PETSC_COMM_SELF, PETSC_ERR_ARG_OUTOFRANGE, "No label named %s was present in this dm", name);
6781 }
6782 
6783 
6784 /*@
6785   DMCopyLabels - Copy labels from one mesh to another with a superset of the points
6786 
6787   Collective on DM
6788 
6789   Input Parameter:
6790 . dmA - The DM object with initial labels
6791 
6792   Output Parameter:
6793 . dmB - The DM object with copied labels
6794 
6795   Level: intermediate
6796 
6797   Note: This is typically used when interpolating or otherwise adding to a mesh
6798 
6799 .keywords: mesh
6800 .seealso: DMGetCoordinates(), DMGetCoordinatesLocal(), DMGetCoordinateDM(), DMGetCoordinateSection()
6801 @*/
6802 PetscErrorCode DMCopyLabels(DM dmA, DM dmB)
6803 {
6804   PetscInt       numLabels, l;
6805   PetscErrorCode ierr;
6806 
6807   PetscFunctionBegin;
6808   if (dmA == dmB) PetscFunctionReturn(0);
6809   ierr = DMGetNumLabels(dmA, &numLabels);CHKERRQ(ierr);
6810   for (l = 0; l < numLabels; ++l) {
6811     DMLabel     label, labelNew;
6812     const char *name;
6813     PetscBool   flg;
6814 
6815     ierr = DMGetLabelName(dmA, l, &name);CHKERRQ(ierr);
6816     ierr = PetscStrcmp(name, "depth", &flg);CHKERRQ(ierr);
6817     if (flg) continue;
6818     ierr = PetscStrcmp(name, "dim", &flg);CHKERRQ(ierr);
6819     if (flg) continue;
6820     ierr = DMGetLabel(dmA, name, &label);CHKERRQ(ierr);
6821     ierr = DMLabelDuplicate(label, &labelNew);CHKERRQ(ierr);
6822     ierr = DMAddLabel(dmB, labelNew);CHKERRQ(ierr);
6823   }
6824   PetscFunctionReturn(0);
6825 }
6826 
6827 /*@
6828   DMGetCoarseDM - Get the coarse mesh from which this was obtained by refinement
6829 
6830   Input Parameter:
6831 . dm - The DM object
6832 
6833   Output Parameter:
6834 . cdm - The coarse DM
6835 
6836   Level: intermediate
6837 
6838 .seealso: DMSetCoarseDM()
6839 @*/
6840 PetscErrorCode DMGetCoarseDM(DM dm, DM *cdm)
6841 {
6842   PetscFunctionBegin;
6843   PetscValidHeaderSpecific(dm, DM_CLASSID, 1);
6844   PetscValidPointer(cdm, 2);
6845   *cdm = dm->coarseMesh;
6846   PetscFunctionReturn(0);
6847 }
6848 
6849 /*@
6850   DMSetCoarseDM - Set the coarse mesh from which this was obtained by refinement
6851 
6852   Input Parameters:
6853 + dm - The DM object
6854 - cdm - The coarse DM
6855 
6856   Level: intermediate
6857 
6858 .seealso: DMGetCoarseDM()
6859 @*/
6860 PetscErrorCode DMSetCoarseDM(DM dm, DM cdm)
6861 {
6862   PetscErrorCode ierr;
6863 
6864   PetscFunctionBegin;
6865   PetscValidHeaderSpecific(dm, DM_CLASSID, 1);
6866   if (cdm) PetscValidHeaderSpecific(cdm, DM_CLASSID, 2);
6867   ierr = PetscObjectReference((PetscObject)cdm);CHKERRQ(ierr);
6868   ierr = DMDestroy(&dm->coarseMesh);CHKERRQ(ierr);
6869   dm->coarseMesh = cdm;
6870   PetscFunctionReturn(0);
6871 }
6872 
6873 /*@
6874   DMGetFineDM - Get the fine mesh from which this was obtained by refinement
6875 
6876   Input Parameter:
6877 . dm - The DM object
6878 
6879   Output Parameter:
6880 . fdm - The fine DM
6881 
6882   Level: intermediate
6883 
6884 .seealso: DMSetFineDM()
6885 @*/
6886 PetscErrorCode DMGetFineDM(DM dm, DM *fdm)
6887 {
6888   PetscFunctionBegin;
6889   PetscValidHeaderSpecific(dm, DM_CLASSID, 1);
6890   PetscValidPointer(fdm, 2);
6891   *fdm = dm->fineMesh;
6892   PetscFunctionReturn(0);
6893 }
6894 
6895 /*@
6896   DMSetFineDM - Set the fine mesh from which this was obtained by refinement
6897 
6898   Input Parameters:
6899 + dm - The DM object
6900 - fdm - The fine DM
6901 
6902   Level: intermediate
6903 
6904 .seealso: DMGetFineDM()
6905 @*/
6906 PetscErrorCode DMSetFineDM(DM dm, DM fdm)
6907 {
6908   PetscErrorCode ierr;
6909 
6910   PetscFunctionBegin;
6911   PetscValidHeaderSpecific(dm, DM_CLASSID, 1);
6912   if (fdm) PetscValidHeaderSpecific(fdm, DM_CLASSID, 2);
6913   ierr = PetscObjectReference((PetscObject)fdm);CHKERRQ(ierr);
6914   ierr = DMDestroy(&dm->fineMesh);CHKERRQ(ierr);
6915   dm->fineMesh = fdm;
6916   PetscFunctionReturn(0);
6917 }
6918 
6919 /*=== DMBoundary code ===*/
6920 
6921 PetscErrorCode DMCopyBoundary(DM dm, DM dmNew)
6922 {
6923   PetscInt       d;
6924   PetscErrorCode ierr;
6925 
6926   PetscFunctionBegin;
6927   for (d = 0; d < dm->Nds; ++d) {
6928     ierr = PetscDSCopyBoundary(dm->probs[d].ds, dmNew->probs[d].ds);CHKERRQ(ierr);
6929   }
6930   PetscFunctionReturn(0);
6931 }
6932 
6933 /*@C
6934   DMAddBoundary - Add a boundary condition to the model
6935 
6936   Input Parameters:
6937 + dm          - The DM, with a PetscDS that matches the problem being constrained
6938 . type        - The type of condition, e.g. DM_BC_ESSENTIAL_ANALYTIC/DM_BC_ESSENTIAL_FIELD (Dirichlet), or DM_BC_NATURAL (Neumann)
6939 . name        - The BC name
6940 . labelname   - The label defining constrained points
6941 . field       - The field to constrain
6942 . numcomps    - The number of constrained field components (0 will constrain all fields)
6943 . comps       - An array of constrained component numbers
6944 . bcFunc      - A pointwise function giving boundary values
6945 . numids      - The number of DMLabel ids for constrained points
6946 . ids         - An array of ids for constrained points
6947 - ctx         - An optional user context for bcFunc
6948 
6949   Options Database Keys:
6950 + -bc_<boundary name> <num> - Overrides the boundary ids
6951 - -bc_<boundary name>_comp <num> - Overrides the boundary components
6952 
6953   Level: developer
6954 
6955 .seealso: DMGetBoundary()
6956 @*/
6957 PetscErrorCode DMAddBoundary(DM dm, DMBoundaryConditionType type, const char name[], const char labelname[], PetscInt field, PetscInt numcomps, const PetscInt *comps, void (*bcFunc)(void), PetscInt numids, const PetscInt *ids, void *ctx)
6958 {
6959   PetscDS        ds;
6960   PetscErrorCode ierr;
6961 
6962   PetscFunctionBegin;
6963   PetscValidHeaderSpecific(dm, DM_CLASSID, 1);
6964   ierr = DMGetDS(dm, &ds);CHKERRQ(ierr);
6965   ierr = PetscDSAddBoundary(ds, type,name, labelname, field, numcomps, comps, bcFunc, numids, ids, ctx);CHKERRQ(ierr);
6966   PetscFunctionReturn(0);
6967 }
6968 
6969 /*@
6970   DMGetNumBoundary - Get the number of registered BC
6971 
6972   Input Parameters:
6973 . dm - The mesh object
6974 
6975   Output Parameters:
6976 . numBd - The number of BC
6977 
6978   Level: intermediate
6979 
6980 .seealso: DMAddBoundary(), DMGetBoundary()
6981 @*/
6982 PetscErrorCode DMGetNumBoundary(DM dm, PetscInt *numBd)
6983 {
6984   PetscDS        ds;
6985   PetscErrorCode ierr;
6986 
6987   PetscFunctionBegin;
6988   PetscValidHeaderSpecific(dm, DM_CLASSID, 1);
6989   ierr = DMGetDS(dm, &ds);CHKERRQ(ierr);
6990   ierr = PetscDSGetNumBoundary(ds, numBd);CHKERRQ(ierr);
6991   PetscFunctionReturn(0);
6992 }
6993 
6994 /*@C
6995   DMGetBoundary - Get a model boundary condition
6996 
6997   Input Parameters:
6998 + dm          - The mesh object
6999 - bd          - The BC number
7000 
7001   Output Parameters:
7002 + type        - The type of condition, e.g. DM_BC_ESSENTIAL_ANALYTIC/DM_BC_ESSENTIAL_FIELD (Dirichlet), or DM_BC_NATURAL (Neumann)
7003 . name        - The BC name
7004 . labelname   - The label defining constrained points
7005 . field       - The field to constrain
7006 . numcomps    - The number of constrained field components
7007 . comps       - An array of constrained component numbers
7008 . bcFunc      - A pointwise function giving boundary values
7009 . numids      - The number of DMLabel ids for constrained points
7010 . ids         - An array of ids for constrained points
7011 - ctx         - An optional user context for bcFunc
7012 
7013   Options Database Keys:
7014 + -bc_<boundary name> <num> - Overrides the boundary ids
7015 - -bc_<boundary name>_comp <num> - Overrides the boundary components
7016 
7017   Level: developer
7018 
7019 .seealso: DMAddBoundary()
7020 @*/
7021 PetscErrorCode DMGetBoundary(DM dm, PetscInt bd, DMBoundaryConditionType *type, const char **name, const char **labelname, PetscInt *field, PetscInt *numcomps, const PetscInt **comps, void (**func)(void), PetscInt *numids, const PetscInt **ids, void **ctx)
7022 {
7023   PetscDS        ds;
7024   PetscErrorCode ierr;
7025 
7026   PetscFunctionBegin;
7027   PetscValidHeaderSpecific(dm, DM_CLASSID, 1);
7028   ierr = DMGetDS(dm, &ds);CHKERRQ(ierr);
7029   ierr = PetscDSGetBoundary(ds, bd, type, name, labelname, field, numcomps, comps, func, numids, ids, ctx);CHKERRQ(ierr);
7030   PetscFunctionReturn(0);
7031 }
7032 
7033 static PetscErrorCode DMPopulateBoundary(DM dm)
7034 {
7035   PetscDS        ds;
7036   DMBoundary    *lastnext;
7037   DSBoundary     dsbound;
7038   PetscErrorCode ierr;
7039 
7040   PetscFunctionBegin;
7041   ierr = DMGetDS(dm, &ds);CHKERRQ(ierr);
7042   dsbound = ds->boundary;
7043   if (dm->boundary) {
7044     DMBoundary next = dm->boundary;
7045 
7046     /* quick check to see if the PetscDS has changed */
7047     if (next->dsboundary == dsbound) PetscFunctionReturn(0);
7048     /* the PetscDS has changed: tear down and rebuild */
7049     while (next) {
7050       DMBoundary b = next;
7051 
7052       next = b->next;
7053       ierr = PetscFree(b);CHKERRQ(ierr);
7054     }
7055     dm->boundary = NULL;
7056   }
7057 
7058   lastnext = &(dm->boundary);
7059   while (dsbound) {
7060     DMBoundary dmbound;
7061 
7062     ierr = PetscNew(&dmbound);CHKERRQ(ierr);
7063     dmbound->dsboundary = dsbound;
7064     ierr = DMGetLabel(dm, dsbound->labelname, &(dmbound->label));CHKERRQ(ierr);
7065     if (!dmbound->label) {ierr = PetscInfo2(dm, "DSBoundary %s wants label %s, which is not in this dm.\n",dsbound->name,dsbound->labelname);CHKERRQ(ierr);}
7066     /* push on the back instead of the front so that it is in the same order as in the PetscDS */
7067     *lastnext = dmbound;
7068     lastnext = &(dmbound->next);
7069     dsbound = dsbound->next;
7070   }
7071   PetscFunctionReturn(0);
7072 }
7073 
7074 PetscErrorCode DMIsBoundaryPoint(DM dm, PetscInt point, PetscBool *isBd)
7075 {
7076   DMBoundary     b;
7077   PetscErrorCode ierr;
7078 
7079   PetscFunctionBegin;
7080   PetscValidHeaderSpecific(dm, DM_CLASSID, 1);
7081   PetscValidPointer(isBd, 3);
7082   *isBd = PETSC_FALSE;
7083   ierr = DMPopulateBoundary(dm);CHKERRQ(ierr);
7084   b = dm->boundary;
7085   while (b && !(*isBd)) {
7086     DMLabel    label = b->label;
7087     DSBoundary dsb = b->dsboundary;
7088 
7089     if (label) {
7090       PetscInt i;
7091 
7092       for (i = 0; i < dsb->numids && !(*isBd); ++i) {
7093         ierr = DMLabelStratumHasPoint(label, dsb->ids[i], point, isBd);CHKERRQ(ierr);
7094       }
7095     }
7096     b = b->next;
7097   }
7098   PetscFunctionReturn(0);
7099 }
7100 
7101 /*@C
7102   DMProjectFunction - This projects the given function into the function space provided.
7103 
7104   Input Parameters:
7105 + dm      - The DM
7106 . time    - The time
7107 . funcs   - The coordinate functions to evaluate, one per field
7108 . ctxs    - Optional array of contexts to pass to each coordinate function.  ctxs itself may be null.
7109 - mode    - The insertion mode for values
7110 
7111   Output Parameter:
7112 . X - vector
7113 
7114    Calling sequence of func:
7115 $    func(PetscInt dim, PetscReal time, const PetscReal x[], PetscInt Nf, PetscScalar u[], void *ctx);
7116 
7117 +  dim - The spatial dimension
7118 .  x   - The coordinates
7119 .  Nf  - The number of fields
7120 .  u   - The output field values
7121 -  ctx - optional user-defined function context
7122 
7123   Level: developer
7124 
7125 .seealso: DMComputeL2Diff()
7126 @*/
7127 PetscErrorCode DMProjectFunction(DM dm, PetscReal time, PetscErrorCode (**funcs)(PetscInt, PetscReal, const PetscReal [], PetscInt, PetscScalar *, void *), void **ctxs, InsertMode mode, Vec X)
7128 {
7129   Vec            localX;
7130   PetscErrorCode ierr;
7131 
7132   PetscFunctionBegin;
7133   PetscValidHeaderSpecific(dm, DM_CLASSID, 1);
7134   ierr = DMGetLocalVector(dm, &localX);CHKERRQ(ierr);
7135   ierr = DMProjectFunctionLocal(dm, time, funcs, ctxs, mode, localX);CHKERRQ(ierr);
7136   ierr = DMLocalToGlobalBegin(dm, localX, mode, X);CHKERRQ(ierr);
7137   ierr = DMLocalToGlobalEnd(dm, localX, mode, X);CHKERRQ(ierr);
7138   ierr = DMRestoreLocalVector(dm, &localX);CHKERRQ(ierr);
7139   PetscFunctionReturn(0);
7140 }
7141 
7142 PetscErrorCode DMProjectFunctionLocal(DM dm, PetscReal time, PetscErrorCode (**funcs)(PetscInt, PetscReal, const PetscReal [], PetscInt, PetscScalar *, void *), void **ctxs, InsertMode mode, Vec localX)
7143 {
7144   PetscErrorCode ierr;
7145 
7146   PetscFunctionBegin;
7147   PetscValidHeaderSpecific(dm,DM_CLASSID,1);
7148   PetscValidHeaderSpecific(localX,VEC_CLASSID,5);
7149   if (!dm->ops->projectfunctionlocal) SETERRQ1(PetscObjectComm((PetscObject)dm),PETSC_ERR_SUP,"DM type %s does not implement DMProjectFunctionLocal",((PetscObject)dm)->type_name);
7150   ierr = (dm->ops->projectfunctionlocal) (dm, time, funcs, ctxs, mode, localX);CHKERRQ(ierr);
7151   PetscFunctionReturn(0);
7152 }
7153 
7154 PetscErrorCode DMProjectFunctionLabel(DM dm, PetscReal time, DMLabel label, PetscInt numIds, const PetscInt ids[], PetscInt Nc, const PetscInt comps[], PetscErrorCode (**funcs)(PetscInt, PetscReal, const PetscReal [], PetscInt, PetscScalar *, void *), void **ctxs, InsertMode mode, Vec X)
7155 {
7156   Vec            localX;
7157   PetscErrorCode ierr;
7158 
7159   PetscFunctionBegin;
7160   PetscValidHeaderSpecific(dm, DM_CLASSID, 1);
7161   ierr = DMGetLocalVector(dm, &localX);CHKERRQ(ierr);
7162   ierr = DMProjectFunctionLabelLocal(dm, time, label, numIds, ids, Nc, comps, funcs, ctxs, mode, localX);CHKERRQ(ierr);
7163   ierr = DMLocalToGlobalBegin(dm, localX, mode, X);CHKERRQ(ierr);
7164   ierr = DMLocalToGlobalEnd(dm, localX, mode, X);CHKERRQ(ierr);
7165   ierr = DMRestoreLocalVector(dm, &localX);CHKERRQ(ierr);
7166   PetscFunctionReturn(0);
7167 }
7168 
7169 PetscErrorCode DMProjectFunctionLabelLocal(DM dm, PetscReal time, DMLabel label, PetscInt numIds, const PetscInt ids[], PetscInt Nc, const PetscInt comps[], PetscErrorCode (**funcs)(PetscInt, PetscReal, const PetscReal [], PetscInt, PetscScalar *, void *), void **ctxs, InsertMode mode, Vec localX)
7170 {
7171   PetscErrorCode ierr;
7172 
7173   PetscFunctionBegin;
7174   PetscValidHeaderSpecific(dm,DM_CLASSID,1);
7175   PetscValidHeaderSpecific(localX,VEC_CLASSID,5);
7176   if (!dm->ops->projectfunctionlabellocal) SETERRQ1(PetscObjectComm((PetscObject)dm),PETSC_ERR_SUP,"DM type %s does not implement DMProjectFunctionLabelLocal",((PetscObject)dm)->type_name);
7177   ierr = (dm->ops->projectfunctionlabellocal) (dm, time, label, numIds, ids, Nc, comps, funcs, ctxs, mode, localX);CHKERRQ(ierr);
7178   PetscFunctionReturn(0);
7179 }
7180 
7181 PetscErrorCode DMProjectFieldLocal(DM dm, PetscReal time, Vec localU,
7182                                    void (**funcs)(PetscInt, PetscInt, PetscInt,
7183                                                   const PetscInt[], const PetscInt[], const PetscScalar[], const PetscScalar[], const PetscScalar[],
7184                                                   const PetscInt[], const PetscInt[], const PetscScalar[], const PetscScalar[], const PetscScalar[],
7185                                                   PetscReal, const PetscReal[], PetscInt, const PetscScalar[], PetscScalar[]),
7186                                    InsertMode mode, Vec localX)
7187 {
7188   PetscErrorCode ierr;
7189 
7190   PetscFunctionBegin;
7191   PetscValidHeaderSpecific(dm,DM_CLASSID,1);
7192   PetscValidHeaderSpecific(localU,VEC_CLASSID,3);
7193   PetscValidHeaderSpecific(localX,VEC_CLASSID,6);
7194   if (!dm->ops->projectfieldlocal) SETERRQ1(PetscObjectComm((PetscObject)dm),PETSC_ERR_SUP,"DM type %s does not implement DMProjectFieldLocal",((PetscObject)dm)->type_name);
7195   ierr = (dm->ops->projectfieldlocal) (dm, time, localU, funcs, mode, localX);CHKERRQ(ierr);
7196   PetscFunctionReturn(0);
7197 }
7198 
7199 PetscErrorCode DMProjectFieldLabelLocal(DM dm, PetscReal time, DMLabel label, PetscInt numIds, const PetscInt ids[], PetscInt Nc, const PetscInt comps[], Vec localU,
7200                                         void (**funcs)(PetscInt, PetscInt, PetscInt,
7201                                                        const PetscInt[], const PetscInt[], const PetscScalar[], const PetscScalar[], const PetscScalar[],
7202                                                        const PetscInt[], const PetscInt[], const PetscScalar[], const PetscScalar[], const PetscScalar[],
7203                                                        PetscReal, const PetscReal[], PetscInt, const PetscScalar[], PetscScalar[]),
7204                                         InsertMode mode, Vec localX)
7205 {
7206   PetscErrorCode ierr;
7207 
7208   PetscFunctionBegin;
7209   PetscValidHeaderSpecific(dm,DM_CLASSID,1);
7210   PetscValidHeaderSpecific(localU,VEC_CLASSID,6);
7211   PetscValidHeaderSpecific(localX,VEC_CLASSID,9);
7212   if (!dm->ops->projectfieldlabellocal) SETERRQ1(PetscObjectComm((PetscObject)dm),PETSC_ERR_SUP,"DM type %s does not implement DMProjectFieldLocal",((PetscObject)dm)->type_name);
7213   ierr = (dm->ops->projectfieldlabellocal)(dm, time, label, numIds, ids, Nc, comps, localU, funcs, mode, localX);CHKERRQ(ierr);
7214   PetscFunctionReturn(0);
7215 }
7216 
7217 /*@C
7218   DMComputeL2Diff - This function computes the L_2 difference between a function u and an FEM interpolant solution u_h.
7219 
7220   Input Parameters:
7221 + dm    - The DM
7222 . time  - The time
7223 . funcs - The functions to evaluate for each field component
7224 . ctxs  - Optional array of contexts to pass to each function, or NULL.
7225 - X     - The coefficient vector u_h, a global vector
7226 
7227   Output Parameter:
7228 . diff - The diff ||u - u_h||_2
7229 
7230   Level: developer
7231 
7232 .seealso: DMProjectFunction(), DMComputeL2FieldDiff(), DMComputeL2GradientDiff()
7233 @*/
7234 PetscErrorCode DMComputeL2Diff(DM dm, PetscReal time, PetscErrorCode (**funcs)(PetscInt, PetscReal, const PetscReal [], PetscInt, PetscScalar *, void *), void **ctxs, Vec X, PetscReal *diff)
7235 {
7236   PetscErrorCode ierr;
7237 
7238   PetscFunctionBegin;
7239   PetscValidHeaderSpecific(dm,DM_CLASSID,1);
7240   PetscValidHeaderSpecific(X,VEC_CLASSID,5);
7241   if (!dm->ops->computel2diff) SETERRQ1(PetscObjectComm((PetscObject)dm),PETSC_ERR_SUP,"DM type %s does not implement DMComputeL2Diff",((PetscObject)dm)->type_name);
7242   ierr = (dm->ops->computel2diff)(dm,time,funcs,ctxs,X,diff);CHKERRQ(ierr);
7243   PetscFunctionReturn(0);
7244 }
7245 
7246 /*@C
7247   DMComputeL2GradientDiff - This function computes the L_2 difference between the gradient of a function u and an FEM interpolant solution grad u_h.
7248 
7249   Input Parameters:
7250 + dm    - The DM
7251 , time  - The time
7252 . funcs - The gradient functions to evaluate for each field component
7253 . ctxs  - Optional array of contexts to pass to each function, or NULL.
7254 . X     - The coefficient vector u_h, a global vector
7255 - n     - The vector to project along
7256 
7257   Output Parameter:
7258 . diff - The diff ||(grad u - grad u_h) . n||_2
7259 
7260   Level: developer
7261 
7262 .seealso: DMProjectFunction(), DMComputeL2Diff()
7263 @*/
7264 PetscErrorCode DMComputeL2GradientDiff(DM dm, PetscReal time, PetscErrorCode (**funcs)(PetscInt, PetscReal, const PetscReal [], const PetscReal[], PetscInt, PetscScalar *, void *), void **ctxs, Vec X, const PetscReal n[], PetscReal *diff)
7265 {
7266   PetscErrorCode ierr;
7267 
7268   PetscFunctionBegin;
7269   PetscValidHeaderSpecific(dm,DM_CLASSID,1);
7270   PetscValidHeaderSpecific(X,VEC_CLASSID,5);
7271   if (!dm->ops->computel2gradientdiff) SETERRQ1(PetscObjectComm((PetscObject)dm),PETSC_ERR_SUP,"DM type %s does not implement DMComputeL2GradientDiff",((PetscObject)dm)->type_name);
7272   ierr = (dm->ops->computel2gradientdiff)(dm,time,funcs,ctxs,X,n,diff);CHKERRQ(ierr);
7273   PetscFunctionReturn(0);
7274 }
7275 
7276 /*@C
7277   DMComputeL2FieldDiff - This function computes the L_2 difference between a function u and an FEM interpolant solution u_h, separated into field components.
7278 
7279   Input Parameters:
7280 + dm    - The DM
7281 . time  - The time
7282 . funcs - The functions to evaluate for each field component
7283 . ctxs  - Optional array of contexts to pass to each function, or NULL.
7284 - X     - The coefficient vector u_h, a global vector
7285 
7286   Output Parameter:
7287 . diff - The array of differences, ||u^f - u^f_h||_2
7288 
7289   Level: developer
7290 
7291 .seealso: DMProjectFunction(), DMComputeL2FieldDiff(), DMComputeL2GradientDiff()
7292 @*/
7293 PetscErrorCode DMComputeL2FieldDiff(DM dm, PetscReal time, PetscErrorCode (**funcs)(PetscInt, PetscReal, const PetscReal [], PetscInt, PetscScalar *, void *), void **ctxs, Vec X, PetscReal diff[])
7294 {
7295   PetscErrorCode ierr;
7296 
7297   PetscFunctionBegin;
7298   PetscValidHeaderSpecific(dm,DM_CLASSID,1);
7299   PetscValidHeaderSpecific(X,VEC_CLASSID,5);
7300   if (!dm->ops->computel2fielddiff) SETERRQ1(PetscObjectComm((PetscObject)dm),PETSC_ERR_SUP,"DM type %s does not implement DMComputeL2FieldDiff",((PetscObject)dm)->type_name);
7301   ierr = (dm->ops->computel2fielddiff)(dm,time,funcs,ctxs,X,diff);CHKERRQ(ierr);
7302   PetscFunctionReturn(0);
7303 }
7304 
7305 /*@C
7306   DMAdaptLabel - Adapt a dm based on a label with values interpreted as coarsening and refining flags.  Specific implementations of DM maybe have
7307                  specialized flags, but all implementations should accept flag values DM_ADAPT_DETERMINE, DM_ADAPT_KEEP, DM_ADAPT_REFINE, and DM_ADAPT_COARSEN.
7308 
7309   Collective on dm
7310 
7311   Input parameters:
7312 + dm - the pre-adaptation DM object
7313 - label - label with the flags
7314 
7315   Output parameters:
7316 . dmAdapt - the adapted DM object: may be NULL if an adapted DM could not be produced.
7317 
7318   Level: intermediate
7319 
7320 .seealso: DMAdaptMetric(), DMCoarsen(), DMRefine()
7321 @*/
7322 PetscErrorCode DMAdaptLabel(DM dm, DMLabel label, DM *dmAdapt)
7323 {
7324   PetscErrorCode ierr;
7325 
7326   PetscFunctionBegin;
7327   PetscValidHeaderSpecific(dm, DM_CLASSID, 1);
7328   PetscValidPointer(label,2);
7329   PetscValidPointer(dmAdapt,3);
7330   *dmAdapt = NULL;
7331   if (!dm->ops->adaptlabel) SETERRQ1(PetscObjectComm((PetscObject)dm),PETSC_ERR_SUP,"DM type %s does not implement DMAdaptLabel",((PetscObject)dm)->type_name);
7332   ierr = (dm->ops->adaptlabel)(dm, label, dmAdapt);CHKERRQ(ierr);
7333   PetscFunctionReturn(0);
7334 }
7335 
7336 /*@C
7337   DMAdaptMetric - Generates a mesh adapted to the specified metric field using the pragmatic library.
7338 
7339   Input Parameters:
7340 + dm - The DM object
7341 . metric - The metric to which the mesh is adapted, defined vertex-wise.
7342 - bdLabel - Label for boundary tags, which will be preserved in the output mesh. bdLabel should be NULL if there is no such label, and should be different from "_boundary_".
7343 
7344   Output Parameter:
7345 . dmAdapt  - Pointer to the DM object containing the adapted mesh
7346 
7347   Note: The label in the adapted mesh will be registered under the name of the input DMLabel object
7348 
7349   Level: advanced
7350 
7351 .seealso: DMAdaptLabel(), DMCoarsen(), DMRefine()
7352 @*/
7353 PetscErrorCode DMAdaptMetric(DM dm, Vec metric, DMLabel bdLabel, DM *dmAdapt)
7354 {
7355   PetscErrorCode ierr;
7356 
7357   PetscFunctionBegin;
7358   PetscValidHeaderSpecific(dm, DM_CLASSID, 1);
7359   PetscValidHeaderSpecific(metric, VEC_CLASSID, 2);
7360   if (bdLabel) PetscValidPointer(bdLabel, 3);
7361   PetscValidPointer(dmAdapt, 4);
7362   *dmAdapt = NULL;
7363   if (!dm->ops->adaptmetric) SETERRQ1(PetscObjectComm((PetscObject)dm),PETSC_ERR_SUP,"DM type %s does not implement DMAdaptMetric",((PetscObject)dm)->type_name);
7364   ierr = (dm->ops->adaptmetric)(dm, metric, bdLabel, dmAdapt);CHKERRQ(ierr);
7365   PetscFunctionReturn(0);
7366 }
7367 
7368 /*@C
7369  DMGetNeighbors - Gets an array containing the MPI rank of all the processes neighbors
7370 
7371  Not Collective
7372 
7373  Input Parameter:
7374  . dm    - The DM
7375 
7376  Output Parameter:
7377  . nranks - the number of neighbours
7378  . ranks - the neighbors ranks
7379 
7380  Notes:
7381  Do not free the array, it is freed when the DM is destroyed.
7382 
7383  Level: beginner
7384 
7385  .seealso: DMDAGetNeighbors(), PetscSFGetRanks()
7386 @*/
7387 PetscErrorCode DMGetNeighbors(DM dm,PetscInt *nranks,const PetscMPIInt *ranks[])
7388 {
7389   PetscErrorCode ierr;
7390 
7391   PetscFunctionBegin;
7392   PetscValidHeaderSpecific(dm,DM_CLASSID,1);
7393   if (!dm->ops->getneighbors) SETERRQ1(PetscObjectComm((PetscObject)dm),PETSC_ERR_SUP,"DM type %s does not implement DMGetNeighbors",((PetscObject)dm)->type_name);
7394   ierr = (dm->ops->getneighbors)(dm,nranks,ranks);CHKERRQ(ierr);
7395   PetscFunctionReturn(0);
7396 }
7397 
7398 #include <petsc/private/matimpl.h> /* Needed because of coloring->ctype below */
7399 
7400 /*
7401     Converts the input vector to a ghosted vector and then calls the standard coloring code.
7402     This has be a different function because it requires DM which is not defined in the Mat library
7403 */
7404 PetscErrorCode  MatFDColoringApply_AIJDM(Mat J,MatFDColoring coloring,Vec x1,void *sctx)
7405 {
7406   PetscErrorCode ierr;
7407 
7408   PetscFunctionBegin;
7409   if (coloring->ctype == IS_COLORING_LOCAL) {
7410     Vec x1local;
7411     DM  dm;
7412     ierr = MatGetDM(J,&dm);CHKERRQ(ierr);
7413     if (!dm) SETERRQ(PetscObjectComm((PetscObject)J),PETSC_ERR_ARG_INCOMP,"IS_COLORING_LOCAL requires a DM");
7414     ierr = DMGetLocalVector(dm,&x1local);CHKERRQ(ierr);
7415     ierr = DMGlobalToLocalBegin(dm,x1,INSERT_VALUES,x1local);CHKERRQ(ierr);
7416     ierr = DMGlobalToLocalEnd(dm,x1,INSERT_VALUES,x1local);CHKERRQ(ierr);
7417     x1   = x1local;
7418   }
7419   ierr = MatFDColoringApply_AIJ(J,coloring,x1,sctx);CHKERRQ(ierr);
7420   if (coloring->ctype == IS_COLORING_LOCAL) {
7421     DM  dm;
7422     ierr = MatGetDM(J,&dm);CHKERRQ(ierr);
7423     ierr = DMRestoreLocalVector(dm,&x1);CHKERRQ(ierr);
7424   }
7425   PetscFunctionReturn(0);
7426 }
7427 
7428 /*@
7429     MatFDColoringUseDM - allows a MatFDColoring object to use the DM associated with the matrix to use a IS_COLORING_LOCAL coloring
7430 
7431     Input Parameter:
7432 .    coloring - the MatFDColoring object
7433 
7434     Developer Notes:
7435     this routine exists because the PETSc Mat library does not know about the DM objects
7436 
7437     Level: advanced
7438 
7439 .seealso: MatFDColoring, MatFDColoringCreate(), ISColoringType
7440 @*/
7441 PetscErrorCode  MatFDColoringUseDM(Mat coloring,MatFDColoring fdcoloring)
7442 {
7443   PetscFunctionBegin;
7444   coloring->ops->fdcoloringapply = MatFDColoringApply_AIJDM;
7445   PetscFunctionReturn(0);
7446 }
7447 
7448 /*@
7449     DMGetCompatibility - determine if two DMs are compatible
7450 
7451     Collective
7452 
7453     Input Parameters:
7454 +    dm - the first DM
7455 -    dm2 - the second DM
7456 
7457     Output Parameters:
7458 +    compatible - whether or not the two DMs are compatible
7459 -    set - whether or not the compatible value was set
7460 
7461     Notes:
7462     Two DMs are deemed compatible if they represent the same parallel decomposition
7463     of the same topology. This implies that the the section (field data) on one
7464     "makes sense" with respect to the topology and parallel decomposition of the other.
7465     Loosely speaking, compatibile DMs represent the same domain, with the same parallel
7466     decomposition, with different data.
7467 
7468     Typically, one would confirm compatibility if intending to simultaneously iterate
7469     over a pair of vectors obtained from different DMs.
7470 
7471     For example, two DMDA objects are compatible if they have the same local
7472     and global sizes and the same stencil width. They can have different numbers
7473     of degrees of freedom per node. Thus, one could use the node numbering from
7474     either DM in bounds for a loop over vectors derived from either DM.
7475 
7476     Consider the operation of summing data living on a 2-dof DMDA to data living
7477     on a 1-dof DMDA, which should be compatible, as in the following snippet.
7478 .vb
7479   ...
7480   ierr = DMGetCompatibility(da1,da2,&compatible,&set);CHKERRQ(ierr);
7481   if (set && compatible)  {
7482     ierr = DMDAVecGetArrayDOF(da1,vec1,&arr1);CHKERRQ(ierr);
7483     ierr = DMDAVecGetArrayDOF(da2,vec2,&arr2);CHKERRQ(ierr);
7484     ierr = DMDAGetCorners(da1,&x,&y,NULL,&m,&n);CHKERRQ(ierr);
7485     for (j=y; j<y+n; ++j) {
7486       for (i=x; i<x+m, ++i) {
7487         arr1[j][i][0] = arr2[j][i][0] + arr2[j][i][1];
7488       }
7489     }
7490     ierr = DMDAVecRestoreArrayDOF(da1,vec1,&arr1);CHKERRQ(ierr);
7491     ierr = DMDAVecRestoreArrayDOF(da2,vec2,&arr2);CHKERRQ(ierr);
7492   } else {
7493     SETERRQ(PetscObjectComm((PetscObject)da1,PETSC_ERR_ARG_INCOMP,"DMDA objects incompatible");
7494   }
7495   ...
7496 .ve
7497 
7498     Checking compatibility might be expensive for a given implementation of DM,
7499     or might be impossible to unambiguously confirm or deny. For this reason,
7500     this function may decline to determine compatibility, and hence users should
7501     always check the "set" output parameter.
7502 
7503     A DM is always compatible with itself.
7504 
7505     In the current implementation, DMs which live on "unequal" communicators
7506     (MPI_UNEQUAL in the terminology of MPI_Comm_compare()) are always deemed
7507     incompatible.
7508 
7509     This function is labeled "Collective," as information about all subdomains
7510     is required on each rank. However, in DM implementations which store all this
7511     information locally, this function may be merely "Logically Collective".
7512 
7513     Developer Notes:
7514     Compatibility is assumed to be a symmetric concept; if DM A is compatible with DM B,
7515     the DM B is compatible with DM A. Thus, this function checks the implementations
7516     of both dm and dm2 (if they are of different types), attempting to determine
7517     compatibility. It is left to DM implementers to ensure that symmetry is
7518     preserved. The simplest way to do this is, when implementing type-specific
7519     logic for this function, to check for existing logic in the implementation
7520     of other DM types and let *set = PETSC_FALSE if found; the logic of this
7521     function will then call that logic.
7522 
7523     Level: advanced
7524 
7525 .seealso: DM, DMDACreateCompatibleDMDA()
7526 @*/
7527 
7528 PetscErrorCode DMGetCompatibility(DM dm,DM dm2,PetscBool *compatible,PetscBool *set)
7529 {
7530   PetscErrorCode ierr;
7531   PetscMPIInt    compareResult;
7532   DMType         type,type2;
7533   PetscBool      sameType;
7534 
7535   PetscFunctionBegin;
7536   PetscValidHeaderSpecific(dm,DM_CLASSID,1);
7537   PetscValidHeaderSpecific(dm2,DM_CLASSID,2);
7538 
7539   /* Declare a DM compatible with itself */
7540   if (dm == dm2) {
7541     *set = PETSC_TRUE;
7542     *compatible = PETSC_TRUE;
7543     PetscFunctionReturn(0);
7544   }
7545 
7546   /* Declare a DM incompatible with a DM that lives on an "unequal"
7547      communicator. Note that this does not preclude compatibility with
7548      DMs living on "congruent" or "similar" communicators, but this must be
7549      determined by the implementation-specific logic */
7550   ierr = MPI_Comm_compare(PetscObjectComm((PetscObject)dm),PetscObjectComm((PetscObject)dm2),&compareResult);CHKERRQ(ierr);
7551   if (compareResult == MPI_UNEQUAL) {
7552     *set = PETSC_TRUE;
7553     *compatible = PETSC_FALSE;
7554     PetscFunctionReturn(0);
7555   }
7556 
7557   /* Pass to the implementation-specific routine, if one exists. */
7558   if (dm->ops->getcompatibility) {
7559     ierr = (*dm->ops->getcompatibility)(dm,dm2,compatible,set);CHKERRQ(ierr);
7560     if (*set) {
7561       PetscFunctionReturn(0);
7562     }
7563   }
7564 
7565   /* If dm and dm2 are of different types, then attempt to check compatibility
7566      with an implementation of this function from dm2 */
7567   ierr = DMGetType(dm,&type);CHKERRQ(ierr);
7568   ierr = DMGetType(dm2,&type2);CHKERRQ(ierr);
7569   ierr = PetscStrcmp(type,type2,&sameType);CHKERRQ(ierr);
7570   if (!sameType && dm2->ops->getcompatibility) {
7571     ierr = (*dm2->ops->getcompatibility)(dm2,dm,compatible,set);CHKERRQ(ierr); /* Note argument order */
7572   } else {
7573     *set = PETSC_FALSE;
7574   }
7575   PetscFunctionReturn(0);
7576 }
7577