elementborderlistfromgrid.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:
/*
  Copyright (C) 2011-2013 by Andreas Lauser

  This file is part of the Open Porous Media project (OPM).

  OPM 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 2 of the License, or
  (at your option) any later version.

  OPM 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 OPM.  If not, see <http://www.gnu.org/licenses/>.
*/
#ifndef EWOMS_ELEMENT_BORDER_LIST_FROM_GRID_HH
#define EWOMS_ELEMENT_BORDER_LIST_FROM_GRID_HH

#include "overlaptypes.hh"
#include "blacklist.hh"

#include <dune/grid/common/datahandleif.hh>
#include <dune/grid/common/gridenums.hh>
#include <dune/istl/bcrsmatrix.hh>
#include <dune/istl/scalarproducts.hh>
#include <dune/istl/operators.hh>
#include <dune/common/version.hh>

#include <algorithm>

namespace Ewoms {
namespace Linear {
template <class GridView, class ElementMapper>
class ElementBorderListFromGrid
{
    typedef BlackList::PeerBlackListedEntry PeerBlackListedEntry;
    typedef BlackList::PeerBlackList PeerBlackList;
    typedef BlackList::PeerBlackLists PeerBlackLists;

    typedef typename GridView::template Codim<0>::Entity Element;

    class BorderListHandle_
        : public Dune::CommDataHandleIF<BorderListHandle_, int>
    {
    public:
        BorderListHandle_(const GridView &gridView,
                          const ElementMapper &map,
                          BlackList &blackList,
                          BorderList &borderList)
            : gridView_(gridView)
            , map_(map)
            , blackList_(blackList)
            , borderList_(borderList)
        {
            auto elemIt = gridView_.template begin<0>();
            const auto& elemEndIt = gridView_.template end<0>();
            for (; elemIt != elemEndIt; ++elemIt) {
                if (elemIt->partitionType() != Dune::InteriorEntity) {
#if DUNE_VERSION_NEWER(DUNE_COMMON, 2, 4)
                    int elemIdx = map_.index(*elemIt);
#else
                    int elemIdx = map_.map(*elemIt);
#endif
                    blackList_.addIndex(elemIdx);
                }
            }
        }

        // data handle methods
        bool contains(int dim, int codim) const
        { return codim == 0; }

        bool fixedsize(int dim, int codim) const
        { return true; }

        template <class EntityType>
        size_t size(const EntityType &e) const
        { return 2; }

        template <class MessageBufferImp, class EntityType>
        void gather(MessageBufferImp &buff, const EntityType &e) const
        {
#if DUNE_VERSION_NEWER(DUNE_COMMON, 2, 4)
            int myIdx = map_.index(e);
#else
            int myIdx = map_.map(e);
#endif
            buff.write(static_cast<int>(gridView_.comm().rank()));
            buff.write(myIdx);
        }

        template <class MessageBufferImp>
        void scatter(MessageBufferImp &buff, const Element &e, size_t n)
        {
            // discard the index if it is not on the process boundary
            bool isInteriorNeighbor = false;
            auto isIt = gridView_.ibegin(e);
            const auto &isEndIt = gridView_.iend(e);
            for (; isIt != isEndIt; ++isIt) {
                if (!isIt->neighbor())
                    continue;
#if DUNE_VERSION_NEWER(DUNE_COMMON, 2, 4)
                else if (isIt->outside().partitionType() == Dune::InteriorEntity) {
#else
                else if (isIt->outside()->partitionType() == Dune::InteriorEntity) {
#endif
                    isInteriorNeighbor = true;
                    break;
                }
            }
            if (!isInteriorNeighbor)
                return;

            BorderIndex bIdx;

#if DUNE_VERSION_NEWER(DUNE_COMMON, 2, 4)
            bIdx.localIdx = map_.index(e);
#else
            bIdx.localIdx = map_.map(e);
#endif
            {
                int tmp;
                buff.read(tmp);
                bIdx.peerRank = tmp;
                peerSet_.insert(tmp);
            }
            {
                int tmp;
                buff.read(tmp);
                bIdx.peerIdx = tmp;
            }
            bIdx.borderDistance = 1;

            borderList_.push_back(bIdx);
        }

        // this template method is needed because the above one only works for codim-0
        // entities (i.e., elements) but the dune grid uses some code which causes the
        // compiler to invoke the scatter method for every codim...
        template <class MessageBufferImp, class EntityType>
        void scatter(MessageBufferImp &buff, const EntityType &e, size_t n)
        { }

        const std::set<ProcessRank>& peerSet() const
        { return peerSet_; }

    private:
        GridView gridView_;
        const ElementMapper &map_;
        std::set<ProcessRank> peerSet_;
        BlackList &blackList_;
        BorderList &borderList_;
    };

    class PeerBlackListHandle_
        : public Dune::CommDataHandleIF<PeerBlackListHandle_, int>
    {
    public:
        PeerBlackListHandle_(const GridView &gridView,
                             const ElementMapper &map,
                             PeerBlackLists& peerBlackLists)
            : gridView_(gridView)
            , map_(map)
            , peerBlackLists_(peerBlackLists)
        {}

        // data handle methods
        bool contains(int dim, int codim) const
        { return codim == 0; }

        bool fixedsize(int dim, int codim) const
        { return true; }

        template <class EntityType>
        size_t size(const EntityType &e) const
        { return 2; }

        template <class MessageBufferImp, class EntityType>
        void gather(MessageBufferImp &buff, const EntityType &e) const
        {
            buff.write(static_cast<int>(gridView_.comm().rank()));
#if DUNE_VERSION_NEWER(DUNE_COMMON, 2, 4)
            buff.write(static_cast<int>(map_.index(e)));
#else
            buff.write(static_cast<int>(map_.map(e)));
#endif
        }

        template <class MessageBufferImp, class EntityType>
        void scatter(MessageBufferImp &buff, const EntityType &e, size_t n)
        {
            int peerRank;
            int peerIdx;
            int localIdx;

            buff.read(peerRank);
            buff.read(peerIdx);
#if DUNE_VERSION_NEWER(DUNE_COMMON, 2, 4)
            localIdx = map_.index(e);
#else
            localIdx = map_.map(e);
#endif

            PeerBlackListedEntry pIdx;
            pIdx.nativeIndexOfPeer = peerIdx;
            pIdx.myOwnNativeIndex = localIdx;

            peerBlackLists_[peerRank].push_back(pIdx);
        }

        const PeerBlackList& peerBlackList(ProcessRank peerRank) const
        { return peerBlackLists_.at(peerRank); }

    private:
        GridView gridView_;
        const ElementMapper &map_;
        PeerBlackLists peerBlackLists_;
    };

public:
    ElementBorderListFromGrid(const GridView &gridView, const ElementMapper &map)
        : gridView_(gridView)
        , map_(map)
    {
        BorderListHandle_ blh(gridView, map, blackList_, borderList_);
        gridView.communicate(blh,
                             Dune::InteriorBorder_All_Interface,
                             Dune::BackwardCommunication);

        PeerBlackListHandle_ pblh(gridView, map, peerBlackLists_);
        gridView.communicate(pblh,
                             Dune::InteriorBorder_All_Interface,
                             Dune::BackwardCommunication);

        auto peerIt = blh.peerSet().begin();
        const auto& peerEndIt = blh.peerSet().end();
        for (; peerIt != peerEndIt; ++peerIt) {
            blackList_.setPeerList(*peerIt, pblh.peerBlackList(*peerIt));
        }
    }

    // Access to the border list.
    const BorderList &borderList() const
    { return borderList_; }

    // Access to the black-list indices.
    const BlackList& blackList() const
    { return blackList_; }

private:
    const GridView gridView_;
    const ElementMapper &map_;

    BorderList borderList_;

    BlackList blackList_;
    PeerBlackLists peerBlackLists_;
};

} // namespace Linear
} // namespace Ewoms

#endif