multidomain/io/vtkoutputmodule.hh

Go to the documentation of this file.

// -*- mode: C++; tab-width: 4; indent-tabs-mode: nil; c-basic-offset: 4 -*-
// vi: set et ts=4 sw=4 sts=4:
//
// SPDX-FileCopyrightInfo: Copyright © DuMux Project contributors, see AUTHORS.md in root folder
// SPDX-License-Identifier: GPL-3.0-or-later
//
#ifndef DUMUX_MULTIDOMAIN_VTK_OUTPUT_MODULE_HH
#define DUMUX_MULTIDOMAIN_VTK_OUTPUT_MODULE_HH
 
#include <tuple>
#include <memory>
#include <utility>
 
#include <dune/common/hybridutilities.hh>
#include <dune/common/indices.hh>
 
#include <dumux/common/typetraits/utility.hh>
#include <dumux/io/vtkoutputmodule.hh>
 
namespace Dumux {
 
template<class MDTraits, template<class GV, class S> class Module = Dumux::VtkOutputModule>
class MultiDomainVtkOutputModule
{
    using MDSolutionVector = typename MDTraits::SolutionVector;
    static constexpr std::size_t numSubDomains = MDTraits::numSubDomains;
 
    template<std::size_t i>
    using GridVariables = typename MDTraits::template SubDomain<i>::GridVariables;
 
    using MDGridVars = typename MDTraits::template TupleOfSharedPtrConst<GridVariables>;
 
    template<std::size_t i>
    using SolutionVector = typename MDTraits::template SubDomain<i>::SolutionVector;
 
    template<std::size_t i>
    using VtkOutputModule = Module<GridVariables<i>, SolutionVector<i>>;
 
    using VtkOutputModuleTuple = typename MDTraits::template TupleOfSharedPtr<VtkOutputModule>;
 
public:
    template<std::size_t i>
    using Type = VtkOutputModule<i>;
 
    template<std::size_t i>
    using PtrType = std::shared_ptr<Type<i>>;
 
    MultiDomainVtkOutputModule() = default;
 
    MultiDomainVtkOutputModule(MDGridVars&& gridVars, const MDSolutionVector& sol,
                               const std::array<std::string, numSubDomains>& name)
    {
        using namespace Dune::Hybrid;
        forEach(std::make_index_sequence<numSubDomains>{}, [&](auto&& id)
        {
            constexpr auto i = std::decay_t<decltype(id)>::value;
            elementAt(vtkOutputModule_, id) = std::make_shared<Type<i>>(*std::get<i>(gridVars), sol[id], name[id]);
        });
    }
 
    void initDefaultOutputFields()
    {
        using namespace Dune::Hybrid;
        forEach(std::make_index_sequence<numSubDomains>{}, [&](auto&& id)
        {
            constexpr auto i = std::decay_t<decltype(id)>::value;
            MDTraits::template SubDomain<i>::IOFields::initOutputModule(*elementAt(vtkOutputModule_, id));
        });
    }
 
    void write(double t, Dune::VTK::OutputType type = Dune::VTK::ascii)
    {
        using namespace Dune::Hybrid;
        forEach(std::make_index_sequence<numSubDomains>{}, [&](auto&& id)
        {
            elementAt(vtkOutputModule_, id)->write(t, type);
        });
    }
 
    template<std::size_t i>
    const Type<i>& operator[] (Dune::index_constant<i> id) const
    { return *Dune::Hybrid::elementAt(vtkOutputModule_, id); }
 
    template<std::size_t i>
    Type<i>& operator[] (Dune::index_constant<i> id)
    { return *Dune::Hybrid::elementAt(vtkOutputModule_, id); }
 
    template<std::size_t i>
    PtrType<i> get(Dune::index_constant<i> id = Dune::index_constant<i>{})
    { return Dune::Hybrid::elementAt(vtkOutputModule_, id); }
 
    template<std::size_t i>
    void set(PtrType<i> p, Dune::index_constant<i> id = Dune::index_constant<i>{})
    { Dune::Hybrid::elementAt(vtkOutputModule_, id) = p; }
 
private:
 
    typename MDTraits::template Tuple<PtrType> vtkOutputModule_;
};
 
} // end namespace Dumux
 
#endif