fem/gridvariablescache.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-FileCopyrightText: Copyright © DuMux Project contributors, see AUTHORS.md in root folder
// SPDX-License-Identifier: GPL-3.0-or-later
//
#ifndef DUMUX_DISCRETIZATION_FE_GRID_LOCAL_VARIABLES_HH
#define DUMUX_DISCRETIZATION_FE_GRID_LOCAL_VARIABLES_HH
 
#include <type_traits>
#include <unordered_map>
#include <utility>
#include <vector>
#include <ranges>
#include <memory>
 
#include <dumux/parallel/parallel_for.hh>
 
#include <dumux/common/concepts/localdofs_.hh>
 
// make the local view function available whenever we use this class
#include <dumux/discretization/localview.hh>
#include <dumux/discretization/cvfe/elementsolution.hh>
#include <dumux/discretization/cvfe/quadraturerules.hh>
#include "elementvariables.hh"
 
namespace Dumux::Experimental {
 
template<class P, class V, class IPD>
struct FEDefaultGridVariablesCacheTraits
{
    using Problem = P;
    using Variables = V;
    using InterpolationPointData = IPD;
 
    template<class GridVariablesCache, bool cachingEnabled>
    using LocalView = FEElementVariables<GridVariablesCache, cachingEnabled>;
};

template<class Traits, bool enableCaching>
class FEGridVariablesCache;
 
// specialization in case of storing the local variables
template<class Traits>
class FEGridVariablesCache<Traits, /*cachingEnabled*/true>
{
    using ThisType = FEGridVariablesCache<Traits, true>;
public:
    using Problem = typename Traits::Problem;

    using Variables = typename Traits::Variables;

    using InterpolationPointData = typename Traits::InterpolationPointData;

    static constexpr bool cachingEnabled = true;

    using LocalView = typename Traits::template LocalView<ThisType, cachingEnabled>;

    using MutableLocalView = typename LocalView::MutableView;
 
    FEGridVariablesCache(const Problem& problem) : problemPtr_(&problem) {}
 
    template<class GridDiscretization, class SolutionVector>
    void init(const GridDiscretization& gridDiscretization, const SolutionVector& sol)
    {
        variables_.resize(gridDiscretization.gridView().size(0));
        ipDataCache_ = std::make_shared<InterpolationPointDataCache>();
        ipDataCache_->resize(gridDiscretization.gridView().size(0));
 
        Dumux::parallelFor(gridDiscretization.gridView().size(0), [&, &problem = problem()](const std::size_t eIdx)
        {
            const auto element = gridDiscretization.element(eIdx);
            const auto elemDisc = localView(gridDiscretization).bindElement(element);
 
            // get the element solution
            auto elemSol = elementSolution(element, sol, gridDiscretization);
 
            variables_[eIdx].resize(Dumux::Detail::LocalDofs::numLocalDofs(elemDisc));
            for (const auto& localDof : localDofs(elemDisc))
                variables_[eIdx][localDof.index()].update(elemSol, problem, elemDisc, ipData(elemDisc, localDof));
 
            ipDataCache_->update(problem, element, elemDisc, variables_[eIdx]);
        });
    }
 
    template<class GridDiscretization, class SolutionVector>
    void update(const GridDiscretization& gridDiscretization, const SolutionVector& sol)
    {
        if constexpr (InterpolationPointData::isSolDependent)
        {
            auto newIpDataCache = std::make_shared<InterpolationPointDataCache>();
            newIpDataCache->resize(gridDiscretization.gridView().size(0));
 
            Dumux::parallelFor(gridDiscretization.gridView().size(0), [&, &problem = problem(), newIpDataCache](const std::size_t eIdx)
            {
                const auto element = gridDiscretization.element(eIdx);
                const auto elemDisc = localView(gridDiscretization).bindElement(element);
 
                // get the element solution
                auto elemSol = elementSolution(element, sol, gridDiscretization);
 
                for (const auto& localDof : localDofs(elemDisc))
                    variables_[eIdx][localDof.index()].update(elemSol, problem, elemDisc, ipData(elemDisc, localDof));
 
                newIpDataCache->update(problem, element, elemDisc, variables_[eIdx]);
            });
 
            ipDataCache_ = std::move(newIpDataCache);
        }
        else
        {
            Dumux::parallelFor(gridDiscretization.gridView().size(0), [&, &problem = problem()](const std::size_t eIdx)
            {
                const auto element = gridDiscretization.element(eIdx);
                const auto elemDisc = localView(gridDiscretization).bindElement(element);
 
                // get the element solution
                auto elemSol = elementSolution(element, sol, gridDiscretization);
 
                for (const auto& localDof : localDofs(elemDisc))
                    variables_[eIdx][localDof.index()].update(elemSol, problem, elemDisc, ipData(elemDisc, localDof));
            });
        }
    }
 
    template<class LocalDof>
    const Variables& variables(const LocalDof& localDof) const
    { return variables_[localDof.elementIndex()][localDof.index()]; }
 
    template<class LocalDof>
    Variables& variables(const LocalDof& localDof)
    { return variables_[localDof.elementIndex()][localDof.index()]; }
 
    const Variables& variables(const std::size_t eIdx, const std::size_t localIdx) const
    { return variables_[eIdx][localIdx]; }
 
    Variables& variables(const std::size_t eIdx, const std::size_t localIdx)
    { return variables_[eIdx][localIdx]; }
 
    const InterpolationPointData& elementCache(std::size_t eIdx, std::size_t qpIdx) const
    { return ipDataCache_->elementCache(eIdx, qpIdx); }
 
    InterpolationPointData& elementCache(std::size_t eIdx, std::size_t qpIdx)
    { return ipDataCache_->elementCache(eIdx, qpIdx); }
 
    const InterpolationPointData& boundaryFaceCache(std::size_t eIdx, int bfIdx, std::size_t qpIdx) const
    { return ipDataCache_->boundaryFaceCache(eIdx, bfIdx, qpIdx); }
 
    InterpolationPointData& boundaryFaceCache(std::size_t eIdx, int bfIdx, std::size_t qpIdx)
    { return ipDataCache_->boundaryFaceCache(eIdx, bfIdx, qpIdx); }
 
    const Problem& problem() const
    { return *problemPtr_; }
 
private:
    class InterpolationPointDataCache
    {
    public:
        struct ElementCache
        {
            std::vector<InterpolationPointData> elementCache;
            std::unordered_map<int, std::vector<InterpolationPointData>> boundaryFaceCache;
 
            template<class Problem, class ElementDiscretization, class ElementVariables>
            void update(const Problem& problem,
                        const typename ElementDiscretization::Element& element,
                        const ElementDiscretization& elemDisc,
                        const ElementVariables& elemVars)
            {
                const auto elemQuadRule = Dumux::CVFE::quadratureRule(elemDisc, element);
                elementCache.resize(std::ranges::size(elemQuadRule));
                for (const auto& qpData : elemQuadRule)
                    elementCache[qpData.ipData().qpIndex()].update(problem, element, elemDisc, elemVars, qpData.ipData());
 
                boundaryFaceCache.clear();
                for (const auto& boundaryFace : boundaryFaces(elemDisc))
                {
                    auto& bfCache = boundaryFaceCache[boundaryFace.index()];
                    const auto quadRule = Dumux::CVFE::quadratureRule(elemDisc, boundaryFace);
                    bfCache.resize(std::ranges::size(quadRule));
                    for (const auto& qpData : quadRule)
                        bfCache[qpData.ipData().qpIndex()].update(problem,
                                                                  element,
                                                                  elemDisc,
                                                                  elemVars,
                                                                  qpData.ipData());
                }
            }
        };
 
        InterpolationPointDataCache()
        {}
 
        void resize(const std::size_t numElements)
        { elementCaches_.resize(numElements); }
 
        template<class Problem, class ElementDiscretization, class ElementVariables>
        void update(const Problem& problem,
                    const typename ElementDiscretization::Element& element,
                    const ElementDiscretization& elemDisc,
                    const ElementVariables& elemVars)
        {
            const auto eIdx = elemDisc.gridDiscretization().elementMapper().index(element);
            elementCaches_[eIdx].update(problem, element, elemDisc, elemVars);
        }
 
        // access operator
        const InterpolationPointData& elementCache(std::size_t eIdx, std::size_t qpIdx) const
        { return elementCaches_[eIdx].elementCache[qpIdx]; }
 
        // access operator
        InterpolationPointData& elementCache(std::size_t eIdx, std::size_t qpIdx)
        { return elementCaches_[eIdx].elementCache[qpIdx]; }
 
        // access operator
        const InterpolationPointData& boundaryFaceCache(std::size_t eIdx, int bfIdx, std::size_t qpIdx) const
        { return elementCaches_[eIdx].boundaryFaceCache.at(bfIdx)[qpIdx]; }
 
        // access operator
        InterpolationPointData& boundaryFaceCache(std::size_t eIdx, int bfIdx, std::size_t qpIdx)
        { return elementCaches_[eIdx].boundaryFaceCache[bfIdx][qpIdx]; }
 
        const ElementCache& cache(std::size_t eIdx) const
        { return elementCaches_[eIdx]; }
 
        ElementCache& cache(std::size_t eIdx)
        { return elementCaches_[eIdx]; }
 
        std::vector<ElementCache> elementCaches_; 
    };
 
public:
    const auto& cache(std::size_t eIdx) const
    { return ipDataCache_->cache(eIdx); }
 
    auto& cache(std::size_t eIdx)
    { return ipDataCache_->cache(eIdx); }
 
private:
    const Problem* problemPtr_;
    std::vector<std::vector<Variables>> variables_;
    std::shared_ptr<InterpolationPointDataCache> ipDataCache_;
};
 
// Specialization when the current local variables are not stored
template<class Traits>
class FEGridVariablesCache<Traits, /*cachingEnabled*/false>
{
    using ThisType = FEGridVariablesCache<Traits, false>;
 
public:
    using Problem = typename Traits::Problem;

    using Variables = typename Traits::Variables;

    static constexpr bool cachingEnabled = false;

    using LocalView = typename Traits::template LocalView<ThisType, cachingEnabled>;

    using MutableLocalView = typename LocalView::MutableView;

    using InterpolationPointData = typename Traits::InterpolationPointData;
 
    FEGridVariablesCache(const Problem& problem) : problemPtr_(&problem) {}
 
    template<class GridDiscretization, class SolutionVector>
    void update(const GridDiscretization& gridDiscretization, const SolutionVector& sol) {}
 
    const Problem& problem() const
    { return *problemPtr_;}
 
private:
    const Problem* problemPtr_;
};
 
} // end namespace Dumux::Experimental
 
#endif