interface.h

Go to the documentation of this file.

/*
  Copyright (C) 2011 - 2024 by the authors of the ASPECT code.
 
  This file is part of ASPECT.
 
  ASPECT is free software; you can redistribute it and/or modify
  it under the terms of the GNU General Public License as published by
  the Free Software Foundation; either version 2, or (at your option)
  any later version.
 
  ASPECT is distributed in the hope that it will be useful,
  but WITHOUT ANY WARRANTY; without even the implied warranty of
  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  GNU General Public License for more details.
 
  You should have received a copy of the GNU General Public License
  along with ASPECT; see the file LICENSE.  If not see
  <http://www.gnu.org/licenses/>.
*/
 
 
#ifndef _aspect_mesh_deformation_interface_h
#define _aspect_mesh_deformation_interface_h
 
#include <aspect/plugins.h>
#include <aspect/simulator_access.h>
#include <aspect/global.h>
 
#include <deal.II/fe/fe_system.h>
#include <deal.II/dofs/dof_handler.h>
#include <deal.II/lac/affine_constraints.h>
#include <deal.II/base/index_set.h>
#include <deal.II/base/mg_level_object.h>
#include <deal.II/lac/la_parallel_vector.h>
#include <deal.II/multigrid/mg_constrained_dofs.h>
#include <deal.II/multigrid/mg_transfer_matrix_free.h>
#include <deal.II/multigrid/mg_transfer_global_coarsening.templates.h>
#include <aspect/simulator/assemblers/interface.h>
 
namespace aspect
{
  namespace Assemblers
  {
    template <int dim>
    class ApplyStabilization: public Assemblers::Interface<dim>,
      public SimulatorAccess<dim>
    {
      public:
        ApplyStabilization(const double stabilization_theta);
 
        void
        execute (internal::Assembly::Scratch::ScratchBase<dim>   &scratch,
                 internal::Assembly::CopyData::CopyDataBase<dim> &data) const override;
 
      private:
        const double free_surface_theta;
    };
  }
 
  template <int dim> class Simulator;
 
  namespace MeshDeformation
  {
    template <int dim>
    class Interface : public Plugins::InterfaceBase
    {
      public:
        virtual bool needs_surface_stabilization() const;
 
 
        virtual
        Tensor<1,dim>
        compute_initial_deformation_on_boundary(const types::boundary_id boundary_indicator,
                                                const Point<dim> &position) const;
 
 
        virtual
        void
        compute_initial_deformation_as_constraints(const Mapping<dim> &mapping,
                                                   const DoFHandler<dim> &mesh_deformation_dof_handler,
                                                   const types::boundary_id boundary_indicator,
                                                   AffineConstraints<double> &constraints) const;
 
        virtual
        void
        compute_velocity_constraints_on_boundary(const DoFHandler<dim> &mesh_deformation_dof_handler,
                                                 AffineConstraints<double> &mesh_velocity_constraints,
                                                 const std::set<types::boundary_id> &boundary_ids) const;
    };
 
 
 
    template <int dim>
    class MeshDeformationHandler: public SimulatorAccess<dim>
    {
      public:
        MeshDeformationHandler(Simulator<dim> &simulator);
 
        ~MeshDeformationHandler() override;
 
        void initialize();
 
        void set_assemblers(const SimulatorAccess<dim> &simulator_access,
                            aspect::Assemblers::Manager<dim> &assemblers) const;
 
        void update();
 
        void execute();
 
        void setup_dofs();
 
        static
        void declare_parameters (ParameterHandler &prm);
 
        void parse_parameters (ParameterHandler &prm);
 
        template <class Archive>
        void save (Archive &ar,
                   const unsigned int version) const;
 
        template <class Archive>
        void load (Archive &ar,
                   const unsigned int version);
 
        BOOST_SERIALIZATION_SPLIT_MEMBER()
 
        
        static
        void
        register_mesh_deformation
        (const std::string &name,
         const std::string &description,
         void (*declare_parameters_function) (ParameterHandler &),
         std::unique_ptr<Interface<dim>> (*factory_function) ());
 
        const std::map<types::boundary_id, std::vector<std::string>> &
        get_active_mesh_deformation_names () const;
 
        const std::map<types::boundary_id,std::vector<std::unique_ptr<Interface<dim>>>> &
        get_active_mesh_deformation_models () const;
 
        const std::set<types::boundary_id> &
        get_active_mesh_deformation_boundary_indicators () const;
 
        const std::set<types::boundary_id> &
        get_tangential_velocity_with_active_mesh_deformation_boundary_indicators () const;
 
        const std::set<types::boundary_id> &
        get_tangential_velocity_without_active_mesh_deformation_boundary_indicators () const;
 
        const std::set<types::boundary_id> &
        get_boundary_indicators_requiring_stabilization () const;
 
        const std::set<types::boundary_id> &
        get_free_surface_boundary_indicators () const;
 
        double get_free_surface_theta () const;
 
        const LinearAlgebra::Vector &
        get_initial_topography () const;
 
        const LinearAlgebra::Vector &
        get_mesh_displacements () const;
 
        const DoFHandler<dim> &
        get_mesh_deformation_dof_handler () const;
 
        template <typename MeshDeformationType,
                  typename = typename std::enable_if_t<std::is_base_of<Interface<dim>,MeshDeformationType>::value>>
        bool
        has_matching_mesh_deformation_object () const;
 
        template <typename MeshDeformationType,
                  typename = typename std::enable_if_t<std::is_base_of<Interface<dim>,MeshDeformationType>::value>>
        const MeshDeformationType &
        get_matching_mesh_deformation_object () const;
 
 
        const Mapping<dim> &
        get_level_mapping(const unsigned int level) const;
 
        static
        void
        write_plugin_graph (std::ostream &output_stream);
 
        DeclException1 (ExcMeshDeformationNameNotFound,
                        std::string,
                        << "Could not find entry <"
                        << arg1
                        << "> among the names of registered mesh deformation objects.");
 
      private:
        void make_initial_constraints ();
 
        void make_constraints ();
 
        void compute_mesh_displacements ();
 
        void compute_mesh_displacements_gmg ();
 
        unsigned int get_mapping_degree () const;
 
        template <unsigned int mesh_deformation_fe_degree>
        void compute_mesh_displacements_gmg_for_degree();
 
        void set_initial_topography ();
 
        void interpolate_mesh_velocity ();
 
        void update_multilevel_deformation ();
 
        Simulator<dim> &sim;
 
        const FESystem<dim> mesh_deformation_fe;
 
        DoFHandler<dim> mesh_deformation_dof_handler;
 
        LinearAlgebra::BlockVector mesh_velocity;
 
        LinearAlgebra::Vector mesh_displacements;
 
        LinearAlgebra::Vector old_mesh_displacements;
 
        LinearAlgebra::Vector initial_topography;
 
        LinearAlgebra::Vector fs_mesh_velocity;
 
        IndexSet mesh_locally_owned;
 
        IndexSet mesh_locally_relevant;
 
        AffineConstraints<double> mesh_velocity_constraints;
 
        AffineConstraints<double> mesh_vertex_constraints;
 
        std::map<types::boundary_id,std::vector<std::unique_ptr<Interface<dim>>>> mesh_deformation_objects;
 
        std::map<types::boundary_id, std::vector<std::string>> mesh_deformation_object_names;
 
        std::set<types::boundary_id> prescribed_mesh_deformation_boundary_indicators;
 
        std::set<types::boundary_id> tangential_velocity_with_prescribed_mesh_deformation_boundary_indicators;
 
        std::set<types::boundary_id> tangential_velocity_without_prescribed_mesh_deformation_boundary_indicators;
 
        std::set<types::boundary_id> tangential_mesh_deformation_boundary_indicators;
 
        std::set<types::boundary_id> zero_mesh_deformation_boundary_indicators;
 
        std::set<types::boundary_id> free_surface_boundary_indicators;
 
        std::set<types::boundary_id> boundary_indicators_requiring_stabilization;
 
        bool include_initial_topography;
 
        double surface_theta;
 
        bool use_automatic_mapping_order;
 
        unsigned int explicit_mapping_order;
 
        MGLevelObject<std::unique_ptr<Mapping<dim>>> level_mappings;
 
        MGLevelObject<::LinearAlgebra::distributed::Vector<double>> level_displacements;
 
        MGTransferType<dim, double> mg_transfer;
 
        MGConstrainedDoFs mg_constrained_dofs;
 
        friend class Simulator<dim>;
        friend class SimulatorAccess<dim>;
    };
 
 
    template <int dim>
    template <class Archive>
    void MeshDeformationHandler<dim>::save (Archive &ar,
                                            const unsigned int) const
    {
      // let all the mesh deformation plugins save their data in a map and then
      // serialize that
      //TODO: for now we assume the same plugins are active before and after restart.
      std::map<std::string,std::string> saved_text;
      for (const auto &boundary_id_and_mesh_deformation_objects : mesh_deformation_objects)
        for (const auto &p : boundary_id_and_mesh_deformation_objects.second)
          p->save (saved_text);
 
      ar &saved_text;
    }
 
 
    template <int dim>
    template <class Archive>
    void MeshDeformationHandler<dim>::load (Archive &ar,
                                            const unsigned int)
    {
      // get the map back out of the stream; then let the mesh deformation plugins
      // that we currently have get their data from there. note that this
      // may not be the same set ofmesh deformation plugins we had when we saved
      // their data
      std::map<std::string,std::string> saved_text;
      ar &saved_text;
 
      for (const auto &boundary_id_and_mesh_deformation_objects : mesh_deformation_objects)
        for (const auto &p : boundary_id_and_mesh_deformation_objects.second)
          p->load (saved_text);
    }
 
 
 
 
    template <int dim>
    template <typename MeshDeformationType, typename>
    inline
    bool
    MeshDeformationHandler<dim>::has_matching_mesh_deformation_object () const
    {
      for (const auto &object_iterator : mesh_deformation_objects)
        for (const auto &p : object_iterator.second)
          if (Plugins::plugin_type_matches<MeshDeformationType>(*p))
            return true;
 
      return false;
    }
 
 
 
    template <int dim>
    template <typename MeshDeformationType, typename>
    inline
    const MeshDeformationType &
    MeshDeformationHandler<dim>::get_matching_mesh_deformation_object () const
    {
      AssertThrow(has_matching_mesh_deformation_object<MeshDeformationType> (),
                  ExcMessage("You asked MeshDeformation::MeshDeformationHandler::get_matching_mesh_deformation_object() for a "
                             "mesh deformation object of type <" + boost::core::demangle(typeid(MeshDeformationType).name()) + "> "
                             "that could not be found in the current model. Activate this "
                             "mesh deformation in the input file."));
 
      for (const auto &object_iterator : mesh_deformation_objects)
        for (const auto &p : object_iterator.second)
          if (Plugins::plugin_type_matches<MeshDeformationType>(*p))
            return Plugins::get_plugin_as_type<MeshDeformationType>(*p);
 
      // We will never get here, because we had the Assert above. Just to avoid warnings.
      typename std::vector<std::unique_ptr<Interface<dim>>>::const_iterator mesh_def;
      return Plugins::get_plugin_as_type<MeshDeformationType>(*(*mesh_def));
 
    }
 
 
    template <int dim>
    std::string
    get_valid_model_names_pattern ();
 
 
 
#define ASPECT_REGISTER_MESH_DEFORMATION_MODEL(classname,name,description) \
  template class classname<2>; \
  template class classname<3>; \
  namespace ASPECT_REGISTER_MESH_DEFORMATION_MODEL_ ## classname \
  { \
    aspect::internal::Plugins::RegisterHelper<aspect::MeshDeformation::Interface<2>,classname<2>> \
    dummy_ ## classname ## _2d (&aspect::MeshDeformation::MeshDeformationHandler<2>::register_mesh_deformation, \
                                name, description); \
    aspect::internal::Plugins::RegisterHelper<aspect::MeshDeformation::Interface<3>,classname<3>> \
    dummy_ ## classname ## _3d (&aspect::MeshDeformation::MeshDeformationHandler<3>::register_mesh_deformation, \
                                name, description); \
  }
  }
}
 
#endif