basegridgeometry.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_BASE_GRID_GEOMETRY_HH
#define DUMUX_DISCRETIZATION_BASE_GRID_GEOMETRY_HH
 
#include <type_traits>
 
#include <dune/grid/common/mcmgmapper.hh>
 
#include <dumux/common/entitymap.hh>
#include <dumux/common/indextraits.hh>
#include <dumux/geometry/boundingboxtree.hh>
#include <dumux/geometry/geometricentityset.hh>
 
// make the local view function available whenever we use the grid geometry
#include <dumux/discretization/localview.hh>
 
namespace Dumux {
 
template<class GV, class Traits>
class BaseGridGeometry
{
    using ElementMap = EntityMap<GV, 0>;
    using ElementSet = GridViewGeometricEntitySet<GV, 0, typename Traits::ElementMapper>;
    using BoundingBoxTree = Dumux::BoundingBoxTree<ElementSet>;
 
    static constexpr int dim = GV::dimension;
    static constexpr int dimWorld = GV::dimensionworld;
 
    using GridIndexType = typename IndexTraits<GV>::GridIndex;
    using Element = typename GV::template Codim<0>::Entity;
 
public:
    using Grid = typename GV::Grid;
    using GridView = GV;
    using GlobalCoordinate = typename Element::Geometry::GlobalCoordinate;
    using ElementMapper = typename Traits::ElementMapper;
    using VertexMapper = typename Traits::VertexMapper;
 
    BaseGridGeometry(const GridView& gridView)
    : gridView_(gridView)
    , elementMapper_(makeElementMapper_(gridView))
    , vertexMapper_(makeVertexMapper_(gridView))
    , bBoxMin_(std::numeric_limits<double>::max())
    , bBoxMax_(-std::numeric_limits<double>::max())
    {
        computeGlobalBoundingBox_();
    }
 
    void update()
    {
        vertexMapper_.update();
        elementMapper_.update();
 
        computeGlobalBoundingBox_();
 
        boundingBoxTree_.release();
        elementMap_.reset();
    }
 
    const GridView& gridView() const
    { return gridView_; }
 
    const VertexMapper &vertexMapper() const
    { return vertexMapper_; }
 
    const ElementMapper &elementMapper() const
    { return elementMapper_; }
 
    VertexMapper &vertexMapper()
    { return vertexMapper_; }
 
    ElementMapper &elementMapper()
    { return elementMapper_; }
 
    const BoundingBoxTree& boundingBoxTree() const
    {
        if(!boundingBoxTree_)
        {
            elementMap(); // make sure the element map is built
            boundingBoxTree_ = std::make_unique<BoundingBoxTree>
                                    ( std::make_shared<ElementSet>(gridView_, elementMapper(), elementMap_) );
        }
 
        return *boundingBoxTree_;
    }
 
    const ElementMap& elementMap() const
    {
        if(!elementMap_)
            elementMap_ = std::make_shared<ElementMap>(gridView_.grid(), elementMapper_);
 
        return *elementMap_;
    }
 
    Element element(GridIndexType eIdx) const
    { return elementMap()[eIdx]; }
 
    const GlobalCoordinate &bBoxMin() const
    { return bBoxMin_; }
 
    const GlobalCoordinate &bBoxMax() const
    { return bBoxMax_; }
 
    bool isPeriodic() const
    { return periodic_; }
 
    void setPeriodic(bool value = true)
    { periodic_ = value; }
 
private:
 
    ElementMapper makeElementMapper_(const GridView& gridView) const
    {
        if constexpr (std::is_constructible<ElementMapper, GridView, Dune::MCMGLayout>())
            return ElementMapper(gridView, Dune::mcmgElementLayout());
        else
            return ElementMapper(gridView);
    }
 
    VertexMapper makeVertexMapper_(const GridView& gridView) const
    {
        if constexpr (std::is_constructible<VertexMapper, GridView, Dune::MCMGLayout>())
            return VertexMapper(gridView, Dune::mcmgVertexLayout());
        else
            return VertexMapper(gridView);
    }
 
    void computeGlobalBoundingBox_()
    {
        // calculate the bounding box of the local partition of the grid view
        for (const auto& vertex : vertices(gridView_))
        {
            for (int i=0; i<dimWorld; i++)
            {
                using std::min;
                using std::max;
                bBoxMin_[i] = min(bBoxMin_[i], vertex.geometry().corner(0)[i]);
                bBoxMax_[i] = max(bBoxMax_[i], vertex.geometry().corner(0)[i]);
            }
        }
 
        // communicate to get the bounding box of the whole domain
        if (gridView_.comm().size() > 1)
        {
            for (int i = 0; i < dimWorld; ++i)
            {
                bBoxMin_[i] = gridView_.comm().min(bBoxMin_[i]);
                bBoxMax_[i] = gridView_.comm().max(bBoxMax_[i]);
            }
        }
    }
 
    const GridView gridView_;
 
    ElementMapper elementMapper_;
    VertexMapper vertexMapper_;
 
    mutable std::unique_ptr<BoundingBoxTree> boundingBoxTree_;
 
    mutable std::shared_ptr<ElementMap> elementMap_;
 
    GlobalCoordinate bBoxMin_;
    GlobalCoordinate bBoxMax_;
 
    bool periodic_ = false;
};
 
} // end namespace Dumux
 
#endif