skeletonfunction.hh

Go to the documentation of this file.

// -*- tab-width: 4; indent-tabs-mode: nil; c-basic-offset: 2 -*-
// vi: set et ts=4 sw=2 sts=2:
 
#ifndef DUNE_GRID_IO_FILE_VTK_SKELETONFUNCTION_HH
#define DUNE_GRID_IO_FILE_VTK_SKELETONFUNCTION_HH
 
#include <string>
#include <vector>
 
#include <dune/common/fvector.hh>
#include <dune/common/shared_ptr.hh>
 
#include <dune/grid/io/file/vtk/functionwriter.hh>
#include <dune/grid/io/file/vtk/pvtuwriter.hh>
#include <dune/grid/io/file/vtk/vtuwriter.hh>
 
namespace Dune {
 
 
  namespace VTK {
 
    //
    //  Prototype for VTKFunktions on the skeleton
    //
 
    template<typename GV, typename RF>
    struct SkeletonFunctionTraits {
      typedef GV GridView;
      typedef typename GV::Intersection Cell;
 
      typedef typename GV::ctype DomainField;
      static const unsigned dimDomain = GV::dimension-1;
      typedef FieldVector<DomainField, dimDomain> Domain;
 
      typedef RF RangeField;
      typedef std::vector<RangeField> Range;
    };
 
    template <typename GV, typename RF>
    class SkeletonFunctionInterface {
    public:
      typedef SkeletonFunctionTraits<GV, RF> Traits;
 
      unsigned dimRange() const;
 
 
      void evaluate(const typename Traits::Cell& c,
                    const typename Traits::Domain& xl,
                    typename Traits::Range& result) const;
    };
 
    //
    //  Class for writing SkeletonFunctions
    //
 
 
    template<typename Func>
    class SkeletonFunctionWriter
      : public FunctionWriterBase<typename Func::Traits::Cell>
    {
      typedef typename Func::Traits::RangeField RF;
 
      shared_ptr<const Func> func;
      std::string name_;
      unsigned dimR;
      shared_ptr<DataArrayWriter<RF> > arraywriter;
 
    public:
      SkeletonFunctionWriter(const shared_ptr<const Func>& func_,
                             const std::string& name, unsigned dimR_)
        : func(func_), name_(name), dimR(dimR_)
      { }
 
      SkeletonFunctionWriter(const shared_ptr<const Func>& func_,
                             const std::string& name)
        : func(func_), name_(name), dimR(func->dimRange())
      { }
 
      virtual std::string name() const { return name_; }
 
      virtual unsigned ncomps() const { return dimR; }
 
      virtual void addArray(PVTUWriter& writer) {
        writer.addArray<RF>(name(), ncomps());
      }
 
      virtual bool beginWrite(VTUWriter& writer, std::size_t nitems) {
        arraywriter.reset(writer.makeArrayWriter<RF>(name(), ncomps(),
                                                     nitems));
        return !arraywriter->writeIsNoop();
      }
 
      virtual void write(const typename Func::Traits::Cell& cell,
                         const typename Func::Traits::Domain& xl) {
        typename Func::Traits::Range result;
        func->evaluate(cell, xl, result);
        for(unsigned d = 0; d < result.size() && d < dimR; ++d)
          arraywriter->write(result[d]);
        for(unsigned d = result.size(); d < dimR; ++d)
          arraywriter->write(0);
      }
 
      virtual void endWrite() {
        arraywriter.reset();
      }
    };
 
  } // namespace VTK
 
 
} // namespace Dune
 
#endif // DUNE_GRID_IO_FILE_VTK_SKELETONFUNCTION_HH