shared_tria.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 at
// the top level of the deal.II distribution.
//
// ---------------------------------------------------------------------

#ifndef dealii_distributed_shared_tria_h
#define dealii_distributed_shared_tria_h


#include <deal.II/base/config.h>
#include <deal.II/base/subscriptor.h>
#include <deal.II/base/smartpointer.h>
#include <deal.II/base/template_constraints.h>
#include <deal.II/grid/tria.h>

#include <deal.II/distributed/tria_base.h>


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

#ifdef DEAL_II_WITH_MPI
#  include <mpi.h>
#endif


DEAL_II_NAMESPACE_OPEN

namespace parallel
{

#ifdef DEAL_II_WITH_MPI


  namespace shared
  {

    template <int dim, int spacedim = dim>
    class Triangulation : public ::parallel::Triangulation<dim,spacedim>
    {
    public:
      typedef typename ::Triangulation<dim,spacedim>::active_cell_iterator active_cell_iterator;
      typedef typename ::Triangulation<dim,spacedim>::cell_iterator        cell_iterator;

      enum Settings
      {
        partition_auto = 0x0,

        partition_metis = 0x1,

        partition_zorder = 0x2,

        partition_zoltan = 0x3,

        partition_custom_signal = 0x4,

        construct_multigrid_hierarchy = 0x8,
      };


      Triangulation (MPI_Comm mpi_communicator,
                     const typename ::Triangulation<dim,spacedim>::MeshSmoothing =
                       (::Triangulation<dim,spacedim>::none),
                     const bool allow_artificial_cells = false,
                     const Settings settings = partition_auto);

      virtual ~Triangulation () = default;

      virtual void execute_coarsening_and_refinement ();

      virtual void create_triangulation (const std::vector< Point< spacedim > > &vertices,
                                         const std::vector< CellData< dim > > &cells,
                                         const SubCellData &subcelldata);

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

      template <class Archive>
      void load (Archive &ar, const unsigned int version);

      const std::vector<types::subdomain_id> &get_true_subdomain_ids_of_cells() const;

      const std::vector<types::subdomain_id> &get_true_level_subdomain_ids_of_cells(const unsigned int level) const;

      bool with_artificial_cells() const;

    protected:

      virtual void update_number_cache ();

    private:

      const Settings settings;

      const bool allow_artificial_cells;

      void partition();

      std::vector<types::subdomain_id> true_subdomain_ids_of_cells;

      std::vector<std::vector<types::subdomain_id>> true_level_subdomain_ids_of_cells;
    };

    template <int dim, int spacedim>
    template <class Archive>
    void
    Triangulation<dim,spacedim>::load (Archive &ar,
                                       const unsigned int version)
    {
      ::Triangulation<dim,spacedim>::load (ar, version);
      partition();
      this->update_number_cache ();
    }
  }
#else

  namespace shared
  {

    template <int dim, int spacedim = dim>
    class Triangulation : public ::parallel::Triangulation<dim,spacedim>
    {
    public:
      Triangulation () = delete;

      const std::vector<types::subdomain_id> &get_true_subdomain_ids_of_cells() const;

      const std::vector<types::subdomain_id> &get_true_level_subdomain_ids_of_cells(const unsigned int level) const;

      bool with_artificial_cells() const;

    private:
      std::vector<types::subdomain_id> true_subdomain_ids_of_cells;

      std::vector<types::subdomain_id> true_level_subdomain_ids_of_cells;
    };
  }


#endif
}

DEAL_II_NAMESPACE_CLOSE

#endif