dof_print_solver_step.h

Go to the documentation of this file.

// -----------------------------------------------------------------------------
//
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception OR LGPL-2.1-or-later
// Copyright (C) 2000 - 2023 by the deal.II authors
//
// This file is part of the deal.II library.
//
// Detailed license information governing the source code and contributions
// can be found in LICENSE.md and CONTRIBUTING.md at the top level directory.
//
// -----------------------------------------------------------------------------
 
#ifndef dealii_dof_print_solver_step_h
#define dealii_dof_print_solver_step_h
 
#include <deal.II/base/config.h>
 
#include <deal.II/base/logstream.h>
 
#include <deal.II/lac/solver_control.h>
#include <deal.II/lac/vector_memory.h>
 
#include <deal.II/numerics/data_out.h>
 
#include <fstream>
#include <iomanip>
#include <sstream>
 
DEAL_II_NAMESPACE_OPEN
 
 
template <int dim, typename SolverType, typename VectorType = Vector<double>>
class DoFPrintSolverStep : public SolverType
{
public:
  DoFPrintSolverStep(SolverControl            &control,
                     VectorMemory<VectorType> &mem,
                     DataOut<dim>             &data_out,
                     const std::string        &basename);
 
  virtual void
  print_vectors(const unsigned int step,
                const VectorType  &x,
                const VectorType  &r,
                const VectorType  &d) const;
 
private:
  DataOut<dim> &out;
 
  const std::string basename;
};
 
 
/* ----------------------- template functions --------------- */
 
template <int dim, typename SolverType, typename VectorType>
DoFPrintSolverStep<dim, SolverType, VectorType>::DoFPrintSolverStep(
  SolverControl            &control,
  VectorMemory<VectorType> &mem,
  DataOut<dim>             &data_out,
  const std::string        &basename)
  : SolverType(control, mem)
  , out(data_out)
  , basename(basename)
{}
 
 
template <int dim, typename SolverType, typename VectorType>
void
DoFPrintSolverStep<dim, SolverType, VectorType>::print_vectors(
  const unsigned int step,
  const VectorType  &x,
  const VectorType  &r,
  const VectorType  &d) const
{
  out.clear_data_vectors();
  out.add_data_vector(x, "solution");
  out.add_data_vector(r, "residual");
  out.add_data_vector(d, "update");
 
  std::ostringstream filename;
  filename << basename << std::setw(3) << std::setfill('0') << step
           << out.default_suffix();
 
  const std::string fname = filename.str();
 
  deallog << "Writing file:" << fname << std::endl;
 
  out.build_patches();
  std::ofstream of(fname);
  out.write(of);
}
 
DEAL_II_NAMESPACE_CLOSE
 
#endif