controlvtkwriter.hh

Go to the documentation of this file.

// SPDX-FileCopyrightText: 2021-2025 The Ikarus Developers ikarus@ibb.uni-stuttgart.de
// SPDX-License-Identifier: LGPL-3.0-or-later
 
#pragma once
#include <string>
 
#include <dune/functions/gridfunctions/discreteglobalbasisfunction.hh>
#include <dune/grid/io/file/vtk/subsamplingvtkwriter.hh>
 
#include <Eigen/Core>
 
#include <ikarus/utils/broadcaster/broadcastermessages.hh>
#include <ikarus/utils/concepts.hh>
#include <ikarus/utils/listener/listener.hh>
 
namespace Ikarus {
 
template <typename B>
class ControlSubsamplingVertexVTKWriter : public Listener
{
  using Basis                     = B;
  static constexpr int components = Basis::LocalView::Tree::degree() == 0 ? 1 : Basis::LocalView::Tree::degree();
 
public:
  ControlSubsamplingVertexVTKWriter(const Basis& basis, int refinementLevels = 0)
      : basis_{basis},
        vtkWriter_(basis.gridView(), Dune::refinementLevels(refinementLevels)) {}
 
  template <typename BC>
  ControlSubsamplingVertexVTKWriter& subscribeTo(BC& bc) {
    this->subscribe(bc, [&](ControlMessages message, const BC::State& state) { this->update(message, state); });
    return *this;
  }
 
  auto setFieldInfo(std::string&& name, Dune::VTK::FieldInfo::Type type, std::size_t size,
                    Dune::VTK::Precision prec = Dune::VTK::Precision::float32) {
    fieldInfo_      = Dune::VTK::FieldInfo(std::move(name), type, size, prec);
    isFieldInfoSet_ = true;
  }
 
  auto setFileNamePrefix(std::string&& name) { prefixString_ = std::move(name); }
 
  void update(ControlMessages message, const Concepts::ControlRoutineState auto& state) {
    assert(isFieldInfoSet_ && "You need to call setFieldInfo first!");
    switch (message) {
      case ControlMessages::SOLUTION_CHANGED: {
        auto disp = Dune::Functions::makeDiscreteGlobalBasisFunction<Dune::FieldVector<double, components>>(
            basis_, state.domain.globalSolution());
        vtkWriter_.addVertexData(disp, fieldInfo_);
        vtkWriter_.write(prefixString_ + std::to_string(step_++));
      } break;
      default:
        break; // default: do nothing when notified
    }
  }
 
private:
  const Basis& basis_;
  Dune::SubsamplingVTKWriter<typename Basis::GridView> vtkWriter_;
  int step_{0};
  Dune::VTK::FieldInfo fieldInfo_{"Default", Dune::VTK::FieldInfo::Type::scalar, 1};
  std::string prefixString_{};
  bool isFieldInfoSet_{false};
};
 
#ifndef DOXYGEN
template <typename Basis>
ControlSubsamplingVertexVTKWriter(const Basis&, int) -> ControlSubsamplingVertexVTKWriter<Basis>;
#endif
 
} // namespace Ikarus