datacollectorinterface.hh

#pragma once
 
#include <cstdint>
#include <vector>
 
#include <dune/grid/common/partitionset.hh>
 
namespace Dune::Vtk
{
 
  template <class GV, class Derived, class Partition = Partitions::InteriorBorder>
  class DataCollectorInterface
  {
  public:
    using GridView = GV;
 
    static constexpr auto partition = Partition{};
 
    enum { dim = GridView::dimension };
 
    enum { dow = GridView::dimensionworld };
 
  public:
    DataCollectorInterface (GridView const& gridView)
      : gridView_(gridView)
    {}
 
    GridView const& gridView () const
    {
      return gridView_;
    }
 
    void update ()
    {
      asDerived().updateImpl();
    }
 
    int ghostLevel () const
    {
      return asDerived().ghostLevelImpl();
    }
 
    std::uint64_t numCells () const
    {
      return asDerived().numCellsImpl();
    }
 
    std::uint64_t numPoints () const
    {
      return asDerived().numPointsImpl();
    }
 
 
    template <class T>
    std::vector<T> points () const
    {
      return asDerived().template pointsImpl<T>();
    }
 
 
    template <class T, class VtkFunction>
    std::vector<T> pointData (VtkFunction const& fct) const
    {
      return asDerived().template pointDataImpl<T>(fct);
    }
 
 
    template <class T, class VtkFunction>
    std::vector<T> cellData (VtkFunction const& fct) const
    {
      return asDerived().template cellDataImpl<T>(fct);
    }
 
    // Return the collective communication object from the grid.
    auto comm () const
    {
      return gridView_.comm();
    }
 
  protected: // cast to derived type
 
    Derived& asDerived ()
    {
      return static_cast<Derived&>(*this);
    }
 
    const Derived& asDerived () const
    {
      return static_cast<const Derived&>(*this);
    }
 
  public: // default implementations
 
    void updateImpl ()
    {
      /* do nothing */
    }
 
    int ghostLevelImpl () const
    {
      return gridView_.overlapSize(0);
    }
 
    // Evaluate `fct` in center of cell.
    template <class T, class VtkFunction>
    std::vector<T> cellDataImpl (VtkFunction const& fct) const;
 
  private:
    GridView gridView_;
  };
 
} // end namespace Dune::Vtk
 
#include "datacollectorinterface.impl.hh"