vtkfunctionwrapper.hh

Go to the documentation of this file.

#pragma once
 
#include <memory>
 
#include <dune/grid/io/file/vtk/function.hh>
 
namespace Dune::Vtk
{
  template <class GridView>
  class VTKFunctionWrapper
  {
    using Entity = typename GridView::template Codim<0>::Entity;
    using Domain = typename Entity::Geometry::LocalCoordinate;
 
  public:
    explicit VTKFunctionWrapper (std::shared_ptr<VTKFunction<GridView> const> fct)
      : fct_(std::move(fct))
    {}
 
    void bind (Entity const& entity)
    {
      entity_ = &entity;
    }
 
    void unbind ()
    {
      entity_ = nullptr;
    }
 
    double evaluate (int comp, Domain const& x) const
    {
      return fct_->evaluate(comp, *entity_, x);
    }
 
    friend VTKFunctionWrapper localFunction (const VTKFunctionWrapper& self)
    {
      return self;
    }
 
  private:
    std::shared_ptr<VTKFunction<GridView> const> fct_;
    Entity const* entity_ = nullptr;
  };
 
} // end namespace Dune::Vtk