evalsolution.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:
/*****************************************************************************
 *   See the file COPYING for full copying permissions.                      *
 *                                                                           *
 *   This program is free software: you can redistribute it and/or modify    *
 *   it under the terms of the GNU General Public License as published by    *
 *   the Free Software Foundation, either version 3 of the License, or       *
 *   (at your option) any later version.                                     *
 *                                                                           *
 *   This program is distributed in the hope that it will be useful,         *
 *   but WITHOUT ANY WARRANTY; without even the implied warranty of          *
 *   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the            *
 *   GNU General Public License for more details.                            *
 *                                                                           *
 *   You should have received a copy of the GNU General Public License       *
 *   along with this program.  If not, see <http://www.gnu.org/licenses/>.   *
 *****************************************************************************/
#ifndef DUMUX_DISCRETIZATION_EVAL_SOLUTION_HH
#define DUMUX_DISCRETIZATION_EVAL_SOLUTION_HH
 
#include <iterator>
#include <algorithm>
#include <type_traits>
 
#include <dune/common/hybridutilities.hh>
#include <dune/localfunctions/lagrange/pqkfactory.hh>
 
#include <dumux/common/typetraits/state.hh>
#include <dumux/common/typetraits/isvalid.hh>
#include <dumux/discretization/box/elementsolution.hh>
#include <dumux/discretization/cellcentered/elementsolution.hh>
 
namespace Dumux {
 
// some implementation details
namespace Detail {
 
template<class ElementSolution>
bool allStatesEqual(const ElementSolution& elemSol, std::true_type hasState)
{
    // determine if all states are the same at all vertices
    auto firstState = elemSol[0].state();
    for (std::size_t i = 1; i < elemSol.size(); ++i)
        if (elemSol[i].state() != firstState)
            return false;
 
    return true;
}
 
template<class ElementSolution>
bool allStatesEqual(const ElementSolution& elemSol, std::false_type hasState)
{
    return true;
}
 
template<class Geometry, class ElementSolution>
auto minDistVertexSol(const Geometry& geometry, const typename Geometry::GlobalCoordinate& globalPos,
                      const ElementSolution& elemSol)
{
    // calculate the distances from the evaluation point to the vertices
    std::vector<typename Geometry::ctype> distances(geometry.corners());
    for (int i = 0; i < geometry.corners(); ++i)
        distances[i] = (geometry.corner(i) - globalPos).two_norm2();
 
    // determine the minimum distance and corresponding vertex
    auto minDistanceIt = std::min_element(distances.begin(), distances.end());
    return elemSol[std::distance(distances.begin(), minDistanceIt)];
}
 
} // end namespace Detail
 
template<class Element, class FVElementGeometry, class PrimaryVariables>
PrimaryVariables evalSolution(const Element& element,
                              const typename Element::Geometry& geometry,
                              const typename FVElementGeometry::GridGeometry& gridGeometry,
                              const BoxElementSolution<FVElementGeometry, PrimaryVariables>& elemSol,
                              const typename Element::Geometry::GlobalCoordinate& globalPos,
                              bool ignoreState = false)
{
    // determine if all states are the same at all vertices
    using HasState = decltype(isValid(Detail::hasState())(elemSol[0]));
    bool allStatesEqual = ignoreState || Detail::allStatesEqual(elemSol, HasState{});
 
    if (allStatesEqual)
    {
        using Scalar = typename PrimaryVariables::value_type;
 
        // interpolate the solution
        const auto& localBasis = gridGeometry.feCache().get(geometry.type()).localBasis();
 
        // evaluate the shape functions at the scv center
        const auto localPos = geometry.local(globalPos);
        std::vector< Dune::FieldVector<Scalar, 1> > shapeValues;
        localBasis.evaluateFunction(localPos, shapeValues);
 
        PrimaryVariables result(0.0);
        for (int i = 0; i < geometry.corners(); ++i)
        {
            auto value = elemSol[i];
            value *= shapeValues[i][0];
            result += value;
        }
 
        // set an arbitrary state if the model requires a state (models constexpr if)
        Dune::Hybrid::ifElse(HasState{}, [&](auto&& evalLazy){
            if (!ignoreState)
                evalLazy(result).setState(evalLazy(elemSol)[0].state());
        });
 
        return result;
    }
    else
    {
        static bool warnedAboutUsingMinDist = false;
        if (!warnedAboutUsingMinDist)
        {
            std::cout << "Warning: Using nearest-neighbor interpolation in evalSolution"
            << "\nbecause not all states are equal and ignoreState is false!"
            << std::endl;
            warnedAboutUsingMinDist = true;
        }
 
        return Detail::minDistVertexSol(geometry, globalPos, elemSol);
    }
}
 
template<class Element, class FVElementGeometry, class PrimaryVariables>
PrimaryVariables evalSolution(const Element& element,
                              const typename Element::Geometry& geometry,
                              const BoxElementSolution<FVElementGeometry, PrimaryVariables>& elemSol,
                              const typename Element::Geometry::GlobalCoordinate& globalPos,
                              bool ignoreState = false)
{
    // determine if all states are the same at all vertices
    using HasState = decltype(isValid(Detail::hasState())(elemSol[0]));
    bool allStatesEqual = ignoreState || Detail::allStatesEqual(elemSol, HasState{});
 
    if (allStatesEqual)
    {
        using Scalar = typename PrimaryVariables::value_type;
        using CoordScalar = typename Element::Geometry::GlobalCoordinate::value_type;
        static constexpr int dim = Element::Geometry::mydimension;
 
        using FeCache = Dune::PQkLocalFiniteElementCache<CoordScalar, Scalar, dim, 1>;
        using ShapeValue = typename FeCache::FiniteElementType::Traits::LocalBasisType::Traits::RangeType;
 
        // obtain local finite element basis
        FeCache feCache;
        const auto& localBasis = feCache.get(geometry.type()).localBasis();
 
        // evaluate the shape functions at the scv center
        const auto localPos = geometry.local(globalPos);
        std::vector< ShapeValue > shapeValues;
        localBasis.evaluateFunction(localPos, shapeValues);
 
        PrimaryVariables result(0.0);
        for (int i = 0; i < geometry.corners(); ++i)
        {
            auto value = elemSol[i];
            value *= shapeValues[i][0];
            result += value;
        }
 
        // set an arbitrary state if the model requires a state (models constexpr if)
        Dune::Hybrid::ifElse(HasState{}, [&](auto&& evalLazy){
            if (!ignoreState)
                evalLazy(result).setState(evalLazy(elemSol)[0].state());
        });
 
        return result;
    }
    else
    {
        static bool warnedAboutUsingMinDist = false;
        if (!warnedAboutUsingMinDist)
        {
            std::cout << "Warning: Using nearest-neighbor interpolation in evalSolution"
            << "\nbecause not all states are equal and ignoreState is false!"
            << std::endl;
            warnedAboutUsingMinDist = true;
        }
 
        return Detail::minDistVertexSol(geometry, globalPos, elemSol);
    }
}
 
template<class Element, class FVElementGeometry, class PrimaryVariables>
PrimaryVariables evalSolution(const Element& element,
                              const typename Element::Geometry& geometry,
                              const typename FVElementGeometry::GridGeometry& gridGeometry,
                              const CCElementSolution<FVElementGeometry, PrimaryVariables>& elemSol,
                              const typename Element::Geometry::GlobalCoordinate& globalPos,
                              bool ignoreState = false)
{
    return elemSol[0];
}
 
template< class Element, class FVElementGeometry, class PrimaryVariables>
PrimaryVariables evalSolution(const Element& element,
                              const typename Element::Geometry& geometry,
                              const CCElementSolution<FVElementGeometry, PrimaryVariables>& elemSol,
                              const typename Element::Geometry::GlobalCoordinate& globalPos,
                              bool ignoreState = false)
{
    return elemSol[0];
}
 
} // namespace Dumux
 
#endif