xref: /petsc/src/dm/impls/plex/plexinterpolate.c (revision 27d0e99a0845f279d25f36eb29f4a4212ca97ece)
1af0996ceSBarry Smith #include <petsc/private/dmpleximpl.h>   /*I      "petscdmplex.h"   I*/
2e8f14785SLisandro Dalcin #include <petsc/private/hashmapi.h>
3e8f14785SLisandro Dalcin #include <petsc/private/hashmapij.h>
4e8f14785SLisandro Dalcin 
5e8f14785SLisandro Dalcin /* HashIJKL */
6e8f14785SLisandro Dalcin 
7e8f14785SLisandro Dalcin #include <petsc/private/hashmap.h>
8e8f14785SLisandro Dalcin 
9e8f14785SLisandro Dalcin typedef struct _PetscHashIJKLKey { PetscInt i, j, k, l; } PetscHashIJKLKey;
10e8f14785SLisandro Dalcin 
11e8f14785SLisandro Dalcin #define PetscHashIJKLKeyHash(key) \
12e8f14785SLisandro Dalcin   PetscHashCombine(PetscHashCombine(PetscHashInt((key).i),PetscHashInt((key).j)), \
13e8f14785SLisandro Dalcin                    PetscHashCombine(PetscHashInt((key).k),PetscHashInt((key).l)))
14e8f14785SLisandro Dalcin 
15e8f14785SLisandro Dalcin #define PetscHashIJKLKeyEqual(k1,k2) \
16e8f14785SLisandro Dalcin   (((k1).i==(k2).i) ? ((k1).j==(k2).j) ? ((k1).k==(k2).k) ? ((k1).l==(k2).l) : 0 : 0 : 0)
17e8f14785SLisandro Dalcin 
18e8f14785SLisandro Dalcin PETSC_HASH_MAP(HashIJKL, PetscHashIJKLKey, PetscInt, PetscHashIJKLKeyHash, PetscHashIJKLKeyEqual, -1)
19e8f14785SLisandro Dalcin 
209f074e33SMatthew G Knepley 
219f074e33SMatthew G Knepley /*
229a5b13a2SMatthew G Knepley   DMPlexGetFaces_Internal - Gets groups of vertices that correspond to faces for the given cell
23cd921712SStefano Zampini   This assumes that the mesh is not interpolated from the depth of point p to the vertices
249f074e33SMatthew G Knepley */
25439ece16SMatthew G. Knepley PetscErrorCode DMPlexGetFaces_Internal(DM dm, PetscInt dim, PetscInt p, PetscInt *numFaces, PetscInt *faceSize, const PetscInt *faces[])
269f074e33SMatthew G Knepley {
279f074e33SMatthew G Knepley   const PetscInt *cone = NULL;
28cd921712SStefano Zampini   PetscInt        coneSize;
299f074e33SMatthew G Knepley   PetscErrorCode  ierr;
309f074e33SMatthew G Knepley 
319f074e33SMatthew G Knepley   PetscFunctionBegin;
329f074e33SMatthew G Knepley   PetscValidHeaderSpecific(dm, DM_CLASSID, 1);
339f074e33SMatthew G Knepley   ierr = DMPlexGetConeSize(dm, p, &coneSize);CHKERRQ(ierr);
349f074e33SMatthew G Knepley   ierr = DMPlexGetCone(dm, p, &cone);CHKERRQ(ierr);
35439ece16SMatthew G. Knepley   ierr = DMPlexGetRawFaces_Internal(dm, dim, coneSize, cone, numFaces, faceSize, faces);CHKERRQ(ierr);
36439ece16SMatthew G. Knepley   PetscFunctionReturn(0);
37439ece16SMatthew G. Knepley }
38439ece16SMatthew G. Knepley 
39439ece16SMatthew G. Knepley /*
40439ece16SMatthew G. Knepley   DMPlexRestoreFaces_Internal - Restores the array
41439ece16SMatthew G. Knepley */
42439ece16SMatthew G. Knepley PetscErrorCode DMPlexRestoreFaces_Internal(DM dm, PetscInt dim, PetscInt p, PetscInt *numFaces, PetscInt *faceSize, const PetscInt *faces[])
43439ece16SMatthew G. Knepley {
44439ece16SMatthew G. Knepley   PetscErrorCode  ierr;
45439ece16SMatthew G. Knepley 
46439ece16SMatthew G. Knepley   PetscFunctionBegin;
47cd921712SStefano Zampini   if (faces) { ierr = DMRestoreWorkArray(dm, 0, MPIU_INT, (void *) faces);CHKERRQ(ierr); }
48439ece16SMatthew G. Knepley   PetscFunctionReturn(0);
49439ece16SMatthew G. Knepley }
50439ece16SMatthew G. Knepley 
51439ece16SMatthew G. Knepley /*
52439ece16SMatthew G. Knepley   DMPlexGetRawFaces_Internal - Gets groups of vertices that correspond to faces for the given cone
53439ece16SMatthew G. Knepley */
54439ece16SMatthew G. Knepley PetscErrorCode DMPlexGetRawFaces_Internal(DM dm, PetscInt dim, PetscInt coneSize, const PetscInt cone[], PetscInt *numFaces, PetscInt *faceSize, const PetscInt *faces[])
55439ece16SMatthew G. Knepley {
56439ece16SMatthew G. Knepley   PetscInt       *facesTmp;
57439ece16SMatthew G. Knepley   PetscInt        maxConeSize, maxSupportSize;
58439ece16SMatthew G. Knepley   PetscErrorCode  ierr;
59439ece16SMatthew G. Knepley 
60439ece16SMatthew G. Knepley   PetscFunctionBegin;
61439ece16SMatthew G. Knepley   PetscValidHeaderSpecific(dm, DM_CLASSID, 1);
62cd921712SStefano Zampini   if (faces && coneSize) PetscValidIntPointer(cone,4);
63439ece16SMatthew G. Knepley   ierr = DMPlexGetMaxSizes(dm, &maxConeSize, &maxSupportSize);CHKERRQ(ierr);
6469291d52SBarry Smith   if (faces) {ierr = DMGetWorkArray(dm, PetscSqr(PetscMax(maxConeSize, maxSupportSize)), MPIU_INT, &facesTmp);CHKERRQ(ierr);}
659f074e33SMatthew G Knepley   switch (dim) {
66c49d9212SMatthew G. Knepley   case 1:
67c49d9212SMatthew G. Knepley     switch (coneSize) {
68c49d9212SMatthew G. Knepley     case 2:
69c49d9212SMatthew G. Knepley       if (faces) {
70c49d9212SMatthew G. Knepley         facesTmp[0] = cone[0]; facesTmp[1] = cone[1];
71c49d9212SMatthew G. Knepley         *faces = facesTmp;
72c49d9212SMatthew G. Knepley       }
73c49d9212SMatthew G. Knepley       if (numFaces) *numFaces = 2;
74c49d9212SMatthew G. Knepley       if (faceSize) *faceSize = 1;
75c49d9212SMatthew G. Knepley       break;
76c49d9212SMatthew G. Knepley     default:
7799836e0eSStefano Zampini       SETERRQ2(PETSC_COMM_SELF, PETSC_ERR_ARG_OUTOFRANGE, "Cone size %D not supported for dimension %D", coneSize, dim);
78c49d9212SMatthew G. Knepley     }
79c49d9212SMatthew G. Knepley     break;
809f074e33SMatthew G Knepley   case 2:
819f074e33SMatthew G Knepley     switch (coneSize) {
829f074e33SMatthew G Knepley     case 3:
839a5b13a2SMatthew G Knepley       if (faces) {
849a5b13a2SMatthew G Knepley         facesTmp[0] = cone[0]; facesTmp[1] = cone[1];
859a5b13a2SMatthew G Knepley         facesTmp[2] = cone[1]; facesTmp[3] = cone[2];
869a5b13a2SMatthew G Knepley         facesTmp[4] = cone[2]; facesTmp[5] = cone[0];
879a5b13a2SMatthew G Knepley         *faces = facesTmp;
889a5b13a2SMatthew G Knepley       }
899a5b13a2SMatthew G Knepley       if (numFaces) *numFaces = 3;
909a5b13a2SMatthew G Knepley       if (faceSize) *faceSize = 2;
919f074e33SMatthew G Knepley       break;
929f074e33SMatthew G Knepley     case 4:
939a5b13a2SMatthew G Knepley       /* Vertices follow right hand rule */
949a5b13a2SMatthew G Knepley       if (faces) {
959a5b13a2SMatthew G Knepley         facesTmp[0] = cone[0]; facesTmp[1] = cone[1];
969a5b13a2SMatthew G Knepley         facesTmp[2] = cone[1]; facesTmp[3] = cone[2];
979a5b13a2SMatthew G Knepley         facesTmp[4] = cone[2]; facesTmp[5] = cone[3];
989a5b13a2SMatthew G Knepley         facesTmp[6] = cone[3]; facesTmp[7] = cone[0];
999a5b13a2SMatthew G Knepley         *faces = facesTmp;
1009a5b13a2SMatthew G Knepley       }
1019a5b13a2SMatthew G Knepley       if (numFaces) *numFaces = 4;
1029a5b13a2SMatthew G Knepley       if (faceSize) *faceSize = 2;
1039f074e33SMatthew G Knepley       break;
1049f074e33SMatthew G Knepley     default:
10599836e0eSStefano Zampini       SETERRQ2(PETSC_COMM_SELF, PETSC_ERR_ARG_OUTOFRANGE, "Cone size %D not supported for dimension %D", coneSize, dim);
1069f074e33SMatthew G Knepley     }
1079f074e33SMatthew G Knepley     break;
1089f074e33SMatthew G Knepley   case 3:
1099f074e33SMatthew G Knepley     switch (coneSize) {
1109f074e33SMatthew G Knepley     case 3:
1119a5b13a2SMatthew G Knepley       if (faces) {
1129a5b13a2SMatthew G Knepley         facesTmp[0] = cone[0]; facesTmp[1] = cone[1];
1139a5b13a2SMatthew G Knepley         facesTmp[2] = cone[1]; facesTmp[3] = cone[2];
1149a5b13a2SMatthew G Knepley         facesTmp[4] = cone[2]; facesTmp[5] = cone[0];
1159a5b13a2SMatthew G Knepley         *faces = facesTmp;
1169a5b13a2SMatthew G Knepley       }
1179a5b13a2SMatthew G Knepley       if (numFaces) *numFaces = 3;
1189a5b13a2SMatthew G Knepley       if (faceSize) *faceSize = 2;
1199f074e33SMatthew G Knepley       break;
1209f074e33SMatthew G Knepley     case 4:
1212e1b13c2SMatthew G. Knepley       /* Vertices of first face follow right hand rule and normal points away from last vertex */
1229a5b13a2SMatthew G Knepley       if (faces) {
1232e1b13c2SMatthew G. Knepley         facesTmp[0] = cone[0]; facesTmp[1]  = cone[1]; facesTmp[2]  = cone[2];
1242e1b13c2SMatthew G. Knepley         facesTmp[3] = cone[0]; facesTmp[4]  = cone[3]; facesTmp[5]  = cone[1];
1252e1b13c2SMatthew G. Knepley         facesTmp[6] = cone[0]; facesTmp[7]  = cone[2]; facesTmp[8]  = cone[3];
1262e1b13c2SMatthew G. Knepley         facesTmp[9] = cone[2]; facesTmp[10] = cone[1]; facesTmp[11] = cone[3];
1279a5b13a2SMatthew G Knepley         *faces = facesTmp;
1289a5b13a2SMatthew G Knepley       }
1299a5b13a2SMatthew G Knepley       if (numFaces) *numFaces = 4;
1309a5b13a2SMatthew G Knepley       if (faceSize) *faceSize = 3;
1319a5b13a2SMatthew G Knepley       break;
1329a5b13a2SMatthew G Knepley     case 8:
133e368ef20SMatthew G. Knepley       /*  7--------6
134e368ef20SMatthew G. Knepley          /|       /|
135e368ef20SMatthew G. Knepley         / |      / |
136e368ef20SMatthew G. Knepley        4--------5  |
137e368ef20SMatthew G. Knepley        |  |     |  |
138e368ef20SMatthew G. Knepley        |  |     |  |
139e368ef20SMatthew G. Knepley        |  1--------2
140e368ef20SMatthew G. Knepley        | /      | /
141e368ef20SMatthew G. Knepley        |/       |/
142e368ef20SMatthew G. Knepley        0--------3
143e368ef20SMatthew G. Knepley        */
1449a5b13a2SMatthew G Knepley       if (faces) {
14551cfd6a4SMatthew G. Knepley         facesTmp[0]  = cone[0]; facesTmp[1]  = cone[1]; facesTmp[2]  = cone[2]; facesTmp[3]  = cone[3]; /* Bottom */
14651cfd6a4SMatthew G. Knepley         facesTmp[4]  = cone[4]; facesTmp[5]  = cone[5]; facesTmp[6]  = cone[6]; facesTmp[7]  = cone[7]; /* Top */
14751cfd6a4SMatthew G. Knepley         facesTmp[8]  = cone[0]; facesTmp[9]  = cone[3]; facesTmp[10] = cone[5]; facesTmp[11] = cone[4]; /* Front */
14851cfd6a4SMatthew G. Knepley         facesTmp[12] = cone[2]; facesTmp[13] = cone[1]; facesTmp[14] = cone[7]; facesTmp[15] = cone[6]; /* Back */
14951cfd6a4SMatthew G. Knepley         facesTmp[16] = cone[3]; facesTmp[17] = cone[2]; facesTmp[18] = cone[6]; facesTmp[19] = cone[5]; /* Right */
15051cfd6a4SMatthew G. Knepley         facesTmp[20] = cone[0]; facesTmp[21] = cone[4]; facesTmp[22] = cone[7]; facesTmp[23] = cone[1]; /* Left */
1519a5b13a2SMatthew G Knepley         *faces = facesTmp;
1529a5b13a2SMatthew G Knepley       }
1539a5b13a2SMatthew G Knepley       if (numFaces) *numFaces = 6;
1549a5b13a2SMatthew G Knepley       if (faceSize) *faceSize = 4;
1559f074e33SMatthew G Knepley       break;
1569f074e33SMatthew G Knepley     default:
15799836e0eSStefano Zampini       SETERRQ2(PETSC_COMM_SELF, PETSC_ERR_ARG_OUTOFRANGE, "Cone size %D not supported for dimension %D", coneSize, dim);
1589f074e33SMatthew G Knepley     }
1599f074e33SMatthew G Knepley     break;
1609f074e33SMatthew G Knepley   default:
16199836e0eSStefano Zampini     SETERRQ1(PETSC_COMM_SELF, PETSC_ERR_ARG_OUTOFRANGE, "Dimension %D not supported", dim);
1629f074e33SMatthew G Knepley   }
1639f074e33SMatthew G Knepley   PetscFunctionReturn(0);
1649f074e33SMatthew G Knepley }
1659f074e33SMatthew G Knepley 
16699836e0eSStefano Zampini /*
16799836e0eSStefano Zampini   DMPlexGetRawFacesHybrid_Internal - Gets groups of vertices that correspond to faces for the given cone using hybrid ordering (prisms)
16899836e0eSStefano Zampini */
16999836e0eSStefano Zampini static PetscErrorCode DMPlexGetRawFacesHybrid_Internal(DM dm, PetscInt dim, PetscInt coneSize, const PetscInt cone[], PetscInt *numFaces, PetscInt *numFacesNotH, PetscInt *faceSize, const PetscInt *faces[])
17099836e0eSStefano Zampini {
17199836e0eSStefano Zampini   PetscInt       *facesTmp;
17299836e0eSStefano Zampini   PetscInt        maxConeSize, maxSupportSize;
17399836e0eSStefano Zampini   PetscErrorCode  ierr;
17499836e0eSStefano Zampini 
17599836e0eSStefano Zampini   PetscFunctionBegin;
17699836e0eSStefano Zampini   PetscValidHeaderSpecific(dm, DM_CLASSID, 1);
17799836e0eSStefano Zampini   if (faces && coneSize) PetscValidIntPointer(cone,4);
17899836e0eSStefano Zampini   ierr = DMPlexGetMaxSizes(dm, &maxConeSize, &maxSupportSize);CHKERRQ(ierr);
17999836e0eSStefano Zampini   if (faces) {ierr = DMGetWorkArray(dm, PetscSqr(PetscMax(maxConeSize, maxSupportSize)), MPIU_INT, &facesTmp);CHKERRQ(ierr);}
18099836e0eSStefano Zampini   switch (dim) {
18199836e0eSStefano Zampini   case 1:
18299836e0eSStefano Zampini     switch (coneSize) {
18399836e0eSStefano Zampini     case 2:
18499836e0eSStefano Zampini       if (faces) {
18599836e0eSStefano Zampini         facesTmp[0] = cone[0]; facesTmp[1] = cone[1];
18699836e0eSStefano Zampini         *faces = facesTmp;
18799836e0eSStefano Zampini       }
18899836e0eSStefano Zampini       if (numFaces)     *numFaces = 2;
18999836e0eSStefano Zampini       if (numFacesNotH) *numFacesNotH = 2;
19099836e0eSStefano Zampini       if (faceSize)     *faceSize = 1;
19199836e0eSStefano Zampini       break;
19299836e0eSStefano Zampini     default:
19399836e0eSStefano Zampini       SETERRQ2(PETSC_COMM_SELF, PETSC_ERR_ARG_OUTOFRANGE, "Cone size %D not supported for dimension %D", coneSize, dim);
19499836e0eSStefano Zampini     }
19599836e0eSStefano Zampini     break;
19699836e0eSStefano Zampini   case 2:
19799836e0eSStefano Zampini     switch (coneSize) {
19899836e0eSStefano Zampini     case 4:
19999836e0eSStefano Zampini       if (faces) {
20099836e0eSStefano Zampini         facesTmp[0] = cone[0]; facesTmp[1] = cone[1];
20199836e0eSStefano Zampini         facesTmp[2] = cone[2]; facesTmp[3] = cone[3];
20299836e0eSStefano Zampini         facesTmp[4] = cone[0]; facesTmp[5] = cone[2];
20399836e0eSStefano Zampini         facesTmp[6] = cone[1]; facesTmp[7] = cone[3];
20499836e0eSStefano Zampini         *faces = facesTmp;
20599836e0eSStefano Zampini       }
20699836e0eSStefano Zampini       if (numFaces)     *numFaces = 4;
20799836e0eSStefano Zampini       if (numFacesNotH) *numFacesNotH = 2;
20899836e0eSStefano Zampini       if (faceSize)     *faceSize = 2;
20999836e0eSStefano Zampini       break;
21099836e0eSStefano Zampini     default:
21199836e0eSStefano Zampini       SETERRQ2(PETSC_COMM_SELF, PETSC_ERR_ARG_OUTOFRANGE, "Cone size %D not supported for dimension %D", coneSize, dim);
21299836e0eSStefano Zampini     }
21399836e0eSStefano Zampini     break;
21499836e0eSStefano Zampini   case 3:
21599836e0eSStefano Zampini     switch (coneSize) {
21699836e0eSStefano Zampini     case 6: /* triangular prism */
21799836e0eSStefano Zampini       if (faces) {
21899836e0eSStefano Zampini         facesTmp[0]  = cone[0]; facesTmp[1]  = cone[1]; facesTmp[2]  = cone[2]; facesTmp[3]  = -1;      /* Bottom */
21999836e0eSStefano Zampini         facesTmp[4]  = cone[3]; facesTmp[5]  = cone[4]; facesTmp[6]  = cone[5]; facesTmp[7]  = -1;      /* Top */
22099836e0eSStefano Zampini         facesTmp[8]  = cone[0]; facesTmp[9]  = cone[1]; facesTmp[10] = cone[3]; facesTmp[11] = cone[4]; /* Back left */
22199836e0eSStefano Zampini         facesTmp[12] = cone[1]; facesTmp[13] = cone[2]; facesTmp[14] = cone[4]; facesTmp[15] = cone[5]; /* Back right */
22299836e0eSStefano Zampini         facesTmp[16] = cone[2]; facesTmp[17] = cone[0]; facesTmp[18] = cone[5]; facesTmp[19] = cone[3]; /* Front */
22399836e0eSStefano Zampini         *faces = facesTmp;
22499836e0eSStefano Zampini       }
22599836e0eSStefano Zampini       if (numFaces)     *numFaces = 5;
22699836e0eSStefano Zampini       if (numFacesNotH) *numFacesNotH = 2;
22799836e0eSStefano Zampini       if (faceSize)     *faceSize = -4;
22899836e0eSStefano Zampini       break;
22999836e0eSStefano Zampini     default:
23099836e0eSStefano Zampini       SETERRQ2(PETSC_COMM_SELF, PETSC_ERR_ARG_OUTOFRANGE, "Cone size %D not supported for dimension %D", coneSize, dim);
23199836e0eSStefano Zampini     }
23299836e0eSStefano Zampini     break;
23399836e0eSStefano Zampini   default:
23499836e0eSStefano Zampini     SETERRQ1(PETSC_COMM_SELF, PETSC_ERR_ARG_OUTOFRANGE, "Dimension %D not supported", dim);
23599836e0eSStefano Zampini   }
23699836e0eSStefano Zampini   PetscFunctionReturn(0);
23799836e0eSStefano Zampini }
23899836e0eSStefano Zampini 
23999836e0eSStefano Zampini static PetscErrorCode DMPlexRestoreRawFacesHybrid_Internal(DM dm, PetscInt dim, PetscInt coneSize, const PetscInt cone[], PetscInt *numFaces, PetscInt *numFacesNotH, PetscInt *faceSize, const PetscInt *faces[])
24099836e0eSStefano Zampini {
24199836e0eSStefano Zampini   PetscErrorCode  ierr;
24299836e0eSStefano Zampini 
24399836e0eSStefano Zampini   PetscFunctionBegin;
24499836e0eSStefano Zampini   if (faces) { ierr = DMRestoreWorkArray(dm, 0, MPIU_INT, (void *) faces);CHKERRQ(ierr); }
24599836e0eSStefano Zampini   PetscFunctionReturn(0);
24699836e0eSStefano Zampini }
24799836e0eSStefano Zampini 
24899836e0eSStefano Zampini static PetscErrorCode DMPlexGetFacesHybrid_Internal(DM dm, PetscInt dim, PetscInt p, PetscInt *numFaces, PetscInt *numFacesNotH, PetscInt *faceSize, const PetscInt *faces[])
24999836e0eSStefano Zampini {
25099836e0eSStefano Zampini   const PetscInt *cone = NULL;
25199836e0eSStefano Zampini   PetscInt        coneSize;
25299836e0eSStefano Zampini   PetscErrorCode  ierr;
25399836e0eSStefano Zampini 
25499836e0eSStefano Zampini   PetscFunctionBegin;
25599836e0eSStefano Zampini   PetscValidHeaderSpecific(dm, DM_CLASSID, 1);
25699836e0eSStefano Zampini   ierr = DMPlexGetConeSize(dm, p, &coneSize);CHKERRQ(ierr);
25799836e0eSStefano Zampini   ierr = DMPlexGetCone(dm, p, &cone);CHKERRQ(ierr);
25899836e0eSStefano Zampini   ierr = DMPlexGetRawFacesHybrid_Internal(dm, dim, coneSize, cone, numFaces, numFacesNotH, faceSize, faces);CHKERRQ(ierr);
25999836e0eSStefano Zampini   PetscFunctionReturn(0);
26099836e0eSStefano Zampini }
26199836e0eSStefano Zampini 
2629a5b13a2SMatthew G Knepley /* This interpolates faces for cells at some stratum */
2639a5b13a2SMatthew G Knepley static PetscErrorCode DMPlexInterpolateFaces_Internal(DM dm, PetscInt cellDepth, DM idm)
2649f074e33SMatthew G Knepley {
265d4efc6f1SMatthew G. Knepley   DMLabel        subpointMap;
2669a5b13a2SMatthew G Knepley   PetscHashIJKL  faceTable;
2679a5b13a2SMatthew G Knepley   PetscInt      *pStart, *pEnd;
2689a5b13a2SMatthew G Knepley   PetscInt       cellDim, depth, faceDepth = cellDepth, numPoints = 0, faceSizeAll = 0, face, c, d;
269e9fa77a1SMatthew G. Knepley   PetscInt       coneSizeH = 0, faceSizeAllH = 0, faceSizeAllT = 0, numCellFacesH = 0, faceT = 0, faceH, pMax = -1, dim, outerloop;
27099836e0eSStefano Zampini   PetscInt       cMax, fMax, eMax, vMax;
2719f074e33SMatthew G Knepley   PetscErrorCode ierr;
2729f074e33SMatthew G Knepley 
2739f074e33SMatthew G Knepley   PetscFunctionBegin;
274c73cfb54SMatthew G. Knepley   ierr = DMGetDimension(dm, &cellDim);CHKERRQ(ierr);
275d4efc6f1SMatthew G. Knepley   /* HACK: I need a better way to determine face dimension, or an alternative to GetFaces() */
276d4efc6f1SMatthew G. Knepley   ierr = DMPlexGetSubpointMap(dm, &subpointMap);CHKERRQ(ierr);
277d4efc6f1SMatthew G. Knepley   if (subpointMap) ++cellDim;
2789a5b13a2SMatthew G Knepley   ierr = DMPlexGetDepth(dm, &depth);CHKERRQ(ierr);
2799a5b13a2SMatthew G Knepley   ++depth;
2809a5b13a2SMatthew G Knepley   ++cellDepth;
2819a5b13a2SMatthew G Knepley   cellDim -= depth - cellDepth;
282dcca6d9dSJed Brown   ierr = PetscMalloc2(depth+1,&pStart,depth+1,&pEnd);CHKERRQ(ierr);
2839a5b13a2SMatthew G Knepley   for (d = depth-1; d >= faceDepth; --d) {
2849a5b13a2SMatthew G Knepley     ierr = DMPlexGetDepthStratum(dm, d, &pStart[d+1], &pEnd[d+1]);CHKERRQ(ierr);
2859f074e33SMatthew G Knepley   }
2869a5b13a2SMatthew G Knepley   ierr = DMPlexGetDepthStratum(dm, -1, NULL, &pStart[faceDepth]);CHKERRQ(ierr);
2879a5b13a2SMatthew G Knepley   pEnd[faceDepth] = pStart[faceDepth];
2889a5b13a2SMatthew G Knepley   for (d = faceDepth-1; d >= 0; --d) {
2899a5b13a2SMatthew G Knepley     ierr = DMPlexGetDepthStratum(dm, d, &pStart[d], &pEnd[d]);CHKERRQ(ierr);
2909f074e33SMatthew G Knepley   }
29199836e0eSStefano Zampini   cMax = fMax = eMax = vMax = PETSC_DETERMINE;
29299836e0eSStefano Zampini   ierr = DMGetDimension(dm, &dim);CHKERRQ(ierr);
29399836e0eSStefano Zampini   if (cellDim == dim) {
29499836e0eSStefano Zampini     ierr = DMPlexGetHybridBounds(dm, &cMax, NULL, NULL, NULL);CHKERRQ(ierr);
29599836e0eSStefano Zampini     pMax = cMax;
29699836e0eSStefano Zampini   } else if (cellDim == dim -1) {
29799836e0eSStefano Zampini     ierr = DMPlexGetHybridBounds(dm, &cMax, &fMax, NULL, NULL);CHKERRQ(ierr);
29899836e0eSStefano Zampini     pMax = fMax;
29999836e0eSStefano Zampini   }
30099836e0eSStefano Zampini   pMax = pMax < 0 ? pEnd[cellDepth] : pMax;
30199836e0eSStefano Zampini   if (pMax < pEnd[cellDepth]) {
30299836e0eSStefano Zampini     const PetscInt *cellFaces, *cone;
30399836e0eSStefano Zampini     PetscInt        numCellFacesT, faceSize, cf;
3049f074e33SMatthew G Knepley 
305e9fa77a1SMatthew G. Knepley     /* First get normal cell face size (we now allow hybrid cells to meet normal cells on either hybrid or normal faces */
3065ebdaad1SMatthew G. Knepley     if (pStart[cellDepth] < pMax) {ierr = DMPlexGetFaces_Internal(dm, cellDim, pStart[cellDepth], NULL, &faceSizeAll, NULL);CHKERRQ(ierr);}
307e9fa77a1SMatthew G. Knepley 
30899836e0eSStefano Zampini     ierr = DMPlexGetConeSize(dm, pMax, &coneSizeH);CHKERRQ(ierr);
30999836e0eSStefano Zampini     ierr = DMPlexGetCone(dm, pMax, &cone);CHKERRQ(ierr);
31099836e0eSStefano Zampini     ierr = DMPlexGetRawFacesHybrid_Internal(dm, cellDim, coneSizeH, cone, &numCellFacesH, &numCellFacesT, &faceSize, &cellFaces);CHKERRQ(ierr);
31199836e0eSStefano Zampini     if (faceSize < 0) {
31299836e0eSStefano Zampini       PetscInt *sizes, minv, maxv;
31399836e0eSStefano Zampini 
31499836e0eSStefano Zampini       /* count vertices of hybrid and non-hybrid faces */
31599836e0eSStefano Zampini       ierr = PetscCalloc1(numCellFacesH, &sizes);CHKERRQ(ierr);
31699836e0eSStefano Zampini       for (cf = 0; cf < numCellFacesT; ++cf) { /* These are the non-hybrid faces */
31799836e0eSStefano Zampini         const PetscInt *cellFace = &cellFaces[-cf*faceSize];
31899836e0eSStefano Zampini         PetscInt       f;
31999836e0eSStefano Zampini 
32099836e0eSStefano Zampini         for (f = 0; f < -faceSize; ++f) sizes[cf] += (cellFace[f] >= 0 ? 1 : 0);
32199836e0eSStefano Zampini       }
32299836e0eSStefano Zampini       ierr = PetscSortInt(numCellFacesT, sizes);CHKERRQ(ierr);
32399836e0eSStefano Zampini       minv = sizes[0];
32499836e0eSStefano Zampini       maxv = sizes[PetscMax(numCellFacesT-1, 0)];
32599836e0eSStefano Zampini       if (minv != maxv) SETERRQ2(PETSC_COMM_SELF, PETSC_ERR_SUP, "Different number of vertices for non-hybrid face %D != %D", minv, maxv);
326e9fa77a1SMatthew G. Knepley       faceSizeAllT = minv;
327580bdb30SBarry Smith       ierr = PetscArrayzero(sizes, numCellFacesH);CHKERRQ(ierr);
32899836e0eSStefano Zampini       for (cf = numCellFacesT; cf < numCellFacesH; ++cf) { /* These are the hybrid faces */
32999836e0eSStefano Zampini         const PetscInt *cellFace = &cellFaces[-cf*faceSize];
33099836e0eSStefano Zampini         PetscInt       f;
33199836e0eSStefano Zampini 
33299836e0eSStefano Zampini         for (f = 0; f < -faceSize; ++f) sizes[cf-numCellFacesT] += (cellFace[f] >= 0 ? 1 : 0);
33399836e0eSStefano Zampini       }
33499836e0eSStefano Zampini       ierr = PetscSortInt(numCellFacesH - numCellFacesT, sizes);CHKERRQ(ierr);
33599836e0eSStefano Zampini       minv = sizes[0];
33699836e0eSStefano Zampini       maxv = sizes[PetscMax(numCellFacesH - numCellFacesT-1, 0)];
33799836e0eSStefano Zampini       ierr = PetscFree(sizes);CHKERRQ(ierr);
33899836e0eSStefano Zampini       if (minv != maxv) SETERRQ2(PETSC_COMM_SELF, PETSC_ERR_SUP, "Different number of vertices for hybrid face %D != %D", minv, maxv);
33999836e0eSStefano Zampini       faceSizeAllH = minv;
3405ebdaad1SMatthew G. Knepley       if (!faceSizeAll) faceSizeAll = faceSizeAllT;
34199836e0eSStefano Zampini     } else { /* the size of the faces in hybrid cells is the same */
342e9fa77a1SMatthew G. Knepley       faceSizeAll = faceSizeAllH = faceSizeAllT = faceSize;
34399836e0eSStefano Zampini     }
34499836e0eSStefano Zampini     ierr = DMPlexRestoreRawFacesHybrid_Internal(dm, cellDim, coneSizeH, cone, &numCellFacesH, &numCellFacesT, &faceSize, &cellFaces);CHKERRQ(ierr);
34599836e0eSStefano Zampini   } else if (pEnd[cellDepth] > pStart[cellDepth]) {
34699836e0eSStefano Zampini     ierr = DMPlexGetFaces_Internal(dm, cellDim, pStart[cellDepth], NULL, &faceSizeAll, NULL);CHKERRQ(ierr);
347a1bcd873SMatthew G. Knepley     faceSizeAllH = faceSizeAllT = faceSizeAll;
34899836e0eSStefano Zampini   }
34999836e0eSStefano Zampini   if (faceSizeAll > 4) SETERRQ1(PETSC_COMM_SELF, PETSC_ERR_ARG_WRONG, "Do not support interpolation of meshes with faces of %D vertices", faceSizeAll);
35099836e0eSStefano Zampini 
351b135d7daSStefano Zampini   /* With hybrid grids, we first iterate on hybrid cells and start numbering the non-hybrid faces
35299836e0eSStefano Zampini      Then, faces for non-hybrid cells are numbered.
35399836e0eSStefano Zampini      This is to guarantee consistent orientations (all 0) of all the points in the cone of the hybrid cells */
35499836e0eSStefano Zampini   ierr = PetscHashIJKLCreate(&faceTable);CHKERRQ(ierr);
35599836e0eSStefano Zampini   for (outerloop = 0, face = pStart[faceDepth]; outerloop < 2; outerloop++) {
35699836e0eSStefano Zampini     PetscInt start, end;
35799836e0eSStefano Zampini 
35899836e0eSStefano Zampini     start = outerloop == 0 ? pMax : pStart[cellDepth];
35999836e0eSStefano Zampini     end = outerloop == 0 ? pEnd[cellDepth] : pMax;
36099836e0eSStefano Zampini     for (c = start; c < end; ++c) {
36199836e0eSStefano Zampini       const PetscInt *cellFaces;
362e9fa77a1SMatthew G. Knepley       PetscInt        numCellFaces, faceSize, faceSizeInc, faceSizeCheck, cf;
36399836e0eSStefano Zampini 
36499836e0eSStefano Zampini       if (c < pMax) {
3659a5b13a2SMatthew G Knepley         ierr = DMPlexGetFaces_Internal(dm, cellDim, c, &numCellFaces, &faceSize, &cellFaces);CHKERRQ(ierr);
3669a5b13a2SMatthew G Knepley         if (faceSize != faceSizeAll) SETERRQ3(PETSC_COMM_SELF, PETSC_ERR_PLIB, "Inconsistent face for cell %D of size %D != %D", c, faceSize, faceSizeAll);
367e9fa77a1SMatthew G. Knepley         faceSizeCheck = faceSizeAll;
36899836e0eSStefano Zampini       } else { /* Hybrid cell */
36999836e0eSStefano Zampini         const PetscInt *cone;
37099836e0eSStefano Zampini         PetscInt        numCellFacesN, coneSize;
37199836e0eSStefano Zampini 
37299836e0eSStefano Zampini         ierr = DMPlexGetConeSize(dm, c, &coneSize);CHKERRQ(ierr);
37399836e0eSStefano Zampini         if (coneSize != coneSizeH) SETERRQ2(PETSC_COMM_SELF, PETSC_ERR_SUP, "Unexpected hybrid coneSize %D != %D", coneSize, coneSizeH);
37499836e0eSStefano Zampini         ierr = DMPlexGetCone(dm, c, &cone);CHKERRQ(ierr);
37599836e0eSStefano Zampini         ierr = DMPlexGetRawFacesHybrid_Internal(dm, cellDim, coneSize, cone, &numCellFaces, &numCellFacesN, &faceSize, &cellFaces);CHKERRQ(ierr);
37699836e0eSStefano Zampini         if (numCellFaces != numCellFacesH) SETERRQ3(PETSC_COMM_SELF, PETSC_ERR_SUP, "Unexpected numCellFaces %D != %D for hybrid cell %D", numCellFaces, numCellFacesH, c);
37799836e0eSStefano Zampini         faceSize = PetscMax(faceSize, -faceSize);
37899836e0eSStefano Zampini         if (faceSize > 4) SETERRQ1(PETSC_COMM_SELF, PETSC_ERR_ARG_WRONG, "Do not support interpolation of meshes with faces of %D vertices", faceSize);
37999836e0eSStefano Zampini         numCellFaces = numCellFacesN; /* process only non-hybrid faces */
380e9fa77a1SMatthew G. Knepley         faceSizeCheck = faceSizeAllT;
38199836e0eSStefano Zampini       }
38299836e0eSStefano Zampini       faceSizeInc = faceSize;
3839f074e33SMatthew G Knepley       for (cf = 0; cf < numCellFaces; ++cf) {
38499836e0eSStefano Zampini         const PetscInt   *cellFace = &cellFaces[cf*faceSizeInc];
38599836e0eSStefano Zampini         PetscInt          faceSizeH = faceSize;
3869a5b13a2SMatthew G Knepley         PetscHashIJKLKey  key;
387e8f14785SLisandro Dalcin         PetscHashIter     iter;
388e8f14785SLisandro Dalcin         PetscBool         missing;
3899f074e33SMatthew G Knepley 
39099836e0eSStefano Zampini         if (faceSizeInc == 2) {
3919a5b13a2SMatthew G Knepley           key.i = PetscMin(cellFace[0], cellFace[1]);
3929a5b13a2SMatthew G Knepley           key.j = PetscMax(cellFace[0], cellFace[1]);
39399836e0eSStefano Zampini           key.k = PETSC_MAX_INT;
39499836e0eSStefano Zampini           key.l = PETSC_MAX_INT;
3959a5b13a2SMatthew G Knepley         } else {
39699836e0eSStefano Zampini           key.i = cellFace[0];
39799836e0eSStefano Zampini           key.j = cellFace[1];
39899836e0eSStefano Zampini           key.k = cellFace[2];
39999836e0eSStefano Zampini           key.l = faceSize > 3 ? (cellFace[3] < 0 ? faceSizeH = 3, PETSC_MAX_INT : cellFace[3]) : PETSC_MAX_INT;
400302440fdSBarry Smith           ierr  = PetscSortInt(faceSize, (PetscInt *) &key);CHKERRQ(ierr);
4019f074e33SMatthew G Knepley         }
40299836e0eSStefano Zampini         /* this check is redundant for non-hybrid meshes */
403e9fa77a1SMatthew G. Knepley         if (faceSizeH != faceSizeCheck) SETERRQ4(PETSC_COMM_SELF, PETSC_ERR_SUP, "Unexpected number of vertices for face %D of point %D -> %D != %D", cf, c, faceSizeH, faceSizeCheck);
404e8f14785SLisandro Dalcin         ierr = PetscHashIJKLPut(faceTable, key, &iter, &missing);CHKERRQ(ierr);
405e9fa77a1SMatthew G. Knepley         if (missing) {
406e9fa77a1SMatthew G. Knepley           ierr = PetscHashIJKLIterSet(faceTable, iter, face++);CHKERRQ(ierr);
407e9fa77a1SMatthew G. Knepley           if (c >= pMax) ++faceT;
408e9fa77a1SMatthew G. Knepley         }
4099a5b13a2SMatthew G Knepley       }
41099836e0eSStefano Zampini       if (c < pMax) {
411439ece16SMatthew G. Knepley         ierr = DMPlexRestoreFaces_Internal(dm, cellDim, c, &numCellFaces, &faceSize, &cellFaces);CHKERRQ(ierr);
41299836e0eSStefano Zampini       } else {
41399836e0eSStefano Zampini         ierr = DMPlexRestoreRawFacesHybrid_Internal(dm, cellDim, coneSizeH, NULL, NULL, NULL, NULL, &cellFaces);CHKERRQ(ierr);
41499836e0eSStefano Zampini       }
41599836e0eSStefano Zampini     }
41699836e0eSStefano Zampini   }
41799836e0eSStefano Zampini   pEnd[faceDepth] = face;
41899836e0eSStefano Zampini 
41999836e0eSStefano Zampini   /* Second pass for hybrid meshes: number hybrid faces */
42099836e0eSStefano Zampini   for (c = pMax; c < pEnd[cellDepth]; ++c) {
42199836e0eSStefano Zampini     const PetscInt *cellFaces, *cone;
42299836e0eSStefano Zampini     PetscInt        numCellFaces, numCellFacesN, faceSize, cf, coneSize;
42399836e0eSStefano Zampini 
42499836e0eSStefano Zampini     ierr = DMPlexGetConeSize(dm, c, &coneSize);CHKERRQ(ierr);
42599836e0eSStefano Zampini     ierr = DMPlexGetCone(dm, c, &cone);CHKERRQ(ierr);
42699836e0eSStefano Zampini     ierr = DMPlexGetRawFacesHybrid_Internal(dm, cellDim, coneSize, cone, &numCellFaces, &numCellFacesN, &faceSize, &cellFaces);CHKERRQ(ierr);
42799836e0eSStefano Zampini     if (numCellFaces != numCellFacesH) SETERRQ2(PETSC_COMM_SELF, PETSC_ERR_SUP, "Unexpected hybrid numCellFaces %D != %D", numCellFaces, numCellFacesH);
42899836e0eSStefano Zampini     faceSize = PetscMax(faceSize, -faceSize);
42999836e0eSStefano Zampini     for (cf = numCellFacesN; cf < numCellFaces; ++cf) { /* These are the hybrid faces */
43099836e0eSStefano Zampini       const PetscInt   *cellFace = &cellFaces[cf*faceSize];
43199836e0eSStefano Zampini       PetscHashIJKLKey  key;
43299836e0eSStefano Zampini       PetscHashIter     iter;
43399836e0eSStefano Zampini       PetscBool         missing;
43499836e0eSStefano Zampini       PetscInt          faceSizeH = faceSize;
43599836e0eSStefano Zampini 
43699836e0eSStefano Zampini       if (faceSize == 2) {
43799836e0eSStefano Zampini         key.i = PetscMin(cellFace[0], cellFace[1]);
43899836e0eSStefano Zampini         key.j = PetscMax(cellFace[0], cellFace[1]);
43999836e0eSStefano Zampini         key.k = PETSC_MAX_INT;
44099836e0eSStefano Zampini         key.l = PETSC_MAX_INT;
44199836e0eSStefano Zampini       } else {
44299836e0eSStefano Zampini         key.i = cellFace[0];
44399836e0eSStefano Zampini         key.j = cellFace[1];
44499836e0eSStefano Zampini         key.k = cellFace[2];
44599836e0eSStefano Zampini         key.l = faceSize > 3 ? (cellFace[3] < 0 ? faceSizeH = 3, PETSC_MAX_INT : cellFace[3]) : PETSC_MAX_INT;
44699836e0eSStefano Zampini         ierr  = PetscSortInt(faceSize, (PetscInt *) &key);CHKERRQ(ierr);
44799836e0eSStefano Zampini       }
44899836e0eSStefano Zampini       if (faceSizeH != faceSizeAllH) SETERRQ4(PETSC_COMM_SELF, PETSC_ERR_SUP, "Unexpected number of vertices for hybrid face %D of point %D -> %D != %D", cf, c, faceSizeH, faceSizeAllH);
44999836e0eSStefano Zampini       ierr = PetscHashIJKLPut(faceTable, key, &iter, &missing);CHKERRQ(ierr);
45099836e0eSStefano Zampini       if (missing) {ierr = PetscHashIJKLIterSet(faceTable, iter, face++);CHKERRQ(ierr);}
45199836e0eSStefano Zampini     }
45299836e0eSStefano Zampini     ierr = DMPlexRestoreRawFacesHybrid_Internal(dm, cellDim, coneSize, cone, &numCellFaces, &numCellFacesN, &faceSize, &cellFaces);CHKERRQ(ierr);
45399836e0eSStefano Zampini   }
45499836e0eSStefano Zampini   faceH = face - pEnd[faceDepth];
45599836e0eSStefano Zampini   if (faceH) {
45699836e0eSStefano Zampini     if (fMax == PETSC_DETERMINE) fMax = pEnd[faceDepth];
45799836e0eSStefano Zampini     else if (eMax == PETSC_DETERMINE) eMax = pEnd[faceDepth];
45899836e0eSStefano Zampini     else SETERRQ3(PETSC_COMM_SELF, PETSC_ERR_PLIB, "Number of unassigned hybrid facets %D for cellDim %D and dimension %D", faceH, cellDim, dim);
4599a5b13a2SMatthew G Knepley   }
4609a5b13a2SMatthew G Knepley   pEnd[faceDepth] = face;
4619a5b13a2SMatthew G Knepley   ierr = PetscHashIJKLDestroy(&faceTable);CHKERRQ(ierr);
4629a5b13a2SMatthew G Knepley   /* Count new points */
4639a5b13a2SMatthew G Knepley   for (d = 0; d <= depth; ++d) {
4649a5b13a2SMatthew G Knepley     numPoints += pEnd[d]-pStart[d];
4659a5b13a2SMatthew G Knepley   }
4669a5b13a2SMatthew G Knepley   ierr = DMPlexSetChart(idm, 0, numPoints);CHKERRQ(ierr);
4679a5b13a2SMatthew G Knepley   /* Set cone sizes */
4689a5b13a2SMatthew G Knepley   for (d = 0; d <= depth; ++d) {
4699a5b13a2SMatthew G Knepley     PetscInt coneSize, p;
4709f074e33SMatthew G Knepley 
4719a5b13a2SMatthew G Knepley     if (d == faceDepth) {
472e9fa77a1SMatthew G. Knepley       /* Now we have two cases: */
473e9fa77a1SMatthew G. Knepley       if (faceSizeAll == faceSizeAllT) {
4749a5b13a2SMatthew G Knepley         /* I see no way to do this if we admit faces of different shapes */
47599836e0eSStefano Zampini         for (p = pStart[d]; p < pEnd[d]-faceH; ++p) {
4769a5b13a2SMatthew G Knepley           ierr = DMPlexSetConeSize(idm, p, faceSizeAll);CHKERRQ(ierr);
4779a5b13a2SMatthew G Knepley         }
47899836e0eSStefano Zampini         for (p = pEnd[d]-faceH; p < pEnd[d]; ++p) {
47999836e0eSStefano Zampini           ierr = DMPlexSetConeSize(idm, p, faceSizeAllH);CHKERRQ(ierr);
48099836e0eSStefano Zampini         }
481e9fa77a1SMatthew G. Knepley       } else if (faceSizeAll == faceSizeAllH) {
482e9fa77a1SMatthew G. Knepley         for (p = pStart[d]; p < pStart[d]+faceT; ++p) {
483e9fa77a1SMatthew G. Knepley           ierr = DMPlexSetConeSize(idm, p, faceSizeAllT);CHKERRQ(ierr);
484e9fa77a1SMatthew G. Knepley         }
485e9fa77a1SMatthew G. Knepley         for (p = pStart[d]+faceT; p < pEnd[d]-faceH; ++p) {
486e9fa77a1SMatthew G. Knepley           ierr = DMPlexSetConeSize(idm, p, faceSizeAll);CHKERRQ(ierr);
487e9fa77a1SMatthew G. Knepley         }
488e9fa77a1SMatthew G. Knepley         for (p = pEnd[d]-faceH; p < pEnd[d]; ++p) {
489e9fa77a1SMatthew G. Knepley           ierr = DMPlexSetConeSize(idm, p, faceSizeAllH);CHKERRQ(ierr);
490e9fa77a1SMatthew G. Knepley         }
491e9fa77a1SMatthew G. Knepley       } else SETERRQ3(PETSC_COMM_SELF, PETSC_ERR_PLIB, "Inconsistent faces sizes N: %D T: %D H: %D", faceSizeAll, faceSizeAllT, faceSizeAllH);
492a014044eSMatthew G Knepley     } else if (d == cellDepth) {
493a014044eSMatthew G Knepley       for (p = pStart[d]; p < pEnd[d]; ++p) {
494a014044eSMatthew G Knepley         /* Number of cell faces may be different from number of cell vertices*/
49599836e0eSStefano Zampini         if (p < pMax) {
496a014044eSMatthew G Knepley           ierr = DMPlexGetFaces_Internal(dm, cellDim, p, &coneSize, NULL, NULL);CHKERRQ(ierr);
49799836e0eSStefano Zampini         } else {
49899836e0eSStefano Zampini           ierr = DMPlexGetFacesHybrid_Internal(dm, cellDim, p, &coneSize, NULL, NULL, NULL);CHKERRQ(ierr);
49999836e0eSStefano Zampini         }
500a014044eSMatthew G Knepley         ierr = DMPlexSetConeSize(idm, p, coneSize);CHKERRQ(ierr);
501a014044eSMatthew G Knepley       }
5029a5b13a2SMatthew G Knepley     } else {
5039a5b13a2SMatthew G Knepley       for (p = pStart[d]; p < pEnd[d]; ++p) {
5049a5b13a2SMatthew G Knepley         ierr = DMPlexGetConeSize(dm, p, &coneSize);CHKERRQ(ierr);
5059a5b13a2SMatthew G Knepley         ierr = DMPlexSetConeSize(idm, p, coneSize);CHKERRQ(ierr);
5069f074e33SMatthew G Knepley       }
5079f074e33SMatthew G Knepley     }
5089f074e33SMatthew G Knepley   }
5099f074e33SMatthew G Knepley   ierr = DMSetUp(idm);CHKERRQ(ierr);
5109f074e33SMatthew G Knepley   /* Get face cones from subsets of cell vertices */
51199836e0eSStefano Zampini   if (faceSizeAll > 4) SETERRQ1(PETSC_COMM_SELF, PETSC_ERR_ARG_WRONG, "Do not support interpolation of meshes with faces of %D vertices", faceSizeAll);
5129a5b13a2SMatthew G Knepley   ierr = PetscHashIJKLCreate(&faceTable);CHKERRQ(ierr);
5139a5b13a2SMatthew G Knepley   for (d = depth; d > cellDepth; --d) {
5149a5b13a2SMatthew G Knepley     const PetscInt *cone;
5159a5b13a2SMatthew G Knepley     PetscInt        p;
5169a5b13a2SMatthew G Knepley 
5179a5b13a2SMatthew G Knepley     for (p = pStart[d]; p < pEnd[d]; ++p) {
5189a5b13a2SMatthew G Knepley       ierr = DMPlexGetCone(dm, p, &cone);CHKERRQ(ierr);
5199a5b13a2SMatthew G Knepley       ierr = DMPlexSetCone(idm, p, cone);CHKERRQ(ierr);
5209a5b13a2SMatthew G Knepley       ierr = DMPlexGetConeOrientation(dm, p, &cone);CHKERRQ(ierr);
5219a5b13a2SMatthew G Knepley       ierr = DMPlexSetConeOrientation(idm, p, cone);CHKERRQ(ierr);
5229f074e33SMatthew G Knepley     }
5239a5b13a2SMatthew G Knepley   }
52499836e0eSStefano Zampini   for (outerloop = 0, face = pStart[faceDepth]; outerloop < 2; outerloop++) {
52599836e0eSStefano Zampini     PetscInt start, end;
5269f074e33SMatthew G Knepley 
52799836e0eSStefano Zampini     start = outerloop == 0 ? pMax : pStart[cellDepth];
52899836e0eSStefano Zampini     end = outerloop == 0 ? pEnd[cellDepth] : pMax;
52999836e0eSStefano Zampini     for (c = start; c < end; ++c) {
53099836e0eSStefano Zampini       const PetscInt *cellFaces;
53199836e0eSStefano Zampini       PetscInt        numCellFaces, faceSize, faceSizeInc, cf;
53299836e0eSStefano Zampini 
53399836e0eSStefano Zampini       if (c < pMax) {
5349a5b13a2SMatthew G Knepley         ierr = DMPlexGetFaces_Internal(dm, cellDim, c, &numCellFaces, &faceSize, &cellFaces);CHKERRQ(ierr);
5359a5b13a2SMatthew G Knepley         if (faceSize != faceSizeAll) SETERRQ3(PETSC_COMM_SELF, PETSC_ERR_PLIB, "Inconsistent face for cell %D of size %D != %D", c, faceSize, faceSizeAll);
53699836e0eSStefano Zampini       } else {
53799836e0eSStefano Zampini         const PetscInt *cone;
53899836e0eSStefano Zampini         PetscInt        numCellFacesN, coneSize;
53999836e0eSStefano Zampini 
54099836e0eSStefano Zampini         ierr = DMPlexGetConeSize(dm, c, &coneSize);CHKERRQ(ierr);
54199836e0eSStefano Zampini         if (coneSize != coneSizeH) SETERRQ2(PETSC_COMM_SELF, PETSC_ERR_SUP, "Unexpected hybrid coneSize %D != %D", coneSize, coneSizeH);
54299836e0eSStefano Zampini         ierr = DMPlexGetCone(dm, c, &cone);CHKERRQ(ierr);
54399836e0eSStefano Zampini         ierr = DMPlexGetRawFacesHybrid_Internal(dm, cellDim, coneSize, cone, &numCellFaces, &numCellFacesN, &faceSize, &cellFaces);CHKERRQ(ierr);
54499836e0eSStefano Zampini         if (numCellFaces != numCellFacesH) SETERRQ3(PETSC_COMM_SELF, PETSC_ERR_SUP, "Unexpected numCellFaces %D != %D for hybrid cell %D", numCellFaces, numCellFacesH, c);
54599836e0eSStefano Zampini         faceSize = PetscMax(faceSize, -faceSize);
54699836e0eSStefano Zampini         if (faceSize > 4) SETERRQ1(PETSC_COMM_SELF, PETSC_ERR_ARG_WRONG, "Do not support interpolation of meshes with faces of %D vertices", faceSize);
54799836e0eSStefano Zampini         numCellFaces = numCellFacesN; /* process only non-hybrid faces */
54899836e0eSStefano Zampini       }
54999836e0eSStefano Zampini       faceSizeInc = faceSize;
5509f074e33SMatthew G Knepley       for (cf = 0; cf < numCellFaces; ++cf) {
55199836e0eSStefano Zampini         const PetscInt  *cellFace = &cellFaces[cf*faceSizeInc];
5529a5b13a2SMatthew G Knepley         PetscHashIJKLKey key;
553e8f14785SLisandro Dalcin         PetscHashIter    iter;
554e8f14785SLisandro Dalcin         PetscBool        missing;
5559f074e33SMatthew G Knepley 
55699836e0eSStefano Zampini         if (faceSizeInc == 2) {
5579a5b13a2SMatthew G Knepley           key.i = PetscMin(cellFace[0], cellFace[1]);
5589a5b13a2SMatthew G Knepley           key.j = PetscMax(cellFace[0], cellFace[1]);
55999836e0eSStefano Zampini           key.k = PETSC_MAX_INT;
56099836e0eSStefano Zampini           key.l = PETSC_MAX_INT;
5619a5b13a2SMatthew G Knepley         } else {
562e8f14785SLisandro Dalcin           key.i = cellFace[0];
563e8f14785SLisandro Dalcin           key.j = cellFace[1];
564e8f14785SLisandro Dalcin           key.k = cellFace[2];
56599836e0eSStefano Zampini           key.l = faceSizeInc > 3 ? (cellFace[3] < 0 ? faceSize = 3, PETSC_MAX_INT : cellFace[3]) : PETSC_MAX_INT;
56699836e0eSStefano Zampini           ierr  = PetscSortInt(faceSizeInc, (PetscInt *) &key);CHKERRQ(ierr);
5679f074e33SMatthew G Knepley         }
568e8f14785SLisandro Dalcin         ierr = PetscHashIJKLPut(faceTable, key, &iter, &missing);CHKERRQ(ierr);
569735a0255SMatthew G. Knepley         if (missing) {
5709a5b13a2SMatthew G Knepley           ierr = DMPlexSetCone(idm, face, cellFace);CHKERRQ(ierr);
571e8f14785SLisandro Dalcin           ierr = PetscHashIJKLIterSet(faceTable, iter, face);CHKERRQ(ierr);
572735a0255SMatthew G. Knepley           ierr = DMPlexInsertCone(idm, c, cf, face++);CHKERRQ(ierr);
5739a5b13a2SMatthew G Knepley         } else {
5749a5b13a2SMatthew G Knepley           const PetscInt *cone;
575735a0255SMatthew G. Knepley           PetscInt        coneSize, ornt, i, j, f;
5769f074e33SMatthew G Knepley 
577e8f14785SLisandro Dalcin           ierr = PetscHashIJKLIterGet(faceTable, iter, &f);CHKERRQ(ierr);
5789a5b13a2SMatthew G Knepley           ierr = DMPlexInsertCone(idm, c, cf, f);CHKERRQ(ierr);
5792e1b13c2SMatthew G. Knepley           /* Orient face: Do not allow reverse orientation at the first vertex */
5809f074e33SMatthew G Knepley           ierr = DMPlexGetConeSize(idm, f, &coneSize);CHKERRQ(ierr);
5819f074e33SMatthew G Knepley           ierr = DMPlexGetCone(idm, f, &cone);CHKERRQ(ierr);
5829a5b13a2SMatthew G Knepley           if (coneSize != faceSize) SETERRQ3(PETSC_COMM_SELF, PETSC_ERR_PLIB, "Invalid number of face vertices %D for face %D should be %D", coneSize, f, faceSize);
5839a5b13a2SMatthew G Knepley           /* - First find the initial vertex */
5849a5b13a2SMatthew G Knepley           for (i = 0; i < faceSize; ++i) if (cellFace[0] == cone[i]) break;
5859a5b13a2SMatthew G Knepley           /* - Try forward comparison */
5869a5b13a2SMatthew G Knepley           for (j = 0; j < faceSize; ++j) if (cellFace[j] != cone[(i+j)%faceSize]) break;
5879a5b13a2SMatthew G Knepley           if (j == faceSize) {
5889a5b13a2SMatthew G Knepley             if ((faceSize == 2) && (i == 1)) ornt = -2;
5899a5b13a2SMatthew G Knepley             else                             ornt = i;
5909a5b13a2SMatthew G Knepley           } else {
5919a5b13a2SMatthew G Knepley             /* - Try backward comparison */
5929a5b13a2SMatthew G Knepley             for (j = 0; j < faceSize; ++j) if (cellFace[j] != cone[(i+faceSize-j)%faceSize]) break;
5932e1b13c2SMatthew G. Knepley             if (j == faceSize) {
5942e1b13c2SMatthew G. Knepley               if (i == 0) ornt = -faceSize;
595dc1a705cSMatthew G. Knepley               else        ornt = -i;
596e9fa77a1SMatthew G. Knepley             } else SETERRQ2(PETSC_COMM_SELF, PETSC_ERR_PLIB, "Could not determine orientation of face %D in cell %D", f, c);
5979a5b13a2SMatthew G Knepley           }
5989a5b13a2SMatthew G Knepley           ierr = DMPlexInsertConeOrientation(idm, c, cf, ornt);CHKERRQ(ierr);
5999f074e33SMatthew G Knepley         }
6009f074e33SMatthew G Knepley       }
60199836e0eSStefano Zampini       if (c < pMax) {
602439ece16SMatthew G. Knepley         ierr = DMPlexRestoreFaces_Internal(dm, cellDim, c, &numCellFaces, &faceSize, &cellFaces);CHKERRQ(ierr);
60399836e0eSStefano Zampini       } else {
60499836e0eSStefano Zampini         ierr = DMPlexRestoreRawFacesHybrid_Internal(dm, cellDim, coneSizeH, NULL, NULL, NULL, NULL, &cellFaces);CHKERRQ(ierr);
6059f074e33SMatthew G Knepley       }
60699836e0eSStefano Zampini     }
60799836e0eSStefano Zampini   }
60899836e0eSStefano Zampini   /* Second pass for hybrid meshes: orient hybrid faces */
60999836e0eSStefano Zampini   for (c = pMax; c < pEnd[cellDepth]; ++c) {
61099836e0eSStefano Zampini     const PetscInt *cellFaces, *cone;
61199836e0eSStefano Zampini     PetscInt        numCellFaces, numCellFacesN, faceSize, cf, coneSize;
61299836e0eSStefano Zampini 
61399836e0eSStefano Zampini     ierr = DMPlexGetConeSize(dm, c, &coneSize);CHKERRQ(ierr);
61499836e0eSStefano Zampini     ierr = DMPlexGetCone(dm, c, &cone);CHKERRQ(ierr);
61599836e0eSStefano Zampini     ierr = DMPlexGetRawFacesHybrid_Internal(dm, cellDim, coneSize, cone, &numCellFaces, &numCellFacesN, &faceSize, &cellFaces);CHKERRQ(ierr);
61699836e0eSStefano Zampini     if (numCellFaces != numCellFacesH) SETERRQ2(PETSC_COMM_SELF, PETSC_ERR_SUP, "Unexpected hybrid numCellFaces %D != %D", numCellFaces, numCellFacesH);
61799836e0eSStefano Zampini     faceSize = PetscMax(faceSize, -faceSize);
61899836e0eSStefano Zampini     for (cf = numCellFacesN; cf < numCellFaces; ++cf) { /* These are the hybrid faces */
61999836e0eSStefano Zampini       const PetscInt   *cellFace = &cellFaces[cf*faceSize];
62099836e0eSStefano Zampini       PetscHashIJKLKey key;
62199836e0eSStefano Zampini       PetscHashIter    iter;
62299836e0eSStefano Zampini       PetscBool        missing;
62399836e0eSStefano Zampini       PetscInt         faceSizeH = faceSize;
62499836e0eSStefano Zampini 
62599836e0eSStefano Zampini       if (faceSize == 2) {
62699836e0eSStefano Zampini         key.i = PetscMin(cellFace[0], cellFace[1]);
62799836e0eSStefano Zampini         key.j = PetscMax(cellFace[0], cellFace[1]);
62899836e0eSStefano Zampini         key.k = PETSC_MAX_INT;
62999836e0eSStefano Zampini         key.l = PETSC_MAX_INT;
63099836e0eSStefano Zampini       } else {
63199836e0eSStefano Zampini         key.i = cellFace[0];
63299836e0eSStefano Zampini         key.j = cellFace[1];
63399836e0eSStefano Zampini         key.k = cellFace[2];
63499836e0eSStefano Zampini         key.l = faceSize > 3 ? (cellFace[3] < 0 ? faceSizeH = 3, PETSC_MAX_INT : cellFace[3]) : PETSC_MAX_INT;
63599836e0eSStefano Zampini         ierr  = PetscSortInt(faceSize, (PetscInt *) &key);CHKERRQ(ierr);
63699836e0eSStefano Zampini       }
63799836e0eSStefano Zampini       if (faceSizeH != faceSizeAllH) SETERRQ4(PETSC_COMM_SELF, PETSC_ERR_SUP, "Unexpected number of vertices for hybrid face %D of point %D -> %D != %D", cf, c, faceSizeH, faceSizeAllH);
63899836e0eSStefano Zampini       ierr = PetscHashIJKLPut(faceTable, key, &iter, &missing);CHKERRQ(ierr);
63999836e0eSStefano Zampini       if (missing) {
64099836e0eSStefano Zampini         ierr = DMPlexSetCone(idm, face, cellFace);CHKERRQ(ierr);
64199836e0eSStefano Zampini         ierr = PetscHashIJKLIterSet(faceTable, iter, face);CHKERRQ(ierr);
64299836e0eSStefano Zampini         ierr = DMPlexInsertCone(idm, c, cf, face++);CHKERRQ(ierr);
64399836e0eSStefano Zampini       } else {
644e9fa77a1SMatthew G. Knepley         PetscInt        fv[4] = {0, 1, 2, 3};
64599836e0eSStefano Zampini         const PetscInt *cone;
64699836e0eSStefano Zampini         PetscInt        coneSize, ornt, i, j, f;
647a74221b0SStefano Zampini         PetscBool       q2h = PETSC_FALSE;
64899836e0eSStefano Zampini 
64999836e0eSStefano Zampini         ierr = PetscHashIJKLIterGet(faceTable, iter, &f);CHKERRQ(ierr);
65099836e0eSStefano Zampini         ierr = DMPlexInsertCone(idm, c, cf, f);CHKERRQ(ierr);
65199836e0eSStefano Zampini         /* Orient face: Do not allow reverse orientation at the first vertex */
65299836e0eSStefano Zampini         ierr = DMPlexGetConeSize(idm, f, &coneSize);CHKERRQ(ierr);
65399836e0eSStefano Zampini         ierr = DMPlexGetCone(idm, f, &cone);CHKERRQ(ierr);
65499836e0eSStefano Zampini         if (coneSize != faceSizeH) SETERRQ3(PETSC_COMM_SELF, PETSC_ERR_PLIB, "Invalid number of face vertices %D for face %D should be %D", coneSize, f, faceSizeH);
655e9fa77a1SMatthew G. Knepley         /* Hybrid faces are stored as tensor products of edges, so to compare them to normal faces, we have to flip */
656a74221b0SStefano Zampini         if (faceSize == 4 && c >= pMax && faceSizeAll != faceSizeAllT && f < pEnd[faceDepth] - faceH) {q2h = PETSC_TRUE; fv[2] = 3; fv[3] = 2;}
65799836e0eSStefano Zampini         /* - First find the initial vertex */
658e9fa77a1SMatthew G. Knepley         for (i = 0; i < faceSizeH; ++i) if (cellFace[fv[0]] == cone[i]) break;
659a74221b0SStefano Zampini         if (q2h) { /* Matt's case: hybrid faces meeting with non-hybrid faces. This is a case that is not (and will not be) supported in general by the refinements */
66099836e0eSStefano Zampini           /* - Try forward comparison */
661e9fa77a1SMatthew G. Knepley           for (j = 0; j < faceSizeH; ++j) if (cellFace[fv[j]] != cone[(i+j)%faceSizeH]) break;
66299836e0eSStefano Zampini           if (j == faceSizeH) {
66399836e0eSStefano Zampini             if ((faceSizeH == 2) && (i == 1)) ornt = -2;
66499836e0eSStefano Zampini             else                              ornt = i;
66599836e0eSStefano Zampini           } else {
66699836e0eSStefano Zampini             /* - Try backward comparison */
667e9fa77a1SMatthew G. Knepley             for (j = 0; j < faceSizeH; ++j) if (cellFace[fv[j]] != cone[(i+faceSizeH-j)%faceSizeH]) break;
66899836e0eSStefano Zampini             if (j == faceSizeH) {
66999836e0eSStefano Zampini               if (i == 0) ornt = -faceSizeH;
67099836e0eSStefano Zampini               else        ornt = -i;
671e9fa77a1SMatthew G. Knepley             } else SETERRQ2(PETSC_COMM_SELF, PETSC_ERR_PLIB, "Could not determine orientation of face %D in cell %D", f, c);
67299836e0eSStefano Zampini           }
673a74221b0SStefano Zampini         } else {
674a74221b0SStefano Zampini           /* when matching hybrid faces in 3D, only few cases are possible.
675a74221b0SStefano Zampini              Face traversal however can no longer follow the usual convention, this seems a serious issue to me */
676a74221b0SStefano Zampini           PetscInt tquad_map[4][4] = { {PETSC_MIN_INT,            0,PETSC_MIN_INT,PETSC_MIN_INT},
677a74221b0SStefano Zampini                                        {           -1,PETSC_MIN_INT,PETSC_MIN_INT,PETSC_MIN_INT},
678a74221b0SStefano Zampini                                        {PETSC_MIN_INT,PETSC_MIN_INT,PETSC_MIN_INT,            1},
679a74221b0SStefano Zampini                                        {PETSC_MIN_INT,PETSC_MIN_INT,           -2,PETSC_MIN_INT} };
680a74221b0SStefano Zampini           PetscInt i2;
681a74221b0SStefano Zampini 
682a74221b0SStefano Zampini           /* find the second vertex */
683a74221b0SStefano Zampini           for (i2 = 0; i2 < faceSizeH; ++i2) if (cellFace[fv[1]] == cone[i2]) break;
684a74221b0SStefano Zampini           switch (faceSizeH) {
685a74221b0SStefano Zampini           case 2:
686a74221b0SStefano Zampini             ornt = i ? -2 : 0;
687a74221b0SStefano Zampini             break;
688a74221b0SStefano Zampini           case 4:
689a74221b0SStefano Zampini             ornt = tquad_map[i][i2];
690a74221b0SStefano Zampini             break;
691a74221b0SStefano Zampini           default:
692a74221b0SStefano Zampini             SETERRQ3(PETSC_COMM_SELF, PETSC_ERR_PLIB, "Unhandled face size %D for face %D in cell %D", faceSizeH, f, c);
693a74221b0SStefano Zampini 
694a74221b0SStefano Zampini           }
695a74221b0SStefano Zampini         }
69699836e0eSStefano Zampini         ierr = DMPlexInsertConeOrientation(idm, c, cf, ornt);CHKERRQ(ierr);
69799836e0eSStefano Zampini       }
69899836e0eSStefano Zampini     }
69999836e0eSStefano Zampini     ierr = DMPlexRestoreRawFacesHybrid_Internal(dm, cellDim, coneSize, cone, &numCellFaces, &numCellFacesN, &faceSize, &cellFaces);CHKERRQ(ierr);
70099836e0eSStefano Zampini   }
70199836e0eSStefano Zampini   if (face != pEnd[faceDepth]) SETERRQ2(PETSC_COMM_SELF, PETSC_ERR_PLIB, "Invalid number of faces %D should be %D", face-pStart[faceDepth], pEnd[faceDepth]-pStart[faceDepth]);
702c907b753SJed Brown   ierr = PetscFree2(pStart,pEnd);CHKERRQ(ierr);
7039a5b13a2SMatthew G Knepley   ierr = PetscHashIJKLDestroy(&faceTable);CHKERRQ(ierr);
7046551a8c7SMatthew G. Knepley   ierr = PetscFree2(pStart,pEnd);CHKERRQ(ierr);
70599836e0eSStefano Zampini   ierr = DMPlexSetHybridBounds(idm, cMax, fMax, eMax, vMax);CHKERRQ(ierr);
7069a5b13a2SMatthew G Knepley   ierr = DMPlexSymmetrize(idm);CHKERRQ(ierr);
7079a5b13a2SMatthew G Knepley   ierr = DMPlexStratify(idm);CHKERRQ(ierr);
7089f074e33SMatthew G Knepley   PetscFunctionReturn(0);
7099f074e33SMatthew G Knepley }
7109f074e33SMatthew G Knepley 
711f80536cbSVaclav Hapla static PetscErrorCode SortRmineRremoteByRemote_Private(PetscSF sf, PetscInt *rmine1[], PetscInt *rremote1[])
712f80536cbSVaclav Hapla {
713f80536cbSVaclav Hapla   PetscInt            nleaves;
714f80536cbSVaclav Hapla   PetscInt            nranks;
715a0d42dbcSVaclav Hapla   const PetscMPIInt  *ranks=NULL;
716a0d42dbcSVaclav Hapla   const PetscInt     *roffset=NULL, *rmine=NULL, *rremote=NULL;
717f80536cbSVaclav Hapla   PetscInt            n, o, r;
718f80536cbSVaclav Hapla   PetscErrorCode      ierr;
719f80536cbSVaclav Hapla 
720f80536cbSVaclav Hapla   PetscFunctionBegin;
721dec1416fSJunchao Zhang   ierr = PetscSFGetRootRanks(sf, &nranks, &ranks, &roffset, &rmine, &rremote);CHKERRQ(ierr);
722f80536cbSVaclav Hapla   nleaves = roffset[nranks];
723f80536cbSVaclav Hapla   ierr = PetscMalloc2(nleaves, rmine1, nleaves, rremote1);CHKERRQ(ierr);
724f80536cbSVaclav Hapla   for (r=0; r<nranks; r++) {
725f80536cbSVaclav Hapla     /* simultaneously sort rank-wise portions of rmine & rremote by values in rremote
726f80536cbSVaclav Hapla        - to unify order with the other side */
727f80536cbSVaclav Hapla     o = roffset[r];
728f80536cbSVaclav Hapla     n = roffset[r+1] - o;
729580bdb30SBarry Smith     ierr = PetscArraycpy(&(*rmine1)[o], &rmine[o], n);CHKERRQ(ierr);
730580bdb30SBarry Smith     ierr = PetscArraycpy(&(*rremote1)[o], &rremote[o], n);CHKERRQ(ierr);
731f80536cbSVaclav Hapla     ierr = PetscSortIntWithArray(n, &(*rremote1)[o], &(*rmine1)[o]);CHKERRQ(ierr);
732f80536cbSVaclav Hapla   }
733f80536cbSVaclav Hapla   PetscFunctionReturn(0);
734f80536cbSVaclav Hapla }
735f80536cbSVaclav Hapla 
736*27d0e99aSVaclav Hapla PetscErrorCode DMPlexOrientInterface_Internal(DM dm)
7372e745bdaSMatthew G. Knepley {
738*27d0e99aSVaclav Hapla   /* Here we only compare first 2 points of the cone. Full cone size would lead to stronger self-checking. */
739*27d0e99aSVaclav Hapla   PetscInt          masterCone[2];
740cae7fe92SVaclav Hapla   PetscInt          (*roots)[2], (*leaves)[2];
7418a650c75SVaclav Hapla   PetscMPIInt       (*rootsRanks)[2], (*leavesRanks)[2];
742*27d0e99aSVaclav Hapla 
743*27d0e99aSVaclav Hapla   PetscSF           sf=NULL;
744a0d42dbcSVaclav Hapla   const PetscInt    *locals=NULL;
745a0d42dbcSVaclav Hapla   const PetscSFNode *remotes=NULL;
7468a650c75SVaclav Hapla   PetscInt           nroots, nleaves, p, c;
747f80536cbSVaclav Hapla   PetscInt           nranks, n, o, r;
748a0d42dbcSVaclav Hapla   const PetscMPIInt *ranks=NULL;
749a0d42dbcSVaclav Hapla   const PetscInt    *roffset=NULL;
750a0d42dbcSVaclav Hapla   PetscInt          *rmine1=NULL, *rremote1=NULL; /* rmine and rremote copies simultaneously sorted by rank and rremote */
751a0d42dbcSVaclav Hapla   const PetscInt    *cone=NULL;
752adeface4SVaclav Hapla   PetscInt           coneSize, ind0;
7532e745bdaSMatthew G. Knepley   MPI_Comm           comm;
754*27d0e99aSVaclav Hapla   PetscMPIInt        rank,size;
7552e745bdaSMatthew G. Knepley   PetscInt           debug = 0;
7562e745bdaSMatthew G. Knepley   PetscErrorCode     ierr;
7572e745bdaSMatthew G. Knepley 
7582e745bdaSMatthew G. Knepley   PetscFunctionBegin;
759df6a9fadSVaclav Hapla   ierr = DMGetPointSF(dm, &sf);CHKERRQ(ierr);
7602e745bdaSMatthew G. Knepley   ierr = PetscSFGetGraph(sf, &nroots, &nleaves, &locals, &remotes);CHKERRQ(ierr);
7613ede9f65SMatthew G. Knepley   if (nroots < 0) PetscFunctionReturn(0);
762f80536cbSVaclav Hapla   ierr = PetscSFSetUp(sf);CHKERRQ(ierr);
763dec1416fSJunchao Zhang   ierr = PetscSFGetRootRanks(sf, &nranks, &ranks, &roffset, NULL, NULL);CHKERRQ(ierr);
764dc21a0bfSVaclav Hapla   ierr = DMViewFromOptions(dm, NULL, "-before_fix_dm_view");CHKERRQ(ierr);
765*27d0e99aSVaclav Hapla #if defined(PETSC_USE_DEBUG)
766dc21a0bfSVaclav Hapla   ierr = DMPlexCheckPointSF(dm);CHKERRQ(ierr);
767dc21a0bfSVaclav Hapla #endif
768f80536cbSVaclav Hapla   ierr = SortRmineRremoteByRemote_Private(sf, &rmine1, &rremote1);CHKERRQ(ierr);
7698a650c75SVaclav Hapla   ierr = PetscMalloc4(nroots, &roots, nroots, &leaves, nroots, &rootsRanks, nroots, &leavesRanks);CHKERRQ(ierr);
7702e745bdaSMatthew G. Knepley   ierr = PetscObjectGetComm((PetscObject) dm, &comm);CHKERRQ(ierr);
7712e745bdaSMatthew G. Knepley   ierr = MPI_Comm_rank(comm, &rank);CHKERRQ(ierr);
772*27d0e99aSVaclav Hapla   ierr = MPI_Comm_size(comm, &size);CHKERRQ(ierr);
7739e24d8a0SVaclav Hapla   for (p = 0; p < nroots; ++p) {
7749e24d8a0SVaclav Hapla     ierr = DMPlexGetConeSize(dm, p, &coneSize);CHKERRQ(ierr);
7759e24d8a0SVaclav Hapla     ierr = DMPlexGetCone(dm, p, &cone);CHKERRQ(ierr);
776*27d0e99aSVaclav Hapla     if (coneSize < 2) {
777*27d0e99aSVaclav Hapla       for (c = 0; c < 2; c++) {
778*27d0e99aSVaclav Hapla         roots[p][c] = -1;
779*27d0e99aSVaclav Hapla         rootsRanks[p][c] = -1;
780*27d0e99aSVaclav Hapla       }
781*27d0e99aSVaclav Hapla       continue;
782*27d0e99aSVaclav Hapla     }
7832e745bdaSMatthew G. Knepley     /* Translate all points to root numbering */
7848a650c75SVaclav Hapla     for (c = 0; c < 2; c++) {
7858a650c75SVaclav Hapla       ierr = PetscFindInt(cone[c], nleaves, locals, &ind0);CHKERRQ(ierr);
7868a650c75SVaclav Hapla       if (ind0 < 0) {
7878a650c75SVaclav Hapla         roots[p][c] = cone[c];
7888a650c75SVaclav Hapla         rootsRanks[p][c] = rank;
789c8148b97SVaclav Hapla       } else {
7908a650c75SVaclav Hapla         roots[p][c] = remotes[ind0].index;
7918a650c75SVaclav Hapla         rootsRanks[p][c] = remotes[ind0].rank;
7928a650c75SVaclav Hapla       }
7932e745bdaSMatthew G. Knepley     }
7942e745bdaSMatthew G. Knepley   }
7959e24d8a0SVaclav Hapla   for (p = 0; p < nroots; ++p) {
7968ccb905dSVaclav Hapla     for (c = 0; c < 2; c++) {
7978ccb905dSVaclav Hapla       leaves[p][c] = -2;
7988ccb905dSVaclav Hapla       leavesRanks[p][c] = -2;
7998ccb905dSVaclav Hapla     }
800c8148b97SVaclav Hapla   }
8012e745bdaSMatthew G. Knepley   ierr = PetscSFBcastBegin(sf, MPIU_2INT, roots, leaves);CHKERRQ(ierr);
8028a650c75SVaclav Hapla   ierr = PetscSFBcastBegin(sf, MPI_2INT, rootsRanks, leavesRanks);CHKERRQ(ierr);
8032e745bdaSMatthew G. Knepley   ierr = PetscSFBcastEnd(sf, MPIU_2INT, roots, leaves);CHKERRQ(ierr);
8048a650c75SVaclav Hapla   ierr = PetscSFBcastEnd(sf, MPI_2INT, rootsRanks, leavesRanks);CHKERRQ(ierr);
805c8148b97SVaclav Hapla   if (debug) {ierr = PetscSynchronizedFlush(comm, NULL);CHKERRQ(ierr);}
806*27d0e99aSVaclav Hapla   if (debug && rank == 0) {ierr = PetscSynchronizedPrintf(comm, "Referenced roots\n");CHKERRQ(ierr);}
8079e24d8a0SVaclav Hapla   for (p = 0; p < nroots; ++p) {
8089e24d8a0SVaclav Hapla     if (leaves[p][0] < 0) continue;
8099e24d8a0SVaclav Hapla     ierr = DMPlexGetConeSize(dm, p, &coneSize);CHKERRQ(ierr);
8109e24d8a0SVaclav Hapla     ierr = DMPlexGetCone(dm, p, &cone);CHKERRQ(ierr);
811*27d0e99aSVaclav Hapla     if (debug) {ierr = PetscSynchronizedPrintf(comm, "[%d]  %4D: cone=[%4D %4D] roots=[(%d,%4D) (%d,%4D)] leaves=[(%d,%4D) (%d,%4D)]", rank, p, cone[0], cone[1], rootsRanks[p][0], roots[p][0], rootsRanks[p][1], roots[p][1], leavesRanks[p][0], leaves[p][0], leavesRanks[p][1], leaves[p][1]);CHKERRQ(ierr);}
81282f5c0aeSVaclav Hapla     if ((leaves[p][0] != roots[p][0]) || (leaves[p][1] != roots[p][1]) || (leavesRanks[p][0] != rootsRanks[p][0]) || (leavesRanks[p][1] != rootsRanks[p][1])) {
813*27d0e99aSVaclav Hapla       /* Translate these two leaves to my cone points; masterCone means desired order p's cone points */
814f80536cbSVaclav Hapla       for (c = 0; c < 2; c++) {
815*27d0e99aSVaclav Hapla         if (leavesRanks[p][c] == rank) {
816*27d0e99aSVaclav Hapla           /* A local leave is just taken as it is */
817*27d0e99aSVaclav Hapla           masterCone[c] = leaves[p][c];
818*27d0e99aSVaclav Hapla           continue;
819*27d0e99aSVaclav Hapla         }
820f80536cbSVaclav Hapla         /* Find index of rank leavesRanks[p][c] among remote ranks */
821f80536cbSVaclav Hapla         /* No need for PetscMPIIntCast because these integers were originally cast from PetscMPIInt. */
822f80536cbSVaclav Hapla         ierr = PetscFindMPIInt((PetscMPIInt)leavesRanks[p][c], nranks, ranks, &r);CHKERRQ(ierr);
823*27d0e99aSVaclav Hapla         if (PetscUnlikely(r < 0)) SETERRQ7(PETSC_COMM_SELF, PETSC_ERR_PLIB, "Point %D cone[%D]=%D root (%d,%D) leave (%d,%D): leave rank not found among remote ranks",p,c,cone[c],rootsRanks[p][c],roots[p][c],leavesRanks[p][c],leaves[p][c]);
824*27d0e99aSVaclav Hapla         if (PetscUnlikely(ranks[r] < 0 || ranks[r] >= size)) SETERRQ5(PETSC_COMM_SELF, PETSC_ERR_PLIB, "p=%D c=%D commsize=%d: ranks[%D] = %d makes no sense",p,c,size,r,ranks[r]);
825f80536cbSVaclav Hapla         /* Find point leaves[p][c] among remote points aimed at rank leavesRanks[p][c] */
826f80536cbSVaclav Hapla         o = roffset[r];
827f80536cbSVaclav Hapla         n = roffset[r+1] - o;
828f80536cbSVaclav Hapla         ierr = PetscFindInt(leaves[p][c], n, &rremote1[o], &ind0);CHKERRQ(ierr);
829*27d0e99aSVaclav Hapla         if (PetscUnlikely(ind0 < 0)) SETERRQ7(PETSC_COMM_SELF, PETSC_ERR_PLIB, "Point %D cone[%D]=%D root (%d,%D) leave (%d,%D): corresponding remote point not found - it seems there is missing connection in point SF!",p,c,cone[c],rootsRanks[p][c],roots[p][c],leavesRanks[p][c],leaves[p][c]);
830f80536cbSVaclav Hapla         /* Get the corresponding local point */
831f80536cbSVaclav Hapla         masterCone[c] = rmine1[o+ind0];CHKERRQ(ierr);
832f80536cbSVaclav Hapla       }
833*27d0e99aSVaclav Hapla       if (debug) {ierr = PetscSynchronizedPrintf(comm, " masterCone=[%4D %4D]\n", masterCone[0], masterCone[1]);CHKERRQ(ierr);}
834*27d0e99aSVaclav Hapla       /* Set the desired order of p's cone points and fix orientations accordingly */
835f80536cbSVaclav Hapla       /* Vaclav's note: Here we only compare first 2 points of the cone. Full cone size would lead to stronger self-checking. */
836f80536cbSVaclav Hapla       ierr = DMPlexOrientCell(dm, p, 2, masterCone);CHKERRQ(ierr);
837*27d0e99aSVaclav Hapla     } else if (debug) {ierr = PetscSynchronizedPrintf(comm, " ==\n");CHKERRQ(ierr);}
83823aaf07bSVaclav Hapla   }
839*27d0e99aSVaclav Hapla   if (debug) {
840*27d0e99aSVaclav Hapla     ierr = PetscSynchronizedFlush(comm, NULL);CHKERRQ(ierr);
841*27d0e99aSVaclav Hapla     ierr = MPI_Barrier(comm);CHKERRQ(ierr);
8422e745bdaSMatthew G. Knepley   }
843adeface4SVaclav Hapla   ierr = DMViewFromOptions(dm, NULL, "-after_fix_dm_view");CHKERRQ(ierr);
8448a650c75SVaclav Hapla   ierr = PetscFree4(roots, leaves, rootsRanks, leavesRanks);CHKERRQ(ierr);
8457c7bb832SVaclav Hapla   ierr = PetscFree2(rmine1, rremote1);CHKERRQ(ierr);
8462e745bdaSMatthew G. Knepley   PetscFunctionReturn(0);
8472e745bdaSMatthew G. Knepley }
8482e745bdaSMatthew G. Knepley 
8492e72742cSMatthew G. Knepley static PetscErrorCode IntArrayViewFromOptions(MPI_Comm comm, const char opt[], const char name[], const char idxname[], const char valname[], PetscInt n, const PetscInt a[])
8507bffde78SMichael Lange {
8512e72742cSMatthew G. Knepley   PetscInt       idx;
8522e72742cSMatthew G. Knepley   PetscMPIInt    rank;
8532e72742cSMatthew G. Knepley   PetscBool      flg;
8547bffde78SMichael Lange   PetscErrorCode ierr;
8557bffde78SMichael Lange 
8567bffde78SMichael Lange   PetscFunctionBegin;
8572e72742cSMatthew G. Knepley   ierr = PetscOptionsHasName(NULL, NULL, opt, &flg);CHKERRQ(ierr);
8582e72742cSMatthew G. Knepley   if (!flg) PetscFunctionReturn(0);
8592e72742cSMatthew G. Knepley   ierr = MPI_Comm_rank(comm, &rank);CHKERRQ(ierr);
8602e72742cSMatthew G. Knepley   ierr = PetscSynchronizedPrintf(comm, "[%d]%s:\n", rank, name);CHKERRQ(ierr);
8612e72742cSMatthew G. Knepley   for (idx = 0; idx < n; ++idx) {ierr = PetscSynchronizedPrintf(comm, "[%d]%s %D %s %D\n", rank, idxname, idx, valname, a[idx]);CHKERRQ(ierr);}
8622e72742cSMatthew G. Knepley   ierr = PetscSynchronizedFlush(comm, NULL);CHKERRQ(ierr);
8632e72742cSMatthew G. Knepley   PetscFunctionReturn(0);
8642e72742cSMatthew G. Knepley }
8652e72742cSMatthew G. Knepley 
8662e72742cSMatthew G. Knepley static PetscErrorCode SFNodeArrayViewFromOptions(MPI_Comm comm, const char opt[], const char name[], const char idxname[], PetscInt n, const PetscSFNode a[])
8672e72742cSMatthew G. Knepley {
8682e72742cSMatthew G. Knepley   PetscInt       idx;
8692e72742cSMatthew G. Knepley   PetscMPIInt    rank;
8702e72742cSMatthew G. Knepley   PetscBool      flg;
8712e72742cSMatthew G. Knepley   PetscErrorCode ierr;
8722e72742cSMatthew G. Knepley 
8732e72742cSMatthew G. Knepley   PetscFunctionBegin;
8742e72742cSMatthew G. Knepley   ierr = PetscOptionsHasName(NULL, NULL, opt, &flg);CHKERRQ(ierr);
8752e72742cSMatthew G. Knepley   if (!flg) PetscFunctionReturn(0);
8762e72742cSMatthew G. Knepley   ierr = MPI_Comm_rank(comm, &rank);CHKERRQ(ierr);
8772e72742cSMatthew G. Knepley   ierr = PetscSynchronizedPrintf(comm, "[%d]%s:\n", rank, name);CHKERRQ(ierr);
8782e72742cSMatthew G. Knepley   if (idxname) {
8792e72742cSMatthew G. Knepley     for (idx = 0; idx < n; ++idx) {ierr = PetscSynchronizedPrintf(comm, "[%d]%s %D rank %D index %D\n", rank, idxname, idx, a[idx].rank, a[idx].index);CHKERRQ(ierr);}
8802e72742cSMatthew G. Knepley   } else {
8812e72742cSMatthew G. Knepley     for (idx = 0; idx < n; ++idx) {ierr = PetscSynchronizedPrintf(comm, "[%d]rank %D index %D\n", rank, a[idx].rank, a[idx].index);CHKERRQ(ierr);}
8822e72742cSMatthew G. Knepley   }
8832e72742cSMatthew G. Knepley   ierr = PetscSynchronizedFlush(comm, NULL);CHKERRQ(ierr);
8842e72742cSMatthew G. Knepley   PetscFunctionReturn(0);
8852e72742cSMatthew G. Knepley }
8862e72742cSMatthew G. Knepley 
8872e72742cSMatthew G. Knepley static PetscErrorCode DMPlexMapToLocalPoint(PetscHMapIJ roothash, const PetscInt localPoints[], PetscMPIInt rank, PetscSFNode remotePoint, PetscInt *localPoint)
8882e72742cSMatthew G. Knepley {
8892e72742cSMatthew G. Knepley   PetscErrorCode ierr;
8902e72742cSMatthew G. Knepley 
8912e72742cSMatthew G. Knepley   PetscFunctionBegin;
8922e72742cSMatthew G. Knepley   if (remotePoint.rank == rank) {
8932e72742cSMatthew G. Knepley     *localPoint = remotePoint.index;
8942e72742cSMatthew G. Knepley   } else {
8952e72742cSMatthew G. Knepley     PetscHashIJKey key;
8962e72742cSMatthew G. Knepley     PetscInt       root;
8972e72742cSMatthew G. Knepley 
8982e72742cSMatthew G. Knepley     key.i = remotePoint.index;
8992e72742cSMatthew G. Knepley     key.j = remotePoint.rank;
9002e72742cSMatthew G. Knepley     ierr = PetscHMapIJGet(roothash, key, &root);CHKERRQ(ierr);
9012e72742cSMatthew G. Knepley     if (root >= 0) {
9022e72742cSMatthew G. Knepley       *localPoint = localPoints[root];
9032e72742cSMatthew G. Knepley     } else PetscFunctionReturn(1);
9042e72742cSMatthew G. Knepley   }
9052e72742cSMatthew G. Knepley   PetscFunctionReturn(0);
9062e72742cSMatthew G. Knepley }
9072e72742cSMatthew G. Knepley 
9082e72742cSMatthew G. Knepley /*@
9092e72742cSMatthew G. Knepley   DMPlexInterpolatePointSF - Insert interpolated points in the overlap into the PointSF in parallel, following local interpolation
9102e72742cSMatthew G. Knepley 
911d083f849SBarry Smith   Collective on dm
9122e72742cSMatthew G. Knepley 
9132e72742cSMatthew G. Knepley   Input Parameters:
9142e72742cSMatthew G. Knepley + dm      - The interpolated DM
9152e72742cSMatthew G. Knepley - pointSF - The initial SF without interpolated points
9162e72742cSMatthew G. Knepley 
9172e72742cSMatthew G. Knepley   Output Parameter:
9182e72742cSMatthew G. Knepley . pointSF - The SF including interpolated points
9192e72742cSMatthew G. Knepley 
9202e72742cSMatthew G. Knepley   Level: intermediate
9212e72742cSMatthew G. Knepley 
9222e72742cSMatthew G. Knepley    Note: All debugging for this process can be turned on with the options: -dm_interp_pre_view -petscsf_interp_pre_view -petscsection_interp_candidate_view -petscsection_interp_candidate_remote_view -petscsection_interp_claim_view -petscsf_interp_pre_view -dmplex_interp_debug
9232e72742cSMatthew G. Knepley 
9242e72742cSMatthew G. Knepley .seealso: DMPlexInterpolate(), DMPlexUninterpolate()
9252e72742cSMatthew G. Knepley @*/
926e53487d3SMatthew G. Knepley PetscErrorCode DMPlexInterpolatePointSF(DM dm, PetscSF pointSF)
9272e72742cSMatthew G. Knepley {
9282e72742cSMatthew G. Knepley   /*
9292e72742cSMatthew G. Knepley        Okay, the algorithm is:
9302e72742cSMatthew G. Knepley          - Take each point in the overlap (root)
9312e72742cSMatthew G. Knepley          - Look at the neighboring points in the overlap (candidates)
9322e72742cSMatthew G. Knepley          - Send these candidate points to neighbors
9332e72742cSMatthew G. Knepley          - Neighbor checks for edge between root and candidate
9342e72742cSMatthew G. Knepley          - If edge is found, it replaces candidate point with edge point
9352e72742cSMatthew G. Knepley          - Send back the overwritten candidates (claims)
9362e72742cSMatthew G. Knepley          - Original guy checks for edges, different from original candidate, and gets its own edge
9372e72742cSMatthew G. Knepley          - This pair is put into SF
9382e72742cSMatthew G. Knepley 
9392e72742cSMatthew G. Knepley        We need a new algorithm that tolerates groups larger than 2.
9402e72742cSMatthew G. Knepley          - Take each point in the overlap (root)
9412e72742cSMatthew G. Knepley          - Find all collections of points in the overlap which make faces (do early join)
9422e72742cSMatthew G. Knepley          - Send collections as candidates (add size as first number)
94366aa2a39SMatthew G. Knepley            - Make sure to send collection to all owners of all overlap points in collection
9442e72742cSMatthew G. Knepley          - Neighbor check for face in collections
9452e72742cSMatthew G. Knepley          - If face is found, it replaces candidate point with face point
9462e72742cSMatthew G. Knepley          - Send back the overwritten candidates (claims)
9472e72742cSMatthew G. Knepley          - Original guy checks for faces, different from original candidate, and gets its own face
9482e72742cSMatthew G. Knepley          - This pair is put into SF
9492e72742cSMatthew G. Knepley   */
9502e72742cSMatthew G. Knepley   PetscHMapI         leafhash;
9512e72742cSMatthew G. Knepley   PetscHMapIJ        roothash;
9522e72742cSMatthew G. Knepley   const PetscInt    *localPoints, *rootdegree;
9532e72742cSMatthew G. Knepley   const PetscSFNode *remotePoints;
9542e72742cSMatthew G. Knepley   PetscSFNode       *candidates, *candidatesRemote, *claims;
9552e72742cSMatthew G. Knepley   PetscSection       candidateSection, candidateSectionRemote, claimSection;
9562e72742cSMatthew G. Knepley   PetscInt           numLeaves, l, numRoots, r, candidatesSize, candidatesRemoteSize;
9572e72742cSMatthew G. Knepley   PetscMPIInt        size, rank;
9582e72742cSMatthew G. Knepley   PetscHashIJKey     key;
9592e72742cSMatthew G. Knepley   PetscBool          debug = PETSC_FALSE;
9602e72742cSMatthew G. Knepley   PetscErrorCode     ierr;
9612e72742cSMatthew G. Knepley 
9622e72742cSMatthew G. Knepley   PetscFunctionBegin;
9632e72742cSMatthew G. Knepley   ierr = PetscOptionsHasName(NULL, ((PetscObject) dm)->prefix, "-dmplex_interp_debug", &debug);CHKERRQ(ierr);
9649852e123SBarry Smith   ierr = MPI_Comm_size(PetscObjectComm((PetscObject) dm), &size);CHKERRQ(ierr);
9657bffde78SMichael Lange   ierr = MPI_Comm_rank(PetscObjectComm((PetscObject) dm), &rank);CHKERRQ(ierr);
9667bffde78SMichael Lange   ierr = PetscSFGetGraph(pointSF, &numRoots, &numLeaves, &localPoints, &remotePoints);CHKERRQ(ierr);
9679852e123SBarry Smith   if (size < 2 || numRoots < 0) PetscFunctionReturn(0);
968f240e3b7SVaclav Hapla   ierr = DMPlexGetOverlap(dm, &r);CHKERRQ(ierr);
969f240e3b7SVaclav Hapla   if (r) SETERRQ(PetscObjectComm((PetscObject)dm), PETSC_ERR_SUP, "Interpolation of overlapped DMPlex not implemented yet");
9702e72742cSMatthew G. Knepley   ierr = PetscObjectViewFromOptions((PetscObject) dm, NULL, "-dm_interp_pre_view");CHKERRQ(ierr);
9712e72742cSMatthew G. Knepley   ierr = PetscObjectViewFromOptions((PetscObject) pointSF, NULL, "-petscsf_interp_pre_view");CHKERRQ(ierr);
97225afeb17SMatthew G. Knepley   ierr = PetscLogEventBegin(DMPLEX_InterpolateSF,dm,0,0,0);CHKERRQ(ierr);
9737bffde78SMichael Lange   /* Build hashes of points in the SF for efficient lookup */
974e8f14785SLisandro Dalcin   ierr = PetscHMapICreate(&leafhash);CHKERRQ(ierr);
975e8f14785SLisandro Dalcin   ierr = PetscHMapIJCreate(&roothash);CHKERRQ(ierr);
9762e72742cSMatthew G. Knepley   for (l = 0; l < numLeaves; ++l) {
9772e72742cSMatthew G. Knepley     key.i = remotePoints[l].index;
9782e72742cSMatthew G. Knepley     key.j = remotePoints[l].rank;
9792e72742cSMatthew G. Knepley     ierr = PetscHMapISet(leafhash, localPoints[l], l);CHKERRQ(ierr);
9802e72742cSMatthew G. Knepley     ierr = PetscHMapIJSet(roothash, key, l);CHKERRQ(ierr);
9817bffde78SMichael Lange   }
98266aa2a39SMatthew G. Knepley   /* Compute root degree to identify shared points */
9832e72742cSMatthew G. Knepley   ierr = PetscSFComputeDegreeBegin(pointSF, &rootdegree);CHKERRQ(ierr);
9842e72742cSMatthew G. Knepley   ierr = PetscSFComputeDegreeEnd(pointSF, &rootdegree);CHKERRQ(ierr);
9852e72742cSMatthew G. Knepley   ierr = IntArrayViewFromOptions(PetscObjectComm((PetscObject) dm), "-interp_root_degree_view", "Root degree", "point", "degree", numRoots, rootdegree);CHKERRQ(ierr);
98666aa2a39SMatthew G. Knepley   /* Build a section / SFNode array of candidate points (face bd points) in the cone(support(leaf)),
98766aa2a39SMatthew G. Knepley      where each candidate is defined by a set of remote points (roots) for the other points that define the face. */
9887bffde78SMichael Lange   ierr = PetscSectionCreate(PetscObjectComm((PetscObject) dm), &candidateSection);CHKERRQ(ierr);
9897bffde78SMichael Lange   ierr = PetscSectionSetChart(candidateSection, 0, numRoots);CHKERRQ(ierr);
9907bffde78SMichael Lange   {
99166aa2a39SMatthew G. Knepley     PetscHMapIJ facehash;
9922e72742cSMatthew G. Knepley 
99366aa2a39SMatthew G. Knepley     ierr = PetscHMapIJCreate(&facehash);CHKERRQ(ierr);
9942e72742cSMatthew G. Knepley     for (l = 0; l < numLeaves; ++l) {
9952e72742cSMatthew G. Knepley       const PetscInt    localPoint = localPoints[l];
9962e72742cSMatthew G. Knepley       const PetscInt   *support;
9972e72742cSMatthew G. Knepley       PetscInt          supportSize, s;
9982e72742cSMatthew G. Knepley 
9992e72742cSMatthew G. Knepley       if (debug) {ierr = PetscSynchronizedPrintf(PetscObjectComm((PetscObject) dm), "[%d]  Checking local point %D\n", rank, localPoint);CHKERRQ(ierr);}
10002e72742cSMatthew G. Knepley       ierr = DMPlexGetSupportSize(dm, localPoint, &supportSize);CHKERRQ(ierr);
10012e72742cSMatthew G. Knepley       ierr = DMPlexGetSupport(dm, localPoint, &support);CHKERRQ(ierr);
10022e72742cSMatthew G. Knepley       for (s = 0; s < supportSize; ++s) {
10032e72742cSMatthew G. Knepley         const PetscInt  face = support[s];
10042e72742cSMatthew G. Knepley         const PetscInt *cone;
10052e72742cSMatthew G. Knepley         PetscInt        coneSize, c, f, root;
10062e72742cSMatthew G. Knepley         PetscBool       isFace = PETSC_TRUE;
10072e72742cSMatthew G. Knepley 
10082e72742cSMatthew G. Knepley         /* Only add face once */
10092e72742cSMatthew G. Knepley         if (debug) {ierr = PetscSynchronizedPrintf(PetscObjectComm((PetscObject) dm), "[%d]    Support point %D\n", rank, face);CHKERRQ(ierr);}
101066aa2a39SMatthew G. Knepley         key.i = localPoint;
101166aa2a39SMatthew G. Knepley         key.j = face;
101266aa2a39SMatthew G. Knepley         ierr = PetscHMapIJGet(facehash, key, &f);CHKERRQ(ierr);
10132e72742cSMatthew G. Knepley         if (f >= 0) continue;
10142e72742cSMatthew G. Knepley         ierr = DMPlexGetConeSize(dm, face, &coneSize);CHKERRQ(ierr);
10152e72742cSMatthew G. Knepley         ierr = DMPlexGetCone(dm, face, &cone);CHKERRQ(ierr);
10162e72742cSMatthew G. Knepley         /* If a cone point does not map to leaves on any proc, then do not put face in SF */
10172e72742cSMatthew G. Knepley         for (c = 0; c < coneSize; ++c) {
10182e72742cSMatthew G. Knepley           if (debug) {ierr = PetscSynchronizedPrintf(PetscObjectComm((PetscObject) dm), "[%d]      Cone point %D\n", rank, cone[c]);CHKERRQ(ierr);}
10192e72742cSMatthew G. Knepley           ierr = PetscHMapIGet(leafhash, cone[c], &root);CHKERRQ(ierr);
10202e72742cSMatthew G. Knepley           if (!rootdegree[cone[c]] && (root < 0)) {isFace = PETSC_FALSE; break;}
10212e72742cSMatthew G. Knepley         }
10222e72742cSMatthew G. Knepley         if (isFace) {
102366aa2a39SMatthew G. Knepley           if (debug) {ierr = PetscSynchronizedPrintf(PetscObjectComm((PetscObject) dm), "[%d]    Found shared face %D\n", rank, face);CHKERRQ(ierr);}
102466aa2a39SMatthew G. Knepley           ierr = PetscHMapIJSet(facehash, key, l);CHKERRQ(ierr);
10252e72742cSMatthew G. Knepley           ierr = PetscSectionAddDof(candidateSection, localPoint, coneSize);CHKERRQ(ierr);
10267bffde78SMichael Lange         }
10277bffde78SMichael Lange       }
10282e72742cSMatthew G. Knepley     }
10292e72742cSMatthew G. Knepley     if (debug) {ierr = PetscSynchronizedFlush(PetscObjectComm((PetscObject) dm), NULL);CHKERRQ(ierr);}
103066aa2a39SMatthew G. Knepley     ierr = PetscHMapIJClear(facehash);CHKERRQ(ierr);
10317bffde78SMichael Lange     ierr = PetscSectionSetUp(candidateSection);CHKERRQ(ierr);
10327bffde78SMichael Lange     ierr = PetscSectionGetStorageSize(candidateSection, &candidatesSize);CHKERRQ(ierr);
10337bffde78SMichael Lange     ierr = PetscMalloc1(candidatesSize, &candidates);CHKERRQ(ierr);
10342e72742cSMatthew G. Knepley     for (l = 0; l < numLeaves; ++l) {
10352e72742cSMatthew G. Knepley       const PetscInt    localPoint = localPoints[l];
10362e72742cSMatthew G. Knepley       const PetscInt   *support;
10372e72742cSMatthew G. Knepley       PetscInt          supportSize, s, offset, idx = 0;
10382e72742cSMatthew G. Knepley 
10392e72742cSMatthew G. Knepley       if (debug) {ierr = PetscSynchronizedPrintf(PetscObjectComm((PetscObject) dm), "[%d]  Checking local point %D\n", rank, localPoint);CHKERRQ(ierr);}
10402e72742cSMatthew G. Knepley       ierr = PetscSectionGetOffset(candidateSection, localPoint, &offset);CHKERRQ(ierr);
10412e72742cSMatthew G. Knepley       ierr = DMPlexGetSupportSize(dm, localPoint, &supportSize);CHKERRQ(ierr);
10422e72742cSMatthew G. Knepley       ierr = DMPlexGetSupport(dm, localPoint, &support);CHKERRQ(ierr);
10432e72742cSMatthew G. Knepley       for (s = 0; s < supportSize; ++s) {
10442e72742cSMatthew G. Knepley         const PetscInt  face = support[s];
10452e72742cSMatthew G. Knepley         const PetscInt *cone;
10462e72742cSMatthew G. Knepley         PetscInt        coneSize, c, f, root;
10472e72742cSMatthew G. Knepley         PetscBool       isFace = PETSC_TRUE;
10482e72742cSMatthew G. Knepley 
10492e72742cSMatthew G. Knepley         /* Only add face once */
10502e72742cSMatthew G. Knepley         if (debug) {ierr = PetscSynchronizedPrintf(PetscObjectComm((PetscObject) dm), "[%d]    Support point %D\n", rank, face);CHKERRQ(ierr);}
105166aa2a39SMatthew G. Knepley         key.i = localPoint;
105266aa2a39SMatthew G. Knepley         key.j = face;
105366aa2a39SMatthew G. Knepley         ierr = PetscHMapIJGet(facehash, key, &f);CHKERRQ(ierr);
10542e72742cSMatthew G. Knepley         if (f >= 0) continue;
10552e72742cSMatthew G. Knepley         ierr = DMPlexGetConeSize(dm, face, &coneSize);CHKERRQ(ierr);
10562e72742cSMatthew G. Knepley         ierr = DMPlexGetCone(dm, face, &cone);CHKERRQ(ierr);
10572e72742cSMatthew G. Knepley         /* If a cone point does not map to leaves on any proc, then do not put face in SF */
10582e72742cSMatthew G. Knepley         for (c = 0; c < coneSize; ++c) {
10592e72742cSMatthew G. Knepley           if (debug) {ierr = PetscSynchronizedPrintf(PetscObjectComm((PetscObject) dm), "[%d]      Cone point %D\n", rank, cone[c]);CHKERRQ(ierr);}
10602e72742cSMatthew G. Knepley           ierr = PetscHMapIGet(leafhash, cone[c], &root);CHKERRQ(ierr);
10612e72742cSMatthew G. Knepley           if (!rootdegree[cone[c]] && (root < 0)) {isFace = PETSC_FALSE; break;}
10622e72742cSMatthew G. Knepley         }
10632e72742cSMatthew G. Knepley         if (isFace) {
106466aa2a39SMatthew G. Knepley           if (debug) {ierr = PetscSynchronizedPrintf(PetscObjectComm((PetscObject) dm), "[%d]    Adding shared face %D at idx %D\n", rank, face, idx);CHKERRQ(ierr);}
106566aa2a39SMatthew G. Knepley           ierr = PetscHMapIJSet(facehash, key, l);CHKERRQ(ierr);
10662e72742cSMatthew G. Knepley           candidates[offset+idx].rank    = -1;
10672e72742cSMatthew G. Knepley           candidates[offset+idx++].index = coneSize-1;
10682e72742cSMatthew G. Knepley           for (c = 0; c < coneSize; ++c) {
10692e72742cSMatthew G. Knepley             if (cone[c] == localPoint) continue;
10702e72742cSMatthew G. Knepley             if (rootdegree[cone[c]]) {
10712e72742cSMatthew G. Knepley               candidates[offset+idx].rank    = rank;
10722e72742cSMatthew G. Knepley               candidates[offset+idx++].index = cone[c];
10732e72742cSMatthew G. Knepley             } else {
10742e72742cSMatthew G. Knepley               ierr = PetscHMapIGet(leafhash, cone[c], &root);CHKERRQ(ierr);
10752e72742cSMatthew G. Knepley               if (root < 0) SETERRQ1(PETSC_COMM_SELF, PETSC_ERR_PLIB, "Cannot locate local point %D in SF", cone[c]);
10762e72742cSMatthew G. Knepley               candidates[offset+idx++] = remotePoints[root];
10777bffde78SMichael Lange             }
10787bffde78SMichael Lange           }
10792e72742cSMatthew G. Knepley         }
10802e72742cSMatthew G. Knepley       }
10812e72742cSMatthew G. Knepley     }
10822e72742cSMatthew G. Knepley     if (debug) {ierr = PetscSynchronizedFlush(PetscObjectComm((PetscObject) dm), NULL);CHKERRQ(ierr);}
108366aa2a39SMatthew G. Knepley     ierr = PetscHMapIJDestroy(&facehash);CHKERRQ(ierr);
10842e72742cSMatthew G. Knepley     ierr = PetscObjectViewFromOptions((PetscObject) candidateSection, NULL, "-petscsection_interp_candidate_view");CHKERRQ(ierr);
10852e72742cSMatthew G. Knepley     ierr = SFNodeArrayViewFromOptions(PetscObjectComm((PetscObject) dm), "-petscsection_interp_candidate_view", "Candidates", NULL, candidatesSize, candidates);CHKERRQ(ierr);
10867bffde78SMichael Lange   }
10877bffde78SMichael Lange   /* Gather candidate section / array pair into the root partition via inverse(multi(pointSF)). */
10882e72742cSMatthew G. Knepley   /*   Note that this section is indexed by offsets into leaves, not by point number */
10897bffde78SMichael Lange   {
10907bffde78SMichael Lange     PetscSF   sfMulti, sfInverse, sfCandidates;
10917bffde78SMichael Lange     PetscInt *remoteOffsets;
10922e72742cSMatthew G. Knepley 
10937bffde78SMichael Lange     ierr = PetscSFGetMultiSF(pointSF, &sfMulti);CHKERRQ(ierr);
10947bffde78SMichael Lange     ierr = PetscSFCreateInverseSF(sfMulti, &sfInverse);CHKERRQ(ierr);
10957bffde78SMichael Lange     ierr = PetscSectionCreate(PetscObjectComm((PetscObject) dm), &candidateSectionRemote);CHKERRQ(ierr);
10967bffde78SMichael Lange     ierr = PetscSFDistributeSection(sfInverse, candidateSection, &remoteOffsets, candidateSectionRemote);CHKERRQ(ierr);
10977bffde78SMichael Lange     ierr = PetscSFCreateSectionSF(sfInverse, candidateSection, remoteOffsets, candidateSectionRemote, &sfCandidates);CHKERRQ(ierr);
10987bffde78SMichael Lange     ierr = PetscSectionGetStorageSize(candidateSectionRemote, &candidatesRemoteSize);CHKERRQ(ierr);
10997bffde78SMichael Lange     ierr = PetscMalloc1(candidatesRemoteSize, &candidatesRemote);CHKERRQ(ierr);
11007bffde78SMichael Lange     ierr = PetscSFBcastBegin(sfCandidates, MPIU_2INT, candidates, candidatesRemote);CHKERRQ(ierr);
11017bffde78SMichael Lange     ierr = PetscSFBcastEnd(sfCandidates, MPIU_2INT, candidates, candidatesRemote);CHKERRQ(ierr);
11027bffde78SMichael Lange     ierr = PetscSFDestroy(&sfInverse);CHKERRQ(ierr);
11037bffde78SMichael Lange     ierr = PetscSFDestroy(&sfCandidates);CHKERRQ(ierr);
11047bffde78SMichael Lange     ierr = PetscFree(remoteOffsets);CHKERRQ(ierr);
11052e72742cSMatthew G. Knepley 
11062e72742cSMatthew G. Knepley     ierr = PetscObjectViewFromOptions((PetscObject) candidateSectionRemote, NULL, "-petscsection_interp_candidate_remote_view");CHKERRQ(ierr);
11072e72742cSMatthew G. Knepley     ierr = SFNodeArrayViewFromOptions(PetscObjectComm((PetscObject) dm), "-petscsection_interp_candidate_remote_view", "Remote Candidates", NULL, candidatesRemoteSize, candidatesRemote);CHKERRQ(ierr);
11087bffde78SMichael Lange   }
11092e72742cSMatthew G. Knepley   /* */
11107bffde78SMichael Lange   {
11112e72742cSMatthew G. Knepley     PetscInt idx;
11122e72742cSMatthew G. Knepley     /* There is a section point for every leaf attached to a given root point */
11132e72742cSMatthew G. Knepley     for (r = 0, idx = 0; r < numRoots; ++r) {
11142e72742cSMatthew G. Knepley       PetscInt deg;
11152e72742cSMatthew G. Knepley       for (deg = 0; deg < rootdegree[r]; ++deg, ++idx) {
11162e72742cSMatthew G. Knepley         PetscInt offset, dof, d;
11172e72742cSMatthew G. Knepley 
11187bffde78SMichael Lange         ierr = PetscSectionGetDof(candidateSectionRemote, idx, &dof);CHKERRQ(ierr);
11197bffde78SMichael Lange         ierr = PetscSectionGetOffset(candidateSectionRemote, idx, &offset);CHKERRQ(ierr);
11202e72742cSMatthew G. Knepley         for (d = 0; d < dof; ++d) {
11212e72742cSMatthew G. Knepley           const PetscInt  sizeInd   = offset+d;
11222e72742cSMatthew G. Knepley           const PetscInt  numPoints = candidatesRemote[sizeInd].index;
11232e72742cSMatthew G. Knepley           const PetscInt *join      = NULL;
11242e72742cSMatthew G. Knepley           PetscInt        points[1024], p, joinSize;
11252e72742cSMatthew G. Knepley 
11262e72742cSMatthew G. Knepley           points[0] = r;
11272e72742cSMatthew G. Knepley           for (p = 0; p < numPoints; ++p) {
11282e72742cSMatthew G. Knepley             ierr = DMPlexMapToLocalPoint(roothash, localPoints, rank, candidatesRemote[offset+(++d)], &points[p+1]);
11292e72742cSMatthew G. Knepley             if (ierr) {d += numPoints-1 - p; break;} /* We got a point not in our overlap */
11302e72742cSMatthew G. Knepley             if (debug) {ierr = PetscSynchronizedPrintf(PetscObjectComm((PetscObject) dm), "[%d]  Checking local candidate %D\n", rank, points[p+1]);CHKERRQ(ierr);}
11317bffde78SMichael Lange           }
11322e72742cSMatthew G. Knepley           if (ierr) continue;
11332e72742cSMatthew G. Knepley           ierr = DMPlexGetJoin(dm, numPoints+1, points, &joinSize, &join);CHKERRQ(ierr);
11347bffde78SMichael Lange           if (joinSize == 1) {
11352e72742cSMatthew G. Knepley             if (debug) {ierr = PetscSynchronizedPrintf(PetscObjectComm((PetscObject) dm), "[%d]    Adding face %D at idx %D\n", rank, join[0], sizeInd);CHKERRQ(ierr);}
11362e72742cSMatthew G. Knepley             candidatesRemote[sizeInd].rank  = rank;
11372e72742cSMatthew G. Knepley             candidatesRemote[sizeInd].index = join[0];
11387bffde78SMichael Lange           }
11392e72742cSMatthew G. Knepley           ierr = DMPlexRestoreJoin(dm, numPoints+1, points, &joinSize, &join);CHKERRQ(ierr);
11407bffde78SMichael Lange         }
11417bffde78SMichael Lange       }
11427bffde78SMichael Lange     }
11432e72742cSMatthew G. Knepley     if (debug) {ierr = PetscSynchronizedFlush(PetscObjectComm((PetscObject) dm), NULL);CHKERRQ(ierr);}
11447bffde78SMichael Lange   }
11457bffde78SMichael Lange   /* Push claims back to receiver via the MultiSF and derive new pointSF mapping on receiver */
11467bffde78SMichael Lange   {
11477bffde78SMichael Lange     PetscSF         sfMulti, sfClaims, sfPointNew;
11487bffde78SMichael Lange     PetscSFNode    *remotePointsNew;
11492e72742cSMatthew G. Knepley     PetscHMapI      claimshash;
11502e72742cSMatthew G. Knepley     PetscInt       *remoteOffsets, *localPointsNew;
11512e72742cSMatthew G. Knepley     PetscInt        claimsSize, pStart, pEnd, root, numLocalNew, p, d;
11522e72742cSMatthew G. Knepley 
11537bffde78SMichael Lange     ierr = PetscSFGetMultiSF(pointSF, &sfMulti);CHKERRQ(ierr);
11547bffde78SMichael Lange     ierr = PetscSectionCreate(PetscObjectComm((PetscObject) dm), &claimSection);CHKERRQ(ierr);
11557bffde78SMichael Lange     ierr = PetscSFDistributeSection(sfMulti, candidateSectionRemote, &remoteOffsets, claimSection);CHKERRQ(ierr);
11567bffde78SMichael Lange     ierr = PetscSFCreateSectionSF(sfMulti, candidateSectionRemote, remoteOffsets, claimSection, &sfClaims);CHKERRQ(ierr);
11572e72742cSMatthew G. Knepley     ierr = PetscSectionGetStorageSize(claimSection, &claimsSize);CHKERRQ(ierr);
11582e72742cSMatthew G. Knepley     ierr = PetscMalloc1(claimsSize, &claims);CHKERRQ(ierr);
11597bffde78SMichael Lange     ierr = PetscSFBcastBegin(sfClaims, MPIU_2INT, candidatesRemote, claims);CHKERRQ(ierr);
11607bffde78SMichael Lange     ierr = PetscSFBcastEnd(sfClaims, MPIU_2INT, candidatesRemote, claims);CHKERRQ(ierr);
11617bffde78SMichael Lange     ierr = PetscSFDestroy(&sfClaims);CHKERRQ(ierr);
11627bffde78SMichael Lange     ierr = PetscFree(remoteOffsets);CHKERRQ(ierr);
11632e72742cSMatthew G. Knepley     ierr = PetscObjectViewFromOptions((PetscObject) claimSection, NULL, "-petscsection_interp_claim_view");CHKERRQ(ierr);
11642e72742cSMatthew G. Knepley     ierr = SFNodeArrayViewFromOptions(PetscObjectComm((PetscObject) dm), "-petscsection_interp_claim_view", "Claims", NULL, claimsSize, claims);CHKERRQ(ierr);
11657bffde78SMichael Lange     /* Walk the original section of local supports and add an SF entry for each updated item */
1166e8f14785SLisandro Dalcin     ierr = PetscHMapICreate(&claimshash);CHKERRQ(ierr);
11677bffde78SMichael Lange     for (p = 0; p < numRoots; ++p) {
11682e72742cSMatthew G. Knepley       PetscInt dof, offset;
11692e72742cSMatthew G. Knepley 
11702e72742cSMatthew G. Knepley       if (debug) {ierr = PetscSynchronizedPrintf(PetscObjectComm((PetscObject) dm), "[%d]  Checking root for claims %D\n", rank, p);CHKERRQ(ierr);}
11717bffde78SMichael Lange       ierr = PetscSectionGetDof(candidateSection, p, &dof);CHKERRQ(ierr);
11727bffde78SMichael Lange       ierr = PetscSectionGetOffset(candidateSection, p, &offset);CHKERRQ(ierr);
11732e72742cSMatthew G. Knepley       for (d = 0; d < dof;) {
11742e72742cSMatthew G. Knepley         if (claims[offset+d].rank >= 0) {
11752e72742cSMatthew G. Knepley           const PetscInt  faceInd   = offset+d;
11762e72742cSMatthew G. Knepley           const PetscInt  numPoints = candidates[faceInd].index;
11772e72742cSMatthew G. Knepley           const PetscInt *join      = NULL;
11782e72742cSMatthew G. Knepley           PetscInt        joinSize, points[1024], c;
11792e72742cSMatthew G. Knepley 
11802e72742cSMatthew G. Knepley           if (debug) {ierr = PetscSynchronizedPrintf(PetscObjectComm((PetscObject) dm), "[%d]    Found claim for remote point (%D, %D)\n", rank, claims[faceInd].rank, claims[faceInd].index);CHKERRQ(ierr);}
11812e72742cSMatthew G. Knepley           points[0] = p;
11822e72742cSMatthew G. Knepley           if (debug) {ierr = PetscSynchronizedPrintf(PetscObjectComm((PetscObject) dm), "[%d]      point %D\n", rank, points[0]);CHKERRQ(ierr);}
11832e72742cSMatthew G. Knepley           for (c = 0, ++d; c < numPoints; ++c, ++d) {
1184e8f14785SLisandro Dalcin             key.i = candidates[offset+d].index;
1185e8f14785SLisandro Dalcin             key.j = candidates[offset+d].rank;
1186e8f14785SLisandro Dalcin             ierr = PetscHMapIJGet(roothash, key, &root);CHKERRQ(ierr);
11872e72742cSMatthew G. Knepley             points[c+1] = localPoints[root];
11882e72742cSMatthew G. Knepley             if (debug) {ierr = PetscSynchronizedPrintf(PetscObjectComm((PetscObject) dm), "[%d]      point %D\n", rank, points[c+1]);CHKERRQ(ierr);}
11892e72742cSMatthew G. Knepley           }
11902e72742cSMatthew G. Knepley           ierr = DMPlexGetJoin(dm, numPoints+1, points, &joinSize, &join);CHKERRQ(ierr);
11912e72742cSMatthew G. Knepley           if (joinSize == 1) {
11922e72742cSMatthew G. Knepley             if (debug) {ierr = PetscSynchronizedPrintf(PetscObjectComm((PetscObject) dm), "[%d]    Found local face %D\n", rank, join[0]);CHKERRQ(ierr);}
11932e72742cSMatthew G. Knepley             ierr = PetscHMapISet(claimshash, join[0], faceInd);CHKERRQ(ierr);
11942e72742cSMatthew G. Knepley           }
11952e72742cSMatthew G. Knepley           ierr = DMPlexRestoreJoin(dm, numPoints+1, points, &joinSize, &join);CHKERRQ(ierr);
11962e72742cSMatthew G. Knepley         } else d += claims[offset+d].index+1;
11977bffde78SMichael Lange       }
11987bffde78SMichael Lange     }
11992e72742cSMatthew G. Knepley     if (debug) {ierr = PetscSynchronizedFlush(PetscObjectComm((PetscObject) dm), NULL);CHKERRQ(ierr);}
12007bffde78SMichael Lange     /* Create new pointSF from hashed claims */
1201e8f14785SLisandro Dalcin     ierr = PetscHMapIGetSize(claimshash, &numLocalNew);CHKERRQ(ierr);
12027bffde78SMichael Lange     ierr = DMPlexGetChart(dm, &pStart, &pEnd);CHKERRQ(ierr);
12037bffde78SMichael Lange     ierr = PetscMalloc1(numLeaves + numLocalNew, &localPointsNew);CHKERRQ(ierr);
12047bffde78SMichael Lange     ierr = PetscMalloc1(numLeaves + numLocalNew, &remotePointsNew);CHKERRQ(ierr);
12057bffde78SMichael Lange     for (p = 0; p < numLeaves; ++p) {
12067bffde78SMichael Lange       localPointsNew[p] = localPoints[p];
12077bffde78SMichael Lange       remotePointsNew[p].index = remotePoints[p].index;
12087bffde78SMichael Lange       remotePointsNew[p].rank  = remotePoints[p].rank;
12097bffde78SMichael Lange     }
1210f3190fc2SToby Isaac     p = numLeaves;
1211e8f14785SLisandro Dalcin     ierr = PetscHMapIGetKeys(claimshash, &p, localPointsNew);CHKERRQ(ierr);
1212f3190fc2SToby Isaac     ierr = PetscSortInt(numLocalNew, &localPointsNew[numLeaves]);CHKERRQ(ierr);
12137bffde78SMichael Lange     for (p = numLeaves; p < numLeaves + numLocalNew; ++p) {
12142e72742cSMatthew G. Knepley       PetscInt offset;
1215e8f14785SLisandro Dalcin       ierr = PetscHMapIGet(claimshash, localPointsNew[p], &offset);CHKERRQ(ierr);
12167bffde78SMichael Lange       remotePointsNew[p] = claims[offset];
12177bffde78SMichael Lange     }
12187bffde78SMichael Lange     ierr = PetscSFCreate(PetscObjectComm((PetscObject) dm), &sfPointNew);CHKERRQ(ierr);
12197bffde78SMichael Lange     ierr = PetscSFSetGraph(sfPointNew, pEnd-pStart, numLeaves+numLocalNew, localPointsNew, PETSC_OWN_POINTER, remotePointsNew, PETSC_OWN_POINTER);CHKERRQ(ierr);
12207bffde78SMichael Lange     ierr = DMSetPointSF(dm, sfPointNew);CHKERRQ(ierr);
12217bffde78SMichael Lange     ierr = PetscSFDestroy(&sfPointNew);CHKERRQ(ierr);
1222e8f14785SLisandro Dalcin     ierr = PetscHMapIDestroy(&claimshash);CHKERRQ(ierr);
12237bffde78SMichael Lange   }
1224e8f14785SLisandro Dalcin   ierr = PetscHMapIDestroy(&leafhash);CHKERRQ(ierr);
1225e8f14785SLisandro Dalcin   ierr = PetscHMapIJDestroy(&roothash);CHKERRQ(ierr);
12267bffde78SMichael Lange   ierr = PetscSectionDestroy(&candidateSection);CHKERRQ(ierr);
12277bffde78SMichael Lange   ierr = PetscSectionDestroy(&candidateSectionRemote);CHKERRQ(ierr);
12287bffde78SMichael Lange   ierr = PetscSectionDestroy(&claimSection);CHKERRQ(ierr);
12297bffde78SMichael Lange   ierr = PetscFree(candidates);CHKERRQ(ierr);
12307bffde78SMichael Lange   ierr = PetscFree(candidatesRemote);CHKERRQ(ierr);
12317bffde78SMichael Lange   ierr = PetscFree(claims);CHKERRQ(ierr);
123225afeb17SMatthew G. Knepley   ierr = PetscLogEventEnd(DMPLEX_InterpolateSF,dm,0,0,0);CHKERRQ(ierr);
12337bffde78SMichael Lange   PetscFunctionReturn(0);
12347bffde78SMichael Lange }
12357bffde78SMichael Lange 
1236248eb259SVaclav Hapla /*@
123780330477SMatthew G. Knepley   DMPlexInterpolate - Take in a cell-vertex mesh and return one with all intermediate faces, edges, etc.
123880330477SMatthew G. Knepley 
1239d083f849SBarry Smith   Collective on dm
124080330477SMatthew G. Knepley 
1241e56d480eSMatthew G. Knepley   Input Parameters:
1242e56d480eSMatthew G. Knepley + dm - The DMPlex object with only cells and vertices
124310a67516SMatthew G. Knepley - dmInt - The interpolated DM
124480330477SMatthew G. Knepley 
124580330477SMatthew G. Knepley   Output Parameter:
12464e3744c5SMatthew G. Knepley . dmInt - The complete DMPlex object
124780330477SMatthew G. Knepley 
124880330477SMatthew G. Knepley   Level: intermediate
124980330477SMatthew G. Knepley 
125095452b02SPatrick Sanan   Notes:
125195452b02SPatrick Sanan     It does not copy over the coordinates.
125243eeeb2dSStefano Zampini 
125343eeeb2dSStefano Zampini .seealso: DMPlexUninterpolate(), DMPlexCreateFromCellList(), DMPlexCopyCoordinates()
125480330477SMatthew G. Knepley @*/
12559f074e33SMatthew G Knepley PetscErrorCode DMPlexInterpolate(DM dm, DM *dmInt)
12569f074e33SMatthew G Knepley {
12579a5b13a2SMatthew G Knepley   DM             idm, odm = dm;
12587bffde78SMichael Lange   PetscSF        sfPoint;
12592e1b13c2SMatthew G. Knepley   PetscInt       depth, dim, d;
126010a67516SMatthew G. Knepley   const char    *name;
1261b325530aSVaclav Hapla   PetscBool      flg=PETSC_TRUE;
12629f074e33SMatthew G Knepley   PetscErrorCode ierr;
12639f074e33SMatthew G Knepley 
12649f074e33SMatthew G Knepley   PetscFunctionBegin;
126510a67516SMatthew G. Knepley   PetscValidHeaderSpecific(dm, DM_CLASSID, 1);
126610a67516SMatthew G. Knepley   PetscValidPointer(dmInt, 2);
1267a72f3261SMatthew G. Knepley   ierr = PetscLogEventBegin(DMPLEX_Interpolate,dm,0,0,0);CHKERRQ(ierr);
12682e1b13c2SMatthew G. Knepley   ierr = DMPlexGetDepth(dm, &depth);CHKERRQ(ierr);
1269c73cfb54SMatthew G. Knepley   ierr = DMGetDimension(dm, &dim);CHKERRQ(ierr);
1270b21b8912SMatthew G. Knepley   if ((depth == dim) || (dim <= 1)) {
127176b791aaSMatthew G. Knepley     ierr = PetscObjectReference((PetscObject) dm);CHKERRQ(ierr);
127276b791aaSMatthew G. Knepley     idm  = dm;
1273b21b8912SMatthew G. Knepley   } else {
12749a5b13a2SMatthew G Knepley     for (d = 1; d < dim; ++d) {
12759a5b13a2SMatthew G Knepley       /* Create interpolated mesh */
127610a67516SMatthew G. Knepley       ierr = DMCreate(PetscObjectComm((PetscObject)dm), &idm);CHKERRQ(ierr);
12779a5b13a2SMatthew G Knepley       ierr = DMSetType(idm, DMPLEX);CHKERRQ(ierr);
1278c73cfb54SMatthew G. Knepley       ierr = DMSetDimension(idm, dim);CHKERRQ(ierr);
12797bffde78SMichael Lange       if (depth > 0) {
12807bffde78SMichael Lange         ierr = DMPlexInterpolateFaces_Internal(odm, 1, idm);CHKERRQ(ierr);
12817bffde78SMichael Lange         ierr = DMGetPointSF(odm, &sfPoint);CHKERRQ(ierr);
12822e72742cSMatthew G. Knepley         ierr = DMPlexInterpolatePointSF(idm, sfPoint);CHKERRQ(ierr);
12837bffde78SMichael Lange       }
12849a5b13a2SMatthew G Knepley       if (odm != dm) {ierr = DMDestroy(&odm);CHKERRQ(ierr);}
12859a5b13a2SMatthew G Knepley       odm = idm;
12869f074e33SMatthew G Knepley     }
128710a67516SMatthew G. Knepley     ierr = PetscObjectGetName((PetscObject) dm,  &name);CHKERRQ(ierr);
128810a67516SMatthew G. Knepley     ierr = PetscObjectSetName((PetscObject) idm,  name);CHKERRQ(ierr);
128910a67516SMatthew G. Knepley     ierr = DMPlexCopyCoordinates(dm, idm);CHKERRQ(ierr);
12905d80c0bfSVaclav Hapla     ierr = DMCopyLabels(dm, idm, PETSC_COPY_VALUES, PETSC_FALSE);CHKERRQ(ierr);
1291b325530aSVaclav Hapla     ierr = PetscOptionsGetBool(((PetscObject)dm)->options, ((PetscObject)dm)->prefix, "-dm_plex_interpolate_orient_interfaces", &flg, NULL);CHKERRQ(ierr);
1292*27d0e99aSVaclav Hapla     if (flg) {ierr = DMPlexOrientInterface_Internal(idm);CHKERRQ(ierr);}
129384699f70SSatish Balay   }
129443eeeb2dSStefano Zampini   {
129543eeeb2dSStefano Zampini     PetscBool            isper;
129643eeeb2dSStefano Zampini     const PetscReal      *maxCell, *L;
129743eeeb2dSStefano Zampini     const DMBoundaryType *bd;
129843eeeb2dSStefano Zampini 
129943eeeb2dSStefano Zampini     ierr = DMGetPeriodicity(dm,&isper,&maxCell,&L,&bd);CHKERRQ(ierr);
130043eeeb2dSStefano Zampini     ierr = DMSetPeriodicity(idm,isper,maxCell,L,bd);CHKERRQ(ierr);
130143eeeb2dSStefano Zampini   }
13029a5b13a2SMatthew G Knepley   *dmInt = idm;
1303a72f3261SMatthew G. Knepley   ierr = PetscLogEventEnd(DMPLEX_Interpolate,dm,0,0,0);CHKERRQ(ierr);
13049f074e33SMatthew G Knepley   PetscFunctionReturn(0);
13059f074e33SMatthew G Knepley }
130607b0a1fcSMatthew G Knepley 
130780330477SMatthew G. Knepley /*@
130880330477SMatthew G. Knepley   DMPlexCopyCoordinates - Copy coordinates from one mesh to another with the same vertices
130980330477SMatthew G. Knepley 
1310d083f849SBarry Smith   Collective on dmA
131180330477SMatthew G. Knepley 
131280330477SMatthew G. Knepley   Input Parameter:
131380330477SMatthew G. Knepley . dmA - The DMPlex object with initial coordinates
131480330477SMatthew G. Knepley 
131580330477SMatthew G. Knepley   Output Parameter:
131680330477SMatthew G. Knepley . dmB - The DMPlex object with copied coordinates
131780330477SMatthew G. Knepley 
131880330477SMatthew G. Knepley   Level: intermediate
131980330477SMatthew G. Knepley 
132080330477SMatthew G. Knepley   Note: This is typically used when adding pieces other than vertices to a mesh
132180330477SMatthew G. Knepley 
132265f90189SJed Brown .seealso: DMCopyLabels(), DMGetCoordinates(), DMGetCoordinatesLocal(), DMGetCoordinateDM(), DMGetCoordinateSection()
132380330477SMatthew G. Knepley @*/
132407b0a1fcSMatthew G Knepley PetscErrorCode DMPlexCopyCoordinates(DM dmA, DM dmB)
132507b0a1fcSMatthew G Knepley {
132607b0a1fcSMatthew G Knepley   Vec            coordinatesA, coordinatesB;
132743eeeb2dSStefano Zampini   VecType        vtype;
132807b0a1fcSMatthew G Knepley   PetscSection   coordSectionA, coordSectionB;
132907b0a1fcSMatthew G Knepley   PetscScalar   *coordsA, *coordsB;
13300bedd6beSMatthew G. Knepley   PetscInt       spaceDim, Nf, vStartA, vStartB, vEndA, vEndB, coordSizeB, v, d;
133143eeeb2dSStefano Zampini   PetscInt       cStartA, cEndA, cStartB, cEndB, cS, cE;
133243eeeb2dSStefano Zampini   PetscBool      lc = PETSC_FALSE;
133307b0a1fcSMatthew G Knepley   PetscErrorCode ierr;
133407b0a1fcSMatthew G Knepley 
133507b0a1fcSMatthew G Knepley   PetscFunctionBegin;
133643eeeb2dSStefano Zampini   PetscValidHeaderSpecific(dmA, DM_CLASSID, 1);
133743eeeb2dSStefano Zampini   PetscValidHeaderSpecific(dmB, DM_CLASSID, 2);
133876b791aaSMatthew G. Knepley   if (dmA == dmB) PetscFunctionReturn(0);
133907b0a1fcSMatthew G Knepley   ierr = DMPlexGetDepthStratum(dmA, 0, &vStartA, &vEndA);CHKERRQ(ierr);
134007b0a1fcSMatthew G Knepley   ierr = DMPlexGetDepthStratum(dmB, 0, &vStartB, &vEndB);CHKERRQ(ierr);
134107b0a1fcSMatthew G Knepley   if ((vEndA-vStartA) != (vEndB-vStartB)) SETERRQ2(PETSC_COMM_SELF, PETSC_ERR_ARG_SIZ, "The number of vertices in first DM %d != %d in the second DM", vEndA-vStartA, vEndB-vStartB);
134243eeeb2dSStefano Zampini   ierr = DMPlexGetHeightStratum(dmA, 0, &cStartA, &cEndA);CHKERRQ(ierr);
134343eeeb2dSStefano Zampini   ierr = DMPlexGetHeightStratum(dmB, 0, &cStartB, &cEndB);CHKERRQ(ierr);
134469d8a9ceSMatthew G. Knepley   ierr = DMGetCoordinateSection(dmA, &coordSectionA);CHKERRQ(ierr);
134569d8a9ceSMatthew G. Knepley   ierr = DMGetCoordinateSection(dmB, &coordSectionB);CHKERRQ(ierr);
1346972bc18aSToby Isaac   if (coordSectionA == coordSectionB) PetscFunctionReturn(0);
13470bedd6beSMatthew G. Knepley   ierr = PetscSectionGetNumFields(coordSectionA, &Nf);CHKERRQ(ierr);
13480bedd6beSMatthew G. Knepley   if (!Nf) PetscFunctionReturn(0);
13490bedd6beSMatthew G. Knepley   if (Nf > 1) SETERRQ1(PETSC_COMM_SELF, PETSC_ERR_ARG_SIZ, "The number of coordinate fields must be 1, not %D", Nf);
1350df26b574SMatthew G. Knepley   if (!coordSectionB) {
1351df26b574SMatthew G. Knepley     PetscInt dim;
1352df26b574SMatthew G. Knepley 
1353df26b574SMatthew G. Knepley     ierr = PetscSectionCreate(PetscObjectComm((PetscObject) coordSectionA), &coordSectionB);CHKERRQ(ierr);
1354df26b574SMatthew G. Knepley     ierr = DMGetCoordinateDim(dmA, &dim);CHKERRQ(ierr);
1355df26b574SMatthew G. Knepley     ierr = DMSetCoordinateSection(dmB, dim, coordSectionB);CHKERRQ(ierr);
1356df26b574SMatthew G. Knepley     ierr = PetscObjectDereference((PetscObject) coordSectionB);CHKERRQ(ierr);
1357df26b574SMatthew G. Knepley   }
135807b0a1fcSMatthew G Knepley   ierr = PetscSectionSetNumFields(coordSectionB, 1);CHKERRQ(ierr);
135907b0a1fcSMatthew G Knepley   ierr = PetscSectionGetFieldComponents(coordSectionA, 0, &spaceDim);CHKERRQ(ierr);
136007b0a1fcSMatthew G Knepley   ierr = PetscSectionSetFieldComponents(coordSectionB, 0, spaceDim);CHKERRQ(ierr);
136143eeeb2dSStefano Zampini   ierr = PetscSectionGetChart(coordSectionA, &cS, &cE);CHKERRQ(ierr);
136243eeeb2dSStefano Zampini   if (cStartA <= cS && cS < cEndA) { /* localized coordinates */
1363367003a6SStefano Zampini     if ((cEndA-cStartA) != (cEndB-cStartB)) SETERRQ2(PETSC_COMM_SELF, PETSC_ERR_ARG_SIZ, "The number of cells in first DM %D != %D in the second DM", cEndA-cStartA, cEndB-cStartB);
136443eeeb2dSStefano Zampini     cS = cS - cStartA + cStartB;
136543eeeb2dSStefano Zampini     cE = vEndB;
136643eeeb2dSStefano Zampini     lc = PETSC_TRUE;
136743eeeb2dSStefano Zampini   } else {
136843eeeb2dSStefano Zampini     cS = vStartB;
136943eeeb2dSStefano Zampini     cE = vEndB;
137043eeeb2dSStefano Zampini   }
137143eeeb2dSStefano Zampini   ierr = PetscSectionSetChart(coordSectionB, cS, cE);CHKERRQ(ierr);
137207b0a1fcSMatthew G Knepley   for (v = vStartB; v < vEndB; ++v) {
137307b0a1fcSMatthew G Knepley     ierr = PetscSectionSetDof(coordSectionB, v, spaceDim);CHKERRQ(ierr);
137407b0a1fcSMatthew G Knepley     ierr = PetscSectionSetFieldDof(coordSectionB, v, 0, spaceDim);CHKERRQ(ierr);
137507b0a1fcSMatthew G Knepley   }
137643eeeb2dSStefano Zampini   if (lc) { /* localized coordinates */
137743eeeb2dSStefano Zampini     PetscInt c;
137843eeeb2dSStefano Zampini 
137943eeeb2dSStefano Zampini     for (c = cS-cStartB; c < cEndB-cStartB; c++) {
138043eeeb2dSStefano Zampini       PetscInt dof;
138143eeeb2dSStefano Zampini 
138243eeeb2dSStefano Zampini       ierr = PetscSectionGetDof(coordSectionA, c + cStartA, &dof);CHKERRQ(ierr);
138343eeeb2dSStefano Zampini       ierr = PetscSectionSetDof(coordSectionB, c + cStartB, dof);CHKERRQ(ierr);
138443eeeb2dSStefano Zampini       ierr = PetscSectionSetFieldDof(coordSectionB, c + cStartB, 0, dof);CHKERRQ(ierr);
138543eeeb2dSStefano Zampini     }
138643eeeb2dSStefano Zampini   }
138707b0a1fcSMatthew G Knepley   ierr = PetscSectionSetUp(coordSectionB);CHKERRQ(ierr);
138807b0a1fcSMatthew G Knepley   ierr = PetscSectionGetStorageSize(coordSectionB, &coordSizeB);CHKERRQ(ierr);
138907b0a1fcSMatthew G Knepley   ierr = DMGetCoordinatesLocal(dmA, &coordinatesA);CHKERRQ(ierr);
13908b9ced59SLisandro Dalcin   ierr = VecCreate(PETSC_COMM_SELF, &coordinatesB);CHKERRQ(ierr);
139107b0a1fcSMatthew G Knepley   ierr = PetscObjectSetName((PetscObject) coordinatesB, "coordinates");CHKERRQ(ierr);
139207b0a1fcSMatthew G Knepley   ierr = VecSetSizes(coordinatesB, coordSizeB, PETSC_DETERMINE);CHKERRQ(ierr);
139343eeeb2dSStefano Zampini   ierr = VecGetBlockSize(coordinatesA, &d);CHKERRQ(ierr);
139443eeeb2dSStefano Zampini   ierr = VecSetBlockSize(coordinatesB, d);CHKERRQ(ierr);
139543eeeb2dSStefano Zampini   ierr = VecGetType(coordinatesA, &vtype);CHKERRQ(ierr);
139643eeeb2dSStefano Zampini   ierr = VecSetType(coordinatesB, vtype);CHKERRQ(ierr);
139707b0a1fcSMatthew G Knepley   ierr = VecGetArray(coordinatesA, &coordsA);CHKERRQ(ierr);
139807b0a1fcSMatthew G Knepley   ierr = VecGetArray(coordinatesB, &coordsB);CHKERRQ(ierr);
139907b0a1fcSMatthew G Knepley   for (v = 0; v < vEndB-vStartB; ++v) {
140043eeeb2dSStefano Zampini     PetscInt offA, offB;
140143eeeb2dSStefano Zampini 
140243eeeb2dSStefano Zampini     ierr = PetscSectionGetOffset(coordSectionA, v + vStartA, &offA);CHKERRQ(ierr);
140343eeeb2dSStefano Zampini     ierr = PetscSectionGetOffset(coordSectionB, v + vStartB, &offB);CHKERRQ(ierr);
140407b0a1fcSMatthew G Knepley     for (d = 0; d < spaceDim; ++d) {
140543eeeb2dSStefano Zampini       coordsB[offB+d] = coordsA[offA+d];
140643eeeb2dSStefano Zampini     }
140743eeeb2dSStefano Zampini   }
140843eeeb2dSStefano Zampini   if (lc) { /* localized coordinates */
140943eeeb2dSStefano Zampini     PetscInt c;
141043eeeb2dSStefano Zampini 
141143eeeb2dSStefano Zampini     for (c = cS-cStartB; c < cEndB-cStartB; c++) {
141243eeeb2dSStefano Zampini       PetscInt dof, offA, offB;
141343eeeb2dSStefano Zampini 
141443eeeb2dSStefano Zampini       ierr = PetscSectionGetOffset(coordSectionA, c + cStartA, &offA);CHKERRQ(ierr);
141543eeeb2dSStefano Zampini       ierr = PetscSectionGetOffset(coordSectionB, c + cStartB, &offB);CHKERRQ(ierr);
141643eeeb2dSStefano Zampini       ierr = PetscSectionGetDof(coordSectionA, c + cStartA, &dof);CHKERRQ(ierr);
1417580bdb30SBarry Smith       ierr = PetscArraycpy(coordsB + offB,coordsA + offA,dof);CHKERRQ(ierr);
141807b0a1fcSMatthew G Knepley     }
141907b0a1fcSMatthew G Knepley   }
142007b0a1fcSMatthew G Knepley   ierr = VecRestoreArray(coordinatesA, &coordsA);CHKERRQ(ierr);
142107b0a1fcSMatthew G Knepley   ierr = VecRestoreArray(coordinatesB, &coordsB);CHKERRQ(ierr);
142207b0a1fcSMatthew G Knepley   ierr = DMSetCoordinatesLocal(dmB, coordinatesB);CHKERRQ(ierr);
142307b0a1fcSMatthew G Knepley   ierr = VecDestroy(&coordinatesB);CHKERRQ(ierr);
142407b0a1fcSMatthew G Knepley   PetscFunctionReturn(0);
142507b0a1fcSMatthew G Knepley }
14265c386225SMatthew G. Knepley 
14274e3744c5SMatthew G. Knepley /*@
14284e3744c5SMatthew G. Knepley   DMPlexUninterpolate - Take in a mesh with all intermediate faces, edges, etc. and return a cell-vertex mesh
14294e3744c5SMatthew G. Knepley 
1430d083f849SBarry Smith   Collective on dm
14314e3744c5SMatthew G. Knepley 
14324e3744c5SMatthew G. Knepley   Input Parameter:
14334e3744c5SMatthew G. Knepley . dm - The complete DMPlex object
14344e3744c5SMatthew G. Knepley 
14354e3744c5SMatthew G. Knepley   Output Parameter:
14364e3744c5SMatthew G. Knepley . dmUnint - The DMPlex object with only cells and vertices
14374e3744c5SMatthew G. Knepley 
14384e3744c5SMatthew G. Knepley   Level: intermediate
14394e3744c5SMatthew G. Knepley 
144095452b02SPatrick Sanan   Notes:
144195452b02SPatrick Sanan     It does not copy over the coordinates.
144243eeeb2dSStefano Zampini 
144343eeeb2dSStefano Zampini .seealso: DMPlexInterpolate(), DMPlexCreateFromCellList(), DMPlexCopyCoordinates()
14444e3744c5SMatthew G. Knepley @*/
14454e3744c5SMatthew G. Knepley PetscErrorCode DMPlexUninterpolate(DM dm, DM *dmUnint)
14464e3744c5SMatthew G. Knepley {
14474e3744c5SMatthew G. Knepley   DM             udm;
1448c9f63434SStefano Zampini   PetscInt       dim, vStart, vEnd, cStart, cEnd, cMax, c, maxConeSize = 0, *cone;
14494e3744c5SMatthew G. Knepley   PetscErrorCode ierr;
14504e3744c5SMatthew G. Knepley 
14514e3744c5SMatthew G. Knepley   PetscFunctionBegin;
145243eeeb2dSStefano Zampini   PetscValidHeaderSpecific(dm, DM_CLASSID, 1);
145343eeeb2dSStefano Zampini   PetscValidPointer(dmUnint, 2);
1454c73cfb54SMatthew G. Knepley   ierr = DMGetDimension(dm, &dim);CHKERRQ(ierr);
14554e3744c5SMatthew G. Knepley   if (dim <= 1) {
14564e3744c5SMatthew G. Knepley     ierr = PetscObjectReference((PetscObject) dm);CHKERRQ(ierr);
1457595d4abbSMatthew G. Knepley     *dmUnint = dm;
1458595d4abbSMatthew G. Knepley     PetscFunctionReturn(0);
14594e3744c5SMatthew G. Knepley   }
1460595d4abbSMatthew G. Knepley   ierr = DMPlexGetDepthStratum(dm, 0, &vStart, &vEnd);CHKERRQ(ierr);
1461595d4abbSMatthew G. Knepley   ierr = DMPlexGetHeightStratum(dm, 0, &cStart, &cEnd);CHKERRQ(ierr);
1462c9f63434SStefano Zampini   ierr = DMPlexGetHybridBounds(dm, &cMax, NULL, NULL, NULL);CHKERRQ(ierr);
14634e3744c5SMatthew G. Knepley   ierr = DMCreate(PetscObjectComm((PetscObject) dm), &udm);CHKERRQ(ierr);
14644e3744c5SMatthew G. Knepley   ierr = DMSetType(udm, DMPLEX);CHKERRQ(ierr);
1465c73cfb54SMatthew G. Knepley   ierr = DMSetDimension(udm, dim);CHKERRQ(ierr);
14664e3744c5SMatthew G. Knepley   ierr = DMPlexSetChart(udm, cStart, vEnd);CHKERRQ(ierr);
14674e3744c5SMatthew G. Knepley   for (c = cStart; c < cEnd; ++c) {
1468595d4abbSMatthew G. Knepley     PetscInt *closure = NULL, closureSize, cl, coneSize = 0;
14694e3744c5SMatthew G. Knepley 
14704e3744c5SMatthew G. Knepley     ierr = DMPlexGetTransitiveClosure(dm, c, PETSC_TRUE, &closureSize, &closure);CHKERRQ(ierr);
14714e3744c5SMatthew G. Knepley     for (cl = 0; cl < closureSize*2; cl += 2) {
14724e3744c5SMatthew G. Knepley       const PetscInt p = closure[cl];
14734e3744c5SMatthew G. Knepley 
14744e3744c5SMatthew G. Knepley       if ((p >= vStart) && (p < vEnd)) ++coneSize;
14754e3744c5SMatthew G. Knepley     }
14764e3744c5SMatthew G. Knepley     ierr = DMPlexRestoreTransitiveClosure(dm, c, PETSC_TRUE, &closureSize, &closure);CHKERRQ(ierr);
14774e3744c5SMatthew G. Knepley     ierr = DMPlexSetConeSize(udm, c, coneSize);CHKERRQ(ierr);
1478595d4abbSMatthew G. Knepley     maxConeSize = PetscMax(maxConeSize, coneSize);
14794e3744c5SMatthew G. Knepley   }
14804e3744c5SMatthew G. Knepley   ierr = DMSetUp(udm);CHKERRQ(ierr);
1481785e854fSJed Brown   ierr = PetscMalloc1(maxConeSize, &cone);CHKERRQ(ierr);
14824e3744c5SMatthew G. Knepley   for (c = cStart; c < cEnd; ++c) {
1483595d4abbSMatthew G. Knepley     PetscInt *closure = NULL, closureSize, cl, coneSize = 0;
14844e3744c5SMatthew G. Knepley 
14854e3744c5SMatthew G. Knepley     ierr = DMPlexGetTransitiveClosure(dm, c, PETSC_TRUE, &closureSize, &closure);CHKERRQ(ierr);
14864e3744c5SMatthew G. Knepley     for (cl = 0; cl < closureSize*2; cl += 2) {
14874e3744c5SMatthew G. Knepley       const PetscInt p = closure[cl];
14884e3744c5SMatthew G. Knepley 
14894e3744c5SMatthew G. Knepley       if ((p >= vStart) && (p < vEnd)) cone[coneSize++] = p;
14904e3744c5SMatthew G. Knepley     }
14914e3744c5SMatthew G. Knepley     ierr = DMPlexRestoreTransitiveClosure(dm, c, PETSC_TRUE, &closureSize, &closure);CHKERRQ(ierr);
14924e3744c5SMatthew G. Knepley     ierr = DMPlexSetCone(udm, c, cone);CHKERRQ(ierr);
14934e3744c5SMatthew G. Knepley   }
14944e3744c5SMatthew G. Knepley   ierr = PetscFree(cone);CHKERRQ(ierr);
1495c9f63434SStefano Zampini   ierr = DMPlexSetHybridBounds(udm, cMax, PETSC_DETERMINE, PETSC_DETERMINE, PETSC_DETERMINE);CHKERRQ(ierr);
14964e3744c5SMatthew G. Knepley   ierr = DMPlexSymmetrize(udm);CHKERRQ(ierr);
14974e3744c5SMatthew G. Knepley   ierr = DMPlexStratify(udm);CHKERRQ(ierr);
14985c7de58aSMatthew G. Knepley   /* Reduce SF */
14995c7de58aSMatthew G. Knepley   {
15005c7de58aSMatthew G. Knepley     PetscSF            sfPoint, sfPointUn;
15015c7de58aSMatthew G. Knepley     const PetscSFNode *remotePoints;
15025c7de58aSMatthew G. Knepley     const PetscInt    *localPoints;
15035c7de58aSMatthew G. Knepley     PetscSFNode       *remotePointsUn;
15045c7de58aSMatthew G. Knepley     PetscInt          *localPointsUn;
15055c7de58aSMatthew G. Knepley     PetscInt           vEnd, numRoots, numLeaves, l;
15065c7de58aSMatthew G. Knepley     PetscInt           numLeavesUn = 0, n = 0;
15075c7de58aSMatthew G. Knepley     PetscErrorCode     ierr;
15085c7de58aSMatthew G. Knepley 
15095c7de58aSMatthew G. Knepley     /* Get original SF information */
15105c7de58aSMatthew G. Knepley     ierr = DMGetPointSF(dm, &sfPoint);CHKERRQ(ierr);
15115c7de58aSMatthew G. Knepley     ierr = DMGetPointSF(udm, &sfPointUn);CHKERRQ(ierr);
15125c7de58aSMatthew G. Knepley     ierr = DMPlexGetDepthStratum(dm, 0, NULL, &vEnd);CHKERRQ(ierr);
15135c7de58aSMatthew G. Knepley     ierr = PetscSFGetGraph(sfPoint, &numRoots, &numLeaves, &localPoints, &remotePoints);CHKERRQ(ierr);
15145c7de58aSMatthew G. Knepley     /* Allocate space for cells and vertices */
15155c7de58aSMatthew G. Knepley     for (l = 0; l < numLeaves; ++l) if (localPoints[l] < vEnd) numLeavesUn++;
15165c7de58aSMatthew G. Knepley     /* Fill in leaves */
15175c7de58aSMatthew G. Knepley     if (vEnd >= 0) {
15185c7de58aSMatthew G. Knepley       ierr = PetscMalloc1(numLeavesUn, &remotePointsUn);CHKERRQ(ierr);
15195c7de58aSMatthew G. Knepley       ierr = PetscMalloc1(numLeavesUn, &localPointsUn);CHKERRQ(ierr);
15205c7de58aSMatthew G. Knepley       for (l = 0; l < numLeaves; l++) {
15215c7de58aSMatthew G. Knepley         if (localPoints[l] < vEnd) {
15225c7de58aSMatthew G. Knepley           localPointsUn[n]        = localPoints[l];
15235c7de58aSMatthew G. Knepley           remotePointsUn[n].rank  = remotePoints[l].rank;
15245c7de58aSMatthew G. Knepley           remotePointsUn[n].index = remotePoints[l].index;
15255c7de58aSMatthew G. Knepley           ++n;
15265c7de58aSMatthew G. Knepley         }
15275c7de58aSMatthew G. Knepley       }
15285c7de58aSMatthew G. Knepley       if (n != numLeavesUn) SETERRQ2(PETSC_COMM_SELF, PETSC_ERR_PLIB, "Inconsistent number of leaves %d != %d", n, numLeavesUn);
15295c7de58aSMatthew G. Knepley       ierr = PetscSFSetGraph(sfPointUn, vEnd, numLeavesUn, localPointsUn, PETSC_OWN_POINTER, remotePointsUn, PETSC_OWN_POINTER);CHKERRQ(ierr);
15305c7de58aSMatthew G. Knepley     }
15315c7de58aSMatthew G. Knepley   }
153243eeeb2dSStefano Zampini   {
153343eeeb2dSStefano Zampini     PetscBool            isper;
153443eeeb2dSStefano Zampini     const PetscReal      *maxCell, *L;
153543eeeb2dSStefano Zampini     const DMBoundaryType *bd;
153643eeeb2dSStefano Zampini 
153743eeeb2dSStefano Zampini     ierr = DMGetPeriodicity(dm,&isper,&maxCell,&L,&bd);CHKERRQ(ierr);
153843eeeb2dSStefano Zampini     ierr = DMSetPeriodicity(udm,isper,maxCell,L,bd);CHKERRQ(ierr);
153943eeeb2dSStefano Zampini   }
154043eeeb2dSStefano Zampini 
15414e3744c5SMatthew G. Knepley   *dmUnint = udm;
15424e3744c5SMatthew G. Knepley   PetscFunctionReturn(0);
15434e3744c5SMatthew G. Knepley }
1544