gridmanager_sub.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_MANAGER_SUB_HH
#define DUMUX_IO_GRID_MANAGER_SUB_HH
 
#include <memory>
#include <utility>
#include <algorithm>
#include <stack>
#include <vector>
#include <numeric>
 
#include <dune/common/shared_ptr.hh>
#include <dune/common/concept.hh>
#include <dune/grid/yaspgrid.hh>
 
// SubGrid specific includes
#if HAVE_DUNE_SUBGRID
#include <dune/subgrid/subgrid.hh>
#include <dumux/io/rasterimagereader.hh>
#include <dumux/io/rasterimagewriter.hh>
#endif
 
#include <dumux/io/grid/gridmanager_base.hh>
 
#include <dumux/common/parameters.hh>
#include <dumux/common/boundaryflag.hh>
 
#if HAVE_DUNE_SUBGRID
namespace Dumux {
namespace Concept {
 
template<class Element>
struct ElementSelector
{
    template<class F>
    auto require(F&& f) -> decltype(
        bool(f(std::declval<const Element&>()))
    );
};
} // end namespace Concept
 
template <class HostGrid, class HostGridManager = GridManager<HostGrid>>
class SubGridManagerBase
: public GridManagerBase<Dune::SubGrid<HostGrid::dimension, HostGrid>>
{
    static constexpr int dim = HostGrid::dimension;
    using HostElement = typename HostGrid::template Codim<0>::Entity;
    using GlobalPosition = typename HostElement::Geometry::GlobalCoordinate;
 
public:
    using Grid = Dune::SubGrid<dim, HostGrid>;
 
    template<class ES,
             typename std::enable_if_t<Dune::models<Concept::ElementSelector<HostElement>, ES>(), int> = 0>
    void init(HostGrid& hostGrid,
              const ES& selector,
              const std::string& paramGroup = "")
    {
        this->gridPtr() = std::make_unique<Grid>(hostGrid);
        updateSubGrid_(selector);
        loadBalance();
    }
 
    template<class ES,
             typename std::enable_if_t<Dune::models<Concept::ElementSelector<HostElement>, ES>(), int> = 0>
    void init(const ES& selector,
              const std::string& paramGroup = "")
    {
        initHostGrid_(paramGroup);
        this->gridPtr() = std::make_unique<Grid>(hostGridManager_->grid());
        updateSubGrid_(selector);
        loadBalance();
    }
 
    void loadBalance()
    {
        if (Dune::MPIHelper::getCommunication().size() > 1)
            this->grid().loadBalance();
    }
 
    template<class ES,
             typename std::enable_if_t<Dune::models<Concept::ElementSelector<HostElement>, ES>(), int> = 0>
    void update(const ES& selector)
    {
        updateSubGrid_(selector);
        loadBalance();
    }
 
protected:
 
    template<class ES,
             typename std::enable_if_t<Dune::models<Concept::ElementSelector<HostElement>, ES>(), int> = 0>
    void updateSubGrid_(const ES& selector)
    {
        auto& subgridPtr = this->gridPtr();
 
        // A container to store the host grid elements' ids.
        std::set<typename HostGrid::Traits::GlobalIdSet::IdType> elementsForSubgrid;
        const auto& globalIDset = subgridPtr->getHostGrid().globalIdSet();
 
        // Construct the subgrid.
        subgridPtr->createBegin();
 
        // Loop over all elements of the host grid and use the selector to
        // choose which elements to add to the subgrid.
        auto hostGridView = subgridPtr->getHostGrid().leafGridView();
        for (const auto& e : elements(hostGridView))
            if (selector(e))
                elementsForSubgrid.insert(globalIDset.template id<0>(e));
 
        if (elementsForSubgrid.empty())
            DUNE_THROW(Dune::GridError, "No elements in subgrid");
 
        subgridPtr->insertSetPartial(elementsForSubgrid);
        subgridPtr->createEnd();
    }
 
    void initHostGrid_(const std::string& paramGroup)
    {
        hostGridManager_ = std::make_unique<HostGridManager>();
        hostGridManager_->init(paramGroup);
    }
 
    void initHostGrid_(const GlobalPosition& lowerLeft,
                       const GlobalPosition& upperRight,
                       const std::array<int, dim>& cells,
                       const std::string& paramGroup,
                       const int overlap = 1,
                       const std::bitset<dim> periodic = std::bitset<dim>{})
    {
        hostGridManager_ = std::make_unique<HostGridManager>();
        hostGridManager_->init(lowerLeft, upperRight, cells, paramGroup, overlap, periodic);
    }
 
    HostGrid& hostGrid_()
    {
        return hostGridManager_->grid();
    }
 
    std::unique_ptr<HostGridManager> hostGridManager_;
};
 
template<int dim, class HostGrid>
class GridManager<Dune::SubGrid<dim, HostGrid>>
: public SubGridManagerBase<HostGrid, GridManager<HostGrid>>
{};
 
template<int dim, class Coordinates>
class GridManager<Dune::SubGrid<dim, Dune::YaspGrid<dim, Coordinates>>>
: public SubGridManagerBase<Dune::YaspGrid<dim, Coordinates>, GridManager<Dune::YaspGrid<dim, Coordinates>>>
{
    using ParentType = SubGridManagerBase<Dune::YaspGrid<dim, Coordinates>,
                                          GridManager<Dune::YaspGrid<dim, Coordinates>>>;
public:
    using typename ParentType::Grid;
    using ParentType::init;
 
    void init(const std::string& paramGroup = "")
    {
        const auto overlap = getParamFromGroup<int>(paramGroup, "Grid.Overlap", 1);
        const auto periodic = getParamFromGroup<std::bitset<dim>>(paramGroup, "Grid.Periodic", std::bitset<dim>{});
 
        // check if there is an image file we can construct the element selector from
        if (hasParamInGroup(paramGroup, "Grid.Image"))
        {
            const auto imgFileName = getParamFromGroup<std::string>(paramGroup, "Grid.Image");
            const auto ext = this->getFileExtension(imgFileName);
            if (ext == "pbm")
            {
                if (dim != 2)
                    DUNE_THROW(Dune::GridError, "Portable Bitmap Format only supports dim == 2");
 
                // read image
                const auto img = NetPBMReader::readPBM(imgFileName);
                createGridFromImage_(img, paramGroup, overlap, periodic);
            }
            else
                DUNE_THROW(Dune::IOError, "The SubGridManager doesn't support image files with extension: *." << ext);
 
        }
        else if (hasParamInGroup(paramGroup, "Grid.BinaryMask"))
        {
            const auto maskFileName = getParamFromGroup<std::string>(paramGroup, "Grid.BinaryMask");
            std::ifstream mask(maskFileName, std::ios_base::binary);
            std::vector<char> buffer(std::istreambuf_iterator<char>(mask), std::istreambuf_iterator<char>{});
            const auto cells = getParamFromGroup<std::array<int, dim>>(paramGroup, "Grid.Cells");
            if (const auto c = std::accumulate(cells.begin(), cells.end(), 1, std::multiplies<int>{}); c != buffer.size())
                DUNE_THROW(Dune::IOError, "Grid dimensions doesn't match number of cells specified " << c << ":" << buffer.size());
 
            maybePostProcessBinaryMask_(buffer, cells, paramGroup);
 
            using GlobalPosition = typename ParentType::Grid::template Codim<0>::Geometry::GlobalCoordinate;
            const auto lowerLeft = GlobalPosition(0.0);
            const auto upperRight = [&]{
                if (hasParamInGroup(paramGroup, "Grid.PixelDimensions"))
                {
                    auto upperRight = getParamFromGroup<GlobalPosition>(paramGroup, "Grid.PixelDimensions");
                    for (int i = 0; i < upperRight.size(); ++i)
                        upperRight[i] *= cells[i];
                    upperRight += lowerLeft;
                    return upperRight;
                }
                else
                    return getParamFromGroup<GlobalPosition>(paramGroup, "Grid.UpperRight");
            }();
 
            // construct the host grid
            this->initHostGrid_(lowerLeft, upperRight, cells, paramGroup, overlap, periodic);
 
            // check if the marker is customized, per default
            // we use all cells that are encoded as 0
            const char marker = getParamFromGroup<char>(paramGroup, "Grid.Marker", 0);
            const auto elementSelector = [&](const auto& element)
            {
                auto level0 = element;
                while(level0.hasFather())
                    level0 = level0.father();
                const auto eIdx = this->hostGrid_().levelGridView(0).indexSet().index(level0);
                return buffer[eIdx] != marker;
            };
 
            // create the grid
            this->gridPtr() = std::make_unique<Grid>(this->hostGrid_());
            this->updateSubGrid_(elementSelector);
            this->loadBalance();
        }
        else
        {
            std::cerr << "No element selector provided and Grid.Image or Grid.BinaryMask not found" << std::endl;
            std::cerr << "Constructing sub-grid with all elements of the host grid" << std::endl;
 
            // create host grid
            using GlobalPosition = typename ParentType::Grid::template Codim<0>::Geometry::GlobalCoordinate;
            const auto lowerLeft = getParamFromGroup<GlobalPosition>(paramGroup, "Grid.LowerLeft", GlobalPosition(0.0));
            const auto upperRight = getParamFromGroup<GlobalPosition>(paramGroup, "Grid.UpperRight");
            const auto cells = getParamFromGroup<std::array<int, dim>>(paramGroup, "Grid.Cells");
            this->initHostGrid_(lowerLeft, upperRight, cells, paramGroup, overlap, periodic);
            const auto elementSelector = [](const auto& element){ return true; };
            // create the grid
            this->gridPtr() = std::make_unique<Grid>(this->hostGrid_());
            this->updateSubGrid_(elementSelector);
            this->loadBalance();
        }
    }
 
private:
    template<class Img>
    void createGridFromImage_(const Img& img, const std::string& paramGroup, const int overlap, const std::bitset<dim> periodic)
    {
        using GlobalPosition = typename ParentType::Grid::template Codim<0>::Geometry::GlobalCoordinate;
        const bool repeated = hasParamInGroup(paramGroup,"Grid.Repeat");
 
        // get the number of cells
        const std::array<int, dim> imageDimensions{static_cast<int>(img.header().nCols), static_cast<int>(img.header().nRows)};
        std::array<int, dim> cells{imageDimensions[0], imageDimensions[1]};
 
        std::array<int, dim> repeatsDefault; repeatsDefault.fill(1);
        const auto numRepeats = getParamFromGroup<std::array<int, dim>>(paramGroup, "Grid.Repeat", repeatsDefault);
        for (int i = 0; i < dim; i++)
            cells[i] = cells[i] * numRepeats[i];
 
        // get the corner coordinates
        const auto [lowerLeft, upperRight] = [&]()
        {
            const auto lowerLeft = getParamFromGroup<GlobalPosition>(paramGroup, "Grid.LowerLeft", GlobalPosition(0.0));
            if (hasParamInGroup(paramGroup, "Grid.PixelDimensions"))
            {
                auto upperRight = getParamFromGroup<GlobalPosition>(paramGroup, "Grid.PixelDimensions");
                for (int i = 0; i < upperRight.size(); ++i)
                    upperRight[i] *= cells[i];
                upperRight += lowerLeft;
                return std::make_pair(lowerLeft, upperRight);
            }
            else
                return std::make_pair(lowerLeft, getParamFromGroup<GlobalPosition>(paramGroup, "Grid.UpperRight"));
        }();
 
        // construct the host grid
        this->initHostGrid_(lowerLeft, upperRight, cells, paramGroup, overlap, periodic);
 
        // check if the marker is customized, per default
        // we mark all cells that are encoded as 0
        const bool marked = getParamFromGroup<bool>(paramGroup, "Grid.Marker", false);
 
        // Create the element selector for a single image
        const auto elementSelector = [&](const auto& element)
        {
            auto eIdx = this->hostGrid_().leafGridView().indexSet().index(element);
 
            // if the hostgrid was refined, get the index of the original, un-refined
            // host grid element which fits with the image's indices
            if (element.hasFather())
            {
                auto level0Element = element.father();
                while (level0Element.hasFather())
                    level0Element = level0Element.father();
 
                assert(level0Element.level() == 0);
                eIdx = this->hostGrid_().levelGridView(0).indexSet().index(level0Element);
            }
            return img[eIdx] == marked;
        };
 
        // Create the element selector for a repeated image
        const auto repeatedElementSelector = [&](const auto& element)
        {
            auto eIdx = this->hostGrid_().leafGridView().indexSet().index(element);
 
            // if the hostgrid was refined, get the index of the original, un-refined
            // host grid element which fits with the image's indices
            if (element.hasFather())
            {
                auto level0Element = element.father();
                while (level0Element.hasFather())
                    level0Element = level0Element.father();
 
                assert(level0Element.level() == 0);
                eIdx = this->hostGrid_().levelGridView(0).indexSet().index(level0Element);
            }
 
            // figure out the size of the repeat, and the size of the target repeated grid
            const int numCols = imageDimensions[0];
            const int numRows = imageDimensions[1];
 
            // map the eIdx to the original img index
            const int repeatUnitIndex = eIdx % (numCols * numRepeats[0] * numRows);
            const int imgI = repeatUnitIndex % numCols;
            const int imgJ = repeatUnitIndex / (numCols * numRepeats[0]);
            return img[ (imgJ * numCols + imgI) ] == marked;
        };
 
        // create the grid
        if (repeated)
        {
            this->gridPtr() = std::make_unique<Grid>(this->hostGrid_());
            this->updateSubGrid_(repeatedElementSelector);
        }
        else
        {
            this->gridPtr() = std::make_unique<Grid>(this->hostGrid_());
            this->updateSubGrid_(elementSelector);
        }
 
        this->loadBalance();
    }
 
private:
    void maybePostProcessBinaryMask_(std::vector<char>& mask, const std::array<int, dim>& cells, const std::string& paramGroup) const
    {
        // implements pruning for directional connectivity scenario (e.g. flow from left to right)
        // all cells that don't connect the boundaries in X  (or Y or Z) direction  are removed
        if (hasParamInGroup(paramGroup, "Grid.KeepOnlyConnected"))
        {
            std::vector<int> marker(mask.size(), 0);
            const std::string direction = getParamFromGroup<std::string>(paramGroup, "Grid.KeepOnlyConnected");
            if constexpr (dim == 3)
            {
                if (direction == "Y")
                {
                    std::cout << "Keeping only cells connected to both boundaries in y-direction" << std::endl;
                    flood_(mask, marker, cells, 0, 0, 0, cells[0], 1, cells[2], 1);
                    flood_(mask, marker, cells, 0, cells[1]-1, 0, cells[0], cells[1], cells[2], 2);
                }
                else if (direction == "Z")
                {
                    std::cout << "Keeping only cells connected to both boundaries in z-direction" << std::endl;
                    flood_(mask, marker, cells, 0, 0, 0, cells[0], cells[1], 1, 1);
                    flood_(mask, marker, cells, 0, 0, cells[2]-1, cells[0], cells[1], cells[2], 2);
                }
                else
                {
                    std::cout << "Keeping only cells connected to both boundaries in x-direction" << std::endl;
                    flood_(mask, marker, cells, 0, 0, 0, 1, cells[1], cells[2], 1);
                    flood_(mask, marker, cells, cells[0]-1, 0, 0, cells[0], cells[1], cells[2], 2);
                }
 
                for (unsigned int i = 0; i < cells[0]; ++i)
                    for (unsigned int j = 0; j < cells[1]; ++j)
                        for (unsigned int k = 0; k < cells[2]; ++k)
                            if (const auto ijk = getIJK_(i, j, k, cells); marker[ijk] < 2)
                                mask[ijk] = false;
            }
 
            else if constexpr (dim == 2)
            {
                if (direction == "Y")
                {
                    std::cout << "Keeping only cells connected to both boundaries in y-direction" << std::endl;
                    flood_(mask, marker, cells, 0, 0, cells[0], 1, 1);
                    flood_(mask, marker, cells, 0, cells[1]-1, cells[0], cells[1], 2);
                }
                else
                {
                    std::cout << "Keeping only cells connected to both boundaries in x-direction" << std::endl;
                    flood_(mask, marker, cells, 0, 0, 1, cells[1], 1);
                    flood_(mask, marker, cells, cells[0]-1, 0, cells[0], cells[1], 2);
                }
 
                for (unsigned int i = 0; i < cells[0]; ++i)
                    for (unsigned int j = 0; j < cells[1]; ++j)
                        if (const auto ij = getIJ_(i, j, cells); marker[ij] < 2)
                            mask[ij] = false;
            }
            else
                DUNE_THROW(Dune::NotImplemented, "KeepOnlyConnected only implemented for 2D and 3D");
        }
    }
 
    void flood_(std::vector<char>& mask, std::vector<int>& markers, const std::array<int, 3>& cells,
                unsigned int iMin, unsigned int jMin, unsigned int kMin,
                unsigned int iMax, unsigned int jMax, unsigned int kMax,
                int marker) const
    {
        // flood-fill with marker starting from all seed cells in the range
        for (unsigned int i = iMin; i < iMax; ++i)
            for (unsigned int j = jMin; j < jMax; ++j)
                for (unsigned int k = kMin; k < kMax; ++k)
                    fill_(mask, markers, cells, i, j, k, marker);
    }
 
    void flood_(std::vector<char>& mask, std::vector<int>& markers, const std::array<int, 2>& cells,
                unsigned int iMin, unsigned int jMin,
                unsigned int iMax, unsigned int jMax,
                int marker) const
    {
        // flood-fill with marker starting from all seed cells in the range
        for (unsigned int i = iMin; i < iMax; ++i)
            for (unsigned int j = jMin; j < jMax; ++j)
                fill_(mask, markers, cells, i, j, marker);
    }
 
    void fill_(std::vector<char>& mask, std::vector<int>& markers, const std::array<int, 3>& cells,
               unsigned int i, unsigned int j, unsigned int k, int marker) const
    {
        std::stack<std::tuple<unsigned int, unsigned int, unsigned int>> st;
        st.push(std::make_tuple(i, j, k));
        while(!st.empty())
        {
            std::tie(i, j, k) = st.top();
            st.pop();
 
            if (i >= cells[0] || j >= cells[1] || k >= cells[2])
                continue;
 
            const auto ijk = getIJK_(i, j, k, cells);
            if (!mask[ijk] || markers[ijk] >= marker)
               continue;
 
            markers[ijk] += 1;
 
            // we rely on -1 wrapping over for unsigned int = 0
            st.push(std::make_tuple(i+1, j, k));
            st.push(std::make_tuple(i-1, j, k));
            st.push(std::make_tuple(i, j+1, k));
            st.push(std::make_tuple(i, j-1, k));
            st.push(std::make_tuple(i, j, k+1));
            st.push(std::make_tuple(i, j, k-1));
        }
    }
 
    void fill_(std::vector<char>& mask, std::vector<int>& markers, const std::array<int, 2>& cells,
               unsigned int i, unsigned int j, int marker) const
    {
        std::stack<std::tuple<unsigned int, unsigned int>> st;
        st.push(std::make_tuple(i, j));
        while(!st.empty())
        {
            std::tie(i, j) = st.top();
            st.pop();
 
            if (i >= cells[0] || j >= cells[1])
                continue;
 
            const auto ij = getIJ_(i, j, cells);
            if (!mask[ij] || markers[ij] >= marker)
               continue;
 
            markers[ij] += 1;
 
            // we rely on -1 wrapping over for unsigned int = 0
            st.push(std::make_tuple(i+1, j));
            st.push(std::make_tuple(i-1, j));
            st.push(std::make_tuple(i, j+1));
            st.push(std::make_tuple(i, j-1));
        }
    }
 
    int getIJK_(int i, int j, int k, const std::array<int, 3>& cells) const
    { return i + j*cells[0] + k*cells[0]*cells[1]; }
 
    int getIJ_(int i, int j, const std::array<int, 2>& cells) const
    { return i + j*cells[0]; }
};
 
template<int dim, class HostGrid>
class BoundaryFlag<Dune::SubGrid<dim, HostGrid>>
{
public:
    BoundaryFlag() : flag_(-1) {}
 
    template<class Intersection>
    BoundaryFlag(const Intersection& i) : flag_(-1) {}
 
    using value_type = int;
 
    value_type get() const
    { DUNE_THROW(Dune::NotImplemented, "Sub-grid doesn't implement boundary segment indices!"); }
 
private:
    int flag_;
};
 
} // end namespace Dumux
#endif // HAVE_DUNE_SUBGRID
#endif