utils.h

// Copyright (C) 2019-2020 Garth N. Wells
//
// This file is part of DOLFINx (https://www.fenicsproject.org)
//
// SPDX-License-Identifier:    LGPL-3.0-or-later
 
#pragma once
 
#include <basix/mdspan.hpp>
#include <dolfinx/graph/AdjacencyList.h>
#include <dolfinx/graph/partition.h>
#include <functional>
#include <mpi.h>
#include <span>
 
namespace dolfinx::fem
{
class ElementDofLayout;
}
 
namespace dolfinx::mesh
{
enum class CellType;
class Mesh;
class Topology;
 
enum class GhostMode : int
{
  none,
  shared_facet,
  shared_vertex
};
 
// See https://github.com/doxygen/doxygen/issues/9552
/*
*/
using CellPartitionFunction = std::function<graph::AdjacencyList<std::int32_t>(
    MPI_Comm comm, int nparts, int tdim,
    const graph::AdjacencyList<std::int64_t>& cells)>;
 
graph::AdjacencyList<std::int64_t>
extract_topology(const CellType& cell_type, const fem::ElementDofLayout& layout,
                 const graph::AdjacencyList<std::int64_t>& cells);
 
std::vector<double> h(const Mesh& mesh, std::span<const std::int32_t> entities,
                      int dim);
 
std::vector<double> cell_normals(const Mesh& mesh, int dim,
                                 std::span<const std::int32_t> entities);
 
std::vector<double> compute_midpoints(const Mesh& mesh, int dim,
                                      std::span<const std::int32_t> entities);
 
std::vector<std::int32_t> locate_entities(
    const Mesh& mesh, int dim,
    const std::function<std::vector<std::int8_t>(
        std::experimental::mdspan<
            const double,
            std::experimental::extents<
                std::size_t, 3, std::experimental::dynamic_extent>>)>& marker);
 
std::vector<std::int32_t> locate_entities_boundary(
    const Mesh& mesh, int dim,
    const std::function<std::vector<std::int8_t>(
        std::experimental::mdspan<
            const double,
            std::experimental::extents<
                std::size_t, 3, std::experimental::dynamic_extent>>)>& marker);
 
std::vector<std::int32_t>
entities_to_geometry(const Mesh& mesh, int dim,
                     std::span<const std::int32_t> entities, bool orient);
 
std::vector<std::int32_t> exterior_facet_indices(const Topology& topology);
 
CellPartitionFunction create_cell_partitioner(mesh::GhostMode ghost_mode
                                              = mesh::GhostMode::none,
                                              const graph::partition_fn& partfn
                                              = &graph::partition_graph);
 
std::vector<std::int32_t> compute_incident_entities(
    const Mesh& mesh, std::span<const std::int32_t> entities, int d0, int d1);
 
} // namespace dolfinx::mesh