vertexborderlistfromgrid.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_VERTEX_BORDER_LIST_FROM_GRID_HH
#define EWOMS_VERTEX_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 VertexMapper>
class VertexBorderListFromGrid
    : public Dune::CommDataHandleIF<VertexBorderListFromGrid<GridView, VertexMapper>,
                                    int>
{
    static const int dimWorld = GridView::dimensionworld;

public:
    VertexBorderListFromGrid(const GridView &gridView, const VertexMapper &map)
        : gridView_(gridView), map_(map)
    {
        gridView.communicate(*this,
                             Dune::InteriorBorder_InteriorBorder_Interface,
                             Dune::ForwardCommunication);

        auto vIt = gridView.template begin<dimWorld>();
        const auto& vEndIt = gridView.template end<dimWorld >();
        for (; vIt != vEndIt; ++vIt) {
            if (vIt->partitionType() != Dune::InteriorEntity
                && vIt->partitionType() != Dune::BorderEntity)
            {
#if DUNE_VERSION_NEWER(DUNE_COMMON, 2, 4)
                int vIdx = map_.index(*vIt);
#else
                int vIdx = map_.map(*vIt);
#endif
                blackList_.addIndex(vIdx);
            }
        }
    }

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

    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)
    {
        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;
        }
        {
            int tmp;
            buff.read(tmp);
            bIdx.peerIdx = tmp;
        }
        bIdx.borderDistance = 0;

        borderList_.push_back(bIdx);
    }

    // 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 VertexMapper &map_;
    BorderList borderList_;
    BlackList blackList_;
};

} // namespace Linear
} // namespace Ewoms

#endif