VTKFile.h

// Copyright (C) 2020 Garth N. Wells
//
// This file is part of DOLFINx (https://www.fenicsproject.org)
//
// SPDX-License-Identifier:    LGPL-3.0-or-later
 
#pragma once
 
#include <dolfinx/common/MPI.h>
#include <dolfinx/fem/Function.h>
#include <functional>
#include <memory>
#include <string>
 
namespace pugi
{
class xml_document;
}
 
namespace dolfinx
{
namespace function
{
template <typename T>
class Function;
}
 
namespace mesh
{
class Mesh;
} // namespace mesh
 
namespace io
{
 
 
 
class VTKFile
{
public:
  VTKFile(MPI_Comm comm, const std::string filename,
          const std::string file_mode);
 
  ~VTKFile();
 
  void close();
 
  void flush();
 
  void write(const mesh::Mesh& mesh, double time = 0.0);
 
  void write(
      const std::vector<std::reference_wrapper<const fem::Function<double>>>& u,
      double t);
 
  void write(
      const std::vector<
          std::reference_wrapper<const fem::Function<std::complex<double>>>>& u,
      double t);
 
private:
  std::unique_ptr<pugi::xml_document> _pvd_xml;
 
  std::string _filename;
 
  // MPI communicator
  dolfinx::MPI::Comm _comm;
};
} // namespace io
} // namespace dolfinx