cvfegridfunction.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_IO_CVFE_GRID_FUNCTION_HH
#define DUMUX_IO_CVFE_GRID_FUNCTION_HH
 
#include <optional>
 
#include <dumux/discretization/elementsolution.hh>
#include <dumux/discretization/evalsolution.hh>
 
namespace Dumux::IO {

template<class GridGeometry, class SolutionVector>
class CVFEGridFunction
{
    using GridView = typename GridGeometry::GridView;
    using Element = typename GridView::template Codim<0>::Entity;
 
public:
    class LocalFunction
    {
    public:
        LocalFunction(const GridGeometry& gg, const SolutionVector& sol)
        : gridGeometry_(&gg), sol_(&sol)
        {}
 
        void bind(const Element& element)
        { element_.emplace(element); }
 
        void unbind()
        { element_.reset(); }

        auto operator()(const typename Element::Geometry::LocalCoordinate& localPos) const
        {
            const auto elemSol = elementSolution(*element_, *sol_, *gridGeometry_);
            return evalSolutionAtLocalPos(
                *element_, element_->geometry(),
                *gridGeometry_, elemSol, localPos
            );
        }
 
    private:
        const GridGeometry* gridGeometry_;
        const SolutionVector* sol_;
        std::optional<Element> element_;
    };
 
    CVFEGridFunction(const GridGeometry& gg, const SolutionVector& sol)
    : gridGeometry_(&gg), sol_(&sol)
    {}

    friend LocalFunction localFunction(const CVFEGridFunction& f)
    { return LocalFunction(*f.gridGeometry_, *f.sol_); }
 
private:
    const GridGeometry* gridGeometry_;
    const SolutionVector* sol_;
};

template<class GridGeometry, class SolutionVector>
auto cvfeGridFunction(const GridGeometry& gg, const SolutionVector& sol)
{ return CVFEGridFunction<GridGeometry, SolutionVector>{gg, sol}; }
 
} // namespace Dumux::IO
 
#endif