functionwriter.hh

// -*- 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_FUNCTIONWRITER_HH
#define DUNE_GRID_IO_FILE_VTK_FUNCTIONWRITER_HH
 
#include <cstddef>
#include <memory>
#include <string>
#include <typeinfo>
#include <vector>
 
#include <dune/common/exceptions.hh>
#include <dune/common/fvector.hh>
 
#include <dune/geometry/referenceelements.hh>
 
#include <dune/grid/io/file/vtk/common.hh>
#include <dune/grid/io/file/vtk/dataarraywriter.hh>
#include <dune/grid/io/file/vtk/pvtuwriter.hh>
#include <dune/grid/io/file/vtk/vtuwriter.hh>
 
namespace Dune
{
 
  namespace VTK {
 
    template<typename Cell_>
    class FunctionWriterBase {
      typedef typename Cell_::Geometry::ctype DF;
      static const unsigned mydim = Cell_::mydimension;
      typedef ReferenceElements<DF, mydim> Refelems;
 
    public:
      typedef FieldVector<DF, mydim> Domain;
      typedef Cell_ Cell;
 
      virtual std::string name() const = 0;
 
      virtual unsigned ncomps() const = 0;
 
      virtual void addArray(PVTUWriter& writer) = 0;
      virtual bool beginWrite(VTUWriter& writer, std::size_t nitems) = 0;
 
      virtual void write(const Cell& cell, const Domain& xl) {
        DUNE_THROW(NotImplemented, "FunctionWriterBase::write(const Cell&, "
                   "const Domain&): Either the derived class " <<
                   typeid(*this).name() << " failed to implement this method "
                   "or this method is not meant to be called on the derived "
                   "class and was called in error.");
      }
 
      virtual void write(const Cell& cell, unsigned cornerIndex) {
        write(cell,
              Refelems::general(cell.type()).position(cornerIndex, mydim));
      }
      virtual void endWrite() = 0;
      virtual ~FunctionWriterBase() {}
    };
 
    //
    //  A Generic Function writer for VTKFunctions
    //
 
    template<typename Func>
    class VTKFunctionWriter
      : public FunctionWriterBase<typename Func::Entity>
    {
      typedef FunctionWriterBase<typename Func::Entity> Base;
      std::shared_ptr<const Func> func;
      VTK::Precision precision_;
      std::shared_ptr<DataArrayWriter> arraywriter;
 
    public:
        VTKFunctionWriter(const std::shared_ptr<const Func>& func_,
                          VTK::Precision prec = VTK::Precision::float32)
        : func(func_), precision_(prec)
      { }
 
      virtual std::string name() const { return func->name(); }
 
      virtual unsigned ncomps() const {
        if(func->ncomps() == 2) return 3;
        else return func->ncomps();
      }
 
      virtual void addArray(PVTUWriter& writer) {
        writer.addArray(name(), ncomps(), precision_);
      }
 
      virtual bool beginWrite(VTUWriter& writer, std::size_t nitems) {
        arraywriter.reset(writer.makeArrayWriter(name(), ncomps(),
                                                 nitems, precision_));
        return !arraywriter->writeIsNoop();
      }
 
      virtual void write(const typename Base::Cell& cell,
                         const typename Base::Domain& xl) {
        for(int d = 0; d < func->ncomps(); ++d)
          arraywriter->write(func->evaluate(d, cell, xl));
        for(unsigned d = func->ncomps(); d < ncomps(); ++d)
          arraywriter->write(0);
      }
 
      virtual void endWrite() {
        arraywriter.reset();
      }
    };
 
    //
    //  Writers for the grid information
    //
 
    template<typename Cell>
    class CoordinatesWriter
      : public FunctionWriterBase<Cell>
    {
      typedef FunctionWriterBase<Cell> Base;
 
      VTK::Precision precision_;
      std::shared_ptr<DataArrayWriter> arraywriter;
 
    public:
      explicit CoordinatesWriter(VTK::Precision prec = VTK::Precision::float32)
      : precision_(prec)
      {}
 
      virtual std::string name() const { return "Coordinates"; }
 
      virtual unsigned ncomps() const { return 3; }
 
      virtual void addArray(PVTUWriter& writer) {
        writer.addArray(name(), ncomps(), precision_);
      }
 
      virtual bool beginWrite(VTUWriter& writer, std::size_t nitems) {
        arraywriter.reset(writer.makeArrayWriter(name(), ncomps(),
                                                 nitems, precision_));
        return !arraywriter->writeIsNoop();
      }
      virtual void write(const typename Base::Cell& cell,
                         const typename Base::Domain& xl) {
        FieldVector<typename Base::Cell::Geometry::ctype, Base::Cell::Geometry::coorddimension> xg
          = cell.geometry().global(xl);
        for(unsigned d = 0; d < 3 && d < Base::Cell::Geometry::coorddimension; ++d)
          arraywriter->write(xg[d]);
        for(unsigned d = Base::Cell::Geometry::coorddimension; d < 3; ++d)
          arraywriter->write(0);
      }
      virtual void endWrite() {
        arraywriter.reset();
      }
    };
 
    template<typename IteratorFactory>
    class ConformingConnectivityWriter
      : public FunctionWriterBase<typename IteratorFactory::Cell>
    {
      typedef FunctionWriterBase<typename IteratorFactory::Cell> Base;
      static const unsigned mydim = Base::Cell::mydimension;
 
      const IteratorFactory& factory;
      std::shared_ptr<DataArrayWriter> arraywriter;
      std::vector<unsigned> pointIndices;
 
    public:
      ConformingConnectivityWriter(const IteratorFactory& factory_)
        : factory(factory_)
      { }
 
      virtual std::string name() const { return "connectivity"; }
 
      virtual unsigned ncomps() const { return 1; }
 
      virtual void addArray(PVTUWriter& writer) {
        writer.addArray(name(), ncomps(), Precision::int32);
      }
 
      virtual bool beginWrite(VTUWriter& writer, std::size_t nitems) {
        arraywriter.reset(writer.makeArrayWriter(name(), ncomps(),
                                                 nitems, Precision::int32));
        if(arraywriter->writeIsNoop())
          return false;
 
        pointIndices.resize(factory.indexSet().size(mydim));
        const typename IteratorFactory::PointIterator& pend =
          factory.endPoints();
        typename IteratorFactory::PointIterator pit = factory.beginPoints();
        unsigned counter = 0;
        while(pit != pend) {
          pointIndices[factory.indexSet().subIndex
                         (pit->cell(), pit->duneIndex(), mydim)] = counter;
          ++counter;
          ++pit;
        }
        return true;
      }
      virtual void write(const typename Base::Cell& cell, unsigned cornerIndex)
      {
        // if pointIndices is empty, we're in writeIsNoop mode
        if(pointIndices.size() == 0)
          return;
        arraywriter->write(pointIndices[factory.indexSet().subIndex
                                          (cell, cornerIndex, mydim)]);
      }
      virtual void endWrite() {
        arraywriter.reset();
        pointIndices.clear();
      }
    };
 
    template<typename Cell>
    class NonConformingConnectivityWriter
      : public FunctionWriterBase<Cell>
    {
      std::shared_ptr<DataArrayWriter> arraywriter;
      unsigned counter;
 
    public:
      virtual std::string name() const { return "connectivity"; }
 
      virtual unsigned ncomps() const { return 1; }
 
      virtual void addArray(PVTUWriter& writer) {
        writer.addArray(name(), ncomps(), Precision::int32);
      }
 
      virtual bool beginWrite(VTUWriter& writer, std::size_t nitems) {
        arraywriter.reset(writer.makeArrayWriter(name(), ncomps(),
                                                 nitems, Precision::int32));
        counter = 0;
        return !arraywriter->writeIsNoop();
      }
      virtual void write(const Cell& cell, unsigned cornerIndex)
      {
        arraywriter->write(counter);
        ++counter;
      }
      virtual void endWrite() {
        arraywriter.reset();
      }
    };
 
    template<typename Cell>
    class OffsetsWriter
      : public FunctionWriterBase<Cell>
    {
      typedef FunctionWriterBase<Cell> Base;
 
      std::shared_ptr<DataArrayWriter> arraywriter;
      unsigned offset;
 
    public:
      virtual std::string name() const { return "offsets"; }
 
      virtual unsigned ncomps() const { return 1; }
 
      virtual void addArray(PVTUWriter& writer) {
        writer.addArray(name(), ncomps(), Precision::int32);
      }
 
      virtual bool beginWrite(VTUWriter& writer, std::size_t nitems) {
        arraywriter.reset(writer.makeArrayWriter(name(), ncomps(),
                                                 nitems, Precision::int32));
        offset = 0;
        return !arraywriter->writeIsNoop();
      }
      virtual void write(const Cell& cell, const typename Base::Domain&) {
        offset += cell.geometry().corners();
        arraywriter->write(offset);
      }
      virtual void endWrite() {
        arraywriter.reset();
      }
    };
 
    template<typename Cell>
    class TypesWriter
      : public FunctionWriterBase<Cell>
    {
      typedef FunctionWriterBase<Cell> Base;
 
      std::shared_ptr<DataArrayWriter> arraywriter;
 
    public:
      virtual std::string name() const { return "types"; }
 
      virtual unsigned ncomps() const { return 1; }
 
      virtual void addArray(PVTUWriter& writer) {
        writer.addArray(name(), ncomps(), Precision::uint8);
      }
 
      virtual bool beginWrite(VTUWriter& writer, std::size_t nitems) {
        arraywriter.reset(writer.makeArrayWriter
                            ( name(), ncomps(), nitems, Precision::uint8));
        return !arraywriter->writeIsNoop();
      }
      virtual void write(const Cell& cell, const typename Base::Domain&) {
        arraywriter->write(geometryType(cell.type()));
      }
      virtual void endWrite() {
        arraywriter.reset();
      }
    };
 
  } // namespace VTK
 
 
} // namespace Dune
 
#endif // DUNE_GRID_IO_FILE_VTK_FUNCTIONWRITER_HH