localdof.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_CVFE_LOCAL_DOF_HH
#define DUMUX_CVFE_LOCAL_DOF_HH
 
#include <ranges>
 
#include <dumux/common/indextraits.hh>
 
namespace Dumux::CVFE {

template<class LocalIndex, class GridIndex>
class LocalDof
{
public:
    using LocalIndexType = LocalIndex;
    using GridIndexType = GridIndex;
 
    LocalDof(LocalIndex index, GridIndex dofIndex, GridIndex eIdx)
    : index_(index), dofIndex_(dofIndex), eIdx_(eIdx) {}
    LocalIndex index() const { return index_; }
    GridIndex dofIndex() const { return dofIndex_; }
    GridIndex elementIndex() const { return eIdx_; }
private:
    LocalIndex index_;
    GridIndex dofIndex_;
    GridIndex eIdx_;
};
 
 
} // end namespace Dumux::CVFE
 
namespace Dumux {

template<class FVElementGeometry>
inline auto localDofs(const FVElementGeometry& fvGeometry)
{
    using LocalIndexType = typename IndexTraits<typename FVElementGeometry::GridGeometry::GridView>::LocalIndex;
    using GridIndexType = typename IndexTraits<typename FVElementGeometry::GridGeometry::GridView>::GridIndex;
 
    auto scvRange = scvs(fvGeometry);
    return std::ranges::transform_view(
        std::ranges::subrange(scvRange.begin(), scvRange.end()),
        [](const auto& scv) {
            return CVFE::LocalDof<LocalIndexType, GridIndexType>{
                static_cast<LocalIndexType>(scv.localDofIndex()),
                static_cast<GridIndexType>(scv.dofIndex()),
                static_cast<GridIndexType>(scv.elementIndex())
            };
        }
    );
}

template<class FVElementGeometry>
inline auto cvLocalDofs(const FVElementGeometry& fvGeometry)
{
    // As default all dofs are cv dofs
    return localDofs(fvGeometry);
}

template<class FVElementGeometry, class LocalDof>
inline std::ranges::range auto
scvs(const FVElementGeometry& fvGeometry, const LocalDof& localDof)
{
    assert(fvGeometry.numScv() > localDof.index());
    return std::views::single(1) | std::views::transform(
        [&](const auto i) -> const typename FVElementGeometry::SubControlVolume& { return fvGeometry.scv(localDof.index()); }
    );
}
 
} // end namespace Dumux
#endif