vtk.hh

Go to the documentation of this file.

/* * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * */
/*                                                                           */
/*  This file is part of the library KASKADE 7                               */
/*  https://www.zib.de/research/projects/kaskade7-finite-element-toolbox     */
/*                                                                           */
/*  Copyright (C) 2002-2022 Zuse Institute Berlin                            */
/*                                                                           */
/*  KASKADE 7 is distributed under the terms of the ZIB Academic License.    */
/*    see $KASKADE/academic.txt                                              */
/*                                                                           */
/* * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * */
 
#ifndef VTK_HH
#define VTK_HH
 
#include "io/check_endianness.hh"
#include "io/iobase.hh"
#include "io/vtkwriter.hh"
#include "fem/variables.hh"
#include "utilities/deprecation.hh"
 
//---------------------------------------------------------------------
 
namespace Kaskade
{
 
  template <class Function>
  void writeVTK(Function const& f, std::string const& filename, IoOptions options, std::string fname)
  {
    using namespace boost::fusion;
    
    if (fname.empty())
      fname = filename;
    
    // Construct a variable set (trivially consisting of only one function) to be handed to the
    // VTK writer.
    auto vsDescription = makeVariableSetDescription(
                                makeSpaceList(&f.space()),
                                make_vector(Variable<SpaceIndex<0>,
                                                     Components<Function::components>>(fname)));
    auto fs = vsDescription.variableSet();
    component<0>(fs) = f;
    
    // write the VTK file
    writeVTK(fs,filename,options);
  }
 
 
  //---------------------------------------------------------------------
 
  class TimeSeriesVTKWriter
  {
  public:
    TimeSeriesVTKWriter(std::string const& basename, IoOptions options_= ioOptions_default)
    : base(basename), pvd((basename+".pvd").c_str()), count(0), options(options_)
    {
      pvd << "<?xml version=\"1.0\"?>\n"
          << "<VTKFile type=\"Collection\" version=\"0.1\" byte_order=\"" <<  VTKWriterDetail::vtkEndianness() << "\">\n"
          << "  <Collection>\n";
    }
 
    template <class VariableSet>
    void add(double time, VariableSet const& vars)
    {
      auto fname = base+"-"+paddedString(count,4);
      writeVTK(vars,fname,options);
 
      pvd << "    <DataSet timestep=\"" << time << "\" file=\"" << fname << ".vtu\" />\n";
      ++count;
    }
 
    ~TimeSeriesVTKWriter()
    {
      pvd << "  </Collection>\n"
          << "</VTKFile>\n";
    }
 
  private:
    std::string base;
    std::ofstream pvd;
    int count;
    IoOptions options;
  };
  
  
  //---------------------------------------------------------------------
  //---------------------------------------------------------------------
  
 
 
  
} // end namespace Kaskade
//---------------------------------------------------------------------
 
#endif