gnuplot.hh

Go to the documentation of this file.

// SPDX-FileCopyrightText: Copyright (C) DUNE Project contributors, see file LICENSE.md in module root
// SPDX-License-Identifier: LicenseRef-GPL-2.0-only-with-DUNE-exception
// -*- tab-width: 4; indent-tabs-mode: nil; c-basic-offset: 2 -*-
// vi: set et ts=4 sw=2 sts=2:
#ifndef DUNE_IO_GNUPLOT_HH
#define DUNE_IO_GNUPLOT_HH
 
#include <vector>
#include <string>
#include <iostream>
#include <fstream>
 
#include <dune/common/fvector.hh>
 
#include <dune/grid/common/grid.hh>
 
namespace Dune {
 
  template<class GridView>
  class GnuplotWriter {
 
    typedef typename GridView::Grid::ctype ctype;
 
    constexpr static int dimworld = GridView::dimensionworld;
 
  public:
    GnuplotWriter (const GridView & gv) : _is(gv.indexSet()), _gv(gv)
    {
      static_assert(dimworld==1 || dimworld==2, "GnuPlot export only works for worlddim==1 and worlddim==2");
      // allocate _data buffer
      _data.resize(_is.size(0)*2);
    }
 
    template <class DataContainer>
    void addCellData(const DataContainer& data, const std::string & name)
    {
      if (dimworld!=1)
        DUNE_THROW(IOError, "Gnuplot cell data writing is only supported for grids in a 1d world!");
      addData(cellData, data, name);
    }
 
    template <class DataContainer>
    void addVertexData(const DataContainer& data, const std::string & name)
    {
      addData(vertexData, data, name);
    }
 
    void write(const std::string& filename) const;
 
  private:
    enum DataType { vertexData, cellData };
    const typename GridView::IndexSet & _is;
    const GridView _gv;
    std::vector< std::vector< float > > _data;
    std::vector< std::string > _names;
 
    template <class DataContainer>
    void addData(DataType t, const DataContainer& data, const std::string & name);
 
    void writeRow(std::ostream & file,
                  const FieldVector<ctype, dimworld>& position,
                  const std::vector<float> & data) const;
  };
 
  template<class G>
  class LeafGnuplotWriter : public GnuplotWriter<typename G::LeafGridView>
  {
  public:
    LeafGnuplotWriter (const G& grid)
      : GnuplotWriter<typename G::LeafGridView>(grid.leafGridView())
    {}
  };
 
  template<class G>
  class LevelGnuplotWriter : public GnuplotWriter<typename G::LevelGridView>
  {
  public:
    LevelGnuplotWriter (const G& grid, int level)
      : GnuplotWriter<typename G::LevelGridView>(grid.levelGridView(level))
    {}
  };
 
}
 
#include "gnuplot/gnuplot.cc"
 
#endif // DUNE_IO_GNUPLOT_HH