refinementindicator.hh

Go to the documentation of this file.

// -*- tab-width: 8; indent-tabs-mode: nil; c-basic-offset: 4 -*-
// vi: set ts=8 sw=4 et sts=4:
 
// SPDX-FileCopyrightText: Copyright © DUNE-FUFEM Project contributors, see file AUTHORS.md
// SPDX-License-Identifier: LicenseRef-GPL-2.0-only-with-DUNE-exception OR LGPL-3.0-or-later
 
#ifndef DUNE_FUFEM_ESTIMATORS_REFINEMENT_INDICATOR_HH
#define DUNE_FUFEM_ESTIMATORS_REFINEMENT_INDICATOR_HH
 
#include <array>
#include <map>
 
template <class GridType>
class RefinementIndicator
{
 
    typedef double field_type;
    typedef typename GridType::LocalIdSet IdSet;
    typedef typename GridType::template Codim<0>::Entity Element;
    typedef typename IdSet::IdType IdType;
 
    enum {dim = GridType::dimension};
 
public:
 
    RefinementIndicator(const GridType& grid)
        : grid_(&grid)
    {}
 
    void clear() {
        for (size_t i=0; i<indicators_.size(); i++)
            indicators_[i].clear();
    }
 
    void set(const Element& element, unsigned int codim, unsigned int subEntity, const field_type);
 
    field_type value (const Element& element, unsigned int codim, unsigned int subEntity) const;
 
private:
 
    const GridType* grid_;
 
    // One map per codimension
    std::array<std::map<IdType, field_type>, dim+1> indicators_;
 
};
 
 
// Set indicator associated with a subentity of an element
template <class GridType>
void RefinementIndicator<GridType>::
set(const Element& element, unsigned int codim, unsigned int subEntity, const field_type value)
{
    IdType id = grid_->localIdSet().subId(element, subEntity, codim);
 
    indicators_[codim][id] = value;
}
 
 
/* Get indicator associated with a subentity of an element */
template <class GridType>
typename RefinementIndicator<GridType>::field_type RefinementIndicator<GridType>::
value (const Element& element, unsigned int codim, unsigned int subEntity) const
{
    IdType id = grid_->localIdSet().subId(element, subEntity, codim);
    typename std::map<IdType,field_type>::const_iterator indicatorEntry = indicators_[codim].find(id);
    return (indicatorEntry != indicators_[codim].end()) ? indicatorEntry->second : 0;
}
 
template <class Grid>
auto makeRefinementIndicator(const Grid& grid) {
  return RefinementIndicator<Grid>(grid);
}
 
#endif