controlvtkwriter.hh

Go to the documentation of this file.

// SPDX-FileCopyrightText: 2021-2025 The Ikarus Developers mueller@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/listener/listener.hh>
 
#pragma GCC diagnostic push
#pragma GCC diagnostic ignored "-Wswitch-enum"
 
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, const Eigen::VectorXd& sol, int refinementLevels = 0)
      : basis_{&basis},
        vtkWriter_(basis.gridView(), Dune::refinementLevels(refinementLevels)),
        solution_{&sol} {}
 
  template <typename BC>
  ControlSubsamplingVertexVTKWriter& subscribeTo(BC& bc) {
    this->subscribe(bc, [&](ControlMessages message) { this->updateImpl(message); });
    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 updateImpl(ControlMessages message) {
    assert(isFieldInfoSet_ && "You need to call setFieldInfo first!");
    switch (message) {
      case ControlMessages::SOLUTION_CHANGED: {
        auto disp = Dune::Functions::makeDiscreteGlobalBasisFunction<Dune::FieldVector<double, components>>(*basis_,
                                                                                                            *solution_);
        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_;
  const Eigen::VectorXd* solution_;
  int step_{0};
  Dune::VTK::FieldInfo fieldInfo_{"Default", Dune::VTK::FieldInfo::Type::scalar, 1};
  std::string prefixString_{};
  bool isFieldInfoSet_{false};
};
} // namespace Ikarus
#pragma GCC diagnostic pop