InterpolationDataTransfer.hpp

#pragma once

#include <map>
#include <memory>

#include <amdis/DataTransfer.hpp>

#include <dune/typetree/treecontainer.hh>

namespace AMDiS {

namespace tag {

  struct interpolation_datatransfer
      : datatransfer
  {};

} // end namespace tag


// forward declaration
template <class Node, class Container, class Basis>
class NodeDataTransfer;


template <class Basis, class Container>
class InterpolationDataTransfer
{
  using LocalView = typename Basis::LocalView;
  using Tree = typename LocalView::Tree;
  using GridView = typename Basis::GridView;
  using Grid = typename GridView::Grid;

  using Element = typename GridView::template Codim<0>::Entity;
  using Geometry = typename Element::Geometry;
  using LocalCoordinate = typename Geometry::LocalCoordinate;
  using IdType = typename Grid::LocalIdSet::IdType;

  template <class Node>
  using NodeElementData = typename NodeDataTransfer<Node, Container, Basis>::NodeElementData;
  using ElementData = Dune::TypeTree::TreeContainer<NodeElementData,Tree>;

public:
  InterpolationDataTransfer() = default;

  void preAdapt(Basis const& basis, Container const& coeff, bool mightCoarsen);

  // [[expects: basis is updated in gridView]]
  // [[expects: comm is updated in basis]]
  void adapt(Basis const& basis, Container& coeff);

  void postAdapt(Container& coeff);

private:
  using PersistentContainer = std::map<IdType, ElementData>;
  PersistentContainer persistentContainer_{};

  template <class Node>
  using NDT = NodeDataTransfer<Node, Container, Basis>;
  using NodeDataTransferContainer = Dune::TypeTree::TreeContainer<NDT,Tree>;
  NodeDataTransferContainer nodeDataTransfer_{};
};

template <>
struct DataTransferFactory<tag::interpolation_datatransfer>
{
  template <class Basis, class T, template <class> class Impl,
    REQUIRES(Concepts::GlobalBasis<Basis>)>
  static DataTransfer<Basis,VectorFacade<T,Impl>> create(Basis const&, VectorFacade<T,Impl> const&)
  {
    return InterpolationDataTransfer<Basis, VectorFacade<T,Impl>>{};
  }
};


} // end namespace AMDiS

#include "InterpolationDataTransfer.inc.hpp"