periodicdofmap.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-FileCopyrightText: Copyright © DuMux Project contributors, see AUTHORS.md in root folder
// SPDX-License-Identifier: GPL-3.0-or-later
//
#ifndef DUMUX_DISCRETIZATION_PERIODIC_DOF_MAP_HH
#define DUMUX_DISCRETIZATION_PERIODIC_DOF_MAP_HH
 
#include <algorithm>
#include <cstddef>
#include <type_traits>
#include <unordered_map>
#include <utility>
#include <vector>
 
namespace Dumux::Detail {

template<class Index>
void addPeriodicallyMappedDof(std::unordered_map<Index, std::vector<Index>>& periodicDofMap,
                              std::type_identity_t<Index> dofIdx, std::type_identity_t<Index> periodicDofIdx)
{
    auto& periodicDofs = periodicDofMap[dofIdx];
    if (std::ranges::find(periodicDofs, periodicDofIdx) == periodicDofs.end())
        periodicDofs.push_back(periodicDofIdx);
}

template<class Index>
void closePeriodicDofMap(std::unordered_map<Index, std::vector<Index>>& periodicDofMap)
{
    std::unordered_map<Index, std::vector<Index>> closedMap;
    for (const auto& entry : periodicDofMap)
    {
        if (closedMap.contains(entry.first))
            continue;
 
        std::vector<Index> group{ entry.first };
        for (std::size_t k = 0; k < group.size(); ++k)
            if (const auto it = periodicDofMap.find(group[k]); it != periodicDofMap.end())
                for (const auto periodicDofIdx : it->second)
                    if (std::ranges::find(group, periodicDofIdx) == group.end())
                        group.push_back(periodicDofIdx);
 
        std::ranges::sort(group);
        for (const auto dofIdx : group)
        {
            auto& periodicDofs = closedMap[dofIdx];
            for (const auto periodicDofIdx : group)
                if (periodicDofIdx != dofIdx)
                    periodicDofs.push_back(periodicDofIdx);
        }
    }
 
    periodicDofMap = std::move(closedMap);
}
 
} // end namespace Dumux::Detail
 
#endif