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 <dolfinx/common/MPI.h>
#include <dolfinx/graph/AdjacencyList.h>
#include <dolfinx/graph/partition.h>
#include <functional>
#include <xtl/xspan.hpp>
 
namespace dolfinx
{
namespace fem
{
class ElementDofLayout;
}
 
namespace mesh
{
enum class CellType;
enum class GhostMode : int;
class Mesh;
 
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,
                      const xtl::span<const std::int32_t>& entities, int dim);
 
xt::xtensor<double, 2>
cell_normals(const Mesh& mesh, int dim,
             const xtl::span<const std::int32_t>& entities);
 
xt::xtensor<double, 2> midpoints(const mesh::Mesh& mesh, int dim,
                                 const xtl::span<const std::int32_t>& entities);
 
std::vector<std::int32_t> locate_entities(
    const mesh::Mesh& mesh, int dim,
    const std::function<xt::xtensor<bool, 1>(const xt::xtensor<double, 2>&)>&
        marker);
 
std::vector<std::int32_t> locate_entities_boundary(
    const mesh::Mesh& mesh, int dim,
    const std::function<xt::xtensor<bool, 1>(const xt::xtensor<double, 2>&)>&
        marker);
 
xt::xtensor<std::int32_t, 2>
entities_to_geometry(const mesh::Mesh& mesh, int dim,
                     const xtl::span<const std::int32_t>& entity_list,
                     bool orient);
 
std::vector<std::int32_t> exterior_facet_indices(const Mesh& mesh);
 
graph::AdjacencyList<std::int32_t>
partition_cells_graph(MPI_Comm comm, int n, int tdim,
                      const graph::AdjacencyList<std::int64_t>& cells,
                      mesh::GhostMode ghost_mode);
 
graph::AdjacencyList<std::int32_t>
partition_cells_graph(MPI_Comm comm, int n, int tdim,
                      const graph::AdjacencyList<std::int64_t>& cells,
                      mesh::GhostMode ghost_mode,
                      const graph::partition_fn& partfn);
 
} // namespace mesh
} // namespace dolfinx