griddata.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:
//
// SPDX-FileCopyrightInfo: Copyright © DuMux Project contributors, see AUTHORS.md in root folder
// SPDX-License-Identifier: GPL-3.0-or-later
//
#ifndef DUMUX_IO_GRID_DATA_HH
#define DUMUX_IO_GRID_DATA_HH
 
#include <vector>
#include <memory>
#include <type_traits>
 
#include <dune/common/exceptions.hh>
#include <dune/common/parallel/communication.hh>
#include <dune/common/parallel/mpihelper.hh>
#include <dune/grid/common/gridfactory.hh>
#include <dune/grid/io/file/dgfparser/parser.hh>
#include <dune/grid/io/file/dgfparser/gridptr.hh>
#include <dumux/io/vtk/vtkreader.hh>
 
// UGGrid specific includes
#if HAVE_DUNE_UGGRID
#include <dune/grid/uggrid.hh>
#endif
 
#include "gmshgriddatahandle.hh"
#include "vtkgriddatahandle.hh"
 
namespace Dumux {
 
namespace Detail {
 
template<class Grid>
struct isUG : public std::false_type {};
 
#if HAVE_DUNE_UGGRID
template<int dim>
struct isUG<Dune::UGGrid<dim>> : public std::true_type {};
#endif
 
} // end namespace Details
 
template <class Grid>
class GridData
{
    using Intersection = typename Grid::LeafIntersection;
    using Element = typename Grid::template Codim<0>::Entity;
    using Vertex = typename Grid::template Codim<Grid::dimension>::Entity;
    using GmshDataHandle = GmshGridDataHandle<Grid, Dune::GridFactory<Grid>, std::vector<int>>;
    using VtkDataHandle = VtkGridDataHandle<Grid, Dune::GridFactory<Grid>, std::vector<double>>;
 
    enum DataSourceType { dgf, gmsh, vtk };
 
public:
    GridData(std::shared_ptr<Grid> grid, std::shared_ptr<Dune::GridFactory<Grid>> factory,
             std::vector<int>&& elementMarkers, std::vector<int>&& boundaryMarkers, std::vector<int>&& faceMarkers = std::vector<int>{})
    : gridPtr_(grid)
    , gridFactory_(factory)
    , elementMarkers_(elementMarkers)
    , boundaryMarkers_(boundaryMarkers)
    , faceMarkers_(faceMarkers)
    , dataSourceType_(DataSourceType::gmsh)
    {}
 
    GridData(Dune::GridPtr<Grid> grid)
    : dgfGrid_(grid)
    , dataSourceType_(DataSourceType::dgf)
    {}
 
    GridData(std::shared_ptr<Grid> grid, std::shared_ptr<Dune::GridFactory<Grid>> factory,
             VTKReader::Data&& cellData, VTKReader::Data&& pointData)
    : gridPtr_(grid)
    , gridFactory_(factory)
    , cellData_(cellData)
    , pointData_(pointData)
    , dataSourceType_(DataSourceType::vtk)
    {}
 
 
    // \{
 
    const std::vector<double>& parameters(const Vertex& vertex) const
    {
        if (dataSourceType_ == DataSourceType::dgf)
        {
            if (vertex.level() != 0)
                DUNE_THROW(Dune::IOError, "You can only obtain parameters for level 0 vertices!");
 
            return dgfGrid_.parameters(vertex);
        }
        else
            DUNE_THROW(Dune::InvalidStateException, "The parameters method is only available if the grid was constructed with a DGF file.");
    }
 
    const std::vector<double>& parameters(const Element& element) const
    {
        if (dataSourceType_ == DataSourceType::dgf)
        {
            if (element.hasFather())
            {
                auto level0Element = element;
                while(level0Element.hasFather())
                    level0Element = level0Element.father();
 
                return dgfGrid_.parameters(level0Element);
            }
            else
            {
                return dgfGrid_.parameters(element);
            }
        }
        else
            DUNE_THROW(Dune::InvalidStateException, "The parameters method is only available if the grid was constructed with a DGF file.");
    }
 
    template <class GridImp, class IntersectionImp>
    const Dune::DGFBoundaryParameter::type& parameters(const Dune::Intersection<GridImp, IntersectionImp>& intersection) const
    {
        if (dataSourceType_ == DataSourceType::dgf)
            return dgfGrid_.parameters(intersection);
        else
            DUNE_THROW(Dune::InvalidStateException, "The parameters method is only available if the grid was constructed with a DGF file.");
    }
 
    // \}
 
    // \{
 
    int getBoundaryDomainMarker(int boundarySegmentIndex) const
    {
        if (dataSourceType_ != DataSourceType::gmsh)
            DUNE_THROW(Dune::InvalidStateException, "Domain markers are only available for gmsh grids.");
        if (boundarySegmentIndex >= boundaryMarkers_.size())
            DUNE_THROW(Dune::RangeError, "Boundary segment index "<< boundarySegmentIndex << " bigger than number of boundary segments in grid.\n"
                                         "Make sure to call this function only for boundaries that were defined as physical entities in gmsh.");
        return boundaryMarkers_[boundarySegmentIndex];
    }
 
    int getBoundaryDomainMarker(const Intersection& intersection) const
    { return getBoundaryDomainMarker(intersection.boundarySegmentIndex()); }
 
    bool wasInserted(const Intersection& intersection) const
    { return gridFactory_->wasInserted(intersection); }
 
    int getElementDomainMarker(const Element& element) const
    {
        if (dataSourceType_ != DataSourceType::gmsh)
            DUNE_THROW(Dune::InvalidStateException, "Domain markers are only available for gmsh grids.");
 
        // parameters are only given for level 0 elements
        auto level0element = element;
        while (level0element.hasFather())
            level0element = level0element.father();
 
        // in the parallel case the data is load balanced and then accessed with indices of the index set
        // for UGGrid element data is read on all processes since UGGrid can't communicate element data (yet)
        if (gridPtr_->comm().size() > 1 && !Detail::isUG<Grid>::value)
            return elementMarkers_[gridPtr_->levelGridView(0).indexSet().index(level0element)];
        else
            return elementMarkers_[gridFactory_->insertionIndex(level0element)];
    }
 
    template<bool ug = Detail::isUG<Grid>::value, typename std::enable_if_t<!ug, int> = 0>
    GmshDataHandle createGmshDataHandle()
    {
        return GmshDataHandle(*gridPtr_, *gridFactory_, elementMarkers_, boundaryMarkers_, faceMarkers_);
    }
 
    template<bool ug = Detail::isUG<Grid>::value, typename std::enable_if_t<ug, int> = 0>
    GmshDataHandle createGmshDataHandle()
    {
        return GmshDataHandle(*gridPtr_, *gridFactory_, elementMarkers_, boundaryMarkers_);
    }
 
    // \}
 
    // \{
 
    double getParameter(const Element& element, const std::string& fieldName) const
    { return getArrayParameter<double, 1>(element, fieldName)[0]; }
 
    template<class T, std::size_t size>
    std::array<T, size> getArrayParameter(const Element& element, const std::string& fieldName) const
    {
        if (dataSourceType_ != DataSourceType::vtk)
            DUNE_THROW(Dune::InvalidStateException, "This access function is only available for data from VTK files.");
 
        if (cellData_.count(fieldName) == 0)
            DUNE_THROW(Dune::IOError, "No field with the name " << fieldName << " found in cell data.");
 
        // parameters are only given for level 0 elements
        auto level0element = element;
        while (level0element.hasFather())
            level0element = level0element.father();
 
        // different index depending on whether we have communicated user data (parallel setting) or not (sequential setting)
        const auto index = [&]{
            if (gridPtr_->comm().size() > 1)
                return gridPtr_->levelGridView(0).indexSet().index(level0element);
            else
                return gridFactory_->insertionIndex(level0element);
        }();
 
        std::array<T, size> param;
        const auto& data = cellData_.at(fieldName);
 
        if (const std::size_t nc = data.size()/gridPtr_->levelGridView(0).size(0); size != nc)
            DUNE_THROW(Dune::IOError, "Array size " << size << " does not match number of components of field " << nc);
 
        for (std::size_t i = 0; i < size; ++i)
            param[i] = static_cast<T>(data[i + size*index]);
        return param;
    }
 
    double getParameter(const Vertex& vertex, const std::string& fieldName) const
    { return getArrayParameter<double, 1>(vertex, fieldName)[0]; }
 
    template<class T, std::size_t size>
    std::array<T, size> getArrayParameter(const Vertex& vertex, const std::string& fieldName) const
    {
        if (dataSourceType_ != DataSourceType::vtk)
            DUNE_THROW(Dune::InvalidStateException, "This access function is only available for data from VTK files.");
 
        if (vertex.level() != 0)
            DUNE_THROW(Dune::IOError, "You can only obtain parameters for level 0 vertices!");
 
        if (pointData_.count(fieldName) == 0)
            DUNE_THROW(Dune::IOError, "No field with the name " << fieldName << " found in point data");
 
        // different index depending on whether we have communicated user data (parallel setting) or not (sequential setting)
        const auto index = [&]{
            if (gridPtr_->comm().size() > 1)
                return gridPtr_->levelGridView(0).indexSet().index(vertex);
            else
                return gridFactory_->insertionIndex(vertex);
        }();
 
        std::array<T, size> param;
        const auto& data = pointData_.at(fieldName);
 
        if (const std::size_t nc = data.size()/gridPtr_->levelGridView(0).size(Grid::dimension); size != nc)
            DUNE_THROW(Dune::IOError, "Array size " << size << " does not match number of components of field " << nc);
 
        for (std::size_t i = 0; i < size; ++i)
            param[i] = static_cast<T>(data[i + size*index]);
        return param;
    }
 
    VtkDataHandle createVtkDataHandle()
    {
        return VtkDataHandle(*gridPtr_, *gridFactory_, cellData_, pointData_);
    }
 
    // \}
 
private:
    // grid and grid factor for gmsh grid data / vtk grid data
    std::shared_ptr<Grid> gridPtr_;
    std::shared_ptr<Dune::GridFactory<Grid>> gridFactory_;
 
    std::vector<int> elementMarkers_;
    std::vector<int> boundaryMarkers_;
    std::vector<int> faceMarkers_;
 
    VTKReader::Data cellData_, pointData_;
 
    // dgf grid data
    Dune::GridPtr<Grid> dgfGrid_;
 
    // specify which type of data we have
    // TODO unfortunately all grid readers provide different data types, should be streamlined (changes in Dune)
    DataSourceType dataSourceType_;
};
 
} // namespace Dumux
 
#endif