add explicit part_id attribute to InterPartAdjacencyGroup

Author: majosm

meshmode/distributed.py

@@ -173,7 +173,7 @@ def make_remote_group_infos(
                 inter_part_adj_groups=[
                     fagrp for fagrp in local_vol_mesh.facial_adjacency_groups[igrp]
                     if isinstance(fagrp, InterPartAdjacencyGroup)
-                    and fagrp.boundary_tag.part_id == remote_part_id],
+                    and fagrp.part_id == remote_part_id],
                 vol_elem_indices=np.concatenate([
                     actx.to_numpy(batch.from_element_indices)
                     for batch in bdry_conn.groups[igrp].batches]),
@@ -450,7 +450,7 @@ def get_connected_parts(mesh: Mesh) -> "Set[PartID]":
     assert mesh.facial_adjacency_groups is not None
 
     return {
-            grp.boundary_tag.part_id
+            grp.part_id
             for fagrp_list in mesh.facial_adjacency_groups
             for grp in fagrp_list
             if isinstance(grp, InterPartAdjacencyGroup)}

meshmode/mesh/__init__.py

@@ -784,6 +784,10 @@ class InterPartAdjacencyGroup(BoundaryAdjacencyGroup):
         The boundary tag identifier of this group. Will be an instance of
         :class:`~meshmode.mesh.BTAG_PARTITION`.
 
+    .. attribute:: part_id
+
+        The identifier of the neighboring part.
+
     .. attribute:: elements
 
         Group-local element numbers.
@@ -819,13 +823,15 @@ class InterPartAdjacencyGroup(BoundaryAdjacencyGroup):
     .. versionadded:: 2017.1
     """
 
+    part_id: PartID
     neighbors: np.ndarray
     neighbor_faces: np.ndarray
     aff_map: AffineMap
 
     def __eq__(self, other):
         return (
             super().__eq__(other)
+            and np.array_equal(self.part_id, other.part_id)
             and np.array_equal(self.neighbors, other.neighbors)
             and np.array_equal(self.neighbor_faces, other.neighbor_faces)
             and self.aff_map == other.aff_map)
@@ -837,6 +843,7 @@ def as_python(self):
         return self._as_python(
             igroup=self.igroup,
             boundary_tag=_boundary_tag_as_python(self.boundary_tag),
+            part_id=self.part_id,
             elements=_numpy_array_as_python(self.elements),
             element_faces=_numpy_array_as_python(self.element_faces),
             neighbors=_numpy_array_as_python(self.neighbors),

meshmode/mesh/processing.py

@@ -374,6 +374,7 @@ def _create_self_to_other_adjacency_groups(
                         InterPartAdjacencyGroup(
                             igroup=igrp,
                             boundary_tag=BTAG_PARTITION(neighbor_part_id),
+                            part_id=neighbor_part_id,
                             elements=elements,
                             element_faces=element_faces,
                             neighbors=neighbors,