grid_tools_cache.h

// ---------------------------------------------------------------------
//
// Copyright (C) 2017 by the deal.II authors
//
// This file is part of the deal.II library.
//
// The deal.II library is free software; you can use it, redistribute
// it, and/or modify it under the terms of the GNU Lesser General
// Public License as published by the Free Software Foundation; either
// version 2.1 of the License, or (at your option) any later version.
// The full text of the license can be found in the file LICENSE at
// the top level of the deal.II distribution.
//
// ---------------------------------------------------------------------

#ifndef dealii_grid_grid_tools_cache_h
#define dealii_grid_grid_tools_cache_h


#include <deal.II/base/config.h>
#include <deal.II/base/subscriptor.h>
#include <deal.II/base/exceptions.h>
#include <deal.II/base/point.h>

#include <deal.II/grid/tria.h>
#include <deal.II/grid/tria_iterator.h>
#include <deal.II/grid/tria_accessor.h>
#include <deal.II/grid/grid_tools_cache_update_flags.h>
#include <deal.II/fe/mapping_q1.h>

#include <deal.II/numerics/kdtree.h>

#include <boost/signals2.hpp>

#include <cmath>

DEAL_II_NAMESPACE_OPEN

namespace GridTools
{

  template <int dim, int spacedim = dim>
  class Cache : public Subscriptor
  {
  public:
    Cache (const Triangulation<dim,spacedim> &tria,
           const Mapping<dim,spacedim> &mapping=StaticMappingQ1<dim,spacedim>::mapping);

    ~Cache();

    void mark_for_update(const CacheUpdateFlags &flags=update_all);


    const std::vector< std::set<typename Triangulation<dim,spacedim>::active_cell_iterator> >
    &get_vertex_to_cell_map() const;

    const std::vector<std::vector<Tensor<1,spacedim>>>
    &get_vertex_to_cell_centers_directions() const;

    const std::map<unsigned int, Point<spacedim> >
    &get_used_vertices() const;

    const Triangulation<dim,spacedim> &get_triangulation() const;

    const Mapping<dim,spacedim> &get_mapping() const;

#ifdef DEAL_II_WITH_NANOFLANN

    const KDTree<spacedim> &get_vertex_kdtree() const;
#endif

  private:
    mutable CacheUpdateFlags update_flags;

    SmartPointer<const Triangulation<dim,spacedim>,
                 Cache<dim,spacedim>> tria;

    SmartPointer<const Mapping<dim,spacedim>,
                 Cache<dim,spacedim>> mapping;


    mutable std::vector< std::set<typename Triangulation<dim,spacedim>::active_cell_iterator> >
    vertex_to_cells;

    mutable std::vector<std::vector<Tensor<1,spacedim>>> vertex_to_cell_centers;

#ifdef DEAL_II_WITH_NANOFLANN

    mutable KDTree<spacedim> vertex_kdtree;
#endif

    mutable std::map<unsigned int, Point<spacedim>> used_vertices;

    boost::signals2::connection tria_signal;
  };



  // Inline functions
  template<int dim, int spacedim>
  inline const Triangulation<dim, spacedim> &
  Cache<dim,spacedim>::get_triangulation() const
  {
    return *tria;
  }



  template<int dim, int spacedim>
  inline const Mapping<dim, spacedim> &
  Cache<dim,spacedim>::get_mapping() const
  {
    return *mapping;
  }
}



DEAL_II_NAMESPACE_CLOSE

#endif