fluxoversurface.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_FLUX_OVER_SURFACE_STAGGERED_HH
#define DUMUX_FLUX_OVER_SURFACE_STAGGERED_HH
 
#include <numeric>
#include <functional>
#include <type_traits>
#include <vector>
 
#include <dune/common/exceptions.hh>
#include <dune/common/fvector.hh>
#include <dune/geometry/type.hh>
#include <dune/geometry/multilineargeometry.hh>
 
#include <dumux/common/parameters.hh>
#include <dumux/geometry/makegeometry.hh>
#include <dumux/geometry/intersectspointgeometry.hh>
#include <dumux/discretization/extrusion.hh>
 
namespace Dumux {
 
template<class GridVariables, class SolutionVector, class ModelTraits, class LocalResidual>
class FluxOverSurface
{
    using Scalar = typename GridVariables::Scalar;
    using GridGeometry = typename GridVariables::GridGeometry;
    using FVElementGeometry = typename GridGeometry::LocalView;
    using SubControlVolumeFace = typename FVElementGeometry::SubControlVolumeFace;
    using Extrusion = Extrusion_t<GridGeometry>;
    using GridView = typename GridGeometry::GridView;
    using VolumeVariables = typename GridVariables::VolumeVariables;
    using Element = typename GridView::template Codim<0>::Entity;
 
    using CellCenterPrimaryVariables = std::decay_t<decltype(std::declval<SolutionVector>()[GridGeometry::cellCenterIdx()][0])>;
 
    enum {
        // Grid and world dimension
        dim = GridView::dimension,
        dimWorld = GridView::dimensionworld
    };
 
    using GlobalPosition = typename Element::Geometry::GlobalCoordinate;
 
    static constexpr auto surfaceDim = dimWorld - 1;
    using SurfaceGeometryType = Dune::MultiLinearGeometry< Scalar, surfaceDim, dimWorld >;
 
    template<int mydim, int coordDim, class Scalar=double>
    class SurfaceData
    {
        using Geo = SurfaceGeometryType;
    public:
 
        SurfaceData() {}
 
        SurfaceData(Geo&& geo)
        {
            values_.resize(1);
            geometries_.push_back(std::move(geo));
        }
 
        SurfaceData(std::vector<Geo>&& geo)
        {
            values_.resize(geo.size());
            geometries_ = std::forward<decltype(geo)>(geo);
        }
 
        void addValue(int surfaceIdx, const CellCenterPrimaryVariables& value)
        {
            values_[surfaceIdx] += value;
        }
 
        void addSubSurface(Geo&& geo)
        {
            values_.emplace_back(0.0);
            geometries_.push_back(std::move(geo));
        }
 
        auto& subSurfaces() const
        {
            return geometries_;
        }
 
        CellCenterPrimaryVariables& value(int surfaceIdx) const
        {
            return values_[surfaceIdx];
        }
 
        auto& values() const
        {
            return values_;
        }
 
        void printSurfaceBoundaries(int surfaceIdx) const
        {
            const auto& geometry = geometries_[surfaceIdx];
            for(int i = 0; i < geometry.corners(); ++i)
                std::cout << "(" << geometry.corner(i) << ")  ";
        }
 
        void resetValues()
        {
            std::fill(values_.begin(), values_.end(), CellCenterPrimaryVariables(0.0));
        }
 
    private:
 
        std::vector<Geo> geometries_;
        std::vector<CellCenterPrimaryVariables> values_;
    };
 
public:
 
    using SurfaceList = std::vector<SurfaceGeometryType>;
 
    template<class Sol>
    FluxOverSurface(const GridVariables& gridVariables,
                    const Sol& sol)
    : gridVariables_(gridVariables),
      sol_(sol)
    {
        static_assert(std::is_same<Sol, SolutionVector>::value, "Make sure that sol has the same type as SolutionVector."
                                                                "Use StaggeredVtkOutputModule<GridVariables, decltype(sol)> when calling the constructor.");
 
        verbose_  = getParamFromGroup<bool>(problem_().paramGroup(), "FluxOverSurface.Verbose", false);
    }
 
    void addSurface(const std::string& name, SurfaceList&& surfaces)
    {
        surfaces_[name] = SurfaceData<surfaceDim,dim>(std::forward<decltype(surfaces)>(surfaces));
    }
 
    void addSurface(const std::string& name, const GlobalPosition& p0, const GlobalPosition& p1)
    {
        surfaces_[name].addSubSurface(makeSurface(std::vector<GlobalPosition>{p0, p1}));
    }
 
    void addSurface(const std::string& name,
                    const GlobalPosition& p0,
                    const GlobalPosition& p1,
                    const GlobalPosition& p2,
                    const GlobalPosition& p3)
    {
        surfaces_[name].addSubSurface(makeSurface(std::vector<GlobalPosition>{p0, p1, p2, p3}));
    }
 
    static SurfaceGeometryType makeSurface(const std::vector<Dune::FieldVector<Scalar, 2>>& corners)
    {
            return SurfaceGeometryType(Dune::GeometryTypes::line, corners);
    }
 
    static SurfaceGeometryType makeSurface(const std::vector<Dune::FieldVector<Scalar, 3>>& corners)
    {
        return makeDuneQuadrilaterial(corners);
    }
 
    void calculateMassOrMoleFluxes()
    {
        auto fluxType = [this](const auto& element,
                               const auto& fvGeometry,
                               const auto& elemVolVars,
                               const auto& elemFaceVars,
                               const auto& scvf,
                               const auto& elemFluxVarsCache)
        {
            LocalResidual localResidual(&problem_());
 
            if (scvf.boundary())
                return localResidual.computeBoundaryFluxForCellCenter(problem_(), element, fvGeometry, scvf, elemVolVars, elemFaceVars, /*elemBcTypes*/{}, elemFluxVarsCache);
            else
                return localResidual.computeFluxForCellCenter(problem_(), element, fvGeometry, elemVolVars, elemFaceVars, scvf, elemFluxVarsCache);
        };
 
        calculateFluxes(fluxType);
    }
 
    void calculateVolumeFluxes()
    {
        auto fluxType = [](const auto& element,
                           const auto& fvGeometry,
                           const auto& elemVolVars,
                           const auto& elemFaceVars,
                           const auto& scvf,
                           const auto& elemFluxVarsCache)
        {
            CellCenterPrimaryVariables result(0.0);
            const auto& insideVolVars = elemVolVars[scvf.insideScvIdx()];
            const auto& outsideVolVars = elemVolVars[scvf.outsideScvIdx()];
            const Scalar extrusionFactor = harmonicMean(insideVolVars.extrusionFactor(), outsideVolVars.extrusionFactor());
            result[0] = elemFaceVars[scvf].velocitySelf() * Extrusion::area(scvf) * extrusionFactor * scvf.directionSign();
            return result;
        };
 
        calculateFluxes(fluxType);
    }
 
    template<class FluxType>
    void calculateFluxes(const FluxType& fluxType)
    {
        // make sure to reset all the values of the surfaces, in case this method has been called already before
        for(auto&& surface : surfaces_)
            surface.second.resetValues();
 
        // make sure not to iterate over the same dofs twice
        std::vector<bool> dofVisited(problem_().gridGeometry().numFaceDofs(), false);
 
        auto fvGeometry = localView(problem_().gridGeometry());
        auto elemVolVars = localView(gridVariables_.curGridVolVars());
        auto elemFaceVars = localView(gridVariables_.curGridFaceVars());
        auto elemFluxVarsCache = localView(gridVariables_.gridFluxVarsCache());
 
        for(auto&& element : elements(problem_().gridGeometry().gridView()))
        {
            fvGeometry.bind(element);
            elemVolVars.bind(element, fvGeometry, sol_);
            elemFaceVars.bind(element, fvGeometry, sol_);
            elemFluxVarsCache.bind(element, fvGeometry, elemVolVars);
 
            for(auto && scvf : scvfs(fvGeometry))
            {
                const auto dofIdx = scvf.dofIndex();
                // do nothing of the dof was already visited
                if(dofVisited[dofIdx])
                    continue;
 
                dofVisited[dofIdx] = true;
 
                // iterate through all surfaces and check if the flux at the given position
                // should be accounted for in the respective surface
                for(auto&& surface : surfaces_)
                {
                    const auto& subSurfaces = surface.second.subSurfaces();
 
                    for(int surfaceIdx = 0; surfaceIdx < subSurfaces.size(); ++surfaceIdx)
                    {
                        if(intersectsPointGeometry(scvf.center(), subSurfaces[surfaceIdx]))
                        {
                            const auto result = fluxType(element, fvGeometry, elemVolVars, elemFaceVars, scvf, elemFluxVarsCache);
 
                            surface.second.addValue(surfaceIdx, result);
 
                            if(verbose_)
                            {
                                std::cout << "Flux at face "  << scvf.center() << " (" << surface.first << "): " << result;
                                std::cout << " (directionIndex: " << scvf.directionIndex() << "; surface boundaries: ";
                                surface.second.printSurfaceBoundaries(surfaceIdx);
                                std::cout << ", surfaceIdx " << surfaceIdx << ")" << std::endl;
                            }
                        }
                    }
                }
            }
        }
    }
 
    auto& values(const std::string& name) const
    {
        return surfaces_.at(name).values();
    }
 
    auto netFlux(const std::string& name) const
    {
        const auto& surfaceResults = values(name);
        return std::accumulate(surfaceResults.begin(), surfaceResults.end(), CellCenterPrimaryVariables(0.0));
    }
 
private:
 
    const auto& problem_() const { return gridVariables_.curGridVolVars().problem(); }
 
    std::map<std::string, SurfaceData<surfaceDim ,dim> > surfaces_;
    const GridVariables& gridVariables_;
    const SolutionVector& sol_;
    bool verbose_;
};
 
} //end namespace
 
#endif