PointListAdaptor.h

Go to the documentation of this file.

//* This file is part of the MOOSE framework
//* https://mooseframework.inl.gov
//*
//* All rights reserved, see COPYRIGHT for full restrictions
//* https://github.com/idaholab/moose/blob/master/COPYRIGHT
//*
//* Licensed under LGPL 2.1, please see LICENSE for details
//* https://www.gnu.org/licenses/lgpl-2.1.html
 
#pragma once
 
#include "libmesh/nanoflann.hpp"
#include "libmesh/utility.h"
#include "libmesh/point.h"
 
#include <iterator>
 
template <typename PointObject>
class PointListAdaptor
{
public:
  using Iterator = typename std::vector<PointObject>::const_iterator;
 
  PointListAdaptor(Iterator begin, Iterator end)
    : _begin(begin), _end(end), _size(std::distance(begin, end))
  {
  }
 
private:
  const Iterator _begin;
 
  const Iterator _end;
 
  std::size_t _size;
 
public:
  using coord_t = Real;
 
  inline size_t kdtree_get_point_count() const { return _size; }
 
  const Point & getPoint(const PointObject & item) const;
 
  inline coord_t kdtree_distance(const coord_t * p1, const size_t idx_p2, size_t /*size*/) const
  {
    mooseAssert(idx_p2 < _size,
                "The point index should be less than"
                "total number of points used to build"
                "the KDTree.");
 
    auto it = _begin;
    std::advance(it, idx_p2);
    const Point & p2 = getPoint(*it);
 
    coord_t dist = 0.0;
 
    for (const auto i : make_range(Moose::dim))
      dist += Utility::pow<2>(p1[i] - p2(i));
 
    return dist;
  }
 
  inline coord_t kdtree_get_pt(const size_t idx, int dim) const
  {
    mooseAssert(dim < (int)Moose::dim,
                "The required component number should be less than the LIBMESH_DIM.");
    mooseAssert(idx < _size,
                "The index of the point should be less"
                "than total number of points used to"
                "construct the KDTree.");
 
    auto it = _begin;
    std::advance(it, idx);
    const Point & p = getPoint(*it);
 
    return p(dim);
  }
 
  template <class BBOX>
  bool kdtree_get_bbox(BBOX & /* bb */) const
  {
    return false;
  }
};
 
// Specialization for PointListAdaptor<Point> (provide your own for custom types)
template <>
inline const Point &
PointListAdaptor<Point>::getPoint(const Point & item) const
{
  return item;
}