elementdata.hh

Go to the documentation of this file.

// -*- tab-width: 4; indent-tabs-mode: nil; c-basic-offset: 2 -*-
// vi: set et ts=4 sw=2 sts=2:
 
// 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_GRID_ELEMENTDATA_HH
#define DUNE_FUFEM_GRID_ELEMENTDATA_HH
 
 
#include <cstddef>
#include <type_traits>
#include <utility>
#include <vector>
 
#include <dune/common/exceptions.hh>
 
#include <dune/grid/common/gridfactory.hh>
#include <dune/grid/common/mcmgmapper.hh>
#include <dune/grid/io/file/gmshreader.hh>
 
 
namespace Dune::Fufem {
 
 
 
template<class GV, class T=int>
class ElementData
{
public:
 
  using GridView = GV;
  using Grid = typename GridView::Grid;
  using Mapper = Dune::MultipleCodimMultipleGeomTypeMapper<GridView>;
  using Element = typename GridView::template Codim<0>::Entity;
 
  ElementData(const std::vector<T>& coarseData, const GridView& gridView)
    : coarseData_(coarseData)
    , mapper_(gridView, Dune::mcmgElementLayout())
  {
    updateData();
  }
 
  ElementData(const Dune::GmshReader<Grid>& reader, const Dune::GridFactory<Grid>& factory, const GridView& gridView)
    : coarseData_()
    , mapper_(gridView, Dune::mcmgElementLayout())
  {
    if (not reader.hasElementData())
      DUNE_THROW(Dune::InvalidStateException, "The GmshReader passed to ElementData has no stored element data.");
    const auto& rawElementData = reader.elementData();
    const auto& grid = mapper_.gridView().grid();
    auto coarseGridView = grid.levelGridView(0);
    auto coarseMapper = Dune::MultipleCodimMultipleGeomTypeMapper(coarseGridView, Dune::mcmgElementLayout());
    coarseData_.resize(coarseMapper.size());
    for(const auto& coarseElement : elements(coarseGridView))
      coarseData_[coarseMapper.index(coarseElement)] = rawElementData[factory.insertionIndex(coarseElement)];
    updateData();
  }
 
  ElementData(const Dune::GmshReader<Grid>& reader, const Dune::GridFactory<Grid>& factory, const Grid& grid)
  requires (std::is_same_v<GridView, typename Grid::LeafGridView>)
    : ElementData(reader, factory, grid.leafGridView())
  {}
 
  ElementData(std::vector<T>&& coarseData, const GridView& gridView)
    : coarseData_(std::move(coarseData))
    , mapper_(gridView, Dune::mcmgElementLayout())
  {
    updateData();
  }
 
  void update(const GridView& gridView)
  {
    mapper_.update(gridView);
    updateData();
  }
 
  const T& operator()(const Element& element) const
  {
    return data_[mapper_.index(element)];
  }
 
  const T& operator[](std::size_t elementIndex) const
  {
    return data_[elementIndex];
  }
 
  std::size_t size() const
  {
    return data_.size();
  }
 
  const std::vector<T>& data() const
  {
    return data_;
  }
 
  const std::vector<T>& coarseData() const
  {
    return coarseData_;
  }
 
private:
 
  template<class F>
  static void forEachDescendent(const Element& element, const GridView& gridView, F&& f)
  {
    if (gridView.contains(element))
      f(element);
    else
      for(const auto& childElement : descendantElements(element, element.level()+1))
        forEachDescendent(childElement, gridView, f);
  }
 
  void updateData()
  {
    data_.resize(mapper_.size());
    const auto& grid = mapper_.gridView().grid();
    auto coarseGridView = grid.levelGridView(0);
    auto coarseMapper = Dune::MultipleCodimMultipleGeomTypeMapper(coarseGridView, Dune::mcmgElementLayout());
    for(const auto& coarseElement : elements(coarseGridView))
    {
      auto coarseValue = coarseData_[coarseMapper.index(coarseElement)];
      forEachDescendent(coarseElement, mapper_.gridView(), [&](const auto& fineElement){
        data_[mapper_.index(fineElement)] = coarseValue;
      });
    }
  }
 
  std::vector<T> coarseData_;
  std::vector<T> data_;
  Mapper mapper_;
};
 
template<class GridView>
ElementData(const Dune::GmshReader<typename GridView::Grid>&, const Dune::GridFactory<typename GridView::Grid>&, const GridView&)
  -> ElementData<GridView, int>;
 
template<class Grid>
ElementData(const Dune::GmshReader<Grid>&, const Dune::GridFactory<Grid>&, const Grid&)
  -> ElementData<typename Grid::LeafGridView, int>;
 
 
 
} // namespace Dune::Fufem
 
 
#endif // DUNE_FUFEM_GRID_ELEMENTDATA_HH