DOLFIN-X
DOLFIN-X C++ interface
utils.h
1 // Copyright (C) 2012-2020 Chris Richardson
2 //
3 // This file is part of DOLFINX (https://www.fenicsproject.org)
4 //
5 // SPDX-License-Identifier: LGPL-3.0-or-later
6 
7 #pragma once
8 
9 #include <Eigen/Core>
10 #include <cstdint>
11 #include <dolfinx/common/MPI.h>
12 #include <map>
13 #include <memory>
14 #include <set>
15 #include <vector>
16 
17 namespace dolfinx
18 {
19 
20 namespace mesh
21 {
22 class Mesh;
23 enum class GhostMode;
24 } // namespace mesh
25 
26 namespace common
27 {
28 class IndexMap;
29 }
30 
31 namespace refinement
32 {
33 
40 std::pair<MPI_Comm, std::map<std::int32_t, std::vector<int>>>
41 compute_edge_sharing(const mesh::Mesh& mesh);
42 
50  const MPI_Comm& neighbor_comm,
51  const std::vector<std::vector<std::int32_t>>& marked_for_update,
52  std::vector<bool>& marked_edges, const common::IndexMap& map_e);
53 
62 std::pair<std::map<std::int32_t, std::int64_t>,
63  Eigen::Array<double, Eigen::Dynamic, Eigen::Dynamic, Eigen::RowMajor>>
65  const MPI_Comm& neighbor_comm,
66  const std::map<std::int32_t, std::vector<std::int32_t>>& shared_edges,
67  const mesh::Mesh& mesh, const std::vector<bool>& marked_edges);
68 
77 mesh::Mesh
78 partition(const mesh::Mesh& old_mesh,
79  const graph::AdjacencyList<std::int64_t>& cell_topology,
80  const Eigen::Array<double, Eigen::Dynamic, Eigen::Dynamic,
81  Eigen::RowMajor>& new_vertex_coordinates,
82  bool redistribute, mesh::GhostMode ghost_mode);
83 
92 std::vector<std::int64_t>
93 adjust_indices(const std::shared_ptr<const common::IndexMap>& index_map,
94  std::int32_t n);
95 
96 } // namespace refinement
97 } // namespace dolfinx
GhostMode
Enum for different partitioning ghost modes.
Definition: Mesh.h:48
mesh::Mesh partition(const mesh::Mesh &old_mesh, const graph::AdjacencyList< std::int64_t > &cell_topology, const Eigen::Array< double, Eigen::Dynamic, Eigen::Dynamic, Eigen::RowMajor > &new_vertex_coordinates, bool redistribute, mesh::GhostMode ghost_mode)
Use vertex and topology data to partition new mesh across processes.
Definition: utils.cpp:312
std::vector< std::int64_t > adjust_indices(const std::shared_ptr< const common::IndexMap > &index_map, std::int32_t n)
Adjust indices to account for extra n values on each process This is a utility to help add new topolo...
Definition: utils.cpp:284
std::pair< std::map< std::int32_t, std::int64_t >, Eigen::Array< double, Eigen::Dynamic, Eigen::Dynamic, Eigen::RowMajor > > create_new_vertices(const MPI_Comm &neighbor_comm, const std::map< std::int32_t, std::vector< std::int32_t >> &shared_edges, const mesh::Mesh &mesh, const std::vector< bool > &marked_edges)
Add new vertex for each marked edge, and create new_vertex_coordinates and global_edge->new_vertex ma...
Definition: utils.cpp:188
std::pair< MPI_Comm, std::map< std::int32_t, std::vector< int > > > compute_edge_sharing(const mesh::Mesh &mesh)
Compute the sharing of edges between processes. The resulting MPI_Comm is over the neighborhood of sh...
Definition: utils.cpp:109
void update_logical_edgefunction(const MPI_Comm &neighbor_comm, const std::vector< std::vector< std::int32_t >> &marked_for_update, std::vector< bool > &marked_edges, const common::IndexMap &map_e)
Transfer marked edges between processes.
Definition: utils.cpp:155