utils.h

// Copyright (C) 2019-2021 Garth N. Wells and Jørgen S. Dokken
//
// This file is part of DOLFINx (https://www.fenicsproject.org)
//
// SPDX-License-Identifier:    LGPL-3.0-or-later
 
#pragma once
 
#include <array>
#include <dolfinx/graph/AdjacencyList.h>
#include <vector>
#include <xtensor/xfixed.hpp>
#include <xtensor/xshape.hpp>
#include <xtensor/xtensor.hpp>
#include <xtl/xspan.hpp>
 
namespace dolfinx::mesh
{
class Mesh;
}
 
namespace dolfinx::geometry
{
class BoundingBoxTree;
 
BoundingBoxTree
create_midpoint_tree(const mesh::Mesh& mesh, int tdim,
                     const xtl::span<const std::int32_t>& entity_indices);
 
std::vector<std::array<int, 2>>
compute_collisions(const BoundingBoxTree& tree0, const BoundingBoxTree& tree1);
 
graph::AdjacencyList<std::int32_t>
compute_collisions(const BoundingBoxTree& tree,
                   const xt::xtensor<double, 2>& points);
 
std::vector<std::int32_t> compute_closest_entity(
    const BoundingBoxTree& tree, const BoundingBoxTree& midpoint_tree,
    const mesh::Mesh& mesh, const xt::xtensor<double, 2>& points);
 
double compute_squared_distance_bbox(
    const xt::xtensor_fixed<double, xt::xshape<2, 3>>& b,
    const xt::xtensor_fixed<double, xt::xshape<3>>& x);
 
xt::xtensor<double, 2>
shortest_vector(const mesh::Mesh& mesh, int dim,
                const xtl::span<const std::int32_t>& entities,
                const xt::xtensor<double, 2>& points);
 
xt::xtensor<double, 1>
squared_distance(const mesh::Mesh& mesh, int dim,
                 const xtl::span<const std::int32_t>& entities,
                 const xt::xtensor<double, 2>& points);
 
graph::AdjacencyList<int> compute_colliding_cells(
    const mesh::Mesh& mesh,
    const graph::AdjacencyList<std::int32_t>& candidate_cells,
    const xt::xtensor<double, 2>& points);
} // namespace dolfinx::geometry