tria_base.h

// ---------------------------------------------------------------------
//
// Copyright (C) 2008 - 2018 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.md at
// the top level directory of deal.II.
//
// ---------------------------------------------------------------------

#ifndef dealii_distributed_tria_base_h
#define dealii_distributed_tria_base_h


#include <deal.II/base/config.h>

#include <deal.II/base/mpi.h>
#include <deal.II/base/smartpointer.h>
#include <deal.II/base/subscriptor.h>
#include <deal.II/base/template_constraints.h>

#include <deal.II/grid/tria.h>

#include <functional>
#include <list>
#include <set>
#include <tuple>
#include <utility>
#include <vector>


DEAL_II_NAMESPACE_OPEN

namespace parallel
{
  template <int dim, int spacedim = dim>
  class Triangulation : public ::Triangulation<dim, spacedim>
  {
  public:
    Triangulation(
      MPI_Comm mpi_communicator,
      const typename ::Triangulation<dim, spacedim>::MeshSmoothing
                 smooth_grid = (::Triangulation<dim, spacedim>::none),
      const bool check_for_distorted_cells = false);

    virtual ~Triangulation() override;

    virtual MPI_Comm
    get_communicator() const;

    virtual void
    copy_triangulation(
      const ::Triangulation<dim, spacedim> &old_tria) override;

    const std::vector<unsigned int> &
    n_locally_owned_active_cells_per_processor() const;


    unsigned int
    n_locally_owned_active_cells() const;

    virtual types::global_dof_index
    n_global_active_cells() const override;

    virtual std::size_t
    memory_consumption() const override;


    virtual unsigned int
    n_global_levels() const override;

    types::subdomain_id
    locally_owned_subdomain() const override;

    const std::set<types::subdomain_id> &
    ghost_owners() const;

    const std::set<types::subdomain_id> &
    level_ghost_owners() const;

    virtual std::map<unsigned int, std::set<::types::subdomain_id>>
    compute_vertices_with_ghost_neighbors() const;

  protected:
    MPI_Comm mpi_communicator;

    types::subdomain_id my_subdomain;

    types::subdomain_id n_subdomains;

    struct NumberCache
    {
      std::vector<unsigned int> n_locally_owned_active_cells;
      types::global_dof_index n_global_active_cells;
      unsigned int n_global_levels;
      std::set<types::subdomain_id> ghost_owners;
      std::set<types::subdomain_id> level_ghost_owners;

      NumberCache();
    };

    NumberCache number_cache;

    virtual void
    update_number_cache();

    void
    fill_level_ghost_owners();
  };

} // namespace parallel

DEAL_II_NAMESPACE_CLOSE

#endif