geometricentityset.hh

Go to the documentation of this file.

/*****************************************************************************
 *   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_GRIDVIEW_GEOMETRIC_ENTITY_SET_HH
#define DUMUX_GRIDVIEW_GEOMETRIC_ENTITY_SET_HH
 
#include <memory>
#include <dune/grid/common/mcmgmapper.hh>
#include <dumux/common/entitymap.hh>
 
namespace Dumux {
 
template <class GridView, int codim = 0, class Mapper = Dune::MultipleCodimMultipleGeomTypeMapper<GridView>>
class GridViewGeometricEntitySet
{
    using EntityMap = Dumux::EntityMap<GridView, codim>;
public:
    using Entity = typename GridView::template Codim<codim>::Entity;
 
    GridViewGeometricEntitySet(const GridView& gridView)
    : GridViewGeometricEntitySet(gridView, Mapper(gridView, Dune::mcmgLayout(Dune::Codim<codim>())))
    {}
 
    GridViewGeometricEntitySet(const GridView& gridView, const Mapper& mapper)
    : gridView_(gridView)
    , mapper_(mapper)
    , entityMap_(std::make_shared<EntityMap>(gridView.grid(), mapper_))
    {}
 
    GridViewGeometricEntitySet(const GridView& gridView,
                               const Mapper& mapper,
                               std::shared_ptr<const EntityMap> entityMap)
    : gridView_(gridView)
    , mapper_(mapper)
    , entityMap_(entityMap)
    {}
 
    enum { dimensionworld = GridView::dimensionworld };
 
    using ctype = typename GridView::ctype;
 
    decltype(auto) size() const
    { return gridView_.size(codim); }
 
    decltype(auto) begin() const
    { return entities(gridView_, Dune::Codim<codim>()).begin(); }
 
    decltype(auto) end() const
    { return entities(gridView_, Dune::Codim<codim>()).end(); }
 
    std::size_t index(const Entity& e) const
    { return mapper_.index(e); }
 
    Entity entity(std::size_t index) const
    { return (*entityMap_)[index]; }
 
private:
    GridView gridView_;
    Mapper mapper_;
    std::shared_ptr<const EntityMap> entityMap_;
 
};
 
} // end namespace Dumux
 
#endif