/export/development/ViennaFEM/viennafem/io/vtk_writer.hpp

Go to the documentation of this file.

#ifndef VIENNAFEM_IO_VTKWRITER_HPP
#define VIENNAFEM_IO_VTKWRITER_HPP

/* =========================================================================
   Copyright (c) 2012, Institute for Microelectronics,
                       Institute for Analysis and Scientific Computing,
                       TU Wien.
                             -----------------
               ViennaFEM - The Vienna Finite Element Method Library
                             -----------------

   Author:     Karl Rupp                          rupp@iue.tuwien.ac.at

   License:    MIT (X11), see file LICENSE in the ViennaFEM base directory
============================================================================ */


// include necessary system headers
#include <iostream>

#include "viennafem/forwards.h"

// ViennaGrid includes:
#include "viennagrid/domain.hpp"
#include "viennagrid/iterators.hpp"
#include "viennagrid/io/vtk_writer.hpp"

// ViennaData includes:
#include "viennadata/api.hpp"

namespace viennafem
{
  namespace io
  {

    template <typename VectorType,
              typename DomainType>
    void write_solution_to_VTK_file(VectorType const & result,
                                    std::string filename,
                                    DomainType const & domain,
                                    long id)
    {
      typedef typename DomainType::config_type                                              ConfigType;
      typedef typename viennagrid::result_of::ncell<ConfigType, 0>::type               VertexType;
      typedef typename viennagrid::result_of::const_ncell_range<DomainType, 0>::type    VertexContainer;
      typedef typename viennagrid::result_of::iterator<VertexContainer>::type               VertexIterator;
      
      typedef viennafem::mapping_key          MappingKeyType;
      typedef viennafem::boundary_key         BoundaryKeyType;
      
      MappingKeyType map_key(id);
      BoundaryKeyType bnd_key(id);
      
      
      std::cout << "* write_solution_to_VTK_file(): Writing result on mesh for later export" << std::endl;
      VertexContainer vertices = viennagrid::ncells<0>(domain);
      for (VertexIterator vit = vertices.begin();
          vit != vertices.end();
          ++vit)
      {
        long cur_index = viennadata::access<MappingKeyType, long>(map_key)(*vit);
        if (cur_index > -1)
          viennadata::access<std::string, double>("vtk_data")(*vit) = result[cur_index];
        else //use Dirichlet boundary data:
          viennadata::access<std::string, double>("vtk_data")(*vit) = 
            viennadata::access<BoundaryKeyType, double>(bnd_key)(*vit);
      }

      std::cout << "* write_solution_to_VTK_file(): Writing data to '"
                << filename
                << "' (can be viewed with e.g. Paraview)" << std::endl;

      viennagrid::io::vtk_writer<DomainType> my_vtk_writer;
      viennagrid::io::add_scalar_data_on_vertices<std::string, double>(my_vtk_writer, "vtk_data", "fem_result");
      my_vtk_writer(domain, filename);  
    }


  }
}
#endif