codimonegridadapter.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_FACETCOUPLING_CODIM_ONE_GRID_ADAPTER_HH
#define DUMUX_FACETCOUPLING_CODIM_ONE_GRID_ADAPTER_HH
 
#include <cassert>
#include <vector>
#include <memory>
 
#include <dune/grid/common/mcmgmapper.hh>
#include <dune/geometry/referenceelements.hh>
 
#include <dumux/common/indextraits.hh>
 
namespace Dumux {

template<class Embeddings, int bulkGridId = 0, int facetGridId = 1>
class CodimOneGridAdapter
{
    // Extract some types of the facet-conforming grid of codimension one
    using FacetGridView = typename Embeddings::template GridView<facetGridId>;
    using FacetGridVertex = typename FacetGridView::template Codim<FacetGridView::dimension>::Entity;
    using FacetGridElement = typename FacetGridView::template Codim<0>::Entity;
    using FacetGridIndexType = typename IndexTraits<FacetGridView>::GridIndex;
 
    // Extract some types of the bulk grid
    using BulkGridView = typename Embeddings::template GridView<bulkGridId>;
    using BulkMapper = Dune::MultipleCodimMultipleGeomTypeMapper<BulkGridView>;
    using BulkReferenceElements = typename Dune::ReferenceElements<typename BulkGridView::ctype, BulkGridView::dimension>;
    using BulkGridElement = typename BulkGridView::template Codim<0>::Entity;
    using BulkGridIntersection = typename BulkGridView::Intersection;
    using BulkGridVertex = typename BulkGridView::template Codim<BulkGridView::dimension>::Entity;
    using BulkIndexType = typename IndexTraits<BulkGridView>::GridIndex;
 
    // check if provided id combination makes sense
    static_assert( int(FacetGridView::dimension) == int(BulkGridView::dimension) - 1,
                   "Grid dimension mismatch! Please check the provided domain ids!" );
    static_assert( int(FacetGridView::dimensionworld) == int(BulkGridView::dimensionworld),
                   "Grid world dimension mismatch! All grids must have the same world dimension" );
 
public:

    CodimOneGridAdapter(std::shared_ptr<const Embeddings> embeddings)
    : embeddingsPtr_(embeddings)
    , bulkVertexMapper_(embeddings->template gridView<bulkGridId>(), Dune::mcmgVertexLayout())
    {
        // bulk insertion to grid index map
        const auto& bulkGridView = embeddings->template gridView<bulkGridId>();
        bulkInsertionToGridVIdx_.resize(bulkGridView.size(BulkGridView::dimension));
        for (const auto& v : vertices(bulkGridView))
            bulkInsertionToGridVIdx_[embeddings->template insertionIndex<bulkGridId>(v)] = bulkVertexMapper_.index(v);
 
        // Determine map from the hierarchy's vertex idx to bulk insertion idx
        // There is one unique set of vertex indices within the hierarchy.
        // Obtain the hierarchy indices that make up the bulk grid. These
        // are ordered corresponding to their insertion (thus loopIdx = insertionIdx)
        hierarchyToBulkInsertionIdx_.resize(embeddingsPtr_->numVerticesInHierarchy());
        bulkGridHasHierarchyVertex_.resize(embeddingsPtr_->numVerticesInHierarchy(), false);
        const auto& bulkHierarchyIndices = embeddingsPtr_->gridHierarchyIndices(bulkGridId);
        for (std::size_t insIdx = 0; insIdx < bulkHierarchyIndices.size(); ++insIdx)
        {
            hierarchyToBulkInsertionIdx_[ bulkHierarchyIndices[insIdx] ] = insIdx;
            bulkGridHasHierarchyVertex_[ bulkHierarchyIndices[insIdx] ] = true;
        }
 
        // determine which bulk vertices lie on facet elements
        bulkVertexIsOnFacetGrid_.resize(bulkGridView.size(BulkGridView::dimension), false);
        const auto& facetGridView = embeddings->template gridView<facetGridId>();
        for (const auto& v : vertices(facetGridView))
        {
            const auto insIdx = embeddings->template insertionIndex<facetGridId>(v);
            const auto hierarchyInsIdx = embeddings->gridHierarchyIndices(facetGridId)[insIdx];
 
            if (bulkGridHasHierarchyVertex_[hierarchyInsIdx])
                bulkVertexIsOnFacetGrid_[ getBulkGridVertexIndex_(hierarchyInsIdx) ] = true;
        }
 
        // determine the bulk vertex indices that make up facet elements & connectivity
        facetElementCorners_.resize(facetGridView.size(0));
        facetElementsAtBulkVertex_.resize(bulkGridView.size(BulkGridView::dimension));
 
        std::size_t facetElementCounter = 0;
        for (const auto& element : elements(facetGridView))
        {
            if (isEmbedded(element))
            {
                // obtain the bulk vertex indices of the corners of this element
                const auto numCorners = element.subEntities(FacetGridView::dimension);
                std::vector<BulkIndexType> cornerIndices(numCorners);
                for (int i = 0; i < numCorners; ++i)
                    cornerIndices[i] = bulkGridVertexIndex(element.template subEntity<FacetGridView::dimension>(i));
 
                // update connectivity map facetVertex -> facetElements
                for (auto bulkVIdx : cornerIndices)
                    facetElementsAtBulkVertex_[bulkVIdx].push_back(facetElementCounter);
 
                // update facet elements (identified by corners - store them sorted!)
                std::sort(cornerIndices.begin(), cornerIndices.end());
                facetElementCorners_[facetElementCounter] = std::move(cornerIndices);
            }
 
            facetElementCounter++;
        }
    }

    BulkIndexType bulkGridVertexIndex(const FacetGridVertex& v) const
    {
        const auto insIdx = embeddingsPtr_->template insertionIndex<facetGridId>(v);
        const auto hierarchyInsIdx = embeddingsPtr_->gridHierarchyIndices(facetGridId)[insIdx];
        return getBulkGridVertexIndex_(hierarchyInsIdx);
    }

    bool isOnFacetGrid(const BulkGridVertex& v) const
    {
        const auto bulkInsIdx = embeddingsPtr_->template insertionIndex<bulkGridId>(v);
        const auto bulkVIdx = bulkInsertionToGridVIdx_[bulkInsIdx];
        return bulkVertexIsOnFacetGrid_[bulkVIdx];
    }

    bool isOnFacetGrid(const BulkGridElement& element, const BulkGridIntersection& intersection) const
    {
        // Intersection lies on facet grid, if the corners of the intersection make up a facet element
        const auto refElement = BulkReferenceElements::general(element.type());
        const auto numCorners = intersection.geometry().corners();
        const auto facetIdx = intersection.indexInInside();
 
        std::vector<BulkIndexType> cornerIndices(numCorners);
        for (int i = 0; i < numCorners; ++i)
            cornerIndices[i] = bulkVertexMapper_.subIndex( element,
                                                           refElement.subEntity(facetIdx, 1, i, BulkGridView::dimension),
                                                           BulkGridView::dimension );
 
        return composeFacetElement(cornerIndices);
    }

    template<class IndexStorage>
    bool composeFacetElement(const IndexStorage& bulkVertexIndices) const
    {
        // set up a vector containing all element indices the vertices are connected to
        std::vector<std::size_t> facetElemIndices;
        for (auto bulkVIdx : bulkVertexIndices)
            facetElemIndices.insert( facetElemIndices.end(),
                                     facetElementsAtBulkVertex_[bulkVIdx].begin(),
                                     facetElementsAtBulkVertex_[bulkVIdx].end() );
 
        // if no facet elements are connected to the vertices this is not on facet grid
        if (facetElemIndices.size() == 0)
            return false;
 
        // make the container unique
        std::sort(facetElemIndices.begin(), facetElemIndices.end());
        facetElemIndices.erase(std::unique(facetElemIndices.begin(), facetElemIndices.end()), facetElemIndices.end());
 
        // check if given indices make up a facet element
        auto cornerIndexCopy = bulkVertexIndices;
        std::sort(cornerIndexCopy.begin(), cornerIndexCopy.end());
        for (const auto& facetElemIdx : facetElemIndices)
        {
            const auto& facetElemCorners = facetElementCorners_[facetElemIdx];
            if (facetElemCorners.size() != cornerIndexCopy.size())
                continue;
 
            if ( std::equal(cornerIndexCopy.begin(), cornerIndexCopy.end(),
                            facetElemCorners.begin(), facetElemCorners.end()) )
                return true;
        }
 
        // no corresponding facet element found
        return false;
    }

    bool isEmbedded(const FacetGridElement& e) const
    { return numEmbedments(e) > 0; }

    std::size_t numEmbedments(const FacetGridElement& e) const
    { return embeddingsPtr_->template adjoinedEntityIndices<facetGridId>(e).size(); }
 
private:
    BulkIndexType getBulkGridVertexIndex_(BulkIndexType hierarchyInsertionIdx) const
    {
        assert(bulkGridHasHierarchyVertex_[hierarchyInsertionIdx]);
        return bulkInsertionToGridVIdx_[ hierarchyToBulkInsertionIdx_[hierarchyInsertionIdx] ];
    }
 
    // shared pointer to the embedment data
    std::shared_ptr<const Embeddings> embeddingsPtr_;
 
    // vertex mapper of the bulk grid
    BulkMapper bulkVertexMapper_;
 
    // data stored on grid vertices
    std::vector<bool> bulkVertexIsOnFacetGrid_;
    std::vector<BulkIndexType> bulkInsertionToGridVIdx_;
    std::vector<BulkIndexType> hierarchyToBulkInsertionIdx_;
    std::vector<bool> bulkGridHasHierarchyVertex_;
 
    // data stored for elements on the codim one grid
    std::vector< std::vector<BulkIndexType> > facetElementsAtBulkVertex_;
    std::vector< std::vector<BulkIndexType> > facetElementCorners_;
};
 
} // end namespace Dumux
 
#endif