augment_sparsity_on_interface.h

Go to the documentation of this file.

#ifndef AUGMENT_SPARSITY_ON_INTERFACE_H
#define AUGMENT_SPARSITY_ON_INTERFACE_H

#include "libmesh/ghosting_functor.h"
#include "libmesh/mesh_base.h"

using libMesh::Elem;
using libMesh::GhostingFunctor;
using libMesh::MeshBase;
using libMesh::boundary_id_type;
using libMesh::processor_id_type;

// Convenient typedef for a map for (element, side id) --> element neighbor
typedef std::map<std::pair<const Elem *, unsigned char>, const Elem *> ElementSideMap;

// And the inverse map, but without sides in the key
typedef std::map<const Elem *, const Elem *> ElementMap;

class AugmentSparsityOnInterface : public GhostingFunctor
{
private:

  MeshBase & _mesh;

  ElementSideMap _lower_to_upper;

  ElementMap _upper_to_lower;

  boundary_id_type _crack_boundary_lower, _crack_boundary_upper;

  bool _initialized;

public:

  AugmentSparsityOnInterface(MeshBase & mesh,
                             boundary_id_type crack_boundary_lower,
                             boundary_id_type crack_boundary_upper);

  const ElementSideMap & get_lower_to_upper() const;

  virtual void operator() (const MeshBase::const_element_iterator & range_begin,
                           const MeshBase::const_element_iterator & range_end,
                           processor_id_type p,
                           map_type & coupled_elements) override;

  virtual void mesh_reinit () override;

  virtual void redistribute () override
  { this->mesh_reinit(); }


};

#endif