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 <filesystem>
#include <functional>
#include <memory>
#include <string>
 
namespace pugi
{
class xml_document;
}
 
namespace dolfinx::fem
{
template <typename T>
class Function;
}
 
namespace dolfinx::mesh
{
class Mesh;
}
 
namespace dolfinx::io
{
 
class VTKFile
{
public:
  VTKFile(MPI_Comm comm, const std::filesystem::path& filename,
          const std::string& file_mode);
 
  ~VTKFile();
 
  void close();
 
  void flush();
 
  void write(const mesh::Mesh& mesh, double time = 0.0);
 
  template <typename T>
  void
  write(const std::vector<std::reference_wrapper<const fem::Function<T>>>& u,
        double t)
  {
    write_functions(u, t);
  }
 
private:
  void write_functions(
      const std::vector<std::reference_wrapper<const fem::Function<double>>>& u,
      double t);
  void write_functions(
      const std::vector<
          std::reference_wrapper<const fem::Function<std::complex<double>>>>& u,
      double t);
 
  std::unique_ptr<pugi::xml_document> _pvd_xml;
 
  std::filesystem::path _filename;
 
  // MPI communicator
  dolfinx::MPI::Comm _comm;
};
} // namespace dolfinx::io