xref: /petsc/src/dm/interface/dm.c (revision a041468a873ecb0eaa683d63e3e590db31f0074c)
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 = DMGetDS(dm, &ds);CHKERRQ(ierr);
4265   ierr = DMGetNumFields(dm, &Nf);CHKERRQ(ierr);
4266   for (f = Nf; f < numFields; ++f) {
4267     PetscContainer obj;
4268 
4269     ierr = PetscContainerCreate(PetscObjectComm((PetscObject) dm), &obj);CHKERRQ(ierr);
4270     ierr = DMAddField(dm, NULL, (PetscObject) obj);CHKERRQ(ierr);
4271     ierr = PetscDSSetDiscretization(ds, f, (PetscObject) obj);CHKERRQ(ierr);
4272     ierr = PetscContainerDestroy(&obj);CHKERRQ(ierr);
4273   }
4274   PetscFunctionReturn(0);
4275 }
4276 
4277 /*@
4278   DMGetField - Return the discretization object for a given DM field
4279 
4280   Not collective
4281 
4282   Input Parameters:
4283 + dm - The DM
4284 - f  - The field number
4285 
4286   Output Parameters:
4287 + label - The label indicating the support of the field, or NULL for the entire mesh
4288 - field - The discretization object
4289 
4290   Level: intermediate
4291 
4292 .seealso: DMAddField(), DMSetField()
4293 @*/
4294 PetscErrorCode DMGetField(DM dm, PetscInt f, DMLabel *label, PetscObject *field)
4295 {
4296   PetscFunctionBegin;
4297   PetscValidHeaderSpecific(dm, DM_CLASSID, 1);
4298   PetscValidPointer(field, 3);
4299   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);
4300   if (label) *label = dm->fields[f].label;
4301   if (field) *field = dm->fields[f].disc;
4302   PetscFunctionReturn(0);
4303 }
4304 
4305 /*@
4306   DMSetField - Set the discretization object for a given DM field
4307 
4308   Logically collective on DM
4309 
4310   Input Parameters:
4311 + dm    - The DM
4312 . f     - The field number
4313 . label - The label indicating the support of the field, or NULL for the entire mesh
4314 - field - The discretization object
4315 
4316   Level: intermediate
4317 
4318 .seealso: DMAddField(), DMGetField()
4319 @*/
4320 PetscErrorCode DMSetField(DM dm, PetscInt f, DMLabel label, PetscObject field)
4321 {
4322   PetscErrorCode ierr;
4323 
4324   PetscFunctionBegin;
4325   PetscValidHeaderSpecific(dm, DM_CLASSID, 1);
4326   if (label) PetscValidHeaderSpecific(label, DMLABEL_CLASSID, 3);
4327   PetscValidHeader(field, 4);
4328   if (f < 0) SETERRQ1(PETSC_COMM_SELF, PETSC_ERR_ARG_OUTOFRANGE, "Field number %d must be non-negative", f);
4329   ierr = DMFieldEnlarge_Static(dm, f+1);CHKERRQ(ierr);
4330   ierr = DMLabelDestroy(&dm->fields[f].label);CHKERRQ(ierr);
4331   ierr = PetscObjectDestroy(&dm->fields[f].disc);CHKERRQ(ierr);
4332   dm->fields[f].label = label;
4333   dm->fields[f].disc  = field;
4334   ierr = PetscObjectReference((PetscObject) label);CHKERRQ(ierr);
4335   ierr = PetscObjectReference((PetscObject) field);CHKERRQ(ierr);
4336   PetscFunctionReturn(0);
4337 }
4338 
4339 /*@
4340   DMAddField - Add the discretization object for the given DM field
4341 
4342   Logically collective on DM
4343 
4344   Input Parameters:
4345 + dm    - The DM
4346 . label - The label indicating the support of the field, or NULL for the entire mesh
4347 - field - The discretization object
4348 
4349   Level: intermediate
4350 
4351 .seealso: DMSetField(), DMGetField()
4352 @*/
4353 PetscErrorCode DMAddField(DM dm, DMLabel label, PetscObject field)
4354 {
4355   PetscInt       Nf = dm->Nf;
4356   PetscErrorCode ierr;
4357 
4358   PetscFunctionBegin;
4359   PetscValidHeaderSpecific(dm, DM_CLASSID, 1);
4360   if (label) PetscValidHeaderSpecific(label, DMLABEL_CLASSID, 3);
4361   PetscValidHeader(field, 3);
4362   ierr = DMFieldEnlarge_Static(dm, Nf+1);CHKERRQ(ierr);
4363   dm->fields[Nf].label = label;
4364   dm->fields[Nf].disc  = field;
4365   ierr = PetscObjectReference((PetscObject) label);CHKERRQ(ierr);
4366   ierr = PetscObjectReference((PetscObject) field);CHKERRQ(ierr);
4367   PetscFunctionReturn(0);
4368 }
4369 
4370 /*@
4371   DMCopyFields - Copy the discretizations for the DM into another DM
4372 
4373   Collective on DM
4374 
4375   Input Parameter:
4376 . dm - The DM
4377 
4378   Output Parameter:
4379 . newdm - The DM
4380 
4381   Level: advanced
4382 
4383 .seealso: DMGetField(), DMSetField(), DMAddField(), DMCopyDS(), DMGetDS(), DMGetCellDS()
4384 @*/
4385 PetscErrorCode DMCopyFields(DM dm, DM newdm)
4386 {
4387   PetscInt       Nf, f;
4388   PetscErrorCode ierr;
4389 
4390   PetscFunctionBegin;
4391   if (dm == newdm) PetscFunctionReturn(0);
4392   ierr = DMGetNumFields(dm, &Nf);CHKERRQ(ierr);
4393   ierr = DMClearFields(newdm);CHKERRQ(ierr);
4394   for (f = 0; f < Nf; ++f) {
4395     DMLabel     label;
4396     PetscObject field;
4397 
4398     ierr = DMGetField(dm, f, &label, &field);CHKERRQ(ierr);
4399     ierr = DMSetField(newdm, f, label, field);CHKERRQ(ierr);
4400   }
4401   PetscFunctionReturn(0);
4402 }
4403 
4404 static PetscErrorCode DMDSEnlarge_Static(DM dm, PetscInt NdsNew)
4405 {
4406   DMSpace       *tmpd;
4407   PetscInt       Nds = dm->Nds, s;
4408   PetscErrorCode ierr;
4409 
4410   PetscFunctionBegin;
4411   if (Nds >= NdsNew) PetscFunctionReturn(0);
4412   ierr = PetscMalloc1(NdsNew, &tmpd);CHKERRQ(ierr);
4413   for (s = 0; s < Nds; ++s) tmpd[s] = dm->probs[s];
4414   for (s = Nds; s < NdsNew; ++s) {tmpd[s].ds = NULL; tmpd[s].label = NULL;}
4415   ierr = PetscFree(dm->probs);CHKERRQ(ierr);
4416   dm->Nds   = NdsNew;
4417   dm->probs = tmpd;
4418   PetscFunctionReturn(0);
4419 }
4420 
4421 /*@
4422   DMGetNumDS - Get the number of discrete systems in the DM
4423 
4424   Not collective
4425 
4426   Input Parameter:
4427 . dm - The DM
4428 
4429   Output Parameter:
4430 . Nds - The number of PetscDS objects
4431 
4432   Level: intermediate
4433 
4434 .seealso: DMGetDS(), DMGetCellDS()
4435 @*/
4436 PetscErrorCode DMGetNumDS(DM dm, PetscInt *Nds)
4437 {
4438   PetscFunctionBegin;
4439   PetscValidHeaderSpecific(dm, DM_CLASSID, 1);
4440   PetscValidPointer(Nds, 2);
4441   *Nds = dm->Nds;
4442   PetscFunctionReturn(0);
4443 }
4444 
4445 /*@
4446   DMClearDS - Remove all discrete systems from the DM
4447 
4448   Logically collective on DM
4449 
4450   Input Parameter:
4451 . dm - The DM
4452 
4453   Level: intermediate
4454 
4455 .seealso: DMGetNumDS(), DMGetDS(), DMSetField()
4456 @*/
4457 PetscErrorCode DMClearDS(DM dm)
4458 {
4459   PetscInt       s;
4460   PetscErrorCode ierr;
4461 
4462   PetscFunctionBegin;
4463   PetscValidHeaderSpecific(dm, DM_CLASSID, 1);
4464   for (s = 0; s < dm->Nds; ++s) {
4465     ierr = PetscDSDestroy(&dm->probs[s].ds);CHKERRQ(ierr);
4466     ierr = DMLabelDestroy(&dm->probs[s].label);CHKERRQ(ierr);
4467   }
4468   ierr = PetscFree(dm->probs);CHKERRQ(ierr);
4469   dm->probs = NULL;
4470   dm->Nds   = 0;
4471   PetscFunctionReturn(0);
4472 }
4473 
4474 /*@
4475   DMGetDS - Get the default PetscDS
4476 
4477   Not collective
4478 
4479   Input Parameter:
4480 . dm    - The DM
4481 
4482   Output Parameter:
4483 . prob - The default PetscDS
4484 
4485   Level: intermediate
4486 
4487 .seealso: DMGetCellDS(), DMGetRegionDS()
4488 @*/
4489 PetscErrorCode DMGetDS(DM dm, PetscDS *prob)
4490 {
4491   PetscFunctionBeginHot;
4492   PetscValidHeaderSpecific(dm, DM_CLASSID, 1);
4493   PetscValidPointer(prob, 2);
4494   if (dm->Nds) *prob = dm->probs[0].ds;
4495   else         *prob = NULL;
4496   PetscFunctionReturn(0);
4497 }
4498 
4499 /*@
4500   DMGetCellDS - Get the PetscDS defined on a given cell
4501 
4502   Not collective
4503 
4504   Input Parameters:
4505 + dm    - The DM
4506 - point - Cell for the DS
4507 
4508   Output Parameter:
4509 . prob - The PetscDS defined on the given cell
4510 
4511   Level: developer
4512 
4513 .seealso: DMGetDS(), DMSetDS()
4514 @*/
4515 PetscErrorCode DMGetCellDS(DM dm, PetscInt point, PetscDS *prob)
4516 {
4517   PetscDS        probDef = NULL;
4518   PetscInt       s;
4519   PetscErrorCode ierr;
4520 
4521   PetscFunctionBeginHot;
4522   PetscValidHeaderSpecific(dm, DM_CLASSID, 1);
4523   PetscValidPointer(prob, 3);
4524   *prob = NULL;
4525   for (s = 0; s < dm->Nds; ++s) {
4526     PetscInt val;
4527 
4528     if (!dm->probs[s].label) {probDef = dm->probs[s].ds;}
4529     else {
4530       ierr = DMLabelGetValue(dm->probs[s].label, point, &val);CHKERRQ(ierr);
4531       if (val >= 0) {*prob = dm->probs[s].ds; break;}
4532     }
4533   }
4534   if (!*prob) *prob = probDef;
4535   PetscFunctionReturn(0);
4536 }
4537 
4538 /*@
4539   DMGetRegionDS - Get the PetscDS for a given mesh region, defined by a DMLabel
4540 
4541   Not collective
4542 
4543   Input Parameters:
4544 + dm    - The DM
4545 - label - The DMLabel defining the mesh region, or NULL for the entire mesh
4546 
4547   Output Parameter:
4548 . prob - The PetscDS defined on the given region
4549 
4550   Note: If the label is missing, this function returns an error
4551 
4552   Level: advanced
4553 
4554 .seealso: DMGetRegionNumDS(), DMSetRegionDS(), DMGetDS(), DMGetCellDS()
4555 @*/
4556 PetscErrorCode DMGetRegionDS(DM dm, DMLabel label, PetscDS *ds)
4557 {
4558   PetscInt Nds = dm->Nds, s;
4559 
4560   PetscFunctionBegin;
4561   PetscValidHeaderSpecific(dm, DM_CLASSID, 1);
4562   if (label) PetscValidHeaderSpecific(label, DMLABEL_CLASSID, 2);
4563   PetscValidPointer(ds, 3);
4564   *ds = NULL;
4565   for (s = 0; s < Nds; ++s) {
4566     if (dm->probs[s].label == label) {*ds = dm->probs[s].ds; PetscFunctionReturn(0);}
4567   }
4568   SETERRQ(PetscObjectComm((PetscObject) dm), PETSC_ERR_ARG_WRONG, "Label not found in DM");
4569 }
4570 
4571 /*@
4572   DMGetRegionNumDS - Get the PetscDS for a given mesh region, defined by the region number
4573 
4574   Not collective
4575 
4576   Input Parameters:
4577 + dm  - The DM
4578 - num - The region number, in [0, Nds)
4579 
4580   Output Parameters:
4581 + label - The region label
4582 - prob  - The PetscDS defined on the given region
4583 
4584   Level: advanced
4585 
4586 .seealso: DMGetRegionDS(), DMSetRegionDS(), DMGetDS(), DMGetCellDS()
4587 @*/
4588 PetscErrorCode DMGetRegionNumDS(DM dm, PetscInt num, DMLabel *label, PetscDS *ds)
4589 {
4590   PetscInt       Nds;
4591   PetscErrorCode ierr;
4592 
4593   PetscFunctionBegin;
4594   PetscValidHeaderSpecific(dm, DM_CLASSID, 1);
4595   ierr = DMGetNumDS(dm, &Nds);CHKERRQ(ierr);
4596   if ((num < 0) || (num >= Nds)) SETERRQ2(PETSC_COMM_SELF, PETSC_ERR_ARG_OUTOFRANGE, "Region number %D is not in [0, %D)", num, Nds);
4597   if (label) {
4598     PetscValidPointer(label, 3);
4599     *label = dm->probs[num].label;
4600   }
4601   if (ds) {
4602     PetscValidPointer(ds, 4);
4603     *ds = dm->probs[num].ds;
4604   }
4605   PetscFunctionReturn(0);
4606 }
4607 
4608 /*@
4609   DMSetRegionDS - Set the PetscDS for a given mesh region, defined by a DMLabel
4610 
4611   Collective on DM
4612 
4613   Input Parameters:
4614 + dm    - The DM
4615 . label - The DMLabel defining the mesh region, or NULL for the entire mesh
4616 - prob - The PetscDS defined on the given cell
4617 
4618   Note: If the label has a DS defined, it will be replaced. Otherwise, it will be added to the DM.
4619 
4620   Level: advanced
4621 
4622 .seealso: DMGetRegionDS(), DMGetDS(), DMGetCellDS()
4623 @*/
4624 PetscErrorCode DMSetRegionDS(DM dm, DMLabel label, PetscDS ds)
4625 {
4626   PetscInt       Nds = dm->Nds, s;
4627   PetscErrorCode ierr;
4628 
4629   PetscFunctionBegin;
4630   PetscValidHeaderSpecific(dm, DM_CLASSID, 1);
4631   if (label) PetscValidHeaderSpecific(label, DMLABEL_CLASSID, 2);
4632   PetscValidHeaderSpecific(ds, PETSCDS_CLASSID, 3);
4633   for (s = 0; s < Nds; ++s) {
4634     if (dm->probs[s].label == label) {
4635       dm->probs[s].ds = ds;
4636       PetscFunctionReturn(0);
4637     }
4638   }
4639   ierr = DMDSEnlarge_Static(dm, Nds+1);CHKERRQ(ierr);
4640   ierr = PetscObjectReference((PetscObject) label);CHKERRQ(ierr);
4641   ierr = PetscObjectReference((PetscObject) ds);CHKERRQ(ierr);
4642   if (!label) {
4643     /* Put the NULL label at the front, so it is returned as the default */
4644     for (s = Nds-1; s >=0; --s) dm->probs[s+1] = dm->probs[s];
4645     Nds = 0;
4646   }
4647   dm->probs[Nds].label = label;
4648   dm->probs[Nds].ds    = ds;
4649   PetscFunctionReturn(0);
4650 }
4651 
4652 /*@
4653   DMCreateDS - Create the discrete systems for the DM based upon the fields added to the DM
4654 
4655   Collective on DM
4656 
4657   Input Parameter:
4658 . dm - The DM
4659 
4660   Note: If the label has a DS defined, it will be replaced. Otherwise, it will be added to the DM.
4661 
4662   Level: intermediate
4663 
4664 .seealso: DMSetField, DMAddField(), DMGetDS(), DMGetCellDS(), DMGetRegionDS(), DMSetRegionDS()
4665 @*/
4666 PetscErrorCode DMCreateDS(DM dm)
4667 {
4668   MPI_Comm       comm;
4669   PetscDS        prob, probh = NULL;
4670   PetscInt       dimEmbed, f, s, field = 0, fieldh = 0;
4671   PetscBool      doSetup = PETSC_TRUE;
4672   PetscErrorCode ierr;
4673 
4674   PetscFunctionBegin;
4675   PetscValidHeaderSpecific(dm, DM_CLASSID, 1);
4676   if (!dm->fields) PetscFunctionReturn(0);
4677   /* Can only handle two label cases right now:
4678    1) NULL
4679    2) Hybrid cells
4680   */
4681   ierr = PetscObjectGetComm((PetscObject) dm, &comm);CHKERRQ(ierr);
4682   ierr = DMGetCoordinateDim(dm, &dimEmbed);CHKERRQ(ierr);
4683   /* Create default DS */
4684   ierr = DMGetRegionDS(dm, NULL, &prob);CHKERRQ(ierr);
4685   ierr = PetscDSSetCoordinateDimension(prob, dimEmbed);CHKERRQ(ierr);
4686   /* Optionally create hybrid DS */
4687   for (f = 0; f < dm->Nf; ++f) {
4688     DMLabel  label = dm->fields[f].label;
4689     PetscInt lStart, lEnd;
4690 
4691     if (label) {
4692       DM       plex;
4693       PetscInt depth, pMax[4];
4694 
4695       ierr = DMConvert(dm, DMPLEX, &plex);CHKERRQ(ierr);
4696       ierr = DMPlexGetDepth(plex, &depth);CHKERRQ(ierr);
4697       ierr = DMPlexGetHybridBounds(plex, depth >= 0 ? &pMax[depth] : NULL, depth>1 ? &pMax[depth-1] : NULL, depth>2 ? &pMax[1] : NULL, &pMax[0]);CHKERRQ(ierr);
4698       ierr = DMDestroy(&plex);CHKERRQ(ierr);
4699 
4700       ierr = DMLabelGetBounds(label, &lStart, &lEnd);CHKERRQ(ierr);
4701       if (lStart < pMax[depth]) SETERRQ(PETSC_COMM_SELF, PETSC_ERR_SUP, "Only support labels over hybrid cells right now");
4702       ierr = PetscDSCreate(comm, &probh);CHKERRQ(ierr);
4703       ierr = DMSetRegionDS(dm, label, probh);CHKERRQ(ierr);
4704       ierr = PetscDSSetHybrid(probh, PETSC_TRUE);CHKERRQ(ierr);
4705       ierr = PetscDSSetCoordinateDimension(probh, dimEmbed);CHKERRQ(ierr);
4706       break;
4707     }
4708   }
4709   /* Set fields in DSes */
4710   for (f = 0; f < dm->Nf; ++f) {
4711     DMLabel     label = dm->fields[f].label;
4712     PetscObject disc  = dm->fields[f].disc;
4713 
4714     if (!label) {
4715       ierr = PetscDSSetDiscretization(prob,  field++,  disc);CHKERRQ(ierr);
4716       if (probh) {
4717         PetscFE subfe;
4718 
4719         ierr = PetscFEGetHeightSubspace((PetscFE) disc, 1, &subfe);CHKERRQ(ierr);
4720         ierr = PetscDSSetDiscretization(probh, fieldh++, (PetscObject) subfe);CHKERRQ(ierr);
4721       }
4722     } else {
4723       ierr = PetscDSSetDiscretization(probh, fieldh++, disc);CHKERRQ(ierr);
4724     }
4725     /* We allow people to have placeholder fields and construct the Section by hand */
4726     {
4727       PetscClassId id;
4728 
4729       ierr = PetscObjectGetClassId(disc, &id);CHKERRQ(ierr);
4730       if ((id != PETSCFE_CLASSID) && (id != PETSCFV_CLASSID)) doSetup = PETSC_FALSE;
4731     }
4732   }
4733   ierr = PetscDSDestroy(&probh);CHKERRQ(ierr);
4734   /* Setup DSes */
4735   if (doSetup) {
4736     for (s = 0; s < dm->Nds; ++s) {ierr = PetscDSSetUp(dm->probs[s].ds);CHKERRQ(ierr);}
4737   }
4738   PetscFunctionReturn(0);
4739 }
4740 
4741 /*@
4742   DMCopyDS - Copy the discrete systems for the DM into another DM
4743 
4744   Collective on DM
4745 
4746   Input Parameter:
4747 . dm - The DM
4748 
4749   Output Parameter:
4750 . newdm - The DM
4751 
4752   Level: advanced
4753 
4754 .seealso: DMCopyFields(), DMAddField(), DMGetDS(), DMGetCellDS(), DMGetRegionDS(), DMSetRegionDS()
4755 @*/
4756 PetscErrorCode DMCopyDS(DM dm, DM newdm)
4757 {
4758   PetscInt       Nds, s;
4759   PetscErrorCode ierr;
4760 
4761   PetscFunctionBegin;
4762   if (dm == newdm) PetscFunctionReturn(0);
4763   ierr = DMGetNumDS(dm, &Nds);CHKERRQ(ierr);
4764   ierr = DMClearDS(newdm);CHKERRQ(ierr);
4765   for (s = 0; s < Nds; ++s) {
4766     DMLabel label;
4767     PetscDS ds;
4768 
4769     ierr = DMGetRegionNumDS(dm, s, &label, &ds);CHKERRQ(ierr);
4770     ierr = DMSetRegionDS(newdm, label, ds);CHKERRQ(ierr);
4771   }
4772   PetscFunctionReturn(0);
4773 }
4774 
4775 /*@
4776   DMCopyDisc - Copy the fields and discrete systems for the DM into another DM
4777 
4778   Collective on DM
4779 
4780   Input Parameter:
4781 . dm - The DM
4782 
4783   Output Parameter:
4784 . newdm - The DM
4785 
4786   Level: advanced
4787 
4788 .seealso: DMCopyFields(), DMCopyDS()
4789 @*/
4790 PetscErrorCode DMCopyDisc(DM dm, DM newdm)
4791 {
4792   PetscErrorCode ierr;
4793 
4794   PetscFunctionBegin;
4795   if (dm == newdm) PetscFunctionReturn(0);
4796   ierr = DMCopyFields(dm, newdm);CHKERRQ(ierr);
4797   ierr = DMCopyDS(dm, newdm);CHKERRQ(ierr);
4798   PetscFunctionReturn(0);
4799 }
4800 
4801 PetscErrorCode DMRestrictHook_Coordinates(DM dm,DM dmc,void *ctx)
4802 {
4803   DM dm_coord,dmc_coord;
4804   PetscErrorCode ierr;
4805   Vec coords,ccoords;
4806   Mat inject;
4807   PetscFunctionBegin;
4808   ierr = DMGetCoordinateDM(dm,&dm_coord);CHKERRQ(ierr);
4809   ierr = DMGetCoordinateDM(dmc,&dmc_coord);CHKERRQ(ierr);
4810   ierr = DMGetCoordinates(dm,&coords);CHKERRQ(ierr);
4811   ierr = DMGetCoordinates(dmc,&ccoords);CHKERRQ(ierr);
4812   if (coords && !ccoords) {
4813     ierr = DMCreateGlobalVector(dmc_coord,&ccoords);CHKERRQ(ierr);
4814     ierr = PetscObjectSetName((PetscObject)ccoords,"coordinates");CHKERRQ(ierr);
4815     ierr = DMCreateInjection(dmc_coord,dm_coord,&inject);CHKERRQ(ierr);
4816     ierr = MatRestrict(inject,coords,ccoords);CHKERRQ(ierr);
4817     ierr = MatDestroy(&inject);CHKERRQ(ierr);
4818     ierr = DMSetCoordinates(dmc,ccoords);CHKERRQ(ierr);
4819     ierr = VecDestroy(&ccoords);CHKERRQ(ierr);
4820   }
4821   PetscFunctionReturn(0);
4822 }
4823 
4824 static PetscErrorCode DMSubDomainHook_Coordinates(DM dm,DM subdm,void *ctx)
4825 {
4826   DM dm_coord,subdm_coord;
4827   PetscErrorCode ierr;
4828   Vec coords,ccoords,clcoords;
4829   VecScatter *scat_i,*scat_g;
4830   PetscFunctionBegin;
4831   ierr = DMGetCoordinateDM(dm,&dm_coord);CHKERRQ(ierr);
4832   ierr = DMGetCoordinateDM(subdm,&subdm_coord);CHKERRQ(ierr);
4833   ierr = DMGetCoordinates(dm,&coords);CHKERRQ(ierr);
4834   ierr = DMGetCoordinates(subdm,&ccoords);CHKERRQ(ierr);
4835   if (coords && !ccoords) {
4836     ierr = DMCreateGlobalVector(subdm_coord,&ccoords);CHKERRQ(ierr);
4837     ierr = PetscObjectSetName((PetscObject)ccoords,"coordinates");CHKERRQ(ierr);
4838     ierr = DMCreateLocalVector(subdm_coord,&clcoords);CHKERRQ(ierr);
4839     ierr = PetscObjectSetName((PetscObject)clcoords,"coordinates");CHKERRQ(ierr);
4840     ierr = DMCreateDomainDecompositionScatters(dm_coord,1,&subdm_coord,NULL,&scat_i,&scat_g);CHKERRQ(ierr);
4841     ierr = VecScatterBegin(scat_i[0],coords,ccoords,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr);
4842     ierr = VecScatterBegin(scat_g[0],coords,clcoords,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr);
4843     ierr = VecScatterEnd(scat_i[0],coords,ccoords,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr);
4844     ierr = VecScatterEnd(scat_g[0],coords,clcoords,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr);
4845     ierr = DMSetCoordinates(subdm,ccoords);CHKERRQ(ierr);
4846     ierr = DMSetCoordinatesLocal(subdm,clcoords);CHKERRQ(ierr);
4847     ierr = VecScatterDestroy(&scat_i[0]);CHKERRQ(ierr);
4848     ierr = VecScatterDestroy(&scat_g[0]);CHKERRQ(ierr);
4849     ierr = VecDestroy(&ccoords);CHKERRQ(ierr);
4850     ierr = VecDestroy(&clcoords);CHKERRQ(ierr);
4851     ierr = PetscFree(scat_i);CHKERRQ(ierr);
4852     ierr = PetscFree(scat_g);CHKERRQ(ierr);
4853   }
4854   PetscFunctionReturn(0);
4855 }
4856 
4857 /*@
4858   DMGetDimension - Return the topological dimension of the DM
4859 
4860   Not collective
4861 
4862   Input Parameter:
4863 . dm - The DM
4864 
4865   Output Parameter:
4866 . dim - The topological dimension
4867 
4868   Level: beginner
4869 
4870 .seealso: DMSetDimension(), DMCreate()
4871 @*/
4872 PetscErrorCode DMGetDimension(DM dm, PetscInt *dim)
4873 {
4874   PetscFunctionBegin;
4875   PetscValidHeaderSpecific(dm, DM_CLASSID, 1);
4876   PetscValidPointer(dim, 2);
4877   *dim = dm->dim;
4878   PetscFunctionReturn(0);
4879 }
4880 
4881 /*@
4882   DMSetDimension - Set the topological dimension of the DM
4883 
4884   Collective on dm
4885 
4886   Input Parameters:
4887 + dm - The DM
4888 - dim - The topological dimension
4889 
4890   Level: beginner
4891 
4892 .seealso: DMGetDimension(), DMCreate()
4893 @*/
4894 PetscErrorCode DMSetDimension(DM dm, PetscInt dim)
4895 {
4896   PetscDS        ds;
4897   PetscErrorCode ierr;
4898 
4899   PetscFunctionBegin;
4900   PetscValidHeaderSpecific(dm, DM_CLASSID, 1);
4901   PetscValidLogicalCollectiveInt(dm, dim, 2);
4902   dm->dim = dim;
4903   ierr = DMGetDS(dm, &ds);CHKERRQ(ierr);
4904   if (ds->dimEmbed < 0) {ierr = PetscDSSetCoordinateDimension(ds, dm->dim);CHKERRQ(ierr);}
4905   PetscFunctionReturn(0);
4906 }
4907 
4908 /*@
4909   DMGetDimPoints - Get the half-open interval for all points of a given dimension
4910 
4911   Collective on DM
4912 
4913   Input Parameters:
4914 + dm - the DM
4915 - dim - the dimension
4916 
4917   Output Parameters:
4918 + pStart - The first point of the given dimension
4919 . pEnd - The first point following points of the given dimension
4920 
4921   Note:
4922   The points are vertices in the Hasse diagram encoding the topology. This is explained in
4923   http://arxiv.org/abs/0908.4427. If no points exist of this dimension in the storage scheme,
4924   then the interval is empty.
4925 
4926   Level: intermediate
4927 
4928 .keywords: point, Hasse Diagram, dimension
4929 .seealso: DMPLEX, DMPlexGetDepthStratum(), DMPlexGetHeightStratum()
4930 @*/
4931 PetscErrorCode DMGetDimPoints(DM dm, PetscInt dim, PetscInt *pStart, PetscInt *pEnd)
4932 {
4933   PetscInt       d;
4934   PetscErrorCode ierr;
4935 
4936   PetscFunctionBegin;
4937   PetscValidHeaderSpecific(dm,DM_CLASSID,1);
4938   ierr = DMGetDimension(dm, &d);CHKERRQ(ierr);
4939   if ((dim < 0) || (dim > d)) SETERRQ2(PetscObjectComm((PetscObject) dm), PETSC_ERR_ARG_OUTOFRANGE, "Invalid dimension %d 1", dim, d);
4940   ierr = (*dm->ops->getdimpoints)(dm, dim, pStart, pEnd);CHKERRQ(ierr);
4941   PetscFunctionReturn(0);
4942 }
4943 
4944 /*@
4945   DMSetCoordinates - Sets into the DM a global vector that holds the coordinates
4946 
4947   Collective on DM
4948 
4949   Input Parameters:
4950 + dm - the DM
4951 - c - coordinate vector
4952 
4953   Notes:
4954   The coordinates do include those for ghost points, which are in the local vector.
4955 
4956   The vector c should be destroyed by the caller.
4957 
4958   Level: intermediate
4959 
4960 .keywords: distributed array, get, corners, nodes, local indices, coordinates
4961 .seealso: DMSetCoordinatesLocal(), DMGetCoordinates(), DMGetCoordinatesLocal(), DMGetCoordinateDM()
4962 @*/
4963 PetscErrorCode DMSetCoordinates(DM dm, Vec c)
4964 {
4965   PetscErrorCode ierr;
4966 
4967   PetscFunctionBegin;
4968   PetscValidHeaderSpecific(dm,DM_CLASSID,1);
4969   PetscValidHeaderSpecific(c,VEC_CLASSID,2);
4970   ierr            = PetscObjectReference((PetscObject) c);CHKERRQ(ierr);
4971   ierr            = VecDestroy(&dm->coordinates);CHKERRQ(ierr);
4972   dm->coordinates = c;
4973   ierr            = VecDestroy(&dm->coordinatesLocal);CHKERRQ(ierr);
4974   ierr            = DMCoarsenHookAdd(dm,DMRestrictHook_Coordinates,NULL,NULL);CHKERRQ(ierr);
4975   ierr            = DMSubDomainHookAdd(dm,DMSubDomainHook_Coordinates,NULL,NULL);CHKERRQ(ierr);
4976   PetscFunctionReturn(0);
4977 }
4978 
4979 /*@
4980   DMSetCoordinatesLocal - Sets into the DM a local vector that holds the coordinates
4981 
4982   Not collective
4983 
4984    Input Parameters:
4985 +  dm - the DM
4986 -  c - coordinate vector
4987 
4988   Notes:
4989   The coordinates of ghost points can be set using DMSetCoordinates()
4990   followed by DMGetCoordinatesLocal(). This is intended to enable the
4991   setting of ghost coordinates outside of the domain.
4992 
4993   The vector c should be destroyed by the caller.
4994 
4995   Level: intermediate
4996 
4997 .keywords: distributed array, get, corners, nodes, local indices, coordinates
4998 .seealso: DMGetCoordinatesLocal(), DMSetCoordinates(), DMGetCoordinates(), DMGetCoordinateDM()
4999 @*/
5000 PetscErrorCode DMSetCoordinatesLocal(DM dm, Vec c)
5001 {
5002   PetscErrorCode ierr;
5003 
5004   PetscFunctionBegin;
5005   PetscValidHeaderSpecific(dm,DM_CLASSID,1);
5006   PetscValidHeaderSpecific(c,VEC_CLASSID,2);
5007   ierr = PetscObjectReference((PetscObject) c);CHKERRQ(ierr);
5008   ierr = VecDestroy(&dm->coordinatesLocal);CHKERRQ(ierr);
5009 
5010   dm->coordinatesLocal = c;
5011 
5012   ierr = VecDestroy(&dm->coordinates);CHKERRQ(ierr);
5013   PetscFunctionReturn(0);
5014 }
5015 
5016 /*@
5017   DMGetCoordinates - Gets a global vector with the coordinates associated with the DM.
5018 
5019   Collective on DM
5020 
5021   Input Parameter:
5022 . dm - the DM
5023 
5024   Output Parameter:
5025 . c - global coordinate vector
5026 
5027   Note:
5028   This is a borrowed reference, so the user should NOT destroy this vector
5029 
5030   Each process has only the local coordinates (does NOT have the ghost coordinates).
5031 
5032   For DMDA, in two and three dimensions coordinates are interlaced (x_0,y_0,x_1,y_1,...)
5033   and (x_0,y_0,z_0,x_1,y_1,z_1...)
5034 
5035   Level: intermediate
5036 
5037 .keywords: distributed array, get, corners, nodes, local indices, coordinates
5038 .seealso: DMSetCoordinates(), DMGetCoordinatesLocal(), DMGetCoordinateDM()
5039 @*/
5040 PetscErrorCode DMGetCoordinates(DM dm, Vec *c)
5041 {
5042   PetscErrorCode ierr;
5043 
5044   PetscFunctionBegin;
5045   PetscValidHeaderSpecific(dm,DM_CLASSID,1);
5046   PetscValidPointer(c,2);
5047   if (!dm->coordinates && dm->coordinatesLocal) {
5048     DM cdm = NULL;
5049 
5050     ierr = DMGetCoordinateDM(dm, &cdm);CHKERRQ(ierr);
5051     ierr = DMCreateGlobalVector(cdm, &dm->coordinates);CHKERRQ(ierr);
5052     ierr = PetscObjectSetName((PetscObject) dm->coordinates, "coordinates");CHKERRQ(ierr);
5053     ierr = DMLocalToGlobalBegin(cdm, dm->coordinatesLocal, INSERT_VALUES, dm->coordinates);CHKERRQ(ierr);
5054     ierr = DMLocalToGlobalEnd(cdm, dm->coordinatesLocal, INSERT_VALUES, dm->coordinates);CHKERRQ(ierr);
5055   }
5056   *c = dm->coordinates;
5057   PetscFunctionReturn(0);
5058 }
5059 
5060 /*@
5061   DMGetCoordinatesLocalSetUp - Prepares a local vector of coordinates, so that DMGetCoordinatesLocalNoncollective() can be used as non-collective afterwards.
5062 
5063   Collective on DM
5064 
5065   Input Parameter:
5066 . dm - the DM
5067 
5068   Level: advanced
5069 
5070 .keywords: distributed array, get, corners, nodes, local indices, coordinates
5071 .seealso: DMGetCoordinatesLocalNoncollective()
5072 @*/
5073 PetscErrorCode DMGetCoordinatesLocalSetUp(DM dm)
5074 {
5075   DM cdm = NULL;
5076   PetscErrorCode ierr;
5077 
5078   PetscFunctionBegin;
5079   PetscValidHeaderSpecific(dm,DM_CLASSID,1);
5080   if (!dm->coordinatesLocal && dm->coordinates) {
5081     ierr = DMGetCoordinateDM(dm, &cdm);CHKERRQ(ierr);
5082     ierr = DMCreateLocalVector(cdm, &dm->coordinatesLocal);CHKERRQ(ierr);
5083     ierr = PetscObjectSetName((PetscObject) dm->coordinatesLocal, "coordinates");CHKERRQ(ierr);
5084     ierr = DMGlobalToLocalBegin(cdm, dm->coordinates, INSERT_VALUES, dm->coordinatesLocal);CHKERRQ(ierr);
5085     ierr = DMGlobalToLocalEnd(cdm, dm->coordinates, INSERT_VALUES, dm->coordinatesLocal);CHKERRQ(ierr);
5086   }
5087   PetscFunctionReturn(0);
5088 }
5089 
5090 /*@
5091   DMGetCoordinatesLocal - Gets a local vector with the coordinates associated with the DM.
5092 
5093   Collective on DM
5094 
5095   Input Parameter:
5096 . dm - the DM
5097 
5098   Output Parameter:
5099 . c - coordinate vector
5100 
5101   Note:
5102   This is a borrowed reference, so the user should NOT destroy this vector
5103 
5104   Each process has the local and ghost coordinates
5105 
5106   For DMDA, in two and three dimensions coordinates are interlaced (x_0,y_0,x_1,y_1,...)
5107   and (x_0,y_0,z_0,x_1,y_1,z_1...)
5108 
5109   Level: intermediate
5110 
5111 .keywords: distributed array, get, corners, nodes, local indices, coordinates
5112 .seealso: DMSetCoordinatesLocal(), DMGetCoordinates(), DMSetCoordinates(), DMGetCoordinateDM(), DMGetCoordinatesLocalNoncollective()
5113 @*/
5114 PetscErrorCode DMGetCoordinatesLocal(DM dm, Vec *c)
5115 {
5116   PetscErrorCode ierr;
5117 
5118   PetscFunctionBegin;
5119   PetscValidHeaderSpecific(dm,DM_CLASSID,1);
5120   PetscValidPointer(c,2);
5121   ierr = DMGetCoordinatesLocalSetUp(dm);CHKERRQ(ierr);
5122   *c = dm->coordinatesLocal;
5123   PetscFunctionReturn(0);
5124 }
5125 
5126 /*@
5127   DMGetCoordinatesLocalNoncollective - Non-collective version of DMGetCoordinatesLocal(). Fails if global coordinates have been set and DMGetCoordinatesLocalSetUp() not called.
5128 
5129   Not collective
5130 
5131   Input Parameter:
5132 . dm - the DM
5133 
5134   Output Parameter:
5135 . c - coordinate vector
5136 
5137   Level: advanced
5138 
5139 .keywords: distributed array, get, corners, nodes, local indices, coordinates
5140 .seealso: DMGetCoordinatesLocalSetUp(), DMGetCoordinatesLocal(), DMSetCoordinatesLocal(), DMGetCoordinates(), DMSetCoordinates(), DMGetCoordinateDM()
5141 @*/
5142 PetscErrorCode DMGetCoordinatesLocalNoncollective(DM dm, Vec *c)
5143 {
5144   PetscFunctionBegin;
5145   PetscValidHeaderSpecific(dm,DM_CLASSID,1);
5146   PetscValidPointer(c,2);
5147   if (!dm->coordinatesLocal && dm->coordinates) SETERRQ(PetscObjectComm((PetscObject)dm), PETSC_ERR_ARG_WRONGSTATE, "DMGetCoordinatesLocalSetUp() has not been called");
5148   *c = dm->coordinatesLocal;
5149   PetscFunctionReturn(0);
5150 }
5151 
5152 /*@
5153   DMGetCoordinatesLocalTuple - Gets a local vector with the coordinates of specified points and section describing its layout.
5154 
5155   Not collective
5156 
5157   Input Parameter:
5158 + dm - the DM
5159 - p - the IS of points whose coordinates will be returned
5160 
5161   Output Parameter:
5162 + pCoordSection - the PetscSection describing the layout of pCoord, i.e. each point corresponds to one point in p, and DOFs correspond to coordinates
5163 - pCoord - the Vec with coordinates of points in p
5164 
5165   Note:
5166   DMGetCoordinatesLocalSetUp() must be called first. This function employs DMGetCoordinatesLocalNoncollective() so it is not collective.
5167 
5168   This creates a new vector, so the user SHOULD destroy this vector
5169 
5170   Each process has the local and ghost coordinates
5171 
5172   For DMDA, in two and three dimensions coordinates are interlaced (x_0,y_0,x_1,y_1,...)
5173   and (x_0,y_0,z_0,x_1,y_1,z_1...)
5174 
5175   Level: advanced
5176 
5177 .keywords: distributed array, get, corners, nodes, local indices, coordinates
5178 .seealso: DMSetCoordinatesLocal(), DMGetCoordinatesLocal(), DMGetCoordinatesLocalNoncollective(), DMGetCoordinatesLocalSetUp(), DMGetCoordinates(), DMSetCoordinates(), DMGetCoordinateDM()
5179 @*/
5180 PetscErrorCode DMGetCoordinatesLocalTuple(DM dm, IS p, PetscSection *pCoordSection, Vec *pCoord)
5181 {
5182   PetscSection        cs, newcs;
5183   Vec                 coords;
5184   const PetscScalar   *arr;
5185   PetscScalar         *newarr=NULL;
5186   PetscInt            n;
5187   PetscErrorCode      ierr;
5188 
5189   PetscFunctionBegin;
5190   PetscValidHeaderSpecific(dm, DM_CLASSID, 1);
5191   PetscValidHeaderSpecific(p, IS_CLASSID, 2);
5192   if (pCoordSection) PetscValidPointer(pCoordSection, 3);
5193   if (pCoord) PetscValidPointer(pCoord, 4);
5194   if (!dm->coordinatesLocal) SETERRQ(PetscObjectComm((PetscObject)dm), PETSC_ERR_ARG_WRONGSTATE, "DMGetCoordinatesLocalSetUp() has not been called or coordinates not set");
5195   if (!dm->coordinateDM || !dm->coordinateDM->defaultSection) SETERRQ(PetscObjectComm((PetscObject)dm), PETSC_ERR_ARG_WRONGSTATE, "DM not supported");
5196   cs = dm->coordinateDM->defaultSection;
5197   coords = dm->coordinatesLocal;
5198   ierr = VecGetArrayRead(coords, &arr);CHKERRQ(ierr);
5199   ierr = PetscSectionExtractDofsFromArray(cs, MPIU_SCALAR, arr, p, &newcs, pCoord ? ((void**)&newarr) : NULL);CHKERRQ(ierr);
5200   ierr = VecRestoreArrayRead(coords, &arr);CHKERRQ(ierr);
5201   if (pCoord) {
5202     ierr = PetscSectionGetStorageSize(newcs, &n);CHKERRQ(ierr);
5203     /* set array in two steps to mimic PETSC_OWN_POINTER */
5204     ierr = VecCreateSeqWithArray(PetscObjectComm((PetscObject)p), 1, n, NULL, pCoord);CHKERRQ(ierr);
5205     ierr = VecReplaceArray(*pCoord, newarr);CHKERRQ(ierr);
5206   } else {
5207     ierr = PetscFree(newarr);CHKERRQ(ierr);
5208   }
5209   if (pCoordSection) {*pCoordSection = newcs;}
5210   else               {ierr = PetscSectionDestroy(&newcs);CHKERRQ(ierr);}
5211   PetscFunctionReturn(0);
5212 }
5213 
5214 PetscErrorCode DMGetCoordinateField(DM dm, DMField *field)
5215 {
5216   PetscErrorCode ierr;
5217 
5218   PetscFunctionBegin;
5219   PetscValidHeaderSpecific(dm,DM_CLASSID,1);
5220   PetscValidPointer(field,2);
5221   if (!dm->coordinateField) {
5222     if (dm->ops->createcoordinatefield) {
5223       ierr = (*dm->ops->createcoordinatefield)(dm,&dm->coordinateField);CHKERRQ(ierr);
5224     }
5225   }
5226   *field = dm->coordinateField;
5227   PetscFunctionReturn(0);
5228 }
5229 
5230 PetscErrorCode DMSetCoordinateField(DM dm, DMField field)
5231 {
5232   PetscErrorCode ierr;
5233 
5234   PetscFunctionBegin;
5235   PetscValidHeaderSpecific(dm,DM_CLASSID,1);
5236   if (field) PetscValidHeaderSpecific(field,DMFIELD_CLASSID,2);
5237   ierr = PetscObjectReference((PetscObject)field);CHKERRQ(ierr);
5238   ierr = DMFieldDestroy(&dm->coordinateField);CHKERRQ(ierr);
5239   dm->coordinateField = field;
5240   PetscFunctionReturn(0);
5241 }
5242 
5243 /*@
5244   DMGetCoordinateDM - Gets the DM that prescribes coordinate layout and scatters between global and local coordinates
5245 
5246   Collective on DM
5247 
5248   Input Parameter:
5249 . dm - the DM
5250 
5251   Output Parameter:
5252 . cdm - coordinate DM
5253 
5254   Level: intermediate
5255 
5256 .keywords: distributed array, get, corners, nodes, local indices, coordinates
5257 .seealso: DMSetCoordinateDM(), DMSetCoordinates(), DMSetCoordinatesLocal(), DMGetCoordinates(), DMGetCoordinatesLocal()
5258 @*/
5259 PetscErrorCode DMGetCoordinateDM(DM dm, DM *cdm)
5260 {
5261   PetscErrorCode ierr;
5262 
5263   PetscFunctionBegin;
5264   PetscValidHeaderSpecific(dm,DM_CLASSID,1);
5265   PetscValidPointer(cdm,2);
5266   if (!dm->coordinateDM) {
5267     DM cdm;
5268 
5269     if (!dm->ops->createcoordinatedm) SETERRQ(PetscObjectComm((PetscObject)dm), PETSC_ERR_SUP, "Unable to create coordinates for this DM");
5270     ierr = (*dm->ops->createcoordinatedm)(dm, &cdm);CHKERRQ(ierr);
5271     /* Just in case the DM sets the coordinate DM when creating it (DMP4est can do this, because it may not setup
5272      * until the call to CreateCoordinateDM) */
5273     ierr = DMDestroy(&dm->coordinateDM);CHKERRQ(ierr);
5274     dm->coordinateDM = cdm;
5275   }
5276   *cdm = dm->coordinateDM;
5277   PetscFunctionReturn(0);
5278 }
5279 
5280 /*@
5281   DMSetCoordinateDM - Sets the DM that prescribes coordinate layout and scatters between global and local coordinates
5282 
5283   Logically Collective on DM
5284 
5285   Input Parameters:
5286 + dm - the DM
5287 - cdm - coordinate DM
5288 
5289   Level: intermediate
5290 
5291 .keywords: distributed array, get, corners, nodes, local indices, coordinates
5292 .seealso: DMGetCoordinateDM(), DMSetCoordinates(), DMSetCoordinatesLocal(), DMGetCoordinates(), DMGetCoordinatesLocal()
5293 @*/
5294 PetscErrorCode DMSetCoordinateDM(DM dm, DM cdm)
5295 {
5296   PetscErrorCode ierr;
5297 
5298   PetscFunctionBegin;
5299   PetscValidHeaderSpecific(dm,DM_CLASSID,1);
5300   PetscValidHeaderSpecific(cdm,DM_CLASSID,2);
5301   ierr = PetscObjectReference((PetscObject)cdm);CHKERRQ(ierr);
5302   ierr = DMDestroy(&dm->coordinateDM);CHKERRQ(ierr);
5303   dm->coordinateDM = cdm;
5304   PetscFunctionReturn(0);
5305 }
5306 
5307 /*@
5308   DMGetCoordinateDim - Retrieve the dimension of embedding space for coordinate values.
5309 
5310   Not Collective
5311 
5312   Input Parameter:
5313 . dm - The DM object
5314 
5315   Output Parameter:
5316 . dim - The embedding dimension
5317 
5318   Level: intermediate
5319 
5320 .keywords: mesh, coordinates
5321 .seealso: DMSetCoordinateDim(), DMGetCoordinateSection(), DMGetCoordinateDM(), DMGetSection(), DMSetSection()
5322 @*/
5323 PetscErrorCode DMGetCoordinateDim(DM dm, PetscInt *dim)
5324 {
5325   PetscFunctionBegin;
5326   PetscValidHeaderSpecific(dm, DM_CLASSID, 1);
5327   PetscValidPointer(dim, 2);
5328   if (dm->dimEmbed == PETSC_DEFAULT) {
5329     dm->dimEmbed = dm->dim;
5330   }
5331   *dim = dm->dimEmbed;
5332   PetscFunctionReturn(0);
5333 }
5334 
5335 /*@
5336   DMSetCoordinateDim - Set the dimension of the embedding space for coordinate values.
5337 
5338   Not Collective
5339 
5340   Input Parameters:
5341 + dm  - The DM object
5342 - dim - The embedding dimension
5343 
5344   Level: intermediate
5345 
5346 .keywords: mesh, coordinates
5347 .seealso: DMGetCoordinateDim(), DMSetCoordinateSection(), DMGetCoordinateSection(), DMGetSection(), DMSetSection()
5348 @*/
5349 PetscErrorCode DMSetCoordinateDim(DM dm, PetscInt dim)
5350 {
5351   PetscDS        ds;
5352   PetscErrorCode ierr;
5353 
5354   PetscFunctionBegin;
5355   PetscValidHeaderSpecific(dm,DM_CLASSID,1);
5356   dm->dimEmbed = dim;
5357   ierr = DMGetDS(dm, &ds);CHKERRQ(ierr);
5358   ierr = PetscDSSetCoordinateDimension(ds, dim);CHKERRQ(ierr);
5359   PetscFunctionReturn(0);
5360 }
5361 
5362 /*@
5363   DMGetCoordinateSection - Retrieve the layout of coordinate values over the mesh.
5364 
5365   Collective on DM
5366 
5367   Input Parameter:
5368 . dm - The DM object
5369 
5370   Output Parameter:
5371 . section - The PetscSection object
5372 
5373   Level: intermediate
5374 
5375 .keywords: mesh, coordinates
5376 .seealso: DMGetCoordinateDM(), DMGetSection(), DMSetSection()
5377 @*/
5378 PetscErrorCode DMGetCoordinateSection(DM dm, PetscSection *section)
5379 {
5380   DM             cdm;
5381   PetscErrorCode ierr;
5382 
5383   PetscFunctionBegin;
5384   PetscValidHeaderSpecific(dm, DM_CLASSID, 1);
5385   PetscValidPointer(section, 2);
5386   ierr = DMGetCoordinateDM(dm, &cdm);CHKERRQ(ierr);
5387   ierr = DMGetSection(cdm, section);CHKERRQ(ierr);
5388   PetscFunctionReturn(0);
5389 }
5390 
5391 /*@
5392   DMSetCoordinateSection - Set the layout of coordinate values over the mesh.
5393 
5394   Not Collective
5395 
5396   Input Parameters:
5397 + dm      - The DM object
5398 . dim     - The embedding dimension, or PETSC_DETERMINE
5399 - section - The PetscSection object
5400 
5401   Level: intermediate
5402 
5403 .keywords: mesh, coordinates
5404 .seealso: DMGetCoordinateSection(), DMGetSection(), DMSetSection()
5405 @*/
5406 PetscErrorCode DMSetCoordinateSection(DM dm, PetscInt dim, PetscSection section)
5407 {
5408   DM             cdm;
5409   PetscErrorCode ierr;
5410 
5411   PetscFunctionBegin;
5412   PetscValidHeaderSpecific(dm,DM_CLASSID,1);
5413   PetscValidHeaderSpecific(section,PETSC_SECTION_CLASSID,3);
5414   ierr = DMGetCoordinateDM(dm, &cdm);CHKERRQ(ierr);
5415   ierr = DMSetSection(cdm, section);CHKERRQ(ierr);
5416   if (dim == PETSC_DETERMINE) {
5417     PetscInt d = PETSC_DEFAULT;
5418     PetscInt pStart, pEnd, vStart, vEnd, v, dd;
5419 
5420     ierr = PetscSectionGetChart(section, &pStart, &pEnd);CHKERRQ(ierr);
5421     ierr = DMGetDimPoints(dm, 0, &vStart, &vEnd);CHKERRQ(ierr);
5422     pStart = PetscMax(vStart, pStart);
5423     pEnd   = PetscMin(vEnd, pEnd);
5424     for (v = pStart; v < pEnd; ++v) {
5425       ierr = PetscSectionGetDof(section, v, &dd);CHKERRQ(ierr);
5426       if (dd) {d = dd; break;}
5427     }
5428     if (d >= 0) {ierr = DMSetCoordinateDim(dm, d);CHKERRQ(ierr);}
5429   }
5430   PetscFunctionReturn(0);
5431 }
5432 
5433 /*@C
5434   DMGetPeriodicity - Get the description of mesh periodicity
5435 
5436   Input Parameters:
5437 . dm      - The DM object
5438 
5439   Output Parameters:
5440 + per     - Whether the DM is periodic or not
5441 . maxCell - Over distances greater than this, we can assume a point has crossed over to another sheet, when trying to localize cell coordinates
5442 . L       - If we assume the mesh is a torus, this is the length of each coordinate
5443 - bd      - This describes the type of periodicity in each topological dimension
5444 
5445   Level: developer
5446 
5447 .seealso: DMGetPeriodicity()
5448 @*/
5449 PetscErrorCode DMGetPeriodicity(DM dm, PetscBool *per, const PetscReal **maxCell, const PetscReal **L, const DMBoundaryType **bd)
5450 {
5451   PetscFunctionBegin;
5452   PetscValidHeaderSpecific(dm,DM_CLASSID,1);
5453   if (per)     *per     = dm->periodic;
5454   if (L)       *L       = dm->L;
5455   if (maxCell) *maxCell = dm->maxCell;
5456   if (bd)      *bd      = dm->bdtype;
5457   PetscFunctionReturn(0);
5458 }
5459 
5460 /*@C
5461   DMSetPeriodicity - Set the description of mesh periodicity
5462 
5463   Input Parameters:
5464 + dm      - The DM object
5465 . per     - Whether the DM is periodic or not. If maxCell is not provided, coordinates need to be localized
5466 . maxCell - Over distances greater than this, we can assume a point has crossed over to another sheet, when trying to localize cell coordinates
5467 . L       - If we assume the mesh is a torus, this is the length of each coordinate
5468 - bd      - This describes the type of periodicity in each topological dimension
5469 
5470   Level: developer
5471 
5472 .seealso: DMGetPeriodicity()
5473 @*/
5474 PetscErrorCode DMSetPeriodicity(DM dm, PetscBool per, const PetscReal maxCell[], const PetscReal L[], const DMBoundaryType bd[])
5475 {
5476   PetscInt       dim, d;
5477   PetscErrorCode ierr;
5478 
5479   PetscFunctionBegin;
5480   PetscValidHeaderSpecific(dm,DM_CLASSID,1);
5481   PetscValidLogicalCollectiveBool(dm,per,2);
5482   if (maxCell) {
5483     PetscValidPointer(maxCell,3);
5484     PetscValidPointer(L,4);
5485     PetscValidPointer(bd,5);
5486   }
5487   ierr = PetscFree3(dm->L,dm->maxCell,dm->bdtype);CHKERRQ(ierr);
5488   ierr = DMGetDimension(dm, &dim);CHKERRQ(ierr);
5489   if (maxCell) {
5490     ierr = PetscMalloc3(dim,&dm->L,dim,&dm->maxCell,dim,&dm->bdtype);CHKERRQ(ierr);
5491     for (d = 0; d < dim; ++d) {dm->L[d] = L[d]; dm->maxCell[d] = maxCell[d]; dm->bdtype[d] = bd[d];}
5492   }
5493   dm->periodic = per;
5494   PetscFunctionReturn(0);
5495 }
5496 
5497 /*@
5498   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.
5499 
5500   Input Parameters:
5501 + dm     - The DM
5502 . in     - The input coordinate point (dim numbers)
5503 - endpoint - Include the endpoint L_i
5504 
5505   Output Parameter:
5506 . out - The localized coordinate point
5507 
5508   Level: developer
5509 
5510 .seealso: DMLocalizeCoordinates(), DMLocalizeAddCoordinate()
5511 @*/
5512 PetscErrorCode DMLocalizeCoordinate(DM dm, const PetscScalar in[], PetscBool endpoint, PetscScalar out[])
5513 {
5514   PetscInt       dim, d;
5515   PetscErrorCode ierr;
5516 
5517   PetscFunctionBegin;
5518   ierr = DMGetCoordinateDim(dm, &dim);CHKERRQ(ierr);
5519   if (!dm->maxCell) {
5520     for (d = 0; d < dim; ++d) out[d] = in[d];
5521   } else {
5522     if (endpoint) {
5523       for (d = 0; d < dim; ++d) {
5524         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)) {
5525           out[d] = in[d] - dm->L[d]*(PetscFloorReal(PetscRealPart(in[d])/dm->L[d]) - 1);
5526         } else {
5527           out[d] = in[d] - dm->L[d]*PetscFloorReal(PetscRealPart(in[d])/dm->L[d]);
5528         }
5529       }
5530     } else {
5531       for (d = 0; d < dim; ++d) {
5532         out[d] = in[d] - dm->L[d]*PetscFloorReal(PetscRealPart(in[d])/dm->L[d]);
5533       }
5534     }
5535   }
5536   PetscFunctionReturn(0);
5537 }
5538 
5539 /*
5540   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.
5541 
5542   Input Parameters:
5543 + dm     - The DM
5544 . dim    - The spatial dimension
5545 . anchor - The anchor point, the input point can be no more than maxCell away from it
5546 - in     - The input coordinate point (dim numbers)
5547 
5548   Output Parameter:
5549 . out - The localized coordinate point
5550 
5551   Level: developer
5552 
5553   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
5554 
5555 .seealso: DMLocalizeCoordinates(), DMLocalizeAddCoordinate()
5556 */
5557 PetscErrorCode DMLocalizeCoordinate_Internal(DM dm, PetscInt dim, const PetscScalar anchor[], const PetscScalar in[], PetscScalar out[])
5558 {
5559   PetscInt d;
5560 
5561   PetscFunctionBegin;
5562   if (!dm->maxCell) {
5563     for (d = 0; d < dim; ++d) out[d] = in[d];
5564   } else {
5565     for (d = 0; d < dim; ++d) {
5566       if ((dm->bdtype[d] != DM_BOUNDARY_NONE) && (PetscAbsScalar(anchor[d] - in[d]) > dm->maxCell[d])) {
5567         out[d] = PetscRealPart(anchor[d]) > PetscRealPart(in[d]) ? dm->L[d] + in[d] : in[d] - dm->L[d];
5568       } else {
5569         out[d] = in[d];
5570       }
5571     }
5572   }
5573   PetscFunctionReturn(0);
5574 }
5575 PetscErrorCode DMLocalizeCoordinateReal_Internal(DM dm, PetscInt dim, const PetscReal anchor[], const PetscReal in[], PetscReal out[])
5576 {
5577   PetscInt d;
5578 
5579   PetscFunctionBegin;
5580   if (!dm->maxCell) {
5581     for (d = 0; d < dim; ++d) out[d] = in[d];
5582   } else {
5583     for (d = 0; d < dim; ++d) {
5584       if ((dm->bdtype[d] != DM_BOUNDARY_NONE) && (PetscAbsReal(anchor[d] - in[d]) > dm->maxCell[d])) {
5585         out[d] = anchor[d] > in[d] ? dm->L[d] + in[d] : in[d] - dm->L[d];
5586       } else {
5587         out[d] = in[d];
5588       }
5589     }
5590   }
5591   PetscFunctionReturn(0);
5592 }
5593 
5594 /*
5595   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.
5596 
5597   Input Parameters:
5598 + dm     - The DM
5599 . dim    - The spatial dimension
5600 . anchor - The anchor point, the input point can be no more than maxCell away from it
5601 . in     - The input coordinate delta (dim numbers)
5602 - out    - The input coordinate point (dim numbers)
5603 
5604   Output Parameter:
5605 . out    - The localized coordinate in + out
5606 
5607   Level: developer
5608 
5609   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
5610 
5611 .seealso: DMLocalizeCoordinates(), DMLocalizeCoordinate()
5612 */
5613 PetscErrorCode DMLocalizeAddCoordinate_Internal(DM dm, PetscInt dim, const PetscScalar anchor[], const PetscScalar in[], PetscScalar out[])
5614 {
5615   PetscInt d;
5616 
5617   PetscFunctionBegin;
5618   if (!dm->maxCell) {
5619     for (d = 0; d < dim; ++d) out[d] += in[d];
5620   } else {
5621     for (d = 0; d < dim; ++d) {
5622       if ((dm->bdtype[d] != DM_BOUNDARY_NONE) && (PetscAbsScalar(anchor[d] - in[d]) > dm->maxCell[d])) {
5623         out[d] += PetscRealPart(anchor[d]) > PetscRealPart(in[d]) ? dm->L[d] + in[d] : in[d] - dm->L[d];
5624       } else {
5625         out[d] += in[d];
5626       }
5627     }
5628   }
5629   PetscFunctionReturn(0);
5630 }
5631 
5632 /*@
5633   DMGetCoordinatesLocalizedLocal - Check if the DM coordinates have been localized for cells on this process
5634 
5635   Not collective
5636 
5637   Input Parameter:
5638 . dm - The DM
5639 
5640   Output Parameter:
5641   areLocalized - True if localized
5642 
5643   Level: developer
5644 
5645 .seealso: DMLocalizeCoordinates(), DMGetCoordinatesLocalized(), DMSetPeriodicity()
5646 @*/
5647 PetscErrorCode DMGetCoordinatesLocalizedLocal(DM dm,PetscBool *areLocalized)
5648 {
5649   DM             cdm;
5650   PetscSection   coordSection;
5651   PetscInt       cStart, cEnd, sStart, sEnd, c, dof;
5652   PetscBool      isPlex, alreadyLocalized;
5653   PetscErrorCode ierr;
5654 
5655   PetscFunctionBegin;
5656   PetscValidHeaderSpecific(dm, DM_CLASSID, 1);
5657   PetscValidPointer(areLocalized, 2);
5658   *areLocalized = PETSC_FALSE;
5659 
5660   /* We need some generic way of refering to cells/vertices */
5661   ierr = DMGetCoordinateDM(dm, &cdm);CHKERRQ(ierr);
5662   ierr = DMGetCoordinateSection(dm, &coordSection);CHKERRQ(ierr);
5663   ierr = PetscObjectTypeCompare((PetscObject) cdm, DMPLEX, &isPlex);CHKERRQ(ierr);
5664   if (!isPlex) SETERRQ(PetscObjectComm((PetscObject) cdm), PETSC_ERR_ARG_WRONG, "Coordinate localization requires a DMPLEX coordinate DM");
5665 
5666   ierr = DMPlexGetHeightStratum(cdm, 0, &cStart, &cEnd);CHKERRQ(ierr);
5667   ierr = PetscSectionGetChart(coordSection, &sStart, &sEnd);CHKERRQ(ierr);
5668   alreadyLocalized = PETSC_FALSE;
5669   for (c = cStart; c < cEnd; ++c) {
5670     if (c < sStart || c >= sEnd) continue;
5671     ierr = PetscSectionGetDof(coordSection, c, &dof);CHKERRQ(ierr);
5672     if (dof) { alreadyLocalized = PETSC_TRUE; break; }
5673   }
5674   *areLocalized = alreadyLocalized;
5675   PetscFunctionReturn(0);
5676 }
5677 
5678 /*@
5679   DMGetCoordinatesLocalized - Check if the DM coordinates have been localized for cells
5680 
5681   Collective on dm
5682 
5683   Input Parameter:
5684 . dm - The DM
5685 
5686   Output Parameter:
5687   areLocalized - True if localized
5688 
5689   Level: developer
5690 
5691 .seealso: DMLocalizeCoordinates(), DMSetPeriodicity(), DMGetCoordinatesLocalizedLocal()
5692 @*/
5693 PetscErrorCode DMGetCoordinatesLocalized(DM dm,PetscBool *areLocalized)
5694 {
5695   PetscBool      localized;
5696   PetscErrorCode ierr;
5697 
5698   PetscFunctionBegin;
5699   PetscValidHeaderSpecific(dm, DM_CLASSID, 1);
5700   PetscValidPointer(areLocalized, 2);
5701   ierr = DMGetCoordinatesLocalizedLocal(dm,&localized);CHKERRQ(ierr);
5702   ierr = MPIU_Allreduce(&localized,areLocalized,1,MPIU_BOOL,MPI_LOR,PetscObjectComm((PetscObject)dm));CHKERRQ(ierr);
5703   PetscFunctionReturn(0);
5704 }
5705 
5706 /*@
5707   DMLocalizeCoordinates - If a mesh is periodic, create local coordinates for cells having periodic faces
5708 
5709   Collective on dm
5710 
5711   Input Parameter:
5712 . dm - The DM
5713 
5714   Level: developer
5715 
5716 .seealso: DMSetPeriodicity(), DMLocalizeCoordinate(), DMLocalizeAddCoordinate()
5717 @*/
5718 PetscErrorCode DMLocalizeCoordinates(DM dm)
5719 {
5720   DM             cdm;
5721   PetscSection   coordSection, cSection;
5722   Vec            coordinates,  cVec;
5723   PetscScalar   *coords, *coords2, *anchor, *localized;
5724   PetscInt       Nc, vStart, vEnd, v, sStart, sEnd, newStart = PETSC_MAX_INT, newEnd = PETSC_MIN_INT, dof, d, off, off2, bs, coordSize;
5725   PetscBool      alreadyLocalized, alreadyLocalizedGlobal;
5726   PetscInt       maxHeight = 0, h;
5727   PetscInt       *pStart = NULL, *pEnd = NULL;
5728   PetscErrorCode ierr;
5729 
5730   PetscFunctionBegin;
5731   PetscValidHeaderSpecific(dm, DM_CLASSID, 1);
5732   if (!dm->periodic) PetscFunctionReturn(0);
5733   ierr = DMGetCoordinatesLocalized(dm, &alreadyLocalized);CHKERRQ(ierr);
5734   if (alreadyLocalized) PetscFunctionReturn(0);
5735 
5736   /* We need some generic way of refering to cells/vertices */
5737   ierr = DMGetCoordinateDM(dm, &cdm);CHKERRQ(ierr);
5738   {
5739     PetscBool isplex;
5740 
5741     ierr = PetscObjectTypeCompare((PetscObject) cdm, DMPLEX, &isplex);CHKERRQ(ierr);
5742     if (isplex) {
5743       ierr = DMPlexGetDepthStratum(cdm, 0, &vStart, &vEnd);CHKERRQ(ierr);
5744       ierr = DMPlexGetMaxProjectionHeight(cdm,&maxHeight);CHKERRQ(ierr);
5745       ierr = DMGetWorkArray(dm,2*(maxHeight + 1),MPIU_INT,&pStart);CHKERRQ(ierr);
5746       pEnd = &pStart[maxHeight + 1];
5747       newStart = vStart;
5748       newEnd   = vEnd;
5749       for (h = 0; h <= maxHeight; h++) {
5750         ierr = DMPlexGetHeightStratum(cdm, h, &pStart[h], &pEnd[h]);CHKERRQ(ierr);
5751         newStart = PetscMin(newStart,pStart[h]);
5752         newEnd   = PetscMax(newEnd,pEnd[h]);
5753       }
5754     } else SETERRQ(PetscObjectComm((PetscObject) cdm), PETSC_ERR_ARG_WRONG, "Coordinate localization requires a DMPLEX coordinate DM");
5755   }
5756   ierr = DMGetCoordinatesLocal(dm, &coordinates);CHKERRQ(ierr);
5757   if (!coordinates) SETERRQ(PetscObjectComm((PetscObject)dm),PETSC_ERR_SUP,"Missing local coordinates vector");
5758   ierr = DMGetCoordinateSection(dm, &coordSection);CHKERRQ(ierr);
5759   ierr = VecGetBlockSize(coordinates, &bs);CHKERRQ(ierr);
5760   ierr = PetscSectionGetChart(coordSection,&sStart,&sEnd);CHKERRQ(ierr);
5761 
5762   ierr = PetscSectionCreate(PetscObjectComm((PetscObject) dm), &cSection);CHKERRQ(ierr);
5763   ierr = PetscSectionSetNumFields(cSection, 1);CHKERRQ(ierr);
5764   ierr = PetscSectionGetFieldComponents(coordSection, 0, &Nc);CHKERRQ(ierr);
5765   ierr = PetscSectionSetFieldComponents(cSection, 0, Nc);CHKERRQ(ierr);
5766   ierr = PetscSectionSetChart(cSection, newStart, newEnd);CHKERRQ(ierr);
5767 
5768   ierr = DMGetWorkArray(dm, 2 * bs, MPIU_SCALAR, &anchor);CHKERRQ(ierr);
5769   localized = &anchor[bs];
5770   alreadyLocalized = alreadyLocalizedGlobal = PETSC_TRUE;
5771   for (h = 0; h <= maxHeight; h++) {
5772     PetscInt cStart = pStart[h], cEnd = pEnd[h], c;
5773 
5774     for (c = cStart; c < cEnd; ++c) {
5775       PetscScalar *cellCoords = NULL;
5776       PetscInt     b;
5777 
5778       if (c < sStart || c >= sEnd) alreadyLocalized = PETSC_FALSE;
5779       ierr = DMPlexVecGetClosure(cdm, coordSection, coordinates, c, &dof, &cellCoords);CHKERRQ(ierr);
5780       for (b = 0; b < bs; ++b) anchor[b] = cellCoords[b];
5781       for (d = 0; d < dof/bs; ++d) {
5782         ierr = DMLocalizeCoordinate_Internal(dm, bs, anchor, &cellCoords[d*bs], localized);CHKERRQ(ierr);
5783         for (b = 0; b < bs; b++) {
5784           if (cellCoords[d*bs + b] != localized[b]) break;
5785         }
5786         if (b < bs) break;
5787       }
5788       if (d < dof/bs) {
5789         if (c >= sStart && c < sEnd) {
5790           PetscInt cdof;
5791 
5792           ierr = PetscSectionGetDof(coordSection, c, &cdof);CHKERRQ(ierr);
5793           if (cdof != dof) alreadyLocalized = PETSC_FALSE;
5794         }
5795         ierr = PetscSectionSetDof(cSection, c, dof);CHKERRQ(ierr);
5796         ierr = PetscSectionSetFieldDof(cSection, c, 0, dof);CHKERRQ(ierr);
5797       }
5798       ierr = DMPlexVecRestoreClosure(cdm, coordSection, coordinates, c, &dof, &cellCoords);CHKERRQ(ierr);
5799     }
5800   }
5801   ierr = MPI_Allreduce(&alreadyLocalized,&alreadyLocalizedGlobal,1,MPIU_BOOL,MPI_LAND,PetscObjectComm((PetscObject)dm));CHKERRQ(ierr);
5802   if (alreadyLocalizedGlobal) {
5803     ierr = DMRestoreWorkArray(dm, 2 * bs, MPIU_SCALAR, &anchor);CHKERRQ(ierr);
5804     ierr = PetscSectionDestroy(&cSection);CHKERRQ(ierr);
5805     ierr = DMRestoreWorkArray(dm,2*(maxHeight + 1),MPIU_INT,&pStart);CHKERRQ(ierr);
5806     PetscFunctionReturn(0);
5807   }
5808   for (v = vStart; v < vEnd; ++v) {
5809     ierr = PetscSectionGetDof(coordSection, v, &dof);CHKERRQ(ierr);
5810     ierr = PetscSectionSetDof(cSection, v, dof);CHKERRQ(ierr);
5811     ierr = PetscSectionSetFieldDof(cSection, v, 0, dof);CHKERRQ(ierr);
5812   }
5813   ierr = PetscSectionSetUp(cSection);CHKERRQ(ierr);
5814   ierr = PetscSectionGetStorageSize(cSection, &coordSize);CHKERRQ(ierr);
5815   ierr = VecCreate(PETSC_COMM_SELF, &cVec);CHKERRQ(ierr);
5816   ierr = PetscObjectSetName((PetscObject)cVec,"coordinates");CHKERRQ(ierr);
5817   ierr = VecSetBlockSize(cVec, bs);CHKERRQ(ierr);
5818   ierr = VecSetSizes(cVec, coordSize, PETSC_DETERMINE);CHKERRQ(ierr);
5819   ierr = VecSetType(cVec, VECSTANDARD);CHKERRQ(ierr);
5820   ierr = VecGetArrayRead(coordinates, (const PetscScalar**)&coords);CHKERRQ(ierr);
5821   ierr = VecGetArray(cVec, &coords2);CHKERRQ(ierr);
5822   for (v = vStart; v < vEnd; ++v) {
5823     ierr = PetscSectionGetDof(coordSection, v, &dof);CHKERRQ(ierr);
5824     ierr = PetscSectionGetOffset(coordSection, v, &off);CHKERRQ(ierr);
5825     ierr = PetscSectionGetOffset(cSection,     v, &off2);CHKERRQ(ierr);
5826     for (d = 0; d < dof; ++d) coords2[off2+d] = coords[off+d];
5827   }
5828   for (h = 0; h <= maxHeight; h++) {
5829     PetscInt cStart = pStart[h], cEnd = pEnd[h], c;
5830 
5831     for (c = cStart; c < cEnd; ++c) {
5832       PetscScalar *cellCoords = NULL;
5833       PetscInt     b, cdof;
5834 
5835       ierr = PetscSectionGetDof(cSection,c,&cdof);CHKERRQ(ierr);
5836       if (!cdof) continue;
5837       ierr = DMPlexVecGetClosure(cdm, coordSection, coordinates, c, &dof, &cellCoords);CHKERRQ(ierr);
5838       ierr = PetscSectionGetOffset(cSection, c, &off2);CHKERRQ(ierr);
5839       for (b = 0; b < bs; ++b) anchor[b] = cellCoords[b];
5840       for (d = 0; d < dof/bs; ++d) {ierr = DMLocalizeCoordinate_Internal(dm, bs, anchor, &cellCoords[d*bs], &coords2[off2+d*bs]);CHKERRQ(ierr);}
5841       ierr = DMPlexVecRestoreClosure(cdm, coordSection, coordinates, c, &dof, &cellCoords);CHKERRQ(ierr);
5842     }
5843   }
5844   ierr = DMRestoreWorkArray(dm, 2 * bs, MPIU_SCALAR, &anchor);CHKERRQ(ierr);
5845   ierr = DMRestoreWorkArray(dm,2*(maxHeight + 1),MPIU_INT,&pStart);CHKERRQ(ierr);
5846   ierr = VecRestoreArrayRead(coordinates, (const PetscScalar**)&coords);CHKERRQ(ierr);
5847   ierr = VecRestoreArray(cVec, &coords2);CHKERRQ(ierr);
5848   ierr = DMSetCoordinateSection(dm, PETSC_DETERMINE, cSection);CHKERRQ(ierr);
5849   ierr = DMSetCoordinatesLocal(dm, cVec);CHKERRQ(ierr);
5850   ierr = VecDestroy(&cVec);CHKERRQ(ierr);
5851   ierr = PetscSectionDestroy(&cSection);CHKERRQ(ierr);
5852   PetscFunctionReturn(0);
5853 }
5854 
5855 /*@
5856   DMLocatePoints - Locate the points in v in the mesh and return a PetscSF of the containing cells
5857 
5858   Collective on Vec v (see explanation below)
5859 
5860   Input Parameters:
5861 + dm - The DM
5862 . v - The Vec of points
5863 . ltype - The type of point location, e.g. DM_POINTLOCATION_NONE or DM_POINTLOCATION_NEAREST
5864 - cells - Points to either NULL, or a PetscSF with guesses for which cells contain each point.
5865 
5866   Output Parameter:
5867 + v - The Vec of points, which now contains the nearest mesh points to the given points if DM_POINTLOCATION_NEAREST is used
5868 - cells - The PetscSF containing the ranks and local indices of the containing points.
5869 
5870 
5871   Level: developer
5872 
5873   Notes:
5874   To do a search of the local cells of the mesh, v should have PETSC_COMM_SELF as its communicator.
5875   To do a search of all the cells in the distributed mesh, v should have the same communicator as dm.
5876 
5877   If *cellSF is NULL on input, a PetscSF will be created.
5878   If *cellSF is not NULL on input, it should point to an existing PetscSF, whose graph will be used as initial guesses.
5879 
5880   An array that maps each point to its containing cell can be obtained with
5881 
5882 $    const PetscSFNode *cells;
5883 $    PetscInt           nFound;
5884 $    const PetscInt    *found;
5885 $
5886 $    PetscSFGetGraph(cellSF,NULL,&nFound,&found,&cells);
5887 
5888   Where cells[i].rank is the rank of the cell containing point found[i] (or i if found == NULL), and cells[i].index is
5889   the index of the cell in its rank's local numbering.
5890 
5891 .keywords: point location, mesh
5892 .seealso: DMSetCoordinates(), DMSetCoordinatesLocal(), DMGetCoordinates(), DMGetCoordinatesLocal(), DMPointLocationType
5893 @*/
5894 PetscErrorCode DMLocatePoints(DM dm, Vec v, DMPointLocationType ltype, PetscSF *cellSF)
5895 {
5896   PetscErrorCode ierr;
5897 
5898   PetscFunctionBegin;
5899   PetscValidHeaderSpecific(dm,DM_CLASSID,1);
5900   PetscValidHeaderSpecific(v,VEC_CLASSID,2);
5901   PetscValidPointer(cellSF,4);
5902   if (*cellSF) {
5903     PetscMPIInt result;
5904 
5905     PetscValidHeaderSpecific(*cellSF,PETSCSF_CLASSID,4);
5906     ierr = MPI_Comm_compare(PetscObjectComm((PetscObject)v),PetscObjectComm((PetscObject)*cellSF),&result);CHKERRQ(ierr);
5907     if (result != MPI_IDENT && result != MPI_CONGRUENT) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_INCOMP,"cellSF must have a communicator congruent to v's");
5908   } else {
5909     ierr = PetscSFCreate(PetscObjectComm((PetscObject)v),cellSF);CHKERRQ(ierr);
5910   }
5911   ierr = PetscLogEventBegin(DM_LocatePoints,dm,0,0,0);CHKERRQ(ierr);
5912   if (dm->ops->locatepoints) {
5913     ierr = (*dm->ops->locatepoints)(dm,v,ltype,*cellSF);CHKERRQ(ierr);
5914   } else SETERRQ(PetscObjectComm((PetscObject)dm), PETSC_ERR_SUP, "Point location not available for this DM");
5915   ierr = PetscLogEventEnd(DM_LocatePoints,dm,0,0,0);CHKERRQ(ierr);
5916   PetscFunctionReturn(0);
5917 }
5918 
5919 /*@
5920   DMGetOutputDM - Retrieve the DM associated with the layout for output
5921 
5922   Collective on dm
5923 
5924   Input Parameter:
5925 . dm - The original DM
5926 
5927   Output Parameter:
5928 . odm - The DM which provides the layout for output
5929 
5930   Level: intermediate
5931 
5932 .seealso: VecView(), DMGetGlobalSection()
5933 @*/
5934 PetscErrorCode DMGetOutputDM(DM dm, DM *odm)
5935 {
5936   PetscSection   section;
5937   PetscBool      hasConstraints, ghasConstraints;
5938   PetscErrorCode ierr;
5939 
5940   PetscFunctionBegin;
5941   PetscValidHeaderSpecific(dm,DM_CLASSID,1);
5942   PetscValidPointer(odm,2);
5943   ierr = DMGetSection(dm, &section);CHKERRQ(ierr);
5944   ierr = PetscSectionHasConstraints(section, &hasConstraints);CHKERRQ(ierr);
5945   ierr = MPI_Allreduce(&hasConstraints, &ghasConstraints, 1, MPIU_BOOL, MPI_LOR, PetscObjectComm((PetscObject) dm));CHKERRQ(ierr);
5946   if (!ghasConstraints) {
5947     *odm = dm;
5948     PetscFunctionReturn(0);
5949   }
5950   if (!dm->dmBC) {
5951     PetscSection newSection, gsection;
5952     PetscSF      sf;
5953 
5954     ierr = DMClone(dm, &dm->dmBC);CHKERRQ(ierr);
5955     ierr = DMCopyDisc(dm, dm->dmBC);CHKERRQ(ierr);
5956     ierr = PetscSectionClone(section, &newSection);CHKERRQ(ierr);
5957     ierr = DMSetSection(dm->dmBC, newSection);CHKERRQ(ierr);
5958     ierr = PetscSectionDestroy(&newSection);CHKERRQ(ierr);
5959     ierr = DMGetPointSF(dm->dmBC, &sf);CHKERRQ(ierr);
5960     ierr = PetscSectionCreateGlobalSection(section, sf, PETSC_TRUE, PETSC_FALSE, &gsection);CHKERRQ(ierr);
5961     ierr = DMSetGlobalSection(dm->dmBC, gsection);CHKERRQ(ierr);
5962     ierr = PetscSectionDestroy(&gsection);CHKERRQ(ierr);
5963   }
5964   *odm = dm->dmBC;
5965   PetscFunctionReturn(0);
5966 }
5967 
5968 /*@
5969   DMGetOutputSequenceNumber - Retrieve the sequence number/value for output
5970 
5971   Input Parameter:
5972 . dm - The original DM
5973 
5974   Output Parameters:
5975 + num - The output sequence number
5976 - val - The output sequence value
5977 
5978   Level: intermediate
5979 
5980   Note: This is intended for output that should appear in sequence, for instance
5981   a set of timesteps in an HDF5 file, or a set of realizations of a stochastic system.
5982 
5983 .seealso: VecView()
5984 @*/
5985 PetscErrorCode DMGetOutputSequenceNumber(DM dm, PetscInt *num, PetscReal *val)
5986 {
5987   PetscFunctionBegin;
5988   PetscValidHeaderSpecific(dm,DM_CLASSID,1);
5989   if (num) {PetscValidPointer(num,2); *num = dm->outputSequenceNum;}
5990   if (val) {PetscValidPointer(val,3);*val = dm->outputSequenceVal;}
5991   PetscFunctionReturn(0);
5992 }
5993 
5994 /*@
5995   DMSetOutputSequenceNumber - Set the sequence number/value for output
5996 
5997   Input Parameters:
5998 + dm - The original DM
5999 . num - The output sequence number
6000 - val - The output sequence value
6001 
6002   Level: intermediate
6003 
6004   Note: This is intended for output that should appear in sequence, for instance
6005   a set of timesteps in an HDF5 file, or a set of realizations of a stochastic system.
6006 
6007 .seealso: VecView()
6008 @*/
6009 PetscErrorCode DMSetOutputSequenceNumber(DM dm, PetscInt num, PetscReal val)
6010 {
6011   PetscFunctionBegin;
6012   PetscValidHeaderSpecific(dm,DM_CLASSID,1);
6013   dm->outputSequenceNum = num;
6014   dm->outputSequenceVal = val;
6015   PetscFunctionReturn(0);
6016 }
6017 
6018 /*@C
6019   DMOutputSequenceLoad - Retrieve the sequence value from a Viewer
6020 
6021   Input Parameters:
6022 + dm   - The original DM
6023 . name - The sequence name
6024 - num  - The output sequence number
6025 
6026   Output Parameter:
6027 . val  - The output sequence value
6028 
6029   Level: intermediate
6030 
6031   Note: This is intended for output that should appear in sequence, for instance
6032   a set of timesteps in an HDF5 file, or a set of realizations of a stochastic system.
6033 
6034 .seealso: DMGetOutputSequenceNumber(), DMSetOutputSequenceNumber(), VecView()
6035 @*/
6036 PetscErrorCode DMOutputSequenceLoad(DM dm, PetscViewer viewer, const char *name, PetscInt num, PetscReal *val)
6037 {
6038   PetscBool      ishdf5;
6039   PetscErrorCode ierr;
6040 
6041   PetscFunctionBegin;
6042   PetscValidHeaderSpecific(dm,DM_CLASSID,1);
6043   PetscValidHeaderSpecific(viewer,PETSC_VIEWER_CLASSID,2);
6044   PetscValidPointer(val,4);
6045   ierr = PetscObjectTypeCompare((PetscObject) viewer, PETSCVIEWERHDF5, &ishdf5);CHKERRQ(ierr);
6046   if (ishdf5) {
6047 #if defined(PETSC_HAVE_HDF5)
6048     PetscScalar value;
6049 
6050     ierr = DMSequenceLoad_HDF5_Internal(dm, name, num, &value, viewer);CHKERRQ(ierr);
6051     *val = PetscRealPart(value);
6052 #endif
6053   } else SETERRQ(PETSC_COMM_SELF, PETSC_ERR_ARG_WRONG, "Invalid viewer; open viewer with PetscViewerHDF5Open()");
6054   PetscFunctionReturn(0);
6055 }
6056 
6057 /*@
6058   DMGetUseNatural - Get the flag for creating a mapping to the natural order on distribution
6059 
6060   Not collective
6061 
6062   Input Parameter:
6063 . dm - The DM
6064 
6065   Output Parameter:
6066 . useNatural - The flag to build the mapping to a natural order during distribution
6067 
6068   Level: beginner
6069 
6070 .seealso: DMSetUseNatural(), DMCreate()
6071 @*/
6072 PetscErrorCode DMGetUseNatural(DM dm, PetscBool *useNatural)
6073 {
6074   PetscFunctionBegin;
6075   PetscValidHeaderSpecific(dm, DM_CLASSID, 1);
6076   PetscValidPointer(useNatural, 2);
6077   *useNatural = dm->useNatural;
6078   PetscFunctionReturn(0);
6079 }
6080 
6081 /*@
6082   DMSetUseNatural - Set the flag for creating a mapping to the natural order after distribution
6083 
6084   Collective on dm
6085 
6086   Input Parameters:
6087 + dm - The DM
6088 - useNatural - The flag to build the mapping to a natural order during distribution
6089 
6090   Note: This also causes the map to be build after DMCreateSubDM() and DMCreateSuperDM()
6091 
6092   Level: beginner
6093 
6094 .seealso: DMGetUseNatural(), DMCreate(), DMPlexDistribute(), DMCreateSubDM(), DMCreateSuperDM()
6095 @*/
6096 PetscErrorCode DMSetUseNatural(DM dm, PetscBool useNatural)
6097 {
6098   PetscFunctionBegin;
6099   PetscValidHeaderSpecific(dm, DM_CLASSID, 1);
6100   PetscValidLogicalCollectiveBool(dm, useNatural, 2);
6101   dm->useNatural = useNatural;
6102   PetscFunctionReturn(0);
6103 }
6104 
6105 
6106 /*@C
6107   DMCreateLabel - Create a label of the given name if it does not already exist
6108 
6109   Not Collective
6110 
6111   Input Parameters:
6112 + dm   - The DM object
6113 - name - The label name
6114 
6115   Level: intermediate
6116 
6117 .keywords: mesh
6118 .seealso: DMLabelCreate(), DMHasLabel(), DMGetLabelValue(), DMSetLabelValue(), DMGetStratumIS()
6119 @*/
6120 PetscErrorCode DMCreateLabel(DM dm, const char name[])
6121 {
6122   DMLabelLink    next  = dm->labels->next;
6123   PetscBool      flg   = PETSC_FALSE;
6124   const char    *lname;
6125   PetscErrorCode ierr;
6126 
6127   PetscFunctionBegin;
6128   PetscValidHeaderSpecific(dm, DM_CLASSID, 1);
6129   PetscValidCharPointer(name, 2);
6130   while (next) {
6131     ierr = PetscObjectGetName((PetscObject) next->label, &lname);CHKERRQ(ierr);
6132     ierr = PetscStrcmp(name, lname, &flg);CHKERRQ(ierr);
6133     if (flg) break;
6134     next = next->next;
6135   }
6136   if (!flg) {
6137     DMLabelLink tmpLabel;
6138 
6139     ierr = PetscCalloc1(1, &tmpLabel);CHKERRQ(ierr);
6140     ierr = DMLabelCreate(PETSC_COMM_SELF, name, &tmpLabel->label);CHKERRQ(ierr);
6141     tmpLabel->output = PETSC_TRUE;
6142     tmpLabel->next   = dm->labels->next;
6143     dm->labels->next = tmpLabel;
6144   }
6145   PetscFunctionReturn(0);
6146 }
6147 
6148 /*@C
6149   DMGetLabelValue - Get the value in a Sieve Label for the given point, with 0 as the default
6150 
6151   Not Collective
6152 
6153   Input Parameters:
6154 + dm   - The DM object
6155 . name - The label name
6156 - point - The mesh point
6157 
6158   Output Parameter:
6159 . value - The label value for this point, or -1 if the point is not in the label
6160 
6161   Level: beginner
6162 
6163 .keywords: mesh
6164 .seealso: DMLabelGetValue(), DMSetLabelValue(), DMGetStratumIS()
6165 @*/
6166 PetscErrorCode DMGetLabelValue(DM dm, const char name[], PetscInt point, PetscInt *value)
6167 {
6168   DMLabel        label;
6169   PetscErrorCode ierr;
6170 
6171   PetscFunctionBegin;
6172   PetscValidHeaderSpecific(dm, DM_CLASSID, 1);
6173   PetscValidCharPointer(name, 2);
6174   ierr = DMGetLabel(dm, name, &label);CHKERRQ(ierr);
6175   if (!label) SETERRQ1(PETSC_COMM_SELF, PETSC_ERR_ARG_WRONG, "No label named %s was found", name);
6176   ierr = DMLabelGetValue(label, point, value);CHKERRQ(ierr);
6177   PetscFunctionReturn(0);
6178 }
6179 
6180 /*@C
6181   DMSetLabelValue - Add a point to a Sieve Label with given value
6182 
6183   Not Collective
6184 
6185   Input Parameters:
6186 + dm   - The DM object
6187 . name - The label name
6188 . point - The mesh point
6189 - value - The label value for this point
6190 
6191   Output Parameter:
6192 
6193   Level: beginner
6194 
6195 .keywords: mesh
6196 .seealso: DMLabelSetValue(), DMGetStratumIS(), DMClearLabelValue()
6197 @*/
6198 PetscErrorCode DMSetLabelValue(DM dm, const char name[], PetscInt point, PetscInt value)
6199 {
6200   DMLabel        label;
6201   PetscErrorCode ierr;
6202 
6203   PetscFunctionBegin;
6204   PetscValidHeaderSpecific(dm, DM_CLASSID, 1);
6205   PetscValidCharPointer(name, 2);
6206   ierr = DMGetLabel(dm, name, &label);CHKERRQ(ierr);
6207   if (!label) {
6208     ierr = DMCreateLabel(dm, name);CHKERRQ(ierr);
6209     ierr = DMGetLabel(dm, name, &label);CHKERRQ(ierr);
6210   }
6211   ierr = DMLabelSetValue(label, point, value);CHKERRQ(ierr);
6212   PetscFunctionReturn(0);
6213 }
6214 
6215 /*@C
6216   DMClearLabelValue - Remove a point from a Sieve Label with given value
6217 
6218   Not Collective
6219 
6220   Input Parameters:
6221 + dm   - The DM object
6222 . name - The label name
6223 . point - The mesh point
6224 - value - The label value for this point
6225 
6226   Output Parameter:
6227 
6228   Level: beginner
6229 
6230 .keywords: mesh
6231 .seealso: DMLabelClearValue(), DMSetLabelValue(), DMGetStratumIS()
6232 @*/
6233 PetscErrorCode DMClearLabelValue(DM dm, const char name[], PetscInt point, PetscInt value)
6234 {
6235   DMLabel        label;
6236   PetscErrorCode ierr;
6237 
6238   PetscFunctionBegin;
6239   PetscValidHeaderSpecific(dm, DM_CLASSID, 1);
6240   PetscValidCharPointer(name, 2);
6241   ierr = DMGetLabel(dm, name, &label);CHKERRQ(ierr);
6242   if (!label) PetscFunctionReturn(0);
6243   ierr = DMLabelClearValue(label, point, value);CHKERRQ(ierr);
6244   PetscFunctionReturn(0);
6245 }
6246 
6247 /*@C
6248   DMGetLabelSize - Get the number of different integer ids in a Label
6249 
6250   Not Collective
6251 
6252   Input Parameters:
6253 + dm   - The DM object
6254 - name - The label name
6255 
6256   Output Parameter:
6257 . size - The number of different integer ids, or 0 if the label does not exist
6258 
6259   Level: beginner
6260 
6261 .keywords: mesh
6262 .seealso: DMLabelGetNumValues(), DMSetLabelValue()
6263 @*/
6264 PetscErrorCode DMGetLabelSize(DM dm, const char name[], PetscInt *size)
6265 {
6266   DMLabel        label;
6267   PetscErrorCode ierr;
6268 
6269   PetscFunctionBegin;
6270   PetscValidHeaderSpecific(dm, DM_CLASSID, 1);
6271   PetscValidCharPointer(name, 2);
6272   PetscValidPointer(size, 3);
6273   ierr  = DMGetLabel(dm, name, &label);CHKERRQ(ierr);
6274   *size = 0;
6275   if (!label) PetscFunctionReturn(0);
6276   ierr = DMLabelGetNumValues(label, size);CHKERRQ(ierr);
6277   PetscFunctionReturn(0);
6278 }
6279 
6280 /*@C
6281   DMGetLabelIdIS - Get the integer ids in a label
6282 
6283   Not Collective
6284 
6285   Input Parameters:
6286 + mesh - The DM object
6287 - name - The label name
6288 
6289   Output Parameter:
6290 . ids - The integer ids, or NULL if the label does not exist
6291 
6292   Level: beginner
6293 
6294 .keywords: mesh
6295 .seealso: DMLabelGetValueIS(), DMGetLabelSize()
6296 @*/
6297 PetscErrorCode DMGetLabelIdIS(DM dm, const char name[], IS *ids)
6298 {
6299   DMLabel        label;
6300   PetscErrorCode ierr;
6301 
6302   PetscFunctionBegin;
6303   PetscValidHeaderSpecific(dm, DM_CLASSID, 1);
6304   PetscValidCharPointer(name, 2);
6305   PetscValidPointer(ids, 3);
6306   ierr = DMGetLabel(dm, name, &label);CHKERRQ(ierr);
6307   *ids = NULL;
6308  if (label) {
6309     ierr = DMLabelGetValueIS(label, ids);CHKERRQ(ierr);
6310   } else {
6311     /* returning an empty IS */
6312     ierr = ISCreateGeneral(PETSC_COMM_SELF,0,NULL,PETSC_USE_POINTER,ids);CHKERRQ(ierr);
6313   }
6314   PetscFunctionReturn(0);
6315 }
6316 
6317 /*@C
6318   DMGetStratumSize - Get the number of points in a label stratum
6319 
6320   Not Collective
6321 
6322   Input Parameters:
6323 + dm - The DM object
6324 . name - The label name
6325 - value - The stratum value
6326 
6327   Output Parameter:
6328 . size - The stratum size
6329 
6330   Level: beginner
6331 
6332 .keywords: mesh
6333 .seealso: DMLabelGetStratumSize(), DMGetLabelSize(), DMGetLabelIds()
6334 @*/
6335 PetscErrorCode DMGetStratumSize(DM dm, const char name[], PetscInt value, PetscInt *size)
6336 {
6337   DMLabel        label;
6338   PetscErrorCode ierr;
6339 
6340   PetscFunctionBegin;
6341   PetscValidHeaderSpecific(dm, DM_CLASSID, 1);
6342   PetscValidCharPointer(name, 2);
6343   PetscValidPointer(size, 4);
6344   ierr  = DMGetLabel(dm, name, &label);CHKERRQ(ierr);
6345   *size = 0;
6346   if (!label) PetscFunctionReturn(0);
6347   ierr = DMLabelGetStratumSize(label, value, size);CHKERRQ(ierr);
6348   PetscFunctionReturn(0);
6349 }
6350 
6351 /*@C
6352   DMGetStratumIS - Get the points in a label stratum
6353 
6354   Not Collective
6355 
6356   Input Parameters:
6357 + dm - The DM object
6358 . name - The label name
6359 - value - The stratum value
6360 
6361   Output Parameter:
6362 . points - The stratum points, or NULL if the label does not exist or does not have that value
6363 
6364   Level: beginner
6365 
6366 .keywords: mesh
6367 .seealso: DMLabelGetStratumIS(), DMGetStratumSize()
6368 @*/
6369 PetscErrorCode DMGetStratumIS(DM dm, const char name[], PetscInt value, IS *points)
6370 {
6371   DMLabel        label;
6372   PetscErrorCode ierr;
6373 
6374   PetscFunctionBegin;
6375   PetscValidHeaderSpecific(dm, DM_CLASSID, 1);
6376   PetscValidCharPointer(name, 2);
6377   PetscValidPointer(points, 4);
6378   ierr    = DMGetLabel(dm, name, &label);CHKERRQ(ierr);
6379   *points = NULL;
6380   if (!label) PetscFunctionReturn(0);
6381   ierr = DMLabelGetStratumIS(label, value, points);CHKERRQ(ierr);
6382   PetscFunctionReturn(0);
6383 }
6384 
6385 /*@C
6386   DMSetStratumIS - Set the points in a label stratum
6387 
6388   Not Collective
6389 
6390   Input Parameters:
6391 + dm - The DM object
6392 . name - The label name
6393 . value - The stratum value
6394 - points - The stratum points
6395 
6396   Level: beginner
6397 
6398 .keywords: mesh
6399 .seealso: DMLabelSetStratumIS(), DMGetStratumSize()
6400 @*/
6401 PetscErrorCode DMSetStratumIS(DM dm, const char name[], PetscInt value, IS points)
6402 {
6403   DMLabel        label;
6404   PetscErrorCode ierr;
6405 
6406   PetscFunctionBegin;
6407   PetscValidHeaderSpecific(dm, DM_CLASSID, 1);
6408   PetscValidCharPointer(name, 2);
6409   PetscValidPointer(points, 4);
6410   ierr = DMGetLabel(dm, name, &label);CHKERRQ(ierr);
6411   if (!label) PetscFunctionReturn(0);
6412   ierr = DMLabelSetStratumIS(label, value, points);CHKERRQ(ierr);
6413   PetscFunctionReturn(0);
6414 }
6415 
6416 /*@C
6417   DMClearLabelStratum - Remove all points from a stratum from a Sieve Label
6418 
6419   Not Collective
6420 
6421   Input Parameters:
6422 + dm   - The DM object
6423 . name - The label name
6424 - value - The label value for this point
6425 
6426   Output Parameter:
6427 
6428   Level: beginner
6429 
6430 .keywords: mesh
6431 .seealso: DMLabelClearStratum(), DMSetLabelValue(), DMGetStratumIS(), DMClearLabelValue()
6432 @*/
6433 PetscErrorCode DMClearLabelStratum(DM dm, const char name[], PetscInt value)
6434 {
6435   DMLabel        label;
6436   PetscErrorCode ierr;
6437 
6438   PetscFunctionBegin;
6439   PetscValidHeaderSpecific(dm, DM_CLASSID, 1);
6440   PetscValidCharPointer(name, 2);
6441   ierr = DMGetLabel(dm, name, &label);CHKERRQ(ierr);
6442   if (!label) PetscFunctionReturn(0);
6443   ierr = DMLabelClearStratum(label, value);CHKERRQ(ierr);
6444   PetscFunctionReturn(0);
6445 }
6446 
6447 /*@
6448   DMGetNumLabels - Return the number of labels defined by the mesh
6449 
6450   Not Collective
6451 
6452   Input Parameter:
6453 . dm   - The DM object
6454 
6455   Output Parameter:
6456 . numLabels - the number of Labels
6457 
6458   Level: intermediate
6459 
6460 .keywords: mesh
6461 .seealso: DMGetLabelValue(), DMSetLabelValue(), DMGetStratumIS()
6462 @*/
6463 PetscErrorCode DMGetNumLabels(DM dm, PetscInt *numLabels)
6464 {
6465   DMLabelLink next = dm->labels->next;
6466   PetscInt  n    = 0;
6467 
6468   PetscFunctionBegin;
6469   PetscValidHeaderSpecific(dm, DM_CLASSID, 1);
6470   PetscValidPointer(numLabels, 2);
6471   while (next) {++n; next = next->next;}
6472   *numLabels = n;
6473   PetscFunctionReturn(0);
6474 }
6475 
6476 /*@C
6477   DMGetLabelName - Return the name of nth label
6478 
6479   Not Collective
6480 
6481   Input Parameters:
6482 + dm - The DM object
6483 - n  - the label number
6484 
6485   Output Parameter:
6486 . name - the label name
6487 
6488   Level: intermediate
6489 
6490 .keywords: mesh
6491 .seealso: DMGetLabelValue(), DMSetLabelValue(), DMGetStratumIS()
6492 @*/
6493 PetscErrorCode DMGetLabelName(DM dm, PetscInt n, const char **name)
6494 {
6495   DMLabelLink    next = dm->labels->next;
6496   PetscInt       l    = 0;
6497   PetscErrorCode ierr;
6498 
6499   PetscFunctionBegin;
6500   PetscValidHeaderSpecific(dm, DM_CLASSID, 1);
6501   PetscValidPointer(name, 3);
6502   while (next) {
6503     if (l == n) {
6504       ierr = PetscObjectGetName((PetscObject) next->label, name);CHKERRQ(ierr);
6505       PetscFunctionReturn(0);
6506     }
6507     ++l;
6508     next = next->next;
6509   }
6510   SETERRQ1(PETSC_COMM_SELF, PETSC_ERR_ARG_OUTOFRANGE, "Label %D does not exist in this DM", n);
6511 }
6512 
6513 /*@C
6514   DMHasLabel - Determine whether the mesh has a label of a given name
6515 
6516   Not Collective
6517 
6518   Input Parameters:
6519 + dm   - The DM object
6520 - name - The label name
6521 
6522   Output Parameter:
6523 . hasLabel - PETSC_TRUE if the label is present
6524 
6525   Level: intermediate
6526 
6527 .keywords: mesh
6528 .seealso: DMCreateLabel(), DMGetLabelValue(), DMSetLabelValue(), DMGetStratumIS()
6529 @*/
6530 PetscErrorCode DMHasLabel(DM dm, const char name[], PetscBool *hasLabel)
6531 {
6532   DMLabelLink    next = dm->labels->next;
6533   const char    *lname;
6534   PetscErrorCode ierr;
6535 
6536   PetscFunctionBegin;
6537   PetscValidHeaderSpecific(dm, DM_CLASSID, 1);
6538   PetscValidCharPointer(name, 2);
6539   PetscValidPointer(hasLabel, 3);
6540   *hasLabel = PETSC_FALSE;
6541   while (next) {
6542     ierr = PetscObjectGetName((PetscObject) next->label, &lname);CHKERRQ(ierr);
6543     ierr = PetscStrcmp(name, lname, hasLabel);CHKERRQ(ierr);
6544     if (*hasLabel) break;
6545     next = next->next;
6546   }
6547   PetscFunctionReturn(0);
6548 }
6549 
6550 /*@C
6551   DMGetLabel - Return the label of a given name, or NULL
6552 
6553   Not Collective
6554 
6555   Input Parameters:
6556 + dm   - The DM object
6557 - name - The label name
6558 
6559   Output Parameter:
6560 . label - The DMLabel, or NULL if the label is absent
6561 
6562   Level: intermediate
6563 
6564 .keywords: mesh
6565 .seealso: DMCreateLabel(), DMHasLabel(), DMGetLabelValue(), DMSetLabelValue(), DMGetStratumIS()
6566 @*/
6567 PetscErrorCode DMGetLabel(DM dm, const char name[], DMLabel *label)
6568 {
6569   DMLabelLink    next = dm->labels->next;
6570   PetscBool      hasLabel;
6571   const char    *lname;
6572   PetscErrorCode ierr;
6573 
6574   PetscFunctionBegin;
6575   PetscValidHeaderSpecific(dm, DM_CLASSID, 1);
6576   PetscValidCharPointer(name, 2);
6577   PetscValidPointer(label, 3);
6578   *label = NULL;
6579   while (next) {
6580     ierr = PetscObjectGetName((PetscObject) next->label, &lname);CHKERRQ(ierr);
6581     ierr = PetscStrcmp(name, lname, &hasLabel);CHKERRQ(ierr);
6582     if (hasLabel) {
6583       *label = next->label;
6584       break;
6585     }
6586     next = next->next;
6587   }
6588   PetscFunctionReturn(0);
6589 }
6590 
6591 /*@C
6592   DMGetLabelByNum - Return the nth label
6593 
6594   Not Collective
6595 
6596   Input Parameters:
6597 + dm - The DM object
6598 - n  - the label number
6599 
6600   Output Parameter:
6601 . label - the label
6602 
6603   Level: intermediate
6604 
6605 .keywords: mesh
6606 .seealso: DMGetLabelValue(), DMSetLabelValue(), DMGetStratumIS()
6607 @*/
6608 PetscErrorCode DMGetLabelByNum(DM dm, PetscInt n, DMLabel *label)
6609 {
6610   DMLabelLink next = dm->labels->next;
6611   PetscInt    l    = 0;
6612 
6613   PetscFunctionBegin;
6614   PetscValidHeaderSpecific(dm, DM_CLASSID, 1);
6615   PetscValidPointer(label, 3);
6616   while (next) {
6617     if (l == n) {
6618       *label = next->label;
6619       PetscFunctionReturn(0);
6620     }
6621     ++l;
6622     next = next->next;
6623   }
6624   SETERRQ1(PETSC_COMM_SELF, PETSC_ERR_ARG_OUTOFRANGE, "Label %D does not exist in this DM", n);
6625 }
6626 
6627 /*@C
6628   DMAddLabel - Add the label to this mesh
6629 
6630   Not Collective
6631 
6632   Input Parameters:
6633 + dm   - The DM object
6634 - label - The DMLabel
6635 
6636   Level: developer
6637 
6638 .keywords: mesh
6639 .seealso: DMCreateLabel(), DMHasLabel(), DMGetLabelValue(), DMSetLabelValue(), DMGetStratumIS()
6640 @*/
6641 PetscErrorCode DMAddLabel(DM dm, DMLabel label)
6642 {
6643   DMLabelLink    tmpLabel;
6644   PetscBool      hasLabel;
6645   const char    *lname;
6646   PetscErrorCode ierr;
6647 
6648   PetscFunctionBegin;
6649   PetscValidHeaderSpecific(dm, DM_CLASSID, 1);
6650   ierr = PetscObjectGetName((PetscObject) label, &lname);CHKERRQ(ierr);
6651   ierr = DMHasLabel(dm, lname, &hasLabel);CHKERRQ(ierr);
6652   if (hasLabel) SETERRQ1(PETSC_COMM_SELF, PETSC_ERR_ARG_OUTOFRANGE, "Label %s already exists in this DM", lname);
6653   ierr = PetscCalloc1(1, &tmpLabel);CHKERRQ(ierr);
6654   tmpLabel->label  = label;
6655   tmpLabel->output = PETSC_TRUE;
6656   tmpLabel->next   = dm->labels->next;
6657   dm->labels->next = tmpLabel;
6658   PetscFunctionReturn(0);
6659 }
6660 
6661 /*@C
6662   DMRemoveLabel - Remove the label from this mesh
6663 
6664   Not Collective
6665 
6666   Input Parameters:
6667 + dm   - The DM object
6668 - name - The label name
6669 
6670   Output Parameter:
6671 . label - The DMLabel, or NULL if the label is absent
6672 
6673   Level: developer
6674 
6675 .keywords: mesh
6676 .seealso: DMCreateLabel(), DMHasLabel(), DMGetLabelValue(), DMSetLabelValue(), DMGetStratumIS()
6677 @*/
6678 PetscErrorCode DMRemoveLabel(DM dm, const char name[], DMLabel *label)
6679 {
6680   DMLabelLink    next = dm->labels->next;
6681   DMLabelLink    last = NULL;
6682   PetscBool      hasLabel;
6683   const char    *lname;
6684   PetscErrorCode ierr;
6685 
6686   PetscFunctionBegin;
6687   PetscValidHeaderSpecific(dm, DM_CLASSID, 1);
6688   ierr   = DMHasLabel(dm, name, &hasLabel);CHKERRQ(ierr);
6689   *label = NULL;
6690   if (!hasLabel) PetscFunctionReturn(0);
6691   while (next) {
6692     ierr = PetscObjectGetName((PetscObject) next->label, &lname);CHKERRQ(ierr);
6693     ierr = PetscStrcmp(name, lname, &hasLabel);CHKERRQ(ierr);
6694     if (hasLabel) {
6695       if (last) last->next       = next->next;
6696       else      dm->labels->next = next->next;
6697       next->next = NULL;
6698       *label     = next->label;
6699       ierr = PetscStrcmp(name, "depth", &hasLabel);CHKERRQ(ierr);
6700       if (hasLabel) {
6701         dm->depthLabel = NULL;
6702       }
6703       ierr = PetscFree(next);CHKERRQ(ierr);
6704       break;
6705     }
6706     last = next;
6707     next = next->next;
6708   }
6709   PetscFunctionReturn(0);
6710 }
6711 
6712 /*@C
6713   DMGetLabelOutput - Get the output flag for a given label
6714 
6715   Not Collective
6716 
6717   Input Parameters:
6718 + dm   - The DM object
6719 - name - The label name
6720 
6721   Output Parameter:
6722 . output - The flag for output
6723 
6724   Level: developer
6725 
6726 .keywords: mesh
6727 .seealso: DMSetLabelOutput(), DMCreateLabel(), DMHasLabel(), DMGetLabelValue(), DMSetLabelValue(), DMGetStratumIS()
6728 @*/
6729 PetscErrorCode DMGetLabelOutput(DM dm, const char name[], PetscBool *output)
6730 {
6731   DMLabelLink    next = dm->labels->next;
6732   const char    *lname;
6733   PetscErrorCode ierr;
6734 
6735   PetscFunctionBegin;
6736   PetscValidHeaderSpecific(dm, DM_CLASSID, 1);
6737   PetscValidPointer(name, 2);
6738   PetscValidPointer(output, 3);
6739   while (next) {
6740     PetscBool flg;
6741 
6742     ierr = PetscObjectGetName((PetscObject) next->label, &lname);CHKERRQ(ierr);
6743     ierr = PetscStrcmp(name, lname, &flg);CHKERRQ(ierr);
6744     if (flg) {*output = next->output; PetscFunctionReturn(0);}
6745     next = next->next;
6746   }
6747   SETERRQ1(PETSC_COMM_SELF, PETSC_ERR_ARG_OUTOFRANGE, "No label named %s was present in this dm", name);
6748 }
6749 
6750 /*@C
6751   DMSetLabelOutput - Set the output flag for a given label
6752 
6753   Not Collective
6754 
6755   Input Parameters:
6756 + dm     - The DM object
6757 . name   - The label name
6758 - output - The flag for output
6759 
6760   Level: developer
6761 
6762 .keywords: mesh
6763 .seealso: DMGetLabelOutput(), DMCreateLabel(), DMHasLabel(), DMGetLabelValue(), DMSetLabelValue(), DMGetStratumIS()
6764 @*/
6765 PetscErrorCode DMSetLabelOutput(DM dm, const char name[], PetscBool output)
6766 {
6767   DMLabelLink    next = dm->labels->next;
6768   const char    *lname;
6769   PetscErrorCode ierr;
6770 
6771   PetscFunctionBegin;
6772   PetscValidHeaderSpecific(dm, DM_CLASSID, 1);
6773   PetscValidPointer(name, 2);
6774   while (next) {
6775     PetscBool flg;
6776 
6777     ierr = PetscObjectGetName((PetscObject) next->label, &lname);CHKERRQ(ierr);
6778     ierr = PetscStrcmp(name, lname, &flg);CHKERRQ(ierr);
6779     if (flg) {next->output = output; PetscFunctionReturn(0);}
6780     next = next->next;
6781   }
6782   SETERRQ1(PETSC_COMM_SELF, PETSC_ERR_ARG_OUTOFRANGE, "No label named %s was present in this dm", name);
6783 }
6784 
6785 
6786 /*@
6787   DMCopyLabels - Copy labels from one mesh to another with a superset of the points
6788 
6789   Collective on DM
6790 
6791   Input Parameter:
6792 . dmA - The DM object with initial labels
6793 
6794   Output Parameter:
6795 . dmB - The DM object with copied labels
6796 
6797   Level: intermediate
6798 
6799   Note: This is typically used when interpolating or otherwise adding to a mesh
6800 
6801 .keywords: mesh
6802 .seealso: DMGetCoordinates(), DMGetCoordinatesLocal(), DMGetCoordinateDM(), DMGetCoordinateSection()
6803 @*/
6804 PetscErrorCode DMCopyLabels(DM dmA, DM dmB)
6805 {
6806   PetscInt       numLabels, l;
6807   PetscErrorCode ierr;
6808 
6809   PetscFunctionBegin;
6810   if (dmA == dmB) PetscFunctionReturn(0);
6811   ierr = DMGetNumLabels(dmA, &numLabels);CHKERRQ(ierr);
6812   for (l = 0; l < numLabels; ++l) {
6813     DMLabel     label, labelNew;
6814     const char *name;
6815     PetscBool   flg;
6816 
6817     ierr = DMGetLabelName(dmA, l, &name);CHKERRQ(ierr);
6818     ierr = PetscStrcmp(name, "depth", &flg);CHKERRQ(ierr);
6819     if (flg) continue;
6820     ierr = PetscStrcmp(name, "dim", &flg);CHKERRQ(ierr);
6821     if (flg) continue;
6822     ierr = DMGetLabel(dmA, name, &label);CHKERRQ(ierr);
6823     ierr = DMLabelDuplicate(label, &labelNew);CHKERRQ(ierr);
6824     ierr = DMAddLabel(dmB, labelNew);CHKERRQ(ierr);
6825   }
6826   PetscFunctionReturn(0);
6827 }
6828 
6829 /*@
6830   DMGetCoarseDM - Get the coarse mesh from which this was obtained by refinement
6831 
6832   Input Parameter:
6833 . dm - The DM object
6834 
6835   Output Parameter:
6836 . cdm - The coarse DM
6837 
6838   Level: intermediate
6839 
6840 .seealso: DMSetCoarseDM()
6841 @*/
6842 PetscErrorCode DMGetCoarseDM(DM dm, DM *cdm)
6843 {
6844   PetscFunctionBegin;
6845   PetscValidHeaderSpecific(dm, DM_CLASSID, 1);
6846   PetscValidPointer(cdm, 2);
6847   *cdm = dm->coarseMesh;
6848   PetscFunctionReturn(0);
6849 }
6850 
6851 /*@
6852   DMSetCoarseDM - Set the coarse mesh from which this was obtained by refinement
6853 
6854   Input Parameters:
6855 + dm - The DM object
6856 - cdm - The coarse DM
6857 
6858   Level: intermediate
6859 
6860 .seealso: DMGetCoarseDM()
6861 @*/
6862 PetscErrorCode DMSetCoarseDM(DM dm, DM cdm)
6863 {
6864   PetscErrorCode ierr;
6865 
6866   PetscFunctionBegin;
6867   PetscValidHeaderSpecific(dm, DM_CLASSID, 1);
6868   if (cdm) PetscValidHeaderSpecific(cdm, DM_CLASSID, 2);
6869   ierr = PetscObjectReference((PetscObject)cdm);CHKERRQ(ierr);
6870   ierr = DMDestroy(&dm->coarseMesh);CHKERRQ(ierr);
6871   dm->coarseMesh = cdm;
6872   PetscFunctionReturn(0);
6873 }
6874 
6875 /*@
6876   DMGetFineDM - Get the fine mesh from which this was obtained by refinement
6877 
6878   Input Parameter:
6879 . dm - The DM object
6880 
6881   Output Parameter:
6882 . fdm - The fine DM
6883 
6884   Level: intermediate
6885 
6886 .seealso: DMSetFineDM()
6887 @*/
6888 PetscErrorCode DMGetFineDM(DM dm, DM *fdm)
6889 {
6890   PetscFunctionBegin;
6891   PetscValidHeaderSpecific(dm, DM_CLASSID, 1);
6892   PetscValidPointer(fdm, 2);
6893   *fdm = dm->fineMesh;
6894   PetscFunctionReturn(0);
6895 }
6896 
6897 /*@
6898   DMSetFineDM - Set the fine mesh from which this was obtained by refinement
6899 
6900   Input Parameters:
6901 + dm - The DM object
6902 - fdm - The fine DM
6903 
6904   Level: intermediate
6905 
6906 .seealso: DMGetFineDM()
6907 @*/
6908 PetscErrorCode DMSetFineDM(DM dm, DM fdm)
6909 {
6910   PetscErrorCode ierr;
6911 
6912   PetscFunctionBegin;
6913   PetscValidHeaderSpecific(dm, DM_CLASSID, 1);
6914   if (fdm) PetscValidHeaderSpecific(fdm, DM_CLASSID, 2);
6915   ierr = PetscObjectReference((PetscObject)fdm);CHKERRQ(ierr);
6916   ierr = DMDestroy(&dm->fineMesh);CHKERRQ(ierr);
6917   dm->fineMesh = fdm;
6918   PetscFunctionReturn(0);
6919 }
6920 
6921 /*=== DMBoundary code ===*/
6922 
6923 PetscErrorCode DMCopyBoundary(DM dm, DM dmNew)
6924 {
6925   PetscInt       d;
6926   PetscErrorCode ierr;
6927 
6928   PetscFunctionBegin;
6929   for (d = 0; d < dm->Nds; ++d) {
6930     ierr = PetscDSCopyBoundary(dm->probs[d].ds, dmNew->probs[d].ds);CHKERRQ(ierr);
6931   }
6932   PetscFunctionReturn(0);
6933 }
6934 
6935 /*@C
6936   DMAddBoundary - Add a boundary condition to the model
6937 
6938   Input Parameters:
6939 + dm          - The DM, with a PetscDS that matches the problem being constrained
6940 . type        - The type of condition, e.g. DM_BC_ESSENTIAL_ANALYTIC/DM_BC_ESSENTIAL_FIELD (Dirichlet), or DM_BC_NATURAL (Neumann)
6941 . name        - The BC name
6942 . labelname   - The label defining constrained points
6943 . field       - The field to constrain
6944 . numcomps    - The number of constrained field components (0 will constrain all fields)
6945 . comps       - An array of constrained component numbers
6946 . bcFunc      - A pointwise function giving boundary values
6947 . numids      - The number of DMLabel ids for constrained points
6948 . ids         - An array of ids for constrained points
6949 - ctx         - An optional user context for bcFunc
6950 
6951   Options Database Keys:
6952 + -bc_<boundary name> <num> - Overrides the boundary ids
6953 - -bc_<boundary name>_comp <num> - Overrides the boundary components
6954 
6955   Level: developer
6956 
6957 .seealso: DMGetBoundary()
6958 @*/
6959 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)
6960 {
6961   PetscDS        ds;
6962   PetscErrorCode ierr;
6963 
6964   PetscFunctionBegin;
6965   PetscValidHeaderSpecific(dm, DM_CLASSID, 1);
6966   ierr = DMGetDS(dm, &ds);CHKERRQ(ierr);
6967   ierr = PetscDSAddBoundary(ds, type,name, labelname, field, numcomps, comps, bcFunc, numids, ids, ctx);CHKERRQ(ierr);
6968   PetscFunctionReturn(0);
6969 }
6970 
6971 /*@
6972   DMGetNumBoundary - Get the number of registered BC
6973 
6974   Input Parameters:
6975 . dm - The mesh object
6976 
6977   Output Parameters:
6978 . numBd - The number of BC
6979 
6980   Level: intermediate
6981 
6982 .seealso: DMAddBoundary(), DMGetBoundary()
6983 @*/
6984 PetscErrorCode DMGetNumBoundary(DM dm, PetscInt *numBd)
6985 {
6986   PetscDS        ds;
6987   PetscErrorCode ierr;
6988 
6989   PetscFunctionBegin;
6990   PetscValidHeaderSpecific(dm, DM_CLASSID, 1);
6991   ierr = DMGetDS(dm, &ds);CHKERRQ(ierr);
6992   ierr = PetscDSGetNumBoundary(ds, numBd);CHKERRQ(ierr);
6993   PetscFunctionReturn(0);
6994 }
6995 
6996 /*@C
6997   DMGetBoundary - Get a model boundary condition
6998 
6999   Input Parameters:
7000 + dm          - The mesh object
7001 - bd          - The BC number
7002 
7003   Output Parameters:
7004 + type        - The type of condition, e.g. DM_BC_ESSENTIAL_ANALYTIC/DM_BC_ESSENTIAL_FIELD (Dirichlet), or DM_BC_NATURAL (Neumann)
7005 . name        - The BC name
7006 . labelname   - The label defining constrained points
7007 . field       - The field to constrain
7008 . numcomps    - The number of constrained field components
7009 . comps       - An array of constrained component numbers
7010 . bcFunc      - A pointwise function giving boundary values
7011 . numids      - The number of DMLabel ids for constrained points
7012 . ids         - An array of ids for constrained points
7013 - ctx         - An optional user context for bcFunc
7014 
7015   Options Database Keys:
7016 + -bc_<boundary name> <num> - Overrides the boundary ids
7017 - -bc_<boundary name>_comp <num> - Overrides the boundary components
7018 
7019   Level: developer
7020 
7021 .seealso: DMAddBoundary()
7022 @*/
7023 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)
7024 {
7025   PetscDS        ds;
7026   PetscErrorCode ierr;
7027 
7028   PetscFunctionBegin;
7029   PetscValidHeaderSpecific(dm, DM_CLASSID, 1);
7030   ierr = DMGetDS(dm, &ds);CHKERRQ(ierr);
7031   ierr = PetscDSGetBoundary(ds, bd, type, name, labelname, field, numcomps, comps, func, numids, ids, ctx);CHKERRQ(ierr);
7032   PetscFunctionReturn(0);
7033 }
7034 
7035 static PetscErrorCode DMPopulateBoundary(DM dm)
7036 {
7037   PetscDS        ds;
7038   DMBoundary    *lastnext;
7039   DSBoundary     dsbound;
7040   PetscErrorCode ierr;
7041 
7042   PetscFunctionBegin;
7043   ierr = DMGetDS(dm, &ds);CHKERRQ(ierr);
7044   dsbound = ds->boundary;
7045   if (dm->boundary) {
7046     DMBoundary next = dm->boundary;
7047 
7048     /* quick check to see if the PetscDS has changed */
7049     if (next->dsboundary == dsbound) PetscFunctionReturn(0);
7050     /* the PetscDS has changed: tear down and rebuild */
7051     while (next) {
7052       DMBoundary b = next;
7053 
7054       next = b->next;
7055       ierr = PetscFree(b);CHKERRQ(ierr);
7056     }
7057     dm->boundary = NULL;
7058   }
7059 
7060   lastnext = &(dm->boundary);
7061   while (dsbound) {
7062     DMBoundary dmbound;
7063 
7064     ierr = PetscNew(&dmbound);CHKERRQ(ierr);
7065     dmbound->dsboundary = dsbound;
7066     ierr = DMGetLabel(dm, dsbound->labelname, &(dmbound->label));CHKERRQ(ierr);
7067     if (!dmbound->label) {ierr = PetscInfo2(dm, "DSBoundary %s wants label %s, which is not in this dm.\n",dsbound->name,dsbound->labelname);CHKERRQ(ierr);}
7068     /* push on the back instead of the front so that it is in the same order as in the PetscDS */
7069     *lastnext = dmbound;
7070     lastnext = &(dmbound->next);
7071     dsbound = dsbound->next;
7072   }
7073   PetscFunctionReturn(0);
7074 }
7075 
7076 PetscErrorCode DMIsBoundaryPoint(DM dm, PetscInt point, PetscBool *isBd)
7077 {
7078   DMBoundary     b;
7079   PetscErrorCode ierr;
7080 
7081   PetscFunctionBegin;
7082   PetscValidHeaderSpecific(dm, DM_CLASSID, 1);
7083   PetscValidPointer(isBd, 3);
7084   *isBd = PETSC_FALSE;
7085   ierr = DMPopulateBoundary(dm);CHKERRQ(ierr);
7086   b = dm->boundary;
7087   while (b && !(*isBd)) {
7088     DMLabel    label = b->label;
7089     DSBoundary dsb = b->dsboundary;
7090 
7091     if (label) {
7092       PetscInt i;
7093 
7094       for (i = 0; i < dsb->numids && !(*isBd); ++i) {
7095         ierr = DMLabelStratumHasPoint(label, dsb->ids[i], point, isBd);CHKERRQ(ierr);
7096       }
7097     }
7098     b = b->next;
7099   }
7100   PetscFunctionReturn(0);
7101 }
7102 
7103 /*@C
7104   DMProjectFunction - This projects the given function into the function space provided.
7105 
7106   Input Parameters:
7107 + dm      - The DM
7108 . time    - The time
7109 . funcs   - The coordinate functions to evaluate, one per field
7110 . ctxs    - Optional array of contexts to pass to each coordinate function.  ctxs itself may be null.
7111 - mode    - The insertion mode for values
7112 
7113   Output Parameter:
7114 . X - vector
7115 
7116    Calling sequence of func:
7117 $    func(PetscInt dim, PetscReal time, const PetscReal x[], PetscInt Nf, PetscScalar u[], void *ctx);
7118 
7119 +  dim - The spatial dimension
7120 .  x   - The coordinates
7121 .  Nf  - The number of fields
7122 .  u   - The output field values
7123 -  ctx - optional user-defined function context
7124 
7125   Level: developer
7126 
7127 .seealso: DMComputeL2Diff()
7128 @*/
7129 PetscErrorCode DMProjectFunction(DM dm, PetscReal time, PetscErrorCode (**funcs)(PetscInt, PetscReal, const PetscReal [], PetscInt, PetscScalar *, void *), void **ctxs, InsertMode mode, Vec X)
7130 {
7131   Vec            localX;
7132   PetscErrorCode ierr;
7133 
7134   PetscFunctionBegin;
7135   PetscValidHeaderSpecific(dm, DM_CLASSID, 1);
7136   ierr = DMGetLocalVector(dm, &localX);CHKERRQ(ierr);
7137   ierr = DMProjectFunctionLocal(dm, time, funcs, ctxs, mode, localX);CHKERRQ(ierr);
7138   ierr = DMLocalToGlobalBegin(dm, localX, mode, X);CHKERRQ(ierr);
7139   ierr = DMLocalToGlobalEnd(dm, localX, mode, X);CHKERRQ(ierr);
7140   ierr = DMRestoreLocalVector(dm, &localX);CHKERRQ(ierr);
7141   PetscFunctionReturn(0);
7142 }
7143 
7144 PetscErrorCode DMProjectFunctionLocal(DM dm, PetscReal time, PetscErrorCode (**funcs)(PetscInt, PetscReal, const PetscReal [], PetscInt, PetscScalar *, void *), void **ctxs, InsertMode mode, Vec localX)
7145 {
7146   PetscErrorCode ierr;
7147 
7148   PetscFunctionBegin;
7149   PetscValidHeaderSpecific(dm,DM_CLASSID,1);
7150   PetscValidHeaderSpecific(localX,VEC_CLASSID,5);
7151   if (!dm->ops->projectfunctionlocal) SETERRQ1(PetscObjectComm((PetscObject)dm),PETSC_ERR_SUP,"DM type %s does not implement DMProjectFunctionLocal",((PetscObject)dm)->type_name);
7152   ierr = (dm->ops->projectfunctionlocal) (dm, time, funcs, ctxs, mode, localX);CHKERRQ(ierr);
7153   PetscFunctionReturn(0);
7154 }
7155 
7156 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)
7157 {
7158   Vec            localX;
7159   PetscErrorCode ierr;
7160 
7161   PetscFunctionBegin;
7162   PetscValidHeaderSpecific(dm, DM_CLASSID, 1);
7163   ierr = DMGetLocalVector(dm, &localX);CHKERRQ(ierr);
7164   ierr = DMProjectFunctionLabelLocal(dm, time, label, numIds, ids, Nc, comps, funcs, ctxs, mode, localX);CHKERRQ(ierr);
7165   ierr = DMLocalToGlobalBegin(dm, localX, mode, X);CHKERRQ(ierr);
7166   ierr = DMLocalToGlobalEnd(dm, localX, mode, X);CHKERRQ(ierr);
7167   ierr = DMRestoreLocalVector(dm, &localX);CHKERRQ(ierr);
7168   PetscFunctionReturn(0);
7169 }
7170 
7171 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)
7172 {
7173   PetscErrorCode ierr;
7174 
7175   PetscFunctionBegin;
7176   PetscValidHeaderSpecific(dm,DM_CLASSID,1);
7177   PetscValidHeaderSpecific(localX,VEC_CLASSID,5);
7178   if (!dm->ops->projectfunctionlabellocal) SETERRQ1(PetscObjectComm((PetscObject)dm),PETSC_ERR_SUP,"DM type %s does not implement DMProjectFunctionLabelLocal",((PetscObject)dm)->type_name);
7179   ierr = (dm->ops->projectfunctionlabellocal) (dm, time, label, numIds, ids, Nc, comps, funcs, ctxs, mode, localX);CHKERRQ(ierr);
7180   PetscFunctionReturn(0);
7181 }
7182 
7183 PetscErrorCode DMProjectFieldLocal(DM dm, PetscReal time, Vec localU,
7184                                    void (**funcs)(PetscInt, PetscInt, PetscInt,
7185                                                   const PetscInt[], const PetscInt[], const PetscScalar[], const PetscScalar[], const PetscScalar[],
7186                                                   const PetscInt[], const PetscInt[], const PetscScalar[], const PetscScalar[], const PetscScalar[],
7187                                                   PetscReal, const PetscReal[], PetscInt, const PetscScalar[], PetscScalar[]),
7188                                    InsertMode mode, Vec localX)
7189 {
7190   PetscErrorCode ierr;
7191 
7192   PetscFunctionBegin;
7193   PetscValidHeaderSpecific(dm,DM_CLASSID,1);
7194   PetscValidHeaderSpecific(localU,VEC_CLASSID,3);
7195   PetscValidHeaderSpecific(localX,VEC_CLASSID,6);
7196   if (!dm->ops->projectfieldlocal) SETERRQ1(PetscObjectComm((PetscObject)dm),PETSC_ERR_SUP,"DM type %s does not implement DMProjectFieldLocal",((PetscObject)dm)->type_name);
7197   ierr = (dm->ops->projectfieldlocal) (dm, time, localU, funcs, mode, localX);CHKERRQ(ierr);
7198   PetscFunctionReturn(0);
7199 }
7200 
7201 PetscErrorCode DMProjectFieldLabelLocal(DM dm, PetscReal time, DMLabel label, PetscInt numIds, const PetscInt ids[], PetscInt Nc, const PetscInt comps[], Vec localU,
7202                                         void (**funcs)(PetscInt, PetscInt, PetscInt,
7203                                                        const PetscInt[], const PetscInt[], const PetscScalar[], const PetscScalar[], const PetscScalar[],
7204                                                        const PetscInt[], const PetscInt[], const PetscScalar[], const PetscScalar[], const PetscScalar[],
7205                                                        PetscReal, const PetscReal[], PetscInt, const PetscScalar[], PetscScalar[]),
7206                                         InsertMode mode, Vec localX)
7207 {
7208   PetscErrorCode ierr;
7209 
7210   PetscFunctionBegin;
7211   PetscValidHeaderSpecific(dm,DM_CLASSID,1);
7212   PetscValidHeaderSpecific(localU,VEC_CLASSID,6);
7213   PetscValidHeaderSpecific(localX,VEC_CLASSID,9);
7214   if (!dm->ops->projectfieldlabellocal) SETERRQ1(PetscObjectComm((PetscObject)dm),PETSC_ERR_SUP,"DM type %s does not implement DMProjectFieldLocal",((PetscObject)dm)->type_name);
7215   ierr = (dm->ops->projectfieldlabellocal)(dm, time, label, numIds, ids, Nc, comps, localU, funcs, mode, localX);CHKERRQ(ierr);
7216   PetscFunctionReturn(0);
7217 }
7218 
7219 /*@C
7220   DMComputeL2Diff - This function computes the L_2 difference between a function u and an FEM interpolant solution u_h.
7221 
7222   Input Parameters:
7223 + dm    - The DM
7224 . time  - The time
7225 . funcs - The functions to evaluate for each field component
7226 . ctxs  - Optional array of contexts to pass to each function, or NULL.
7227 - X     - The coefficient vector u_h, a global vector
7228 
7229   Output Parameter:
7230 . diff - The diff ||u - u_h||_2
7231 
7232   Level: developer
7233 
7234 .seealso: DMProjectFunction(), DMComputeL2FieldDiff(), DMComputeL2GradientDiff()
7235 @*/
7236 PetscErrorCode DMComputeL2Diff(DM dm, PetscReal time, PetscErrorCode (**funcs)(PetscInt, PetscReal, const PetscReal [], PetscInt, PetscScalar *, void *), void **ctxs, Vec X, PetscReal *diff)
7237 {
7238   PetscErrorCode ierr;
7239 
7240   PetscFunctionBegin;
7241   PetscValidHeaderSpecific(dm,DM_CLASSID,1);
7242   PetscValidHeaderSpecific(X,VEC_CLASSID,5);
7243   if (!dm->ops->computel2diff) SETERRQ1(PetscObjectComm((PetscObject)dm),PETSC_ERR_SUP,"DM type %s does not implement DMComputeL2Diff",((PetscObject)dm)->type_name);
7244   ierr = (dm->ops->computel2diff)(dm,time,funcs,ctxs,X,diff);CHKERRQ(ierr);
7245   PetscFunctionReturn(0);
7246 }
7247 
7248 /*@C
7249   DMComputeL2GradientDiff - This function computes the L_2 difference between the gradient of a function u and an FEM interpolant solution grad u_h.
7250 
7251   Input Parameters:
7252 + dm    - The DM
7253 , time  - The time
7254 . funcs - The gradient functions to evaluate for each field component
7255 . ctxs  - Optional array of contexts to pass to each function, or NULL.
7256 . X     - The coefficient vector u_h, a global vector
7257 - n     - The vector to project along
7258 
7259   Output Parameter:
7260 . diff - The diff ||(grad u - grad u_h) . n||_2
7261 
7262   Level: developer
7263 
7264 .seealso: DMProjectFunction(), DMComputeL2Diff()
7265 @*/
7266 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)
7267 {
7268   PetscErrorCode ierr;
7269 
7270   PetscFunctionBegin;
7271   PetscValidHeaderSpecific(dm,DM_CLASSID,1);
7272   PetscValidHeaderSpecific(X,VEC_CLASSID,5);
7273   if (!dm->ops->computel2gradientdiff) SETERRQ1(PetscObjectComm((PetscObject)dm),PETSC_ERR_SUP,"DM type %s does not implement DMComputeL2GradientDiff",((PetscObject)dm)->type_name);
7274   ierr = (dm->ops->computel2gradientdiff)(dm,time,funcs,ctxs,X,n,diff);CHKERRQ(ierr);
7275   PetscFunctionReturn(0);
7276 }
7277 
7278 /*@C
7279   DMComputeL2FieldDiff - This function computes the L_2 difference between a function u and an FEM interpolant solution u_h, separated into field components.
7280 
7281   Input Parameters:
7282 + dm    - The DM
7283 . time  - The time
7284 . funcs - The functions to evaluate for each field component
7285 . ctxs  - Optional array of contexts to pass to each function, or NULL.
7286 - X     - The coefficient vector u_h, a global vector
7287 
7288   Output Parameter:
7289 . diff - The array of differences, ||u^f - u^f_h||_2
7290 
7291   Level: developer
7292 
7293 .seealso: DMProjectFunction(), DMComputeL2FieldDiff(), DMComputeL2GradientDiff()
7294 @*/
7295 PetscErrorCode DMComputeL2FieldDiff(DM dm, PetscReal time, PetscErrorCode (**funcs)(PetscInt, PetscReal, const PetscReal [], PetscInt, PetscScalar *, void *), void **ctxs, Vec X, PetscReal diff[])
7296 {
7297   PetscErrorCode ierr;
7298 
7299   PetscFunctionBegin;
7300   PetscValidHeaderSpecific(dm,DM_CLASSID,1);
7301   PetscValidHeaderSpecific(X,VEC_CLASSID,5);
7302   if (!dm->ops->computel2fielddiff) SETERRQ1(PetscObjectComm((PetscObject)dm),PETSC_ERR_SUP,"DM type %s does not implement DMComputeL2FieldDiff",((PetscObject)dm)->type_name);
7303   ierr = (dm->ops->computel2fielddiff)(dm,time,funcs,ctxs,X,diff);CHKERRQ(ierr);
7304   PetscFunctionReturn(0);
7305 }
7306 
7307 /*@C
7308   DMAdaptLabel - Adapt a dm based on a label with values interpreted as coarsening and refining flags.  Specific implementations of DM maybe have
7309                  specialized flags, but all implementations should accept flag values DM_ADAPT_DETERMINE, DM_ADAPT_KEEP, DM_ADAPT_REFINE, and DM_ADAPT_COARSEN.
7310 
7311   Collective on dm
7312 
7313   Input parameters:
7314 + dm - the pre-adaptation DM object
7315 - label - label with the flags
7316 
7317   Output parameters:
7318 . dmAdapt - the adapted DM object: may be NULL if an adapted DM could not be produced.
7319 
7320   Level: intermediate
7321 
7322 .seealso: DMAdaptMetric(), DMCoarsen(), DMRefine()
7323 @*/
7324 PetscErrorCode DMAdaptLabel(DM dm, DMLabel label, DM *dmAdapt)
7325 {
7326   PetscErrorCode ierr;
7327 
7328   PetscFunctionBegin;
7329   PetscValidHeaderSpecific(dm, DM_CLASSID, 1);
7330   PetscValidPointer(label,2);
7331   PetscValidPointer(dmAdapt,3);
7332   *dmAdapt = NULL;
7333   if (!dm->ops->adaptlabel) SETERRQ1(PetscObjectComm((PetscObject)dm),PETSC_ERR_SUP,"DM type %s does not implement DMAdaptLabel",((PetscObject)dm)->type_name);
7334   ierr = (dm->ops->adaptlabel)(dm, label, dmAdapt);CHKERRQ(ierr);
7335   PetscFunctionReturn(0);
7336 }
7337 
7338 /*@C
7339   DMAdaptMetric - Generates a mesh adapted to the specified metric field using the pragmatic library.
7340 
7341   Input Parameters:
7342 + dm - The DM object
7343 . metric - The metric to which the mesh is adapted, defined vertex-wise.
7344 - 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_".
7345 
7346   Output Parameter:
7347 . dmAdapt  - Pointer to the DM object containing the adapted mesh
7348 
7349   Note: The label in the adapted mesh will be registered under the name of the input DMLabel object
7350 
7351   Level: advanced
7352 
7353 .seealso: DMAdaptLabel(), DMCoarsen(), DMRefine()
7354 @*/
7355 PetscErrorCode DMAdaptMetric(DM dm, Vec metric, DMLabel bdLabel, DM *dmAdapt)
7356 {
7357   PetscErrorCode ierr;
7358 
7359   PetscFunctionBegin;
7360   PetscValidHeaderSpecific(dm, DM_CLASSID, 1);
7361   PetscValidHeaderSpecific(metric, VEC_CLASSID, 2);
7362   if (bdLabel) PetscValidPointer(bdLabel, 3);
7363   PetscValidPointer(dmAdapt, 4);
7364   *dmAdapt = NULL;
7365   if (!dm->ops->adaptmetric) SETERRQ1(PetscObjectComm((PetscObject)dm),PETSC_ERR_SUP,"DM type %s does not implement DMAdaptMetric",((PetscObject)dm)->type_name);
7366   ierr = (dm->ops->adaptmetric)(dm, metric, bdLabel, dmAdapt);CHKERRQ(ierr);
7367   PetscFunctionReturn(0);
7368 }
7369 
7370 /*@C
7371  DMGetNeighbors - Gets an array containing the MPI rank of all the processes neighbors
7372 
7373  Not Collective
7374 
7375  Input Parameter:
7376  . dm    - The DM
7377 
7378  Output Parameter:
7379  . nranks - the number of neighbours
7380  . ranks - the neighbors ranks
7381 
7382  Notes:
7383  Do not free the array, it is freed when the DM is destroyed.
7384 
7385  Level: beginner
7386 
7387  .seealso: DMDAGetNeighbors(), PetscSFGetRanks()
7388 @*/
7389 PetscErrorCode DMGetNeighbors(DM dm,PetscInt *nranks,const PetscMPIInt *ranks[])
7390 {
7391   PetscErrorCode ierr;
7392 
7393   PetscFunctionBegin;
7394   PetscValidHeaderSpecific(dm,DM_CLASSID,1);
7395   if (!dm->ops->getneighbors) SETERRQ1(PetscObjectComm((PetscObject)dm),PETSC_ERR_SUP,"DM type %s does not implement DMGetNeighbors",((PetscObject)dm)->type_name);
7396   ierr = (dm->ops->getneighbors)(dm,nranks,ranks);CHKERRQ(ierr);
7397   PetscFunctionReturn(0);
7398 }
7399 
7400 #include <petsc/private/matimpl.h> /* Needed because of coloring->ctype below */
7401 
7402 /*
7403     Converts the input vector to a ghosted vector and then calls the standard coloring code.
7404     This has be a different function because it requires DM which is not defined in the Mat library
7405 */
7406 PetscErrorCode  MatFDColoringApply_AIJDM(Mat J,MatFDColoring coloring,Vec x1,void *sctx)
7407 {
7408   PetscErrorCode ierr;
7409 
7410   PetscFunctionBegin;
7411   if (coloring->ctype == IS_COLORING_LOCAL) {
7412     Vec x1local;
7413     DM  dm;
7414     ierr = MatGetDM(J,&dm);CHKERRQ(ierr);
7415     if (!dm) SETERRQ(PetscObjectComm((PetscObject)J),PETSC_ERR_ARG_INCOMP,"IS_COLORING_LOCAL requires a DM");
7416     ierr = DMGetLocalVector(dm,&x1local);CHKERRQ(ierr);
7417     ierr = DMGlobalToLocalBegin(dm,x1,INSERT_VALUES,x1local);CHKERRQ(ierr);
7418     ierr = DMGlobalToLocalEnd(dm,x1,INSERT_VALUES,x1local);CHKERRQ(ierr);
7419     x1   = x1local;
7420   }
7421   ierr = MatFDColoringApply_AIJ(J,coloring,x1,sctx);CHKERRQ(ierr);
7422   if (coloring->ctype == IS_COLORING_LOCAL) {
7423     DM  dm;
7424     ierr = MatGetDM(J,&dm);CHKERRQ(ierr);
7425     ierr = DMRestoreLocalVector(dm,&x1);CHKERRQ(ierr);
7426   }
7427   PetscFunctionReturn(0);
7428 }
7429 
7430 /*@
7431     MatFDColoringUseDM - allows a MatFDColoring object to use the DM associated with the matrix to use a IS_COLORING_LOCAL coloring
7432 
7433     Input Parameter:
7434 .    coloring - the MatFDColoring object
7435 
7436     Developer Notes:
7437     this routine exists because the PETSc Mat library does not know about the DM objects
7438 
7439     Level: advanced
7440 
7441 .seealso: MatFDColoring, MatFDColoringCreate(), ISColoringType
7442 @*/
7443 PetscErrorCode  MatFDColoringUseDM(Mat coloring,MatFDColoring fdcoloring)
7444 {
7445   PetscFunctionBegin;
7446   coloring->ops->fdcoloringapply = MatFDColoringApply_AIJDM;
7447   PetscFunctionReturn(0);
7448 }
7449 
7450 /*@
7451     DMGetCompatibility - determine if two DMs are compatible
7452 
7453     Collective
7454 
7455     Input Parameters:
7456 +    dm - the first DM
7457 -    dm2 - the second DM
7458 
7459     Output Parameters:
7460 +    compatible - whether or not the two DMs are compatible
7461 -    set - whether or not the compatible value was set
7462 
7463     Notes:
7464     Two DMs are deemed compatible if they represent the same parallel decomposition
7465     of the same topology. This implies that the the section (field data) on one
7466     "makes sense" with respect to the topology and parallel decomposition of the other.
7467     Loosely speaking, compatibile DMs represent the same domain, with the same parallel
7468     decomposition, with different data.
7469 
7470     Typically, one would confirm compatibility if intending to simultaneously iterate
7471     over a pair of vectors obtained from different DMs.
7472 
7473     For example, two DMDA objects are compatible if they have the same local
7474     and global sizes and the same stencil width. They can have different numbers
7475     of degrees of freedom per node. Thus, one could use the node numbering from
7476     either DM in bounds for a loop over vectors derived from either DM.
7477 
7478     Consider the operation of summing data living on a 2-dof DMDA to data living
7479     on a 1-dof DMDA, which should be compatible, as in the following snippet.
7480 .vb
7481   ...
7482   ierr = DMGetCompatibility(da1,da2,&compatible,&set);CHKERRQ(ierr);
7483   if (set && compatible)  {
7484     ierr = DMDAVecGetArrayDOF(da1,vec1,&arr1);CHKERRQ(ierr);
7485     ierr = DMDAVecGetArrayDOF(da2,vec2,&arr2);CHKERRQ(ierr);
7486     ierr = DMDAGetCorners(da1,&x,&y,NULL,&m,&n);CHKERRQ(ierr);
7487     for (j=y; j<y+n; ++j) {
7488       for (i=x; i<x+m, ++i) {
7489         arr1[j][i][0] = arr2[j][i][0] + arr2[j][i][1];
7490       }
7491     }
7492     ierr = DMDAVecRestoreArrayDOF(da1,vec1,&arr1);CHKERRQ(ierr);
7493     ierr = DMDAVecRestoreArrayDOF(da2,vec2,&arr2);CHKERRQ(ierr);
7494   } else {
7495     SETERRQ(PetscObjectComm((PetscObject)da1,PETSC_ERR_ARG_INCOMP,"DMDA objects incompatible");
7496   }
7497   ...
7498 .ve
7499 
7500     Checking compatibility might be expensive for a given implementation of DM,
7501     or might be impossible to unambiguously confirm or deny. For this reason,
7502     this function may decline to determine compatibility, and hence users should
7503     always check the "set" output parameter.
7504 
7505     A DM is always compatible with itself.
7506 
7507     In the current implementation, DMs which live on "unequal" communicators
7508     (MPI_UNEQUAL in the terminology of MPI_Comm_compare()) are always deemed
7509     incompatible.
7510 
7511     This function is labeled "Collective," as information about all subdomains
7512     is required on each rank. However, in DM implementations which store all this
7513     information locally, this function may be merely "Logically Collective".
7514 
7515     Developer Notes:
7516     Compatibility is assumed to be a symmetric concept; if DM A is compatible with DM B,
7517     the DM B is compatible with DM A. Thus, this function checks the implementations
7518     of both dm and dm2 (if they are of different types), attempting to determine
7519     compatibility. It is left to DM implementers to ensure that symmetry is
7520     preserved. The simplest way to do this is, when implementing type-specific
7521     logic for this function, to check for existing logic in the implementation
7522     of other DM types and let *set = PETSC_FALSE if found; the logic of this
7523     function will then call that logic.
7524 
7525     Level: advanced
7526 
7527 .seealso: DM, DMDACreateCompatibleDMDA()
7528 @*/
7529 
7530 PetscErrorCode DMGetCompatibility(DM dm,DM dm2,PetscBool *compatible,PetscBool *set)
7531 {
7532   PetscErrorCode ierr;
7533   PetscMPIInt    compareResult;
7534   DMType         type,type2;
7535   PetscBool      sameType;
7536 
7537   PetscFunctionBegin;
7538   PetscValidHeaderSpecific(dm,DM_CLASSID,1);
7539   PetscValidHeaderSpecific(dm2,DM_CLASSID,2);
7540 
7541   /* Declare a DM compatible with itself */
7542   if (dm == dm2) {
7543     *set = PETSC_TRUE;
7544     *compatible = PETSC_TRUE;
7545     PetscFunctionReturn(0);
7546   }
7547 
7548   /* Declare a DM incompatible with a DM that lives on an "unequal"
7549      communicator. Note that this does not preclude compatibility with
7550      DMs living on "congruent" or "similar" communicators, but this must be
7551      determined by the implementation-specific logic */
7552   ierr = MPI_Comm_compare(PetscObjectComm((PetscObject)dm),PetscObjectComm((PetscObject)dm2),&compareResult);CHKERRQ(ierr);
7553   if (compareResult == MPI_UNEQUAL) {
7554     *set = PETSC_TRUE;
7555     *compatible = PETSC_FALSE;
7556     PetscFunctionReturn(0);
7557   }
7558 
7559   /* Pass to the implementation-specific routine, if one exists. */
7560   if (dm->ops->getcompatibility) {
7561     ierr = (*dm->ops->getcompatibility)(dm,dm2,compatible,set);CHKERRQ(ierr);
7562     if (*set) {
7563       PetscFunctionReturn(0);
7564     }
7565   }
7566 
7567   /* If dm and dm2 are of different types, then attempt to check compatibility
7568      with an implementation of this function from dm2 */
7569   ierr = DMGetType(dm,&type);CHKERRQ(ierr);
7570   ierr = DMGetType(dm2,&type2);CHKERRQ(ierr);
7571   ierr = PetscStrcmp(type,type2,&sameType);CHKERRQ(ierr);
7572   if (!sameType && dm2->ops->getcompatibility) {
7573     ierr = (*dm2->ops->getcompatibility)(dm2,dm,compatible,set);CHKERRQ(ierr); /* Note argument order */
7574   } else {
7575     *set = PETSC_FALSE;
7576   }
7577   PetscFunctionReturn(0);
7578 }
7579